Linux中统计文件夹内各种文件类型的数量

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>
#include <sys/stat.h>
#include <string.h>
enum {FTW_F=1,FTW_D,FTW_DNR,FTW_NS};
static long nreg,ndir,nblk,nchr,nfifo,nslink,nsock;
typedef int (*Myfunc)(const char*,const struct stat*,int);
static int accumulate(const char*,const struct stat*,int);
static int walk(Myfunc);
static int start(const char*,Myfunc);
int main(int argc,char** argv)
{
if(argc!=2)
{
printf("Usage:traverser director");
exit(-1);
}
start(argv[1],accumulate);
printf("nreg %d/nndir %d/nnblk %d/nnchr %d/nnfifo %d/nnslink %d/nnsock %d/n",nreg,ndir,nblk,nchr,nfifo,nslink,nsock);
}
static char* fullpath;
static int
start(const char* pathname,Myfunc func)
{
int len;
len=256;
fullpath=malloc(len);
strcpy(fullpath,pathname);

return walk(func);
}
static int
walk(Myfunc func)
{
struct stat statbuf;
DIR* dp;
struct dirent* entry;
char* ptr;
int ret;
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)))
{
return func(fullpath,&statbuf,FTW_DNR);
}
while(entry=readdir(dp))
{
if( strcmp(entry->d_name,".")==0 || strcmp(entry->d_name,"..")==0)
{
continue;
}
strcpy(ptr,entry->d_name);
if((ret=walk(func))!=0)
{
break;
}
}
*--ptr=0;
closedir(dp);
return ret;
}
static int
accumulate(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:
printf("for S_IFDIR for %s/n",pathname);
exit(-1);
}
break;
case FTW_D:
ndir++;
break;
case FTW_DNR:
printf("Can't read directory %s/n",pathname);
return -1;
case FTW_NS:
printf("stat error for %s/n",pathname);
return -1;
}
return 0;
}

这段代码用来统计文件夹中各种文件的数量。

通过这个练习，熟悉了文件类型的判断。

这段代码中有些很巧妙的技巧：

1.递归时，同一个函数的循环中，为了消除平凡的字符串加减，用另外一根指针指向会改变的部分，然后每次将新的文件名，从这里开始增加，直接覆盖前一个名字。

2.一次递归结束，进行回溯时，用ptr[-1]=0来去掉slash后面的文件名。如/etc/httpd,ptr指向h。