diff options
Diffstat (limited to 'nss/makedb.c')
| -rw-r--r-- | nss/makedb.c | 580 |
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; +} |