Annotation of embedaddon/pimd/libite/dir.c, revision 1.1
1.1 ! misho 1: /* Functions for operating on files in directories.
! 2: *
! 3: * Copyright (c) 2008-2014 Joachim Nilsson <troglobit@gmail.com>
! 4: *
! 5: * Permission to use, copy, modify, and/or distribute this software for any
! 6: * purpose with or without fee is hereby granted, provided that the above
! 7: * copyright notice and this permission notice appear in all copies.
! 8: *
! 9: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
! 10: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
! 11: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
! 12: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
! 13: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
! 14: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
! 15: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
! 16: */
! 17:
! 18: #include <assert.h>
! 19: #include <dirent.h>
! 20: #include <stdio.h>
! 21: #include <stdlib.h>
! 22: #include <string.h>
! 23:
! 24: static const char *matcher_type = NULL;
! 25: static int (*matcher_filter) (const char *file) = NULL;
! 26: static int matcher(const struct dirent *entry)
! 27: {
! 28: char *pos = strrchr(entry->d_name, '.');
! 29:
! 30: if (matcher_filter && !matcher_filter(entry->d_name))
! 31: /* User matcher overrides the rest. */
! 32: return 0;
! 33:
! 34: /* Skip current dir "." from list of files. */
! 35: if ((1 == strlen(entry->d_name) && entry->d_name[0] == '.') ||
! 36: (2 == strlen(entry->d_name) && !strcmp(entry->d_name, "..")))
! 37: return 0;
! 38:
! 39: /* filetype == "" */
! 40: if (matcher_type[0] == 0)
! 41: return 1;
! 42:
! 43: /* Entry has no "." */
! 44: if (!pos)
! 45: return 0;
! 46:
! 47: return !strcmp(pos, matcher_type);
! 48: }
! 49:
! 50: /**
! 51: * dir - List all files of a certain type in the given directory.
! 52: * @dir: Base directory for dir operation.
! 53: * @type: File type suffix, e.g. ".cfg".
! 54: * @filter: Optional file name filter.
! 55: * @list: Pointer to an array of file names.
! 56: * @strip: Flag, if set dir() strips the file type.
! 57: *
! 58: * This function returns a @list of files, matching the @type suffix,
! 59: * in the given directory @dir.
! 60: *
! 61: * The @list argument is a pointer to where to store the dynamically
! 62: * allocated list of file names. This list should be free'd by first
! 63: * calling free() on each file name and then on the list itself.
! 64: *
! 65: * If @filter is not %NULL it will be called for each file found. If
! 66: * @filter returns non-zero the @file argument will be included in the
! 67: * resulting @list. If @filter returns zero for given @file it will
! 68: * be discarded.
! 69: *
! 70: * If the @strip flag is set the resulting @list of files has their
! 71: * file type stripped, including the dot. So a match "config0.cfg"
! 72: * would be returned as "config0".
! 73: *
! 74: * Returns:
! 75: * Number of files in @list, zero if no matching files of @type.
! 76: */
! 77: int dir(const char *dir, const char *type, int (*filter) (const char *file), char ***list, int strip)
! 78: {
! 79: int i, n, num = 0;
! 80: char **files;
! 81: struct dirent **namelist;
! 82:
! 83: assert(list);
! 84:
! 85: if (!dir)
! 86: /* Assuming current directory */
! 87: dir = ".";
! 88: if (!type)
! 89: /* Assuming all files. */
! 90: type = "";
! 91:
! 92: matcher_type = type;
! 93: matcher_filter = filter;
! 94: n = scandir(dir, &namelist, matcher, alphasort);
! 95: if (n < 0) {
! 96: perror("scandir");
! 97: } else if (n > 0) {
! 98: files = (char **)malloc(n * sizeof(char *));
! 99: for (i = 0; i < n; i++) {
! 100: if (files) {
! 101: char *name = namelist[i]->d_name;
! 102: char *type = strrchr(name, '.');
! 103:
! 104: if (type && strip)
! 105: *type = 0;
! 106:
! 107: files[i] = strdup(name);
! 108: num++;
! 109: }
! 110: free(namelist[i]);
! 111: }
! 112: if (num)
! 113: *list = files;
! 114: }
! 115:
! 116: if (namelist)
! 117: free(namelist);
! 118:
! 119: return num;
! 120: }
! 121:
! 122: #ifdef UNITTEST
! 123: #include <errno.h>
! 124: #include <sys/stat.h>
! 125: #include <fcntl.h>
! 126: #include "lite.h"
! 127:
! 128: #define DIR_TYPE_IMAGE ".img"
! 129: #define DIR_TYPE_SYSLOG ""
! 130: #define DIR_TYPE_CONFIG ".cfg"
! 131: #define STARTUP_CONFIG "startup-config.cfg"
! 132:
! 133: int simulate_files(int creat)
! 134: {
! 135: int i;
! 136: char *files[] =
! 137: { "config0.cfg", "config1.cfg", "config2.cfg", "config3.cfg",
! 138: "rr109.img", "rx100.img", "rx107.img", "rm957.img", "messages"
! 139: };
! 140:
! 141: for (i = 0; i < NELEMS(files); i++) {
! 142: if (creat)
! 143: touch(files[i]);
! 144: else
! 145: erase(files[i]);
! 146: }
! 147:
! 148: if (creat)
! 149: symlink("config2.cfg", STARTUP_CONFIG);
! 150: else
! 151: erase(STARTUP_CONFIG);
! 152:
! 153: return 0;
! 154: }
! 155:
! 156: static int cfg_dir_filter(const char *file)
! 157: {
! 158: /* Skip the STARTUP_CONFIG file, it is a symbolic link to the
! 159: * current startup configuration. */
! 160: return ! !strcmp(file, STARTUP_CONFIG);
! 161: }
! 162:
! 163: int main(int argc, char *argv[])
! 164: {
! 165: int i, num;
! 166: char *type = DIR_TYPE_CONFIG;
! 167: char **files;
! 168: int once = 1;
! 169:
! 170: int is_startup_config(const char *entry) {
! 171: static char file[80];
! 172:
! 173: if (once) {
! 174: int len = readlink(STARTUP_CONFIG, file, sizeof(file));
! 175:
! 176: if (len == -1)
! 177: return 0;
! 178:
! 179: file[len] = 0;
! 180: once = 0; /* Only once per call to dir() */
! 181: }
! 182: //printf ("Comparing link %s with entry %s\n", file, entry);
! 183: return !strcmp(file, entry);
! 184: }
! 185:
! 186: simulate_files(1);
! 187:
! 188: if (argc >= 2) {
! 189: if (!strcasecmp("CONFIG", argv[1])) {
! 190: type = DIR_TYPE_CONFIG;
! 191: system("ls -l *" DIR_TYPE_CONFIG);
! 192: }
! 193: if (!strcasecmp("IMAGE", argv[1])) {
! 194: type = DIR_TYPE_IMAGE;
! 195: system("ls -l *" DIR_TYPE_IMAGE);
! 196: }
! 197: if (!strcasecmp("SYSLOG", argv[1])) {
! 198: type = DIR_TYPE_SYSLOG;
! 199: system("ls -l *");
! 200: }
! 201: }
! 202:
! 203: num = dir(NULL, type, cfg_dir_filter, &files, 0);
! 204: if (num) {
! 205: for (i = 0; i < num; i++) {
! 206: printf("%s", files[i]);
! 207: if (is_startup_config(files[i]))
! 208: printf(" --> startup-config");
! 209: printf("\n");
! 210:
! 211: free(files[i]);
! 212: }
! 213: free(files);
! 214: }
! 215:
! 216: simulate_files(0);
! 217:
! 218: return 0;
! 219: }
! 220: #endif /* UNITTEST */
! 221:
! 222: /**
! 223: * Local Variables:
! 224: * compile-command: "make V=1 -f dir.mk"
! 225: * version-control: t
! 226: * indent-tabs-mode: t
! 227: * c-file-style: "linux"
! 228: * End:
! 229: */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to dir.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / pimd / libite