]> arthur.barton.de Git - ngircd-alex.git/blobdiff - src/ngircd/conn.c
New function Simple_Message(). Better error reporting to clients on connect.
[ngircd-alex.git] / src / ngircd / conn.c
index 3975a920429dcb7b2bbf911bbd1b3272ffc1b3c9..882251bcd80f5abf8fe776ed152647649543e4b9 100644 (file)
@@ -1,6 +1,6 @@
 /*
  * ngIRCd -- The Next Generation IRC Daemon
- * Copyright (c)2001,2002 by Alexander Barton (alex@barton.de)
+ * Copyright (c)2001-2003 by Alexander Barton (alex@barton.de)
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -16,7 +16,7 @@
 
 #include "portab.h"
 
-static char UNUSED id[] = "$Id: conn.c,v 1.113 2002/12/30 17:14:59 alex Exp $";
+static char UNUSED id[] = "$Id: conn.c,v 1.119 2003/03/07 17:16:49 alex Exp $";
 
 #include "imp.h"
 #include <assert.h>
@@ -40,7 +40,11 @@ static char UNUSED id[] = "$Id: conn.c,v 1.113 2002/12/30 17:14:59 alex Exp $";
 #endif
 
 #ifdef HAVE_STDINT_H
-#include <stdint.h>                    /* u.a. fuer Mac OS X */
+#include <stdint.h>                    /* e.g. for Mac OS X */
+#endif
+
+#ifdef USE_TCPWRAP
+#include <tcpd.h>                      /* for TCP Wrappers */
 #endif
 
 #include "defines.h"
@@ -59,6 +63,10 @@ static char UNUSED id[] = "$Id: conn.c,v 1.113 2002/12/30 17:14:59 alex Exp $";
 #include "parse.h"
 #include "tool.h"
 
+#ifdef RENDEZVOUS
+#include "rendezvous.h"
+#endif
+
 #include "exp.h"
 
 
@@ -78,11 +86,17 @@ LOCAL VOID Init_Conn_Struct PARAMS(( CONN_ID Idx ));
 LOCAL BOOLEAN Init_Socket PARAMS(( INT Sock ));
 LOCAL VOID New_Server PARAMS(( INT Server, CONN_ID Idx ));
 LOCAL VOID Read_Resolver_Result PARAMS(( INT r_fd ));
+LOCAL VOID Simple_Message PARAMS(( INT Sock, CHAR *Msg ));
 
 LOCAL fd_set My_Listeners;
 LOCAL fd_set My_Sockets;
 LOCAL fd_set My_Connects;
 
+#ifdef USE_TCPWRAP
+INT allow_severity = LOG_INFO;
+INT deny_severity = LOG_ERR;
+#endif
+
 
 GLOBAL VOID
 Conn_Init( VOID )
@@ -132,8 +146,13 @@ Conn_Exit( VOID )
        CONN_ID idx;
        INT i;
 
-       /* Sockets schliessen */
        Log( LOG_DEBUG, "Shutting down all connections ..." );
+
+#ifdef RENDEZVOUS
+       Rendezvous_UnregisterListeners( );
+#endif
+       
+       /* Sockets schliessen */
        for( i = 0; i < Conn_MaxFD + 1; i++ )
        {
                if( FD_ISSET( i, &My_Sockets ))
@@ -195,6 +214,10 @@ Conn_ExitListeners( VOID )
 
        INT i;
 
+#ifdef RENDEZVOUS
+       Rendezvous_UnregisterListeners( );
+#endif
+       
        Log( LOG_INFO, "Shutting down all listening sockets ..." );
        for( i = 0; i < Conn_MaxFD + 1; i++ )
        {
@@ -214,7 +237,10 @@ Conn_NewListener( CONST UINT Port )
 
        struct sockaddr_in addr;
        INT sock;
-
+#ifdef RENDEZVOUS
+       CHAR name[CLIENT_ID_LEN], *info;
+#endif
+       
        /* Server-"Listen"-Socket initialisieren */
        memset( &addr, 0, sizeof( addr ));
        addr.sin_family = AF_INET;
@@ -255,6 +281,34 @@ Conn_NewListener( CONST UINT Port )
 
        Log( LOG_INFO, "Now listening on port %d (socket %d).", Port, sock );
 
+#ifdef RENDEZVOUS
+       /* Get best server description text */
+       if( ! Conf_ServerInfo[0] ) info = Conf_ServerName;
+       else
+       {
+               /* Use server info string */
+               info = NULL;
+               if( Conf_ServerInfo[0] == '[' )
+               {
+                       /* Cut off leading hostname part in "[]" */
+                       info = strchr( Conf_ServerInfo, ']' );
+                       if( info )
+                       {
+                               info++;
+                               while( *info == ' ' ) info++;
+                       }
+               }
+               if( ! info ) info = Conf_ServerInfo;
+       }
+
+       /* Add port number to description if non-standard */
+       if( Port != 6667 ) snprintf( name, sizeof( name ), "%s (port %u)", info, Port );
+       else strlcpy( name, info, sizeof( name ));
+
+       /* Register service */
+       Rendezvous_Register( name, RENDEZVOUS_TYPE, Port );
+#endif
+
        return TRUE;
 } /* Conn_NewListener */
 
@@ -285,6 +339,10 @@ Conn_Handler( VOID )
        {
                timeout = TRUE;
 
+#ifdef RENDEZVOUS
+               Rendezvous_Handler( );
+#endif
+
                /* Should the configuration be reloaded? */
                if( NGIRCd_SignalRehash ) NGIRCd_Rehash( );
 
@@ -529,6 +587,17 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
        assert( Idx > NONE );
        assert( My_Connections[Idx].sock > NONE );
 
+       /* Is this link already shutting down? */
+       if( My_Connections[Idx].options & CONN_ISCLOSING )
+       {
+               /* Conn_Close() has been called recursively for this link;
+                * probabe reason: Try_Write() failed  -- see below. */
+               return;
+       }
+
+       /* Mark link as "closing" */
+       My_Connections[Idx].options |= CONN_ISCLOSING;
+
        /* Search client, if any */
        c = Client_GetFromConn( Idx );
 
@@ -550,7 +619,7 @@ Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
        }
 
        /* Try to write out the write buffer */
-       Try_Write( Idx );
+       (VOID)Try_Write( Idx );
 
        /* Shut down socket */
        if( close( My_Connections[Idx].sock ) != 0 )
@@ -698,6 +767,7 @@ Handle_Write( CONN_ID Idx )
        /* Daten aus Schreibpuffer versenden bzw. Connection aufbauen */
 
        INT len, res, err;
+       CLIENT *c;
 
        assert( Idx > NONE );
        assert( My_Connections[Idx].sock > NONE );
@@ -721,8 +791,10 @@ Handle_Write( CONN_ID Idx )
                        if( res != 0 ) Log( LOG_CRIT, "getsockopt (connection %d): %s!", Idx, strerror( errno ));
                        else Log( LOG_CRIT, "Can't connect socket to \"%s:%d\" (connection %d): %s!", My_Connections[Idx].host, Conf_Server[Conf_GetServer( Idx )].port, Idx, strerror( err ));
 
-                       /* Socket etc. pp. aufraeumen */
+                       /* Clean up socket, connection and client structures */
                        FD_CLR( My_Connections[Idx].sock, &My_Sockets );
+                       c = Client_GetFromConn( Idx );
+                       if( c ) Client_DestroyNow( c );
                        close( My_Connections[Idx].sock );
                        Init_Conn_Struct( Idx );
 
@@ -774,6 +846,9 @@ New_Connection( INT Sock )
        /* Neue Client-Verbindung von Listen-Socket annehmen und
         * CLIENT-Struktur anlegen. */
 
+#ifdef USE_TCPWRAP
+       struct request_info req;
+#endif
        struct sockaddr_in new_addr;
        INT new_sock, new_sock_len;
        RES_STAT *s;
@@ -792,6 +867,19 @@ New_Connection( INT Sock )
                Log( LOG_CRIT, "Can't accept connection: %s!", strerror( errno ));
                return;
        }
+       
+#ifdef USE_TCPWRAP
+       /* Validate socket using TCP Wrappers */
+       request_init( &req, RQ_DAEMON, PACKAGE, RQ_FILE, new_sock, RQ_CLIENT_SIN, &new_addr, NULL );
+       if( ! hosts_access( &req ))
+       {
+               /* Access denied! */
+               Log( deny_severity, "Refused connection from %s (by TCP Wrappers)!", inet_ntoa( new_addr.sin_addr ));
+               Simple_Message( new_sock, "ERROR :Connection refused" );
+               close( new_sock );
+               return;
+       }
+#endif
 
        /* Socket initialisieren */
        Init_Socket( new_sock );
@@ -812,6 +900,7 @@ New_Connection( INT Sock )
                        {
                                /* Mehr Verbindungen duerfen wir leider nicht mehr annehmen ... */
                                Log( LOG_ALERT, "Can't accept connection: limit (%d) reached!", Pool_Size );
+                               Simple_Message( new_sock, "ERROR :Connection limit reached" );
                                close( new_sock );
                                return;
                        }
@@ -820,6 +909,7 @@ New_Connection( INT Sock )
                if( new_size < Pool_Size )
                {
                        Log( LOG_ALERT, "Can't accespt connection: limit (%d) reached -- overflow!", Pool_Size );
+                       Simple_Message( new_sock, "ERROR :Connection limit reached" );
                        close( new_sock );
                        return;
                }
@@ -836,6 +926,7 @@ New_Connection( INT Sock )
                        {
                                /* Offenbar steht kein weiterer Sepeicher zur Verfuegung :-( */
                                Log( LOG_EMERG, "Can't allocate memory! [New_Connection]" );
+                               Simple_Message( new_sock, "ERROR: Internal error" );
                                close( new_sock );
                                return;
                        }
@@ -863,6 +954,7 @@ New_Connection( INT Sock )
        if( ! c )
        {
                Log( LOG_ALERT, "Can't accept connection: can't create client structure!" );
+               Simple_Message( new_sock, "ERROR :Internal error" );
                close( new_sock );
                return;
        }
@@ -1162,9 +1254,6 @@ Check_Servers( VOID )
        CONN_ID idx;
        INT i, n;
 
-       /* Don't connect in "passive mode" */
-       if( NGIRCd_Passive ) return;
-
        /* Serach all connections, are there results from the resolver? */
        for( idx = 0; idx < Pool_Size; idx++ )
        {
@@ -1177,8 +1266,8 @@ Check_Servers( VOID )
        /* Check all configured servers */
        for( i = 0; i < MAX_SERVERS; i++ )
        {
-               /* Valid outgoing server which isn't already connected? */
-               if(( ! Conf_Server[i].host[0] ) || ( ! Conf_Server[i].port > 0 ) || ( Conf_Server[i].conn_id > NONE )) continue;
+               /* Valid outgoing server which isn't already connected or disabled? */
+               if(( ! Conf_Server[i].host[0] ) || ( ! Conf_Server[i].port > 0 ) || ( Conf_Server[i].conn_id > NONE ) || ( Conf_Server[i].flags & CONF_SFLAG_DISABLED )) continue;
 
                /* Is there already a connection in this group? */
                if( Conf_Server[i].group > NONE )
@@ -1438,4 +1527,19 @@ Read_Resolver_Result( INT r_fd )
 } /* Read_Resolver_Result */
 
 
+LOCAL VOID
+Simple_Message( INT Sock, CHAR *Msg )
+{
+       /* Write "simple" message to socket, without using compression
+        * or even the connection write buffers. Used e.g. for error
+        * messages by New_Connection(). */
+
+       assert( Sock > NONE );
+       assert( Msg != NULL );
+
+       (VOID)send( Sock, Msg, strlen( Msg ), 0 );
+       (VOID)send( Sock, "\r\n", 2, 0 );
+} /* Simple_Error */
+
+
 /* -eof- */