libsysio/tests/test_list.c

   1 /*
   2  *    This Cplant(TM) source code is the property of Sandia National
   3  *    Laboratories.
   4  *
   5  *    This Cplant(TM) source code is copyrighted by Sandia National
   6  *    Laboratories.
   7  *
   8  *    The redistribution of this Cplant(TM) source code is subject to the
   9  *    terms of the GNU Lesser General Public License
  10  *    (see cit/LGPL or http://www.gnu.org/licenses/lgpl.html)
  11  *
  12  *    Cplant(TM) Copyright 1998-2003 Sandia Corporation.
  13  *    Under the terms of Contract DE-AC04-94AL85000, there is a non-exclusive
  14  *    license for use of this work by or on behalf of the US Government.
  15  *    Export of this program may require a license from the United States
  16  *    Government.
  17  */
  18
  19 /*
  20  * This library is free software; you can redistribute it and/or
  21  * modify it under the terms of the GNU Lesser General Public
  22  * License as published by the Free Software Foundation; either
  23  * version 2.1 of the License, or (at your option) any later version.
  24  *
  25  * This library is distributed in the hope that it will be useful,
  26  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  27  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  28  * Lesser General Public License for more details.
  29  *
  30  * You should have received a copy of the GNU Lesser General Public
  31  * License along with this library; if not, write to the Free Software
  32  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  33  *
  34  * Questions or comments about this library should be sent to:
  35  *
  36  * Lee Ward
  37  * Sandia National Laboratories, New Mexico
  38  * P.O. Box 5800
  39  * Albuquerque, NM 87185-1110
  40  *
  41  * lee@sandia.gov
  42  */
  43
  44 #define _BSD_SOURCE
  45
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <string.h>
  49 #include <unistd.h>
  50 #include <errno.h>
  51 #include <sys/types.h>
  52 #include <sys/stat.h>
  53 #include <fcntl.h>
  54 #include <dirent.h>
  55 #include <sys/uio.h>
  56 #include <sys/queue.h>
  57
  58 #include "xtio.h"
  59 #include "test.h"
  60
  61 /*
  62  * Stat files.
  63  *
  64  * Usage: test_list [path...]
  65  *
  66  * Without any path arguments, the program reads from standard-in, dealing with
  67  * each line as an absolute or relative path until EOF.
  68  */
  69
  70 static int listit(const char *path);
  71 static void usage(void);
  72
  73 int
  74 main(int argc, char *const argv[])
  75 {
  76         int     i;
  77         int     err;
  78         int     n;
  79         extern int _test_sysio_startup(void);
  80
  81         /*
  82          * Parse command line arguments.
  83          */
  84         while ((i = getopt(argc, argv, "")) != -1)
  85                 switch (i) {
  86
  87                 default:
  88                         usage();
  89                 }
  90
  91         /*
  92          * Init sysio lib.
  93          */
  94         err = _test_sysio_startup();
  95         if (err) {
  96                 errno = -err;
  97                 perror("sysio startup");
  98                 exit(1);
  99         }
 100
 101         n = argc - optind;
 102
 103         /*
 104          * Try path(s) listed on command-line.
 105          */
 106         while (optind < argc) {
 107                 const char *path;
 108
 109                 path = argv[optind++];
 110                 (void )listit(path);
 111         }
 112
 113         /*
 114          * If no command-line arguments, read from stdin until EOF.
 115          */
 116         if (!n) {
 117                 int     doflush;
 118                 static char buf[4096];
 119                 size_t  len;
 120                 char    *cp;
 121                 char    c;
 122
 123                 doflush = 0;
 124                 while (fgets(buf, sizeof(buf), stdin) != NULL) {
 125                         len = strlen(buf);
 126                         cp = buf + len - 1;
 127                         c = *cp;
 128                         *cp = '\0';
 129                         if (!doflush)
 130                                 listit(buf);
 131                         doflush = c == '\n' ? 0 : 1;
 132                 }
 133         }
 134
 135         /*
 136          * Clean up.
 137          */
 138         _test_sysio_shutdown();
 139
 140         return 0;
 141 }
 142
 143 static int
 144 listit(const char *path)
 145 {
 146         int     fd;
 147         size_t  n;
 148         struct dirent *buf, *dp;
 149         off_t   base;
 150         ssize_t cc;
 151
 152         fd = open(path, O_RDONLY);
 153         if (fd < 0) {
 154                 perror(path);
 155                 return -1;
 156         }
 157
 158         n = 16 * 1024;
 159         buf = malloc(n);
 160         if (!buf) {
 161                 perror(path);
 162                 cc = -1;
 163                 goto out;
 164         }
 165
 166         base = 0;
 167         while ((cc = getdirentries(fd, (char *)buf, n, &base)) > 0) {
 168                 dp = buf;
 169                 while (cc > 0) {
 170                         (void )printf("\t%s: ino %llu type %u\n",
 171                                       dp->d_name,
 172                                       (unsigned long long )dp->d_ino,
 173                                       (int )dp->d_type);
 174                         cc -= dp->d_reclen;
 175                         dp = (struct dirent *)((char *)dp + dp->d_reclen);
 176                 }
 177                 if (!base)
 178                         break;
 179         }
 180
 181 out:
 182         if (cc < 0)
 183                 perror(path);
 184
 185         free(buf);
 186         {
 187                 int     oerrno = errno;
 188
 189                 if (close(fd) != 0) {
 190                         perror(path);
 191                         if (cc < 0)
 192                                 errno = oerrno;
 193                         else
 194                                 cc = -1;
 195                 }
 196         }
 197
 198         return (int )cc;
 199 }
 200
 201 static void
 202 usage()
 203 {
 204
 205         (void )fprintf(stderr,
 206                        "Usage: list_path"
 207                        " [<path> ...\n]");
 208
 209         exit(1);
 210 }