linux系统编程之文件与IO（四）：目录访问相关系统调用
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系统调用
[code]
filename: mk_rm_dir.c
#include <sys/stat.h>
int mkdir(const char *path, mode_t mode);
return:
S 0
F -1
note:
mode权限至少要有执行权限。
[/code]
[code]
#include <unistd.h>
int rmdir(const char *pathname);
return:
S 0
F -1
note:
pathname目录必须是空目录。
1.1.1 实例
#include <unistd.h> #include <sys/stat.h> #include <stdio.h> #include <assert.h> #define MODE (S_IRUSR | S_IWUSR | S_IXUSR | S_IXGRP | S_IXOTH) int main(int argc, char *argv[]) { char *pname; assert(argc == 2); pname = argv[1]; assert(mkdir(pname, MODE) == 0); printf("create %s successful!\n", pname); assert(rmdir(pname) == 0); printf("rm %s\n", pname); return 0; }
测试：
[qtlldr@qtldr editing]$ ./mk_rm_dir testdir
create testdir successful!
rm testdir
[qtlldr@qtldr editing]$
1.2 chdir, getcwd系统调用
#include <unistd.h>
int chdir(const char *pathname);
return:
S 0
F -1
#include <unistd.h>
char *getpwd(char *buf, size_t size);
return:
S buf
F NULL
buf是缓冲地址，size是buf的长度。该缓冲必须有足够的长度以容纳绝对路径名加上一个null终止符。
1.2.1 实例
[code]
filename:ch_get_dir.c
#include <unistd.h> #include <stdio.h> #include <string.h> #include <assert.h> #define BUFSIZE (50) int main(void) { char buf[BUFSIZE]; memset((void *)buf, '\0', sizeof buf); assert(chdir("/tmp") == 0); printf("chdir to /tmp successful\n"); assert(getcwd(buf, BUFSIZE) != NULL); printf("now the directory is %s\n", buf); return 0; }
测试：
[qtlldr@qtldr editing]$ ./ch_get_dir
chdir to /tmp successful
now the directory is /tmp
[qtlldr@qtldr editing]$
1.3 opendir, closedir, readdir,
#include <sys/type.s>
#include <dirent.h>
DIR *opendir(const char *dirname);
return:
S DIR指针
F NULL
note:
DIR是一种目录结构，类似FILE。
#include <sys/types.h>
#include <dirent.h>
struct dirent *readir(DIR *dirp);
return:
S 一个指向保存目录流下一个目录项的dirent指针
F NULL
note:
struct dirent {
char d_name[NAME + 1]; /* \0结尾的文件名 */
}
到达目录尾或出错返回NULL，但是到达目录尾不会设置errno，出错则设置。
如果在readir的同时有其他进程在目录中创建或者删除文件爱你，readdir不保证能列处该目录中所有文件。
#include <sys/types.h>
#include <dirent.h>
int closedir(DIR *dirp);
return:
S 0
F -1
1.3.1 实例:递归便利目录
filename:help.txt 帮助文档
本程序只为学习linux目录操作而写
printdir
输出目录文件或者统计目录中的文件数目
语法：
printdir [option] <files...>
选项：
-l
输出目录下的文件名
-c
统计目录下的文件
-d n
指定最大层次，最大为30
默认行为：
如果没有指定选项，那么只输出该目录下的文件名
BUG:
-l与 -c选项不能同时使用，如果同时使用统计出错。(以后会修正）
本程序只为学习linux目录操作而写
filename:printdir.c
#include <unistd.h> #include <fcntl.h> #include <sys/stat.h> #include <sys/types.h> #include <dirent.h> #include <stdio.h> #include <string.h> #include <stdlib.h> #define INDENT_DEPTH (4) /* 列举文件时的缩进数 */ #define DEPTH_MAX (30) /* 递归便利的最大层次 */ #define HELPFILE ("help.txt") typedef int count_t; struct nfiletype { count_t ndir; count_t nreg; count_t nchr; count_t nfifo; count_t nsock; count_t nchar; count_t nblock; count_t nlink; count_t ntotol; count_t nunknow; };/*记录各个类型文件的数目*/ int DEPTH = 20; /* 递归层级限制 */ int idepth_count = 1; int idepth_print = 1; static struct nfiletype *count_files(const char *pathname, struct nfiletype *nfile); static void printdir(const char *pathname, int indent); int main(int argc, char **argv) { int opt; int depth_opt; int count_flag = 0; int print_flag = 0; char *parg = NULL; struct nfiletype nfiles = {0}; int fd_help; char buf_help[BUFSIZ]; int nread_help; char *filename_help = HELPFILE; while ((opt = getopt(argc, argv, "lhd:c")) != -1) { switch (opt) { case 'l': print_flag = 1; break; case 'c': count_flag = 1; break; case 'd': depth_opt = strtol(optarg, NULL, 10); DEPTH = depth_opt <= DEPTH_MAX ? depth_opt: DEPTH; break; case ':': printf("option needs a value\n"); break; case '?': printf("unknown option :%c\n", optopt); break; case 'h': fd_help = open(filename_help, O_RDONLY); if (fd_help != -1) { while ((nread_help = read(fd_help, buf_help, BUFSIZ)) > 0) { write(1, buf_help, nread_help); } close(fd_help); } else { fprintf(stderr, "open %s failed!\n", filename_help); } return 0; } } /* 如果没有选项，那么默认是打印目录 */ if (!print_flag && !count_flag) print_flag = 1; for( ; optind < argc; optind++) { parg = argv[optind]; if (print_flag) { //printf("DEBUG-- printdir --%s\n", parg); printdir(parg, 4); } if (count_flag) { memset((void *)&nfiles, '\0', sizeof nfiles); //printf("DEBUG-- count_files--%s\n", parg); count_files(parg, &nfiles); printf("In the %s there are :\n", parg); printf(" directory %d\n", nfiles.ndir); printf(" regular file %d\n", nfiles.nreg); printf(" specal character file %d\n", nfiles.nchr); printf(" special block file %d\n", nfiles.nblock); printf(" fifo file %d\n", nfiles.nfifo); printf(" sock file %d\n", nfiles.nsock); printf(" link file %d\n", nfiles.nlink); printf(" unknown file %d\n", nfiles.nunknow); printf("Total %d\n", nfiles.ntotol); } } return 0; } /* *function: 对该目录下的文件类型进行统计 * input arg: * pathname:目录名指针 * nfile:记录文件类型数目的结构体指针 * return： * 记录文件类型数目的结构体指针 */ static struct nfiletype *count_files(const char *pathname, struct nfiletype *nfile) { DIR *dp; struct dirent *entry; struct stat statbuf; //printf("DEBUG-- in count_files -- %s\n", pathname); /* 层次控制 */ if (idepth_count > DEPTH) return NULL; idepth_count++; if ((dp = opendir(pathname)) == NULL) { fprintf(stderr, "can not open %s\n", pathname); return NULL; } chdir(pathname); while ((entry = readdir(dp)) != NULL) { /* 跳过 . 和 .. */ if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue; /* 取得文件信息 */ if (lstat(entry->d_name, &statbuf) == -1) { fprintf(stderr, "can not test the %s's type\n", entry->d_name); return NULL; } /* 统计文件数目 */ if (S_ISDIR(statbuf.st_mode)) { /* 是目录就递归吧 */ //printf("DEBUG -- directory %s\n", entry->d_name); count_files(entry->d_name, nfile); nfile->ndir++; } else if (S_ISREG(statbuf.st_mode)) { //printf("DEBUG -- regular file %s\n", entry->d_name); nfile->nreg++; } else if (S_ISCHR(statbuf.st_mode)) nfile->nchr++; else if (S_ISBLK(statbuf.st_mode)) nfile->nblock++; else if (S_ISLNK(statbuf.st_mode)) nfile->nlink++; else if (S_ISFIFO(statbuf.st_mode)) nfile->nfifo++; else if (S_ISSOCK(statbuf.st_mode)) nfile->nsock++; else nfile->nunknow++; nfile->ntotol++; } chdir(".."); closedir(dp); return nfile; } /* nblock; *function:列出目录中的文件 nlink; *input arg: ntotol; * pathname: 目录名 *return: * void */ static void printdir(const char *pathname, int indent) { DIR *dp; struct dirent *entry; struct stat statbuf; /* 层次控制 */ if (idepth_print > DEPTH) return ; idepth_print++; if ((dp = opendir(pathname)) == NULL) { fprintf(stderr, "can not open %s\n", pathname); return ; } chdir(pathname); while ((entry = readdir(dp)) != NULL) { /* 跳过 . 和 .. */ if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue; if (lstat(entry->d_name, &statbuf) == -1) { fprintf(stderr, "can not test the %s's type\n", entry->d_name); return ; } if (S_ISDIR(statbuf.st_mode)) { /* 是目录就递归吧 */ printf("%*s%s/\n", indent," ", entry->d_name); printdir(entry->d_name, indent + INDENT_DEPTH); } else { printf("%*s%s\n", indent," ", entry->d_name); } } chdir(".."); closedir(dp); }
注：本文转自：http://blog.chinaunix.net/uid-26242642-id-2779802.html
