source/lib/util_sock.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Tim Potter      2000-2001
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23
  24 /* the last IP received from */
  25 struct in_addr lastip;
  26
  27 /* the last port received from */
  28 int lastport=0;
  29
  30 int smb_read_error = 0;
  31
  32 static char *get_socket_addr(int fd)
  33 {
  34         struct sockaddr sa;
  35         struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
  36         socklen_t length = sizeof(sa);
  37         static fstring addr_buf;
  38
  39         fstrcpy(addr_buf,"0.0.0.0");
  40
  41         if (fd == -1) {
  42                 return addr_buf;
  43         }
  44
  45         if (getsockname(fd, &sa, &length) < 0) {
  46                 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
  47                 return addr_buf;
  48         }
  49
  50         fstrcpy(addr_buf,(char *)inet_ntoa(sockin->sin_addr));
  51
  52         return addr_buf;
  53 }
  54
  55 static int get_socket_port(int fd)
  56 {
  57         struct sockaddr sa;
  58         struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
  59         socklen_t length = sizeof(sa);
  60
  61         if (fd == -1)
  62                 return -1;
  63
  64         if (getsockname(fd, &sa, &length) < 0) {
  65                 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
  66                 return -1;
  67         }
  68
  69         return ntohs(sockin->sin_port);
  70 }
  71
  72 /****************************************************************************
  73  Determine if a file descriptor is in fact a socket.
  74 ****************************************************************************/
  75
  76 BOOL is_a_socket(int fd)
  77 {
  78         int v;
  79         socklen_t l;
  80         l = sizeof(int);
  81         return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
  82 }
  83
  84 enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
  85
  86 typedef struct smb_socket_option {
  87         const char *name;
  88         int level;
  89         int option;
  90         int value;
  91         int opttype;
  92 } smb_socket_option;
  93
  94 static const smb_socket_option socket_options[] = {
  95   {"SO_KEEPALIVE",      SOL_SOCKET,    SO_KEEPALIVE,    0,                 OPT_BOOL},
  96   {"SO_REUSEADDR",      SOL_SOCKET,    SO_REUSEADDR,    0,                 OPT_BOOL},
  97   {"SO_BROADCAST",      SOL_SOCKET,    SO_BROADCAST,    0,                 OPT_BOOL},
  98 #ifdef TCP_NODELAY
  99   {"TCP_NODELAY",       IPPROTO_TCP,   TCP_NODELAY,     0,                 OPT_BOOL},
 100 #endif
 101 #ifdef IPTOS_LOWDELAY
 102   {"IPTOS_LOWDELAY",    IPPROTO_IP,    IP_TOS,          IPTOS_LOWDELAY,    OPT_ON},
 103 #endif
 104 #ifdef IPTOS_THROUGHPUT
 105   {"IPTOS_THROUGHPUT",  IPPROTO_IP,    IP_TOS,          IPTOS_THROUGHPUT,  OPT_ON},
 106 #endif
 107 #ifdef SO_REUSEPORT
 108   {"SO_REUSEPORT",      SOL_SOCKET,    SO_REUSEPORT,    0,                 OPT_BOOL},
 109 #endif
 110 #ifdef SO_SNDBUF
 111   {"SO_SNDBUF",         SOL_SOCKET,    SO_SNDBUF,       0,                 OPT_INT},
 112 #endif
 113 #ifdef SO_RCVBUF
 114   {"SO_RCVBUF",         SOL_SOCKET,    SO_RCVBUF,       0,                 OPT_INT},
 115 #endif
 116 #ifdef SO_SNDLOWAT
 117   {"SO_SNDLOWAT",       SOL_SOCKET,    SO_SNDLOWAT,     0,                 OPT_INT},
 118 #endif
 119 #ifdef SO_RCVLOWAT
 120   {"SO_RCVLOWAT",       SOL_SOCKET,    SO_RCVLOWAT,     0,                 OPT_INT},
 121 #endif
 122 #ifdef SO_SNDTIMEO
 123   {"SO_SNDTIMEO",       SOL_SOCKET,    SO_SNDTIMEO,     0,                 OPT_INT},
 124 #endif
 125 #ifdef SO_RCVTIMEO
 126   {"SO_RCVTIMEO",       SOL_SOCKET,    SO_RCVTIMEO,     0,                 OPT_INT},
 127 #endif
 128   {NULL,0,0,0,0}};
 129
 130 /****************************************************************************
 131  Print socket options.
 132 ****************************************************************************/
 133
 134 static void print_socket_options(int s)
 135 {
 136         int value;
 137         socklen_t vlen = 4;
 138         const smb_socket_option *p = &socket_options[0];
 139
 140         /* wrapped in if statement to prevent streams leak in SCO Openserver 5.0 */
 141         /* reported on samba-technical  --jerry */
 142         if ( DEBUGLEVEL >= 5 ) {
 143         for (; p->name != NULL; p++) {
 144                 if (getsockopt(s, p->level, p->option, (void *)&value, &vlen) == -1) {
 145                         DEBUG(5,("Could not test socket option %s.\n", p->name));
 146                 } else {
 147                         DEBUG(5,("socket option %s = %d\n",p->name,value));
 148                         }
 149                 }
 150         }
 151  }
 152
 153 /****************************************************************************
 154  Set user socket options.
 155 ****************************************************************************/
 156
 157 void set_socket_options(int fd, const char *options)
 158 {
 159         fstring tok;
 160
 161         while (next_token(&options,tok," \t,", sizeof(tok))) {
 162                 int ret=0,i;
 163                 int value = 1;
 164                 char *p;
 165                 BOOL got_value = False;
 166
 167                 if ((p = strchr_m(tok,'='))) {
 168                         *p = 0;
 169                         value = atoi(p+1);
 170                         got_value = True;
 171                 }
 172
 173                 for (i=0;socket_options[i].name;i++)
 174                         if (strequal(socket_options[i].name,tok))
 175                                 break;
 176
 177                 if (!socket_options[i].name) {
 178                         DEBUG(0,("Unknown socket option %s\n",tok));
 179                         continue;
 180                 }
 181
 182                 switch (socket_options[i].opttype) {
 183                 case OPT_BOOL:
 184                 case OPT_INT:
 185                         ret = setsockopt(fd,socket_options[i].level,
 186                                                 socket_options[i].option,(char *)&value,sizeof(int));
 187                         break;
 188
 189                 case OPT_ON:
 190                         if (got_value)
 191                                 DEBUG(0,("syntax error - %s does not take a value\n",tok));
 192
 193                         {
 194                                 int on = socket_options[i].value;
 195                                 ret = setsockopt(fd,socket_options[i].level,
 196                                                         socket_options[i].option,(char *)&on,sizeof(int));
 197                         }
 198                         break;
 199                 }
 200
 201                 if (ret != 0)
 202                         DEBUG(0,("Failed to set socket option %s (Error %s)\n",tok, strerror(errno) ));
 203         }
 204
 205         print_socket_options(fd);
 206 }
 207
 208 /****************************************************************************
 209  Read from a socket.
 210 ****************************************************************************/
 211
 212 ssize_t read_udp_socket(int fd,char *buf,size_t len)
 213 {
 214         ssize_t ret;
 215         struct sockaddr_in sock;
 216         socklen_t socklen = sizeof(sock);
 217
 218         memset((char *)&sock,'\0',socklen);
 219         memset((char *)&lastip,'\0',sizeof(lastip));
 220         ret = (ssize_t)sys_recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
 221         if (ret <= 0) {
 222                 DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
 223                 return(0);
 224         }
 225
 226         lastip = sock.sin_addr;
 227         lastport = ntohs(sock.sin_port);
 228
 229         DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %lu\n",
 230                         inet_ntoa(lastip), lastport, (unsigned long)ret));
 231
 232         return(ret);
 233 }
 234
 235 /****************************************************************************
 236  Read data from a socket with a timout in msec.
 237  mincount = if timeout, minimum to read before returning
 238  maxcount = number to be read.
 239  time_out = timeout in milliseconds
 240 ****************************************************************************/
 241
 242 ssize_t read_socket_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
 243 {
 244         fd_set fds;
 245         int selrtn;
 246         ssize_t readret;
 247         size_t nread = 0;
 248         struct timeval timeout;
 249
 250         /* just checking .... */
 251         if (maxcnt <= 0)
 252                 return(0);
 253
 254         smb_read_error = 0;
 255
 256         /* Blocking read */
 257         if (time_out <= 0) {
 258                 if (mincnt == 0) mincnt = maxcnt;
 259
 260                 while (nread < mincnt) {
 261                         readret = sys_read(fd, buf + nread, maxcnt - nread);
 262
 263                         if (readret == 0) {
 264                                 DEBUG(5,("read_socket_with_timeout: blocking read. EOF from client.\n"));
 265                                 smb_read_error = READ_EOF;
 266                                 return -1;
 267                         }
 268
 269                         if (readret == -1) {
 270                                 DEBUG(0,("read_socket_with_timeout: read error = %s.\n", strerror(errno) ));
 271                                 smb_read_error = READ_ERROR;
 272                                 return -1;
 273                         }
 274                         nread += readret;
 275                 }
 276                 return((ssize_t)nread);
 277         }
 278
 279         /* Most difficult - timeout read */
 280         /* If this is ever called on a disk file and
 281            mincnt is greater then the filesize then
 282            system performance will suffer severely as
 283            select always returns true on disk files */
 284
 285         /* Set initial timeout */
 286         timeout.tv_sec = (time_t)(time_out / 1000);
 287         timeout.tv_usec = (long)(1000 * (time_out % 1000));
 288
 289         for (nread=0; nread < mincnt; ) {
 290                 FD_ZERO(&fds);
 291                 FD_SET(fd,&fds);
 292
 293                 selrtn = sys_select_intr(fd+1,&fds,NULL,NULL,&timeout);
 294
 295                 /* Check if error */
 296                 if (selrtn == -1) {
 297                         /* something is wrong. Maybe the socket is dead? */
 298                         DEBUG(0,("read_socket_with_timeout: timeout read. select error = %s.\n", strerror(errno) ));
 299                         smb_read_error = READ_ERROR;
 300                         return -1;
 301                 }
 302
 303                 /* Did we timeout ? */
 304                 if (selrtn == 0) {
 305                         DEBUG(10,("read_socket_with_timeout: timeout read. select timed out.\n"));
 306                         smb_read_error = READ_TIMEOUT;
 307                         return -1;
 308                 }
 309
 310                 readret = sys_read(fd, buf+nread, maxcnt-nread);
 311
 312                 if (readret == 0) {
 313                         /* we got EOF on the file descriptor */
 314                         DEBUG(5,("read_socket_with_timeout: timeout read. EOF from client.\n"));
 315                         smb_read_error = READ_EOF;
 316                         return -1;
 317                 }
 318
 319                 if (readret == -1) {
 320                         /* the descriptor is probably dead */
 321                         DEBUG(0,("read_socket_with_timeout: timeout read. read error = %s.\n", strerror(errno) ));
 322                         smb_read_error = READ_ERROR;
 323                         return -1;
 324                 }
 325
 326                 nread += readret;
 327         }
 328
 329         /* Return the number we got */
 330         return (ssize_t)nread;
 331 }
 332
 333 /****************************************************************************
 334  Read data from the client, reading exactly N bytes.
 335 ****************************************************************************/
 336
 337 ssize_t read_data(int fd,char *buffer,size_t N)
 338 {
 339         ssize_t ret;
 340         size_t total=0;
 341
 342         smb_read_error = 0;
 343
 344         while (total < N) {
 345                 ret = sys_read(fd,buffer + total,N - total);
 346
 347                 if (ret == 0) {
 348                         DEBUG(10,("read_data: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
 349                         smb_read_error = READ_EOF;
 350                         return 0;
 351                 }
 352
 353                 if (ret == -1) {
 354                         DEBUG(0,("read_data: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
 355                         smb_read_error = READ_ERROR;
 356                         return -1;
 357                 }
 358                 total += ret;
 359         }
 360         return (ssize_t)total;
 361 }
 362
 363 /****************************************************************************
 364  Read data from a socket, reading exactly N bytes.
 365 ****************************************************************************/
 366
 367 static ssize_t read_socket_data(int fd,char *buffer,size_t N)
 368 {
 369         ssize_t ret;
 370         size_t total=0;
 371
 372         smb_read_error = 0;
 373
 374         while (total < N) {
 375                 ret = sys_read(fd,buffer + total,N - total);
 376
 377                 if (ret == 0) {
 378                         DEBUG(10,("read_socket_data: recv of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
 379                         smb_read_error = READ_EOF;
 380                         return 0;
 381                 }
 382
 383                 if (ret == -1) {
 384                         DEBUG(0,("read_socket_data: recv failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
 385                         smb_read_error = READ_ERROR;
 386                         return -1;
 387                 }
 388                 total += ret;
 389         }
 390         return (ssize_t)total;
 391 }
 392
 393 /****************************************************************************
 394  Write data to a fd.
 395 ****************************************************************************/
 396
 397 ssize_t write_data(int fd,char *buffer,size_t N)
 398 {
 399         size_t total=0;
 400         ssize_t ret;
 401
 402         while (total < N) {
 403                 ret = sys_write(fd,buffer + total,N - total);
 404
 405                 if (ret == -1) {
 406                         DEBUG(0,("write_data: write failure. Error = %s\n", strerror(errno) ));
 407                         return -1;
 408                 }
 409                 if (ret == 0)
 410                         return total;
 411
 412                 total += ret;
 413         }
 414         return (ssize_t)total;
 415 }
 416
 417 /****************************************************************************
 418  Write data to a socket - use send rather than write.
 419 ****************************************************************************/
 420
 421 static ssize_t write_socket_data(int fd,char *buffer,size_t N)
 422 {
 423         size_t total=0;
 424         ssize_t ret;
 425
 426         while (total < N) {
 427                 ret = sys_send(fd,buffer + total,N - total,0);
 428
 429                 if (ret == -1) {
 430                         DEBUG(0,("write_socket_data: write failure. Error = %s\n", strerror(errno) ));
 431                         return -1;
 432                 }
 433                 if (ret == 0)
 434                         return total;
 435
 436                 total += ret;
 437         }
 438         return (ssize_t)total;
 439 }
 440
 441 /****************************************************************************
 442  Write to a socket.
 443 ****************************************************************************/
 444
 445 ssize_t write_socket(int fd,char *buf,size_t len)
 446 {
 447         ssize_t ret=0;
 448
 449         DEBUG(6,("write_socket(%d,%d)\n",fd,(int)len));
 450         ret = write_socket_data(fd,buf,len);
 451
 452         DEBUG(6,("write_socket(%d,%d) wrote %d\n",fd,(int)len,(int)ret));
 453         if(ret <= 0)
 454                 DEBUG(0,("write_socket: Error writing %d bytes to socket %d: ERRNO = %s\n",
 455                         (int)len, fd, strerror(errno) ));
 456
 457         return(ret);
 458 }
 459
 460 /****************************************************************************
 461  Send a keepalive packet (rfc1002).
 462 ****************************************************************************/
 463
 464 BOOL send_keepalive(int client)
 465 {
 466         unsigned char buf[4];
 467
 468         buf[0] = SMBkeepalive;
 469         buf[1] = buf[2] = buf[3] = 0;
 470
 471         return(write_socket_data(client,(char *)buf,4) == 4);
 472 }
 473
 474
 475 /****************************************************************************
 476  Read 4 bytes of a smb packet and return the smb length of the packet.
 477  Store the result in the buffer.
 478  This version of the function will return a length of zero on receiving
 479  a keepalive packet.
 480  Timeout is in milliseconds.
 481 ****************************************************************************/
 482
 483 static ssize_t read_smb_length_return_keepalive(int fd,char *inbuf,unsigned int timeout)
 484 {
 485         ssize_t len=0;
 486         int msg_type;
 487         BOOL ok = False;
 488
 489         while (!ok) {
 490                 if (timeout > 0)
 491                         ok = (read_socket_with_timeout(fd,inbuf,4,4,timeout) == 4);
 492                 else
 493                         ok = (read_socket_data(fd,inbuf,4) == 4);
 494
 495                 if (!ok)
 496                         return(-1);
 497
 498                 len = smb_len(inbuf);
 499                 msg_type = CVAL(inbuf,0);
 500
 501                 if (msg_type == SMBkeepalive)
 502                         DEBUG(5,("Got keepalive packet\n"));
 503         }
 504
 505         DEBUG(10,("got smb length of %lu\n",(unsigned long)len));
 506
 507         return(len);
 508 }
 509
 510 /****************************************************************************
 511  Read 4 bytes of a smb packet and return the smb length of the packet.
 512  Store the result in the buffer. This version of the function will
 513  never return a session keepalive (length of zero).
 514  Timeout is in milliseconds.
 515 ****************************************************************************/
 516
 517 ssize_t read_smb_length(int fd,char *inbuf,unsigned int timeout)
 518 {
 519         ssize_t len;
 520
 521         for(;;) {
 522                 len = read_smb_length_return_keepalive(fd, inbuf, timeout);
 523
 524                 if(len < 0)
 525                         return len;
 526
 527                 /* Ignore session keepalives. */
 528                 if(CVAL(inbuf,0) != SMBkeepalive)
 529                         break;
 530         }
 531
 532         DEBUG(10,("read_smb_length: got smb length of %lu\n",
 533                   (unsigned long)len));
 534
 535         return len;
 536 }
 537
 538 /****************************************************************************
 539  Read an smb from a fd. Note that the buffer *MUST* be of size
 540  BUFFER_SIZE+SAFETY_MARGIN.
 541  The timeout is in milliseconds.
 542  This function will return on receipt of a session keepalive packet.
 543  Doesn't check the MAC on signed packets.
 544 ****************************************************************************/
 545
 546 BOOL receive_smb_raw(int fd,char *buffer, unsigned int timeout)
 547 {
 548         ssize_t len,ret;
 549
 550         smb_read_error = 0;
 551
 552         memset(buffer,'\0',smb_size + 100);
 553
 554         len = read_smb_length_return_keepalive(fd,buffer,timeout);
 555         if (len < 0) {
 556                 DEBUG(10,("receive_smb_raw: length < 0!\n"));
 557
 558                 /*
 559                  * Correct fix. smb_read_error may have already been
 560                  * set. Only set it here if not already set. Global
 561                  * variables still suck :-). JRA.
 562                  */
 563
 564                 if (smb_read_error == 0)
 565                         smb_read_error = READ_ERROR;
 566                 return False;
 567         }
 568
 569         /*
 570          * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
 571          * of header. Don't print the error if this fits.... JRA.
 572          */
 573
 574         if (len > (BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
 575                 DEBUG(0,("Invalid packet length! (%lu bytes).\n",(unsigned long)len));
 576                 if (len > BUFFER_SIZE + (SAFETY_MARGIN/2)) {
 577
 578                         /*
 579                          * Correct fix. smb_read_error may have already been
 580                          * set. Only set it here if not already set. Global
 581                          * variables still suck :-). JRA.
 582                          */
 583
 584                         if (smb_read_error == 0)
 585                                 smb_read_error = READ_ERROR;
 586                         return False;
 587                 }
 588         }
 589
 590         if(len > 0) {
 591                 ret = read_socket_data(fd,buffer+4,len);
 592                 if (ret != len) {
 593                         if (smb_read_error == 0)
 594                                 smb_read_error = READ_ERROR;
 595                         return False;
 596                 }
 597
 598                 /* not all of samba3 properly checks for packet-termination of strings. This
 599                    ensures that we don't run off into empty space. */
 600                 SSVAL(buffer+4,len, 0);
 601         }
 602
 603         return True;
 604 }
 605
 606 /****************************************************************************
 607  Wrapper for receive_smb_raw().
 608  Checks the MAC on signed packets.
 609 ****************************************************************************/
 610
 611 BOOL receive_smb(int fd,char *buffer, unsigned int timeout)
 612 {
 613         if (!receive_smb_raw(fd, buffer, timeout)) {
 614                 return False;
 615         }
 616
 617         /* Check the incoming SMB signature. */
 618         if (!srv_check_sign_mac(buffer, True)) {
 619                 DEBUG(0, ("receive_smb: SMB Signature verification failed on incoming packet!\n"));
 620                 if (smb_read_error == 0)
 621                         smb_read_error = READ_BAD_SIG;
 622                 return False;
 623         };
 624
 625         return(True);
 626 }
 627
 628 /****************************************************************************
 629  Send an smb to a fd.
 630 ****************************************************************************/
 631
 632 BOOL send_smb(int fd,char *buffer)
 633 {
 634         size_t len;
 635         size_t nwritten=0;
 636         ssize_t ret;
 637
 638         /* Sign the outgoing packet if required. */
 639         srv_calculate_sign_mac(buffer);
 640
 641         len = smb_len(buffer) + 4;
 642
 643         while (nwritten < len) {
 644                 ret = write_socket(fd,buffer+nwritten,len - nwritten);
 645                 if (ret <= 0) {
 646                         DEBUG(0,("Error writing %d bytes to client. %d. (%s)\n",
 647                                 (int)len,(int)ret, strerror(errno) ));
 648                         return False;
 649                 }
 650                 nwritten += ret;
 651         }
 652
 653         return True;
 654 }
 655
 656 /****************************************************************************
 657  Open a socket of the specified type, port, and address for incoming data.
 658 ****************************************************************************/
 659
 660 int open_socket_in( int type, int port, int dlevel, uint32 socket_addr, BOOL rebind )
 661 {
 662         struct sockaddr_in sock;
 663         int res;
 664
 665         memset( (char *)&sock, '\0', sizeof(sock) );
 666
 667 #ifdef HAVE_SOCK_SIN_LEN
 668         sock.sin_len         = sizeof(sock);
 669 #endif
 670         sock.sin_port        = htons( port );
 671         sock.sin_family      = AF_INET;
 672         sock.sin_addr.s_addr = socket_addr;
 673
 674         res = socket( AF_INET, type, 0 );
 675         if( res == -1 ) {
 676                 if( DEBUGLVL(0) ) {
 677                         dbgtext( "open_socket_in(): socket() call failed: " );
 678                         dbgtext( "%s\n", strerror( errno ) );
 679                 }
 680                 return -1;
 681         }
 682
 683         /* This block sets/clears the SO_REUSEADDR and possibly SO_REUSEPORT. */
 684         {
 685                 int val = rebind ? 1 : 0;
 686                 if( setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val)) == -1 ) {
 687                         if( DEBUGLVL( dlevel ) ) {
 688                                 dbgtext( "open_socket_in(): setsockopt: " );
 689                                 dbgtext( "SO_REUSEADDR = %s ", val?"True":"False" );
 690                                 dbgtext( "on port %d failed ", port );
 691                                 dbgtext( "with error = %s\n", strerror(errno) );
 692                         }
 693                 }
 694 #ifdef SO_REUSEPORT
 695                 if( setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val)) == -1 ) {
 696                         if( DEBUGLVL( dlevel ) ) {
 697                                 dbgtext( "open_socket_in(): setsockopt: ");
 698                                 dbgtext( "SO_REUSEPORT = %s ", val?"True":"False" );
 699                                 dbgtext( "on port %d failed ", port );
 700                                 dbgtext( "with error = %s\n", strerror(errno) );
 701                         }
 702                 }
 703 #endif /* SO_REUSEPORT */
 704         }
 705
 706         /* now we've got a socket - we need to bind it */
 707         if( bind( res, (struct sockaddr *)&sock, sizeof(sock) ) == -1 ) {
 708                 if( DEBUGLVL(dlevel) && (port == SMB_PORT1 || port == SMB_PORT2 || port == NMB_PORT) ) {
 709                         dbgtext( "bind failed on port %d ", port );
 710                         dbgtext( "socket_addr = %s.\n", inet_ntoa( sock.sin_addr ) );
 711                         dbgtext( "Error = %s\n", strerror(errno) );
 712                 }
 713                 close( res );
 714                 return( -1 );
 715         }
 716
 717         DEBUG( 10, ( "bind succeeded on port %d\n", port ) );
 718
 719         return( res );
 720  }
 721
 722 /****************************************************************************
 723  Create an outgoing socket. timeout is in milliseconds.
 724 **************************************************************************/
 725
 726 int open_socket_out(int type, struct in_addr *addr, int port ,int timeout)
 727 {
 728         struct sockaddr_in sock_out;
 729         int res,ret;
 730         int connect_loop = 10;
 731         int increment = 10;
 732
 733         /* create a socket to write to */
 734         res = socket(PF_INET, type, 0);
 735         if (res == -1) {
 736                 DEBUG(0,("socket error (%s)\n", strerror(errno)));
 737                 return -1;
 738         }
 739
 740         if (type != SOCK_STREAM)
 741                 return(res);
 742
 743         memset((char *)&sock_out,'\0',sizeof(sock_out));
 744         putip((char *)&sock_out.sin_addr,(char *)addr);
 745
 746         sock_out.sin_port = htons( port );
 747         sock_out.sin_family = PF_INET;
 748
 749         /* set it non-blocking */
 750         set_blocking(res,False);
 751
 752         DEBUG(3,("Connecting to %s at port %d\n",inet_ntoa(*addr),port));
 753
 754         /* and connect it to the destination */
 755   connect_again:
 756
 757         ret = connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out));
 758
 759         /* Some systems return EAGAIN when they mean EINPROGRESS */
 760         if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
 761                         errno == EAGAIN) && (connect_loop < timeout) ) {
 762                 smb_msleep(connect_loop);
 763                 timeout -= connect_loop;
 764                 connect_loop += increment;
 765                 if (increment < 250) {
 766                         /* After 8 rounds we end up at a max of 255 msec */
 767                         increment *= 1.5;
 768                 }
 769                 goto connect_again;
 770         }
 771
 772         if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
 773                         errno == EAGAIN)) {
 774                 DEBUG(1,("timeout connecting to %s:%d\n",inet_ntoa(*addr),port));
 775                 close(res);
 776                 return -1;
 777         }
 778
 779 #ifdef EISCONN
 780
 781         if (ret < 0 && errno == EISCONN) {
 782                 errno = 0;
 783                 ret = 0;
 784         }
 785 #endif
 786
 787         if (ret < 0) {
 788                 DEBUG(2,("error connecting to %s:%d (%s)\n",
 789                                 inet_ntoa(*addr),port,strerror(errno)));
 790                 close(res);
 791                 return -1;
 792         }
 793
 794         /* set it blocking again */
 795         set_blocking(res,True);
 796
 797         return res;
 798 }
 799
 800 /****************************************************************************
 801  Open a connected UDP socket to host on port
 802 **************************************************************************/
 803
 804 int open_udp_socket(const char *host, int port)
 805 {
 806         int type = SOCK_DGRAM;
 807         struct sockaddr_in sock_out;
 808         int res;
 809         struct in_addr *addr;
 810
 811         addr = interpret_addr2(host);
 812
 813         res = socket(PF_INET, type, 0);
 814         if (res == -1) {
 815                 return -1;
 816         }
 817
 818         memset((char *)&sock_out,'\0',sizeof(sock_out));
 819         putip((char *)&sock_out.sin_addr,(char *)addr);
 820         sock_out.sin_port = htons(port);
 821         sock_out.sin_family = PF_INET;
 822
 823         if (connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out))) {
 824                 close(res);
 825                 return -1;
 826         }
 827
 828         return res;
 829 }
 830
 831
 832 /* the following 3 client_*() functions are nasty ways of allowing
 833    some generic functions to get info that really should be hidden in
 834    particular modules */
 835 static int client_fd = -1;
 836
 837 void client_setfd(int fd)
 838 {
 839         client_fd = fd;
 840 }
 841
 842 char *client_name(void)
 843 {
 844         return get_peer_name(client_fd,False);
 845 }
 846
 847 char *client_addr(void)
 848 {
 849         return get_peer_addr(client_fd);
 850 }
 851
 852 char *client_socket_addr(void)
 853 {
 854         return get_socket_addr(client_fd);
 855 }
 856
 857 int client_socket_port(void)
 858 {
 859         return get_socket_port(client_fd);
 860 }
 861
 862 struct in_addr *client_inaddr(struct sockaddr *sa)
 863 {
 864         struct sockaddr_in *sockin = (struct sockaddr_in *) (sa);
 865         socklen_t  length = sizeof(*sa);
 866
 867         if (getpeername(client_fd, sa, &length) < 0) {
 868                 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
 869                 return NULL;
 870         }
 871
 872         return &sockin->sin_addr;
 873 }
 874
 875 /*******************************************************************
 876  Matchname - determine if host name matches IP address. Used to
 877  confirm a hostname lookup to prevent spoof attacks.
 878 ******************************************************************/
 879
 880 static BOOL matchname(char *remotehost,struct in_addr  addr)
 881 {
 882         struct hostent *hp;
 883         int     i;
 884
 885         if ((hp = sys_gethostbyname(remotehost)) == 0) {
 886                 DEBUG(0,("sys_gethostbyname(%s): lookup failure.\n", remotehost));
 887                 return False;
 888         }
 889
 890         /*
 891          * Make sure that gethostbyname() returns the "correct" host name.
 892          * Unfortunately, gethostbyname("localhost") sometimes yields
 893          * "localhost.domain". Since the latter host name comes from the
 894          * local DNS, we just have to trust it (all bets are off if the local
 895          * DNS is perverted). We always check the address list, though.
 896          */
 897
 898         if (!strequal(remotehost, hp->h_name)
 899             && !strequal(remotehost, "localhost")) {
 900                 DEBUG(0,("host name/name mismatch: %s != %s\n",
 901                          remotehost, hp->h_name));
 902                 return False;
 903         }
 904
 905         /* Look up the host address in the address list we just got. */
 906         for (i = 0; hp->h_addr_list[i]; i++) {
 907                 if (memcmp(hp->h_addr_list[i], (char *) & addr, sizeof(addr)) == 0)
 908                         return True;
 909         }
 910
 911         /*
 912          * The host name does not map to the original host address. Perhaps
 913          * someone has compromised a name server. More likely someone botched
 914          * it, but that could be dangerous, too.
 915          */
 916
 917         DEBUG(0,("host name/address mismatch: %s != %s\n",
 918                  inet_ntoa(addr), hp->h_name));
 919         return False;
 920 }
 921
 922 /*******************************************************************
 923  Return the DNS name of the remote end of a socket.
 924 ******************************************************************/
 925
 926 char *get_peer_name(int fd, BOOL force_lookup)
 927 {
 928         static pstring name_buf;
 929         pstring tmp_name;
 930         static fstring addr_buf;
 931         struct hostent *hp;
 932         struct in_addr addr;
 933         char *p;
 934
 935         /* reverse lookups can be *very* expensive, and in many
 936            situations won't work because many networks don't link dhcp
 937            with dns. To avoid the delay we avoid the lookup if
 938            possible */
 939         if (!lp_hostname_lookups() && (force_lookup == False)) {
 940                 return get_peer_addr(fd);
 941         }
 942
 943         p = get_peer_addr(fd);
 944
 945         /* it might be the same as the last one - save some DNS work */
 946         if (strcmp(p, addr_buf) == 0)
 947                 return name_buf;
 948
 949         pstrcpy(name_buf,"UNKNOWN");
 950         if (fd == -1)
 951                 return name_buf;
 952
 953         fstrcpy(addr_buf, p);
 954
 955         addr = *interpret_addr2(p);
 956
 957         /* Look up the remote host name. */
 958         if ((hp = gethostbyaddr((char *)&addr.s_addr, sizeof(addr.s_addr), AF_INET)) == 0) {
 959                 DEBUG(1,("Gethostbyaddr failed for %s\n",p));
 960                 pstrcpy(name_buf, p);
 961         } else {
 962                 pstrcpy(name_buf,(char *)hp->h_name);
 963                 if (!matchname(name_buf, addr)) {
 964                         DEBUG(0,("Matchname failed on %s %s\n",name_buf,p));
 965                         pstrcpy(name_buf,"UNKNOWN");
 966                 }
 967         }
 968
 969         /* can't pass the same source and dest strings in when you
 970            use --enable-developer or the clobber_region() call will
 971            get you */
 972
 973         pstrcpy( tmp_name, name_buf );
 974         alpha_strcpy(name_buf, tmp_name, "_-.", sizeof(name_buf));
 975         if (strstr(name_buf,"..")) {
 976                 pstrcpy(name_buf, "UNKNOWN");
 977         }
 978
 979         return name_buf;
 980 }
 981
 982 /*******************************************************************
 983  Return the IP addr of the remote end of a socket as a string.
 984  ******************************************************************/
 985
 986 char *get_peer_addr(int fd)
 987 {
 988         struct sockaddr sa;
 989         struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
 990         socklen_t length = sizeof(sa);
 991         static fstring addr_buf;
 992
 993         fstrcpy(addr_buf,"0.0.0.0");
 994
 995         if (fd == -1) {
 996                 return addr_buf;
 997         }
 998
 999         if (getpeername(fd, &sa, &length) < 0) {
1000                 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
1001                 return addr_buf;
1002         }
1003
1004         fstrcpy(addr_buf,(char *)inet_ntoa(sockin->sin_addr));
1005
1006         return addr_buf;
1007 }
1008
1009 /*******************************************************************
1010  Create protected unix domain socket.
1011
1012  Some unixes cannot set permissions on a ux-dom-sock, so we
1013  have to make sure that the directory contains the protection
1014  permissions instead.
1015  ******************************************************************/
1016
1017 int create_pipe_sock(const char *socket_dir,
1018                      const char *socket_name,
1019                      mode_t dir_perms)
1020 {
1021 #ifdef HAVE_UNIXSOCKET
1022         struct sockaddr_un sunaddr;
1023         struct stat st;
1024         int sock;
1025         mode_t old_umask;
1026         pstring path;
1027
1028         old_umask = umask(0);
1029
1030         /* Create the socket directory or reuse the existing one */
1031
1032         if (lstat(socket_dir, &st) == -1) {
1033                 if (errno == ENOENT) {
1034                         /* Create directory */
1035                         if (mkdir(socket_dir, dir_perms) == -1) {
1036                                 DEBUG(0, ("error creating socket directory "
1037                                         "%s: %s\n", socket_dir,
1038                                         strerror(errno)));
1039                                 goto out_umask;
1040                         }
1041                 } else {
1042                         DEBUG(0, ("lstat failed on socket directory %s: %s\n",
1043                                 socket_dir, strerror(errno)));
1044                         goto out_umask;
1045                 }
1046         } else {
1047                 /* Check ownership and permission on existing directory */
1048                 if (!S_ISDIR(st.st_mode)) {
1049                         DEBUG(0, ("socket directory %s isn't a directory\n",
1050                                 socket_dir));
1051                         goto out_umask;
1052                 }
1053                 if ((st.st_uid != sec_initial_uid()) ||
1054                                 ((st.st_mode & 0777) != dir_perms)) {
1055                         DEBUG(0, ("invalid permissions on socket directory "
1056                                 "%s\n", socket_dir));
1057                         goto out_umask;
1058                 }
1059         }
1060
1061         /* Create the socket file */
1062
1063         sock = socket(AF_UNIX, SOCK_STREAM, 0);
1064
1065         if (sock == -1) {
1066                 perror("socket");
1067                 goto out_umask;
1068         }
1069
1070         pstr_sprintf(path, "%s/%s", socket_dir, socket_name);
1071
1072         unlink(path);
1073         memset(&sunaddr, 0, sizeof(sunaddr));
1074         sunaddr.sun_family = AF_UNIX;
1075         safe_strcpy(sunaddr.sun_path, path, sizeof(sunaddr.sun_path)-1);
1076
1077         if (bind(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) == -1) {
1078                 DEBUG(0, ("bind failed on pipe socket %s: %s\n", path,
1079                         strerror(errno)));
1080                 goto out_close;
1081         }
1082
1083         if (listen(sock, 5) == -1) {
1084                 DEBUG(0, ("listen failed on pipe socket %s: %s\n", path,
1085                         strerror(errno)));
1086                 goto out_close;
1087         }
1088
1089         umask(old_umask);
1090         return sock;
1091
1092 out_close:
1093         close(sock);
1094
1095 out_umask:
1096         umask(old_umask);
1097         return -1;
1098
1099 #else
1100         DEBUG(0, ("create_pipe_sock: No Unix sockets on this system\n"));
1101         return -1;
1102 #endif /* HAVE_UNIXSOCKET */
1103 }