]> arthur.barton.de Git - ngircd-alex.git/blob - src/ngircd/conn.c
d694c5e9e2b22a903cf2e9bbe6bc525248309ab8
[ngircd-alex.git] / src / ngircd / conn.c
1 /*
2  * ngIRCd -- The Next Generation IRC Daemon
3  * Copyright (c)2001,2002 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 ngIRCd beteiligten Autoren finden Sie in der Datei AUTHORS.
11  *
12  * $Id: conn.c,v 1.80 2002/10/09 17:07:22 alex Exp $
13  *
14  * connect.h: Verwaltung aller Netz-Verbindungen ("connections")
15  */
16
17
18 #include "portab.h"
19
20 #include "imp.h"
21 #include <assert.h>
22 #include <stdarg.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <unistd.h>
26 #include <errno.h>
27 #include <fcntl.h>
28 #include <string.h>
29 #include <sys/socket.h>
30 #include <sys/time.h>
31 #include <sys/types.h>
32 #include <time.h>
33 #include <netinet/in.h>
34
35 #ifdef HAVE_ARPA_INET_H
36 #include <arpa/inet.h>
37 #else
38 #define PF_INET AF_INET
39 #endif
40
41 #ifdef HAVE_STDINT_H
42 #include <stdint.h>                     /* u.a. fuer Mac OS X */
43 #endif
44
45 #include "exp.h"
46 #include "conn.h"
47
48 #include "imp.h"
49 #include "ngircd.h"
50 #include "client.h"
51 #include "resolve.h"
52 #include "conf.h"
53 #include "log.h"
54 #include "parse.h"
55 #include "tool.h"
56
57 #include "exp.h"
58
59
60 #define SERVER_WAIT (NONE - 1)
61
62
63 typedef struct _Connection
64 {
65         INT sock;                       /* Socket Handle */
66         struct sockaddr_in addr;        /* Adresse des Client */
67         RES_STAT *res_stat;             /* "Resolver-Status", s.o. */
68         CHAR host[HOST_LEN];            /* Hostname */
69         CHAR rbuf[READBUFFER_LEN];      /* Lesepuffer */
70         INT rdatalen;                   /* Laenge der Daten im Lesepuffer */
71         CHAR wbuf[WRITEBUFFER_LEN];     /* Schreibpuffer */
72         INT wdatalen;                   /* Laenge der Daten im Schreibpuffer */
73         INT our_server;                 /* wenn von uns zu connectender Server: ID */
74         time_t lastdata;                /* Letzte Aktivitaet */
75         time_t lastping;                /* Letzter PING */
76         time_t lastprivmsg;             /* Letzte PRIVMSG */
77         time_t delaytime;               /* Nicht beachten bis ("penalty") */
78         LONG bytes_in, bytes_out;       /* Counter fuer Statistik */
79 } CONNECTION;
80
81
82 LOCAL VOID Handle_Read PARAMS(( INT sock ));
83 LOCAL BOOLEAN Handle_Write PARAMS(( CONN_ID Idx ));
84 LOCAL VOID New_Connection PARAMS(( INT Sock ));
85 LOCAL CONN_ID Socket2Index PARAMS(( INT Sock ));
86 LOCAL VOID Read_Request PARAMS(( CONN_ID Idx ));
87 LOCAL BOOLEAN Try_Write PARAMS(( CONN_ID Idx ));
88 LOCAL VOID Handle_Buffer PARAMS(( CONN_ID Idx ));
89 LOCAL VOID Check_Connections PARAMS(( VOID ));
90 LOCAL VOID Check_Servers PARAMS(( VOID ));
91 LOCAL VOID Init_Conn_Struct PARAMS(( INT Idx ));
92 LOCAL BOOLEAN Init_Socket PARAMS(( INT Sock ));
93 LOCAL VOID New_Server PARAMS(( INT Server, CONN_ID Idx ));
94 LOCAL VOID Read_Resolver_Result PARAMS(( INT r_fd ));
95
96
97 LOCAL fd_set My_Listeners;
98 LOCAL fd_set My_Sockets;
99 LOCAL fd_set My_Connects;
100
101 LOCAL CONNECTION My_Connections[MAX_CONNECTIONS];
102
103
104 GLOBAL VOID
105 Conn_Init( VOID )
106 {
107         /* Modul initialisieren: statische Strukturen "ausnullen". */
108
109         CONN_ID i;
110
111         /* zu Beginn haben wir keine Verbindungen */
112         FD_ZERO( &My_Listeners );
113         FD_ZERO( &My_Sockets );
114         FD_ZERO( &My_Connects );
115
116         Conn_MaxFD = 0;
117
118         /* Connection-Struktur initialisieren */
119         for( i = 0; i < MAX_CONNECTIONS; i++ ) Init_Conn_Struct( i );
120 } /* Conn_Init */
121
122
123 GLOBAL VOID
124 Conn_Exit( VOID )
125 {
126         /* Modul abmelden: alle noch offenen Connections
127          * schliessen und freigeben. */
128
129         CONN_ID idx;
130         INT i;
131
132         /* Sockets schliessen */
133         Log( LOG_DEBUG, "Shutting down all connections ..." );
134         for( i = 0; i < Conn_MaxFD + 1; i++ )
135         {
136                 if( FD_ISSET( i, &My_Sockets ))
137                 {
138                         for( idx = 0; idx < MAX_CONNECTIONS; idx++ )
139                         {
140                                 if( My_Connections[idx].sock == i ) break;
141                         }
142                         if( FD_ISSET( i, &My_Listeners ))
143                         {
144                                 close( i );
145                                 Log( LOG_DEBUG, "Listening socket %d closed.", i );
146                         }
147                         else if( FD_ISSET( i, &My_Connects ))
148                         {
149                                 close( i );
150                                 Log( LOG_DEBUG, "Connection %d closed during creation (socket %d).", idx, i );
151                         }
152                         else if( idx < MAX_CONNECTIONS )
153                         {
154                                 if( NGIRCd_Restart ) Conn_Close( idx, NULL, "Server going down (restarting)", TRUE );
155                                 else Conn_Close( idx, NULL, "Server going down", TRUE );
156                         }
157                         else
158                         {
159                                 Log( LOG_WARNING, "Closing unknown connection %d ...", i );
160                                 close( i );
161                         }
162                 }
163         }
164 } /* Conn_Exit */
165
166
167 GLOBAL BOOLEAN
168 Conn_NewListener( CONST UINT Port )
169 {
170         /* Neuen Listen-Socket erzeugen: der Server wartet dann auf
171          * dem angegebenen Port auf Verbindungen. Kann der Listen-
172          * Socket nicht erteugt werden, so wird NULL geliefert.*/
173
174         struct sockaddr_in addr;
175         INT sock;
176
177         /* Server-"Listen"-Socket initialisieren */
178         memset( &addr, 0, sizeof( addr ));
179         addr.sin_family = AF_INET;
180         addr.sin_port = htons( Port );
181         addr.sin_addr.s_addr = htonl( INADDR_ANY );
182
183         /* Socket erzeugen */
184         sock = socket( PF_INET, SOCK_STREAM, 0);
185         if( sock < 0 )
186         {
187                 Log( LOG_CRIT, "Can't create socket: %s!", strerror( errno ));
188                 return FALSE;
189         }
190
191         if( ! Init_Socket( sock )) return FALSE;
192
193         /* an Port binden */
194         if( bind( sock, (struct sockaddr *)&addr, (socklen_t)sizeof( addr )) != 0 )
195         {
196                 Log( LOG_CRIT, "Can't bind socket: %s!", strerror( errno ));
197                 close( sock );
198                 return FALSE;
199         }
200
201         /* in "listen mode" gehen :-) */
202         if( listen( sock, 10 ) != 0 )
203         {
204                 Log( LOG_CRIT, "Can't listen on soecket: %s!", strerror( errno ));
205                 close( sock );
206                 return FALSE;
207         }
208
209         /* Neuen Listener in Strukturen einfuegen */
210         FD_SET( sock, &My_Listeners );
211         FD_SET( sock, &My_Sockets );
212
213         if( sock > Conn_MaxFD ) Conn_MaxFD = sock;
214
215         Log( LOG_INFO, "Now listening on port %d (socket %d).", Port, sock );
216
217         return TRUE;
218 } /* Conn_NewListener */
219
220
221 GLOBAL VOID
222 Conn_Handler( VOID )
223 {
224         /* "Hauptschleife": Aktive Verbindungen ueberwachen. Folgende Aktionen
225          * werden dabei durchgefuehrt, bis der Server terminieren oder neu
226          * starten soll:
227          *
228          *  - neue Verbindungen annehmen,
229          *  - Server-Verbindungen aufbauen,
230          *  - geschlossene Verbindungen loeschen,
231          *  - volle Schreibpuffer versuchen zu schreiben,
232          *  - volle Lesepuffer versuchen zu verarbeiten,
233          *  - Antworten von Resolver Sub-Prozessen annehmen.
234          */
235
236         fd_set read_sockets, write_sockets;
237         struct timeval tv;
238         time_t start, t;
239         INT i;
240
241         start = time( NULL );
242         while(( ! NGIRCd_Quit ) && ( ! NGIRCd_Restart ))
243         {
244                 Check_Servers( );
245
246                 Check_Connections( );
247
248                 /* noch volle Lese-Buffer suchen */
249                 for( i = 0; i < MAX_CONNECTIONS; i++ )
250                 {
251                         if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].rdatalen > 0 ))
252                         {
253                                 /* Kann aus dem Buffer noch ein Befehl extrahiert werden? */
254                                 Handle_Buffer( i );
255                         }
256                 }
257
258                 /* noch volle Schreib-Puffer suchen */
259                 FD_ZERO( &write_sockets );
260                 for( i = 0; i < MAX_CONNECTIONS; i++ )
261                 {
262                         if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].wdatalen > 0 ))
263                         {
264                                 /* Socket der Verbindung in Set aufnehmen */
265                                 FD_SET( My_Connections[i].sock, &write_sockets );
266                         }
267                 }
268                 /* Sockets mit im Aufbau befindlichen ausgehenden Verbindungen suchen */
269                 for( i = 0; i < MAX_CONNECTIONS; i++ )
270                 {
271                         if(( My_Connections[i].sock > NONE ) && ( FD_ISSET( My_Connections[i].sock, &My_Connects ))) FD_SET( My_Connections[i].sock, &write_sockets );
272                 }
273
274                 /* von welchen Sockets koennte gelesen werden? */
275                 t = time( NULL );
276                 read_sockets = My_Sockets;
277                 for( i = 0; i < MAX_CONNECTIONS; i++ )
278                 {
279                         if(( My_Connections[i].sock > NONE ) && ( My_Connections[i].host[0] == '\0' ))
280                         {
281                                 /* Hier muss noch auf den Resolver Sub-Prozess gewartet werden */
282                                 FD_CLR( My_Connections[i].sock, &read_sockets );
283                         }
284                         if(( My_Connections[i].sock > NONE ) && ( FD_ISSET( My_Connections[i].sock, &My_Connects )))
285                         {
286                                 /* Hier laeuft noch ein asyncrones connect() */
287                                 FD_CLR( My_Connections[i].sock, &read_sockets );
288                         }
289                         if( My_Connections[i].delaytime > t )
290                         {
291                                 /* Fuer die Verbindung ist eine "Penalty-Zeit" gesetzt */
292                                 FD_CLR( My_Connections[i].sock, &read_sockets );
293                                 FD_CLR( My_Connections[i].sock, &write_sockets );
294                         }
295                 }
296                 for( i = 0; i < Conn_MaxFD + 1; i++ )
297                 {
298                         /* Pipes von Resolver Sub-Prozessen aufnehmen */
299                         if( FD_ISSET( i, &Resolver_FDs ))
300                         {
301                                 FD_SET( i, &read_sockets );
302                         }
303                 }
304
305                 /* Timeout initialisieren */
306                 tv.tv_sec = 1;
307                 tv.tv_usec = 0;
308                 
309                 /* Auf Aktivitaet warten */
310                 i = select( Conn_MaxFD + 1, &read_sockets, &write_sockets, NULL, &tv );
311                 if( i == 0 )
312                 {
313                         /* keine Veraenderung an den Sockets */
314                         continue;
315                 }
316                 if( i == -1 )
317                 {
318                         /* Fehler (z.B. Interrupt) */
319                         if( errno != EINTR )
320                         {
321                                 Log( LOG_EMERG, "select(): %s!", strerror( errno ));
322                                 Log( LOG_ALERT, "%s exiting due to fatal errors!", PACKAGE );
323                                 exit( 1 );
324                         }
325                         continue;
326                 }
327
328                 /* Koennen Daten geschrieben werden? */
329                 for( i = 0; i < Conn_MaxFD + 1; i++ )
330                 {
331                         if( FD_ISSET( i, &write_sockets )) Handle_Write( Socket2Index( i ));
332                 }
333
334                 /* Daten zum Lesen vorhanden? */
335                 for( i = 0; i < Conn_MaxFD + 1; i++ )
336                 {
337                         if( FD_ISSET( i, &read_sockets )) Handle_Read( i );
338                 }
339         }
340 } /* Conn_Handler */
341
342
343 #ifdef PROTOTYPES
344 GLOBAL BOOLEAN
345 Conn_WriteStr( CONN_ID Idx, CHAR *Format, ... )
346 #else
347 GLOBAL BOOLEAN
348 Conn_WriteStr( Idx, Format, va_alist )
349 CONN_ID Idx;
350 CHAR *Format;
351 va_dcl
352 #endif
353 {
354         /* String in Socket schreiben. CR+LF wird von dieser Funktion
355          * automatisch angehaengt. Im Fehlerfall wird dir Verbindung
356          * getrennt und FALSE geliefert. */
357
358         CHAR buffer[COMMAND_LEN];
359         BOOLEAN ok;
360         va_list ap;
361
362         assert( Idx >= 0 );
363         assert( Format != NULL );
364
365 #ifdef PROTOTYPES
366         va_start( ap, Format );
367 #else
368         va_start( ap );
369 #endif
370         if( vsnprintf( buffer, COMMAND_LEN - 2, Format, ap ) == COMMAND_LEN - 2 )
371         {
372                 Log( LOG_CRIT, "Text too long to send (connection %d)!", Idx );
373                 Conn_Close( Idx, "Text too long to send!", NULL, FALSE );
374                 return FALSE;
375         }
376
377 #ifdef SNIFFER
378         if( NGIRCd_Sniffer ) Log( LOG_DEBUG, " -> connection %d: '%s'.", Idx, buffer );
379 #endif
380
381         strcat( buffer, "\r\n" );
382         ok = Conn_Write( Idx, buffer, strlen( buffer ));
383
384         va_end( ap );
385         return ok;
386 } /* Conn_WriteStr */
387
388
389 GLOBAL BOOLEAN
390 Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
391 {
392         /* Daten in Socket schreiben. Bei "fatalen" Fehlern wird
393          * der Client disconnectiert und FALSE geliefert. */
394
395         assert( Idx >= 0 );
396         assert( Data != NULL );
397         assert( Len > 0 );
398
399         /* Ist der entsprechende Socket ueberhaupt noch offen?
400          * In einem "Handler-Durchlauf" kann es passieren, dass
401          * dem nicht mehr so ist, wenn einer von mehreren
402          * Conn_Write()'s fehlgeschlagen ist. In diesem Fall
403          * wird hier einfach ein Fehler geliefert. */
404         if( My_Connections[Idx].sock <= NONE )
405         {
406                 Log( LOG_DEBUG, "Skipped write on closed socket (connection %d).", Idx );
407                 return FALSE;
408         }
409
410         /* pruefen, ob Daten im Schreibpuffer sind. Wenn ja, zunaechst
411          * pruefen, ob diese gesendet werden koennen */
412         if( My_Connections[Idx].wdatalen > 0 )
413         {
414                 if( ! Try_Write( Idx )) return FALSE;
415         }
416
417         /* pruefen, ob im Schreibpuffer genuegend Platz ist */
418         if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 )
419         {
420                 /* der Puffer ist dummerweise voll ... */
421                 Log( LOG_NOTICE, "Write buffer overflow (connection %d)!", Idx );
422                 Conn_Close( Idx, "Write buffer overflow!", NULL, FALSE );
423                 return FALSE;
424         }
425
426         /* Daten in Puffer kopieren */
427         memcpy( My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen, Data, Len );
428         My_Connections[Idx].wdatalen += Len;
429
430         /* pruefen, on Daten vorhanden sind und geschrieben werden koennen */
431         if( My_Connections[Idx].wdatalen > 0 )
432         {
433                 if( ! Try_Write( Idx )) return FALSE;
434         }
435
436         return TRUE;
437 } /* Conn_Write */
438
439
440 GLOBAL VOID
441 Conn_Close( CONN_ID Idx, CHAR *LogMsg, CHAR *FwdMsg, BOOLEAN InformClient )
442 {
443         /* Verbindung schliessen. Evtl. noch von Resolver
444          * Sub-Prozessen offene Pipes werden geschlossen. */
445
446         CLIENT *c;
447
448         assert( Idx >= 0 );
449         assert( My_Connections[Idx].sock > NONE );
450
451         c = Client_GetFromConn( Idx );
452
453         if( InformClient )
454         {
455                 /* Statistik an Client melden, wenn User */
456                 if(( c != NULL ) && ( Client_Type( c ) == CLIENT_USER ))
457                 {
458                         Conn_WriteStr( Idx, "NOTICE %s :%sConnection statistics: %.1f kb received, %.1f kb sent.", Client_ThisServer( ), NOTICE_TXTPREFIX, (DOUBLE)My_Connections[Idx].bytes_in / 1024,  (DOUBLE)My_Connections[Idx].bytes_out / 1024 );
459                 }
460
461                 /* ERROR an Client schicken (von RFC so vorgesehen!) */
462                 if( FwdMsg ) Conn_WriteStr( Idx, "ERROR :%s", FwdMsg );
463                 else Conn_WriteStr( Idx, "ERROR :Closing connection." );
464                 if( My_Connections[Idx].sock == NONE ) return;
465         }
466
467         if( close( My_Connections[Idx].sock ) != 0 )
468         {
469                 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 ));
470         }
471         else
472         {
473                 Log( LOG_INFO, "Connection %d with %s:%d closed.", Idx, inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port ));
474         }
475
476         if( c ) Client_Destroy( c, LogMsg, FwdMsg, TRUE );
477
478         if( My_Connections[Idx].res_stat )
479         {
480                 /* Resolver-Strukturen freigeben, wenn noch nicht geschehen */
481                 FD_CLR( My_Connections[Idx].res_stat->pipe[0], &Resolver_FDs );
482                 close( My_Connections[Idx].res_stat->pipe[0] );
483                 close( My_Connections[Idx].res_stat->pipe[1] );
484                 free( My_Connections[Idx].res_stat );
485         }
486
487         /* Startzeit des naechsten Connect-Versuchs modifizieren? */
488         if(( My_Connections[Idx].our_server >= 0 ) && ( Conf_Server[My_Connections[Idx].our_server].lasttry <  time( NULL ) - Conf_ConnectRetry ))
489         {
490                 /* Okay, die Verbindung stand schon "genuegend lange":
491                  * lasttry-Zeitpunkt so setzen, dass der naechste
492                  * Verbindungsversuch in RECONNECT_DELAY Sekunden
493                  * gestartet wird. */
494                 Conf_Server[My_Connections[Idx].our_server].lasttry = time( NULL ) - Conf_ConnectRetry + RECONNECT_DELAY;
495         }
496
497         FD_CLR( My_Connections[Idx].sock, &My_Sockets );
498         FD_CLR( My_Connections[Idx].sock, &My_Connects );
499         My_Connections[Idx].sock = NONE;
500 } /* Conn_Close */
501
502
503 GLOBAL VOID
504 Conn_UpdateIdle( CONN_ID Idx )
505 {
506         /* Idle-Timer zuruecksetzen */
507
508         assert( Idx >= 0 );
509         My_Connections[Idx].lastprivmsg = time( NULL );
510 }
511
512
513 GLOBAL time_t
514 Conn_GetIdle( CONN_ID Idx )
515 {
516         /* Idle-Time einer Verbindung liefern (in Sekunden) */
517
518         assert( Idx >= 0 );
519         return time( NULL ) - My_Connections[Idx].lastprivmsg;
520 } /* Conn_GetIdle */
521
522
523 GLOBAL time_t
524 Conn_LastPing( CONN_ID Idx )
525 {
526         /* Zeitpunkt des letzten PING liefern */
527
528         assert( Idx >= 0 );
529         return My_Connections[Idx].lastping;
530 } /* Conn_LastPing */
531
532
533 GLOBAL VOID
534 Conn_SetPenalty( CONN_ID Idx, time_t Seconds )
535 {
536         /* Penalty-Delay fuer eine Verbindung (in Sekunden) setzen;
537          * waehrend dieser Zeit wird der entsprechende Socket vom Server
538          * bei Lese-Operationen komplett ignoriert. Der Delay kann mit
539          * dieser Funktion nur erhoeht, nicht aber verringert werden. */
540         
541         time_t t;
542         
543         assert( Idx >= 0 );
544         assert( Seconds >= 0 );
545         
546         t = time( NULL ) + Seconds;
547         if( t > My_Connections[Idx].delaytime ) My_Connections[Idx].delaytime = t;
548 } /* Conn_SetPenalty */
549
550
551 LOCAL BOOLEAN
552 Try_Write( CONN_ID Idx )
553 {
554         /* Versuchen, Daten aus dem Schreib-Puffer in den
555          * Socket zu schreiben. */
556
557         fd_set write_socket;
558
559         assert( Idx >= 0 );
560         assert( My_Connections[Idx].sock > NONE );
561         assert( My_Connections[Idx].wdatalen > 0 );
562
563         FD_ZERO( &write_socket );
564         FD_SET( My_Connections[Idx].sock, &write_socket );
565         if( select( My_Connections[Idx].sock + 1, NULL, &write_socket, NULL, 0 ) == -1 )
566         {
567                 /* Fehler! */
568                 if( errno != EINTR )
569                 {
570                         Log( LOG_ALERT, "select() failed: %s!", strerror( errno ));
571                         Conn_Close( Idx, "Server error!", NULL, FALSE );
572                         return FALSE;
573                 }
574         }
575
576         if( FD_ISSET( My_Connections[Idx].sock, &write_socket )) return Handle_Write( Idx );
577         else return TRUE;
578 } /* Try_Write */
579
580
581 LOCAL VOID
582 Handle_Read( INT Sock )
583 {
584         /* Aktivitaet auf einem Socket verarbeiten:
585          *  - neue Clients annehmen,
586          *  - Daten von Clients verarbeiten,
587          *  - Resolver-Rueckmeldungen annehmen. */
588
589         CONN_ID idx;
590
591         assert( Sock >= 0 );
592
593         if( FD_ISSET( Sock, &My_Listeners ))
594         {
595                 /* es ist einer unserer Listener-Sockets: es soll
596                  * also eine neue Verbindung aufgebaut werden. */
597
598                 New_Connection( Sock );
599         }
600         else if( FD_ISSET( Sock, &Resolver_FDs ))
601         {
602                 /* Rueckmeldung von einem Resolver Sub-Prozess */
603
604                 Read_Resolver_Result( Sock );
605         }
606         else
607         {
608                 /* Ein Client Socket: entweder ein User oder Server */
609
610                 idx = Socket2Index( Sock );
611                 Read_Request( idx );
612         }
613 } /* Handle_Read */
614
615
616 LOCAL BOOLEAN
617 Handle_Write( CONN_ID Idx )
618 {
619         /* Daten aus Schreibpuffer versenden bzw. Connection aufbauen */
620
621         INT len, res, err;
622
623         assert( Idx >= 0 );
624         assert( My_Connections[Idx].sock > NONE );
625
626         if( FD_ISSET( My_Connections[Idx].sock, &My_Connects ))
627         {
628                 /* es soll nichts geschrieben werden, sondern ein
629                  * connect() hat ein Ergebnis geliefert */
630
631                 FD_CLR( My_Connections[Idx].sock, &My_Connects );
632
633                 /* Ergebnis des connect() ermitteln */
634                 len = sizeof( err );
635                 res = getsockopt( My_Connections[Idx].sock, SOL_SOCKET, SO_ERROR, &err, &len );
636                 assert( len == sizeof( err ));
637
638                 /* Fehler aufgetreten? */
639                 if(( res != 0 ) || ( err != 0 ))
640                 {
641                         /* Fehler! */
642                         if( res != 0 ) Log( LOG_CRIT, "getsockopt (connection %d): %s!", Idx, strerror( errno ));
643                         else Log( LOG_CRIT, "Can't connect socket to \"%s:%d\" (connection %d): %s!", My_Connections[Idx].host, Conf_Server[My_Connections[Idx].our_server].port, Idx, strerror( err ));
644
645                         /* Socket etc. pp. aufraeumen */
646                         FD_CLR( My_Connections[Idx].sock, &My_Sockets );
647                         close( My_Connections[Idx].sock );
648                         Init_Conn_Struct( Idx );
649
650                         /* Bei Server-Verbindungen lasttry-Zeitpunkt auf "jetzt" setzen */
651                         Conf_Server[My_Connections[Idx].our_server].lasttry = time( NULL );
652
653                         return FALSE;
654                 }
655                 Log( LOG_DEBUG, "Connection %d with \"%s:%d\" established, now sendig PASS and SERVER ...", Idx, My_Connections[Idx].host, Conf_Server[My_Connections[Idx].our_server].port );
656
657                 /* PASS und SERVER verschicken */
658                 Conn_WriteStr( Idx, "PASS %s %s", Conf_Server[My_Connections[Idx].our_server].pwd, NGIRCd_ProtoID );
659                 Conn_WriteStr( Idx, "SERVER %s :%s", Conf_ServerName, Conf_ServerInfo );
660
661                 return TRUE;
662         }
663
664         assert( My_Connections[Idx].wdatalen > 0 );
665
666         /* Daten schreiben */
667         len = send( My_Connections[Idx].sock, My_Connections[Idx].wbuf, My_Connections[Idx].wdatalen, 0 );
668         if( len < 0 )
669         {
670                 /* Oops, ein Fehler! */
671                 Log( LOG_ERR, "Write error (buffer) on connection %d: %s!", Idx, strerror( errno ));
672                 Conn_Close( Idx, "Write error (buffer)!", NULL, FALSE );
673                 return FALSE;
674         }
675
676         /* Connection-Statistik aktualisieren */
677         My_Connections[Idx].bytes_out += len;
678
679         /* Puffer anpassen */
680         My_Connections[Idx].wdatalen -= len;
681         memmove( My_Connections[Idx].wbuf, My_Connections[Idx].wbuf + len, My_Connections[Idx].wdatalen );
682
683         return TRUE;
684 } /* Handle_Write */
685
686
687 LOCAL VOID
688 New_Connection( INT Sock )
689 {
690         /* Neue Client-Verbindung von Listen-Socket annehmen und
691          * CLIENT-Struktur anlegen. */
692
693         struct sockaddr_in new_addr;
694         INT new_sock, new_sock_len;
695         RES_STAT *s;
696         CONN_ID idx;
697         CLIENT *c;
698
699         assert( Sock >= 0 );
700
701         new_sock_len = sizeof( new_addr );
702         new_sock = accept( Sock, (struct sockaddr *)&new_addr, (socklen_t *)&new_sock_len );
703         if( new_sock < 0 )
704         {
705                 Log( LOG_CRIT, "Can't accept connection: %s!", strerror( errno ));
706                 return;
707         }
708
709         /* Freie Connection-Struktur suschen */
710         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock == NONE ) break;
711         if( idx >= MAX_CONNECTIONS )
712         {
713                 Log( LOG_ALERT, "Can't accept connection: limit reached (%d)!", MAX_CONNECTIONS );
714                 close( new_sock );
715                 return;
716         }
717
718         /* Client-Struktur initialisieren */
719         c = Client_NewLocal( idx, inet_ntoa( new_addr.sin_addr ), CLIENT_UNKNOWN, FALSE );
720         if( ! c )
721         {
722                 Log( LOG_ALERT, "Can't accept connection: can't create client structure!" );
723                 close( new_sock );
724                 return;
725         }
726
727         /* Verbindung registrieren */
728         Init_Conn_Struct( idx );
729         My_Connections[idx].sock = new_sock;
730         My_Connections[idx].addr = new_addr;
731
732         /* Neuen Socket registrieren */
733         FD_SET( new_sock, &My_Sockets );
734         if( new_sock > Conn_MaxFD ) Conn_MaxFD = new_sock;
735
736         Log( LOG_INFO, "Accepted connection %d from %s:%d on socket %d.", idx, inet_ntoa( new_addr.sin_addr ), ntohs( new_addr.sin_port), Sock );
737
738         /* Hostnamen ermitteln */
739         s = Resolve_Addr( &new_addr );
740         if( s )
741         {
742                 /* Sub-Prozess wurde asyncron gestartet */
743                 Conn_WriteStr( idx, "NOTICE AUTH :%sLooking up your hostname ...", NOTICE_TXTPREFIX );
744                 My_Connections[idx].res_stat = s;
745         }
746         else
747         {
748                 /* kann Namen nicht aufloesen, daher wird die IP-Adresse verwendet */
749                 strcpy( My_Connections[idx].host, inet_ntoa( new_addr.sin_addr ));
750                 Client_SetHostname( c, My_Connections[idx].host );
751         }
752         
753         /* Penalty-Zeit setzen */
754         Conn_SetPenalty( idx, 1 );
755 } /* New_Connection */
756
757
758 LOCAL CONN_ID
759 Socket2Index( INT Sock )
760 {
761         /* zum Socket passende Connection suchen */
762
763         CONN_ID idx;
764
765         assert( Sock >= 0 );
766
767         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock == Sock ) break;
768
769         assert( idx < MAX_CONNECTIONS );
770         return idx;
771 } /* Socket2Index */
772
773
774 LOCAL VOID
775 Read_Request( CONN_ID Idx )
776 {
777         /* Daten von Socket einlesen und entsprechend behandeln.
778          * Tritt ein Fehler auf, so wird der Socket geschlossen. */
779
780         INT len;
781
782         assert( Idx >= 0 );
783         assert( My_Connections[Idx].sock > NONE );
784
785         if( READBUFFER_LEN - My_Connections[Idx].rdatalen - 2 < 0 )
786         {
787                 /* Der Lesepuffer ist voll */
788                 Log( LOG_ERR, "Read buffer overflow (connection %d): %d bytes!", Idx, My_Connections[Idx].rdatalen );
789                 Conn_Close( Idx, "Read buffer overflow!", NULL, FALSE );
790                 return;
791         }
792
793         len = recv( My_Connections[Idx].sock, My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen, READBUFFER_LEN - My_Connections[Idx].rdatalen - 2, 0 );
794
795         if( len == 0 )
796         {
797                 /* Socket wurde geschlossen */
798                 Log( LOG_INFO, "%s:%d is closing the connection ...", inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port));
799                 Conn_Close( Idx, "Socket closed!", "Client closed connection", FALSE );
800                 return;
801         }
802
803         if( len < 0 )
804         {
805                 /* Fehler beim Lesen */
806                 Log( LOG_ERR, "Read error on connection %d: %s!", Idx, strerror( errno ));
807                 Conn_Close( Idx, "Read error!", "Client closed connection", FALSE );
808                 return;
809         }
810
811         /* Connection-Statistik aktualisieren */
812         My_Connections[Idx].bytes_in += len;
813
814         /* Lesebuffer updaten */
815         My_Connections[Idx].rdatalen += len;
816         assert( My_Connections[Idx].rdatalen < READBUFFER_LEN );
817         My_Connections[Idx].rbuf[My_Connections[Idx].rdatalen] = '\0';
818
819         /* Timestamp aktualisieren */
820         My_Connections[Idx].lastdata = time( NULL );
821
822         Handle_Buffer( Idx );
823 } /* Read_Request */
824
825
826 LOCAL VOID
827 Handle_Buffer( CONN_ID Idx )
828 {
829         /* Daten im Lese-Puffer einer Verbindung verarbeiten. */
830
831         CHAR *ptr, *ptr1, *ptr2;
832         INT len, delta;
833
834         /* Eine komplette Anfrage muss mit CR+LF enden, vgl.
835          * RFC 2812. Haben wir eine? */
836         ptr = strstr( My_Connections[Idx].rbuf, "\r\n" );
837
838         if( ptr ) delta = 2;
839 #ifndef STRICT_RFC
840         else
841         {
842                 /* Nicht RFC-konforme Anfrage mit nur CR oder LF? Leider
843                  * machen soetwas viele Clients, u.a. "mIRC" :-( */
844                 ptr1 = strchr( My_Connections[Idx].rbuf, '\r' );
845                 ptr2 = strchr( My_Connections[Idx].rbuf, '\n' );
846                 delta = 1;
847                 if( ptr1 && ptr2 ) ptr = ptr1 > ptr2 ? ptr2 : ptr1;
848                 else if( ptr1 ) ptr = ptr1;
849                 else if( ptr2 ) ptr = ptr2;
850         }
851 #endif
852
853         if( ptr )
854         {
855                 /* Ende der Anfrage wurde gefunden */
856                 *ptr = '\0';
857                 len = ( ptr - My_Connections[Idx].rbuf ) + delta;
858                 if( len > ( COMMAND_LEN - 1 ))
859                 {
860                         /* Eine Anfrage darf(!) nicht laenger als 512 Zeichen
861                         * (incl. CR+LF!) werden; vgl. RFC 2812. Wenn soetwas
862                         * empfangen wird, wird der Client disconnectiert. */
863                         Log( LOG_ERR, "Request too long (connection %d): %d bytes (max. %d expected)!", Idx, My_Connections[Idx].rdatalen, COMMAND_LEN - 1 );
864                         Conn_Close( Idx, NULL, "Request too long", TRUE );
865                         return;
866                 }
867
868                 if( len > delta )
869                 {
870                         /* Es wurde ein Request gelesen */
871                         if( ! Parse_Request( Idx, My_Connections[Idx].rbuf )) return;
872                 }
873
874                 /* Puffer anpassen */
875                 My_Connections[Idx].rdatalen -= len;
876                 memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen );
877         }
878 } /* Handle_Buffer */
879
880
881 LOCAL VOID
882 Check_Connections( VOID )
883 {
884         /* Pruefen, ob Verbindungen noch "alive" sind. Ist dies
885          * nicht der Fall, zunaechst PING-PONG spielen und, wenn
886          * auch das nicht "hilft", Client disconnectieren. */
887
888         CLIENT *c;
889         INT i;
890
891         for( i = 0; i < MAX_CONNECTIONS; i++ )
892         {
893                 if( My_Connections[i].sock == NONE ) continue;
894
895                 c = Client_GetFromConn( i );
896                 if( c && (( Client_Type( c ) == CLIENT_USER ) || ( Client_Type( c ) == CLIENT_SERVER ) || ( Client_Type( c ) == CLIENT_SERVICE )))
897                 {
898                         /* verbundener User, Server oder Service */
899                         if( My_Connections[i].lastping > My_Connections[i].lastdata )
900                         {
901                                 /* es wurde bereits ein PING gesendet */
902                                 if( My_Connections[i].lastping < time( NULL ) - Conf_PongTimeout )
903                                 {
904                                         /* Timeout */
905                                         Log( LOG_DEBUG, "Connection %d: Ping timeout: %d seconds.", i, Conf_PongTimeout );
906                                         Conn_Close( i, NULL, "Ping timeout", TRUE );
907                                 }
908                         }
909                         else if( My_Connections[i].lastdata < time( NULL ) - Conf_PingTimeout )
910                         {
911                                 /* es muss ein PING gesendet werden */
912                                 Log( LOG_DEBUG, "Connection %d: sending PING ...", i );
913                                 My_Connections[i].lastping = time( NULL );
914                                 Conn_WriteStr( i, "PING :%s", Client_ID( Client_ThisServer( )));
915                         }
916                 }
917                 else
918                 {
919                         /* noch nicht vollstaendig aufgebaute Verbindung */
920                         if( My_Connections[i].lastdata < time( NULL ) - Conf_PingTimeout )
921                         {
922                                 /* Timeout */
923                                 Log( LOG_DEBUG, "Connection %d timed out ...", i );
924                                 Conn_Close( i, NULL, "Timeout", FALSE );
925                         }
926                 }
927         }
928 } /* Check_Connections */
929
930
931 LOCAL VOID
932 Check_Servers( VOID )
933 {
934         /* Pruefen, ob Server-Verbindungen aufgebaut werden
935          * muessen bzw. koennen */
936
937         INT idx, i, n;
938         RES_STAT *s;
939
940         /* Wenn "Passive-Mode" aktiv: nicht verbinden */
941         if( NGIRCd_Passive ) return;
942
943         for( i = 0; i < Conf_Server_Count; i++ )
944         {
945                 /* Ist ein Hostname und Port definiert? */
946                 if(( ! Conf_Server[i].host[0] ) || ( ! Conf_Server[i].port > 0 )) continue;
947
948                 /* Haben wir schon eine Verbindung? */
949                 for( n = 0; n < MAX_CONNECTIONS; n++ )
950                 {
951                         if( My_Connections[n].sock == NONE ) continue;
952                         
953                         /* Verbindung zu diesem Server? */
954                         if( My_Connections[n].our_server == i )
955                         {
956                                 /* Komplett aufgebaute Verbindung? */
957                                 if( My_Connections[n].sock > NONE ) break;
958
959                                 /* IP schon aufgeloest? */
960                                 if( My_Connections[n].res_stat == NULL ) New_Server( i, n );
961                         }
962
963                         /* Verbindung in dieser Server-Gruppe? */
964                         if(( My_Connections[n].our_server != NONE ) && ( Conf_Server[i].group != NONE ))
965                         {
966                                 if( Conf_Server[My_Connections[n].our_server].group == Conf_Server[i].group ) break;
967                         }
968                 }
969                 if( n < MAX_CONNECTIONS ) continue;
970
971                 /* Wann war der letzte Connect-Versuch? */
972                 if( Conf_Server[i].lasttry > time( NULL ) - Conf_ConnectRetry ) continue;
973
974                 /* Okay, Verbindungsaufbau versuchen */
975                 Conf_Server[i].lasttry = time( NULL );
976
977                 /* Freie Connection-Struktur suschen */
978                 for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock == NONE ) break;
979                 if( idx >= MAX_CONNECTIONS )
980                 {
981                         Log( LOG_ALERT, "Can't establist server connection: connection limit reached (%d)!", MAX_CONNECTIONS );
982                         return;
983                 }
984                 Log( LOG_DEBUG, "Preparing connection %d for \"%s\" ...", idx, Conf_Server[i].host );
985
986                 /* Verbindungs-Struktur initialisieren */
987                 Init_Conn_Struct( idx );
988                 My_Connections[idx].sock = SERVER_WAIT;
989                 My_Connections[idx].our_server = i;
990
991                 /* Hostnamen in IP aufloesen */
992                 s = Resolve_Name( Conf_Server[i].host );
993                 if( s )
994                 {
995                         /* Sub-Prozess wurde asyncron gestartet */
996                         My_Connections[idx].res_stat = s;
997                 }
998                 else
999                 {
1000                         /* kann Namen nicht aufloesen: nun versuchen wir einfach,
1001                          * den "Text" direkt als IP-Adresse zu verwenden ... */
1002                         strcpy( Conf_Server[My_Connections[idx].our_server].ip, Conf_Server[i].host );
1003                 }
1004         }
1005 } /* Check_Servers */
1006
1007
1008 LOCAL VOID
1009 New_Server( INT Server, CONN_ID Idx )
1010 {
1011         /* Neue Server-Verbindung aufbauen */
1012
1013         struct sockaddr_in new_addr;
1014         struct in_addr inaddr;
1015         INT new_sock;
1016         CLIENT *c;
1017
1018         assert( Server >= 0 );
1019         assert( Idx >= 0 );
1020
1021         /* Wurde eine gueltige IP-Adresse gefunden? */
1022         if( ! Conf_Server[Server].ip[0] )
1023         {
1024                 /* Nein. Verbindung wieder freigeben: */
1025                 Init_Conn_Struct( Idx );
1026                 Log( LOG_ERR, "Can't connect to \"%s\" (connection %d): ip address unknown!", Conf_Server[Server].host, Idx );
1027                 return;
1028         }
1029
1030         Log( LOG_INFO, "Establishing connection to \"%s\", %s, port %d (connection %d) ... ", Conf_Server[Server].host, Conf_Server[Server].ip, Conf_Server[Server].port, Idx );
1031
1032 #ifdef HAVE_INET_ATON
1033         if( inet_aton( Conf_Server[Server].ip, &inaddr ) == 0 )
1034 #else
1035         memset( &inaddr, 0, sizeof( inaddr ));
1036         inaddr.s_addr = inet_addr( Conf_Server[Server].ip );
1037         if( inaddr.s_addr == (unsigned)-1 )
1038 #endif
1039         {
1040                 /* Konnte Adresse nicht konvertieren */
1041                 Init_Conn_Struct( Idx );
1042                 Log( LOG_ERR, "Can't connect to \"%s\" (connection %d): can't convert ip address %s!", Conf_Server[Server].host, Idx, Conf_Server[Server].ip );
1043                 return;
1044         }
1045
1046         memset( &new_addr, 0, sizeof( new_addr ));
1047         new_addr.sin_family = AF_INET;
1048         new_addr.sin_addr = inaddr;
1049         new_addr.sin_port = htons( Conf_Server[Server].port );
1050
1051         new_sock = socket( PF_INET, SOCK_STREAM, 0 );
1052         if ( new_sock < 0 )
1053         {
1054                 Init_Conn_Struct( Idx );
1055                 Log( LOG_CRIT, "Can't create socket: %s!", strerror( errno ));
1056                 return;
1057         }
1058
1059         if( ! Init_Socket( new_sock )) return;
1060
1061         connect( new_sock, (struct sockaddr *)&new_addr, sizeof( new_addr ));
1062         if( errno != EINPROGRESS )
1063         {
1064
1065                 close( new_sock );
1066                 Init_Conn_Struct( Idx );
1067                 Log( LOG_CRIT, "Can't connect socket: %s!", strerror( errno ));
1068                 return;
1069         }
1070
1071         /* Client-Struktur initialisieren */
1072         c = Client_NewLocal( Idx, inet_ntoa( new_addr.sin_addr ), CLIENT_UNKNOWNSERVER, FALSE );
1073         if( ! c )
1074         {
1075                 close( new_sock );
1076                 Init_Conn_Struct( Idx );
1077                 Log( LOG_ALERT, "Can't establish connection: can't create client structure!" );
1078                 return;
1079         }
1080         Client_SetIntroducer( c, c );
1081         Client_SetToken( c, TOKEN_OUTBOUND );
1082
1083         /* Verbindung registrieren */
1084         My_Connections[Idx].sock = new_sock;
1085         My_Connections[Idx].addr = new_addr;
1086         strcpy( My_Connections[Idx].host, Conf_Server[Server].host );
1087
1088         /* Neuen Socket registrieren */
1089         FD_SET( new_sock, &My_Sockets );
1090         FD_SET( new_sock, &My_Connects );
1091         if( new_sock > Conn_MaxFD ) Conn_MaxFD = new_sock;
1092 } /* New_Server */
1093
1094
1095 LOCAL VOID
1096 Init_Conn_Struct( INT Idx )
1097 {
1098         /* Connection-Struktur initialisieren */
1099
1100         My_Connections[Idx].sock = NONE;
1101         My_Connections[Idx].res_stat = NULL;
1102         My_Connections[Idx].host[0] = '\0';
1103         My_Connections[Idx].rbuf[0] = '\0';
1104         My_Connections[Idx].rdatalen = 0;
1105         My_Connections[Idx].wbuf[0] = '\0';
1106         My_Connections[Idx].wdatalen = 0;
1107         My_Connections[Idx].our_server = NONE;
1108         My_Connections[Idx].lastdata = time( NULL );
1109         My_Connections[Idx].lastping = 0;
1110         My_Connections[Idx].lastprivmsg = time( NULL );
1111         My_Connections[Idx].delaytime = 0;
1112         My_Connections[Idx].bytes_in = 0;
1113         My_Connections[Idx].bytes_out = 0;
1114 } /* Init_Conn_Struct */
1115
1116
1117 LOCAL BOOLEAN
1118 Init_Socket( INT Sock )
1119 {
1120         /* Socket-Optionen setzen */
1121
1122         INT on = 1;
1123
1124 #ifdef O_NONBLOCK       /* A/UX kennt das nicht? */
1125         if( fcntl( Sock, F_SETFL, O_NONBLOCK ) != 0 )
1126         {
1127                 Log( LOG_CRIT, "Can't enable non-blocking mode: %s!", strerror( errno ));
1128                 close( Sock );
1129                 return FALSE;
1130         }
1131 #endif
1132         if( setsockopt( Sock, SOL_SOCKET, SO_REUSEADDR, &on, (socklen_t)sizeof( on )) != 0)
1133         {
1134                 Log( LOG_ERR, "Can't set socket options: %s!", strerror( errno ));
1135                 /* dieser Fehler kann ignoriert werden. */
1136         }
1137
1138         return TRUE;
1139 } /* Init_Socket */
1140
1141
1142 LOCAL VOID
1143 Read_Resolver_Result( INT r_fd )
1144 {
1145         /* Ergebnis von Resolver Sub-Prozess aus Pipe lesen
1146          * und entsprechende Connection aktualisieren */
1147
1148         CHAR result[HOST_LEN];
1149         CLIENT *c;
1150         INT len, i;
1151
1152         FD_CLR( r_fd, &Resolver_FDs );
1153
1154         /* Anfrage vom Parent lesen */
1155         len = read( r_fd, result, HOST_LEN - 1 );
1156         if( len < 0 )
1157         {
1158                 /* Fehler beim Lesen aus der Pipe */
1159                 close( r_fd );
1160                 Log( LOG_CRIT, "Resolver: Can't read result: %s!", strerror( errno ));
1161                 return;
1162         }
1163         result[len] = '\0';
1164
1165         /* zugehoerige Connection suchen */
1166         for( i = 0; i < MAX_CONNECTIONS; i++ )
1167         {
1168                 if(( My_Connections[i].sock != NONE ) && ( My_Connections[i].res_stat ) && ( My_Connections[i].res_stat->pipe[0] == r_fd )) break;
1169         }
1170         if( i >= MAX_CONNECTIONS )
1171         {
1172                 /* Opsa! Keine passende Connection gefunden!? Vermutlich
1173                 * wurde sie schon wieder geschlossen. */
1174                 close( r_fd );
1175                 Log( LOG_DEBUG, "Resolver: Got result for unknown connection!?" );
1176                 return;
1177         }
1178
1179         /* Aufraeumen */
1180         close( My_Connections[i].res_stat->pipe[0] );
1181         close( My_Connections[i].res_stat->pipe[1] );
1182         free( My_Connections[i].res_stat );
1183         My_Connections[i].res_stat = NULL;
1184
1185         if( My_Connections[i].sock > NONE )
1186         {
1187                 /* Eingehende Verbindung: Hostnamen setzen */
1188                 c = Client_GetFromConn( i );
1189                 assert( c != NULL );
1190                 strcpy( My_Connections[i].host, result );
1191                 Client_SetHostname( c, result );
1192
1193                 Conn_WriteStr( i, "NOTICE AUTH :%sGot your hostname.", NOTICE_TXTPREFIX );
1194         }
1195         else
1196         {
1197                 /* Ausgehende Verbindung (=Server): IP setzen */
1198                 assert( My_Connections[i].our_server >= 0 );
1199                 strcpy( Conf_Server[My_Connections[i].our_server].ip, result );
1200         }
1201 } /* Read_Resolver_Result */
1202
1203
1204 /* -eof- */