00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023 #include "includes.h"
00024
00025
00026
00027
00028 static int client_fd = -1;
00029
00030 static char client_ip_string[16];
00031
00032 void client_setfd(int fd)
00033 {
00034 client_fd = fd;
00035 safe_strcpy(client_ip_string, get_peer_addr(client_fd), sizeof(client_ip_string)-1);
00036 }
00037
00038 static char *get_socket_addr(int fd)
00039 {
00040 struct sockaddr sa;
00041 struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
00042 socklen_t length = sizeof(sa);
00043 static fstring addr_buf;
00044
00045 fstrcpy(addr_buf,"0.0.0.0");
00046
00047 if (fd == -1) {
00048 return addr_buf;
00049 }
00050
00051 if (getsockname(fd, &sa, &length) < 0) {
00052 DEBUG(0,("getsockname failed. Error was %s\n", strerror(errno) ));
00053 return addr_buf;
00054 }
00055
00056 fstrcpy(addr_buf,(char *)inet_ntoa(sockin->sin_addr));
00057
00058 return addr_buf;
00059 }
00060
00061 static int get_socket_port(int fd)
00062 {
00063 struct sockaddr sa;
00064 struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
00065 socklen_t length = sizeof(sa);
00066
00067 if (fd == -1)
00068 return -1;
00069
00070 if (getsockname(fd, &sa, &length) < 0) {
00071 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
00072 return -1;
00073 }
00074
00075 return ntohs(sockin->sin_port);
00076 }
00077
00078 char *client_name(void)
00079 {
00080 return get_peer_name(client_fd,False);
00081 }
00082
00083 char *client_addr(void)
00084 {
00085 return get_peer_addr(client_fd);
00086 }
00087
00088 char *client_socket_addr(void)
00089 {
00090 return get_socket_addr(client_fd);
00091 }
00092
00093 int client_socket_port(void)
00094 {
00095 return get_socket_port(client_fd);
00096 }
00097
00098 struct in_addr *client_inaddr(struct sockaddr *sa)
00099 {
00100 struct sockaddr_in *sockin = (struct sockaddr_in *) (sa);
00101 socklen_t length = sizeof(*sa);
00102
00103 if (getpeername(client_fd, sa, &length) < 0) {
00104 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
00105 return NULL;
00106 }
00107
00108 return &sockin->sin_addr;
00109 }
00110
00111
00112 struct in_addr lastip;
00113
00114
00115 int lastport=0;
00116
00117 int smb_read_error = 0;
00118
00119
00120
00121
00122
00123 BOOL is_a_socket(int fd)
00124 {
00125 int v;
00126 socklen_t l;
00127 l = sizeof(int);
00128 return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
00129 }
00130
00131 enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
00132
00133 typedef struct smb_socket_option {
00134 const char *name;
00135 int level;
00136 int option;
00137 int value;
00138 int opttype;
00139 } smb_socket_option;
00140
00141 static const smb_socket_option socket_options[] = {
00142 {"SO_KEEPALIVE", SOL_SOCKET, SO_KEEPALIVE, 0, OPT_BOOL},
00143 {"SO_REUSEADDR", SOL_SOCKET, SO_REUSEADDR, 0, OPT_BOOL},
00144 {"SO_BROADCAST", SOL_SOCKET, SO_BROADCAST, 0, OPT_BOOL},
00145 #ifdef TCP_NODELAY
00146 {"TCP_NODELAY", IPPROTO_TCP, TCP_NODELAY, 0, OPT_BOOL},
00147 #endif
00148 #ifdef TCP_KEEPCNT
00149 {"TCP_KEEPCNT", IPPROTO_TCP, TCP_KEEPCNT, 0, OPT_INT},
00150 #endif
00151 #ifdef TCP_KEEPIDLE
00152 {"TCP_KEEPIDLE", IPPROTO_TCP, TCP_KEEPIDLE, 0, OPT_INT},
00153 #endif
00154 #ifdef TCP_KEEPINTVL
00155 {"TCP_KEEPINTVL", IPPROTO_TCP, TCP_KEEPINTVL, 0, OPT_INT},
00156 #endif
00157 #ifdef IPTOS_LOWDELAY
00158 {"IPTOS_LOWDELAY", IPPROTO_IP, IP_TOS, IPTOS_LOWDELAY, OPT_ON},
00159 #endif
00160 #ifdef IPTOS_THROUGHPUT
00161 {"IPTOS_THROUGHPUT", IPPROTO_IP, IP_TOS, IPTOS_THROUGHPUT, OPT_ON},
00162 #endif
00163 #ifdef SO_REUSEPORT
00164 {"SO_REUSEPORT", SOL_SOCKET, SO_REUSEPORT, 0, OPT_BOOL},
00165 #endif
00166 #ifdef SO_SNDBUF
00167 {"SO_SNDBUF", SOL_SOCKET, SO_SNDBUF, 0, OPT_INT},
00168 #endif
00169 #ifdef SO_RCVBUF
00170 {"SO_RCVBUF", SOL_SOCKET, SO_RCVBUF, 0, OPT_INT},
00171 #endif
00172 #ifdef SO_SNDLOWAT
00173 {"SO_SNDLOWAT", SOL_SOCKET, SO_SNDLOWAT, 0, OPT_INT},
00174 #endif
00175 #ifdef SO_RCVLOWAT
00176 {"SO_RCVLOWAT", SOL_SOCKET, SO_RCVLOWAT, 0, OPT_INT},
00177 #endif
00178 #ifdef SO_SNDTIMEO
00179 {"SO_SNDTIMEO", SOL_SOCKET, SO_SNDTIMEO, 0, OPT_INT},
00180 #endif
00181 #ifdef SO_RCVTIMEO
00182 {"SO_RCVTIMEO", SOL_SOCKET, SO_RCVTIMEO, 0, OPT_INT},
00183 #endif
00184 #ifdef TCP_FASTACK
00185 {"TCP_FASTACK", IPPROTO_TCP, TCP_FASTACK, 0, OPT_INT},
00186 #endif
00187 {NULL,0,0,0,0}};
00188
00189
00190
00191
00192
00193 static void print_socket_options(int s)
00194 {
00195 int value;
00196 socklen_t vlen = 4;
00197 const smb_socket_option *p = &socket_options[0];
00198
00199
00200
00201 if ( DEBUGLEVEL >= 5 ) {
00202 for (; p->name != NULL; p++) {
00203 if (getsockopt(s, p->level, p->option, (void *)&value, &vlen) == -1) {
00204 DEBUG(5,("Could not test socket option %s.\n", p->name));
00205 } else {
00206 DEBUG(5,("socket option %s = %d\n",p->name,value));
00207 }
00208 }
00209 }
00210 }
00211
00212
00213
00214
00215
00216 void set_socket_options(int fd, const char *options)
00217 {
00218 fstring tok;
00219
00220 while (next_token(&options,tok," \t,", sizeof(tok))) {
00221 int ret=0,i;
00222 int value = 1;
00223 char *p;
00224 BOOL got_value = False;
00225
00226 if ((p = strchr_m(tok,'='))) {
00227 *p = 0;
00228 value = atoi(p+1);
00229 got_value = True;
00230 }
00231
00232 for (i=0;socket_options[i].name;i++)
00233 if (strequal(socket_options[i].name,tok))
00234 break;
00235
00236 if (!socket_options[i].name) {
00237 DEBUG(0,("Unknown socket option %s\n",tok));
00238 continue;
00239 }
00240
00241 switch (socket_options[i].opttype) {
00242 case OPT_BOOL:
00243 case OPT_INT:
00244 ret = setsockopt(fd,socket_options[i].level,
00245 socket_options[i].option,(char *)&value,sizeof(int));
00246 break;
00247
00248 case OPT_ON:
00249 if (got_value)
00250 DEBUG(0,("syntax error - %s does not take a value\n",tok));
00251
00252 {
00253 int on = socket_options[i].value;
00254 ret = setsockopt(fd,socket_options[i].level,
00255 socket_options[i].option,(char *)&on,sizeof(int));
00256 }
00257 break;
00258 }
00259
00260 if (ret != 0)
00261 DEBUG(0,("Failed to set socket option %s (Error %s)\n",tok, strerror(errno) ));
00262 }
00263
00264 print_socket_options(fd);
00265 }
00266
00267
00268
00269
00270
00271 ssize_t read_udp_socket(int fd,char *buf,size_t len)
00272 {
00273 ssize_t ret;
00274 struct sockaddr_in sock;
00275 socklen_t socklen = sizeof(sock);
00276
00277 memset((char *)&sock,'\0',socklen);
00278 memset((char *)&lastip,'\0',sizeof(lastip));
00279 ret = (ssize_t)sys_recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
00280 if (ret <= 0) {
00281
00282 if (errno == EAGAIN) {
00283 DEBUG(10,("read socket returned EAGAIN. ERRNO=%s\n",strerror(errno)));
00284 } else {
00285 DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
00286 }
00287 return(0);
00288 }
00289
00290 lastip = sock.sin_addr;
00291 lastport = ntohs(sock.sin_port);
00292
00293 DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %lu\n",
00294 inet_ntoa(lastip), lastport, (unsigned long)ret));
00295
00296 return(ret);
00297 }
00298
00299 #if 0
00300
00301 Socket routines from HEAD - maybe re-enable in future. JRA.
00302
00303
00304
00305
00306
00307 static BOOL timeout_until(struct timeval *timeout, const struct timeval *endtime)
00308 {
00309 struct timeval now;
00310 SMB_BIG_INT t_dif;
00311
00312 GetTimeOfDay(&now);
00313
00314 t_dif = usec_time_diff(endtime, &now);
00315 if (t_dif <= 0) {
00316 return False;
00317 }
00318
00319 timeout->tv_sec = (t_dif / (SMB_BIG_INT)1000000);
00320 timeout->tv_usec = (t_dif % (SMB_BIG_INT)1000000);
00321 return True;
00322 }
00323
00324
00325
00326
00327
00328
00329 ssize_t read_data_until(int fd,char *buffer,size_t N, const struct timeval *endtime)
00330 {
00331 ssize_t ret;
00332 size_t total=0;
00333
00334 smb_read_error = 0;
00335
00336 while (total < N) {
00337
00338 if (endtime != NULL) {
00339 fd_set r_fds;
00340 struct timeval timeout;
00341 int selrtn;
00342
00343 if (!timeout_until(&timeout, endtime)) {
00344 DEBUG(10,("read_data_until: read timed out\n"));
00345 smb_read_error = READ_TIMEOUT;
00346 return -1;
00347 }
00348
00349 FD_ZERO(&r_fds);
00350 FD_SET(fd, &r_fds);
00351
00352
00353 selrtn = sys_select_intr(fd+1, &r_fds, NULL, NULL, &timeout);
00354 if (selrtn == -1) {
00355
00356 DEBUG(0,("read_data_until: select error = %s.\n", strerror(errno) ));
00357 smb_read_error = READ_ERROR;
00358 return -1;
00359 }
00360
00361
00362 if (selrtn == 0) {
00363 DEBUG(10,("read_data_until: select timed out.\n"));
00364 smb_read_error = READ_TIMEOUT;
00365 return -1;
00366 }
00367 }
00368
00369 ret = sys_read(fd,buffer + total,N - total);
00370
00371 if (ret == 0) {
00372 DEBUG(10,("read_data_until: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
00373 smb_read_error = READ_EOF;
00374 return 0;
00375 }
00376
00377 if (ret == -1) {
00378 if (errno == EAGAIN) {
00379
00380 continue;
00381 }
00382 DEBUG(0,("read_data_until: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
00383 smb_read_error = READ_ERROR;
00384 return -1;
00385 }
00386 total += ret;
00387 }
00388 return (ssize_t)total;
00389 }
00390 #endif
00391
00392
00393
00394
00395
00396
00397
00398
00399 ssize_t read_socket_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
00400 {
00401 fd_set fds;
00402 int selrtn;
00403 ssize_t readret;
00404 size_t nread = 0;
00405 struct timeval timeout;
00406
00407
00408 if (maxcnt <= 0)
00409 return(0);
00410
00411 smb_read_error = 0;
00412
00413
00414 if (time_out == 0) {
00415 if (mincnt == 0) {
00416 mincnt = maxcnt;
00417 }
00418
00419 while (nread < mincnt) {
00420 readret = sys_read(fd, buf + nread, maxcnt - nread);
00421
00422 if (readret == 0) {
00423 DEBUG(5,("read_socket_with_timeout: blocking read. EOF from client.\n"));
00424 smb_read_error = READ_EOF;
00425 return -1;
00426 }
00427
00428 if (readret == -1) {
00429 if (fd == client_fd) {
00430
00431 DEBUG(0,("read_socket_with_timeout: client %s read error = %s.\n",
00432 client_ip_string, strerror(errno) ));
00433 } else {
00434 DEBUG(0,("read_socket_with_timeout: read error = %s.\n", strerror(errno) ));
00435 }
00436 smb_read_error = READ_ERROR;
00437 return -1;
00438 }
00439 nread += readret;
00440 }
00441 return((ssize_t)nread);
00442 }
00443
00444
00445
00446
00447
00448
00449
00450
00451 timeout.tv_sec = (time_t)(time_out / 1000);
00452 timeout.tv_usec = (long)(1000 * (time_out % 1000));
00453
00454 for (nread=0; nread < mincnt; ) {
00455 FD_ZERO(&fds);
00456 FD_SET(fd,&fds);
00457
00458 selrtn = sys_select_intr(fd+1,&fds,NULL,NULL,&timeout);
00459
00460
00461 if (selrtn == -1) {
00462
00463 if (fd == client_fd) {
00464
00465 DEBUG(0,("read_socket_with_timeout: timeout read for client %s. select error = %s.\n",
00466 client_ip_string, strerror(errno) ));
00467 } else {
00468 DEBUG(0,("read_socket_with_timeout: timeout read. select error = %s.\n", strerror(errno) ));
00469 }
00470 smb_read_error = READ_ERROR;
00471 return -1;
00472 }
00473
00474
00475 if (selrtn == 0) {
00476 DEBUG(10,("read_socket_with_timeout: timeout read. select timed out.\n"));
00477 smb_read_error = READ_TIMEOUT;
00478 return -1;
00479 }
00480
00481 readret = sys_read(fd, buf+nread, maxcnt-nread);
00482
00483 if (readret == 0) {
00484
00485 DEBUG(5,("read_socket_with_timeout: timeout read. EOF from client.\n"));
00486 smb_read_error = READ_EOF;
00487 return -1;
00488 }
00489
00490 if (readret == -1) {
00491
00492 if (fd == client_fd) {
00493
00494 DEBUG(0,("read_socket_with_timeout: timeout read to client %s. read error = %s.\n",
00495 client_ip_string, strerror(errno) ));
00496 } else {
00497 DEBUG(0,("read_socket_with_timeout: timeout read. read error = %s.\n", strerror(errno) ));
00498 }
00499 smb_read_error = READ_ERROR;
00500 return -1;
00501 }
00502
00503 nread += readret;
00504 }
00505
00506
00507 return (ssize_t)nread;
00508 }
00509
00510
00511
00512
00513
00514 ssize_t read_data(int fd,char *buffer,size_t N)
00515 {
00516 ssize_t ret;
00517 size_t total=0;
00518
00519 smb_read_error = 0;
00520
00521 while (total < N) {
00522 ret = sys_read(fd,buffer + total,N - total);
00523
00524 if (ret == 0) {
00525 DEBUG(10,("read_data: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
00526 smb_read_error = READ_EOF;
00527 return 0;
00528 }
00529
00530 if (ret == -1) {
00531 if (fd == client_fd) {
00532
00533 DEBUG(0,("read_data: read failure for %d bytes to client %s. Error = %s\n",
00534 (int)(N - total), client_ip_string, strerror(errno) ));
00535 } else {
00536 DEBUG(0,("read_data: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
00537 }
00538 smb_read_error = READ_ERROR;
00539 return -1;
00540 }
00541 total += ret;
00542 }
00543 return (ssize_t)total;
00544 }
00545
00546
00547
00548
00549
00550 ssize_t write_data(int fd, const char *buffer, size_t N)
00551 {
00552 size_t total=0;
00553 ssize_t ret;
00554
00555 while (total < N) {
00556 ret = sys_write(fd,buffer + total,N - total);
00557
00558 if (ret == -1) {
00559 if (fd == client_fd) {
00560
00561 DEBUG(0,("write_data: write failure in writing to client %s. Error %s\n",
00562 client_ip_string, strerror(errno) ));
00563 } else {
00564 DEBUG(0,("write_data: write failure. Error = %s\n", strerror(errno) ));
00565 }
00566 return -1;
00567 }
00568
00569 if (ret == 0) {
00570 return total;
00571 }
00572
00573 total += ret;
00574 }
00575 return (ssize_t)total;
00576 }
00577
00578
00579
00580
00581
00582 BOOL send_keepalive(int client)
00583 {
00584 unsigned char buf[4];
00585
00586 buf[0] = SMBkeepalive;
00587 buf[1] = buf[2] = buf[3] = 0;
00588
00589 return(write_data(client,(char *)buf,4) == 4);
00590 }
00591
00592
00593
00594
00595
00596
00597
00598
00599
00600
00601 static ssize_t read_smb_length_return_keepalive(int fd, char *inbuf, unsigned int timeout)
00602 {
00603 ssize_t len=0;
00604 int msg_type;
00605 BOOL ok = False;
00606
00607 while (!ok) {
00608 if (timeout > 0)
00609 ok = (read_socket_with_timeout(fd,inbuf,4,4,timeout) == 4);
00610 else
00611 ok = (read_data(fd,inbuf,4) == 4);
00612
00613 if (!ok)
00614 return(-1);
00615
00616 len = smb_len(inbuf);
00617 msg_type = CVAL(inbuf,0);
00618
00619 if (msg_type == SMBkeepalive)
00620 DEBUG(5,("Got keepalive packet\n"));
00621 }
00622
00623 DEBUG(10,("got smb length of %lu\n",(unsigned long)len));
00624
00625 return(len);
00626 }
00627
00628
00629
00630
00631
00632
00633
00634
00635 ssize_t read_smb_length(int fd, char *inbuf, unsigned int timeout)
00636 {
00637 ssize_t len;
00638
00639 for(;;) {
00640 len = read_smb_length_return_keepalive(fd, inbuf, timeout);
00641
00642 if(len < 0)
00643 return len;
00644
00645
00646 if(CVAL(inbuf,0) != SMBkeepalive)
00647 break;
00648 }
00649
00650 DEBUG(10,("read_smb_length: got smb length of %lu\n",
00651 (unsigned long)len));
00652
00653 return len;
00654 }
00655
00656
00657
00658
00659
00660
00661
00662
00663 BOOL receive_smb_raw(int fd, char *buffer, size_t buflen, unsigned int timeout)
00664 {
00665 ssize_t len,ret;
00666
00667 smb_read_error = 0;
00668
00669 len = read_smb_length_return_keepalive(fd,buffer,timeout);
00670 if (len < 0) {
00671 DEBUG(10,("receive_smb_raw: length < 0!\n"));
00672
00673
00674
00675
00676
00677
00678
00679 if (smb_read_error == 0)
00680 smb_read_error = READ_ERROR;
00681 return False;
00682 }
00683
00684 if (len > buflen) {
00685 DEBUG(0,("Invalid packet length! (%lu bytes).\n",(unsigned long)len));
00686
00687
00688
00689
00690
00691
00692
00693 if (smb_read_error == 0)
00694 smb_read_error = READ_ERROR;
00695 return False;
00696 }
00697
00698 if(len > 0) {
00699 if (timeout > 0) {
00700 ret = read_socket_with_timeout(fd,buffer+4,len,len,timeout);
00701 } else {
00702 ret = read_data(fd,buffer+4,len);
00703 }
00704
00705 if (ret != len) {
00706 if (smb_read_error == 0) {
00707 smb_read_error = READ_ERROR;
00708 }
00709 return False;
00710 }
00711
00712
00713
00714 SSVAL(buffer+4,len, 0);
00715 }
00716
00717 return True;
00718 }
00719
00720
00721
00722
00723
00724
00725 BOOL receive_smb(int fd, char *buffer, size_t buflen, unsigned int timeout)
00726 {
00727 if (!receive_smb_raw(fd, buffer, buflen, timeout)) {
00728 return False;
00729 }
00730
00731
00732 if (!srv_check_sign_mac(buffer, True)) {
00733 DEBUG(0, ("receive_smb: SMB Signature verification failed on incoming packet!\n"));
00734 if (smb_read_error == 0)
00735 smb_read_error = READ_BAD_SIG;
00736 return False;
00737 };
00738
00739 return(True);
00740 }
00741
00742
00743
00744
00745
00746 BOOL send_smb(int fd, char *buffer)
00747 {
00748 size_t len;
00749 size_t nwritten=0;
00750 ssize_t ret;
00751
00752
00753 srv_calculate_sign_mac(buffer);
00754
00755 len = smb_len(buffer) + 4;
00756
00757 while (nwritten < len) {
00758 ret = write_data(fd,buffer+nwritten,len - nwritten);
00759 if (ret <= 0) {
00760 DEBUG(0,("Error writing %d bytes to client. %d. (%s)\n",
00761 (int)len,(int)ret, strerror(errno) ));
00762 return False;
00763 }
00764 nwritten += ret;
00765 }
00766
00767 return True;
00768 }
00769
00770
00771
00772
00773
00774 int open_socket_in( int type, int port, int dlevel, uint32 socket_addr, BOOL rebind )
00775 {
00776 struct sockaddr_in sock;
00777 int res;
00778
00779 memset( (char *)&sock, '\0', sizeof(sock) );
00780
00781 #ifdef HAVE_SOCK_SIN_LEN
00782 sock.sin_len = sizeof(sock);
00783 #endif
00784 sock.sin_port = htons( port );
00785 sock.sin_family = AF_INET;
00786 sock.sin_addr.s_addr = socket_addr;
00787
00788 res = socket( AF_INET, type, 0 );
00789 if( res == -1 ) {
00790 if( DEBUGLVL(0) ) {
00791 dbgtext( "open_socket_in(): socket() call failed: " );
00792 dbgtext( "%s\n", strerror( errno ) );
00793 }
00794 return -1;
00795 }
00796
00797
00798 {
00799 int val = rebind ? 1 : 0;
00800 if( setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val)) == -1 ) {
00801 if( DEBUGLVL( dlevel ) ) {
00802 dbgtext( "open_socket_in(): setsockopt: " );
00803 dbgtext( "SO_REUSEADDR = %s ", val?"True":"False" );
00804 dbgtext( "on port %d failed ", port );
00805 dbgtext( "with error = %s\n", strerror(errno) );
00806 }
00807 }
00808 #ifdef SO_REUSEPORT
00809 if( setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val)) == -1 ) {
00810 if( DEBUGLVL( dlevel ) ) {
00811 dbgtext( "open_socket_in(): setsockopt: ");
00812 dbgtext( "SO_REUSEPORT = %s ", val?"True":"False" );
00813 dbgtext( "on port %d failed ", port );
00814 dbgtext( "with error = %s\n", strerror(errno) );
00815 }
00816 }
00817 #endif
00818 }
00819
00820
00821 if( bind( res, (struct sockaddr *)&sock, sizeof(sock) ) == -1 ) {
00822 if( DEBUGLVL(dlevel) && (port == SMB_PORT1 || port == SMB_PORT2 || port == NMB_PORT) ) {
00823 dbgtext( "bind failed on port %d ", port );
00824 dbgtext( "socket_addr = %s.\n", inet_ntoa( sock.sin_addr ) );
00825 dbgtext( "Error = %s\n", strerror(errno) );
00826 }
00827 close( res );
00828 return( -1 );
00829 }
00830
00831 DEBUG( 10, ( "bind succeeded on port %d\n", port ) );
00832
00833 return( res );
00834 }
00835
00836
00837
00838
00839
00840 int open_socket_out(int type, struct in_addr *addr, int port ,int timeout)
00841 {
00842 struct sockaddr_in sock_out;
00843 int res,ret;
00844 int connect_loop = 10;
00845 int increment = 10;
00846
00847
00848 res = socket(PF_INET, type, 0);
00849 if (res == -1) {
00850 DEBUG(0,("socket error (%s)\n", strerror(errno)));
00851 return -1;
00852 }
00853
00854 if (type != SOCK_STREAM)
00855 return(res);
00856
00857 memset((char *)&sock_out,'\0',sizeof(sock_out));
00858 putip((char *)&sock_out.sin_addr,(char *)addr);
00859
00860 sock_out.sin_port = htons( port );
00861 sock_out.sin_family = PF_INET;
00862
00863
00864 set_blocking(res,False);
00865
00866 DEBUG(3,("Connecting to %s at port %d\n",inet_ntoa(*addr),port));
00867
00868
00869 connect_again:
00870
00871 ret = connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out));
00872
00873
00874 if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
00875 errno == EAGAIN) && (connect_loop < timeout) ) {
00876 smb_msleep(connect_loop);
00877 timeout -= connect_loop;
00878 connect_loop += increment;
00879 if (increment < 250) {
00880
00881 increment *= 1.5;
00882 }
00883 goto connect_again;
00884 }
00885
00886 if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
00887 errno == EAGAIN)) {
00888 DEBUG(1,("timeout connecting to %s:%d\n",inet_ntoa(*addr),port));
00889 close(res);
00890 return -1;
00891 }
00892
00893 #ifdef EISCONN
00894
00895 if (ret < 0 && errno == EISCONN) {
00896 errno = 0;
00897 ret = 0;
00898 }
00899 #endif
00900
00901 if (ret < 0) {
00902 DEBUG(2,("error connecting to %s:%d (%s)\n",
00903 inet_ntoa(*addr),port,strerror(errno)));
00904 close(res);
00905 return -1;
00906 }
00907
00908
00909 set_blocking(res,True);
00910
00911 return res;
00912 }
00913
00914
00915
00916
00917
00918
00919
00920 BOOL open_any_socket_out(struct sockaddr_in *addrs, int num_addrs,
00921 int timeout, int *fd_index, int *fd)
00922 {
00923 int i, resulting_index, res;
00924 int *sockets;
00925 BOOL good_connect;
00926
00927 fd_set r_fds, wr_fds;
00928 struct timeval tv;
00929 int maxfd;
00930
00931 int connect_loop = 10000;
00932
00933 timeout *= 1000;
00934
00935 sockets = SMB_MALLOC_ARRAY(int, num_addrs);
00936
00937 if (sockets == NULL)
00938 return False;
00939
00940 resulting_index = -1;
00941
00942 for (i=0; i<num_addrs; i++)
00943 sockets[i] = -1;
00944
00945 for (i=0; i<num_addrs; i++) {
00946 sockets[i] = socket(PF_INET, SOCK_STREAM, 0);
00947 if (sockets[i] < 0)
00948 goto done;
00949 set_blocking(sockets[i], False);
00950 }
00951
00952 connect_again:
00953 good_connect = False;
00954
00955 for (i=0; i<num_addrs; i++) {
00956
00957 if (sockets[i] == -1)
00958 continue;
00959
00960 if (connect(sockets[i], (struct sockaddr *)&(addrs[i]),
00961 sizeof(*addrs)) == 0) {
00962
00963
00964 resulting_index = i;
00965 goto done;
00966 }
00967
00968 if (errno == EINPROGRESS || errno == EALREADY ||
00969 #ifdef EISCONN
00970 errno == EISCONN ||
00971 #endif
00972 errno == EAGAIN || errno == EINTR) {
00973
00974
00975 good_connect = True;
00976 } else if (errno != 0) {
00977
00978 close(sockets[i]);
00979 sockets[i] = -1;
00980 }
00981 }
00982
00983 if (!good_connect) {
00984
00985 goto done;
00986 }
00987
00988
00989
00990 maxfd = 0;
00991 FD_ZERO(&wr_fds);
00992 FD_ZERO(&r_fds);
00993
00994 for (i=0; i<num_addrs; i++) {
00995 if (sockets[i] == -1)
00996 continue;
00997 FD_SET(sockets[i], &wr_fds);
00998 FD_SET(sockets[i], &r_fds);
00999 if (sockets[i]>maxfd)
01000 maxfd = sockets[i];
01001 }
01002
01003 tv.tv_sec = 0;
01004 tv.tv_usec = connect_loop;
01005
01006 res = sys_select_intr(maxfd+1, &r_fds, &wr_fds, NULL, &tv);
01007
01008 if (res < 0)
01009 goto done;
01010
01011 if (res == 0)
01012 goto next_round;
01013
01014 for (i=0; i<num_addrs; i++) {
01015
01016 if (sockets[i] == -1)
01017 continue;
01018
01019
01020
01021
01022
01023 if (FD_ISSET(sockets[i], &r_fds) &&
01024 FD_ISSET(sockets[i], &wr_fds)) {
01025
01026 close(sockets[i]);
01027 sockets[i] = -1;
01028 continue;
01029 }
01030
01031 if (!FD_ISSET(sockets[i], &r_fds) &&
01032 FD_ISSET(sockets[i], &wr_fds)) {
01033
01034 resulting_index = i;
01035 goto done;
01036 }
01037 }
01038
01039 next_round:
01040
01041 timeout -= connect_loop;
01042 if (timeout <= 0)
01043 goto done;
01044 connect_loop *= 1.5;
01045 if (connect_loop > timeout)
01046 connect_loop = timeout;
01047 goto connect_again;
01048
01049 done:
01050 for (i=0; i<num_addrs; i++) {
01051 if (i == resulting_index)
01052 continue;
01053 if (sockets[i] >= 0)
01054 close(sockets[i]);
01055 }
01056
01057 if (resulting_index >= 0) {
01058 *fd_index = resulting_index;
01059 *fd = sockets[*fd_index];
01060 set_blocking(*fd, True);
01061 }
01062
01063 free(sockets);
01064
01065 return (resulting_index >= 0);
01066 }
01067
01068
01069
01070
01071 int open_udp_socket(const char *host, int port)
01072 {
01073 int type = SOCK_DGRAM;
01074 struct sockaddr_in sock_out;
01075 int res;
01076 struct in_addr *addr;
01077
01078 addr = interpret_addr2(host);
01079
01080 res = socket(PF_INET, type, 0);
01081 if (res == -1) {
01082 return -1;
01083 }
01084
01085 memset((char *)&sock_out,'\0',sizeof(sock_out));
01086 putip((char *)&sock_out.sin_addr,(char *)addr);
01087 sock_out.sin_port = htons(port);
01088 sock_out.sin_family = PF_INET;
01089
01090 if (connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out))) {
01091 close(res);
01092 return -1;
01093 }
01094
01095 return res;
01096 }
01097
01098
01099
01100
01101
01102
01103
01104 static BOOL matchname(char *remotehost,struct in_addr addr)
01105 {
01106 struct hostent *hp;
01107 int i;
01108
01109 if ((hp = sys_gethostbyname(remotehost)) == 0) {
01110 DEBUG(0,("sys_gethostbyname(%s): lookup failure.\n", remotehost));
01111 return False;
01112 }
01113
01114
01115
01116
01117
01118
01119
01120
01121
01122 if (!strequal(remotehost, hp->h_name)
01123 && !strequal(remotehost, "localhost")) {
01124 DEBUG(0,("host name/name mismatch: %s != %s\n",
01125 remotehost, hp->h_name));
01126 return False;
01127 }
01128
01129
01130 for (i = 0; hp->h_addr_list[i]; i++) {
01131 if (memcmp(hp->h_addr_list[i], (char *) & addr, sizeof(addr)) == 0)
01132 return True;
01133 }
01134
01135
01136
01137
01138
01139
01140
01141 DEBUG(0,("host name/address mismatch: %s != %s\n",
01142 inet_ntoa(addr), hp->h_name));
01143 return False;
01144 }
01145
01146
01147
01148
01149
01150 char *get_peer_name(int fd, BOOL force_lookup)
01151 {
01152 static pstring name_buf;
01153 pstring tmp_name;
01154 static fstring addr_buf;
01155 struct hostent *hp;
01156 struct in_addr addr;
01157 char *p;
01158
01159
01160
01161
01162
01163 if (!lp_hostname_lookups() && (force_lookup == False)) {
01164 return get_peer_addr(fd);
01165 }
01166
01167 p = get_peer_addr(fd);
01168
01169
01170 if (strcmp(p, addr_buf) == 0)
01171 return name_buf;
01172
01173 pstrcpy(name_buf,"UNKNOWN");
01174 if (fd == -1)
01175 return name_buf;
01176
01177 fstrcpy(addr_buf, p);
01178
01179 addr = *interpret_addr2(p);
01180
01181
01182 if ((hp = gethostbyaddr((char *)&addr.s_addr, sizeof(addr.s_addr), AF_INET)) == 0) {
01183 DEBUG(1,("Gethostbyaddr failed for %s\n",p));
01184 pstrcpy(name_buf, p);
01185 } else {
01186 pstrcpy(name_buf,(char *)hp->h_name);
01187 if (!matchname(name_buf, addr)) {
01188 DEBUG(0,("Matchname failed on %s %s\n",name_buf,p));
01189 pstrcpy(name_buf,"UNKNOWN");
01190 }
01191 }
01192
01193
01194
01195
01196
01197 pstrcpy( tmp_name, name_buf );
01198 alpha_strcpy(name_buf, tmp_name, "_-.", sizeof(name_buf));
01199 if (strstr(name_buf,"..")) {
01200 pstrcpy(name_buf, "UNKNOWN");
01201 }
01202
01203 return name_buf;
01204 }
01205
01206
01207
01208
01209
01210 char *get_peer_addr(int fd)
01211 {
01212 struct sockaddr sa;
01213 struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
01214 socklen_t length = sizeof(sa);
01215 static fstring addr_buf;
01216
01217 fstrcpy(addr_buf,"0.0.0.0");
01218
01219 if (fd == -1) {
01220 return addr_buf;
01221 }
01222
01223 if (getpeername(fd, &sa, &length) < 0) {
01224 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
01225 return addr_buf;
01226 }
01227
01228 fstrcpy(addr_buf,(char *)inet_ntoa(sockin->sin_addr));
01229
01230 return addr_buf;
01231 }
01232
01233
01234
01235
01236
01237
01238
01239
01240
01241 int create_pipe_sock(const char *socket_dir,
01242 const char *socket_name,
01243 mode_t dir_perms)
01244 {
01245 #ifdef HAVE_UNIXSOCKET
01246 struct sockaddr_un sunaddr;
01247 struct stat st;
01248 int sock;
01249 mode_t old_umask;
01250 pstring path;
01251
01252 old_umask = umask(0);
01253
01254
01255
01256 if (lstat(socket_dir, &st) == -1) {
01257 if (errno == ENOENT) {
01258
01259 if (mkdir(socket_dir, dir_perms) == -1) {
01260 DEBUG(0, ("error creating socket directory "
01261 "%s: %s\n", socket_dir,
01262 strerror(errno)));
01263 goto out_umask;
01264 }
01265 } else {
01266 DEBUG(0, ("lstat failed on socket directory %s: %s\n",
01267 socket_dir, strerror(errno)));
01268 goto out_umask;
01269 }
01270 } else {
01271
01272 if (!S_ISDIR(st.st_mode)) {
01273 DEBUG(0, ("socket directory %s isn't a directory\n",
01274 socket_dir));
01275 goto out_umask;
01276 }
01277 if ((st.st_uid != sec_initial_uid()) ||
01278 ((st.st_mode & 0777) != dir_perms)) {
01279 DEBUG(0, ("invalid permissions on socket directory "
01280 "%s\n", socket_dir));
01281 goto out_umask;
01282 }
01283 }
01284
01285
01286
01287 sock = socket(AF_UNIX, SOCK_STREAM, 0);
01288
01289 if (sock == -1) {
01290 perror("socket");
01291 goto out_umask;
01292 }
01293
01294 pstr_sprintf(path, "%s/%s", socket_dir, socket_name);
01295
01296 unlink(path);
01297 memset(&sunaddr, 0, sizeof(sunaddr));
01298 sunaddr.sun_family = AF_UNIX;
01299 safe_strcpy(sunaddr.sun_path, path, sizeof(sunaddr.sun_path)-1);
01300
01301 if (bind(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) == -1) {
01302 DEBUG(0, ("bind failed on pipe socket %s: %s\n", path,
01303 strerror(errno)));
01304 goto out_close;
01305 }
01306
01307 if (listen(sock, 5) == -1) {
01308 DEBUG(0, ("listen failed on pipe socket %s: %s\n", path,
01309 strerror(errno)));
01310 goto out_close;
01311 }
01312
01313 umask(old_umask);
01314 return sock;
01315
01316 out_close:
01317 close(sock);
01318
01319 out_umask:
01320 umask(old_umask);
01321 return -1;
01322
01323 #else
01324 DEBUG(0, ("create_pipe_sock: No Unix sockets on this system\n"));
01325 return -1;
01326 #endif
01327 }