【每日一坑 4】 查找文件

xuefu

长长的c又来了，有点乱没来得急整理。。。

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. #include <stdlib.h>
   
4. #include <dirent.h>
   
5. #include <errno.h>
   
6. #include <string.h>
   
7. #include <sys/stat.h>
   
8. 
   
9. #define PATH_SIZE 4096
   
10. 
    
11. char *path;
    
12. 
    
13. int show_txt(char *);
    
14. 
    
15. int main(int argc, const char *argv[])
    
16. {
    
17.   if (argc != 2) {
    
18.     printf("Usage: ./a.out <path of dir>\n");
    
19.     exit(1);
    
20.   }
    
21. 
    
22.   path = malloc(PATH_SIZE);
    
23.   strcpy(path, argv[1]);
    
24. 
    
25.   show_txt(path);
    
26. 
    
27.   free(path);
    
28. 
    
29.   return 0;
    
30. }
    
31. 
    
32. int show_txt(char *path)
    
33. {
    
34.   struct stat statbuf;
    
35.   struct dirent *dirp;
    
36.   DIR *dp;
    
37.   char *ptr;
    
38. 
    
39.   int len = strlen(path);
    
40. 
    
41.   if (lstat(path, &statbuf) < 0) {  /* lstat error */
    
42.     perror("lstat error");
    
43.     exit(1);
    
44.   }
    
45. 
    
46.   if (S_ISDIR(statbuf.st_mode) == 0) {  /* not a directory */
    
47.       if (strcmp(&path[len-4], ".txt") == 0) {
    
48.       printf("%s\n", path);
    
49.     }
    
50.     return 0;
    
51.   }
    
52. 
    
53.   ptr = path + len;
    
54.   *ptr++ = '/';
    
55.   *ptr = 0;
    
56. 
    
57.   if ((dp = opendir(path)) == NULL) { /* open directory error */
    
58.     perror("opendir error");
    
59.     exit(1);
    
60.   }
    
61. 
    
62.   /* iterate the directory(dp)'s each element*/
    
63.   while ((dirp = readdir(dp)) != NULL) {  
    
64.     /* ignore the present directory and the last directory */
    
65.     if (strcmp(dirp->d_name, ".") == 0 || strcmp(dirp->d_name, "..") == 0)
    
66.       continue;
    
67. 
    
68.     strcpy(ptr, dirp->d_name);  /* append the element's name to path */
    
69.     show_txt(path); /* call the show_txt(char*) recursively */
    
70.   }
    
71. 
    
72.   ptr[-1] = 0;
    
73.   closedir(dp);
    
74.   return 0;
    
75. }
    

复制代码