现在的位置: 首页 > 综合 > 正文

目录扫描的实现

2019年07月19日 ⁄ 综合 ⁄ 共 1209字 ⁄ 字号 评论关闭

/*************************************************************************
    > File Name: dir_scan.c
    > Author: mahao
    > Mail: mahao@hust.edu.cn
    > Created Time: 2013年11月12日 星期二 22时04分31秒
 ************************************************************************/

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

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

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

    chdir(dir);

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

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

    printf("Directory scan of %s\n",topdir);
    printdir(topdir,0);
    printf("done.\n");

    exit(0);

抱歉!评论已关闭.