git.sesse.net Git - mlt/blob - mlt/src/valerie/valerie_remote.c

   1 /*
   2  * valerie_remote.c -- Remote Parser
   3  * Copyright (C) 2002-2003 Ushodaya Enterprises Limited
   4  * Author: Charles Yates <charles.yates@pandora.be>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 /* System header files */
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <signal.h>
  26 #include <errno.h>
  27 #include <pthread.h>
  28
  29 /* Application header files */
  30 #include "valerie_remote.h"
  31 #include "valerie_socket.h"
  32 #include "valerie_tokeniser.h"
  33 #include "valerie_util.h"
  34
  35 /** Private valerie_remote structure.
  36 */
  37
  38 typedef struct
  39 {
  40         int terminated;
  41         char *server;
  42         int port;
  43         valerie_socket socket;
  44         valerie_socket status;
  45         pthread_t thread;
  46         valerie_parser parser;
  47         pthread_mutex_t mutex;
  48         int connected;
  49 }
  50 *valerie_remote, valerie_remote_t;
  51
  52 /** Forward declarations.
  53 */
  54
  55 static valerie_response valerie_remote_connect( valerie_remote );
  56 static valerie_response valerie_remote_execute( valerie_remote, char * );
  57 static void valerie_remote_close( valerie_remote );
  58 static int valerie_remote_read_response( valerie_socket, valerie_response );
  59
  60 /** DV Parser constructor.
  61 */
  62
  63 valerie_parser valerie_parser_init_remote( char *server, int port )
  64 {
  65         valerie_parser parser = malloc( sizeof( valerie_parser_t ) );
  66         valerie_remote remote = malloc( sizeof( valerie_remote_t ) );
  67
  68         if ( parser != NULL )
  69         {
  70                 memset( parser, 0, sizeof( valerie_parser_t ) );
  71
  72                 parser->connect = (parser_connect)valerie_remote_connect;
  73                 parser->execute = (parser_execute)valerie_remote_execute;
  74                 parser->close = (parser_close)valerie_remote_close;
  75                 parser->real = remote;
  76
  77                 if ( remote != NULL )
  78                 {
  79                         memset( remote, 0, sizeof( valerie_remote_t ) );
  80                         remote->parser = parser;
  81                         remote->server = strdup( server );
  82                         remote->port = port;
  83                         pthread_mutex_init( &remote->mutex, NULL );
  84                 }
  85         }
  86         return parser;
  87 }
  88
  89 /** Thread for receiving and distributing the status information.
  90 */
  91
  92 static void *valerie_remote_status_thread( void *arg )
  93 {
  94         valerie_remote remote = arg;
  95         char temp[ 10240 ];
  96         int length = 0;
  97         int offset = 0;
  98         valerie_tokeniser tokeniser = valerie_tokeniser_init( );
  99         valerie_notifier notifier = valerie_parser_get_notifier( remote->parser );
 100         valerie_status_t status;
 101         int index = 0;
 102
 103         valerie_socket_write_data( remote->status, "STATUS\r\n", 8 );
 104
 105         while ( !remote->terminated &&
 106                         ( length = valerie_socket_read_data( remote->status, temp + offset, sizeof( temp ) ) ) >= 0 )
 107         {
 108                 if ( strchr( temp, '\n' ) == NULL )
 109                 {
 110                         offset = length;
 111                         continue;
 112                 }
 113                 offset = 0;
 114                 valerie_tokeniser_parse_new( tokeniser, temp, "\n" );
 115                 for ( index = 0; index < valerie_tokeniser_count( tokeniser ); index ++ )
 116                 {
 117                         char *line = valerie_tokeniser_get_string( tokeniser, index );
 118                         if ( line[ strlen( line ) - 1 ] == '\r' )
 119                         {
 120                                 valerie_util_chomp( line );
 121                                 valerie_status_parse( &status, line );
 122                                 valerie_notifier_put( notifier, &status );
 123                         }
 124                         else
 125                         {
 126                                 strcpy( temp, line );
 127                                 offset = strlen( temp );
 128                         }
 129                 }
 130         }
 131
 132         valerie_notifier_disconnected( notifier );
 133         valerie_tokeniser_close( tokeniser );
 134         remote->terminated = 1;
 135
 136         return NULL;
 137 }
 138
 139 /** Forward reference.
 140 */
 141
 142 static void valerie_remote_disconnect( valerie_remote remote );
 143
 144 /** Connect to the server.
 145 */
 146
 147 static valerie_response valerie_remote_connect( valerie_remote remote )
 148 {
 149         valerie_response response = NULL;
 150
 151         valerie_remote_disconnect( remote );
 152
 153         if ( !remote->connected )
 154         {
 155                 signal( SIGPIPE, SIG_IGN );
 156
 157                 remote->socket = valerie_socket_init( remote->server, remote->port );
 158                 remote->status = valerie_socket_init( remote->server, remote->port );
 159
 160                 if ( valerie_socket_connect( remote->socket ) == 0 )
 161                 {
 162                         response = valerie_response_init( );
 163                         valerie_remote_read_response( remote->socket, response );
 164                 }
 165
 166                 if ( response != NULL && valerie_socket_connect( remote->status ) == 0 )
 167                 {
 168                         valerie_response status_response = valerie_response_init( );
 169                         valerie_remote_read_response( remote->status, status_response );
 170                         if ( valerie_response_get_error_code( status_response ) == 100 )
 171                                 pthread_create( &remote->thread, NULL, valerie_remote_status_thread, remote );
 172                         valerie_response_close( status_response );
 173                         remote->connected = 1;
 174                 }
 175         }
 176
 177         return response;
 178 }
 179
 180 /** Execute the command.
 181 */
 182
 183 static valerie_response valerie_remote_execute( valerie_remote remote, char *command )
 184 {
 185         valerie_response response = NULL;
 186         pthread_mutex_lock( &remote->mutex );
 187         if ( valerie_socket_write_data( remote->socket, command, strlen( command ) ) == strlen( command ) )
 188         {
 189                 response = valerie_response_init( );
 190                 valerie_socket_write_data( remote->socket, "\r\n", 2 );
 191                 valerie_remote_read_response( remote->socket, response );
 192         }
 193         pthread_mutex_unlock( &remote->mutex );
 194         return response;
 195 }
 196
 197 /** Disconnect.
 198 */
 199
 200 static void valerie_remote_disconnect( valerie_remote remote )
 201 {
 202         if ( remote != NULL && remote->terminated )
 203         {
 204                 pthread_join( remote->thread, NULL );
 205                 valerie_socket_close( remote->status );
 206                 valerie_socket_close( remote->socket );
 207                 remote->connected = 0;
 208                 remote->terminated = 0;
 209         }
 210 }
 211
 212 /** Close the parser.
 213 */
 214
 215 static void valerie_remote_close( valerie_remote remote )
 216 {
 217         if ( remote != NULL )
 218         {
 219                 remote->terminated = 1;
 220                 valerie_remote_disconnect( remote );
 221                 pthread_mutex_destroy( &remote->mutex );
 222                 free( remote->server );
 223                 free( remote );
 224         }
 225 }
 226
 227 /** Read response.
 228 */
 229
 230 static int valerie_remote_read_response( valerie_socket socket, valerie_response response )
 231 {
 232         char temp[ 10240 ];
 233         int length;
 234         int terminated = 0;
 235
 236         while ( !terminated && ( length = valerie_socket_read_data( socket, temp, 10240 ) ) >= 0 )
 237         {
 238                 int position = 0;
 239                 temp[ length ] = '\0';
 240                 valerie_response_write( response, temp, length );
 241                 position = valerie_response_count( response ) - 1;
 242                 if ( position < 0 || temp[ strlen( temp ) - 1 ] != '\n' )
 243                         continue;
 244                 switch( valerie_response_get_error_code( response ) )
 245                 {
 246                         case 201:
 247                         case 500:
 248                                 terminated = !strcmp( valerie_response_get_line( response, position ), "" );
 249                                 break;
 250                         case 202:
 251                                 terminated = valerie_response_count( response ) >= 2;
 252                                 break;
 253                         default:
 254                                 terminated = 1;
 255                                 break;
 256                 }
 257         }
 258
 259         return 0;
 260 }