遍历目录

#include <stdio.h>
#include <string.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/dir.h>

#define MAX_PATH 1024


void fsize(char * name);
void dirwalk(char * dir, void (*fcn)(char *));

int main(int argc, char **argv)
{
    if(argc == 1)
    {
        fsize(".");
    }
    else
    {
        while(--argc > 0)
        {
            fsize(*++argv);
        }
    }

    return 0;
}

void fsize(char * name)
{
    printf("In the function [%s]/n", __FUNCTION__);
    struct stat stbuf;

    if(stat(name, &stbuf) == -1)
    {
        fprintf(stderr, "fsize: can't access %s/n", name);
        return;
    }
    if((stbuf.st_mode & S_IFMT) == S_IFDIR)
    {
        dirwalk(name, fsize);
    }
    else
    {
        printf("stbuf.st_size=%ld, name=%s/n", stbuf.st_size, name);
        i++;
    }
}

void dirwalk(char * dir, void (*fcn)(char *))
{
    printf("In the function [%s]/n", __FUNCTION__);
    char name[MAX_PATH];
    struct dirent * dp;
    DIR * dfd;

    if((dfd = opendir(dir)) == NULL)
    {
        fprintf(stderr, "dirwalk: can't open %s/n", dir);
        return;
    }
    while((dp = readdir(dfd)) != NULL)
    {
        if(strcmp(dp->d_name, ".") == 0 || strcmp(dp->d_name, "..") == 0)
        {
            printf("dp->d_name = %s/n", dp->d_name);
            continue;
        }
        if(strlen(dir) + strlen(dp->d_name) + 2 > sizeof(name))
        {
            fprintf(stderr, "dirwalk: name %s%s too long/n", dir, dp->d_name);
        }
        else
        {
            sprintf(name, "%s/%s", dir, dp->d_name);
            (*fcn)(name);
        }
    }
    closedir(dfd);
}

你可能感兴趣的:(遍历目录)