在准备阅读一个开源项目的代码前，可以大约看看整个项目共有多少代码，估计项目的规模。我就写了一个简单的程序来达到此目的，其中的一些代码参考了apue中的代码。
代码如下:
View Code
1 //程序功能:统计一个文件夹(一个项目)中所有文件的有效代码行数(除去空白行)。 2 3 #include <apue.h> 4 #include <dirent.h> 5 #include <limits.h> 6 #include <string.h> 7 #include <fcntl.h> 8 #include <unistd.h> 9 #define MAXBUF 5000 10 11 typedef int Myfunc(const char *,const struct stat *,int); 12 13 static Myfunc myfunc; 14 static int myftw(char *,Myfunc *); 15 static int dopath(Myfunc *); 16 static int sum_lines=0;//the sum of no empty lines 17 18 static long nreg, ndir, nblk, nchr, nfifo, nslink, nsock, ntot; 19 20 21 int 22 main(int argc, char *argv[]) 23 { 24 int ret; 25 26 if (argc != 2) 27 err_quit("usage: ftw <starting-pathname>"); 28 29 ret = myftw(argv[1], myfunc); /* does it all */ 30 31 printf("\n\nSum lines: %d lines\n\n\n",sum_lines); 32 exit(ret); 33 } 34 35 /* 36 * Descend through the hierarchy, starting at "pathname". 37 * The caller's func() is called for every file. 38 */ 39 #define FTW_F 1 /* file other than directory */ 40 #define FTW_D 2 /* directory */ 41 #define FTW_DNR 3 /* directory that can't be read */ 42 #define FTW_NS 4 /* file that we can't stat */ 43 44 static char *fullpath; /* contains full pathname for every file */ 45 46 static int /* we return whatever func() returns */ 47 myftw(char *pathname, Myfunc *func) 48 { 49 int len; 50 fullpath = path_alloc(&len); /* malloc's for PATH_MAX+1 bytes */ 51 /* ({Prog pathalloc}) */ 52 strncpy(fullpath, pathname, len); /* protect against */ 53 fullpath[len-1] = 0; /* buffer overrun */ 54 55 return(dopath(func)); 56 } 57 58 static int dopath(Myfunc * func) 59 { 60 struct stat statbuf; 61 struct dirent *dirp; 62 DIR *dp; 63 int ret; 64 char *ptr; 65 66 if(lstat(fullpath,&statbuf)<0) 67 return (func(fullpath,&statbuf,FTW_NS)); 68 if(S_ISDIR(statbuf.st_mode)==0) 69 return (func(fullpath,&statbuf,FTW_F)); 70 71 if( (ret=func(fullpath,&statbuf,FTW_D))!=0 ) 72 return ret; 73 74 ptr=fullpath+strlen(fullpath); 75 *ptr++='/'; 76 *ptr=0; 77 78 if((dp=opendir(fullpath))==NULL) 79 return (func(fullpath,&statbuf,FTW_DNR)); 80 while((dirp=readdir(dp))!=NULL) 81 { 82 if(strcmp(dirp->d_name,".")==0 || strcmp(dirp->d_name,"..")==0) 83 continue; 84 strcpy(ptr,dirp->d_name); 85 if(ret=dopath(func)!=0) 86 return ret; 87 } 88 ptr[-1]=0; 89 if(closedir(dp)<0) 90 err_ret("can't close directory %s",fullpath); 91 return ret; 92 } 93 94 95 static int 96 myfunc(const char *pathname, const struct stat *statptr, int type) 97 { 98 switch (type) { 99 case FTW_F: 100 { 101 int fd=0; 102 int num=0; 103 char buf[MAXBUF]; 104 int file_lines=0;//current file lines 105 if((fd=open(pathname,O_RDONLY))<0) 106 err_sys("error open for %s\n",pathname); 107 while((num=read(fd,buf,4096))!=0) 108 { 109 int index; 110 char previous_c=buf[0]; 111 char current_c; 112 for(index=1;index<num;index++) 113 { 114 current_c=buf[index]; 115 if(previous_c!='\n' && current_c=='\n') 116 sum_lines++,file_lines++; 117 previous_c=current_c; 118 } 119 } 120 if(close(fd)==-1) 121 err_sys("error close for %s",pathname); 122 printf("%s:%d lines\n",pathname,file_lines); 123 } 124 break; 125 126 case FTW_D: 127 ndir++; 128 break; 129 130 case FTW_DNR: 131 err_ret("can't read directory %s", pathname); 132 break; 133 134 case FTW_NS: 135 err_ret("stat error for %s", pathname); 136 break; 137 138 default: 139 err_dump("unknown type %d for pathname %s", type, pathname); 140 } 141 142 return(0); 143 }
编译方法:
gcc -g -o count_line count_line.c fig2_15.c error.c
其中fig2_15.c和error.c来自apue，分别定义了一些简单的函数和错误处理函数。
输入样例:在redis这个开源项目的文件夹之下，输入:
./count_line src
输出:
1 src/rio.h:85 lines 2 src/crc16.c:85 lines 3 src/zipmap.h:46 lines 4 src/intset.h:46 lines 5 src/crc64.c:186 lines 6 src/adlist.h:82 lines 7 .....省略 8 9 Sum lines: 36108 lines
参考资料:apue
如果你觉得我的文章对你有帮助，请推荐一下，非常感谢!