1 /* $OpenBSD: mtree.c,v 1.28 2022/12/04 23:50:51 cheloha Exp $ */
2 /* $NetBSD: mtree.c,v 1.7 1996/09/05 23:29:22 thorpej Exp $ */
3
4 /*-
5 * Copyright (c) 1989, 1990, 1993
6 * The Regents of the University of California. All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of the University nor the names of its contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33 #include <sys/stat.h>
34 #include <err.h>
35 #include <errno.h>
36 #include <unistd.h>
37 #include <stdio.h>
38 #include <limits.h>
39 #include <fts.h>
40 #include <grp.h>
41 #include <pwd.h>
42 #include "mtree.h"
43 #include "extern.h"
44
45 extern u_int32_t crc_total;
46
47 int ftsoptions = FTS_PHYSICAL;
48 int cflag, dflag, eflag, iflag, lflag, nflag, qflag, rflag, sflag, tflag,
49 uflag, Uflag;
50 u_int keys;
51 char fullpath[PATH_MAX];
52
53 static void usage(void);
54
55 int
main(int argc,char * argv[])56 main(int argc, char *argv[])
57 {
58 int ch;
59 char *dir, *p;
60 int status;
61
62 dir = NULL;
63 keys = KEYDEFAULT;
64 while ((ch = getopt(argc, argv, "cdef:iK:k:lnp:qrs:tUux")) != -1)
65 switch(ch) {
66 case 'c':
67 cflag = 1;
68 break;
69 case 'd':
70 dflag = 1;
71 break;
72 case 'e':
73 eflag = 1;
74 break;
75 case 'f':
76 if (!(freopen(optarg, "r", stdin)))
77 error("%s: %s", optarg, strerror(errno));
78 break;
79 case 'i':
80 iflag = 1;
81 break;
82 case 'K':
83 while ((p = strsep(&optarg, " \t,")) != NULL)
84 if (*p != '\0')
85 keys |= parsekey(p, NULL);
86 break;
87 case 'k':
88 keys = F_TYPE;
89 while ((p = strsep(&optarg, " \t,")) != NULL)
90 if (*p != '\0')
91 keys |= parsekey(p, NULL);
92 break;
93 case 'l':
94 lflag = 1;
95 break;
96 case 'n':
97 nflag = 1;
98 break;
99 case 'p':
100 dir = optarg;
101 break;
102 case 'q':
103 qflag = 1;
104 break;
105 case 'r':
106 rflag = 1;
107 break;
108 case 's':
109 sflag = 1;
110 crc_total = ~strtol(optarg, &p, 0);
111 if (*p)
112 error("illegal seed value -- %s", optarg);
113 break;
114 case 't':
115 tflag = 1;
116 break;
117 case 'U':
118 Uflag = 1;
119 uflag = 1;
120 break;
121 case 'u':
122 uflag = 1;
123 break;
124 case 'x':
125 ftsoptions |= FTS_XDEV;
126 break;
127 default:
128 usage();
129 }
130 argc -= optind;
131 argv += optind;
132
133 if (argc)
134 usage();
135
136 /*
137 * If uflag is set we can't make any pledges because we must be able
138 * to chown and place setugid bits. Make sure that we're pledged
139 * when -c was specified.
140 */
141 if (!uflag || cflag) {
142 if (rflag && tflag) {
143 if (pledge("stdio rpath cpath getpw fattr", NULL) == -1)
144 err(1, "pledge");
145 } else if (rflag && !tflag) {
146 if (pledge("stdio rpath cpath getpw", NULL) == -1)
147 err(1, "pledge");
148 } else if (!rflag && tflag) {
149 if (pledge("stdio rpath getpw fattr", NULL) == -1)
150 err(1, "pledge");
151 } else {
152 if (pledge("stdio rpath getpw", NULL) == -1)
153 err(1, "pledge");
154 }
155 }
156
157 /* Keep passwd and group files open for faster lookups. */
158 setpassent(1);
159 setgroupent(1);
160
161 if (dir && chdir(dir))
162 error("%s: %s", dir, strerror(errno));
163
164 if ((cflag || sflag) && !getcwd(fullpath, sizeof fullpath))
165 error("getcwd: %s", strerror(errno));
166
167 if (lflag == 1 && uflag == 1)
168 error("-l and -u flags are mutually exclusive");
169
170 if (cflag) {
171 cwalk();
172 exit(0);
173 }
174 status = verify();
175 if (Uflag && (status == MISMATCHEXIT))
176 status = 0;
177 exit(status);
178 }
179
180 static void
usage(void)181 usage(void)
182 {
183 (void)fprintf(stderr,
184 "usage: mtree [-cdeilnqrtUux] [-f spec] [-K keywords] "
185 "[-k keywords] [-p path]\n"
186 " [-s seed]\n");
187 exit(1);
188 }
189