[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;  
}