Generate database from textual representation.
[kopensolaris-gnu/glibc.git] / nss / makedb.c
1 /* Create simple DB database from textual input.
2    Copyright (C) 1996, 1997, 1998, 1999, 2000 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 <dlfcn.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 <stdint.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <sys/stat.h>
34 #include "nss_db/dummy-db.h"
35
36 /* Get libc version number.  */
37 #include "../version.h"
38
39 #define PACKAGE _libc_intl_domainname
40
41 /* If non-zero convert key to lower case.  */
42 static int to_lowercase;
43
44 /* If non-zero print content of input file, one entry per line.  */
45 static int do_undo;
46
47 /* If non-zero do not print informational messages.  */
48 static int be_quiet;
49
50 /* Name of output file.  */
51 static const char *output_name;
52
53 /* Various definitions for the libdb handling.  */
54 enum {
55   nodb,
56   db24,
57   db27
58 } libdb_version;
59 static int (*db_open) (const char *, int,
60                        uint32_t, int, void *, void *, void **);
61
62 /* Constants which vary from version to version are actually variables
63    here.  */
64 static int db_first;
65 static int db_next;
66 static int db_nooverwrite;
67 static int db_truncate;
68
69
70 /* Name and version of program.  */
71 static void print_version (FILE *stream, struct argp_state *state);
72 void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version;
73
74 /* Definitions of arguments for argp functions.  */
75 static const struct argp_option options[] =
76 {
77   { "fold-case", 'f', NULL, 0, N_("Convert key to lower case") },
78   { "output", 'o', N_("NAME"), 0, N_("Write output to file NAME") },
79   { "quiet", 'q', NULL, 0,
80     N_("Do not print messages while building database") },
81   { "undo", 'u', NULL, 0,
82     N_("Print content of database file, one entry a line") },
83   { NULL, 0, NULL, 0, NULL }
84 };
85
86 /* Short description of program.  */
87 static const char doc[] = N_("Create simple DB database from textual input.");
88
89 /* Strings for arguments in help texts.  */
90 static const char args_doc[] = N_("\
91 INPUT-FILE OUTPUT-FILE\n-o OUTPUT-FILE INPUT-FILE\n-u INPUT-FILE");
92
93 /* Prototype for option handler.  */
94 static error_t parse_opt __P ((int key, char *arg, struct argp_state *state));
95
96 /* Function to print some extra text in the help message.  */
97 static char *more_help __P ((int key, const char *text, void *input));
98
99 /* Data structure to communicate with argp functions.  */
100 static struct argp argp =
101 {
102   options, parse_opt, args_doc, doc, NULL, more_help
103 };
104
105
106 /* Prototypes for local functions.  */
107 static int process_input __P ((FILE *input, const char *inname, NSS_DB *output,
108                                int to_lowercase, int be_quiet));
109 static int print_database __P ((NSS_DB *db));
110 static NSS_DB *dbopen (const char *fname, int oper, int mode);
111 int main __P ((int argc, char *argv[]));
112
113
114 int
115 main (argc, argv)
116      int argc;
117      char *argv[];
118 {
119   const char *input_name;
120   FILE *input_file;
121   NSS_DB *db_file;
122   int status;
123   int remaining;
124   int mode = 0666;
125
126   /* Set locale via LC_ALL.  */
127   setlocale (LC_ALL, "");
128
129   /* Set the text message domain.  */
130   textdomain (_libc_intl_domainname);
131
132   /* Initialize local variables.  */
133   input_name = NULL;
134
135   /* Parse and process arguments.  */
136   argp_parse (&argp, argc, argv, 0, &remaining, NULL);
137
138   /* Determine file names.  */
139   if (do_undo || output_name != NULL)
140     {
141       if (remaining + 1 != argc)
142         {
143         wrong_arguments:
144           error (0, 0, gettext ("wrong number of arguments"));
145           argp_help (&argp, stdout, ARGP_HELP_SEE,
146                      program_invocation_short_name);
147           exit (1);
148         }
149       input_name = argv[remaining];
150     }
151   else
152     {
153       if (remaining + 2 != argc)
154         goto wrong_arguments;
155
156       input_name = argv[remaining++];
157       output_name = argv[remaining];
158     }
159
160   /* Special handling if we are asked to print the database.  */
161   if (do_undo)
162     {
163       db_file = dbopen (input_name, DB_RDONLY, 0666);
164       if (db_file == NULL)
165         error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
166                input_name,
167                (errno == EINVAL ? gettext ("incorrectly formatted file")
168                 : strerror (errno)));
169
170       status = print_database (db_file);
171
172       db_file->close (db_file->db, 0);
173
174       return status;
175     }
176
177   /* Open input file.  */
178   if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
179     input_file = stdin;
180   else
181     {
182       struct stat st;
183
184       input_file = fopen (input_name, "r");
185       if (input_file == NULL)
186         error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
187                input_name);
188
189       /* Get the access rights from the source file.  The output file should
190          have the same.  */
191       if (fstat (fileno (input_file), &st) >= 0)
192         mode = st.st_mode & ACCESSPERMS;
193     }
194
195   /* Open output file.  This must not be standard output so we don't
196      handle "-" and "/dev/stdout" special.  */
197   db_file = dbopen (output_name, DB_CREATE | db_truncate, mode);
198   if (db_file == NULL)
199     error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"),
200            output_name);
201
202   /* Start the real work.  */
203   status = process_input (input_file, input_name, db_file, to_lowercase,
204                           be_quiet);
205
206   /* Close files.  */
207   if (input_file != stdin)
208     fclose (input_file);
209   db_file->close (db_file->db, 0);
210
211   return status;
212 }
213
214
215 /* Handle program arguments.  */
216 static error_t
217 parse_opt (int key, char *arg, struct argp_state *state)
218 {
219   switch (key)
220     {
221     case 'f':
222       to_lowercase = 1;
223       break;
224     case 'o':
225       output_name = arg;
226       break;
227     case 'q':
228       be_quiet = 1;
229       break;
230     case 'u':
231       do_undo = 1;
232       break;
233     default:
234       return ARGP_ERR_UNKNOWN;
235     }
236   return 0;
237 }
238
239
240 static char *
241 more_help (int key, const char *text, void *input)
242 {
243   switch (key)
244     {
245     case ARGP_KEY_HELP_EXTRA:
246       /* We print some extra information.  */
247       return strdup (gettext ("\
248 Report bugs using the `glibcbug' script to <bugs@gnu.org>.\n"));
249     default:
250       break;
251     }
252   return (char *) text;
253 }
254
255 /* Print the version information.  */
256 static void
257 print_version (FILE *stream, struct argp_state *state)
258 {
259   fprintf (stream, "makedb (GNU %s) %s\n", PACKAGE, VERSION);
260   fprintf (stream, gettext ("\
261 Copyright (C) %s Free Software Foundation, Inc.\n\
262 This is free software; see the source for copying conditions.  There is NO\n\
263 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\
264 "), "1999");
265   fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper");
266 }
267
268
269 static int
270 process_input (input, inname, output, to_lowercase, be_quiet)
271      FILE *input;
272      const char *inname;
273      NSS_DB *output;
274      int to_lowercase;
275      int be_quiet;
276 {
277   char *line;
278   size_t linelen;
279   int status;
280   size_t linenr;
281
282   line = NULL;
283   linelen = 0;
284   status = EXIT_SUCCESS;
285   linenr = 0;
286
287   while (!feof (input))
288     {
289       DBT key;
290       DBT val;
291       char *cp;
292       int n;
293
294       n = getline (&line, &linelen, input);
295       if (n < 0)
296         /* This means end of file or some bug.  */
297         break;
298       if (n == 0)
299         /* Short read.  Probably interrupted system call. */
300         continue;
301
302       ++linenr;
303
304       if (line[n - 1] == '\n')
305         /* Remove trailing newline.  */
306         line[--n] = '\0';
307
308       cp = line;
309       while (isspace (*cp))
310         ++cp;
311
312       if (*cp == '#')
313         /* First non-space character in line '#': it's a comment.  */
314         continue;
315
316       key.data = cp;
317       while (*cp != '\0' && !isspace (*cp))
318         {
319           if (to_lowercase)
320             *cp = tolower (*cp);
321           ++cp;
322         }
323
324       if (key.data == cp)
325         /* It's an empty line.  */
326         continue;
327
328       key.size = cp - (char *) key.data;
329       key.flags = 0;
330
331       while (isspace (*cp))
332         ++cp;
333
334       val.data = cp;
335       val.size = (&line[n] - cp) + 1;
336       val.flags = 0;
337
338       /* Store the value.  */
339       status = output->put (output->db, NULL, &key, &val, db_nooverwrite);
340       if (status != 0)
341         {
342           if (status == DB_KEYEXIST)
343             {
344               if (!be_quiet)
345                 error_at_line (0, 0, inname, linenr,
346                                gettext ("duplicate key"));
347               /* This is no real error.  Just give a warning.  */
348               status = 0;
349               continue;
350             }
351           else
352             error (0, status, gettext ("while writing database file"));
353
354           status = EXIT_FAILURE;
355
356           clearerr (input);
357           break;
358         }
359     }
360
361   if (ferror (input))
362     {
363       error (0, 0, gettext ("problems while reading `%s'"), inname);
364       status = EXIT_FAILURE;
365     }
366
367   return status;
368 }
369
370
371 static int
372 print_database (db)
373      NSS_DB *db;
374 {
375   DBT key;
376   DBT val;
377   NSS_DBC *cursor;
378   int status;
379
380   status = db->cursor (db->db, NULL, &cursor);
381   if (status != 0)
382     {
383       error (0, status, gettext ("while reading database"));
384       return EXIT_FAILURE;
385     }
386
387   key.flags = 0;
388   val.flags = 0;
389   status = cursor->c_get (cursor->cursor, &key, &val, db_first);
390   while (status == 0)
391     {
392       printf ("%.*s %s\n", (int) key.size, (char *) key.data,
393               (char *) val.data);
394
395       status = cursor->c_get (cursor->cursor, &key, &val, db_next);
396     }
397
398   if (status != DB_NOTFOUND)
399     {
400       error (0, status, gettext ("while reading database"));
401       return EXIT_FAILURE;
402     }
403
404   return EXIT_SUCCESS;
405 }
406
407
408 static int
409 load_db (void)
410 {
411   static const char *libnames[] = { "libdb.so.3" };
412   int x;
413   void *handle;
414
415   for(x = 0; x < sizeof (libnames) / sizeof (libnames[0]); ++x)
416     {
417       handle = dlopen (libnames[x], RTLD_LAZY);
418       if (handle == NULL)
419         continue;
420
421       db_open = dlsym (handle, "db_open");
422       if (db_open != NULL)
423         {
424           /* Alright, we got a library.  Now find out which version it is.  */
425           const char *(*db_version) (int *, int *, int *);
426
427           db_version = dlsym (handle, "db_version");
428           if (db_version != NULL)
429             {
430               /* Call the function and get the information.  */
431               int major, minor, subminor;
432
433               DL_CALL_FCT (db_version, (&major, &minor, &subminor));
434               if (major == 2)
435                 {
436                   /* We currently cannot handle other versions than the
437                      2.x series.  */
438                   if (minor < 6 || (minor == 6 && subminor < 4))
439                     {
440                       libdb_version = db24;
441                       db_first = DB24_FIRST;
442                       db_next = DB24_NEXT;
443                       db_nooverwrite = DB24_NOOVERWRITE;
444                       db_truncate = DB24_TRUNCATE;
445                     }
446                   else
447                     {
448                       libdb_version = db27;
449                       db_first = DB27_FIRST;
450                       db_next = DB27_NEXT;
451                       db_nooverwrite = DB27_NOOVERWRITE;
452                       db_truncate = DB27_TRUNCATE;
453                     }
454                 }
455             }
456
457           if (libdb_version != nodb)
458             return 0;
459         }
460
461       dlclose (handle);
462     }
463
464   (void) dlerror ();
465   return 1;
466 }
467
468
469 static int
470 db_cursor (void *db, void *txn, NSS_DBC **dbcp)
471 {
472   void *ptr;
473   NSS_DBC *dbc = NULL;
474   int ret;
475
476   switch (libdb_version)
477     {
478     case db24:
479       ret = ((struct db24 *) db)->cursor (db, txn, &ptr);
480       break;
481     case db27:
482       ret = ((struct db27 *) db)->cursor (db, txn, &ptr, 0);
483       break;
484     default:
485       abort ();
486     }
487
488   if (ret == 0)
489     {
490       dbc = (NSS_DBC *) malloc (sizeof (NSS_DBC));
491       if (dbc == NULL)
492         error (EXIT_FAILURE, errno, gettext ("while reading database"));
493       dbc->cursor = ptr;
494
495       switch (libdb_version)
496         {
497         case db24:
498           dbc->c_get =
499             (int (*) (void *, void *, void *, uint32_t))
500             ((struct dbc24 *) ptr)->c_get;
501           break;
502         case db27:
503           dbc->c_get =
504             (int (*) (void *, void *, void *, uint32_t))
505             ((struct dbc27 *) ptr)->c_get;
506           break;
507         default:
508           abort ();
509         }
510     }
511
512   *dbcp = dbc;
513
514   return ret;
515 }
516
517
518 static NSS_DB *
519 dbopen (const char *fname, int oper, int mode)
520 {
521   int err;
522   void *odb;
523   NSS_DB *db;
524
525   /* First load the shared object.  */
526   load_db ();
527
528   if (db_open == NULL)
529     return NULL;
530
531   /* Actually open the database.  */
532   err = DL_CALL_FCT (db_open, (fname, DB_BTREE, oper, mode, NULL, NULL, &odb));
533   if (err != 0)
534     {
535       errno = err;
536       return NULL;
537     }
538
539   /* Construct the object we pass up.  */
540   db = (NSS_DB *) malloc (sizeof (NSS_DB));
541   if (db != NULL)
542     {
543       db->db = odb;
544
545       /* The functions are at different positions for the different
546          versions.  Sigh.  */
547       switch (libdb_version)
548         {
549         case db24:
550           db->close =
551             (int (*) (void *, uint32_t)) ((struct db24 *) odb)->close;
552           db->fd =
553             (int (*) (void *, int *)) ((struct db24 *) odb)->fd;
554           db->get =
555             (int (*) (void *, void *, void *, void *, uint32_t))
556             ((struct db24 *) odb)->get;
557           db->put =
558             (int (*) (void *, void *, void *, void *, uint32_t))
559             ((struct db24 *) odb)->put;
560           break;
561         case db27:
562           db->close =
563             (int (*) (void *, uint32_t)) ((struct db27 *) odb)->close;
564           db->fd =
565             (int (*) (void *, int *)) ((struct db27 *) odb)->fd;
566           db->get =
567             (int (*) (void *, void *, void *, void *, uint32_t))
568             ((struct db27 *) odb)->get;
569           db->put =
570             (int (*) (void *, void *, void *, void *, uint32_t))
571             ((struct db27 *) odb)->put;
572           break;
573         default:
574           abort ();
575         }
576       db->cursor = db_cursor;
577     }
578
579   return db;
580 }