修改《Unix环境高级编程第二版》程序清单4-7

修改了《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