9.自己实现linux中的tree

运行效果:

代码：

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <dirent.h>
#include <sys/stat.h>

void MyTree(char szPath[],int deep)
{
    //目录用于遍历
    DIR *pDir;
    //用于存储文件信息
    struct dirent *pDent;
    //子目录名字
    char szSubPath[PATH_MAX];
    
    //用于判断文件的类型
    struct stat stInfo;
    //用来遍历层数，格式输出
    int i;
    
    //打开当前目录，获取目录列表
    pDir = opendir(szPath);
    //如果打开失败
    if(pDir == NULL)
    {
        perror("Fail to opendir");
        return;
    }
    
    //不断循环
    while(1)
    {
        //依次读取文件信息
        pDent = readdir(pDir);
        //如果读取失败，当前递归结束
        if(pDent == NULL)
        {
            break;
        }
        
        //如果获得了文件信息
        
        //过滤掉当前目录(.)和上一层目录 (..)
        if(strcmp(pDent->d_name,".")==0 || strcmp(pDent->d_name,"..")==0)
        {
            continue;
        }
        
        //格式输出
        for(i=0;i!=deep;i++)
        {
            printf("│   ");
        }
        
        //如果是目录文件
        if(pDent->d_type == DT_DIR)
        {
            //红色
            printf("├── 33[0;31m%s33[0m
",pDent->d_name);
            sprintf(szSubPath,"%s/%s",szPath,pDent->d_name);
            MyTree(szSubPath,deep+1);
        }
        else
        {
            stat(pDent->d_name,&stInfo);
            //判断是否是普通文件
            if(S_ISREG(stInfo.st_mode))
            {
                //判断是否有执行权限
                if(stInfo.st_mode & 0100)
                {
                    //青色
                    printf("├── 33[1;32m%s33[0m
",pDent->d_name);
                }
                //不能执行的文件
                else
                {
                    printf("├── %s
",pDent->d_name);
                }
            }
            //是否是一个块文件
            else if(S_ISFIFO(stInfo.st_mode))
            {
                    printf("├── 33[1;42m%s33[0m
",pDent->d_name);
            }
            //其他文件
            else
            {
                printf("├── %s
",pDent->d_name);
            }
            
        }
    }
    closedir(pDir);
}

int main(int argc,char **argv)
{
        //char szPath[PATH_MAX];
        
        
        
        if(argc == 2)
        {
            chdir(argv[1]);
        }
        
        //getcwd(szPath,PATH_MAX);
        //MyTree(szPath,0);
        MyTree(argv[1],0);
        
        return 0;
}