- bessere Dokumentation des Modules bzw. der Funktionen.
[ngircd-alex.git] / src / ngircd / conn.c
1 /*
2  * ngIRCd -- The Next Generation IRC Daemon
3  * Copyright (c)2001 by Alexander Barton (alex@barton.de)
4  *
5  * Dieses Programm ist freie Software. Sie koennen es unter den Bedingungen
6  * der GNU General Public License (GPL), wie von der Free Software Foundation
7  * herausgegeben, weitergeben und/oder modifizieren, entweder unter Version 2
8  * der Lizenz oder (wenn Sie es wuenschen) jeder spaeteren Version.
9  * Naehere Informationen entnehmen Sie bitter der Datei COPYING. Eine Liste
10  * der an comBase beteiligten Autoren finden Sie in der Datei AUTHORS.
11  *
12  * $Id: conn.c,v 1.21 2001/12/29 22:33:36 alex Exp $
13  *
14  * connect.h: Verwaltung aller Netz-Verbindungen ("connections")
15  *
16  * $Log: conn.c,v $
17  * Revision 1.21  2001/12/29 22:33:36  alex
18  * - bessere Dokumentation des Modules bzw. der Funktionen.
19  *
20  * Revision 1.20  2001/12/29 22:09:43  alex
21  * - kleinere Aenderungen ("clean-ups") bei Logging (Resolver).
22  *
23  * Revision 1.19  2001/12/29 21:53:57  alex
24  * - Da hatte ich mich wohl ein wenig verrannt; jetzt sollte der Resolver
25  *   aber tatsaechlich funktionieren.
26  *
27  * Revision 1.18  2001/12/29 20:17:25  alex
28  * - asyncronen Resolver (IP->Name) implementiert, dadurch div. Aenderungen.
29  *
30  * Revision 1.17  2001/12/29 03:06:16  alex
31  * - Loglevel (nochmal) angepasst.
32  *
33  * Revision 1.16  2001/12/27 19:32:44  alex
34  * - bei "Null-Requests" wird nichts mehr geloggt. Uberfluessig, da normal.
35  *
36  * Revision 1.15  2001/12/27 16:35:04  alex
37  * - vergessene Variable bei Ping-Timeout-Logmeldung ergaenzt. Opsa.
38  *
39  * Revision 1.14  2001/12/26 14:45:37  alex
40  * - "Code Cleanups".
41  *
42  * Revision 1.13  2001/12/26 03:36:57  alex
43  * - Verbindungen mit Lesefehlern werden nun korrekt terminiert.
44  *
45  * Revision 1.12  2001/12/26 03:20:53  alex
46  * - PING/PONG-Timeout implementiert.
47  *
48  * Revision 1.11  2001/12/25 23:15:16  alex
49  * - buffer werden nun periodisch geprueft, keine haengenden Clients mehr.
50  *
51  * Revision 1.10  2001/12/25 22:03:47  alex
52  * - Conn_Close() eingefuehrt: war die lokale Funktion Close_Connection().
53  *
54  * Revision 1.9  2001/12/24 01:32:33  alex
55  * - in Conn_WriteStr() wurde das CR+LF nicht angehaengt!
56  * - Fehler-Ausgaben vereinheitlicht.
57  *
58  * Revision 1.8  2001/12/23 22:02:54  alex
59  * - Conn_WriteStr() nimmt nun variable Parameter,
60  * - diverse kleinere Aenderungen.
61  *
62  * Revision 1.7  2001/12/21 22:24:25  alex
63  * - kleinere Aenderungen an den Log-Meldungen,
64  * - Parse_Request() wird aufgerufen.
65  *
66  * Revision 1.6  2001/12/15 00:11:55  alex
67  * - Lese- und Schreib-Puffer implementiert.
68  * - einige neue (Unter-)Funktionen eingefuehrt.
69  * - diverse weitere kleinere Aenderungen.
70  *
71  * Revision 1.5  2001/12/14 08:16:47  alex
72  * - Begonnen, Client-spezifische Lesepuffer zu implementieren.
73  * - Umstellung auf Datentyp "CONN_ID".
74  *
75  * Revision 1.4  2001/12/13 02:04:16  alex
76  * - boesen "Speicherschiesser" in Log() gefixt.
77  *
78  * Revision 1.3  2001/12/13 01:33:09  alex
79  * - Conn_Handler() unterstuetzt nun einen Timeout.
80  * - fuer Verbindungen werden keine FILE-Handles mehr benutzt.
81  * - kleinere "Code Cleanups" ;-)
82  *
83  * Revision 1.2  2001/12/12 23:32:02  alex
84  * - diverse Erweiterungen und Verbesserungen (u.a. sind nun mehrere
85  *   Verbindungen und Listen-Sockets moeglich).
86  *
87  * Revision 1.1  2001/12/12 17:18:38  alex
88  * - Modul zur Verwaltung aller Netzwerk-Verbindungen begonnen.
89  */
90
91
92 #include <portab.h>
93 #include "global.h"
94
95 #include <imp.h>
96 #include <assert.h>
97 #include <stdarg.h>
98 #include <stdio.h>
99 #include <unistd.h>
100 #include <errno.h>
101 #include <fcntl.h>
102 #include <string.h>
103 #include <sys/socket.h>
104 #include <sys/time.h>
105 #include <sys/types.h>
106 #include <time.h>
107 #include <netinet/in.h>
108 #include <arpa/inet.h>
109 #include <netdb.h>
110
111 #ifdef HAVE_STDINT_H
112 #include <stdint.h>                     /* u.a. fuer Mac OS X */
113 #endif
114
115 #include "ngircd.h"
116 #include "client.h"
117 #include "conf.h"
118 #include "log.h"
119 #include "parse.h"
120 #include "tool.h"
121
122 #include <exp.h>
123 #include "conn.h"
124
125
126 #define MAX_CONNECTIONS 100             /* max. Anzahl von Verbindungen an diesem Server */
127
128 #define MAX_CMDLEN 512                  /* max. Laenge eines Befehls, vgl. RFC 2812, 3.2 */
129
130 #define READBUFFER_LEN 2 * MAX_CMDLEN   /* Laenge des Lesepuffers je Verbindung (Bytes) */
131 #define WRITEBUFFER_LEN 4096            /* Laenge des Schreibpuffers je Verbindung (Bytes) */
132
133 #define HOST_LEN 256                    /* max. Laenge eines Hostnamen */
134
135
136 typedef struct _Res_Stat
137 {
138         INT pid;                        /* PID des Child-Prozess */
139         INT pipe[2];                    /* Pipe fuer IPC */
140 } RES_STAT;
141
142
143 typedef struct _Connection
144 {
145         INT sock;                       /* Socket Handle */
146         struct sockaddr_in addr;        /* Adresse des Client */
147         RES_STAT *res_stat;             /* "Resolver-Status", s.o. */
148         CHAR host[HOST_LEN];            /* Hostname */
149         CHAR rbuf[READBUFFER_LEN + 1];  /* Lesepuffer */
150         INT rdatalen;                   /* Laenge der Daten im Lesepuffer */
151         CHAR wbuf[WRITEBUFFER_LEN + 1]; /* Schreibpuffer */
152         INT wdatalen;                   /* Laenge der Daten im Schreibpuffer */
153         time_t lastdata;                /* Letzte Aktivitaet */
154         time_t lastping;                /* Letzter PING */
155 } CONNECTION;
156
157
158 LOCAL VOID Handle_Read( INT sock );
159 LOCAL BOOLEAN Handle_Write( CONN_ID Idx );
160 LOCAL VOID New_Connection( INT Sock );
161 LOCAL CONN_ID Socket2Index( INT Sock );
162 LOCAL VOID Read_Request( CONN_ID Idx );
163 LOCAL BOOLEAN Try_Write( CONN_ID Idx );
164 LOCAL VOID Handle_Buffer( CONN_ID Idx );
165 LOCAL VOID Check_Connections( VOID );
166 LOCAL VOID Init_Conn_Struct( INT Idx );
167
168 LOCAL RES_STAT *Resolve( struct sockaddr_in *Addr );
169 LOCAL VOID Read_Resolver_Result( INT r_fd );
170 LOCAL VOID Do_Resolve( struct sockaddr_in *Addr, INT w_fd );
171
172
173 LOCAL fd_set My_Listeners;
174 LOCAL fd_set My_Sockets;
175 LOCAL fd_set My_Resolvers;
176
177 LOCAL INT My_Max_Fd;
178
179 LOCAL CONNECTION My_Connections[MAX_CONNECTIONS];
180
181
182 GLOBAL VOID Conn_Init( VOID )
183 {
184         /* Modul initialisieren: statische Strukturen "ausnullen". */
185
186         CONN_ID i;
187
188         /* zu Beginn haben wir keine Verbindungen */
189         FD_ZERO( &My_Listeners );
190         FD_ZERO( &My_Sockets );
191         FD_ZERO( &My_Resolvers );
192
193         My_Max_Fd = 0;
194
195         /* Connection-Struktur initialisieren */
196         for( i = 0; i < MAX_CONNECTIONS; i++ ) Init_Conn_Struct( i );
197 } /* Conn_Init */
198
199
200 GLOBAL VOID Conn_Exit( VOID )
201 {
202         /* Modul abmelden: alle noch offenen Connections
203          * schliessen und freigeben. */
204
205         CONN_ID idx;
206         INT i;
207
208         /* Sockets schliessen */
209         for( i = 0; i < My_Max_Fd + 1; i++ )
210         {
211                 if( FD_ISSET( i, &My_Sockets ))
212                 {
213                         for( idx = 0; idx < MAX_CONNECTIONS; idx++ )
214                         {
215                                 if( My_Connections[idx].sock == i ) break;
216                         }
217                         if( idx < MAX_CONNECTIONS ) Conn_Close( idx, "Server going down ..." );
218                         else if( FD_ISSET( i, &My_Listeners ))
219                         {
220                                 close( i );
221                                 Log( LOG_INFO, "Listening socket %d closed.", i );
222                         }
223                         else
224                         {
225                                 close( i );
226                                 Log( LOG_WARNING, "Unknown connection %d closed.", i );
227                         }
228                 }
229         }
230 } /* Conn_Exit */
231
232
233 GLOBAL BOOLEAN Conn_NewListener( CONST INT Port )
234 {
235         /* Neuen Listen-Socket erzeugen: der Server wartet dann auf
236          * dem angegebenen Port auf Verbindungen. Kann der Listen-
237          * Socket nicht erteugt werden, so wird NULL geliefert.*/
238
239         struct sockaddr_in addr;
240         INT sock, on = 1;
241
242         /* Server-"Listen"-Socket initialisieren */
243         memset( &addr, 0, sizeof( addr ));
244         addr.sin_family = AF_INET;
245         addr.sin_port = htons( Port );
246         addr.sin_addr.s_addr = htonl( INADDR_ANY );
247
248         /* Socket erzeugen */
249         sock = socket( PF_INET, SOCK_STREAM, 0);
250         if( sock < 0 )
251         {
252                 Log( LOG_ALERT, "Can't create socket: %s!", strerror( errno ));
253                 return FALSE;
254         }
255
256         /* Socket-Optionen setzen */
257         if( fcntl( sock, F_SETFL, O_NONBLOCK ) != 0 )
258         {
259                 Log( LOG_ALERT, "Can't enable non-blocking mode: %s!", strerror( errno ));
260                 close( sock );
261                 return FALSE;
262         }
263         if( setsockopt( sock, SOL_SOCKET, SO_REUSEADDR, &on, (socklen_t)sizeof( on )) != 0)
264         {
265                 Log( LOG_CRIT, "Can't set socket options: %s!", strerror( errno ));
266                 /* dieser Fehler kann ignoriert werden. */
267         }
268
269         /* an Port binden */
270         if( bind( sock, (struct sockaddr *)&addr, (socklen_t)sizeof( addr )) != 0 )
271         {
272                 Log( LOG_ALERT, "Can't bind socket: %s!", strerror( errno ));
273                 close( sock );
274                 return FALSE;
275         }
276
277         /* in "listen mode" gehen :-) */
278         if( listen( sock, 10 ) != 0 )
279         {
280                 Log( LOG_ALERT, "Can't listen on soecket: %s!", strerror( errno ));
281                 close( sock );
282                 return FALSE;
283         }
284
285         /* Neuen Listener in Strukturen einfuegen */
286         FD_SET( sock, &My_Listeners );
287         FD_SET( sock, &My_Sockets );
288
289         if( sock > My_Max_Fd ) My_Max_Fd = sock;
290
291         Log( LOG_INFO, "Now listening on port %d, socket %d.", Port, sock );
292
293         return TRUE;
294 } /* Conn_NewListener */
295
296
297 GLOBAL VOID Conn_Handler( INT Timeout )
298 {
299         /* Aktive Verbindungen ueberwachen. Mindestens alle "Timeout"
300          * Sekunden wird die Funktion verlassen. Folgende Aktionen
301          * werden durchgefuehrt:
302          *  - neue Verbindungen annehmen,
303          *  - geschlossene Verbindungen loeschen,
304          *  - volle Schreibpuffer versuchen zu schreiben,
305          *  - volle Lesepuffer versuchen zu verarbeiten,
306          *  - Antworten von Resolver Sub-Prozessen annehmen.
307          */
308
309         fd_set read_sockets, write_sockets;
310         struct timeval tv;
311         time_t start;
312         INT i;
313
314         start = time( NULL );
315         while(( time( NULL ) - start < Timeout ) && ( ! NGIRCd_Quit ))
316         {
317                 Check_Connections( );
318
319                 /* Timeout initialisieren */
320                 tv.tv_sec = 0;
321                 tv.tv_usec = 50000;
322
323                 /* noch volle Lese-Buffer suchen */
324                 for( i = 0; i < MAX_CONNECTIONS; i++ )
325                 {
326                         if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].rdatalen > 0 ))
327                         {
328                                 /* Kann aus dem Buffer noch ein Befehl extrahiert werden? */
329                                 Handle_Buffer( i );
330                         }
331                 }
332                 
333                 /* noch volle Schreib-Puffer suchen */
334                 FD_ZERO( &write_sockets );
335                 for( i = 0; i < MAX_CONNECTIONS; i++ )
336                 {
337                         if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].wdatalen > 0 ))
338                         {
339                                 /* Socket der Verbindung in Set aufnehmen */
340                                 FD_SET( My_Connections[i].sock, &write_sockets );
341                         }
342                 }
343
344                 /* von welchen Sockets koennte gelesen werden? */
345                 read_sockets = My_Sockets;
346                 for( i = 0; i < MAX_CONNECTIONS; i++ )
347                 {
348                         if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].host[0] == '\0' ))
349                         {
350                                 /* Hier muss noch auf den Resolver Sub-Prozess gewartet werden */
351                                 FD_CLR( My_Connections[i].sock, &read_sockets );
352                         }
353                 }
354                 for( i = 0; i < My_Max_Fd + 1; i++ )
355                 {
356                         /* Pipes von Resolver Sub-Prozessen aufnehmen */
357                         if( FD_ISSET( i, &My_Resolvers ))
358                         {
359                                 FD_SET( i, &read_sockets );
360                         }
361                 }
362
363                 /* Auf Aktivitaet warten */
364                 if( select( My_Max_Fd + 1, &read_sockets, &write_sockets, NULL, &tv ) == -1 )
365                 {
366                         if( errno != EINTR ) Log( LOG_ALERT, "select(): %s!", strerror( errno ));
367                         return;
368                 }
369
370                 /* Koennen Daten geschrieben werden? */
371                 for( i = 0; i < My_Max_Fd + 1; i++ )
372                 {
373                         if( FD_ISSET( i, &write_sockets )) Handle_Write( Socket2Index( i ));
374                 }
375
376                 /* Daten zum Lesen vorhanden? */
377                 for( i = 0; i < My_Max_Fd + 1; i++ )
378                 {
379                         if( FD_ISSET( i, &read_sockets )) Handle_Read( i );
380                 }
381         }
382 } /* Conn_Handler */
383
384
385 GLOBAL BOOLEAN Conn_WriteStr( CONN_ID Idx, CHAR *Format, ... )
386 {
387         /* String in Socket schreiben. CR+LF wird von dieser Funktion
388          * automatisch angehaengt. Im Fehlerfall wird dir Verbindung
389          * getrennt und FALSE geliefert. */
390
391         CHAR buffer[MAX_CMDLEN];
392         BOOLEAN ok;
393         va_list ap;
394
395         va_start( ap, Format );
396         if( vsnprintf( buffer, MAX_CMDLEN - 2, Format, ap ) == MAX_CMDLEN - 2 )
397         {
398                 Log( LOG_ALERT, "String too long to send (connection %d)!", Idx );
399                 Conn_Close( Idx, "Server error: String too long to send!" );
400                 return FALSE;
401         }
402
403 #ifdef SNIFFER
404         Log( LOG_DEBUG, " -> connection %d: '%s'.", Idx, buffer );
405 #endif
406
407         strcat( buffer, "\r\n" );
408         ok = Conn_Write( Idx, buffer, strlen( buffer ));
409
410         va_end( ap );
411         return ok;
412 } /* Conn_WriteStr */
413
414
415 GLOBAL BOOLEAN Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
416 {
417         /* Daten in Socket schreiben. Bei "fatalen" Fehlern wird
418          * der Client disconnectiert und FALSE geliefert. */
419
420         assert( Idx >= 0 );
421         assert( My_Connections[Idx].sock >= 0 );
422         assert( Data != NULL );
423         assert( Len > 0 );
424
425         /* pruefen, ob Daten im Schreibpuffer sind. Wenn ja, zunaechst
426          * pruefen, ob diese gesendet werden koennen */
427         if( My_Connections[Idx].wdatalen > 0 )
428         {
429                 if( ! Try_Write( Idx )) return FALSE;
430         }
431
432         /* pruefen, ob im Schreibpuffer genuegend Platz ist */
433         if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 )
434         {
435                 /* der Puffer ist dummerweise voll ... */
436                 Log( LOG_NOTICE, "Write buffer overflow (connection %d)!", Idx );
437                 Conn_Close( Idx, NULL );
438                 return FALSE;
439         }
440
441         /* Daten in Puffer kopieren */
442         memcpy( My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen, Data, Len );
443         My_Connections[Idx].wdatalen += Len;
444
445         /* pruefen, on Daten vorhanden sind und geschrieben werden koennen */
446         if( My_Connections[Idx].wdatalen > 0 )
447         {
448                 if( ! Try_Write( Idx )) return FALSE;
449         }
450
451         return TRUE;
452 } /* Conn_Write */
453
454
455 GLOBAL VOID Conn_Close( CONN_ID Idx, CHAR *Msg )
456 {
457         /* Verbindung schliessen. Evtl. noch von Resolver
458          * Sub-Prozessen offene Pipes werden geschlossen. */
459
460         assert( Idx >= 0 );
461         assert( My_Connections[Idx].sock >= 0 );
462
463         if( Msg ) Conn_WriteStr( Idx, "ERROR :%s", Msg );
464
465         if( close( My_Connections[Idx].sock ) != 0 )
466         {
467                 Log( LOG_ERR, "Error closing connection %d with %s:%d - %s!", Idx, inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port), strerror( errno ));
468                 return;
469         }
470         else
471         {
472                 Log( LOG_NOTICE, "Connection %d with %s:%d closed.", Idx, inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port ));
473         }
474
475         Client_Destroy( Client_GetFromConn( Idx ));
476
477         if( My_Connections[Idx].res_stat )
478         {
479                 /* Resolver-Strukturen freigeben, wenn noch nicht geschehen */
480                 FD_CLR( My_Connections[Idx].res_stat->pipe[0], &My_Resolvers );
481                 close( My_Connections[Idx].res_stat->pipe[0] );
482                 close( My_Connections[Idx].res_stat->pipe[1] );
483                 free( My_Connections[Idx].res_stat );
484         }
485         
486         FD_CLR( My_Connections[Idx].sock, &My_Sockets );
487         My_Connections[Idx].sock = NONE;
488 } /* Conn_Close */
489
490
491 LOCAL BOOLEAN Try_Write( CONN_ID Idx )
492 {
493         /* Versuchen, Daten aus dem Schreib-Puffer in den
494          * Socket zu schreiben. */
495
496         fd_set write_socket;
497
498         assert( Idx >= 0 );
499         assert( My_Connections[Idx].sock >= 0 );
500         assert( My_Connections[Idx].wdatalen > 0 );
501
502         FD_ZERO( &write_socket );
503         FD_SET( My_Connections[Idx].sock, &write_socket );
504         if( select( My_Connections[Idx].sock + 1, NULL, &write_socket, NULL, 0 ) == -1 )
505         {
506                 /* Fehler! */
507                 if( errno != EINTR )
508                 {
509                         Log( LOG_ALERT, "select(): %s!", strerror( errno ));
510                         Conn_Close( Idx, NULL );
511                         return FALSE;
512                 }
513         }
514
515         if( FD_ISSET( My_Connections[Idx].sock, &write_socket )) return Handle_Write( Idx );
516         else return TRUE;
517 } /* Try_Write */
518
519
520 LOCAL VOID Handle_Read( INT Sock )
521 {
522         /* Aktivitaet auf einem Socket verarbeiten:
523          *  - neue Clients annehmen,
524          *  - Daten von Clients verarbeiten,
525          *  - Resolver-Rueckmeldungen annehmen. */
526
527         CONN_ID idx;
528
529         assert( Sock >= 0 );
530         
531         if( FD_ISSET( Sock, &My_Listeners ))
532         {
533                 /* es ist einer unserer Listener-Sockets: es soll
534                  * also eine neue Verbindung aufgebaut werden. */
535
536                 New_Connection( Sock );
537         }
538         else if( FD_ISSET( Sock, &My_Resolvers ))
539         {
540                 /* Rueckmeldung von einem Resolver Sub-Prozess */
541
542                 Read_Resolver_Result( Sock );
543         }
544         else
545         {
546                 /* Ein Client Socket: entweder ein User oder Server */
547
548                 idx = Socket2Index( Sock );
549                 Read_Request( idx );
550         }
551 } /* Handle_Read */
552
553
554 LOCAL BOOLEAN Handle_Write( CONN_ID Idx )
555 {
556         /* Daten aus Schreibpuffer versenden */
557
558         INT len;
559
560         assert( Idx >= 0 );
561         assert( My_Connections[Idx].sock >= 0 );
562         assert( My_Connections[Idx].wdatalen > 0 );
563
564         /* Daten schreiben */
565         len = send( My_Connections[Idx].sock, My_Connections[Idx].wbuf, My_Connections[Idx].wdatalen, 0 );
566         if( len < 0 )
567         {
568                 /* Oops, ein Fehler! */
569                 Log( LOG_ALERT, "Write error (buffer) on connection %d: %s!", Idx, strerror( errno ));
570                 Conn_Close( Idx, NULL );
571                 return FALSE;
572         }
573
574         /* Puffer anpassen */
575         My_Connections[Idx].wdatalen -= len;
576         memmove( My_Connections[Idx].wbuf, My_Connections[Idx].wbuf + len, My_Connections[Idx].wdatalen );
577
578         return TRUE;
579 } /* Handle_Write */
580
581
582 LOCAL VOID New_Connection( INT Sock )
583 {
584         /* Neue Client-Verbindung von Listen-Socket annehmen und
585          * CLIENT-Struktur anlegen. */
586
587         struct sockaddr_in new_addr;
588         INT new_sock, new_sock_len;
589         RES_STAT *s;
590         CONN_ID idx;
591
592         assert( Sock >= 0 );
593
594         new_sock_len = sizeof( new_addr );
595         new_sock = accept( Sock, (struct sockaddr *)&new_addr, (socklen_t *)&new_sock_len );
596         if( new_sock < 0 )
597         {
598                 Log( LOG_CRIT, "Can't accept connection: %s!", strerror( errno ));
599                 return;
600         }
601
602         /* Freie Connection-Struktur suschen */
603         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock < 0 ) break;
604         if( idx >= MAX_CONNECTIONS )
605         {
606                 Log( LOG_ALERT, "Can't accept connection: limit reached (%d)!", MAX_CONNECTIONS );
607                 close( new_sock );
608                 return;
609         }
610
611         /* Client-Struktur initialisieren */
612         if( ! Client_NewLocal( idx, inet_ntoa( new_addr.sin_addr )))
613         {
614                 Log( LOG_ALERT, "Can't accept connection: can't create client structure!" );
615                 close( new_sock );
616                 return;
617         }
618
619         /* Verbindung registrieren */
620         Init_Conn_Struct( idx );
621         My_Connections[idx].sock = new_sock;
622         My_Connections[idx].addr = new_addr;
623
624         /* Neuen Socket registrieren */
625         FD_SET( new_sock, &My_Sockets );
626         if( new_sock > My_Max_Fd ) My_Max_Fd = new_sock;
627
628         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 );
629
630         /* Hostnamen ermitteln */
631         s = Resolve( &new_addr );
632         if( s )
633         {
634                 /* Sub-Prozess wurde asyncron gestartet */
635                 My_Connections[idx].res_stat = s;
636         }
637         else
638         {
639                 /* kann Namen nicht aufloesen */
640                 strcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr ));
641         }
642 } /* New_Connection */
643
644
645 LOCAL CONN_ID Socket2Index( INT Sock )
646 {
647         /* zum Socket passende Connection suchen */
648
649         CONN_ID idx;
650
651         assert( Sock >= 0 );
652
653         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock == Sock ) break;
654
655         assert( idx < MAX_CONNECTIONS );
656         return idx;
657 } /* Socket2Index */
658
659
660 LOCAL VOID Read_Request( CONN_ID Idx )
661 {
662         /* Daten von Socket einlesen und entsprechend behandeln.
663          * Tritt ein Fehler auf, so wird der Socket geschlossen. */
664
665         INT len;
666
667         assert( Idx >= 0 );
668         assert( My_Connections[Idx].sock >= 0 );
669
670         len = recv( My_Connections[Idx].sock, My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen, READBUFFER_LEN - My_Connections[Idx].rdatalen, 0 );
671         My_Connections[Idx].rbuf[READBUFFER_LEN] = '\0';
672
673         if( len == 0 )
674         {
675                 /* Socket wurde geschlossen */
676                 Log( LOG_INFO, "%s:%d is closing the connection ...", inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port));
677                 Conn_Close( Idx, NULL );
678                 return;
679         }
680
681         if( len < 0 )
682         {
683                 /* Fehler beim Lesen */
684                 Log( LOG_ALERT, "Read error on connection %d: %s!", Idx, strerror( errno ));
685                 Conn_Close( Idx, NULL );
686                 return;
687         }
688
689         /* Lesebuffer updaten */
690         My_Connections[Idx].rdatalen += len;
691         assert( My_Connections[Idx].rdatalen <= READBUFFER_LEN );
692         My_Connections[Idx].rbuf[My_Connections[Idx].rdatalen] = '\0';
693
694         if( My_Connections[Idx].rdatalen > MAX_CMDLEN )
695         {
696                 /* Eine Anfrage darf(!) nicht laenger als 512 Zeichen
697                  * (incl. CR+LF!) werden; vgl. RFC 2812. Wenn soetwas
698                  * empfangen wird, wird der Client disconnectiert. */
699                 Log( LOG_ALERT, "Request too long (connection %d)!", Idx );
700                 Conn_Close( Idx, "Request too long!" );
701                 return;
702         }
703
704         /* Timestamp aktualisieren */
705         My_Connections[Idx].lastdata = time( NULL );
706
707         Handle_Buffer( Idx );
708 } /* Read_Request */
709
710
711 LOCAL VOID Handle_Buffer( CONN_ID Idx )
712 {
713         /* Daten im Lese-Puffer einer Verbindung verarbeiten. */
714
715         CHAR *ptr, *ptr1, *ptr2;
716         INT len, delta;
717         
718         /* Eine komplette Anfrage muss mit CR+LF enden, vgl.
719          * RFC 2812. Haben wir eine? */
720         ptr = strstr( My_Connections[Idx].rbuf, "\r\n" );
721
722         if( ptr ) delta = 2;
723         else
724         {
725                 /* Nicht RFC-konforme Anfrage mit nur CR oder LF? Leider
726                  * machen soetwas viele Clients, u.a. "mIRC" :-( */
727                 ptr1 = strchr( My_Connections[Idx].rbuf, '\r' );
728                 ptr2 = strchr( My_Connections[Idx].rbuf, '\n' );
729                 delta = 1;
730                 if( ptr1 && ptr2 ) ptr = ptr1 > ptr2 ? ptr2 : ptr1;
731                 else if( ptr1 ) ptr = ptr1;
732                 else if( ptr2 ) ptr = ptr2;
733         }
734
735         if( ptr )
736         {
737                 /* Ende der Anfrage wurde gefunden */
738                 *ptr = '\0';
739                 len = ( ptr - My_Connections[Idx].rbuf ) + delta;
740                 if( len > delta )
741                 {
742                         /* Es wurde ein Request gelesen */
743                         if( ! Parse_Request( Idx, My_Connections[Idx].rbuf )) return;
744                 }
745
746                 /* Puffer anpassen */
747                 My_Connections[Idx].rdatalen -= len;
748                 memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen );
749         }
750 } /* Handle_Buffer */
751
752
753 LOCAL VOID Check_Connections( VOID )
754 {
755         /* Pruefen, ob Verbindungen noch "alive" sind. Ist dies
756          * nicht der Fall, zunaechst PING-PONG spielen und, wenn
757          * auch das nicht "hilft", Client disconnectieren. */
758
759         INT i;
760
761         for( i = 0; i < MAX_CONNECTIONS; i++ )
762         {
763                 if( My_Connections[i].sock != NONE )
764                 {
765                         if( My_Connections[i].lastping > My_Connections[i].lastdata )
766                         {
767                                 /* es wurde bereits ein PING gesendet */
768                                 if( My_Connections[i].lastping < time( NULL ) - Conf_PongTimeout )
769                                 {
770                                         /* Timeout */
771                                         Log( LOG_INFO, "Connection %d: Ping timeout.", i );
772                                         Conn_Close( i, "Ping timeout" );
773                                 }
774                         }
775                         else if( My_Connections[i].lastdata < time( NULL ) - Conf_PingTimeout )
776                         {
777                                 /* es muss ein PING gesendet werden */
778                                 Log( LOG_DEBUG, "Connection %d: sending PING ...", i );
779                                 My_Connections[i].lastping = time( NULL );
780                                 Conn_WriteStr( i, "PING :%s", This_Server->nick );
781                         }
782                 }
783         }
784 } /* Conn_Check */
785
786
787 LOCAL VOID Init_Conn_Struct( INT Idx )
788 {
789         /* Connection-Struktur initialisieren */
790
791         My_Connections[Idx].sock = NONE;
792         My_Connections[Idx].res_stat = NULL;
793         My_Connections[Idx].host[0] = '\0';
794         My_Connections[Idx].rbuf[0] = '\0';
795         My_Connections[Idx].rdatalen = 0;
796         My_Connections[Idx].wbuf[0] = '\0';
797         My_Connections[Idx].wdatalen = 0;
798         My_Connections[Idx].lastdata = time( NULL );
799         My_Connections[Idx].lastping = 0;
800 } /* Init_Conn_Struct */
801
802
803 LOCAL RES_STAT *Resolve( struct sockaddr_in *Addr )
804 {
805         /* Hostnamen (asyncron!) aufloesen. Bei Fehler, z.B. wenn der
806          * Child-Prozess nicht erzeugt werden kann, wird NULL geliefert.
807          * Der Host kann dann nicht aufgeloest werden. */
808
809         RES_STAT *s;
810         INT pid;
811
812         /* Speicher anfordern */
813         s = malloc( sizeof( RES_STAT ));
814         if( ! s )
815         {
816                 Log( LOG_ALERT, "Resolver: Can't alloc memory!" );
817                 return NULL;
818         }
819
820         /* Pipe fuer Antwort initialisieren */
821         if( pipe( s->pipe ) != 0 )
822         {
823                 free( s );
824                 Log( LOG_ALERT, "Resolver: Can't create output pipe: %s!", strerror( errno ));
825                 return NULL;
826         }
827
828         /* Sub-Prozess erzeugen */
829         pid = fork( );
830         if( pid > 0 )
831         {
832                 /* Haupt-Prozess */
833                 Log( LOG_DEBUG, "Resolver for %s created (PID %d).", inet_ntoa( Addr->sin_addr ), pid );
834                 FD_SET( s->pipe[0], &My_Resolvers );
835                 if( s->pipe[0] > My_Max_Fd ) My_Max_Fd = s->pipe[0];
836                 s->pid = pid;
837                 return s;
838         }
839         else if( pid == 0 )
840         {
841                 /* Sub-Prozess */
842                 Log_Init_Resolver( );
843                 Do_Resolve( Addr, s->pipe[1] );
844                 Log_Exit_Resolver( );
845                 exit( 0 );
846         }
847         else
848         {
849                 /* Fehler */
850                 free( s );
851                 Log( LOG_ALERT, "Resolver: Can't fork: %s!", strerror( errno ));
852                 return NULL;
853         }
854 } /* Resolve */
855
856
857 LOCAL VOID Read_Resolver_Result( INT r_fd )
858 {
859         /* Ergebnis von Resolver Sub-Prozess aus Pipe lesen
860          * und entsprechende Connection aktualisieren */
861
862         CHAR hostname[HOST_LEN];
863         CLIENT *c;
864         INT i;
865
866         FD_CLR( r_fd, &My_Resolvers );
867
868         /* Anfrage vom Parent lesen */
869         if( read( r_fd, hostname, HOST_LEN) < 0 )
870         {
871                 /* Fehler beim Lesen aus der Pipe */
872                 close( r_fd );
873                 Log( LOG_ALERT, "Resolver: Can't read result: %s!", strerror( errno ));
874                 return;
875         }
876
877         /* zugehoerige Connection suchen */
878         for( i = 0; i < MAX_CONNECTIONS; i++ )
879         {
880                 if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].res_stat ) && ( My_Connections[i].res_stat->pipe[0] == r_fd )) break;
881         }
882
883         if( i >= MAX_CONNECTIONS )
884         {
885                 /* Opsa! Keine passende Connection gefunden!? */
886                 close( r_fd );
887                 Log( LOG_ALERT, "Resolver: Got result for unknown connection!?" );
888                 return;
889         }
890
891         /* Aufraeumen */
892         close( My_Connections[i].res_stat->pipe[0] );
893         close( My_Connections[i].res_stat->pipe[1] );
894         free( My_Connections[i].res_stat );
895         My_Connections[i].res_stat = NULL;
896         
897         /* Hostnamen setzen */
898         strcpy( My_Connections[i].host, hostname );
899         c = Client_GetFromConn( i );
900         if( c ) Client_SetHostname( c, hostname );
901 } /* Read_Resolver_Result */
902
903
904 LOCAL VOID Do_Resolve( struct sockaddr_in *Addr, INT w_fd )
905 {
906         /* Resolver Sub-Prozess: aufzuloesenden Namen aus
907          * der Pipe lesen, Ergebnis in Pipe schreiben. */
908
909         CHAR hostname[HOST_LEN];
910         struct hostent *h;
911
912         Log_Resolver( LOG_DEBUG, "Now resolving %s ...", inet_ntoa( Addr->sin_addr ));
913
914         /* Namen aufloesen */
915         h = gethostbyaddr( (CHAR *)&Addr->sin_addr, sizeof( Addr->sin_addr ), AF_INET );
916         if( h ) strcpy( hostname, h->h_name );
917         else
918         {
919                 Log_Resolver( LOG_WARNING, "Resolver: Can't resolve host name (code %d)!", h_errno );
920                 strcpy( hostname, inet_ntoa( Addr->sin_addr ));
921         }
922
923         /* Antwort an Parent schreiben */
924         if( write( w_fd, hostname, strlen( hostname ) + 1 ) != ( strlen( hostname ) + 1 ))
925         {
926                 Log_Resolver( LOG_ALERT, "Resolver: Can't write to parent: %s!", strerror( errno ));
927                 close( w_fd );
928                 return;
929         }
930
931         Log_Resolver( LOG_DEBUG, "Ok, translated %s to \"%s\".", inet_ntoa( Addr->sin_addr ), hostname );
932 } /* Do_Resolve */
933
934
935 /* -eof- */