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 /** Initialise a server structure.
52 miracle_server
miracle_server_init( char *id
)
54 miracle_server server
= malloc( sizeof( miracle_server_t
) );
57 memset( server
, 0, sizeof( miracle_server_t
) );
59 server
->port
= DEFAULT_TCP_PORT
;
65 void miracle_server_set_config( miracle_server server
, char *config
)
69 free( server
->config
);
70 server
->config
= config
!= NULL ?
strdup( config
) : NULL
;
74 /** Set the port of the server.
77 void miracle_server_set_port( miracle_server server
, int port
)
82 void miracle_server_set_proxy( miracle_server server
, char *proxy
)
84 valerie_tokeniser tokeniser
= valerie_tokeniser_init( );
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
);
94 /** Wait for a connection.
97 static int miracle_server_wait_for_connect( miracle_server server
)
102 /* Wait for a 1 second. */
107 FD_SET( server
->socket
, &rfds
);
109 return select( server
->socket
+ 1, &rfds
, NULL
, NULL
, &tv
);
112 /** Run the server thread.
115 static void *miracle_server_run( void *arg
)
117 miracle_server server
= arg
;
118 pthread_t cmd_parse_info
;
119 connection_t
*tmp
= NULL
;
120 pthread_attr_t thread_attributes
;
123 socksize
= sizeof( struct sockaddr
);
125 miracle_log( LOG_NOTICE
, "%s version %s listening on port %i", server
->id
, VERSION
, server
->port
);
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 ); */
135 while ( !server
->shutdown
)
137 /* Wait for a new connection. */
138 if ( miracle_server_wait_for_connect( server
) )
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. */
143 tmp
= (connection_t
*) malloc( sizeof(connection_t
) );
144 tmp
->parser
= server
->parser
;
145 tmp
->fd
= accept( server
->socket
, (struct sockaddr
*) &(tmp
->sin
), &socksize
);
147 /* Pass the connection to a parser thread :-/ */
149 pthread_create( &cmd_parse_info
, &thread_attributes
, parser_thread
, tmp
);
153 miracle_log( LOG_NOTICE
, "%s version %s server terminated.", server
->id
, VERSION
);
158 /** Execute the server thread.
161 int miracle_server_execute( miracle_server server
)
164 valerie_response response
= NULL
;
166 struct sockaddr_in ServerAddr
;
169 ServerAddr
.sin_family
= AF_INET
;
170 ServerAddr
.sin_port
= htons( server
->port
);
171 ServerAddr
.sin_addr
.s_addr
= INADDR_ANY
;
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 );
177 if ( server
->socket
== -1 )
179 server
->shutdown
= 1;
181 miracle_log( LOG_ERR
, "%s unable to create socket.", server
->id
);
185 setsockopt( server
->socket
, SOL_SOCKET
, SO_REUSEADDR
, (char *)&flag
, sizeof( int ) );
187 if ( bind( server
->socket
, (struct sockaddr
*) &ServerAddr
, sizeof (ServerAddr
) ) != 0 )
189 server
->shutdown
= 1;
191 miracle_log( LOG_ERR
, "%s unable to bind to port %d.", server
->id
, server
->port
);
195 if ( listen( server
->socket
, 5 ) != 0 )
197 server
->shutdown
= 1;
199 miracle_log( LOG_ERR
, "%s unable to listen on port %d.", server
->id
, server
->port
);
203 fcntl( server
->socket
, F_SETFL
, O_NONBLOCK
);
205 if ( !server
->proxy
)
207 miracle_log( LOG_NOTICE
, "Starting server on %d.", server
->port
);
208 server
->parser
= miracle_parser_init_local( );
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
);
216 response
= valerie_parser_connect( server
->parser
);
218 if ( response
!= NULL
&& valerie_response_get_error_code( response
) == 100 )
220 /* read configuration file */
221 if ( response
!= NULL
&& !server
->proxy
&& server
->config
!= NULL
)
223 valerie_response_close( response
);
224 response
= valerie_parser_run( server
->parser
, server
->config
);
226 if ( valerie_response_count( response
) > 1 )
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
) );
235 if ( response
!= NULL
)
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
);
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
);
253 miracle_log( LOG_CRIT
, "Failed to launch TCP listener thread" );
261 miracle_log( LOG_ERR
, "Error connecting to parser. Processing stopped." );
262 server
->shutdown
= 1;
269 /** Shutdown the server.
272 void miracle_server_shutdown( miracle_server server
)
274 if ( server
!= NULL
&& !server
->shutdown
)
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
);
285 /** Close the server.
288 void miracle_server_close( miracle_server server
)
290 if ( server
!= NULL
)
292 miracle_server_shutdown( server
);