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