sbase/ln.c

104 lines
2.2 KiB
C

/* See LICENSE file for copyright and license details. */
#include <sys/stat.h>
#include <errno.h>
#include <fcntl.h>
#include <libgen.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
static void
usage(void)
{
eprintf("usage: %s [-f] [-L | -P | -s] target [name]\n"
" %s [-f] [-L | -P | -s] target ... dir\n", argv0, argv0);
}
int
main(int argc, char *argv[])
{
char *targetdir = ".", *target = NULL;
int ret = 0, sflag = 0, fflag = 0, dirfd = AT_FDCWD,
hastarget = 0, flags = AT_SYMLINK_FOLLOW;
struct stat st, tst;
ARGBEGIN {
case 'f':
fflag = 1;
break;
case 'L':
flags |= AT_SYMLINK_FOLLOW;
break;
case 'P':
flags &= ~AT_SYMLINK_FOLLOW;
break;
case 's':
sflag = 1;
break;
default:
usage();
} ARGEND
if (!argc)
usage();
if (argc > 1) {
if (!stat(argv[argc - 1], &st) && S_ISDIR(st.st_mode)) {
if ((dirfd = open(argv[argc - 1], O_RDONLY)) < 0)
eprintf("open %s:", argv[argc - 1]);
targetdir = argv[argc - 1];
if (targetdir[strlen(targetdir) - 1] == '/')
targetdir[strlen(targetdir) - 1] = '\0';
} else if (argc == 2) {
hastarget = 1;
target = argv[argc - 1];
} else {
eprintf("%s: not a directory\n", argv[argc - 1]);
}
argv[argc - 1] = NULL;
argc--;
}
for (; *argv; argc--, argv++) {
if (!hastarget)
target = basename(*argv);
if (!sflag) {
if (stat(*argv, &st) < 0) {
weprintf("stat %s:", *argv);
ret = 1;
continue;
} else if (fstatat(dirfd, target, &tst, AT_SYMLINK_NOFOLLOW) < 0) {
if (errno != ENOENT) {
weprintf("fstatat %s %s:", targetdir, target);
ret = 1;
continue;
}
} else if (st.st_dev == tst.st_dev && st.st_ino == tst.st_ino) {
if (!fflag) {
weprintf("%s and %s/%s are the same file\n",
*argv, targetdir, target);
ret = 1;
}
continue;
}
}
if (fflag && unlinkat(dirfd, target, 0) < 0 && errno != ENOENT) {
weprintf("unlinkat %s %s:", targetdir, target);
ret = 1;
continue;
}
if ((sflag ? symlinkat(*argv, dirfd, target) :
linkat(AT_FDCWD, *argv, dirfd, target, flags)) < 0) {
weprintf("%s %s <- %s/%s:", sflag ? "symlinkat" : "linkat",
*argv, targetdir, target);
ret = 1;
}
}
return ret;
}