fbt/sbase
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Use *at functions with appropriate flags instead of lstat/lchown

Browse Source
This commit is contained in:
Michael Forney
2019-11-01 20:43:32 -07:00
parent 3e160b616a
commit 039b54aa51
4 changed files with 29 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
libutil/cp.c
View File

@@ -26,24 +26,18 @@ int
cp(const char *s1, const char *s2, int depth)
{
DIR *dp;
int f1, f2;
int f1, f2, flags = 0;
struct dirent *d;
struct stat st;
struct timespec times[2];
ssize_t r;
int (*statf)(const char *, struct stat *);
char target[PATH_MAX], ns1[PATH_MAX], ns2[PATH_MAX], *statf_name;
char target[PATH_MAX], ns1[PATH_MAX], ns2[PATH_MAX];
if (cp_follow == 'P' || (cp_follow == 'H' && depth)) {
statf_name = "lstat";
statf = lstat;
} else {
statf_name = "stat";
statf = stat;
}
if (cp_follow == 'P' || (cp_follow == 'H' && depth))
flags |= AT_SYMLINK_NOFOLLOW;
if (statf(s1, &st) < 0) {
weprintf("%s %s:", statf_name, s1);
if (fstatat(AT_FDCWD, s1, &st, flags) < 0) {
weprintf("stat %s:", s1);
cp_status = 1;
return 0;
}

28
libutil/recurse.c
View File

@@ -1,6 +1,7 @@
/* 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>
@@ -21,20 +22,15 @@ recurse(const char *path, void *data, struct recursor *r)
struct history *new, *h;
struct stat st, dst;
DIR *dp;
int (*statf)(const char *, struct stat *);
char subpath[PATH_MAX], *statf_name;
char subpath[PATH_MAX];
int flags = 0;
if (r->follow == 'P' || (r->follow == 'H' && r->depth)) {
statf_name = "lstat";
statf = lstat;
} else {
statf_name = "stat";
statf = stat;
}
if (r->follow == 'P' || (r->follow == 'H' && r->depth))
flags |= AT_SYMLINK_NOFOLLOW;
if (statf(path, &st) < 0) {
if (fstatat(AT_FDCWD, path, &st, flags) < 0) {
if (!(r->flags & SILENT)) {
weprintf("%s %s:", statf_name, path);
weprintf("stat %s:", path);
recurse_status = 1;
}
return;
@@ -66,19 +62,17 @@ recurse(const char *path, void *data, struct recursor *r)
return;
}
while ((d = readdir(dp))) {
if (r->follow == 'H') {
statf_name = "lstat";
statf = lstat;
}
if (r->follow == 'H')
flags |= AT_SYMLINK_NOFOLLOW;
if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
continue;
estrlcpy(subpath, path, sizeof(subpath));
if (path[strlen(path) - 1] != '/')
estrlcat(subpath, "/", sizeof(subpath));
estrlcat(subpath, d->d_name, sizeof(subpath));
if (statf(subpath, &dst) < 0) {
if (fstatat(AT_FDCWD, subpath, &dst, flags) < 0) {
if (!(r->flags & SILENT)) {
weprintf("%s %s:", statf_name, subpath);
weprintf("stat %s:", subpath);
recurse_status = 1;
}
} else if ((r->flags & SAMEDEV) && dst.st_dev != st.st_dev) {