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.8 2001/12/23 22:02:54 alex Exp $
  13  *
  14  * connect.h: Verwaltung aller Netz-Verbindungen ("connections")
  15  *
  16  * $Log: conn.c,v $
  17  * Revision 1.8  2001/12/23 22:02:54  alex
  18  * - Conn_WriteStr() nimmt nun variable Parameter,
  19  * - diverse kleinere Aenderungen.
  20  *
  21  * Revision 1.7  2001/12/21 22:24:25  alex
  22  * - kleinere Aenderungen an den Log-Meldungen,
  23  * - Parse_Request() wird aufgerufen.
  24  *
  25  * Revision 1.6  2001/12/15 00:11:55  alex
  26  * - Lese- und Schreib-Puffer implementiert.
  27  * - einige neue (Unter-)Funktionen eingefuehrt.
  28  * - diverse weitere kleinere Aenderungen.
  29  *
  30  * Revision 1.5  2001/12/14 08:16:47  alex
  31  * - Begonnen, Client-spezifische Lesepuffer zu implementieren.
  32  * - Umstellung auf Datentyp "CONN_ID".
  33  *
  34  * Revision 1.4  2001/12/13 02:04:16  alex
  35  * - boesen "Speicherschiesser" in Log() gefixt.
  36  *
  37  * Revision 1.3  2001/12/13 01:33:09  alex
  38  * - Conn_Handler() unterstuetzt nun einen Timeout.
  39  * - fuer Verbindungen werden keine FILE-Handles mehr benutzt.
  40  * - kleinere "Code Cleanups" ;-)
  41  *
  42  * Revision 1.2  2001/12/12 23:32:02  alex
  43  * - diverse Erweiterungen und Verbesserungen (u.a. sind nun mehrere
  44  *   Verbindungen und Listen-Sockets moeglich).
  45  *
  46  * Revision 1.1  2001/12/12 17:18:38  alex
  47  * - Modul zur Verwaltung aller Netzwerk-Verbindungen begonnen.
  48  */
  49
  50
  51 #include <portab.h>
  52 #include "global.h"
  53
  54 #include <imp.h>
  55 #include <assert.h>
  56 #include <stdarg.h>
  57 #include <stdio.h>
  58 #include <unistd.h>
  59 #include <errno.h>
  60 #include <fcntl.h>
  61 #include <string.h>
  62 #include <sys/socket.h>
  63 #include <sys/time.h>
  64 #include <sys/types.h>
  65 #include <netinet/in.h>
  66 #include <arpa/inet.h>
  67
  68 #ifdef HAVE_STDINT_H
  69 #include <stdint.h>                     /* u.a. fuer Mac OS X */
  70 #endif
  71
  72 #include "ngircd.h"
  73 #include "client.h"
  74 #include "log.h"
  75 #include "parse.h"
  76 #include "tool.h"
  77
  78 #include <exp.h>
  79 #include "conn.h"
  80
  81
  82 #define MAX_CONNECTIONS 100             /* max. Anzahl von Verbindungen an diesem Server */
  83
  84 #define MAX_CMDLEN 512                  /* max. Laenge eines Befehls, vgl. RFC 2812, 3.2 */
  85
  86 #define READBUFFER_LEN 2 * MAX_CMDLEN   /* Laenge des Lesepuffers je Verbindung (Bytes) */
  87 #define WRITEBUFFER_LEN 4096            /* Laenge des Schreibpuffers je Verbindung (Bytes) */
  88
  89
  90 typedef struct _Connection
  91 {
  92         INT sock;                       /* Socket Handle */
  93         struct sockaddr_in addr;        /* Adresse des Client */
  94         CHAR rbuf[READBUFFER_LEN + 1];  /* Lesepuffer */
  95         INT rdatalen;                   /* Laenge der Daten im Lesepuffer */
  96         CHAR wbuf[WRITEBUFFER_LEN + 1]; /* Schreibpuffer */
  97         INT wdatalen;                   /* Laenge der Daten im Schreibpuffer */
  98 } CONNECTION;
  99
 100
 101 LOCAL VOID Handle_Read( INT sock );
 102 LOCAL BOOLEAN Handle_Write( CONN_ID Idx );
 103 LOCAL VOID New_Connection( INT Sock );
 104 LOCAL CONN_ID Socket2Index( INT Sock );
 105 LOCAL VOID Close_Connection( CONN_ID Idx, CHAR *Msg );
 106 LOCAL VOID Read_Request( CONN_ID Idx );
 107 LOCAL BOOLEAN Try_Write( CONN_ID Idx );
 108
 109
 110 LOCAL fd_set My_Listener;
 111 LOCAL fd_set My_Sockets;
 112
 113 LOCAL INT My_Max_Fd;
 114
 115 LOCAL CONNECTION My_Connections[MAX_CONNECTIONS];
 116
 117
 118 GLOBAL VOID Conn_Init( VOID )
 119 {
 120         CONN_ID i;
 121
 122         /* zu Beginn haben wir keine Verbindungen */
 123         FD_ZERO( &My_Listener );
 124         FD_ZERO( &My_Sockets );
 125
 126         My_Max_Fd = 0;
 127
 128         /* Connection-Struktur initialisieren */
 129         for( i = 0; i < MAX_CONNECTIONS; i++ ) My_Connections[i].sock = NONE;
 130 } /* Conn_Init */
 131
 132
 133 GLOBAL VOID Conn_Exit( VOID )
 134 {
 135         CONN_ID idx;
 136         INT i;
 137
 138         /* Sockets schliessen */
 139         for( i = 0; i < My_Max_Fd + 1; i++ )
 140         {
 141                 if( FD_ISSET( i, &My_Sockets ))
 142                 {
 143                         for( idx = 0; idx < MAX_CONNECTIONS; idx++ )
 144                         {
 145                                 if( My_Connections[idx].sock == i ) break;
 146                         }
 147                         if( idx < MAX_CONNECTIONS ) Close_Connection( idx, "Server going down ..." );
 148                         else if( FD_ISSET( i, &My_Listener ))
 149                         {
 150                                 close( i );
 151                                 Log( LOG_INFO, "Listening socket %d closed.", i );
 152                         }
 153                         else
 154                         {
 155                                 close( i );
 156                                 Log( LOG_WARNING, "Unknown connection %d closed.", i );
 157                         }
 158                 }
 159         }
 160 } /* Conn_Exit */
 161
 162
 163 GLOBAL BOOLEAN Conn_New_Listener( CONST INT Port )
 164 {
 165         /* Neuen Listen-Socket erzeugen: der Server wartet dann
 166          * auf dem angegebenen Port auf Verbindungen. */
 167
 168         struct sockaddr_in addr;
 169         INT sock, on = 1;
 170
 171         /* Server-"Listen"-Socket initialisieren */
 172         memset( &addr, 0, sizeof( addr ));
 173         addr.sin_family = AF_INET;
 174         addr.sin_port = htons( Port );
 175         addr.sin_addr.s_addr = htonl( INADDR_ANY );
 176
 177         /* Socket erzeugen */
 178         sock = socket( PF_INET, SOCK_STREAM, 0);
 179         if( sock < 0 )
 180         {
 181                 Log( LOG_ALERT, "Can't create socket: %s", strerror( errno ));
 182                 return FALSE;
 183         }
 184
 185         /* Socket-Optionen setzen */
 186         if( fcntl( sock, F_SETFL, O_NONBLOCK ) != 0 )
 187         {
 188                 Log( LOG_ALERT, "Can't enable non-blocking mode: %s", strerror( errno ));
 189                 close( sock );
 190                 return FALSE;
 191         }
 192         if( setsockopt( sock, SOL_SOCKET, SO_REUSEADDR, &on, (socklen_t)sizeof( on )) != 0)
 193         {
 194                 Log( LOG_CRIT, "Can't set socket options: %s", strerror( errno ));
 195                 /* dieser Fehler kann ignoriert werden. */
 196         }
 197
 198         /* an Port binden */
 199         if( bind( sock, (struct sockaddr *)&addr, (socklen_t)sizeof( addr )) != 0 )
 200         {
 201                 Log( LOG_ALERT, "Can't bind socket: %s", strerror( errno ));
 202                 close( sock );
 203                 return FALSE;
 204         }
 205
 206         /* in "listen mode" gehen :-) */
 207         if( listen( sock, 10 ) != 0 )
 208         {
 209                 Log( LOG_ALERT, "Can't listen on soecket: %s", strerror( errno ));
 210                 close( sock );
 211                 return FALSE;
 212         }
 213
 214         /* Neuen Listener in Strukturen einfuegen */
 215         FD_SET( sock, &My_Listener );
 216         FD_SET( sock, &My_Sockets );
 217
 218         if( sock > My_Max_Fd ) My_Max_Fd = sock;
 219
 220         Log( LOG_INFO, "Now listening on port %d, socket %d.", Port, sock );
 221
 222         return TRUE;
 223 } /* Conn_New_Listener */
 224
 225
 226 GLOBAL VOID Conn_Handler( INT Timeout )
 227 {
 228         fd_set read_sockets, write_sockets;
 229         struct timeval tv;
 230         INT i;
 231
 232         /* Timeout initialisieren */
 233         tv.tv_sec = Timeout;
 234         tv.tv_usec = 0;
 235
 236         /* noch volle Schreib-Puffer suchen */
 237         FD_ZERO( &write_sockets );
 238         for( i = 0; i < MAX_CONNECTIONS; i++ )
 239         {
 240                 if(( My_Connections[i].sock >= 0 ) && ( My_Connections[i].wdatalen > 0 ))
 241                 {
 242                         /* Socket der Verbindung in Set aufnehmen */
 243                         FD_SET( My_Connections[i].sock, &write_sockets );
 244                 }
 245         }
 246
 247         read_sockets = My_Sockets;
 248         if( select( My_Max_Fd + 1, &read_sockets, &write_sockets, NULL, &tv ) == -1 )
 249         {
 250                 if( errno != EINTR ) Log( LOG_ALERT, "select(): %s", strerror( errno ));
 251                 return;
 252         }
 253
 254         /* Koennen Daten geschrieben werden? */
 255         for( i = 0; i < My_Max_Fd + 1; i++ )
 256         {
 257                 if( FD_ISSET( i, &write_sockets )) Handle_Write( Socket2Index( i ));
 258         }
 259
 260         /* Daten zum Lesen vorhanden? */
 261         for( i = 0; i < My_Max_Fd + 1; i++ )
 262         {
 263                 if( FD_ISSET( i, &read_sockets )) Handle_Read( i );
 264         }
 265 } /* Conn_Handler */
 266
 267
 268 GLOBAL BOOLEAN Conn_WriteStr( CONN_ID Idx, CHAR *Format, ... )
 269 {
 270         /* String in Socket schreiben. CR+LF wird von dieser Funktion
 271          * automatisch angehaengt. Im Fehlerfall wird dir Verbindung
 272          * getrennt und FALSE geliefert. */
 273
 274         CHAR buffer[MAX_CMDLEN];
 275         BOOLEAN ok;
 276         va_list ap;
 277
 278         va_start( ap, Format );
 279         if( vsnprintf( buffer, MAX_CMDLEN, Format, ap ) == MAX_CMDLEN )
 280         {
 281                 Log( LOG_ALERT, "String too long to send (connection %d)!", Idx );
 282                 Close_Connection( Idx, "Server error: String too long to send!" );
 283                 return FALSE;
 284         }
 285         ok = Conn_Write( Idx, buffer, strlen( buffer ));
 286
 287 #ifdef DEBUG
 288         Log( LOG_DEBUG, " -> connection %d: '%s'.", Idx, buffer );
 289 #endif
 290
 291         va_end( ap );
 292         return ok;
 293 } /* Conn_WriteStr */
 294
 295
 296 GLOBAL BOOLEAN Conn_Write( CONN_ID Idx, CHAR *Data, INT Len )
 297 {
 298         /* Daten in Socket schreiben. Bei "fatalen" Fehlern wird
 299          * der Client disconnectiert und FALSE geliefert. */
 300
 301         assert( Idx >= 0 );
 302         assert( My_Connections[Idx].sock >= 0 );
 303         assert( Data != NULL );
 304         assert( Len > 0 );
 305
 306         /* pruefen, ob Daten im Schreibpuffer sind. Wenn ja, zunaechst
 307          * pruefen, ob diese gesendet werden koennen */
 308         if( My_Connections[Idx].wdatalen > 0 )
 309         {
 310                 if( ! Try_Write( Idx )) return FALSE;
 311         }
 312
 313         /* pruefen, ob im Schreibpuffer genuegend Platz ist */
 314         if( WRITEBUFFER_LEN - My_Connections[Idx].wdatalen - Len <= 0 )
 315         {
 316                 /* der Puffer ist dummerweise voll ... */
 317                 Log( LOG_NOTICE, "Write buffer overflow (connection %d)!", Idx );
 318                 Close_Connection( Idx, NULL );
 319                 return FALSE;
 320         }
 321
 322         /* Daten in Puffer kopieren */
 323         memcpy( My_Connections[Idx].wbuf + My_Connections[Idx].wdatalen, Data, Len );
 324         My_Connections[Idx].wdatalen += Len;
 325
 326         /* pruefen, on Daten vorhanden sind und geschrieben werden koennen */
 327         if( My_Connections[Idx].wdatalen > 0 )
 328         {
 329                 if( ! Try_Write( Idx )) return FALSE;
 330         }
 331
 332         return TRUE;
 333 } /* Conn_Write */
 334
 335
 336 LOCAL BOOLEAN Try_Write( CONN_ID Idx )
 337 {
 338         /* Versuchen, Daten aus dem Schreib-Puffer in den
 339          * Socket zu schreiben. */
 340
 341         fd_set write_socket;
 342
 343         assert( Idx >= 0 );
 344         assert( My_Connections[Idx].sock >= 0 );
 345         assert( My_Connections[Idx].wdatalen > 0 );
 346
 347         FD_ZERO( &write_socket );
 348         FD_SET( My_Connections[Idx].sock, &write_socket );
 349         if( select( My_Connections[Idx].sock + 1, NULL, &write_socket, NULL, 0 ) == -1 )
 350         {
 351                 /* Fehler! */
 352                 if( errno != EINTR )
 353                 {
 354                         Log( LOG_ALERT, "select(): %s", strerror( errno ));
 355                         Close_Connection( Idx, NULL );
 356                         return FALSE;
 357                 }
 358         }
 359
 360         if( FD_ISSET( My_Connections[Idx].sock, &write_socket )) return Handle_Write( Idx );
 361         else return TRUE;
 362 } /* Try_Write */
 363
 364
 365 LOCAL VOID Handle_Read( INT Sock )
 366 {
 367         /* Aktivitaet auf einem Socket verarbeiten */
 368
 369         CONN_ID idx;
 370
 371         assert( Sock >= 0 );
 372
 373         if( FD_ISSET( Sock, &My_Listener ))
 374         {
 375                 /* es ist einer unserer Listener-Sockets: es soll
 376                  * also eine neue Verbindung aufgebaut werden. */
 377
 378                 New_Connection( Sock );
 379         }
 380         else
 381         {
 382                 /* Ein Client Socket: entweder ein User oder Server */
 383
 384                 idx = Socket2Index( Sock );
 385                 Read_Request( idx );
 386         }
 387 } /* Handle_Read */
 388
 389
 390 LOCAL BOOLEAN Handle_Write( CONN_ID Idx )
 391 {
 392         /* Daten aus Schreibpuffer versenden */
 393
 394         INT len;
 395
 396         assert( Idx >= 0 );
 397         assert( My_Connections[Idx].sock >= 0 );
 398         assert( My_Connections[Idx].wdatalen > 0 );
 399
 400         /* Daten schreiben */
 401         len = send( My_Connections[Idx].sock, My_Connections[Idx].wbuf, My_Connections[Idx].wdatalen, 0 );
 402         if( len < 0 )
 403         {
 404                 /* Oops, ein Fehler! */
 405                 Log( LOG_ALERT, "Write error (buffer) on connection %d: %s", Idx, strerror( errno ));
 406                 Close_Connection( Idx, NULL );
 407                 return FALSE;
 408         }
 409
 410         /* Puffer anpassen */
 411         My_Connections[Idx].wdatalen -= len;
 412         memmove( My_Connections[Idx].wbuf, My_Connections[Idx].wbuf + len, My_Connections[Idx].wdatalen );
 413
 414         return TRUE;
 415 } /* Handle_Write */
 416
 417
 418 LOCAL VOID New_Connection( INT Sock )
 419 {
 420         /* Neue Client-Verbindung von Listen-Socket annehmen und
 421          * CLIENT-Struktur anlegen. */
 422
 423         struct sockaddr_in new_addr;
 424         INT new_sock, new_sock_len;
 425         CONN_ID idx;
 426
 427         assert( Sock >= 0 );
 428
 429         new_sock_len = sizeof( new_addr );
 430         new_sock = accept( Sock, (struct sockaddr *)&new_addr, (socklen_t *)&new_sock_len );
 431         if( new_sock < 0 )
 432         {
 433                 Log( LOG_CRIT, "Can't accept connection: %s", strerror( errno ));
 434                 return;
 435         }
 436
 437         /* Freie Connection-Struktur suschen */
 438         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock < 0 ) break;
 439         if( idx >= MAX_CONNECTIONS )
 440         {
 441                 Log( LOG_ALERT, "Can't accept connection: limit reached (%d)!", MAX_CONNECTIONS );
 442                 close( new_sock );
 443                 return;
 444         }
 445
 446         /* Client-Struktur initialisieren */
 447         if( ! Client_New_Local( idx, inet_ntoa( new_addr.sin_addr )))
 448         {
 449                 Log( LOG_ALERT, "Can't accept connection: can't create client structure!" );
 450                 close( new_sock );
 451                 return;
 452         }
 453
 454         /* Verbindung registrieren */
 455         My_Connections[idx].sock = new_sock;
 456         My_Connections[idx].addr = new_addr;
 457         My_Connections[idx].rdatalen = 0;
 458         My_Connections[idx].wdatalen = 0;
 459
 460         /* Neuen Socket registrieren */
 461         FD_SET( new_sock, &My_Sockets );
 462
 463         if( new_sock > My_Max_Fd ) My_Max_Fd = new_sock;
 464
 465         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 );
 466 } /* New_Connection */
 467
 468
 469 LOCAL CONN_ID Socket2Index( INT Sock )
 470 {
 471         /* zum Socket passende Connection suchen */
 472
 473         CONN_ID idx;
 474
 475         assert( Sock >= 0 );
 476
 477         for( idx = 0; idx < MAX_CONNECTIONS; idx++ ) if( My_Connections[idx].sock == Sock ) break;
 478
 479         assert( idx < MAX_CONNECTIONS );
 480         return idx;
 481 } /* Socket2Index */
 482
 483
 484 LOCAL VOID Close_Connection( CONN_ID Idx, CHAR *Msg )
 485 {
 486         /* Verbindung schlie§en */
 487
 488         assert( Idx >= 0 );
 489         assert( My_Connections[Idx].sock >= 0 );
 490
 491         if( Msg ) Conn_WriteStr( Idx, "ERROR :%s", Msg );
 492
 493         if( close( My_Connections[Idx].sock ) != 0 )
 494         {
 495                 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 ));
 496         }
 497         else
 498         {
 499                 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 ));
 500         }
 501
 502         Client_Destroy( Client_GetFromConn( Idx ));
 503
 504         FD_CLR( My_Connections[Idx].sock, &My_Sockets );
 505         My_Connections[Idx].sock = NONE;
 506 } /* Close_Connection */
 507
 508
 509 LOCAL VOID Read_Request( CONN_ID Idx )
 510 {
 511         /* Daten von Socket einlesen und entsprechend behandeln.
 512          * Tritt ein Fehler auf, so wird der Socket geschlossen. */
 513
 514         INT len;
 515         CHAR *ptr;
 516
 517         assert( Idx >= 0 );
 518         assert( My_Connections[Idx].sock >= 0 );
 519
 520         len = recv( My_Connections[Idx].sock, My_Connections[Idx].rbuf + My_Connections[Idx].rdatalen, READBUFFER_LEN - My_Connections[Idx].rdatalen, 0 );
 521         My_Connections[Idx].rbuf[READBUFFER_LEN] = '\0';
 522
 523         if( len == 0 )
 524         {
 525                 /* Socket wurde geschlossen */
 526                 Log( LOG_INFO, "%s:%d is closing the connection ...", inet_ntoa( My_Connections[Idx].addr.sin_addr ), ntohs( My_Connections[Idx].addr.sin_port));
 527                 Close_Connection( Idx, NULL );
 528                 return;
 529         }
 530
 531         if( len < 0 )
 532         {
 533                 /* Fehler beim Lesen */
 534                 Log( LOG_ALERT, "Read error on connection %d!", Idx );
 535                 Close_Connection( Idx, "Read error!" );
 536                 return;
 537         }
 538
 539         My_Connections[Idx].rdatalen += len;
 540         assert( My_Connections[Idx].rdatalen <= READBUFFER_LEN );
 541         My_Connections[Idx].rbuf[My_Connections[Idx].rdatalen] = '\0';
 542
 543         if( My_Connections[Idx].rdatalen > MAX_CMDLEN )
 544         {
 545                 /* Eine Anfrage darf(!) nicht laenger als 512 Zeichen
 546                  * (incl. CR+LF!) werden; vgl. RFC 2812. Wenn soetwas
 547                  * empfangen wird, wird der Client disconnectiert. */
 548                 Log( LOG_ALERT, "Request too long (connection %d)!", Idx );
 549                 Close_Connection( Idx, "Request too long!" );
 550                 return;
 551         }
 552
 553         /* Eine komplette Anfrage muss mit CR+LF enden, vgl.
 554          * RFC 2812. Haben wir eine? */
 555         ptr = strstr( My_Connections[Idx].rbuf, "\r\n" );
 556         if( ptr )
 557         {
 558                 /* Ende der Anfrage (CR+LF) wurde gefunden */
 559                 *ptr = '\0';
 560                 len = ( ptr - My_Connections[Idx].rbuf ) + 2;
 561                 if( len > 2 )
 562                 {
 563                         /* Es wurde ein Request gelesen */
 564                         if( ! Parse_Request( Idx, My_Connections[Idx].rbuf )) return;
 565                 }
 566                 else Log( LOG_DEBUG, "Got null-request (connection %d).", Idx );
 567
 568                 /* Puffer anpassen */
 569                 My_Connections[Idx].rdatalen -= len;
 570                 memmove( My_Connections[Idx].rbuf, My_Connections[Idx].rbuf + len, My_Connections[Idx].rdatalen );
 571         }
 572 } /* Read_Request */
 573
 574
 575 /* -eof- */