diff options
-rw-r--r-- | bitlbee.c | 26 | ||||
-rw-r--r-- | irc.c | 18 | ||||
-rw-r--r-- | irc.h | 4 | ||||
-rw-r--r-- | log.c | 112 |
4 files changed, 88 insertions, 72 deletions
@@ -199,7 +199,7 @@ gboolean bitlbee_io_current_client_read( GIOChannel *source, GIOCondition condit if( !irc_process( irc ) ) { log_message( LOGLVL_INFO, "Destroying connection with fd %d.", irc->fd ); - irc_free( irc ); + irc_abort( irc ); return FALSE; } @@ -207,7 +207,7 @@ gboolean bitlbee_io_current_client_read( GIOChannel *source, GIOCondition condit if( irc->readbuffer && ( strlen( irc->readbuffer ) > 1024 ) ) { log_message( LOGLVL_ERROR, "Maximum line length exceeded." ); - irc_free( irc ); + irc_abort( irc ); return FALSE; } @@ -226,25 +226,25 @@ gboolean bitlbee_io_current_client_write( GIOChannel *source, GIOCondition condi size = strlen( irc->sendbuffer ); st = write( irc->fd, irc->sendbuffer, size ); - if( st <= 0 ) + if( st == 0 || ( st < 0 && !sockerr_again() ) ) { - if( sockerr_again() ) - { - return TRUE; - } - else - { - irc_free( irc ); - return FALSE; - } + irc_free( irc ); + return FALSE; + } + else if( st < 0 ) /* && sockerr_again() */ + { + return TRUE; } if( st == size ) { g_free( irc->sendbuffer ); irc->sendbuffer = NULL; - irc->w_watch_source_id = 0; + + if( irc->status == USTATUS_SHUTDOWN ) + irc_free( irc ); + return( FALSE ); } else @@ -150,6 +150,20 @@ irc_t *irc_new( int fd ) return( irc ); } +void irc_abort( irc_t *irc ) +{ + irc->status = USTATUS_SHUTDOWN; + if( irc->sendbuffer ) + { + g_source_remove( irc->r_watch_source_id ); + irc->r_watch_source_id = g_timeout_add_full( G_PRIORITY_HIGH, 1000, (GSourceFunc) irc_free, irc, NULL ); + } + else + { + irc_free( irc ); + } +} + static gboolean irc_free_userhash( gpointer key, gpointer value, gpointer data ) { g_free( key ); @@ -158,7 +172,7 @@ static gboolean irc_free_userhash( gpointer key, gpointer value, gpointer data ) } /* Because we have no garbage collection, this is quite annoying */ -void irc_free(irc_t * irc) +void irc_free( irc_t * irc ) { account_t *account, *accounttmp; user_t *user, *usertmp; @@ -495,7 +509,7 @@ int irc_exec( irc_t *irc, char **cmd ) else if( g_strcasecmp( cmd[0], "QUIT" ) == 0 ) { irc_write( irc, "ERROR :%s%s", cmd[1]?"Quit: ":"", cmd[1]?cmd[1]:"Client Quit" ); - g_io_channel_close( irc->io_channel ); + /* g_io_channel_close( irc->io_channel ); */ return( 0 ); } @@ -43,7 +43,8 @@ typedef enum USTATUS_OFFLINE, USTATUS_AUTHORIZED, USTATUS_LOGGED_IN, - USTATUS_IDENTIFIED + USTATUS_IDENTIFIED, + USTATUS_SHUTDOWN } irc_status_t; typedef struct channel @@ -103,6 +104,7 @@ typedef struct irc extern GSList *irc_connection_list; irc_t *irc_new( int fd ); +void irc_abort( irc_t *irc ); void irc_free( irc_t *irc ); int irc_exec( irc_t *irc, char **cmd ); @@ -37,11 +37,11 @@ static void log_console(int level, char *logmessage); void log_init(void) { openlog("bitlbee", LOG_PID, LOG_DAEMON); - logoutput.informational=&log_null; - logoutput.warning=&log_null; - logoutput.error=&log_null; + logoutput.informational = &log_null; + logoutput.warning = &log_null; + logoutput.error = &log_null; #ifdef DEBUG - logoutput.debug=&log_null; + logoutput.debug = &log_null; #endif return; @@ -50,46 +50,46 @@ void log_init(void) { void log_link(int level, int output) { /* I know it's ugly, but it works and I didn't feel like messing with pointer to function pointers */ - if(level==LOGLVL_INFO) { - if(output==LOGOUTPUT_NULL) - logoutput.informational=&log_null; - else if(output==LOGOUTPUT_IRC) - logoutput.informational=&log_irc; - else if(output==LOGOUTPUT_SYSLOG) - logoutput.informational=&log_syslog; - else if(output==LOGOUTPUT_CONSOLE) - logoutput.informational=&log_console; + if(level == LOGLVL_INFO) { + if(output == LOGOUTPUT_NULL) + logoutput.informational = &log_null; + else if(output == LOGOUTPUT_IRC) + logoutput.informational = &log_irc; + else if(output == LOGOUTPUT_SYSLOG) + logoutput.informational = &log_syslog; + else if(output == LOGOUTPUT_CONSOLE) + logoutput.informational = &log_console; } - else if(level==LOGLVL_WARNING) { - if(output==LOGOUTPUT_NULL) - logoutput.warning=&log_null; - else if(output==LOGOUTPUT_IRC) - logoutput.warning=&log_irc; - else if(output==LOGOUTPUT_SYSLOG) - logoutput.warning=&log_syslog; - else if(output==LOGOUTPUT_CONSOLE) - logoutput.warning=&log_console; + else if(level == LOGLVL_WARNING) { + if(output == LOGOUTPUT_NULL) + logoutput.warning = &log_null; + else if(output == LOGOUTPUT_IRC) + logoutput.warning = &log_irc; + else if(output == LOGOUTPUT_SYSLOG) + logoutput.warning = &log_syslog; + else if(output == LOGOUTPUT_CONSOLE) + logoutput.warning = &log_console; } - else if(level==LOGLVL_ERROR) { - if(output==LOGOUTPUT_NULL) - logoutput.error=&log_null; - else if(output==LOGOUTPUT_IRC) - logoutput.error=&log_irc; - else if(output==LOGOUTPUT_SYSLOG) - logoutput.error=&log_syslog; - else if(output==LOGOUTPUT_CONSOLE) - logoutput.error=&log_console; + else if(level == LOGLVL_ERROR) { + if(output == LOGOUTPUT_NULL) + logoutput.error = &log_null; + else if(output == LOGOUTPUT_IRC) + logoutput.error = &log_irc; + else if(output == LOGOUTPUT_SYSLOG) + logoutput.error = &log_syslog; + else if(output == LOGOUTPUT_CONSOLE) + logoutput.error = &log_console; } #ifdef DEBUG - else if(level==LOGLVL_DEBUG) { - if(output==LOGOUTPUT_NULL) - logoutput.debug=&log_null; - else if(output==LOGOUTPUT_IRC) - logoutput.debug=&log_irc; - else if(output==LOGOUTPUT_SYSLOG) - logoutput.debug=&log_syslog; - else if(output==LOGOUTPUT_CONSOLE) - logoutput.debug=&log_console; + else if(level == LOGLVL_DEBUG) { + if(output == LOGOUTPUT_NULL) + logoutput.debug = &log_null; + else if(output == LOGOUTPUT_IRC) + logoutput.debug = &log_irc; + else if(output == LOGOUTPUT_SYSLOG) + logoutput.debug = &log_syslog; + else if(output == LOGOUTPUT_CONSOLE) + logoutput.debug = &log_console; } #endif return; @@ -105,14 +105,14 @@ void log_message(int level, char *message, ... ) { msgstring = g_strdup_vprintf(message, ap); va_end(ap); - if(level==LOGLVL_INFO) + if(level == LOGLVL_INFO) (*(logoutput.informational))(level, msgstring); - if(level==LOGLVL_WARNING) + if(level == LOGLVL_WARNING) (*(logoutput.warning))(level, msgstring); - if(level==LOGLVL_ERROR) + if(level == LOGLVL_ERROR) (*(logoutput.error))(level, msgstring); #ifdef DEBUG - if(level==LOGLVL_DEBUG) + if(level == LOGLVL_DEBUG) (*(logoutput.debug))(level, msgstring); #endif @@ -132,14 +132,14 @@ static void log_null(int level, char *message) { } static void log_irc(int level, char *message) { - if(level==LOGLVL_ERROR) + if(level == LOGLVL_ERROR) irc_write_all(1, "ERROR :Error: %s", message); - if(level==LOGLVL_WARNING) + if(level == LOGLVL_WARNING) irc_write_all(0, "ERROR :Warning: %s", message); - if(level==LOGLVL_INFO) + if(level == LOGLVL_INFO) irc_write_all(0, "ERROR :Informational: %s", message); #ifdef DEBUG - if(level==LOGLVL_DEBUG) + if(level == LOGLVL_DEBUG) irc_write_all(0, "ERROR :Debug: %s", message); #endif @@ -147,28 +147,28 @@ static void log_irc(int level, char *message) { } static void log_syslog(int level, char *message) { - if(level==LOGLVL_ERROR) + if(level == LOGLVL_ERROR) syslog(LOG_ERR, "%s", message); - if(level==LOGLVL_WARNING) + if(level == LOGLVL_WARNING) syslog(LOG_WARNING, "%s", message); - if(level==LOGLVL_INFO) + if(level == LOGLVL_INFO) syslog(LOG_INFO, "%s", message); #ifdef DEBUG - if(level==LOGLVL_DEBUG) + if(level == LOGLVL_DEBUG) syslog(LOG_DEBUG, "%s", message); #endif return; } static void log_console(int level, char *message) { - if(level==LOGLVL_ERROR) + if(level == LOGLVL_ERROR) fprintf(stderr, "Error: %s\n", message); - if(level==LOGLVL_WARNING) + if(level == LOGLVL_WARNING) fprintf(stderr, "Warning: %s\n", message); - if(level==LOGLVL_INFO) + if(level == LOGLVL_INFO) fprintf(stdout, "Informational: %s\n", message); #ifdef DEBUG - if(level==LOGLVL_DEBUG) + if(level == LOGLVL_DEBUG) fprintf(stdout, "Debug: %s\n", message); #endif return; |