db/makedb.c

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