[Linux]C语言遍历文件和目录

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>

#define MAX_PATH_LENGTH		512
#define MAX_FILE_EXTENSION	9

unsigned long visit_dirs = 0;
unsigned long visit_files = 0;

void listdir(char *path){
DIR 		*ptr_dir;
struct dirent 	*dir_entry;
int 		i = 0;
char		*child_path;
char		*file_path;

child_path = (char*)malloc(sizeof(char)*MAX_PATH_LENGTH);
if(child_path == NULL){
printf("allocate memory for path failed.\n");
return;
}
memset(child_path, 0, sizeof(char)*MAX_PATH_LENGTH);

file_path = (char*)malloc(sizeof(char)*MAX_PATH_LENGTH);
if(file_path == NULL){
printf("allocate memory for file path failed.\n");
free(child_path);
child_path = NULL;
return;
}
memset(file_path, 0, sizeof(char)*MAX_PATH_LENGTH);

ptr_dir = opendir(path);
while((dir_entry = readdir(ptr_dir)) != NULL){
if(dir_entry->d_type & DT_DIR){
if(strcmp(dir_entry->d_name,".") == 0 ||
strcmp(dir_entry->d_name,"..") == 0){
continue;
}

sprintf(child_path, "%s/%s", path, dir_entry->d_name);
printf("[DIR]%s\n", child_path);

visit_dirs++;

listdir(child_path);
}

if(dir_entry->d_type & DT_REG){
sprintf(file_path, "%s/%s", path, dir_entry->d_name);
printf("[FILE]%s\n", file_path);
visit_files++;
}
}

free(child_path);
child_path = NULL;

free(file_path);
file_path = NULL;
}

int main(int argc, char* argv[]){
if(argc == 2){
listdir(argv[1]);
printf("Total DIR: %ld, Total FILE: %ld\n", visit_dirs, visit_files);
}else{
printf("Usage: listdir <dir>\n");
return;
}

return 0;
}