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