/
githubmirror
/
gawk
Обзор
Документация
Войти
/
githubmirror
/
gawk
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
extension/readdir.c
343 строки
7 KB
Arnold D. Robbins
API cleanups and doc additions.
14 окт 2012, 20:56
14 окт 2012, 20:56
a892293
Код
Авторство
О чём код?
/* * readdir.c --- Provide an input parser to read directories * * Arnold Robbins * arnold@skeeve.com * Written 7/2012 * * Andrew Schorr and Arnold Robbins: further fixes 8/2012. */ /* * Copyright (C) 2012 the Free Software Foundation, Inc. * * This file is part of GAWK, the GNU implementation of the * AWK Programming Language. * * GAWK is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 3 of the License, or * (at your option) any later version. * * GAWK is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA */ #ifdef HAVE_CONFIG_H #include <config.h> #endif #define _BSD_SOURCE #include <stdio.h> #include <errno.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/types.h> #include <sys/stat.h> #ifdef HAVE_DIRENT_H #include <dirent.h> #else #error Cannot compile the dirent extension on this system! #endif #include "gawkapi.h" #include "gettext.h" #define _(msgid) gettext(msgid) #define N_(msgid) msgid static const gawk_api_t *api; /* for convenience macros to work */ static awk_ext_id_t *ext_id; static const char *ext_version = "readdir extension: version 1.0"; static awk_bool_t init_readdir(void); static awk_bool_t (*init_func)(void) = init_readdir; int plugin_is_GPL_compatible; enum { NEVER_DO_INFO, USE_DIRENT_INFO, USE_STAT_INFO } do_ftype = #ifdef DT_BLK USE_DIRENT_INFO #else NEVER_DO_INFO #endif ; /* ftype --- return type of file as a single character string */ static const char * ftype(struct dirent *entry) { struct stat sbuf; #ifdef DT_BLK switch (entry->d_type) { case DT_BLK: return "b"; case DT_CHR: return "c"; case DT_DIR: return "d"; case DT_FIFO: return "p"; case DT_LNK: return "l"; case DT_REG: return "f"; case DT_SOCK: return "s"; default: case DT_UNKNOWN: /* * Could be that filesystem doesn't support d_type, * even if the OS does. (E.g., XFS on GNU/Linux). * So let lstat() do it. */ break; } #endif if (do_ftype < USE_STAT_INFO) /* * Avoid "/u" since user did not insist on file type info, * and it does not seem to be supported by dirent on this * filesystem. */ return NULL; /* Should we set ERRNO here? */ if (lstat(entry->d_name, & sbuf) < 0) return "u"; switch (sbuf.st_mode & S_IFMT) { case S_IFREG: return "f"; case S_IFBLK: return "b"; case S_IFCHR: return "c"; case S_IFDIR: return "d"; #ifdef S_IFSOCK case S_IFSOCK: return "s"; #endif #ifdef S_IFIFO case S_IFIFO: return "p"; #endif #ifdef S_IFLNK case S_IFLNK: return "l"; #endif #ifdef S_IFDOOR /* Solaris weirdness */ case S_IFDOOR: return "D"; #endif /* S_IFDOOR */ } return "u"; } /* data type for the opaque pointer: */ typedef struct open_directory { DIR *dp; char *buf; } open_directory_t; /* dir_get_record --- get one record at a time out of a directory */ static int dir_get_record(char **out, awk_input_buf_t *iobuf, int *errcode, char **rt_start, size_t *rt_len) { DIR *dp; struct dirent *dirent; int len; open_directory_t *the_dir; /* * The caller sets *errcode to 0, so we should set it only if an * error occurs. */ if (out == NULL || iobuf == NULL || iobuf->opaque == NULL) return EOF; the_dir = (open_directory_t *) iobuf->opaque; dp = the_dir->dp; /* * Initialize errno, since readdir does not set it to zero on EOF. */ errno = 0; dirent = readdir(dp); if (dirent == NULL) { *errcode = errno; /* in case there was an error */ return EOF; } #ifdef ZOS_USS len = sprintf(the_dir->buf, "%lu/%s", (unsigned long) dirent->d_ino, dirent->d_name); #else len = sprintf(the_dir->buf, "%llu/%s", (unsigned long long) dirent->d_ino, dirent->d_name); #endif if (do_ftype != NEVER_DO_INFO) { const char *ftstr = ftype(dirent); if (ftstr) len += sprintf(the_dir->buf + len, "/%s", ftstr); } *out = the_dir->buf; *rt_len = 0; /* set RT to "" */ return len; } /* dir_close --- close up when done */ static void dir_close(awk_input_buf_t *iobuf) { open_directory_t *the_dir; if (iobuf == NULL || iobuf->opaque == NULL) return; the_dir = (open_directory_t *) iobuf->opaque; closedir(the_dir->dp); free(the_dir->buf); free(the_dir); iobuf->fd = -1; } /* dir_can_take_file --- return true if we want the file */ static awk_bool_t dir_can_take_file(const awk_input_buf_t *iobuf) { if (iobuf == NULL) return 0; return (iobuf->fd != INVALID_HANDLE && S_ISDIR(iobuf->sbuf.st_mode)); } /* * dir_take_control_of --- set up input parser. * We can assume that dir_can_take_file just returned true, * and no state has changed since then. */ static awk_bool_t dir_take_control_of(awk_input_buf_t *iobuf) { DIR *dp; open_directory_t *the_dir; size_t size; errno = 0; #ifdef HAVE_FDOPENDIR dp = fdopendir(iobuf->fd); #else dp = opendir(iobuf->name); if (dp != NULL) iobuf->fd = dirfd(dp); #endif if (dp == NULL) { warning(ext_id, _("dir_take_control_of: opendir/fdopendir failed: %s"), strerror(errno)); update_ERRNO_int(errno); return 0; } emalloc(the_dir, open_directory_t *, sizeof(open_directory_t), "dir_take_control_of"); the_dir->dp = dp; size = sizeof(struct dirent) + 21 /* max digits in inode */ + 2 /* slashes */; emalloc(the_dir->buf, char *, size, "dir_take_control_of"); iobuf->opaque = the_dir; iobuf->get_record = dir_get_record; iobuf->close_func = dir_close; return 1; } static awk_input_parser_t readdir_parser = { "readdir", dir_can_take_file, dir_take_control_of, NULL }; #ifdef TEST_DUPLICATE static awk_input_parser_t readdir_parser2 = { "readdir2", dir_can_take_file, dir_take_control_of, NULL }; #endif /* init_readdir --- set things ups */ static awk_bool_t init_readdir() { register_input_parser(& readdir_parser); #ifdef TEST_DUPLICATE register_input_parser(& readdir_parser2); #endif return 1; } /* do_readdir_do_ftype --- enable / disable ftype where need to do stat */ static awk_value_t * do_readdir_do_ftype(int nargs, awk_value_t *result) { awk_value_t flag; make_number(1.0, result); if (nargs < 1) { warning(ext_id, _("readdir_do_ftype: called with no arguments")); update_ERRNO_int(EINVAL); make_number(0.0, result); goto out; } else if (do_lint && nargs > 3) lintwarn(ext_id, _("readdir_do_ftype: called with more than one argument")); if (! get_argument(0, AWK_STRING, & flag)) { warning(ext_id, _("readdir_do_ftype: could not get argument")); update_ERRNO_int(EINVAL); make_number(0.0, result); goto out; } if (strcmp(flag.str_value.str, "never") == 0) do_ftype = NEVER_DO_INFO; else if (strcmp(flag.str_value.str, "dirent") == 0) do_ftype = USE_DIRENT_INFO; else if (strcmp(flag.str_value.str, "stat") == 0) do_ftype = USE_STAT_INFO; else { update_ERRNO_int(EINVAL); make_number(0.0, result); } out: return result; } static awk_ext_func_t func_table[] = { { "readdir_do_ftype", do_readdir_do_ftype, 1 }, }; /* define the dl_load function using the boilerplate macro */ dl_load_func(func_table, readdir, "")