Changeset 6738a67 for ipc.c

Timestamp:

2008-07-16T23:22:52Z (16 years ago)

Author:

Sven Moritz Hallberg <pesco@…>

Branches:

master

Children:

9b55485

Parents:

9730d72 (diff), 6a78c0e (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

merge in latest trunk

File:

• ipc.c (modified) (13 diffs)

ipc.c

@@ -33 +33 @@
 
 GSList *child_list = NULL;
-static char *statefile = NULL;
 
 static void ipc_master_cmd_client( irc_t *data, char **cmd )
@@ -63 +62 @@
 }
 
+static void ipc_master_cmd_deaf( irc_t *data, char **cmd )
+{
+        if( global.conf->runmode == RUNMODE_DAEMON )
+        {
+                b_event_remove( global.listen_watch_source_id );
+                close( global.listen_socket );
+                
+                global.listen_socket = global.listen_watch_source_id = -1;
+        
+                ipc_to_children_str( "OPERMSG :Closed listening socket, waiting "
+                                     "for all users to disconnect." );
+        }
+        else
+        {
+                ipc_to_children_str( "OPERMSG :The DEAF command only works in "
+                                     "normal daemon mode. Try DIE instead." );
+        }
+}
+
 void ipc_master_cmd_rehash( irc_t *data, char **cmd )
 {
@@ -98 +116 @@
         { "hello",      0, ipc_master_cmd_client,     0 },
         { "die",        0, ipc_master_cmd_die,        0 },
+        { "deaf",       0, ipc_master_cmd_deaf,       0 },
         { "wallops",    1, NULL,                      IPC_CMD_TO_CHILDREN },
         { "wall",       1, NULL,                      IPC_CMD_TO_CHILDREN },
@@ -209 +228 @@
 }
 
+/* Return just one line. Returns NULL if something broke, an empty string
+   on temporary "errors" (EAGAIN and friends). */
 static char *ipc_readline( int fd )
 {
-        char *buf, *eol;
+        char buf[513], *eol;
         int size;
-        
-        buf = g_new0( char, 513 );
         
         /* Because this is internal communication, it should be pretty safe
@@ -221 +240 @@
            sockets and limites message length, messages should always be
            complete. Saves us quite a lot of code and buffering. */
-        size = recv( fd, buf, 512, MSG_PEEK );
+        size = recv( fd, buf, sizeof( buf ) - 1, MSG_PEEK );
         if( size == 0 || ( size < 0 && !sockerr_again() ) )
                 return NULL;
@@ -229 +248 @@
                 buf[size] = 0;
         
-        eol = strstr( buf, "\r\n" );
-        if( eol == NULL )
+        if( ( eol = strstr( buf, "\r\n" ) ) == NULL )
                 return NULL;
         else
                 size = eol - buf + 2;
-        
-        g_free( buf );
-        buf = g_new0( char, size + 1 );
         
         if( recv( fd, buf, size, 0 ) != size )
                 return NULL;
         else
-                buf[size-2] = 0;
-        
-        return buf;
+                return g_strndup( buf, size - 2 );
 }
 
@@ -254 +267 @@
                 cmd = irc_parse_line( buf );
                 if( cmd )
+                {
                         ipc_command_exec( data, cmd, ipc_master_commands );
-        }
-        else
+                        g_free( cmd );
+                }
+                g_free( buf );
+        }
+        else
+        {
+                ipc_master_free_fd( source );
+        }
+        
+        return TRUE;
+}
+
+gboolean ipc_child_read( gpointer data, gint source, b_input_condition cond )
+{
+        char *buf, **cmd;
+        
+        if( ( buf = ipc_readline( source ) ) )
+        {
+                cmd = irc_parse_line( buf );
+                if( cmd )
+                {
+                        ipc_command_exec( data, cmd, ipc_child_commands );
+                        g_free( cmd );
+                }
+                g_free( buf );
+        }
+        else
+        {
+                ipc_child_disable();
+        }
+        
+        return TRUE;
+}
+
+void ipc_to_master( char **cmd )
+{
+        if( global.conf->runmode == RUNMODE_FORKDAEMON )
+        {
+                char *s = irc_build_line( cmd );
+                ipc_to_master_str( "%s", s );
+                g_free( s );
+        }
+        else if( global.conf->runmode == RUNMODE_DAEMON )
+        {
+                ipc_command_exec( NULL, cmd, ipc_master_commands );
+        }
+}
+
+void ipc_to_master_str( char *format, ... )
+{
+        char *msg_buf;
+        va_list params;
+
+        va_start( params, format );
+        msg_buf = g_strdup_vprintf( format, params );
+        va_end( params );
+        
+        if( strlen( msg_buf ) > 512 )
+        {
+                /* Don't send it, it's too long... */
+        }
+        else if( global.conf->runmode == RUNMODE_FORKDAEMON )
+        {
+                if( global.listen_socket >= 0 )
+                        if( write( global.listen_socket, msg_buf, strlen( msg_buf ) ) <= 0 )
+                                ipc_child_disable();
+        }
+        else if( global.conf->runmode == RUNMODE_DAEMON )
+        {
+                char **cmd, *s;
+                
+                if( ( s = strchr( msg_buf, '\r' ) ) )
+                        *s = 0;
+                
+                cmd = irc_parse_line( msg_buf );
+                ipc_command_exec( NULL, cmd, ipc_master_commands );
+                g_free( cmd );
+        }
+        
+        g_free( msg_buf );
+}
+
+void ipc_to_children( char **cmd )
+{
+        if( global.conf->runmode == RUNMODE_FORKDAEMON )
+        {
+                char *msg_buf = irc_build_line( cmd );
+                ipc_to_children_str( "%s", msg_buf );
+                g_free( msg_buf );
+        }
+        else if( global.conf->runmode == RUNMODE_DAEMON )
         {
                 GSList *l;
-                struct bitlbee_child *c;
-                
-                for( l = child_list; l; l = l->next )
+                
+                for( l = irc_connection_list; l; l = l->next )
+                        ipc_command_exec( l->data, cmd, ipc_child_commands );
+        }
+}
+
+void ipc_to_children_str( char *format, ... )
+{
+        char *msg_buf;
+        va_list params;
+
+        va_start( params, format );
+        msg_buf = g_strdup_vprintf( format, params );
+        va_end( params );
+        
+        if( strlen( msg_buf ) > 512 )
+        {
+                /* Don't send it, it's too long... */
+        }
+        else if( global.conf->runmode == RUNMODE_FORKDAEMON )
+        {
+                int msg_len = strlen( msg_buf );
+                GSList *l, *next;
+                
+                for( l = child_list; l; l = next )
                 {
-                        c = l->data;
-                        if( c->ipc_fd == source )
+                        struct bitlbee_child *c = l->data;
+                        
+                        next = l->next;
+                        if( write( c->ipc_fd, msg_buf, msg_len ) <= 0 )
                         {
                                 ipc_master_free_one( c );
                                 child_list = g_slist_remove( child_list, c );
-                                break;
                         }
-                }
-        }
-        
-        return TRUE;
-}
-
-gboolean ipc_child_read( gpointer data, gint source, b_input_condition cond )
-{
-        char *buf, **cmd;
-        
-        if( ( buf = ipc_readline( source ) ) )
-        {
-                cmd = irc_parse_line( buf );
-                if( cmd )
-                        ipc_command_exec( data, cmd, ipc_child_commands );
-        }
-        else
-        {
-                b_event_remove( global.listen_watch_source_id );
-                close( global.listen_socket );
-                
-                global.listen_socket = -1;
-        }
-        
-        return TRUE;
-}
-
-void ipc_to_master( char **cmd )
-{
-        if( global.conf->runmode == RUNMODE_FORKDAEMON )
-        {
-                char *s = irc_build_line( cmd );
-                ipc_to_master_str( "%s", s );
-                g_free( s );
-        }
-        else if( global.conf->runmode == RUNMODE_DAEMON )
-        {
-                ipc_command_exec( NULL, cmd, ipc_master_commands );
-        }
-}
-
-void ipc_to_master_str( char *format, ... )
-{
-        char *msg_buf;
-        va_list params;
-
-        va_start( params, format );
-        msg_buf = g_strdup_vprintf( format, params );
-        va_end( params );
-        
-        if( strlen( msg_buf ) > 512 )
-        {
-                /* Don't send it, it's too long... */
-        }
-        else if( global.conf->runmode == RUNMODE_FORKDAEMON )
-        {
-                write( global.listen_socket, msg_buf, strlen( msg_buf ) );
-        }
-        else if( global.conf->runmode == RUNMODE_DAEMON )
-        {
-                char **cmd, *s;
-                
-                if( ( s = strchr( msg_buf, '\r' ) ) )
-                        *s = 0;
-                
-                cmd = irc_parse_line( msg_buf );
-                ipc_command_exec( NULL, cmd, ipc_master_commands );
-                g_free( cmd );
-        }
-        
-        g_free( msg_buf );
-}
-
-void ipc_to_children( char **cmd )
-{
-        if( global.conf->runmode == RUNMODE_FORKDAEMON )
-        {
-                char *msg_buf = irc_build_line( cmd );
-                ipc_to_children_str( "%s", msg_buf );
-                g_free( msg_buf );
-        }
-        else if( global.conf->runmode == RUNMODE_DAEMON )
-        {
-                GSList *l;
-                
-                for( l = irc_connection_list; l; l = l->next )
-                        ipc_command_exec( l->data, cmd, ipc_child_commands );
-        }
-}
-
-void ipc_to_children_str( char *format, ... )
-{
-        char *msg_buf;
-        va_list params;
-
-        va_start( params, format );
-        msg_buf = g_strdup_vprintf( format, params );
-        va_end( params );
-        
-        if( strlen( msg_buf ) > 512 )
-        {
-                /* Don't send it, it's too long... */
-        }
-        else if( global.conf->runmode == RUNMODE_FORKDAEMON )
-        {
-                int msg_len = strlen( msg_buf );
-                GSList *l;
-                
-                for( l = child_list; l; l = l->next )
-                {
-                        struct bitlbee_child *c = l->data;
-                        write( c->ipc_fd, msg_buf, msg_len );
                 }
         }
@@ -410 +424 @@
 }
 
+void ipc_master_free_fd( int fd )
+{
+        GSList *l;
+        struct bitlbee_child *c;
+        
+        for( l = child_list; l; l = l->next )
+        {
+                c = l->data;
+                if( c->ipc_fd == fd )
+                {
+                        ipc_master_free_one( c );
+                        child_list = g_slist_remove( child_list, c );
+                        break;
+                }
+        }
+}
+
 void ipc_master_free_all()
 {
@@ -421 +452 @@
 }
 
+void ipc_child_disable()
+{
+        b_event_remove( global.listen_watch_source_id );
+        close( global.listen_socket );
+        
+        global.listen_socket = -1;
+}
+
+#ifndef _WIN32
 char *ipc_master_save_state()
 {
@@ -461 +501 @@
 }
 
-void ipc_master_set_statefile( char *fn )
-{
-        statefile = g_strdup( fn );
-}
-
 
 static gboolean new_ipc_client( gpointer data, gint serversock, b_input_condition cond )
@@ -486 +521 @@
 }
 
-#ifndef _WIN32
 int ipc_master_listen_socket()
 {
@@ -523 +557 @@
 }
 #else
+int ipc_master_listen_socket()
+{
         /* FIXME: Open named pipe \\.\BITLBEE */
+        return 0;
+}
 #endif
 
-int ipc_master_load_state()
+int ipc_master_load_state( char *statefile )
 {
         struct bitlbee_child *child;
@@ -534 +572 @@
         if( statefile == NULL )
                 return 0;
+        
         fp = fopen( statefile, "r" );
         unlink( statefile );    /* Why do it later? :-) */