about summary refs log tree commit diff
path: root/db/makedb.c
diff options
context:
space:
mode:
Diffstat (limited to 'db/makedb.c')
-rw-r--r--db/makedb.c325
1 files changed, 325 insertions, 0 deletions
diff --git a/db/makedb.c b/db/makedb.c
new file mode 100644
index 0000000000..b1ed982275
--- /dev/null
+++ b/db/makedb.c
@@ -0,0 +1,325 @@
+/* makedb -- create simple DB database from textual input.
+Copyright (C) 1996 Free Software Foundation, Inc.
+This file is part of the GNU C Library.
+Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.
+
+The GNU C Library is free software; you can redistribute it and/or
+modify it under the terms of the GNU Library General Public License as
+published by the Free Software Foundation; either version 2 of the
+License, or (at your option) any later version.
+
+The GNU C Library 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
+Library General Public License for more details.
+
+You should have received a copy of the GNU Library General Public
+License along with the GNU C Library; see the file COPYING.LIB.  If
+not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+Boston, MA 02111-1307, USA.  */
+
+#include <db.h>
+#include <ctype.h>
+#include <errno.h>
+#include <error.h>
+#include <fcntl.h>
+#include <getopt.h>
+#include <libintl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+/* Get libc version number.  */
+#include "../version.h"
+
+
+/* Long options.  */
+static const struct option long_options[] =
+{
+  { "help", no_argument, NULL, 'h' },
+  { "fold-case", no_argument, NULL, 'f' },
+  { "output", required_argument, NULL, 'o' },
+  { "undo", no_argument, NULL, 'u' },
+  { "version", no_argument, NULL, 'V' },
+  { NULL, }
+};
+
+/* Prototypes for local functions.  */
+static void usage __P ((int status)) __attribute__ ((noreturn));
+static int process_input __P ((FILE *input, const char *inname, DB *output,
+			       int to_lowercase));
+static int print_database __P ((DB *db));
+
+
+int
+main (argc, argv)
+     int argc;
+     char *argv[];
+{
+  const char *output_name;
+  const char *input_name;
+  FILE *input_file;
+  DB *db_file;
+  int do_help;
+  int do_version;
+  int to_lowercase;
+  int do_undo;
+  int status;
+  int opt;
+
+  /* Set locale via LC_ALL.  */
+  setlocale (LC_ALL, "");
+
+  /* Set the text message domain.  */
+  textdomain (_libc_intl_domainname);
+
+  /* Initialize local variables.  */
+  do_help = 0;
+  do_version = 0;
+  to_lowercase = 0;
+  do_undo = 0;
+  output_name = NULL;
+
+  while ((opt = getopt_long (argc, argv, "fho:uV", long_options, NULL)) != EOF)
+    switch (opt)
+      {
+      case '\0':        /* Long option.  */
+        break;
+      case 'h':
+        do_help = 1;
+        break;
+      case 'f':
+	to_lowercase = 1;
+	break;
+      case 'o':
+        output_name = optarg;
+        break;
+      case 'u':
+	do_undo = 1;
+	break;
+      case 'V':
+        do_version = 1;
+        break;
+      default:
+        usage (EXIT_FAILURE);
+      }
+
+  /* Version information is requested.  */
+  if (do_version)
+    printf ("%s - GNU %s %s\n", program_invocation_name, "libc", VERSION);
+
+  /* Help is requested.  */
+  if (do_help)
+    usage (EXIT_SUCCESS);
+  else if (do_version)
+    exit (EXIT_SUCCESS);
+
+  /* Determine file names.  */
+  if (do_undo || output_name != NULL)
+    {
+      if (optind + 1 != argc)
+	{
+	wrong_arguments:
+	  error (0, 0, gettext ("wrong number of arguments"));
+	  usage (EXIT_FAILURE);
+	}
+      input_name = argv[optind];
+    }
+  else
+    {
+      if (optind + 2 != argc)
+	goto wrong_arguments;
+
+      input_name = argv[optind++];
+      output_name = argv[optind];
+    }
+
+  /* Special handling if we are asked to print the database.  */
+  if (do_undo)
+    {
+      db_file = dbopen (input_name, O_RDONLY, 0666, DB_BTREE, NULL);
+      if (db_file == NULL)
+	error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
+	       input_name,
+	       errno == EFTYPE ? gettext ("incorrectly formatted file")
+			       : strerror (errno));
+
+      status = print_database (db_file);
+
+      db_file->close (db_file);
+
+      return status;
+    }
+
+  /* Open input file.  */
+  if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
+    input_file = stdin;
+  else
+    {
+      input_file = fopen (input_name, "r");
+      if (input_file == NULL)
+	error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
+	       input_name);
+    }
+
+  /* Open output file.  This must not be standard output so we don't
+     handle "-" and "/dev/stdout" special.  */
+  db_file = dbopen (output_name, O_CREAT | O_RDWR | O_TRUNC, 0666,
+		    DB_BTREE, NULL);
+  if (db_file == NULL)
+    error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"));
+
+  /* Start the real work.  */
+  status = process_input (input_file, input_name, db_file, to_lowercase);
+
+  /* Close files.  */
+  if (input_file != stdin)
+    fclose (input_file);
+  db_file->close (db_file);
+
+  return status;
+}
+
+
+static void
+usage (status)
+     int status;
+{
+  if (status != EXIT_SUCCESS)
+    fprintf (stderr, gettext ("Try `%s --help' for more information.\n"),
+             program_invocation_name);
+  else
+    printf (gettext ("\
+Usage: %s [OPTION]... INPUT-FILE OUTPUT-FILE\n\
+       %s [OPTION]... -o OUTPUT-FILE INPUT-FILE\n\
+       %s [OPTION]... -u INPUT-FILE\n\
+Mandatory arguments to long options are mandatory for short options too.\n\
+  -f, --fold-case     convert key to lower case\n\
+  -h, --help          display this help and exit\n\
+  -o, --output=NAME   write output to file NAME\n\
+  -u, --undo          print content of database file, one entry a line\n\
+  -V, --version       output version information and exit\n\
+If INPUT-FILE is -, input is read from standard input.\n"),
+	    program_invocation_name, program_invocation_name,
+	    program_invocation_name);
+
+  exit (status);
+}
+
+
+static int
+process_input (input, inname, output, to_lowercase)
+     FILE *input;
+     const char *inname;
+     DB *output;
+     int to_lowercase;
+{
+  char *line;
+  size_t linelen;
+  int status;
+  size_t linenr;
+
+  line = NULL;
+  linelen = 0;
+  status = EXIT_SUCCESS;
+  linenr = 0;
+
+  while (!feof (input))
+    {
+      DBT key;
+      DBT val;
+      char *cp;
+      int n;
+
+      n = getline (&line, &linelen, input);
+      if (n < 0)
+	/* This means end of file or some bug.  */
+	break;
+      if (n == 0)
+	/* Short read.  Probably interrupted system call. */
+	continue;
+
+      ++linenr;
+
+      if (line[n - 1] == '\n')
+	/* Remove trailing newline.  */
+	line[--n] = '\0';
+
+      cp = line;
+      while (isspace (*cp))
+	++cp;
+
+      if (*cp == '#')
+	/* First non-space character in line '#': it's a comment.  */
+	continue;
+
+      key.data = cp;
+      while (*cp != '\0' && !isspace (*cp))
+	{
+	  if (to_lowercase)
+	    *cp = tolower (*cp);
+	  ++cp;
+	}
+
+      if (key.data == cp)
+	/* It's an empty line.  */
+	continue;
+
+      key.size = cp - (char *) key.data;
+
+      while (isspace (*cp))
+	++cp;
+
+      val.data = cp;
+      val.size = &line[n] - cp;
+
+      /* Store the value.  */
+      status = output->put (output, &key, &val, R_NOOVERWRITE);
+      if (status != 0)
+	{
+	  if (status == 1)
+	    error_at_line (0, 0, inname, linenr, gettext ("duplicate key"));
+	  else
+	    error (0, errno, gettext ("while writing data base file"));
+
+	  status = EXIT_FAILURE;
+	  clearerr (input);
+	  break;
+	}
+    }
+
+  if (ferror (input))
+    {
+      error (0, 0, gettext ("problems while reading `%s'"));
+      status = EXIT_FAILURE;
+    }
+
+  return status;
+}
+
+
+static int
+print_database (db)
+     DB *db;
+{
+  DBT key;
+  DBT val;
+  int no_more;
+
+  no_more = db->seq (db, &key, &val, R_FIRST);
+  while (!no_more)
+    {
+      printf ("%.*s %.*s\n", (int) key.size, (char *) key.data, (int) val.size,
+	      (char *) val.data);
+
+      no_more = db->seq (db, &key, &val, R_NEXT);
+    }
+
+  if (no_more == -1)
+    {
+      error (0, errno, gettext ("while reading database"));
+      return EXIT_FAILURE;
+    }
+
+  return EXIT_SUCCESS;
+}