Annotation of src/usr.bin/rdistd/server.c, Revision 1.42
1.42 ! millert 1: /* $OpenBSD: server.c,v 1.41 2016/03/30 17:03:06 millert Exp $ */
1.4 deraadt 2:
1.1 dm 3: /*
4: * Copyright (c) 1983 Regents of the University of California.
5: * All rights reserved.
6: *
7: * Redistribution and use in source and binary forms, with or without
8: * modification, are permitted provided that the following conditions
9: * are met:
10: * 1. Redistributions of source code must retain the above copyright
11: * notice, this list of conditions and the following disclaimer.
12: * 2. Redistributions in binary form must reproduce the above copyright
13: * notice, this list of conditions and the following disclaimer in the
14: * documentation and/or other materials provided with the distribution.
1.15 millert 15: * 3. Neither the name of the University nor the names of its contributors
1.1 dm 16: * may be used to endorse or promote products derived from this software
17: * without specific prior written permission.
18: *
19: * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22: * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29: * SUCH DAMAGE.
30: */
1.28 guenther 31:
1.35 guenther 32: #include <ctype.h>
1.28 guenther 33: #include <dirent.h>
1.35 guenther 34: #include <errno.h>
35: #include <fcntl.h>
36: #include <grp.h>
37: #include <limits.h>
38: #include <stdio.h>
39: #include <stdlib.h>
40: #include <string.h>
41: #include <time.h>
42: #include <unistd.h>
1.28 guenther 43:
1.35 guenther 44: #include "server.h"
1.1 dm 45:
46: /*
47: * Server routines
48: */
49:
50: char tempname[sizeof _RDIST_TMP + 1]; /* Tmp file name */
51: char buf[BUFSIZ]; /* general purpose buffer */
1.34 deraadt 52: char target[PATH_MAX]; /* target/source directory name */
1.1 dm 53: char *ptarget; /* pointer to end of target name */
54: int catname = 0; /* cat name to target name */
55: char *sptarget[32]; /* stack of saved ptarget's for directories */
56: char *fromhost = NULL; /* Client hostname */
1.22 krw 57: static int64_t min_freespace = 0; /* Minimium free space on a filesystem */
58: static int64_t min_freefiles = 0; /* Minimium free # files on a filesystem */
1.1 dm 59: int oumask; /* Old umask */
60:
1.14 millert 61: static int cattarget(char *);
1.29 guenther 62: static int setownership(char *, int, uid_t, gid_t, int);
1.14 millert 63: static int setfilemode(char *, int, int, int);
64: static int fchog(int, char *, char *, char *, int);
65: static int removefile(struct stat *, int);
66: static void doclean(char *);
67: static void clean(char *);
68: static void dospecial(char *);
69: static void docmdspecial(void);
70: static void query(char *);
71: static int chkparent(char *, opt_t);
72: static char *savetarget(char *, opt_t);
1.19 millert 73: static void recvfile(char *, opt_t, int, char *, char *, time_t, time_t, off_t);
1.14 millert 74: static void recvdir(opt_t, int, char *, char *);
75: static void recvlink(char *, opt_t, int, off_t);
76: static void hardlink(char *);
77: static void setconfig(char *);
78: static void recvit(char *, int);
79: static void dochmog(char *);
80: static void settarget(char *, int);
81:
1.1 dm 82: /*
83: * Cat "string" onto the target buffer with error checking.
84: */
1.14 millert 85: static int
86: cattarget(char *string)
1.1 dm 87: {
88: if (strlen(string) + strlen(target) + 2 > sizeof(target)) {
89: message(MT_INFO, "target buffer is not large enough.");
90: return(-1);
91: }
92: if (!ptarget) {
93: message(MT_INFO, "NULL target pointer set.");
94: return(-10);
95: }
96:
1.14 millert 97: (void) snprintf(ptarget, sizeof(target) - (ptarget - target),
98: "/%s", string);
1.1 dm 99:
100: return(0);
101: }
102:
103: /*
104: * Set uid and gid ownership of a file.
105: */
1.14 millert 106: static int
1.33 guenther 107: setownership(char *file, int fd, uid_t uid, gid_t gid, int islink)
1.1 dm 108: {
1.38 guenther 109: static int is_root = -1;
1.1 dm 110: int status = -1;
111:
112: /*
113: * We assume only the Superuser can change uid ownership.
114: */
1.38 guenther 115: switch (is_root) {
116: case -1:
117: is_root = getuid() == 0;
118: if (is_root)
119: break;
120: /* FALLTHROUGH */
121: case 0:
1.14 millert 122: uid = -1;
1.38 guenther 123: break;
124: case 1:
125: break;
126: }
1.14 millert 127:
1.33 guenther 128: if (fd != -1 && !islink)
1.29 guenther 129: status = fchown(fd, uid, gid);
1.38 guenther 130: else
131: status = fchownat(AT_FDCWD, file, uid, gid,
132: AT_SYMLINK_NOFOLLOW);
1.1 dm 133:
1.14 millert 134: if (status < 0) {
1.29 guenther 135: if (uid == (uid_t)-1)
1.1 dm 136: message(MT_NOTICE, "%s: chgrp %d failed: %s",
1.14 millert 137: target, gid, SYSERR);
138: else
1.38 guenther 139: message(MT_NOTICE, "%s: chown %d:%d failed: %s",
1.14 millert 140: target, uid, gid, SYSERR);
141: return(-1);
1.1 dm 142: }
143:
144: return(0);
145: }
146:
147: /*
148: * Set mode of a file
149: */
1.14 millert 150: static int
1.33 guenther 151: setfilemode(char *file, int fd, int mode, int islink)
1.1 dm 152: {
153: int status = -1;
154:
155: if (mode == -1)
156: return(0);
157:
1.33 guenther 158: if (islink)
1.30 guenther 159: status = fchmodat(AT_FDCWD, file, mode, AT_SYMLINK_NOFOLLOW);
1.14 millert 160:
1.33 guenther 161: if (fd != -1 && !islink)
1.1 dm 162: status = fchmod(fd, mode);
163:
1.33 guenther 164: if (status < 0 && !islink)
1.1 dm 165: status = chmod(file, mode);
166:
167: if (status < 0) {
168: message(MT_NOTICE, "%s: chmod failed: %s", target, SYSERR);
169: return(-1);
170: }
171:
172: return(0);
173: }
174: /*
175: * Change owner, group and mode of file.
176: */
1.14 millert 177: static int
178: fchog(int fd, char *file, char *owner, char *group, int mode)
1.1 dm 179: {
1.14 millert 180: static struct group *gr = NULL;
1.10 mpech 181: int i;
1.14 millert 182: struct stat st;
1.29 guenther 183: uid_t uid;
184: gid_t gid;
185: gid_t primegid = (gid_t)-2;
1.1 dm 186:
187: uid = userid;
188: if (userid == 0) { /* running as root; take anything */
189: if (*owner == ':') {
1.29 guenther 190: uid = (uid_t) atoi(owner + 1);
1.14 millert 191: } else if (pw == NULL || strcmp(owner, pw->pw_name) != 0) {
1.1 dm 192: if ((pw = getpwnam(owner)) == NULL) {
193: if (mode != -1 && IS_ON(mode, S_ISUID)) {
194: message(MT_NOTICE,
195: "%s: unknown login name \"%s\", clearing setuid",
196: target, owner);
197: mode &= ~S_ISUID;
198: uid = 0;
199: } else
200: message(MT_NOTICE,
201: "%s: unknown login name \"%s\"",
202: target, owner);
1.14 millert 203: } else
204: uid = pw->pw_uid;
1.1 dm 205: } else {
1.14 millert 206: uid = pw->pw_uid;
207: primegid = pw->pw_gid;
1.1 dm 208: }
209: if (*group == ':') {
1.29 guenther 210: gid = (gid_t)atoi(group + 1);
1.1 dm 211: goto ok;
212: }
213: } else { /* not root, setuid only if user==owner */
214: struct passwd *lupw;
215:
216: if (mode != -1) {
217: if (IS_ON(mode, S_ISUID) &&
218: strcmp(locuser, owner) != 0)
219: mode &= ~S_ISUID;
220: if (mode)
221: mode &= ~S_ISVTX; /* and strip sticky too */
222: }
223:
224: if ((lupw = getpwnam(locuser)) != NULL)
225: primegid = lupw->pw_gid;
226: }
227:
1.29 guenther 228: gid = (gid_t)-1;
1.14 millert 229: if (gr == NULL || strcmp(group, gr->gr_name) != 0) {
230: if ((*group == ':' &&
231: (getgrgid(gid = atoi(group + 1)) == NULL))
232: || ((gr = (struct group *)getgrnam(group)) == NULL)) {
1.1 dm 233: if (mode != -1 && IS_ON(mode, S_ISGID)) {
234: message(MT_NOTICE,
235: "%s: unknown group \"%s\", clearing setgid",
236: target, group);
237: mode &= ~S_ISGID;
238: } else
239: message(MT_NOTICE,
240: "%s: unknown group \"%s\"",
241: target, group);
1.14 millert 242: } else
243: gid = gr->gr_gid;
244: } else
245: gid = gr->gr_gid;
1.1 dm 246:
247: if (userid && gid >= 0 && gid != primegid) {
248: if (gr)
249: for (i = 0; gr->gr_mem[i] != NULL; i++)
1.14 millert 250: if (strcmp(locuser, gr->gr_mem[i]) == 0)
1.1 dm 251: goto ok;
252: if (mode != -1 && IS_ON(mode, S_ISGID)) {
253: message(MT_NOTICE,
254: "%s: user %s not in group %s, clearing setgid",
255: target, locuser, group);
256: mode &= ~S_ISGID;
257: }
1.29 guenther 258: gid = (gid_t)-1;
1.1 dm 259: }
260: ok:
1.14 millert 261: if (stat(file, &st) == -1) {
262: error("%s: Stat failed %s", file, SYSERR);
263: return -1;
264: }
1.1 dm 265: /*
266: * Set uid and gid ownership. If that fails, strip setuid and
267: * setgid bits from mode. Once ownership is set, successful
268: * or otherwise, set the new file mode.
269: */
1.14 millert 270: if (setownership(file, fd, uid, gid, S_ISLNK(st.st_mode)) < 0) {
1.1 dm 271: if (mode != -1 && IS_ON(mode, S_ISUID)) {
272: message(MT_NOTICE,
273: "%s: chown failed, clearing setuid", target);
274: mode &= ~S_ISUID;
275: }
276: if (mode != -1 && IS_ON(mode, S_ISGID)) {
277: message(MT_NOTICE,
278: "%s: chown failed, clearing setgid", target);
279: mode &= ~S_ISGID;
280: }
281: }
1.14 millert 282: (void) setfilemode(file, fd, mode, S_ISLNK(st.st_mode));
1.1 dm 283:
284:
285: return(0);
286: }
287:
288: /*
289: * Remove a file or directory (recursively) and send back an acknowledge
290: * or an error message.
291: */
1.14 millert 292: static int
293: removefile(struct stat *statb, int silent)
1.1 dm 294: {
295: DIR *d;
1.28 guenther 296: static struct dirent *dp;
1.10 mpech 297: char *cp;
1.1 dm 298: struct stat stb;
299: char *optarget;
300: int len, failures = 0;
301:
302: switch (statb->st_mode & S_IFMT) {
303: case S_IFREG:
304: case S_IFLNK:
1.14 millert 305: case S_IFCHR:
306: case S_IFBLK:
307: case S_IFSOCK:
308: case S_IFIFO:
1.1 dm 309: if (unlink(target) < 0) {
310: if (errno == ETXTBSY) {
1.14 millert 311: if (!silent)
312: message(MT_REMOTE|MT_NOTICE,
313: "%s: unlink failed: %s",
314: target, SYSERR);
1.1 dm 315: return(0);
316: } else {
317: error("%s: unlink failed: %s", target, SYSERR);
318: return(-1);
319: }
320: }
321: goto removed;
322:
323: case S_IFDIR:
324: break;
325:
326: default:
327: error("%s: not a plain file", target);
328: return(-1);
329: }
330:
331: errno = 0;
332: if ((d = opendir(target)) == NULL) {
333: error("%s: opendir failed: %s", target, SYSERR);
334: return(-1);
335: }
336:
337: optarget = ptarget;
338: len = ptarget - target;
1.14 millert 339: while ((dp = readdir(d)) != NULL) {
1.26 guenther 340: if (dp->d_name[0] == '.' && (dp->d_name[1] == '\0' ||
1.29 guenther 341: (dp->d_name[1] == '.' && dp->d_name[2] == '\0')))
1.1 dm 342: continue;
343:
1.34 deraadt 344: if (len + 1 + (int)strlen(dp->d_name) >= PATH_MAX - 1) {
1.14 millert 345: if (!silent)
346: message(MT_REMOTE|MT_WARNING,
347: "%s/%s: Name too long",
348: target, dp->d_name);
1.1 dm 349: continue;
350: }
351: ptarget = optarget;
352: *ptarget++ = '/';
1.17 otto 353: cp = dp->d_name;
1.14 millert 354: while ((*ptarget++ = *cp++) != '\0')
355: continue;
1.1 dm 356: ptarget--;
357: if (lstat(target, &stb) < 0) {
1.14 millert 358: if (!silent)
359: message(MT_REMOTE|MT_WARNING,
360: "%s: lstat failed: %s",
361: target, SYSERR);
1.1 dm 362: continue;
363: }
1.14 millert 364: if (removefile(&stb, 0) < 0)
1.1 dm 365: ++failures;
366: }
367: (void) closedir(d);
368: ptarget = optarget;
369: *ptarget = CNULL;
370:
371: if (failures)
372: return(-1);
373:
374: if (rmdir(target) < 0) {
375: error("%s: rmdir failed: %s", target, SYSERR);
376: return(-1);
377: }
378: removed:
1.14 millert 379: #if NEWWAY
380: if (!silent)
381: message(MT_CHANGE|MT_REMOTE, "%s: removed", target);
382: #else
1.5 millert 383: /*
384: * We use MT_NOTICE instead of MT_CHANGE because this function is
385: * sometimes called by other functions that are suppose to return a
386: * single ack() back to the client (rdist). This is a kludge until
387: * the Rdist protocol is re-done. Sigh.
388: */
389: message(MT_NOTICE|MT_REMOTE, "%s: removed", target);
1.14 millert 390: #endif
1.1 dm 391: return(0);
392: }
393:
394: /*
395: * Check the current directory (initialized by the 'T' command to server())
396: * for extraneous files and remove them.
397: */
1.14 millert 398: static void
399: doclean(char *cp)
1.1 dm 400: {
401: DIR *d;
1.28 guenther 402: struct dirent *dp;
1.1 dm 403: struct stat stb;
404: char *optarget, *ep;
405: int len;
406: opt_t opts;
1.34 deraadt 407: char targ[PATH_MAX*4];
1.1 dm 408:
409: opts = strtol(cp, &ep, 8);
410: if (*ep != CNULL) {
411: error("clean: options not delimited");
412: return;
413: }
414: if ((d = opendir(target)) == NULL) {
415: error("%s: opendir failed: %s", target, SYSERR);
416: return;
417: }
418: ack();
419:
420: optarget = ptarget;
421: len = ptarget - target;
1.14 millert 422: while ((dp = readdir(d)) != NULL) {
1.27 guenther 423: if (dp->d_name[0] == '.' && (dp->d_name[1] == '\0' ||
1.29 guenther 424: (dp->d_name[1] == '.' && dp->d_name[2] == '\0')))
1.1 dm 425: continue;
426:
1.34 deraadt 427: if (len + 1 + (int)strlen(dp->d_name) >= PATH_MAX - 1) {
1.1 dm 428: message(MT_REMOTE|MT_WARNING, "%s/%s: Name too long",
429: target, dp->d_name);
430: continue;
431: }
432: ptarget = optarget;
433: *ptarget++ = '/';
1.17 otto 434: cp = dp->d_name;
1.14 millert 435: while ((*ptarget++ = *cp++) != '\0')
436: continue;
1.1 dm 437: ptarget--;
438: if (lstat(target, &stb) < 0) {
439: message(MT_REMOTE|MT_WARNING, "%s: lstat failed: %s",
440: target, SYSERR);
441: continue;
442: }
443:
1.14 millert 444: ENCODE(targ, dp->d_name);
445: (void) sendcmd(CC_QUERY, "%s", targ);
1.1 dm 446: (void) remline(cp = buf, sizeof(buf), TRUE);
447:
448: if (*cp != CC_YES)
449: continue;
450:
451: if (IS_ON(opts, DO_VERIFY))
452: message(MT_REMOTE|MT_INFO, "%s: need to remove",
453: target);
454: else
1.14 millert 455: (void) removefile(&stb, 0);
1.1 dm 456: }
457: (void) closedir(d);
458:
459: ptarget = optarget;
460: *ptarget = CNULL;
461: }
462:
463: /*
464: * Frontend to doclean().
465: */
1.14 millert 466: static void
467: clean(char *cp)
1.1 dm 468: {
469: doclean(cp);
470: (void) sendcmd(CC_END, NULL);
471: (void) response();
472: }
473:
474: /*
475: * Execute a shell command to handle special cases.
476: * We can't really set an alarm timeout here since we
477: * have no idea how long the command should take.
478: */
1.14 millert 479: static void
480: dospecial(char *xcmd)
1.1 dm 481: {
1.14 millert 482: char cmd[BUFSIZ];
483: if (DECODE(cmd, xcmd) == -1) {
484: error("dospecial: Cannot decode command.");
485: return;
486: }
1.1 dm 487: runcommand(cmd);
488: }
489:
490: /*
491: * Do a special cmd command. This differs from normal special
492: * commands in that it's done after an entire command has been updated.
493: * The list of updated target files is sent one at a time with RC_FILE
494: * commands. Each one is added to an environment variable defined by
495: * E_FILES. When an RC_COMMAND is finally received, the E_FILES variable
496: * is stuffed into our environment and a normal dospecial() command is run.
497: */
1.14 millert 498: static void
499: docmdspecial(void)
1.1 dm 500: {
1.10 mpech 501: char *cp;
1.1 dm 502: char *cmd, *env = NULL;
503: int n;
1.13 millert 504: size_t len;
1.1 dm 505:
506: /* We're ready */
507: ack();
508:
509: for ( ; ; ) {
510: n = remline(cp = buf, sizeof(buf), FALSE);
511: if (n <= 0) {
512: error("cmdspecial: premature end of input.");
513: return;
514: }
515:
516: switch (*cp++) {
517: case RC_FILE:
518: if (env == NULL) {
519: len = (2 * sizeof(E_FILES)) + strlen(cp) + 10;
1.40 mmcc 520: env = xmalloc(len);
1.11 deraadt 521: (void) snprintf(env, len, "export %s;%s=%s",
1.1 dm 522: E_FILES, E_FILES, cp);
523: } else {
1.13 millert 524: len = strlen(env) + 1 + strlen(cp) + 1;
1.39 mmcc 525: env = xrealloc(env, len);
1.13 millert 526: (void) strlcat(env, ":", len);
527: (void) strlcat(env, cp, len);
1.1 dm 528: }
529: ack();
530: break;
531:
532: case RC_COMMAND:
533: if (env) {
1.13 millert 534: len = strlen(env) + 1 + strlen(cp) + 1;
1.39 mmcc 535: env = xrealloc(env, len);
1.13 millert 536: (void) strlcat(env, ";", len);
537: (void) strlcat(env, cp, len);
1.1 dm 538: cmd = env;
539: } else
540: cmd = cp;
541:
542: dospecial(cmd);
543: if (env)
544: (void) free(env);
545: return;
546:
547: default:
548: error("Unknown cmdspecial command '%s'.", cp);
549: return;
550: }
551: }
552: }
553:
554: /*
555: * Query. Check to see if file exists. Return one of the following:
556: *
557: * QC_ONNFS - resides on a NFS
558: * QC_ONRO - resides on a Read-Only filesystem
559: * QC_NO - doesn't exist
560: * QC_YESsize mtime - exists and its a regular file (size & mtime of file)
561: * QC_YES - exists and its a directory or symbolic link
562: * QC_ERRMSGmessage - error message
563: */
1.14 millert 564: static void
565: query(char *xname)
1.1 dm 566: {
567: static struct stat stb;
568: int s = -1, stbvalid = 0;
1.34 deraadt 569: char name[PATH_MAX];
1.14 millert 570:
571: if (DECODE(name, xname) == -1) {
572: error("query: Cannot decode filename");
573: return;
574: }
1.1 dm 575:
576: if (catname && cattarget(name) < 0)
577: return;
578:
579: if (IS_ON(options, DO_CHKNFS)) {
580: s = is_nfs_mounted(target, &stb, &stbvalid);
581: if (s > 0)
582: (void) sendcmd(QC_ONNFS, NULL);
583:
1.9 mpech 584: /* Either the above check was true or an error occurred */
1.1 dm 585: /* and is_nfs_mounted sent the error message */
586: if (s != 0) {
587: *ptarget = CNULL;
588: return;
589: }
590: }
591:
592: if (IS_ON(options, DO_CHKREADONLY)) {
593: s = is_ro_mounted(target, &stb, &stbvalid);
594: if (s > 0)
595: (void) sendcmd(QC_ONRO, NULL);
596:
1.9 mpech 597: /* Either the above check was true or an error occurred */
1.1 dm 598: /* and is_ro_mounted sent the error message */
599: if (s != 0) {
600: *ptarget = CNULL;
601: return;
602: }
603: }
604:
605: if (IS_ON(options, DO_CHKSYM)) {
606: if (is_symlinked(target, &stb, &stbvalid) > 0) {
607: (void) sendcmd(QC_SYM, NULL);
608: return;
609: }
610: }
611:
612: /*
1.37 guenther 613: * If stbvalid is false, "stb" is not valid because the stat()
614: * by is_*_mounted() either failed or does not match "target".
1.1 dm 615: */
616: if (!stbvalid && lstat(target, &stb) < 0) {
617: if (errno == ENOENT)
618: (void) sendcmd(QC_NO, NULL);
619: else
620: error("%s: lstat failed: %s", target, SYSERR);
621: *ptarget = CNULL;
622: return;
623: }
624:
625: switch (stb.st_mode & S_IFMT) {
626: case S_IFLNK:
627: case S_IFDIR:
628: case S_IFREG:
1.23 guenther 629: (void) sendcmd(QC_YES, "%lld %lld %o %s %s",
630: (long long) stb.st_size,
631: (long long) stb.st_mtime,
1.1 dm 632: stb.st_mode & 07777,
633: getusername(stb.st_uid, target, options),
634: getgroupname(stb.st_gid, target, options));
635: break;
636:
637: default:
638: error("%s: not a file or directory", target);
639: break;
640: }
641: *ptarget = CNULL;
642: }
643:
644: /*
645: * Check to see if parent directory exists and create one if not.
646: */
1.14 millert 647: static int
648: chkparent(char *name, opt_t opts)
1.1 dm 649: {
1.10 mpech 650: char *cp;
1.1 dm 651: struct stat stb;
652: int r = -1;
653:
1.36 guenther 654: debugmsg(DM_CALL, "chkparent(%s, %#x) start\n", name, opts);
1.1 dm 655:
656: cp = strrchr(name, '/');
657: if (cp == NULL || cp == name)
658: return(0);
659:
660: *cp = CNULL;
661:
662: if (lstat(name, &stb) < 0) {
663: if (errno == ENOENT && chkparent(name, opts) >= 0) {
664: if (mkdir(name, 0777 & ~oumask) == 0) {
665: message(MT_NOTICE, "%s: mkdir", name);
666: r = 0;
667: } else
668: debugmsg(DM_MISC,
1.36 guenther 669: "chkparent(%s, %#04o) mkdir fail: %s\n",
1.1 dm 670: name, opts, SYSERR);
671: }
672: } else /* It exists */
673: r = 0;
674:
675: /* Put back what we took away */
676: *cp = '/';
677:
678: return(r);
679: }
680:
681: /*
682: * Save a copy of 'file' by renaming it.
683: */
1.14 millert 684: static char *
685: savetarget(char *file, opt_t opts)
1.1 dm 686: {
1.34 deraadt 687: static char savefile[PATH_MAX];
1.1 dm 688:
1.34 deraadt 689: if (strlen(file) + sizeof(SAVE_SUFFIX) + 1 > PATH_MAX) {
1.1 dm 690: error("%s: Cannot save: Save name too long", file);
1.7 millert 691: return(NULL);
1.1 dm 692: }
693:
1.14 millert 694: if (IS_ON(opts, DO_HISTORY)) {
695: int i;
696: struct stat st;
697: /*
698: * There is a race here, but the worst that can happen
699: * is to lose a version of the file
700: */
701: for (i = 1; i < 1000; i++) {
702: (void) snprintf(savefile, sizeof(savefile),
703: "%s;%.3d", file, i);
1.38 guenther 704: if (lstat(savefile, &st) == -1 && errno == ENOENT)
1.14 millert 705: break;
1.1 dm 706:
1.14 millert 707: }
708: if (i == 1000) {
709: message(MT_NOTICE,
710: "%s: More than 1000 versions for %s; reusing 1\n",
711: savefile, SYSERR);
712: i = 1;
713: (void) snprintf(savefile, sizeof(savefile),
714: "%s;%.3d", file, i);
715: }
716: }
717: else {
718: (void) snprintf(savefile, sizeof(savefile), "%s%s",
719: file, SAVE_SUFFIX);
720:
721: if (unlink(savefile) != 0 && errno != ENOENT) {
722: message(MT_NOTICE, "%s: remove failed: %s",
723: savefile, SYSERR);
724: return(NULL);
725: }
1.1 dm 726: }
727:
728: if (rename(file, savefile) != 0 && errno != ENOENT) {
729: error("%s -> %s: rename failed: %s",
730: file, savefile, SYSERR);
1.7 millert 731: return(NULL);
1.1 dm 732: }
733:
734: return(savefile);
735: }
736:
737: /*
738: * Receive a file
739: */
1.14 millert 740: static void
1.19 millert 741: recvfile(char *new, opt_t opts, int mode, char *owner, char *group,
742: time_t mtime, time_t atime, off_t size)
1.1 dm 743: {
1.19 millert 744: int f, wrerr, olderrno;
1.1 dm 745: off_t i;
1.10 mpech 746: char *cp;
1.1 dm 747: char *savefile = NULL;
1.5 millert 748: static struct stat statbuff;
1.1 dm 749:
750: /*
1.19 millert 751: * Create temporary file
752: */
1.41 millert 753: if (chkparent(new, opts) < 0 || (f = mkstemp(new)) < 0) {
754: error("%s: create failed: %s", new, SYSERR);
755: return;
1.19 millert 756: }
757:
758: /*
1.1 dm 759: * Receive the file itself
760: */
761: ack();
762: wrerr = 0;
763: olderrno = 0;
764: for (i = 0; i < size; i += BUFSIZ) {
1.22 krw 765: off_t amt = BUFSIZ;
1.1 dm 766:
767: cp = buf;
768: if (i + amt > size)
769: amt = size - i;
770: do {
1.22 krw 771: ssize_t j;
1.1 dm 772:
773: j = readrem(cp, amt);
774: if (j <= 0) {
775: (void) close(f);
776: (void) unlink(new);
777: fatalerr(
1.9 mpech 778: "Read error occurred while receiving file.");
1.1 dm 779: finish();
780: }
781: amt -= j;
782: cp += j;
783: } while (amt > 0);
784: amt = BUFSIZ;
785: if (i + amt > size)
786: amt = size - i;
1.14 millert 787: if (wrerr == 0 && xwrite(f, buf, amt) != amt) {
1.1 dm 788: olderrno = errno;
789: wrerr++;
790: }
791: }
792:
793: if (response() < 0) {
794: (void) close(f);
795: (void) unlink(new);
796: return;
797: }
1.2 dm 798:
1.1 dm 799: if (wrerr) {
800: error("%s: Write error: %s", new, strerror(olderrno));
801: (void) close(f);
802: (void) unlink(new);
803: return;
804: }
805:
806: /*
807: * Do file comparison if enabled
808: */
809: if (IS_ON(opts, DO_COMPARE)) {
810: FILE *f1, *f2;
811: int c;
812:
813: errno = 0; /* fopen is not a syscall */
814: if ((f1 = fopen(target, "r")) == NULL) {
815: error("%s: open for read failed: %s", target, SYSERR);
816: (void) close(f);
817: (void) unlink(new);
818: return;
819: }
820: errno = 0;
821: if ((f2 = fopen(new, "r")) == NULL) {
822: error("%s: open for read failed: %s", new, SYSERR);
1.14 millert 823: (void) fclose(f1);
1.1 dm 824: (void) close(f);
825: (void) unlink(new);
826: return;
827: }
828: while ((c = getc(f1)) == getc(f2))
829: if (c == EOF) {
830: debugmsg(DM_MISC,
831: "Files are the same '%s' '%s'.",
832: target, new);
833: (void) fclose(f1);
834: (void) fclose(f2);
835: (void) close(f);
836: (void) unlink(new);
837: /*
838: * This isn't an error per-se, but we
839: * need to indicate to the master that
840: * the file was not updated.
841: */
1.42 ! millert 842: error(NULL);
1.1 dm 843: return;
844: }
845: debugmsg(DM_MISC, "Files are different '%s' '%s'.",
846: target, new);
847: (void) fclose(f1);
848: (void) fclose(f2);
849: if (IS_ON(opts, DO_VERIFY)) {
850: message(MT_REMOTE|MT_INFO, "%s: need to update",
851: target);
852: (void) close(f);
853: (void) unlink(new);
854: return;
855: }
856: }
857:
858: /*
859: * Set owner, group, and file mode
860: */
861: if (fchog(f, new, owner, group, mode) < 0) {
862: (void) close(f);
863: (void) unlink(new);
864: return;
865: }
866: (void) close(f);
867:
868: /*
869: * Perform utimes() after file is closed to make
870: * certain OS's, such as NeXT 2.1, happy.
871: */
1.24 deraadt 872: if (setfiletime(new, time(NULL), mtime) < 0)
1.1 dm 873: message(MT_NOTICE, "%s: utimes failed: %s", new, SYSERR);
874:
875: /*
876: * Try to save target file from being over-written
877: */
878: if (IS_ON(opts, DO_SAVETARGETS))
1.14 millert 879: if ((savefile = savetarget(target, opts)) == NULL) {
1.1 dm 880: (void) unlink(new);
881: return;
882: }
883:
884: /*
1.5 millert 885: * If the target is a directory, we need to remove it first
886: * before we can rename the new file.
887: */
888: if ((stat(target, &statbuff) == 0) && S_ISDIR(statbuff.st_mode)) {
889: char *saveptr = ptarget;
890:
891: ptarget = &target[strlen(target)];
1.14 millert 892: removefile(&statbuff, 0);
1.5 millert 893: ptarget = saveptr;
894: }
895:
896: /*
1.1 dm 897: * Install new (temporary) file as the actual target
898: */
899: if (rename(new, target) < 0) {
1.25 guenther 900: static const char fmt[] = "%s -> %s: rename failed: %s";
1.14 millert 901: struct stat stb;
1.1 dm 902: /*
903: * If the rename failed due to "Text file busy", then
904: * try to rename the target file and retry the rename.
905: */
1.14 millert 906: switch (errno) {
907: case ETXTBSY:
1.1 dm 908: /* Save the target */
1.14 millert 909: if ((savefile = savetarget(target, opts)) != NULL) {
1.1 dm 910: /* Retry installing new file as target */
911: if (rename(new, target) < 0) {
1.14 millert 912: error(fmt, new, target, SYSERR);
1.1 dm 913: /* Try to put back save file */
914: if (rename(savefile, target) < 0)
1.14 millert 915: error(fmt,
916: savefile, target, SYSERR);
917: (void) unlink(new);
1.1 dm 918: } else
919: message(MT_NOTICE, "%s: renamed to %s",
920: target, savefile);
1.14 millert 921: /*
922: * XXX: We should remove the savefile here.
923: * But we are nice to nfs clients and
924: * we keep it.
925: */
926: }
927: break;
928: case EISDIR:
929: /*
930: * See if target is a directory and remove it if it is
931: */
932: if (lstat(target, &stb) == 0) {
933: if (S_ISDIR(stb.st_mode)) {
934: char *optarget = ptarget;
935: for (ptarget = target; *ptarget;
936: ptarget++);
937: /* If we failed to remove, we'll catch
938: it later */
939: (void) removefile(&stb, 1);
940: ptarget = optarget;
941: }
1.1 dm 942: }
1.14 millert 943: if (rename(new, target) >= 0)
944: break;
945: /*FALLTHROUGH*/
946:
947: default:
948: error(fmt, new, target, SYSERR);
1.1 dm 949: (void) unlink(new);
1.14 millert 950: break;
1.1 dm 951: }
952: }
953:
954: if (IS_ON(opts, DO_COMPARE))
955: message(MT_REMOTE|MT_CHANGE, "%s: updated", target);
956: else
957: ack();
958: }
959:
960: /*
961: * Receive a directory
962: */
1.14 millert 963: static void
964: recvdir(opt_t opts, int mode, char *owner, char *group)
1.1 dm 965: {
966: static char lowner[100], lgroup[100];
1.10 mpech 967: char *cp;
1.1 dm 968: struct stat stb;
969: int s;
970:
971: s = lstat(target, &stb);
972: if (s == 0) {
973: /*
974: * If target is not a directory, remove it
975: */
976: if (!S_ISDIR(stb.st_mode)) {
977: if (IS_ON(opts, DO_VERIFY))
978: message(MT_NOTICE, "%s: need to remove",
979: target);
980: else {
981: if (unlink(target) < 0) {
982: error("%s: remove failed: %s",
983: target, SYSERR);
984: return;
985: }
986: }
987: s = -1;
988: errno = ENOENT;
989: } else {
990: if (!IS_ON(opts, DO_NOCHKMODE) &&
991: (stb.st_mode & 07777) != mode) {
992: if (IS_ON(opts, DO_VERIFY))
993: message(MT_NOTICE,
1.36 guenther 994: "%s: need to chmod to %#04o",
1.1 dm 995: target, mode);
1.36 guenther 996: else if (chmod(target, mode) != 0)
997: message(MT_NOTICE,
998: "%s: chmod from %#04o to %#04o failed: %s",
999: target,
1000: stb.st_mode & 07777,
1001: mode,
1002: SYSERR);
1003: else
1004: message(MT_NOTICE,
1005: "%s: chmod from %#04o to %#04o",
1006: target,
1007: stb.st_mode & 07777,
1008: mode);
1.1 dm 1009: }
1010:
1011: /*
1012: * Check ownership and set if necessary
1013: */
1014: lowner[0] = CNULL;
1015: lgroup[0] = CNULL;
1016:
1017: if (!IS_ON(opts, DO_NOCHKOWNER) && owner) {
1018: int o;
1019:
1020: o = (owner[0] == ':') ? opts & DO_NUMCHKOWNER :
1021: opts;
1.14 millert 1022: if ((cp = getusername(stb.st_uid, target, o))
1023: != NULL)
1.1 dm 1024: if (strcmp(owner, cp))
1.11 deraadt 1025: (void) strlcpy(lowner, cp,
1.14 millert 1026: sizeof(lowner));
1.1 dm 1027: }
1028: if (!IS_ON(opts, DO_NOCHKGROUP) && group) {
1029: int o;
1030:
1031: o = (group[0] == ':') ? opts & DO_NUMCHKGROUP :
1032: opts;
1.14 millert 1033: if ((cp = getgroupname(stb.st_gid, target, o))
1034: != NULL)
1.1 dm 1035: if (strcmp(group, cp))
1.11 deraadt 1036: (void) strlcpy(lgroup, cp,
1.14 millert 1037: sizeof(lgroup));
1.1 dm 1038: }
1039:
1040: /*
1041: * Need to set owner and/or group
1042: */
1043: #define PRN(n) ((n[0] == ':') ? n+1 : n)
1044: if (lowner[0] != CNULL || lgroup[0] != CNULL) {
1045: if (lowner[0] == CNULL &&
1046: (cp = getusername(stb.st_uid,
1047: target, opts)))
1.11 deraadt 1048: (void) strlcpy(lowner, cp,
1.14 millert 1049: sizeof(lowner));
1.1 dm 1050: if (lgroup[0] == CNULL &&
1051: (cp = getgroupname(stb.st_gid,
1052: target, opts)))
1.11 deraadt 1053: (void) strlcpy(lgroup, cp,
1.14 millert 1054: sizeof(lgroup));
1.1 dm 1055:
1056: if (IS_ON(opts, DO_VERIFY))
1057: message(MT_NOTICE,
1.12 millert 1058: "%s: need to chown from %s:%s to %s:%s",
1.1 dm 1059: target,
1060: PRN(lowner), PRN(lgroup),
1061: PRN(owner), PRN(group));
1062: else {
1063: if (fchog(-1, target, owner,
1064: group, -1) == 0)
1065: message(MT_NOTICE,
1.12 millert 1066: "%s: chown from %s:%s to %s:%s",
1.1 dm 1067: target,
1068: PRN(lowner),
1069: PRN(lgroup),
1070: PRN(owner),
1071: PRN(group));
1072: }
1073: }
1074: #undef PRN
1075: ack();
1076: return;
1077: }
1078: }
1079:
1080: if (IS_ON(opts, DO_VERIFY)) {
1081: ack();
1082: return;
1083: }
1084:
1085: /*
1086: * Create the directory
1087: */
1088: if (s < 0) {
1089: if (errno == ENOENT) {
1090: if (mkdir(target, mode) == 0 ||
1.7 millert 1091: (chkparent(target, opts) == 0 &&
1092: mkdir(target, mode) == 0)) {
1.1 dm 1093: message(MT_NOTICE, "%s: mkdir", target);
1094: (void) fchog(-1, target, owner, group, mode);
1095: ack();
1096: } else {
1097: error("%s: mkdir failed: %s", target, SYSERR);
1098: ptarget = sptarget[--catname];
1099: *ptarget = CNULL;
1100: }
1101: return;
1102: }
1103: }
1104: error("%s: lstat failed: %s", target, SYSERR);
1105: ptarget = sptarget[--catname];
1106: *ptarget = CNULL;
1107: }
1108:
1109: /*
1110: * Receive a link
1111: */
1.14 millert 1112: static void
1113: recvlink(char *new, opt_t opts, int mode, off_t size)
1.1 dm 1114: {
1.34 deraadt 1115: char tbuf[PATH_MAX], dbuf[BUFSIZ];
1.1 dm 1116: struct stat stb;
1117: char *optarget;
1.14 millert 1118: int uptodate;
1.1 dm 1119: off_t i;
1120:
1121: /*
1122: * Read basic link info
1123: */
1124: ack();
1125: (void) remline(buf, sizeof(buf), TRUE);
1126:
1127: if (response() < 0) {
1128: err();
1129: return;
1130: }
1131:
1.20 deraadt 1132: if (DECODE(dbuf, buf) == -1) {
1133: error("recvlink: cannot decode symlink target");
1134: return;
1135: }
1136:
1.14 millert 1137: uptodate = 0;
1.16 mpech 1138: if ((i = readlink(target, tbuf, sizeof(tbuf)-1)) != -1) {
1.14 millert 1139: tbuf[i] = '\0';
1.20 deraadt 1140: if (i == size && strncmp(dbuf, tbuf, (int) size) == 0)
1.14 millert 1141: uptodate = 1;
1142: }
1143: mode &= 0777;
1144:
1145: if (IS_ON(opts, DO_VERIFY) || uptodate) {
1146: if (uptodate)
1.42 ! millert 1147: message(MT_REMOTE|MT_INFO, NULL);
1.14 millert 1148: else
1149: message(MT_REMOTE|MT_INFO, "%s: need to update",
1150: target);
1151: if (IS_ON(opts, DO_COMPARE))
1152: return;
1153: (void) sendcmd(C_END, NULL);
1154: (void) response();
1155: return;
1156: }
1157:
1.1 dm 1158: /*
1159: * Make new symlink using a temporary name
1160: */
1.41 millert 1161: if (chkparent(new, opts) < 0 || mktemp(new) == NULL ||
1162: symlink(dbuf, new) < 0) {
1163: error("%s -> %s: symlink failed: %s", new, dbuf, SYSERR);
1164: return;
1.1 dm 1165: }
1166:
1167: /*
1168: * See if target is a directory and remove it if it is
1169: */
1170: if (lstat(target, &stb) == 0) {
1171: if (S_ISDIR(stb.st_mode)) {
1172: optarget = ptarget;
1173: for (ptarget = target; *ptarget; ptarget++);
1.14 millert 1174: if (removefile(&stb, 0) < 0) {
1.1 dm 1175: ptarget = optarget;
1176: (void) unlink(new);
1177: (void) sendcmd(C_END, NULL);
1178: (void) response();
1179: return;
1180: }
1181: ptarget = optarget;
1182: }
1183: }
1184:
1185: /*
1186: * Install link as the target
1187: */
1188: if (rename(new, target) < 0) {
1189: error("%s -> %s: symlink rename failed: %s",
1190: new, target, SYSERR);
1191: (void) unlink(new);
1192: (void) sendcmd(C_END, NULL);
1193: (void) response();
1194: return;
1195: }
1196:
1.14 millert 1197: message(MT_REMOTE|MT_CHANGE, "%s: updated", target);
1.1 dm 1198:
1199: /*
1200: * Indicate end of receive operation
1201: */
1202: (void) sendcmd(C_END, NULL);
1203: (void) response();
1204: }
1205:
1206: /*
1207: * Creat a hard link to existing file.
1208: */
1.14 millert 1209: static void
1210: hardlink(char *cmd)
1.1 dm 1211: {
1212: struct stat stb;
1213: int exists = 0;
1.14 millert 1214: char *xoldname, *xnewname;
1.1 dm 1215: char *cp = cmd;
1216: static char expbuf[BUFSIZ];
1.14 millert 1217: char oldname[BUFSIZ], newname[BUFSIZ];
1.1 dm 1218:
1219: /* Skip over opts */
1220: (void) strtol(cp, &cp, 8);
1221: if (*cp++ != ' ') {
1222: error("hardlink: options not delimited");
1223: return;
1224: }
1225:
1.14 millert 1226: xoldname = strtok(cp, " ");
1227: if (xoldname == NULL) {
1.1 dm 1228: error("hardlink: oldname name not delimited");
1229: return;
1230: }
1231:
1.14 millert 1232: if (DECODE(oldname, xoldname) == -1) {
1233: error("hardlink: Cannot decode oldname");
1234: return;
1235: }
1236:
1237: xnewname = strtok(NULL, " ");
1238: if (xnewname == NULL) {
1.1 dm 1239: error("hardlink: new name not specified");
1240: return;
1241: }
1242:
1.14 millert 1243: if (DECODE(newname, xnewname) == -1) {
1244: error("hardlink: Cannot decode newname");
1245: return;
1246: }
1247:
1.13 millert 1248: if (exptilde(expbuf, oldname, sizeof(expbuf)) == NULL) {
1.1 dm 1249: error("hardlink: tilde expansion failed");
1250: return;
1251: }
1252:
1253: if (catname && cattarget(newname) < 0) {
1254: error("Cannot set newname target.");
1255: return;
1256: }
1257:
1258: if (lstat(target, &stb) == 0) {
1259: int mode = stb.st_mode & S_IFMT;
1260:
1261: if (mode != S_IFREG && mode != S_IFLNK) {
1262: error("%s: not a regular file", target);
1263: return;
1264: }
1265: exists = 1;
1266: }
1267:
1268: if (chkparent(target, options) < 0 ) {
1269: error("%s: no parent: %s ", target, SYSERR);
1270: return;
1271: }
1272: if (exists && (unlink(target) < 0)) {
1273: error("%s: unlink failed: %s", target, SYSERR);
1274: return;
1275: }
1.38 guenther 1276: if (linkat(AT_FDCWD, expbuf, AT_FDCWD, target, 0) < 0) {
1.1 dm 1277: error("%s: cannot link to %s: %s", target, oldname, SYSERR);
1278: return;
1279: }
1280: ack();
1281: }
1282:
1283: /*
1284: * Set configuration information.
1285: *
1286: * A key letter is followed immediately by the value
1287: * to set. The keys are:
1288: * SC_FREESPACE - Set minimium free space of filesystem
1289: * SC_FREEFILES - Set minimium free number of files of filesystem
1290: */
1.14 millert 1291: static void
1292: setconfig(char *cmd)
1.1 dm 1293: {
1.10 mpech 1294: char *cp = cmd;
1.1 dm 1295: char *estr;
1.22 krw 1296: const char *errstr;
1.1 dm 1297:
1298: switch (*cp++) {
1299: case SC_HOSTNAME: /* Set hostname */
1300: /*
1301: * Only use info if we don't know who this is.
1302: */
1303: if (!fromhost) {
1.8 millert 1304: fromhost = xstrdup(cp);
1.14 millert 1305: message(MT_SYSLOG, "startup for %s", fromhost);
1.1 dm 1306: setproctitle("serving %s", cp);
1307: }
1308: break;
1309:
1310: case SC_FREESPACE: /* Minimium free space */
1.22 krw 1311: min_freespace = (int64_t)strtonum(cp, 0, LLONG_MAX, &errstr);
1312: if (errstr)
1313: fatalerr("Minimum free space is %s: '%s'", errstr,
1314: optarg);
1.1 dm 1315: break;
1316:
1317: case SC_FREEFILES: /* Minimium free files */
1.22 krw 1318: min_freefiles = (int64_t)strtonum(cp, 0, LLONG_MAX, &errstr);
1319: if (errstr)
1320: fatalerr("Minimum free files is %s: '%s'", errstr,
1321: optarg);
1.1 dm 1322: break;
1323:
1324: case SC_LOGGING: /* Logging options */
1.14 millert 1325: if ((estr = msgparseopts(cp, TRUE)) != NULL) {
1.1 dm 1326: fatalerr("Bad message option string (%s): %s",
1327: cp, estr);
1328: return;
1329: }
1330: break;
1331:
1.14 millert 1332: case SC_DEFOWNER:
1333: (void) strlcpy(defowner, cp, sizeof(defowner));
1334: break;
1335:
1336: case SC_DEFGROUP:
1337: (void) strlcpy(defgroup, cp, sizeof(defgroup));
1338: break;
1339:
1.1 dm 1340: default:
1341: message(MT_NOTICE, "Unknown config command \"%s\".", cp-1);
1342: return;
1343: }
1344: }
1345:
1346: /*
1347: * Receive something
1348: */
1.14 millert 1349: static void
1350: recvit(char *cmd, int type)
1.1 dm 1351: {
1.19 millert 1352: int mode;
1.1 dm 1353: opt_t opts;
1354: off_t size;
1355: time_t mtime, atime;
1356: char *owner, *group, *file;
1.34 deraadt 1357: char new[PATH_MAX];
1358: char fileb[PATH_MAX];
1.22 krw 1359: int64_t freespace = -1, freefiles = -1;
1.1 dm 1360: char *cp = cmd;
1361:
1362: /*
1363: * Get rdist option flags
1364: */
1365: opts = strtol(cp, &cp, 8);
1366: if (*cp++ != ' ') {
1367: error("recvit: options not delimited");
1368: return;
1369: }
1370:
1371: /*
1372: * Get file mode
1373: */
1374: mode = strtol(cp, &cp, 8);
1375: if (*cp++ != ' ') {
1376: error("recvit: mode not delimited");
1377: return;
1378: }
1379:
1380: /*
1381: * Get file size
1382: */
1.22 krw 1383: size = (off_t) strtoll(cp, &cp, 10);
1.1 dm 1384: if (*cp++ != ' ') {
1385: error("recvit: size not delimited");
1386: return;
1387: }
1388:
1389: /*
1390: * Get modification time
1391: */
1.23 guenther 1392: mtime = (time_t) strtoll(cp, &cp, 10);
1.1 dm 1393: if (*cp++ != ' ') {
1394: error("recvit: mtime not delimited");
1395: return;
1396: }
1397:
1398: /*
1399: * Get access time
1400: */
1.23 guenther 1401: atime = (time_t) strtoll(cp, &cp, 10);
1.1 dm 1402: if (*cp++ != ' ') {
1403: error("recvit: atime not delimited");
1404: return;
1405: }
1406:
1407: /*
1408: * Get file owner name
1409: */
1410: owner = strtok(cp, " ");
1411: if (owner == NULL) {
1412: error("recvit: owner name not delimited");
1413: return;
1414: }
1415:
1416: /*
1417: * Get file group name
1418: */
1.7 millert 1419: group = strtok(NULL, " ");
1.1 dm 1420: if (group == NULL) {
1421: error("recvit: group name not delimited");
1422: return;
1423: }
1424:
1425: /*
1.14 millert 1426: * Get file name. Can't use strtok() since there could
1.1 dm 1427: * be white space in the file name.
1428: */
1.14 millert 1429: if (DECODE(fileb, group + strlen(group) + 1) == -1) {
1430: error("recvit: Cannot decode file name");
1431: return;
1432: }
1433:
1434: if (fileb[0] == '\0') {
1.1 dm 1435: error("recvit: no file name");
1436: return;
1437: }
1.14 millert 1438: file = fileb;
1.1 dm 1439:
1440: debugmsg(DM_MISC,
1.36 guenther 1441: "recvit: opts = %#x mode = %#04o size = %lld mtime = %lld",
1.23 guenther 1442: opts, mode, (long long) size, (long long)mtime);
1.1 dm 1443: debugmsg(DM_MISC,
1444: "recvit: owner = '%s' group = '%s' file = '%s' catname = %d isdir = %d",
1445: owner, group, file, catname, (type == S_IFDIR) ? 1 : 0);
1446:
1447: if (type == S_IFDIR) {
1.14 millert 1448: if ((size_t) catname >= sizeof(sptarget)) {
1.1 dm 1449: error("%s: too many directory levels", target);
1450: return;
1451: }
1452: sptarget[catname] = ptarget;
1453: if (catname++) {
1454: *ptarget++ = '/';
1.14 millert 1455: while ((*ptarget++ = *file++) != '\0')
1456: continue;
1.1 dm 1457: ptarget--;
1458: }
1459: } else {
1460: /*
1461: * Create name of temporary file
1462: */
1463: if (catname && cattarget(file) < 0) {
1464: error("Cannot set file name.");
1465: return;
1466: }
1467: file = strrchr(target, '/');
1468: if (file == NULL)
1.14 millert 1469: (void) strlcpy(new, tempname, sizeof(new));
1.1 dm 1470: else if (file == target)
1.14 millert 1471: (void) snprintf(new, sizeof(new), "/%s", tempname);
1.1 dm 1472: else {
1473: *file = CNULL;
1.14 millert 1474: (void) snprintf(new, sizeof(new), "%s/%s", target,
1475: tempname);
1.1 dm 1476: *file = '/';
1477: }
1478: }
1479:
1480: /*
1481: * Check to see if there is enough free space and inodes
1482: * to install this file.
1483: */
1484: if (min_freespace || min_freefiles) {
1485: /* Convert file size to kilobytes */
1.22 krw 1486: int64_t fsize = (int64_t)size / 1024;
1.1 dm 1487:
1488: if (getfilesysinfo(target, &freespace, &freefiles) != 0)
1489: return;
1490:
1491: /*
1492: * filesystem values < 0 indicate unsupported or unavailable
1493: * information.
1494: */
1495: if (min_freespace && (freespace >= 0) &&
1496: (freespace - fsize < min_freespace)) {
1497: error(
1.22 krw 1498: "%s: Not enough free space on filesystem: min %lld "
1499: "free %lld", target, min_freespace, freespace);
1.1 dm 1500: return;
1501: }
1502: if (min_freefiles && (freefiles >= 0) &&
1503: (freefiles - 1 < min_freefiles)) {
1504: error(
1.22 krw 1505: "%s: Not enough free files on filesystem: min %lld free "
1506: "%lld", target, min_freefiles, freefiles);
1.1 dm 1507: return;
1508: }
1509: }
1510:
1511: /*
1512: * Call appropriate receive function to receive file
1513: */
1514: switch (type) {
1515: case S_IFDIR:
1516: recvdir(opts, mode, owner, group);
1517: break;
1518:
1519: case S_IFLNK:
1520: recvlink(new, opts, mode, size);
1521: break;
1522:
1523: case S_IFREG:
1.19 millert 1524: recvfile(new, opts, mode, owner, group, mtime, atime, size);
1.1 dm 1525: break;
1526:
1527: default:
1528: error("%d: unknown file type", type);
1529: break;
1530: }
1531: }
1532:
1533: /*
1.14 millert 1534: * Chmog something
1535: */
1536: static void
1537: dochmog(char *cmd)
1538: {
1539: int mode;
1540: opt_t opts;
1541: char *owner, *group, *file;
1542: char *cp = cmd;
1.34 deraadt 1543: char fileb[PATH_MAX];
1.14 millert 1544:
1545: /*
1546: * Get rdist option flags
1547: */
1548: opts = strtol(cp, &cp, 8);
1549: if (*cp++ != ' ') {
1550: error("dochmog: options not delimited");
1551: return;
1552: }
1553:
1554: /*
1555: * Get file mode
1556: */
1557: mode = strtol(cp, &cp, 8);
1558: if (*cp++ != ' ') {
1559: error("dochmog: mode not delimited");
1560: return;
1561: }
1562:
1563: /*
1564: * Get file owner name
1565: */
1566: owner = strtok(cp, " ");
1567: if (owner == NULL) {
1568: error("dochmog: owner name not delimited");
1569: return;
1570: }
1571:
1572: /*
1573: * Get file group name
1574: */
1575: group = strtok(NULL, " ");
1576: if (group == NULL) {
1577: error("dochmog: group name not delimited");
1578: return;
1579: }
1580:
1581: /*
1582: * Get file name. Can't use strtok() since there could
1583: * be white space in the file name.
1584: */
1585: if (DECODE(fileb, group + strlen(group) + 1) == -1) {
1586: error("dochmog: Cannot decode file name");
1587: return;
1588: }
1589:
1590: if (fileb[0] == '\0') {
1591: error("dochmog: no file name");
1592: return;
1593: }
1594: file = fileb;
1595:
1596: debugmsg(DM_MISC,
1.36 guenther 1597: "dochmog: opts = %#x mode = %#04o", opts, mode);
1.14 millert 1598: debugmsg(DM_MISC,
1599: "dochmog: owner = '%s' group = '%s' file = '%s' catname = %d",
1600: owner, group, file, catname);
1601:
1602: if (catname && cattarget(file) < 0) {
1603: error("Cannot set newname target.");
1604: return;
1605: }
1606:
1607: (void) fchog(-1, target, owner, group, mode);
1608:
1609: ack();
1610: }
1611:
1612: /*
1.1 dm 1613: * Set target information
1614: */
1.14 millert 1615: static void
1616: settarget(char *cmd, int isdir)
1.1 dm 1617: {
1618: char *cp = cmd;
1619: opt_t opts;
1.14 millert 1620: char file[BUFSIZ];
1.1 dm 1621:
1622: catname = isdir;
1623:
1624: /*
1625: * Parse options for this target
1626: */
1627: opts = strtol(cp, &cp, 8);
1628: if (*cp++ != ' ') {
1629: error("settarget: options not delimited");
1630: return;
1631: }
1632: options = opts;
1633:
1.14 millert 1634: if (DECODE(file, cp) == -1) {
1635: error("settarget: Cannot decode target name");
1636: return;
1637: }
1638:
1.1 dm 1639: /*
1640: * Handle target
1641: */
1.13 millert 1642: if (exptilde(target, cp, sizeof(target)) == NULL)
1.1 dm 1643: return;
1644: ptarget = target;
1645: while (*ptarget)
1646: ptarget++;
1647:
1648: ack();
1649: }
1650:
1651: /*
1652: * Cleanup in preparation for exiting.
1653: */
1.14 millert 1654: void
1655: cleanup(int dummy)
1.1 dm 1656: {
1657: /* We don't need to do anything */
1658: }
1659:
1660: /*
1661: * Server routine to read requests and process them.
1662: */
1.14 millert 1663: void
1664: server(void)
1.1 dm 1665: {
1666: static char cmdbuf[BUFSIZ];
1.10 mpech 1667: char *cp;
1.35 guenther 1668: int n, proto_version;
1.1 dm 1669:
1.14 millert 1670: if (setjmp(finish_jmpbuf))
1.1 dm 1671: return;
1672: (void) signal(SIGHUP, sighandler);
1673: (void) signal(SIGINT, sighandler);
1674: (void) signal(SIGQUIT, sighandler);
1675: (void) signal(SIGTERM, sighandler);
1676: (void) signal(SIGPIPE, sighandler);
1677: (void) umask(oumask = umask(0));
1.14 millert 1678: (void) strlcpy(tempname, _RDIST_TMP, sizeof(tempname));
1.1 dm 1679: if (fromhost) {
1680: message(MT_SYSLOG, "Startup for %s", fromhost);
1681: #if defined(SETARGS)
1682: setproctitle("Serving %s", fromhost);
1683: #endif /* SETARGS */
1684: }
1685:
1686: /*
1687: * Let client know we want it to send it's version number
1688: */
1689: (void) sendcmd(S_VERSION, NULL);
1690:
1691: if (remline(cmdbuf, sizeof(cmdbuf), TRUE) < 0) {
1692: error("server: expected control record");
1693: return;
1694: }
1695:
1.14 millert 1696: if (cmdbuf[0] != S_VERSION || !isdigit((unsigned char)cmdbuf[1])) {
1.1 dm 1697: error("Expected version command, received: \"%s\".", cmdbuf);
1698: return;
1699: }
1700:
1701: proto_version = atoi(&cmdbuf[1]);
1702: if (proto_version != VERSION) {
1703: error("Protocol version %d is not supported.", proto_version);
1704: return;
1705: }
1706:
1707: /* Version number is okay */
1708: ack();
1709:
1710: /*
1711: * Main command loop
1712: */
1713: for ( ; ; ) {
1714: n = remline(cp = cmdbuf, sizeof(cmdbuf), TRUE);
1.14 millert 1715: if (n == -1) /* EOF */
1.1 dm 1716: return;
1717: if (n == 0) {
1718: error("server: expected control record");
1719: continue;
1720: }
1721:
1722: switch (*cp++) {
1723: case C_SETCONFIG: /* Configuration info */
1724: setconfig(cp);
1725: ack();
1726: continue;
1727:
1728: case C_DIRTARGET: /* init target file/directory name */
1729: settarget(cp, TRUE);
1730: continue;
1731:
1732: case C_TARGET: /* init target file/directory name */
1733: settarget(cp, FALSE);
1734: continue;
1735:
1736: case C_RECVREG: /* Transfer a regular file. */
1737: recvit(cp, S_IFREG);
1738: continue;
1739:
1740: case C_RECVDIR: /* Transfer a directory. */
1741: recvit(cp, S_IFDIR);
1742: continue;
1743:
1744: case C_RECVSYMLINK: /* Transfer symbolic link. */
1745: recvit(cp, S_IFLNK);
1746: continue;
1747:
1748: case C_RECVHARDLINK: /* Transfer hard link. */
1749: hardlink(cp);
1750: continue;
1751:
1752: case C_END: /* End of transfer */
1753: *ptarget = CNULL;
1754: if (catname <= 0) {
1755: error("server: too many '%c's", C_END);
1756: continue;
1757: }
1758: ptarget = sptarget[--catname];
1759: *ptarget = CNULL;
1760: ack();
1761: continue;
1762:
1763: case C_CLEAN: /* Clean. Cleanup a directory */
1764: clean(cp);
1765: continue;
1766:
1767: case C_QUERY: /* Query file/directory */
1768: query(cp);
1769: continue;
1770:
1771: case C_SPECIAL: /* Special. Execute commands */
1772: dospecial(cp);
1773: continue;
1774:
1775: case C_CMDSPECIAL: /* Cmd Special. Execute commands */
1776: docmdspecial();
1777: continue;
1778:
1.14 millert 1779: case C_CHMOG: /* Set owner, group, mode */
1780: dochmog(cp);
1.1 dm 1781: continue;
1782:
1783: case C_ERRMSG: /* Normal error message */
1784: if (cp && *cp)
1785: message(MT_NERROR|MT_NOREMOTE, "%s", cp);
1786: continue;
1787:
1788: case C_FERRMSG: /* Fatal error message */
1789: if (cp && *cp)
1790: message(MT_FERROR|MT_NOREMOTE, "%s", cp);
1791: return;
1792:
1793: default:
1794: error("server: unknown command '%s'", cp - 1);
1795: case CNULL:
1796: continue;
1797: }
1798: }
1799: }