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