tree命令的模拟

1. include <fcntl.h>  
2. #include <unistd.h>  
3. #include <stdio.h>  
4. #include <dirent.h>  
5. #include <string.h>  
6. #include <sys/stat.h>  
7. #include <sys/types.h>  
8. #include <stdlib.h>  
9.   
10. void scan_dir(char *dir, int depth)  
11. {  
12.    DIR *dp;  
13.    struct dirent *entry;  
14.    struct stat statbuf;  
15.    if((dp = opendir(dir)) == NULL) {  
16.       puts("Cannot open directioy");  
17.       return;  
18.    }  
19.    chdir(dir);  
20.    while((entry = readdir(dp)) != NULL) {  
21.       lstat(entry->d_name, &statbuf);  
22.       if(S_IFDIR & statbuf.st_mode) {  
23.          if(strcmp(".", entry->d_name) == 0 || strcmp("..", entry->d_name) == 0)  
24.         continue;  
25.          printf("%*s%s/\n",depth,"",entry->d_name);  
26.          scan_dir(entry->d_name,depth+4);  
27.       }  
28.       else  
29.          printf("%*s%s\n", depth, "", entry->d_name);  
30.    }  
31.    chdir("..");  
32.    closedir(dp);  
33. }  
34.   
35. int main(int argc, char *argv[])  
36. {  
37.     puts("Scan the directory:");  
38.     scan_dir(argv[1], 0);  
39.     puts("Scanning ended.");  
40.     return 0;  
41. }