/[cvs]/eggdrop1.9/src/net.c
ViewVC logotype

Annotation of /eggdrop1.9/src/net.c

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph


Revision 1.25 - (hide annotations) (download) (as text)
Fri Oct 27 19:35:51 2000 UTC (18 years, 7 months ago) by fabian
Branch: MAIN
Changes since 1.24: +95 -31 lines
File MIME type: text/x-chdr
fabian: applied save_data patch

1 fabian 1.11 /*
2 segfault 1.1 * net.c -- handles:
3 fabian 1.11 * all raw network i/o
4 segfault 1.1 *
5 fabian 1.25 * $Id: net.c,v 1.24 2000/10/27 19:34:54 fabian Exp $
6 fabian 1.11 */
7     /*
8 segfault 1.1 * This is hereby released into the public domain.
9     * Robey Pointer, robey@netcom.com
10     */
11    
12 fabian 1.16 #include <fcntl.h>
13 segfault 1.1 #include "main.h"
14     #include <limits.h>
15     #include <string.h>
16     #include <netdb.h>
17     #include <sys/socket.h>
18     #if HAVE_SYS_SELECT_H
19 fabian 1.16 # include <sys/select.h>
20 segfault 1.1 #endif
21     #include <netinet/in.h>
22     #include <arpa/inet.h> /* is this really necessary? */
23     #include <errno.h>
24     #if HAVE_UNISTD_H
25 fabian 1.16 # include <unistd.h>
26 segfault 1.1 #endif
27     #include <setjmp.h>
28    
29     #if !HAVE_GETDTABLESIZE
30 fabian 1.16 # ifdef FD_SETSIZE
31     # define getdtablesize() FD_SETSIZE
32     # else
33     # define getdtablesize() 200
34     # endif
35 segfault 1.1 #endif
36    
37 fabian 1.16 extern struct dcc_t *dcc;
38     extern int backgrd, use_stderr, resolve_timeout, dcc_total,
39     otraffic_irc_today, otraffic_bn_today,
40     otraffic_dcc_today, otraffic_filesys_today,
41     otraffic_trans_today, otraffic_unknown_today;
42    
43     char hostname[121] = ""; /* Hostname can be specified in the config
44     file */
45     char myip[121] = ""; /* IP can be specified in the config file */
46     char firewall[121] = ""; /* Socks server for firewall */
47     int firewallport = 1080; /* Default port of Sock4/5 firewalls */
48     char botuser[21] = "eggdrop"; /* Username of the user running the bot */
49     int dcc_sanitycheck = 0; /* We should do some sanity checking on dcc
50     connections. */
51     sock_list *socklist = NULL; /* Enough to be safe */
52     int MAXSOCKS = 0;
53     jmp_buf alarmret; /* Env buffer for alarm() returns */
54 segfault 1.1
55 fabian 1.16 /* Types of proxy */
56 segfault 1.1 #define PROXY_SOCKS 1
57     #define PROXY_SUN 2
58    
59    
60 fabian 1.16 /* I need an UNSIGNED long for dcc type stuff
61     */
62 segfault 1.1 IP my_atoul(char *s)
63     {
64     IP ret = 0;
65    
66     while ((*s >= '0') && (*s <= '9')) {
67     ret *= 10;
68     ret += ((*s) - '0');
69     s++;
70     }
71     return ret;
72     }
73    
74 fabian 1.16 /* Initialize the socklist
75     */
76 segfault 1.1 void init_net()
77     {
78     int i;
79    
80     for (i = 0; i < MAXSOCKS; i++) {
81     socklist[i].flags = SOCK_UNUSED;
82     }
83     }
84    
85     int expmem_net()
86     {
87     int i, tot = 0;
88    
89 fabian 1.9 Context;
90 segfault 1.1 for (i = 0; i < MAXSOCKS; i++) {
91     if (!(socklist[i].flags & SOCK_UNUSED)) {
92     if (socklist[i].inbuf != NULL)
93     tot += strlen(socklist[i].inbuf) + 1;
94     if (socklist[i].outbuf != NULL)
95     tot += socklist[i].outbuflen;
96     }
97     }
98     return tot;
99     }
100    
101 fabian 1.16 /* Get my ip number
102     */
103 segfault 1.1 IP getmyip()
104     {
105     struct hostent *hp;
106     char s[121];
107     IP ip;
108     struct in_addr *in;
109    
110 fabian 1.16 /* Could be pre-defined */
111 segfault 1.1 if (myip[0]) {
112     if ((myip[strlen(myip) - 1] >= '0') && (myip[strlen(myip) - 1] <= '9'))
113     return (IP) inet_addr(myip);
114     }
115 fabian 1.16 /* Also could be pre-defined */
116 segfault 1.1 if (hostname[0])
117     hp = gethostbyname(hostname);
118     else {
119     gethostname(s, 120);
120     hp = gethostbyname(s);
121     }
122     if (hp == NULL)
123     fatal("Hostname self-lookup failed.", 0);
124     in = (struct in_addr *) (hp->h_addr_list[0]);
125     ip = (IP) (in->s_addr);
126     return ip;
127     }
128    
129     void neterror(char *s)
130     {
131     switch (errno) {
132     case EADDRINUSE:
133     strcpy(s, "Address already in use");
134     break;
135     case EADDRNOTAVAIL:
136     strcpy(s, "Address invalid on remote machine");
137     break;
138     case EAFNOSUPPORT:
139     strcpy(s, "Address family not supported");
140     break;
141     case EALREADY:
142     strcpy(s, "Socket already in use");
143     break;
144     case EBADF:
145     strcpy(s, "Socket descriptor is bad");
146     break;
147     case ECONNREFUSED:
148     strcpy(s, "Connection refused");
149     break;
150     case EFAULT:
151     strcpy(s, "Namespace segment violation");
152     break;
153     case EINPROGRESS:
154     strcpy(s, "Operation in progress");
155     break;
156     case EINTR:
157     strcpy(s, "Timeout");
158     break;
159     case EINVAL:
160     strcpy(s, "Invalid namespace");
161     break;
162     case EISCONN:
163     strcpy(s, "Socket already connected");
164     break;
165     case ENETUNREACH:
166     strcpy(s, "Network unreachable");
167     break;
168     case ENOTSOCK:
169     strcpy(s, "File descriptor, not a socket");
170     break;
171     case ETIMEDOUT:
172     strcpy(s, "Connection timed out");
173     break;
174     case ENOTCONN:
175     strcpy(s, "Socket is not connected");
176     break;
177     case EHOSTUNREACH:
178     strcpy(s, "Host is unreachable");
179     break;
180     case EPIPE:
181     strcpy(s, "Broken pipe");
182     break;
183     #ifdef ECONNRESET
184     case ECONNRESET:
185     strcpy(s, "Connection reset by peer");
186     break;
187     #endif
188     #ifdef EACCES
189     case EACCES:
190     strcpy(s, "Permission denied");
191     break;
192     #endif
193     case 0:
194     strcpy(s, "Error 0");
195     break;
196     default:
197     sprintf(s, "Unforseen error %d", errno);
198     break;
199     }
200     }
201    
202 fabian 1.25 /* Sets/Unsets options for a specific socket.
203     *
204     * Returns: 0 - on success
205     * -1 - socket not found
206     * -2 - illegal operation
207     */
208     int sockoptions(int sock, int operation, int sock_options)
209     {
210     int i;
211    
212     Context;
213     for (i = 0; i < MAXSOCKS; i++)
214     if (socklist[i].sock == sock) {
215     if (operation == EGG_OPTION_SET)
216     socklist[i].flags |= sock_options;
217     else if (operation == EGG_OPTION_UNSET)
218     socklist[i].flags &= ~sock_options;
219     else
220     return -2;
221     return 0;
222     }
223     return -1;
224     }
225    
226 fabian 1.16 /* Return a free entry in the socket entry
227     */
228 fabian 1.3 int allocsock(int sock, int options)
229 segfault 1.1 {
230     int i;
231    
232     for (i = 0; i < MAXSOCKS; i++) {
233     if (socklist[i].flags & SOCK_UNUSED) {
234     /* yay! there is table space */
235     socklist[i].inbuf = socklist[i].outbuf = NULL;
236 fabian 1.25 socklist[i].inbuflen = socklist[i].outbuflen = 0;
237 segfault 1.1 socklist[i].flags = options;
238     socklist[i].sock = sock;
239 fabian 1.3 return i;
240     }
241     }
242     fatal("Socket table is full!", 0);
243 fabian 1.16 return -1; /* Never reached */
244 fabian 1.3 }
245    
246 fabian 1.16 /* Request a normal socket for i/o
247     */
248 fabian 1.3 void setsock(int sock, int options)
249     {
250     int i = allocsock(sock, options);
251     int parm;
252    
253 fabian 1.16 if (((sock != STDOUT) || backgrd) &&
254     !(socklist[i].flags & SOCK_NONSOCK)) {
255     parm = 1;
256 fabian 1.3 setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (void *) &parm, sizeof(int));
257 segfault 1.1
258 fabian 1.16 parm = 0;
259     setsockopt(sock, SOL_SOCKET, SO_LINGER, (void *) &parm, sizeof(int));
260     }
261     if (options & SOCK_LISTEN) {
262     /* Tris says this lets us grab the same port again next time */
263     parm = 1;
264     setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void *) &parm, sizeof(int));
265     }
266     /* Yay async i/o ! */
267     fcntl(sock, F_SETFL, O_NONBLOCK);
268 segfault 1.1 }
269    
270     int getsock(int options)
271     {
272     int sock = socket(AF_INET, SOCK_STREAM, 0);
273    
274 fabian 1.24 if (sock >= 0)
275     setsock(sock, options);
276     else
277     putlog(LOG_MISC, "*", "Warning: Can't create new socket!");
278 segfault 1.1 return sock;
279     }
280    
281 fabian 1.16 /* Done with a socket
282     */
283 fabian 1.24 void killsock(register int sock)
284 segfault 1.1 {
285 fabian 1.24 register int i;
286 segfault 1.1
287 fabian 1.24 /* Ignore invalid sockets. */
288     if (sock < 0)
289     return;
290 segfault 1.1 for (i = 0; i < MAXSOCKS; i++) {
291 fabian 1.2 if ((socklist[i].sock == sock) && !(socklist[i].flags & SOCK_UNUSED)) {
292 segfault 1.1 close(socklist[i].sock);
293     if (socklist[i].inbuf != NULL) {
294     nfree(socklist[i].inbuf);
295     socklist[i].inbuf = NULL;
296     }
297     if (socklist[i].outbuf != NULL) {
298     nfree(socklist[i].outbuf);
299     socklist[i].outbuf = NULL;
300     socklist[i].outbuflen = 0;
301     }
302     socklist[i].flags = SOCK_UNUSED;
303     return;
304     }
305     }
306     putlog(LOG_MISC, "*", "Attempt to kill un-allocated socket %d !!", sock);
307     }
308    
309 fabian 1.16 /* Send connection request to proxy
310     */
311 segfault 1.1 static int proxy_connect(int sock, char *host, int port, int proxy)
312     {
313     unsigned char x[10];
314     struct hostent *hp;
315     char s[30];
316     int i;
317    
318     /* socks proxy */
319     if (proxy == PROXY_SOCKS) {
320     /* numeric IP? */
321     if ((host[strlen(host) - 1] >= '0') && (host[strlen(host) - 1] <= '9')) {
322     IP ip = ((IP) inet_addr(host)); /* drummer */
323 fabian 1.18 egg_memcpy(x, &ip, 4); /* Beige@Efnet */
324 segfault 1.1 } else {
325     /* no, must be host.domain */
326     if (!setjmp(alarmret)) {
327     alarm(resolve_timeout);
328     hp = gethostbyname(host);
329     alarm(0);
330     } else {
331     hp = NULL;
332     }
333     if (hp == NULL) {
334     killsock(sock);
335     return -2;
336     }
337 fabian 1.18 egg_memcpy(x, hp->h_addr, hp->h_length);
338 segfault 1.1 }
339     for (i = 0; i < MAXSOCKS; i++) {
340     if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == sock))
341     socklist[i].flags |= SOCK_PROXYWAIT; /* drummer */
342     }
343     sprintf(s, "\004\001%c%c%c%c%c%c%s", (port >> 8) % 256, (port % 256),
344     x[0], x[1], x[2], x[3], botuser);
345     tputs(sock, s, strlen(botuser) + 9); /* drummer */
346     } else if (proxy == PROXY_SUN) {
347     sprintf(s, "%s %d\n", host, port);
348     tputs(sock, s, strlen(s)); /* drummer */
349     }
350     return sock;
351     }
352    
353 fabian 1.16 /* Starts a connection attempt to a socket
354 fabian 1.11 *
355 fabian 1.3 * If given a normal hostname, this will be resolved to the corresponding
356     * IP address first. PLEASE try to use the non-blocking dns functions
357     * instead and then call this function with the IP address to avoid blocking.
358 fabian 1.11 *
359 segfault 1.1 * returns <0 if connection refused:
360     * -1 neterror() type error
361 fabian 1.16 * -2 can't resolve hostname
362     */
363 segfault 1.1 int open_telnet_raw(int sock, char *server, int sport)
364     {
365     struct sockaddr_in name;
366     struct hostent *hp;
367     char host[121];
368     int i, port;
369     volatile int proxy;
370    
371     /* firewall? use socks */
372     if (firewall[0]) {
373     if (firewall[0] == '!') {
374     proxy = PROXY_SUN;
375     strcpy(host, &firewall[1]);
376     } else {
377     proxy = PROXY_SOCKS;
378     strcpy(host, firewall);
379     }
380     port = firewallport;
381     } else {
382     proxy = 0;
383     strcpy(host, server);
384     port = sport;
385     }
386     /* patch by tris for multi-hosted machines: */
387 fabian 1.18 egg_bzero((char *) &name, sizeof(struct sockaddr_in));
388 segfault 1.1
389     name.sin_family = AF_INET;
390     name.sin_addr.s_addr = (myip[0] ? getmyip() : INADDR_ANY);
391 fabian 1.4 if (bind(sock, (struct sockaddr *) &name, sizeof(name)) < 0)
392 segfault 1.1 return -1;
393 fabian 1.18 egg_bzero((char *) &name, sizeof(struct sockaddr_in));
394 segfault 1.1
395     name.sin_family = AF_INET;
396 fabian 1.21 name.sin_port = htons(port);
397 fabian 1.16 /* Numeric IP? */
398 segfault 1.1 if ((host[strlen(host) - 1] >= '0') && (host[strlen(host) - 1] <= '9'))
399     name.sin_addr.s_addr = inet_addr(host);
400     else {
401 fabian 1.16 /* No, must be host.domain */
402     debug0("WARNING: open_telnet_raw() is about to block in gethostbyname()!");
403 segfault 1.1 if (!setjmp(alarmret)) {
404     alarm(resolve_timeout);
405     hp = gethostbyname(host);
406     alarm(0);
407     } else {
408     hp = NULL;
409     }
410 fabian 1.4 if (hp == NULL)
411 segfault 1.1 return -2;
412 fabian 1.18 egg_memcpy(&name.sin_addr, hp->h_addr, hp->h_length);
413 segfault 1.1 name.sin_family = hp->h_addrtype;
414     }
415     for (i = 0; i < MAXSOCKS; i++) {
416     if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == sock))
417 fabian 1.17 socklist[i].flags = (socklist[i].flags & ~SOCK_VIRTUAL) | SOCK_CONNECT;
418 segfault 1.1 }
419     if (connect(sock, (struct sockaddr *) &name,
420     sizeof(struct sockaddr_in)) < 0) {
421     if (errno == EINPROGRESS) {
422 fabian 1.16 /* Firewall? announce connect attempt to proxy */
423 segfault 1.1 if (firewall[0])
424     return proxy_connect(sock, server, sport, proxy);
425     return sock; /* async success! */
426 fabian 1.4 } else
427 segfault 1.1 return -1;
428     }
429 fabian 1.16 /* Synchronous? :/ */
430 segfault 1.1 if (firewall[0])
431     return proxy_connect(sock, server, sport, proxy);
432     return sock;
433     }
434    
435 fabian 1.16 /* Ordinary non-binary connection attempt */
436 segfault 1.1 int open_telnet(char *server, int port)
437     {
438 fabian 1.4 int sock = getsock(0),
439     ret = open_telnet_raw(sock, server, port);
440    
441     if (ret < 0)
442     killsock(sock);
443     return ret;
444 segfault 1.1 }
445    
446 fabian 1.16 /* Returns a socket number for a listening socket that will accept any
447 fabian 1.19 * connection on a certain address -- port # is returned in port
448 fabian 1.16 */
449 fabian 1.19 int open_address_listen(IP addr, int *port)
450 segfault 1.1 {
451 fabian 1.13 int sock;
452     unsigned int addrlen;
453 segfault 1.1 struct sockaddr_in name;
454    
455     if (firewall[0]) {
456     /* FIXME: can't do listen port thru firewall yet */
457 fabian 1.6 putlog(LOG_MISC, "*", "!! Cant open a listen port (you are using a firewall)");
458 segfault 1.1 return -1;
459     }
460 fabian 1.24
461 segfault 1.1 sock = getsock(SOCK_LISTEN);
462 fabian 1.24 if (sock < 1)
463     return -1;
464    
465 fabian 1.18 egg_bzero((char *) &name, sizeof(struct sockaddr_in));
466 segfault 1.1 name.sin_family = AF_INET;
467 fabian 1.19 name.sin_port = htons(*port); /* 0 = just assign us a port */
468     name.sin_addr.s_addr = addr;
469 segfault 1.1 if (bind(sock, (struct sockaddr *) &name, sizeof(name)) < 0) {
470     killsock(sock);
471     return -1;
472     }
473     /* what port are we on? */
474     addrlen = sizeof(name);
475     if (getsockname(sock, (struct sockaddr *) &name, &addrlen) < 0) {
476     killsock(sock);
477     return -1;
478     }
479 fabian 1.19 *port = ntohs(name.sin_port);
480 segfault 1.1 if (listen(sock, 1) < 0) {
481     killsock(sock);
482     return -1;
483     }
484     return sock;
485 fabian 1.19 }
486    
487     /* Returns a socket number for a listening socket that will accept any
488     * connection -- port # is returned in port
489     */
490     inline int open_listen(int *port)
491     {
492     return open_address_listen(myip[0] ? getmyip() : INADDR_ANY, port);
493 segfault 1.1 }
494    
495 fabian 1.3 /* Given a network-style IP address, returns the hostname. The hostname
496     * will be in the "##.##.##.##" format if there was an error.
497     *
498     * NOTE: This function is depreciated. Try using the async dns approach
499 fabian 1.16 * instead.
500     */
501 segfault 1.1 char *hostnamefromip(unsigned long ip)
502     {
503     struct hostent *hp;
504     unsigned long addr = ip;
505     unsigned char *p;
506 fabian 1.3 static char s[UHOSTLEN];
507 segfault 1.1
508     if (!setjmp(alarmret)) {
509     alarm(resolve_timeout);
510     hp = gethostbyaddr((char *) &addr, sizeof(addr), AF_INET);
511     alarm(0);
512     } else {
513     hp = NULL;
514     }
515     if (hp == NULL) {
516     p = (unsigned char *) &addr;
517     sprintf(s, "%u.%u.%u.%u", p[0], p[1], p[2], p[3]);
518     return s;
519     }
520 fabian 1.3 strncpy(s, hp->h_name, UHOSTLEN - 1);
521     s[UHOSTLEN - 1] = 0;
522 segfault 1.1 return s;
523     }
524    
525 fabian 1.5 /* Returns the given network byte order IP address in the
526 fabian 1.16 * dotted format - "##.##.##.##"
527     */
528 fabian 1.3 char *iptostr(IP ip)
529     {
530     struct in_addr a;
531    
532     a.s_addr = ip;
533     return inet_ntoa(a);
534     }
535    
536 fabian 1.16 /* Short routine to answer a connect received on a socket made previously
537 segfault 1.1 * by open_listen ... returns hostname of the caller & the new socket
538 fabian 1.16 * does NOT dispose of old "public" socket!
539     */
540     int answer(int sock, char *caller, unsigned long *ip, unsigned short *port,
541     int binary)
542 segfault 1.1 {
543 fabian 1.13 int new_sock;
544     unsigned int addrlen;
545 segfault 1.1 struct sockaddr_in from;
546 fabian 1.25
547 fabian 1.8 addrlen = sizeof(struct sockaddr);
548 segfault 1.1 new_sock = accept(sock, (struct sockaddr *) &from, &addrlen);
549     if (new_sock < 0)
550     return -1;
551     if (ip != NULL) {
552     *ip = from.sin_addr.s_addr;
553 fabian 1.16 /* This is now done asynchronously. We now only provide the IP address.
554     *
555 fabian 1.3 * strncpy(caller, hostnamefromip(*ip), 120);
556     */
557     strncpy(caller, iptostr(*ip), 120);
558 segfault 1.1 caller[120] = 0;
559 fabian 1.21 *ip = ntohl(*ip);
560 segfault 1.1 }
561     if (port != NULL)
562 fabian 1.21 *port = ntohs(from.sin_port);
563 fabian 1.16 /* Set up all the normal socket crap */
564 segfault 1.1 setsock(new_sock, (binary ? SOCK_BINARY : 0));
565     return new_sock;
566     }
567    
568 fabian 1.16 /* Like open_telnet, but uses server & port specifications of dcc
569     */
570 segfault 1.1 int open_telnet_dcc(int sock, char *server, char *port)
571     {
572     int p;
573     unsigned long addr;
574     char sv[121];
575     unsigned char c[4];
576    
577     if (port != NULL)
578     p = atoi(port);
579     else
580     p = 2000;
581     if (server != NULL)
582     addr = my_atoul(server);
583     else
584     addr = 0L;
585     if (addr < (1 << 24))
586     return -3; /* fake address */
587     c[0] = (addr >> 24) & 0xff;
588     c[1] = (addr >> 16) & 0xff;
589     c[2] = (addr >> 8) & 0xff;
590     c[3] = addr & 0xff;
591     sprintf(sv, "%u.%u.%u.%u", c[0], c[1], c[2], c[3]);
592     /* strcpy(sv,hostnamefromip(addr)); */
593     p = open_telnet_raw(sock, sv, p);
594     return p;
595     }
596    
597 fabian 1.16 /* Attempts to read from all the sockets in socklist
598 segfault 1.1 * fills s with up to 511 bytes if available, and returns the array index
599 fabian 1.16 *
600     * on EOF: returns -1, with socket in len
601     * on socket error: returns -2
602     * if nothing is ready: returns -3
603     */
604 segfault 1.1 static int sockread(char *s, int *len)
605     {
606     fd_set fd;
607     int fds, i, x;
608     struct timeval t;
609     int grab = 511;
610    
611     fds = getdtablesize();
612     #ifdef FD_SETSIZE
613     if (fds > FD_SETSIZE)
614 fabian 1.16 fds = FD_SETSIZE; /* Fixes YET ANOTHER freebsd bug!!! */
615 segfault 1.1 #endif
616     /* timeout: 1 sec */
617     t.tv_sec = 1;
618     t.tv_usec = 0;
619     FD_ZERO(&fd);
620 fabian 1.17
621 segfault 1.1 for (i = 0; i < MAXSOCKS; i++)
622 fabian 1.17 if (!(socklist[i].flags & (SOCK_UNUSED | SOCK_VIRTUAL))) {
623 segfault 1.1 if ((socklist[i].sock == STDOUT) && !backgrd)
624     FD_SET(STDIN, &fd);
625     else
626     FD_SET(socklist[i].sock, &fd);
627     }
628     #ifdef HPUX_HACKS
629     #ifndef HPUX10_HACKS
630     x = select(fds, (int *) &fd, (int *) NULL, (int *) NULL, &t);
631     #else
632     x = select(fds, &fd, NULL, NULL, &t);
633     #endif
634     #else
635     x = select(fds, &fd, NULL, NULL, &t);
636     #endif
637     if (x > 0) {
638 fabian 1.16 /* Something happened */
639 segfault 1.1 for (i = 0; i < MAXSOCKS; i++) {
640     if ((!(socklist[i].flags & SOCK_UNUSED)) &&
641     ((FD_ISSET(socklist[i].sock, &fd)) ||
642     ((socklist[i].sock == STDOUT) && (!backgrd) &&
643     (FD_ISSET(STDIN, &fd))))) {
644     if (socklist[i].flags & (SOCK_LISTEN | SOCK_CONNECT)) {
645 fabian 1.16 /* Listening socket -- don't read, just return activity */
646     /* Same for connection attempt */
647 segfault 1.1 /* (for strong connections, require a read to succeed first) */
648     if (socklist[i].flags & SOCK_PROXYWAIT) { /* drummer */
649 fabian 1.16 /* Hang around to get the return code from proxy */
650 segfault 1.1 grab = 10;
651     } else if (!(socklist[i].flags & SOCK_STRONGCONN)) {
652     debug1("net: connect! sock %d", socklist[i].sock);
653     s[0] = 0;
654     *len = 0;
655     return i;
656     }
657 fabian 1.3 } else if (socklist[i].flags & SOCK_PASS) {
658     s[0] = 0;
659     *len = 0;
660     return i;
661 segfault 1.1 }
662     if ((socklist[i].sock == STDOUT) && !backgrd)
663     x = read(STDIN, s, grab);
664     else
665     x = read(socklist[i].sock, s, grab);
666     if (x <= 0) { /* eof */
667     if (x == EAGAIN) {
668     s[0] = 0;
669     *len = 0;
670     return -3;
671     }
672     *len = socklist[i].sock;
673     socklist[i].flags &= ~SOCK_CONNECT;
674     debug1("net: eof!(read) socket %d", socklist[i].sock);
675     return -1;
676     }
677     s[x] = 0;
678     *len = x;
679     if (socklist[i].flags & SOCK_PROXYWAIT) {
680     debug2("net: socket: %d proxy errno: %d", socklist[i].sock, s[1]);
681     socklist[i].flags &= ~(SOCK_CONNECT | SOCK_PROXYWAIT);
682     switch (s[1]) {
683 fabian 1.16 case 90: /* Success */
684 segfault 1.1 s[0] = 0;
685     *len = 0;
686     return i;
687 fabian 1.16 case 91: /* Failed */
688 segfault 1.1 errno = ECONNREFUSED;
689     break;
690 fabian 1.16 case 92: /* No identd */
691     case 93: /* Identd said wrong username */
692     /* A better error message would be "socks misconfigured"
693     * or "identd not working" but this is simplest.
694     */
695 segfault 1.1 errno = ENETUNREACH;
696     break;
697     }
698     *len = socklist[i].sock;
699     return -1;
700     }
701     return i;
702     }
703     }
704     } else if (x == -1)
705     return -2; /* socket error */
706     else {
707     s[0] = 0;
708     *len = 0;
709     }
710     return -3;
711     }
712    
713     /* sockgets: buffer and read from sockets
714     *
715 fabian 1.16 * Attempts to read from all registered sockets for up to one second. if
716 segfault 1.1 * after one second, no complete data has been received from any of the
717     * sockets, 's' will be empty, 'len' will be 0, and sockgets will return -3.
718     * if there is returnable data received from a socket, the data will be
719     * in 's' (null-terminated if non-binary), the length will be returned
720     * in len, and the socket number will be returned.
721     * normal sockets have their input buffered, and each call to sockgets
722     * will return one line terminated with a '\n'. binary sockets are not
723     * buffered and return whatever coems in as soon as it arrives.
724     * listening sockets will return an empty string when a connection comes in.
725     * connecting sockets will return an empty string on a successful connect,
726     * or EOF on a failed connect.
727     * if an EOF is detected from any of the sockets, that socket number will be
728     * put in len, and -1 will be returned.
729     * the maximum length of the string returned is 512 (including null)
730 fabian 1.25 *
731     * Returns -4 if we handled something that shouldn't be handled by the
732     * dcc functions. Simply ignore it.
733 segfault 1.1 */
734    
735     int sockgets(char *s, int *len)
736     {
737     char xx[514], *p, *px;
738     int ret, i, data = 0;
739    
740 fabian 1.9 Context;
741 segfault 1.1 for (i = 0; i < MAXSOCKS; i++) {
742 fabian 1.16 /* Check for stored-up data waiting to be processed */
743 fabian 1.25 if (!(socklist[i].flags & SOCK_UNUSED) &&
744     !(socklist[i].flags & SOCK_BUFFER) && (socklist[i].inbuf != NULL)) {
745     if (!(socklist[i].flags & SOCK_BINARY)) {
746     /* look for \r too cos windows can't follow RFCs */
747     p = strchr(socklist[i].inbuf, '\n');
748     if (p == NULL)
749     p = strchr(socklist[i].inbuf, '\r');
750     if (p != NULL) {
751     *p = 0;
752     if (strlen(socklist[i].inbuf) > 510)
753     socklist[i].inbuf[510] = 0;
754     strcpy(s, socklist[i].inbuf);
755     px = (char *) nmalloc(strlen(p + 1) + 1);
756     strcpy(px, p + 1);
757     nfree(socklist[i].inbuf);
758     if (px[0])
759     socklist[i].inbuf = px;
760     else {
761     nfree(px);
762     socklist[i].inbuf = NULL;
763     }
764     /* Strip CR if this was CR/LF combo */
765     if (s[strlen(s) - 1] == '\r')
766     s[strlen(s) - 1] = 0;
767     *len = strlen(s);
768     return socklist[i].sock;
769     }
770     } else {
771     /* Handling buffered binary data (must have been SOCK_BUFFER before). */
772     if (socklist[i].inbuflen <= 510) {
773     *len = socklist[i].inbuflen;
774     egg_memcpy(s, socklist[i].inbuf, socklist[i].inbuflen);
775     nfree(socklist[i].inbuf);
776     socklist[i].inbuflen = 0;
777     } else {
778     /* Split up into chunks of 510 bytes. */
779     *len = 510;
780     egg_memcpy(s, socklist[i].inbuf, *len);
781     egg_memcpy(socklist[i].inbuf, socklist[i].inbuf + *len, *len);
782     socklist[i].inbuflen -= *len;
783     socklist[i].inbuf = nrealloc(socklist[i].inbuf,
784     socklist[i].inbuflen);
785 segfault 1.1 }
786     return socklist[i].sock;
787     }
788     }
789 fabian 1.16 /* Also check any sockets that might have EOF'd during write */
790 segfault 1.1 if (!(socklist[i].flags & SOCK_UNUSED)
791     && (socklist[i].flags & SOCK_EOFD)) {
792 fabian 1.9 Context;
793 segfault 1.1 s[0] = 0;
794     *len = socklist[i].sock;
795     return -1;
796     }
797     }
798 fabian 1.16 /* No pent-up data of any worth -- down to business */
799 fabian 1.9 Context;
800 segfault 1.1 *len = 0;
801     ret = sockread(xx, len);
802     if (ret < 0) {
803     s[0] = 0;
804     return ret;
805     }
806 fabian 1.16 /* Binary, listening and passed on sockets don't get buffered. */
807 segfault 1.1 if (socklist[ret].flags & SOCK_CONNECT) {
808     if (socklist[ret].flags & SOCK_STRONGCONN) {
809     socklist[ret].flags &= ~SOCK_STRONGCONN;
810 fabian 1.25 /* Buffer any data that came in, for future read. */
811     socklist[ret].inbuflen = *len;
812     socklist[ret].inbuf = (char *) nmalloc(*len + 1);
813     /* It might be binary data. You never know. */
814     egg_memcpy(socklist[ret].inbuf, xx, *len);
815     socklist[ret].inbuf[*len] = 0;
816 segfault 1.1 }
817     socklist[ret].flags &= ~SOCK_CONNECT;
818     s[0] = 0;
819     return socklist[ret].sock;
820     }
821     if (socklist[ret].flags & SOCK_BINARY) {
822 fabian 1.18 egg_memcpy(s, xx, *len);
823 segfault 1.1 return socklist[ret].sock;
824     }
825 fabian 1.3 if ((socklist[ret].flags & SOCK_LISTEN) ||
826     (socklist[ret].flags & SOCK_PASS))
827 segfault 1.1 return socklist[ret].sock;
828 fabian 1.25 if (socklist[ret].flags & SOCK_BUFFER) {
829     socklist[ret].inbuf = (char *) nrealloc(socklist[ret].inbuf,
830     socklist[ret].inbuflen + *len + 1);
831     egg_memcpy(socklist[ret].inbuf + socklist[ret].inbuflen, xx, *len);
832     socklist[ret].inbuflen += *len;
833     /* We don't know whether it's binary data. Make sure normal strings
834     will be handled properly later on too. */
835     socklist[ret].inbuf[socklist[ret].inbuflen] = 0;
836     return -4; /* Ignore this one. */
837     }
838 fabian 1.9 Context;
839 fabian 1.16 /* Might be necessary to prepend stored-up data! */
840 segfault 1.1 if (socklist[ret].inbuf != NULL) {
841     p = socklist[ret].inbuf;
842     socklist[ret].inbuf = (char *) nmalloc(strlen(p) + strlen(xx) + 1);
843     strcpy(socklist[ret].inbuf, p);
844     strcat(socklist[ret].inbuf, xx);
845     nfree(p);
846     if (strlen(socklist[ret].inbuf) < 512) {
847     strcpy(xx, socklist[ret].inbuf);
848     nfree(socklist[ret].inbuf);
849     socklist[ret].inbuf = NULL;
850 fabian 1.25 socklist[ret].inbuflen = 0;
851 segfault 1.1 } else {
852     p = socklist[ret].inbuf;
853 fabian 1.25 socklist[ret].inbuflen = strlen(p) - 510;
854     socklist[ret].inbuf = (char *) nmalloc(socklist[ret].inbuflen + 1);
855 segfault 1.1 strcpy(socklist[ret].inbuf, p + 510);
856     *(p + 510) = 0;
857     strcpy(xx, p);
858     nfree(p);
859     /* (leave the rest to be post-pended later) */
860     }
861     }
862 fabian 1.9 Context;
863 fabian 1.16 /* Look for EOL marker; if it's there, i have something to show */
864 segfault 1.1 p = strchr(xx, '\n');
865     if (p == NULL)
866     p = strchr(xx, '\r');
867     if (p != NULL) {
868     *p = 0;
869     strcpy(s, xx);
870     strcpy(xx, p + 1);
871     if (s[strlen(s) - 1] == '\r')
872     s[strlen(s) - 1] = 0;
873     data = 1; /* DCC_CHAT may now need to process a
874 fabian 1.16 blank line */
875 segfault 1.1 /* NO! */
876     /* if (!s[0]) strcpy(s," "); */
877     } else {
878     s[0] = 0;
879     if (strlen(xx) >= 510) {
880 fabian 1.16 /* String is too long, so just insert fake \n */
881 segfault 1.1 strcpy(s, xx);
882     xx[0] = 0;
883     data = 1;
884     }
885     }
886 fabian 1.9 Context;
887 segfault 1.1 *len = strlen(s);
888 fabian 1.16 /* Anything left that needs to be saved? */
889 segfault 1.1 if (!xx[0]) {
890     if (data)
891     return socklist[ret].sock;
892     else
893     return -3;
894     }
895 fabian 1.9 Context;
896 fabian 1.16 /* Prepend old data back */
897 segfault 1.1 if (socklist[ret].inbuf != NULL) {
898 fabian 1.9 Context;
899 segfault 1.1 p = socklist[ret].inbuf;
900 fabian 1.25 socklist[ret].inbuflen = strlen(p) + strlen(xx);
901     socklist[ret].inbuf = (char *) nmalloc(socklist[ret].inbuflen + 1);
902 segfault 1.1 strcpy(socklist[ret].inbuf, xx);
903     strcat(socklist[ret].inbuf, p);
904     nfree(p);
905     } else {
906 fabian 1.9 Context;
907 fabian 1.25 socklist[ret].inbuflen = strlen(xx);
908     socklist[ret].inbuf = (char *) nmalloc(socklist[ret].inbuflen + 1);
909 segfault 1.1 strcpy(socklist[ret].inbuf, xx);
910     }
911 fabian 1.9 Context;
912 segfault 1.1 if (data) {
913 fabian 1.9 Context;
914 segfault 1.1 return socklist[ret].sock;
915     } else {
916 fabian 1.9 Context;
917 segfault 1.1 return -3;
918     }
919     }
920    
921 fabian 1.16 /* Dump something to a socket
922     *
923     * NOTE: Do NOT put Contexts in here if you want DEBUG to be meaningful!!
924     */
925 fabian 1.12 void tputs(register int z, char *s, unsigned int len)
926 segfault 1.1 {
927 fabian 1.12 register int i, x, idx;
928 segfault 1.1 char *p;
929 fabian 1.12 static int inhere = 0;
930 segfault 1.1
931     if (z < 0)
932     return; /* um... HELLO?! sanity check please! */
933     if (((z == STDOUT) || (z == STDERR)) && (!backgrd || use_stderr)) {
934     write(z, s, len);
935     return;
936     }
937     for (i = 0; i < MAXSOCKS; i++) {
938     if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == z)) {
939 fabian 1.10 for (idx = 0; idx < dcc_total; idx++) {
940     if (dcc[idx].sock == z) {
941     if (dcc[idx].type) {
942     if (dcc[idx].type->name) {
943     if (!strncmp(dcc[idx].type->name, "BOT", 3)) {
944     otraffic_bn_today += len;
945     break;
946     } else if (!strcmp(dcc[idx].type->name, "SERVER")) {
947     otraffic_irc_today += len;
948     break;
949     } else if (!strncmp(dcc[idx].type->name, "CHAT", 4)) {
950     otraffic_dcc_today += len;
951     break;
952     } else if (!strncmp(dcc[idx].type->name, "FILES", 5)) {
953     otraffic_filesys_today += len;
954     break;
955     } else if (!strcmp(dcc[idx].type->name, "SEND")) {
956     otraffic_trans_today += len;
957     break;
958     } else if (!strncmp(dcc[idx].type->name, "GET", 3)) {
959     otraffic_trans_today += len;
960     break;
961     } else {
962     otraffic_unknown_today += len;
963     break;
964     }
965     }
966     }
967     }
968     }
969    
970 segfault 1.1 if (socklist[i].outbuf != NULL) {
971 fabian 1.16 /* Already queueing: just add it */
972 segfault 1.1 p = (char *) nrealloc(socklist[i].outbuf, socklist[i].outbuflen + len);
973 fabian 1.18 egg_memcpy(p + socklist[i].outbuflen, s, len);
974 segfault 1.1 socklist[i].outbuf = p;
975     socklist[i].outbuflen += len;
976     return;
977     }
978 fabian 1.16 /* Try. */
979 segfault 1.1 x = write(z, s, len);
980     if (x == (-1))
981     x = 0;
982     if (x < len) {
983 fabian 1.16 /* Socket is full, queue it */
984 segfault 1.1 socklist[i].outbuf = (char *) nmalloc(len - x);
985 fabian 1.18 egg_memcpy(socklist[i].outbuf, &s[x], len - x);
986 segfault 1.1 socklist[i].outbuflen = len - x;
987     }
988     return;
989     }
990     }
991 fabian 1.12 /* Make sure we don't cause a crash by looping here */
992     if (!inhere) {
993     inhere = 1;
994 fabian 1.16
995 fabian 1.12 putlog(LOG_MISC, "*", "!!! writing to nonexistent socket: %d", z);
996     s[strlen(s) - 1] = 0;
997     putlog(LOG_MISC, "*", "!-> '%s'", s);
998 fabian 1.16
999 fabian 1.12 inhere = 0;
1000     }
1001 segfault 1.1 }
1002    
1003     /* tputs might queue data for sockets, let's dump as much of it as
1004 fabian 1.16 * possible.
1005     */
1006 segfault 1.1 void dequeue_sockets()
1007     {
1008     int i, x;
1009    
1010     for (i = 0; i < MAXSOCKS; i++) {
1011     if (!(socklist[i].flags & SOCK_UNUSED) &&
1012 fabian 1.22 socklist[i].outbuf != NULL) {
1013 fabian 1.16 /* Trick tputs into doing the work */
1014 segfault 1.1 x = write(socklist[i].sock, socklist[i].outbuf,
1015     socklist[i].outbuflen);
1016     if ((x < 0) && (errno != EAGAIN)
1017     #ifdef EBADSLT
1018     && (errno != EBADSLT)
1019     #endif
1020     #ifdef ENOTCONN
1021     && (errno != ENOTCONN)
1022     #endif
1023     ) {
1024 fabian 1.16 /* This detects an EOF during writing */
1025 segfault 1.1 debug3("net: eof!(write) socket %d (%s,%d)", socklist[i].sock,
1026     strerror(errno), errno);
1027     socklist[i].flags |= SOCK_EOFD;
1028     } else if (x == socklist[i].outbuflen) {
1029 fabian 1.16 /* If the whole buffer was sent, nuke it */
1030 segfault 1.1 nfree(socklist[i].outbuf);
1031     socklist[i].outbuf = NULL;
1032     socklist[i].outbuflen = 0;
1033     } else if (x > 0) {
1034     char *p = socklist[i].outbuf;
1035    
1036 fabian 1.16 /* This removes any sent bytes from the beginning of the buffer */
1037 segfault 1.1 socklist[i].outbuf = (char *) nmalloc(socklist[i].outbuflen - x);
1038 fabian 1.18 egg_memcpy(socklist[i].outbuf, p + x, socklist[i].outbuflen - x);
1039 segfault 1.1 socklist[i].outbuflen -= x;
1040     nfree(p);
1041     }
1042 fabian 1.15
1043     /* All queued data was sent. Call handler if one exists and the
1044     * dcc entry wants it.
1045     */
1046     if (!socklist[i].outbuf) {
1047     int idx = findanyidx(socklist[i].sock);
1048    
1049 fabian 1.22 if (idx > 0 && dcc[idx].type && dcc[idx].type->outdone)
1050 fabian 1.15 dcc[idx].type->outdone(idx);
1051     }
1052 segfault 1.1 }
1053     }
1054     }
1055 fabian 1.16
1056 segfault 1.1
1057 fabian 1.16 /*
1058     * Debugging stuff
1059     */
1060 segfault 1.1
1061     void tell_netdebug(int idx)
1062     {
1063     int i;
1064     char s[80];
1065    
1066     dprintf(idx, "Open sockets:");
1067     for (i = 0; i < MAXSOCKS; i++) {
1068     if (!(socklist[i].flags & SOCK_UNUSED)) {
1069     sprintf(s, " %d", socklist[i].sock);
1070     if (socklist[i].flags & SOCK_BINARY)
1071     strcat(s, " (binary)");
1072     if (socklist[i].flags & SOCK_LISTEN)
1073     strcat(s, " (listen)");
1074 fabian 1.3 if (socklist[i].flags & SOCK_PASS)
1075     strcat(s, " (passed on)");
1076 segfault 1.1 if (socklist[i].flags & SOCK_CONNECT)
1077     strcat(s, " (connecting)");
1078     if (socklist[i].flags & SOCK_STRONGCONN)
1079     strcat(s, " (strong)");
1080     if (socklist[i].flags & SOCK_NONSOCK)
1081     strcat(s, " (file)");
1082     if (socklist[i].inbuf != NULL)
1083     sprintf(&s[strlen(s)], " (inbuf: %04X)", strlen(socklist[i].inbuf));
1084     if (socklist[i].outbuf != NULL)
1085     sprintf(&s[strlen(s)], " (outbuf: %06lX)", socklist[i].outbuflen);
1086     strcat(s, ",");
1087     dprintf(idx, "%s", s);
1088     }
1089     }
1090     dprintf(idx, " done.\n");
1091     }
1092    
1093     /* Security-flavoured sanity checking on DCC connections of all sorts can be
1094     * done with this routine. Feed it the proper information from your DCC
1095     * before you attempt the connection, and this will make an attempt at
1096     * figuring out if the connection is really that person, or someone screwing
1097     * around. It's not foolproof, but anything that fails this check probably
1098     * isn't going to work anyway due to masquerading firewalls, NAT routers,
1099 fabian 1.16 * or bugs in mIRC.
1100     */
1101 segfault 1.1 int sanitycheck_dcc(char *nick, char *from, char *ipaddy, char *port)
1102     {
1103     /* According to the latest RFC, the clients SHOULD be able to handle
1104 fabian 1.16 * DNS names that are up to 255 characters long. This is not broken.
1105     */
1106 fabian 1.3 char badaddress[16];
1107 segfault 1.1 IP ip = my_atoul(ipaddy);
1108     int prt = atoi(port);
1109    
1110     /* It is disabled HERE so we only have to check in *one* spot! */
1111     if (!dcc_sanitycheck)
1112     return 1;
1113 fabian 1.9 Context; /* This should be pretty solid, but
1114 fabian 1.16 something _might_ break. */
1115 segfault 1.1 if (prt < 1) {
1116     putlog(LOG_MISC, "*", "ALERT: (%s!%s) specified an impossible port of %u!",
1117     nick, from, prt);
1118     return 0;
1119     }
1120 fabian 1.3 sprintf(badaddress, "%u.%u.%u.%u", (ip >> 24) & 0xff, (ip >> 16) & 0xff,
1121     (ip >> 8) & 0xff, ip & 0xff);
1122 segfault 1.1 if (ip < (1 << 24)) {
1123     putlog(LOG_MISC, "*", "ALERT: (%s!%s) specified an impossible IP of %s!",
1124     nick, from, badaddress);
1125     return 0;
1126     }
1127 fabian 1.3 return 1;
1128     }
1129    
1130     int hostsanitycheck_dcc(char *nick, char *from, IP ip, char *dnsname,
1131     char *prt)
1132     {
1133     /* According to the latest RFC, the clients SHOULD be able to handle
1134 fabian 1.16 * DNS names that are up to 255 characters long. This is not broken.
1135     */
1136 fabian 1.23 char hostn[256], badaddress[16];
1137 fabian 1.3
1138     /* It is disabled HERE so we only have to check in *one* spot! */
1139     if (!dcc_sanitycheck)
1140     return 1;
1141 fabian 1.9 Context; /* This should be pretty solid, but
1142 fabian 1.16 something _might_ break. */
1143 fabian 1.3 sprintf(badaddress, "%u.%u.%u.%u", (ip >> 24) & 0xff, (ip >> 16) & 0xff,
1144     (ip >> 8) & 0xff, ip & 0xff);
1145 segfault 1.1 /* These should pad like crazy with zeros, since 120 bytes or so is
1146     * where the routines providing our data currently lose interest. I'm
1147 fabian 1.16 * using the n-variant in case someone changes that...
1148     */
1149 fabian 1.23 strncpy(hostn, extracthostname(from), 255);
1150     hostn[255] = 0;
1151     if (!egg_strcasecmp(hostn, dnsname)) {
1152 segfault 1.1 putlog(LOG_DEBUG, "*", "DNS information for submitted IP checks out.");
1153     return 1;
1154     }
1155     if (!strcmp(badaddress, dnsname))
1156 fabian 1.23 putlog(LOG_MISC, "*", "ALERT: (%s!%s) sent a DCC request with bogus IP "
1157     "information of %s port %u!", nick, from, badaddress, prt);
1158 segfault 1.1 else
1159     return 1; /* <- usually happens when we have
1160 fabian 1.16 a user with an unresolved hostmask! */
1161 segfault 1.1 return 0;
1162 fabian 1.15 }
1163    
1164     /* Checks wether the referenced socket has data queued.
1165     *
1166     * Returns true if the incoming/outgoing (depending on 'type') queues
1167     * contain data, otherwise false.
1168     */
1169     int sock_has_data(int type, int sock)
1170     {
1171     int ret = 0, i;
1172    
1173     for (i = 0; i < MAXSOCKS; i++)
1174 fabian 1.22 if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].sock == sock)
1175 fabian 1.15 break;
1176     if (i < MAXSOCKS) {
1177     switch (type) {
1178     case SOCK_DATA_OUTGOING:
1179     ret = (socklist[i].outbuf != NULL);
1180     break;
1181     case SOCK_DATA_INCOMING:
1182     ret = (socklist[i].inbuf != NULL);
1183     break;
1184     }
1185 fabian 1.22 } else
1186     debug1("sock_has_data: could not find socket #%d, returning false.", sock);
1187 fabian 1.15 return ret;
1188 segfault 1.1 }

webmaster@eggheads.org
ViewVC Help
Powered by ViewVC 1.1.23