]> arthur.barton.de Git - ngircd-alex.git/blobdiff - src/ngircd/conn.c
- bessere Dokumentation des Modules bzw. der Funktionen.
[ngircd-alex.git] / src / ngircd / conn.c
index dccdb6a606cae7186a090345262b9f153a825344..d2479db52a8310eca4961e89a938ae61f5358f94 100644 (file)
@@ -9,11 +9,24 @@
  * Naehere Informationen entnehmen Sie bitter der Datei COPYING. Eine Liste
  * der an comBase beteiligten Autoren finden Sie in der Datei AUTHORS.
  *
- * $Id: conn.c,v 1.17 2001/12/29 03:06:16 alex Exp $
+ * $Id: conn.c,v 1.21 2001/12/29 22:33:36 alex Exp $
  *
  * connect.h: Verwaltung aller Netz-Verbindungen ("connections")
  *
  * $Log: conn.c,v $
+ * Revision 1.21  2001/12/29 22:33:36  alex
+ * - bessere Dokumentation des Modules bzw. der Funktionen.
+ *
+ * Revision 1.20  2001/12/29 22:09:43  alex
+ * - kleinere Aenderungen ("clean-ups") bei Logging (Resolver).
+ *
+ * Revision 1.19  2001/12/29 21:53:57  alex
+ * - Da hatte ich mich wohl ein wenig verrannt; jetzt sollte der Resolver
+ *   aber tatsaechlich funktionieren.
+ *
+ * Revision 1.18  2001/12/29 20:17:25  alex
+ * - asyncronen Resolver (IP->Name) implementiert, dadurch div. Aenderungen.
+ *
  * Revision 1.17  2001/12/29 03:06:16  alex
  * - Loglevel (nochmal) angepasst.
  *
 #include <time.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
+#include <netdb.h>
 
 #ifdef HAVE_STDINT_H
 #include <stdint.h>                    /* u.a. fuer Mac OS X */
 #define READBUFFER_LEN 2 * MAX_CMDLEN  /* Laenge des Lesepuffers je Verbindung (Bytes) */
 #define WRITEBUFFER_LEN 4096           /* Laenge des Schreibpuffers je Verbindung (Bytes) */
 
+#define HOST_LEN 256                   /* max. Laenge eines Hostnamen */
+
+
+typedef struct _Res_Stat
+{
+       INT pid;                        /* PID des Child-Prozess */
+       INT pipe[2];                    /* Pipe fuer IPC */
+} RES_STAT;
+
 
 typedef struct _Connection
 {
        INT sock;                       /* Socket Handle */
        struct sockaddr_in addr;        /* Adresse des Client */
+       RES_STAT *res_stat;             /* "Resolver-Status", s.o. */
+       CHAR host[HOST_LEN];            /* Hostname */
        CHAR rbuf[READBUFFER_LEN + 1];  /* Lesepuffer */
        INT rdatalen;                   /* Laenge der Daten im Lesepuffer */
        CHAR wbuf[WRITEBUFFER_LEN + 1]; /* Schreibpuffer */
@@ -138,10 +163,16 @@ LOCAL VOID Read_Request( CONN_ID Idx );
 LOCAL BOOLEAN Try_Write( CONN_ID Idx );
 LOCAL VOID Handle_Buffer( CONN_ID Idx );
 LOCAL VOID Check_Connections( VOID );
+LOCAL VOID Init_Conn_Struct( INT Idx );
+
+LOCAL RES_STAT *Resolve( struct sockaddr_in *Addr );
+LOCAL VOID Read_Resolver_Result( INT r_fd );
+LOCAL VOID Do_Resolve( struct sockaddr_in *Addr, INT w_fd );
 
 
-LOCAL fd_set My_Listener;
+LOCAL fd_set My_Listeners;
 LOCAL fd_set My_Sockets;
+LOCAL fd_set My_Resolvers;
 
 LOCAL INT My_Max_Fd;
 
@@ -150,21 +181,27 @@ LOCAL CONNECTION My_Connections[MAX_CONNECTIONS];
 
 GLOBAL VOID Conn_Init( VOID )
 {
+       /* Modul initialisieren: statische Strukturen "ausnullen". */
+
        CONN_ID i;
 
        /* zu Beginn haben wir keine Verbindungen */
-       FD_ZERO( &My_Listener );
+       FD_ZERO( &My_Listeners );
        FD_ZERO( &My_Sockets );
+       FD_ZERO( &My_Resolvers );
 
        My_Max_Fd = 0;
 
        /* Connection-Struktur initialisieren */
-       for( i = 0; i < MAX_CONNECTIONS; i++ ) My_Connections[i].sock = NONE;
+       for( i = 0; i < MAX_CONNECTIONS; i++ ) Init_Conn_Struct( i );
 } /* Conn_Init */
 
 
 GLOBAL VOID Conn_Exit( VOID )
 {
+       /* Modul abmelden: alle noch offenen Connections
+        * schliessen und freigeben. */
+
        CONN_ID idx;
        INT i;
 
@@ -178,7 +215,7 @@ GLOBAL VOID Conn_Exit( VOID )
                                if( My_Connections[idx].sock == i ) break;
                        }
                        if( idx < MAX_CONNECTIONS ) Conn_Close( idx, "Server going down ..." );
-                       else if( FD_ISSET( i, &My_Listener ))
+                       else if( FD_ISSET( i, &My_Listeners ))
                        {
                                close( i );
                                Log( LOG_INFO, "Listening socket %d closed.", i );
@@ -195,8 +232,9 @@ GLOBAL VOID Conn_Exit( VOID )
 
 GLOBAL BOOLEAN Conn_NewListener( CONST INT Port )
 {
-       /* Neuen Listen-Socket erzeugen: der Server wartet dann
-        * auf dem angegebenen Port auf Verbindungen. */
+       /* Neuen Listen-Socket erzeugen: der Server wartet dann auf
+        * dem angegebenen Port auf Verbindungen. Kann der Listen-
+        * Socket nicht erteugt werden, so wird NULL geliefert.*/
 
        struct sockaddr_in addr;
        INT sock, on = 1;
@@ -245,7 +283,7 @@ GLOBAL BOOLEAN Conn_NewListener( CONST INT Port )
        }
 
        /* Neuen Listener in Strukturen einfuegen */
-       FD_SET( sock, &My_Listener );
+       FD_SET( sock, &My_Listeners );
        FD_SET( sock, &My_Sockets );
 
        if( sock > My_Max_Fd ) My_Max_Fd = sock;
@@ -258,6 +296,16 @@ GLOBAL BOOLEAN Conn_NewListener( CONST INT Port )
 
 GLOBAL VOID Conn_Handler( INT Timeout )
 {
+       /* Aktive Verbindungen ueberwachen. Mindestens alle "Timeout"
+        * Sekunden wird die Funktion verlassen. Folgende Aktionen
+        * werden durchgefuehrt:
+        *  - neue Verbindungen annehmen,
+        *  - geschlossene Verbindungen loeschen,
+        *  - volle Schreibpuffer versuchen zu schreiben,
+        *  - volle Lesepuffer versuchen zu verarbeiten,
+        *  - Antworten von Resolver Sub-Prozessen annehmen.
+        */
+
        fd_set read_sockets, write_sockets;
        struct timeval tv;
        time_t start;
@@ -292,8 +340,27 @@ GLOBAL VOID Conn_Handler( INT Timeout )
                                FD_SET( My_Connections[i].sock, &write_sockets );
                        }
                }
-               
+
+               /* von welchen Sockets koennte gelesen werden? */
                read_sockets = My_Sockets;
+               for( i = 0; i < MAX_CONNECTIONS; i++ )
+               {
+                       if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].host[0] == '\0' ))
+                       {
+                               /* Hier muss noch auf den Resolver Sub-Prozess gewartet werden */
+                               FD_CLR( My_Connections[i].sock, &read_sockets );
+                       }
+               }
+               for( i = 0; i < My_Max_Fd + 1; i++ )
+               {
+                       /* Pipes von Resolver Sub-Prozessen aufnehmen */
+                       if( FD_ISSET( i, &My_Resolvers ))
+                       {
+                               FD_SET( i, &read_sockets );
+                       }
+               }
+
+               /* Auf Aktivitaet warten */
                if( select( My_Max_Fd + 1, &read_sockets, &write_sockets, NULL, &tv ) == -1 )
                {
                        if( errno != EINTR ) Log( LOG_ALERT, "select(): %s!", strerror( errno ));
@@ -387,7 +454,8 @@ GLOBAL BOOLEAN Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
 
 GLOBAL VOID Conn_Close( CONN_ID Idx, CHAR *Msg )
 {
-       /* Verbindung schliessen */
+       /* Verbindung schliessen. Evtl. noch von Resolver
+        * Sub-Prozessen offene Pipes werden geschlossen. */
 
        assert( Idx >= 0 );
        assert( My_Connections[Idx].sock >= 0 );
@@ -406,6 +474,15 @@ GLOBAL VOID Conn_Close( CONN_ID Idx, CHAR *Msg )
 
        Client_Destroy( Client_GetFromConn( Idx ));
 
+       if( My_Connections[Idx].res_stat )
+       {
+               /* Resolver-Strukturen freigeben, wenn noch nicht geschehen */
+               FD_CLR( My_Connections[Idx].res_stat->pipe[0], &My_Resolvers );
+               close( My_Connections[Idx].res_stat->pipe[0] );
+               close( My_Connections[Idx].res_stat->pipe[1] );
+               free( My_Connections[Idx].res_stat );
+       }
+       
        FD_CLR( My_Connections[Idx].sock, &My_Sockets );
        My_Connections[Idx].sock = NONE;
 } /* Conn_Close */
@@ -442,19 +519,28 @@ LOCAL BOOLEAN Try_Write( CONN_ID Idx )
 
 LOCAL VOID Handle_Read( INT Sock )
 {
-       /* Aktivitaet auf einem Socket verarbeiten */
+       /* Aktivitaet auf einem Socket verarbeiten:
+        *  - neue Clients annehmen,
+        *  - Daten von Clients verarbeiten,
+        *  - Resolver-Rueckmeldungen annehmen. */
 
        CONN_ID idx;
 
        assert( Sock >= 0 );
-
-       if( FD_ISSET( Sock, &My_Listener ))
+       
+       if( FD_ISSET( Sock, &My_Listeners ))
        {
                /* es ist einer unserer Listener-Sockets: es soll
                 * also eine neue Verbindung aufgebaut werden. */
 
                New_Connection( Sock );
        }
+       else if( FD_ISSET( Sock, &My_Resolvers ))
+       {
+               /* Rueckmeldung von einem Resolver Sub-Prozess */
+
+               Read_Resolver_Result( Sock );
+       }
        else
        {
                /* Ein Client Socket: entweder ein User oder Server */
@@ -500,6 +586,7 @@ LOCAL VOID New_Connection( INT Sock )
 
        struct sockaddr_in new_addr;
        INT new_sock, new_sock_len;
+       RES_STAT *s;
        CONN_ID idx;
 
        assert( Sock >= 0 );
@@ -530,19 +617,28 @@ LOCAL VOID New_Connection( INT Sock )
        }
 
        /* Verbindung registrieren */
+       Init_Conn_Struct( idx );
        My_Connections[idx].sock = new_sock;
        My_Connections[idx].addr = new_addr;
-       My_Connections[idx].rdatalen = 0;
-       My_Connections[idx].wdatalen = 0;
-       My_Connections[idx].lastdata = time( NULL );
-       My_Connections[idx].lastping = 0;
 
        /* Neuen Socket registrieren */
        FD_SET( new_sock, &My_Sockets );
-
        if( new_sock > My_Max_Fd ) My_Max_Fd = new_sock;
 
        Log( LOG_NOTICE, "Accepted connection %d from %s:%d on socket %d.", idx, inet_ntoa( new_addr.sin_addr ), ntohs( new_addr.sin_port), Sock );
+
+       /* Hostnamen ermitteln */
+       s = Resolve( &new_addr );
+       if( s )
+       {
+               /* Sub-Prozess wurde asyncron gestartet */
+               My_Connections[idx].res_stat = s;
+       }
+       else
+       {
+               /* kann Namen nicht aufloesen */
+               strcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr ));
+       }
 } /* New_Connection */
 
 
@@ -614,6 +710,8 @@ LOCAL VOID Read_Request( CONN_ID Idx )
 
 LOCAL VOID Handle_Buffer( CONN_ID Idx )
 {
+       /* Daten im Lese-Puffer einer Verbindung verarbeiten. */
+
        CHAR *ptr, *ptr1, *ptr2;
        INT len, delta;
        
@@ -654,7 +752,9 @@ LOCAL VOID Handle_Buffer( CONN_ID Idx )
 
 LOCAL VOID Check_Connections( VOID )
 {
-       /* Pruefen, ob Verbindungen noch "alive" sind */
+       /* Pruefen, ob Verbindungen noch "alive" sind. Ist dies
+        * nicht der Fall, zunaechst PING-PONG spielen und, wenn
+        * auch das nicht "hilft", Client disconnectieren. */
 
        INT i;
 
@@ -684,4 +784,152 @@ LOCAL VOID Check_Connections( VOID )
 } /* Conn_Check */
 
 
+LOCAL VOID Init_Conn_Struct( INT Idx )
+{
+       /* Connection-Struktur initialisieren */
+
+       My_Connections[Idx].sock = NONE;
+       My_Connections[Idx].res_stat = NULL;
+       My_Connections[Idx].host[0] = '\0';
+       My_Connections[Idx].rbuf[0] = '\0';
+       My_Connections[Idx].rdatalen = 0;
+       My_Connections[Idx].wbuf[0] = '\0';
+       My_Connections[Idx].wdatalen = 0;
+       My_Connections[Idx].lastdata = time( NULL );
+       My_Connections[Idx].lastping = 0;
+} /* Init_Conn_Struct */
+
+
+LOCAL RES_STAT *Resolve( struct sockaddr_in *Addr )
+{
+       /* Hostnamen (asyncron!) aufloesen. Bei Fehler, z.B. wenn der
+        * Child-Prozess nicht erzeugt werden kann, wird NULL geliefert.
+        * Der Host kann dann nicht aufgeloest werden. */
+
+       RES_STAT *s;
+       INT pid;
+
+       /* Speicher anfordern */
+       s = malloc( sizeof( RES_STAT ));
+       if( ! s )
+       {
+               Log( LOG_ALERT, "Resolver: Can't alloc memory!" );
+               return NULL;
+       }
+
+       /* Pipe fuer Antwort initialisieren */
+       if( pipe( s->pipe ) != 0 )
+       {
+               free( s );
+               Log( LOG_ALERT, "Resolver: Can't create output pipe: %s!", strerror( errno ));
+               return NULL;
+       }
+
+       /* Sub-Prozess erzeugen */
+       pid = fork( );
+       if( pid > 0 )
+       {
+               /* Haupt-Prozess */
+               Log( LOG_DEBUG, "Resolver for %s created (PID %d).", inet_ntoa( Addr->sin_addr ), pid );
+               FD_SET( s->pipe[0], &My_Resolvers );
+               if( s->pipe[0] > My_Max_Fd ) My_Max_Fd = s->pipe[0];
+               s->pid = pid;
+               return s;
+       }
+       else if( pid == 0 )
+       {
+               /* Sub-Prozess */
+               Log_Init_Resolver( );
+               Do_Resolve( Addr, s->pipe[1] );
+               Log_Exit_Resolver( );
+               exit( 0 );
+       }
+       else
+       {
+               /* Fehler */
+               free( s );
+               Log( LOG_ALERT, "Resolver: Can't fork: %s!", strerror( errno ));
+               return NULL;
+       }
+} /* Resolve */
+
+
+LOCAL VOID Read_Resolver_Result( INT r_fd )
+{
+       /* Ergebnis von Resolver Sub-Prozess aus Pipe lesen
+        * und entsprechende Connection aktualisieren */
+
+       CHAR hostname[HOST_LEN];
+       CLIENT *c;
+       INT i;
+
+       FD_CLR( r_fd, &My_Resolvers );
+
+       /* Anfrage vom Parent lesen */
+       if( read( r_fd, hostname, HOST_LEN) < 0 )
+       {
+               /* Fehler beim Lesen aus der Pipe */
+               close( r_fd );
+               Log( LOG_ALERT, "Resolver: Can't read result: %s!", strerror( errno ));
+               return;
+       }
+
+       /* zugehoerige Connection suchen */
+       for( i = 0; i < MAX_CONNECTIONS; i++ )
+       {
+               if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].res_stat ) && ( My_Connections[i].res_stat->pipe[0] == r_fd )) break;
+       }
+
+       if( i >= MAX_CONNECTIONS )
+       {
+               /* Opsa! Keine passende Connection gefunden!? */
+               close( r_fd );
+               Log( LOG_ALERT, "Resolver: Got result for unknown connection!?" );
+               return;
+       }
+
+       /* Aufraeumen */
+       close( My_Connections[i].res_stat->pipe[0] );
+       close( My_Connections[i].res_stat->pipe[1] );
+       free( My_Connections[i].res_stat );
+       My_Connections[i].res_stat = NULL;
+       
+       /* Hostnamen setzen */
+       strcpy( My_Connections[i].host, hostname );
+       c = Client_GetFromConn( i );
+       if( c ) Client_SetHostname( c, hostname );
+} /* Read_Resolver_Result */
+
+
+LOCAL VOID Do_Resolve( struct sockaddr_in *Addr, INT w_fd )
+{
+       /* Resolver Sub-Prozess: aufzuloesenden Namen aus
+        * der Pipe lesen, Ergebnis in Pipe schreiben. */
+
+       CHAR hostname[HOST_LEN];
+       struct hostent *h;
+
+       Log_Resolver( LOG_DEBUG, "Now resolving %s ...", inet_ntoa( Addr->sin_addr ));
+
+       /* Namen aufloesen */
+       h = gethostbyaddr( (CHAR *)&Addr->sin_addr, sizeof( Addr->sin_addr ), AF_INET );
+       if( h ) strcpy( hostname, h->h_name );
+       else
+       {
+               Log_Resolver( LOG_WARNING, "Resolver: Can't resolve host name (code %d)!", h_errno );
+               strcpy( hostname, inet_ntoa( Addr->sin_addr ));
+       }
+
+       /* Antwort an Parent schreiben */
+       if( write( w_fd, hostname, strlen( hostname ) + 1 ) != ( strlen( hostname ) + 1 ))
+       {
+               Log_Resolver( LOG_ALERT, "Resolver: Can't write to parent: %s!", strerror( errno ));
+               close( w_fd );
+               return;
+       }
+
+       Log_Resolver( LOG_DEBUG, "Ok, translated %s to \"%s\".", inet_ntoa( Addr->sin_addr ), hostname );
+} /* Do_Resolve */
+
+
 /* -eof- */