You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

195 lines
3.4 KiB

/* See LICENSE file for copyright and license details. */
#include <grp.h>
#include <pwd.h>
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include "util.h"
static int Dflag = 0;
static gid_t group;
static uid_t owner;
static mode_t mode = 0755;
static void
make_dir(char *dir, int was_missing)
{
if (!mkdir(dir, was_missing ? 0755 : mode)) {
if (!was_missing && (lchown(dir, owner, group) < 0))
eprintf("lchmod %s:", dir);
} else if (errno != EEXIST) {
eprintf("mkdir %s:", dir);
}
}
static void
make_dirs(char *dir, int was_missing)
{
char *p;
for (p = strchr(dir + (dir[0] == '/'), '/'); p; p = strchr(p + 1, '/')) {
*p = '\0';
make_dir(dir, was_missing);
*p = '/';
}
make_dir(dir, was_missing);
}
static int
install(const char *s1, const char *s2, int depth)
{
int f1, f2;
if ((f1 = open(s1, O_RDONLY)) < 0)
eprintf("open %s:", s1);
if ((f2 = creat(s2, 0600)) < 0) {
if (unlink(s2) < 0 && errno != ENOENT)
eprintf("unlink %s:", s2);
if ((f2 = creat(s2, 0600)) < 0)
eprintf("creat %s:", s2);
}
if (concat(f1, s1, f2, s2) < 0)
goto fail;
if (fchmod(f2, mode) < 0) {
weprintf("fchmod %s:", s2);
goto fail;
}
if (fchown(f2, owner, group) < 0) {
weprintf("fchown %s:", s2);
goto fail;
}
close(f1);
close(f2);
return 0;
fail:
unlink(s2);
exit(1);
}
static void
usage(void)
{
eprintf("usage: %s [-g group] [-o owner] [-m mode] (-d dir ... | [-D] (-t dest source ... | source ... dest))\n", argv0);
}
int
main(int argc, char *argv[])
{
int dflag = 0;
char *gflag = 0;
char *oflag = 0;
char *mflag = 0;
char *tflag = 0;
struct group *gr;
struct passwd *pw;
struct stat st;
char *p;
ARGBEGIN {
case 'c':
/* no-op for compatibility */
break;
case 'd':
dflag = 1;
break;
case 'D':
Dflag = 1;
break;
case 's':
/* no-op for compatibility */
break;
case 'g':
gflag = EARGF(usage());
break;
case 'o':
oflag = EARGF(usage());
break;
case 'm':
mflag = EARGF(usage());
break;
case 't':
tflag = EARGF(usage());
break;
default:
usage();
} ARGEND
if (argc < 1 + (!tflag & !dflag) || dflag & (Dflag | !!tflag))
usage();
if (gflag) {
errno = 0;
gr = getgrnam(gflag);
if (gr) {
group = gr->gr_gid;
} else {
if (errno)
eprintf("getgrnam %s:", gflag);
group = estrtonum(gflag, 0, UINT_MAX);
}
} else {
group = getgid();
}
if (oflag) {
errno = 0;
pw = getpwnam(oflag);
if (pw) {
owner = pw->pw_uid;
} else {
if (errno)
eprintf("getpwnam %s:", oflag);
owner = estrtonum(oflag, 0, UINT_MAX);
}
} else {
owner = getuid();
}
if (mflag)
mode = parsemode(mflag, mode, 0);
if (dflag) {
for (; *argv; argc--, argv++)
make_dirs(*argv, 0);
return 0;
}
if (tflag) {
argv = memmove(argv - 1, argv, argc * sizeof(*argv));
argv[argc++] = tflag;
}
if (tflag || argc > 2) {
if (stat(argv[argc - 1], &st) < 0) {
if ((errno == ENOENT) && Dflag) {
make_dirs(argv[argc - 1], 1);
} else {
eprintf("stat %s:", argv[argc - 1]);
}
} else if (!S_ISDIR(st.st_mode)) {
eprintf("%s: not a directory\n", argv[argc - 1]);
}
}
if (stat(argv[argc - 1], &st) < 0) {
if (errno != ENOENT)
eprintf("stat %s:", argv[argc - 1]);
if (tflag || Dflag || argc > 2) {
if ((p = strrchr(argv[argc - 1], '/')) != NULL) {
*p = '\0';
make_dirs(argv[argc - 1], 1);
*p = '/';
}
}
}
enmasse(argc, argv, install);
return 0;
}