test_file_traversal.c File Reference

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <dirent.h>
#include <errno.h>
#include "gwlib/gwlib.h"

Go to the source code of this file.

Functions
static int	count_file (const char *filename, const struct stat *sb, int tflag, void *ftwbuf)
static int	for_each_file (const Octstr *dir_s, int ignore_err, int(*cb)(const char *, const struct stat *, int, void *))
int	main (int argc, char **argv)

Variables
static Counter *	counter

Function Documentation

count_file()

static int count_file ( const char *  filename, const struct stat *  sb, int  tflag, void *  ftwbuf  )

static

Definition at line 96 of file test_file_traversal.c.

References counter, and counter_increase().

Referenced by main().

{
    counter_increase(counter);

    return 0;
}

for_each_file()

static int for_each_file ( const Octstr *  dir_s, int  ignore_err, int(*)(const char *, const struct stat *, int, void *)  cb  )

static

Definition at line 120 of file test_file_traversal.c.

References error(), filename, octstr_destroy(), octstr_format(), and octstr_get_cstr.

Referenced by main().

{
    DIR *dir;
    struct dirent *ent;
    int ret = 0;
#ifndef _DIRENT_HAVE_D_TYPE
    struct stat stat;
#endif

    if ((dir = opendir(octstr_get_cstr(dir_s))) == NULL) {
        error(errno, "Could not open directory `%s'", octstr_get_cstr(dir_s));
        return -1;
    }
    while ((ent = readdir(dir)) != NULL) {
        Octstr *filename;
        if (!(strcmp((char*)ent->d_name, "." ) != 0 && strcmp((char*)ent->d_name, ".." ) != 0))
            continue;
        filename = octstr_format("%S/%s", dir_s, ent->d_name);
#ifdef _DIRENT_HAVE_D_TYPE
        if (ent->d_type == DT_DIR && for_each_file(filename, ignore_err, cb) == -1) {
            ret = -1;
        } else if (ent->d_type == DT_REG && cb != NULL) {
            cb(octstr_get_cstr(filename), NULL, 0, NULL);
        }
#else
        if (lstat(octstr_get_cstr(filename), &stat) == -1) {
            if (!ignore_err)
                error(errno, "Could not get stat for `%s'", octstr_get_cstr(filename));
            ret = -1;
        } else if (S_ISDIR(stat.st_mode) && for_each_file(filename, ignore_err, cb) == -1) {
            ret = -1;
        } else if (S_ISREG(stat.st_mode) && cb != NULL) {
            cb(octstr_get_cstr(filename), &stat, 0, NULL);
        }
#endif
        octstr_destroy(filename);
        if (ret == -1 && ignore_err)
            ret = 0;
        else if (ret == -1)
            break;
    }
    closedir(dir);

    return ret;
}

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 168 of file test_file_traversal.c.

References count_file(), counter, counter_create(), counter_destroy(), counter_set(), counter_value(), debug(), for_each_file(), gwlib_init(), gwlib_shutdown(), octstr_create, octstr_destroy(), and start.

{
    Octstr *os1;
    Octstr *os2;
    time_t start, diff;

    gwlib_init();

    os1 = octstr_create(argv[1]);
    os2 = octstr_create(argv[2]);

    counter = counter_create();
    start = time(NULL);
    for_each_file(os1, 1, count_file);
    diff = (time(NULL) - start);
    debug("",0,"file count: %ld in %lds", (long) counter_value(counter), (long) diff);

#ifdef HAVE_NFTW
    counter_set(counter, 0);
    start = time(NULL);
    for_each_file2(os2, 1, count_file2);
    diff = (time(NULL) - start);
    debug("",0,"file count: %ld in %lds", (long) counter_value(counter), (long) diff);
#endif

    counter_destroy(counter);
    octstr_destroy(os1);
    octstr_destroy(os2);
    gwlib_shutdown();
    return 0;
}

Variable Documentation

counter

Counter* counter

static

Definition at line 75 of file test_file_traversal.c.

Referenced by count_file(), and main().