|
Message-ID: <YewY55arOjKae3xZ@pirotess> Date: Sat, 22 Jan 2022 15:47:03 +0100 From: Ismael Luceno <ismael@...ev.co.uk> To: musl@...ts.openwall.com Cc: Rich Felker <dalias@...c.org> Subject: Re: [PATCH 2/2] nftw: implement FTW_ACTIONRETVAL (GNU extension) I noticed a small bug, I'll submit a v2. On 22/Jan/2022 14:26, Ismael Luceno wrote: > Signed-off-by: Ismael Luceno <ismael@...ev.co.uk> > --- > include/ftw.h | 7 +++++++ > src/misc/nftw.c | 32 ++++++++++++++++++++++---------- > 2 files changed, 29 insertions(+), 10 deletions(-) > > diff --git a/include/ftw.h b/include/ftw.h > index b15c062a8389..2e77dc76e11b 100644 > --- a/include/ftw.h > +++ b/include/ftw.h > @@ -21,6 +21,13 @@ extern "C" { > #define FTW_CHDIR 4 > #define FTW_DEPTH 8 > > +#define FTW_ACTIONRETVAL 16 > +/* return values for callback */ > +#define FTW_CONTINUE 0 > +#define FTW_STOP 1 > +#define FTW_SKIP_SUBTREE 2 > +#define FTW_SKIP_SIBLINGS 3 > + > struct FTW { > int base; > int level; > diff --git a/src/misc/nftw.c b/src/misc/nftw.c > index 0a35686e663d..4b934adcfb0c 100644 > --- a/src/misc/nftw.c > +++ b/src/misc/nftw.c > @@ -26,7 +26,7 @@ static int do_nftw(char *path, int (*fn)(const char *, const struct stat *, int, > struct stat st; > struct history new; > int type; > - int r; > + int r, r2; > int dfd; > int err; > struct FTW lev; > @@ -72,12 +72,18 @@ static int do_nftw(char *path, int (*fn)(const char *, const struct stat *, int, > if (!fd_limit) close(dfd); > } > > - if (!(flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) > - return r; > + if (!(flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) { > + if ((flags & FTW_ACTIONRETVAL)) { > + if (r == FTW_SKIP_SUBTREE) > + return 0; > + /* FTW_SKIP_SIBLINGS is saved for returning */ > + } else > + return r; > + } > > for (; h; h = h->chain) > if (h->dev == st.st_dev && h->ino == st.st_ino) > - return 0; > + return r; > > if ((type == FTW_D || type == FTW_DP) && fd_limit) { > if (dfd < 0) { > @@ -101,9 +107,12 @@ static int do_nftw(char *path, int (*fn)(const char *, const struct stat *, int, > } > path[j]='/'; > strcpy(path+j+1, de->d_name); > - if ((r=do_nftw(path, fn, fd_limit-1, flags, &new))) { > + if ((r2=do_nftw(path, fn, fd_limit-1, flags, &new))) { > + if ((flags & FTW_ACTIONRETVAL) > + && r2 == FTW_SKIP_SIBLINGS) > + break; > closedir(d); > - return r; > + return r2; > } > } > closedir(d); > @@ -114,10 +123,13 @@ static int do_nftw(char *path, int (*fn)(const char *, const struct stat *, int, > } > > path[l] = 0; > - if ((flags & FTW_DEPTH) && (r=fn(path, &st, type, &lev))) > - return r; > - > - return 0; > + if (flags & FTW_DEPTH) { > + r = fn(path, &st, type, &lev); > + /* ignore FTW_SKIP_SUBTREE, the caller is broken */ > + if ((flags & FTW_ACTIONRETVAL) && r == FTW_SKIP_SUBTREE) > + r = 0; > + } > + return r; > } > > int nftw(const char *path, int (*fn)(const char *, const struct stat *, int, struct FTW *), int fd_limit, int flags) > -- > 2.33.0 > -- Ismael Luceno http://iodev.co.uk/
Powered by blists - more mailing lists
Confused about mailing lists and their use? Read about mailing lists on Wikipedia and check out these guidelines on proper formatting of your messages.