1 |
/* |
2 |
* ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd). |
3 |
* s_bsd.c: Network functions. |
4 |
* |
5 |
* Copyright (C) 2002 by the past and present ircd coders, and others. |
6 |
* |
7 |
* This program is free software; you can redistribute it and/or modify |
8 |
* it under the terms of the GNU General Public License as published by |
9 |
* the Free Software Foundation; either version 2 of the License, or |
10 |
* (at your option) any later version. |
11 |
* |
12 |
* This program is distributed in the hope that it will be useful, |
13 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
15 |
* GNU General Public License for more details. |
16 |
* |
17 |
* You should have received a copy of the GNU General Public License |
18 |
* along with this program; if not, write to the Free Software |
19 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 |
20 |
* USA |
21 |
* |
22 |
* $Id$ |
23 |
*/ |
24 |
|
25 |
#include "stdinc.h" |
26 |
#ifndef _WIN32 |
27 |
#include <netinet/in_systm.h> |
28 |
#include <netinet/ip.h> |
29 |
#include <netinet/tcp.h> |
30 |
#endif |
31 |
#include "fdlist.h" |
32 |
#include "s_bsd.h" |
33 |
#include "client.h" |
34 |
#include "common.h" |
35 |
#include "dbuf.h" |
36 |
#include "event.h" |
37 |
#include "irc_string.h" |
38 |
#include "irc_getnameinfo.h" |
39 |
#include "irc_getaddrinfo.h" |
40 |
#include "ircd.h" |
41 |
#include "list.h" |
42 |
#include "listener.h" |
43 |
#include "numeric.h" |
44 |
#include "packet.h" |
45 |
#include "irc_res.h" |
46 |
#include "inet_misc.h" |
47 |
#include "restart.h" |
48 |
#include "s_auth.h" |
49 |
#include "s_conf.h" |
50 |
#include "s_log.h" |
51 |
#include "s_serv.h" |
52 |
#include "s_stats.h" |
53 |
#include "send.h" |
54 |
#include "memory.h" |
55 |
#include "s_user.h" |
56 |
#include "hook.h" |
57 |
|
58 |
static const char *comm_err_str[] = { "Comm OK", "Error during bind()", |
59 |
"Error during DNS lookup", "connect timeout", "Error during connect()", |
60 |
"Comm Error" }; |
61 |
|
62 |
struct Callback *setup_socket_cb = NULL; |
63 |
|
64 |
static void comm_connect_callback(fde_t *fd, int status); |
65 |
static PF comm_connect_timeout; |
66 |
static void comm_connect_dns_callback(void *vptr, struct DNSReply *reply); |
67 |
static PF comm_connect_tryconnect; |
68 |
|
69 |
extern void init_netio(void); |
70 |
|
71 |
/* check_can_use_v6() |
72 |
* Check if the system can open AF_INET6 sockets |
73 |
*/ |
74 |
void |
75 |
check_can_use_v6(void) |
76 |
{ |
77 |
#ifdef IPV6 |
78 |
int v6; |
79 |
|
80 |
if ((v6 = socket(AF_INET6, SOCK_STREAM, 0)) < 0) |
81 |
ServerInfo.can_use_v6 = 0; |
82 |
else |
83 |
{ |
84 |
ServerInfo.can_use_v6 = 1; |
85 |
#ifdef _WIN32 |
86 |
closesocket(v6); |
87 |
#else |
88 |
close(v6); |
89 |
#endif |
90 |
} |
91 |
#else |
92 |
ServerInfo.can_use_v6 = 0; |
93 |
#endif |
94 |
} |
95 |
|
96 |
/* get_sockerr - get the error value from the socket or the current errno |
97 |
* |
98 |
* Get the *real* error from the socket (well try to anyway..). |
99 |
* This may only work when SO_DEBUG is enabled but its worth the |
100 |
* gamble anyway. |
101 |
*/ |
102 |
int |
103 |
get_sockerr(int fd) |
104 |
{ |
105 |
#ifndef _WIN32 |
106 |
int errtmp = errno; |
107 |
#else |
108 |
int errtmp = WSAGetLastError(); |
109 |
#endif |
110 |
#ifdef SO_ERROR |
111 |
int err = 0; |
112 |
socklen_t len = sizeof(err); |
113 |
|
114 |
if (-1 < fd && !getsockopt(fd, SOL_SOCKET, SO_ERROR, (char*) &err, (socklen_t *)&len)) |
115 |
{ |
116 |
if (err) |
117 |
errtmp = err; |
118 |
} |
119 |
errno = errtmp; |
120 |
#endif |
121 |
return errtmp; |
122 |
} |
123 |
|
124 |
/* |
125 |
* report_error - report an error from an errno. |
126 |
* Record error to log and also send a copy to all *LOCAL* opers online. |
127 |
* |
128 |
* text is a *format* string for outputing error. It must |
129 |
* contain only two '%s', the first will be replaced |
130 |
* by the sockhost from the client_p, and the latter will |
131 |
* be taken from sys_errlist[errno]. |
132 |
* |
133 |
* client_p if not NULL, is the *LOCAL* client associated with |
134 |
* the error. |
135 |
* |
136 |
* Cannot use perror() within daemon. stderr is closed in |
137 |
* ircd and cannot be used. And, worse yet, it might have |
138 |
* been reassigned to a normal connection... |
139 |
* |
140 |
* Actually stderr is still there IFF ircd was run with -s --Rodder |
141 |
*/ |
142 |
|
143 |
void |
144 |
report_error(int level, const char* text, const char* who, int error) |
145 |
{ |
146 |
who = (who) ? who : ""; |
147 |
|
148 |
sendto_realops_flags(UMODE_DEBUG, level, text, who, strerror(error)); |
149 |
log_oper_action(LOG_IOERR_TYPE, NULL, "%s %s %s\n", who, text, strerror(error)); |
150 |
ilog(L_ERROR, text, who, strerror(error)); |
151 |
} |
152 |
|
153 |
/* |
154 |
* setup_socket() |
155 |
* |
156 |
* Set the socket non-blocking, and other wonderful bits. |
157 |
*/ |
158 |
static void * |
159 |
setup_socket(va_list args) |
160 |
{ |
161 |
int fd = va_arg(args, int); |
162 |
int opt = 1; |
163 |
|
164 |
setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char *) &opt, sizeof(opt)); |
165 |
|
166 |
#ifdef IPTOS_LOWDELAY |
167 |
opt = IPTOS_LOWDELAY; |
168 |
setsockopt(fd, IPPROTO_IP, IP_TOS, (char *) &opt, sizeof(opt)); |
169 |
#endif |
170 |
|
171 |
#ifndef _WIN32 |
172 |
fcntl(fd, F_SETFL, fcntl(fd, F_GETFL, 0) | O_NONBLOCK); |
173 |
#endif |
174 |
|
175 |
return NULL; |
176 |
} |
177 |
|
178 |
/* |
179 |
* init_comm() |
180 |
* |
181 |
* Initializes comm subsystem. |
182 |
*/ |
183 |
void |
184 |
init_comm(void) |
185 |
{ |
186 |
setup_socket_cb = register_callback("setup_socket", setup_socket); |
187 |
init_netio(); |
188 |
} |
189 |
|
190 |
/* |
191 |
* close_connection |
192 |
* Close the physical connection. This function must make |
193 |
* MyConnect(client_p) == FALSE, and set client_p->from == NULL. |
194 |
*/ |
195 |
void |
196 |
close_connection(struct Client *client_p) |
197 |
{ |
198 |
struct ConfItem *conf; |
199 |
struct AccessItem *aconf; |
200 |
struct ClassItem *aclass; |
201 |
|
202 |
assert(NULL != client_p); |
203 |
|
204 |
if (IsServer(client_p)) |
205 |
{ |
206 |
ServerStats->is_sv++; |
207 |
ServerStats->is_sbs += client_p->localClient->send.bytes; |
208 |
ServerStats->is_sbr += client_p->localClient->recv.bytes; |
209 |
ServerStats->is_sti += CurrentTime - client_p->firsttime; |
210 |
|
211 |
/* XXX Does this even make any sense at all anymore? |
212 |
* scheduling a 'quick' reconnect could cause a pile of |
213 |
* nick collides under TSora protocol... -db |
214 |
*/ |
215 |
/* |
216 |
* If the connection has been up for a long amount of time, schedule |
217 |
* a 'quick' reconnect, else reset the next-connect cycle. |
218 |
*/ |
219 |
if ((conf = find_conf_exact(SERVER_TYPE, |
220 |
client_p->name, client_p->username, |
221 |
client_p->host))) |
222 |
{ |
223 |
/* |
224 |
* Reschedule a faster reconnect, if this was a automatically |
225 |
* connected configuration entry. (Note that if we have had |
226 |
* a rehash in between, the status has been changed to |
227 |
* CONF_ILLEGAL). But only do this if it was a "good" link. |
228 |
*/ |
229 |
aconf = (struct AccessItem *)map_to_conf(conf); |
230 |
aclass = (struct ClassItem *)map_to_conf(aconf->class_ptr); |
231 |
aconf->hold = time(NULL); |
232 |
aconf->hold += (aconf->hold - client_p->since > HANGONGOODLINK) ? |
233 |
HANGONRETRYDELAY : ConFreq(aclass); |
234 |
if (nextconnect > aconf->hold) |
235 |
nextconnect = aconf->hold; |
236 |
} |
237 |
} |
238 |
else if (IsClient(client_p)) |
239 |
{ |
240 |
ServerStats->is_cl++; |
241 |
ServerStats->is_cbs += client_p->localClient->send.bytes; |
242 |
ServerStats->is_cbr += client_p->localClient->recv.bytes; |
243 |
ServerStats->is_cti += CurrentTime - client_p->firsttime; |
244 |
} |
245 |
else |
246 |
ServerStats->is_ni++; |
247 |
|
248 |
if (!IsDead(client_p)) |
249 |
{ |
250 |
/* attempt to flush any pending dbufs. Evil, but .. -- adrian */ |
251 |
/* there is still a chance that we might send data to this socket |
252 |
* even if it is marked as blocked (COMM_SELECT_READ handler is called |
253 |
* before COMM_SELECT_WRITE). Let's try, nothing to lose.. -adx |
254 |
*/ |
255 |
ClearSendqBlocked(client_p); |
256 |
send_queued_write(client_p); |
257 |
} |
258 |
|
259 |
#ifdef HAVE_LIBCRYPTO |
260 |
if (client_p->localClient->fd.ssl) |
261 |
SSL_shutdown(client_p->localClient->fd.ssl); |
262 |
#endif |
263 |
if (client_p->localClient->fd.flags.open) |
264 |
fd_close(&client_p->localClient->fd); |
265 |
|
266 |
if (HasServlink(client_p)) |
267 |
{ |
268 |
if (client_p->localClient->ctrlfd.flags.open) |
269 |
fd_close(&client_p->localClient->ctrlfd); |
270 |
} |
271 |
|
272 |
dbuf_clear(&client_p->localClient->buf_sendq); |
273 |
dbuf_clear(&client_p->localClient->buf_recvq); |
274 |
|
275 |
MyFree(client_p->localClient->passwd); |
276 |
detach_conf(client_p, CONF_TYPE); |
277 |
client_p->from = NULL; /* ...this should catch them! >:) --msa */ |
278 |
} |
279 |
|
280 |
#ifdef HAVE_LIBCRYPTO |
281 |
/* |
282 |
* ssl_handshake - let OpenSSL initialize the protocol. Register for |
283 |
* read/write events if necessary. |
284 |
*/ |
285 |
static void |
286 |
ssl_handshake(int fd, struct Client *client_p) |
287 |
{ |
288 |
int ret = SSL_accept(client_p->localClient->fd.ssl); |
289 |
|
290 |
if (ret <= 0) |
291 |
switch (SSL_get_error(client_p->localClient->fd.ssl, ret)) |
292 |
{ |
293 |
case SSL_ERROR_WANT_WRITE: |
294 |
comm_setselect(&client_p->localClient->fd, COMM_SELECT_WRITE, |
295 |
(PF *) ssl_handshake, client_p, 0); |
296 |
return; |
297 |
|
298 |
case SSL_ERROR_WANT_READ: |
299 |
comm_setselect(&client_p->localClient->fd, COMM_SELECT_READ, |
300 |
(PF *) ssl_handshake, client_p, 0); |
301 |
return; |
302 |
|
303 |
default: |
304 |
exit_client(client_p, client_p, "Error during SSL handshake"); |
305 |
return; |
306 |
} |
307 |
|
308 |
execute_callback(auth_cb, client_p); |
309 |
} |
310 |
#endif |
311 |
|
312 |
/* |
313 |
* add_connection - creates a client which has just connected to us on |
314 |
* the given fd. The sockhost field is initialized with the ip# of the host. |
315 |
* An unique id is calculated now, in case it is needed for auth. |
316 |
* The client is sent to the auth module for verification, and not put in |
317 |
* any client list yet. |
318 |
*/ |
319 |
void |
320 |
add_connection(struct Listener* listener, int fd) |
321 |
{ |
322 |
struct Client *new_client; |
323 |
socklen_t len = sizeof(struct irc_ssaddr); |
324 |
struct irc_ssaddr irn; |
325 |
assert(NULL != listener); |
326 |
|
327 |
/* |
328 |
* get the client socket name from the socket |
329 |
* the client has already been checked out in accept_connection |
330 |
*/ |
331 |
|
332 |
memset(&irn, 0, sizeof(irn)); |
333 |
if (getpeername(fd, (struct sockaddr *)&irn, (socklen_t *)&len)) |
334 |
{ |
335 |
#ifdef _WIN32 |
336 |
errno = WSAGetLastError(); |
337 |
#endif |
338 |
report_error(L_ALL, "Failed in adding new connection %s :%s", |
339 |
get_listener_name(listener), errno); |
340 |
ServerStats->is_ref++; |
341 |
#ifdef _WIN32 |
342 |
closesocket(fd); |
343 |
#else |
344 |
close(fd); |
345 |
#endif |
346 |
return; |
347 |
} |
348 |
|
349 |
#ifdef IPV6 |
350 |
remove_ipv6_mapping(&irn); |
351 |
#else |
352 |
irn.ss_len = len; |
353 |
#endif |
354 |
new_client = make_client(NULL); |
355 |
fd_open(&new_client->localClient->fd, fd, 1, |
356 |
(listener->flags & LISTENER_SSL) ? |
357 |
"Incoming SSL connection" : "Incoming connection"); |
358 |
memset(&new_client->localClient->ip, 0, sizeof(struct irc_ssaddr)); |
359 |
|
360 |
/* |
361 |
* copy address to 'sockhost' as a string, copy it to host too |
362 |
* so we have something valid to put into error messages... |
363 |
*/ |
364 |
memcpy(&new_client->localClient->ip, &irn, sizeof(struct irc_ssaddr)); |
365 |
|
366 |
irc_getnameinfo((struct sockaddr*)&new_client->localClient->ip, |
367 |
new_client->localClient->ip.ss_len, new_client->sockhost, |
368 |
HOSTIPLEN, NULL, 0, NI_NUMERICHOST); |
369 |
new_client->localClient->aftype = new_client->localClient->ip.ss.ss_family; |
370 |
|
371 |
*new_client->host = '\0'; |
372 |
#ifdef IPV6 |
373 |
if (*new_client->sockhost == ':') |
374 |
strlcat(new_client->host, "0", HOSTLEN+1); |
375 |
|
376 |
if (new_client->localClient->aftype == AF_INET6 && |
377 |
ConfigFileEntry.dot_in_ip6_addr == 1) |
378 |
{ |
379 |
strlcat(new_client->host, new_client->sockhost,HOSTLEN+1); |
380 |
strlcat(new_client->host, ".", HOSTLEN+1); |
381 |
} else |
382 |
#endif |
383 |
strlcat(new_client->host, new_client->sockhost,HOSTLEN+1); |
384 |
|
385 |
new_client->localClient->listener = listener; |
386 |
++listener->ref_count; |
387 |
|
388 |
connect_id++; |
389 |
new_client->connect_id = connect_id; |
390 |
|
391 |
#ifdef HAVE_LIBCRYPTO |
392 |
if ((listener->flags & LISTENER_SSL)) |
393 |
{ |
394 |
if ((new_client->localClient->fd.ssl = SSL_new(ServerInfo.ctx)) == NULL) |
395 |
{ |
396 |
ilog(L_CRIT, "SSL_new() ERROR! -- %s", |
397 |
ERR_error_string(ERR_get_error(), NULL)); |
398 |
|
399 |
SetDead(new_client); |
400 |
exit_client(new_client, new_client, "SSL_new failed"); |
401 |
return; |
402 |
} |
403 |
|
404 |
SSL_set_fd(new_client->localClient->fd.ssl, fd); |
405 |
ssl_handshake(0, new_client); |
406 |
} |
407 |
else |
408 |
#endif |
409 |
execute_callback(auth_cb, new_client); |
410 |
} |
411 |
|
412 |
/* |
413 |
* stolen from squid - its a neat (but overused! :) routine which we |
414 |
* can use to see whether we can ignore this errno or not. It is |
415 |
* generally useful for non-blocking network IO related errnos. |
416 |
* -- adrian |
417 |
*/ |
418 |
int |
419 |
ignoreErrno(int ierrno) |
420 |
{ |
421 |
switch (ierrno) |
422 |
{ |
423 |
case EINPROGRESS: |
424 |
case EWOULDBLOCK: |
425 |
#if EAGAIN != EWOULDBLOCK |
426 |
case EAGAIN: |
427 |
#endif |
428 |
case EALREADY: |
429 |
case EINTR: |
430 |
#ifdef ERESTART |
431 |
case ERESTART: |
432 |
#endif |
433 |
return 1; |
434 |
default: |
435 |
return 0; |
436 |
} |
437 |
} |
438 |
|
439 |
/* |
440 |
* comm_settimeout() - set the socket timeout |
441 |
* |
442 |
* Set the timeout for the fd |
443 |
*/ |
444 |
void |
445 |
comm_settimeout(fde_t *fd, time_t timeout, PF *callback, void *cbdata) |
446 |
{ |
447 |
assert(fd->flags.open); |
448 |
|
449 |
fd->timeout = CurrentTime + (timeout / 1000); |
450 |
fd->timeout_handler = callback; |
451 |
fd->timeout_data = cbdata; |
452 |
} |
453 |
|
454 |
/* |
455 |
* comm_setflush() - set a flush function |
456 |
* |
457 |
* A flush function is simply a function called if found during |
458 |
* comm_timeouts(). Its basically a second timeout, except in this case |
459 |
* I'm too lazy to implement multiple timeout functions! :-) |
460 |
* its kinda nice to have it separate, since this is designed for |
461 |
* flush functions, and when comm_close() is implemented correctly |
462 |
* with close functions, we _actually_ don't call comm_close() here .. |
463 |
* -- originally Adrian's notes |
464 |
* comm_close() is replaced with fd_close() in fdlist.c |
465 |
*/ |
466 |
void |
467 |
comm_setflush(fde_t *fd, time_t timeout, PF *callback, void *cbdata) |
468 |
{ |
469 |
assert(fd->flags.open); |
470 |
|
471 |
fd->flush_timeout = CurrentTime + (timeout / 1000); |
472 |
fd->flush_handler = callback; |
473 |
fd->flush_data = cbdata; |
474 |
} |
475 |
|
476 |
/* |
477 |
* comm_checktimeouts() - check the socket timeouts |
478 |
* |
479 |
* All this routine does is call the given callback/cbdata, without closing |
480 |
* down the file descriptor. When close handlers have been implemented, |
481 |
* this will happen. |
482 |
*/ |
483 |
void |
484 |
comm_checktimeouts(void *notused) |
485 |
{ |
486 |
int i; |
487 |
fde_t *F; |
488 |
PF *hdl; |
489 |
void *data; |
490 |
|
491 |
for (i = 0; i < FD_HASH_SIZE; i++) |
492 |
for (F = fd_hash[i]; F != NULL; F = fd_next_in_loop) |
493 |
{ |
494 |
assert(F->flags.open); |
495 |
fd_next_in_loop = F->hnext; |
496 |
|
497 |
/* check flush functions */ |
498 |
if (F->flush_handler && F->flush_timeout > 0 && |
499 |
F->flush_timeout < CurrentTime) |
500 |
{ |
501 |
hdl = F->flush_handler; |
502 |
data = F->flush_data; |
503 |
comm_setflush(F, 0, NULL, NULL); |
504 |
hdl(F, data); |
505 |
} |
506 |
|
507 |
/* check timeouts */ |
508 |
if (F->timeout_handler && F->timeout > 0 && |
509 |
F->timeout < CurrentTime) |
510 |
{ |
511 |
/* Call timeout handler */ |
512 |
hdl = F->timeout_handler; |
513 |
data = F->timeout_data; |
514 |
comm_settimeout(F, 0, NULL, NULL); |
515 |
hdl(F, data); |
516 |
} |
517 |
} |
518 |
} |
519 |
|
520 |
/* |
521 |
* void comm_connect_tcp(int fd, const char *host, unsigned short port, |
522 |
* struct sockaddr *clocal, int socklen, |
523 |
* CNCB *callback, void *data, int aftype, int timeout) |
524 |
* Input: An fd to connect with, a host and port to connect to, |
525 |
* a local sockaddr to connect from + length(or NULL to use the |
526 |
* default), a callback, the data to pass into the callback, the |
527 |
* address family. |
528 |
* Output: None. |
529 |
* Side-effects: A non-blocking connection to the host is started, and |
530 |
* if necessary, set up for selection. The callback given |
531 |
* may be called now, or it may be called later. |
532 |
*/ |
533 |
void |
534 |
comm_connect_tcp(fde_t *fd, const char *host, unsigned short port, |
535 |
struct sockaddr *clocal, int socklen, CNCB *callback, |
536 |
void *data, int aftype, int timeout) |
537 |
{ |
538 |
struct addrinfo hints, *res; |
539 |
char portname[PORTNAMELEN+1]; |
540 |
|
541 |
assert(callback); |
542 |
fd->connect.callback = callback; |
543 |
fd->connect.data = data; |
544 |
|
545 |
fd->connect.hostaddr.ss.ss_family = aftype; |
546 |
fd->connect.hostaddr.ss_port = htons(port); |
547 |
|
548 |
/* Note that we're using a passed sockaddr here. This is because |
549 |
* generally you'll be bind()ing to a sockaddr grabbed from |
550 |
* getsockname(), so this makes things easier. |
551 |
* XXX If NULL is passed as local, we should later on bind() to the |
552 |
* virtual host IP, for completeness. |
553 |
* -- adrian |
554 |
*/ |
555 |
if ((clocal != NULL) && (bind(fd->fd, clocal, socklen) < 0)) |
556 |
{ |
557 |
/* Failure, call the callback with COMM_ERR_BIND */ |
558 |
comm_connect_callback(fd, COMM_ERR_BIND); |
559 |
/* ... and quit */ |
560 |
return; |
561 |
} |
562 |
|
563 |
/* Next, if we have been given an IP, get the addr and skip the |
564 |
* DNS check (and head direct to comm_connect_tryconnect(). |
565 |
*/ |
566 |
memset(&hints, 0, sizeof(hints)); |
567 |
hints.ai_family = AF_UNSPEC; |
568 |
hints.ai_socktype = SOCK_STREAM; |
569 |
hints.ai_flags = AI_PASSIVE | AI_NUMERICHOST; |
570 |
|
571 |
snprintf(portname, PORTNAMELEN, "%d", port); |
572 |
|
573 |
if (irc_getaddrinfo(host, portname, &hints, &res)) |
574 |
{ |
575 |
/* Send the DNS request, for the next level */ |
576 |
fd->dns_query = MyMalloc(sizeof(struct DNSQuery)); |
577 |
fd->dns_query->ptr = fd; |
578 |
fd->dns_query->callback = comm_connect_dns_callback; |
579 |
gethost_byname(host, fd->dns_query); |
580 |
} |
581 |
else |
582 |
{ |
583 |
/* We have a valid IP, so we just call tryconnect */ |
584 |
/* Make sure we actually set the timeout here .. */ |
585 |
assert(res != NULL); |
586 |
memcpy(&fd->connect.hostaddr, res->ai_addr, res->ai_addrlen); |
587 |
fd->connect.hostaddr.ss_len = res->ai_addrlen; |
588 |
fd->connect.hostaddr.ss.ss_family = res->ai_family; |
589 |
irc_freeaddrinfo(res); |
590 |
comm_settimeout(fd, timeout*1000, comm_connect_timeout, NULL); |
591 |
comm_connect_tryconnect(fd, NULL); |
592 |
} |
593 |
} |
594 |
|
595 |
/* |
596 |
* comm_connect_callback() - call the callback, and continue with life |
597 |
*/ |
598 |
static void |
599 |
comm_connect_callback(fde_t *fd, int status) |
600 |
{ |
601 |
CNCB *hdl; |
602 |
|
603 |
/* This check is gross..but probably necessary */ |
604 |
if (fd->connect.callback == NULL) |
605 |
return; |
606 |
|
607 |
/* Clear the connect flag + handler */ |
608 |
hdl = fd->connect.callback; |
609 |
fd->connect.callback = NULL; |
610 |
|
611 |
/* Clear the timeout handler */ |
612 |
comm_settimeout(fd, 0, NULL, NULL); |
613 |
|
614 |
/* Call the handler */ |
615 |
hdl(fd, status, fd->connect.data); |
616 |
} |
617 |
|
618 |
/* |
619 |
* comm_connect_timeout() - this gets called when the socket connection |
620 |
* times out. This *only* can be called once connect() is initially |
621 |
* called .. |
622 |
*/ |
623 |
static void |
624 |
comm_connect_timeout(fde_t *fd, void *notused) |
625 |
{ |
626 |
/* error! */ |
627 |
comm_connect_callback(fd, COMM_ERR_TIMEOUT); |
628 |
} |
629 |
|
630 |
/* |
631 |
* comm_connect_dns_callback() - called at the completion of the DNS request |
632 |
* |
633 |
* The DNS request has completed, so if we've got an error, return it, |
634 |
* otherwise we initiate the connect() |
635 |
*/ |
636 |
static void |
637 |
comm_connect_dns_callback(void *vptr, struct DNSReply *reply) |
638 |
{ |
639 |
fde_t *F = vptr; |
640 |
|
641 |
if (reply == NULL) |
642 |
{ |
643 |
MyFree(F->dns_query); |
644 |
F->dns_query = NULL; |
645 |
comm_connect_callback(F, COMM_ERR_DNS); |
646 |
return; |
647 |
} |
648 |
|
649 |
/* No error, set a 10 second timeout */ |
650 |
comm_settimeout(F, 30*1000, comm_connect_timeout, NULL); |
651 |
|
652 |
/* Copy over the DNS reply info so we can use it in the connect() */ |
653 |
/* |
654 |
* Note we don't fudge the refcount here, because we aren't keeping |
655 |
* the DNS record around, and the DNS cache is gone anyway.. |
656 |
* -- adrian |
657 |
*/ |
658 |
memcpy(&F->connect.hostaddr, &reply->addr, reply->addr.ss_len); |
659 |
/* The cast is hacky, but safe - port offset is same on v4 and v6 */ |
660 |
((struct sockaddr_in *) &F->connect.hostaddr)->sin_port = |
661 |
F->connect.hostaddr.ss_port; |
662 |
F->connect.hostaddr.ss_len = reply->addr.ss_len; |
663 |
|
664 |
/* Now, call the tryconnect() routine to try a connect() */ |
665 |
MyFree(F->dns_query); |
666 |
F->dns_query = NULL; |
667 |
comm_connect_tryconnect(F, NULL); |
668 |
} |
669 |
|
670 |
/* static void comm_connect_tryconnect(int fd, void *notused) |
671 |
* Input: The fd, the handler data(unused). |
672 |
* Output: None. |
673 |
* Side-effects: Try and connect with pending connect data for the FD. If |
674 |
* we succeed or get a fatal error, call the callback. |
675 |
* Otherwise, it is still blocking or something, so register |
676 |
* to select for a write event on this FD. |
677 |
*/ |
678 |
static void |
679 |
comm_connect_tryconnect(fde_t *fd, void *notused) |
680 |
{ |
681 |
int retval; |
682 |
|
683 |
/* This check is needed or re-entrant s_bsd_* like sigio break it. */ |
684 |
if (fd->connect.callback == NULL) |
685 |
return; |
686 |
|
687 |
/* Try the connect() */ |
688 |
retval = connect(fd->fd, (struct sockaddr *) &fd->connect.hostaddr, |
689 |
fd->connect.hostaddr.ss_len); |
690 |
|
691 |
/* Error? */ |
692 |
if (retval < 0) |
693 |
{ |
694 |
#ifdef _WIN32 |
695 |
errno = WSAGetLastError(); |
696 |
#endif |
697 |
/* |
698 |
* If we get EISCONN, then we've already connect()ed the socket, |
699 |
* which is a good thing. |
700 |
* -- adrian |
701 |
*/ |
702 |
if (errno == EISCONN) |
703 |
comm_connect_callback(fd, COMM_OK); |
704 |
else if (ignoreErrno(errno)) |
705 |
/* Ignore error? Reschedule */ |
706 |
comm_setselect(fd, COMM_SELECT_WRITE, comm_connect_tryconnect, |
707 |
NULL, 0); |
708 |
else |
709 |
/* Error? Fail with COMM_ERR_CONNECT */ |
710 |
comm_connect_callback(fd, COMM_ERR_CONNECT); |
711 |
return; |
712 |
} |
713 |
|
714 |
/* If we get here, we've suceeded, so call with COMM_OK */ |
715 |
comm_connect_callback(fd, COMM_OK); |
716 |
} |
717 |
|
718 |
/* |
719 |
* comm_errorstr() - return an error string for the given error condition |
720 |
*/ |
721 |
const char * |
722 |
comm_errstr(int error) |
723 |
{ |
724 |
if (error < 0 || error >= COMM_ERR_MAX) |
725 |
return "Invalid error number!"; |
726 |
return comm_err_str[error]; |
727 |
} |
728 |
|
729 |
/* |
730 |
* comm_open() - open a socket |
731 |
* |
732 |
* This is a highly highly cut down version of squid's comm_open() which |
733 |
* for the most part emulates socket(), *EXCEPT* it fails if we're about |
734 |
* to run out of file descriptors. |
735 |
*/ |
736 |
int |
737 |
comm_open(fde_t *F, int family, int sock_type, int proto, const char *note) |
738 |
{ |
739 |
int fd; |
740 |
|
741 |
/* First, make sure we aren't going to run out of file descriptors */ |
742 |
if (number_fd >= hard_fdlimit) |
743 |
{ |
744 |
errno = ENFILE; |
745 |
return -1; |
746 |
} |
747 |
|
748 |
/* |
749 |
* Next, we try to open the socket. We *should* drop the reserved FD |
750 |
* limit if/when we get an error, but we can deal with that later. |
751 |
* XXX !!! -- adrian |
752 |
*/ |
753 |
fd = socket(family, sock_type, proto); |
754 |
if (fd < 0) |
755 |
{ |
756 |
#ifdef _WIN32 |
757 |
errno = WSAGetLastError(); |
758 |
#endif |
759 |
return -1; /* errno will be passed through, yay.. */ |
760 |
} |
761 |
|
762 |
execute_callback(setup_socket_cb, fd); |
763 |
|
764 |
/* update things in our fd tracking */ |
765 |
fd_open(F, fd, 1, note); |
766 |
return 0; |
767 |
} |
768 |
|
769 |
/* |
770 |
* comm_accept() - accept an incoming connection |
771 |
* |
772 |
* This is a simple wrapper for accept() which enforces FD limits like |
773 |
* comm_open() does. Returned fd must be either closed or tagged with |
774 |
* fd_open (this function no longer does it). |
775 |
*/ |
776 |
int |
777 |
comm_accept(struct Listener *lptr, struct irc_ssaddr *pn) |
778 |
{ |
779 |
int newfd; |
780 |
socklen_t addrlen = sizeof(struct irc_ssaddr); |
781 |
|
782 |
if (number_fd >= hard_fdlimit) |
783 |
{ |
784 |
errno = ENFILE; |
785 |
return -1; |
786 |
} |
787 |
|
788 |
/* |
789 |
* Next, do the accept(). if we get an error, we should drop the |
790 |
* reserved fd limit, but we can deal with that when comm_open() |
791 |
* also does it. XXX -- adrian |
792 |
*/ |
793 |
newfd = accept(lptr->fd.fd, (struct sockaddr *)pn, (socklen_t *)&addrlen); |
794 |
if (newfd < 0) |
795 |
{ |
796 |
#ifdef _WIN32 |
797 |
errno = WSAGetLastError(); |
798 |
#endif |
799 |
return -1; |
800 |
} |
801 |
|
802 |
#ifdef IPV6 |
803 |
remove_ipv6_mapping(pn); |
804 |
#else |
805 |
pn->ss_len = addrlen; |
806 |
#endif |
807 |
|
808 |
execute_callback(setup_socket_cb, newfd); |
809 |
|
810 |
/* .. and return */ |
811 |
return newfd; |
812 |
} |
813 |
|
814 |
/* |
815 |
* remove_ipv6_mapping() - Removes IPv4-In-IPv6 mapping from an address |
816 |
* This function should really inspect the struct itself rather than relying |
817 |
* on inet_pton and inet_ntop. OSes with IPv6 mapping listening on both |
818 |
* AF_INET and AF_INET6 map AF_INET connections inside AF_INET6 structures |
819 |
* |
820 |
*/ |
821 |
#ifdef IPV6 |
822 |
void |
823 |
remove_ipv6_mapping(struct irc_ssaddr *addr) |
824 |
{ |
825 |
if (addr->ss.ss_family == AF_INET6) |
826 |
{ |
827 |
struct sockaddr_in6 *v6; |
828 |
|
829 |
v6 = (struct sockaddr_in6*)addr; |
830 |
if (IN6_IS_ADDR_V4MAPPED(&v6->sin6_addr)) |
831 |
{ |
832 |
char v4ip[HOSTIPLEN]; |
833 |
struct sockaddr_in *v4 = (struct sockaddr_in*)addr; |
834 |
inetntop(AF_INET6, &v6->sin6_addr, v4ip, HOSTIPLEN); |
835 |
inet_pton(AF_INET, v4ip, &v4->sin_addr); |
836 |
addr->ss.ss_family = AF_INET; |
837 |
addr->ss_len = sizeof(struct sockaddr_in); |
838 |
} |
839 |
else |
840 |
addr->ss_len = sizeof(struct sockaddr_in6); |
841 |
} |
842 |
else |
843 |
addr->ss_len = sizeof(struct sockaddr_in); |
844 |
} |
845 |
#endif |