f252377b7ed88a03318c75c48e4c6c075a52f8bd
[jra/samba/.git] / source / lib / util_sock.c
1 /*
2    Unix SMB/CIFS implementation.
3    Samba utility functions
4    Copyright (C) Andrew Tridgell 1992-1998
5    Copyright (C) Tim Potter      2000-2001
6    Copyright (C) Jeremy Allison  1992-2007
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23
24 /****************************************************************************
25  Return true if a string could be an IPv4 address.
26 ****************************************************************************/
27
28 bool is_ipaddress_v4(const char *str)
29 {
30         int ret = -1;
31         struct in_addr dest;
32
33         ret = inet_pton(AF_INET, str, &dest);
34         if (ret > 0) {
35                 return true;
36         }
37         return false;
38 }
39
40 /****************************************************************************
41  Return true if a string could be an IPv4 or IPv6 address.
42 ****************************************************************************/
43
44 bool is_ipaddress(const char *str)
45 {
46         int ret = -1;
47
48 #if defined(HAVE_IPV6)
49         if (strchr_m(str, ':')) {
50                 char addr[INET6_ADDRSTRLEN];
51                 struct in6_addr dest6;
52                 const char *sp = str;
53                 char *p = strchr_m(str, '%');
54
55                 /*
56                  * Cope with link-local.
57                  * This is IP:v6:addr%ifname.
58                  */
59
60                 if (p && (p > str) && (if_nametoindex(p+1) != 0)) {
61                         strlcpy(addr, str,
62                                 MIN(PTR_DIFF(p,str)+1,
63                                         sizeof(addr)));
64                         sp = addr;
65                 }
66                 ret = inet_pton(AF_INET6, sp, &dest6);
67                 if (ret > 0) {
68                         return true;
69                 }
70         }
71 #endif
72         return is_ipaddress_v4(str);
73 }
74
75 /****************************************************************************
76  Is a sockaddr_storage a broadcast address ?
77 ****************************************************************************/
78
79 bool is_broadcast_addr(const struct sockaddr_storage *pss)
80 {
81 #if defined(HAVE_IPV6)
82         if (pss->ss_family == AF_INET6) {
83                 const struct in6_addr *sin6 =
84                         &((const struct sockaddr_in6 *)pss)->sin6_addr;
85                 return IN6_IS_ADDR_MULTICAST(sin6);
86         }
87 #endif
88         if (pss->ss_family == AF_INET) {
89                 uint32_t addr =
90                 ntohl(((const struct sockaddr_in *)pss)->sin_addr.s_addr);
91                 return addr == INADDR_BROADCAST;
92         }
93         return false;
94 }
95
96 /*******************************************************************
97  Wrap getaddrinfo...
98 ******************************************************************/
99
100 static bool interpret_string_addr_internal(struct addrinfo **ppres,
101                                         const char *str, int flags)
102 {
103         int ret;
104         struct addrinfo hints;
105
106         memset(&hints, '\0', sizeof(hints));
107         /* By default make sure it supports TCP. */
108         hints.ai_socktype = SOCK_STREAM;
109         hints.ai_flags = flags;
110
111         /* Linux man page on getaddinfo() says port will be
112            uninitialized when service string in NULL */
113
114         ret = getaddrinfo(str, NULL,
115                         &hints,
116                         ppres);
117
118         if (ret) {
119                 DEBUG(3,("interpret_string_addr_internal: getaddrinfo failed "
120                         "for name %s [%s]\n",
121                         str,
122                         gai_strerror(ret) ));
123                 return false;
124         }
125         return true;
126 }
127
128 /****************************************************************************
129  Interpret an internet address or name into an IP address in 4 byte form.
130  RETURNS IN NETWORK BYTE ORDER (big endian).
131 ****************************************************************************/
132
133 uint32 interpret_addr(const char *str)
134 {
135         uint32 ret;
136
137         /* If it's in the form of an IP address then
138          * get the lib to interpret it */
139         if (is_ipaddress_v4(str)) {
140                 struct in_addr dest;
141
142                 if (inet_pton(AF_INET, str, &dest) <= 0) {
143                         /* Error - this shouldn't happen ! */
144                         DEBUG(0,("interpret_addr: inet_pton failed "
145                                 "host %s\n",
146                                 str));
147                         return 0;
148                 }
149                 ret = dest.s_addr; /* NETWORK BYTE ORDER ! */
150         } else {
151                 /* Otherwise assume it's a network name of some sort and use
152                         getadddrinfo. */
153                 struct addrinfo *res = NULL;
154                 struct addrinfo *res_list = NULL;
155                 if (!interpret_string_addr_internal(&res_list,
156                                         str,
157                                         AI_ADDRCONFIG)) {
158                         DEBUG(3,("interpret_addr: Unknown host. %s\n",str));
159                         return 0;
160                 }
161
162                 /* Find the first IPv4 address. */
163                 for (res = res_list; res; res = res->ai_next) {
164                         if (res->ai_family != AF_INET) {
165                                 continue;
166                         }
167                         if (res->ai_addr == NULL) {
168                                 continue;
169                         }
170                         break;
171                 }
172                 if(res == NULL) {
173                         DEBUG(3,("interpret_addr: host address is "
174                                 "invalid for host %s\n",str));
175                         if (res_list) {
176                                 freeaddrinfo(res_list);
177                         }
178                         return 0;
179                 }
180                 putip((char *)&ret,
181                         &((struct sockaddr_in *)res->ai_addr)->sin_addr.s_addr);
182                 if (res_list) {
183                         freeaddrinfo(res_list);
184                 }
185         }
186
187         /* This is so bogus - all callers need fixing... JRA. */
188         if (ret == (uint32)-1) {
189                 return 0;
190         }
191
192         return ret;
193 }
194
195 /*******************************************************************
196  A convenient addition to interpret_addr().
197 ******************************************************************/
198
199 struct in_addr *interpret_addr2(struct in_addr *ip, const char *str)
200 {
201         uint32 a = interpret_addr(str);
202         ip->s_addr = a;
203         return ip;
204 }
205
206 /*******************************************************************
207  Map a text hostname or IP address (IPv4 or IPv6) into a
208  struct sockaddr_storage.
209 ******************************************************************/
210
211 bool interpret_string_addr(struct sockaddr_storage *pss,
212                 const char *str,
213                 int flags)
214 {
215         char addr[INET6_ADDRSTRLEN];
216         struct addrinfo *res = NULL;
217 #if defined(HAVE_IPV6)
218         unsigned int scope_id = 0;
219
220         if (strchr_m(str, ':')) {
221                 char *p = strchr_m(str, '%');
222
223                 /*
224                  * Cope with link-local.
225                  * This is IP:v6:addr%ifname.
226                  */
227
228                 if (p && (p > str) && ((scope_id = if_nametoindex(p+1)) != 0)) {
229                         strlcpy(addr, str,
230                                 MIN(PTR_DIFF(p,str)+1,
231                                         sizeof(addr)));
232                         str = addr;
233                 }
234         }
235 #endif
236
237         zero_addr(pss);
238
239         if (!interpret_string_addr_internal(&res, str, flags|AI_ADDRCONFIG)) {
240                 return false;
241         }
242         if (!res) {
243                 return false;
244         }
245         /* Copy the first sockaddr. */
246         memcpy(pss, res->ai_addr, res->ai_addrlen);
247
248 #if defined(HAVE_IPV6)
249         if (pss->ss_family == AF_INET6 && scope_id) {
250                 struct sockaddr_in6 *ps6 = (struct sockaddr_in6 *)pss;
251                 if (IN6_IS_ADDR_LINKLOCAL(&ps6->sin6_addr) &&
252                                 ps6->sin6_scope_id == 0) {
253                         ps6->sin6_scope_id = scope_id;
254                 }
255         }
256 #endif
257
258         freeaddrinfo(res);
259         return true;
260 }
261
262 /*******************************************************************
263  Check if an IPv7 is 127.0.0.1
264 ******************************************************************/
265
266 bool is_loopback_ip_v4(struct in_addr ip)
267 {
268         struct in_addr a;
269         a.s_addr = htonl(INADDR_LOOPBACK);
270         return(ip.s_addr == a.s_addr);
271 }
272
273 /*******************************************************************
274  Check if a struct sockaddr_storage is the loopback address.
275 ******************************************************************/
276
277 bool is_loopback_addr(const struct sockaddr_storage *pss)
278 {
279 #if defined(HAVE_IPV6)
280         if (pss->ss_family == AF_INET6) {
281                 struct in6_addr *pin6 =
282                         &((struct sockaddr_in6 *)pss)->sin6_addr;
283                 return IN6_IS_ADDR_LOOPBACK(pin6);
284         }
285 #endif
286         if (pss->ss_family == AF_INET) {
287                 struct in_addr *pin = &((struct sockaddr_in *)pss)->sin_addr;
288                 return is_loopback_ip_v4(*pin);
289         }
290         return false;
291 }
292
293 /*******************************************************************
294  Check if an IPv4 is 0.0.0.0.
295 ******************************************************************/
296
297 bool is_zero_ip_v4(struct in_addr ip)
298 {
299         uint32 a;
300         putip((char *)&a,(char *)&ip);
301         return(a == 0);
302 }
303
304 /*******************************************************************
305  Check if a struct sockaddr_storage has an unspecified address.
306 ******************************************************************/
307
308 bool is_zero_addr(const struct sockaddr_storage *pss)
309 {
310 #if defined(HAVE_IPV6)
311         if (pss->ss_family == AF_INET6) {
312                 struct in6_addr *pin6 =
313                         &((struct sockaddr_in6 *)pss)->sin6_addr;
314                 return IN6_IS_ADDR_UNSPECIFIED(pin6);
315         }
316 #endif
317         if (pss->ss_family == AF_INET) {
318                 struct in_addr *pin = &((struct sockaddr_in *)pss)->sin_addr;
319                 return is_zero_ip_v4(*pin);
320         }
321         return false;
322 }
323
324 /*******************************************************************
325  Set an IP to 0.0.0.0.
326 ******************************************************************/
327
328 void zero_ip_v4(struct in_addr *ip)
329 {
330         memset(ip, '\0', sizeof(struct in_addr));
331 }
332
333 /*******************************************************************
334  Set an address to INADDR_ANY.
335 ******************************************************************/
336
337 void zero_addr(struct sockaddr_storage *pss)
338 {
339         memset(pss, '\0', sizeof(*pss));
340         /* Ensure we're at least a valid sockaddr-storage. */
341         pss->ss_family = AF_INET;
342 }
343
344 /*******************************************************************
345  Are two IPs on the same subnet - IPv4 version ?
346 ********************************************************************/
347
348 bool same_net_v4(struct in_addr ip1,struct in_addr ip2,struct in_addr mask)
349 {
350         uint32 net1,net2,nmask;
351
352         nmask = ntohl(mask.s_addr);
353         net1  = ntohl(ip1.s_addr);
354         net2  = ntohl(ip2.s_addr);
355
356         return((net1 & nmask) == (net2 & nmask));
357 }
358
359 /*******************************************************************
360  Convert an IPv4 struct in_addr to a struct sockaddr_storage.
361 ********************************************************************/
362
363 void in_addr_to_sockaddr_storage(struct sockaddr_storage *ss,
364                 struct in_addr ip)
365 {
366         struct sockaddr_in *sa = (struct sockaddr_in *)ss;
367         memset(ss, '\0', sizeof(*ss));
368         ss->ss_family = AF_INET;
369         sa->sin_addr = ip;
370 }
371
372 #if defined(HAVE_IPV6)
373 /*******************************************************************
374  Convert an IPv6 struct in_addr to a struct sockaddr_storage.
375 ********************************************************************/
376
377  void in6_addr_to_sockaddr_storage(struct sockaddr_storage *ss,
378                 struct in6_addr ip)
379 {
380         struct sockaddr_in6 *sa = (struct sockaddr_in6 *)ss;
381         memset(ss, '\0', sizeof(*ss));
382         ss->ss_family = AF_INET6;
383         sa->sin6_addr = ip;
384 }
385 #endif
386
387 /*******************************************************************
388  Are two IPs on the same subnet?
389 ********************************************************************/
390
391 bool same_net(const struct sockaddr_storage *ip1,
392                 const struct sockaddr_storage *ip2,
393                 const struct sockaddr_storage *mask)
394 {
395         if (ip1->ss_family != ip2->ss_family) {
396                 /* Never on the same net. */
397                 return false;
398         }
399
400 #if defined(HAVE_IPV6)
401         if (ip1->ss_family == AF_INET6) {
402                 struct sockaddr_in6 ip1_6 = *(struct sockaddr_in6 *)ip1;
403                 struct sockaddr_in6 ip2_6 = *(struct sockaddr_in6 *)ip2;
404                 struct sockaddr_in6 mask_6 = *(struct sockaddr_in6 *)mask;
405                 char *p1 = (char *)&ip1_6.sin6_addr;
406                 char *p2 = (char *)&ip2_6.sin6_addr;
407                 char *m = (char *)&mask_6.sin6_addr;
408                 int i;
409
410                 for (i = 0; i < sizeof(struct in6_addr); i++) {
411                         *p1++ &= *m;
412                         *p2++ &= *m;
413                         m++;
414                 }
415                 return (memcmp(&ip1_6.sin6_addr,
416                                 &ip2_6.sin6_addr,
417                                 sizeof(struct in6_addr)) == 0);
418         }
419 #endif
420         if (ip1->ss_family == AF_INET) {
421                 return same_net_v4(((const struct sockaddr_in *)ip1)->sin_addr,
422                                 ((const struct sockaddr_in *)ip2)->sin_addr,
423                                 ((const struct sockaddr_in *)mask)->sin_addr);
424         }
425         return false;
426 }
427
428 /*******************************************************************
429  Are two sockaddr_storage's the same family and address ? Ignore port etc.
430 ********************************************************************/
431
432 bool addr_equal(const struct sockaddr_storage *ip1,
433                 const struct sockaddr_storage *ip2)
434 {
435         if (ip1->ss_family != ip2->ss_family) {
436                 /* Never the same. */
437                 return false;
438         }
439
440 #if defined(HAVE_IPV6)
441         if (ip1->ss_family == AF_INET6) {
442                 return (memcmp(&((const struct sockaddr_in6 *)ip1)->sin6_addr,
443                                 &((const struct sockaddr_in6 *)ip2)->sin6_addr,
444                                 sizeof(struct in6_addr)) == 0);
445         }
446 #endif
447         if (ip1->ss_family == AF_INET) {
448                 return (memcmp(&((const struct sockaddr_in *)ip1)->sin_addr,
449                                 &((const struct sockaddr_in *)ip2)->sin_addr,
450                                 sizeof(struct in_addr)) == 0);
451         }
452         return false;
453 }
454
455 /****************************************************************************
456  Is an IP address the INADDR_ANY or in6addr_any value ?
457 ****************************************************************************/
458
459 bool is_address_any(const struct sockaddr_storage *psa)
460 {
461 #if defined(HAVE_IPV6)
462         if (psa->ss_family == AF_INET6) {
463                 struct sockaddr_in6 *si6 = (struct sockaddr_in6 *)psa;
464                 if (memcmp(&in6addr_any,
465                                 &si6->sin6_addr,
466                                 sizeof(in6addr_any)) == 0) {
467                         return true;
468                 }
469                 return false;
470         }
471 #endif
472         if (psa->ss_family == AF_INET) {
473                 struct sockaddr_in *si = (struct sockaddr_in *)psa;
474                 if (si->sin_addr.s_addr == INADDR_ANY) {
475                         return true;
476                 }
477                 return false;
478         }
479         return false;
480 }
481
482 /****************************************************************************
483  Get a port number in host byte order from a sockaddr_storage.
484 ****************************************************************************/
485
486 uint16_t get_sockaddr_port(const struct sockaddr_storage *pss)
487 {
488         uint16_t port = 0;
489
490         if (pss->ss_family != AF_INET) {
491 #if defined(HAVE_IPV6)
492                 /* IPv6 */
493                 const struct sockaddr_in6 *sa6 =
494                         (const struct sockaddr_in6 *)pss;
495                 port = ntohs(sa6->sin6_port);
496 #endif
497         } else {
498                 const struct sockaddr_in *sa =
499                         (const struct sockaddr_in *)pss;
500                 port = ntohs(sa->sin_port);
501         }
502         return port;
503 }
504
505 /****************************************************************************
506  Print out an IPv4 or IPv6 address from a struct sockaddr_storage.
507 ****************************************************************************/
508
509 static char *print_sockaddr_len(char *dest,
510                         size_t destlen,
511                         const struct sockaddr_storage *psa,
512                         socklen_t psalen)
513 {
514         if (destlen > 0) {
515                 dest[0] = '\0';
516         }
517         (void)sys_getnameinfo((const struct sockaddr *)psa,
518                         psalen,
519                         dest, destlen,
520                         NULL, 0,
521                         NI_NUMERICHOST);
522         return dest;
523 }
524
525 /****************************************************************************
526  Print out an IPv4 or IPv6 address from a struct sockaddr_storage.
527 ****************************************************************************/
528
529 char *print_sockaddr(char *dest,
530                         size_t destlen,
531                         const struct sockaddr_storage *psa)
532 {
533         return print_sockaddr_len(dest, destlen, psa,
534                         sizeof(struct sockaddr_storage));
535 }
536
537 /****************************************************************************
538  Print out a canonical IPv4 or IPv6 address from a struct sockaddr_storage.
539 ****************************************************************************/
540
541 char *print_canonical_sockaddr(TALLOC_CTX *ctx,
542                         const struct sockaddr_storage *pss)
543 {
544         char addr[INET6_ADDRSTRLEN];
545         char *dest = NULL;
546         int ret;
547
548         /* Linux getnameinfo() man pages says port is unitialized if
549            service name is NULL. */
550
551         ret = sys_getnameinfo((const struct sockaddr *)pss,
552                         sizeof(struct sockaddr_storage),
553                         addr, sizeof(addr),
554                         NULL, 0,
555                         NI_NUMERICHOST);
556         if (ret != 0) {
557                 return NULL;
558         }
559
560         if (pss->ss_family != AF_INET) {
561 #if defined(HAVE_IPV6)
562                 dest = talloc_asprintf(ctx, "[%s]", addr);
563 #else
564                 return NULL;
565 #endif
566         } else {
567                 dest = talloc_asprintf(ctx, "%s", addr);
568         }
569         
570         return dest;
571 }
572
573 /****************************************************************************
574  Return the string of an IP address (IPv4 or IPv6).
575 ****************************************************************************/
576
577 static const char *get_socket_addr(int fd, char *addr_buf, size_t addr_len)
578 {
579         struct sockaddr_storage sa;
580         socklen_t length = sizeof(sa);
581
582         /* Ok, returning a hard coded IPv4 address
583          * is bogus, but it's just as bogus as a
584          * zero IPv6 address. No good choice here.
585          */
586
587         strlcpy(addr_buf, "0.0.0.0", addr_len);
588
589         if (fd == -1) {
590                 return addr_buf;
591         }
592
593         if (getsockname(fd, (struct sockaddr *)&sa, &length) < 0) {
594                 DEBUG(0,("getsockname failed. Error was %s\n",
595                         strerror(errno) ));
596                 return addr_buf;
597         }
598
599         return print_sockaddr_len(addr_buf, addr_len, &sa, length);
600 }
601
602 #if 0
603 /* Not currently used. JRA. */
604 /****************************************************************************
605  Return the port number we've bound to on a socket.
606 ****************************************************************************/
607
608 static int get_socket_port(int fd)
609 {
610         struct sockaddr_storage sa;
611         socklen_t length = sizeof(sa);
612
613         if (fd == -1) {
614                 return -1;
615         }
616
617         if (getsockname(fd, (struct sockaddr *)&sa, &length) < 0) {
618                 DEBUG(0,("getpeername failed. Error was %s\n",
619                         strerror(errno) ));
620                 return -1;
621         }
622
623 #if defined(HAVE_IPV6)
624         if (sa.ss_family == AF_INET6) {
625                 return ntohs(((struct sockaddr_in6 *)&sa)->sin6_port);
626         }
627 #endif
628         if (sa.ss_family == AF_INET) {
629                 return ntohs(((struct sockaddr_in *)&sa)->sin_port);
630         }
631         return -1;
632 }
633 #endif
634
635 void set_sockaddr_port(struct sockaddr_storage *psa, uint16 port)
636 {
637 #if defined(HAVE_IPV6)
638         if (psa->ss_family == AF_INET6) {
639                 ((struct sockaddr_in6 *)psa)->sin6_port = htons(port);
640         }
641 #endif
642         if (psa->ss_family == AF_INET) {
643                 ((struct sockaddr_in *)psa)->sin_port = htons(port);
644         }
645 }
646
647 const char *client_name(int fd)
648 {
649         return get_peer_name(fd,false);
650 }
651
652 const char *client_addr(int fd, char *addr, size_t addrlen)
653 {
654         return get_peer_addr(fd,addr,addrlen);
655 }
656
657 const char *client_socket_addr(int fd, char *addr, size_t addr_len)
658 {
659         return get_socket_addr(fd, addr, addr_len);
660 }
661
662 #if 0
663 /* Not currently used. JRA. */
664 int client_socket_port(int fd)
665 {
666         return get_socket_port(fd);
667 }
668 #endif
669
670 /****************************************************************************
671  Accessor functions to make thread-safe code easier later...
672 ****************************************************************************/
673
674 void set_smb_read_error(enum smb_read_errors *pre,
675                         enum smb_read_errors newerr)
676 {
677         if (pre) {
678                 *pre = newerr;
679         }
680 }
681
682 void cond_set_smb_read_error(enum smb_read_errors *pre,
683                         enum smb_read_errors newerr)
684 {
685         if (pre && *pre == SMB_READ_OK) {
686                 *pre = newerr;
687         }
688 }
689
690 /****************************************************************************
691  Determine if a file descriptor is in fact a socket.
692 ****************************************************************************/
693
694 bool is_a_socket(int fd)
695 {
696         int v;
697         socklen_t l;
698         l = sizeof(int);
699         return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
700 }
701
702 enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
703
704 typedef struct smb_socket_option {
705         const char *name;
706         int level;
707         int option;
708         int value;
709         int opttype;
710 } smb_socket_option;
711
712 static const smb_socket_option socket_options[] = {
713   {"SO_KEEPALIVE", SOL_SOCKET, SO_KEEPALIVE, 0, OPT_BOOL},
714   {"SO_REUSEADDR", SOL_SOCKET, SO_REUSEADDR, 0, OPT_BOOL},
715   {"SO_BROADCAST", SOL_SOCKET, SO_BROADCAST, 0, OPT_BOOL},
716 #ifdef TCP_NODELAY
717   {"TCP_NODELAY", IPPROTO_TCP, TCP_NODELAY, 0, OPT_BOOL},
718 #endif
719 #ifdef TCP_KEEPCNT
720   {"TCP_KEEPCNT", IPPROTO_TCP, TCP_KEEPCNT, 0, OPT_INT},
721 #endif
722 #ifdef TCP_KEEPIDLE
723   {"TCP_KEEPIDLE", IPPROTO_TCP, TCP_KEEPIDLE, 0, OPT_INT},
724 #endif
725 #ifdef TCP_KEEPINTVL
726   {"TCP_KEEPINTVL", IPPROTO_TCP, TCP_KEEPINTVL, 0, OPT_INT},
727 #endif
728 #ifdef IPTOS_LOWDELAY
729   {"IPTOS_LOWDELAY", IPPROTO_IP, IP_TOS, IPTOS_LOWDELAY, OPT_ON},
730 #endif
731 #ifdef IPTOS_THROUGHPUT
732   {"IPTOS_THROUGHPUT", IPPROTO_IP, IP_TOS, IPTOS_THROUGHPUT, OPT_ON},
733 #endif
734 #ifdef SO_REUSEPORT
735   {"SO_REUSEPORT", SOL_SOCKET, SO_REUSEPORT, 0, OPT_BOOL},
736 #endif
737 #ifdef SO_SNDBUF
738   {"SO_SNDBUF", SOL_SOCKET, SO_SNDBUF, 0, OPT_INT},
739 #endif
740 #ifdef SO_RCVBUF
741   {"SO_RCVBUF", SOL_SOCKET, SO_RCVBUF, 0, OPT_INT},
742 #endif
743 #ifdef SO_SNDLOWAT
744   {"SO_SNDLOWAT", SOL_SOCKET, SO_SNDLOWAT, 0, OPT_INT},
745 #endif
746 #ifdef SO_RCVLOWAT
747   {"SO_RCVLOWAT", SOL_SOCKET, SO_RCVLOWAT, 0, OPT_INT},
748 #endif
749 #ifdef SO_SNDTIMEO
750   {"SO_SNDTIMEO", SOL_SOCKET, SO_SNDTIMEO, 0, OPT_INT},
751 #endif
752 #ifdef SO_RCVTIMEO
753   {"SO_RCVTIMEO", SOL_SOCKET, SO_RCVTIMEO, 0, OPT_INT},
754 #endif
755 #ifdef TCP_FASTACK
756   {"TCP_FASTACK", IPPROTO_TCP, TCP_FASTACK, 0, OPT_INT},
757 #endif
758   {NULL,0,0,0,0}};
759
760 /****************************************************************************
761  Print socket options.
762 ****************************************************************************/
763
764 static void print_socket_options(int s)
765 {
766         int value;
767         socklen_t vlen = 4;
768         const smb_socket_option *p = &socket_options[0];
769
770         /* wrapped in if statement to prevent streams
771          * leak in SCO Openserver 5.0 */
772         /* reported on samba-technical  --jerry */
773         if ( DEBUGLEVEL >= 5 ) {
774                 for (; p->name != NULL; p++) {
775                         if (getsockopt(s, p->level, p->option,
776                                                 (void *)&value, &vlen) == -1) {
777                                 DEBUG(5,("Could not test socket option %s.\n",
778                                                         p->name));
779                         } else {
780                                 DEBUG(5,("socket option %s = %d\n",
781                                                         p->name,value));
782                         }
783                 }
784         }
785  }
786
787 /****************************************************************************
788  Set user socket options.
789 ****************************************************************************/
790
791 void set_socket_options(int fd, const char *options)
792 {
793         TALLOC_CTX *ctx = talloc_stackframe();
794         char *tok;
795
796         while (next_token_talloc(ctx, &options, &tok," \t,")) {
797                 int ret=0,i;
798                 int value = 1;
799                 char *p;
800                 bool got_value = false;
801
802                 if ((p = strchr_m(tok,'='))) {
803                         *p = 0;
804                         value = atoi(p+1);
805                         got_value = true;
806                 }
807
808                 for (i=0;socket_options[i].name;i++)
809                         if (strequal(socket_options[i].name,tok))
810                                 break;
811
812                 if (!socket_options[i].name) {
813                         DEBUG(0,("Unknown socket option %s\n",tok));
814                         continue;
815                 }
816
817                 switch (socket_options[i].opttype) {
818                 case OPT_BOOL:
819                 case OPT_INT:
820                         ret = setsockopt(fd,socket_options[i].level,
821                                         socket_options[i].option,
822                                         (char *)&value,sizeof(int));
823                         break;
824
825                 case OPT_ON:
826                         if (got_value)
827                                 DEBUG(0,("syntax error - %s "
828                                         "does not take a value\n",tok));
829
830                         {
831                                 int on = socket_options[i].value;
832                                 ret = setsockopt(fd,socket_options[i].level,
833                                         socket_options[i].option,
834                                         (char *)&on,sizeof(int));
835                         }
836                         break;
837                 }
838
839                 if (ret != 0) {
840                         /* be aware that some systems like Solaris return
841                          * EINVAL to a setsockopt() call when the client
842                          * sent a RST previously - no need to worry */
843                         DEBUG(2,("Failed to set socket option %s (Error %s)\n",
844                                 tok, strerror(errno) ));
845                 }
846         }
847
848         TALLOC_FREE(ctx);
849         print_socket_options(fd);
850 }
851
852 /****************************************************************************
853  Read from a socket.
854 ****************************************************************************/
855
856 ssize_t read_udp_v4_socket(int fd,
857                         char *buf,
858                         size_t len,
859                         struct sockaddr_storage *psa)
860 {
861         ssize_t ret;
862         socklen_t socklen = sizeof(*psa);
863         struct sockaddr_in *si = (struct sockaddr_in *)psa;
864
865         memset((char *)psa,'\0',socklen);
866
867         ret = (ssize_t)sys_recvfrom(fd,buf,len,0,
868                         (struct sockaddr *)psa,&socklen);
869         if (ret <= 0) {
870                 /* Don't print a low debug error for a non-blocking socket. */
871                 if (errno == EAGAIN) {
872                         DEBUG(10,("read_udp_v4_socket: returned EAGAIN\n"));
873                 } else {
874                         DEBUG(2,("read_udp_v4_socket: failed. errno=%s\n",
875                                 strerror(errno)));
876                 }
877                 return 0;
878         }
879
880         if (psa->ss_family != AF_INET) {
881                 DEBUG(2,("read_udp_v4_socket: invalid address family %d "
882                         "(not IPv4)\n", (int)psa->ss_family));
883                 return 0;
884         }
885
886         DEBUG(10,("read_udp_v4_socket: ip %s port %d read: %lu\n",
887                         inet_ntoa(si->sin_addr),
888                         si->sin_port,
889                         (unsigned long)ret));
890
891         return ret;
892 }
893
894 /****************************************************************************
895  Read data from a socket with a timout in msec.
896  mincount = if timeout, minimum to read before returning
897  maxcount = number to be read.
898  time_out = timeout in milliseconds
899 ****************************************************************************/
900
901 NTSTATUS read_socket_with_timeout(int fd, char *buf,
902                                   size_t mincnt, size_t maxcnt,
903                                   unsigned int time_out,
904                                   size_t *size_ret)
905 {
906         fd_set fds;
907         int selrtn;
908         ssize_t readret;
909         size_t nread = 0;
910         struct timeval timeout;
911         char addr[INET6_ADDRSTRLEN];
912
913         /* just checking .... */
914         if (maxcnt <= 0)
915                 return NT_STATUS_OK;
916
917         /* Blocking read */
918         if (time_out == 0) {
919                 if (mincnt == 0) {
920                         mincnt = maxcnt;
921                 }
922
923                 while (nread < mincnt) {
924                         readret = sys_read(fd, buf + nread, maxcnt - nread);
925
926                         if (readret == 0) {
927                                 DEBUG(5,("read_socket_with_timeout: "
928                                         "blocking read. EOF from client.\n"));
929                                 return NT_STATUS_END_OF_FILE;
930                         }
931
932                         if (readret == -1) {
933                                 if (fd == get_client_fd()) {
934                                         /* Try and give an error message
935                                          * saying what client failed. */
936                                         DEBUG(0,("read_socket_with_timeout: "
937                                                 "client %s read error = %s.\n",
938                                                 get_peer_addr(fd,addr,sizeof(addr)),
939                                                 strerror(errno) ));
940                                 } else {
941                                         DEBUG(0,("read_socket_with_timeout: "
942                                                 "read error = %s.\n",
943                                                 strerror(errno) ));
944                                 }
945                                 return map_nt_error_from_unix(errno);
946                         }
947                         nread += readret;
948                 }
949                 goto done;
950         }
951
952         /* Most difficult - timeout read */
953         /* If this is ever called on a disk file and
954            mincnt is greater then the filesize then
955            system performance will suffer severely as
956            select always returns true on disk files */
957
958         /* Set initial timeout */
959         timeout.tv_sec = (time_t)(time_out / 1000);
960         timeout.tv_usec = (long)(1000 * (time_out % 1000));
961
962         for (nread=0; nread < mincnt; ) {
963                 FD_ZERO(&fds);
964                 FD_SET(fd,&fds);
965
966                 selrtn = sys_select_intr(fd+1,&fds,NULL,NULL,&timeout);
967
968                 /* Check if error */
969                 if (selrtn == -1) {
970                         /* something is wrong. Maybe the socket is dead? */
971                         if (fd == get_client_fd()) {
972                                 /* Try and give an error message saying
973                                  * what client failed. */
974                                 DEBUG(0,("read_socket_with_timeout: timeout "
975                                 "read for client %s. select error = %s.\n",
976                                 get_peer_addr(fd,addr,sizeof(addr)),
977                                 strerror(errno) ));
978                         } else {
979                                 DEBUG(0,("read_socket_with_timeout: timeout "
980                                 "read. select error = %s.\n",
981                                 strerror(errno) ));
982                         }
983                         return map_nt_error_from_unix(errno);
984                 }
985
986                 /* Did we timeout ? */
987                 if (selrtn == 0) {
988                         DEBUG(10,("read_socket_with_timeout: timeout read. "
989                                 "select timed out.\n"));
990                         return NT_STATUS_IO_TIMEOUT;
991                 }
992
993                 readret = sys_read(fd, buf+nread, maxcnt-nread);
994
995                 if (readret == 0) {
996                         /* we got EOF on the file descriptor */
997                         DEBUG(5,("read_socket_with_timeout: timeout read. "
998                                 "EOF from client.\n"));
999                         return NT_STATUS_END_OF_FILE;
1000                 }
1001
1002                 if (readret == -1) {
1003                         /* the descriptor is probably dead */
1004                         if (fd == get_client_fd()) {
1005                                 /* Try and give an error message
1006                                  * saying what client failed. */
1007                                 DEBUG(0,("read_socket_with_timeout: timeout "
1008                                         "read to client %s. read error = %s.\n",
1009                                         get_peer_addr(fd,addr,sizeof(addr)),
1010                                         strerror(errno) ));
1011                         } else {
1012                                 DEBUG(0,("read_socket_with_timeout: timeout "
1013                                         "read. read error = %s.\n",
1014                                         strerror(errno) ));
1015                         }
1016                         return map_nt_error_from_unix(errno);
1017                 }
1018
1019                 nread += readret;
1020         }
1021
1022  done:
1023         /* Return the number we got */
1024         if (size_ret) {
1025                 *size_ret = nread;
1026         }
1027         return NT_STATUS_OK;
1028 }
1029
1030 /****************************************************************************
1031  Read data from the client, reading exactly N bytes.
1032 ****************************************************************************/
1033
1034 NTSTATUS read_data(int fd, char *buffer, size_t N)
1035 {
1036         return read_socket_with_timeout(fd, buffer, N, N, 0, NULL);
1037 }
1038
1039 /****************************************************************************
1040  Write data to a fd.
1041 ****************************************************************************/
1042
1043 ssize_t write_data(int fd, const char *buffer, size_t N)
1044 {
1045         size_t total=0;
1046         ssize_t ret;
1047         char addr[INET6_ADDRSTRLEN];
1048
1049         while (total < N) {
1050                 ret = sys_write(fd,buffer + total,N - total);
1051
1052                 if (ret == -1) {
1053                         if (fd == get_client_fd()) {
1054                                 /* Try and give an error message saying
1055                                  * what client failed. */
1056                                 DEBUG(0,("write_data: write failure in "
1057                                         "writing to client %s. Error %s\n",
1058                                         get_peer_addr(fd,addr,sizeof(addr)),
1059                                         strerror(errno) ));
1060                         } else {
1061                                 DEBUG(0,("write_data: write failure. "
1062                                         "Error = %s\n", strerror(errno) ));
1063                         }
1064                         return -1;
1065                 }
1066
1067                 if (ret == 0) {
1068                         return total;
1069                 }
1070
1071                 total += ret;
1072         }
1073         return (ssize_t)total;
1074 }
1075
1076 /****************************************************************************
1077  Send a keepalive packet (rfc1002).
1078 ****************************************************************************/
1079
1080 bool send_keepalive(int client)
1081 {
1082         unsigned char buf[4];
1083
1084         buf[0] = SMBkeepalive;
1085         buf[1] = buf[2] = buf[3] = 0;
1086
1087         return(write_data(client,(char *)buf,4) == 4);
1088 }
1089
1090 /****************************************************************************
1091  Read 4 bytes of a smb packet and return the smb length of the packet.
1092  Store the result in the buffer.
1093  This version of the function will return a length of zero on receiving
1094  a keepalive packet.
1095  Timeout is in milliseconds.
1096 ****************************************************************************/
1097
1098 NTSTATUS read_smb_length_return_keepalive(int fd, char *inbuf,
1099                                           unsigned int timeout,
1100                                           size_t *len)
1101 {
1102         int msg_type;
1103         NTSTATUS status;
1104
1105         status = read_socket_with_timeout(fd, inbuf, 4, 4, timeout, NULL);
1106
1107         if (!NT_STATUS_IS_OK(status)) {
1108                 return status;
1109         }
1110
1111         *len = smb_len(inbuf);
1112         msg_type = CVAL(inbuf,0);
1113
1114         if (msg_type == SMBkeepalive) {
1115                 DEBUG(5,("Got keepalive packet\n"));
1116         }
1117
1118         DEBUG(10,("got smb length of %lu\n",(unsigned long)(*len)));
1119
1120         return NT_STATUS_OK;
1121 }
1122
1123 /****************************************************************************
1124  Read 4 bytes of a smb packet and return the smb length of the packet.
1125  Store the result in the buffer. This version of the function will
1126  never return a session keepalive (length of zero).
1127  Timeout is in milliseconds.
1128 ****************************************************************************/
1129
1130 NTSTATUS read_smb_length(int fd, char *inbuf, unsigned int timeout,
1131                          size_t *len)
1132 {
1133         uint8_t msgtype = SMBkeepalive;
1134
1135         while (msgtype == SMBkeepalive) {
1136                 NTSTATUS status;
1137
1138                 status = read_smb_length_return_keepalive(fd, inbuf, timeout,
1139                                                           len);
1140                 if (!NT_STATUS_IS_OK(status)) {
1141                         return status;
1142                 }
1143
1144                 msgtype = CVAL(inbuf, 0);
1145         }
1146
1147         DEBUG(10,("read_smb_length: got smb length of %lu\n",
1148                   (unsigned long)len));
1149
1150         return NT_STATUS_OK;
1151 }
1152
1153 /****************************************************************************
1154  Read an smb from a fd. Note that the buffer *MUST* be of size
1155  BUFFER_SIZE+SAFETY_MARGIN.
1156  The timeout is in milliseconds.
1157  This function will return on receipt of a session keepalive packet.
1158  maxlen is the max number of bytes to return, not including the 4 byte
1159  length. If zero it means BUFFER_SIZE+SAFETY_MARGIN limit.
1160  Doesn't check the MAC on signed packets.
1161 ****************************************************************************/
1162
1163 NTSTATUS receive_smb_raw(int fd, char *buffer, unsigned int timeout,
1164                          size_t maxlen, size_t *p_len)
1165 {
1166         size_t len;
1167         NTSTATUS status;
1168
1169         status = read_smb_length_return_keepalive(fd,buffer,timeout,&len);
1170
1171         if (!NT_STATUS_IS_OK(status)) {
1172                 DEBUG(10, ("receive_smb_raw: %s!\n", nt_errstr(status)));
1173                 return status;
1174         }
1175
1176         /*
1177          * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
1178          * of header. Don't print the error if this fits.... JRA.
1179          */
1180
1181         if (len > (BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
1182                 DEBUG(0,("Invalid packet length! (%lu bytes).\n",
1183                                         (unsigned long)len));
1184                 if (len > BUFFER_SIZE + (SAFETY_MARGIN/2)) {
1185                         return NT_STATUS_INVALID_PARAMETER;
1186                 }
1187         }
1188
1189         if(len > 0) {
1190                 if (maxlen) {
1191                         len = MIN(len,maxlen);
1192                 }
1193
1194                 status = read_socket_with_timeout(
1195                         fd, buffer+4, len, len, timeout, &len);
1196
1197                 if (!NT_STATUS_IS_OK(status)) {
1198                         return status;
1199                 }
1200
1201                 /* not all of samba3 properly checks for packet-termination
1202                  * of strings. This ensures that we don't run off into
1203                  * empty space. */
1204                 SSVAL(buffer+4,len, 0);
1205         }
1206
1207         *p_len = len;
1208         return NT_STATUS_OK;
1209 }
1210
1211 /****************************************************************************
1212  Open a socket of the specified type, port, and address for incoming data.
1213 ****************************************************************************/
1214
1215 int open_socket_in(int type,
1216                 uint16_t port,
1217                 int dlevel,
1218                 const struct sockaddr_storage *psock,
1219                 bool rebind)
1220 {
1221         struct sockaddr_storage sock;
1222         int res;
1223         socklen_t slen = sizeof(struct sockaddr_in);
1224
1225         sock = *psock;
1226
1227 #if defined(HAVE_IPV6)
1228         if (sock.ss_family == AF_INET6) {
1229                 ((struct sockaddr_in6 *)&sock)->sin6_port = htons(port);
1230                 slen = sizeof(struct sockaddr_in6);
1231         }
1232 #endif
1233         if (sock.ss_family == AF_INET) {
1234                 ((struct sockaddr_in *)&sock)->sin_port = htons(port);
1235         }
1236
1237         res = socket(sock.ss_family, type, 0 );
1238         if( res == -1 ) {
1239                 if( DEBUGLVL(0) ) {
1240                         dbgtext( "open_socket_in(): socket() call failed: " );
1241                         dbgtext( "%s\n", strerror( errno ) );
1242                 }
1243                 return -1;
1244         }
1245
1246         /* This block sets/clears the SO_REUSEADDR and possibly SO_REUSEPORT. */
1247         {
1248                 int val = rebind ? 1 : 0;
1249                 if( setsockopt(res,SOL_SOCKET,SO_REUSEADDR,
1250                                         (char *)&val,sizeof(val)) == -1 ) {
1251                         if( DEBUGLVL( dlevel ) ) {
1252                                 dbgtext( "open_socket_in(): setsockopt: " );
1253                                 dbgtext( "SO_REUSEADDR = %s ",
1254                                                 val?"true":"false" );
1255                                 dbgtext( "on port %d failed ", port );
1256                                 dbgtext( "with error = %s\n", strerror(errno) );
1257                         }
1258                 }
1259 #ifdef SO_REUSEPORT
1260                 if( setsockopt(res,SOL_SOCKET,SO_REUSEPORT,
1261                                         (char *)&val,sizeof(val)) == -1 ) {
1262                         if( DEBUGLVL( dlevel ) ) {
1263                                 dbgtext( "open_socket_in(): setsockopt: ");
1264                                 dbgtext( "SO_REUSEPORT = %s ",
1265                                                 val?"true":"false");
1266                                 dbgtext( "on port %d failed ", port);
1267                                 dbgtext( "with error = %s\n", strerror(errno));
1268                         }
1269                 }
1270 #endif /* SO_REUSEPORT */
1271         }
1272
1273         /* now we've got a socket - we need to bind it */
1274         if (bind(res, (struct sockaddr *)&sock, slen) == -1 ) {
1275                 if( DEBUGLVL(dlevel) && (port == SMB_PORT1 ||
1276                                 port == SMB_PORT2 || port == NMB_PORT) ) {
1277                         char addr[INET6_ADDRSTRLEN];
1278                         print_sockaddr(addr, sizeof(addr),
1279                                         &sock);
1280                         dbgtext( "bind failed on port %d ", port);
1281                         dbgtext( "socket_addr = %s.\n", addr);
1282                         dbgtext( "Error = %s\n", strerror(errno));
1283                 }
1284                 close(res);
1285                 return -1;
1286         }
1287
1288         DEBUG( 10, ( "bind succeeded on port %d\n", port ) );
1289         return( res );
1290  }
1291
1292 /****************************************************************************
1293  Create an outgoing socket. timeout is in milliseconds.
1294 **************************************************************************/
1295
1296 int open_socket_out(int type,
1297                 const struct sockaddr_storage *pss,
1298                 uint16_t port,
1299                 int timeout)
1300 {
1301         char addr[INET6_ADDRSTRLEN];
1302         struct sockaddr_storage sock_out = *pss;
1303         int res,ret;
1304         int connect_loop = 10;
1305         int increment = 10;
1306
1307         /* create a socket to write to */
1308         res = socket(pss->ss_family, type, 0);
1309         if (res == -1) {
1310                 DEBUG(0,("socket error (%s)\n", strerror(errno)));
1311                 return -1;
1312         }
1313
1314         if (type != SOCK_STREAM) {
1315                 return res;
1316         }
1317
1318 #if defined(HAVE_IPV6)
1319         if (pss->ss_family == AF_INET6) {
1320                 struct sockaddr_in6 *psa6 = (struct sockaddr_in6 *)&sock_out;
1321                 psa6->sin6_port = htons(port);
1322                 if (psa6->sin6_scope_id == 0 &&
1323                                 IN6_IS_ADDR_LINKLOCAL(&psa6->sin6_addr)) {
1324                         setup_linklocal_scope_id(&sock_out);
1325                 }
1326         }
1327 #endif
1328         if (pss->ss_family == AF_INET) {
1329                 struct sockaddr_in *psa = (struct sockaddr_in *)&sock_out;
1330                 psa->sin_port = htons(port);
1331         }
1332
1333         /* set it non-blocking */
1334         set_blocking(res,false);
1335
1336         print_sockaddr(addr, sizeof(addr), &sock_out);
1337         DEBUG(3,("Connecting to %s at port %u\n",
1338                                 addr,
1339                                 (unsigned int)port));
1340
1341         /* and connect it to the destination */
1342   connect_again:
1343
1344         ret = sys_connect(res, (struct sockaddr *)&sock_out);
1345
1346         /* Some systems return EAGAIN when they mean EINPROGRESS */
1347         if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
1348                         errno == EAGAIN) && (connect_loop < timeout) ) {
1349                 smb_msleep(connect_loop);
1350                 timeout -= connect_loop;
1351                 connect_loop += increment;
1352                 if (increment < 250) {
1353                         /* After 8 rounds we end up at a max of 255 msec */
1354                         increment *= 1.5;
1355                 }
1356                 goto connect_again;
1357         }
1358
1359         if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
1360                         errno == EAGAIN)) {
1361                 DEBUG(1,("timeout connecting to %s:%u\n",
1362                                         addr,
1363                                         (unsigned int)port));
1364                 close(res);
1365                 return -1;
1366         }
1367
1368 #ifdef EISCONN
1369         if (ret < 0 && errno == EISCONN) {
1370                 errno = 0;
1371                 ret = 0;
1372         }
1373 #endif
1374
1375         if (ret < 0) {
1376                 DEBUG(2,("error connecting to %s:%d (%s)\n",
1377                                 addr,
1378                                 (unsigned int)port,
1379                                 strerror(errno)));
1380                 close(res);
1381                 return -1;
1382         }
1383
1384         /* set it blocking again */
1385         set_blocking(res,true);
1386
1387         return res;
1388 }
1389
1390 /****************************************************************************
1391  Create an outgoing TCP socket to any of the addrs. This is for
1392  simultaneous connects to port 445 and 139 of a host or even a variety
1393  of DC's all of which are equivalent for our purposes.
1394 **************************************************************************/
1395
1396 bool open_any_socket_out(struct sockaddr_storage *addrs, int num_addrs,
1397                          int timeout, int *fd_index, int *fd)
1398 {
1399         int i, resulting_index, res;
1400         int *sockets;
1401         bool good_connect;
1402
1403         fd_set r_fds, wr_fds;
1404         struct timeval tv;
1405         int maxfd;
1406
1407         int connect_loop = 10000; /* 10 milliseconds */
1408
1409         timeout *= 1000;        /* convert to microseconds */
1410
1411         sockets = SMB_MALLOC_ARRAY(int, num_addrs);
1412
1413         if (sockets == NULL)
1414                 return false;
1415
1416         resulting_index = -1;
1417
1418         for (i=0; i<num_addrs; i++)
1419                 sockets[i] = -1;
1420
1421         for (i=0; i<num_addrs; i++) {
1422                 sockets[i] = socket(addrs[i].ss_family, SOCK_STREAM, 0);
1423                 if (sockets[i] < 0)
1424                         goto done;
1425                 set_blocking(sockets[i], false);
1426         }
1427
1428  connect_again:
1429         good_connect = false;
1430
1431         for (i=0; i<num_addrs; i++) {
1432                 const struct sockaddr * a = 
1433                     (const struct sockaddr *)&(addrs[i]);
1434
1435                 if (sockets[i] == -1)
1436                         continue;
1437
1438                 if (sys_connect(sockets[i], a) == 0) {
1439                         /* Rather unlikely as we are non-blocking, but it
1440                          * might actually happen. */
1441                         resulting_index = i;
1442                         goto done;
1443                 }
1444
1445                 if (errno == EINPROGRESS || errno == EALREADY ||
1446 #ifdef EISCONN
1447                         errno == EISCONN ||
1448 #endif
1449                     errno == EAGAIN || errno == EINTR) {
1450                         /* These are the error messages that something is
1451                            progressing. */
1452                         good_connect = true;
1453                 } else if (errno != 0) {
1454                         /* There was a direct error */
1455                         close(sockets[i]);
1456                         sockets[i] = -1;
1457                 }
1458         }
1459
1460         if (!good_connect) {
1461                 /* All of the connect's resulted in real error conditions */
1462                 goto done;
1463         }
1464
1465         /* Lets see if any of the connect attempts succeeded */
1466
1467         maxfd = 0;
1468         FD_ZERO(&wr_fds);
1469         FD_ZERO(&r_fds);
1470
1471         for (i=0; i<num_addrs; i++) {
1472                 if (sockets[i] == -1)
1473                         continue;
1474                 FD_SET(sockets[i], &wr_fds);
1475                 FD_SET(sockets[i], &r_fds);
1476                 if (sockets[i]>maxfd)
1477                         maxfd = sockets[i];
1478         }
1479
1480         tv.tv_sec = 0;
1481         tv.tv_usec = connect_loop;
1482
1483         res = sys_select_intr(maxfd+1, &r_fds, &wr_fds, NULL, &tv);
1484
1485         if (res < 0)
1486                 goto done;
1487
1488         if (res == 0)
1489                 goto next_round;
1490
1491         for (i=0; i<num_addrs; i++) {
1492
1493                 if (sockets[i] == -1)
1494                         continue;
1495
1496                 /* Stevens, Network Programming says that if there's a
1497                  * successful connect, the socket is only writable. Upon an
1498                  * error, it's both readable and writable. */
1499
1500                 if (FD_ISSET(sockets[i], &r_fds) &&
1501                     FD_ISSET(sockets[i], &wr_fds)) {
1502                         /* readable and writable, so it's an error */
1503                         close(sockets[i]);
1504                         sockets[i] = -1;
1505                         continue;
1506                 }
1507
1508                 if (!FD_ISSET(sockets[i], &r_fds) &&
1509                     FD_ISSET(sockets[i], &wr_fds)) {
1510                         /* Only writable, so it's connected */
1511                         resulting_index = i;
1512                         goto done;
1513                 }
1514         }
1515
1516  next_round:
1517
1518         timeout -= connect_loop;
1519         if (timeout <= 0)
1520                 goto done;
1521         connect_loop *= 1.5;
1522         if (connect_loop > timeout)
1523                 connect_loop = timeout;
1524         goto connect_again;
1525
1526  done:
1527         for (i=0; i<num_addrs; i++) {
1528                 if (i == resulting_index)
1529                         continue;
1530                 if (sockets[i] >= 0)
1531                         close(sockets[i]);
1532         }
1533
1534         if (resulting_index >= 0) {
1535                 *fd_index = resulting_index;
1536                 *fd = sockets[*fd_index];
1537                 set_blocking(*fd, true);
1538         }
1539
1540         free(sockets);
1541
1542         return (resulting_index >= 0);
1543 }
1544 /****************************************************************************
1545  Open a connected UDP socket to host on port
1546 **************************************************************************/
1547
1548 int open_udp_socket(const char *host, int port)
1549 {
1550         int type = SOCK_DGRAM;
1551         struct sockaddr_in sock_out;
1552         int res;
1553         struct in_addr addr;
1554
1555         (void)interpret_addr2(&addr, host);
1556
1557         res = socket(PF_INET, type, 0);
1558         if (res == -1) {
1559                 return -1;
1560         }
1561
1562         memset((char *)&sock_out,'\0',sizeof(sock_out));
1563         putip((char *)&sock_out.sin_addr,(char *)&addr);
1564         sock_out.sin_port = htons(port);
1565         sock_out.sin_family = PF_INET;
1566
1567         if (sys_connect(res,(struct sockaddr *)&sock_out)) {
1568                 close(res);
1569                 return -1;
1570         }
1571
1572         return res;
1573 }
1574
1575 /*******************************************************************
1576  Return the IP addr of the remote end of a socket as a string.
1577  Optionally return the struct sockaddr_storage.
1578  ******************************************************************/
1579
1580 static const char *get_peer_addr_internal(int fd,
1581                                 char *addr_buf,
1582                                 size_t addr_buf_len,
1583                                 struct sockaddr_storage *pss,
1584                                 socklen_t *plength)
1585 {
1586         struct sockaddr_storage ss;
1587         socklen_t length = sizeof(ss);
1588
1589         strlcpy(addr_buf,"0.0.0.0",addr_buf_len);
1590
1591         if (fd == -1) {
1592                 return addr_buf;
1593         }
1594
1595         if (pss == NULL) {
1596                 pss = &ss;
1597         }
1598         if (plength == NULL) {
1599                 plength = &length;
1600         }
1601
1602         if (getpeername(fd, (struct sockaddr *)pss, plength) < 0) {
1603                 DEBUG(0,("getpeername failed. Error was %s\n",
1604                                         strerror(errno) ));
1605                 return addr_buf;
1606         }
1607
1608         print_sockaddr_len(addr_buf,
1609                         addr_buf_len,
1610                         pss,
1611                         *plength);
1612         return addr_buf;
1613 }
1614
1615 /*******************************************************************
1616  Matchname - determine if host name matches IP address. Used to
1617  confirm a hostname lookup to prevent spoof attacks.
1618 ******************************************************************/
1619
1620 static bool matchname(const char *remotehost,
1621                 const struct sockaddr_storage *pss,
1622                 socklen_t len)
1623 {
1624         struct addrinfo *res = NULL;
1625         struct addrinfo *ailist = NULL;
1626         char addr_buf[INET6_ADDRSTRLEN];
1627         bool ret = interpret_string_addr_internal(&ailist,
1628                         remotehost,
1629                         AI_ADDRCONFIG|AI_CANONNAME);
1630
1631         if (!ret || ailist == NULL) {
1632                 DEBUG(3,("matchname: getaddrinfo failed for "
1633                         "name %s [%s]\n",
1634                         remotehost,
1635                         gai_strerror(ret) ));
1636                 return false;
1637         }
1638
1639         /*
1640          * Make sure that getaddrinfo() returns the "correct" host name.
1641          */
1642
1643         if (ailist->ai_canonname == NULL ||
1644                 (!strequal(remotehost, ailist->ai_canonname) &&
1645                  !strequal(remotehost, "localhost"))) {
1646                 DEBUG(0,("matchname: host name/name mismatch: %s != %s\n",
1647                          remotehost,
1648                          ailist->ai_canonname ?
1649                                  ailist->ai_canonname : "(NULL)"));
1650                 freeaddrinfo(ailist);
1651                 return false;
1652         }
1653
1654         /* Look up the host address in the address list we just got. */
1655         for (res = ailist; res; res = res->ai_next) {
1656                 if (!res->ai_addr) {
1657                         continue;
1658                 }
1659                 if (addr_equal((const struct sockaddr_storage *)res->ai_addr,
1660                                         pss)) {
1661                         freeaddrinfo(ailist);
1662                         return true;
1663                 }
1664         }
1665
1666         /*
1667          * The host name does not map to the original host address. Perhaps
1668          * someone has compromised a name server. More likely someone botched
1669          * it, but that could be dangerous, too.
1670          */
1671
1672         DEBUG(0,("matchname: host name/address mismatch: %s != %s\n",
1673                 print_sockaddr_len(addr_buf,
1674                         sizeof(addr_buf),
1675                         pss,
1676                         len),
1677                  ailist->ai_canonname ? ailist->ai_canonname : "(NULL)"));
1678
1679         if (ailist) {
1680                 freeaddrinfo(ailist);
1681         }
1682         return false;
1683 }
1684
1685 /*******************************************************************
1686  Deal with the singleton cache.
1687 ******************************************************************/
1688
1689 struct name_addr_pair {
1690         struct sockaddr_storage ss;
1691         const char *name;
1692 };
1693
1694 /*******************************************************************
1695  Lookup a name/addr pair. Returns memory allocated from memcache.
1696 ******************************************************************/
1697
1698 static bool lookup_nc(struct name_addr_pair *nc)
1699 {
1700         DATA_BLOB tmp;
1701
1702         ZERO_STRUCTP(nc);
1703
1704         if (!memcache_lookup(
1705                         NULL, SINGLETON_CACHE,
1706                         data_blob_string_const("get_peer_name"),
1707                         &tmp)) {
1708                 return false;
1709         }
1710
1711         memcpy(&nc->ss, tmp.data, sizeof(nc->ss));
1712         nc->name = (const char *)tmp.data + sizeof(nc->ss);
1713         return true;
1714 }
1715
1716 /*******************************************************************
1717  Save a name/addr pair.
1718 ******************************************************************/
1719
1720 static void store_nc(const struct name_addr_pair *nc)
1721 {
1722         DATA_BLOB tmp;
1723         size_t namelen = strlen(nc->name);
1724
1725         tmp = data_blob(NULL, sizeof(nc->ss) + namelen + 1);
1726         if (!tmp.data) {
1727                 return;
1728         }
1729         memcpy(tmp.data, &nc->ss, sizeof(nc->ss));
1730         memcpy(tmp.data+sizeof(nc->ss), nc->name, namelen+1);
1731
1732         memcache_add(NULL, SINGLETON_CACHE,
1733                         data_blob_string_const("get_peer_name"),
1734                         tmp);
1735         data_blob_free(&tmp);
1736 }
1737
1738 /*******************************************************************
1739  Return the DNS name of the remote end of a socket.
1740 ******************************************************************/
1741
1742 const char *get_peer_name(int fd, bool force_lookup)
1743 {
1744         struct name_addr_pair nc;
1745         char addr_buf[INET6_ADDRSTRLEN];
1746         struct sockaddr_storage ss;
1747         socklen_t length = sizeof(ss);
1748         const char *p;
1749         int ret;
1750         char name_buf[MAX_DNS_NAME_LENGTH];
1751         char tmp_name[MAX_DNS_NAME_LENGTH];
1752
1753         /* reverse lookups can be *very* expensive, and in many
1754            situations won't work because many networks don't link dhcp
1755            with dns. To avoid the delay we avoid the lookup if
1756            possible */
1757         if (!lp_hostname_lookups() && (force_lookup == false)) {
1758                 length = sizeof(nc.ss);
1759                 nc.name = get_peer_addr_internal(fd, addr_buf, sizeof(addr_buf),
1760                         &nc.ss, &length);
1761                 store_nc(&nc);
1762                 lookup_nc(&nc);
1763                 return nc.name ? nc.name : "UNKNOWN";
1764         }
1765
1766         lookup_nc(&nc);
1767
1768         memset(&ss, '\0', sizeof(ss));
1769         p = get_peer_addr_internal(fd, addr_buf, sizeof(addr_buf), &ss, &length);
1770
1771         /* it might be the same as the last one - save some DNS work */
1772         if (addr_equal(&ss, &nc.ss)) {
1773                 return nc.name ? nc.name : "UNKNOWN";
1774         }
1775
1776         /* Not the same. We need to lookup. */
1777         if (fd == -1) {
1778                 return "UNKNOWN";
1779         }
1780
1781         /* Look up the remote host name. */
1782         ret = sys_getnameinfo((struct sockaddr *)&ss,
1783                         length,
1784                         name_buf,
1785                         sizeof(name_buf),
1786                         NULL,
1787                         0,
1788                         0);
1789
1790         if (ret) {
1791                 DEBUG(1,("get_peer_name: getnameinfo failed "
1792                         "for %s with error %s\n",
1793                         p,
1794                         gai_strerror(ret)));
1795                 strlcpy(name_buf, p, sizeof(name_buf));
1796         } else {
1797                 if (!matchname(name_buf, &ss, length)) {
1798                         DEBUG(0,("Matchname failed on %s %s\n",name_buf,p));
1799                         strlcpy(name_buf,"UNKNOWN",sizeof(name_buf));
1800                 }
1801         }
1802
1803         /* can't pass the same source and dest strings in when you
1804            use --enable-developer or the clobber_region() call will
1805            get you */
1806
1807         strlcpy(tmp_name, name_buf, sizeof(tmp_name));
1808         alpha_strcpy(name_buf, tmp_name, "_-.", sizeof(name_buf));
1809         if (strstr(name_buf,"..")) {
1810                 strlcpy(name_buf, "UNKNOWN", sizeof(name_buf));
1811         }
1812
1813         nc.name = name_buf;
1814         nc.ss = ss;
1815
1816         store_nc(&nc);
1817         lookup_nc(&nc);
1818         return nc.name ? nc.name : "UNKNOWN";
1819 }
1820
1821 /*******************************************************************
1822  Return the IP addr of the remote end of a socket as a string.
1823  ******************************************************************/
1824
1825 const char *get_peer_addr(int fd, char *addr, size_t addr_len)
1826 {
1827         return get_peer_addr_internal(fd, addr, addr_len, NULL, NULL);
1828 }
1829
1830 /*******************************************************************
1831  Create protected unix domain socket.
1832
1833  Some unixes cannot set permissions on a ux-dom-sock, so we
1834  have to make sure that the directory contains the protection
1835  permissions instead.
1836  ******************************************************************/
1837
1838 int create_pipe_sock(const char *socket_dir,
1839                      const char *socket_name,
1840                      mode_t dir_perms)
1841 {
1842 #ifdef HAVE_UNIXSOCKET
1843         struct sockaddr_un sunaddr;
1844         struct stat st;
1845         int sock;
1846         mode_t old_umask;
1847         char *path = NULL;
1848
1849         old_umask = umask(0);
1850
1851         /* Create the socket directory or reuse the existing one */
1852
1853         if (lstat(socket_dir, &st) == -1) {
1854                 if (errno == ENOENT) {
1855                         /* Create directory */
1856                         if (mkdir(socket_dir, dir_perms) == -1) {
1857                                 DEBUG(0, ("error creating socket directory "
1858                                         "%s: %s\n", socket_dir,
1859                                         strerror(errno)));
1860                                 goto out_umask;
1861                         }
1862                 } else {
1863                         DEBUG(0, ("lstat failed on socket directory %s: %s\n",
1864                                 socket_dir, strerror(errno)));
1865                         goto out_umask;
1866                 }
1867         } else {
1868                 /* Check ownership and permission on existing directory */
1869                 if (!S_ISDIR(st.st_mode)) {
1870                         DEBUG(0, ("socket directory %s isn't a directory\n",
1871                                 socket_dir));
1872                         goto out_umask;
1873                 }
1874                 if ((st.st_uid != sec_initial_uid()) ||
1875                                 ((st.st_mode & 0777) != dir_perms)) {
1876                         DEBUG(0, ("invalid permissions on socket directory "
1877                                 "%s\n", socket_dir));
1878                         goto out_umask;
1879                 }
1880         }
1881
1882         /* Create the socket file */
1883
1884         sock = socket(AF_UNIX, SOCK_STREAM, 0);
1885
1886         if (sock == -1) {
1887                 DEBUG(0, ("create_pipe_sock: socket error %s\n",
1888                         strerror(errno) ));
1889                 goto out_close;
1890         }
1891
1892         if (asprintf(&path, "%s/%s", socket_dir, socket_name) == -1) {
1893                 goto out_close;
1894         }
1895
1896         unlink(path);
1897         memset(&sunaddr, 0, sizeof(sunaddr));
1898         sunaddr.sun_family = AF_UNIX;
1899         strlcpy(sunaddr.sun_path, path, sizeof(sunaddr.sun_path));
1900
1901         if (bind(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) == -1) {
1902                 DEBUG(0, ("bind failed on pipe socket %s: %s\n", path,
1903                         strerror(errno)));
1904                 goto out_close;
1905         }
1906
1907         if (listen(sock, 5) == -1) {
1908                 DEBUG(0, ("listen failed on pipe socket %s: %s\n", path,
1909                         strerror(errno)));
1910                 goto out_close;
1911         }
1912
1913         SAFE_FREE(path);
1914
1915         umask(old_umask);
1916         return sock;
1917
1918 out_close:
1919         SAFE_FREE(path);
1920         if (sock != -1)
1921                 close(sock);
1922
1923 out_umask:
1924         umask(old_umask);
1925         return -1;
1926
1927 #else
1928         DEBUG(0, ("create_pipe_sock: No Unix sockets on this system\n"));
1929         return -1;
1930 #endif /* HAVE_UNIXSOCKET */
1931 }
1932
1933 /****************************************************************************
1934  Get my own canonical name, including domain.
1935 ****************************************************************************/
1936
1937 const char *get_mydnsfullname(void)
1938 {
1939         struct addrinfo *res = NULL;
1940         char my_hostname[HOST_NAME_MAX];
1941         bool ret;
1942         DATA_BLOB tmp;
1943
1944         if (memcache_lookup(NULL, SINGLETON_CACHE,
1945                         data_blob_string_const("get_mydnsfullname"),
1946                         &tmp)) {
1947                 SMB_ASSERT(tmp.length > 0);
1948                 return (const char *)tmp.data;
1949         }
1950
1951         /* get my host name */
1952         if (gethostname(my_hostname, sizeof(my_hostname)) == -1) {
1953                 DEBUG(0,("get_mydnsfullname: gethostname failed\n"));
1954                 return NULL;
1955         }
1956
1957         /* Ensure null termination. */
1958         my_hostname[sizeof(my_hostname)-1] = '\0';
1959
1960         ret = interpret_string_addr_internal(&res,
1961                                 my_hostname,
1962                                 AI_ADDRCONFIG|AI_CANONNAME);
1963
1964         if (!ret || res == NULL) {
1965                 DEBUG(3,("get_mydnsfullname: getaddrinfo failed for "
1966                         "name %s [%s]\n",
1967                         my_hostname,
1968                         gai_strerror(ret) ));
1969                 return NULL;
1970         }
1971
1972         /*
1973          * Make sure that getaddrinfo() returns the "correct" host name.
1974          */
1975
1976         if (res->ai_canonname == NULL) {
1977                 DEBUG(3,("get_mydnsfullname: failed to get "
1978                         "canonical name for %s\n",
1979                         my_hostname));
1980                 freeaddrinfo(res);
1981                 return NULL;
1982         }
1983
1984         /* This copies the data, so we must do a lookup
1985          * afterwards to find the value to return.
1986          */
1987
1988         memcache_add(NULL, SINGLETON_CACHE,
1989                         data_blob_string_const("get_mydnsfullname"),
1990                         data_blob_string_const(res->ai_canonname));
1991
1992         if (!memcache_lookup(NULL, SINGLETON_CACHE,
1993                         data_blob_string_const("get_mydnsfullname"),
1994                         &tmp)) {
1995                 tmp = data_blob_talloc(talloc_tos(), res->ai_canonname,
1996                                 strlen(res->ai_canonname) + 1);
1997         }
1998
1999         freeaddrinfo(res);
2000
2001         return (const char *)tmp.data;
2002 }
2003
2004 /************************************************************
2005  Is this my name ?
2006 ************************************************************/
2007
2008 bool is_myname_or_ipaddr(const char *s)
2009 {
2010         TALLOC_CTX *ctx = talloc_tos();
2011         char *name = NULL;
2012         const char *dnsname;
2013         char *servername = NULL;
2014
2015         if (!s) {
2016                 return false;
2017         }
2018
2019         /* Santize the string from '\\name' */
2020         name = talloc_strdup(ctx, s);
2021         if (!name) {
2022                 return false;
2023         }
2024
2025         servername = strrchr_m(name, '\\' );
2026         if (!servername) {
2027                 servername = name;
2028         } else {
2029                 servername++;
2030         }
2031
2032         /* Optimize for the common case */
2033         if (strequal(servername, global_myname())) {
2034                 return true;
2035         }
2036
2037         /* Check for an alias */
2038         if (is_myname(servername)) {
2039                 return true;
2040         }
2041
2042         /* Check for loopback */
2043         if (strequal(servername, "127.0.0.1") ||
2044                         strequal(servername, "::1")) {
2045                 return true;
2046         }
2047
2048         if (strequal(servername, "localhost")) {
2049                 return true;
2050         }
2051
2052         /* Maybe it's my dns name */
2053         dnsname = get_mydnsfullname();
2054         if (dnsname && strequal(servername, dnsname)) {
2055                 return true;
2056         }
2057
2058         /* Handle possible CNAME records - convert to an IP addr. */
2059         if (!is_ipaddress(servername)) {
2060                 /* Use DNS to resolve the name, but only the first address */
2061                 struct sockaddr_storage ss;
2062                 if (interpret_string_addr(&ss, servername,0)) {
2063                         print_sockaddr(name,
2064                                         sizeof(name),
2065                                         &ss);
2066                         servername = name;
2067                 }
2068         }
2069
2070         /* Maybe its an IP address? */
2071         if (is_ipaddress(servername)) {
2072                 struct sockaddr_storage ss;
2073                 struct iface_struct *nics;
2074                 int i, n;
2075
2076                 if (!interpret_string_addr(&ss, servername, AI_NUMERICHOST)) {
2077                         return false;
2078                 }
2079
2080                 if (is_zero_addr(&ss) || is_loopback_addr(&ss)) {
2081                         return false;
2082                 }
2083
2084                 nics = TALLOC_ARRAY(ctx, struct iface_struct,
2085                                         MAX_INTERFACES);
2086                 if (!nics) {
2087                         return false;
2088                 }
2089                 n = get_interfaces(nics, MAX_INTERFACES);
2090                 for (i=0; i<n; i++) {
2091                         if (addr_equal(&nics[i].ip, &ss)) {
2092                                 TALLOC_FREE(nics);
2093                                 return true;
2094                         }
2095                 }
2096                 TALLOC_FREE(nics);
2097         }
2098
2099         /* No match */
2100         return false;
2101 }