6ff8fe6261976fa343ad4884a3460841bdc057d7
[exim.git] / src / src / ip.c
1 /*************************************************
2 *     Exim - an Internet mail transport agent    *
3 *************************************************/
4
5 /* Copyright (c) University of Cambridge 1995 - 2018 */
6 /* Copyright (c) The Exim Maintainers 2020 - 2021 */
7 /* See the file NOTICE for conditions of use and distribution. */
8 /* SPDX-License-Identifier: GPL-2.0-or-later */
9
10 /* Functions for doing things with sockets. With the advent of IPv6 this has
11 got messier, so that it's worth pulling out the code into separate functions
12 that other parts of Exim can call, especially as there are now several
13 different places in the code where sockets are used. */
14
15
16 #include "exim.h"
17
18
19 #if defined(TCP_FASTOPEN)
20 # if defined(MSG_FASTOPEN) || defined(EXIM_TFO_CONNECTX) || defined(EXIM_TFO_FREEBSD)
21 #  define EXIM_SUPPORT_TFO
22 # endif
23 #endif
24
25 /*************************************************
26 *             Create a socket                    *
27 *************************************************/
28
29 /* Socket creation happens in a number of places so it's packaged here for
30 convenience.
31
32 Arguments:
33   type       SOCK_DGRAM or SOCK_STREAM
34   af         AF_INET or AF_INET6
35
36 Returns:     socket number or -1 on failure
37 */
38
39 int
40 ip_socket(int type, int af)
41 {
42 int sock = socket(af, type, 0);
43 if (sock < 0)
44   log_write(0, LOG_MAIN, "IPv%c socket creation failed: %s",
45     (af == AF_INET6)? '6':'4', strerror(errno));
46 return sock;
47 }
48
49
50
51
52 #if HAVE_IPV6
53 /*************************************************
54 *      Convert printing address to numeric       *
55 *************************************************/
56
57 /* This function converts the textual form of an IP address into a numeric form
58 in an appropriate structure in an IPv6 environment. The getaddrinfo() function
59 can (apparently) handle more complicated addresses (e.g. those containing
60 scopes) than inet_pton() in some environments. We use hints to tell it that the
61 input must be a numeric address.
62
63 However, apparently some operating systems (or libraries) don't support
64 getaddrinfo(), so there is a build-time option to revert to inet_pton() (which
65 does not support scopes).
66
67 Arguments:
68   address     textual form of the address
69   addr        where to copy back the answer
70
71 Returns:      nothing - failure provokes a panic-die
72 */
73
74 static void
75 ip_addrinfo(const uschar *address, struct sockaddr_in6 *saddr)
76 {
77 #ifdef IPV6_USE_INET_PTON
78
79   if (inet_pton(AF_INET6, CCS address, &saddr->sin6_addr) != 1)
80     log_write(0, LOG_MAIN|LOG_PANIC_DIE, "unable to parse \"%s\" as an "
81       "IP address", address);
82   saddr->sin6_family = AF_INET6;
83
84 #else
85
86   int rc;
87   struct addrinfo hints, *res;
88   memset(&hints, 0, sizeof(hints));
89   hints.ai_family = AF_INET6;
90   hints.ai_socktype = SOCK_STREAM;
91   hints.ai_flags = AI_NUMERICHOST;
92   if ((rc = getaddrinfo(CCS address, NULL, &hints, &res)) != 0 || res == NULL)
93     log_write(0, LOG_MAIN|LOG_PANIC_DIE, "unable to parse \"%s\" as an "
94       "IP address: %s", address,
95       (rc == 0)? "NULL result returned" : gai_strerror(rc));
96   memcpy(saddr, res->ai_addr, res->ai_addrlen);
97   freeaddrinfo(res);
98
99 #endif
100 }
101 #endif  /* HAVE_IPV6 */
102
103
104 /*************************************************
105 *         Bind socket to interface and port      *
106 *************************************************/
107
108 int
109 ip_addr(void * sin_, int af, const uschar * address, int port)
110 {
111 union sockaddr_46 * sin = sin_;
112 memset(sin, 0, sizeof(*sin));
113
114 /* Setup code when using an IPv6 socket. The wildcard address is ":", to
115 ensure an IPv6 socket is used. */
116
117 #if HAVE_IPV6
118 if (af == AF_INET6)
119   {
120   if (address[0] == ':' && address[1] == 0)
121     {
122     sin->v6.sin6_family = AF_INET6;
123     sin->v6.sin6_addr = in6addr_any;
124     }
125   else
126     ip_addrinfo(address, &sin->v6);  /* Panic-dies on error */
127   sin->v6.sin6_port = htons(port);
128   return sizeof(sin->v6);
129   }
130 else
131 #endif    /* HAVE_IPV6 */
132
133 /* Setup code when using IPv4 socket. The wildcard address is "". */
134
135   {
136   sin->v4.sin_family = AF_INET;
137   sin->v4.sin_port = htons(port);
138   sin->v4.sin_addr.s_addr = address[0] == 0
139     ? (S_ADDR_TYPE)INADDR_ANY
140     : (S_ADDR_TYPE)inet_addr(CS address);
141   return sizeof(sin->v4);
142   }
143 }
144
145
146
147 /* This function binds a socket to a local interface address and port. For a
148 wildcard IPv6 bind, the address is ":".
149
150 Arguments:
151   sock           the socket
152   af             AF_INET or AF_INET6 - the socket type
153   address        the IP address, in text form
154   port           the IP port (host order)
155
156 Returns:         the result of bind()
157 */
158
159 int
160 ip_bind(int sock, int af, uschar *address, int port)
161 {
162 union sockaddr_46 sin;
163 int s_len = ip_addr(&sin, af, address, port);
164 int rc = bind(sock, (struct sockaddr *)&sin, s_len);
165 if (rc < 0)
166   log_write(0, LOG_MAIN, "bind of [%s]:%d failed", address, port);
167 return rc;
168 }
169
170
171
172 /*************************************************
173 *        Connect socket to remote host           *
174 *************************************************/
175
176 /* This function connects a socket to a remote address and port. The socket may
177 or may not have previously been bound to a local interface. The socket is not
178 closed, even in cases of error. It is expected that the calling function, which
179 created the socket, will be the one that closes it.
180
181 Arguments:
182   sock        the socket
183   af          AF_INET6 or AF_INET for the socket type
184   address     the remote address, in text form
185   port        the remote port
186   timeout     a timeout (zero for indefinite timeout)
187   fastopen_blob    non-null iff TCP_FASTOPEN can be used; may indicate early-data to
188                 be sent in SYN segment.  Any such data must be idempotent.
189
190 Returns:      0 on success; -1 on failure, with errno set
191 */
192
193 int
194 ip_connect(int sock, int af, const uschar *address, int port, int timeout,
195   const blob * fastopen_blob)
196 {
197 struct sockaddr_in s_in4;
198 struct sockaddr *s_ptr;
199 int s_len, rc, save_errno;
200
201 /* For an IPv6 address, use an IPv6 sockaddr structure. */
202
203 #if HAVE_IPV6
204 struct sockaddr_in6 s_in6;
205 if (af == AF_INET6)
206   {
207   memset(&s_in6, 0, sizeof(s_in6));
208   ip_addrinfo(address, &s_in6);   /* Panic-dies on error */
209   s_in6.sin6_port = htons(port);
210   s_ptr = (struct sockaddr *)&s_in6;
211   s_len = sizeof(s_in6);
212   }
213 else
214 #endif    /* HAVE_IPV6 */
215
216 /* For an IPv4 address, use an IPv4 sockaddr structure, even on a system with
217 IPv6 support. */
218
219   {
220   memset(&s_in4, 0, sizeof(s_in4));
221   s_in4.sin_family = AF_INET;
222   s_in4.sin_port = htons(port);
223   s_in4.sin_addr.s_addr = (S_ADDR_TYPE)inet_addr(CCS address);
224   s_ptr = (struct sockaddr *)&s_in4;
225   s_len = sizeof(s_in4);
226   }
227
228 /* If no connection timeout is set, just call connect() without setting a
229 timer, thereby allowing the inbuilt OS timeout to operate. */
230
231 callout_address = string_sprintf("[%s]:%d", address, port);
232 sigalrm_seen = FALSE;
233 if (timeout > 0) ALARM(timeout);
234
235 #ifdef EXIM_SUPPORT_TFO
236 /* TCP Fast Open, if the system has a cookie from a previous call to
237 this peer, can send data in the SYN packet.  The peer can send data
238 before it gets our ACK of its SYN,ACK - the latter is useful for
239 the SMTP banner.  Other (than SMTP) cases of TCP connections can
240 possibly use the data-on-syn, so support that too. */
241
242 if (fastopen_blob && f.tcp_fastopen_ok)
243   {
244 # ifdef MSG_FASTOPEN
245   /* This is a Linux implementation. */
246
247   if ((rc = sendto(sock, fastopen_blob->data, fastopen_blob->len,
248                     MSG_FASTOPEN | MSG_DONTWAIT, s_ptr, s_len)) >= 0)
249         /* seen for with-data, experimental TFO option, with-cookie case */
250         /* seen for with-data, proper TFO opt, with-cookie case */
251     {
252     DEBUG(D_transport|D_v)
253       debug_printf(" TFO mode connection attempt to %s, %lu data\n",
254         address, (unsigned long)fastopen_blob->len);
255     /*XXX also seen on successful TFO, sigh */
256     tcp_out_fastopen = fastopen_blob->len > 0 ?  TFO_ATTEMPTED_DATA : TFO_ATTEMPTED_NODATA;
257     }
258   else switch (errno)
259     {
260     case EINPROGRESS:   /* expected if we had no cookie for peer */
261         /* seen for no-data, proper TFO option, both cookie-request and with-cookie cases */
262         /*  apparently no visibility of the diffference at this point */
263         /* seen for with-data, proper TFO opt, cookie-req */
264         /*   with netwk delay, post-conn tcp_info sees unacked 1 for R, 2 for C; code in smtp_out.c */
265         /* ? older Experimental TFO option behaviour ? */
266       DEBUG(D_transport|D_v) debug_printf(" TFO mode sendto, %s data: EINPROGRESS\n",
267         fastopen_blob->len > 0 ? "with"  : "no");
268       if (!fastopen_blob->data)
269         {
270         tcp_out_fastopen = TFO_ATTEMPTED_NODATA;                /* we tried; unknown if useful yet */
271         rc = 0;
272         }
273       else                                      /* queue unsent data */
274         rc = send(sock, fastopen_blob->data, fastopen_blob->len, 0);
275       break;
276
277     case EOPNOTSUPP:
278       DEBUG(D_transport)
279         debug_printf("Tried TCP Fast Open but apparently not enabled by sysctl\n");
280       goto legacy_connect;
281
282     case EPIPE:
283       DEBUG(D_transport)
284         debug_printf("Tried TCP Fast Open but kernel too old to support it\n");
285       goto legacy_connect;
286     }
287
288 # elif defined(EXIM_TFO_FREEBSD)
289   /* Re: https://people.freebsd.org/~pkelsey/tfo-tools/tfo-client.c */
290
291   if (setsockopt(sock, IPPROTO_TCP, TCP_FASTOPEN, &on, sizeof(on)) < 0)
292     {
293     DEBUG(D_transport)
294       debug_printf("Tried TCP Fast Open but apparently not enabled by sysctl\n");
295     goto legacy_connect;
296     }
297   if ((rc = sendto(sock, fastopen_blob->data, fastopen_blob->len, 0,
298                     s_ptr, s_len)) >= 0)
299     {
300     DEBUG(D_transport|D_v)
301       debug_printf(" TFO mode connection attempt to %s, %lu data\n",
302         address, (unsigned long)fastopen_blob->len);
303     tcp_out_fastopen = fastopen_blob->len > 0 ?  TFO_ATTEMPTED_DATA : TFO_ATTEMPTED_NODATA;
304     }
305
306 # elif defined(EXIM_TFO_CONNECTX)
307   /* MacOS */
308   sa_endpoints_t ends = {
309     .sae_srcif = 0, .sae_srcaddr = NULL, .sae_srcaddrlen = 0,
310     .sae_dstaddr = s_ptr, .sae_dstaddrlen = s_len };
311   struct iovec iov = {
312     .iov_base = fastopen_blob->data, .iov_len = fastopen_blob->len };
313   size_t len;
314
315   if ((rc = connectx(sock, &ends, SAE_ASSOCID_ANY,
316              CONNECT_DATA_IDEMPOTENT, &iov, 1, &len, NULL)) == 0)
317     {
318     DEBUG(D_transport|D_v)
319       debug_printf(" TFO mode connection attempt to %s, %lu data\n",
320         address, (unsigned long)fastopen_blob->len);
321     tcp_out_fastopen = fastopen_blob->len > 0 ?  TFO_ATTEMPTED_DATA : TFO_ATTEMPTED_NODATA;
322
323     if (len != fastopen_blob->len)
324       DEBUG(D_transport|D_v)
325         debug_printf(" only queued %lu data!\n", (unsigned long)len);
326     }
327   else if (errno == EINPROGRESS)
328     {
329     DEBUG(D_transport|D_v) debug_printf(" TFO mode connectx, %s data: EINPROGRESS\n",
330       fastopen_blob->len > 0 ? "with"  : "no");
331     if (!fastopen_blob->data)
332       {
333       tcp_out_fastopen = TFO_ATTEMPTED_NODATA;          /* we tried; unknown if useful yet */
334       rc = 0;
335       }
336     else        /* assume that no data was queued; block in send */
337       rc = send(sock, fastopen_blob->data, fastopen_blob->len, 0);
338     }
339 # endif
340   }
341 else
342 #endif  /*EXIM_SUPPORT_TFO*/
343   {
344 #if defined(EXIM_SUPPORT_TFO) && !defined(EXIM_TFO_CONNECTX)
345 legacy_connect:
346 #endif
347
348   DEBUG(D_transport|D_v) if (fastopen_blob)
349     debug_printf(" non-TFO mode connection attempt to %s, %lu data\n",
350       address, (unsigned long)fastopen_blob->len);
351   if ((rc = connect(sock, s_ptr, s_len)) >= 0)
352     if (  fastopen_blob && fastopen_blob->data && fastopen_blob->len
353        && send(sock, fastopen_blob->data, fastopen_blob->len, 0) < 0)
354         rc = -1;
355   }
356
357 save_errno = errno;
358 ALARM_CLR(0);
359
360 /* There is a testing facility for simulating a connection timeout, as I
361 can't think of any other way of doing this. It converts a connection refused
362 into a timeout if the timeout is set to 999999. */
363
364 if (f.running_in_test_harness  && save_errno == ECONNREFUSED && timeout == 999999)
365   {
366   rc = -1;
367   save_errno = EINTR;
368   sigalrm_seen = TRUE;
369   }
370
371 /* Success */
372
373 if (rc >= 0)
374   return 0;
375
376 /* A failure whose error code is "Interrupted system call" is in fact
377 an externally applied timeout if the signal handler has been run. */
378
379 errno = save_errno == EINTR && sigalrm_seen ? ETIMEDOUT : save_errno;
380 return -1;
381 }
382
383
384
385 /*************************************************
386 *    Create connected socket to remote host      *
387 *************************************************/
388
389 /* Create a socket and connect to host (name or number, ipv6 ok)
390    at one of port-range.
391
392 Arguments:
393   type          SOCK_DGRAM or SOCK_STREAM
394   af            AF_INET6 or AF_INET for the socket type
395   hostname      host name, or ip address (as text)
396   portlo,porthi the remote port range
397   timeout       a timeout
398   connhost      if not NULL, host_item to be filled in with connection details
399   errstr        pointer for allocated string on error
400   fastopen_blob with SOCK_STREAM, if non-null, request TCP Fast Open.
401                 Additionally, optional idempotent early-data to send
402
403 Return:
404   socket fd, or -1 on failure (having allocated an error string)
405 */
406 int
407 ip_connectedsocket(int type, const uschar * hostname, int portlo, int porthi,
408       int timeout, host_item * connhost, uschar ** errstr, const blob * fastopen_blob)
409 {
410 int namelen;
411 host_item shost;
412 int af = 0, fd, fd4 = -1, fd6 = -1;
413
414 shost.next = NULL;
415 shost.address = NULL;
416 shost.port = portlo;
417 shost.mx = -1;
418
419 namelen = Ustrlen(hostname);
420
421 /* Anything enclosed in [] must be an IP address. */
422
423 if (hostname[0] == '[' &&
424     hostname[namelen - 1] == ']')
425   {
426   uschar * host = string_copyn(hostname+1, namelen-2);
427   if (string_is_ip_address(host, NULL) == 0)
428     {
429     *errstr = string_sprintf("malformed IP address \"%s\"", hostname);
430     return -1;
431     }
432   shost.name = shost.address = host;
433   }
434
435 /* Otherwise check for an unadorned IP address */
436
437 else if (string_is_ip_address(hostname, NULL) != 0)
438   shost.name = shost.address = string_copyn(hostname, namelen);
439
440 /* Otherwise lookup IP address(es) from the name */
441
442 else
443   {
444   shost.name = string_copyn(hostname, namelen);
445   if (host_find_byname(&shost, NULL, HOST_FIND_QUALIFY_SINGLE,
446       NULL, FALSE) != HOST_FOUND)
447     {
448     *errstr = string_sprintf("no IP address found for host %s", shost.name);
449     return -1;
450     }
451   }
452
453 /* Try to connect to the server - test each IP till one works */
454
455 for (host_item * h = &shost; h; h = h->next)
456   {
457   fd = Ustrchr(h->address, ':') != 0
458     ? fd6 < 0 ? (fd6 = ip_socket(type, af = AF_INET6)) : fd6
459     : fd4 < 0 ? (fd4 = ip_socket(type, af = AF_INET )) : fd4;
460
461   if (fd < 0)
462     {
463     *errstr = string_sprintf("failed to create socket: %s", strerror(errno));
464     goto bad;
465     }
466
467   for (int port = portlo; port <= porthi; port++)
468     if (ip_connect(fd, af, h->address, port, timeout, fastopen_blob) == 0)
469       {
470       if (fd6 >= 0 && fd != fd6) close(fd6);
471       if (fd4 >= 0 && fd != fd4) close(fd4);
472       if (connhost)
473         {
474         h->port = port;
475         *connhost = *h;
476         connhost->next = NULL;
477         }
478       return fd;
479       }
480   }
481
482 *errstr = string_sprintf("failed to connect to any address for %s: %s",
483   hostname, strerror(errno));
484
485 bad:
486   close(fd4); close(fd6); return -1;
487 }
488
489
490 /*XXX TFO? */
491 int
492 ip_tcpsocket(const uschar * hostport, uschar ** errstr, int tmo,
493   host_item * connhost)
494 {
495 int scan;
496 uschar hostname[256];
497 unsigned int portlow, porthigh;
498
499 /* extract host and port part */
500 scan = sscanf(CS hostport, "%255s %u-%u", hostname, &portlow, &porthigh);
501 if (scan != 3)
502   {
503   if (scan != 2)
504     {
505     *errstr = string_sprintf("invalid socket '%s'", hostport);
506     return -1;
507     }
508   porthigh = portlow;
509   }
510
511 return ip_connectedsocket(SOCK_STREAM, hostname, portlow, porthigh,
512                           tmo, connhost, errstr, NULL);
513 }
514
515 int
516 ip_unixsocket(const uschar * path, uschar ** errstr)
517 {
518 int sock;
519 struct sockaddr_un server;
520
521 if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
522   {
523   *errstr = US"can't open UNIX socket.";
524   return -1;
525   }
526
527 callout_address = string_copy(path);
528 server.sun_family = AF_UNIX;
529 Ustrncpy(US server.sun_path, path, sizeof(server.sun_path)-1);
530 server.sun_path[sizeof(server.sun_path)-1] = '\0';
531 if (connect(sock, (struct sockaddr *) &server, sizeof(server)) < 0)
532   {
533   int err = errno;
534   (void)close(sock);
535   *errstr = string_sprintf("unable to connect to UNIX socket (%s): %s",
536                 path, strerror(err));
537   return -1;
538   }
539 return sock;
540 }
541
542 /* spec is either an absolute path (with a leading /), or
543 a host (name or IP) and port (whitespace-separated).
544 The port can be a range, dash-separated, or a single number.
545
546 For a TCP socket, optionally fill in a  host_item.
547 */
548 int
549 ip_streamsocket(const uschar * spec, uschar ** errstr, int tmo,
550   host_item * connhost)
551 {
552 return *spec == '/'
553   ? ip_unixsocket(spec, errstr) : ip_tcpsocket(spec, errstr, tmo, connhost);
554 }
555
556 /*************************************************
557 *         Set keepalive on a socket              *
558 *************************************************/
559
560 /* Can be called for both incoming and outgoing sockets.
561
562 Arguments:
563   sock       the socket
564   address    the remote host address, for failure logging
565   torf       true for outgoing connection, false for incoming
566
567 Returns:     nothing
568 */
569
570 void
571 ip_keepalive(int sock, const uschar *address, BOOL torf)
572 {
573 int fodder = 1;
574 if (setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE,
575     US (&fodder), sizeof(fodder)) != 0)
576   log_write(0, LOG_MAIN, "setsockopt(SO_KEEPALIVE) on connection %s %s "
577     "failed: %s", torf? "to":"from", address, strerror(errno));
578 }
579
580
581
582 /*************************************************
583 *         Receive from a socket with timeout     *
584 *************************************************/
585
586 /*
587 Arguments:
588   fd          the file descriptor
589   timelimit   the timeout endpoint, seconds-since-epoch
590 Returns:      TRUE => ready for i/o
591               FALSE => timed out, or other error
592 */
593 BOOL
594 fd_ready(int fd, time_t timelimit)
595 {
596 int rc, time_left = timelimit - time(NULL);
597
598 if (time_left <= 0)
599   {
600   errno = ETIMEDOUT;
601   return FALSE;
602   }
603 /* Wait until the socket is ready */
604
605 do
606   {
607   /*DEBUG(D_transport) debug_printf("waiting for data on fd\n");*/
608   rc = poll_one_fd(fd, POLLIN, time_left * 1000);
609
610   /* If some interrupt arrived, just retry. We presume this to be rare,
611   but it can happen (e.g. the SIGUSR1 signal sent by exiwhat causes
612   select() to exit).
613
614   Aug 2004: Somebody set up a cron job that ran exiwhat every 2 minutes, making
615   the interrupt not at all rare. Since the timeout is typically more than 2
616   minutes, the effect was to block the timeout completely. To prevent this
617   happening again, we do an explicit time test and adjust the timeout
618   accordingly */
619
620   if (rc < 0 && errno == EINTR)
621     {
622     DEBUG(D_transport) debug_printf("EINTR while waiting for socket data\n");
623
624     /* Watch out, 'continue' jumps to the condition, not to the loops top */
625     if ((time_left = timelimit - time(NULL)) > 0) continue;
626     }
627
628   if (rc <= 0)
629     {
630     errno = ETIMEDOUT;
631     return FALSE;
632     }
633
634   /* Checking the FD_ISSET is not enough, if we're interrupted, the
635   select_inset may still contain the 'input'. */
636   }
637 while (rc < 0);
638 return TRUE;
639 }
640
641 /* The timeout is implemented using select(), and we loop to cover select()
642 getting interrupted, and the possibility of select() returning with a positive
643 result but no ready descriptor. Is this in fact possible?
644
645 Arguments:
646   cctx        the connection context (socket fd, possibly TLS context)
647   buffer      to read into
648   bufsize     the buffer size
649   timelimit   the timeout endpoint, seconds-since-epoch
650
651 Returns:      > 0 => that much data read
652               <= 0 on error or EOF; errno set - zero for EOF
653 */
654
655 int
656 ip_recv(client_conn_ctx * cctx, uschar * buffer, int buffsize, time_t timelimit)
657 {
658 int rc;
659
660 if (!fd_ready(cctx->sock, timelimit))
661   return -1;
662
663 /* The socket is ready, read from it (via TLS if it's active). On EOF (i.e.
664 close down of the connection), set errno to zero; otherwise leave it alone. */
665
666 #ifndef DISABLE_TLS
667 if (cctx->tls_ctx)                                      /* client TLS */
668   rc = tls_read(cctx->tls_ctx, buffer, buffsize);
669 else if (tls_in.active.sock == cctx->sock)              /* server TLS */
670   rc = tls_read(NULL, buffer, buffsize);
671 else
672 #endif
673   rc = recv(cctx->sock, buffer, buffsize, 0);
674
675 if (rc > 0) return rc;
676 if (rc == 0) errno = 0;
677 return -1;
678 }
679
680
681
682
683 /*************************************************
684 *    Lookup address family of potential socket   *
685 *************************************************/
686
687 /* Given a file-descriptor, check to see if it's a socket and, if so,
688 return the address family; detects IPv4 vs IPv6.  If not a socket then
689 return -1.
690
691 The value 0 is typically AF_UNSPEC, which should not be seen on a connected
692 fd.  If the return is -1, the errno will be from getsockname(); probably
693 ENOTSOCK or ECONNRESET.
694
695 Arguments:     socket-or-not fd
696 Returns:       address family or -1
697 */
698
699 int
700 ip_get_address_family(int fd)
701 {
702 struct sockaddr_storage ss;
703 socklen_t sslen = sizeof(ss);
704
705 if (getsockname(fd, (struct sockaddr *) &ss, &sslen) < 0)
706   return -1;
707
708 return (int) ss.ss_family;
709 }
710
711
712
713
714 /*************************************************
715 *       Lookup DSCP settings for a socket        *
716 *************************************************/
717
718 struct dscp_name_tableentry {
719   const uschar *name;
720   int value;
721 };
722 /* Keep both of these tables sorted! */
723 static struct dscp_name_tableentry dscp_table[] = {
724 #ifdef IPTOS_DSCP_AF11
725     { CUS"af11", IPTOS_DSCP_AF11 },
726     { CUS"af12", IPTOS_DSCP_AF12 },
727     { CUS"af13", IPTOS_DSCP_AF13 },
728     { CUS"af21", IPTOS_DSCP_AF21 },
729     { CUS"af22", IPTOS_DSCP_AF22 },
730     { CUS"af23", IPTOS_DSCP_AF23 },
731     { CUS"af31", IPTOS_DSCP_AF31 },
732     { CUS"af32", IPTOS_DSCP_AF32 },
733     { CUS"af33", IPTOS_DSCP_AF33 },
734     { CUS"af41", IPTOS_DSCP_AF41 },
735     { CUS"af42", IPTOS_DSCP_AF42 },
736     { CUS"af43", IPTOS_DSCP_AF43 },
737     { CUS"ef", IPTOS_DSCP_EF },
738 #endif
739 #ifdef IPTOS_LOWCOST
740     { CUS"lowcost", IPTOS_LOWCOST },
741 #endif
742     { CUS"lowdelay", IPTOS_LOWDELAY },
743 #ifdef IPTOS_MINCOST
744     { CUS"mincost", IPTOS_MINCOST },
745 #endif
746     { CUS"reliability", IPTOS_RELIABILITY },
747     { CUS"throughput", IPTOS_THROUGHPUT }
748 };
749 static int dscp_table_size =
750   sizeof(dscp_table) / sizeof(struct dscp_name_tableentry);
751
752 /* DSCP values change by protocol family, and so do the options used for
753 setsockopt(); this utility does all the lookups.  It takes an unexpanded
754 option string, expands it, strips off affix whitespace, then checks if it's
755 a number.  If all of what's left is a number, then that's how the option will
756 be parsed and success/failure is a range check.  If it's not all a number,
757 then it must be a supported keyword.
758
759 Arguments:
760   dscp_name   a string, so far unvalidated
761   af          address_family in use
762   level       setsockopt level to use
763   optname     setsockopt name to use
764   dscp_value  value for dscp_name
765
766 Returns: TRUE if okay to setsockopt(), else FALSE
767
768 *level and *optname may be set even if FALSE is returned
769 */
770
771 BOOL
772 dscp_lookup(const uschar *dscp_name, int af,
773     int *level, int *optname, int *dscp_value)
774 {
775 uschar *dscp_lookup, *p;
776 int first, last;
777 long rawlong;
778
779 if (af == AF_INET)
780   {
781   *level = IPPROTO_IP;
782   *optname = IP_TOS;
783   }
784 #if HAVE_IPV6 && defined(IPV6_TCLASS)
785 else if (af == AF_INET6)
786   {
787   *level = IPPROTO_IPV6;
788   *optname = IPV6_TCLASS;
789   }
790 #endif
791 else
792   {
793   DEBUG(D_transport)
794     debug_printf("Unhandled address family %d in dscp_lookup()\n", af);
795   return FALSE;
796   }
797 if (!dscp_name)
798   {
799   DEBUG(D_transport)
800     debug_printf("[empty DSCP]\n");
801   return FALSE;
802   }
803 dscp_lookup = expand_string(US dscp_name);
804 if (dscp_lookup == NULL || *dscp_lookup == '\0')
805   return FALSE;
806
807 p = dscp_lookup + Ustrlen(dscp_lookup) - 1;
808 while (isspace(*p)) *p-- = '\0';
809 while (isspace(*dscp_lookup) && dscp_lookup < p) dscp_lookup++;
810 if (*dscp_lookup == '\0')
811   return FALSE;
812
813 rawlong = Ustrtol(dscp_lookup, &p, 0);
814 if (p != dscp_lookup && *p == '\0')
815   {
816   /* We have six bits available, which will end up shifted to fit in 0xFC mask.
817   RFC 2597 defines the values unshifted. */
818   if (rawlong < 0 || rawlong > 0x3F)
819     {
820     DEBUG(D_transport)
821       debug_printf("DSCP value %ld out of range, ignored.\n", rawlong);
822     return FALSE;
823     }
824   *dscp_value = rawlong << 2;
825   return TRUE;
826   }
827
828 first = 0;
829 last = dscp_table_size;
830 while (last > first)
831   {
832   int middle = (first + last)/2;
833   int c = Ustrcmp(dscp_lookup, dscp_table[middle].name);
834   if (c == 0)
835     {
836     *dscp_value = dscp_table[middle].value;
837     return TRUE;
838     }
839   else if (c > 0)
840     first = middle + 1;
841   else
842     last = middle;
843   }
844 return FALSE;
845 }
846
847 void
848 dscp_list_to_stream(FILE *stream)
849 {
850 for (int i = 0; i < dscp_table_size; ++i)
851   fprintf(stream, "%s\n", dscp_table[i].name);
852 }
853
854
855 /* End of ip.c */
856 /* vi: aw ai sw=2
857 */