Linux 获取制定目录的文件数（递归）

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <dirent.h>
#include <string.h>
/*
获取指定文件数目（absolute path)
*/
int getFileNum(char* root)
{
	DIR *dir = NULL;
	//打开目录
	dir = opendir(root);
	if(dir == NULL)
	{
		perror("opendir");
		exit(1);
	}
	
	//读取目录
	struct dirent* ptr = NULL;
	char path[1024] = {0};
	int total = 0;
	while ((ptr = readdir(dir))!=NULL) {
		
		 if(strcmp(ptr->d_name,".") == 0||strcmp(ptr->d_name,"..")==0){
		  	continue;
	  	}
	  	
	  	//递归
	  	if(ptr->d_type == DT_DIR){
	  		sprintf(path,"%s/%s",root,ptr->d_name);
	  		total += getFileNum(path);
	  	}
	  	
	  	if(ptr->d_type == DT_REG){
	  		
	  		total++;
	  	}
	}
	
	closedir(dir);
	return total;
}

int main(int argc,char* argv[])
{
	if(argc<2){
		printf("./fileNum path");
		exit(1);
	}
	
	int num = 0;
	num = getFileNum(argv[1]);
	
	printf("%s has %d file\n",argv[1],num);
	
	return 0;
	
}