research.m1stereo.tv/gitweb/ - melted/blob - src/miracle/miracle_server.c

   1 /*
   2  * miracle_server.c -- DV Server
   3  * Copyright (C) 2002-2003 Ushodaya Enterprises Limited
   4  * Author: Charles Yates <charles.yates@pandora.be>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program 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
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * 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
  27 #include <fcntl.h>
  28 #include <pthread.h>
  29 #include <signal.h>
  30 #include <stdlib.h>
  31 #include <unistd.h>
  32
  33 #include <string.h>
  34 #include <netinet/in.h>
  35 #include <netdb.h>
  36 #include <errno.h>
  37 #include <arpa/inet.h>
  38
  39 /* Application header files */
  40 #include "miracle_server.h"
  41 #include "miracle_connection.h"
  42 #include "miracle_local.h"
  43 #include "miracle_log.h"
  44 #include <valerie/valerie_remote.h>
  45 #include <valerie/valerie_tokeniser.h>
  46
  47 #define VERSION "0.0.1"
  48
  49 /** Initialise a server structure.
  50 */
  51
  52 miracle_server miracle_server_init( char *id )
  53 {
  54         miracle_server server = malloc( sizeof( miracle_server_t ) );
  55         if ( server != NULL )
  56         {
  57                 memset( server, 0, sizeof( miracle_server_t ) );
  58                 server->id = id;
  59                 server->port = DEFAULT_TCP_PORT;
  60                 server->socket = -1;
  61         }
  62         return server;
  63 }
  64
  65 void miracle_server_set_config( miracle_server server, char *config )
  66 {
  67         if ( server != NULL )
  68         {
  69                 free( server->config );
  70                 server->config = config != NULL ? strdup( config ) : NULL;
  71         }
  72 }
  73
  74 /** Set the port of the server.
  75 */
  76
  77 void miracle_server_set_port( miracle_server server, int port )
  78 {
  79         server->port = port;
  80 }
  81
  82 void miracle_server_set_proxy( miracle_server server, char *proxy )
  83 {
  84         valerie_tokeniser tokeniser = valerie_tokeniser_init( );
  85         server->proxy = 1;
  86         server->remote_port = DEFAULT_TCP_PORT;
  87         valerie_tokeniser_parse_new( tokeniser, proxy, ":" );
  88         strcpy( server->remote_server, valerie_tokeniser_get_string( tokeniser, 0 ) );
  89         if ( valerie_tokeniser_count( tokeniser ) == 2 )
  90                 server->remote_port = atoi( valerie_tokeniser_get_string( tokeniser, 1 ) );
  91         valerie_tokeniser_close( tokeniser );
  92 }
  93
  94 /** Wait for a connection.
  95 */
  96
  97 static int miracle_server_wait_for_connect( miracle_server server )
  98 {
  99     struct timeval tv;
 100     fd_set rfds;
 101
 102     /* Wait for a 1 second. */
 103     tv.tv_sec = 1;
 104     tv.tv_usec = 0;
 105
 106     FD_ZERO( &rfds );
 107     FD_SET( server->socket, &rfds );
 108
 109     return select( server->socket + 1, &rfds, NULL, NULL, &tv);
 110 }
 111
 112 /** Run the server thread.
 113 */
 114
 115 static void *miracle_server_run( void *arg )
 116 {
 117         miracle_server server = arg;
 118         pthread_t cmd_parse_info;
 119         connection_t *tmp = NULL;
 120         pthread_attr_t thread_attributes;
 121         int socksize;
 122
 123         socksize = sizeof( struct sockaddr );
 124
 125         miracle_log( LOG_NOTICE, "%s version %s listening on port %i", server->id, VERSION, server->port );
 126
 127         /* Create the initial thread. We want all threads to be created detached so
 128            their resources get freed automatically. (CY: ... hmmph...) */
 129         pthread_attr_init( &thread_attributes );
 130         pthread_attr_setdetachstate( &thread_attributes, PTHREAD_CREATE_DETACHED );
 131         pthread_attr_init( &thread_attributes );
 132         pthread_attr_setinheritsched( &thread_attributes, PTHREAD_INHERIT_SCHED );
 133         /* pthread_attr_setschedpolicy( &thread_attributes, SCHED_RR ); */
 134
 135         while ( !server->shutdown )
 136         {
 137                 /* Wait for a new connection. */
 138                 if ( miracle_server_wait_for_connect( server ) )
 139                 {
 140                         /* Create a new block of data to hold a copy of the incoming connection for
 141                            our server thread. The thread should free this when it terminates. */
 142
 143                         tmp = (connection_t*) malloc( sizeof(connection_t) );
 144                         tmp->parser = server->parser;
 145                         tmp->fd = accept( server->socket, (struct sockaddr*) &(tmp->sin), &socksize );
 146
 147                         /* Pass the connection to a parser thread :-/ */
 148                         if ( tmp->fd != -1 )
 149                                 pthread_create( &cmd_parse_info, &thread_attributes, parser_thread, tmp );
 150                 }
 151         }
 152
 153         miracle_log( LOG_NOTICE, "%s version %s server terminated.", server->id, VERSION );
 154
 155         return NULL;
 156 }
 157
 158 /** Execute the server thread.
 159 */
 160
 161 int miracle_server_execute( miracle_server server )
 162 {
 163         int error = 0;
 164         valerie_response response = NULL;
 165         int index = 0;
 166         struct sockaddr_in ServerAddr;
 167         int flag = 1;
 168
 169         ServerAddr.sin_family = AF_INET;
 170         ServerAddr.sin_port = htons( server->port );
 171         ServerAddr.sin_addr.s_addr = INADDR_ANY;
 172
 173         /* Create socket, and bind to port. Listen there. Backlog = 5
 174            should be sufficient for listen (). */
 175         server->socket = socket( AF_INET, SOCK_STREAM, 0 );
 176
 177         if ( server->socket == -1 )
 178         {
 179                 server->shutdown = 1;
 180                 perror( "socket" );
 181                 miracle_log( LOG_ERR, "%s unable to create socket.", server->id );
 182                 return -1;
 183         }
 184
 185     setsockopt( server->socket, SOL_SOCKET, SO_REUSEADDR, (char *)&flag, sizeof( int ) );
 186
 187         if ( bind( server->socket, (struct sockaddr *) &ServerAddr, sizeof (ServerAddr) ) != 0 )
 188         {
 189                 server->shutdown = 1;
 190                 perror( "bind" );
 191                 miracle_log( LOG_ERR, "%s unable to bind to port %d.", server->id, server->port );
 192                 return -1;
 193         }
 194
 195         if ( listen( server->socket, 5 ) != 0 )
 196         {
 197                 server->shutdown = 1;
 198                 perror( "listen" );
 199                 miracle_log( LOG_ERR, "%s unable to listen on port %d.", server->id, server->port );
 200                 return -1;
 201         }
 202
 203         fcntl( server->socket, F_SETFL, O_NONBLOCK );
 204
 205         if ( !server->proxy )
 206         {
 207                 miracle_log( LOG_NOTICE, "Starting server on %d.", server->port );
 208                 server->parser = miracle_parser_init_local( );
 209         }
 210         else
 211         {
 212                 miracle_log( LOG_NOTICE, "Starting proxy for %s:%d on %d.", server->remote_server, server->remote_port, server->port );
 213                 server->parser = valerie_parser_init_remote( server->remote_server, server->remote_port );
 214         }
 215
 216         response = valerie_parser_connect( server->parser );
 217
 218         if ( response != NULL && valerie_response_get_error_code( response ) == 100 )
 219         {
 220                 /* read configuration file */
 221                 if ( response != NULL && !server->proxy && server->config != NULL )
 222                 {
 223                         valerie_response_close( response );
 224                         response = valerie_parser_run( server->parser, server->config );
 225
 226                         if ( valerie_response_count( response ) > 1 )
 227                         {
 228                                 if ( valerie_response_get_error_code( response ) > 299 )
 229                                         miracle_log( LOG_ERR, "Error evaluating server configuration. Processing stopped." );
 230                                 for ( index = 0; index < valerie_response_count( response ); index ++ )
 231                                         miracle_log( LOG_DEBUG, "%4d: %s", index, valerie_response_get_line( response, index ) );
 232                         }
 233                 }
 234
 235                 if ( response != NULL )
 236                 {
 237                         pthread_attr_t attr;
 238                         int result;
 239                         pthread_attr_init( &attr );
 240                         pthread_attr_setdetachstate( &attr, PTHREAD_CREATE_JOINABLE );
 241                         pthread_attr_setinheritsched( &attr, PTHREAD_EXPLICIT_SCHED );
 242                         pthread_attr_setschedpolicy( &attr, SCHED_FIFO );
 243                         pthread_attr_setscope( &attr, PTHREAD_SCOPE_SYSTEM );
 244                         valerie_response_close( response );
 245                         result = pthread_create( &server->thread, &attr, miracle_server_run, server );
 246                         if ( result )
 247                         {
 248                                 miracle_log( LOG_WARNING, "Failed to schedule realtime (%s)", strerror(errno) );
 249                                 pthread_attr_setschedpolicy( &attr, SCHED_OTHER );
 250                                 result = pthread_create( &server->thread, &attr, miracle_server_run, server );
 251                                 if ( result )
 252                                 {
 253                                         miracle_log( LOG_CRIT, "Failed to launch TCP listener thread" );
 254                                         error = -1;
 255                                 }
 256                         }
 257                 }
 258         }
 259         else
 260         {
 261                 miracle_log( LOG_ERR, "Error connecting to parser. Processing stopped." );
 262                 server->shutdown = 1;
 263                 error = -1;
 264         }
 265
 266         return error;
 267 }
 268
 269 /** Shutdown the server.
 270 */
 271
 272 void miracle_server_shutdown( miracle_server server )
 273 {
 274         if ( server != NULL && !server->shutdown )
 275         {
 276                 server->shutdown = 1;
 277                 pthread_join( server->thread, NULL );
 278                 miracle_server_set_config( server, NULL );
 279                 valerie_parser_close( server->parser );
 280                 server->parser = NULL;
 281                 close( server->socket );
 282         }
 283 }
 284
 285 /** Close the server.
 286 */
 287
 288 void miracle_server_close( miracle_server server )
 289 {
 290         if ( server != NULL )
 291         {
 292                 miracle_server_shutdown( server );
 293                 free( server );
 294         }
 295 }