|
1 /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ |
|
2 /* This Source Code Form is subject to the terms of the Mozilla Public |
|
3 * License, v. 2.0. If a copy of the MPL was not distributed with this |
|
4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
|
5 /* |
|
6 ** Netscape portable install command. |
|
7 ** |
|
8 ** Brendan Eich, 7/20/95 |
|
9 */ |
|
10 #include <stdio.h> /* OSF/1 requires this before grp.h, so put it first */ |
|
11 #include <assert.h> |
|
12 #include <fcntl.h> |
|
13 #include <errno.h> |
|
14 #include <dirent.h> |
|
15 #include <limits.h> |
|
16 #include <grp.h> |
|
17 #include <pwd.h> |
|
18 #include <stdio.h> |
|
19 #include <stdlib.h> |
|
20 #include <string.h> |
|
21 #include <unistd.h> |
|
22 #include <utime.h> |
|
23 #include <sys/types.h> |
|
24 #include <sys/stat.h> |
|
25 #include "pathsub.h" |
|
26 |
|
27 #ifdef HAVE_GETOPT_H |
|
28 #include <getopt.h> |
|
29 #endif |
|
30 |
|
31 #ifdef SUNOS4 |
|
32 #include "sunos4.h" |
|
33 #endif |
|
34 |
|
35 #ifdef NEXTSTEP |
|
36 #include <bsd/libc.h> |
|
37 #endif |
|
38 |
|
39 #ifdef __QNX__ |
|
40 #include <unix.h> |
|
41 #endif |
|
42 |
|
43 #ifdef NEED_S_ISLNK |
|
44 #if !defined(S_ISLNK) && defined(S_IFLNK) |
|
45 #define S_ISLNK(a) (((a) & S_IFMT) == S_IFLNK) |
|
46 #endif |
|
47 #endif |
|
48 |
|
49 #ifndef _DIRECTORY_SEPARATOR |
|
50 #define _DIRECTORY_SEPARATOR "/" |
|
51 #endif /* _DIRECTORY_SEPARATOR */ |
|
52 |
|
53 #ifdef NEED_FCHMOD_PROTO |
|
54 extern int fchmod(int fildes, mode_t mode); |
|
55 #endif |
|
56 |
|
57 static void |
|
58 usage(void) |
|
59 { |
|
60 fprintf(stderr, |
|
61 "usage: %s [-C cwd] [-L linkprefix] [-m mode] [-o owner] [-g group]\n" |
|
62 " %*s [-DdltR] file [file ...] directory\n", |
|
63 program, (int) strlen(program), ""); |
|
64 exit(2); |
|
65 } |
|
66 |
|
67 static int |
|
68 mkdirs(char *path, mode_t mode) |
|
69 { |
|
70 char *cp; |
|
71 struct stat sb; |
|
72 int res; |
|
73 int l; |
|
74 |
|
75 /* strip trailing "/." */ |
|
76 l = strlen(path); |
|
77 if(l > 1 && path[l - 1] == '.' && path[l - 2] == '/') |
|
78 path[l - 2] = 0; |
|
79 |
|
80 while (*path == '/' && path[1] == '/') |
|
81 path++; |
|
82 for (cp = strrchr(path, '/'); cp && cp != path && *(cp - 1) == '/'; cp--); |
|
83 if (cp && cp != path) { |
|
84 *cp = '\0'; |
|
85 if ((lstat(path, &sb) < 0 || !S_ISDIR(sb.st_mode)) && |
|
86 mkdirs(path, mode) < 0) { |
|
87 return -1; |
|
88 } |
|
89 *cp = '/'; |
|
90 } |
|
91 |
|
92 res = mkdir(path, mode); |
|
93 if ((res != 0) && (errno == EEXIST)) |
|
94 return 0; |
|
95 else |
|
96 return res; |
|
97 } |
|
98 |
|
99 static uid_t |
|
100 touid(char *owner) |
|
101 { |
|
102 struct passwd *pw; |
|
103 uid_t uid; |
|
104 char *cp; |
|
105 |
|
106 pw = getpwnam(owner); |
|
107 if (pw) |
|
108 return pw->pw_uid; |
|
109 uid = strtol(owner, &cp, 0); |
|
110 if (uid == 0 && cp == owner) |
|
111 fail("cannot find uid for %s", owner); |
|
112 return uid; |
|
113 } |
|
114 |
|
115 static gid_t |
|
116 togid(char *group) |
|
117 { |
|
118 struct group *gr; |
|
119 gid_t gid; |
|
120 char *cp; |
|
121 |
|
122 gr = getgrnam(group); |
|
123 if (gr) |
|
124 return gr->gr_gid; |
|
125 gid = strtol(group, &cp, 0); |
|
126 if (gid == 0 && cp == group) |
|
127 fail("cannot find gid for %s", group); |
|
128 return gid; |
|
129 } |
|
130 |
|
131 static void |
|
132 copyfile( char *name, char *toname, mode_t mode, char *group, char *owner, |
|
133 int dotimes, uid_t uid, gid_t gid ) |
|
134 { |
|
135 int fromfd, tofd = -1, cc, wc, exists; |
|
136 char buf[BUFSIZ], *bp; |
|
137 struct stat sb, tosb; |
|
138 struct utimbuf utb; |
|
139 |
|
140 exists = (lstat(toname, &tosb) == 0); |
|
141 |
|
142 fromfd = open(name, O_RDONLY); |
|
143 if (fromfd < 0 || fstat(fromfd, &sb) < 0) |
|
144 fail("cannot access %s", name); |
|
145 if (exists) { |
|
146 if (S_ISREG(tosb.st_mode)) { |
|
147 /* See if we can open it. This is more reliable than 'access'. */ |
|
148 tofd = open(toname, O_CREAT | O_WRONLY, 0666); |
|
149 } |
|
150 if (tofd < 0) { |
|
151 (void) (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(toname); |
|
152 } |
|
153 } |
|
154 if (tofd < 0) { |
|
155 tofd = open(toname, O_CREAT | O_WRONLY, 0666); |
|
156 if (tofd < 0) |
|
157 fail("cannot create %s", toname); |
|
158 } |
|
159 |
|
160 bp = buf; |
|
161 while ((cc = read(fromfd, bp, sizeof buf)) > 0) |
|
162 { |
|
163 while ((wc = write(tofd, bp, (unsigned int)cc)) > 0) |
|
164 { |
|
165 if ((cc -= wc) == 0) |
|
166 break; |
|
167 bp += wc; |
|
168 } |
|
169 if (wc < 0) |
|
170 fail("cannot write to %s", toname); |
|
171 } |
|
172 if (cc < 0) |
|
173 fail("cannot read from %s", name); |
|
174 |
|
175 if (ftruncate(tofd, sb.st_size) < 0) |
|
176 fail("cannot truncate %s", toname); |
|
177 #if !defined(VMS) |
|
178 if (dotimes) |
|
179 { |
|
180 utb.actime = sb.st_atime; |
|
181 utb.modtime = sb.st_mtime; |
|
182 if (utime(toname, &utb) < 0) |
|
183 fail("cannot set times of %s", toname); |
|
184 } |
|
185 #ifdef HAVE_FCHMOD |
|
186 if (fchmod(tofd, mode) < 0) |
|
187 #else |
|
188 if (chmod(toname, mode) < 0) |
|
189 #endif |
|
190 fail("cannot change mode of %s", toname); |
|
191 #endif |
|
192 if ((owner || group) && fchown(tofd, uid, gid) < 0) |
|
193 fail("cannot change owner of %s", toname); |
|
194 |
|
195 /* Must check for delayed (NFS) write errors on close. */ |
|
196 if (close(tofd) < 0) |
|
197 fail("cannot write to %s", toname); |
|
198 close(fromfd); |
|
199 #if defined(VMS) |
|
200 if (chmod(toname, (mode & (S_IREAD | S_IWRITE))) < 0) |
|
201 fail("cannot change mode of %s", toname); |
|
202 if (dotimes) |
|
203 { |
|
204 utb.actime = sb.st_atime; |
|
205 utb.modtime = sb.st_mtime; |
|
206 if (utime(toname, &utb) < 0) |
|
207 fail("cannot set times of %s", toname); |
|
208 } |
|
209 #endif |
|
210 } |
|
211 |
|
212 static void |
|
213 copydir( char *from, char *to, mode_t mode, char *group, char *owner, |
|
214 int dotimes, uid_t uid, gid_t gid) |
|
215 { |
|
216 int i; |
|
217 DIR *dir; |
|
218 struct dirent *ep; |
|
219 struct stat sb; |
|
220 char *base, *destdir, *direntry, *destentry; |
|
221 |
|
222 base = xbasename(from); |
|
223 |
|
224 /* create destination directory */ |
|
225 destdir = xmalloc((unsigned int)(strlen(to) + 1 + strlen(base) + 1)); |
|
226 sprintf(destdir, "%s%s%s", to, _DIRECTORY_SEPARATOR, base); |
|
227 if (mkdirs(destdir, mode) != 0) { |
|
228 fail("cannot make directory %s\n", destdir); |
|
229 free(destdir); |
|
230 return; |
|
231 } |
|
232 |
|
233 if (!(dir = opendir(from))) { |
|
234 fail("cannot open directory %s\n", from); |
|
235 free(destdir); |
|
236 return; |
|
237 } |
|
238 |
|
239 direntry = xmalloc((unsigned int)PATH_MAX); |
|
240 destentry = xmalloc((unsigned int)PATH_MAX); |
|
241 |
|
242 while ((ep = readdir(dir))) |
|
243 { |
|
244 if (strcmp(ep->d_name, ".") == 0 || strcmp(ep->d_name, "..") == 0) |
|
245 continue; |
|
246 |
|
247 sprintf(direntry, "%s/%s", from, ep->d_name); |
|
248 sprintf(destentry, "%s%s%s", destdir, _DIRECTORY_SEPARATOR, ep->d_name); |
|
249 |
|
250 if (stat(direntry, &sb) == 0 && S_ISDIR(sb.st_mode)) |
|
251 copydir( direntry, destdir, mode, group, owner, dotimes, uid, gid ); |
|
252 else |
|
253 copyfile( direntry, destentry, mode, group, owner, dotimes, uid, gid ); |
|
254 } |
|
255 |
|
256 free(destdir); |
|
257 free(direntry); |
|
258 free(destentry); |
|
259 closedir(dir); |
|
260 } |
|
261 |
|
262 int |
|
263 main(int argc, char **argv) |
|
264 { |
|
265 int onlydir, dodir, dolink, dorelsymlink, dotimes, opt, len, lplen, tdlen, bnlen, exists, fromfd, tofd, cc, wc; |
|
266 mode_t mode = 0755; |
|
267 char *linkprefix, *owner, *group, *cp, *cwd, *todir, *toname, *name, *base, *linkname, *bp, buf[BUFSIZ]; |
|
268 uid_t uid; |
|
269 gid_t gid; |
|
270 struct stat sb, tosb, fromsb; |
|
271 struct utimbuf utb; |
|
272 |
|
273 program = argv[0]; |
|
274 cwd = linkname = linkprefix = owner = group = 0; |
|
275 onlydir = dodir = dolink = dorelsymlink = dotimes = lplen = 0; |
|
276 |
|
277 while ((opt = getopt(argc, argv, "C:DdlL:Rm:o:g:t")) != EOF) { |
|
278 switch (opt) { |
|
279 case 'C': |
|
280 cwd = optarg; |
|
281 break; |
|
282 case 'D': |
|
283 onlydir = 1; |
|
284 break; |
|
285 case 'd': |
|
286 dodir = 1; |
|
287 break; |
|
288 case 'L': |
|
289 linkprefix = optarg; |
|
290 lplen = strlen(linkprefix); |
|
291 dolink = 1; |
|
292 break; |
|
293 case 'R': |
|
294 dolink = dorelsymlink = 1; |
|
295 break; |
|
296 case 'm': |
|
297 mode = strtoul(optarg, &cp, 8); |
|
298 if (mode == 0 && cp == optarg) |
|
299 usage(); |
|
300 break; |
|
301 case 'o': |
|
302 owner = optarg; |
|
303 break; |
|
304 case 'g': |
|
305 group = optarg; |
|
306 break; |
|
307 case 't': |
|
308 dotimes = 1; |
|
309 break; |
|
310 default: |
|
311 usage(); |
|
312 } |
|
313 } |
|
314 |
|
315 argc -= optind; |
|
316 argv += optind; |
|
317 if (argc < 2 - onlydir) |
|
318 usage(); |
|
319 |
|
320 todir = argv[argc-1]; |
|
321 if ((stat(todir, &sb) < 0 || !S_ISDIR(sb.st_mode)) && |
|
322 mkdirs(todir, 0777) < 0) { |
|
323 fail("cannot make directory %s", todir); |
|
324 } |
|
325 if (onlydir) |
|
326 return 0; |
|
327 |
|
328 if (!cwd) { |
|
329 #ifndef NEEDS_GETCWD |
|
330 #ifndef GETCWD_CANT_MALLOC |
|
331 cwd = getcwd(0, PATH_MAX); |
|
332 #else |
|
333 cwd = malloc(PATH_MAX + 1); |
|
334 cwd = getcwd(cwd, PATH_MAX); |
|
335 #endif |
|
336 #else |
|
337 cwd = malloc(PATH_MAX + 1); |
|
338 cwd = getwd(cwd); |
|
339 #endif |
|
340 } |
|
341 |
|
342 xchdir(todir); |
|
343 #ifndef NEEDS_GETCWD |
|
344 #ifndef GETCWD_CANT_MALLOC |
|
345 todir = getcwd(0, PATH_MAX); |
|
346 #else |
|
347 todir = malloc(PATH_MAX + 1); |
|
348 todir = getcwd(todir, PATH_MAX); |
|
349 #endif |
|
350 #else |
|
351 todir = malloc(PATH_MAX + 1); |
|
352 todir = getwd(todir); |
|
353 #endif |
|
354 tdlen = strlen(todir); |
|
355 xchdir(cwd); |
|
356 tdlen = strlen(todir); |
|
357 |
|
358 uid = owner ? touid(owner) : (uid_t)(-1); |
|
359 gid = group ? togid(group) : (gid_t)(-1); |
|
360 |
|
361 while (--argc > 0) { |
|
362 name = *argv++; |
|
363 len = strlen(name); |
|
364 base = xbasename(name); |
|
365 bnlen = strlen(base); |
|
366 toname = xmalloc((unsigned int)(tdlen + 1 + bnlen + 1)); |
|
367 sprintf(toname, "%s%s%s", todir, _DIRECTORY_SEPARATOR, base); |
|
368 exists = (lstat(toname, &tosb) == 0); |
|
369 |
|
370 if (dodir) { |
|
371 /* -d means create a directory, always */ |
|
372 if (exists && !S_ISDIR(tosb.st_mode)) { |
|
373 (void) unlink(toname); |
|
374 exists = 0; |
|
375 } |
|
376 if (!exists && mkdir(toname, mode) < 0) |
|
377 fail("cannot make directory %s", toname); |
|
378 if ((owner || group) && chown(toname, uid, gid) < 0) |
|
379 fail("cannot change owner of %s", toname); |
|
380 } else if (dolink) { |
|
381 if (access(name, R_OK) != 0) { |
|
382 fail("cannot access %s", name); |
|
383 } |
|
384 if (*name == '/') { |
|
385 /* source is absolute pathname, link to it directly */ |
|
386 linkname = 0; |
|
387 } else { |
|
388 if (linkprefix) { |
|
389 /* -L prefixes names with a $cwd arg. */ |
|
390 len += lplen + 1; |
|
391 linkname = xmalloc((unsigned int)(len + 1)); |
|
392 sprintf(linkname, "%s/%s", linkprefix, name); |
|
393 } else if (dorelsymlink) { |
|
394 /* Symlink the relative path from todir to source name. */ |
|
395 linkname = xmalloc(PATH_MAX); |
|
396 |
|
397 if (*todir == '/') { |
|
398 /* todir is absolute: skip over common prefix. */ |
|
399 lplen = relatepaths(todir, cwd, linkname); |
|
400 strcpy(linkname + lplen, name); |
|
401 } else { |
|
402 /* todir is named by a relative path: reverse it. */ |
|
403 reversepath(todir, name, len, linkname); |
|
404 xchdir(cwd); |
|
405 } |
|
406 |
|
407 len = strlen(linkname); |
|
408 } |
|
409 name = linkname; |
|
410 } |
|
411 |
|
412 /* Check for a pre-existing symlink with identical content. */ |
|
413 if ((exists && (!S_ISLNK(tosb.st_mode) || |
|
414 readlink(toname, buf, sizeof buf) != len || |
|
415 strncmp(buf, name, (unsigned int)len) != 0)) || |
|
416 ((stat(name, &fromsb) == 0) && |
|
417 (fromsb.st_mtime > tosb.st_mtime))) { |
|
418 (void) (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(toname); |
|
419 exists = 0; |
|
420 } |
|
421 if (!exists && symlink(name, toname) < 0) |
|
422 fail("cannot make symbolic link %s", toname); |
|
423 #ifdef HAVE_LCHOWN |
|
424 if ((owner || group) && lchown(toname, uid, gid) < 0) |
|
425 fail("cannot change owner of %s", toname); |
|
426 #endif |
|
427 |
|
428 if (linkname) { |
|
429 free(linkname); |
|
430 linkname = 0; |
|
431 } |
|
432 } else { |
|
433 /* Copy from name to toname, which might be the same file. */ |
|
434 if( stat(name, &sb) == 0 && S_IFDIR & sb.st_mode ) |
|
435 { |
|
436 /* then is directory: must explicitly create destination dir */ |
|
437 /* and manually copy files over */ |
|
438 copydir( name, todir, mode, group, owner, dotimes, uid, gid ); |
|
439 } |
|
440 else |
|
441 { |
|
442 copyfile(name, toname, mode, group, owner, dotimes, uid, gid); |
|
443 } |
|
444 } |
|
445 |
|
446 free(toname); |
|
447 } |
|
448 |
|
449 free(cwd); |
|
450 free(todir); |
|
451 return 0; |
|
452 } |