/* 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 Lesser General Public License as published by the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with the GNU C Library; 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; /* 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 (int key, char *arg, struct argp_state *state); /* Function to print some extra text in the help message. */ static char *more_help (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 (FILE *input, const char *inname, NSS_DB *output, int to_lowercase, int be_quiet); static int print_database (NSS_DB *db); int main (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]; } /* First load the shared object to initialize version dependend variables. */ if (load_db () != NSS_STATUS_SUCCESS) error (EXIT_FAILURE, 0, gettext ("No usable database library found.")); /* Special handling if we are asked to print the database. */ if (do_undo) { dbopen (input_name, db_rdonly, 0666, &db_file); 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. */ dbopen (output_name, DB_CREATE | db_truncate, mode, &db_file); 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; }