- defer the error message from the options parsing until after the
[rsync/rsync.git] / clientserver.c
CommitLineData
3591c066
AT
1/*
2 Copyright (C) Andrew Tridgell 1998
3
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17*/
18
19/* the socket based protocol for setting up a connection wit rsyncd */
20
21#include "rsync.h"
22
23extern int module_id;
24extern int read_only;
25extern int verbose;
26extern int rsync_port;
27
28int start_socket_client(char *host, char *path, int argc, char *argv[])
29{
30 int fd, i;
31 char *sargs[MAX_ARGS];
32 int sargc=0;
e42c9458 33 char line[MAXPATHLEN];
bcb7e502 34 char *p, *user=NULL;
13c5fc0e 35 extern int remote_version;
8d9dc9f9
AT
36 extern int am_client;
37 extern int am_sender;
3591c066 38
bcb7e502
AT
39 p = strchr(host, '@');
40 if (p) {
41 user = host;
42 host = p+1;
43 *p = 0;
44 }
45
46 if (!user) user = getenv("USER");
47 if (!user) user = getenv("LOGNAME");
48
8d9dc9f9
AT
49 am_client = 1;
50
3591c066
AT
51 fd = open_socket_out(host, rsync_port);
52 if (fd == -1) {
53 exit_cleanup(1);
54 }
55
56 server_options(sargs,&sargc);
57
58 sargs[sargc++] = ".";
59
60 if (path && *path)
61 sargs[sargc++] = path;
62
63 sargs[sargc] = NULL;
64
91eee594
AT
65 io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION);
66
3591c066
AT
67 if (!read_line(fd, line, sizeof(line)-1)) {
68 return -1;
69 }
70
13c5fc0e 71 if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) {
3591c066
AT
72 return -1;
73 }
74
2c91d3d3
AT
75 p = strchr(path,'/');
76 if (p) *p = 0;
77 io_printf(fd,"%s\n",path);
78 if (p) *p = '/';
79
3591c066
AT
80 while (1) {
81 if (!read_line(fd, line, sizeof(line)-1)) {
82 return -1;
83 }
31593dd6 84
31593dd6 85 if (strncmp(line,"@RSYNCD: AUTHREQD ",18) == 0) {
bcb7e502 86 auth_client(fd, user, line+18);
31593dd6
AT
87 continue;
88 }
31593dd6 89
3591c066
AT
90 if (strcmp(line,"@RSYNCD: OK") == 0) break;
91 rprintf(FINFO,"%s\n", line);
92 }
93
94 for (i=0;i<sargc;i++) {
95 io_printf(fd,"%s\n", sargs[i]);
96 }
97 io_printf(fd,"\n");
98
8d9dc9f9
AT
99 if (remote_version > 17 && !am_sender)
100 io_start_multiplex_in(fd);
101
3591c066
AT
102 return client_run(fd, fd, -1, argc, argv);
103}
104
105
106
107static int rsync_module(int fd, int i)
108{
109 int argc=0;
110 char *argv[MAX_ARGS];
111 char **argp;
e42c9458 112 char line[MAXPATHLEN];
1a016bfd
AT
113 uid_t uid = (uid_t)-2;
114 gid_t gid = (gid_t)-2;
8ef4ffd6 115 char *p;
56c473b7
AT
116 char *addr = client_addr(fd);
117 char *host = client_name(fd);
874895d5 118 char *name = lp_name(i);
d0d56395 119 char *user;
874895d5 120 int start_glob=0;
41979ff8 121 int ret;
7b372642
AT
122 char *request=NULL;
123 extern int am_sender;
8d9dc9f9 124 extern int remote_version;
943882a2 125 extern int am_root;
56c473b7
AT
126
127 if (!allow_access(addr, host, lp_hosts_allow(i), lp_hosts_deny(i))) {
128 rprintf(FERROR,"rsync denied on module %s from %s (%s)\n",
874895d5 129 name, client_name(fd), client_addr(fd));
c8e78d87 130 io_printf(fd,"@ERROR: access denied to %s from %s (%s)\n",
874895d5 131 name, client_name(fd), client_addr(fd));
56c473b7
AT
132 return -1;
133 }
3591c066 134
0c515f17 135 if (!claim_connection(lp_lock_file(), lp_max_connections())) {
8d72ef6e
AT
136 if (errno) {
137 rprintf(FERROR,"failed to open lock file %s : %s\n",
138 lp_lock_file(), strerror(errno));
139 io_printf(fd,"@ERROR: failed to open lock file %s : %s\n",
140 lp_lock_file(), strerror(errno));
141 } else {
142 rprintf(FERROR,"max connections (%d) reached\n",
143 lp_max_connections());
144 io_printf(fd,"@ERROR: max connections (%d) reached - try again later\n", lp_max_connections());
145 }
0c515f17
AT
146 return -1;
147 }
148
31593dd6 149
d0d56395
AT
150 user = auth_server(fd, i, addr, "@RSYNCD: AUTHREQD ");
151
152 if (!user) {
153 rprintf(FERROR,"auth failed on module %s from %s (%s)\n",
154 name, client_name(fd), client_addr(fd));
155 io_printf(fd,"@ERROR: auth failed on module %s\n",name);
156 return -1;
157 }
158
3591c066
AT
159 module_id = i;
160
161 if (lp_read_only(i))
162 read_only = 1;
163
8ef4ffd6
AT
164 p = lp_uid(i);
165 if (!name_to_uid(p, &uid)) {
166 if (!isdigit(*p)) {
167 rprintf(FERROR,"Invalid uid %s\n", p);
c8e78d87 168 io_printf(fd,"@ERROR: invalid uid\n");
8ef4ffd6
AT
169 return -1;
170 }
171 uid = atoi(p);
172 }
173
174 p = lp_gid(i);
175 if (!name_to_gid(p, &gid)) {
176 if (!isdigit(*p)) {
177 rprintf(FERROR,"Invalid gid %s\n", p);
c8e78d87 178 io_printf(fd,"@ERROR: invalid gid\n");
8ef4ffd6
AT
179 return -1;
180 }
181 gid = atoi(p);
182 }
183
8f3a2d54 184 p = lp_exclude_from(i);
2b6b4d53 185 add_exclude_file(p, 1, 0);
8f3a2d54 186
5805327b 187 p = lp_exclude(i);
8f3a2d54
AT
188 add_exclude_line(p);
189
1a016bfd
AT
190 log_open();
191
3591c066 192 if (chroot(lp_path(i))) {
1a016bfd 193 rprintf(FERROR,"chroot %s failed\n", lp_path(i));
3591c066
AT
194 io_printf(fd,"@ERROR: chroot failed\n");
195 return -1;
196 }
197
198 if (chdir("/")) {
1a016bfd 199 rprintf(FERROR,"chdir %s failed\n", lp_path(i));
3591c066
AT
200 io_printf(fd,"@ERROR: chdir failed\n");
201 return -1;
202 }
203
1a016bfd
AT
204 if (setgid(gid) || getgid() != gid) {
205 rprintf(FERROR,"setgid %d failed\n", gid);
3591c066
AT
206 io_printf(fd,"@ERROR: setgid failed\n");
207 return -1;
208 }
209
1a016bfd
AT
210 if (setuid(uid) || getuid() != uid) {
211 rprintf(FERROR,"setuid %d failed\n", uid);
3591c066
AT
212 io_printf(fd,"@ERROR: setuid failed\n");
213 return -1;
214 }
215
943882a2
AT
216 am_root = (getuid() == 0);
217
3591c066
AT
218 io_printf(fd,"@RSYNCD: OK\n");
219
220 argv[argc++] = "rsyncd";
221
222 while (1) {
223 if (!read_line(fd, line, sizeof(line)-1)) {
224 return -1;
225 }
226
227 if (!*line) break;
228
874895d5
AT
229 p = line;
230
874895d5 231 argv[argc] = strdup(p);
3591c066
AT
232 if (!argv[argc]) {
233 return -1;
234 }
235
874895d5 236 if (start_glob) {
1a016bfd 237 if (start_glob == 1) {
7b372642 238 request = strdup(p);
1a016bfd
AT
239 start_glob++;
240 }
087bf010 241 glob_expand(name, argv, &argc, MAX_ARGS);
874895d5
AT
242 } else {
243 argc++;
244 }
245
246 if (strcmp(line,".") == 0) {
247 start_glob = 1;
248 }
249
3591c066
AT
250 if (argc == MAX_ARGS) {
251 return -1;
252 }
253 }
254
41979ff8 255 ret = parse_arguments(argc, argv);
3591c066 256
7b372642 257 if (request) {
d0d56395
AT
258 if (*user) {
259 rprintf(FINFO,"rsync %s %s from %s@%s (%s)\n",
260 am_sender?"on":"to",
261 request, user, host, addr);
262 } else {
263 rprintf(FINFO,"rsync %s %s from %s (%s)\n",
264 am_sender?"on":"to",
265 request, host, addr);
266 }
7b372642
AT
267 free(request);
268 }
269
3591c066
AT
270 /* don't allow the logs to be flooded too fast */
271 if (verbose > 1) verbose = 1;
272
273 argc -= optind;
274 argp = argv + optind;
275 optind = 0;
276
8d9dc9f9
AT
277 if (remote_version > 17 && am_sender)
278 io_start_multiplex_out(fd);
279
41979ff8
AT
280 if (!ret) {
281 rprintf(FERROR,"Error parsing options (unsupported option?) - aborting\n");
282 exit_cleanup(1);
283 }
284
3591c066
AT
285 start_server(fd, fd, argc, argp);
286
287 return 0;
288}
289
7a6421fa
AT
290/* send a list of available modules to the client. Don't list those
291 with "list = False". */
3591c066
AT
292static void send_listing(int fd)
293{
294 int n = lp_numservices();
295 int i;
296
297 for (i=0;i<n;i++)
298 if (lp_list(i))
299 io_printf(fd, "%-15s\t%s\n", lp_name(i), lp_comment(i));
300}
301
302/* this is called when a socket connection is established to a client
303 and we want to start talking. The setup of the system is done from
304 here */
305static int start_daemon(int fd)
306{
7a6421fa 307 char line[200];
3591c066 308 char *motd;
8ef4ffd6 309 int i = -1;
4cdf25e4 310 extern char *config_file;
13c5fc0e 311 extern int remote_version;
4cdf25e4 312
f9e940ef 313 if (!lp_load(config_file, 0)) {
4cdf25e4
AT
314 exit_cleanup(1);
315 }
3591c066
AT
316
317 set_socket_options(fd,"SO_KEEPALIVE");
a6801c39
AT
318 set_socket_options(fd,lp_socket_options());
319
3591c066
AT
320
321 io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION);
322
323 motd = lp_motd_file();
324 if (*motd) {
325 FILE *f = fopen(motd,"r");
326 while (f && !feof(f)) {
327 int len = fread(line, 1, sizeof(line)-1, f);
328 if (len > 0) {
329 line[len] = 0;
330 io_printf(fd,"%s", line);
331 }
332 }
333 if (f) fclose(f);
334 io_printf(fd,"\n");
335 }
336
91eee594
AT
337 if (!read_line(fd, line, sizeof(line)-1)) {
338 return -1;
339 }
340
341 if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) {
c8e78d87 342 io_printf(fd,"@ERROR: protocol startup error\n");
91eee594
AT
343 return -1;
344 }
345
8ef4ffd6 346 while (i == -1) {
3591c066
AT
347 line[0] = 0;
348 if (!read_line(fd, line, sizeof(line)-1)) {
349 return -1;
350 }
351
352 if (!*line || strcmp(line,"#list")==0) {
353 send_listing(fd);
354 return -1;
355 }
356
357 if (*line == '#') {
358 /* it's some sort of command that I don't understand */
c8e78d87 359 io_printf(fd,"@ERROR: Unknown command '%s'\n", line);
3591c066
AT
360 return -1;
361 }
362
363 i = lp_number(line);
364 if (i == -1) {
c8e78d87 365 io_printf(fd,"@ERROR: Unknown module '%s'\n", line);
3591c066
AT
366 return -1;
367 }
3591c066
AT
368 }
369
8ef4ffd6 370 return rsync_module(fd, i);
3591c066
AT
371}
372
373
374int daemon_main(void)
375{
f9e940ef 376 extern char *config_file;
1a016bfd 377
5243c216
AT
378 /* this ensures that we don't call getcwd after the chroot,
379 which doesn't work on platforms that use popen("pwd","r")
380 for getcwd */
381 push_dir("/", 0);
382
3591c066 383 if (is_a_socket(STDIN_FILENO)) {
41979ff8
AT
384 int i;
385
386 /* we are running via inetd - close off stdout and
387 stderr so that library functions (and getopt) don't
388 try to use them. Redirect them to /dev/null */
389 for (i=1;i<3;i++) {
390 close(i);
391 open("/dev/null", O_RDWR);
392 }
3591c066
AT
393 return start_daemon(STDIN_FILENO);
394 }
395
396 become_daemon();
397
f9e940ef
AT
398 if (!lp_load(config_file, 1)) {
399 fprintf(stderr,"failed to load config file %s\n", config_file);
400 exit_cleanup(1);
401 }
402
403 log_open();
404
e42c9458
AT
405 rprintf(FINFO,"rsyncd version %s starting\n",VERSION);
406
8ef4ffd6
AT
407 start_accept_loop(rsync_port, start_daemon);
408 return -1;
3591c066
AT
409}
410