gst-plugins-mythtv/myth_file_transfer.c
author abnerf
Tue Sep 26 15:30:52 2006 +0100 (2006-09-26)
branchtrunk
changeset 12 343d4e3c03ab
parent 2 bd3829c2e9c9
permissions -rwxr-xr-x
[svn r13] svn tree clean.
leo_sobral@2
     1
/* vim: set sw=2: -*- Mode: C; tab-width: 2; indent-tabs-mode: t; c-basic-offset: 2; c-indent-level: 2-*- */
leo_sobral@2
     2
/**
leo_sobral@2
     3
 * GStreamer plug-in properties:
leo_sobral@2
     4
 * - location (backend server hostname/URL) [ex.: myth://192.168.1.73:28722/1000_1092091.nuv]
leo_sobral@2
     5
 * - path (qurl - remote file to be opened)
leo_sobral@2
     6
 * - port number
leo_sobral@2
     7
 *   @author Rosfran Lins Borges <rosfran.borges@indt.org.br>
leo_sobral@2
     8
 */
leo_sobral@2
     9
leo_sobral@2
    10
#include "myth_file_transfer.h"
leo_sobral@2
    11
#include "myth_uri.h"
leo_sobral@2
    12
#include "myth_livetv.h"
leo_sobral@2
    13
#include <gmyth/gmyth_util.h>
leo_sobral@2
    14
#include <gmyth/gmyth_socket.h>
leo_sobral@2
    15
#include <gmyth/gmyth_stringlist.h>
leo_sobral@2
    16
leo_sobral@2
    17
#include <unistd.h>
leo_sobral@2
    18
#include <glib.h>
leo_sobral@2
    19
leo_sobral@2
    20
#include <arpa/inet.h>
leo_sobral@2
    21
#include <sys/types.h>
leo_sobral@2
    22
#include <sys/socket.h>
leo_sobral@2
    23
#include <netdb.h>
leo_sobral@2
    24
#include <errno.h>
leo_sobral@2
    25
#include <stdlib.h>
leo_sobral@2
    26
leo_sobral@2
    27
#define MYTHTV_QUERY_HEADER "QUERY_FILETRANSFER"
leo_sobral@2
    28
#define MYTHTV_RECORDER_HEADER "QUERY_RECORDER"
leo_sobral@2
    29
leo_sobral@2
    30
/* default values to the file transfer parameters */
leo_sobral@2
    31
#define MYTHTV_USER_READ_AHEAD	FALSE
leo_sobral@2
    32
#define MYTHTV_RETRIES			1
leo_sobral@2
    33
#define MYTHTV_FILE_SIZE		-1
leo_sobral@2
    34
leo_sobral@2
    35
#define MYTHTV_BUFFER_SIZE		2048
leo_sobral@2
    36
leo_sobral@2
    37
#define MYTHTV_VERSION			30
leo_sobral@2
    38
leo_sobral@2
    39
#define MYTHTV_TRANSFER_MAX_WAITS	700
leo_sobral@2
    40
leo_sobral@2
    41
#ifdef MYTHTV_ENABLE_DEBUG
leo_sobral@2
    42
#define MYTHTV_ENABLE_DEBUG	1
leo_sobral@2
    43
#else
leo_sobral@2
    44
#undef MYTHTV_ENABLE_DEBUG
leo_sobral@2
    45
#endif
leo_sobral@2
    46
leo_sobral@2
    47
/* this NDEBUG is to maintain compatibility with GMyth library */
leo_sobral@2
    48
#ifndef NDEBUG
leo_sobral@2
    49
#define MYTHTV_ENABLE_DEBUG	1
leo_sobral@2
    50
#endif
leo_sobral@2
    51
leo_sobral@2
    52
static guint wait_to_transfer = 0;
leo_sobral@2
    53
leo_sobral@2
    54
enum myth_sock_types {
leo_sobral@2
    55
  MYTH_PLAYBACK_TYPE = 0,
leo_sobral@2
    56
  MYTH_MONITOR_TYPE,
leo_sobral@2
    57
  MYTH_FILETRANSFER_TYPE,
leo_sobral@2
    58
  MYTH_RINGBUFFER_TYPE
leo_sobral@2
    59
};
leo_sobral@2
    60
leo_sobral@2
    61
static GStaticMutex mutex = G_STATIC_MUTEX_INIT;
leo_sobral@2
    62
leo_sobral@2
    63
static void myth_file_transfer_class_init          (MythFileTransferClass *klass);
leo_sobral@2
    64
static void myth_file_transfer_init                (MythFileTransfer *object);
leo_sobral@2
    65
leo_sobral@2
    66
static void myth_file_transfer_dispose  (GObject *object);
leo_sobral@2
    67
static void myth_file_transfer_finalize (GObject *object);
leo_sobral@2
    68
leo_sobral@2
    69
static GMythSocket *myth_connect_to_transfer_backend( MythFileTransfer **transfer, guint sock_type );
leo_sobral@2
    70
static void* myth_init_io_watchers( void *data );
leo_sobral@2
    71
leo_sobral@2
    72
void myth_file_transfer_close( MythFileTransfer *transfer );
leo_sobral@2
    73
leo_sobral@2
    74
G_DEFINE_TYPE(MythFileTransfer, myth_file_transfer, G_TYPE_OBJECT)
leo_sobral@2
    75
leo_sobral@2
    76
static guint64
leo_sobral@2
    77
mmyth_util_decode_long_long( GMythStringList *strlist, guint offset  )
leo_sobral@2
    78
{
leo_sobral@2
    79
leo_sobral@2
    80
  guint64 ret_value = 0LL;
leo_sobral@2
    81
leo_sobral@2
    82
  g_return_val_if_fail( strlist != NULL, ret_value );
leo_sobral@2
    83
leo_sobral@2
    84
  if ( offset < gmyth_string_list_length( strlist ))
leo_sobral@2
    85
    g_printerr( "[%s] Offset is lower than the GMythStringList (offset = %d)!\n", __FUNCTION__, offset );
leo_sobral@2
    86
  g_return_val_if_fail( offset < gmyth_string_list_length( strlist ), ret_value );
leo_sobral@2
    87
leo_sobral@2
    88
  gint l1 = gmyth_string_list_get_int( strlist, offset );
leo_sobral@2
    89
  gint l2 = gmyth_string_list_get_int( strlist, offset + 1 );
leo_sobral@2
    90
leo_sobral@2
    91
  ret_value = ((guint64)(l2) & 0xffffffffLL) | ((guint64)(l1) << 32);
leo_sobral@2
    92
leo_sobral@2
    93
  return ret_value;
leo_sobral@2
    94
leo_sobral@2
    95
}
leo_sobral@2
    96
leo_sobral@2
    97
static void
leo_sobral@2
    98
myth_file_transfer_class_init (MythFileTransferClass *klass)
leo_sobral@2
    99
{
leo_sobral@2
   100
  GObjectClass *gobject_class;
leo_sobral@2
   101
leo_sobral@2
   102
  gobject_class = (GObjectClass *) klass;
leo_sobral@2
   103
leo_sobral@2
   104
  gobject_class->dispose  = myth_file_transfer_dispose;
leo_sobral@2
   105
  gobject_class->finalize = myth_file_transfer_finalize;
leo_sobral@2
   106
}
leo_sobral@2
   107
leo_sobral@2
   108
  static void
leo_sobral@2
   109
myth_file_transfer_init (MythFileTransfer *myth_file_transfer)
leo_sobral@2
   110
{ 
leo_sobral@2
   111
  g_return_if_fail( myth_file_transfer != NULL );
leo_sobral@2
   112
  myth_file_transfer->mythtv_version = MYTHTV_VERSION;
leo_sobral@2
   113
}
leo_sobral@2
   114
leo_sobral@2
   115
static void
leo_sobral@2
   116
myth_file_transfer_dispose  (GObject *object)
leo_sobral@2
   117
{
leo_sobral@2
   118
  MythFileTransfer *myth_file_transfer = MYTH_FILE_TRANSFER(object);
leo_sobral@2
   119
leo_sobral@2
   120
  myth_file_transfer_close( myth_file_transfer );
leo_sobral@2
   121
leo_sobral@2
   122
  G_OBJECT_CLASS (myth_file_transfer_parent_class)->dispose (object);
leo_sobral@2
   123
}
leo_sobral@2
   124
leo_sobral@2
   125
  static void
leo_sobral@2
   126
myth_file_transfer_finalize (GObject *object)
leo_sobral@2
   127
{
leo_sobral@2
   128
  g_signal_handlers_destroy (object);
leo_sobral@2
   129
leo_sobral@2
   130
  G_OBJECT_CLASS (myth_file_transfer_parent_class)->finalize (object);
leo_sobral@2
   131
}
leo_sobral@2
   132
leo_sobral@2
   133
  MythFileTransfer*
leo_sobral@2
   134
myth_file_transfer_new (gint num, GString *uri_str, gshort port, gint mythtv_version)
leo_sobral@2
   135
{
leo_sobral@2
   136
  MythFileTransfer *transfer = MYTH_FILE_TRANSFER ( g_object_new (
leo_sobral@2
   137
	MYTH_FILE_TRANSFER_TYPE, FALSE ));
leo_sobral@2
   138
leo_sobral@2
   139
  if ( mythtv_version > 0 )
leo_sobral@2
   140
    transfer->mythtv_version = mythtv_version;
leo_sobral@2
   141
leo_sobral@2
   142
  transfer->card_id = num;
leo_sobral@2
   143
leo_sobral@2
   144
  transfer->rec_id = -1;
leo_sobral@2
   145
leo_sobral@2
   146
  transfer->recordernum = 0;
leo_sobral@2
   147
  transfer->uri = myth_uri_new ( uri_str->str );
leo_sobral@2
   148
leo_sobral@2
   149
  transfer->hostname = g_string_new( myth_uri_gethost(transfer->uri) );
leo_sobral@2
   150
  g_print( "\t--> transfer->hostname = %s\n", transfer->hostname->str );
leo_sobral@2
   151
leo_sobral@2
   152
  if ( port >= 0 )
leo_sobral@2
   153
    transfer->port = port;
leo_sobral@2
   154
  else
leo_sobral@2
   155
    transfer->port = myth_uri_getport( transfer->uri );
leo_sobral@2
   156
leo_sobral@2
   157
  g_print( "\t--> transfer->port = %d\n", transfer->port );
leo_sobral@2
   158
leo_sobral@2
   159
  transfer->readposition = 0;
leo_sobral@2
   160
  transfer->filesize = MYTHTV_FILE_SIZE;
leo_sobral@2
   161
  transfer->timeoutisfast = FALSE;
leo_sobral@2
   162
leo_sobral@2
   163
  transfer->userreadahead = MYTHTV_USER_READ_AHEAD;
leo_sobral@2
   164
  transfer->retries = MYTHTV_RETRIES;  
leo_sobral@2
   165
leo_sobral@2
   166
  transfer->live_tv = FALSE;
leo_sobral@2
   167
leo_sobral@2
   168
  transfer->query = g_string_new( MYTHTV_QUERY_HEADER );
leo_sobral@2
   169
  g_string_append_printf ( transfer->query, " %d", transfer->recordernum );
leo_sobral@2
   170
  g_print( "\t--> transfer->query = %s\n", transfer->query->str );
leo_sobral@2
   171
leo_sobral@2
   172
  transfer->control_sock = NULL;
leo_sobral@2
   173
  transfer->event_sock = NULL;
leo_sobral@2
   174
  transfer->sock = NULL;
leo_sobral@2
   175
leo_sobral@2
   176
  return transfer;
leo_sobral@2
   177
}
leo_sobral@2
   178
leo_sobral@2
   179
gboolean
leo_sobral@2
   180
myth_file_transfer_livetv_setup( MythFileTransfer **transfer, GMythSocket *live_socket )
leo_sobral@2
   181
{
leo_sobral@2
   182
	(*transfer)->sock = live_socket;
leo_sobral@2
   183
	g_object_ref( live_socket );
leo_sobral@2
   184
leo_sobral@2
   185
	return TRUE;
leo_sobral@2
   186
}
leo_sobral@2
   187
leo_sobral@2
   188
gboolean
leo_sobral@2
   189
myth_file_transfer_playback_setup( MythFileTransfer **transfer, gboolean live_tv )
leo_sobral@2
   190
{
leo_sobral@2
   191
leo_sobral@2
   192
  gboolean ret = TRUE;
leo_sobral@2
   193
leo_sobral@2
   194
  (*transfer)->live_tv = live_tv;
leo_sobral@2
   195
leo_sobral@2
   196
  printf("[%s] Running config to the %s...\n", __FUNCTION__, live_tv ? "LiveTV" : "FileTransfer" );
leo_sobral@2
   197
leo_sobral@2
   198
  /* configure the control socket */
leo_sobral@2
   199
  if ((*transfer)->control_sock == NULL) { 
leo_sobral@2
   200
leo_sobral@2
   201
    if ( myth_connect_to_transfer_backend ( transfer, MYTH_PLAYBACK_TYPE ) == NULL ) {
leo_sobral@2
   202
      g_printerr( "Connection to backend failed (Control Socket).\n" );
leo_sobral@2
   203
      ret = FALSE;
leo_sobral@2
   204
    }
leo_sobral@2
   205
leo_sobral@2
   206
  } else {
leo_sobral@2
   207
    g_warning("Remote transfer control socket already created.\n");
leo_sobral@2
   208
  }
leo_sobral@2
   209
leo_sobral@2
   210
  return ret;
leo_sobral@2
   211
leo_sobral@2
   212
}
leo_sobral@2
   213
leo_sobral@2
   214
gboolean
leo_sobral@2
   215
myth_file_transfer_setup( MythFileTransfer **transfer, gboolean live_tv )
leo_sobral@2
   216
{
leo_sobral@2
   217
  GMythStringList *strlist = NULL;
leo_sobral@2
   218
leo_sobral@2
   219
  gboolean ret = TRUE;
leo_sobral@2
   220
leo_sobral@2
   221
  (*transfer)->live_tv = live_tv;
leo_sobral@2
   222
leo_sobral@2
   223
  printf("[%s] Running config to the %s...\n", __FUNCTION__, live_tv ? "LiveTV" : "FileTransfer" );
leo_sobral@2
   224
leo_sobral@2
   225
#if 0
leo_sobral@2
   226
  /* configure the event socket */
leo_sobral@2
   227
  if ((*transfer)->event_sock == NULL) { 
leo_sobral@2
   228
leo_sobral@2
   229
    if ( myth_connect_to_transfer_backend ( transfer, MYTH_MONITOR_TYPE ) == NULL ) {
leo_sobral@2
   230
      g_printerr( "Connection to backend failed (Event Socket).\n" );
leo_sobral@2
   231
      ret = FALSE;
leo_sobral@2
   232
    }
leo_sobral@2
   233
leo_sobral@2
   234
  } else {
leo_sobral@2
   235
    g_warning("Remote transfer control socket already created.\n");
leo_sobral@2
   236
  }
leo_sobral@2
   237
#endif
leo_sobral@2
   238
leo_sobral@2
   239
  /* configure the socket */
leo_sobral@2
   240
  if ( (*transfer)->sock == NULL ) { 
leo_sobral@2
   241
leo_sobral@2
   242
    //if ( live_tv == FALSE ) {
leo_sobral@2
   243
leo_sobral@2
   244
    if ( myth_connect_to_transfer_backend ( transfer, MYTH_FILETRANSFER_TYPE ) == NULL ) {
leo_sobral@2
   245
      g_printerr ("Connection to backend failed (Raw Transfer Socket).\n");
leo_sobral@2
   246
      ret = FALSE;
leo_sobral@2
   247
    }
leo_sobral@2
   248
leo_sobral@2
   249
    if ( !(*transfer)->live_tv && (*transfer)->control_sock != NULL) {
leo_sobral@2
   250
      strlist = gmyth_string_list_new();
leo_sobral@2
   251
      g_string_printf ( (*transfer)->query, "%s %d", MYTHTV_QUERY_HEADER, (*transfer)->recordernum );
leo_sobral@2
   252
leo_sobral@2
   253
      gmyth_string_list_append_string( strlist, (*transfer)->query );
leo_sobral@2
   254
      gmyth_string_list_append_char_array( strlist, "IS_OPEN" );
leo_sobral@2
   255
leo_sobral@2
   256
      gmyth_socket_write_stringlist( (*transfer)->control_sock, strlist );
leo_sobral@2
   257
      gmyth_socket_read_stringlist( (*transfer)->control_sock, strlist );
leo_sobral@2
   258
leo_sobral@2
   259
      if ( strlist!=NULL && gmyth_string_list_get_int( strlist, 0 ) == 1 ) {
leo_sobral@2
   260
	g_print( "[%s] Remote Myth FileTransfer socket is open!\n", __FUNCTION__ );
leo_sobral@2
   261
      } else {
leo_sobral@2
   262
	g_print( "[%s] Remote Myth FileTransfer socket is CLOSED! See the MythTV Server Backend for configuration details...\n", __FUNCTION__ );
leo_sobral@2
   263
	ret = FALSE;
leo_sobral@2
   264
      }
leo_sobral@2
   265
    }
leo_sobral@2
   266
leo_sobral@2
   267
  } else {
leo_sobral@2
   268
    g_warning("Remote transfer (raw) socket already created.\n");
leo_sobral@2
   269
  }
leo_sobral@2
   270
leo_sobral@2
   271
  return ret;
leo_sobral@2
   272
}
leo_sobral@2
   273
leo_sobral@2
   274
static GMythSocket *
leo_sobral@2
   275
myth_connect_to_transfer_backend( MythFileTransfer **transfer, guint sock_type )
leo_sobral@2
   276
{
leo_sobral@2
   277
  GMythSocket *sock = NULL;
leo_sobral@2
   278
leo_sobral@2
   279
  g_return_val_if_fail( transfer != NULL && *transfer != NULL, NULL );
leo_sobral@2
   280
  g_return_val_if_fail( (*transfer)->uri != NULL, NULL );
leo_sobral@2
   281
leo_sobral@2
   282
  g_static_mutex_lock (&mutex);
leo_sobral@2
   283
leo_sobral@2
   284
  gchar *path_dir = myth_uri_getpath( (*transfer)->uri );
leo_sobral@2
   285
  //g_print( "\t--> %s: path_dir = %s\n", __FUNCTION__, path_dir );
leo_sobral@2
   286
leo_sobral@2
   287
  gchar *stype = g_strdup( "" );
leo_sobral@2
   288
leo_sobral@2
   289
  //  if ( (*transfer)->live_tv == FALSE ) {
leo_sobral@2
   290
leo_sobral@2
   291
  sock = gmyth_socket_new();
leo_sobral@2
   292
leo_sobral@2
   293
  gmyth_socket_connect( &sock, (*transfer)->hostname->str, (*transfer)->port );
leo_sobral@2
   294
leo_sobral@2
   295
  /*
leo_sobral@2
   296
     } else {
leo_sobral@2
   297
     sock = (*transfer)->sock;
leo_sobral@2
   298
     }
leo_sobral@2
   299
     */
leo_sobral@2
   300
#ifdef MYTHTV_ENABLE_DEBUG
leo_sobral@2
   301
leo_sobral@2
   302
  g_print( "[%s] --> Creating socket... (%s, %d)\n", __FUNCTION__, (*transfer)->hostname->str, (*transfer)->port );
leo_sobral@2
   303
#endif
leo_sobral@2
   304
leo_sobral@2
   305
  GMythStringList *strlist = NULL;
leo_sobral@2
   306
leo_sobral@2
   307
  GString *hostname = g_string_new( myth_uri_gethost( (*transfer)->uri ) );
leo_sobral@2
   308
  GString *base_str = g_string_new( "" );
leo_sobral@2
   309
leo_sobral@2
   310
  if ( gmyth_socket_check_protocol_version_number (sock, (*transfer)->mythtv_version) ) {
leo_sobral@2
   311
leo_sobral@2
   312
    if (sock == NULL) {
leo_sobral@2
   313
      stype = (sock_type==MYTH_PLAYBACK_TYPE) ? "control socket" : "file data socket";
leo_sobral@2
   314
      g_printerr( "FileTransfer, open_socket(%s): \n"
leo_sobral@2
   315
	  "\t\t\tCould not connect to server \"%s\" @ port %d\n", stype, 
leo_sobral@2
   316
	  (*transfer)->hostname->str, (*transfer)->port );
leo_sobral@2
   317
      g_object_unref(sock);
leo_sobral@2
   318
      g_static_mutex_unlock (&mutex);
leo_sobral@2
   319
      return NULL;
leo_sobral@2
   320
    }
leo_sobral@2
   321
leo_sobral@2
   322
    hostname = gmyth_socket_get_local_hostname();
leo_sobral@2
   323
leo_sobral@2
   324
    g_print( "[%s] local hostname = %s\n", __FUNCTION__, hostname->str  );
leo_sobral@2
   325
leo_sobral@2
   326
    if ( sock_type == MYTH_PLAYBACK_TYPE )
leo_sobral@2
   327
    {
leo_sobral@2
   328
      (*transfer)->control_sock = sock;
leo_sobral@2
   329
      g_string_printf( base_str, "ANN Playback %s %d", hostname->str, FALSE );
leo_sobral@2
   330
leo_sobral@2
   331
      gmyth_socket_send_command( (*transfer)->control_sock, base_str );
leo_sobral@2
   332
      GString *resp = gmyth_socket_receive_response( (*transfer)->control_sock );
leo_sobral@2
   333
      g_print( "[%s] Got Playback response from %s: %s\n", __FUNCTION__, base_str->str, resp->str );
leo_sobral@2
   334
    }
leo_sobral@2
   335
    else if ( sock_type == MYTH_MONITOR_TYPE )
leo_sobral@2
   336
    {
leo_sobral@2
   337
      (*transfer)->event_sock = sock;
leo_sobral@2
   338
      g_string_printf( base_str, "ANN Monitor %s %d", hostname->str, TRUE );
leo_sobral@2
   339
leo_sobral@2
   340
      gmyth_socket_send_command( (*transfer)->event_sock, base_str );
leo_sobral@2
   341
      GString *resp = gmyth_socket_receive_response( (*transfer)->event_sock );
leo_sobral@2
   342
      g_print( "[%s] Got Monitor response from %s: %s\n", __FUNCTION__, base_str->str, resp->str );
leo_sobral@2
   343
      g_thread_create( myth_init_io_watchers, (void*)(*transfer), FALSE, NULL );
leo_sobral@2
   344
leo_sobral@2
   345
      g_printerr( "[%s] Watch listener function to the IO control channel on thread %p.\n", __FUNCTION__, g_thread_self() );
leo_sobral@2
   346
leo_sobral@2
   347
    }
leo_sobral@2
   348
    else if ( sock_type == MYTH_FILETRANSFER_TYPE )
leo_sobral@2
   349
    {
leo_sobral@2
   350
      (*transfer)->sock = sock;
leo_sobral@2
   351
      strlist = gmyth_string_list_new();
leo_sobral@2
   352
      //g_string_printf( base_str, "ANN FileTransfer %s %d %d", hostname->str,
leo_sobral@2
   353
      //  		transfer->userreadahead, transfer->retries );
leo_sobral@2
   354
      g_string_printf( base_str, "ANN FileTransfer %s", hostname->str );
leo_sobral@2
   355
leo_sobral@2
   356
      gmyth_string_list_append_string( strlist, base_str );
leo_sobral@2
   357
      gmyth_string_list_append_char_array( strlist, path_dir );
leo_sobral@2
   358
leo_sobral@2
   359
      gmyth_socket_write_stringlist( (*transfer)->sock, strlist );
leo_sobral@2
   360
      gmyth_socket_read_stringlist( (*transfer)->sock, strlist );
leo_sobral@2
   361
leo_sobral@2
   362
      /* socket number, where all the stream data comes from - got from the MythTV remote backend */
leo_sobral@2
   363
      (*transfer)->recordernum = gmyth_string_list_get_int( strlist, 1 );
leo_sobral@2
   364
leo_sobral@2
   365
      /* Myth URI stream file size - decoded using two 8-bytes sequences (64 bits/long long types) */
leo_sobral@2
   366
      (*transfer)->filesize = mmyth_util_decode_long_long( strlist, 2 );
leo_sobral@2
   367
leo_sobral@2
   368
      printf( "[%s] ***** Received: recordernum = %d, filesize = %" G_GUINT64_FORMAT "\n", __FUNCTION__,
leo_sobral@2
   369
	  (*transfer)->recordernum, (*transfer)->filesize );
leo_sobral@2
   370
leo_sobral@2
   371
      if ( (*transfer)->filesize <= 0 ) {
leo_sobral@2
   372
	g_print( "[%s] Got filesize equals to %llu is lesser than 0 [invalid stream file]\n", __FUNCTION__, (*transfer)->filesize );
leo_sobral@2
   373
	g_object_unref(sock);
leo_sobral@2
   374
	sock = NULL; 
leo_sobral@2
   375
      }
leo_sobral@2
   376
    }
leo_sobral@2
   377
    else if ( sock_type == MYTH_RINGBUFFER_TYPE )
leo_sobral@2
   378
    {
leo_sobral@2
   379
      (*transfer)->sock = sock;
leo_sobral@2
   380
      //myth_file_transfer_spawntv( (*transfer), NULL );      
leo_sobral@2
   381
leo_sobral@2
   382
      strlist = gmyth_string_list_new();
leo_sobral@2
   383
      g_string_printf( base_str, "ANN RingBuffer %s %d", hostname->str, (*transfer)->card_id );
leo_sobral@2
   384
leo_sobral@2
   385
      gmyth_socket_send_command( (*transfer)->sock, base_str );
leo_sobral@2
   386
      GString *resp = gmyth_socket_receive_response( (*transfer)->sock );
leo_sobral@2
   387
      g_print( "[%s] Got RingBuffer response from %s: %s\n", __FUNCTION__, base_str->str, resp->str );
leo_sobral@2
   388
leo_sobral@2
   389
    }
leo_sobral@2
   390
leo_sobral@2
   391
  }
leo_sobral@2
   392
leo_sobral@2
   393
  printf("[%s] ANN %s sent: %s\n", (sock_type==MYTH_PLAYBACK_TYPE) ? "Playback" : (sock_type==MYTH_FILETRANSFER_TYPE) ? "FileTransfer" : "Monitor", __FUNCTION__, base_str->str);
leo_sobral@2
   394
leo_sobral@2
   395
  if ( strlist != NULL )
leo_sobral@2
   396
    g_object_unref( strlist );
leo_sobral@2
   397
leo_sobral@2
   398
  g_static_mutex_unlock (&mutex);
leo_sobral@2
   399
leo_sobral@2
   400
  return sock;
leo_sobral@2
   401
}    
leo_sobral@2
   402
leo_sobral@2
   403
void
leo_sobral@2
   404
myth_file_transfer_spawntv ( MythFileTransfer *file_transfer, 
leo_sobral@2
   405
    GString *tvchain_id )
leo_sobral@2
   406
{
leo_sobral@2
   407
  GMythStringList *str_list;
leo_sobral@2
   408
leo_sobral@2
   409
  g_debug ("myth_file_transfer_spawntv.\n");
leo_sobral@2
   410
leo_sobral@2
   411
  str_list = gmyth_string_list_new ();
leo_sobral@2
   412
leo_sobral@2
   413
  g_string_printf( file_transfer->query, "%s %d", MYTHTV_RECORDER_HEADER, 
leo_sobral@2
   414
      file_transfer->card_id );
leo_sobral@2
   415
  gmyth_string_list_append_string (str_list, file_transfer->query);
leo_sobral@2
   416
  gmyth_string_list_append_string (str_list, g_string_new ("SPAWN_LIVETV"));
leo_sobral@2
   417
  if (tvchain_id!=NULL) {
leo_sobral@2
   418
    gmyth_string_list_append_string (str_list, tvchain_id);
leo_sobral@2
   419
    gmyth_string_list_append_int (str_list, FALSE); // PIP = FALSE (0)
leo_sobral@2
   420
  }
leo_sobral@2
   421
leo_sobral@2
   422
  gmyth_socket_sendreceive_stringlist ( file_transfer->sock, str_list );
leo_sobral@2
   423
leo_sobral@2
   424
  //GString *str = NULL;
leo_sobral@2
   425
leo_sobral@2
   426
  //if (str_list!=NULL && (str = gmyth_string_list_get_string( str_list, 0 )) != NULL && strcasecmp( str->str, "ok" ) != 0 ) {
leo_sobral@2
   427
  //  g_print( "[%s]\t\tSpawnLiveTV is OK!\n", __FUNCTION__ );
leo_sobral@2
   428
  //}
leo_sobral@2
   429
  if (str_list!=NULL)
leo_sobral@2
   430
    g_object_unref (str_list);
leo_sobral@2
   431
leo_sobral@2
   432
}
leo_sobral@2
   433
leo_sobral@2
   434
gboolean
leo_sobral@2
   435
myth_file_transfer_is_recording ( MythFileTransfer *file_transfer )
leo_sobral@2
   436
{
leo_sobral@2
   437
  gboolean ret = TRUE;
leo_sobral@2
   438
  
leo_sobral@2
   439
  GMythStringList *str_list = gmyth_string_list_new ();
leo_sobral@2
   440
leo_sobral@2
   441
  g_debug ( "[%s]\n", __FUNCTION__ );
leo_sobral@2
   442
  g_static_mutex_lock (&mutex);
leo_sobral@2
   443
leo_sobral@2
   444
  g_string_printf( file_transfer->query, "%s %d", MYTHTV_RECORDER_HEADER, 
leo_sobral@2
   445
      file_transfer->rec_id >= 0 ? file_transfer->rec_id : file_transfer->card_id );
leo_sobral@2
   446
  gmyth_string_list_append_string (str_list, file_transfer->query);
leo_sobral@2
   447
  gmyth_string_list_append_string (str_list, g_string_new ("IS_RECORDING"));
leo_sobral@2
   448
leo_sobral@2
   449
  gmyth_socket_sendreceive_stringlist ( file_transfer->control_sock, str_list );
leo_sobral@2
   450
leo_sobral@2
   451
  if ( str_list != NULL && gmyth_string_list_length(str_list) > 0 ) 
leo_sobral@2
   452
  {
leo_sobral@2
   453
    GString *str = NULL;
leo_sobral@2
   454
    if ( ( str = gmyth_string_list_get_string( str_list, 0 ) ) != NULL && strcmp( str->str, "bad" )!= 0 ) {
leo_sobral@2
   455
      gint is_rec = gmyth_string_list_get_int( str_list, 0 );
leo_sobral@2
   456
      if ( is_rec != 0 )
leo_sobral@2
   457
	ret = TRUE;
leo_sobral@2
   458
      else
leo_sobral@2
   459
	ret = FALSE;
leo_sobral@2
   460
    }
leo_sobral@2
   461
  }  
leo_sobral@2
   462
  g_print( "[%s] %s, stream is %s being recorded!\n", __FUNCTION__, ret ? "YES" : "NO", ret ? "" : "NOT" );
leo_sobral@2
   463
  g_static_mutex_unlock (&mutex);
leo_sobral@2
   464
leo_sobral@2
   465
  if ( str_list != NULL )
leo_sobral@2
   466
    g_object_unref (str_list);
leo_sobral@2
   467
leo_sobral@2
   468
  return ret;
leo_sobral@2
   469
leo_sobral@2
   470
}
leo_sobral@2
   471
leo_sobral@2
   472
guint64
leo_sobral@2
   473
myth_file_transfer_get_file_position ( MythFileTransfer *file_transfer )
leo_sobral@2
   474
{
leo_sobral@2
   475
  guint64 pos = 0;
leo_sobral@2
   476
leo_sobral@2
   477
  GMythStringList *str_list = gmyth_string_list_new ();
leo_sobral@2
   478
leo_sobral@2
   479
  g_debug ( "[%s]\n", __FUNCTION__ );
leo_sobral@2
   480
  g_static_mutex_lock (&mutex);
leo_sobral@2
   481
leo_sobral@2
   482
  g_string_printf( file_transfer->query, "%s %d", MYTHTV_RECORDER_HEADER, 
leo_sobral@2
   483
      file_transfer->rec_id >= 0 ? file_transfer->rec_id : file_transfer->card_id );
leo_sobral@2
   484
leo_sobral@2
   485
  gmyth_string_list_append_string (str_list, file_transfer->query);
leo_sobral@2
   486
  gmyth_string_list_append_string (str_list, g_string_new ("GET_FILE_POSITION"));
leo_sobral@2
   487
leo_sobral@2
   488
  gmyth_socket_sendreceive_stringlist ( file_transfer->control_sock, str_list );
leo_sobral@2
   489
leo_sobral@2
   490
  if ( str_list != NULL && gmyth_string_list_length(str_list) > 0 ) 
leo_sobral@2
   491
  {
leo_sobral@2
   492
    GString *str = NULL;
leo_sobral@2
   493
    if ( ( str = gmyth_string_list_get_string( str_list, 0 ) ) != NULL && strcmp ( str->str, "bad" ) != 0 )
leo_sobral@2
   494
      pos = gmyth_util_decode_long_long( str_list, 0 );
leo_sobral@2
   495
  } 
leo_sobral@2
   496
  g_static_mutex_unlock (&mutex);
leo_sobral@2
   497
leo_sobral@2
   498
#ifndef MYTHTV_ENABLE_DEBUG
leo_sobral@2
   499
leo_sobral@2
   500
  g_print( "[%s] Got file position = %llu\n", __FUNCTION__, pos );
leo_sobral@2
   501
#endif
leo_sobral@2
   502
  if (str_list!=NULL)
leo_sobral@2
   503
    g_object_unref (str_list);
leo_sobral@2
   504
leo_sobral@2
   505
  return pos;
leo_sobral@2
   506
leo_sobral@2
   507
}
leo_sobral@2
   508
leo_sobral@2
   509
  glong
leo_sobral@2
   510
myth_file_transfer_get_recordernum( MythFileTransfer *transfer )
leo_sobral@2
   511
{
leo_sobral@2
   512
  return transfer->recordernum;
leo_sobral@2
   513
}
leo_sobral@2
   514
leo_sobral@2
   515
  glong
leo_sobral@2
   516
myth_file_transfer_get_filesize( MythFileTransfer *transfer )
leo_sobral@2
   517
{
leo_sobral@2
   518
  return transfer->filesize;
leo_sobral@2
   519
}
leo_sobral@2
   520
leo_sobral@2
   521
  gboolean
leo_sobral@2
   522
myth_file_transfer_isopen( MythFileTransfer *transfer )
leo_sobral@2
   523
{
leo_sobral@2
   524
  return (transfer->sock != NULL && transfer->control_sock != NULL);
leo_sobral@2
   525
}
leo_sobral@2
   526
leo_sobral@2
   527
  void
leo_sobral@2
   528
myth_file_transfer_close( MythFileTransfer *transfer )
leo_sobral@2
   529
{
leo_sobral@2
   530
  GMythStringList *strlist;
leo_sobral@2
   531
leo_sobral@2
   532
  if (transfer->control_sock == NULL)
leo_sobral@2
   533
    return;
leo_sobral@2
   534
leo_sobral@2
   535
  strlist = gmyth_string_list_new( );
leo_sobral@2
   536
leo_sobral@2
   537
  g_string_printf( transfer->query, "%s %d", MYTHTV_QUERY_HEADER, 
leo_sobral@2
   538
      transfer->recordernum );
leo_sobral@2
   539
  gmyth_string_list_append_string( strlist, transfer->query );
leo_sobral@2
   540
  gmyth_string_list_append_char_array( strlist, "DONE" );
leo_sobral@2
   541
leo_sobral@2
   542
leo_sobral@2
   543
  if ( gmyth_socket_sendreceive_stringlist(transfer->control_sock, strlist) <= 0 )
leo_sobral@2
   544
  {
leo_sobral@2
   545
    g_printerr( "Remote file timeout.\n" );
leo_sobral@2
   546
  }
leo_sobral@2
   547
leo_sobral@2
   548
  if (transfer->sock)
leo_sobral@2
   549
  {
leo_sobral@2
   550
    g_object_unref( transfer->sock );
leo_sobral@2
   551
    transfer->sock = NULL;
leo_sobral@2
   552
  }
leo_sobral@2
   553
leo_sobral@2
   554
  if (transfer->control_sock)
leo_sobral@2
   555
  {
leo_sobral@2
   556
    g_object_unref( transfer->control_sock );
leo_sobral@2
   557
    transfer->control_sock = NULL;
leo_sobral@2
   558
  } 
leo_sobral@2
   559
leo_sobral@2
   560
}
leo_sobral@2
   561
leo_sobral@2
   562
  void
leo_sobral@2
   563
myth_file_transfer_reset_controlsock( MythFileTransfer *transfer )
leo_sobral@2
   564
{
leo_sobral@2
   565
  if (transfer->control_sock == NULL)
leo_sobral@2
   566
  {
leo_sobral@2
   567
    g_printerr( "myth_file_transfer_reset_controlsock(): Called with no control socket" );
leo_sobral@2
   568
    return;
leo_sobral@2
   569
  }
leo_sobral@2
   570
leo_sobral@2
   571
  GString *str = gmyth_socket_receive_response( transfer->control_sock );
leo_sobral@2
   572
leo_sobral@2
   573
  g_string_free( str, TRUE );
leo_sobral@2
   574
}
leo_sobral@2
   575
leo_sobral@2
   576
void
leo_sobral@2
   577
myth_file_transfer_reset_sock( MythFileTransfer *transfer )
leo_sobral@2
   578
{
leo_sobral@2
   579
  if ( transfer->sock == NULL )
leo_sobral@2
   580
  {
leo_sobral@2
   581
    g_printerr( "myth_file_transfer_reset_sock(): Called with no raw socket" );
leo_sobral@2
   582
    return;
leo_sobral@2
   583
  }
leo_sobral@2
   584
leo_sobral@2
   585
  GString *str = gmyth_socket_receive_response( transfer->sock );
leo_sobral@2
   586
leo_sobral@2
   587
  g_string_free( str, TRUE );
leo_sobral@2
   588
}
leo_sobral@2
   589
leo_sobral@2
   590
void
leo_sobral@2
   591
myth_file_transfer_reset( MythFileTransfer *transfer )
leo_sobral@2
   592
{
leo_sobral@2
   593
  myth_file_transfer_reset_controlsock( transfer );
leo_sobral@2
   594
  myth_file_transfer_reset_sock( transfer );
leo_sobral@2
   595
}
leo_sobral@2
   596
leo_sobral@2
   597
guint64
leo_sobral@2
   598
myth_file_transfer_seek(MythFileTransfer *transfer, guint64 pos, gint whence)
leo_sobral@2
   599
{
leo_sobral@2
   600
  if (transfer->sock == NULL)
leo_sobral@2
   601
  {
leo_sobral@2
   602
    g_printerr( "[%s] myth_file_transfer_seek(): Called with no socket", __FUNCTION__ );
leo_sobral@2
   603
    return 0;
leo_sobral@2
   604
  }
leo_sobral@2
   605
leo_sobral@2
   606
  if (transfer->control_sock == NULL)
leo_sobral@2
   607
    return 0;
leo_sobral@2
   608
leo_sobral@2
   609
  // if (!controlSock->isOpen() || controlSock->error())
leo_sobral@2
   610
  //   return 0;
leo_sobral@2
   611
leo_sobral@2
   612
  GMythStringList *strlist = gmyth_string_list_new();
leo_sobral@2
   613
  g_string_printf (transfer->query, "%s %d", MYTHTV_QUERY_HEADER, transfer->recordernum);
leo_sobral@2
   614
  gmyth_string_list_append_string( strlist, transfer->query );  
leo_sobral@2
   615
  gmyth_string_list_append_char_array( strlist, "SEEK" );
leo_sobral@2
   616
  gmyth_string_list_append_uint64( strlist, pos );
leo_sobral@2
   617
  //  gmyth_string_list_append_int( strlist, whence );
leo_sobral@2
   618
leo_sobral@2
   619
  if (pos > 0 )
leo_sobral@2
   620
    gmyth_string_list_append_uint64( strlist, pos );
leo_sobral@2
   621
  else
leo_sobral@2
   622
    gmyth_string_list_append_uint64( strlist, transfer->readposition );
leo_sobral@2
   623
leo_sobral@2
   624
  gmyth_socket_sendreceive_stringlist( transfer->control_sock, strlist );
leo_sobral@2
   625
leo_sobral@2
   626
  guint64 retval = gmyth_string_list_get_uint64(strlist, 0);
leo_sobral@2
   627
  transfer->readposition = retval;
leo_sobral@2
   628
  g_print( "[%s] got reading position pointer from the streaming = %llu\n", 
leo_sobral@2
   629
      __FUNCTION__, retval );
leo_sobral@2
   630
leo_sobral@2
   631
  //myth_file_transfer_reset( transfer );
leo_sobral@2
   632
leo_sobral@2
   633
  return retval;
leo_sobral@2
   634
}
leo_sobral@2
   635
leo_sobral@2
   636
static gboolean
leo_sobral@2
   637
myth_control_sock_listener( GIOChannel *source, GIOCondition condition, gpointer data )
leo_sobral@2
   638
{
leo_sobral@2
   639
leo_sobral@2
   640
  GIOStatus ret;
leo_sobral@2
   641
  GError *err = NULL;
leo_sobral@2
   642
  gchar *msg = g_strdup("");
leo_sobral@2
   643
leo_sobral@2
   644
  gsize len;
leo_sobral@2
   645
  if (condition & G_IO_HUP)
leo_sobral@2
   646
    g_error ("Read end of pipe died!\n");
leo_sobral@2
   647
  ret = g_io_channel_read_line ( source, &msg, &len, NULL, &err);
leo_sobral@2
   648
  if ( ret == G_IO_STATUS_ERROR )
leo_sobral@2
   649
    g_error ("[%s] Error reading: %s\n", __FUNCTION__, err != NULL ? err->message : "" );
leo_sobral@2
   650
  g_print ("\n\n\n\n\n\n[%s]\t\tEVENT: Read %u bytes: %s\n\n\n\n\n", __FUNCTION__, len, msg != NULL ? msg : "" );
leo_sobral@2
   651
  if ( msg != NULL )
leo_sobral@2
   652
    g_free (msg);
leo_sobral@2
   653
leo_sobral@2
   654
  return TRUE;
leo_sobral@2
   655
leo_sobral@2
   656
}
leo_sobral@2
   657
leo_sobral@2
   658
static void*
leo_sobral@2
   659
myth_init_io_watchers( void *data )
leo_sobral@2
   660
{
leo_sobral@2
   661
  MythFileTransfer *transfer = (MythFileTransfer*)data;
leo_sobral@2
   662
  GMainContext *context = g_main_context_new();
leo_sobral@2
   663
  GMainLoop *loop = g_main_loop_new( NULL, FALSE );
leo_sobral@2
   664
leo_sobral@2
   665
  GSource *source = NULL;
leo_sobral@2
   666
leo_sobral@2
   667
  if ( transfer->event_sock->sd_io_ch != NULL )
leo_sobral@2
   668
    source = g_io_create_watch( transfer->event_sock->sd_io_ch, G_IO_IN | G_IO_HUP );
leo_sobral@2
   669
leo_sobral@2
   670
  g_source_set_callback ( source, (GSourceFunc)myth_control_sock_listener, NULL, NULL );
leo_sobral@2
   671
leo_sobral@2
   672
  g_source_attach( source, context );
leo_sobral@2
   673
leo_sobral@2
   674
  if (source==NULL)
leo_sobral@2
   675
    g_printerr( "[%s] Error adding watch listener function to the IO control channel!\n", __FUNCTION__ );
leo_sobral@2
   676
leo_sobral@2
   677
  g_main_loop_run( loop );
leo_sobral@2
   678
leo_sobral@2
   679
  g_source_unref( source );
leo_sobral@2
   680
leo_sobral@2
   681
  g_main_loop_unref( loop );
leo_sobral@2
   682
leo_sobral@2
   683
  g_main_context_unref( context );
leo_sobral@2
   684
leo_sobral@2
   685
  return NULL;
leo_sobral@2
   686
}
leo_sobral@2
   687
leo_sobral@2
   688
  gint 
leo_sobral@2
   689
myth_file_transfer_read(MythFileTransfer *transfer, void *data, gint size, gboolean read_unlimited)
leo_sobral@2
   690
{
leo_sobral@2
   691
  gint recv = 0;
leo_sobral@2
   692
  gsize bytes_read = 0;
leo_sobral@2
   693
leo_sobral@2
   694
  gint sent = 0;
leo_sobral@2
   695
  //guint zerocnt = 0;
leo_sobral@2
   696
  gboolean response = FALSE;
leo_sobral@2
   697
leo_sobral@2
   698
  GIOChannel *io_channel;
leo_sobral@2
   699
  GIOChannel *io_channel_control;
leo_sobral@2
   700
leo_sobral@2
   701
  GIOCondition io_cond;
leo_sobral@2
   702
  GIOCondition io_cond_control;
leo_sobral@2
   703
  GIOStatus io_status = G_IO_STATUS_NORMAL, io_status_control = G_IO_STATUS_NORMAL;   
leo_sobral@2
   704
leo_sobral@2
   705
  gint buf_len = MYTHTV_BUFFER_SIZE;
leo_sobral@2
   706
leo_sobral@2
   707
  GMythStringList *strlist = NULL;
leo_sobral@2
   708
  GError *error = NULL;
leo_sobral@2
   709
leo_sobral@2
   710
  gchar *trash = g_strdup("");
leo_sobral@2
   711
leo_sobral@2
   712
  g_return_val_if_fail ( data != NULL, -2 );
leo_sobral@2
   713
leo_sobral@2
   714
  /* gets the size of the entire file, if the size requested is lesser than 0 */
leo_sobral@2
   715
  if ( size <= 0 )
leo_sobral@2
   716
    size = transfer->filesize;
leo_sobral@2
   717
leo_sobral@2
   718
  io_channel = transfer->sock->sd_io_ch;
leo_sobral@2
   719
  io_channel_control = transfer->control_sock->sd_io_ch;
leo_sobral@2
   720
leo_sobral@2
   721
  //g_io_channel_set_flags( io_channel, G_IO_FLAG_APPEND | 
leo_sobral@2
   722
  //    G_IO_STATUS_AGAIN | G_IO_FLAG_IS_READABLE | G_IO_FLAG_IS_WRITEABLE | 
leo_sobral@2
   723
  //    G_IO_FLAG_IS_SEEKABLE, NULL );
leo_sobral@2
   724
leo_sobral@2
   725
  io_status = g_io_channel_set_encoding( io_channel, NULL, &error );
leo_sobral@2
   726
  if ( io_status == G_IO_STATUS_NORMAL )
leo_sobral@2
   727
    g_print( "[%s] Setting encoding to binary data socket).\n", __FUNCTION__ );
leo_sobral@2
   728
leo_sobral@2
   729
  io_cond = g_io_channel_get_buffer_condition( io_channel );  
leo_sobral@2
   730
leo_sobral@2
   731
  io_cond_control = g_io_channel_get_buffer_condition( io_channel );
leo_sobral@2
   732
leo_sobral@2
   733
  if ( transfer->sock == NULL || ( io_status == G_IO_STATUS_ERROR ) )
leo_sobral@2
   734
  {
leo_sobral@2
   735
    g_printerr( "myth_file_transfer_read(): Called with no raw socket.\n" );
leo_sobral@2
   736
    recv = -1;
leo_sobral@2
   737
    goto cleanup;
leo_sobral@2
   738
  }
leo_sobral@2
   739
leo_sobral@2
   740
  if ( transfer->control_sock == NULL || ( io_status_control == G_IO_STATUS_ERROR ) )
leo_sobral@2
   741
  {
leo_sobral@2
   742
    g_printerr( "myth_file_transfer_read(): Called with no control socket.\n" );
leo_sobral@2
   743
    recv = -1;
leo_sobral@2
   744
    goto cleanup;
leo_sobral@2
   745
  }
leo_sobral@2
   746
leo_sobral@2
   747
  /*
leo_sobral@2
   748
     if (!controlSock->isOpen() || controlSock->error())
leo_sobral@2
   749
     return -1;
leo_sobral@2
   750
     */
leo_sobral@2
   751
leo_sobral@2
   752
  if ( ( io_cond & G_IO_IN ) != 0 ) {
leo_sobral@2
   753
    do 
leo_sobral@2
   754
    {
leo_sobral@2
   755
leo_sobral@2
   756
      io_status = g_io_channel_read_line( io_channel, &trash, &bytes_read, NULL, &error);
leo_sobral@2
   757
leo_sobral@2
   758
      g_print( "[%s] cleaning buffer on IO binary channel... (%s)\n", __FUNCTION__, trash );
leo_sobral@2
   759
      io_cond = g_io_channel_get_buffer_condition( io_channel );
leo_sobral@2
   760
leo_sobral@2
   761
    } while ( ( io_cond & G_IO_IN ) != 0 && ( io_status != G_IO_STATUS_ERROR ) );
leo_sobral@2
   762
leo_sobral@2
   763
    if ( trash!= NULL )
leo_sobral@2
   764
      g_free( trash );
leo_sobral@2
   765
  }
leo_sobral@2
   766
leo_sobral@2
   767
  if ( ( io_cond_control & G_IO_IN ) != 0 ) {
leo_sobral@2
   768
    GMythStringList *strlist_tmp = gmyth_string_list_new();
leo_sobral@2
   769
    gmyth_socket_read_stringlist( transfer->control_sock, strlist_tmp );
leo_sobral@2
   770
    g_object_unref( strlist_tmp );
leo_sobral@2
   771
  }
leo_sobral@2
   772
leo_sobral@2
   773
  wait_to_transfer = 0;
leo_sobral@2
   774
leo_sobral@2
   775
  while ( transfer->live_tv && ( myth_file_transfer_get_file_position( transfer ) < 4096 ) &&
leo_sobral@2
   776
  		wait_to_transfer++ < MYTHTV_TRANSFER_MAX_WAITS )
leo_sobral@2
   777
  	g_usleep( 1000*50 ); /* waits just for 2/10 second */
leo_sobral@2
   778
leo_sobral@2
   779
  //g_thread_create( myth_init_io_watchers, (void*)transfer, FALSE, NULL );
leo_sobral@2
   780
  //g_printerr( "[%s] Watch listener function to the IO control channel on thread %p.\n", __FUNCTION__, g_thread_self() );
leo_sobral@2
   781
leo_sobral@2
   782
  g_static_mutex_lock (&mutex);
leo_sobral@2
   783
  strlist = gmyth_string_list_new();
leo_sobral@2
   784
leo_sobral@2
   785
  g_string_printf ( transfer->query, "%s %d", /*transfer->live_tv ? MYTHTV_RECORDER_HEADER :*/  MYTHTV_QUERY_HEADER, 
leo_sobral@2
   786
      /* transfer->live_tv ? transfer->card_id :*/ transfer->recordernum ); // transfer->recordernum
leo_sobral@2
   787
  g_print( "\t[%s] Transfer_query = %s\n", __FUNCTION__, transfer->query->str );
leo_sobral@2
   788
  strlist = gmyth_string_list_new();
leo_sobral@2
   789
leo_sobral@2
   790
  gmyth_string_list_append_string( strlist, transfer->query );
leo_sobral@2
   791
  gmyth_string_list_append_char_array( strlist, /*transfer->live_tv ? "REQUEST_BLOCK_RINGBUF" :*/ "REQUEST_BLOCK" );
leo_sobral@2
   792
  gmyth_string_list_append_int( strlist, size );
leo_sobral@2
   793
leo_sobral@2
   794
  gmyth_socket_write_stringlist( transfer->control_sock, strlist );
leo_sobral@2
   795
  sent = size;
leo_sobral@2
   796
leo_sobral@2
   797
  //data = (void*)g_new0( gchar, size );
leo_sobral@2
   798
leo_sobral@2
   799
  g_io_channel_flush( io_channel_control, NULL );
leo_sobral@2
   800
//  g_io_channel_flush( io_channel, NULL );
leo_sobral@2
   801
leo_sobral@2
   802
  io_cond = g_io_channel_get_buffer_condition( io_channel );
leo_sobral@2
   803
leo_sobral@2
   804
  while ( ( recv < sent ) )//&& ( io_cond & G_IO_IN ) != 0 )
leo_sobral@2
   805
  {
leo_sobral@2
   806
    do
leo_sobral@2
   807
    {
leo_sobral@2
   808
      //while ( ( io_cond & G_IO_IN ) == 0 ) {
leo_sobral@2
   809
      //usleep(200);
leo_sobral@2
   810
      //
leo_sobral@2
   811
      //io_cond = g_io_channel_get_buffer_condition( io_channel );
leo_sobral@2
   812
      //
leo_sobral@2
   813
leo_sobral@2
   814
      buf_len = ( sent - recv ) > MYTHTV_BUFFER_SIZE ? MYTHTV_BUFFER_SIZE : ( sent - recv );
leo_sobral@2
   815
leo_sobral@2
   816
      io_status = g_io_channel_read_chars( io_channel, data + recv, 
leo_sobral@2
   817
	  buf_len, &bytes_read, &error );
leo_sobral@2
   818
      /*
leo_sobral@2
   819
	 GString *sss = g_string_new("");
leo_sobral@2
   820
	 sss = g_string_append_len( sss, (gchar*)data+recv, bytes_read );
leo_sobral@2
   821
leo_sobral@2
   822
	 g_print( "[%s] Reading buffer (length = %d)\n", __FUNCTION__, bytes_read);
leo_sobral@2
   823
	 */
leo_sobral@2
   824
      if ( bytes_read > 0 )
leo_sobral@2
   825
      {
leo_sobral@2
   826
	if ( bytes_read <= buf_len )
leo_sobral@2
   827
	  recv += bytes_read;
leo_sobral@2
   828
      } 
leo_sobral@2
   829
leo_sobral@2
   830
      if ( io_status == G_IO_STATUS_EOF ) {
leo_sobral@2
   831
	g_printerr( "[%s] got EOS!", __FUNCTION__ );
leo_sobral@2
   832
	break;
leo_sobral@2
   833
      } else if ( io_status == G_IO_STATUS_ERROR ) {
leo_sobral@2
   834
	g_printerr( "[%s] myth_file_transfer_read(): socket error.\n", __FUNCTION__ );
leo_sobral@2
   835
	break;
leo_sobral@2
   836
      }
leo_sobral@2
   837
leo_sobral@2
   838
      /* increase buffer size, to allow get more data (do not obey to the buffer size) */
leo_sobral@2
   839
      if ( read_unlimited == TRUE ) {
leo_sobral@2
   840
	//if ( recv > buf_len )
leo_sobral@2
   841
	// sent += (bytes_read - buf_len) + 1;
leo_sobral@2
   842
      }
leo_sobral@2
   843
      if ( bytes_read == buf_len )
leo_sobral@2
   844
	break;
leo_sobral@2
   845
leo_sobral@2
   846
      /* verify if the input (read) buffer is ready to receive data */
leo_sobral@2
   847
      io_cond = g_io_channel_get_buffer_condition( io_channel );
leo_sobral@2
   848
leo_sobral@2
   849
      g_print( "[%s]\t io_cond %s prepared for reading! (G_IO_IN) !!!\n\n", __FUNCTION__, 
leo_sobral@2
   850
      		( ( io_cond & G_IO_IN ) != 0 ) ? "IS" : "IS NOT" );
leo_sobral@2
   851
leo_sobral@2
   852
    } while ( recv < sent && ( ( io_cond & G_IO_IN ) != 0 ) && ( io_status == G_IO_STATUS_NORMAL ) );
leo_sobral@2
   853
leo_sobral@2
   854
    io_cond_control = g_io_channel_get_buffer_condition( io_channel_control );
leo_sobral@2
   855
    if ( ( io_status == G_IO_STATUS_EOF ) || ( ( io_cond_control & G_IO_IN ) != 0 ) )
leo_sobral@2
   856
    {
leo_sobral@2
   857
      gmyth_socket_read_stringlist( transfer->control_sock, strlist );
leo_sobral@2
   858
      sent = gmyth_string_list_get_int( strlist,  0 ); // -1 on backend error      
leo_sobral@2
   859
      g_print( "[%s]\t sent = %d, io_cond %s prepared for reading! (G_IO_IN) !!!\n\n", __FUNCTION__, 
leo_sobral@2
   860
	  sent, ( ( io_cond & G_IO_IN ) != 0 ) ? "IS" : "IS NOT" );
leo_sobral@2
   861
      response = TRUE;
leo_sobral@2
   862
    }
leo_sobral@2
   863
  }
leo_sobral@2
   864
leo_sobral@2
   865
  if ( ( ( error == NULL ) && ( response == FALSE ) ) || 
leo_sobral@2
   866
  		( io_status == G_IO_STATUS_EOF ) || ( ( io_cond & G_IO_IN ) == 0 ) )
leo_sobral@2
   867
  {
leo_sobral@2
   868
    if ( gmyth_socket_read_stringlist( transfer->control_sock, strlist ) > 0 )
leo_sobral@2
   869
    {
leo_sobral@2
   870
      if ( strlist != NULL && gmyth_string_list_length(strlist) > 0 ) {
leo_sobral@2
   871
	sent = gmyth_string_list_get_int( strlist, 0 ); // -1 on backend error
leo_sobral@2
   872
	g_print( "[%s]\t sent = %d, io_cond %s prepared for reading! (G_IO_IN) !!!\n\n", __FUNCTION__, 
leo_sobral@2
   873
	    sent, ( ( io_cond & G_IO_IN ) != 0 ) ? "IS" : "IS NOT" );
leo_sobral@2
   874
      }
leo_sobral@2
   875
    }
leo_sobral@2
   876
    else
leo_sobral@2
   877
    {
leo_sobral@2
   878
      g_printerr ( "myth_file_transfer_read(): No response from control socket.");
leo_sobral@2
   879
      sent = -1;
leo_sobral@2
   880
    }
leo_sobral@2
   881
leo_sobral@2
   882
    if ( error!=NULL ) {
leo_sobral@2
   883
      g_printerr( "[%s] Error occurred: (%d, %s)\n", __FUNCTION__, error->code, error->message );
leo_sobral@2
   884
      g_error_free( error );
leo_sobral@2
   885
    }
leo_sobral@2
   886
  }
leo_sobral@2
   887
leo_sobral@2
   888
cleanup:
leo_sobral@2
   889
leo_sobral@2
   890
  if ( trash != NULL )
leo_sobral@2
   891
    g_free( trash );
leo_sobral@2
   892
leo_sobral@2
   893
  if ( strlist != NULL )
leo_sobral@2
   894
    g_object_unref( strlist );
leo_sobral@2
   895
leo_sobral@2
   896
  g_static_mutex_unlock (&mutex);
leo_sobral@2
   897
  g_print( "myth_file_transfer_read(): reqd=%d, rcvd=%d, rept=%d, "\
leo_sobral@2
   898
      "(rcvd and rept MUST be the same!)\n", size, 
leo_sobral@2
   899
      recv, sent );
leo_sobral@2
   900
leo_sobral@2
   901
  //if ( sent != recv ) {
leo_sobral@2
   902
  //  recv = -1;
leo_sobral@2
   903
  //}
leo_sobral@2
   904
leo_sobral@2
   905
  if ( error != NULL )
leo_sobral@2
   906
    g_printerr( "ERROR: %s [msg = %s, code = %d]\n", __FUNCTION__, error->message, 
leo_sobral@2
   907
	error->code );
leo_sobral@2
   908
leo_sobral@2
   909
  return recv;
leo_sobral@2
   910
}
leo_sobral@2
   911
leo_sobral@2
   912
  void 
leo_sobral@2
   913
myth_file_transfer_settimeout( MythFileTransfer *transfer, gboolean fast )
leo_sobral@2
   914
{
leo_sobral@2
   915
leo_sobral@2
   916
  GMythStringList *strlist = NULL;
leo_sobral@2
   917
leo_sobral@2
   918
  if ( transfer->timeoutisfast == fast )
leo_sobral@2
   919
    return;
leo_sobral@2
   920
leo_sobral@2
   921
  if ( transfer->sock == NULL )
leo_sobral@2
   922
  {
leo_sobral@2
   923
    g_printerr( "myth_file_transfer_settimeout(): Called with no socket" );
leo_sobral@2
   924
    return;
leo_sobral@2
   925
  }
leo_sobral@2
   926
leo_sobral@2
   927
  if ( transfer->control_sock == NULL )
leo_sobral@2
   928
    return;
leo_sobral@2
   929
leo_sobral@2
   930
  strlist = gmyth_string_list_new(); 
leo_sobral@2
   931
  gmyth_string_list_append_string( strlist, transfer->query );
leo_sobral@2
   932
  gmyth_string_list_append_char_array( strlist, "SET_TIMEOUT" );
leo_sobral@2
   933
  gmyth_string_list_append_int( strlist, fast ); 
leo_sobral@2
   934
leo_sobral@2
   935
  gmyth_socket_write_stringlist( transfer->control_sock, strlist );
leo_sobral@2
   936
  gmyth_socket_read_stringlist( transfer->control_sock, strlist );
leo_sobral@2
   937
leo_sobral@2
   938
  transfer->timeoutisfast = fast;
leo_sobral@2
   939
leo_sobral@2
   940
}
leo_sobral@2
   941
leo_sobral@2
   942
#ifdef DO_TESTING
leo_sobral@2
   943
leo_sobral@2
   944
  int
leo_sobral@2
   945
main( int argc, char *argv[] )
leo_sobral@2
   946
{
leo_sobral@2
   947
  g_type_init();
leo_sobral@2
   948
leo_sobral@2
   949
  MythFileTransfer *file_transfer = myth_file_transfer_new( 1, 
leo_sobral@2
   950
      g_string_new("myth://192.168.1.109:6543/jshks.nuv"), -1, MYTHTV_VERSION );
leo_sobral@2
   951
  myth_file_transfer_setup( &file_transfer );
leo_sobral@2
   952
  gchar *data = g_strdup("");
leo_sobral@2
   953
leo_sobral@2
   954
  gint num = myth_file_transfer_read( file_transfer, data, -1 );
leo_sobral@2
   955
leo_sobral@2
   956
  return 0;
leo_sobral@2
   957
leo_sobral@2
   958
}
leo_sobral@2
   959
leo_sobral@2
   960
#endif