Moved inline functions from rsync.h into ifuncs.h.
[rsync/rsync.git] / socket.c
CommitLineData
0f78b815 1/*
d5d4b282 2 * Socket functions used in rsync.
362099a5 3 *
0f78b815
WD
4 * Copyright (C) 1992-2001 Andrew Tridgell <tridge@samba.org>
5 * Copyright (C) 2001, 2002 Martin Pool <mbp@samba.org>
ba2133d6 6 * Copyright (C) 2003-2007 Wayne Davison
0f78b815
WD
7 *
8 * This program is free software; you can redistribute it and/or modify
8e41b68e
WD
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
0f78b815
WD
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
e7c67065 18 * You should have received a copy of the GNU General Public License along
4fd842f9 19 * with this program; if not, visit the http://fsf.org website.
0f78b815
WD
20 */
21
22/* This file is now converted to use the new-style getaddrinfo()
362099a5
MP
23 * interface, which supports IPv6 but is also supported on recent
24 * IPv4-only machines. On systems that don't have that interface, we
0f78b815 25 * emulate it using the KAME implementation. */
bc2e93eb 26
f0fca04e 27#include "rsync.h"
1b42f628 28#include "ifuncs.h"
5899b8cf
WD
29#include <netinet/in_systm.h>
30#include <netinet/ip.h>
0f5c1c19 31#include <netinet/tcp.h>
f0fca04e 32
2c7d63c7
WD
33extern char *bind_address;
34extern int default_af_hint;
9a5a8673 35
44e604f4 36#ifdef HAVE_SIGACTION
2b28968d
WD
37static struct sigaction sigact;
38#endif
39
a7dc44d2
MP
40/**
41 * Establish a proxy connection on an open socket to a web proxy by
a3a84107
WD
42 * using the CONNECT method. If proxy_user and proxy_pass are not NULL,
43 * they are used to authenticate to the proxy using the "Basic"
44 * proxy-authorization protocol
a7dc44d2 45 **/
a3a84107
WD
46static int establish_proxy_connection(int fd, char *host, int port,
47 char *proxy_user, char *proxy_pass)
4c3b4b25 48{
a3a84107
WD
49 char *cp, buffer[1024];
50 char *authhdr, authbuf[1024];
51 int len;
52
53 if (proxy_user && proxy_pass) {
893c4cc0
WD
54 stringjoin(buffer, sizeof buffer,
55 proxy_user, ":", proxy_pass, NULL);
a3a84107 56 len = strlen(buffer);
4c3b4b25 57
8030b28f 58 if ((len*8 + 5) / 6 >= (int)sizeof authbuf - 3) {
a3a84107
WD
59 rprintf(FERROR,
60 "authentication information is too long\n");
61 return -1;
62 }
63
6854bf69 64 base64_encode(buffer, len, authbuf, 1);
a3a84107
WD
65 authhdr = "\r\nProxy-Authorization: Basic ";
66 } else {
67 *authbuf = '\0';
68 authhdr = "";
69 }
70
71 snprintf(buffer, sizeof buffer, "CONNECT %s:%d HTTP/1.0%s%s\r\n\r\n",
72 host, port, authhdr, authbuf);
73 len = strlen(buffer);
74 if (write(fd, buffer, len) != len) {
d62bcc17 75 rsyserr(FERROR, errno, "failed to write to proxy");
4c3b4b25
AT
76 return -1;
77 }
78
a3a84107 79 for (cp = buffer; cp < &buffer[sizeof buffer - 1]; cp++) {
4c3b4b25 80 if (read(fd, cp, 1) != 1) {
d62bcc17 81 rsyserr(FERROR, errno, "failed to read from proxy");
4c3b4b25
AT
82 return -1;
83 }
84 if (*cp == '\n')
85 break;
86 }
87
88 if (*cp != '\n')
89 cp++;
90 *cp-- = '\0';
91 if (*cp == '\r')
92 *cp = '\0';
93 if (strncmp(buffer, "HTTP/", 5) != 0) {
4ccfd96c 94 rprintf(FERROR, "bad response from proxy -- %s\n",
4c3b4b25
AT
95 buffer);
96 return -1;
97 }
2dc7b8bd 98 for (cp = &buffer[5]; isDigit(cp) || *cp == '.'; cp++) {}
4c3b4b25
AT
99 while (*cp == ' ')
100 cp++;
101 if (*cp != '2') {
4ccfd96c 102 rprintf(FERROR, "bad response from proxy -- %s\n",
4c3b4b25
AT
103 buffer);
104 return -1;
105 }
106 /* throw away the rest of the HTTP header */
107 while (1) {
a3a84107 108 for (cp = buffer; cp < &buffer[sizeof buffer - 1]; cp++) {
4c3b4b25 109 if (read(fd, cp, 1) != 1) {
d62bcc17
WD
110 rsyserr(FERROR, errno,
111 "failed to read from proxy");
4c3b4b25
AT
112 return -1;
113 }
114 if (*cp == '\n')
115 break;
116 }
9c07d253 117 if (cp > buffer && *cp == '\n')
4c3b4b25 118 cp--;
9c07d253 119 if (cp == buffer && (*cp == '\n' || *cp == '\r'))
4c3b4b25
AT
120 break;
121 }
122 return 0;
123}
124
125
f8be7d42
MP
126/**
127 * Try to set the local address for a newly-created socket. Return -1
128 * if this fails.
129 **/
e028b9ff 130int try_bind_local(int s, int ai_family, int ai_socktype,
4313d6f9 131 const char *bind_addr)
f8be7d42
MP
132{
133 int error;
134 struct addrinfo bhints, *bres_all, *r;
135
a3a84107 136 memset(&bhints, 0, sizeof bhints);
f8be7d42
MP
137 bhints.ai_family = ai_family;
138 bhints.ai_socktype = ai_socktype;
139 bhints.ai_flags = AI_PASSIVE;
4313d6f9 140 if ((error = getaddrinfo(bind_addr, NULL, &bhints, &bres_all))) {
f8be7d42 141 rprintf(FERROR, RSYNC_NAME ": getaddrinfo %s: %s\n",
4313d6f9 142 bind_addr, gai_strerror(error));
f8be7d42
MP
143 return -1;
144 }
145
146 for (r = bres_all; r; r = r->ai_next) {
9ec75284 147 if (bind(s, r->ai_addr, r->ai_addrlen) == -1)
f8be7d42 148 continue;
6b2d24de 149 freeaddrinfo(bres_all);
f8be7d42
MP
150 return s;
151 }
152
153 /* no error message; there might be some problem that allows
154 * creation of the socket but not binding, perhaps if the
155 * machine has no ipv6 address of this name. */
6b2d24de 156 freeaddrinfo(bres_all);
f8be7d42
MP
157 return -1;
158}
159
eecd22ff 160
d5d4b282
MP
161/**
162 * Open a socket to a tcp remote host with the specified port .
06963d0f 163 *
d5d4b282
MP
164 * Based on code from Warren. Proxy support by Stephen Rothwell.
165 * getaddrinfo() rewrite contributed by KAME.net.
06963d0f 166 *
d5d4b282
MP
167 * Now that we support IPv6 we need to look up the remote machine's
168 * address first, using @p af_hint to set a preference for the type
169 * of address. Then depending on whether it has v4 or v6 addresses we
170 * try to open a connection.
06963d0f 171 *
d5d4b282
MP
172 * The loop allows for machines with some addresses which may not be
173 * reachable, perhaps because we can't e.g. route ipv6 to that network
174 * but we can get ip4 packets through.
175 *
4313d6f9 176 * @param bind_addr Local address to use. Normally NULL to bind
d5d4b282
MP
177 * the wildcard address.
178 *
179 * @param af_hint Address family, e.g. AF_INET or AF_INET6.
06963d0f 180 **/
4313d6f9 181int open_socket_out(char *host, int port, const char *bind_addr,
d5d4b282 182 int af_hint)
bc2e93eb 183{
f0fca04e 184 int type = SOCK_STREAM;
a3a84107 185 int error, s;
06963d0f
MP
186 struct addrinfo hints, *res0, *res;
187 char portbuf[10];
a3a84107 188 char *h, *cp;
4c3b4b25
AT
189 int proxied = 0;
190 char buffer[1024];
a3a84107 191 char *proxy_user = NULL, *proxy_pass = NULL;
4c3b4b25 192
660c6fbd 193 /* if we have a RSYNC_PROXY env variable then redirect our
a3a84107 194 * connetcion via a web proxy at the given address. */
4c3b4b25 195 h = getenv("RSYNC_PROXY");
9c07d253 196 proxied = h != NULL && *h != '\0';
4c3b4b25
AT
197
198 if (proxied) {
a3a84107
WD
199 strlcpy(buffer, h, sizeof buffer);
200
201 /* Is the USER:PASS@ prefix present? */
b31c92ed 202 if ((cp = strrchr(buffer, '@')) != NULL) {
a3a84107
WD
203 *cp++ = '\0';
204 /* The remainder is the HOST:PORT part. */
205 h = cp;
206
207 if ((cp = strchr(buffer, ':')) == NULL) {
208 rprintf(FERROR,
209 "invalid proxy specification: should be USER:PASS@HOST:PORT\n");
210 return -1;
211 }
212 *cp++ = '\0';
213
214 proxy_user = buffer;
215 proxy_pass = cp;
216 } else {
217 /* The whole buffer is the HOST:PORT part. */
218 h = buffer;
219 }
220
221 if ((cp = strchr(h, ':')) == NULL) {
660c6fbd
MP
222 rprintf(FERROR,
223 "invalid proxy specification: should be HOST:PORT\n");
4c3b4b25
AT
224 return -1;
225 }
226 *cp++ = '\0';
a3a84107 227 strlcpy(portbuf, cp, sizeof portbuf);
7bea78ce
MP
228 if (verbose >= 2) {
229 rprintf(FINFO, "connection via http proxy %s port %s\n",
230 h, portbuf);
231 }
4c3b4b25 232 } else {
a3a84107 233 snprintf(portbuf, sizeof portbuf, "%d", port);
4c3b4b25 234 h = host;
4c3b4b25 235 }
f0fca04e 236
a3a84107 237 memset(&hints, 0, sizeof hints);
d5d4b282 238 hints.ai_family = af_hint;
06963d0f
MP
239 hints.ai_socktype = type;
240 error = getaddrinfo(h, portbuf, &hints, &res0);
241 if (error) {
d5d4b282
MP
242 rprintf(FERROR, RSYNC_NAME ": getaddrinfo: %s %s: %s\n",
243 h, portbuf, gai_strerror(error));
f0fca04e
AT
244 return -1;
245 }
246
06963d0f 247 s = -1;
2d6dbe29
MP
248 /* Try to connect to all addresses for this machine until we get
249 * through. It might e.g. be multi-homed, or have both IPv4 and IPv6
250 * addresses. We need to create a socket for each record, since the
251 * address record tells us what protocol to use to try to connect. */
06963d0f
MP
252 for (res = res0; res; res = res->ai_next) {
253 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
254 if (s < 0)
255 continue;
f0fca04e 256
4313d6f9 257 if (bind_addr
a3a84107 258 && try_bind_local(s, res->ai_family, type,
4313d6f9 259 bind_addr) == -1) {
a3a84107
WD
260 close(s);
261 s = -1;
262 continue;
263 }
06963d0f
MP
264 if (connect(s, res->ai_addr, res->ai_addrlen) < 0) {
265 close(s);
266 s = -1;
267 continue;
268 }
a3a84107
WD
269 if (proxied
270 && establish_proxy_connection(s, host, port,
271 proxy_user, proxy_pass) != 0) {
06963d0f
MP
272 close(s);
273 s = -1;
274 continue;
a3a84107
WD
275 }
276 break;
4c3b4b25 277 }
06963d0f
MP
278 freeaddrinfo(res0);
279 if (s < 0) {
d62bcc17 280 rsyserr(FERROR, errno, "failed to connect to %s", h);
f0fca04e
AT
281 return -1;
282 }
06963d0f 283 return s;
f0fca04e
AT
284}
285
286
eecd22ff
MP
287/**
288 * Open an outgoing socket, but allow for it to be intercepted by
289 * $RSYNC_CONNECT_PROG, which will execute a program across a TCP
290 * socketpair rather than really opening a socket.
291 *
292 * We use this primarily in testing to detect TCP flow bugs, but not
293 * cause security problems by really opening remote connections.
294 *
295 * This is based on the Samba LIBSMB_PROG feature.
06963d0f 296 *
4313d6f9 297 * @param bind_addr Local address to use. Normally NULL to get the stack default.
eecd22ff 298 **/
4313d6f9 299int open_socket_out_wrapped(char *host, int port, const char *bind_addr,
9c07d253 300 int af_hint)
eecd22ff 301{
df5cd107 302 char *prog = getenv("RSYNC_CONNECT_PROG");
eecd22ff 303
4eff3051 304 if (prog && strchr(prog, '%')) {
e5f1a96f
WD
305 int hlen = strlen(host);
306 int len = strlen(prog) + 1;
307 char *f, *t;
308 for (f = prog; *f; f++) {
309 if (*f != '%')
310 continue;
311 /* Compute more than enough room. */
312 if (f[1] == '%')
313 f++;
314 else
315 len += hlen;
316 }
317 f = prog;
318 if (!(prog = new_array(char, len)))
319 out_of_memory("open_socket_out_wrapped");
320 for (t = prog; *f; f++) {
321 if (*f == '%') {
322 switch (*++f) {
323 case '%':
324 /* Just skips the extra '%'. */
325 break;
326 case 'H':
327 memcpy(t, host, hlen);
328 t += hlen;
329 continue;
330 default:
331 f--; /* pass % through */
332 break;
333 }
334 }
335 *t++ = *f;
336 }
337 *t = '\0';
338 }
339
df5cd107
WD
340 if (verbose >= 2) {
341 rprintf(FINFO, "%sopening tcp connection to %s port %d\n",
342 prog ? "Using RSYNC_CONNECT_PROG instead of " : "",
343 host, port);
344 }
345 if (prog)
9c07d253 346 return sock_exec(prog);
4313d6f9 347 return open_socket_out(host, port, bind_addr, af_hint);
eecd22ff
MP
348}
349
350
351
06963d0f 352/**
2c7d63c7
WD
353 * Open one or more sockets for incoming data using the specified type,
354 * port, and address.
06963d0f 355 *
2c7d63c7
WD
356 * The getaddrinfo() call may return several address results, e.g. for
357 * the machine's IPv4 and IPv6 name.
9c07d253 358 *
2c7d63c7
WD
359 * We return an array of file-descriptors to the sockets, with a trailing
360 * -1 value to indicate the end of the list.
9c07d253 361 *
4313d6f9 362 * @param bind_addr Local address to bind, or NULL to allow it to
06963d0f
MP
363 * default.
364 **/
4313d6f9 365static int *open_socket_in(int type, int port, const char *bind_addr,
b0fd253a 366 int af_hint)
f0fca04e 367{
2c7d63c7 368 int one = 1;
5c6d4632 369 int s, *socks, maxs, i, ecnt;
13e29995 370 struct addrinfo hints, *all_ai, *resp;
5c6d4632 371 char portbuf[10], **errmsgs;
06963d0f
MP
372 int error;
373
a3a84107 374 memset(&hints, 0, sizeof hints);
d5d4b282 375 hints.ai_family = af_hint;
06963d0f
MP
376 hints.ai_socktype = type;
377 hints.ai_flags = AI_PASSIVE;
a3a84107 378 snprintf(portbuf, sizeof portbuf, "%d", port);
4313d6f9 379 error = getaddrinfo(bind_addr, portbuf, &hints, &all_ai);
06963d0f 380 if (error) {
7ef6aa64 381 rprintf(FERROR, RSYNC_NAME ": getaddrinfo: bind address %s: %s\n",
4313d6f9 382 bind_addr, gai_strerror(error));
b0fd253a
WD
383 return NULL;
384 }
385
386 /* Count max number of sockets we might open. */
387 for (maxs = 0, resp = all_ai; resp; resp = resp->ai_next, maxs++) {}
2c7d63c7 388
5c6d4632
WD
389 socks = new_array(int, maxs + 1);
390 errmsgs = new_array(char *, maxs);
391 if (!socks || !errmsgs)
2c7d63c7 392 out_of_memory("open_socket_in");
06963d0f 393
13e29995
MP
394 /* We may not be able to create the socket, if for example the
395 * machine knows about IPv6 in the C library, but not in the
396 * kernel. */
5c6d4632 397 for (resp = all_ai, i = ecnt = 0; resp; resp = resp->ai_next) {
13e29995
MP
398 s = socket(resp->ai_family, resp->ai_socktype,
399 resp->ai_protocol);
400
b0fd253a 401 if (s == -1) {
5c6d4632
WD
402 int r = asprintf(&errmsgs[ecnt++],
403 "socket(%d,%d,%d) failed: %s\n",
404 (int)resp->ai_family, (int)resp->ai_socktype,
405 (int)resp->ai_protocol, strerror(errno));
406 if (r < 0)
407 out_of_memory("open_socket_in");
13e29995
MP
408 /* See if there's another address that will work... */
409 continue;
b0fd253a 410 }
9c07d253 411
13e29995
MP
412 setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
413 (char *)&one, sizeof one);
9c07d253 414
b0fd253a
WD
415#ifdef IPV6_V6ONLY
416 if (resp->ai_family == AF_INET6) {
dcd08dc5
WD
417 if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY,
418 (char *)&one, sizeof one) < 0
419 && default_af_hint != AF_INET6) {
420 close(s);
421 continue;
422 }
b0fd253a
WD
423 }
424#endif
425
e028b9ff
WD
426 /* Now we've got a socket - we need to bind it. */
427 if (bind(s, resp->ai_addr, resp->ai_addrlen) < 0) {
13e29995 428 /* Nope, try another */
5c6d4632 429 int r = asprintf(&errmsgs[ecnt++],
e2d774cd
WD
430 "bind() failed: %s (address-family %d)\n",
431 strerror(errno), (int)resp->ai_family);
5c6d4632
WD
432 if (r < 0)
433 out_of_memory("open_socket_in");
13e29995
MP
434 close(s);
435 continue;
b8771f96 436 }
e23d790f 437
2c7d63c7 438 socks[i++] = s;
f0fca04e 439 }
2c7d63c7 440 socks[i] = -1;
f0fca04e 441
b0fd253a
WD
442 if (all_ai)
443 freeaddrinfo(all_ai);
b8771f96 444
5c6d4632
WD
445 /* Only output the socket()/bind() messages if we were totally
446 * unsuccessful, or if the daemon is being run with -vv. */
447 for (s = 0; s < ecnt; s++) {
448 if (!i || verbose > 1)
332cf6df 449 rwrite(FLOG, errmsgs[s], strlen(errmsgs[s]), 0);
5c6d4632
WD
450 free(errmsgs[s]);
451 }
452 free(errmsgs);
453
2c7d63c7 454 if (!i) {
b0fd253a 455 rprintf(FERROR,
2c7d63c7
WD
456 "unable to bind any inbound sockets on port %d\n",
457 port);
b0fd253a
WD
458 free(socks);
459 return NULL;
460 }
461 return socks;
f0fca04e
AT
462}
463
464
7c1b4daa
MP
465/*
466 * Determine if a file descriptor is in fact a socket
467 */
f0fca04e
AT
468int is_a_socket(int fd)
469{
ac2a1a44 470 int v;
a3a84107
WD
471 socklen_t l = sizeof (int);
472
473 /* Parameters to getsockopt, setsockopt etc are very
474 * unstandardized across platforms, so don't be surprised if
475 * there are compiler warnings on e.g. SCO OpenSwerver or AIX.
476 * It seems they all eventually get the right idea.
477 *
478 * Debian says: ``The fifth argument of getsockopt and
479 * setsockopt is in reality an int [*] (and this is what BSD
480 * 4.* and libc4 and libc5 have). Some POSIX confusion
481 * resulted in the present socklen_t. The draft standard has
482 * not been adopted yet, but glibc2 already follows it and
483 * also has socklen_t [*]. See also accept(2).''
484 *
485 * We now return to your regularly scheduled programming. */
9c07d253 486 return getsockopt(fd, SOL_SOCKET, SO_TYPE, (char *)&v, &l) == 0;
f0fca04e
AT
487}
488
489
067669da
WD
490static RETSIGTYPE sigchld_handler(UNUSED(int val))
491{
ca20c7fd
WD
492#ifdef WNOHANG
493 while (waitpid(-1, NULL, WNOHANG) > 0) {}
494#endif
44e604f4 495#ifndef HAVE_SIGACTION
cb984e62 496 signal(SIGCHLD, sigchld_handler);
2b28968d 497#endif
ca20c7fd
WD
498}
499
500
39993af5 501void start_accept_loop(int port, int (*fn)(int, int))
f0fca04e 502{
b0fd253a 503 fd_set deffds;
d8d36af4 504 int *sp, maxfd, i;
f0fca04e 505
44e604f4 506#ifdef HAVE_SIGACTION
2b28968d
WD
507 sigact.sa_flags = SA_NOCLDSTOP;
508#endif
509
f0fca04e 510 /* open an incoming socket */
b0fd253a
WD
511 sp = open_socket_in(SOCK_STREAM, port, bind_address, default_af_hint);
512 if (sp == NULL)
65417579 513 exit_cleanup(RERR_SOCKETIO);
f0fca04e
AT
514
515 /* ready to listen */
b0fd253a 516 FD_ZERO(&deffds);
2c7d63c7
WD
517 for (i = 0, maxfd = -1; sp[i] >= 0; i++) {
518 if (listen(sp[i], 5) < 0) {
d62bcc17 519 rsyserr(FERROR, errno, "listen() on socket failed");
4f5b0756 520#ifdef INET6
2c7d63c7
WD
521 if (errno == EADDRINUSE && i > 0) {
522 rprintf(FINFO,
880570f2 523 "Try using --ipv4 or --ipv6 to avoid this listen() error.\n");
2c7d63c7
WD
524 }
525#endif
b0fd253a
WD
526 exit_cleanup(RERR_SOCKETIO);
527 }
528 FD_SET(sp[i], &deffds);
529 if (maxfd < sp[i])
530 maxfd = sp[i];
f0fca04e
AT
531 }
532
f0fca04e 533 /* now accept incoming connections - forking a new process
a3a84107 534 * for each incoming connection */
f0fca04e
AT
535 while (1) {
536 fd_set fds;
c4a5c57d 537 pid_t pid;
f0fca04e 538 int fd;
2d6dbe29 539 struct sockaddr_storage addr;
d54765c4 540 socklen_t addrlen = sizeof addr;
f0fca04e 541
15b84e14 542 /* close log file before the potentially very long select so
a3a84107
WD
543 * file can be trimmed by another process instead of growing
544 * forever */
8ee6adef 545 logfile_close();
45a83540 546
b0fd253a
WD
547#ifdef FD_COPY
548 FD_COPY(&deffds, &fds);
549#else
550 fds = deffds;
551#endif
f0fca04e 552
b0fd253a 553 if (select(maxfd + 1, &fds, NULL, NULL, NULL) != 1)
9c07d253 554 continue;
f0fca04e 555
2c7d63c7 556 for (i = 0, fd = -1; sp[i] >= 0; i++) {
b0fd253a
WD
557 if (FD_ISSET(sp[i], &fds)) {
558 fd = accept(sp[i], (struct sockaddr *)&addr,
559 &addrlen);
560 break;
561 }
562 }
f0fca04e 563
b0fd253a 564 if (fd < 0)
9c07d253 565 continue;
f0fca04e 566
2b28968d 567 SIGACTION(SIGCHLD, sigchld_handler);
31f440e6 568
c4a5c57d 569 if ((pid = fork()) == 0) {
9f639210 570 int ret;
2c7d63c7
WD
571 for (i = 0; sp[i] >= 0; i++)
572 close(sp[i]);
1da05366 573 /* Re-open log file in child before possibly giving
8ee6adef
WD
574 * up privileges (see logfile_close() above). */
575 logfile_reopen();
9f639210
DD
576 ret = fn(fd, fd);
577 close_all();
578 _exit(ret);
c4a5c57d 579 } else if (pid < 0) {
d62bcc17
WD
580 rsyserr(FERROR, errno,
581 "could not create child server process");
c4a5c57d
MP
582 close(fd);
583 /* This might have happened because we're
584 * overloaded. Sleep briefly before trying to
585 * accept again. */
586 sleep(2);
bd37c666 587 } else {
79845f28 588 /* Parent doesn't need this fd anymore. */
bd37c666 589 close(fd);
f0fca04e 590 }
f0fca04e 591 }
f0fca04e
AT
592}
593
594
595enum SOCK_OPT_TYPES {OPT_BOOL,OPT_INT,OPT_ON};
596
597struct
598{
599 char *name;
600 int level;
601 int option;
602 int value;
603 int opttype;
604} socket_options[] = {
605 {"SO_KEEPALIVE", SOL_SOCKET, SO_KEEPALIVE, 0, OPT_BOOL},
606 {"SO_REUSEADDR", SOL_SOCKET, SO_REUSEADDR, 0, OPT_BOOL},
607 {"SO_BROADCAST", SOL_SOCKET, SO_BROADCAST, 0, OPT_BOOL},
608#ifdef TCP_NODELAY
609 {"TCP_NODELAY", IPPROTO_TCP, TCP_NODELAY, 0, OPT_BOOL},
610#endif
611#ifdef IPTOS_LOWDELAY
612 {"IPTOS_LOWDELAY", IPPROTO_IP, IP_TOS, IPTOS_LOWDELAY, OPT_ON},
613#endif
614#ifdef IPTOS_THROUGHPUT
615 {"IPTOS_THROUGHPUT", IPPROTO_IP, IP_TOS, IPTOS_THROUGHPUT, OPT_ON},
616#endif
617#ifdef SO_SNDBUF
618 {"SO_SNDBUF", SOL_SOCKET, SO_SNDBUF, 0, OPT_INT},
619#endif
620#ifdef SO_RCVBUF
621 {"SO_RCVBUF", SOL_SOCKET, SO_RCVBUF, 0, OPT_INT},
622#endif
623#ifdef SO_SNDLOWAT
624 {"SO_SNDLOWAT", SOL_SOCKET, SO_SNDLOWAT, 0, OPT_INT},
625#endif
626#ifdef SO_RCVLOWAT
627 {"SO_RCVLOWAT", SOL_SOCKET, SO_RCVLOWAT, 0, OPT_INT},
628#endif
629#ifdef SO_SNDTIMEO
630 {"SO_SNDTIMEO", SOL_SOCKET, SO_SNDTIMEO, 0, OPT_INT},
631#endif
632#ifdef SO_RCVTIMEO
633 {"SO_RCVTIMEO", SOL_SOCKET, SO_RCVTIMEO, 0, OPT_INT},
634#endif
635 {NULL,0,0,0,0}};
636
9c07d253 637
f0fca04e 638
a7dc44d2
MP
639/**
640 * Set user socket options
641 **/
f0fca04e
AT
642void set_socket_options(int fd, char *options)
643{
644 char *tok;
9c07d253
WD
645
646 if (!options || !*options)
647 return;
a6801c39 648
f0fca04e 649 options = strdup(options);
f0fca04e 650
9c07d253
WD
651 if (!options)
652 out_of_memory("set_socket_options");
653
654 for (tok = strtok(options, " \t,"); tok; tok = strtok(NULL," \t,")) {
f0fca04e
AT
655 int ret=0,i;
656 int value = 1;
657 char *p;
658 int got_value = 0;
659
660 if ((p = strchr(tok,'='))) {
661 *p = 0;
662 value = atoi(p+1);
663 got_value = 1;
664 }
665
9c07d253 666 for (i = 0; socket_options[i].name; i++) {
f0fca04e
AT
667 if (strcmp(socket_options[i].name,tok)==0)
668 break;
9c07d253 669 }
f0fca04e
AT
670
671 if (!socket_options[i].name) {
672 rprintf(FERROR,"Unknown socket option %s\n",tok);
673 continue;
674 }
675
676 switch (socket_options[i].opttype) {
677 case OPT_BOOL:
678 case OPT_INT:
679 ret = setsockopt(fd,socket_options[i].level,
a3a84107
WD
680 socket_options[i].option,
681 (char *)&value, sizeof (int));
f0fca04e 682 break;
9c07d253 683
f0fca04e
AT
684 case OPT_ON:
685 if (got_value)
4ccfd96c 686 rprintf(FERROR,"syntax error -- %s does not take a value\n",tok);
f0fca04e
AT
687
688 {
689 int on = socket_options[i].value;
690 ret = setsockopt(fd,socket_options[i].level,
a3a84107
WD
691 socket_options[i].option,
692 (char *)&on, sizeof (int));
f0fca04e 693 }
9c07d253 694 break;
f0fca04e 695 }
9c07d253 696
d62bcc17
WD
697 if (ret != 0) {
698 rsyserr(FERROR, errno,
699 "failed to set socket option %s", tok);
700 }
f0fca04e
AT
701 }
702
703 free(options);
704}
705
a7dc44d2
MP
706/**
707 * Become a daemon, discarding the controlling terminal
708 **/
f0fca04e
AT
709void become_daemon(void)
710{
b11ed3b1
AT
711 int i;
712
c46ded46 713 if (fork()) {
f0fca04e 714 _exit(0);
c46ded46 715 }
f0fca04e
AT
716
717 /* detach from the terminal */
4f5b0756 718#ifdef HAVE_SETSID
f0fca04e 719 setsid();
4f5b0756 720#elif defined TIOCNOTTY
c46ded46
AT
721 i = open("/dev/tty", O_RDWR);
722 if (i >= 0) {
9c07d253 723 ioctl(i, (int)TIOCNOTTY, (char *)0);
c46ded46 724 close(i);
f0fca04e 725 }
f0fca04e 726#endif
b11ed3b1 727 /* make sure that stdin, stdout an stderr don't stuff things
a3a84107 728 * up (library functions, for example) */
9c07d253
WD
729 for (i = 0; i < 3; i++) {
730 close(i);
b11ed3b1
AT
731 open("/dev/null", O_RDWR);
732 }
bc2e93eb 733}
ff8b29b8 734
eecd22ff 735
a7dc44d2
MP
736/**
737 * This is like socketpair but uses tcp. It is used by the Samba
738 * regression test code.
9c07d253 739 *
a7dc44d2
MP
740 * The function guarantees that nobody else can attach to the socket,
741 * or if they do that this function fails and the socket gets closed
742 * returns 0 on success, -1 on failure the resulting file descriptors
743 * are symmetrical.
744 **/
eecd22ff
MP
745static int socketpair_tcp(int fd[2])
746{
747 int listener;
748 struct sockaddr_in sock;
749 struct sockaddr_in sock2;
a3a84107 750 socklen_t socklen = sizeof sock;
eb8ffa90 751 int connect_done = 0;
9c07d253 752
eecd22ff
MP
753 fd[0] = fd[1] = listener = -1;
754
a3a84107 755 memset(&sock, 0, sizeof sock);
9c07d253
WD
756
757 if ((listener = socket(PF_INET, SOCK_STREAM, 0)) == -1)
758 goto failed;
eecd22ff 759
a3a84107 760 memset(&sock2, 0, sizeof sock2);
4f5b0756 761#ifdef HAVE_SOCKADDR_IN_LEN
a3a84107 762 sock2.sin_len = sizeof sock2;
eecd22ff 763#endif
a3a84107 764 sock2.sin_family = PF_INET;
eecd22ff 765
a3a84107 766 bind(listener, (struct sockaddr *)&sock2, sizeof sock2);
eecd22ff 767
9c07d253
WD
768 if (listen(listener, 1) != 0)
769 goto failed;
eecd22ff 770
9c07d253
WD
771 if (getsockname(listener, (struct sockaddr *)&sock, &socklen) != 0)
772 goto failed;
eecd22ff 773
9c07d253
WD
774 if ((fd[1] = socket(PF_INET, SOCK_STREAM, 0)) == -1)
775 goto failed;
eecd22ff
MP
776
777 set_nonblocking(fd[1]);
778
779 sock.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
780
a3a84107 781 if (connect(fd[1], (struct sockaddr *)&sock, sizeof sock) == -1) {
9c07d253
WD
782 if (errno != EINPROGRESS)
783 goto failed;
784 } else
eecd22ff 785 connect_done = 1;
eecd22ff 786
9c07d253
WD
787 if ((fd[0] = accept(listener, (struct sockaddr *)&sock, &socklen)) == -1)
788 goto failed;
eecd22ff
MP
789
790 close(listener);
ab217f7f
WD
791 listener = -1;
792
793 set_blocking(fd[1]);
794
eecd22ff 795 if (connect_done == 0) {
a3a84107 796 if (connect(fd[1], (struct sockaddr *)&sock, sizeof sock) != 0
9c07d253
WD
797 && errno != EISCONN)
798 goto failed;
eecd22ff
MP
799 }
800
eecd22ff
MP
801 /* all OK! */
802 return 0;
803
804 failed:
9c07d253
WD
805 if (fd[0] != -1)
806 close(fd[0]);
807 if (fd[1] != -1)
808 close(fd[1]);
809 if (listener != -1)
810 close(listener);
eecd22ff
MP
811 return -1;
812}
813
814
d02984bb
MP
815
816/**
817 * Run a program on a local tcp socket, so that we can talk to it's
255810c0
MP
818 * stdin and stdout. This is used to fake a connection to a daemon
819 * for testing -- not for the normal case of running SSH.
d02984bb
MP
820 *
821 * @return a socket which is attached to a subprocess running
822 * "prog". stdin and stdout are attached. stderr is left attached to
823 * the original stderr
824 **/
eecd22ff
MP
825int sock_exec(const char *prog)
826{
827 int fd[2];
9c07d253 828
eecd22ff 829 if (socketpair_tcp(fd) != 0) {
d62bcc17 830 rsyserr(FERROR, errno, "socketpair_tcp failed");
eecd22ff
MP
831 return -1;
832 }
df5cd107
WD
833 if (verbose >= 2)
834 rprintf(FINFO, "Running socket program: \"%s\"\n", prog);
eecd22ff
MP
835 if (fork() == 0) {
836 close(fd[0]);
837 close(0);
838 close(1);
839 dup(fd[1]);
840 dup(fd[1]);
9c07d253 841 exit(system(prog));
eecd22ff 842 }
9c07d253 843 close(fd[1]);
eecd22ff
MP
844 return fd[0];
845}