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