Linux--tree命令实现

一个简单的实现。
参考

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

void printdir(const char* dir, int depth){
    DIR *dp;
    struct dirent *entry;
    struct stat statbuf;

    if((dp = opendir(dir)) == NULL){
        fprintf(stderr, "can't open directory %s\n", dir);
        return;
    }

    chdir(dir);
    while((entry = readdir(dp)) != NULL){
        lstat(entry->d_name, &statbuf);
        if(!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, "..")) continue;
        int index = depth;
        while(index--) printf("|  ");
        printf("|--%s\n", entry->d_name);
        index = depth;
        if(S_ISDIR(statbuf.st_mode)){
            printdir(entry->d_name, depth + 1);
        }
    }
    chdir("..");
    closedir(dp);
}

int main(int argc, char* argv[]){
    char *topdir = ".";
    if(argc >= 2 ) topdir = argv[1];

    printf("%s\n", topdir);
    printdir(topdir, 0);
    printf("\n");
    return 0;
}