зеркало из https://github.com/mozilla/gecko-dev.git
408 строки
9.6 KiB
C
408 строки
9.6 KiB
C
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
/*
|
|
** Netscape portable install command.
|
|
*/
|
|
#include <stdio.h> /* OSF/1 requires this before grp.h, so put it first */
|
|
#include <assert.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#if defined(_WINDOWS)
|
|
#include <windows.h>
|
|
typedef unsigned int mode_t;
|
|
#else
|
|
#include <grp.h>
|
|
#include <pwd.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <utime.h>
|
|
#endif
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include "pathsub.h"
|
|
|
|
#define HAVE_LCHOWN
|
|
|
|
#if defined(AIX) || defined(BSDI) || defined(HPUX) || defined(LINUX) || defined(SUNOS4) || defined(SCO) || defined(UNIXWARE) || defined(NTO) || defined(DARWIN) || defined(BEOS) || defined(__riscos__)
|
|
#undef HAVE_LCHOWN
|
|
#endif
|
|
|
|
#define HAVE_FCHMOD
|
|
|
|
#if defined(BEOS)
|
|
#undef HAVE_FCHMOD
|
|
#endif
|
|
|
|
#ifdef LINUX
|
|
#include <getopt.h>
|
|
#endif
|
|
|
|
#if defined(SCO) || defined(UNIXWARE) || defined(SNI) || defined(NCR) || defined(NEC)
|
|
#if !defined(S_ISLNK) && defined(S_IFLNK)
|
|
#define S_ISLNK(a) (((a) & S_IFMT) == S_IFLNK)
|
|
#endif
|
|
#endif
|
|
|
|
#if defined(SNI)
|
|
extern int fchmod(int fildes, mode_t mode);
|
|
#endif
|
|
|
|
|
|
#ifdef GETCWD_CANT_MALLOC
|
|
/*
|
|
* this should probably go into a utility library in case other applications
|
|
* need it.
|
|
*/
|
|
static char *
|
|
getcwd_do_malloc(char *path, int len) {
|
|
|
|
if (!path) {
|
|
path = malloc(PATH_MAX +1);
|
|
if (!path) return NULL;
|
|
}
|
|
return getcwd(path, PATH_MAX);
|
|
}
|
|
#define GETCWD getcwd_do_malloc
|
|
#else
|
|
#define GETCWD getcwd
|
|
#endif
|
|
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
fprintf(stderr,
|
|
"usage: %s [-C cwd] [-L linkprefix] [-m mode] [-o owner] [-g group]\n"
|
|
" %*s [-DdltR] file [file ...] directory\n",
|
|
program, (int)strlen(program), "");
|
|
exit(2);
|
|
}
|
|
|
|
/* this is more-or-less equivalent to mkdir -p */
|
|
static int
|
|
mkdirs(char *path, mode_t mode)
|
|
{
|
|
char * cp;
|
|
int rv;
|
|
struct stat sb;
|
|
|
|
if (!path || !path[0])
|
|
fail("Null pointer or empty string passed to mkdirs()");
|
|
while (*path == '/' && path[1] == '/')
|
|
path++;
|
|
for (cp = strrchr(path, '/'); cp && cp != path && *(cp - 1) == '/'; cp--);
|
|
if (cp && cp != path) {
|
|
*cp = '\0';
|
|
if ((stat(path, &sb) < 0 || !S_ISDIR(sb.st_mode)) &&
|
|
mkdirs(path, mode) < 0) {
|
|
return -1;
|
|
}
|
|
*cp = '/';
|
|
}
|
|
rv = mkdir(path, mode);
|
|
if (rv) {
|
|
if (errno != EEXIST)
|
|
fail("mkdirs cannot make %s", path);
|
|
fprintf(stderr, "directory creation race: %s\n", path);
|
|
if (!stat(path, &sb) && S_ISDIR(sb.st_mode))
|
|
rv = 0;
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
static uid_t
|
|
touid(char *owner)
|
|
{
|
|
struct passwd *pw;
|
|
uid_t uid;
|
|
char *cp;
|
|
|
|
if (!owner || !owner[0])
|
|
fail("Null pointer or empty string passed to touid()");
|
|
pw = getpwnam(owner);
|
|
if (pw)
|
|
return pw->pw_uid;
|
|
uid = strtol(owner, &cp, 0);
|
|
if (uid == 0 && cp == owner)
|
|
fail("cannot find uid for %s", owner);
|
|
return uid;
|
|
}
|
|
|
|
static gid_t
|
|
togid(char *group)
|
|
{
|
|
struct group *gr;
|
|
gid_t gid;
|
|
char *cp;
|
|
|
|
if (!group || !group[0])
|
|
fail("Null pointer or empty string passed to togid()");
|
|
gr = getgrnam(group);
|
|
if (gr)
|
|
return gr->gr_gid;
|
|
gid = strtol(group, &cp, 0);
|
|
if (gid == 0 && cp == group)
|
|
fail("cannot find gid for %s", group);
|
|
return gid;
|
|
}
|
|
|
|
void * const uninit = (void *)0xdeadbeef;
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
char * base = uninit;
|
|
char * bp = uninit;
|
|
char * cp = uninit;
|
|
char * cwd = 0;
|
|
char * group = 0;
|
|
char * linkname = 0;
|
|
char * linkprefix = 0;
|
|
char * name = uninit;
|
|
char * owner = 0;
|
|
char * todir = uninit;
|
|
char * toname = uninit;
|
|
|
|
int bnlen = -1;
|
|
int cc = 0;
|
|
int dodir = 0;
|
|
int dolink = 0;
|
|
int dorelsymlink = 0;
|
|
int dotimes = 0;
|
|
int exists = 0;
|
|
int fromfd = -1;
|
|
int len = -1;
|
|
int lplen = 0;
|
|
int onlydir = 0;
|
|
int opt = -1;
|
|
int tdlen = -1;
|
|
int tofd = -1;
|
|
int wc = -1;
|
|
|
|
mode_t mode = 0755;
|
|
|
|
uid_t uid = -1;
|
|
gid_t gid = -1;
|
|
|
|
struct stat sb;
|
|
struct stat tosb;
|
|
struct utimbuf utb;
|
|
char buf[BUFSIZ];
|
|
|
|
program = strrchr(argv[0], '/');
|
|
if (!program)
|
|
program = strrchr(argv[0], '\\');
|
|
program = program ? program+1 : argv[0];
|
|
|
|
|
|
while ((opt = getopt(argc, argv, "C:DdlL:Rm:o:g:t")) != EOF) {
|
|
switch (opt) {
|
|
case 'C': cwd = optarg; break;
|
|
case 'D': onlydir = 1; break;
|
|
case 'd': dodir = 1; break;
|
|
case 'l': dolink = 1; break;
|
|
case 'L':
|
|
linkprefix = optarg;
|
|
lplen = strlen(linkprefix);
|
|
dolink = 1;
|
|
break;
|
|
case 'R': dolink = dorelsymlink = 1; break;
|
|
case 'm':
|
|
mode = strtoul(optarg, &cp, 8);
|
|
if (mode == 0 && cp == optarg)
|
|
usage();
|
|
break;
|
|
case 'o': owner = optarg; break;
|
|
case 'g': group = optarg; break;
|
|
case 't': dotimes = 1; break;
|
|
default: usage();
|
|
}
|
|
}
|
|
|
|
argc -= optind;
|
|
argv += optind;
|
|
if (argc < 2 - onlydir)
|
|
usage();
|
|
|
|
todir = argv[argc-1];
|
|
if ((stat(todir, &sb) < 0 || !S_ISDIR(sb.st_mode)) &&
|
|
mkdirs(todir, 0777) < 0) {
|
|
fail("cannot mkdir -p %s", todir);
|
|
}
|
|
if (onlydir)
|
|
return 0;
|
|
|
|
if (!cwd) {
|
|
cwd = GETCWD(0, PATH_MAX);
|
|
if (!cwd)
|
|
fail("could not get CWD");
|
|
}
|
|
|
|
/* make sure we can get into todir. */
|
|
xchdir(todir);
|
|
todir = GETCWD(0, PATH_MAX);
|
|
if (!todir)
|
|
fail("could not get CWD in todir");
|
|
tdlen = strlen(todir);
|
|
|
|
/* back to original directory. */
|
|
xchdir(cwd);
|
|
|
|
uid = owner ? touid(owner) : -1;
|
|
gid = group ? togid(group) : -1;
|
|
|
|
while (--argc > 0) {
|
|
name = *argv++;
|
|
len = strlen(name);
|
|
base = xbasename(name);
|
|
bnlen = strlen(base);
|
|
toname = (char*)xmalloc(tdlen + 1 + bnlen + 1);
|
|
sprintf(toname, "%s/%s", todir, base);
|
|
retry:
|
|
exists = (lstat(toname, &tosb) == 0);
|
|
|
|
if (dodir) {
|
|
/* -d means create a directory, always */
|
|
if (exists && !S_ISDIR(tosb.st_mode)) {
|
|
int rv = unlink(toname);
|
|
if (rv)
|
|
fail("cannot unlink %s", toname);
|
|
exists = 0;
|
|
}
|
|
if (!exists && mkdir(toname, mode) < 0) {
|
|
/* we probably have two nsinstall programs in a race here. */
|
|
if (errno == EEXIST && !stat(toname, &sb) &&
|
|
S_ISDIR(sb.st_mode)) {
|
|
fprintf(stderr, "directory creation race: %s\n", toname);
|
|
goto retry;
|
|
}
|
|
fail("cannot make directory %s", toname);
|
|
}
|
|
if ((owner || group) && chown(toname, uid, gid) < 0)
|
|
fail("cannot change owner of %s", toname);
|
|
} else if (dolink) {
|
|
if (*name == '/') {
|
|
/* source is absolute pathname, link to it directly */
|
|
linkname = 0;
|
|
} else {
|
|
if (linkprefix) {
|
|
/* -L implies -l and prefixes names with a $cwd arg. */
|
|
len += lplen + 1;
|
|
linkname = (char*)xmalloc(len + 1);
|
|
sprintf(linkname, "%s/%s", linkprefix, name);
|
|
} else if (dorelsymlink) {
|
|
/* Symlink the relative path from todir to source name. */
|
|
linkname = (char*)xmalloc(PATH_MAX);
|
|
|
|
if (*todir == '/') {
|
|
/* todir is absolute: skip over common prefix. */
|
|
lplen = relatepaths(todir, cwd, linkname);
|
|
strcpy(linkname + lplen, name);
|
|
} else {
|
|
/* todir is named by a relative path: reverse it. */
|
|
reversepath(todir, name, len, linkname);
|
|
xchdir(cwd);
|
|
}
|
|
|
|
len = strlen(linkname);
|
|
}
|
|
name = linkname;
|
|
}
|
|
|
|
/* Check for a pre-existing symlink with identical content. */
|
|
if (exists &&
|
|
(!S_ISLNK(tosb.st_mode) ||
|
|
readlink(toname, buf, sizeof buf) != len ||
|
|
strncmp(buf, name, len) != 0)) {
|
|
int rmrv;
|
|
rmrv = (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(toname);
|
|
if (rmrv < 0) {
|
|
fail("destination exists, cannot remove %s", toname);
|
|
}
|
|
exists = 0;
|
|
}
|
|
if (!exists && symlink(name, toname) < 0) {
|
|
if (errno == EEXIST) {
|
|
fprintf(stderr, "symlink creation race: %s\n", toname);
|
|
fail("symlink was attempted in working directory %s "
|
|
"from %s to %s.\n", cwd, name, toname);
|
|
goto retry;
|
|
}
|
|
diagnosePath(toname);
|
|
fail("cannot make symbolic link %s", toname);
|
|
}
|
|
#ifdef HAVE_LCHOWN
|
|
if ((owner || group) && lchown(toname, uid, gid) < 0)
|
|
fail("cannot change owner of %s", toname);
|
|
#endif
|
|
|
|
if (linkname) {
|
|
free(linkname);
|
|
linkname = 0;
|
|
}
|
|
} else {
|
|
/* Copy from name to toname, which might be the same file. */
|
|
fromfd = open(name, O_RDONLY);
|
|
if (fromfd < 0 || fstat(fromfd, &sb) < 0)
|
|
fail("cannot access %s", name);
|
|
if (exists &&
|
|
(!S_ISREG(tosb.st_mode) || access(toname, W_OK) < 0)) {
|
|
int rmrv;
|
|
rmrv = (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(toname);
|
|
if (rmrv < 0) {
|
|
fail("destination exists, cannot remove %s", toname);
|
|
}
|
|
}
|
|
tofd = open(toname, O_CREAT | O_WRONLY, 0666);
|
|
if (tofd < 0)
|
|
fail("cannot create %s", toname);
|
|
|
|
bp = buf;
|
|
while ((cc = read(fromfd, bp, sizeof buf)) > 0) {
|
|
while ((wc = write(tofd, bp, cc)) > 0) {
|
|
if ((cc -= wc) == 0)
|
|
break;
|
|
bp += wc;
|
|
}
|
|
if (wc < 0)
|
|
fail("cannot write to %s", toname);
|
|
}
|
|
if (cc < 0)
|
|
fail("cannot read from %s", name);
|
|
|
|
if (ftruncate(tofd, sb.st_size) < 0)
|
|
fail("cannot truncate %s", toname);
|
|
if (dotimes) {
|
|
utb.actime = sb.st_atime;
|
|
utb.modtime = sb.st_mtime;
|
|
if (utime(toname, &utb) < 0)
|
|
fail("cannot set times of %s", toname);
|
|
}
|
|
#ifdef HAVE_FCHMOD
|
|
if (fchmod(tofd, mode) < 0)
|
|
#else
|
|
if (chmod(toname, mode) < 0)
|
|
#endif
|
|
fail("cannot change mode of %s", toname);
|
|
|
|
if ((owner || group) && fchown(tofd, uid, gid) < 0)
|
|
fail("cannot change owner of %s", toname);
|
|
|
|
/* Must check for delayed (NFS) write errors on close. */
|
|
if (close(tofd) < 0)
|
|
fail("close reports write error on %s", toname);
|
|
close(fromfd);
|
|
}
|
|
|
|
free(toname);
|
|
}
|
|
|
|
free(cwd);
|
|
free(todir);
|
|
return 0;
|
|
}
|
|
|