/* Copyright 2008 Rob Landley <rob@landley.net>
*
* See http://opengroup.org/onlinepubs/9699919799/utilities/cp.html
+ * And http://opengroup.org/onlinepubs/9699919799/utilities/mv.html
+ * And http://refspecs.linuxfoundation.org/LSB_5.0.0/LSB-Core-generic/LSB-Core-generic.html#INSTALL
*
- * TODO: sHLP
+ * Posix says "cp -Rf dir file" shouldn't delete file, but our -f does.
+ *
+ * Deviations from posix: -adlnrsvF, --preserve... about half the
+ * functionality in this cp isn't in posix. Posix is stuck in the 1970's.
+ *
+ * TODO: --preserve=links
+ * TODO: what's this _CP_mode system.posix_acl_ business? We chmod()?
-// This is subtle: MV options must be in same order (right to left) as CP
-// for FLAG_X macros to work out right.
+// options shared between mv/cp must be in same order (right to left)
+// for FLAG macros to work out right in shared infrastructure.
-USE_CP(NEWTOY(cp, "<2RHLPp"USE_CP_MORE("rdaslvn")"fi"USE_CP_MORE("[-ni]"), TOYFLAG_BIN))
-USE_CP_MV(OLDTOY(mv, cp, "<2"USE_CP_MORE("vn")"fi"USE_CP_MORE("[-ni]"), TOYFLAG_BIN))
+USE_CP(NEWTOY(cp, "<2"USE_CP_PRESERVE("(preserve):;")"RHLPprdaslvnF(remove-destination)fi[-HLPd][-ni]", TOYFLAG_BIN))
+USE_MV(NEWTOY(mv, "<2vnF(remove-destination)fi[-ni]", TOYFLAG_BIN))
+USE_INSTALL(NEWTOY(install, "<1cdDpsvm:o:g:", TOYFLAG_USR|TOYFLAG_BIN))
config CP
bool "cp"
default y
help
- usage: cp [-fipRHLP] SOURCE... DEST
+ usage: cp [-adlnrsvfipRHLP] SOURCE... DEST
Copy files from SOURCE to DEST. If more than one SOURCE, DEST must
be a directory.
- -f force copy by deleting destination file
+ -f delete destination files we can't write to
+ -F delete any existing destination file first (--remove-destination)
-i interactive, prompt before overwriting existing DEST
- -p preserve timestamps, ownership, and permissions
+ -p preserve timestamps, ownership, and mode
-R recurse into subdirectories (DEST must be a directory)
-H Follow symlinks listed on command line
-L Follow all symlinks
-P Do not follow symlinks [default]
-
-config CP_MORE
- bool "cp -rdavsl options"
- default y
- depends on CP
- help
- usage: cp [-rdavsl]
-
-a same as -dpr
-d don't dereference symlinks
-l hard link instead of copy
-s symlink instead of copy
-v verbose
-config CP_MV
- bool "mv"
+config CP_PRESERVE
+ bool "cp --preserve support"
default y
depends on CP
help
- usage: mv [-fi] SOURCE... DEST"
+ usage: cp [--preserve=motcxa]
- -f force copy by deleting destination file
- -i interactive, prompt before overwriting existing DEST
+ --preserve takes either a comma separated list of attributes, or the first
+ letter(s) of:
+
+ mode - permissions (ignore umask for rwx, copy suid and sticky bit)
+ ownership - user and group
+ timestamps - file creation, modification, and access times.
+ context - security context
+ xattr - extended attributes
+ all - all of the above
-config CP_MV_MORE
- bool
+config MV
+ bool "mv"
default y
- depends on CP_MV && CP_MORE
help
- usage: mv [-vn]
+ usage: mv [-fivn] SOURCE... DEST"
+ -f force copy by deleting destination file
+ -i interactive, prompt before overwriting existing DEST
-v verbose
-n no clobber (don't overwrite DEST)
+
+config INSTALL
+ bool "install"
+ default y
+ help
+ usage: install [-dDpsv] [-o USER] [-g GROUP] [-m MODE] [SOURCE...] DEST
+
+ Copy files and set attributes.
+
+ -d Act like mkdir -p
+ -D Create leading directories for DEST
+ -g Make copy belong to GROUP
+ -m Set permissions to MODE
+ -o Make copy belong to USER
+ -p Preserve timestamps
+ -s Call "strip -p"
+ -v Verbose
*/
+#define FORCE_FLAGS
#define FOR_cp
#include "toys.h"
-// TODO: PLHlsd
-
GLOBALS(
+ union {
+ struct {
+ // install's options
+ char *group;
+ char *user;
+ char *mode;
+ } i;
+ struct {
+ char *preserve;
+ } c;
+ };
+
char *destname;
struct stat top;
+ int (*callback)(struct dirtree *try);
+ uid_t uid;
+ gid_t gid;
+ int pflags;
)
+struct cp_preserve {
+ char *name;
+} static const cp_preserve[] = TAGGED_ARRAY(CP,
+ {"mode"}, {"ownership"}, {"timestamps"}, {"context"}, {"xattr"},
+);
+
// Callback from dirtree_read() for each file/directory under a source dir.
int cp_node(struct dirtree *try)
if (!dirtree_notdotdot(try)) return 0;
// If returning from COMEAGAIN, jump straight to -p logic at end.
- if (S_ISDIR(try->st.st_mode) && try->data == -1) {
+ if (S_ISDIR(try->st.st_mode) && try->again) {
fdout = try->extra;
err = 0;
} else {
// -d is only the same as -r for symlinks, not for directories
- if (S_ISLNK(try->st.st_mode) & (flags & FLAG_d)) flags |= FLAG_r;
+ if (S_ISLNK(try->st.st_mode) && (flags & FLAG_d)) flags |= FLAG_r;
// Detect recursive copies via repeated top node (cp -R .. .) or
// identical source/target (fun with hardlinks).
return 0;
}
- // Handle -inv
+ // Handle -invF
if (!faccessat(cfd, catch, F_OK, 0) && !S_ISDIR(cst.st_mode)) {
char *s;
- if (S_ISDIR(try->st.st_dev)) {
+ if (S_ISDIR(try->st.st_mode)) {
error_msg("dir at '%s'", s = dirtree_path(try, 0));
free(s);
return 0;
+ } else if ((flags & FLAG_F) && unlinkat(cfd, catch, 0)) {
+ error_msg("unlink '%s'", catch);
+ return 0;
} else if (flags & FLAG_n) return 0;
else if (flags & FLAG_i) {
- fprintf(stderr, "cp: overwrite '%s'", s = dirtree_path(try, 0));
+ fprintf(stderr, "%s: overwrite '%s'", toys.which->name,
+ s = dirtree_path(try, 0));
free(s);
- if (!yesno("", 1)) return 0;
+ if (!yesno(1)) return 0;
}
}
if (flags & FLAG_v) {
char *s = dirtree_path(try, 0);
- printf("cp '%s'\n", s);
+ printf("%s '%s'\n", toys.which->name, s);
free(s);
}
if (!mkdirat(cfd, catch, try->st.st_mode | 0200) || errno == EEXIST)
if (-1 != (try->extra = openat(cfd, catch, O_NOFOLLOW)))
- if (!fstat(try->extra, &st2))
- if (S_ISDIR(st2.st_mode)) return DIRTREE_COMEAGAIN;
+ if (!fstat(try->extra, &st2) && S_ISDIR(st2.st_mode))
+ return DIRTREE_COMEAGAIN
+ | (DIRTREE_SYMFOLLOW*!!(toys.optflags&FLAG_L));
// Hardlink
if (*or->name == '/') dotdots = 0;
if (dotdots) {
- char *s2 = xmprintf("% *c%s", 3*dotdots, ' ', s);
+ char *s2 = xmprintf("%*c%s", 3*dotdots, ' ', s);
free(s);
s = s2;
while(dotdots--) {
// make symlink, or make block/char/fifo/socket
if (S_ISLNK(try->st.st_mode)
- ? (0 < (i = readlinkat(tfd, try->name, toybuf, sizeof(toybuf))) &&
- sizeof(toybuf) > i && !symlinkat(toybuf, cfd, catch))
+ ? ((i = readlinkat0(tfd, try->name, toybuf, sizeof(toybuf))) &&
+ !symlinkat(toybuf, cfd, catch))
: !mknodat(cfd, catch, try->st.st_mode, try->st.st_rdev))
{
err = 0;
if (fdin < 0) {
catch = try->name;
break;
- } else {
- fdout = openat(cfd, catch, O_RDWR|O_CREAT|O_TRUNC, try->st.st_mode);
- if (fdout >= 0) {
- xsendfile(fdin, fdout);
- err = 0;
+ }
+ fdout = openat(cfd, catch, O_RDWR|O_CREAT|O_TRUNC, try->st.st_mode);
+ if (fdout >= 0) {
+ xsendfile(fdin, fdout);
+ err = 0;
+ }
+
+ // We only copy xattrs for files because there's no flistxattrat()
+ if (TT.pflags&(_CP_xattr|_CP_context)) {
+ ssize_t listlen = flistxattr(fdin, 0, 0), len;
+ char *name, *value, *list;
+
+ if (listlen>0) {
+ list = xmalloc(listlen);
+ flistxattr(fdin, list, listlen);
+ list[listlen-1] = 0; // I do not trust this API.
+ for (name = list; name-list < listlen; name += strlen(name)+1) {
+ if (!(TT.pflags&_CP_xattr) && strncmp(name, "security.", 9))
+ continue;
+ if ((len = fgetxattr(fdin, name, 0, 0))>0) {
+ value = xmalloc(len);
+ if (len == fgetxattr(fdin, name, value, len))
+ if (fsetxattr(fdout, name, value, len, 0))
+ perror_msg("%s setxattr(%s=%s)", catch, name, value);
+ free(value);
+ }
+ }
+ free(list);
}
- close(fdin);
}
+
+ close(fdin);
}
} while (err && (flags & (FLAG_f|FLAG_n)) && !unlinkat(cfd, catch, 0));
}
+ // Did we make a thing?
if (fdout != -1) {
- if (flags & (FLAG_a|FLAG_p)) {
- struct timespec times[2];
+ int rc;
- // Inability to set these isn't fatal, some require root access.
+ // Inability to set --preserve isn't fatal, some require root access.
- times[0] = try->st.st_atim;
- times[1] = try->st.st_mtim;
+ // ownership
+ if (TT.pflags & _CP_ownership) {
+ // permission bits already correct for mknod and don't apply to symlink
// If we can't get a filehandle to the actual object, use racy functions
- if (fdout == AT_FDCWD) {
+ if (fdout == AT_FDCWD)
fchownat(cfd, catch, try->st.st_uid, try->st.st_gid,
- AT_SYMLINK_NOFOLLOW);
- utimensat(cfd, catch, times, AT_SYMLINK_NOFOLLOW);
- // permission bits already correct for mknod, don't apply to symlink
- } else {
- fchown(fdout, try->st.st_uid, try->st.st_gid);
- futimens(fdout, times);
- fchmod(fdout, try->st.st_mode);
+ AT_SYMLINK_NOFOLLOW);
+ else rc = fchown(fdout, try->st.st_uid, try->st.st_gid);
+ if (rc && !geteuid()) {
+ char *pp;
+
+ perror_msg("chown '%s'", pp = dirtree_path(try, 0));
+ free(pp);
}
}
- if (fdout != AT_FDCWD) xclose(fdout);
+ // timestamp
+ if (TT.pflags & _CP_timestamps) {
+ struct timespec times[] = {try->st.st_atim, try->st.st_mtim};
- if (toys.which->name[0] == 'm')
- if (unlinkat(tfd, try->name, S_ISDIR(try->st.st_mode) ? AT_REMOVEDIR : 0))
+ if (fdout == AT_FDCWD) utimensat(cfd, catch, times, AT_SYMLINK_NOFOLLOW);
+ else futimens(fdout, times);
+ }
+
+ // mode comes last because other syscalls can strip suid bit
+ if (fdout != AT_FDCWD) {
+ if (TT.pflags & _CP_mode) fchmod(fdout, try->st.st_mode);
+ xclose(fdout);
+ }
+
+ if (CFG_MV && toys.which->name[0] == 'm')
+ if (unlinkat(tfd, try->name, S_ISDIR(try->st.st_mode) ? AT_REMOVEDIR :0))
err = "%s";
}
- if (err) perror_msg(err, catch);
+ if (err) {
+ char *f = 0;
+
+ if (catch == try->name) {
+ f = dirtree_path(try, 0);
+ while (try->parent) try = try->parent;
+ catch = xmprintf("%s%s", TT.destname, f+strlen(try->name));
+ free(f);
+ f = catch;
+ }
+ perror_msg(err, catch);
+ free(f);
+ }
return 0;
}
int i, destdir = !stat(destname, &TT.top) && S_ISDIR(TT.top.st_mode);
if (toys.optc>1 && !destdir) error_exit("'%s' not directory", destname);
- if (toys.which->name[0] == 'm') toys.optflags |= FLAG_d|FLAG_p|FLAG_R;
- if (toys.optflags & (FLAG_a|FLAG_p)) umask(0);
- // Loop through sources
+ if (toys.optflags & (FLAG_a|FLAG_p)) {
+ TT.pflags = CP_mode|CP_ownership|CP_timestamps;
+ umask(0);
+ }
+ // Not using comma_args() (yet?) because interpeting as letters.
+ if (CFG_CP_PRESERVE && (toys.optflags & FLAG_preserve)) {
+ char *pre = xstrdup(TT.c.preserve), *s;
+
+ if (comma_scan(pre, "all", 1)) TT.pflags = ~0;
+ for (i=0; i<ARRAY_LEN(cp_preserve); i++)
+ if (comma_scan(pre, cp_preserve[i].name, 1)) TT.pflags |= 1<<i;
+ if (*pre) {
+
+ // Try to interpret as letters, commas won't set anything this doesn't.
+ for (s = TT.c.preserve; *s; s++) {
+ for (i=0; i<ARRAY_LEN(cp_preserve); i++)
+ if (*s == *cp_preserve[i].name) break;
+ if (i == ARRAY_LEN(cp_preserve)) {
+ if (*s == 'a') TT.pflags = ~0;
+ else break;
+ } else TT.pflags |= 1<<i;
+ }
+ if (*s) error_exit("bad --preserve=%s", pre);
+ }
+ free(pre);
+ }
+ if (!TT.callback) TT.callback = cp_node;
+
+ // Loop through sources
for (i=0; i<toys.optc; i++) {
- struct dirtree *new;
char *src = toys.optargs[i];
int rc = 1;
else TT.destname = destname;
errno = EXDEV;
- if (toys.which->name[0] == 'm') rc = rename(src, TT.destname);
+ if (CFG_MV && toys.which->name[0] == 'm') {
+ int force = toys.optflags & FLAG_f, no_clobber = toys.optflags & FLAG_n;
+
+ if (!force || no_clobber) {
+ struct stat st;
+ int exists = !stat(TT.destname, &st);
+
+ // Prompt if -i or file isn't writable. Technically "is writable" is
+ // more complicated (022 is not writeable by the owner, just everybody
+ // _else_) but I don't care.
+ if (exists && ((toys.optflags & FLAG_i) || !(st.st_mode & 0222))) {
+ fprintf(stderr, "%s: overwrite '%s'", toys.which->name, TT.destname);
+ if (!yesno(1)) rc = 0;
+ else unlink(TT.destname);
+ }
+ // if -n and dest exists, don't try to rename() or copy
+ if (exists && no_clobber) rc = 0;
+ }
+ if (rc) rc = rename(src, TT.destname);
+ }
- // Skip nonexistent sources
+ // Copy if we didn't mv, skipping nonexistent sources
if (rc) {
- if (errno != EXDEV ||
- !(new = dirtree_add_node(0, src, !(toys.optflags & (FLAG_d|FLAG_a)))))
+ if (errno!=EXDEV || dirtree_flagread(src, DIRTREE_SHUTUP+
+ DIRTREE_SYMFOLLOW*!!(toys.optflags&(FLAG_H|FLAG_L)), TT.callback))
perror_msg("bad '%s'", src);
- else dirtree_handle_callback(new, cp_node);
}
if (destdir) free(TT.destname);
}
}
+
+void mv_main(void)
+{
+ toys.optflags |= FLAG_d|FLAG_p|FLAG_R;
+
+ cp_main();
+}
+
+// Export cp flags into install's flag context.
+
+static inline int cp_flag_F(void) { return FLAG_F; };
+static inline int cp_flag_p(void) { return FLAG_p; };
+static inline int cp_flag_v(void) { return FLAG_v; };
+
+// Switch to install's flag context
+#define CLEANUP_cp
+#define FOR_install
+#include <generated/flags.h>
+
+static int install_node(struct dirtree *try)
+{
+ try->st.st_mode = (TT.i.mode)
+ ? string_to_mode(TT.i.mode, try->st.st_mode) : 0755;
+ if (TT.i.group) try->st.st_gid = TT.gid;
+ if (TT.i.user) try->st.st_uid = TT.uid;
+
+ // Always returns 0 because no -r
+ cp_node(try);
+
+ // No -r so always one level deep, so destname as set by cp_node() is correct
+ if (toys.optflags & FLAG_s)
+ if (xrun((char *[]){"strip", "-p", TT.destname, 0})) toys.exitval = 1;
+
+ return 0;
+}
+
+void install_main(void)
+{
+ char **ss;
+ int flags = toys.optflags;
+
+ if (flags & FLAG_d) {
+ for (ss = toys.optargs; *ss; ss++) {
+ if (mkpathat(AT_FDCWD, *ss, 0777, 3)) perror_msg_raw(*ss);
+ if (flags & FLAG_v) printf("%s\n", *ss);
+ }
+
+ return;
+ }
+
+ if (toys.optflags & FLAG_D) {
+ TT.destname = toys.optargs[toys.optc-1];
+ if (mkpathat(AT_FDCWD, TT.destname, 0, 2))
+ perror_exit("-D '%s'", TT.destname);
+ if (toys.optc == 1) return;
+ }
+ if (toys.optc < 2) error_exit("needs 2 args");
+
+ // Translate flags from install to cp
+ toys.optflags = cp_flag_F();
+ if (flags & FLAG_v) toys.optflags |= cp_flag_v();
+ if (flags & (FLAG_p|FLAG_o|FLAG_g)) toys.optflags |= cp_flag_p();
+
+ if (TT.i.user) TT.uid = xgetuid(TT.i.user);
+ if (TT.i.group) TT.gid = xgetgid(TT.i.group);
+
+ TT.callback = install_node;
+ cp_main();
+}
OSDN Git Service
