source: vbox/trunk/src/libs/xpcom18a4/nsprpub/config/nsinstall.c@ 55761

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* ***** BEGIN LICENSE BLOCK *****
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is the Netscape Portable Runtime (NSPR).
 *
 * The Initial Developer of the Original Code is
 * Netscape Communications Corporation.
 * Portions created by the Initial Developer are Copyright (C) 1998-2000
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 *
 * ***** END LICENSE BLOCK ***** */

/*
** Netscape portable install command.
**
** Brendan Eich, 7/20/95
*/
#include <stdio.h>  /* OSF/1 requires this before grp.h, so put it first */
#include <assert.h>
#include <fcntl.h>
#include <grp.h>
#include <pwd.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <utime.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>
#include <errno.h>
#include <stdarg.h>
#ifdef USE_REENTRANT_LIBC
#include "libc_r.h"
#endif /* USE_REENTRANT_LIBC */

#include "pathsub.h"

#define HAVE_FCHMOD

#if defined(BEOS)
#undef HAVE_FCHMOD
#endif

/*
 * Does getcwd() take NULL as the first argument and malloc
 * the result buffer?
 */
#if !defined(DARWIN) && !defined(NEXTSTEP) && !defined(VMS)
#define GETCWD_CAN_MALLOC
#endif

#ifdef NEXTSTEP
#include <bsd/libc.h>

/*
** [email protected]: The getcwd is broken in NEXTSTEP (returns 0),
** when called on a mounted fs. Did anyone notice this? Here's an ugly
** workaround ...
*/
#define getcwd(b,s)   my_getcwd(b,s)

static char *
my_getcwd (char *buf, size_t size)
{
    FILE *pwd = popen("pwd", "r");
    char *result = fgets(buf, size, pwd);

    if (result) {
        buf[strlen(buf)-1] = '\0';
    }
    pclose (pwd);
    return buf;
}
#endif /* NEXTSTEP */

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

#if defined(SCO) || defined(UNIXWARE) || defined(SNI) || defined(NCR) || defined(NEC) || defined(NEXTSTEP)
#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 QNX
#define d_ino d_stat.st_ino
#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);
}

static int
mkdirs(char *path, mode_t mode)
{
    char *cp;
    struct stat sb;
    int res;
    
    while (*path == '/' && path[1] == '/')
        path++;
    while ((cp = strrchr(path, '/')) && cp[1] == '\0')
        *cp = '\0';
    if (cp && cp != path) {
        *cp = '\0';
        if ((stat(path, &sb) < 0 || !S_ISDIR(sb.st_mode)) &&
            mkdirs(path, mode) < 0) {
            return -1;
        }
        *cp = '/';
    }
    res = mkdir(path, mode);
    if ((res != 0) && (errno == EEXIST))
        return 0;
     else
        return res;
}

static uid_t
touid(char *owner)
{
    struct passwd *pw;
    uid_t uid;
    char *cp;

    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;

    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;
}

int
main(int argc, char **argv)
{
    int onlydir, dodir, dolink, dorelsymlink, dotimes, opt, len, lplen, tdlen, bnlen, exists, fromfd, tofd, cc, wc;
    mode_t mode = 0755;
    char *linkprefix, *owner, *group, *cp, *cwd, *todir, *toname, *name, *base, *linkname, *bp, buf[BUFSIZ];
    uid_t uid;
    gid_t gid;
    struct stat sb, tosb;
    struct utimbuf utb;

    program = argv[0];
    cwd = linkname = linkprefix = owner = group = 0;
    onlydir = dodir = dolink = dorelsymlink = dotimes = lplen = 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 make directory %s", todir);
    }
    if (onlydir)
        return 0;

    if (!cwd) {
#ifdef GETCWD_CAN_MALLOC
        cwd = getcwd(0, PATH_MAX);
#else
        cwd = malloc(PATH_MAX + 1);
        cwd = getcwd(cwd, PATH_MAX);
#endif
    }
    xchdir(todir);
#ifdef GETCWD_CAN_MALLOC
    todir = getcwd(0, PATH_MAX);
#else
    todir = malloc(PATH_MAX + 1);
    todir = getcwd(todir, PATH_MAX);
#endif
    tdlen = strlen(todir);
    xchdir(cwd);
    tdlen = strlen(todir);

    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);
        exists = (lstat(toname, &tosb) == 0);

        if (dodir) {
            /* -d means create a directory, always */
            if (exists && !S_ISDIR(tosb.st_mode)) {
                (void) unlink(toname);
                exists = 0;
            }
            if (!exists && mkdir(toname, mode) < 0)
                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)) {
                (void) (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(toname);
                exists = 0;
            }
            if (!exists && symlink(name, toname) < 0)
                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))
                (void) (S_ISDIR(tosb.st_mode) ? rmdir : unlink)(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);
            /*
            ** On OpenVMS we can't chmod() until the file is closed, and we
            ** have to utime() last since fchown/chmod alter the timestamps.
            */
#ifndef VMS
            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);
#endif
            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("cannot write to %s", toname);
            close(fromfd);
#ifdef VMS
            if (chmod(toname, mode) < 0)
                fail("cannot change mode of %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);
            }
#endif
        }

        free(toname);
    }

    free(cwd);
    free(todir);
    return 0;
}

/*
** Pathname subroutines.
**
** Brendan Eich, 8/29/95
*/

char *program;

void
fail(char *format, ...)
{
    int error;
    va_list ap;

#ifdef USE_REENTRANT_LIBC
    R_STRERROR_INIT_R();
#endif

    error = errno;
    fprintf(stderr, "%s: ", program);
    va_start(ap, format);
    vfprintf(stderr, format, ap);
    va_end(ap);
    if (error)

#ifdef USE_REENTRANT_LIBC
    R_STRERROR_R(errno);
        fprintf(stderr, ": %s", r_strerror_r);
#else
        fprintf(stderr, ": %s", strerror(errno));
#endif

    putc('\n', stderr);
    exit(1);
}

char *
getcomponent(char *path, char *name)
{
    if (*path == '\0')
        return 0;
    if (*path == '/') {
        *name++ = '/';
    } else {
        do {
            *name++ = *path++;
        } while (*path != '/' && *path != '\0');
    }
    *name = '\0';
    while (*path == '/')
        path++;
    return path;
}

#ifdef UNIXWARE_READDIR_BUFFER_TOO_SMALL
/* Sigh.  The static buffer in Unixware's readdir is too small. */
struct dirent * readdir(DIR *d)
{
        static struct dirent *buf = NULL;
#define MAX_PATH_LEN 1024


        if(buf == NULL)
                buf = (struct dirent *) malloc(sizeof(struct dirent) + MAX_PATH_LEN)
;
        return(readdir_r(d, buf));
}
#endif

char *
ino2name(ino_t ino, char *dir)
{
    DIR *dp;
    struct dirent *ep;
    char *name;

    dp = opendir("..");
    if (!dp)
        fail("cannot read parent directory");
    for (;;) {
        if (!(ep = readdir(dp)))
            fail("cannot find current directory");
        if (ep->d_ino == ino)
            break;
    }
    name = xstrdup(ep->d_name);
    closedir(dp);
    return name;
}

void *
xmalloc(size_t size)
{
    void *p = malloc(size);
    if (!p)
        fail("cannot allocate %u bytes", size);
    return p;
}

char *
xstrdup(char *s)
{
    return strcpy((char*)xmalloc(strlen(s) + 1), s);
}

char *
xbasename(char *path)
{
    char *cp;

    while ((cp = strrchr(path, '/')) && cp[1] == '\0')
        *cp = '\0';
    if (!cp) return path;
    return cp + 1;
}

void
xchdir(char *dir)
{
    if (chdir(dir) < 0)
        fail("cannot change directory to %s", dir);
}

int
relatepaths(char *from, char *to, char *outpath)
{
    char *cp, *cp2;
    int len;
    char buf[NAME_MAX];

    assert(*from == '/' && *to == '/');
    for (cp = to, cp2 = from; *cp == *cp2; cp++, cp2++)
        if (*cp == '\0')
            break;
    while (cp[-1] != '/')
        cp--, cp2--;
    if (cp - 1 == to) {
        /* closest common ancestor is /, so use full pathname */
        len = strlen(strcpy(outpath, to));
        if (outpath[len] != '/') {
            outpath[len++] = '/';
            outpath[len] = '\0';
        }
    } else {
        len = 0;
        while ((cp2 = getcomponent(cp2, buf)) != 0) {
            strcpy(outpath + len, "../");
            len += 3;
        }
        while ((cp = getcomponent(cp, buf)) != 0) {
            sprintf(outpath + len, "%s/", buf);
            len += strlen(outpath + len);
        }
    }
    return len;
}

void
reversepath(char *inpath, char *name, int len, char *outpath)
{
    char *cp, *cp2;
    char buf[NAME_MAX];
    struct stat sb;

    cp = strcpy(outpath + PATH_MAX - (len + 1), name);
    cp2 = inpath;
    while ((cp2 = getcomponent(cp2, buf)) != 0) {
        if (strcmp(buf, ".") == 0)
            continue;
        if (strcmp(buf, "..") == 0) {
            if (stat(".", &sb) < 0)
                fail("cannot stat current directory");
            name = ino2name(sb.st_ino, "..");
            len = strlen(name);
            cp -= len + 1;
            strcpy(cp, name);
            cp[len] = '/';
            free(name);
            xchdir("..");
        } else {
            cp -= 3;
            strncpy(cp, "../", 3);
            xchdir(buf);
        }
    }
    strcpy(outpath, cp);
}