From 8917a48c31540dd7a99f95dd317d6cf1ac2f6803 Mon Sep 17 00:00:00 2001 From: Tavian Barnes Date: Tue, 8 Nov 2022 11:53:19 -0500 Subject: tests/xtouch: Add some missing POSIX touch features --- tests/tests.sh | 10 ++-- tests/xtouch.c | 145 +++++++++++++++++++++++++++++++++++++++------------------ 2 files changed, 104 insertions(+), 51 deletions(-) (limited to 'tests') diff --git a/tests/tests.sh b/tests/tests.sh index 6a32f62..8d13aca 100755 --- a/tests/tests.sh +++ b/tests/tests.sh @@ -362,12 +362,12 @@ make_loops "$TMP/loops" # Creates a file+directory structure with varying timestamps function make_times() { - "$XTOUCH" -p -t 199112140000 "$1/a" - "$XTOUCH" -p -t 199112140001 "$1/b" - "$XTOUCH" -p -t 199112140002 "$1/c" + "$XTOUCH" -p -t "1991-12-14 00:00" "$1/a" + "$XTOUCH" -p -t "1991-12-14 00:01" "$1/b" + "$XTOUCH" -p -t "1991-12-14 00:02" "$1/c" ln -s a "$1/l" - "$XTOUCH" -p -h -t 199112140003 "$1/l" - "$XTOUCH" -p -t 199112140004 "$1" + "$XTOUCH" -p -h -t "1991-12-14 00:03" "$1/l" + "$XTOUCH" -p -t "1991-12-14 00:04" "$1" } make_times "$TMP/times" diff --git a/tests/xtouch.c b/tests/xtouch.c index 7a248eb..9a91ec7 100644 --- a/tests/xtouch.c +++ b/tests/xtouch.c @@ -14,6 +14,7 @@ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. * ****************************************************************************/ +#include "../src/bfstd.h" #include "../src/xtime.h" #include #include @@ -25,10 +26,36 @@ #include #include -/** Check if a path has a trailing slash. */ -static bool trailing_slash(const char *path) { - size_t len = strlen(path); - return len > 0 && path[len - 1] == '/'; +/** Parsed xtouch arguments. */ +struct args { + /** Simple flags. */ + enum { + /** Don't create nonexistent files (-c). */ + NO_CREATE = 1 << 0, + /** Don't follow symlinks (-h). */ + NO_FOLLOW = 1 << 1, + /** Create any missing parent directories (-p). */ + CREATE_PARENTS = 1 << 2, + } flags; + + /** Timestamps (-r|-t|-d). */ + struct timespec times[2]; + + /** File creation mode (-M; default 0666 & ~umask). */ + mode_t fmode; + /** Directory creation mode (-M; default 0777 & ~umask). */ + mode_t dmode; + /** Parent directory creation mode (0777 & ~umask). */ + mode_t pmode; +}; + +/** Compute flags for fstatat()/utimensat(). */ +static int at_flags(const struct args *args) { + if (args->flags & NO_FOLLOW) { + return AT_SYMLINK_NOFOLLOW; + } else { + return 0; + } } /** Create any parent directories of the given path. */ @@ -62,29 +89,34 @@ static int mkdirs(const char *path, mode_t mode) { } /** Touch one path. */ -static int xtouch(const char *path, const struct timespec times[2], int flags, mode_t mode, bool parents, mode_t pmode) { - int ret = utimensat(AT_FDCWD, path, times, flags); +static int xtouch(const struct args *args, const char *path) { + int ret = utimensat(AT_FDCWD, path, args->times, at_flags(args)); if (ret == 0 || errno != ENOENT) { return ret; } - if (parents && mkdirs(path, pmode) != 0) { - return -1; + if (args->flags & NO_CREATE) { + return 0; + } else if (args->flags & CREATE_PARENTS) { + if (mkdirs(path, args->pmode) != 0) { + return -1; + } } - if (trailing_slash(path)) { - if (mkdir(path, mode) != 0) { + size_t len = strlen(path); + if (len > 0 && path[len - 1] == '/') { + if (mkdir(path, args->dmode) != 0) { return -1; } - return utimensat(AT_FDCWD, path, times, flags); + return utimensat(AT_FDCWD, path, args->times, at_flags(args)); } else { - int fd = open(path, O_WRONLY | O_CREAT, mode); + int fd = open(path, O_WRONLY | O_CREAT, args->fmode); if (fd < 0) { return -1; } - if (futimens(fd, times) != 0) { + if (futimens(fd, args->times) != 0) { int error = errno; close(fd); errno = error; @@ -96,33 +128,52 @@ static int xtouch(const char *path, const struct timespec times[2], int flags, m } int main(int argc, char *argv[]) { - int flags = 0; + mode_t mask = umask(0); + + struct args args = { + .flags = 0, + .times = { + { .tv_nsec = UTIME_OMIT }, + { .tv_nsec = UTIME_OMIT }, + }, + .fmode = 0666 & ~mask, + .dmode = 0777 & ~mask, + .pmode = 0777 & ~mask, + }; + bool atime = false, mtime = false; - bool parents = false; - const char *mstr = NULL; - const char *tstr = NULL; + const char *darg = NULL; + const char *marg = NULL; + const char *rarg = NULL; const char *cmd = argc > 0 ? argv[0] : "xtouch"; int c; - while (c = getopt(argc, argv, ":M:ahmpt:"), c != -1) { + while (c = getopt(argc, argv, ":M:acd:hmpr:t:"), c != -1) { switch (c) { case 'M': - mstr = optarg; + marg = optarg; break; case 'a': atime = true; break; + case 'c': + args.flags |= NO_CREATE; + break; + case 'd': + case 't': + darg = optarg; + break; case 'h': - flags = AT_SYMLINK_NOFOLLOW; + args.flags |= NO_FOLLOW; break; case 'm': mtime = true; break; case 'p': - parents = true; + args.flags |= CREATE_PARENTS; break; - case 't': - tstr = optarg; + case 'r': + rarg = optarg; break; case ':': fprintf(stderr, "%s: Missing argument to -%c\n", cmd, optopt); @@ -133,48 +184,51 @@ int main(int argc, char *argv[]) { } } - mode_t mask = umask(0); - mode_t fmode = 0666 & ~mask; - mode_t dmode = 0777 & ~mask; - mode_t pmode = 0777 & ~mask; - if (mstr) { + if (marg) { char *end; - long mode = strtol(mstr, &end, 8); - if (*mstr && !*end && mode >= 0 && mode < 01000) { - fmode = dmode = mode; + long mode = strtol(marg, &end, 8); + if (*marg && !*end && mode >= 0 && mode < 01000) { + args.fmode = args.dmode = mode; } else { - fprintf(stderr, "%s: Invalid mode '%s'\n", cmd, mstr); + fprintf(stderr, "%s: Invalid mode '%s'\n", cmd, marg); return EXIT_FAILURE; } } - struct timespec ts; - if (tstr) { - if (xgetdate(tstr, &ts) != 0) { - fprintf(stderr, "%s: Parsing time '%s' failed: %s\n", cmd, tstr, strerror(errno)); + struct timespec times[2]; + + if (rarg) { + struct stat buf; + if (fstatat(AT_FDCWD, rarg, &buf, at_flags(&args)) != 0) { + fprintf(stderr, "%s: '%s': %s\n", cmd, rarg, strerror(errno)); return EXIT_FAILURE; } + times[0] = buf.st_atim; + times[1] = buf.st_mtim; + } else if (darg) { + if (xgetdate(darg, ×[0]) != 0) { + fprintf(stderr, "%s: Parsing time '%s' failed: %s\n", cmd, darg, strerror(errno)); + return EXIT_FAILURE; + } + times[1] = times[0]; } else { // Don't use UTIME_NOW, so that multiple paths all get the same timestamp - if (xgettime(&ts) != 0) { + if (xgettime(×[0]) != 0) { perror("xgettime()"); return EXIT_FAILURE; } + times[1] = times[0]; } - struct timespec times[2] = { - { .tv_nsec = UTIME_OMIT }, - { .tv_nsec = UTIME_OMIT }, - }; if (!atime && !mtime) { atime = true; mtime = true; } if (atime) { - times[0] = ts; + args.times[0] = times[0]; } if (mtime) { - times[1] = ts; + args.times[1] = times[1]; } if (optind >= argc) { @@ -185,9 +239,8 @@ int main(int argc, char *argv[]) { int ret = EXIT_SUCCESS; for (; optind < argc; ++optind) { const char *path = argv[optind]; - bool isdir = trailing_slash(path); - if (xtouch(path, times, flags, isdir ? dmode : fmode, parents, pmode) != 0) { - fprintf(stderr, "%s: Touching '%s' failed: %s\n", cmd, path, strerror(errno)); + if (xtouch(&args, path) != 0) { + fprintf(stderr, "%s: '%s': %s\n", cmd, path, strerror(errno)); ret = EXIT_FAILURE; } } -- cgit v1.2.3