source/lib/util_sock.c

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