/[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.57 - (show annotations) (download) (as text)
Fri May 3 07:57:12 2002 UTC (18 years, 5 months ago) by stdarg
Branch: MAIN
Changes since 1.56: +13 -11 lines
File MIME type: text/x-chdr
* Made itraffic_* otraffic_* into a single struct.
* mod-path -> mod_path

1 /*
2 * net.c -- handles:
3 * all raw network i/o
4 *
5 * $Id: net.c,v 1.56 2002/04/01 23:39:36 ite 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>
23 #include <errno.h>
24 #if HAVE_UNISTD_H
25 # include <unistd.h>
26 #endif
27 #include <setjmp.h>
28
29 #include "lib/adns/adns.h"
30 #include "egg_timer.h"
31 #include "logfile.h"
32 #include "dns.h"
33
34 #include "traffic.h" /* egg_traffic_t */
35
36 #if !HAVE_GETDTABLESIZE
37 # ifdef FD_SETSIZE
38 # define getdtablesize() FD_SETSIZE
39 # else
40 # define getdtablesize() 200
41 # endif
42 #endif
43
44 #ifndef IN6_IS_ADDR_V4MAPPED
45 # define IN6_IS_ADDR_V4MAPPED(a) \
46 ((((uint32_t *) (a))[0] == 0) && (((uint32_t *) (a))[1] == 0) && \
47 (((uint32_t *) (a))[2] == htonl (0xffff)))
48 #endif
49
50 extern struct dcc_t *dcc;
51 extern int backgrd, use_stderr, resolve_timeout, dcc_total;
52
53 extern egg_traffic_t traffic;
54
55 extern char natip[];
56
57 char myip[121] = ""; /* IP can be specified in the config file */
58 char myip6[121] = ""; /* IP6 can be specified in the config file */
59 char firewall[121] = ""; /* Socks server for firewall */
60 int firewallport = 1080; /* Default port of Sock4/5 firewalls */
61 char botuser[21] = "eggdrop"; /* Username of the user running the bot */
62 sock_list *socklist = NULL; /* Enough to be safe */
63 int MAXSOCKS = 0;
64 jmp_buf alarmret; /* Env buffer for alarm() returns */
65 IP localipv4addr = 0; /* Cache the local IPv4 address */
66
67 adns_state ads;
68
69 /* Types of proxy */
70 #define PROXY_SOCKS 1
71 #define PROXY_SUN 2
72
73 /* Initialize the socklist
74 */
75 void init_net()
76 {
77 int i;
78 char s[256];
79 struct hostent *hp;
80
81 socklist = (sock_list *)calloc(MAXSOCKS, sizeof(sock_list));
82 for (i = 0; i < MAXSOCKS; i++) {
83 socklist[i].flags = SOCK_UNUSED;
84 }
85
86 gethostname(s, sizeof s);
87 if ((hp = gethostbyname(s)) == NULL)
88 fatal("Hostname self-lookup failed.", 0);
89 localipv4addr = *((IP*) (hp->h_addr_list[0]));
90
91 /* init ADNS */
92 i = adns_init(&ads, adns_if_noautosys, 0);
93 if (i)
94 fatal(adns_strerror(i), 0);
95 }
96
97 /* Get my ip number
98 */
99 IP getmyip()
100 {
101 /* Could be pre-defined */
102 if (myip[0] && (myip[strlen(myip) - 1] >= '0') &&
103 (myip[strlen(myip) - 1] <= '9')) {
104 return (IP) inet_addr(myip);
105 }
106 return localipv4addr;
107 }
108
109 #ifdef IPV6
110 struct in6_addr getmyip6()
111 {
112 struct in6_addr ip;
113
114 /* Could be pre-defined */
115 if (myip6[0])
116 inet_pton(AF_INET6, myip6, &ip);
117 else {
118 /* get system's default IPv6 ip -- FIXME!? */
119 /* is there a know way?! - drummer */
120 ip = in6addr_any;
121 }
122 return ip;
123 }
124
125 struct in6_addr ipv4to6(IP a)
126 {
127 struct in6_addr ip;
128 if (a == INADDR_ANY)
129 return in6addr_any;
130 else {
131 ((uint32_t *)&ip)[0] = 0;
132 ((uint32_t *)&ip)[1] = 0;
133 ((uint16_t *)&ip)[4] = 0;
134 ((uint16_t *)&ip)[5] = 0xffff;
135 ((uint32_t *)&ip)[3] = a;
136 }
137 return ip;
138 }
139
140 #endif
141
142 void neterror(char *s)
143 {
144 int e = errno;
145 char *err = strerror(e);
146
147 /* Calling procs usually use char s[UHOSTLEN] for the error message. */
148 if (err) strlcpy(s, err, UHOSTLEN);
149 else sprintf(s, "Unforeseen error %d", e);
150 }
151
152 /* Sets/Unsets options for a specific socket.
153 *
154 * Returns: 0 - on success
155 * -1 - socket not found
156 * -2 - illegal operation
157 */
158 int sockoptions(int sock, int operation, int sock_options)
159 {
160 int i;
161
162 for (i = 0; i < MAXSOCKS; i++)
163 if ((socklist[i].sock == sock) && !(socklist[i].flags & SOCK_UNUSED)) {
164 if (operation == EGG_OPTION_SET)
165 socklist[i].flags |= sock_options;
166 else if (operation == EGG_OPTION_UNSET)
167 socklist[i].flags &= ~sock_options;
168 else
169 return -2;
170 return 0;
171 }
172 return -1;
173 }
174
175 /* Return a free entry in the socket entry
176 */
177 int allocsock(int sock, int options)
178 {
179 int i;
180
181 for (i = 0; i < MAXSOCKS; i++) {
182 if (socklist[i].flags & SOCK_UNUSED) break;
183 }
184
185 if (i == MAXSOCKS) {
186 int j;
187
188 /* Expand table by 5 */
189 socklist = (sock_list *)realloc(socklist, (MAXSOCKS+5) * sizeof(sock_list));
190 memset(socklist+MAXSOCKS, 0, 5 * sizeof(sock_list));
191 for (j = 0; j < 5; j++) socklist[MAXSOCKS+j].flags = SOCK_UNUSED;
192 MAXSOCKS += 5;
193 }
194
195 memset(socklist+i, 0, sizeof(sock_list));
196 socklist[i].flags = options;
197 socklist[i].sock = sock;
198 return(i);
199 }
200
201 /* Request a normal socket for i/o
202 */
203 void setsock(int sock, int options)
204 {
205 int i = allocsock(sock, options);
206 int parm;
207
208 if (((sock != STDOUT) || backgrd) &&
209 !(socklist[i].flags & SOCK_NONSOCK)) {
210 parm = 1;
211 setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, (void *) &parm, sizeof(int));
212
213 parm = 0;
214 setsockopt(sock, SOL_SOCKET, SO_LINGER, (void *) &parm, sizeof(int));
215 }
216 if (options & SOCK_LISTEN) {
217 /* Tris says this lets us grab the same port again next time */
218 parm = 1;
219 setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void *) &parm, sizeof(int));
220 }
221 /* Yay async i/o ! */
222 fcntl(sock, F_SETFL, O_NONBLOCK);
223 }
224
225 int getsock(int options)
226 {
227 #ifdef IPV6
228 int sock = socket(AF_INET6, SOCK_STREAM, 0);
229 #else
230 int sock = socket(AF_INET, SOCK_STREAM, 0);
231 #endif
232
233 if (sock >= 0)
234 setsock(sock, options);
235 else
236 putlog(LOG_MISC, "*", "Warning: Can't create new socket!");
237 return sock;
238 }
239
240 /* Done with a socket
241 */
242 void killsock(register int sock)
243 {
244 register int i;
245
246 /* Ignore invalid sockets. */
247 if (sock < 0)
248 return;
249 for (i = 0; i < MAXSOCKS; i++) {
250 if ((socklist[i].sock == sock) && !(socklist[i].flags & SOCK_UNUSED)) {
251 close(socklist[i].sock);
252 if (socklist[i].inbuf != NULL)
253 free_null(socklist[i].inbuf);
254 if (socklist[i].outbuf != NULL) {
255 free_null(socklist[i].outbuf);
256 socklist[i].outbuflen = 0;
257 }
258 socklist[i].flags = SOCK_UNUSED;
259 return;
260 }
261 }
262 putlog(LOG_MISC, "*", "Attempt to kill un-allocated socket %d !!", sock);
263 }
264
265 /* Send connection request to proxy
266 */
267 static int proxy_connect(int sock, char *host, int port, int proxy)
268 {
269 unsigned char x[10];
270 struct hostent *hp;
271 char s[256];
272 int i;
273
274 /* socks proxy */
275 if (proxy == PROXY_SOCKS) {
276 /* numeric IP? */
277 if (host[strlen(host) - 1] >= '0' && host[strlen(host) - 1] <= '9') {
278 IP ip = ((IP) inet_addr(host)); /* drummer */
279 memcpy(x, &ip, 4); /* Beige@Efnet */
280 } else {
281 /* no, must be host.domain */
282 if (!setjmp(alarmret)) {
283 alarm(resolve_timeout);
284 hp = gethostbyname(host);
285 alarm(0);
286 } else {
287 hp = NULL;
288 }
289 if (hp == NULL) {
290 killsock(sock);
291 return -2;
292 }
293 memcpy(x, hp->h_addr, hp->h_length);
294 }
295 for (i = 0; i < MAXSOCKS; i++)
296 if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].sock == sock)
297 socklist[i].flags |= SOCK_PROXYWAIT; /* drummer */
298 snprintf(s, sizeof s, "\004\001%c%c%c%c%c%c%s", (port >> 8) % 256,
299 (port % 256), x[0], x[1], x[2], x[3], botuser);
300 tputs(sock, s, strlen(botuser) + 9); /* drummer */
301 } else if (proxy == PROXY_SUN) {
302 snprintf(s, sizeof s, "%s %d\n", host, port);
303 tputs(sock, s, strlen(s)); /* drummer */
304 }
305 return sock;
306 }
307
308 /* Starts a connection attempt to a socket
309 *
310 * If given a normal hostname, this will be resolved to the corresponding
311 * IP address first. PLEASE try to use the non-blocking dns functions
312 * instead and then call this function with the IP address to avoid blocking.
313 *
314 * returns <0 if connection refused:
315 * -1 neterror() type error
316 * -2 can't resolve hostname
317 */
318 int open_telnet_raw(int sock, char *server, int sport)
319 {
320 #ifdef IPV6
321 struct sockaddr_in6 name, vname;
322 #else
323 struct sockaddr_in name, vname;
324 #endif
325 struct hostent *hp;
326 char host[121];
327 int i, port;
328 volatile int proxy;
329
330 debug2("|NET| open_telnet_raw: %s %d", server, sport);
331
332 /* firewall? use socks */
333 if (firewall[0]) {
334 if (firewall[0] == '!') {
335 proxy = PROXY_SUN;
336 strcpy(host, &firewall[1]);
337 } else {
338 proxy = PROXY_SOCKS;
339 strcpy(host, firewall);
340 }
341 port = firewallport;
342 } else {
343 proxy = 0;
344 strcpy(host, server);
345 port = sport;
346 }
347
348 memset((char *) &name, 0, sizeof name);
349 #ifdef IPV6
350 name.sin6_family = AF_INET6;
351 name.sin6_port = htons(port);
352 #else
353 name.sin_family = AF_INET;
354 name.sin_port = htons(port);
355 #endif
356
357 if (!setjmp(alarmret)) {
358 char *p;
359 int type;
360
361 alarm(resolve_timeout);
362
363 if (!strncasecmp("ipv6%", host, 5)) {
364 type = AF_INET6;
365 p = host + 5;
366 debug1("|NET| checking only AAAA record for %s", p);
367 } else if (!strncasecmp("ipv4%", host, 5)) {
368 type = AF_INET;
369 p = host + 5;
370 debug1("|NET| checking only A record for %s", p);
371 } else {
372 type = AF_INET; /* af_preferred */
373 p = host;
374 }
375 alarm(resolve_timeout);
376 #ifndef IPV6
377 hp = gethostbyname(p);
378 #else
379 hp = gethostbyname2(p, type);
380 if (!hp && (p == host))
381 hp = gethostbyname2(p, (type == AF_INET6 ? AF_INET : AF_INET6));
382 #endif
383 alarm(0);
384 } else {
385 hp = NULL;
386 }
387 if (hp == NULL)
388 return -2;
389
390 #ifndef IPV6
391 memcpy(&name.sin_addr, hp->h_addr, hp->h_length);
392 name.sin_family = hp->h_addrtype;
393 #else
394 if (hp->h_addrtype == AF_INET6)
395 memcpy(&name.sin6_addr, hp->h_addr, hp->h_length);
396 else if (hp->h_addrtype == AF_INET)
397 name.sin6_addr = ipv4to6(*((unsigned int*) hp->h_addr));
398 else
399 return -123; /* panic :P */
400 name.sin6_family = AF_INET6;
401 #endif
402
403 memset((char *) &vname, 0, sizeof vname);
404 #ifndef IPV6
405 vname.sin_family = AF_INET;
406 vname.sin_addr.s_addr = (myip[0] ? getmyip() : INADDR_ANY);
407 #else
408 vname.sin6_family = AF_INET6;
409 if (IN6_IS_ADDR_V4MAPPED(&name.sin6_addr))
410 vname.sin6_addr = ipv4to6(myip[0] ? getmyip() : INADDR_ANY);
411 else
412 vname.sin6_addr = (myip6[0] ? getmyip6() : in6addr_any);
413 #endif
414 if (bind(sock, (struct sockaddr *) &vname, sizeof vname) < 0)
415 return -1;
416
417 for (i = 0; i < MAXSOCKS; i++) {
418 if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == sock))
419 socklist[i].flags = (socklist[i].flags & ~SOCK_VIRTUAL) | SOCK_CONNECT;
420 }
421 if (connect(sock, (struct sockaddr *) &name, (sizeof name)) < 0) {
422 if (errno == EINPROGRESS) {
423 /* Firewall? announce connect attempt to proxy */
424 if (firewall[0])
425 return proxy_connect(sock, server, sport, proxy);
426 return sock; /* async success! */
427 } else
428 return -1;
429 }
430 /* Synchronous? :/ */
431 if (firewall[0])
432 return proxy_connect(sock, server, sport, proxy);
433 return sock;
434 }
435
436 /* Ordinary non-binary connection attempt */
437 int open_telnet(char *server, int port)
438 {
439 int sock = getsock(0),
440 ret = open_telnet_raw(sock, server, port);
441
442 if (ret < 0)
443 killsock(sock);
444 return ret;
445 }
446
447 /* Returns a socket number for a listening socket that will accept any
448 * connection on a certain address -- port # is returned in port
449 * "" means on any address
450 */
451 int open_address_listen(char *addr, int *port)
452 {
453 int sock;
454 unsigned int addrlen;
455 #ifdef IPV6
456 struct sockaddr_in6 name;
457 #else
458 struct sockaddr_in name;
459 #endif
460
461 if (firewall[0]) {
462 /* FIXME: can't do listen port thru firewall yet */
463 putlog(LOG_MISC, "*", "!! Cant open a listen port (you are using a firewall)");
464 return -1;
465 }
466
467 sock = getsock(SOCK_LISTEN);
468 if (sock < 1)
469 return -1;
470
471 debug2("|NET| open_address_listen(\"%s\", %d)", addr, *port);
472
473 memset((char *) &name, 0, sizeof name);
474 #ifdef IPV6
475 name.sin6_family = AF_INET6;
476 name.sin6_port = htons(*port); /* 0 = just assign us a port */
477 if (!addr[0])
478 name.sin6_addr = in6addr_any;
479 else if (!inet_pton(AF_INET6, addr, &name.sin6_addr)) {
480 struct in_addr a4;
481 if (inet_aton(addr, &a4))
482 name.sin6_addr = ipv4to6(a4.s_addr);
483 else
484 name.sin6_addr = in6addr_any;
485 }
486 #else
487 name.sin_family = AF_INET;
488 name.sin_port = htons(*port); /* 0 = just assign us a port */
489 if (addr[0])
490 inet_aton(addr, &name.sin_addr);
491 else
492 name.sin_addr.s_addr = INADDR_ANY;
493 #endif
494 if (bind(sock, (struct sockaddr *) &name, sizeof name) < 0) {
495 killsock(sock);
496 if (errno == EADDRNOTAVAIL || errno == EAFNOSUPPORT)
497 return -2;
498 else
499 return -1;
500 }
501 /* what port are we on? */
502 addrlen = sizeof name;
503 if (getsockname(sock, (struct sockaddr *) &name, &addrlen) < 0) {
504 killsock(sock);
505 return -1;
506 }
507 #ifdef IPV6
508 *port = ntohs(name.sin6_port);
509 #else
510 *port = ntohs(name.sin_port);
511 #endif
512 if (listen(sock, 1) < 0) {
513 killsock(sock);
514 return -1;
515 }
516 return sock;
517 }
518
519 /* Returns a socket number for a listening socket that will accept any
520 * connection -- port # is returned in port
521 */
522 inline int open_listen(int *port, int af)
523 {
524 if (af == AF_INET)
525 return open_address_listen(myip, port);
526 #ifdef IPV6
527 else if (af == AF_INET6)
528 return open_address_listen(myip6, port);
529 #endif
530 else
531 return -1;
532 }
533
534 /* Returns the given network byte order IP address in the
535 * dotted format - "##.##.##.##"
536 * (it's IPv4 only, and it's okey - drummer)
537 */
538 char *iptostr(IP ip)
539 {
540 struct in_addr a;
541
542 a.s_addr = ip;
543 return inet_ntoa(a);
544 }
545
546 char *getlocaladdr(int sock)
547 {
548 static char buf[ADDRLEN];
549 #ifdef IPV6
550 struct sockaddr_in6 sa;
551 #else
552 struct sockaddr_in sa;
553 #endif
554 int len = sizeof sa;
555
556 if (sock == -1) { /* assuming IPv4... */
557 sprintf(buf, "%lu", (unsigned long int)
558 ntohl(natip[0] ? inet_addr(natip) : getmyip()));
559 return buf;
560 }
561
562 if (getsockname(sock, (struct sockaddr*) &sa, &len) == -1) {
563 debug1("|NET| getsockname() failed for sock %d", sock);
564 return 0;
565 }
566 #ifdef IPV6
567 if (IN6_IS_ADDR_V4MAPPED(&sa.sin6_addr))
568 sprintf(buf, "%lu",
569 (unsigned long int) ntohl(((uint32_t *)&sa.sin6_addr)[3]));
570 else
571 inet_ntop(AF_INET6, &(sa.sin6_addr), buf, sizeof buf);
572 #else
573 sprintf(buf, "%lu", (unsigned long int) ntohl(sa.sin_addr.s_addr));
574 #endif
575 return buf;
576 }
577
578 /* Short routine to answer a connect received on a socket made previously
579 * by open_listen ... returns hostname of the caller & the new socket
580 * does NOT dispose of old "public" socket!
581 */
582 int answer(int sock, char *caller, char *ip, unsigned short *port,
583 int binary)
584 {
585 int new_sock;
586 unsigned int addrlen;
587 #ifdef IPV6
588 struct sockaddr_in6 from;
589 #else
590 struct sockaddr_in from;
591 #endif
592
593 addrlen = sizeof from;
594 new_sock = accept(sock, (struct sockaddr *) &from, &addrlen);
595 if (new_sock < 0)
596 return -1;
597 if (ip != NULL) {
598 #ifdef IPV6
599 if (IN6_IS_ADDR_V4MAPPED(&from.sin6_addr))
600 inet_ntop(AF_INET, &(((uint32_t *)&from.sin6_addr)[3]), ip, ADDRMAX);
601 else
602 inet_ntop(AF_INET6, &(from.sin6_addr), ip, ADDRMAX);
603 #else
604 inet_ntop(AF_INET, &(from.sin_addr.s_addr), ip, ADDRMAX);
605 #endif
606 /* This is now done asynchronously. We now only provide the IP address.
607 */
608 strlcpy(caller, ip, 121);
609 }
610 if (port != NULL)
611 #ifdef IPV6
612 *port = ntohs(from.sin6_port);
613 #else
614 *port = ntohs(from.sin_port);
615 #endif
616 /* Set up all the normal socket crap */
617 setsock(new_sock, (binary ? SOCK_BINARY : 0));
618 return new_sock;
619 }
620
621 /* Like open_telnet, but uses server & port specifications of dcc
622 */
623 int open_telnet_dcc(int sock, char *server, char *port)
624 {
625 int p;
626 struct in_addr ia;
627
628 debug2("|NET| open_telnet_dcc: %s %s", server, port);
629 if (port != NULL)
630 p = atoi(port);
631 else
632 return -3;
633 if (server == NULL)
634 return -3;
635 /* fix the IPv4 IP format (ie: 167772161 -> 10.0.0.1) */
636 if (inet_aton(server, &ia))
637 return open_telnet_raw(sock, inet_ntoa(ia), p);
638 else
639 return open_telnet_raw(sock, server, p);
640 }
641
642 void egg_dns_gotanswer(int status, adns_answer *aw, char *origname)
643 {
644 char name[UHOSTLEN];
645 #ifdef IPV6
646 char *orign2;
647 #endif
648
649 if (!aw) {
650 debug0("|DNS| egg_dns_gotanswer: ANSWER IS NULL!");
651 return;
652 }
653 if (!origname) {
654 debug0("|DNS| egg_dns_gotanswer: origname is NULL!");
655 return;
656 }
657
658 debug2("|DNS| egg_dns_gotanswer: status=%d adns_answer=%x", status, (int)aw);
659 status = 0;
660 if ((aw->type == adns_r_addr)
661 #ifdef IPV6
662 || (aw->type == adns_r_addr6)
663 #endif
664 ) {
665 if ((aw->status == adns_s_ok) && (aw->nrrs > 0)) {
666 adns_rr_addr *rrp = aw->rrs.untyped;
667 if (rrp->addr.sa.sa_family == AF_INET) {
668 inet_ntop(AF_INET, &(rrp->addr.inet.sin_addr), name, UHOSTLEN-1);
669 status = 1;
670 #ifdef IPV6
671 } else if (rrp->addr.sa.sa_family == AF_INET6) {
672 inet_ntop(AF_INET6, &(rrp->addr.inet6.sin6_addr), name, UHOSTLEN-1);
673 status = 1;
674 #endif
675 }
676 #ifdef IPV6
677 } else if ((aw->type == adns_r_addr /* af_preferred */) &&
678 strncasecmp(origname, "ipv6%", 5) &&
679 strncasecmp(origname, "ipv4%", 5)) {
680 adns_query q6;
681 orign2 = strdup(origname);
682 /* ...it may be AAAA */
683 debug1("|DNS| egg_dns_gotanswer: A failed, checking for AAAA (%s)", origname);
684 adns_submit(ads, origname, adns_r_addr6, 0, orign2, &q6);
685 status = -1;
686 #endif
687 }
688 if (status == 0)
689 strcpy(name, "0.0.0.0");
690 if (status >= 0) {
691 debug3("|DNS| egg_dns_gotanswer: (ipbyhost) host: %s ip: %s status: %d", origname, name, status);
692 call_ipbyhost(origname, name, status);
693 }
694 } else if ((aw->type == adns_r_ptr_ip6) || (aw->type == adns_r_ptr)) {
695 if ((aw->status == adns_s_ok) && (aw->nrrs > 0)) {
696 if (aw->rrs.str) {
697 strlcpy(name, *(aw->rrs.str), UHOSTLEN);
698 status = 1;
699 }
700 }
701 if (!status) {
702 if (origname)
703 strlcpy(name, origname, UHOSTLEN);
704 else
705 strcpy(name, "error");
706 }
707 debug3("|DNS| egg_dns_gotanswer: (hostbyip) ip: %s host: %s status: %d", origname, name, status);
708 call_hostbyip(origname, name, status);
709 } else
710 debug0("|DNS| egg_dns_gotanswer: got unknow type of answer ?!");
711 free(origname);
712 }
713
714 void egg_dns_checkall()
715 {
716 adns_query q, r;
717 adns_answer *answer;
718 char *origname;
719
720 adns_forallqueries_begin(ads);
721 while ((q = adns_forallqueries_next(ads, (void **)&r)) != NULL) {
722 switch (adns_check(ads, &q, &answer, (void **)&origname)) {
723 case 0: /* ok */
724 egg_dns_gotanswer(1, answer, origname);
725 break;
726 case EAGAIN: /* Go into the queue again */
727 break;
728 default: /* failed */
729 egg_dns_gotanswer(0, answer, origname);
730 break;
731 }
732 }
733 }
734
735 /* Attempts to read from all the sockets in socklist
736 * fills s with up to 511 bytes if available, and returns the array index
737 *
738 * on EOF: returns -1, with socket in len
739 * on socket error: returns -2
740 * if nothing is ready: returns -3
741 */
742 static int sockread(char *s, int *len)
743 {
744 fd_set fd, fdw, fde;
745 int fds, i, x, fdtmp;
746 struct timeval t, tnow;
747 struct timeval *pt = &t;
748 int grab = 511;
749 egg_timeval_t howlong;
750
751 fds = getdtablesize();
752 #ifdef FD_SETSIZE
753 if (fds > FD_SETSIZE)
754 fds = FD_SETSIZE; /* Fixes YET ANOTHER freebsd bug!!! */
755 #endif
756
757 if (timer_get_shortest(&howlong)) {
758 /* No timer, default to 1 second. */
759 t.tv_sec = 1;
760 t.tv_usec = 0;
761 }
762 else {
763 t.tv_sec = howlong.sec;
764 t.tv_usec = howlong.usec;
765 }
766
767 FD_ZERO(&fd);
768
769 for (i = 0; i < MAXSOCKS; i++)
770 if (!(socklist[i].flags & (SOCK_UNUSED | SOCK_VIRTUAL))) {
771 if ((socklist[i].sock == STDOUT) && !backgrd)
772 fdtmp = STDIN;
773 else
774 fdtmp = socklist[i].sock;
775 /*
776 * Looks like that having more than a call, in the same
777 * program, to the FD_SET macro, triggers a bug in gcc.
778 * SIGBUS crashing binaries used to be produced on a number
779 * (prolly all?) of 64 bits architectures.
780 * Make your best to avoid to make it happen again.
781 *
782 * ITE
783 */
784 FD_SET(fdtmp , &fd);
785 }
786 tnow.tv_sec = time(NULL);
787 tnow.tv_usec = 0;
788 adns_beforeselect(ads, &fds, &fd, &fdw, &fde, &pt, 0, &tnow);
789 #ifdef HPUX_HACKS
790 #ifndef HPUX10_HACKS
791 x = select(fds, (int *) &fd, (int *) NULL, (int *) NULL, &t);
792 #else
793 x = select(fds, &fd, NULL, NULL, &t);
794 #endif
795 #else
796 x = select(fds, &fd, NULL, NULL, &t);
797 #endif
798 tnow.tv_sec = time(NULL);
799 tnow.tv_usec = 0;
800 adns_afterselect(ads, fds, &fd, &fdw, &fde, &tnow);
801
802 /* dns stuff */
803 egg_dns_checkall();
804
805 if (x > 0) {
806 /* Something happened */
807 for (i = 0; i < MAXSOCKS; i++) {
808 if ((!(socklist[i].flags & SOCK_UNUSED)) &&
809 ((FD_ISSET(socklist[i].sock, &fd)) ||
810 ((socklist[i].sock == STDOUT) && (!backgrd) &&
811 (FD_ISSET(STDIN, &fd))))) {
812 if (socklist[i].flags & (SOCK_LISTEN | SOCK_CONNECT)) {
813 /* Listening socket -- don't read, just return activity */
814 /* Same for connection attempt */
815 /* (for strong connections, require a read to succeed first) */
816 if (socklist[i].flags & SOCK_PROXYWAIT) { /* drummer */
817 /* Hang around to get the return code from proxy */
818 grab = 10;
819 } else if (!(socklist[i].flags & SOCK_STRONGCONN)) {
820 debug1("net: connect! sock %d", socklist[i].sock);
821 s[0] = 0;
822 *len = 0;
823 return i;
824 }
825 } else if (socklist[i].flags & SOCK_PASS) {
826 s[0] = 0;
827 *len = 0;
828 return i;
829 }
830 if ((socklist[i].sock == STDOUT) && !backgrd)
831 x = read(STDIN, s, grab);
832 else
833 x = read(socklist[i].sock, s, grab);
834 if (x <= 0) { /* eof */
835 if (errno == EAGAIN) {
836 s[0] = 0;
837 *len = 0;
838 return -3;
839 }
840 *len = socklist[i].sock;
841 socklist[i].flags &= ~SOCK_CONNECT;
842 debug1("net: eof!(read) socket %d", socklist[i].sock);
843 return -1;
844 }
845 s[x] = 0;
846 *len = x;
847 if (socklist[i].flags & SOCK_PROXYWAIT) {
848 debug2("net: socket: %d proxy errno: %d", socklist[i].sock, s[1]);
849 socklist[i].flags &= ~(SOCK_CONNECT | SOCK_PROXYWAIT);
850 switch (s[1]) {
851 case 90: /* Success */
852 s[0] = 0;
853 *len = 0;
854 return i;
855 case 91: /* Failed */
856 errno = ECONNREFUSED;
857 break;
858 case 92: /* No identd */
859 case 93: /* Identd said wrong username */
860 /* A better error message would be "socks misconfigured"
861 * or "identd not working" but this is simplest.
862 */
863 errno = ENETUNREACH;
864 break;
865 }
866 *len = socklist[i].sock;
867 return -1;
868 }
869 return i;
870 }
871 }
872 } else if (x == -1)
873 return -2; /* socket error */
874 else {
875 s[0] = 0;
876 *len = 0;
877 }
878 return -3;
879 }
880
881 /* sockgets: buffer and read from sockets
882 *
883 * Attempts to read from all registered sockets for up to one second. if
884 * after one second, no complete data has been received from any of the
885 * sockets, 's' will be empty, 'len' will be 0, and sockgets will return -3.
886 * if there is returnable data received from a socket, the data will be
887 * in 's' (null-terminated if non-binary), the length will be returned
888 * in len, and the socket number will be returned.
889 * normal sockets have their input buffered, and each call to sockgets
890 * will return one line terminated with a '\n'. binary sockets are not
891 * buffered and return whatever coems in as soon as it arrives.
892 * listening sockets will return an empty string when a connection comes in.
893 * connecting sockets will return an empty string on a successful connect,
894 * or EOF on a failed connect.
895 * if an EOF is detected from any of the sockets, that socket number will be
896 * put in len, and -1 will be returned.
897 * the maximum length of the string returned is 512 (including null)
898 *
899 * Returns -4 if we handled something that shouldn't be handled by the
900 * dcc functions. Simply ignore it.
901 */
902
903 int sockgets(char *s, int *len)
904 {
905 char xx[514], *p, *px;
906 int ret, i, data = 0;
907
908 for (i = 0; i < MAXSOCKS; i++) {
909 /* Check for stored-up data waiting to be processed */
910 if (!(socklist[i].flags & SOCK_UNUSED) &&
911 !(socklist[i].flags & SOCK_BUFFER) && (socklist[i].inbuf != NULL)) {
912 if (!(socklist[i].flags & SOCK_BINARY)) {
913 /* look for \r too cos windows can't follow RFCs */
914 p = strchr(socklist[i].inbuf, '\n');
915 if (p == NULL)
916 p = strchr(socklist[i].inbuf, '\r');
917 if (p != NULL) {
918 *p = 0;
919 if (strlen(socklist[i].inbuf) > 510)
920 socklist[i].inbuf[510] = 0;
921 strcpy(s, socklist[i].inbuf);
922 /* Uhm... very strange way to code this... */
923 px = (char *) malloc(strlen(p + 1) + 1);
924 strcpy(px, p + 1);
925 free(socklist[i].inbuf);
926 if (px[0])
927 socklist[i].inbuf = px;
928 else {
929 free(px);
930 socklist[i].inbuf = NULL;
931 }
932 /* Strip CR if this was CR/LF combo */
933 if (s[strlen(s) - 1] == '\r')
934 s[strlen(s) - 1] = 0;
935 *len = strlen(s);
936 return socklist[i].sock;
937 }
938 } else {
939 /* Handling buffered binary data (must have been SOCK_BUFFER before). */
940 if (socklist[i].inbuflen <= 510) {
941 *len = socklist[i].inbuflen;
942 memcpy(s, socklist[i].inbuf, socklist[i].inbuflen);
943 free_null(socklist[i].inbuf);
944 socklist[i].inbuflen = 0;
945 } else {
946 /* Split up into chunks of 510 bytes. */
947 *len = 510;
948 memcpy(s, socklist[i].inbuf, *len);
949 memcpy(socklist[i].inbuf, socklist[i].inbuf + *len, *len);
950 socklist[i].inbuflen -= *len;
951 socklist[i].inbuf = realloc(socklist[i].inbuf,
952 socklist[i].inbuflen);
953 }
954 return socklist[i].sock;
955 }
956 }
957 /* Also check any sockets that might have EOF'd during write */
958 if (!(socklist[i].flags & SOCK_UNUSED)
959 && (socklist[i].flags & SOCK_EOFD)) {
960 s[0] = 0;
961 *len = socklist[i].sock;
962 return -1;
963 }
964 }
965 /* No pent-up data of any worth -- down to business */
966 *len = 0;
967 ret = sockread(xx, len);
968 if (ret < 0) {
969 s[0] = 0;
970 return ret;
971 }
972 /* Binary, listening and passed on sockets don't get buffered. */
973 if (socklist[ret].flags & SOCK_CONNECT) {
974 if (socklist[ret].flags & SOCK_STRONGCONN) {
975 socklist[ret].flags &= ~SOCK_STRONGCONN;
976 /* Buffer any data that came in, for future read. */
977 socklist[ret].inbuflen = *len;
978 socklist[ret].inbuf = (char *) malloc(*len + 1);
979 /* It might be binary data. You never know. */
980 memcpy(socklist[ret].inbuf, xx, *len);
981 socklist[ret].inbuf[*len] = 0;
982 }
983 socklist[ret].flags &= ~SOCK_CONNECT;
984 s[0] = 0;
985 return socklist[ret].sock;
986 }
987 if (socklist[ret].flags & SOCK_BINARY) {
988 memcpy(s, xx, *len);
989 return socklist[ret].sock;
990 }
991 if ((socklist[ret].flags & SOCK_LISTEN) ||
992 (socklist[ret].flags & SOCK_PASS))
993 return socklist[ret].sock;
994 if (socklist[ret].flags & SOCK_BUFFER) {
995 socklist[ret].inbuf = (char *) realloc(socklist[ret].inbuf,
996 socklist[ret].inbuflen + *len + 1);
997 memcpy(socklist[ret].inbuf + socklist[ret].inbuflen, xx, *len);
998 socklist[ret].inbuflen += *len;
999 /* We don't know whether it's binary data. Make sure normal strings
1000 will be handled properly later on too. */
1001 socklist[ret].inbuf[socklist[ret].inbuflen] = 0;
1002 return -4; /* Ignore this one. */
1003 }
1004 /* Might be necessary to prepend stored-up data! */
1005 if (socklist[ret].inbuf != NULL) {
1006 p = socklist[ret].inbuf;
1007 socklist[ret].inbuf = (char *) malloc(strlen(p) + strlen(xx) + 1);
1008 strcpy(socklist[ret].inbuf, p);
1009 strcat(socklist[ret].inbuf, xx);
1010 free(p);
1011 if (strlen(socklist[ret].inbuf) < 512) {
1012 strcpy(xx, socklist[ret].inbuf);
1013 free_null(socklist[ret].inbuf);
1014 socklist[ret].inbuflen = 0;
1015 } else {
1016 p = socklist[ret].inbuf;
1017 socklist[ret].inbuflen = strlen(p) - 510;
1018 socklist[ret].inbuf = (char *) malloc(socklist[ret].inbuflen + 1);
1019 strcpy(socklist[ret].inbuf, p + 510);
1020 *(p + 510) = 0;
1021 strcpy(xx, p);
1022 free(p);
1023 /* (leave the rest to be post-pended later) */
1024 }
1025 }
1026 /* Look for EOL marker; if it's there, i have something to show */
1027 p = strchr(xx, '\n');
1028 if (p == NULL)
1029 p = strchr(xx, '\r');
1030 if (p != NULL) {
1031 *p = 0;
1032 strcpy(s, xx);
1033 strcpy(xx, p + 1);
1034 if (s[strlen(s) - 1] == '\r')
1035 s[strlen(s) - 1] = 0;
1036 data = 1; /* DCC_CHAT may now need to process a
1037 blank line */
1038 /* NO! */
1039 /* if (!s[0]) strcpy(s," "); */
1040 } else {
1041 s[0] = 0;
1042 if (strlen(xx) >= 510) {
1043 /* String is too long, so just insert fake \n */
1044 strcpy(s, xx);
1045 xx[0] = 0;
1046 data = 1;
1047 }
1048 }
1049 *len = strlen(s);
1050 /* Anything left that needs to be saved? */
1051 if (!xx[0]) {
1052 if (data)
1053 return socklist[ret].sock;
1054 else
1055 return -3;
1056 }
1057 /* Prepend old data back */
1058 if (socklist[ret].inbuf != NULL) {
1059 p = socklist[ret].inbuf;
1060 socklist[ret].inbuflen = strlen(p) + strlen(xx);
1061 socklist[ret].inbuf = (char *) malloc(socklist[ret].inbuflen + 1);
1062 strcpy(socklist[ret].inbuf, xx);
1063 strcat(socklist[ret].inbuf, p);
1064 free(p);
1065 } else {
1066 socklist[ret].inbuflen = strlen(xx);
1067 socklist[ret].inbuf = (char *) malloc(socklist[ret].inbuflen + 1);
1068 strcpy(socklist[ret].inbuf, xx);
1069 }
1070 if (data) {
1071 return socklist[ret].sock;
1072 } else {
1073 return -3;
1074 }
1075 }
1076
1077 /* Dump something to a socket
1078 */
1079 void tputs(register int z, char *s, unsigned int len)
1080 {
1081 register int i, x, idx;
1082 char *p;
1083 static int inhere = 0;
1084
1085 if (z < 0)
1086 return; /* um... HELLO?! sanity check please! */
1087 if (((z == STDOUT) || (z == STDERR)) && (!backgrd || use_stderr)) {
1088 write(z, s, len);
1089 return;
1090 }
1091 for (i = 0; i < MAXSOCKS; i++) {
1092 if (!(socklist[i].flags & SOCK_UNUSED) && (socklist[i].sock == z)) {
1093 for (idx = 0; idx < dcc_total; idx++) {
1094 if (dcc[idx].sock == z) {
1095 if (dcc[idx].type) {
1096 if (dcc[idx].type->name) {
1097 if (!strncmp(dcc[idx].type->name, "BOT", 3)) {
1098 traffic.out_today.bn += len;
1099 break;
1100 } else if (!strcmp(dcc[idx].type->name, "SERVER")) {
1101 traffic.out_today.irc += len;
1102 break;
1103 } else if (!strncmp(dcc[idx].type->name, "CHAT", 4)) {
1104 traffic.out_today.dcc += len;
1105 break;
1106 } else if (!strncmp(dcc[idx].type->name, "FILES", 5)) {
1107 traffic.out_today.filesys += len;
1108 break;
1109 } else if (!strcmp(dcc[idx].type->name, "SEND")) {
1110 traffic.out_today.trans += len;
1111 break;
1112 } else if (!strncmp(dcc[idx].type->name, "GET", 3)) {
1113 traffic.out_today.trans += len;
1114 break;
1115 } else {
1116 traffic.out_today.unknown += len;
1117 break;
1118 }
1119 }
1120 }
1121 }
1122 }
1123
1124 if (socklist[i].outbuf != NULL) {
1125 /* Already queueing: just add it */
1126 p = (char *) realloc(socklist[i].outbuf, socklist[i].outbuflen + len);
1127 memcpy(p + socklist[i].outbuflen, s, len);
1128 socklist[i].outbuf = p;
1129 socklist[i].outbuflen += len;
1130 return;
1131 }
1132 /* Try. */
1133 x = write(z, s, len);
1134 if (x == (-1))
1135 x = 0;
1136 if (x < len) {
1137 /* Socket is full, queue it */
1138 socklist[i].outbuf = (char *) malloc(len - x);
1139 memcpy(socklist[i].outbuf, &s[x], len - x);
1140 socklist[i].outbuflen = len - x;
1141 }
1142 return;
1143 }
1144 }
1145 /* Make sure we don't cause a crash by looping here */
1146 if (!inhere) {
1147 inhere = 1;
1148
1149 putlog(LOG_MISC, "*", "!!! writing to nonexistent socket: %d", z);
1150 s[strlen(s) - 1] = 0;
1151 putlog(LOG_MISC, "*", "!-> '%s'", s);
1152
1153 inhere = 0;
1154 }
1155 }
1156
1157 /* tputs might queue data for sockets, let's dump as much of it as
1158 * possible.
1159 */
1160 void dequeue_sockets()
1161 {
1162 int i, x;
1163
1164 for (i = 0; i < MAXSOCKS; i++) {
1165 if (!(socklist[i].flags & SOCK_UNUSED) &&
1166 socklist[i].outbuf != NULL) {
1167 /* Trick tputs into doing the work */
1168 x = write(socklist[i].sock, socklist[i].outbuf,
1169 socklist[i].outbuflen);
1170 if ((x < 0) && (errno != EAGAIN)
1171 #ifdef EBADSLT
1172 && (errno != EBADSLT)
1173 #endif
1174 #ifdef ENOTCONN
1175 && (errno != ENOTCONN)
1176 #endif
1177 ) {
1178 /* This detects an EOF during writing */
1179 debug3("net: eof!(write) socket %d (%s,%d)", socklist[i].sock,
1180 strerror(errno), errno);
1181 socklist[i].flags |= SOCK_EOFD;
1182 } else if (x == socklist[i].outbuflen) {
1183 /* If the whole buffer was sent, nuke it */
1184 free_null(socklist[i].outbuf);
1185 socklist[i].outbuflen = 0;
1186 } else if (x > 0) {
1187 char *p = socklist[i].outbuf;
1188
1189 /* This removes any sent bytes from the beginning of the buffer */
1190 socklist[i].outbuf = (char *) malloc(socklist[i].outbuflen - x);
1191 memcpy(socklist[i].outbuf, p + x, socklist[i].outbuflen - x);
1192 socklist[i].outbuflen -= x;
1193 free(p);
1194 }
1195
1196 /* All queued data was sent. Call handler if one exists and the
1197 * dcc entry wants it.
1198 */
1199 if (!socklist[i].outbuf) {
1200 int idx = findanyidx(socklist[i].sock);
1201
1202 if (idx > 0 && dcc[idx].type && dcc[idx].type->outdone)
1203 dcc[idx].type->outdone(idx);
1204 }
1205 }
1206 }
1207 }
1208
1209
1210 /*
1211 * Debugging stuff
1212 */
1213
1214 void tell_netdebug(int idx)
1215 {
1216 int i;
1217 char s[80];
1218
1219 dprintf(idx, "Open sockets:");
1220 for (i = 0; i < MAXSOCKS; i++) {
1221 if (!(socklist[i].flags & SOCK_UNUSED)) {
1222 sprintf(s, " %d", socklist[i].sock);
1223 if (socklist[i].flags & SOCK_BINARY)
1224 strcat(s, " (binary)");
1225 if (socklist[i].flags & SOCK_LISTEN)
1226 strcat(s, " (listen)");
1227 if (socklist[i].flags & SOCK_PASS)
1228 strcat(s, " (passed on)");
1229 if (socklist[i].flags & SOCK_CONNECT)
1230 strcat(s, " (connecting)");
1231 if (socklist[i].flags & SOCK_STRONGCONN)
1232 strcat(s, " (strong)");
1233 if (socklist[i].flags & SOCK_NONSOCK)
1234 strcat(s, " (file)");
1235 if (socklist[i].inbuf != NULL)
1236 sprintf(&s[strlen(s)], " (inbuf: %04X)", strlen(socklist[i].inbuf));
1237 if (socklist[i].outbuf != NULL)
1238 sprintf(&s[strlen(s)], " (outbuf: %06lX)", socklist[i].outbuflen);
1239 strcat(s, ",");
1240 dprintf(idx, "%s", s);
1241 }
1242 }
1243 dprintf(idx, " done.\n");
1244 }
1245
1246 /* Checks wether the referenced socket has data queued.
1247 *
1248 * Returns true if the incoming/outgoing (depending on 'type') queues
1249 * contain data, otherwise false.
1250 */
1251 int sock_has_data(int type, int sock)
1252 {
1253 int ret = 0, i;
1254
1255 for (i = 0; i < MAXSOCKS; i++)
1256 if (!(socklist[i].flags & SOCK_UNUSED) && socklist[i].sock == sock)
1257 break;
1258 if (i < MAXSOCKS) {
1259 switch (type) {
1260 case SOCK_DATA_OUTGOING:
1261 ret = (socklist[i].outbuf != NULL);
1262 break;
1263 case SOCK_DATA_INCOMING:
1264 ret = (socklist[i].inbuf != NULL);
1265 break;
1266 }
1267 } else
1268 debug1("sock_has_data: could not find socket #%d, returning false.", sock);
1269 return ret;
1270 }
1271
1272 /* flush_inbuf():
1273 * checks if there's data in the incoming buffer of an connection
1274 * and flushs the buffer if possible
1275 *
1276 * returns: -1 if the dcc entry wasn't found
1277 * -2 if dcc[idx].type->activity doesn't exist and the data couldn't
1278 * be handled
1279 * 0 if buffer was empty
1280 * otherwise length of flushed buffer
1281 */
1282 int flush_inbuf(int idx)
1283 {
1284 int i, len;
1285 char *inbuf;
1286
1287 assert((idx >= 0) && (idx < dcc_total));
1288 for (i = 0; i < MAXSOCKS; i++) {
1289 if ((dcc[idx].sock == socklist[i].sock)
1290 && !(socklist[i].flags & SOCK_UNUSED)) {
1291 len = socklist[i].inbuflen;
1292 if ((len > 0) && socklist[i].inbuf) {
1293 if (dcc[idx].type && dcc[idx].type->activity) {
1294 inbuf = socklist[i].inbuf;
1295 socklist[i].inbuf = NULL;
1296 dcc[idx].type->activity(idx, inbuf, len);
1297 free(inbuf);
1298 return len;
1299 } else
1300 return -2;
1301 } else
1302 return 0;
1303 }
1304 }
1305 return -1;
1306 }

webmaster@eggheads.org
ViewVC Help
Powered by ViewVC 1.1.23