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

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

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


Revision 1.19 - (show annotations) (download) (as text)
Fri Aug 18 00:25:09 2000 UTC (18 years, 11 months ago) by fabian
Branch: MAIN
CVS Tags: eggdrop105040
Changes since 1.18: +14 -6 lines
File MIME type: text/x-chdr
open_address_listen patch

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

webmaster@eggheads.org
ViewVC Help
Powered by ViewVC 1.1.23