linux/unix 目录操作系统调用 (最后的小测试程序有bug)

j3kljs02398j

TABLE
1. 目录操作相关的系统调用
        1.1 mkdir和rmdir系统调用
        1.1.1 实例
        1.2 chdir, getcwd系统调用
        1.2.1 实例
        1.3 opendir, closedir, readdir,
        1.3.1 实例:递归便利目录



1. 目录操作相关的系统调用

        1.1 mkdir和rmdir系统调用

1. 
   
2. filename: mk_rm_dir.c
   
3. #include <sys/stat.h>
   
4. int mkdir(const char *path, mode_t mode);
   
5. return:
   
6.         S        0
   
7.         F        -1
   
8. note:
   
9.         mode权限至少要有执行权限。
   

复制代码

1. 
   
2. #include <unistd.h>
   
3. int rmdir(const char *pathname);
   
4. return:
   
5.         S        0
   
6.         F        -1
   
7. note:
   
8.         pathname目录必须是空目录。
   

复制代码

1.1.1 实例

1. 
   
2. #include <unistd.h>
   
3. #include <sys/stat.h>
   
4. #include <stdio.h>
   
5. #include <assert.h>
   
6. 
   
7. #define MODE        (S_IRUSR | S_IWUSR | S_IXUSR | S_IXGRP | S_IXOTH)
   
8. 
   
9. int main(int argc, char *argv[])
   
10. {
    
11.         char        *pname;
    
12. 
    
13.         assert(argc == 2);
    
14.         pname = argv[1];
    
15.         assert(mkdir(pname, MODE) == 0);
    
16.         printf("create %s successful!\n", pname);
    
17.         assert(rmdir(pname) == 0);
    
18.         printf("rm %s\n", pname);
    
19.        
    
20.         return 0;
    
21. }
    

复制代码

测试：

1. 
   
2. [qtlldr@qtldr editing]$ ./mk_rm_dir  testdir
   
3. create testdir successful!
   
4. rm testdir
   
5. [qtlldr@qtldr editing]$
   

复制代码

1.2 chdir, getcwd系统调用

1. 
   
2. #include <unistd.h>
   
3. int chdir(const char *pathname);
   
4. return:
   
5.         S        0
   
6.         F        -1
   

复制代码

1. 
   
2. #include <unistd.h>
   
3. char *getpwd(char *buf, size_t size);
   
4. return:
   
5.         S        buf
   
6.         F        NULL
   
7. note:
   
8.         buf是缓冲地址，size是buf的长度。该缓冲必须有足够的长度以容纳绝对路径名加上一个null终止符。
   

复制代码

1.2.1 实例

1. 
   
2. filename:ch_get_dir.c
   
3. #include <unistd.h>
   
4. #include <stdio.h>
   
5. #include <string.h>
   
6. #include <assert.h>
   
7. 
   
8. #define BUFSIZE        (50)
   
9. 
   
10. int main(void)
    
11. {
    
12.         char buf[BUFSIZE];
    
13. 
    
14.         memset((void *)buf, '\0', sizeof buf);
    
15. 
    
16.         assert(chdir("/tmp") == 0);
    
17.         printf("chdir to /tmp successful\n");
    
18.         assert(getcwd(buf, BUFSIZE) != NULL);
    
19.         printf("now the directory is %s\n", buf);
    
20. 
    
21.         return 0;
    
22. }
    

复制代码

测试：

1. 
   
2. [qtlldr@qtldr editing]$ ./ch_get_dir
   
3. chdir to /tmp successful
   
4. now the directory is /tmp
   
5. [qtlldr@qtldr editing]$
   

复制代码

1.3 opendir, closedir, readdir,

1. 
   
2. #include <sys/type.s>
   
3. #include <dirent.h>
   
4. 
   
5. DIR *opendir(const char *dirname);
   
6. return:
   
7.         S        DIR指针
   
8.         F        NULL
   
9. note:
   
10.         DIR是一种目录结构，类似FILE。
    

复制代码

1. 
   
2. #include <sys/types.h>
   
3. #include <dirent.h>
   
4. 
   
5. struct dirent *readir(DIR *dirp);
   
6. return:
   
7.         S        一个指向保存目录流下一个目录项的dirent指针
   
8.         F        NULL
   
9. note:
   
10.         struct dirent {
    
11.                 char        d_name[NAME + 1]; /* \0结尾的文件名 */
    
12.         }
    
13.         到达目录尾或出错返回NULL，但是到达目录尾不会设置errno，出错则设置。
    
14.         如果在readir的同时有其他进程在目录中创建或者删除文件爱你，readdir不保证能列处该目录中所有文件。
    

复制代码

1. 
   
2. #include <sys/types.h>
   
3. #include <dirent.h>
   
4. int closedir(DIR *dirp);
   
5. return:
   
6.         S        0
   
7.         F        -1
   

复制代码

1.3.1 实例:递归便利目录

1. 
   
2. filename:help.txt 帮助文档
   
3. 
   
4. 
   
5.                         本程序只为学习linux目录操作而写
   
6. 
   
7. printdir
   
8.         输出目录文件或者统计目录中的文件数目
   
9. 
   
10. 语法：
    
11.         printdir [option] <files...>
    
12. 选项：
    
13.         -l
    
14.                 输出目录下的文件名
    
15.         -c
    
16.                 统计目录下的文件
    
17.         -d n
    
18.                 指定最大层次，最大为30
    
19. 默认行为：
    
20.         如果没有指定选项，那么只输出该目录下的文件名
    
21.        
    
22. BUG:
    
23.         -l与 -c选项不能同时使用，如果同时使用统计出错。(以后会修正）
    
24. 反馈：
    
25.         qtlldr@gmail.com
    
26. 
    
27. TIME:
    
28. Sat Sep 10 16:58:10 CST 2011
    
29. 
    
30.                         本程序只为学习linux目录操作而写
    

复制代码

1. 
   
2. filename:printdir.c
   
3. 
   
4. #include <unistd.h>
   
5. #include <fcntl.h>
   
6. #include <sys/stat.h>
   
7. #include <sys/types.h>
   
8. #include <dirent.h>
   
9. #include <stdio.h>
   
10. #include <string.h>
    
11. #include <stdlib.h>
    
12. 
    
13. #define INDENT_DEPTH        (4) /* 列举文件时的缩进数 */
    
14. #define DEPTH_MAX (30)        /* 递归便利的最大层次 */
    
15. #define HELPFILE ("help.txt")
    
16. 
    
17. typedef int count_t;
    
18. 
    
19. struct nfiletype {
    
20.         count_t ndir;
    
21.         count_t nreg;
    
22.         count_t nchr;
    
23.         count_t        nfifo;
    
24.         count_t        nsock;
    
25.         count_t nchar;
    
26.         count_t nblock;
    
27.         count_t        nlink;
    
28.         count_t ntotol;
    
29.         count_t nunknow;
    
30. };/*记录各个类型文件的数目*/
    
31. 
    
32. int        DEPTH = 20; /* 递归层级限制 */
    
33. int idepth_count = 1;
    
34. int idepth_print = 1;
    
35. static struct nfiletype
    
36. *count_files(const char *pathname,
    
37.                         struct nfiletype *nfile);
    
38. 
    
39. static void
    
40. printdir(const char *pathname, int indent);
    
41. 
    
42. 
    
43. 
    
44. int main(int argc, char **argv)
    
45. {
    
46.         int opt;
    
47.         int        depth_opt;
    
48.         int        count_flag = 0;
    
49.         int        print_flag = 0;
    
50.         char *parg = NULL;
    
51.         struct nfiletype nfiles = {0};
    
52. 
    
53.         int        fd_help;
    
54.         char buf_help[BUFSIZ];
    
55.         int nread_help;
    
56.         char *filename_help = HELPFILE;
    
57. 
    
58.         while ((opt = getopt(argc, argv, "lhd:c")) != -1) {
    
59.                 switch (opt) {
    
60.                         case 'l':
    
61.                                 print_flag = 1;
    
62.                                 break;
    
63.                         case 'c':
    
64.                                 count_flag = 1;
    
65.                                 break;
    
66.                         case 'd':
    
67.                                 depth_opt = strtol(optarg, NULL, 10);
    
68.                                 DEPTH = depth_opt <= DEPTH_MAX ? depth_opt: DEPTH;
    
69.                                 break;
    
70.                         case ':':
    
71.                                 printf("option needs a value\n");
    
72.                                 break;
    
73.                         case '?':
    
74.                                 printf("unknown option :%c\n", optopt);
    
75.                                 break;
    
76.                         case 'h':
    
77.                                 fd_help = open(filename_help, O_RDONLY);
    
78.                                 if (fd_help != -1) {
    
79.                                         while ((nread_help = read(fd_help, buf_help, BUFSIZ)) > 0) {
    
80.                                                 write(1, buf_help, nread_help);
    
81.                                         }
    
82.                                         close(fd_help);
    
83.                                 } else {
    
84.                                         fprintf(stderr, "open %s failed!\n", filename_help);
    
85.                                 }
    
86.                                 return 0;
    
87.                 }
    
88.         }
    
89.         /* 如果没有选项，那么默认是打印目录 */
    
90.         if (!print_flag && !count_flag)
    
91.                 print_flag = 1;
    
92. 
    
93.         for( ; optind < argc; optind++) {
    
94.                 parg = argv[optind];
    
95.                 if (print_flag) {
    
96.                         //printf("DEBUG-- printdir --%s\n", parg);
    
97.                         printdir(parg, 4);
    
98.                 }
    
99.                 if (count_flag) {
    
100.                         memset((void *)&nfiles, '\0', sizeof nfiles);
     
101.                         //printf("DEBUG-- count_files--%s\n", parg);
     
102.                         count_files(parg, &nfiles);
     
103.                         printf("In the %s there are :\n", parg);
     
104.                         printf("         directory %d\n", nfiles.ndir);
     
105.                         printf("         regular file %d\n", nfiles.nreg);
     
106.                         printf("         specal character file %d\n", nfiles.nchr);
     
107.                         printf("         special block file %d\n", nfiles.nblock);
     
108.                         printf("         fifo file %d\n", nfiles.nfifo);
     
109.                         printf("         sock file %d\n", nfiles.nsock);
     
110.                         printf("         link file %d\n", nfiles.nlink);
     
111.                         printf("         unknown file %d\n", nfiles.nunknow);
     
112.                         printf("Total %d\n", nfiles.ntotol);
     
113.                 }
     
114.         }
     
115.         return 0;
     
116. }
     
117. /*
     
118. *function: 对该目录下的文件类型进行统计
     
119. * input arg:
     
120. *                pathname:目录名指针
     
121. *                nfile:记录文件类型数目的结构体指针
     
122. * return：
     
123. *                记录文件类型数目的结构体指针
     
124. */
     
125. static struct nfiletype
     
126. *count_files(const char *pathname,
     
127.                         struct nfiletype *nfile)
     
128. {
     
129.         DIR *dp;
     
130.         struct dirent *entry;
     
131.         struct stat        statbuf;
     
132. 
     
133.         //printf("DEBUG-- in count_files -- %s\n", pathname);
     
134.         /* 层次控制 */       
     
135.         if (idepth_count > DEPTH)
     
136.                 return NULL;
     
137.         idepth_count++;
     
138. 
     
139.         if ((dp = opendir(pathname)) == NULL) {
     
140.                 fprintf(stderr, "can not open %s\n", pathname);
     
141.                 return NULL;
     
142.         }
     
143.         chdir(pathname);
     
144.         while ((entry = readdir(dp)) != NULL) {
     
145.                 /* 跳过 . 和 .. */
     
146.                 if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
     
147.                         continue;
     
148.                 /* 取得文件信息 */
     
149.                 if (lstat(entry->d_name, &statbuf) == -1) {
     
150.                         fprintf(stderr, "can not test the %s's type\n", entry->d_name);
     
151.                         return NULL;
     
152.                 }
     
153.                 /* 统计文件数目 */
     
154.                 if (S_ISDIR(statbuf.st_mode)) { /* 是目录就递归吧 */
     
155.                         //printf("DEBUG -- directory %s\n", entry->d_name);
     
156.                         count_files(entry->d_name, nfile);
     
157.                         nfile->ndir++;
     
158.                 }
     
159.                 else if (S_ISREG(statbuf.st_mode)) {
     
160.                         //printf("DEBUG -- regular file %s\n", entry->d_name);
     
161.                         nfile->nreg++;
     
162.                 }
     
163.                 else if (S_ISCHR(statbuf.st_mode))
     
164.                         nfile->nchr++;
     
165.                 else if (S_ISBLK(statbuf.st_mode))
     
166.                         nfile->nblock++;
     
167.                 else if (S_ISLNK(statbuf.st_mode))
     
168.                         nfile->nlink++;
     
169.                 else if (S_ISFIFO(statbuf.st_mode))
     
170.                         nfile->nfifo++;
     
171.                 else if (S_ISSOCK(statbuf.st_mode))
     
172.                         nfile->nsock++;
     
173.                 else nfile->nunknow++;
     
174.                 nfile->ntotol++;
     
175.         }
     
176.         chdir("..");
     
177.         closedir(dp);
     
178.         return nfile;
     
179. }
     
180. 
     
181. /*
     
182. nblock; *function:列出目录中的文件
     
183. nlink;  *input arg:
     
184. ntotol; *                pathname: 目录名
     
185. *return:
     
186. *                void
     
187. */
     
188. static void
     
189. printdir(const char *pathname, int indent)
     
190. {
     
191.         DIR *dp;
     
192.         struct dirent *entry;
     
193.         struct stat        statbuf;
     
194. 
     
195.         /* 层次控制 */       
     
196.         if (idepth_print > DEPTH)
     
197.                 return ;
     
198.         idepth_print++;
     
199. 
     
200.         if ((dp = opendir(pathname)) == NULL) {
     
201.                 fprintf(stderr, "can not open %s\n", pathname);
     
202.                 return ;
     
203.         }
     
204.         chdir(pathname);
     
205.         while ((entry = readdir(dp)) != NULL) {
     
206.                 /* 跳过 . 和 .. */
     
207.                 if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
     
208.                         continue;
     
209.                 if (lstat(entry->d_name, &statbuf) == -1) {
     
210.                         fprintf(stderr, "can not test the %s's type\n", entry->d_name);
     
211.                         return ;
     
212.                 }
     
213.                 if (S_ISDIR(statbuf.st_mode)) { /* 是目录就递归吧 */
     
214.                         printf("%*s%s/\n", indent," ",  entry->d_name);
     
215.                         printdir(entry->d_name, indent + INDENT_DEPTH);
     
216.                 }
     
217.                 else {
     
218.                         printf("%*s%s\n", indent," ", entry->d_name);
     
219.                         }
     
220.         }
     
221.         chdir("..");
     
222.         closedir(dp);
     
223. }
     

复制代码

dengcqw

lz，贴出来让大家学习的吗？

j3kljs02398j

回复 2# dengcqw


    自我娱乐