2 * miracle_server.c -- DV Server
3 * Copyright (C) 2002-2003 Ushodaya Enterprises Limited
4 * Author: Charles Yates <charles.yates@pandora.be>
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.
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.
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.
21 /* System header files */
34 #include <netinet/in.h>
37 #include <arpa/inet.h>
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>
47 #define VERSION "0.0.1"
49 static void miracle_command_received( mlt_listener listener, mlt_properties owner, miracle_server this, void **args )
51 if ( listener != NULL )
52 listener( owner, this, ( valerie_response ** )args[ 0 ], ( char * )args[ 1 ] );
55 static void miracle_doc_received( mlt_listener listener, mlt_properties owner, miracle_server this, void **args )
57 if ( listener != NULL )
58 listener( owner, this, ( valerie_response ** )args[ 0 ], ( char * )args[ 1 ], ( char * )args[ 2 ] );
61 static void miracle_push_received( mlt_listener listener, mlt_properties owner, miracle_server this, void **args )
63 if ( listener != NULL )
64 listener( owner, this, ( valerie_response ** )args[ 0 ], ( char * )args[ 1 ], ( mlt_service )args[ 2 ] );
67 /** Initialise a server structure.
70 miracle_server miracle_server_init( char *id )
72 miracle_server server = malloc( sizeof( miracle_server_t ) );
74 memset( server, 0, sizeof( miracle_server_t ) );
75 if ( server != NULL && mlt_properties_init( &server->parent, server ) == 0 )
78 server->port = DEFAULT_TCP_PORT;
81 mlt_events_init( &server->parent );
82 mlt_events_register( &server->parent, "command-received", ( mlt_transmitter )miracle_command_received );
83 mlt_events_register( &server->parent, "doc-received", ( mlt_transmitter )miracle_doc_received );
84 mlt_events_register( &server->parent, "push-received", ( mlt_transmitter )miracle_push_received );
89 void miracle_server_set_config( miracle_server server, char *config )
93 free( server->config );
94 server->config = config != NULL ? strdup( config ) : NULL;
98 /** Set the port of the server.
101 void miracle_server_set_port( miracle_server server, int port )
106 void miracle_server_set_proxy( miracle_server server, char *proxy )
108 valerie_tokeniser tokeniser = valerie_tokeniser_init( );
110 server->remote_port = DEFAULT_TCP_PORT;
111 valerie_tokeniser_parse_new( tokeniser, proxy, ":" );
112 strcpy( server->remote_server, valerie_tokeniser_get_string( tokeniser, 0 ) );
113 if ( valerie_tokeniser_count( tokeniser ) == 2 )
114 server->remote_port = atoi( valerie_tokeniser_get_string( tokeniser, 1 ) );
115 valerie_tokeniser_close( tokeniser );
118 /** Wait for a connection.
121 static int miracle_server_wait_for_connect( miracle_server server )
126 /* Wait for a 1 second. */
131 FD_SET( server->socket, &rfds );
133 return select( server->socket + 1, &rfds, NULL, NULL, &tv);
136 /** Run the server thread.
139 static void *miracle_server_run( void *arg )
141 miracle_server server = arg;
142 pthread_t cmd_parse_info;
143 connection_t *tmp = NULL;
144 pthread_attr_t thread_attributes;
147 socksize = sizeof( struct sockaddr );
149 miracle_log( LOG_NOTICE, "%s version %s listening on port %i", server->id, VERSION, server->port );
151 /* Create the initial thread. We want all threads to be created detached so
152 their resources get freed automatically. (CY: ... hmmph...) */
153 pthread_attr_init( &thread_attributes );
154 pthread_attr_setdetachstate( &thread_attributes, PTHREAD_CREATE_DETACHED );
155 pthread_attr_init( &thread_attributes );
156 pthread_attr_setinheritsched( &thread_attributes, PTHREAD_INHERIT_SCHED );
157 /* pthread_attr_setschedpolicy( &thread_attributes, SCHED_RR ); */
159 while ( !server->shutdown )
161 /* Wait for a new connection. */
162 if ( miracle_server_wait_for_connect( server ) )
164 /* Create a new block of data to hold a copy of the incoming connection for
165 our server thread. The thread should free this when it terminates. */
167 tmp = (connection_t*) malloc( sizeof(connection_t) );
168 tmp->owner = &server->parent;
169 tmp->parser = server->parser;
170 tmp->fd = accept( server->socket, (struct sockaddr*) &(tmp->sin), &socksize );
172 /* Pass the connection to a parser thread :-/ */
174 pthread_create( &cmd_parse_info, &thread_attributes, parser_thread, tmp );
178 miracle_log( LOG_NOTICE, "%s version %s server terminated.", server->id, VERSION );
183 /** Execute the server thread.
186 int miracle_server_execute( miracle_server server )
189 valerie_response response = NULL;
191 struct sockaddr_in ServerAddr;
194 server->shutdown = 0;
196 ServerAddr.sin_family = AF_INET;
197 ServerAddr.sin_port = htons( server->port );
198 ServerAddr.sin_addr.s_addr = INADDR_ANY;
200 /* Create socket, and bind to port. Listen there. Backlog = 5
201 should be sufficient for listen (). */
202 server->socket = socket( AF_INET, SOCK_STREAM, 0 );
204 if ( server->socket == -1 )
206 server->shutdown = 1;
208 miracle_log( LOG_ERR, "%s unable to create socket.", server->id );
212 setsockopt( server->socket, SOL_SOCKET, SO_REUSEADDR, (char *)&flag, sizeof( int ) );
214 if ( bind( server->socket, (struct sockaddr *) &ServerAddr, sizeof (ServerAddr) ) != 0 )
216 server->shutdown = 1;
218 miracle_log( LOG_ERR, "%s unable to bind to port %d.", server->id, server->port );
222 if ( listen( server->socket, 5 ) != 0 )
224 server->shutdown = 1;
226 miracle_log( LOG_ERR, "%s unable to listen on port %d.", server->id, server->port );
230 fcntl( server->socket, F_SETFL, O_NONBLOCK );
232 if ( !server->proxy )
234 miracle_log( LOG_NOTICE, "Starting server on %d.", server->port );
235 server->parser = miracle_parser_init_local( );
239 miracle_log( LOG_NOTICE, "Starting proxy for %s:%d on %d.", server->remote_server, server->remote_port, server->port );
240 server->parser = valerie_parser_init_remote( server->remote_server, server->remote_port );
243 response = valerie_parser_connect( server->parser );
245 if ( response != NULL && valerie_response_get_error_code( response ) == 100 )
247 /* read configuration file */
248 if ( response != NULL && !server->proxy && server->config != NULL )
250 valerie_response_close( response );
251 response = valerie_parser_run( server->parser, server->config );
253 if ( valerie_response_count( response ) > 1 )
255 if ( valerie_response_get_error_code( response ) > 299 )
256 miracle_log( LOG_ERR, "Error evaluating server configuration. Processing stopped." );
257 for ( index = 0; index < valerie_response_count( response ); index ++ )
258 miracle_log( LOG_DEBUG, "%4d: %s", index, valerie_response_get_line( response, index ) );
262 if ( response != NULL )
266 pthread_attr_init( &attr );
267 pthread_attr_setdetachstate( &attr, PTHREAD_CREATE_JOINABLE );
268 pthread_attr_setinheritsched( &attr, PTHREAD_EXPLICIT_SCHED );
269 pthread_attr_setschedpolicy( &attr, SCHED_FIFO );
270 pthread_attr_setscope( &attr, PTHREAD_SCOPE_SYSTEM );
271 valerie_response_close( response );
272 result = pthread_create( &server->thread, &attr, miracle_server_run, server );
275 miracle_log( LOG_WARNING, "Failed to schedule realtime (%s)", strerror(errno) );
276 pthread_attr_setschedpolicy( &attr, SCHED_OTHER );
277 result = pthread_create( &server->thread, &attr, miracle_server_run, server );
280 miracle_log( LOG_CRIT, "Failed to launch TCP listener thread" );
288 miracle_log( LOG_ERR, "Error connecting to parser. Processing stopped." );
289 server->shutdown = 1;
296 /** Shutdown the server.
299 void miracle_server_shutdown( miracle_server server )
301 if ( server != NULL && !server->shutdown )
303 server->shutdown = 1;
304 pthread_join( server->thread, NULL );
305 miracle_server_set_config( server, NULL );
306 valerie_parser_close( server->parser );
307 server->parser = NULL;
308 close( server->socket );
312 /** Close the server.
315 void miracle_server_close( miracle_server server )
317 if ( server != NULL && mlt_properties_dec_ref( &server->parent ) <= 0 )
319 mlt_properties_close( &server->parent );
320 miracle_server_shutdown( server );