fscrawl.c: use FTS to handle symlink-loops.
This commit is contained in:
Henrik Hautakoski
parent
72bdd8bb03
commit
1c6f66d58d
1 changed files with 79 additions and 116 deletions
195
src/fscrawl.c
195
src/fscrawl.c
|
|
@ -12,121 +12,104 @@
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
#include <dirent.h>
|
#include <fts.h>
|
||||||
|
|
||||||
#include "log.h"
|
#include "log.h"
|
||||||
#include "strbuf.h"
|
|
||||||
#include "path.h"
|
#include "path.h"
|
||||||
#include "fscrawl.h"
|
#include "fscrawl.h"
|
||||||
|
|
||||||
#define MAX_DEPTH 0x20
|
#define FTSENT_VALID_TYPE(x) \
|
||||||
|
((x)->fts_info == FTS_D || \
|
||||||
#define isrefdir(x) (!strcmp((x)->d_name, ".") || !strcmp((x)->d_name, ".."))
|
(x)->fts_info == FTS_DC || \
|
||||||
|
(x)->fts_info == FTS_F || \
|
||||||
|
(x)->fts_info == FTS_SL || \
|
||||||
|
(x)->fts_info == FTS_SLNONE)
|
||||||
|
|
||||||
struct __fscrawl {
|
struct __fscrawl {
|
||||||
strbuf_t path;
|
FTS *fts;
|
||||||
DIR *dirs[MAX_DEPTH];
|
|
||||||
unsigned char depth;
|
|
||||||
fs_entry ent;
|
fs_entry ent;
|
||||||
};
|
};
|
||||||
|
|
||||||
static int mvup(struct __fscrawl *f) {
|
static FTS* _fts_open(char *path, int opts) {
|
||||||
|
|
||||||
if (closedir(f->dirs[f->depth]) == -1) {
|
|
||||||
logerrno(LOG_WARN, "ftc: closedir", errno);
|
|
||||||
errno = 0;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (f->depth == 0)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
f->depth--;
|
char *npath[2] = { path, NULL };
|
||||||
|
|
||||||
if (f->path.buf[f->path.len-1] == '/')
|
return fts_open(npath, opts, NULL);
|
||||||
strbuf_reduce(&f->path, 1);
|
|
||||||
strbuf_rchop(&f->path, '/');
|
|
||||||
strbuf_term(&f->path, '/');
|
|
||||||
|
|
||||||
return 1;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static int mvdown(struct __fscrawl *f, const char *dir) {
|
static fs_entry* ftsentcpy(fs_entry *dest, FTSENT *src) {
|
||||||
|
|
||||||
DIR *ds = NULL;
|
int len = src->fts_pathlen - src->fts_namelen;
|
||||||
|
|
||||||
if (f->depth >= MAX_DEPTH-1)
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
strbuf_term(&f->path, '/');
|
if (len > 0) {
|
||||||
strbuf_append_str(&f->path, dir);
|
char *newbase = realloc(dest->base, len + 1);
|
||||||
strbuf_term(&f->path, '/');
|
|
||||||
|
|
||||||
ds = opendir(f->path.buf);
|
|
||||||
|
|
||||||
if (!ds) {
|
|
||||||
if (errno != EACCES)
|
|
||||||
logerrno(LOG_WARN, "ftc: opendir", errno);
|
|
||||||
errno = 0;
|
|
||||||
|
|
||||||
strbuf_reduce(&f->path, 1);
|
if (newbase) {
|
||||||
strbuf_rchop(&f->path, '/');
|
memcpy(newbase, src->fts_path, len);
|
||||||
strbuf_term(&f->path, '/');
|
newbase[len] = '\0';
|
||||||
return 0;
|
|
||||||
}
|
dest->base = newbase;
|
||||||
|
dest->name = src->fts_name;
|
||||||
f->dirs[++f->depth] = ds;
|
dest->dir = src->fts_info == FTS_D;
|
||||||
|
return dest;
|
||||||
return 1;
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int update_base(fs_entry *ent, const char *base, unsigned len) {
|
||||||
|
|
||||||
|
char *p = realloc(ent->base, len + 1);
|
||||||
|
if (p) {
|
||||||
|
memcpy(p, base, len);
|
||||||
|
p[len] = '\0';
|
||||||
|
ent->base = p;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
fscrawl_t fsc_open(const char *path) {
|
fscrawl_t fsc_open(const char *path) {
|
||||||
|
|
||||||
struct __fscrawl *f = malloc(sizeof(struct __fscrawl));
|
struct __fscrawl *f;
|
||||||
|
char *npath;
|
||||||
if (f == NULL)
|
|
||||||
|
f = malloc(sizeof(struct __fscrawl));
|
||||||
|
if (!f)
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
strbuf_init(&f->path);
|
npath = path_normalize(path, NULL, 1);
|
||||||
|
|
||||||
char *npath = path_normalize(path, NULL, 1);
|
|
||||||
|
|
||||||
if (!npath)
|
if (!npath)
|
||||||
return NULL;
|
goto free;
|
||||||
|
|
||||||
/* set the normalized path as the string buffer */
|
f->fts = _fts_open(npath, FTS_LOGICAL);
|
||||||
strbuf_attach(&f->path, npath, strlen(npath), strlen(npath));
|
free(npath);
|
||||||
|
|
||||||
f->depth = 0;
|
if (!f->fts) {
|
||||||
|
logerrno(LOG_WARN, "ftc_open", errno);
|
||||||
|
goto free;
|
||||||
|
}
|
||||||
|
|
||||||
f->dirs[f->depth] = opendir(f->path.buf);
|
f->ent.base = NULL;
|
||||||
|
f->ent.name = NULL;
|
||||||
if (!f->dirs[f->depth]) {
|
f->ent.dir = 0;
|
||||||
|
|
||||||
logerrno(LOG_WARN, "fsc_open", errno);
|
|
||||||
errno = 0;
|
|
||||||
|
|
||||||
strbuf_free(&f->path);
|
|
||||||
free(f);
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
f->ent.name = NULL;
|
|
||||||
f->ent.base = NULL;
|
|
||||||
f->ent.dir = 0;
|
|
||||||
|
|
||||||
return f;
|
return f;
|
||||||
|
free:
|
||||||
|
if (f)
|
||||||
|
free(f);
|
||||||
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
void fsc_close(fscrawl_t f) {
|
void fsc_close(fscrawl_t f) {
|
||||||
|
|
||||||
int i;
|
if (f) {
|
||||||
|
if (f->ent.base)
|
||||||
for(i=0; i < f->depth; i++)
|
free(f->ent.base);
|
||||||
closedir(f->dirs[i]);
|
if (fts_close(f->fts) < 0)
|
||||||
|
logerrno(LOG_WARN, "ftc_close", errno);
|
||||||
strbuf_free(&f->path);
|
free(f);
|
||||||
free(f);
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fs_entry* fsc_cpy(fs_entry *ent) {
|
fs_entry* fsc_cpy(fs_entry *ent) {
|
||||||
|
|
@ -143,48 +126,28 @@ fs_entry* fsc_cpy(fs_entry *ent) {
|
||||||
|
|
||||||
fs_entry* fsc_read(fscrawl_t f) {
|
fs_entry* fsc_read(fscrawl_t f) {
|
||||||
|
|
||||||
struct dirent *ent = NULL;
|
FTSENT *ent;
|
||||||
|
|
||||||
if (!f)
|
if (!f)
|
||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
if (f->ent.dir)
|
for(;;) {
|
||||||
mvdown(f, f->ent.name);
|
ent = fts_read(f->fts);
|
||||||
|
|
||||||
errno = 0;
|
if (ent == NULL) {
|
||||||
for(;;) {
|
if (errno) {
|
||||||
ent = readdir(f->dirs[f->depth]);
|
logerrno(LOG_DEBUG, "fsc_read", errno);
|
||||||
|
|
||||||
if (ent) {
|
|
||||||
if (isrefdir(ent))
|
|
||||||
continue;
|
continue;
|
||||||
|
}
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (errno) {
|
if (ent->fts_namelen > 0 && FTSENT_VALID_TYPE(ent))
|
||||||
if (errno != EACCES)
|
return ftsentcpy(&f->ent, ent);
|
||||||
logerrno(LOG_WARN, "fsc_read", errno);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (mvup(f))
|
|
||||||
continue;
|
|
||||||
|
|
||||||
fsc_close(f);
|
if (ent->fts_info == FTS_ERR)
|
||||||
return NULL;
|
logerrno(LOG_DEBUG, "fsc_read", ent->fts_errno);
|
||||||
}
|
|
||||||
|
|
||||||
f->ent.base = f->path.buf;
|
|
||||||
f->ent.name = &ent->d_name[0];
|
|
||||||
|
|
||||||
if (ent->d_type == DT_UNKNOWN) {
|
|
||||||
size_t tmp = strlen(f->ent.name);
|
|
||||||
strbuf_append(&f->path, f->ent.name, tmp);
|
|
||||||
f->ent.dir = is_dir(f->path.buf);
|
|
||||||
strbuf_reduce(&f->path, tmp);
|
|
||||||
} else {
|
|
||||||
f->ent.dir = ent->d_type == DT_DIR;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return &f->ent;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Reference in a new issue