7 transport_router_registrar* routt;
9 void _build_trusted_sites( transport_router_registrar* router );
11 void sig_hup_handler( int a ) {
12 router_registrar_free( routt );
15 free( router_resource );
20 int main( int argc, char* argv[] ) {
23 fatal_handler( "Usage: %s <path_to_config_file>", argv[0] );
29 jid_get_domain( "client@elroy/test", b );
30 fprintf(stderr, "Domain %s\n", b );
32 config_reader_init( argv[1] );
33 if( conf_reader == NULL ) fatal_handler( "main(): Config is NULL" );
35 /* laod the config options */
36 char* server = config_value("//router/transport/server");
37 char* port = config_value("//router/transport/port");
38 char* username = config_value("//router/transport/username");
39 char* password = config_value("//router/transport/password");
40 router_resource = config_value("//router/transport/resource");
41 char* con_timeout = config_value("//router/transport/connect_timeout" );
42 char* max_retries = config_value("//router/transport/max_reconnect_attempts" );
43 char* component = config_value("//router/component" );
45 fprintf(stderr, "Router connecting as \nserver: %s \nport: %s \nuser:%s \nresource:%s\n",
46 server, port, username, router_resource );
48 int iport = atoi( port );
49 int con_itimeout = atoi( con_timeout );
50 int max_retries_ = atoi(max_retries);
53 icomponent = atoi(component);
56 fatal_handler( "Port is negative or 0" );
61 /* build the router_registrar */
62 transport_router_registrar* router_registrar =
63 router_registrar_init( server, iport, username, password, router_resource, 0, con_itimeout, icomponent );
65 routt = router_registrar;
74 signal(SIGHUP,sig_hup_handler);
78 /* wait for incoming... */
79 while( ++counter <= max_retries_ ) {
81 /* connect to jabber */
82 if( router_registrar_connect( router_registrar ) ) {
83 info_handler( "Connected..." );
84 fprintf(stderr, "- Connected -\n");
86 listen_loop( router_registrar );
88 fatal_handler( "Could not connect to Jabber Server" );
90 /* this should never happen */
91 warning_handler( "Jabber server probably went away, attempting reconnect" );
97 router_registrar_free( router_registrar );
103 transport_router_registrar* router_registrar_init( char* server,
104 int port, char* username, char* password,
105 char* resource, int client_timeout, int con_timeout, int component ) {
107 if( server == NULL ) { return NULL; }
109 /* allocate a new router_registrar object */
110 size_t size = sizeof( transport_router_registrar );
111 transport_router_registrar* router_registrar = (transport_router_registrar*) safe_malloc( size );
113 router_registrar->client_timeout = client_timeout;
114 router_registrar->jabber = jabber_connect_init( server, port, username, password, resource, con_timeout, component );
115 _build_trusted_sites( router_registrar );
116 info_handler( "Trusted stuff %s, %s, %s", router_registrar->trusted_servers[0],
117 router_registrar->trusted_clients[0], router_registrar->trusted_clients[1] );
118 return router_registrar;
122 void _build_trusted_sites( transport_router_registrar* router ) {
124 router->trusted_servers = (char**) safe_malloc(sizeof(char**));
125 router->trusted_clients = (char**) safe_malloc(sizeof(char**));
127 *(router->trusted_servers) = (char*) safe_malloc(ROUTER_MAX_TRUSTED);
128 *(router->trusted_clients) = (char*) safe_malloc(ROUTER_MAX_TRUSTED);
132 char* server = config_value("//router/trusted_domains/server%d", i );
136 router->trusted_servers[i-1] = server;
141 char* client = config_value( "//router/trusted_domains/client%d", i );
144 router->trusted_clients[i-1] = client;
147 if( router->trusted_servers[0] == NULL ||
148 router->trusted_clients[0] == NULL )
150 fatal_handler( "You must specify at least one trusted server and client in the config file");
154 jabber_connect* jabber_connect_init( char* server,
155 int port, char* username, char* password, char* resource, int connect_timeout, int component ) {
157 size_t len = sizeof(jabber_connect);
158 jabber_connect* jabber = (jabber_connect*) safe_malloc( len );
161 jabber->connect_timeout = connect_timeout;
163 jabber->server = strdup(server);
164 jabber->username = strdup(username);
165 jabber->password = strdup(password);
166 jabber->resource = strdup(resource);
168 if( jabber->server == NULL || jabber->username == NULL ||
169 jabber->password == NULL || jabber->resource == NULL ) {
170 fatal_handler( "jabber_init(): Out of Memory" );
174 /* build the transport client */
175 jabber->t_client = client_init( jabber->server, jabber->port, component );
180 /* connect the router_registrar to jabber */
181 int router_registrar_connect( transport_router_registrar* router ) {
182 return j_connect( router->jabber );
185 /* connect a jabber_connect object jabber */
186 int j_connect( jabber_connect* jabber ) {
187 if( jabber == NULL ) { return 0; }
188 return client_connect( jabber->t_client,
189 jabber->username, jabber->password, jabber->resource,
190 jabber->connect_timeout, AUTH_DIGEST );
193 int fill_fd_set( transport_router_registrar* router, fd_set* set ) {
198 int router_fd = router->jabber->t_client->session->sock_obj->sock_fd;
200 FD_SET( router_fd, set );
202 server_class_node* cur_node = router->server_class_list;
203 while( cur_node != NULL ) {
204 int cur_class_fd = cur_node->jabber->t_client->session->sock_obj->sock_fd;
205 if( cur_class_fd > max_fd )
206 max_fd = cur_class_fd;
207 FD_SET( cur_class_fd, set );
208 cur_node = cur_node->next;
216 void listen_loop( transport_router_registrar* router ) {
222 int router_fd = router->jabber->t_client->session->sock_obj->sock_fd;
223 transport_message* cur_msg;
228 int max_fd = fill_fd_set( router, &listen_set );
231 fatal_handler( "fill_fd_set return bogus max_fd: %d", max_fd );
234 info_handler( "Going into select" );
236 if( (select_ret=select(max_fd+ 1, &listen_set, NULL, NULL, NULL)) < 0 ) {
238 warning_handler( "Select returned error %d", select_ret );
239 warning_handler( "Select Error %d on fd %d", errno );
240 perror( "Select Error" );
241 warning_handler( "Errors: EBADF %d, EINTR %d, EINVAL %d, ENOMEM %d",
242 EBADF, EINTR, EINVAL, ENOMEM );
247 info_handler( "Select returned %d", select_ret );
249 if( FD_ISSET( router_fd, &listen_set ) ) {
250 cur_msg = client_recv( router->jabber->t_client, 1 );
252 /* We only process a message if we have some trusted servers and the current
253 message came from one of those servers */
254 if(router->trusted_servers && router->trusted_servers[0]) {
258 char server_buf[256];
259 memset(server_buf,0,256);
260 jid_get_domain( cur_msg->sender, server_buf );
261 info_handler("Received top level message from %s", server_buf );
264 if(router->trusted_servers[i] == NULL)
266 if(!strcmp(router->trusted_servers[i], server_buf)) {
273 router_registrar_handle_msg( router, cur_msg );
275 warning_handler( "Received top level message from unpriveleged sender %s", cur_msg->sender );
278 message_free( cur_msg );
279 if( ++num_handled == select_ret )
283 /* cycle through the children and find any whose fd's are ready for reading */
284 server_class_node* cur_node = router->server_class_list;
285 while( cur_node != NULL ) {
286 int cur_fd = cur_node->jabber->t_client->session->sock_obj->sock_fd;
288 if( FD_ISSET(cur_fd, &listen_set) ) {
290 FD_CLR(cur_fd,&listen_set);
292 cur_msg = client_recv( cur_node->jabber->t_client, 1 );
293 info_handler( "%s received from %s", cur_node->server_class, cur_msg->sender );
294 int handle_ret = server_class_handle_msg( router, cur_node, cur_msg );
296 if( handle_ret == -1 ) {
297 warning_handler( "server_class_handle_msg() returned -1" );
298 cur_node = router->server_class_list; /*start over*/
301 } else if( handle_ret == 0 ) {
302 /* delete and continue */
303 warning_handler( "server_class_handle_msg() returned 0" );
304 server_class_node* tmp_node = cur_node->next;
305 remove_server_class( router, cur_node );
310 info_handler( "%s handled message successfully", cur_node->server_class );
311 /* dont free message here */
312 if( num_handled == select_ret )
315 if( num_handled == select_ret )
317 cur_node = cur_node->next;
319 } /* cycling through the server_class list */
321 } /* no select errors */
326 /* determine where to route top level messages */
327 int router_registrar_handle_msg( transport_router_registrar* router_registrar, transport_message* msg ) {
329 info_handler( "Received class: %s : command %s ", msg->router_class, msg->router_command );
331 if( router_registrar == NULL || msg == NULL ) { return 0; }
333 // user issued a ruoter query
334 /* a query command has router_command="query" and the actual query type
335 is the content of the message */
336 if( !strcmp(msg->router_command,"query")) {
337 info_handler( "Router received query command" );
339 // user issues a servers query
340 if( !strcmp(msg->body, "servers")) {
342 info_handler( "Router received servers query command" );
343 router_return_server_info( router_registrar, msg );
349 info_handler("Looking for server_class_node %s...",msg->router_class);
350 server_class_node* active_class_node = find_server_class( router_registrar, msg->router_class );
352 if( active_class_node == NULL ) {
353 info_handler("Could not find server_class_node %s, creating one.",msg->router_class);
355 /* there is no server_class for msg->router_class so we build it here */
356 if( strcmp( msg->router_command, "register") == 0 ) {
358 info_handler("Adding server_class_node for %s",msg->router_class);
360 init_server_class( router_registrar, msg->sender, msg->router_class );
362 if( active_class_node == NULL ) {
363 fatal_handler( "router_listen(): active_class_node == NULL for %s", msg->sender );
367 if (router_registrar->server_class_list != NULL) {
368 active_class_node->next = router_registrar->server_class_list;
369 router_registrar->server_class_list->prev = active_class_node;
371 router_registrar->server_class_list = active_class_node;
373 //spawn_server_class( (void*) active_class_node );
376 warning_handler( "router_register_handler_msg(): Bad Command [%s] for class [%s]",
377 msg->router_command, msg->router_class );
380 } else if( strcmp( msg->router_command, "register") == 0 ) {
381 /* there is a server_class for msg->router_class so we
382 need to either add a new server_node or update the existing one */
385 server_node* s_node = find_server_node( active_class_node, msg->sender );
387 if( s_node != NULL ) {
388 s_node->available = 1;
389 s_node->upd_time = time(NULL);
390 info_handler( "Found matching registered server: %s. Updating.",
393 s_node = init_server_node( msg->sender );
395 info_handler( "Adding server_node for: %s.", s_node->remote_id );
397 if (s_node == NULL ) {
398 warning_handler( " Could not create new xerver_node for %s.",
403 s_node->next = active_class_node->current_server_node->next;
404 s_node->prev = active_class_node->current_server_node;
406 active_class_node->current_server_node->next->prev = s_node;
407 active_class_node->current_server_node->next = s_node;
411 } else if( strcmp( msg->router_command, "unregister") == 0 ) {
413 if( ! unregister_server_node( active_class_node, msg->sender ) )
414 remove_server_class( router_registrar, active_class_node );
417 warning_handler( "router_register_handler_msg(): Bad Command [%s] for class [%s]",
418 msg->router_command, msg->router_class );
425 /* removes a server class node from the top level router_registrar */
426 int unregister_server_node( server_class_node* active_class_node, char* remote_id ) {
428 server_node* d_node = find_server_node( active_class_node, remote_id );
430 if ( d_node != NULL ) {
432 info_handler( "Removing server_node for: %s.", d_node->remote_id );
434 if ( d_node->next == NULL ) {
435 warning_handler( "NEXT is NULL in ring [%s] -- "
436 "THIS SHOULD NEVER HAPPEN",
441 if ( d_node->prev == NULL ) {
442 warning_handler( "PREV is NULL in a ring [%s] -- "
443 "THIS SHOULD NEVER HAPPEN",
448 if ( d_node->next == d_node && d_node->prev == d_node) {
449 info_handler( "Last node, setting ring to NULL: %s.",
452 active_class_node->current_server_node = NULL;
454 server_node_free( d_node );
458 info_handler( "Nodes remain, splicing: %s, %s",
459 d_node->prev->remote_id,
460 d_node->next->remote_id);
462 info_handler( "d_node => %x, next => %x, prev => %x",
463 d_node, d_node->next, d_node->prev );
466 d_node->prev->next = d_node->next;
467 d_node->next->prev = d_node->prev;
469 info_handler( "prev => %x, prev->next => %x, prev->prev => %x",
470 d_node->prev, d_node->prev->next, d_node->prev->prev );
472 info_handler( "next => %x, next->next => %x, next->prev => %x",
473 d_node->next, d_node->next->next, d_node->next->prev );
475 if (active_class_node->current_server_node == d_node)
476 active_class_node->current_server_node = d_node->next;
479 server_node_free( d_node );
486 server_node * find_server_node ( server_class_node * class, const char * remote_id ) {
488 if ( class == NULL ) {
489 warning_handler(" find_server_node(): bad arg!");
493 server_node * start_node = class->current_server_node;
494 server_node * node = class->current_server_node;
500 if ( strcmp(node->remote_id, remote_id) == 0 )
505 } while ( node != start_node );
510 /* if we return -1, then we just deleted the server_class you were looking for
511 if we return 0, then some other error has occured
512 we return 1 otherwise */
513 int remove_server_class( transport_router_registrar* router, server_class_node* class ) {
517 transport_message * msg = NULL;
518 while ( (msg = client_recv(class->jabber->t_client, 0)) != NULL ) {
519 server_class_handle_msg(router, class, msg);
523 free( class->server_class );
524 class->server_class = NULL;
526 find_server_class( router, router_resource ); /* find deletes for us */
528 if( router->server_class_list == NULL )
533 server_class_node * find_server_class ( transport_router_registrar * router, const char * class_id ) {
535 if ( router == NULL ) {
536 warning_handler(" find_server_class(): bad arg!");
540 info_handler( "Finding server class for %s", class_id );
541 server_class_node * class = router->server_class_list;
542 server_class_node * dead_class = NULL;
544 while ( class != NULL ) {
546 if ( class->server_class == NULL ) {
547 info_handler( "Found an empty server class" );
549 if ( class->prev != NULL ) {
550 class->prev->next = class->next;
551 if( class->next != NULL ) {
552 class->next->prev = class->prev;
556 info_handler( "Empty class is the first on the list" );
557 if( class->next != NULL )
558 router->server_class_list = class->next;
560 else { /* we're the last class node in the class node list */
561 info_handler( "Empty class is the last on the list" );
562 server_class_node_free( router->server_class_list );
563 router->server_class_list = NULL;
572 info_handler( "Tossing our dead class" );
573 server_class_node_free( dead_class );
579 if ( strcmp(class->server_class, class_id) == 0 )
581 info_handler( "%s != %s", class->server_class, class_id );
589 /* builds a new server class and connects to the jabber server with the new resource */
590 server_class_node* init_server_class(
591 transport_router_registrar* router, char* remote_id, char* server_class ) {
593 size_t len = sizeof( server_class_node );
594 server_class_node* node = (server_class_node*) safe_malloc( len );
596 node->jabber = jabber_connect_init( router->jabber->server,
597 router->jabber->port, router->jabber->username,
598 router->jabber->password, server_class, router->jabber->connect_timeout, router->component );
602 node->server_class = strdup( server_class );
603 if( server_class == NULL ) {
604 fatal_handler( "imit_server_class(): out of memory for %s", server_class );
608 info_handler( "Received class to init_server_class: %s", server_class );
609 node->current_server_node = init_server_node( remote_id );
610 if( node->current_server_node == NULL ) {
611 fatal_handler( "init_server_class(): NULL server_node for %s", remote_id );
616 if( ! j_connect( node->jabber ) ) {
617 fatal_handler( "Unable to init server class %s", node->server_class );
621 info_handler( "Jabber address in init for %s : address %x : username %s : resource %s",
622 node->server_class, node->jabber->t_client->session->sock_obj->sock_fd,
623 node->jabber->username, node->jabber->resource );
629 /* builds a new server_node to be added to the ring of server_nodes */
630 server_node* init_server_node( char* remote_id ) {
632 info_handler( "Initing server node for %s", remote_id );
633 server_node* current_server_node;
634 size_t size = sizeof( server_node);
635 current_server_node = (server_node*) safe_malloc( size );
637 current_server_node->remote_id = strdup(remote_id);
638 if( current_server_node->remote_id == NULL ) {
639 fatal_handler("init_server_class(): Out of Memory for %s", remote_id );
643 current_server_node->reg_time = time(NULL);
644 current_server_node->available = 1;
645 current_server_node->next = current_server_node;
646 current_server_node->prev = current_server_node;
649 return current_server_node;
653 int server_class_handle_msg( transport_router_registrar* router,
654 server_class_node* s_node, transport_message* msg ) {
656 if( s_node->current_server_node == NULL ) {
657 /* return error to client ??!*/
658 /* WE have no one to send the message to */
659 warning_handler( "We no longer have any servers for %s : "
660 "no one to send the message to. Sending error message to %s", s_node->server_class, msg->sender );
661 free( msg->recipient );
663 char* rec = strdup( msg->sender );
665 fatal_handler( "class msg_handler: out of memory");
669 info_handler( "Building error message to return for %s", s_node->server_class);
670 msg->recipient = rec;
671 set_msg_error(msg, "cancel", 501);
673 client_send_message( s_node->jabber->t_client, msg );
676 remove_server_class( router, s_node );
681 info_handler( "[%s] Received \nfrom: %s \nto: %s",
682 s_node->server_class, msg->sender, msg->recipient );
684 if( msg->is_error ) {
685 warning_handler( "We've received an error message type: %s : code: %d",
686 msg->error_type, msg->error_code );
688 if( strcmp( msg->error_type, "cancel" ) == 0 ) {
689 warning_handler( "Looks like we've lost a server!" );
690 server_node* dead_node = find_server_node( s_node, msg->sender );
692 if( dead_node != NULL ) {
693 //message_free( msg );
694 transport_message* tmp = dead_node->last_sent;
696 /* copy over last sent, it will be freed in the unregister function */
697 transport_message* tmp2 = message_init( tmp->body, tmp->subject, tmp->thread,
698 tmp->recipient, tmp->sender );
700 message_set_router_info( tmp2, tmp->router_from,
701 tmp->router_to, tmp->router_class, tmp->router_command, tmp->broadcast );
703 if( ! unregister_server_node( s_node, dead_node->remote_id ) ) {
704 /* WE have no one to send the message to */
705 warning_handler( "We no longer have any servers for %s : "
706 "no one to send the message to.", s_node->server_class );
707 free( msg->recipient );
709 char* rec = strdup( msg->router_from );
711 fatal_handler( "class msg_handler: out of memory");
715 info_handler( "Building error message to return for %s", s_node->server_class);
716 msg->recipient = rec;
717 client_send_message( s_node->jabber->t_client, msg );
718 message_free( tmp2 );
730 server_node* c_node = s_node->current_server_node->next;
732 /* not implemented yet */
733 while( ! c_node->available ) {
734 if( c_node == s_node->current_server_node ) {
735 warning_handler("No server_node's are available for %s", s_node->server_class );
736 /* XXX send error message to client */
739 c_node = c_node->next;
741 s_node->current_server_node = c_node;
743 transport_message * new_msg =
744 message_init( msg->body, msg->subject, msg->thread,
745 s_node->current_server_node->remote_id, msg->sender );
747 message_set_router_info( new_msg, msg->sender, NULL, NULL, NULL, 0 );
749 info_handler( "[%s] Routing message from [%s]\nto [%s]", s_node->server_class, msg->sender, new_msg->recipient );
750 //info_handler( "New Message Details: sender:%s recipient: %s \nbody: %s",
751 // new_msg->sender, new_msg->recipient, new_msg->body );
753 message_free( s_node->current_server_node->last_sent );
754 s_node->current_server_node->last_sent = msg;
756 if ( new_msg != NULL && client_send_message( s_node->jabber->t_client, new_msg ) ) {
757 s_node->current_server_node->serve_count++;
758 s_node->current_server_node->la_time = time(NULL);
759 message_free( new_msg ); // XXX
762 message_free( new_msg ); // XXX
768 int router_return_server_info(
769 transport_router_registrar* router, transport_message* msg ) {
771 server_class_node* cur_class = router->server_class_list;
772 growing_buffer* buffer = buffer_init(1024);
774 while( cur_class != NULL ) {
776 server_node* start_node = cur_class->current_server_node;
777 server_node* cur_node = start_node;
778 if( cur_node == NULL ) continue;
784 sprintf( tbuf, "%d", (int)cur_node->reg_time );
785 buffer_add( buffer, tbuf );
786 buffer_add( buffer, " | ");
789 sprintf( tbuf, "%d", (int)cur_node->upd_time );
790 buffer_add( buffer, tbuf );
791 buffer_add( buffer, " | ");
794 sprintf( tbuf, "%d", (int)cur_node->la_time );
795 buffer_add( buffer, tbuf );
796 buffer_add( buffer, " | ");
801 sprintf(sbuf,"%d",cur_node->serve_count);
803 buffer_add( buffer, "#" );
804 buffer_add( buffer, sbuf );
805 buffer_add( buffer, " | ");
807 buffer_add( buffer, cur_class->server_class );
808 buffer_add( buffer, " | ");
810 buffer_add( buffer, cur_node->remote_id );
812 buffer_add( buffer, "\n" );
813 cur_node = cur_node->next;
815 } while( cur_node != start_node );
817 cur_class = cur_class->next;
821 info_handler( "Router returning servers query command: %s", buffer->buf );
823 transport_message* new_msg;
825 if( buffer->buf == NULL || strlen(buffer->buf) == 0 )
826 new_msg = message_init( "0", NULL, NULL, msg->sender, NULL );
828 new_msg = message_init( buffer->buf, NULL, NULL, msg->sender, NULL );
830 client_send_message( router->jabber->t_client, new_msg );
831 message_free( new_msg );
837 int router_registrar_free( transport_router_registrar* router_registrar ) {
838 if( router_registrar == NULL ) return 0;
839 jabber_connect_free( router_registrar->jabber );
841 /* free the server_class list XXX */
842 while( router_registrar->server_class_list != NULL ) {
843 remove_server_class(router_registrar, router_registrar->server_class_list);
846 transport_router_registrar* router = router_registrar;
852 if( router->trusted_servers[i] == NULL &&
853 router->trusted_clients[i] == NULL )
856 if(router->trusted_servers[i] != NULL)
857 free(router->trusted_servers[i]);
858 if(router->trusted_clients[i] != NULL)
859 free(router->trusted_clients[i]);
863 free( router_registrar );
871 int server_class_node_free( server_class_node* node ) {
872 if( node == NULL ) { return 0; }
873 if( node->server_class != NULL )
874 free( node->server_class );
876 jabber_connect_free( node->jabber );
878 /* just in case, free the list */
879 while( node->current_server_node != NULL ) {
880 unregister_server_node( node, node->current_server_node->remote_id );
886 int server_node_free( server_node* node ) {
887 if( node == NULL ) { return 0; }
888 message_free( node->last_sent );
889 free( node->remote_id );
894 int jabber_connect_free( jabber_connect* jabber ) {
895 if( jabber == NULL ) { return 0; }
896 client_free( jabber->t_client );
897 free( jabber->username );
898 free( jabber->password );
899 free( jabber->resource );
900 free( jabber->server );