Various fixes.

[unix-history] / usr / src / usr.bin / xinstall / xinstall.c

/*
 * Copyright (c) 1987 Regents of the University of California.
 * All rights reserved.  The Berkeley software License Agreement
 * specifies the terms and conditions for redistribution.
 */

#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1987 Regents of the University of California.\n\
 All rights reserved.\n";
#endif not lint

#ifndef lint
static char sccsid[] = "@(#)xinstall.c	5.5 (Berkeley) %G%";
#endif not lint

#include <sys/param.h>
#include <sys/stat.h>
#include <sys/file.h>
#include <a.out.h>
#include <grp.h>
#include <pwd.h>
#include <stdio.h>
#include <ctype.h>

#define	YES		1			/* yes/true */
#define	DEF_GROUP	"staff"			/* default group */
#define	DEF_OWNER	"root"			/* default owner */

static int	docopy, dostrip,
		mode = 0755;
static char	*group = DEF_GROUP,
		*owner = DEF_OWNER,
		*path;
extern int	errno;
extern char	*sys_errlist[];

main(argc, argv)
	int	argc;
	char	**argv;
{
	extern char	*optarg;
	extern int	optind;
	register int	to_fd;
	struct stat	from_sb, to_sb;
	struct passwd	*pp;
	struct group	*gp;
	int	ch, devnull;
	char	pbuf[MAXPATHLEN];

	while ((ch = getopt(argc, argv, "cg:m:o:s")) != EOF)
		switch((char)ch) {
		case 'c':
			docopy = YES;
			break;
		case 'g':
			group = optarg;
			break;
		case 'm':
			mode = atoo(optarg);
			break;
		case 'o':
			owner = optarg;
			break;
		case 's':
			dostrip = YES;
			break;
		case '?':
		default:
			usage();
		}
	argc -= optind;
	argv += optind;
	if (argc != 2)
		usage();

	/* get group and owner id's */
	if (!(gp = getgrnam(group))) {
		fprintf(stderr, "install: unknown group %s.\n", group);
		exit(1);
	}
	if (!(pp = getpwnam(owner))) {
		fprintf(stderr, "install: unknown user %s.\n", owner);
		exit(1);
	}

	/* check source */
	if (stat(argv[0], &from_sb)) {
		fprintf(stderr, "install: fstat: %s: %s\n", argv[0], sys_errlist[errno]);
		exit(1);
	}
	/* special case for removing files */
	devnull = !strcmp(argv[0], "/dev/null");
	if (!devnull && !(from_sb.st_mode & S_IFREG)) {
		fprintf(stderr, "install: %s isn't a regular file.\n", argv[0]);
		exit(1);
	}

	/* build target path, find out if target is same as source */
	if (!stat(path = argv[1], &to_sb)) {
		if (to_sb.st_mode & S_IFDIR) {
			char	*C, *rindex();

			(void)sprintf(path = pbuf, "%s/%s", argv[1], (C = rindex(argv[0], '/')) ? ++C : argv[0]);
			if (stat(path, &to_sb))
				goto nocompare;
		}
		if (!(to_sb.st_mode & S_IFREG)) {
			fprintf(stderr, "install: %s isn't a regular file.\n", path);
			exit(1);
		}
		if (to_sb.st_dev == from_sb.st_dev && to_sb.st_ino == from_sb.st_ino) {
			fprintf(stderr, "install: %s and %s are the same file.\n", argv[0], path);
			exit(1);
		}
		/* unlink now... avoid ETXTBSY errors later */
		(void)unlink(path);
	}

nocompare:
	/* open target, set mode, owner, group */
	if ((to_fd = open(path, O_CREAT|O_WRONLY|O_TRUNC, 0)) < 0) {
		fprintf(stderr, "install: %s: %s\n", path, sys_errlist[errno]);
		exit(1);
	}
	if (fchmod(to_fd, mode)) {
		fprintf(stderr, "install: fchmod: %s: %s\n", path, sys_errlist[errno]);
		bad();
	}
	if (fchown(to_fd, pp->pw_uid, gp->gr_gid)) {
		fprintf(stderr, "install: fchown: %s: %s\n", path, sys_errlist[errno]);
		bad();
	}

	if (devnull)
		exit(0);

	if (dostrip) {
		strip(to_fd, argv[0], path);
		if (docopy)
			exit(0);
	}
	else if (docopy) {
		copy(argv[0], to_fd, path);
		exit(0);
	}
	else if (rename(argv[0], path))
		copy(argv[0], to_fd, path);
	(void)unlink(argv[0]);
	exit(0);
}

/*
 * copy --
 *	copy from one file to another
 */
static
copy(from_name, to_fd, to_name)
	register int	to_fd;
	char	*from_name, *to_name;
{
	register int	n, from_fd;
	char	buf[MAXBSIZE];

	if ((from_fd = open(from_name, O_RDONLY, 0)) < 0) {
		fprintf(stderr, "install: open: %s: %s\n", from_name, sys_errlist[errno]);
		bad();
	}
	while ((n = read(from_fd, buf, sizeof(buf))) > 0)
		if (write(to_fd, buf, n) != n) {
			fprintf(stderr, "install: write: %s: %s\n", to_name, sys_errlist[errno]);
			bad();
		}
	if (n == -1) {
		fprintf(stderr, "install: read: %s: %s\n", from_name, sys_errlist[errno]);
		bad();
	}
}

/*
 * strip --
 *	copy file, strip(1)'ing it at the same time
 */
static
strip(to_fd, from_name, to_name)
	register int	to_fd;
	char	*from_name, *to_name;
{
	typedef struct exec	EXEC;
	register long	size;
	register int	n, from_fd;
	EXEC	head;
	char	buf[MAXBSIZE];

	if ((from_fd = open(from_name, O_RDONLY, 0)) < 0) {
		fprintf(stderr, "install: open: %s: %s\n", from_name, sys_errlist[errno]);
		bad();
	}
	if (read(from_fd, (char *)&head, sizeof(head)) < 0 || N_BADMAG(head)) {
		fprintf(stderr, "install: %s not in a.out format.\n", from_name);
		bad();
	}
	if (head.a_syms || head.a_trsize || head.a_drsize) {
		size = (long)head.a_text + head.a_data;
		head.a_syms = head.a_trsize = head.a_drsize = 0;
		if (head.a_magic == ZMAGIC)
			size += getpagesize() - sizeof(EXEC);
		if (write(to_fd, (char *)&head, sizeof(EXEC)) != sizeof(EXEC)) {
			fprintf(stderr, "install: write: %s: %s\n", to_name, sys_errlist[errno]);
			bad();
		}
		for (; size; size -= n)
			if ((n = read(from_fd, buf, (int)MIN(size, sizeof(buf)))) <= 0)
				break;
			else if (write(to_fd, buf, n) != n) {
				fprintf(stderr, "install: write: %s: %s\n", to_name, sys_errlist[errno]);
				bad();
			}
		if (size) {
			fprintf(stderr, "install: read: %s: premature EOF.\n", from_name);
			bad();
		}
		if (n == -1) {
			fprintf(stderr, "install: read: %s: %s\n", from_name, sys_errlist[errno]);
			bad();
		}
	}
}

/*
 * atoo --
 *	octal string to int
 */
static
atoo(str)
	register char	*str;
{
	register int	val;

	for (val = 0; isdigit(*str); ++str)
		val = val * 8 + *str - '0';
	return(val);
}

/*
 * usage --
 *	print a usage message and die
 */
static
usage()
{
	fputs("usage: install [-cs] [-g group] [-m mode] [-o owner] source destination\n", stderr);
	exit(1);
}

/*
 * bad --
 *	remove created target and die
 */
static
bad()
{
	(void)unlink(path);
	exit(1);
}