libgnomevfs2-mythtv/modules/mythtv-method.c
author renatofilho
Fri Dec 01 14:52:05 2006 +0000 (2006-12-01)
branchtrunk
changeset 164 81e9da9ad7db
parent 160 106adb6883dc
child 169 318e94209a11
permissions -rwxr-xr-x
[svn r165] code cleanup
melunko@38
     1
/*
melunko@38
     2
 * @author Hallyson Melo <hallyson.melo@indt.org.br>
melunko@38
     3
 *
melunko@38
     4
 * This program is free software; you can redistribute it and/or modify
melunko@38
     5
 * it under the terms of the GNU Lesser General Public License as published by
melunko@38
     6
 * the Free Software Foundation; either version 2 of the License, or
melunko@38
     7
 * (at your option) any later version.
melunko@38
     8
 *
melunko@38
     9
 * This program is distributed in the hope that it will be useful,
melunko@38
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
melunko@38
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
melunko@38
    12
 * GNU General Public License for more details.
melunko@38
    13
 *
melunko@38
    14
 * You should have received a copy of the GNU Lesser General Public License
melunko@38
    15
 * along with this program; if not, write to the Free Software
melunko@38
    16
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
melunko@38
    17
 */
melunko@38
    18
melunko@38
    19
#ifdef HAVE_CONFIG_H
melunko@38
    20
#include <config.h>
melunko@38
    21
#endif
melunko@38
    22
melunko@38
    23
#include <string.h>
melunko@38
    24
#include <glib.h>
melunko@38
    25
#include <math.h>
melunko@38
    26
melunko@38
    27
#include <libgnomevfs/gnome-vfs-module.h>
melunko@38
    28
#include <libgnomevfs/gnome-vfs-utils.h>
melunko@38
    29
melunko@48
    30
#include "gmyth_file_transfer.h"
melunko@160
    31
#include "gmyth_backendinfo.h"
melunko@38
    32
melunko@38
    33
#define GST_MYTHTV_ID_NUM               1
melunko@38
    34
#define MYTHTV_VERSION_DEFAULT          30
melunko@38
    35
melunko@111
    36
#define MYTHTV_BUFFER_SIZE		1024*64
melunko@111
    37
melunko@38
    38
static GnomeVFSResult do_read (GnomeVFSMethod *method,
melunko@38
    39
                               GnomeVFSMethodHandle *method_handle,
melunko@38
    40
                               gpointer buffer,
melunko@38
    41
                               GnomeVFSFileSize num_bytes,
melunko@38
    42
                               GnomeVFSFileSize *bytes_read,
melunko@38
    43
                               GnomeVFSContext *context);
melunko@38
    44
melunko@38
    45
typedef struct {
melunko@48
    46
    GMythFileTransfer *file_transfer;
melunko@38
    47
    
melunko@38
    48
    gint mythtv_version;
melunko@38
    49
    guint64 content_size;
melunko@38
    50
    guint64 bytes_read;
melunko@111
    51
rosfran@116
    52
    GByteArray *buffer;
melunko@111
    53
    gsize buffer_remain;
melunko@38
    54
} MythtvHandle;
melunko@38
    55
melunko@38
    56
melunko@38
    57
melunko@38
    58
static GnomeVFSResult
melunko@38
    59
do_open (GnomeVFSMethod *method,
melunko@38
    60
         GnomeVFSMethodHandle **method_handle,
melunko@38
    61
         GnomeVFSURI *uri,
melunko@38
    62
         GnomeVFSOpenMode mode,
melunko@38
    63
         GnomeVFSContext *context)
melunko@38
    64
{
melunko@111
    65
    MythtvHandle *myth_handle;
melunko@160
    66
    GMythBackendInfo *backend_info;
melunko@38
    67
    gboolean ret;
melunko@38
    68
melunko@38
    69
    _GNOME_VFS_METHOD_PARAM_CHECK (method_handle != NULL);
melunko@38
    70
    _GNOME_VFS_METHOD_PARAM_CHECK (uri != NULL);
melunko@38
    71
melunko@111
    72
    myth_handle = g_new0 (MythtvHandle, 1);
melunko@111
    73
melunko@38
    74
    if (mode & GNOME_VFS_OPEN_WRITE) {
melunko@38
    75
        return GNOME_VFS_ERROR_NOT_PERMITTED;
melunko@38
    76
    }
melunko@38
    77
renatofilho@149
    78
    if (gnome_vfs_uri_get_host_name (uri) == NULL) {
renatofilho@149
    79
    	return GNOME_VFS_ERROR_INVALID_HOST_NAME;
melunko@38
    80
    }
melunko@38
    81
melunko@38
    82
    /* Initialize mythtv handler*/
melunko@38
    83
    myth_handle->file_transfer = NULL;
melunko@38
    84
    myth_handle->mythtv_version = MYTHTV_VERSION_DEFAULT;
melunko@38
    85
    myth_handle->bytes_read = 0;
melunko@38
    86
    myth_handle->content_size = -1;
melunko@38
    87
melunko@160
    88
    /* Creates and fills out the backend info structure */
melunko@160
    89
    backend_info = gmyth_backend_info_new_full (
renatofilho@164
    90
        gnome_vfs_uri_get_host_name (uri),
renatofilho@164
    91
	    NULL, NULL, NULL,
renatofilho@164
    92
	    gnome_vfs_uri_get_host_port (uri));
melunko@160
    93
melunko@38
    94
    /* Connect to the backend */
melunko@160
    95
    myth_handle->file_transfer = gmyth_file_transfer_new (backend_info);
melunko@38
    96
melunko@111
    97
    g_return_val_if_fail (myth_handle->file_transfer != NULL, GNOME_VFS_ERROR_NOT_OPEN);
melunko@38
    98
melunko@38
    99
    /* sets the Playback monitor connection */
melunko@160
   100
    ret = gmyth_file_transfer_open (myth_handle->file_transfer, gnome_vfs_uri_get_path (uri));
melunko@38
   101
    if (ret == FALSE) {
renatofilho@164
   102
        g_warning ("Mythtv FileTransfer open error\n");
melunko@111
   103
        return GNOME_VFS_ERROR_NOT_OPEN;
melunko@38
   104
    }
melunko@38
   105
melunko@38
   106
    // TODO: Verify if file exists in the backend
melunko@38
   107
melunko@38
   108
    myth_handle->content_size = myth_handle->file_transfer->filesize;
melunko@111
   109
    
rosfran@116
   110
    myth_handle->buffer = g_byte_array_sized_new (MYTHTV_BUFFER_SIZE);
melunko@111
   111
    myth_handle->buffer_remain = 0;
melunko@38
   112
melunko@38
   113
    *method_handle = (GnomeVFSMethodHandle *) myth_handle;
melunko@38
   114
melunko@38
   115
    return GNOME_VFS_OK;
melunko@38
   116
}
melunko@38
   117
melunko@38
   118
static GnomeVFSResult
melunko@38
   119
do_read (GnomeVFSMethod *method,
melunko@38
   120
         GnomeVFSMethodHandle *method_handle,
melunko@38
   121
         gpointer buffer,
melunko@38
   122
         GnomeVFSFileSize num_bytes,
melunko@38
   123
         GnomeVFSFileSize *bytes_read,
melunko@38
   124
         GnomeVFSContext *context)
melunko@38
   125
{
melunko@38
   126
    MythtvHandle *myth_handle = (MythtvHandle *) method_handle;
melunko@38
   127
    GnomeVFSFileSize bytes_to_read;
melunko@38
   128
melunko@38
   129
    *bytes_read = 0;
melunko@38
   130
melunko@38
   131
    if (myth_handle->bytes_read >= myth_handle->content_size)
melunko@38
   132
        return GNOME_VFS_ERROR_EOF;
melunko@38
   133
melunko@38
   134
    // fixme: change this to min math function
melunko@38
   135
    if (num_bytes > myth_handle->content_size - myth_handle->bytes_read)
melunko@38
   136
	    bytes_to_read = myth_handle->content_size - myth_handle->bytes_read;
melunko@38
   137
    else
melunko@38
   138
	    bytes_to_read = num_bytes;
melunko@38
   139
melunko@38
   140
    /* Loop sending the Myth File Transfer request:
melunko@38
   141
    * Retry whilst authentication fails and we supply it. */
rosfran@116
   142
  
renatofilho@149
   143
    if ( bytes_to_read > myth_handle->buffer_remain ) {
renatofilho@149
   144
        GByteArray *tmp_buffer = g_byte_array_new();
melunko@38
   145
renatofilho@149
   146
        gint len = gmyth_file_transfer_read (myth_handle->file_transfer,
renatofilho@149
   147
              tmp_buffer, MYTHTV_BUFFER_SIZE - myth_handle->buffer_remain, TRUE);
melunko@38
   148
renatofilho@149
   149
		if (len < 0) {
renatofilho@149
   150
            g_byte_array_free (tmp_buffer, TRUE);
renatofilho@164
   151
            g_warning ("Fail to read bytes");
rosfran@116
   152
		    return GNOME_VFS_ERROR_IO;
renatofilho@149
   153
        }
melunko@111
   154
renatofilho@149
   155
        myth_handle->buffer = g_byte_array_append (myth_handle->buffer,
renatofilho@149
   156
    		tmp_buffer->data, len);
rosfran@116
   157
rosfran@116
   158
		myth_handle->buffer_remain += len;
rosfran@116
   159
rosfran@116
   160
    	tmp_buffer = NULL;    	
rosfran@116
   161
    }
melunko@38
   162
    
melunko@111
   163
    bytes_to_read = (bytes_to_read > myth_handle->buffer_remain) ? myth_handle->buffer_remain : bytes_to_read;
renatofilho@164
   164
    /* gets the first buffer_size bytes from the byte array buffer variable */ 
rosfran@116
   165
renatofilho@149
   166
    g_memmove (buffer, myth_handle->buffer->data, bytes_to_read);
rosfran@116
   167
melunko@111
   168
    myth_handle->bytes_read += bytes_to_read;
rosfran@116
   169
    myth_handle->buffer_remain -= bytes_to_read;    
rosfran@116
   170
rosfran@116
   171
  	/* flushs the newly buffer got from byte array */
renatofilho@149
   172
  	myth_handle->buffer = g_byte_array_remove_range (myth_handle->buffer, 0, bytes_to_read);
melunko@111
   173
    *bytes_read = bytes_to_read;
melunko@38
   174
  
melunko@38
   175
    return GNOME_VFS_OK;
melunko@38
   176
}
melunko@38
   177
melunko@38
   178
static GnomeVFSResult
melunko@38
   179
do_close (GnomeVFSMethod *method,
melunko@38
   180
          GnomeVFSMethodHandle *method_handle,
melunko@38
   181
          GnomeVFSContext *context)
melunko@38
   182
{
melunko@111
   183
melunko@38
   184
    MythtvHandle *myth_handle = (MythtvHandle *) method_handle;
melunko@38
   185
melunko@111
   186
    if (myth_handle->file_transfer) {
renatofilho@164
   187
	    gmyth_file_transfer_close (myth_handle->file_transfer);
melunko@38
   188
        g_object_unref (myth_handle->file_transfer);
renatofilho@164
   189
    	myth_handle->file_transfer = NULL;
melunko@111
   190
    }
rosfran@127
   191
    
rosfran@127
   192
    if (myth_handle->buffer) {
renatofilho@164
   193
		g_byte_array_free (myth_handle->buffer, TRUE);
renatofilho@164
   194
		myth_handle->buffer = NULL;
rosfran@127
   195
    }
melunko@38
   196
melunko@38
   197
    g_free (myth_handle);
melunko@38
   198
melunko@38
   199
    return GNOME_VFS_OK;
melunko@38
   200
}
melunko@38
   201
melunko@38
   202
static GnomeVFSResult
melunko@38
   203
do_get_file_info (GnomeVFSMethod *method,
melunko@38
   204
                  GnomeVFSURI *uri,
melunko@38
   205
                  GnomeVFSFileInfo *file_info,
melunko@38
   206
                  GnomeVFSFileInfoOptions options,
melunko@38
   207
                  GnomeVFSContext *context)
melunko@38
   208
{
melunko@111
   209
    file_info->name = g_strdup ("fixme.txt");
melunko@38
   210
    file_info->valid_fields = file_info->valid_fields
melunko@38
   211
        | GNOME_VFS_FILE_INFO_FIELDS_TYPE
melunko@38
   212
        | GNOME_VFS_FILE_INFO_FIELDS_MIME_TYPE
melunko@38
   213
        | GNOME_VFS_FILE_INFO_FIELDS_PERMISSIONS;
melunko@38
   214
    file_info->type = GNOME_VFS_FILE_TYPE_REGULAR;
melunko@111
   215
    // fixme: get from file extension?
melunko@111
   216
    file_info->mime_type = g_strdup ("video/x-nuv");
melunko@38
   217
    file_info->permissions =
melunko@38
   218
        GNOME_VFS_PERM_USER_READ |
melunko@38
   219
        GNOME_VFS_PERM_OTHER_READ |
melunko@38
   220
        GNOME_VFS_PERM_GROUP_READ;
melunko@38
   221
melunko@38
   222
    return GNOME_VFS_OK;
melunko@38
   223
}
melunko@38
   224
melunko@38
   225
static gboolean
melunko@38
   226
do_is_local (GnomeVFSMethod *method,
melunko@38
   227
             const GnomeVFSURI *uri)
melunko@38
   228
{
melunko@38
   229
	return FALSE;
melunko@38
   230
}
melunko@38
   231
melunko@38
   232
static GnomeVFSMethod method = {
melunko@38
   233
	sizeof (GnomeVFSMethod),
melunko@38
   234
	do_open,
melunko@38
   235
	NULL,
melunko@38
   236
	do_close,
melunko@38
   237
	do_read,
melunko@38
   238
	NULL,
melunko@38
   239
	NULL,
melunko@38
   240
	NULL,
melunko@38
   241
	NULL,
melunko@38
   242
	NULL,
melunko@38
   243
	NULL,
melunko@38
   244
	NULL,
melunko@38
   245
	do_get_file_info,
melunko@38
   246
	NULL,
melunko@38
   247
	do_is_local,
melunko@38
   248
	NULL,
melunko@38
   249
	NULL,
melunko@38
   250
	NULL,
melunko@38
   251
	NULL,
melunko@38
   252
	NULL,
melunko@38
   253
	NULL,
melunko@38
   254
	NULL,
melunko@38
   255
	NULL,
melunko@38
   256
	NULL,
melunko@38
   257
	NULL,
melunko@38
   258
	NULL,
melunko@38
   259
	NULL,
melunko@38
   260
};
melunko@38
   261
melunko@38
   262
melunko@38
   263
GnomeVFSMethod *
melunko@38
   264
vfs_module_init (const char *method_name, const char *args)
melunko@38
   265
{
melunko@38
   266
	return &method;
melunko@38
   267
}
melunko@38
   268
melunko@38
   269
void
melunko@38
   270
vfs_module_shutdown (GnomeVFSMethod *method)
melunko@38
   271
{
melunko@38
   272
}