fdwalk should return 0 on an empty directory
[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 Lesser General Public
8    License as published by the Free Software Foundation; either
9    version 2.1 of the 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    Lesser General Public License for more details.
15
16    You should have received a copy of the GNU Lesser General Public
17    License along with the GNU C Library; if not, write to the Free
18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19    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 /* Name and version of program.  */
54 static void print_version (FILE *stream, struct argp_state *state);
55 void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version;
56
57 /* Definitions of arguments for argp functions.  */
58 static const struct argp_option options[] =
59 {
60   { "fold-case", 'f', NULL, 0, N_("Convert key to lower case") },
61   { "output", 'o', N_("NAME"), 0, N_("Write output to file NAME") },
62   { "quiet", 'q', NULL, 0,
63     N_("Do not print messages while building database") },
64   { "undo", 'u', NULL, 0,
65     N_("Print content of database file, one entry a line") },
66   { NULL, 0, NULL, 0, NULL }
67 };
68
69 /* Short description of program.  */
70 static const char doc[] = N_("Create simple DB database from textual input.");
71
72 /* Strings for arguments in help texts.  */
73 static const char args_doc[] = N_("\
74 INPUT-FILE OUTPUT-FILE\n-o OUTPUT-FILE INPUT-FILE\n-u INPUT-FILE");
75
76 /* Prototype for option handler.  */
77 static error_t parse_opt (int key, char *arg, struct argp_state *state);
78
79 /* Function to print some extra text in the help message.  */
80 static char *more_help (int key, const char *text, void *input);
81
82 /* Data structure to communicate with argp functions.  */
83 static struct argp argp =
84 {
85   options, parse_opt, args_doc, doc, NULL, more_help
86 };
87
88
89 /* Prototypes for local functions.  */
90 static int process_input (FILE *input, const char *inname, NSS_DB *output,
91                           int to_lowercase, int be_quiet);
92 static int print_database (NSS_DB *db);
93
94
95 int
96 main (int argc, char *argv[])
97 {
98   const char *input_name;
99   FILE *input_file;
100   NSS_DB *db_file;
101   int status;
102   int remaining;
103   int mode = 0666;
104
105   /* Set locale via LC_ALL.  */
106   setlocale (LC_ALL, "");
107
108   /* Set the text message domain.  */
109   textdomain (_libc_intl_domainname);
110
111   /* Initialize local variables.  */
112   input_name = NULL;
113
114   /* Parse and process arguments.  */
115   argp_parse (&argp, argc, argv, 0, &remaining, NULL);
116
117   /* Determine file names.  */
118   if (do_undo || output_name != NULL)
119     {
120       if (remaining + 1 != argc)
121         {
122         wrong_arguments:
123           error (0, 0, gettext ("wrong number of arguments"));
124           argp_help (&argp, stdout, ARGP_HELP_SEE,
125                      program_invocation_short_name);
126           exit (1);
127         }
128       input_name = argv[remaining];
129     }
130   else
131     {
132       if (remaining + 2 != argc)
133         goto wrong_arguments;
134
135       input_name = argv[remaining++];
136       output_name = argv[remaining];
137     }
138
139   /* First load the shared object to initialize version dependend
140      variables.  */
141   if (load_db () != NSS_STATUS_SUCCESS)
142     error (EXIT_FAILURE, 0, gettext ("No usable database library found."));
143
144   /* Special handling if we are asked to print the database.  */
145   if (do_undo)
146     {
147       dbopen (input_name, db_rdonly, 0666, &db_file);
148       if (db_file == NULL)
149         error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
150                input_name,
151                (errno == EINVAL ? gettext ("incorrectly formatted file")
152                 : strerror (errno)));
153
154       status = print_database (db_file);
155
156       db_file->close (db_file->db, 0);
157
158       return status;
159     }
160
161   /* Open input file.  */
162   if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
163     input_file = stdin;
164   else
165     {
166       struct stat st;
167
168       input_file = fopen (input_name, "r");
169       if (input_file == NULL)
170         error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
171                input_name);
172
173       /* Get the access rights from the source file.  The output file should
174          have the same.  */
175       if (fstat (fileno (input_file), &st) >= 0)
176         mode = st.st_mode & ACCESSPERMS;
177     }
178
179   /* Open output file.  This must not be standard output so we don't
180      handle "-" and "/dev/stdout" special.  */
181   dbopen (output_name, DB_CREATE | db_truncate, mode, &db_file);
182   if (db_file == NULL)
183     error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"),
184            output_name);
185
186   /* Start the real work.  */
187   status = process_input (input_file, input_name, db_file, to_lowercase,
188                           be_quiet);
189
190   /* Close files.  */
191   if (input_file != stdin)
192     fclose (input_file);
193   db_file->close (db_file->db, 0);
194
195   return status;
196 }
197
198
199 /* Handle program arguments.  */
200 static error_t
201 parse_opt (int key, char *arg, struct argp_state *state)
202 {
203   switch (key)
204     {
205     case 'f':
206       to_lowercase = 1;
207       break;
208     case 'o':
209       output_name = arg;
210       break;
211     case 'q':
212       be_quiet = 1;
213       break;
214     case 'u':
215       do_undo = 1;
216       break;
217     default:
218       return ARGP_ERR_UNKNOWN;
219     }
220   return 0;
221 }
222
223
224 static char *
225 more_help (int key, const char *text, void *input)
226 {
227   switch (key)
228     {
229     case ARGP_KEY_HELP_EXTRA:
230       /* We print some extra information.  */
231       return strdup (gettext ("\
232 For bug reporting instructions, please see:\n\
233 <http://www.gnu.org/software/libc/bugs.html>.\n"));
234     default:
235       break;
236     }
237   return (char *) text;
238 }
239
240 /* Print the version information.  */
241 static void
242 print_version (FILE *stream, struct argp_state *state)
243 {
244   fprintf (stream, "makedb (GNU %s) %s\n", PACKAGE, VERSION);
245   fprintf (stream, gettext ("\
246 Copyright (C) %s Free Software Foundation, Inc.\n\
247 This is free software; see the source for copying conditions.  There is NO\n\
248 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\
249 "), "2000");
250   fprintf (stream, gettext ("Written by %s.\n"), "Ulrich Drepper");
251 }
252
253
254 static int
255 process_input (input, inname, output, to_lowercase, be_quiet)
256      FILE *input;
257      const char *inname;
258      NSS_DB *output;
259      int to_lowercase;
260      int be_quiet;
261 {
262   char *line;
263   size_t linelen;
264   int status;
265   size_t linenr;
266
267   line = NULL;
268   linelen = 0;
269   status = EXIT_SUCCESS;
270   linenr = 0;
271
272   while (!feof (input))
273     {
274       DBT key;
275       DBT val;
276       char *cp;
277       int n;
278
279       n = getline (&line, &linelen, input);
280       if (n < 0)
281         /* This means end of file or some bug.  */
282         break;
283       if (n == 0)
284         /* Short read.  Probably interrupted system call. */
285         continue;
286
287       ++linenr;
288
289       if (line[n - 1] == '\n')
290         /* Remove trailing newline.  */
291         line[--n] = '\0';
292
293       cp = line;
294       while (isspace (*cp))
295         ++cp;
296
297       if (*cp == '#')
298         /* First non-space character in line '#': it's a comment.  */
299         continue;
300
301       key.data = cp;
302       while (*cp != '\0' && !isspace (*cp))
303         {
304           if (to_lowercase)
305             *cp = tolower (*cp);
306           ++cp;
307         }
308
309       if (key.data == cp)
310         /* It's an empty line.  */
311         continue;
312
313       key.size = cp - (char *) key.data;
314       key.flags = 0;
315
316       while (isspace (*cp))
317         ++cp;
318
319       val.data = cp;
320       val.size = (&line[n] - cp) + 1;
321       val.flags = 0;
322
323       /* Store the value.  */
324       status = output->put (output->db, NULL, &key, &val, db_nooverwrite);
325       if (status != 0)
326         {
327           if (status == db_keyexist)
328             {
329               if (!be_quiet)
330                 error_at_line (0, 0, inname, linenr,
331                                gettext ("duplicate key"));
332               /* This is no real error.  Just give a warning.  */
333               status = 0;
334               continue;
335             }
336           else
337             error (0, status, gettext ("while writing database file"));
338
339           status = EXIT_FAILURE;
340
341           clearerr (input);
342           break;
343         }
344     }
345
346   if (ferror (input))
347     {
348       error (0, 0, gettext ("problems while reading `%s'"), inname);
349       status = EXIT_FAILURE;
350     }
351
352   return status;
353 }
354
355
356 static int
357 print_database (db)
358      NSS_DB *db;
359 {
360   DBT key;
361   DBT val;
362   NSS_DBC *cursor;
363   int status;
364
365   status = db->cursor (db->db, NULL, &cursor);
366   if (status != 0)
367     {
368       error (0, status, gettext ("while reading database"));
369       return EXIT_FAILURE;
370     }
371
372   key.flags = 0;
373   val.flags = 0;
374   status = cursor->c_get (cursor->cursor, &key, &val, db_first);
375   while (status == 0)
376     {
377       printf ("%.*s %s\n", (int) key.size, (char *) key.data,
378               (char *) val.data);
379
380       status = cursor->c_get (cursor->cursor, &key, &val, db_next);
381     }
382
383   if (status != db_notfound)
384     {
385       error (0, status, gettext ("while reading database"));
386       return EXIT_FAILURE;
387     }
388
389   return EXIT_SUCCESS;
390 }