[cpp]
#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;
}
