Linux中tree命令的C语言实现

一、tree命令是干什么的？

       我们经常用tree命令来查看指定目录下的目录文件结构，并以树状结构显示。

二、明确需求:

       1、显示目录下的所有文件

       2、遇到目录则要进入目录，并显示该目录下的所有文件

       3、显示需要使用树状层次显示

三、编程思路：

       <1>、很明显这是一个递归过程，遇到目录则递归调用。

       <2>、需要用到的函数

                 (1)、打开目录

                         DIR *opendir( const char *name );

                         参数   name : 目录路径(可以是相对路径也可以是绝对路径)

                         返回值   :  指向参数指定目录的第一个目录项的目录流指针

                 (2)、读取目录

                         struct dirent *readdir(DIR *dirp);

                         参数  dirp : 目录流指针

                         返回值 ： 指定目录项的目录结构体指针，并使dirp指向下一个目录项

                         注意： 当该目录下的目录项都读取完之后，返回NULL

                 (3)、关闭目录

                         int closedir( DIR *dirp );

四、代码实现

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <dirent.h>

//自定义错误处理函数
void my_error(const char *strerr)
{
    perror(strerr);
    exit(1);
}

//输出目录结构
void PrintDirentStruct(char direntName[], int level)
{
    //定义一个目录流指针
    DIR *p_dir = NULL;

    //定义一个目录结构体指针
    struct dirent *p_dirent = NULL;

    //打开目录，返回一个目录流指针指向第一个目录项
    p_dir = opendir(direntName);
    if(p_dir == NULL)
    {
        my_error("opendir error");
    }

    //循环读取每个目录项, 当返回NULL时读取完毕
    while((p_dirent = readdir(p_dir)) != NULL)
    {
        //备份之前的目录名
        char *backupDirName = NULL;

        if(p_dirent->d_name[0] == '.')
        {
            continue;
        }

        int i;
        for(i = 0; i < level; i++)
        {
            printf("|");
            printf("     ");
        }
        printf("|--- ");
        printf("%s\n", p_dirent->d_name);


        //如果目录项仍是一个目录的话，进入目录
        if(p_dirent->d_type == DT_DIR)
        {
            //当前目录长度
            int curDirentNameLen = strlen(direntName) + 1;

            //备份
            backupDirName = (char *)malloc(curDirentNameLen);
            memset(backupDirName, 0, curDirentNameLen);
            memcpy(backupDirName, direntName, curDirentNameLen);


            strcat(direntName, "/");
            strcat(direntName, p_dirent->d_name);
            PrintDirentStruct(direntName, level + 1);


            //恢复之前的目录名
            memcpy(direntName, backupDirName, curDirentNameLen);
            free(backupDirName);
            backupDirName = NULL;
        }

    }

    closedir(p_dir);

}

int main(int argc , char *argu[])
{
    //目录名
    char direntName[256];
    memset(direntName, 0, sizeof(direntName));

    if(argc == 1)
    {
        direntName[0] = '.';
    }
    else if(argc == 2)
    {
        strcat(direntName, argu[1]);
    }
    else
    {
        my_error("argument error");
    }

    //输出目录结构
    printf("%s\n", direntName);
    PrintDirentStruct(direntName, 0);
    return 0;

}

五、运行截图