aboutsummaryrefslogtreecommitdiff
path: root/nss/makedb.c
diff options
context:
space:
mode:
authorUlrich Drepper <drepper@redhat.com>2000-01-02 08:45:58 +0000
committerUlrich Drepper <drepper@redhat.com>2000-01-02 08:45:58 +0000
commit793bd4d9ebc5198ac292437227bec133fc6f4dfc (patch)
treefcac05d3fe6bd9e1e8ac94df7f3450929640ad12 /nss/makedb.c
parent8d6f1731fcd082e4f744ba9cb4bde4be7c08f1b3 (diff)
downloadglibc-793bd4d9ebc5198ac292437227bec133fc6f4dfc.tar
glibc-793bd4d9ebc5198ac292437227bec133fc6f4dfc.tar.gz
glibc-793bd4d9ebc5198ac292437227bec133fc6f4dfc.tar.bz2
glibc-793bd4d9ebc5198ac292437227bec133fc6f4dfc.zip
Update.
2000-01-02 Ulrich Drepper <drepper@cygnus.com> * nss/nss_db/nss_db.h: Add definitions needed for makedb. * nss/nss_db/dummy-db.h: Likewise. * nss/makedb.c: New file. Copied from file in db2, modified to not depend on being linked against libdb.
Diffstat (limited to 'nss/makedb.c')
-rw-r--r--nss/makedb.c580
1 files changed, 580 insertions, 0 deletions
diff --git a/nss/makedb.c b/nss/makedb.c
new file mode 100644
index 0000000000..d7662585e2
--- /dev/null
+++ b/nss/makedb.c
@@ -0,0 +1,580 @@
+/* Create simple DB database from textual input.
+ Copyright (C) 1996, 1997, 1998, 1999, 2000 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 <argp.h>
+#include <ctype.h>
+#include <dlfcn.h>
+#include <errno.h>
+#include <error.h>
+#include <fcntl.h>
+#include <libintl.h>
+#include <locale.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include "nss_db/dummy-db.h"
+
+/* Get libc version number. */
+#include "../version.h"
+
+#define PACKAGE _libc_intl_domainname
+
+/* If non-zero convert key to lower case. */
+static int to_lowercase;
+
+/* If non-zero print content of input file, one entry per line. */
+static int do_undo;
+
+/* If non-zero do not print informational messages. */
+static int be_quiet;
+
+/* Name of output file. */
+static const char *output_name;
+
+/* Various definitions for the libdb handling. */
+enum {
+ nodb,
+ db24,
+ db27
+} libdb_version;
+static int (*db_open) (const char *, int,
+ uint32_t, int, void *, void *, void **);
+
+/* Constants which vary from version to version are actually variables
+ here. */
+static int db_first;
+static int db_next;
+static int db_nooverwrite;
+static int db_truncate;
+
+
+/* Name and version of program. */
+static void print_version (FILE *stream, struct argp_state *state);
+void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version;
+
+/* Definitions of arguments for argp functions. */
+static const struct argp_option options[] =
+{
+ { "fold-case", 'f', NULL, 0, N_("Convert key to lower case") },
+ { "output", 'o', N_("NAME"), 0, N_("Write output to file NAME") },
+ { "quiet", 'q', NULL, 0,
+ N_("Do not print messages while building database") },
+ { "undo", 'u', NULL, 0,
+ N_("Print content of database file, one entry a line") },
+ { NULL, 0, NULL, 0, NULL }
+};
+
+/* Short description of program. */
+static const char doc[] = N_("Create simple DB database from textual input.");
+
+/* Strings for arguments in help texts. */
+static const char args_doc[] = N_("\
+INPUT-FILE OUTPUT-FILE\n-o OUTPUT-FILE INPUT-FILE\n-u INPUT-FILE");
+
+/* Prototype for option handler. */
+static error_t parse_opt __P ((int key, char *arg, struct argp_state *state));
+
+/* Function to print some extra text in the help message. */
+static char *more_help __P ((int key, const char *text, void *input));
+
+/* Data structure to communicate with argp functions. */
+static struct argp argp =
+{
+ options, parse_opt, args_doc, doc, NULL, more_help
+};
+
+
+/* Prototypes for local functions. */
+static int process_input __P ((FILE *input, const char *inname, NSS_DB *output,
+ int to_lowercase, int be_quiet));
+static int print_database __P ((NSS_DB *db));
+static NSS_DB *dbopen (const char *fname, int oper, int mode);
+int main __P ((int argc, char *argv[]));
+
+
+int
+main (argc, argv)
+ int argc;
+ char *argv[];
+{
+ const char *input_name;
+ FILE *input_file;
+ NSS_DB *db_file;
+ int status;
+ int remaining;
+ int mode = 0666;
+
+ /* Set locale via LC_ALL. */
+ setlocale (LC_ALL, "");
+
+ /* Set the text message domain. */
+ textdomain (_libc_intl_domainname);
+
+ /* Initialize local variables. */
+ input_name = NULL;
+
+ /* Parse and process arguments. */
+ argp_parse (&argp, argc, argv, 0, &remaining, NULL);
+
+ /* Determine file names. */
+ if (do_undo || output_name != NULL)
+ {
+ if (remaining + 1 != argc)
+ {
+ wrong_arguments:
+ error (0, 0, gettext ("wrong number of arguments"));
+ argp_help (&argp, stdout, ARGP_HELP_SEE,
+ program_invocation_short_name);
+ exit (1);
+ }
+ input_name = argv[remaining];
+ }
+ else
+ {
+ if (remaining + 2 != argc)
+ goto wrong_arguments;
+
+ input_name = argv[remaining++];
+ output_name = argv[remaining];
+ }
+
+ /* Special handling if we are asked to print the database. */
+ if (do_undo)
+ {
+ db_file = dbopen (input_name, DB_RDONLY, 0666);
+ if (db_file == NULL)
+ error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
+ input_name,
+ (errno == EINVAL ? gettext ("incorrectly formatted file")
+ : strerror (errno)));
+
+ status = print_database (db_file);
+
+ db_file->close (db_file->db, 0);
+
+ return status;
+ }
+
+ /* Open input file. */
+ if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
+ input_file = stdin;
+ else
+ {
+ struct stat st;
+
+ input_file = fopen (input_name, "r");
+ if (input_file == NULL)
+ error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
+ input_name);
+
+ /* Get the access rights from the source file. The output file should
+ have the same. */
+ if (fstat (fileno (input_file), &st) >= 0)
+ mode = st.st_mode & ACCESSPERMS;
+ }
+
+ /* Open output file. This must not be standard output so we don't
+ handle "-" and "/dev/stdout" special. */
+ db_file = dbopen (output_name, DB_CREATE | db_truncate, mode);
+ if (db_file == NULL)
+ error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"),
+ output_name);
+
+ /* Start the real work. */
+ status = process_input (input_file, input_name, db_file, to_lowercase,
+ be_quiet);
+
+ /* Close files. */
+ if (input_file != stdin)
+ fclose (input_file);
+ db_file->close (db_file->db, 0);
+
+ return status;
+}
+
+
+/* Handle program arguments. */
+static error_t
+parse_opt (int key, char *arg, struct argp_state *state)
+{
+ switch (key)
+ {
+ case 'f':
+ to_lowercase = 1;
+ break;
+ case 'o':
+ output_name = arg;
+ break;
+ case 'q':
+ be_quiet = 1;
+ break;
+ case 'u':
+ do_undo = 1;
+ break;
+ default:
+ return ARGP_ERR_UNKNOWN;
+ }
+ return 0;
+}
+
+
+static char *
+more_help (int key, const char *text, void *input)
+{
+ switch (key)
+ {
+ case ARGP_KEY_HELP_EXTRA:
+ /* We print some extra information. */
+ return strdup (gettext ("\
+Report bugs using the `glibcbug' script to <bugs@gnu.org>.\n"));
+ default:
+ break;
+ }
+ return (char *) text;
+}
+
+/* Print the version information. */
+static void
+print_version (FILE *stream, struct argp_state *state)
+{
+ fprintf (stream, "makedb (GNU %s) %s\n", PACKAGE, VERSION);
+ fprintf (stream, gettext ("\
+Copyright (C) %s Free Software Foundation, Inc.\n\
+This is free software; see the source for copying conditions. There is NO\n\
+warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\
+"), "2000");
+ fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper");
+}
+
+
+static int
+process_input (input, inname, output, to_lowercase, be_quiet)
+ FILE *input;
+ const char *inname;
+ NSS_DB *output;
+ int to_lowercase;
+ int be_quiet;
+{
+ 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;
+ key.flags = 0;
+
+ while (isspace (*cp))
+ ++cp;
+
+ val.data = cp;
+ val.size = (&line[n] - cp) + 1;
+ val.flags = 0;
+
+ /* Store the value. */
+ status = output->put (output->db, NULL, &key, &val, db_nooverwrite);
+ if (status != 0)
+ {
+ if (status == DB_KEYEXIST)
+ {
+ if (!be_quiet)
+ error_at_line (0, 0, inname, linenr,
+ gettext ("duplicate key"));
+ /* This is no real error. Just give a warning. */
+ status = 0;
+ continue;
+ }
+ else
+ error (0, status, gettext ("while writing database file"));
+
+ status = EXIT_FAILURE;
+
+ clearerr (input);
+ break;
+ }
+ }
+
+ if (ferror (input))
+ {
+ error (0, 0, gettext ("problems while reading `%s'"), inname);
+ status = EXIT_FAILURE;
+ }
+
+ return status;
+}
+
+
+static int
+print_database (db)
+ NSS_DB *db;
+{
+ DBT key;
+ DBT val;
+ NSS_DBC *cursor;
+ int status;
+
+ status = db->cursor (db->db, NULL, &cursor);
+ if (status != 0)
+ {
+ error (0, status, gettext ("while reading database"));
+ return EXIT_FAILURE;
+ }
+
+ key.flags = 0;
+ val.flags = 0;
+ status = cursor->c_get (cursor->cursor, &key, &val, db_first);
+ while (status == 0)
+ {
+ printf ("%.*s %s\n", (int) key.size, (char *) key.data,
+ (char *) val.data);
+
+ status = cursor->c_get (cursor->cursor, &key, &val, db_next);
+ }
+
+ if (status != DB_NOTFOUND)
+ {
+ error (0, status, gettext ("while reading database"));
+ return EXIT_FAILURE;
+ }
+
+ return EXIT_SUCCESS;
+}
+
+
+static int
+load_db (void)
+{
+ static const char *libnames[] = { "libdb.so.3" };
+ int x;
+ void *handle;
+
+ for(x = 0; x < sizeof (libnames) / sizeof (libnames[0]); ++x)
+ {
+ handle = dlopen (libnames[x], RTLD_LAZY);
+ if (handle == NULL)
+ continue;
+
+ db_open = dlsym (handle, "db_open");
+ if (db_open != NULL)
+ {
+ /* Alright, we got a library. Now find out which version it is. */
+ const char *(*db_version) (int *, int *, int *);
+
+ db_version = dlsym (handle, "db_version");
+ if (db_version != NULL)
+ {
+ /* Call the function and get the information. */
+ int major, minor, subminor;
+
+ DL_CALL_FCT (db_version, (&major, &minor, &subminor));
+ if (major == 2)
+ {
+ /* We currently cannot handle other versions than the
+ 2.x series. */
+ if (minor < 6 || (minor == 6 && subminor < 4))
+ {
+ libdb_version = db24;
+ db_first = DB24_FIRST;
+ db_next = DB24_NEXT;
+ db_nooverwrite = DB24_NOOVERWRITE;
+ db_truncate = DB24_TRUNCATE;
+ }
+ else
+ {
+ libdb_version = db27;
+ db_first = DB27_FIRST;
+ db_next = DB27_NEXT;
+ db_nooverwrite = DB27_NOOVERWRITE;
+ db_truncate = DB27_TRUNCATE;
+ }
+ }
+ }
+
+ if (libdb_version != nodb)
+ return 0;
+ }
+
+ dlclose (handle);
+ }
+
+ (void) dlerror ();
+ return 1;
+}
+
+
+static int
+db_cursor (void *db, void *txn, NSS_DBC **dbcp)
+{
+ void *ptr;
+ NSS_DBC *dbc = NULL;
+ int ret;
+
+ switch (libdb_version)
+ {
+ case db24:
+ ret = ((struct db24 *) db)->cursor (db, txn, &ptr);
+ break;
+ case db27:
+ ret = ((struct db27 *) db)->cursor (db, txn, &ptr, 0);
+ break;
+ default:
+ abort ();
+ }
+
+ if (ret == 0)
+ {
+ dbc = (NSS_DBC *) malloc (sizeof (NSS_DBC));
+ if (dbc == NULL)
+ error (EXIT_FAILURE, errno, gettext ("while reading database"));
+ dbc->cursor = ptr;
+
+ switch (libdb_version)
+ {
+ case db24:
+ dbc->c_get =
+ (int (*) (void *, void *, void *, uint32_t))
+ ((struct dbc24 *) ptr)->c_get;
+ break;
+ case db27:
+ dbc->c_get =
+ (int (*) (void *, void *, void *, uint32_t))
+ ((struct dbc27 *) ptr)->c_get;
+ break;
+ default:
+ abort ();
+ }
+ }
+
+ *dbcp = dbc;
+
+ return ret;
+}
+
+
+static NSS_DB *
+dbopen (const char *fname, int oper, int mode)
+{
+ int err;
+ void *odb;
+ NSS_DB *db;
+
+ /* First load the shared object. */
+ load_db ();
+
+ if (db_open == NULL)
+ return NULL;
+
+ /* Actually open the database. */
+ err = DL_CALL_FCT (db_open, (fname, DB_BTREE, oper, mode, NULL, NULL, &odb));
+ if (err != 0)
+ {
+ errno = err;
+ return NULL;
+ }
+
+ /* Construct the object we pass up. */
+ db = (NSS_DB *) malloc (sizeof (NSS_DB));
+ if (db != NULL)
+ {
+ db->db = odb;
+
+ /* The functions are at different positions for the different
+ versions. Sigh. */
+ switch (libdb_version)
+ {
+ case db24:
+ db->close =
+ (int (*) (void *, uint32_t)) ((struct db24 *) odb)->close;
+ db->fd =
+ (int (*) (void *, int *)) ((struct db24 *) odb)->fd;
+ db->get =
+ (int (*) (void *, void *, void *, void *, uint32_t))
+ ((struct db24 *) odb)->get;
+ db->put =
+ (int (*) (void *, void *, void *, void *, uint32_t))
+ ((struct db24 *) odb)->put;
+ break;
+ case db27:
+ db->close =
+ (int (*) (void *, uint32_t)) ((struct db27 *) odb)->close;
+ db->fd =
+ (int (*) (void *, int *)) ((struct db27 *) odb)->fd;
+ db->get =
+ (int (*) (void *, void *, void *, void *, uint32_t))
+ ((struct db27 *) odb)->get;
+ db->put =
+ (int (*) (void *, void *, void *, void *, uint32_t))
+ ((struct db27 *) odb)->put;
+ break;
+ default:
+ abort ();
+ }
+ db->cursor = db_cursor;
+ }
+
+ return db;
+}