3 * Copyright (C) 2002-20039 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 "melted_server.h"
41 #include "melted_connection.h"
42 #include "melted_local.h"
43 #include "melted_log.h"
44 #include "melted_commands.h"
45 #include <mvcp/mvcp_remote.h>
46 #include <mvcp/mvcp_tokeniser.h>
48 static void melted_command_received( mlt_listener listener
, mlt_properties owner
, melted_server
this, void **args
)
50 if ( listener
!= NULL
)
51 listener( owner
, this, ( mvcp_response
** )args
[ 0 ], ( char * )args
[ 1 ] );
54 static void melted_doc_received( mlt_listener listener
, mlt_properties owner
, melted_server
this, void **args
)
56 if ( listener
!= NULL
)
57 listener( owner
, this, ( mvcp_response
** )args
[ 0 ], ( char * )args
[ 1 ], ( char * )args
[ 2 ] );
60 static void melted_push_received( mlt_listener listener
, mlt_properties owner
, melted_server
this, void **args
)
62 if ( listener
!= NULL
)
63 listener( owner
, this, ( mvcp_response
** )args
[ 0 ], ( char * )args
[ 1 ], ( mlt_service
)args
[ 2 ] );
66 /** Initialise a server structure.
69 melted_server
melted_server_init( char *id
)
71 melted_server server
= malloc( sizeof( melted_server_t
) );
73 memset( server
, 0, sizeof( melted_server_t
) );
74 if ( server
!= NULL
&& mlt_properties_init( &server
->parent
, server
) == 0 )
77 server
->port
= DEFAULT_TCP_PORT
;
80 mlt_events_init( &server
->parent
);
81 mlt_events_register( &server
->parent
, "command-received", ( mlt_transmitter
)melted_command_received
);
82 mlt_events_register( &server
->parent
, "doc-received", ( mlt_transmitter
)melted_doc_received
);
83 mlt_events_register( &server
->parent
, "push-received", ( mlt_transmitter
)melted_push_received
);
88 const char *melted_server_id( melted_server server
)
90 return server
!= NULL
&& server
->id
!= NULL ? server
->id
: "melted";
93 void melted_server_set_config( melted_server server
, const char *config
)
97 free( server
->config
);
98 server
->config
= config
!= NULL ?
strdup( config
) : NULL
;
102 /** Set the port of the server.
105 void melted_server_set_port( melted_server server
, int port
)
110 void melted_server_set_proxy( melted_server server
, char *proxy
)
112 mvcp_tokeniser tokeniser
= mvcp_tokeniser_init( );
114 server
->remote_port
= DEFAULT_TCP_PORT
;
115 mvcp_tokeniser_parse_new( tokeniser
, proxy
, ":" );
116 strcpy( server
->remote_server
, mvcp_tokeniser_get_string( tokeniser
, 0 ) );
117 if ( mvcp_tokeniser_count( tokeniser
) == 2 )
118 server
->remote_port
= atoi( mvcp_tokeniser_get_string( tokeniser
, 1 ) );
119 mvcp_tokeniser_close( tokeniser
);
122 /** Wait for a connection.
125 static int melted_server_wait_for_connect( melted_server server
)
130 /* Wait for a 1 second. */
135 FD_SET( server
->socket
, &rfds
);
137 return select( server
->socket
+ 1, &rfds
, NULL
, NULL
, &tv
);
140 /** Run the server thread.
143 static void *melted_server_run( void *arg
)
145 melted_server server
= arg
;
146 pthread_t cmd_parse_info
;
147 connection_t
*tmp
= NULL
;
148 pthread_attr_t thread_attributes
;
151 socksize
= sizeof( struct sockaddr
);
153 melted_log( LOG_NOTICE
, "%s version %s listening on port %i", server
->id
, VERSION
, server
->port
);
155 /* Create the initial thread. We want all threads to be created detached so
156 their resources get freed automatically. (CY: ... hmmph...) */
157 pthread_attr_init( &thread_attributes
);
158 pthread_attr_setdetachstate( &thread_attributes
, PTHREAD_CREATE_DETACHED
);
160 while ( !server
->shutdown
)
162 /* Wait for a new connection. */
163 if ( melted_server_wait_for_connect( server
) )
165 /* Create a new block of data to hold a copy of the incoming connection for
166 our server thread. The thread should free this when it terminates. */
168 tmp
= (connection_t
*) malloc( sizeof(connection_t
) );
169 tmp
->owner
= &server
->parent
;
170 tmp
->parser
= server
->parser
;
171 tmp
->fd
= accept( server
->socket
, (struct sockaddr
*) &(tmp
->sin
), &socksize
);
173 /* Pass the connection to a parser thread :-/ */
175 pthread_create( &cmd_parse_info
, &thread_attributes
, parser_thread
, tmp
);
179 melted_log( LOG_NOTICE
, "%s version %s server terminated.", server
->id
, VERSION
);
184 /** Execute the server thread.
187 int melted_server_execute( melted_server server
)
190 mvcp_response response
= NULL
;
192 struct sockaddr_in ServerAddr
;
195 server
->shutdown
= 0;
197 ServerAddr
.sin_family
= AF_INET
;
198 ServerAddr
.sin_port
= htons( server
->port
);
199 ServerAddr
.sin_addr
.s_addr
= INADDR_ANY
;
201 /* Create socket, and bind to port. Listen there. Backlog = 5
202 should be sufficient for listen (). */
203 server
->socket
= socket( AF_INET
, SOCK_STREAM
, 0 );
205 if ( server
->socket
== -1 )
207 server
->shutdown
= 1;
209 melted_log( LOG_ERR
, "%s unable to create socket.", server
->id
);
213 setsockopt( server
->socket
, SOL_SOCKET
, SO_REUSEADDR
, (char *)&flag
, sizeof( int ) );
215 if ( bind( server
->socket
, (struct sockaddr
*) &ServerAddr
, sizeof (ServerAddr
) ) != 0 )
217 server
->shutdown
= 1;
219 melted_log( LOG_ERR
, "%s unable to bind to port %d.", server
->id
, server
->port
);
223 if ( listen( server
->socket
, 5 ) != 0 )
225 server
->shutdown
= 1;
227 melted_log( LOG_ERR
, "%s unable to listen on port %d.", server
->id
, server
->port
);
231 fcntl( server
->socket
, F_SETFL
, O_NONBLOCK
);
233 if ( !server
->proxy
)
235 melted_log( LOG_NOTICE
, "Starting server on %d.", server
->port
);
236 server
->parser
= melted_parser_init_local( );
240 melted_log( LOG_NOTICE
, "Starting proxy for %s:%d on %d.", server
->remote_server
, server
->remote_port
, server
->port
);
241 server
->parser
= mvcp_parser_init_remote( server
->remote_server
, server
->remote_port
);
244 response
= mvcp_parser_connect( server
->parser
);
246 if ( response
!= NULL
&& mvcp_response_get_error_code( response
) == 100 )
248 /* read configuration file */
249 if ( response
!= NULL
&& !server
->proxy
&& server
->config
!= NULL
)
251 mvcp_response_close( response
);
252 response
= mvcp_parser_run( server
->parser
, server
->config
);
254 if ( mvcp_response_count( response
) > 1 )
256 if ( mvcp_response_get_error_code( response
) > 299 )
257 melted_log( LOG_ERR
, "Error evaluating server configuration. Processing stopped." );
258 for ( index
= 0; index
< mvcp_response_count( response
); index
++ )
259 melted_log( LOG_DEBUG
, "%4d: %s", index
, mvcp_response_get_line( response
, index
) );
263 if ( response
!= NULL
)
266 mvcp_response_close( response
);
267 result
= pthread_create( &server
->thread
, NULL
, melted_server_run
, server
);
270 melted_log( LOG_CRIT
, "Failed to launch TCP listener thread" );
277 melted_log( LOG_ERR
, "Error connecting to parser. Processing stopped." );
278 server
->shutdown
= 1;
285 /** Fetch a units properties
288 mlt_properties
melted_server_fetch_unit( melted_server server
, int index
)
290 melted_unit unit
= melted_get_unit( index
);
291 return unit
!= NULL ? unit
->properties
: NULL
;
294 /** Shutdown the server.
297 void melted_server_shutdown( melted_server server
)
299 if ( server
!= NULL
&& !server
->shutdown
)
301 server
->shutdown
= 1;
302 pthread_join( server
->thread
, NULL
);
303 melted_server_set_config( server
, NULL
);
304 mvcp_parser_close( server
->parser
);
305 server
->parser
= NULL
;
306 close( server
->socket
);
310 /** Close the server.
313 void melted_server_close( melted_server server
)
315 if ( server
!= NULL
&& mlt_properties_dec_ref( &server
->parent
) <= 0 )
317 mlt_properties_close( &server
->parent
);
318 melted_server_shutdown( server
);