tor-browser

nsinstall.c (9608B)

---

/* 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(HPUX) || defined(LINUX) || defined(NTO) || defined(DARWIN) || defined(__riscos__)
#undef HAVE_LCHOWN
#endif

#define HAVE_FCHMOD

#ifdef LINUX
#include <getopt.h>
#endif

#if defined(NCR)
#if !defined(S_ISLNK) && defined(S_IFLNK)
#define S_ISLNK(a)	(((a) & S_IFMT) == S_IFLNK)
#endif
#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);
   size_t toname_len = tdlen + 1 + bnlen + 1;
toname = (char*)xmalloc(toname_len);
snprintf(toname, toname_len, "%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);
	    snprintf(linkname, len+1, "%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;
}