melunko@38: /* melunko@38: * @author Hallyson Melo melunko@38: * melunko@38: * This program is free software; you can redistribute it and/or modify melunko@38: * it under the terms of the GNU Lesser General Public License as published by melunko@38: * the Free Software Foundation; either version 2 of the License, or melunko@38: * (at your option) any later version. melunko@38: * melunko@38: * This program is distributed in the hope that it will be useful, melunko@38: * but WITHOUT ANY WARRANTY; without even the implied warranty of melunko@38: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the melunko@38: * GNU General Public License for more details. melunko@38: * melunko@38: * You should have received a copy of the GNU Lesser General Public License melunko@38: * along with this program; if not, write to the Free Software melunko@38: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA melunko@38: */ melunko@38: melunko@38: #ifdef HAVE_CONFIG_H melunko@38: #include melunko@38: #endif melunko@38: melunko@38: #include melunko@38: #include melunko@38: #include melunko@38: melunko@38: #include melunko@38: #include melunko@38: melunko@48: #include "gmyth_file_transfer.h" melunko@160: #include "gmyth_backendinfo.h" melunko@38: melunko@38: #define GST_MYTHTV_ID_NUM 1 melunko@38: #define MYTHTV_VERSION_DEFAULT 30 melunko@38: melunko@111: #define MYTHTV_BUFFER_SIZE 1024*64 melunko@111: melunko@38: static GnomeVFSResult do_read (GnomeVFSMethod *method, melunko@38: GnomeVFSMethodHandle *method_handle, melunko@38: gpointer buffer, melunko@38: GnomeVFSFileSize num_bytes, melunko@38: GnomeVFSFileSize *bytes_read, melunko@38: GnomeVFSContext *context); melunko@38: melunko@38: typedef struct { melunko@48: GMythFileTransfer *file_transfer; melunko@38: melunko@38: gint mythtv_version; melunko@38: guint64 content_size; melunko@38: guint64 bytes_read; melunko@111: rosfran@116: GByteArray *buffer; melunko@111: gsize buffer_remain; melunko@38: } MythtvHandle; melunko@38: melunko@38: melunko@38: melunko@38: static GnomeVFSResult melunko@38: do_open (GnomeVFSMethod *method, melunko@38: GnomeVFSMethodHandle **method_handle, melunko@38: GnomeVFSURI *uri, melunko@38: GnomeVFSOpenMode mode, melunko@38: GnomeVFSContext *context) melunko@38: { melunko@111: MythtvHandle *myth_handle; melunko@160: GMythBackendInfo *backend_info; melunko@38: gboolean ret; melunko@38: melunko@38: _GNOME_VFS_METHOD_PARAM_CHECK (method_handle != NULL); melunko@38: _GNOME_VFS_METHOD_PARAM_CHECK (uri != NULL); melunko@38: melunko@111: myth_handle = g_new0 (MythtvHandle, 1); melunko@111: melunko@38: if (mode & GNOME_VFS_OPEN_WRITE) { melunko@38: return GNOME_VFS_ERROR_NOT_PERMITTED; melunko@38: } melunko@38: renatofilho@149: if (gnome_vfs_uri_get_host_name (uri) == NULL) { renatofilho@149: return GNOME_VFS_ERROR_INVALID_HOST_NAME; melunko@38: } melunko@38: melunko@38: /* Initialize mythtv handler*/ melunko@38: myth_handle->file_transfer = NULL; melunko@38: myth_handle->mythtv_version = MYTHTV_VERSION_DEFAULT; melunko@38: myth_handle->bytes_read = 0; melunko@38: myth_handle->content_size = -1; melunko@38: melunko@160: /* Creates and fills out the backend info structure */ melunko@160: backend_info = gmyth_backend_info_new_full ( renatofilho@164: gnome_vfs_uri_get_host_name (uri), renatofilho@164: NULL, NULL, NULL, renatofilho@164: gnome_vfs_uri_get_host_port (uri)); rosfran@180: rosfran@180: gmyth_backend_info_set_path ( backend_info, gnome_vfs_uri_get_path (uri) ); melunko@160: melunko@38: /* Connect to the backend */ melunko@160: myth_handle->file_transfer = gmyth_file_transfer_new (backend_info); renatofilho@188: g_object_unref (backend_info); melunko@38: melunko@111: g_return_val_if_fail (myth_handle->file_transfer != NULL, GNOME_VFS_ERROR_NOT_OPEN); melunko@38: melunko@38: /* sets the Playback monitor connection */ renatofilho@188: ret = gmyth_file_transfer_open (myth_handle->file_transfer, gnome_vfs_uri_get_path (uri)); melunko@38: if (ret == FALSE) { renatofilho@164: g_warning ("Mythtv FileTransfer open error\n"); melunko@111: return GNOME_VFS_ERROR_NOT_OPEN; melunko@38: } melunko@38: melunko@38: // TODO: Verify if file exists in the backend melunko@38: melunko@38: myth_handle->content_size = myth_handle->file_transfer->filesize; melunko@111: rosfran@116: myth_handle->buffer = g_byte_array_sized_new (MYTHTV_BUFFER_SIZE); melunko@111: myth_handle->buffer_remain = 0; melunko@38: melunko@38: *method_handle = (GnomeVFSMethodHandle *) myth_handle; melunko@38: melunko@38: return GNOME_VFS_OK; melunko@38: } melunko@38: melunko@38: static GnomeVFSResult melunko@38: do_read (GnomeVFSMethod *method, melunko@38: GnomeVFSMethodHandle *method_handle, melunko@38: gpointer buffer, melunko@38: GnomeVFSFileSize num_bytes, melunko@38: GnomeVFSFileSize *bytes_read, melunko@38: GnomeVFSContext *context) melunko@38: { melunko@38: MythtvHandle *myth_handle = (MythtvHandle *) method_handle; melunko@38: GnomeVFSFileSize bytes_to_read; melunko@38: melunko@38: *bytes_read = 0; melunko@38: melunko@38: if (myth_handle->bytes_read >= myth_handle->content_size) melunko@38: return GNOME_VFS_ERROR_EOF; melunko@38: melunko@38: // fixme: change this to min math function melunko@38: if (num_bytes > myth_handle->content_size - myth_handle->bytes_read) melunko@38: bytes_to_read = myth_handle->content_size - myth_handle->bytes_read; melunko@38: else melunko@38: bytes_to_read = num_bytes; melunko@38: melunko@38: /* Loop sending the Myth File Transfer request: melunko@38: * Retry whilst authentication fails and we supply it. */ rosfran@116: renatofilho@149: if ( bytes_to_read > myth_handle->buffer_remain ) { renatofilho@149: GByteArray *tmp_buffer = g_byte_array_new(); melunko@38: renatofilho@149: gint len = gmyth_file_transfer_read (myth_handle->file_transfer, renatofilho@149: tmp_buffer, MYTHTV_BUFFER_SIZE - myth_handle->buffer_remain, TRUE); melunko@38: renatofilho@149: if (len < 0) { renatofilho@149: g_byte_array_free (tmp_buffer, TRUE); renatofilho@164: g_warning ("Fail to read bytes"); rosfran@116: return GNOME_VFS_ERROR_IO; renatofilho@149: } melunko@111: renatofilho@149: myth_handle->buffer = g_byte_array_append (myth_handle->buffer, renatofilho@149: tmp_buffer->data, len); rosfran@116: rosfran@116: myth_handle->buffer_remain += len; rosfran@116: renatofilho@169: g_byte_array_free (tmp_buffer, TRUE); rosfran@116: tmp_buffer = NULL; rosfran@116: } melunko@38: melunko@111: bytes_to_read = (bytes_to_read > myth_handle->buffer_remain) ? myth_handle->buffer_remain : bytes_to_read; renatofilho@164: /* gets the first buffer_size bytes from the byte array buffer variable */ rosfran@116: renatofilho@149: g_memmove (buffer, myth_handle->buffer->data, bytes_to_read); rosfran@116: melunko@111: myth_handle->bytes_read += bytes_to_read; rosfran@116: myth_handle->buffer_remain -= bytes_to_read; rosfran@116: rosfran@116: /* flushs the newly buffer got from byte array */ renatofilho@149: myth_handle->buffer = g_byte_array_remove_range (myth_handle->buffer, 0, bytes_to_read); melunko@111: *bytes_read = bytes_to_read; melunko@38: melunko@38: return GNOME_VFS_OK; melunko@38: } melunko@38: melunko@38: static GnomeVFSResult melunko@38: do_close (GnomeVFSMethod *method, melunko@38: GnomeVFSMethodHandle *method_handle, melunko@38: GnomeVFSContext *context) melunko@38: { melunko@111: melunko@38: MythtvHandle *myth_handle = (MythtvHandle *) method_handle; melunko@38: melunko@111: if (myth_handle->file_transfer) { renatofilho@164: gmyth_file_transfer_close (myth_handle->file_transfer); melunko@38: g_object_unref (myth_handle->file_transfer); renatofilho@164: myth_handle->file_transfer = NULL; melunko@111: } rosfran@127: rosfran@127: if (myth_handle->buffer) { renatofilho@164: g_byte_array_free (myth_handle->buffer, TRUE); renatofilho@164: myth_handle->buffer = NULL; rosfran@127: } melunko@38: melunko@38: g_free (myth_handle); melunko@38: melunko@38: return GNOME_VFS_OK; melunko@38: } melunko@38: melunko@38: static GnomeVFSResult melunko@38: do_get_file_info (GnomeVFSMethod *method, melunko@38: GnomeVFSURI *uri, melunko@38: GnomeVFSFileInfo *file_info, melunko@38: GnomeVFSFileInfoOptions options, melunko@38: GnomeVFSContext *context) melunko@38: { renatofilho@188: GMythFileTransfer *file_transfer = NULL; renatofilho@188: GMythBackendInfo *backend_info = NULL; renatofilho@188: renatofilho@188: file_info->name = g_strdup (gnome_vfs_uri_get_path (uri)); melunko@38: file_info->valid_fields = file_info->valid_fields melunko@38: | GNOME_VFS_FILE_INFO_FIELDS_TYPE melunko@38: | GNOME_VFS_FILE_INFO_FIELDS_MIME_TYPE melunko@38: | GNOME_VFS_FILE_INFO_FIELDS_PERMISSIONS; melunko@38: file_info->type = GNOME_VFS_FILE_TYPE_REGULAR; melunko@111: // fixme: get from file extension? melunko@111: file_info->mime_type = g_strdup ("video/x-nuv"); melunko@38: file_info->permissions = melunko@38: GNOME_VFS_PERM_USER_READ | melunko@38: GNOME_VFS_PERM_OTHER_READ | melunko@38: GNOME_VFS_PERM_GROUP_READ; melunko@38: renatofilho@188: backend_info = gmyth_backend_info_new_full (gnome_vfs_uri_get_host_name (uri), renatofilho@188: gnome_vfs_uri_get_user_name (uri), renatofilho@188: gnome_vfs_uri_get_password (uri), renatofilho@188: NULL, renatofilho@188: gnome_vfs_uri_get_host_port (uri)); renatofilho@188: renatofilho@188: file_transfer = gmyth_file_transfer_new (backend_info); renatofilho@188: if (gmyth_file_transfer_open (file_transfer, gnome_vfs_uri_get_path (uri)) == TRUE) { renatofilho@188: file_info->size = gmyth_file_transfer_get_filesize (file_transfer); renatofilho@188: file_info->block_count = GNOME_VFS_FILE_INFO_FIELDS_BLOCK_COUNT; renatofilho@188: file_info->io_block_size = GNOME_VFS_FILE_INFO_FIELDS_IO_BLOCK_SIZE; renatofilho@188: } renatofilho@188: renatofilho@188: g_object_unref (file_transfer); renatofilho@188: g_object_unref (backend_info); melunko@38: return GNOME_VFS_OK; melunko@38: } melunko@38: melunko@38: static gboolean melunko@38: do_is_local (GnomeVFSMethod *method, melunko@38: const GnomeVFSURI *uri) melunko@38: { melunko@38: return FALSE; melunko@38: } melunko@38: melunko@38: static GnomeVFSMethod method = { melunko@38: sizeof (GnomeVFSMethod), melunko@38: do_open, melunko@38: NULL, melunko@38: do_close, melunko@38: do_read, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: do_get_file_info, melunko@38: NULL, melunko@38: do_is_local, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: NULL, melunko@38: }; melunko@38: melunko@38: melunko@38: GnomeVFSMethod * melunko@38: vfs_module_init (const char *method_name, const char *args) melunko@38: { melunko@38: return &method; melunko@38: } melunko@38: melunko@38: void melunko@38: vfs_module_shutdown (GnomeVFSMethod *method) melunko@38: { melunko@38: }