修改了《Unix环境高级编程第二版》程序清单4-7,使用chdir变换当前工作目录,使用文件名字而不是文件路径进行处理。通过测试,运行效率提高了。测试环境为CentOS 6.5
#include "apue.h"
#include "stdio.h"
#include "fcntl.h"
#include "unistd.h"
#include "errno.h"
#include "sys/stat.h"
#include "dirent.h"
#include "limits.h"
***#include <sys/time.h>***
extern int errno;
typedef int Myfunc(const char *,const struct stat *,int);
static Myfunc myfunc;
static int myftw(char *,Myfunc *);
static int dopath(Myfunc *);
static long nreg,ndir,nblk,nchr,nfifo,nslink,nsock,ntot;
int main(int argc,char *argv[])
{
int ret;
***//添加
long int lTime;//默认微秒数
struct timeval start,end;
gettimeofday(&start,NULL);
//***
if(argc != 2) err_quit("usage: ftw <starting-pathname>");
ret=myftw(argv[1],myfunc);
ntot=nreg+ndir+nblk+nchr+nfifo+nslink+nsock;
if(ntot==0) ntot=1;
printf("regular files = %7ld, %5.2f %%\n",nreg,nreg*100.0/ntot);
printf("diretories files = %7ld, %5.2f %%\n",ndir,ndir*100.0/ntot);
printf("block files = %7ld, %5.2f %%\n",nblk,nblk*100.0/ntot);
printf("char files = %7ld, %5.2f %%\n",nchr,nchr*100.0/ntot);
printf("FIFOs = %7ld, %5.2f %%\n",nfifo,nfifo*100.0/ntot);
printf("symbolic links = %7ld, %5.2f %%\n",nslink,nslink*100.0/ntot);
printf("sockets = %7ld, %5.2f %%\n",nsock,nsock*100.0/ntot);
***//添加
gettimeofday(&end,NULL);
lTime = 1000000*( end.tv_sec-start.tv_sec) + end.tv_usec - start.tv_usec; //默认微秒为单位
printf("time of running: %d\n",lTime);
//***
exit(ret);
}
#define FTW_F 1
#define FTW_D 2
#define FTW_DNR 3
#define FTW_NS 4
static char *fullpath;
static int myftw(char *pathname, Myfunc *func)
{
int len;
fullpath=path_alloc(&len);
strncpy(fullpath,pathname,len);
fullpath[len-1]=0;
return(dopath(func));
}
static int dopath(Myfunc* func)
{
struct stat statbuf;
struct dirent *dirp;
DIR *dp;
int ret;
***//char *ptr;***
if(lstat(fullpath,&statbuf)<0) return(func(fullpath,&statbuf,FTW_NS));
if(S_ISDIR(statbuf.st_mode)==0)
{
return(func(fullpath,&statbuf,FTW_F));
}
if((ret=func(fullpath,&statbuf,FTW_D))!=0) return(ret);
***//ptr=fullpath+strlen(fullpath);
//*ptr++='/';
//*ptr=0;***
if((dp=opendir(fullpath))==NULL) return(func(fullpath,&statbuf,FTW_DNR));
***//添加如下
if(chdir(fullpath)<0) err_sys("chdir error");***
while((dirp=readdir(dp))!=NULL)
{
if(strcmp(dirp->d_name,".")==0 || strcmp(dirp->d_name,"..")==0) continue;
strcpy(fullpath,dirp->d_name);
if((ret=dopath(func))!=0) break;
}
***//ptr[-1]=0;***
***//添加
if(chdir("..")<0) err_sys("chdir error");***
if(closedir(dp)<0) err_ret("can`t close directory %s",fullpath);
return(ret);
}
static int myfunc(const char *pathname,const struct stat * statptr,int type)
{
switch(type)
{
case FTW_F:
switch (statptr->st_mode & S_IFMT)
{
case S_IFREG: nreg++;break;
case S_IFBLK: nblk++;break;
case S_IFCHR: nchr++;break;
case S_IFIFO: nfifo++;break;
case S_IFLNK: nslink++;break;
case S_IFSOCK: nsock++;break;
case S_IFDIR: err_dump("for S_IFDIR for %s",pathname);
}
break;
case FTW_D: ndir++;break;
case FTW_DNR:err_ret("can't read directory %s",pathname);break;
case FTW_NS: err_ret("stat error for %s",pathname);break;
default : err_dump("unknow type %d for pathname %s",type,pathname);
}
return(0);
}
测试结果如下:
[root@SimonXie C]# ./dir / #运行未修改过的程序
regular files = 222903, 81.78 %
diretories files = 25313, 9.29 %
block files = 31, 0.01 %
char files = 160, 0.06 %
FIFOs = 4, 0.00 %
symbolic links = 24091, 8.84 %
sockets = 67, 0.02 %
time of running: 1770686
[root@SimonXie C]# ./dir_v2 / #运行修改过的程序
regular files = 222903, 81.78 %
diretories files = 25313, 9.29 %
block files = 31, 0.01 %
char files = 160, 0.06 %
FIFOs = 4, 0.00 %
symbolic links = 24091, 8.84 %
sockets = 67, 0.02 %
time of running: 1414655