]> git.ipfire.org Git - thirdparty/samba.git/blob
projects / thirdparty / samba.git / blob
? search:


1444981
[thirdparty/samba.git] /
1 /*
2 Unix SMB/Netbios implementation.
3 Version 1.9.
4 Samba utility functions
5 Copyright (C) Andrew Tridgell 1992-1998
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 #ifdef WITH_SSL
25 #include <ssl.h>
26 #undef Realloc /* SSLeay defines this and samba has a function of this name */
27 extern SSL *ssl;
28 extern int sslFd;
29 #endif /* WITH_SSL */
30
31 extern int DEBUGLEVEL;
32
33 /* the last IP received from */
34 struct in_addr lastip;
35
36 /* the last port received from */
37 int lastport=0;
38
39 int smb_read_error = 0;
40
41 /****************************************************************************
42 Determine if a file descriptor is in fact a socket.
43 ****************************************************************************/
44
45 BOOL is_a_socket(int fd)
46 {
47 int v,l;
48 l = sizeof(int);
49 return(getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0);
50 }
51
52 enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
53
54 typedef struct smb_socket_option
55 {
56 char *name;
57 int level;
58 int option;
59 int value;
60 int opttype;
61 } smb_socket_option;
62
63 smb_socket_option socket_options[] = {
64 {"SO_KEEPALIVE", SOL_SOCKET, SO_KEEPALIVE, 0, OPT_BOOL},
65 {"SO_REUSEADDR", SOL_SOCKET, SO_REUSEADDR, 0, OPT_BOOL},
66 {"SO_BROADCAST", SOL_SOCKET, SO_BROADCAST, 0, OPT_BOOL},
67 #ifdef TCP_NODELAY
68 {"TCP_NODELAY", IPPROTO_TCP, TCP_NODELAY, 0, OPT_BOOL},
69 #endif
70 #ifdef IPTOS_LOWDELAY
71 {"IPTOS_LOWDELAY", IPPROTO_IP, IP_TOS, IPTOS_LOWDELAY, OPT_ON},
72 #endif
73 #ifdef IPTOS_THROUGHPUT
74 {"IPTOS_THROUGHPUT", IPPROTO_IP, IP_TOS, IPTOS_THROUGHPUT, OPT_ON},
75 #endif
76 #ifdef SO_REUSEPORT
77 {"SO_REUSEPORT", SOL_SOCKET, SO_REUSEPORT, 0, OPT_BOOL},
78 #endif
79 #ifdef SO_SNDBUF
80 {"SO_SNDBUF", SOL_SOCKET, SO_SNDBUF, 0, OPT_INT},
81 #endif
82 #ifdef SO_RCVBUF
83 {"SO_RCVBUF", SOL_SOCKET, SO_RCVBUF, 0, OPT_INT},
84 #endif
85 #ifdef SO_SNDLOWAT
86 {"SO_SNDLOWAT", SOL_SOCKET, SO_SNDLOWAT, 0, OPT_INT},
87 #endif
88 #ifdef SO_RCVLOWAT
89 {"SO_RCVLOWAT", SOL_SOCKET, SO_RCVLOWAT, 0, OPT_INT},
90 #endif
91 #ifdef SO_SNDTIMEO
92 {"SO_SNDTIMEO", SOL_SOCKET, SO_SNDTIMEO, 0, OPT_INT},
93 #endif
94 #ifdef SO_RCVTIMEO
95 {"SO_RCVTIMEO", SOL_SOCKET, SO_RCVTIMEO, 0, OPT_INT},
96 #endif
97 {NULL,0,0,0,0}};
98
99 /****************************************************************************
100 Print socket options.
101 ****************************************************************************/
102 static void print_socket_options(int s)
103 {
104 int value, vlen = 4;
105 smb_socket_option *p = &socket_options[0];
106
107 for (; p->name != NULL; p++) {
108 if (getsockopt(s, p->level, p->option, (void *)&value, &vlen) == -1) {
109 DEBUG(5,("Could not test socket option %s.\n", p->name));
110 } else {
111 DEBUG(5,("socket option %s = %d\n",p->name,value));
112 }
113 }
114 }
115
116 /****************************************************************************
117 Set user socket options.
118 ****************************************************************************/
119
120 void set_socket_options(int fd, char *options)
121 {
122 fstring tok;
123
124 while (next_token(&options,tok," \t,", sizeof(tok))) {
125 int ret=0,i;
126 int value = 1;
127 char *p;
128 BOOL got_value = False;
129
130 if ((p = strchr(tok,'='))) {
131 *p = 0;
132 value = atoi(p+1);
133 got_value = True;
134 }
135
136 for (i=0;socket_options[i].name;i++)
137 if (strequal(socket_options[i].name,tok))
138 break;
139
140 if (!socket_options[i].name) {
141 DEBUG(0,("Unknown socket option %s\n",tok));
142 continue;
143 }
144
145 switch (socket_options[i].opttype) {
146 case OPT_BOOL:
147 case OPT_INT:
148 ret = setsockopt(fd,socket_options[i].level,
149 socket_options[i].option,(char *)&value,sizeof(int));
150 break;
151
152 case OPT_ON:
153 if (got_value)
154 DEBUG(0,("syntax error - %s does not take a value\n",tok));
155
156 {
157 int on = socket_options[i].value;
158 ret = setsockopt(fd,socket_options[i].level,
159 socket_options[i].option,(char *)&on,sizeof(int));
160 }
161 break;
162 }
163
164 if (ret != 0)
165 DEBUG(0,("Failed to set socket option %s (Error %s)\n",tok, strerror(errno) ));
166 }
167
168 print_socket_options(fd);
169 }
170
171 /****************************************************************************
172 Read from a socket.
173 ****************************************************************************/
174
175 ssize_t read_udp_socket(int fd,char *buf,size_t len)
176 {
177 ssize_t ret;
178 struct sockaddr_in sock;
179 int socklen;
180
181 socklen = sizeof(sock);
182 memset((char *)&sock,'\0',socklen);
183 memset((char *)&lastip,'\0',sizeof(lastip));
184 ret = (ssize_t)recvfrom(fd,buf,len,0,(struct sockaddr *)&sock,&socklen);
185 if (ret <= 0) {
186 DEBUG(2,("read socket failed. ERRNO=%s\n",strerror(errno)));
187 return(0);
188 }
189
190 lastip = sock.sin_addr;
191 lastport = ntohs(sock.sin_port);
192
193 DEBUG(10,("read_udp_socket: lastip %s lastport %d read: %d\n",
194 inet_ntoa(lastip), lastport, ret));
195
196 return(ret);
197 }
198
199 /****************************************************************************
200 Read data from a socket with a timout in msec.
201 mincount = if timeout, minimum to read before returning
202 maxcount = number to be read.
203 time_out = timeout in milliseconds
204 ****************************************************************************/
205
206 static ssize_t read_socket_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
207 {
208 fd_set fds;
209 int selrtn;
210 ssize_t readret;
211 size_t nread = 0;
212 struct timeval timeout;
213
214 /* just checking .... */
215 if (maxcnt <= 0)
216 return(0);
217
218 smb_read_error = 0;
219
220 /* Blocking read */
221 if (time_out <= 0) {
222 if (mincnt == 0) mincnt = maxcnt;
223
224 while (nread < mincnt) {
225 #ifdef WITH_SSL
226 if(fd == sslFd){
227 readret = SSL_read(ssl, buf + nread, maxcnt - nread);
228 }else{
229 readret = read(fd, buf + nread, maxcnt - nread);
230 }
231 #else /* WITH_SSL */
232 readret = read(fd, buf + nread, maxcnt - nread);
233 #endif /* WITH_SSL */
234
235 if (readret == 0) {
236 DEBUG(5,("read_socket_with_timeout: blocking read. EOF from client.\n"));
237 smb_read_error = READ_EOF;
238 return -1;
239 }
240
241 if (readret == -1) {
242 DEBUG(0,("read_socket_with_timeout: read error = %s.\n", strerror(errno) ));
243 smb_read_error = READ_ERROR;
244 return -1;
245 }
246 nread += readret;
247 }
248 return((ssize_t)nread);
249 }
250
251 /* Most difficult - timeout read */
252 /* If this is ever called on a disk file and
253 mincnt is greater then the filesize then
254 system performance will suffer severely as
255 select always returns true on disk files */
256
257 /* Set initial timeout */
258 timeout.tv_sec = (time_t)(time_out / 1000);
259 timeout.tv_usec = (long)(1000 * (time_out % 1000));
260
261 for (nread=0; nread < mincnt; ) {
262 FD_ZERO(&fds);
263 FD_SET(fd,&fds);
264
265 selrtn = sys_select_intr(fd+1,&fds,&timeout);
266
267 /* Check if error */
268 if(selrtn == -1) {
269 /* something is wrong. Maybe the socket is dead? */
270 DEBUG(0,("read_socket_with_timeout: timeout read. select error = %s.\n", strerror(errno) ));
271 smb_read_error = READ_ERROR;
272 return -1;
273 }
274
275 /* Did we timeout ? */
276 if (selrtn == 0) {
277 DEBUG(10,("read_socket_with_timeout: timeout read. select timed out.\n"));
278 smb_read_error = READ_TIMEOUT;
279 return -1;
280 }
281
282 #ifdef WITH_SSL
283 if(fd == sslFd){
284 readret = SSL_read(ssl, buf + nread, maxcnt - nread);
285 }else{
286 readret = read(fd, buf + nread, maxcnt - nread);
287 }
288 #else /* WITH_SSL */
289 readret = read(fd, buf+nread, maxcnt-nread);
290 #endif /* WITH_SSL */
291
292 if (readret == 0) {
293 /* we got EOF on the file descriptor */
294 DEBUG(5,("read_socket_with_timeout: timeout read. EOF from client.\n"));
295 smb_read_error = READ_EOF;
296 return -1;
297 }
298
299 if (readret == -1) {
300 /* the descriptor is probably dead */
301 DEBUG(0,("read_socket_with_timeout: timeout read. read error = %s.\n", strerror(errno) ));
302 smb_read_error = READ_ERROR;
303 return -1;
304 }
305
306 nread += readret;
307 }
308
309 /* Return the number we got */
310 return((ssize_t)nread);
311 }
312
313 /****************************************************************************
314 Read data from a fd with a timout in msec.
315 mincount = if timeout, minimum to read before returning
316 maxcount = number to be read.
317 time_out = timeout in milliseconds
318 ****************************************************************************/
319
320 ssize_t read_with_timeout(int fd,char *buf,size_t mincnt,size_t maxcnt,unsigned int time_out)
321 {
322 fd_set fds;
323 int selrtn;
324 ssize_t readret;
325 size_t nread = 0;
326 struct timeval timeout;
327
328 /* just checking .... */
329 if (maxcnt <= 0)
330 return(0);
331
332 /* Blocking read */
333 if (time_out <= 0) {
334 if (mincnt == 0) mincnt = maxcnt;
335
336 while (nread < mincnt) {
337 #ifdef WITH_SSL
338 if(fd == sslFd){
339 readret = SSL_read(ssl, buf + nread, maxcnt - nread);
340 }else{
341 readret = read(fd, buf + nread, maxcnt - nread);
342 }
343 #else /* WITH_SSL */
344 readret = read(fd, buf + nread, maxcnt - nread);
345 #endif /* WITH_SSL */
346
347 if (readret <= 0)
348 return readret;
349
350 nread += readret;
351 }
352 return((ssize_t)nread);
353 }
354
355 /* Most difficult - timeout read */
356 /* If this is ever called on a disk file and
357 mincnt is greater then the filesize then
358 system performance will suffer severely as
359 select always returns true on disk files */
360
361 /* Set initial timeout */
362 timeout.tv_sec = (time_t)(time_out / 1000);
363 timeout.tv_usec = (long)(1000 * (time_out % 1000));
364
365 for (nread=0; nread < mincnt; ) {
366 FD_ZERO(&fds);
367 FD_SET(fd,&fds);
368
369 selrtn = sys_select_intr(fd+1,&fds,&timeout);
370
371 if(selrtn <= 0)
372 return selrtn;
373
374 #ifdef WITH_SSL
375 if(fd == sslFd){
376 readret = SSL_read(ssl, buf + nread, maxcnt - nread);
377 }else{
378 readret = read(fd, buf + nread, maxcnt - nread);
379 }
380 #else /* WITH_SSL */
381 readret = read(fd, buf+nread, maxcnt-nread);
382 #endif /* WITH_SSL */
383
384 if (readret <= 0)
385 return readret;
386
387 nread += readret;
388 }
389
390 /* Return the number we got */
391 return((ssize_t)nread);
392 }
393
394 /****************************************************************************
395 send a keepalive packet (rfc1002)
396 ****************************************************************************/
397
398 BOOL send_keepalive(int client)
399 {
400 unsigned char buf[4];
401
402 buf[0] = 0x85;
403 buf[1] = buf[2] = buf[3] = 0;
404
405 return(write_socket_data(client,(char *)buf,4) == 4);
406 }
407
408 /****************************************************************************
409 read data from the client, reading exactly N bytes.
410 ****************************************************************************/
411
412 ssize_t read_data(int fd,char *buffer,size_t N)
413 {
414 ssize_t ret;
415 size_t total=0;
416
417 smb_read_error = 0;
418
419 while (total < N)
420 {
421 #ifdef WITH_SSL
422 if(fd == sslFd){
423 ret = SSL_read(ssl, buffer + total, N - total);
424 }else{
425 ret = read(fd,buffer + total,N - total);
426 }
427 #else /* WITH_SSL */
428 ret = read(fd,buffer + total,N - total);
429 #endif /* WITH_SSL */
430
431 if (ret == 0)
432 {
433 DEBUG(10,("read_data: read of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
434 smb_read_error = READ_EOF;
435 return 0;
436 }
437 if (ret == -1)
438 {
439 DEBUG(0,("read_data: read failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
440 smb_read_error = READ_ERROR;
441 return -1;
442 }
443 total += ret;
444 }
445 return (ssize_t)total;
446 }
447
448 /****************************************************************************
449 Read data from a socket, reading exactly N bytes.
450 ****************************************************************************/
451
452 static ssize_t read_socket_data(int fd,char *buffer,size_t N)
453 {
454 ssize_t ret;
455 size_t total=0;
456
457 smb_read_error = 0;
458
459 while (total < N)
460 {
461 #ifdef WITH_SSL
462 if(fd == sslFd){
463 ret = SSL_read(ssl, buffer + total, N - total);
464 }else{
465 ret = read(fd,buffer + total,N - total);
466 }
467 #else /* WITH_SSL */
468 ret = read(fd,buffer + total,N - total);
469 #endif /* WITH_SSL */
470
471 if (ret == 0)
472 {
473 DEBUG(10,("read_socket_data: recv of %d returned 0. Error = %s\n", (int)(N - total), strerror(errno) ));
474 smb_read_error = READ_EOF;
475 return 0;
476 }
477 if (ret == -1)
478 {
479 DEBUG(0,("read_socket_data: recv failure for %d. Error = %s\n", (int)(N - total), strerror(errno) ));
480 smb_read_error = READ_ERROR;
481 return -1;
482 }
483 total += ret;
484 }
485 return (ssize_t)total;
486 }
487
488 /****************************************************************************
489 Write data to a fd.
490 ****************************************************************************/
491
492 ssize_t write_data(int fd,char *buffer,size_t N)
493 {
494 size_t total=0;
495 ssize_t ret;
496
497 while (total < N)
498 {
499 #ifdef WITH_SSL
500 if(fd == sslFd){
501 ret = SSL_write(ssl,buffer + total,N - total);
502 }else{
503 ret = write(fd,buffer + total,N - total);
504 }
505 #else /* WITH_SSL */
506 ret = write(fd,buffer + total,N - total);
507 #endif /* WITH_SSL */
508
509 if (ret == -1) {
510 DEBUG(0,("write_data: write failure. Error = %s\n", strerror(errno) ));
511 return -1;
512 }
513 if (ret == 0) return total;
514
515 total += ret;
516 }
517 return (ssize_t)total;
518 }
519
520 /****************************************************************************
521 Write data to a socket - use send rather than write.
522 ****************************************************************************/
523
524 ssize_t write_socket_data(int fd,char *buffer,size_t N)
525 {
526 size_t total=0;
527 ssize_t ret;
528
529 while (total < N)
530 {
531 #ifdef WITH_SSL
532 if(fd == sslFd){
533 ret = SSL_write(ssl,buffer + total,N - total);
534 }else{
535 ret = send(fd,buffer + total,N - total, 0);
536 }
537 #else /* WITH_SSL */
538 ret = send(fd,buffer + total,N - total,0);
539 #endif /* WITH_SSL */
540
541 if (ret == -1) {
542 DEBUG(0,("write_socket_data: write failure. Error = %s\n", strerror(errno) ));
543 return -1;
544 }
545 if (ret == 0) return total;
546
547 total += ret;
548 }
549 return (ssize_t)total;
550 }
551
552 /****************************************************************************
553 write to a socket
554 ****************************************************************************/
555
556 ssize_t write_socket(int fd,char *buf,size_t len)
557 {
558 ssize_t ret=0;
559
560 DEBUG(6,("write_socket(%d,%d)\n",fd,(int)len));
561 ret = write_socket_data(fd,buf,len);
562
563 DEBUG(6,("write_socket(%d,%d) wrote %d\n",fd,(int)len,(int)ret));
564 if(ret <= 0)
565 DEBUG(0,("write_socket: Error writing %d bytes to socket %d: ERRNO = %s\n",
566 (int)len, fd, strerror(errno) ));
567
568 return(ret);
569 }
570
571 /****************************************************************************
572 read 4 bytes of a smb packet and return the smb length of the packet
573 store the result in the buffer
574 This version of the function will return a length of zero on receiving
575 a keepalive packet.
576 timeout is in milliseconds.
577 ****************************************************************************/
578
579 static ssize_t read_smb_length_return_keepalive(int fd,char *inbuf,unsigned int timeout)
580 {
581 ssize_t len=0;
582 int msg_type;
583 BOOL ok = False;
584
585 while (!ok)
586 {
587 if (timeout > 0)
588 ok = (read_socket_with_timeout(fd,inbuf,4,4,timeout) == 4);
589 else
590 ok = (read_socket_data(fd,inbuf,4) == 4);
591
592 if (!ok)
593 return(-1);
594
595 len = smb_len(inbuf);
596 msg_type = CVAL(inbuf,0);
597
598 if (msg_type == 0x85)
599 DEBUG(5,("Got keepalive packet\n"));
600 }
601
602 DEBUG(10,("got smb length of %d\n",len));
603
604 return(len);
605 }
606
607 /****************************************************************************
608 read 4 bytes of a smb packet and return the smb length of the packet
609 store the result in the buffer. This version of the function will
610 never return a session keepalive (length of zero).
611 timeout is in milliseconds.
612 ****************************************************************************/
613
614 ssize_t read_smb_length(int fd,char *inbuf,unsigned int timeout)
615 {
616 ssize_t len;
617
618 for(;;)
619 {
620 len = read_smb_length_return_keepalive(fd, inbuf, timeout);
621
622 if(len < 0)
623 return len;
624
625 /* Ignore session keepalives. */
626 if(CVAL(inbuf,0) != 0x85)
627 break;
628 }
629
630 DEBUG(10,("read_smb_length: got smb length of %d\n",len));
631
632 return len;
633 }
634
635 /****************************************************************************
636 read an smb from a fd. Note that the buffer *MUST* be of size
637 BUFFER_SIZE+SAFETY_MARGIN.
638 The timeout is in milliseconds.
639 This function will return on a
640 receipt of a session keepalive packet.
641 ****************************************************************************/
642
643 BOOL receive_smb(int fd,char *buffer, unsigned int timeout)
644 {
645 ssize_t len,ret;
646
647 smb_read_error = 0;
648
649 memset(buffer,'\0',smb_size + 100);
650
651 len = read_smb_length_return_keepalive(fd,buffer,timeout);
652 if (len < 0) {
653 DEBUG(10,("receive_smb: length < 0!\n"));
654 return(False);
655 }
656
657 /*
658 * A WRITEX with CAP_LARGE_WRITEX can be 64k worth of data plus 65 bytes
659 * of header. Don't print the error if this fits.... JRA.
660 */
661
662 if (len > (BUFFER_SIZE + LARGE_WRITEX_HDR_SIZE)) {
663 DEBUG(0,("Invalid packet length! (%d bytes).\n",len));
664 if (len > BUFFER_SIZE + (SAFETY_MARGIN/2))
665 exit(1);
666 }
667
668 if(len > 0) {
669 ret = read_socket_data(fd,buffer+4,len);
670 if (ret != len) {
671 smb_read_error = READ_ERROR;
672 return False;
673 }
674 }
675
676 return(True);
677 }
678
679 /****************************************************************************
680 read an smb from a fd ignoring all keepalive packets. Note that the buffer
681 *MUST* be of size BUFFER_SIZE+SAFETY_MARGIN.
682 The timeout is in milliseconds
683
684 This is exactly the same as receive_smb except that it never returns
685 a session keepalive packet (just as receive_smb used to do).
686 receive_smb was changed to return keepalives as the oplock processing means this call
687 should never go into a blocking read.
688 ****************************************************************************/
689
690 BOOL client_receive_smb(int fd,char *buffer, unsigned int timeout)
691 {
692 BOOL ret;
693
694 for(;;)
695 {
696 ret = receive_smb(fd, buffer, timeout);
697
698 if (!ret)
699 {
700 DEBUG(10,("client_receive_smb failed\n"));
701 show_msg(buffer);
702 return ret;
703 }
704
705 /* Ignore session keepalive packets. */
706 if(CVAL(buffer,0) != 0x85)
707 break;
708 }
709 show_msg(buffer);
710 return ret;
711 }
712
713 /****************************************************************************
714 send an null session message to a fd
715 ****************************************************************************/
716
717 BOOL send_null_session_msg(int fd)
718 {
719 ssize_t ret;
720 uint32 blank = 0;
721 size_t len = 4;
722 size_t nwritten=0;
723 char *buffer = (char *)&blank;
724
725 while (nwritten < len)
726 {
727 ret = write_socket(fd,buffer+nwritten,len - nwritten);
728 if (ret <= 0)
729 {
730 DEBUG(0,("send_null_session_msg: Error writing %d bytes to client. %d. Exiting\n",(int)len,(int)ret));
731 exit(1);
732 }
733 nwritten += ret;
734 }
735
736 DEBUG(10,("send_null_session_msg: sent 4 null bytes to client.\n"));
737 return True;
738 }
739
740 /****************************************************************************
741 send an smb to a fd
742 ****************************************************************************/
743
744 BOOL send_smb(int fd,char *buffer)
745 {
746 size_t len;
747 size_t nwritten=0;
748 ssize_t ret;
749 len = smb_len(buffer) + 4;
750
751 while (nwritten < len)
752 {
753 ret = write_socket(fd,buffer+nwritten,len - nwritten);
754 if (ret <= 0)
755 {
756 DEBUG(0,("Error writing %d bytes to client. %d. Exiting\n",(int)len,(int)ret));
757 exit(1);
758 }
759 nwritten += ret;
760 }
761
762 return True;
763 }
764
765 /****************************************************************************
766 send a single packet to a port on another machine
767 ****************************************************************************/
768
769 BOOL send_one_packet(char *buf,int len,struct in_addr ip,int port,int type)
770 {
771 BOOL ret;
772 int out_fd;
773 struct sockaddr_in sock_out;
774
775 /* create a socket to write to */
776 out_fd = socket(AF_INET, type, 0);
777 if (out_fd == -1)
778 {
779 DEBUG(0,("socket failed"));
780 return False;
781 }
782
783 /* set the address and port */
784 memset((char *)&sock_out,'\0',sizeof(sock_out));
785 putip((char *)&sock_out.sin_addr,(char *)&ip);
786 sock_out.sin_port = htons( port );
787 sock_out.sin_family = AF_INET;
788
789 if (DEBUGLEVEL > 0)
790 DEBUG(3,("sending a packet of len %d to (%s) on port %d of type %s\n",
791 len,inet_ntoa(ip),port,type==SOCK_DGRAM?"DGRAM":"STREAM"));
792
793 /* send it */
794 ret = (sendto(out_fd,buf,len,0,(struct sockaddr *)&sock_out,sizeof(sock_out)) >= 0);
795
796 if (!ret)
797 DEBUG(0,("Packet send to %s(%d) failed ERRNO=%s\n",
798 inet_ntoa(ip),port,strerror(errno)));
799
800 close(out_fd);
801 return(ret);
802 }
803
804 /****************************************************************************
805 open a socket of the specified type, port and address for incoming data
806 ****************************************************************************/
807
808 int open_socket_in(int type, int port, int dlevel,uint32 socket_addr, BOOL rebind)
809 {
810 struct hostent *hp;
811 struct sockaddr_in sock;
812 pstring host_name;
813 int res;
814
815 /* get my host name */
816 if (gethostname(host_name, MAXHOSTNAMELEN) == -1)
817 { DEBUG(0,("gethostname failed\n")); return -1; }
818
819 /* get host info */
820 if ((hp = Get_Hostbyname(host_name)) == 0)
821 {
822 DEBUG(0,( "Get_Hostbyname: Unknown host %s\n",host_name));
823 return -1;
824 }
825
826 memset((char *)&sock,'\0',sizeof(sock));
827 memcpy((char *)&sock.sin_addr,(char *)hp->h_addr, hp->h_length);
828
829 #ifdef HAVE_SOCK_SIN_LEN
830 sock.sin_len = sizeof(sock);
831 #endif
832 sock.sin_port = htons( port );
833 sock.sin_family = hp->h_addrtype;
834 sock.sin_addr.s_addr = socket_addr;
835 res = socket(hp->h_addrtype, type, 0);
836 if (res == -1)
837 { DEBUG(0,("socket failed\n")); return -1; }
838
839 {
840 int val=1;
841 if(rebind)
842 val=1;
843 else
844 val=0;
845 if(setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val)) == -1)
846 DEBUG(dlevel,("setsockopt: SO_REUSEADDR=%d on port %d failed with error = %s\n",
847 val, port, strerror(errno) ));
848 #ifdef SO_REUSEPORT
849 if(setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val)) == -1)
850 DEBUG(dlevel,("setsockopt: SO_REUSEPORT=%d on port %d failed with error = %s\n",
851 val, port, strerror(errno) ));
852 #endif /* SO_REUSEPORT */
853 }
854
855 /* now we've got a socket - we need to bind it */
856 if (bind(res, (struct sockaddr * ) &sock,sizeof(sock)) < 0)
857 {
858 if (port) {
859 if (port == SMB_PORT || port == NMB_PORT)
860 DEBUG(dlevel,("bind failed on port %d socket_addr=%s (%s)\n",
861 port,inet_ntoa(sock.sin_addr),strerror(errno)));
862 close(res);
863
864 if (dlevel > 0 && port < 1000)
865 port = 7999;
866
867 if (port >= 1000 && port < 9000)
868 return(open_socket_in(type,port+1,dlevel,socket_addr,rebind));
869 }
870
871 return(-1);
872 }
873 DEBUG(3,("bind succeeded on port %d\n",port));
874
875 return res;
876 }
877
878 /****************************************************************************
879 create an outgoing socket. timeout is in milliseconds.
880 **************************************************************************/
881
882 int open_socket_out(int type, struct in_addr *addr, int port ,int timeout)
883 {
884 struct sockaddr_in sock_out;
885 int res,ret;
886 int connect_loop = 250; /* 250 milliseconds */
887 int loops = (timeout) / connect_loop;
888
889 /* create a socket to write to */
890 res = socket(PF_INET, type, 0);
891 if (res == -1)
892 { DEBUG(0,("socket error\n")); return -1; }
893
894 if (type != SOCK_STREAM) 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 ret = connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out));
910
911 /* Some systems return EAGAIN when they mean EINPROGRESS */
912 if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
913 errno == EAGAIN) && loops--) {
914 msleep(connect_loop);
915 goto connect_again;
916 }
917
918 if (ret < 0 && (errno == EINPROGRESS || errno == EALREADY ||
919 errno == EAGAIN)) {
920 DEBUG(1,("timeout connecting to %s:%d\n",inet_ntoa(*addr),port));
921 close(res);
922 return -1;
923 }
924
925 #ifdef EISCONN
926 if (ret < 0 && errno == EISCONN) {
927 errno = 0;
928 ret = 0;
929 }
930 #endif
931
932 if (ret < 0) {
933 DEBUG(1,("error connecting to %s:%d (%s)\n",
934 inet_ntoa(*addr),port,strerror(errno)));
935 close(res);
936 return -1;
937 }
938
939 /* set it blocking again */
940 set_blocking(res,True);
941
942 return res;
943 }
944
945 /* the following 3 client_*() functions are nasty ways of allowing
946 some generic functions to get info that really should be hidden in
947 particular modules */
948 static int client_fd = -1;
949
950 void client_setfd(int fd)
951 {
952 client_fd = fd;
953 }
954
955 char *client_name(void)
956 {
957 return get_socket_name(client_fd);
958 }
959
960 char *client_addr(void)
961 {
962 return get_socket_addr(client_fd);
963 }
964
965 /*******************************************************************
966 matchname - determine if host name matches IP address. Used to
967 confirm a hostname lookup to prevent spoof attacks
968 ******************************************************************/
969 static BOOL matchname(char *remotehost,struct in_addr addr)
970 {
971 struct hostent *hp;
972 int i;
973
974 if ((hp = Get_Hostbyname(remotehost)) == 0) {
975 DEBUG(0,("Get_Hostbyname(%s): lookup failure.\n", remotehost));
976 return False;
977 }
978
979 /*
980 * Make sure that gethostbyname() returns the "correct" host name.
981 * Unfortunately, gethostbyname("localhost") sometimes yields
982 * "localhost.domain". Since the latter host name comes from the
983 * local DNS, we just have to trust it (all bets are off if the local
984 * DNS is perverted). We always check the address list, though.
985 */
986
987 if (strcasecmp(remotehost, hp->h_name)
988 && strcasecmp(remotehost, "localhost")) {
989 DEBUG(0,("host name/name mismatch: %s != %s\n",
990 remotehost, hp->h_name));
991 return False;
992 }
993
994 /* Look up the host address in the address list we just got. */
995 for (i = 0; hp->h_addr_list[i]; i++) {
996 if (memcmp(hp->h_addr_list[i], (caddr_t) & addr, sizeof(addr)) == 0)
997 return True;
998 }
999
1000 /*
1001 * The host name does not map to the original host address. Perhaps
1002 * someone has compromised a name server. More likely someone botched
1003 * it, but that could be dangerous, too.
1004 */
1005
1006 DEBUG(0,("host name/address mismatch: %s != %s\n",
1007 inet_ntoa(addr), hp->h_name));
1008 return False;
1009 }
1010
1011
1012 /*******************************************************************
1013 return the DNS name of the remote end of a socket
1014 ******************************************************************/
1015 char *get_socket_name(int fd)
1016 {
1017 static pstring name_buf;
1018 static fstring addr_buf;
1019 struct hostent *hp;
1020 struct in_addr addr;
1021 char *p;
1022
1023 p = get_socket_addr(fd);
1024
1025 /* it might be the same as the last one - save some DNS work */
1026 if (strcmp(p, addr_buf) == 0) return name_buf;
1027
1028 pstrcpy(name_buf,"UNKNOWN");
1029 if (fd == -1) return name_buf;
1030
1031 fstrcpy(addr_buf, p);
1032
1033 addr = *interpret_addr2(p);
1034
1035 /* Look up the remote host name. */
1036 if ((hp = gethostbyaddr((char *)&addr.s_addr, sizeof(addr.s_addr), AF_INET)) == 0) {
1037 DEBUG(1,("Gethostbyaddr failed for %s\n",p));
1038 pstrcpy(name_buf, p);
1039 } else {
1040 pstrcpy(name_buf,(char *)hp->h_name);
1041 if (!matchname(name_buf, addr)) {
1042 DEBUG(0,("Matchname failed on %s %s\n",name_buf,p));
1043 pstrcpy(name_buf,"UNKNOWN");
1044 }
1045 }
1046 return name_buf;
1047 }
1048
1049 /*******************************************************************
1050 return the IP addr of the remote end of a socket as a string
1051 ******************************************************************/
1052 char *get_socket_addr(int fd)
1053 {
1054 struct sockaddr sa;
1055 struct sockaddr_in *sockin = (struct sockaddr_in *) (&sa);
1056 int length = sizeof(sa);
1057 static fstring addr_buf;
1058
1059 fstrcpy(addr_buf,"0.0.0.0");
1060
1061 if (fd == -1) {
1062 return addr_buf;
1063 }
1064
1065 if (getpeername(fd, &sa, &length) < 0) {
1066 DEBUG(0,("getpeername failed. Error was %s\n", strerror(errno) ));
1067 return addr_buf;
1068 }
1069
1070 fstrcpy(addr_buf,(char *)inet_ntoa(sockin->sin_addr));
1071
1072 return addr_buf;
1073 }
1074
1075 /*******************************************************************
1076 opens and connects to a unix pipe socket
1077 ******************************************************************/
1078 int open_pipe_sock(char *path)
1079 {
1080 int sock;
1081 struct sockaddr_un sa;
1082
1083 sock = socket(AF_UNIX, SOCK_STREAM, 0);
1084
1085 if (sock < 0)
1086 {
1087 DEBUG(0, ("unix socket open failed\n"));
1088 return sock;
1089 }
1090
1091 ZERO_STRUCT(sa);
1092 sa.sun_family = AF_UNIX;
1093 safe_strcpy(sa.sun_path, path, sizeof(sa.sun_path)-1);
1094
1095 DEBUG(10, ("socket open succeeded. file name: %s\n", sa.sun_path));
1096
1097 if (connect(sock, (struct sockaddr*) &sa, sizeof(sa)) < 0)
1098 {
1099 DEBUG(0,("socket connect to %s failed\n", sa.sun_path));
1100 close(sock);
1101 return -1;
1102 }
1103
1104 return sock;
1105 }
1106
1107 int create_pipe_socket(char *dir, int dir_perms,
1108 char *path, int path_perms)
1109 {
1110 int s;
1111 struct sockaddr_un sa;
1112
1113 DEBUG(0,("create_pipe_socket: %s %d %s %d\n",
1114 dir, dir_perms, path, path_perms));
1115
1116 DEBUG(0,("*** RACE CONDITION. PLEASE SOMEONE EXAMINE create_pipe_Socket AND FIX IT ***\n"));
1117
1118 mkdir(dir, dir_perms);
1119
1120 if (chmod(dir, dir_perms) < 0)
1121 {
1122 DEBUG(0, ("chmod on %s failed\n", dir));
1123 return -1;
1124 }
1125
1126 if (!remove(path))
1127 {
1128 DEBUG(0, ("remove on %s failed\n", path));
1129 }
1130
1131 /* start listening on unix socket */
1132 s = socket(AF_UNIX, SOCK_STREAM, 0);
1133
1134 if (s < 0)
1135 {
1136 DEBUG(0, ("socket open failed\n"));
1137 return -1;
1138 }
1139
1140 ZERO_STRUCT(sa);
1141 sa.sun_family = AF_UNIX;
1142 safe_strcpy(sa.sun_path, path, sizeof(sa.sun_path)-1);
1143
1144 if (bind(s, (struct sockaddr*) &sa, sizeof(sa)) < 0)
1145 {
1146 DEBUG(0, ("socket bind to %s failed\n", sa.sun_path));
1147 close(s);
1148 remove(path);
1149 return -1;
1150 }
1151
1152 if (s == -1)
1153 {
1154 DEBUG(0,("bind failed\n"));
1155 remove(path);
1156 return -1;
1157 }
1158
1159 if (path_perms != 0)
1160 {
1161 chmod(path, path_perms);
1162 }
1163
1164 if (listen(s, 5) == -1)
1165 {
1166 DEBUG(0,("listen failed\n"));
1167 return -1;
1168 }
1169
1170 DEBUG(5,("unix socket opened: %s\n", path));
1171
1172 return s;
1173 }
Mirror of https://github.com/samba-team/samba.git