db/makedb.c

   1 /* makedb -- create simple DB database from textual input.
   2 Copyright (C) 1996 Free Software Foundation, Inc.
   3 This file is part of the GNU C Library.
   4 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
   5
   6 The GNU C Library is free software; you can redistribute it and/or
   7 modify it under the terms of the GNU Library General Public License as
   8 published by the Free Software Foundation; either version 2 of the
   9 License, or (at your option) any later version.
  10
  11 The GNU C Library is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 Library General Public License for more details.
  15
  16 You should have received a copy of the GNU Library General Public
  17 License along with the GNU C Library; see the file COPYING.LIB.  If
  18 not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19 Boston, MA 02111-1307, USA.  */
  20
  21 #include <db.h>
  22 #include <ctype.h>
  23 #include <errno.h>
  24 #include <error.h>
  25 #include <fcntl.h>
  26 #include <getopt.h>
  27 #include <libintl.h>
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31
  32 /* Get libc version number.  */
  33 #include "../version.h"
  34
  35
  36 /* Long options.  */
  37 static const struct option long_options[] =
  38 {
  39   { "help", no_argument, NULL, 'h' },
  40   { "fold-case", no_argument, NULL, 'f' },
  41   { "output", required_argument, NULL, 'o' },
  42   { "quiet", no_argument, NULL, 'q' },
  43   { "undo", no_argument, NULL, 'u' },
  44   { "version", no_argument, NULL, 'V' },
  45   { NULL, 0, NULL, 0}
  46 };
  47
  48 /* Prototypes for local functions.  */
  49 static void usage __P ((int status)) __attribute__ ((noreturn));
  50 static int process_input __P ((FILE *input, const char *inname, DB *output,
  51                                int to_lowercase, int be_quiet));
  52 static int print_database __P ((DB *db));
  53
  54
  55 int
  56 main (argc, argv)
  57      int argc;
  58      char *argv[];
  59 {
  60   const char *output_name;
  61   const char *input_name;
  62   FILE *input_file;
  63   DB *db_file;
  64   int do_help;
  65   int do_version;
  66   int to_lowercase;
  67   int do_undo;
  68   int be_quiet;
  69   int status;
  70   int opt;
  71
  72   /* Set locale via LC_ALL.  */
  73   setlocale (LC_ALL, "");
  74
  75   /* Set the text message domain.  */
  76   textdomain (_libc_intl_domainname);
  77
  78   /* Initialize local variables.  */
  79   do_help = 0;
  80   do_version = 0;
  81   to_lowercase = 0;
  82   do_undo = 0;
  83   be_quiet = 0;
  84   output_name = NULL;
  85
  86   while ((opt = getopt_long (argc, argv, "fho:uV", long_options, NULL)) != EOF)
  87     switch (opt)
  88       {
  89       case '\0':        /* Long option.  */
  90         break;
  91       case 'h':
  92         do_help = 1;
  93         break;
  94       case 'f':
  95         to_lowercase = 1;
  96         break;
  97       case 'o':
  98         output_name = optarg;
  99         break;
 100       case 'q':
 101         be_quiet = 1;
 102         break;
 103       case 'u':
 104         do_undo = 1;
 105         break;
 106       case 'V':
 107         do_version = 1;
 108         break;
 109       default:
 110         usage (EXIT_FAILURE);
 111       }
 112
 113   /* Version information is requested.  */
 114   if (do_version)
 115     printf ("%s - GNU %s %s\n", program_invocation_name, "libc", VERSION);
 116
 117   /* Help is requested.  */
 118   if (do_help)
 119     usage (EXIT_SUCCESS);
 120   else if (do_version)
 121     exit (EXIT_SUCCESS);
 122
 123   /* Determine file names.  */
 124   if (do_undo || output_name != NULL)
 125     {
 126       if (optind + 1 != argc)
 127         {
 128         wrong_arguments:
 129           error (0, 0, gettext ("wrong number of arguments"));
 130           usage (EXIT_FAILURE);
 131         }
 132       input_name = argv[optind];
 133     }
 134   else
 135     {
 136       if (optind + 2 != argc)
 137         goto wrong_arguments;
 138
 139       input_name = argv[optind++];
 140       output_name = argv[optind];
 141     }
 142
 143   /* Special handling if we are asked to print the database.  */
 144   if (do_undo)
 145     {
 146       db_file = dbopen (input_name, O_RDONLY, 0666, DB_BTREE, NULL);
 147       if (db_file == NULL)
 148         error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
 149                input_name,
 150                errno == EFTYPE ? gettext ("incorrectly formatted file")
 151                                : strerror (errno));
 152
 153       status = print_database (db_file);
 154
 155       db_file->close (db_file);
 156
 157       return status;
 158     }
 159
 160   /* Open input file.  */
 161   if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
 162     input_file = stdin;
 163   else
 164     {
 165       input_file = fopen (input_name, "r");
 166       if (input_file == NULL)
 167         error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
 168                input_name);
 169     }
 170
 171   /* Open output file.  This must not be standard output so we don't
 172      handle "-" and "/dev/stdout" special.  */
 173   db_file = dbopen (output_name, O_CREAT | O_RDWR | O_TRUNC, 0666,
 174                     DB_BTREE, NULL);
 175   if (db_file == NULL)
 176     error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"));
 177
 178   /* Start the real work.  */
 179   status = process_input (input_file, input_name, db_file, to_lowercase,
 180                           be_quiet);
 181
 182   /* Close files.  */
 183   if (input_file != stdin)
 184     fclose (input_file);
 185   db_file->close (db_file);
 186
 187   return status;
 188 }
 189
 190
 191 static void
 192 usage (status)
 193      int status;
 194 {
 195   if (status != EXIT_SUCCESS)
 196     fprintf (stderr, gettext ("Try `%s --help' for more information.\n"),
 197              program_invocation_name);
 198   else
 199     printf (gettext ("\
 200 Usage: %s [OPTION]... INPUT-FILE OUTPUT-FILE\n\
 201        %s [OPTION]... -o OUTPUT-FILE INPUT-FILE\n\
 202        %s [OPTION]... -u INPUT-FILE\n\
 203 Mandatory arguments to long options are mandatory for short options too.\n\
 204   -f, --fold-case     convert key to lower case\n\
 205   -h, --help          display this help and exit\n\
 206   -o, --output=NAME   write output to file NAME\n\
 207       --quiet         don't print messages while building database\n\
 208   -u, --undo          print content of database file, one entry a line\n\
 209   -V, --version       output version information and exit\n\
 210 If INPUT-FILE is -, input is read from standard input.\n\
 211 Report bugs to <bug-glibc@gnu.ai.mit.edu>.\n"),
 212             program_invocation_name, program_invocation_name,
 213             program_invocation_name);
 214
 215   exit (status);
 216 }
 217
 218
 219 static int
 220 process_input (input, inname, output, to_lowercase, be_quiet)
 221      FILE *input;
 222      const char *inname;
 223      DB *output;
 224      int to_lowercase;
 225      int be_quiet;
 226 {
 227   char *line;
 228   size_t linelen;
 229   int status;
 230   size_t linenr;
 231
 232   line = NULL;
 233   linelen = 0;
 234   status = EXIT_SUCCESS;
 235   linenr = 0;
 236
 237   while (!feof (input))
 238     {
 239       DBT key;
 240       DBT val;
 241       char *cp;
 242       int n;
 243
 244       n = getline (&line, &linelen, input);
 245       if (n < 0)
 246         /* This means end of file or some bug.  */
 247         break;
 248       if (n == 0)
 249         /* Short read.  Probably interrupted system call. */
 250         continue;
 251
 252       ++linenr;
 253
 254       if (line[n - 1] == '\n')
 255         /* Remove trailing newline.  */
 256         line[--n] = '\0';
 257
 258       cp = line;
 259       while (isspace (*cp))
 260         ++cp;
 261
 262       if (*cp == '#')
 263         /* First non-space character in line '#': it's a comment.  */
 264         continue;
 265
 266       key.data = cp;
 267       while (*cp != '\0' && !isspace (*cp))
 268         {
 269           if (to_lowercase)
 270             *cp = tolower (*cp);
 271           ++cp;
 272         }
 273
 274       if (key.data == cp)
 275         /* It's an empty line.  */
 276         continue;
 277
 278       key.size = cp - (char *) key.data;
 279
 280       while (isspace (*cp))
 281         ++cp;
 282
 283       val.data = cp;
 284       val.size = &line[n] - cp;
 285
 286       /* Store the value.  */
 287       status = output->put (output, &key, &val, R_NOOVERWRITE);
 288       if (status != 0)
 289         {
 290           if (status == 1)
 291             {
 292               if (!be_quiet)
 293                 error_at_line (0, 0, inname, linenr,
 294                                gettext ("duplicate key"));
 295               /* This is no real error.  Just give a warning.  */
 296               status = 0;
 297             }
 298           else
 299             error (0, errno, gettext ("while writing data base file"));
 300
 301           status = status ? EXIT_FAILURE : EXIT_SUCCESS;
 302
 303           clearerr (input);
 304           break;
 305         }
 306     }
 307
 308   if (ferror (input))
 309     {
 310       error (0, 0, gettext ("problems while reading `%s'"));
 311       status = EXIT_FAILURE;
 312     }
 313
 314   return status;
 315 }
 316
 317
 318 static int
 319 print_database (db)
 320      DB *db;
 321 {
 322   DBT key;
 323   DBT val;
 324   int no_more;
 325
 326   no_more = db->seq (db, &key, &val, R_FIRST);
 327   while (!no_more)
 328     {
 329       printf ("%.*s %.*s\n", (int) key.size, (char *) key.data, (int) val.size,
 330               (char *) val.data);
 331
 332       no_more = db->seq (db, &key, &val, R_NEXT);
 333     }
 334
 335   if (no_more == -1)
 336     {
 337       error (0, errno, gettext ("while reading database"));
 338       return EXIT_FAILURE;
 339     }
 340
 341   return EXIT_SUCCESS;
 342 }