[Linux环境编程] Linux系统命令“ls -R”的实现

Linux系统命令“ls -R”的实现

 

一、基本概念

1、“ls -R”的意义

　　递归处理，将指定目录下的所有文件及子目录一并显示。

　　例： ls   -R   ./testdir1/

　　　　./testdir1/:
　　　　test1.c   test1.txt   test2.txt   testdir2

　　　　./testdir1/testdir2:
　　　　test2.c   test3.c      test3.txt   到 test1.c 的链接

　　其中蓝色为目录文件，红色为软连接文件（具体颜色和vimrc配置有关）。

 

二、重要函数与结构体

1、目录操作函数

　　这里的 void rewinddir(DIR *dirp); 函数非常重要，在读取一遍目录后，如果遗漏rewinddir函数会导致指向文件的指针停留在该目录文件流的末尾，从而影响之后二次读取。

 1        #include <sys/types.h>
 2        #include <dirent.h>
 3 
 4        DIR *opendir(const char *name);
 5        DIR *fdopendir(int fd);
 6 
 7 
 8        #include <dirent.h>
 9 
10        struct dirent *readdir(DIR *dirp);
11 
12            struct dirent {
13                ino_t          d_ino;       /* inode number */
14                off_t          d_off;       /* offset to the next dirent */
15                unsigned short d_reclen;    /* length of this record */
16                unsigned char  d_type;      /* type of file; not supported by all file system types */
17                char           d_name[256]; /* filename */
18            };

 1        #include <sys/types.h>
 2        #include <dirent.h>
 3 
 4        void rewinddir(DIR *dirp);

 

2、获取文件信息

　　这里必须使用 int lstat(const char *path, struct stat *buf); 函数，否则在处理链接文件时会将其链接的原文件作为处理对象，而不是其本身。

 1        #include <sys/types.h>
 2        #include <sys/stat.h>
 3        #include <unistd.h>
 4 
 5        int stat(const char *path, struct stat *buf);
 6        int fstat(int fd, struct stat *buf);
 7        int lstat(const char *path, struct stat *buf);
 8 
 9            struct stat {
10                dev_t     st_dev;     /* ID of device containing file */
11                ino_t     st_ino;     /* inode number */
12                mode_t    st_mode;    /* protection */
13                nlink_t   st_nlink;   /* number of hard links */
14                uid_t     st_uid;     /* user ID of owner */
15                gid_t     st_gid;     /* group ID of owner */
16                dev_t     st_rdev;    /* device ID (if special file) */
17                off_t     st_size;    /* total size, in bytes */
18                blksize_t st_blksize; /* blocksize for file system I/O */
19                blkcnt_t  st_blocks;  /* number of 512B blocks allocated */
20                time_t    st_atime;   /* time of last access */
21                time_t    st_mtime;   /* time of last modification */
22                time_t    st_ctime;   /* time of last status change */
23            };

 

3、 文件类型及权限的判断

 1        The following POSIX macros are defined to check the file type using the st_mode field:
 2 
 3            S_ISREG(m)  is it a regular file?
 4            S_ISDIR(m)  directory?
 5            S_ISCHR(m)  character device?
 6            S_ISBLK(m)  block device?
 7            S_ISFIFO(m) FIFO (named pipe)?
 8            S_ISLNK(m)  symbolic link? (Not in POSIX.1-1996.)
 9            S_ISSOCK(m) socket? (Not in POSIX.1-1996.)

 

 

三、执行结果及对比

 

　

　

四、总结

　　总的来说，实现“ls -R”功能所涉及的特殊结构体很少，涉及知识面较窄，代码量少，但细节部分较多，需要一定的经验，经验不足的话调试起来会比较麻烦，总体难度一般。在实现功能的过程中有几点需要注意下：

　　1. 利用 lstat 函数读取文件信息，否则无法有效处理软连接文件；

　　2. 利用 rewinddir 函数重置指向文件流的指针，以便之后二次遍历目录；

　　3. 在读取文件流时，注意屏蔽当前目录（.）、上一级目录（..）和隐藏文件（.*）。

　　目前暂未实现制表功能，有兴趣的朋友可以尝试着通过读取终端显示宽度与最长文件名长度来加以设计。

 

 

五、实现代码

1、 myls2.h 

#ifndef _MYLS2_H_
#define _MYLS2_H_

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<limits.h>
#include<unistd.h>
#include<dirent.h>
#include<sys/stat.h>
#include<sys/types.h>
#include<fcntl.h>
#include<time.h>
#include<pwd.h>
#include<grp.h>

// 处理错误
void error_printf(const char* );

// 处理路径下的文件
void list_dir(const char* );

// 所显示的文件信息
void display_dir(DIR* );

#endif//_MYLS2_H_

 

2、myls2.c

#include "myls2.h"

// 处理错误
void error_printf(const char* funname)
{
    perror(funname);
    exit(EXIT_FAILURE);
}

// 读取路径下的文件
void list_dir(const char* pathname)
{
    char nextpath[PATH_MAX+1];

    DIR* ret_opendir = opendir(pathname); // 打开目录"pathname"
    if(ret_opendir == NULL)
        error_printf("opendir");
    
    printf("%s:\n",pathname); // 显示pathname目录路径
    display_dir(ret_opendir); // 显示pathname目录下所有非隐藏文件名称

    struct dirent* ret_readdir = NULL; // 定义readdir函数返回的结构体变量
    while(ret_readdir = readdir(ret_opendir)) // 判断是否读取到目录尾
    {
        char* filename = ret_readdir->d_name; // 获取文件名

        int end = 0; // 优化显示路径（处理"./test/"与"./test"）
        while(pathname[end])
            end++;
        strcpy(nextpath,pathname);
        if(pathname[end-1] != '/')
            strcat(nextpath,"/");
        strcat(nextpath,filename);

        struct stat file_message = {}; // 定义stat函数返回的结构体变量
        int ret_stat = lstat(nextpath, &file_message); // 获取文件信息
        if(ret_stat == -1) // stat读取文件错误则输出提示信息
            printf("%s error!", filename);
        else if(S_ISDIR(file_message.st_mode)  && filename[0]!='.') // 筛选"."、".."与隐藏文件
        {
            list_dir(nextpath);
        }
    }
    closedir(ret_opendir);
}

// 打印所读取文件的信息
void display_dir(DIR* ret_opendir)
{
    struct dirent* ret_readdir = NULL; // 定义readdir函数返回的结构体变量
    while(ret_readdir = readdir(ret_opendir)) // 判断是否读取到目录尾
    {
        char* filename = ret_readdir->d_name; // 获取文件名
        if(filename[0]!='.') // 不输出当前目录、上一级目录与隐藏文件 
            printf("%s\t",ret_readdir->d_name); // 打印文件名
    }
    rewinddir(ret_opendir); // 非常重要，将文件流的指针拨回起始位置
    puts("");
    puts("");
}

 

3、main_myls2.c

#include "myls2.h"


int main(const char argc, const char** argv)
{
    char path[PATH_MAX+1] = {};

    if(argc == 2 && !(strcmp(argv[1],"-R"))) // 判断命令格式
        strcpy(path,".");
    else if(argc != 3)
    {
        printf("格式有误！ \n");
        exit(EXIT_FAILURE);
    }
    else
        strcpy(path,argv[2]);

    if(!(strcmp(argv[1],"-R")))
    {
        struct stat file_message = {};
        int ret_stat = lstat(path, &file_message);

        if(ret_stat == -1)
               error_printf("stat");

        if(S_ISDIR(file_message.st_mode)) // 判断是否为目录
            list_dir(path);
        else
        printf("It is not dir!");
    }
    else
    {
        printf("error in main!\n");
        exit(EXIT_FAILURE);
    }                    
    return 0;
}