/*	cp 1.12 - copy files				Author: Kees J. Bot
 *	mv      - move files					20 Jul 1993
 *	rm      - remove files
 *	ln      - make a link
 *	cpdir   - copy a directory tree (cp -psmr)
 *	clone   - make a link farm (ln -fmr)
 */
#define nil 0
#include <stdio.h>
#include <sys/types.h>
#include <stdlib.h>
#include <string.h>
#include <stddef.h>
#include <unistd.h>
#include <fcntl.h>
#include <time.h>
#include <sys/stat.h>
#include <utime.h>
#include <dirent.h>
#include <errno.h>
#ifndef DEBUG
#define DEBUG	0
#define NDEBUG	1
#endif
#include <assert.h>


/* Copy files in this size chunks: */
#if __minix && !__minix_vmd
#define CHUNK	(8192 * sizeof(char *))
#else
#define CHUNK	(1024 << (sizeof(int) + sizeof(char *)))
#endif


#ifndef CONFORMING
#define CONFORMING	1	/* Precisely POSIX conforming. */
#endif


#define arraysize(a)	(sizeof(a) / sizeof((a)[0]))
#define arraylimit(a)	((a) + arraysize(a))

char *prog_name;		/* Call name of this program. */
int ex_code= 0;			/* Final exit code. */

typedef enum identity { CP, MV, RM, LN, CPDIR, CLONE } identity_t;
typedef enum action { COPY, MOVE, REMOVE, LINK } action_t;

identity_t identity;		/* How did the user call me? */
action_t action;		/* Copying, moving, or linking. */
int pflag= 0;			/* -p/-s: Make orginal and copy the same. */
int iflag= 0;			/* -i: Interactive overwriting/deleting. */
int fflag= 0;			/* -f: Force. */
int sflag= 0;			/* -s: Make a symbolic link (ln/clone). */
int Sflag= 0;			/* -S: Make a symlink if across devices. */
int mflag= 0;			/* -m: Merge trees, no target dir trickery. */
int rflag= 0;			/* -r/-R: Recursively copy a tree. */
int vflag= 0;			/* -v: Verbose. */
int xflag= 0;			/* -x: Don't traverse past mount points. */
int xdev= 0;			/* Set when moving or linking cross-device. */
int expand= 0;			/* Expand symlinks, ignore links. */
int conforming= CONFORMING;	/* Sometimes standards are a pain. */

int fc_mask;			/* File creation mask. */
int uid, gid;			/* Effective uid & gid. */
int istty;			/* Can have terminal input. */

#ifndef S_ISLNK
/* There were no symlinks in medieval times. */
#define S_ISLNK(mode)			(0)
#define lstat				stat
#define symlink(path1, path2)		(errno= ENOSYS, -1)
#define readlink(path, buf, len)	(errno= ENOSYS, -1)
#endif

void report(const char *label)
{
    if (action == REMOVE && fflag) return;
    fprintf(stderr, "%s: %s: %s\n", prog_name, label, strerror(errno));
    ex_code= 1;
}

void fatal(const char *label)
{
    report(label);
    exit(1);
}

void report2(const char *src, const char *dst)
{
    fprintf(stderr, "%s %s %s: %s\n", prog_name, src, dst, strerror(errno));
    ex_code= 1;
}

#if DEBUG
size_t nchunks= 0;	/* Number of allocated cells. */
#endif

void *allocate(void *mem, size_t size)
/* Like realloc, but with checking of the return value. */
{
#if DEBUG
    if (mem == nil) nchunks++;
#endif
    if ((mem= mem == nil ? malloc(size) : realloc(mem, size)) == nil)
	fatal("malloc()");
    return mem;
}

void deallocate(void *mem)
/* Release a chunk of memory. */
{
    if (mem != nil) {
#if DEBUG
	nchunks--;
#endif
	free(mem);
    }
}

typedef struct pathname {
    char		*path;	/* The actual pathname. */
    size_t		idx;	/* Index for the terminating null byte. */
    size_t		lim;	/* Actual length of the path array. */
} pathname_t;

void path_init(pathname_t *pp)
/* Initialize a pathname to the null string. */
{
    pp->path= allocate(nil, pp->lim= 16);
    pp->path[pp->idx= 0]= 0;
}

void path_add(pathname_t *pp, const char *name)
/* Add a component to a pathname. */
{
    size_t lim;
    char *p;

    lim= pp->idx + strlen(name) + 2;

    if (lim > pp->lim) {
	pp->lim= lim += lim/2;	/* add an extra 50% growing space. */

	pp->path= allocate(pp->path, lim);
    }

    p= pp->path + pp->idx;
    if (p > pp->path && p[-1] != '/') *p++ = '/';

    while (*name != 0) {
	if (*name != '/' || p == pp->path || p[-1] != '/') *p++ = *name;
	name++;
    }
    *p = 0;
    pp->idx= p - pp->path;
}

void path_trunc(pathname_t *pp, size_t didx)
/* Delete part of a pathname to a remembered length. */
{
    pp->path[pp->idx= didx]= 0;
}

#if DEBUG
const char *path_name(const pathname_t *pp)
/* Return the actual name as a C string. */
{
    return pp->path;
}

size_t path_length(const pathname_t *pp)
/* The length of the pathname. */
{
    return pp->idx;
}

void path_drop(pathname_t *pp)
/* Release the storage occupied by the pathname. */
{
    deallocate(pp->path);
}

#else /* !DEBUG */
#define path_name(pp)		((const char *) (pp)->path)
#define path_length(pp)		((pp)->idx)
#define path_drop(pp)		deallocate((void *) (pp)->path)
#endif /* !DEBUG */

char *basename(const char *path)
/* Return the last component of a pathname.  (Note: declassifies a const
 * char * just like strchr.
 */
{
    const char *p= path;

    for (;;) {
	while (*p == '/') p++;			/* Trailing slashes? */

	if (*p == 0) break;

	path= p;
	while (*p != 0 && *p != '/') p++;	/* Skip component. */
    }
    return (char *) path;
}

int affirmative(void)
/* Get a yes/no answer from the suspecting user. */
{
    int c;
    int ok;

    fflush(stdout);
    fflush(stderr);

    while ((c= getchar()) == ' ') {}
    ok= (c == 'y' || c == 'Y');
    while (c != EOF && c != '\n') c= getchar();

    return ok;
}

int writable(const struct stat *stp)
/* True iff the file with the given attributes allows writing.  (And we have
 * a terminal to ask if ok to overwrite.)
 */
{
    if (!istty || uid == 0) return 1;
    if (stp->st_uid == uid) return stp->st_mode & S_IWUSR;
    if (stp->st_gid == gid) return stp->st_mode & S_IWGRP;
    return stp->st_mode & S_IWOTH;
}

#ifndef PATH_MAX
#define PATH_MAX	1024
#endif

static char *link_islink(struct stat *stp, const char *file)
{
    /* Tell if a file, which stat(2) information in '*stp', has been seen
     * earlier by this function under a different name.  If not return a
     * null pointer with errno set to ENOENT, otherwise return the name of
     * the link.  Return a null pointer with an error code in errno for any
     * error, using E2BIG for a too long file name.
     *
     * Use link_islink(nil, nil) to reset all bookkeeping.
     *
     * Call for a file twice to delete it from the store.
     */

    typedef struct link {	/* In-memory link store. */
	struct link	*next;		/* Hash chain on inode number. */
	ino_t		ino;		/* File's inode number. */
	off_t		off;		/* Offset to more info in temp file. */
    } link_t;
    typedef struct dlink {	/* On-disk link store. */
	dev_t		dev;		/* Device number. */
	char		file[PATH_MAX];	/* Name of earlier seen link. */
    } dlink_t;
    static link_t *links[256];		/* Hash list of known links. */
    static int tfd= -1;			/* Temp file for file name storage. */
    static dlink_t dlink;
    link_t *lp, **plp;
    size_t len;
    off_t off;

    if (file == nil) {
	/* Reset everything. */
	for (plp= links; plp < arraylimit(links); plp++) {
	    while ((lp= *plp) != nil) {
		*plp= lp->next;
		free(lp);
	    }
	}
	if (tfd != -1) close(tfd);
	tfd= -1;
	return nil;
    }

    /* The file must be a non-directory with more than one link. */
    if (S_ISDIR(stp->st_mode) || stp->st_nlink <= 1) {
	errno= ENOENT;
	return nil;
    }

    plp= &links[stp->st_ino % arraysize(links)];

    while ((lp= *plp) != nil) {
	if (lp->ino == stp->st_ino) {
	    /* May have seen this link before.  Get it and check. */
	    if (lseek(tfd, lp->off, SEEK_SET) == -1) return nil;
	    if (read(tfd, &dlink, sizeof(dlink)) < 0) return nil;

	    /* Only need to check the device number. */
	    if (dlink.dev == stp->st_dev) {
		if (strcmp(file, dlink.file) == 0) {
		    /* Called twice.  Forget about this link. */
		    *plp= lp->next;
		    free(lp);
		    errno= ENOENT;
		    return nil;
		}

		/* Return the name of the earlier link. */
		return dlink.file;
	    }
	}
	plp= &lp->next;
    }

    /* First time I see this link.  Add it to the store. */
    if (tfd == -1) {
	for (;;) {
	    char *tmp;

	    tmp= tmpnam(nil);
	    tfd= open(tmp, O_RDWR|O_CREAT|O_EXCL, 0600);
	    if (tfd < 0) {
		if (errno != EEXIST) return nil;
	    } else {
		(void) unlink(tmp);
		break;
	    }
	}
    }
    if ((len= strlen(file)) >= PATH_MAX) {
	errno= E2BIG;
	return nil;
    }

    dlink.dev= stp->st_dev;
    strcpy(dlink.file, file);
    len += offsetof(dlink_t, file) + 1;
    if ((off= lseek(tfd, 0, SEEK_END)) == -1) return nil;
    if (write(tfd, &dlink, len) != len) return nil;

    if ((lp= malloc(sizeof(*lp))) == nil) return nil;
    lp->next= nil;
    lp->ino= stp->st_ino;
    lp->off= off;
    *plp= lp;
    errno= ENOENT;
    return nil;
}

int trylink(const char *src, const char *dst, struct stat *srcst,
			struct stat *dstst)
/* Keep the link structure intact if src has been seen before. */
{
    char *olddst;
    int linked;

    if (action == COPY && expand) return 0;

    if ((olddst= link_islink(srcst, dst)) == nil) {
	/* if (errno != ENOENT) ... */
	return 0;
    }

    /* Try to link the file copied earlier to the new file. */
    if (dstst->st_ino != 0) (void) unlink(dst);

    if ((linked= (link(olddst, dst) == 0)) && vflag)
	printf("ln %s %s\n", olddst, dst);

    return linked;
}

int copy(const char *src, const char *dst, struct stat *srcst,
			struct stat *dstst)
/* Copy one file to another and copy (some of) the attributes. */
{
    char buf[CHUNK];
    int srcfd, dstfd;
    ssize_t n;

    assert(srcst->st_ino != 0);

    if (dstst->st_ino == 0) {
	/* The file doesn't exist yet. */

	if (!S_ISREG(srcst->st_mode)) {
	    /* Making a new mode 666 regular file. */
	    srcst->st_mode= (S_IFREG | 0666) & fc_mask;
	} else
	if (!pflag && conforming) {
	    /* Making a new file copying mode with umask applied. */
	    srcst->st_mode &= fc_mask;
	}
    } else {
	/* File exists, ask if ok to overwrite if '-i'. */

	if (iflag || (action == MOVE && !fflag && !writable(dstst))) {
	    fprintf(stderr, "Overwrite %s? (mode = %03o) ",
			dst, dstst->st_mode & 07777);
	    if (!affirmative()) return 0;
	}

	if (action == MOVE) {
	    /* Don't overwrite, remove first. */
	    if (unlink(dst) < 0 && errno != ENOENT) {
		report(dst);
		return 0;
	    }
	} else {
	    /* Overwrite. */
	    if (!pflag) {
		/* Keep the existing mode and ownership. */
		srcst->st_mode= dstst->st_mode;
		srcst->st_uid= dstst->st_uid;
		srcst->st_gid= dstst->st_gid;
	    }
	}
    }

    /* Keep the link structure if possible. */
    if (trylink(src, dst, srcst, dstst)) return 1;

    if ((srcfd= open(src, O_RDONLY)) < 0) {
	report(src);
	return 0;
    }

    dstfd= open(dst, O_WRONLY|O_CREAT|O_TRUNC, srcst->st_mode & 0777);
    if (dstfd < 0 && fflag && errno == EACCES) {
	/* Retry adding a "w" bit. */
	(void) chmod(dst, dstst->st_mode | S_IWUSR);
	dstfd= open(dst, O_WRONLY|O_CREAT|O_TRUNC, 0);
    }
    if (dstfd < 0 && fflag && errno == EACCES) {
	/* Retry after trying to delete. */
	(void) unlink(dst);
	dstfd= open(dst, O_WRONLY|O_CREAT|O_TRUNC, 0);
    }
    if (dstfd < 0) {
	report(dst);
	close(srcfd);
	return 0;
    }

    /* Get current parameters. */
    if (fstat(dstfd, dstst) < 0) {
	report(dst);
	close(srcfd);
	close(dstfd);
	return 0;
    }

    /* Copy the little bytes themselves. */
    while ((n= read(srcfd, buf, sizeof(buf))) > 0) {
	char *bp = buf;
	ssize_t r;

	while (n > 0 && (r= write(dstfd, bp, n)) > 0) {
	    bp += r;
	    n -= r;
	}
	if (r <= 0) {
	    if (r == 0) {
		fprintf(stderr,
		    "%s: Warning: EOF writing to %s\n",
		    prog_name, dst);
		break;
	    }
	    fatal(dst);
	}
    }

    if (n < 0) {
	report(src);
	close(srcfd);
	close(dstfd);
	return 0;
    }

    close(srcfd);
    close(dstfd);

    /* Copy the ownership. */
    if ((pflag || !conforming)
	&& S_ISREG(dstst->st_mode)
	&& (dstst->st_uid != srcst->st_uid
		|| dstst->st_gid != srcst->st_gid)
    ) {
	if (chmod(dst, 0) == 0) dstst->st_mode&= ~07777;
	if (chown(dst, srcst->st_uid, srcst->st_gid) < 0) {
	    if (errno != EPERM) {
		report(dst);
		return 0;
	    }
	} else {
	    dstst->st_uid= srcst->st_uid;
	    dstst->st_gid= srcst->st_gid;
	}
    }

    if (conforming && S_ISREG(dstst->st_mode)
	&& (dstst->st_uid != srcst->st_uid
		|| dstst->st_gid != srcst->st_gid)
    ) {
	/* Suid bits must be cleared in the holy name of
	 * security (and the assumed user stupidity).
	 */
	srcst->st_mode&= ~06000;
    }

    /* Copy the mode. */
    if (S_ISREG(dstst->st_mode) && dstst->st_mode != srcst->st_mode) {
	if (chmod(dst, srcst->st_mode) < 0) {
	    if (errno != EPERM) {
		report(dst);
		return 0;
	    }
	    fprintf(stderr, "%s: Can't change the mode of %s\n",
		prog_name, dst);
	}
    }

    /* Copy the file modification time. */
    if ((pflag || !conforming) && S_ISREG(dstst->st_mode)) {
	struct utimbuf ut;

	ut.actime= action == MOVE ? srcst->st_atime : time(nil);
	ut.modtime= srcst->st_mtime;
	if (utime(dst, &ut) < 0) {
	    if (errno != EPERM) {
		report(dst);
		return 0;
	    }
	    if (pflag) {
		fprintf(stderr,
		    "%s: Can't set the time of %s\n",
		    prog_name, dst);
	    }
	}
    }
    if (vflag) {
	printf(action == COPY ? "cp %s %s\n" : "mv %s %s\n", src, dst);
    }
    return 1;
}

void copy1(const char *src, const char *dst, struct stat *srcst,
			    struct stat *dstst)
/* Inspect the source file and then copy it.  Treatment of symlinks and
 * special files is a bit complicated.  The filetype and link-structure are
 * ignored if (expand && !rflag), symlinks and link-structure are ignored
 * if (expand && rflag), everything is copied precisely if !expand.
 */
{
    int r, linked;

    assert(srcst->st_ino != 0);

    if (srcst->st_ino == dstst->st_ino && srcst->st_dev == dstst->st_dev) {
	fprintf(stderr, "%s: can't copy %s onto itself\n",
	    prog_name, src);
	ex_code= 1;
	return;
    }

    /* You can forget it if the destination is a directory. */
    if (dstst->st_ino != 0 && S_ISDIR(dstst->st_mode)) {
	errno= EISDIR;
	report(dst);
	return;
    }

    if (S_ISREG(srcst->st_mode) || (expand && !rflag)) {
	if (!copy(src, dst, srcst, dstst)) return;

	if (action == MOVE && unlink(src) < 0) {
	    report(src);
	    return;
	}
	return;
    }

    if (dstst->st_ino != 0) {
	if (iflag || (action == MOVE && !fflag && !writable(dstst))) {
	    fprintf(stderr, "Replace %s? (mode = %03o) ",
		dst, dstst->st_mode & 07777);
	    if (!affirmative()) return;
	}
	if (unlink(dst) < 0) {
	    report(dst);
	    return;
	}
	dstst->st_ino= 0;
    }

    /* Apply the file creation mask if so required. */
    if (!pflag && conforming) srcst->st_mode &= fc_mask;

    linked= 0;

    if (S_ISLNK(srcst->st_mode)) {
	char buf[1024+1];

	if ((r= readlink(src, buf, sizeof(buf)-1)) < 0) {
	    report(src);
	    return;
	}
	buf[r]= 0;
	r= symlink(buf, dst);
	if (vflag && r == 0)
	    printf("ln -s %s %s\n", buf, dst);
    } else
    if (trylink(src, dst, srcst, dstst)) {
	linked= 1;
	r= 0;
    } else
    if (S_ISFIFO(srcst->st_mode)) {
	r= mkfifo(dst, srcst->st_mode);
	if (vflag && r == 0)
	    printf("mkfifo %s\n", dst);
    } else
    if (S_ISBLK(srcst->st_mode) || S_ISCHR(srcst->st_mode)) {
	r= mknod(dst, srcst->st_mode, srcst->st_rdev);
	if (vflag && r == 0) {
	    printf("mknod %s %c %d %d\n",
		dst,
		S_ISBLK(srcst->st_mode) ? 'b' : 'c',
		(srcst->st_rdev >> 8) & 0xFF,
		(srcst->st_rdev >> 0) & 0xFF);
	}
    } else {
	fprintf(stderr, "%s: %s: odd filetype %5o (not copied)\n",
	    prog_name, src, srcst->st_mode);
	ex_code= 1;
	return;
    }

    if (r < 0 || lstat(dst, dstst) < 0) {
	report(dst);
	return;
    }

    if (action == MOVE && unlink(src) < 0) {
	report(src);
	(void) unlink(dst);	/* Don't want it twice. */
	return;
    }

    if (linked) return;

    if (S_ISLNK(srcst->st_mode)) return;

    /* Copy the ownership. */
    if ((pflag || !conforming)
	&& (dstst->st_uid != srcst->st_uid
		|| dstst->st_gid != srcst->st_gid)
    ) {
	if (chown(dst, srcst->st_uid, srcst->st_gid) < 0) {
	    if (errno != EPERM) {
		report(dst);
		return;
	    }
	}
    }

    /* Copy the file modification time. */
    if (pflag || !conforming) {
	struct utimbuf ut;

	ut.actime= action == MOVE ? srcst->st_atime : time(nil);
	ut.modtime= srcst->st_mtime;
	if (utime(dst, &ut) < 0) {
	    if (errno != EPERM) {
		report(dst);
		return;
	    }
	    fprintf(stderr, "%s: Can't set the time of %s\n",
		prog_name, dst);
	}
    }
}

void remove1(const char *src, struct stat *srcst)
{
    if (iflag || (!fflag && !writable(srcst))) {
	fprintf(stderr, "Remove %s? (mode = %03o) ", src,
			srcst->st_mode & 07777);
	if (!affirmative()) return;
    }
    if (unlink(src) < 0) {
	report(src);
    } else {
	if (vflag) printf("rm %s\n", src);
    }
}

void link1(const char *src, const char *dst, struct stat *srcst,
			    struct stat *dstst)
{
    pathname_t sym;
    const char *p;

    if (dstst->st_ino != 0 && (iflag || fflag)) {
	if (srcst->st_ino == dstst->st_ino) {
	    if (fflag) return;
	    fprintf(stderr, "%s: Can't link %s onto itself\n",
		prog_name, src);
	    ex_code= 1;
	    return;
	}
	if (iflag) {
	    fprintf(stderr, "Remove %s? ", dst);
	    if (!affirmative()) return;
	}
	errno= EISDIR;
	if (S_ISDIR(dstst->st_mode) || unlink(dst) < 0) {
	    report(dst);
	    return;
	}
    }

    if (!sflag && !(rflag && S_ISLNK(srcst->st_mode)) && !(Sflag && xdev)) {
	/* A normal link. */
	if (link(src, dst) < 0) {
	    if (!Sflag || errno != EXDEV) {
		report2(src, dst);
		return;
	    }
	    /* Can't do a cross-device link, we have to symlink. */
	    xdev= 1;
	} else {
	    if (vflag) printf("ln %s %s\n", src, dst);
	    return;
	}
    }

    /* Do a symlink. */
    if (!rflag && !Sflag) {
	/* We can get away with a "don't care if it works" symlink. */
	if (symlink(src, dst) < 0) {
	    report(dst);
	    return;
	}
	if (vflag) printf("ln -s %s %s\n", src, dst);
	return;
    }

    /* If the source is a symlink then it is simply copied. */
    if (S_ISLNK(srcst->st_mode)) {
	int r;
	char buf[1024+1];

	if ((r= readlink(src, buf, sizeof(buf)-1)) < 0) {
	    report(src);
	    return;
	}
	buf[r]= 0;
	if (symlink(buf, dst) < 0) {
	    report(dst);
	    return;
	}
	if (vflag) printf("ln -s %s %s\n", buf, dst);
	return;
    }

    /* Make a symlink that has to work, i.e. we must be able to access the
     * source now, and the link must work.
     */
    if (dst[0] == '/' && src[0] != '/') {
	/* ln -[rsS] relative/path /full/path. */
	fprintf(stderr,
	    "%s: Symlinking %s to %s is too difficult for me to figure out\n",
	    prog_name, src, dst);
	exit(1);
    }

    /* Count the number of subdirectories in the destination file and
     * add one '..' for each.
     */
    path_init(&sym);
    if (src[0] != '/') {
	p= dst;
	while (*p != 0) {
	    if (p[0] == '.') {
		if (p[1] == '/' || p[1] == 0) {
		    /* A "." component; skip. */
		    do p++; while (*p == '/');
		    continue;
		} else
		if (p[1] == '.' && (p[2] == '/' || p[2] == 0)) {
		    /* A ".." component; oops. */
		    switch (path_length(&sym)) {
		    case 0:
			fprintf(stderr,
	    "%s: Symlinking %s to %s is too difficult for me to figure out\n",
			    prog_name, src, dst);
			exit(1);
		    case 2:
			path_trunc(&sym, 0);
			break;
		    default:
			path_trunc(&sym, path_length(&sym) - 3);
		    }
		    p++;
		    do p++; while (*p == '/');
		    continue;
		}
	    }
	    while (*p != 0 && *p != '/') p++;
	    while (*p == '/') p++;
	    if (*p == 0) break;
	    path_add(&sym, "..");
	}
    }
    path_add(&sym, src);

    if (symlink(path_name(&sym), dst) < 0) {
	report(dst);
    } else {
	if (vflag) printf("ln -s %s %s\n", path_name(&sym), dst);
    }
    path_drop(&sym);
}

typedef struct entrylist {
    struct entrylist	*next;
    char			*name;
} entrylist_t;

int eat_dir(const char *dir, entrylist_t **dlist)
/* Make a linked list of all the names in a directory. */
{
    DIR *dp;
    struct dirent *entry;

    if ((dp= opendir(dir)) == nil) return 0;

    while ((entry= readdir(dp)) != nil) {
	if (strcmp(entry->d_name, ".") == 0) continue;
	if (strcmp(entry->d_name, "..") == 0) continue;

	*dlist= allocate(nil, sizeof(**dlist));
	(*dlist)->name= allocate(nil, strlen(entry->d_name)+1);
	strcpy((*dlist)->name, entry->d_name);
	dlist= &(*dlist)->next;
    }
    closedir(dp);
    *dlist= nil;
    return 1;
}

void chop_dlist(entrylist_t **dlist)
/* Chop an entry of a name list. */
{
    entrylist_t *junk= *dlist;

    *dlist= junk->next;
    deallocate(junk->name);
    deallocate(junk);
}

void drop_dlist(entrylist_t *dlist)
/* Get rid of a whole list. */
{
    while (dlist != nil) chop_dlist(&dlist);
}

void do1(pathname_t *src, pathname_t *dst, int depth)
/* Perform the appropriate action on a source and destination file. */
{
    size_t slashsrc, slashdst;
    struct stat srcst, dstst;
    entrylist_t *dlist;
    static ino_t topdst_ino;
    static dev_t topdst_dev;
    static dev_t topsrc_dev;

#if DEBUG
    if (vflag && depth == 0) {
	char flags[100], *pf= flags;

	if (pflag) *pf++= 'p';
	if (iflag) *pf++= 'i';
	if (fflag) *pf++= 'f';
	if (sflag) *pf++= 's';
	if (Sflag) *pf++= 'S';
	if (mflag) *pf++= 'm';
	if (rflag) *pf++= 'r';
	if (vflag) *pf++= 'v';
	if (xflag) *pf++= 'x';
	if (expand) *pf++= 'L';
	if (conforming) *pf++= 'C';
	*pf= 0;
	printf(": %s -%s %s %s\n", prog_name, flags,
		    path_name(src), path_name(dst));
    }
#endif

    /* st_ino == 0 if not stat()'ed yet, or nonexistent. */
    srcst.st_ino= 0;
    dstst.st_ino= 0;

    if (action != LINK || !sflag || rflag) {
	/* Source must exist unless symlinking. */
	if ((expand ? stat : lstat)(path_name(src), &srcst) < 0) {
	    report(path_name(src));
	    return;
	}
    }

    if (depth == 0) {
	/* First call: Not cross-device yet, first dst not seen yet,
	 * remember top device number.
	 */
	xdev= 0;
	topdst_ino= 0;
	topsrc_dev= srcst.st_dev;
    }

    /* Inspect the intended destination unless removing. */
    if (action != REMOVE) {
	if ((expand ? stat : lstat)(path_name(dst), &dstst) < 0) {
	    if (errno != ENOENT) {
		report(path_name(dst));
		return;
	    }
	}
    }

    if (action == MOVE && !xdev) {
	if (dstst.st_ino != 0 && srcst.st_dev != dstst.st_dev) {
	    /* It's a cross-device rename, i.e. copy and remove. */
	    xdev= 1;
	} else
	if (!mflag || dstst.st_ino == 0 || !S_ISDIR(dstst.st_mode)) {
	    /* Try to simply rename the file (not merging trees). */

	    if (srcst.st_ino == dstst.st_ino) {
		fprintf(stderr,
		    "%s: Can't move %s onto itself\n",
		    prog_name, path_name(src));
		ex_code= 1;
		return;
	    }

	    if (dstst.st_ino != 0) {
		if (iflag || (!fflag && !writable(&dstst))) {
		    fprintf(stderr,
			"Replace %s? (mode = %03o) ",
			path_name(dst),
			dstst.st_mode & 07777);
		    if (!affirmative()) return;
		}
		if (!S_ISDIR(dstst.st_mode))
		    (void) unlink(path_name(dst));
	    }

	    if (rename(path_name(src), path_name(dst)) == 0) {
		/* Success. */
		if (vflag) {
		    printf("mv %s %s\n", path_name(src),
			    path_name(dst));
		}
		return;
	    }
	    if (errno == EXDEV) {
		xdev= 1;
	    } else {
		report2(path_name(src), path_name(dst));
		return;
	    }
	}
    }

    if (srcst.st_ino == 0 || !S_ISDIR(srcst.st_mode)) {
	/* Copy/move/remove/link a single file. */
	switch (action) {
	case COPY:
	case MOVE:
	    copy1(path_name(src), path_name(dst), &srcst, &dstst);
	    break;
	case REMOVE:
	    remove1(path_name(src), &srcst);
	    break;
	case LINK:
	    link1(path_name(src), path_name(dst), &srcst, &dstst);
	    break;
	}
	return;
    }

    /* Recursively copy/move/remove/link a directory if -r or -R. */
    if (!rflag) {
	errno= EISDIR;
	report(path_name(src));
	return;
    }

    /* Ok to remove contents of dir? */
    if (action == REMOVE) {
	if (xflag && topsrc_dev != srcst.st_dev) {
	    /* Don't recurse past a mount point. */
	    return;
	}
	if (iflag) {
	    fprintf(stderr, "Remove contents of %s? ", path_name(src));
	    if (!affirmative()) return;
	}
    }

    /* Gather the names in the source directory. */
    if (!eat_dir(path_name(src), &dlist)) {
	report(path_name(src));
	return;
    }

    /* Check/create the target directory. */
    if (action != REMOVE && dstst.st_ino != 0 && !S_ISDIR(dstst.st_mode)) {
	if (action != MOVE && !fflag) {
	    errno= ENOTDIR;
	    report(path_name(dst));
	    return;
	}
	if (iflag) {
	    fprintf(stderr, "Replace %s? ", path_name(dst));
	    if (!affirmative()) {
		drop_dlist(dlist);
		return;
	    }
	}
	if (unlink(path_name(dst)) < 0) {
	    report(path_name(dst));
	    drop_dlist(dlist);
	    return;
	}
	dstst.st_ino= 0;
    }

    if (action != REMOVE) {
	if (dstst.st_ino == 0) {
	    /* Create a new target directory. */
	    if (!pflag && conforming) srcst.st_mode&= fc_mask;

	    if (mkdir(path_name(dst), srcst.st_mode | S_IRWXU) < 0
		    || stat(path_name(dst), &dstst) < 0) {
		report(path_name(dst));
		drop_dlist(dlist);
		return;
	    }
	    if (vflag) printf("mkdir %s\n", path_name(dst));
	} else {
	    /* Target directory already exists. */
	    if (action == MOVE && !mflag) {
		errno= EEXIST;
		report(path_name(dst));
		drop_dlist(dlist);
		return;
	    }
	    if (!pflag) {
		/* Keep the existing attributes. */
		srcst.st_mode= dstst.st_mode;
		srcst.st_uid= dstst.st_uid;
		srcst.st_gid= dstst.st_gid;
		srcst.st_mtime= dstst.st_mtime;
	    }
	}

	if (topdst_ino == 0) {
	    /* Remember the top destination. */
	    topdst_dev= dstst.st_dev;
	    topdst_ino= dstst.st_ino;
	}

	if (srcst.st_ino == topdst_ino && srcst.st_dev == topdst_dev) {
	    /* E.g. cp -r /shallow /shallow/deep. */
	    fprintf(stderr,
		"%s%s %s/ %s/: infinite recursion avoided\n",
		prog_name, action != MOVE ? " -r" : "",
		path_name(src), path_name(dst));
	    drop_dlist(dlist);
	    return;
	}

	if (xflag && topsrc_dev != srcst.st_dev) {
	    /* Don't recurse past a mount point. */
	    drop_dlist(dlist);
	    return;
	}
    }

    /* Go down. */
    slashsrc= path_length(src);
    slashdst= path_length(dst);

    while (dlist != nil) {
	path_add(src, dlist->name);
	if (action != REMOVE) path_add(dst, dlist->name);

	do1(src, dst, depth+1);

	path_trunc(src, slashsrc);
	path_trunc(dst, slashdst);
	chop_dlist(&dlist);
    }

    if (action == MOVE || action == REMOVE) {
	/* The contents of the source directory should have
	 * been (re)moved above.  Get rid of the empty dir.
	 */
	if (action == REMOVE && iflag) {
	    fprintf(stderr, "Remove directory %s? ",
			    path_name(src));
	    if (!affirmative()) return;
	}
	if (rmdir(path_name(src)) < 0) {
	    if (errno != ENOTEMPTY) report(path_name(src));
	    return;
	}
	if (vflag) printf("rmdir %s\n", path_name(src));
    }

    if (action != REMOVE) {
	/* Set the attributes of a new directory. */
	struct utimbuf ut;

	/* Copy the ownership. */
	if ((pflag || !conforming)
	    && (dstst.st_uid != srcst.st_uid
		|| dstst.st_gid != srcst.st_gid)
	) {
	    if (chown(path_name(dst), srcst.st_uid,
			    srcst.st_gid) < 0) {
		if (errno != EPERM) {
		    report(path_name(dst));
		    return;
		}
	    }
	}

	/* Copy the mode. */
	if (dstst.st_mode != srcst.st_mode) {
	    if (chmod(path_name(dst), srcst.st_mode) < 0) {
		report(path_name(dst));
		return;
	    }
	}

	/* Copy the file modification time. */
	if (dstst.st_mtime != srcst.st_mtime) {
	    ut.actime= action == MOVE ? srcst.st_atime : time(nil);
	    ut.modtime= srcst.st_mtime;
	    if (utime(path_name(dst), &ut) < 0) {
		if (errno != EPERM) {
		    report(path_name(dst));
		    return;
		}
		fprintf(stderr,
		    "%s: Can't set the time of %s\n",
		    prog_name, path_name(dst));
	    }
	}
    }
}

void usage(void)
{
    char *flags1, *flags2;

    switch (identity) {
    case CP:
	flags1= "pifsmrRvx";
	flags2= "pifsrRvx";
	break;
    case MV:
	flags1= "ifsmvx";
	flags2= "ifsvx";
	break;
    case RM:
	fprintf(stderr, "Usage: rm [-ifrRvx] file ...\n");
	exit(1);
    case LN:
	flags1= "ifsSmrRvx";
	flags2= "ifsSrRvx";
	break;
    case CPDIR:
	flags1= "ifvx";
	flags2= nil;
	break;
    case CLONE:
	flags1= "ifsSvx";
	flags2= nil;
	break;
    }
    fprintf(stderr, "Usage: %s [-%s] file1 file2\n", prog_name, flags1);
  if (flags2 != nil)
    fprintf(stderr, "       %s [-%s] file ... dir\n", prog_name, flags2);
    exit(1);
}

void main(int argc, char **argv)
{
    int i;
    char *flags;
    struct stat st;
    pathname_t src, dst;
    size_t slash;

#if DEBUG >= 3
    /* The first argument is the call name while debugging. */
    if (argc < 2) exit(-1);
    argv++;
    argc--;
#endif
#if DEBUG
    vflag= isatty(1);
#endif

    /* Call name of this program. */
    prog_name= basename(argv[0]);

    /* Required action. */
    if (strcmp(prog_name, "cp") == 0) {
	identity= CP;
	action= COPY;
	flags= "pifsmrRvx";
	expand= 1;
    } else
    if (strcmp(prog_name, "mv") == 0) {
	identity= MV;
	action= MOVE;
	flags= "ifsmvx";
	rflag= pflag= 1;
    } else
    if (strcmp(prog_name, "rm") == 0) {
	identity= RM;
	action= REMOVE;
	flags= "ifrRvx";
    } else
    if (strcmp(prog_name, "ln") == 0) {
	identity= LN;
	action= LINK;
	flags= "ifsSmrRvx";
    } else
    if (strcmp(prog_name, "cpdir") == 0) {
	identity= CPDIR;
	action= COPY;
	flags= "pifsmrRvx";
	rflag= mflag= pflag= 1;
	conforming= 0;
    } else
    if (strcmp(prog_name, "clone") == 0) {
	identity= CLONE;
	action= LINK;
	flags= "ifsSmrRvx";
	rflag= mflag= fflag= 1;
    } else {
	fprintf(stderr,
	    "%s: Identity crisis, not called cp, mv, rm, ln, cpdir, or clone\n",
	    prog_name);
	exit(1);
    }

    /* Who am I?, where am I?, how protective am I? */
    uid= geteuid();
    gid= getegid();
    istty= isatty(0);
    fc_mask= ~umask(0);

    /* Gather flags. */
    i= 1;
    while (i < argc && argv[i][0] == '-') {
	char *opt= argv[i++] + 1;

	if (opt[0] == '-' && opt[1] == 0) break;	/* -- */

	while (*opt != 0) {
	    /* Flag supported? */
	    if (strchr(flags, *opt) == nil) usage();

	    switch (*opt++) {
	    case 'p':
		pflag= 1;
		break;
	    case 'i':
		iflag= 1;
		if (action == MOVE) fflag= 0;
		break;
	    case 'f':
		fflag= 1;
		if (action == MOVE) iflag= 0;
		break;
	    case 's':
		if (action == LINK) {
		    sflag= 1;
		} else {
		    /* Forget about POSIX, do it right. */
		    conforming= 0;
		}
		break;
	    case 'S':
		Sflag= 1;
		break;
	    case 'm':
		mflag= 1;
		break;
	    case 'r':
		expand= 0;
		/*FALL THROUGH*/
	    case 'R':
		rflag= 1;
		break;
	    case 'v':
		vflag= 1;
		break;
	    case 'x':
		xflag= 1;
		break;
	    default:
		assert(0);
	    }
	}
    }

    switch (action) {
    case REMOVE:
	if (i == argc) usage();
	break;
    case LINK:
	/* 'ln dir/file' is to be read as 'ln dir/file .'. */
	if ((argc - i) == 1 && action == LINK) argv[argc++]= ".";
	/*FALL THROUGH*/
    default:
	if ((argc - i) < 2) usage();
    }

    path_init(&src);
    path_init(&dst);

    if (action != REMOVE && !mflag
	&& stat(argv[argc-1], &st) >= 0 && S_ISDIR(st.st_mode)
    ) {
	/* The last argument is a directory, this means we have to
	 * throw the whole lot into this directory.  This is the
	 * Right Thing unless you use -r.
	 */
	path_add(&dst, argv[argc-1]);
	slash= path_length(&dst);

	do {
	    path_add(&src, argv[i]);
	    path_add(&dst, basename(argv[i]));

	    do1(&src, &dst, 0);

	    path_trunc(&src, 0);
	    path_trunc(&dst, slash);
	} while (++i < argc-1);
    } else
    if (action == REMOVE || (argc - i) == 2) {
	/* Just two files (or many files for rm). */
	do {
	    path_add(&src, argv[i]);
	    if (action != REMOVE) path_add(&dst, argv[i+1]);

	    do1(&src, &dst, 0);
	    path_trunc(&src, 0);
	} while (action == REMOVE && ++i < argc);
    } else {
	usage();
    }
    path_drop(&src);
    path_drop(&dst);

#if DEBUG
    if (nchunks != 0) {
	fprintf(stderr, "(%ld chunks of memory not freed)\n",
	    (long) nchunks);
    }
#endif
    exit(ex_code);
}
