db2/makedb.c

   1 /* Create simple DB database from textual input.
   2    Copyright (C) 1996, 1997, 1998, 1999 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 <argp.h>
  22 #include <ctype.h>
  23 #include <db.h>
  24 #include <errno.h>
  25 #include <error.h>
  26 #include <fcntl.h>
  27 #include <libintl.h>
  28 #include <locale.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <string.h>
  32 #include <sys/stat.h>
  33
  34 /* Get libc version number.  */
  35 #include "../version.h"
  36
  37 #define PACKAGE _libc_intl_domainname
  38
  39 /* If non-zero convert key to lower case.  */
  40 static int to_lowercase;
  41
  42 /* If non-zero print content of input file, one entry per line.  */
  43 static int do_undo;
  44
  45 /* If non-zero do not print informational messages.  */
  46 static int be_quiet;
  47
  48 /* Name of output file.  */
  49 static const char *output_name;
  50
  51 /* Name and version of program.  */
  52 static void print_version (FILE *stream, struct argp_state *state);
  53 void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version;
  54
  55 /* Definitions of arguments for argp functions.  */
  56 static const struct argp_option options[] =
  57 {
  58   { "fold-case", 'f', NULL, 0, N_("Convert key to lower case") },
  59   { "output", 'o', N_("NAME"), 0, N_("Write output to file NAME") },
  60   { "quiet", 'q', NULL, 0,
  61     N_("Do not print messages while building database") },
  62   { "undo", 'u', NULL, 0,
  63     N_("Print content of database file, one entry a line") },
  64   { NULL, 0, NULL, 0, NULL }
  65 };
  66
  67 /* Short description of program.  */
  68 static const char doc[] = N_("Create simple DB database from textual input.");
  69
  70 /* Strings for arguments in help texts.  */
  71 static const char args_doc[] = N_("\
  72 INPUT-FILE OUTPUT-FILE\n-o OUTPUT-FILE INPUT-FILE\n-u INPUT-FILE");
  73
  74 /* Prototype for option handler.  */
  75 static error_t parse_opt __P ((int key, char *arg, struct argp_state *state));
  76
  77 /* Function to print some extra text in the help message.  */
  78 static char *more_help __P ((int key, const char *text, void *input));
  79
  80 /* Data structure to communicate with argp functions.  */
  81 static struct argp argp =
  82 {
  83   options, parse_opt, args_doc, doc, NULL, more_help
  84 };
  85
  86
  87 /* Prototypes for local functions.  */
  88 static int process_input __P ((FILE *input, const char *inname, DB *output,
  89                                int to_lowercase, int be_quiet));
  90 static int print_database __P ((DB *db));
  91 int main __P ((int argc, char *argv[]));
  92
  93
  94 int
  95 main (argc, argv)
  96      int argc;
  97      char *argv[];
  98 {
  99   const char *input_name;
 100   FILE *input_file;
 101   DB *db_file;
 102   int status;
 103   int remaining;
 104   int mode = 0666;
 105
 106   /* Set locale via LC_ALL.  */
 107   setlocale (LC_ALL, "");
 108
 109   /* Set the text message domain.  */
 110   textdomain (_libc_intl_domainname);
 111
 112   /* Initialize local variables.  */
 113   input_name = NULL;
 114
 115   /* Parse and process arguments.  */
 116   argp_parse (&argp, argc, argv, 0, &remaining, NULL);
 117
 118   /* Determine file names.  */
 119   if (do_undo || output_name != NULL)
 120     {
 121       if (remaining + 1 != argc)
 122         {
 123         wrong_arguments:
 124           error (0, 0, gettext ("wrong number of arguments"));
 125           argp_help (&argp, stdout, ARGP_HELP_SEE,
 126                      program_invocation_short_name);
 127           exit (1);
 128         }
 129       input_name = argv[remaining];
 130     }
 131   else
 132     {
 133       if (remaining + 2 != argc)
 134         goto wrong_arguments;
 135
 136       input_name = argv[remaining++];
 137       output_name = argv[remaining];
 138     }
 139
 140   /* Special handling if we are asked to print the database.  */
 141   if (do_undo)
 142     {
 143       status = db_open (input_name, DB_BTREE, DB_RDONLY, 0666, NULL, NULL,
 144                         &db_file);
 145       if (status != 0)
 146         error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
 147                input_name,
 148                (status == EINVAL ? gettext ("incorrectly formatted file")
 149                 : strerror (status)));
 150
 151       status = print_database (db_file);
 152
 153       db_file->close (db_file, 0);
 154
 155       return status;
 156     }
 157
 158   /* Open input file.  */
 159   if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
 160     input_file = stdin;
 161   else
 162     {
 163       struct stat st;
 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       /* Get the access rights from the source file.  The output file should
 171          have the same.  */
 172       if (fstat (fileno (input_file), &st) >= 0)
 173         mode = st.st_mode & ACCESSPERMS;
 174     }
 175
 176   /* Open output file.  This must not be standard output so we don't
 177      handle "-" and "/dev/stdout" special.  */
 178   status = db_open (output_name, DB_BTREE, DB_CREATE | DB_TRUNCATE, mode,
 179                     NULL, NULL, &db_file);
 180   if (status != 0)
 181     error (EXIT_FAILURE, status, gettext ("cannot open output file `%s'"),
 182            output_name);
 183
 184   /* Start the real work.  */
 185   status = process_input (input_file, input_name, db_file, to_lowercase,
 186                           be_quiet);
 187
 188   /* Close files.  */
 189   if (input_file != stdin)
 190     fclose (input_file);
 191   db_file->close (db_file, 0);
 192
 193   return status;
 194 }
 195
 196
 197 /* Handle program arguments.  */
 198 static error_t
 199 parse_opt (int key, char *arg, struct argp_state *state)
 200 {
 201   switch (key)
 202     {
 203     case 'f':
 204       to_lowercase = 1;
 205       break;
 206     case 'o':
 207       output_name = arg;
 208       break;
 209     case 'q':
 210       be_quiet = 1;
 211       break;
 212     case 'u':
 213       do_undo = 1;
 214       break;
 215     default:
 216       return ARGP_ERR_UNKNOWN;
 217     }
 218   return 0;
 219 }
 220
 221
 222 static char *
 223 more_help (int key, const char *text, void *input)
 224 {
 225   switch (key)
 226     {
 227     case ARGP_KEY_HELP_EXTRA:
 228       /* We print some extra information.  */
 229       return strdup (gettext ("\
 230 Report bugs using the `glibcbug' script to <bugs@gnu.org>.\n"));
 231     default:
 232       break;
 233     }
 234   return (char *) text;
 235 }
 236
 237 /* Print the version information.  */
 238 static void
 239 print_version (FILE *stream, struct argp_state *state)
 240 {
 241   fprintf (stream, "makedb (GNU %s) %s\n", PACKAGE, VERSION);
 242   fprintf (stream, gettext ("\
 243 Copyright (C) %s Free Software Foundation, Inc.\n\
 244 This is free software; see the source for copying conditions.  There is NO\n\
 245 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\
 246 "), "1999");
 247   fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper");
 248 }
 249
 250
 251 static int
 252 process_input (input, inname, output, to_lowercase, be_quiet)
 253      FILE *input;
 254      const char *inname;
 255      DB *output;
 256      int to_lowercase;
 257      int be_quiet;
 258 {
 259   char *line;
 260   size_t linelen;
 261   int status;
 262   size_t linenr;
 263
 264   line = NULL;
 265   linelen = 0;
 266   status = EXIT_SUCCESS;
 267   linenr = 0;
 268
 269   while (!feof (input))
 270     {
 271       DBT key;
 272       DBT val;
 273       char *cp;
 274       int n;
 275
 276       n = getline (&line, &linelen, input);
 277       if (n < 0)
 278         /* This means end of file or some bug.  */
 279         break;
 280       if (n == 0)
 281         /* Short read.  Probably interrupted system call. */
 282         continue;
 283
 284       ++linenr;
 285
 286       if (line[n - 1] == '\n')
 287         /* Remove trailing newline.  */
 288         line[--n] = '\0';
 289
 290       cp = line;
 291       while (isspace (*cp))
 292         ++cp;
 293
 294       if (*cp == '#')
 295         /* First non-space character in line '#': it's a comment.  */
 296         continue;
 297
 298       key.data = cp;
 299       while (*cp != '\0' && !isspace (*cp))
 300         {
 301           if (to_lowercase)
 302             *cp = tolower (*cp);
 303           ++cp;
 304         }
 305
 306       if (key.data == cp)
 307         /* It's an empty line.  */
 308         continue;
 309
 310       key.size = cp - (char *) key.data;
 311       key.flags = 0;
 312
 313       while (isspace (*cp))
 314         ++cp;
 315
 316       val.data = cp;
 317       val.size = (&line[n] - cp) + 1;
 318       val.flags = 0;
 319
 320       /* Store the value.  */
 321       status = output->put (output, NULL, &key, &val, DB_NOOVERWRITE);
 322       if (status != 0)
 323         {
 324           if (status == DB_KEYEXIST)
 325             {
 326               if (!be_quiet)
 327                 error_at_line (0, 0, inname, linenr,
 328                                gettext ("duplicate key"));
 329               /* This is no real error.  Just give a warning.  */
 330               status = 0;
 331               continue;
 332             }
 333           else
 334             error (0, status, gettext ("while writing database file"));
 335
 336           status = EXIT_FAILURE;
 337
 338           clearerr (input);
 339           break;
 340         }
 341     }
 342
 343   if (ferror (input))
 344     {
 345       error (0, 0, gettext ("problems while reading `%s'"), inname);
 346       status = EXIT_FAILURE;
 347     }
 348
 349   return status;
 350 }
 351
 352
 353 static int
 354 print_database (db)
 355      DB *db;
 356 {
 357   DBT key;
 358   DBT val;
 359   DBC *cursor;
 360   int status;
 361
 362   status = db->cursor (db, NULL, &cursor);
 363   if (status != 0)
 364     {
 365       error (0, status, gettext ("while reading database"));
 366       return EXIT_FAILURE;
 367     }
 368
 369   key.flags = 0;
 370   val.flags = 0;
 371   status = cursor->c_get (cursor, &key, &val, DB_FIRST);
 372   while (status == 0)
 373     {
 374       printf ("%.*s %s\n", (int) key.size, (char *) key.data,
 375               (char *) val.data);
 376
 377       status = cursor->c_get (cursor, &key, &val, DB_NEXT);
 378     }
 379
 380   if (status != DB_NOTFOUND)
 381     {
 382       error (0, status, gettext ("while reading database"));
 383       return EXIT_FAILURE;
 384     }
 385
 386   return EXIT_SUCCESS;
 387 }