| /* |
| * Copyright (c) 2003, 2004 Todd C. Miller <Todd.Miller@courtesan.com> |
| * |
| * Permission to use, copy, modify, and distribute this software for any |
| * purpose with or without fee is hereby granted, provided that the above |
| * copyright notice and this permission notice appear in all copies. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
| * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
| * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
| * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
| * |
| * Sponsored in part by the Defense Advanced Research Projects |
| * Agency (DARPA) and Air Force Research Laboratory, Air Force |
| * Materiel Command, USAF, under agreement number F39502-99-1-0512. |
| */ |
| |
| #include<errno.h> |
| #include<fts.h> |
| #include<ftw.h> |
| #include<sys/stat.h> |
| #include<sys/types.h> |
| #include<unistd.h> |
| |
| extern"C" FTS* __fts_open(char*const*,int,int(*)(const FTSENT**,const FTSENT**)); |
| |
| staticint do_nftw(constchar* path, |
| int(*ftw_fn)(constchar*,conststruct stat*,int), |
| int(*nftw_fn)(constchar*,conststruct stat*,int, FTW*), |
| int nfds, |
| int nftw_flags){ |
| // TODO: nfds is currently unused. |
| if(nfds<1){ |
| errno= EINVAL; |
| return-1; |
| } |
| |
| // Translate to fts_open options. |
| int fts_options= FTS_LOGICAL| FTS_COMFOLLOW| FTS_NOCHDIR; |
| if(nftw_fn){ |
| fts_options= FTS_COMFOLLOW|((nftw_flags& FTW_PHYS)? FTS_PHYSICAL: FTS_LOGICAL); |
| if(!(nftw_flags& FTW_CHDIR)) fts_options|= FTS_NOCHDIR; |
| if(nftw_flags& FTW_MOUNT) fts_options|= FTS_XDEV; |
| } |
| bool postorder=(nftw_flags& FTW_DEPTH)!=0; |
| |
| // Call fts_open. |
| char*const paths[2]={const_cast<char*>(path),nullptr}; |
| FTS* fts= __fts_open(paths, fts_options| FTS_FOR_FTW,nullptr); |
| if(fts==nullptr){ |
| return-1; |
| } |
| |
| // Translate fts_read results into ftw/nftw callbacks. |
| int error=0; |
| FTSENT* cur; |
| while(error==0&&(cur= fts_read(fts))!=nullptr){ |
| int fn_flag; |
| switch(cur->fts_info){ |
| case FTS_D: |
| // In the postorder case, we'll translate FTS_DP to FTW_DP later. |
| // In the can't-access case, we'll translate FTS_DNR to FTW_DNR later. |
| if(postorder|| access(cur->fts_path, R_OK)==-1)continue; |
| fn_flag= FTW_D; |
| break; |
| case FTS_DC: |
| // POSIX says nftw "shall not report" directories causing loops (http://b/31152735). |
| continue; |
| case FTS_DNR: |
| fn_flag= FTW_DNR; |
| break; |
| case FTS_DP: |
| if(!postorder)continue; |
| fn_flag= FTW_DP; |
| break; |
| case FTS_F: |
| case FTS_DEFAULT: |
| fn_flag= FTW_F; |
| break; |
| case FTS_NS: |
| case FTS_NSOK: |
| fn_flag= FTW_NS; |
| break; |
| case FTS_SL: |
| fn_flag= FTW_SL; |
| break; |
| case FTS_SLNONE: |
| fn_flag=(nftw_fn!=nullptr)? FTW_SLN: FTW_NS; |
| break; |
| default: |
| error=-1; |
| continue; |
| } |
| |
| // Call the appropriate function. |
| if(nftw_fn!=nullptr){ |
| FTW ftw; |
| ftw.base= cur->fts_pathlen- cur->fts_namelen; |
| ftw.level= cur->fts_level; |
| error= nftw_fn(cur->fts_path, cur->fts_statp, fn_flag,&ftw); |
| }else{ |
| error= ftw_fn(cur->fts_path, cur->fts_statp, fn_flag); |
| } |
| } |
| |
| int saved_errno= errno; |
| if(fts_close(fts)!=0&& error==0){ |
| error=-1; |
| }else{ |
| errno= saved_errno; |
| } |
| return error; |
| } |
| |
| int ftw(constchar* path,int(*ftw_fn)(constchar*,conststruct stat*,int),int nfds){ |
| return do_nftw(path, ftw_fn,nullptr, nfds,0); |
| } |
| |
| int nftw(constchar* path,int(*nftw_fn)(constchar*,conststruct stat*,int, FTW*), |
| int nfds,int nftw_flags){ |
| return do_nftw(path,nullptr, nftw_fn, nfds, nftw_flags); |
| } |
