Commit | Line | Data |
---|---|---|
0f78b815 WD |
1 | /* |
2 | * Routines used to setup various kinds of inter-process pipes. | |
e2ccd357 | 3 | * |
0f78b815 WD |
4 | * Copyright (C) 1996-2000 Andrew Tridgell |
5 | * Copyright (C) 1996 Paul Mackerras | |
6 | * Copyright (C) 2001, 2002 Martin Pool <mbp@samba.org> | |
d3d07a5e | 7 | * Copyright (C) 2004-2008 Wayne Davison |
e2ccd357 | 8 | * |
fcb69e5c | 9 | * This program is free software; you can redistribute it and/or modify |
8e41b68e WD |
10 | * it under the terms of the GNU General Public License as published by |
11 | * the Free Software Foundation; either version 3 of the License, or | |
12 | * (at your option) any later version. | |
e2ccd357 | 13 | * |
fcb69e5c MP |
14 | * This program is distributed in the hope that it will be useful, |
15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
17 | * GNU General Public License for more details. | |
e2ccd357 | 18 | * |
e7c67065 | 19 | * You should have received a copy of the GNU General Public License along |
4fd842f9 | 20 | * with this program; if not, visit the http://fsf.org website. |
fcb69e5c MP |
21 | */ |
22 | ||
b35d0d8e MP |
23 | #include "rsync.h" |
24 | ||
72a6e631 WD |
25 | extern int am_sender; |
26 | extern int am_server; | |
27 | extern int blocking_io; | |
72a6e631 | 28 | extern int filesfrom_fd; |
c5b7aa15 | 29 | extern mode_t orig_umask; |
887d7455 | 30 | extern char *logfile_name; |
ba30fb5c | 31 | extern struct chmod_mode_struct *chmod_modes; |
72a6e631 | 32 | |
b35d0d8e | 33 | /** |
20c1926a | 34 | * Create a child connected to us via its stdin/stdout. |
b35d0d8e | 35 | * |
e2ccd357 WD |
36 | * This is derived from CVS code |
37 | * | |
b35d0d8e MP |
38 | * Note that in the child STDIN is set to blocking and STDOUT |
39 | * is set to non-blocking. This is necessary as rsh relies on stdin being blocking | |
40 | * and ssh relies on stdout being non-blocking | |
41 | * | |
42 | * If blocking_io is set then use blocking io on both fds. That can be | |
43 | * used to cope with badly broken rsh implementations like the one on | |
44 | * Solaris. | |
45 | **/ | |
46 | pid_t piped_child(char **command, int *f_in, int *f_out) | |
47 | { | |
48 | pid_t pid; | |
49 | int to_child_pipe[2]; | |
50 | int from_child_pipe[2]; | |
e2ccd357 | 51 | |
4a19c3b2 | 52 | if (verbose >= 2) |
0b515981 | 53 | print_child_argv("opening connection using:", command); |
b35d0d8e MP |
54 | |
55 | if (fd_pair(to_child_pipe) < 0 || fd_pair(from_child_pipe) < 0) { | |
d62bcc17 | 56 | rsyserr(FERROR, errno, "pipe"); |
b35d0d8e MP |
57 | exit_cleanup(RERR_IPC); |
58 | } | |
59 | ||
b35d0d8e MP |
60 | pid = do_fork(); |
61 | if (pid == -1) { | |
d62bcc17 | 62 | rsyserr(FERROR, errno, "fork"); |
b35d0d8e MP |
63 | exit_cleanup(RERR_IPC); |
64 | } | |
65 | ||
66 | if (pid == 0) { | |
b35d0d8e MP |
67 | if (dup2(to_child_pipe[0], STDIN_FILENO) < 0 || |
68 | close(to_child_pipe[1]) < 0 || | |
69 | close(from_child_pipe[0]) < 0 || | |
70 | dup2(from_child_pipe[1], STDOUT_FILENO) < 0) { | |
d62bcc17 | 71 | rsyserr(FERROR, errno, "Failed to dup/close"); |
b35d0d8e MP |
72 | exit_cleanup(RERR_IPC); |
73 | } | |
74 | if (to_child_pipe[0] != STDIN_FILENO) | |
75 | close(to_child_pipe[0]); | |
76 | if (from_child_pipe[1] != STDOUT_FILENO) | |
77 | close(from_child_pipe[1]); | |
78 | umask(orig_umask); | |
79 | set_blocking(STDIN_FILENO); | |
7d3f8ae2 | 80 | if (blocking_io > 0) |
b35d0d8e | 81 | set_blocking(STDOUT_FILENO); |
b35d0d8e | 82 | execvp(command[0], command); |
45c49b52 | 83 | rsyserr(FERROR, errno, "Failed to exec %s", command[0]); |
b35d0d8e MP |
84 | exit_cleanup(RERR_IPC); |
85 | } | |
86 | ||
87 | if (close(from_child_pipe[1]) < 0 || close(to_child_pipe[0]) < 0) { | |
d62bcc17 | 88 | rsyserr(FERROR, errno, "Failed to close"); |
b35d0d8e MP |
89 | exit_cleanup(RERR_IPC); |
90 | } | |
91 | ||
92 | *f_in = from_child_pipe[0]; | |
93 | *f_out = to_child_pipe[1]; | |
94 | ||
95 | return pid; | |
96 | } | |
97 | ||
391516da | 98 | /* This function forks a child which calls child_main(). First, |
b9f592fb WD |
99 | * however, it has to establish communication paths to and from the |
100 | * newborn child. It creates two socket pairs -- one for writing to | |
101 | * the child (from the parent) and one for reading from the child | |
102 | * (writing to the parent). Since that's four socket ends, each | |
103 | * process has to close the two ends it doesn't need. The remaining | |
104 | * two socket ends are retained for reading and writing. In the | |
105 | * child, the STDIN and STDOUT file descriptors refer to these | |
106 | * sockets. In the parent, the function arguments f_in and f_out are | |
391516da | 107 | * set to refer to these sockets. */ |
b9f592fb | 108 | pid_t local_child(int argc, char **argv, int *f_in, int *f_out, |
b44be3e9 | 109 | int (*child_main)(int, char*[])) |
b35d0d8e MP |
110 | { |
111 | pid_t pid; | |
112 | int to_child_pipe[2]; | |
113 | int from_child_pipe[2]; | |
b35d0d8e | 114 | |
ba30fb5c WD |
115 | /* The parent process is always the sender for a local rsync. */ |
116 | assert(am_sender); | |
117 | ||
b35d0d8e MP |
118 | if (fd_pair(to_child_pipe) < 0 || |
119 | fd_pair(from_child_pipe) < 0) { | |
d62bcc17 | 120 | rsyserr(FERROR, errno, "pipe"); |
b35d0d8e MP |
121 | exit_cleanup(RERR_IPC); |
122 | } | |
123 | ||
8c90957f | 124 | pid = do_fork(); |
b35d0d8e | 125 | if (pid == -1) { |
d62bcc17 | 126 | rsyserr(FERROR, errno, "fork"); |
b35d0d8e MP |
127 | exit_cleanup(RERR_IPC); |
128 | } | |
129 | ||
130 | if (pid == 0) { | |
ba30fb5c | 131 | am_sender = 0; |
e2ccd357 | 132 | am_server = 1; |
ba30fb5c WD |
133 | filesfrom_fd = -1; |
134 | chmod_modes = NULL; /* Let the sending side handle this. */ | |
7c2a9e76 | 135 | |
ddc8110d WD |
136 | /* Let the client side handle this. */ |
137 | if (logfile_name) { | |
138 | logfile_name = NULL; | |
139 | logfile_close(); | |
140 | } | |
141 | ||
b35d0d8e MP |
142 | if (dup2(to_child_pipe[0], STDIN_FILENO) < 0 || |
143 | close(to_child_pipe[1]) < 0 || | |
144 | close(from_child_pipe[0]) < 0 || | |
145 | dup2(from_child_pipe[1], STDOUT_FILENO) < 0) { | |
d62bcc17 | 146 | rsyserr(FERROR, errno, "Failed to dup/close"); |
b35d0d8e MP |
147 | exit_cleanup(RERR_IPC); |
148 | } | |
3896bca4 WD |
149 | if (to_child_pipe[0] != STDIN_FILENO) |
150 | close(to_child_pipe[0]); | |
151 | if (from_child_pipe[1] != STDOUT_FILENO) | |
152 | close(from_child_pipe[1]); | |
44c4492a | 153 | #ifdef ICONV_CONST |
b61665d5 | 154 | setup_iconv(); |
44c4492a | 155 | #endif |
b35d0d8e MP |
156 | child_main(argc, argv); |
157 | } | |
158 | ||
159 | if (close(from_child_pipe[1]) < 0 || | |
160 | close(to_child_pipe[0]) < 0) { | |
e2ccd357 | 161 | rsyserr(FERROR, errno, "Failed to close"); |
b35d0d8e MP |
162 | exit_cleanup(RERR_IPC); |
163 | } | |
164 | ||
165 | *f_in = from_child_pipe[0]; | |
166 | *f_out = to_child_pipe[1]; | |
e2ccd357 | 167 | |
b35d0d8e MP |
168 | return pid; |
169 | } |