oneechanhax/phunix
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
phunix/commands/simple/cp.c
Ben Gras 64695715da *** empty log message ***
2005-08-11 16:18:53 +00:00

1381 lines
31 KiB
C
Executable File
Raw Blame History

/* 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>
#include <sys/dir.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= DIRSIZ + 2);
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 ..\n", olddst);
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 ..\n" : "mv %s ..\n", src);
}
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..\n", src);
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);
}
Reference in New Issue View Git Blame Copy Permalink