156 lines
3.1 KiB
C
156 lines
3.1 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <utime.h>
|
|
|
|
#include "../fs.h"
|
|
#include "../text.h"
|
|
#include "../util.h"
|
|
|
|
int cp_aflag = 0;
|
|
int cp_fflag = 0;
|
|
int cp_pflag = 0;
|
|
int cp_rflag = 0;
|
|
int cp_vflag = 0;
|
|
int cp_status = 0;
|
|
int cp_HLPflag = 'L';
|
|
|
|
int
|
|
cp(const char *s1, const char *s2, int depth)
|
|
{
|
|
FILE *f1, *f2;
|
|
char *ns1, *ns2;
|
|
long size1, size2;
|
|
struct dirent *d;
|
|
struct stat st;
|
|
struct utimbuf ut;
|
|
char buf[PATH_MAX];
|
|
DIR *dp;
|
|
int r;
|
|
|
|
if (cp_vflag)
|
|
printf("'%s' -> '%s'\n", s1, s2);
|
|
|
|
r = (cp_HLPflag == 'P' || (cp_HLPflag == 'H' && depth > 0)) ?
|
|
lstat(s1, &st) : stat(s1, &st);
|
|
if (r < 0) {
|
|
weprintf("%s %s:", (cp_HLPflag == 'P' ||
|
|
(cp_HLPflag == 'H' && depth > 0)) ? "lstat" : "stat", s1);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
|
|
if (S_ISLNK(st.st_mode)) {
|
|
if (readlink(s1, buf, sizeof(buf) - 1) >= 0) {
|
|
if (cp_fflag)
|
|
unlink(s2);
|
|
if (symlink(buf, s2) != 0) {
|
|
weprintf("%s: can't create '%s'\n", argv0, s2);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
}
|
|
goto preserve;
|
|
}
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
|
if (!cp_rflag)
|
|
eprintf("%s: is a directory\n", s1);
|
|
|
|
if (!(dp = opendir(s1)))
|
|
eprintf("opendir %s:", s1);
|
|
|
|
if (mkdir(s2, st.st_mode) < 0 && errno != EEXIST)
|
|
eprintf("mkdir %s:", s2);
|
|
|
|
apathmax(&ns1, &size1);
|
|
apathmax(&ns2, &size2);
|
|
while ((d = readdir(dp))) {
|
|
if (strcmp(d->d_name, ".") && strcmp(d->d_name, "..")) {
|
|
r = snprintf(ns1, size1, "%s/%s", s1, d->d_name);
|
|
if (r >= size1 || r < 0) {
|
|
eprintf("%s/%s: filename too long\n",
|
|
s1, d->d_name);
|
|
}
|
|
r = snprintf(ns2, size2, "%s/%s", s2, d->d_name);
|
|
if (r >= size2 || r < 0) {
|
|
eprintf("%s/%s: filename too long\n",
|
|
s2, d->d_name);
|
|
}
|
|
fnck(ns1, ns2, cp, depth + 1);
|
|
}
|
|
}
|
|
closedir(dp);
|
|
free(ns1);
|
|
free(ns2);
|
|
goto preserve;
|
|
}
|
|
|
|
if (cp_aflag) {
|
|
if (S_ISBLK(st.st_mode) || S_ISCHR(st.st_mode) ||
|
|
S_ISSOCK(st.st_mode) || S_ISFIFO(st.st_mode)) {
|
|
unlink(s2);
|
|
if (mknod(s2, st.st_mode, st.st_rdev) < 0) {
|
|
weprintf("%s: can't create '%s':", argv0, s2);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
goto preserve;
|
|
}
|
|
}
|
|
|
|
if (!(f1 = fopen(s1, "r"))) {
|
|
weprintf("fopen %s:", s1);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
|
|
if (!(f2 = fopen(s2, "w"))) {
|
|
if (cp_fflag) {
|
|
unlink(s2);
|
|
if (!(f2 = fopen(s2, "w"))) {
|
|
weprintf("fopen %s:", s2);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
} else {
|
|
weprintf("fopen %s:", s2);
|
|
cp_status = 1;
|
|
return 0;
|
|
}
|
|
}
|
|
concat(f1, s1, f2, s2);
|
|
/* preserve permissions by default */
|
|
fchmod(fileno(f2), st.st_mode);
|
|
fclose(f2);
|
|
fclose(f1);
|
|
|
|
preserve:
|
|
if (cp_aflag || cp_pflag) {
|
|
if (!(S_ISLNK(st.st_mode))) {
|
|
/* timestamp */
|
|
ut.actime = st.st_atime;
|
|
ut.modtime = st.st_mtime;
|
|
utime(s2, &ut);
|
|
}
|
|
/* preserve owner ? */
|
|
if (S_ISLNK(st.st_mode))
|
|
r = lchown(s2, st.st_uid, st.st_gid);
|
|
else
|
|
r = chown(s2, st.st_uid, st.st_gid);
|
|
if (r < 0) {
|
|
weprintf("cp: can't preserve ownership of '%s':", s2);
|
|
cp_status = 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|