update from main archive 960829
[kopensolaris-gnu/glibc.git] / db / makedb.c
1 /* makedb -- create simple DB database from textual input.
2 Copyright (C) 1996 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
18 not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19 Boston, MA 02111-1307, USA.  */
20
21 #include <db.h>
22 #include <ctype.h>
23 #include <errno.h>
24 #include <error.h>
25 #include <fcntl.h>
26 #include <getopt.h>
27 #include <libintl.h>
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <string.h>
31
32 /* Get libc version number.  */
33 #include "../version.h"
34
35
36 /* Long options.  */
37 static const struct option long_options[] =
38 {
39   { "help", no_argument, NULL, 'h' },
40   { "fold-case", no_argument, NULL, 'f' },
41   { "output", required_argument, NULL, 'o' },
42   { "quiet", no_argument, NULL, 'q' },
43   { "undo", no_argument, NULL, 'u' },
44   { "version", no_argument, NULL, 'V' },
45   { NULL, 0, NULL, 0}
46 };
47
48 /* Prototypes for local functions.  */
49 static void usage __P ((int status)) __attribute__ ((noreturn));
50 static int process_input __P ((FILE *input, const char *inname, DB *output,
51                                int to_lowercase, int be_quiet));
52 static int print_database __P ((DB *db));
53
54
55 int
56 main (argc, argv)
57      int argc;
58      char *argv[];
59 {
60   const char *output_name;
61   const char *input_name;
62   FILE *input_file;
63   DB *db_file;
64   int do_help;
65   int do_version;
66   int to_lowercase;
67   int do_undo;
68   int be_quiet;
69   int status;
70   int opt;
71
72   /* Set locale via LC_ALL.  */
73   setlocale (LC_ALL, "");
74
75   /* Set the text message domain.  */
76   textdomain (_libc_intl_domainname);
77
78   /* Initialize local variables.  */
79   do_help = 0;
80   do_version = 0;
81   to_lowercase = 0;
82   do_undo = 0;
83   be_quiet = 0;
84   output_name = NULL;
85
86   while ((opt = getopt_long (argc, argv, "fho:uV", long_options, NULL)) != EOF)
87     switch (opt)
88       {
89       case '\0':        /* Long option.  */
90         break;
91       case 'h':
92         do_help = 1;
93         break;
94       case 'f':
95         to_lowercase = 1;
96         break;
97       case 'o':
98         output_name = optarg;
99         break;
100       case 'q':
101         be_quiet = 1;
102         break;
103       case 'u':
104         do_undo = 1;
105         break;
106       case 'V':
107         do_version = 1;
108         break;
109       default:
110         usage (EXIT_FAILURE);
111       }
112
113   /* Version information is requested.  */
114   if (do_version)
115     printf ("%s - GNU %s %s\n", program_invocation_name, "libc", VERSION);
116
117   /* Help is requested.  */
118   if (do_help)
119     usage (EXIT_SUCCESS);
120   else if (do_version)
121     exit (EXIT_SUCCESS);
122
123   /* Determine file names.  */
124   if (do_undo || output_name != NULL)
125     {
126       if (optind + 1 != argc)
127         {
128         wrong_arguments:
129           error (0, 0, gettext ("wrong number of arguments"));
130           usage (EXIT_FAILURE);
131         }
132       input_name = argv[optind];
133     }
134   else
135     {
136       if (optind + 2 != argc)
137         goto wrong_arguments;
138
139       input_name = argv[optind++];
140       output_name = argv[optind];
141     }
142
143   /* Special handling if we are asked to print the database.  */
144   if (do_undo)
145     {
146       db_file = dbopen (input_name, O_RDONLY, 0666, DB_BTREE, NULL);
147       if (db_file == NULL)
148         error (EXIT_FAILURE, 0, gettext ("cannot open database file `%s': %s"),
149                input_name,
150                errno == EFTYPE ? gettext ("incorrectly formatted file")
151                                : strerror (errno));
152
153       status = print_database (db_file);
154
155       db_file->close (db_file);
156
157       return status;
158     }
159
160   /* Open input file.  */
161   if (strcmp (input_name, "-") == 0 || strcmp (input_name, "/dev/stdin") == 0)
162     input_file = stdin;
163   else
164     {
165       input_file = fopen (input_name, "r");
166       if (input_file == NULL)
167         error (EXIT_FAILURE, errno, gettext ("cannot open input file `%s'"),
168                input_name);
169     }
170
171   /* Open output file.  This must not be standard output so we don't
172      handle "-" and "/dev/stdout" special.  */
173   db_file = dbopen (output_name, O_CREAT | O_RDWR | O_TRUNC, 0666,
174                     DB_BTREE, NULL);
175   if (db_file == NULL)
176     error (EXIT_FAILURE, errno, gettext ("cannot open output file `%s'"));
177
178   /* Start the real work.  */
179   status = process_input (input_file, input_name, db_file, to_lowercase,
180                           be_quiet);
181
182   /* Close files.  */
183   if (input_file != stdin)
184     fclose (input_file);
185   db_file->close (db_file);
186
187   return status;
188 }
189
190
191 static void
192 usage (status)
193      int status;
194 {
195   if (status != EXIT_SUCCESS)
196     fprintf (stderr, gettext ("Try `%s --help' for more information.\n"),
197              program_invocation_name);
198   else
199     printf (gettext ("\
200 Usage: %s [OPTION]... INPUT-FILE OUTPUT-FILE\n\
201        %s [OPTION]... -o OUTPUT-FILE INPUT-FILE\n\
202        %s [OPTION]... -u INPUT-FILE\n\
203 Mandatory arguments to long options are mandatory for short options too.\n\
204   -f, --fold-case     convert key to lower case\n\
205   -h, --help          display this help and exit\n\
206   -o, --output=NAME   write output to file NAME\n\
207       --quiet         don't print messages while building database\n\
208   -u, --undo          print content of database file, one entry a line\n\
209   -V, --version       output version information and exit\n\
210 If INPUT-FILE is -, input is read from standard input.\n\
211 Report bugs to <bug-glibc@gnu.ai.mit.edu>.\n"),
212             program_invocation_name, program_invocation_name,
213             program_invocation_name);
214
215   exit (status);
216 }
217
218
219 static int
220 process_input (input, inname, output, to_lowercase, be_quiet)
221      FILE *input;
222      const char *inname;
223      DB *output;
224      int to_lowercase;
225      int be_quiet;
226 {
227   char *line;
228   size_t linelen;
229   int status;
230   size_t linenr;
231
232   line = NULL;
233   linelen = 0;
234   status = EXIT_SUCCESS;
235   linenr = 0;
236
237   while (!feof (input))
238     {
239       DBT key;
240       DBT val;
241       char *cp;
242       int n;
243
244       n = getline (&line, &linelen, input);
245       if (n < 0)
246         /* This means end of file or some bug.  */
247         break;
248       if (n == 0)
249         /* Short read.  Probably interrupted system call. */
250         continue;
251
252       ++linenr;
253
254       if (line[n - 1] == '\n')
255         /* Remove trailing newline.  */
256         line[--n] = '\0';
257
258       cp = line;
259       while (isspace (*cp))
260         ++cp;
261
262       if (*cp == '#')
263         /* First non-space character in line '#': it's a comment.  */
264         continue;
265
266       key.data = cp;
267       while (*cp != '\0' && !isspace (*cp))
268         {
269           if (to_lowercase)
270             *cp = tolower (*cp);
271           ++cp;
272         }
273
274       if (key.data == cp)
275         /* It's an empty line.  */
276         continue;
277
278       key.size = cp - (char *) key.data;
279
280       while (isspace (*cp))
281         ++cp;
282
283       val.data = cp;
284       val.size = &line[n] - cp;
285
286       /* Store the value.  */
287       status = output->put (output, &key, &val, R_NOOVERWRITE);
288       if (status != 0)
289         {
290           if (status == 1)
291             {
292               if (!be_quiet)
293                 error_at_line (0, 0, inname, linenr,
294                                gettext ("duplicate key"));
295               /* This is no real error.  Just give a warning.  */
296               status = 0;
297             }
298           else
299             error (0, errno, gettext ("while writing data base file"));
300
301           status = status ? EXIT_FAILURE : EXIT_SUCCESS;
302
303           clearerr (input);
304           break;
305         }
306     }
307
308   if (ferror (input))
309     {
310       error (0, 0, gettext ("problems while reading `%s'"));
311       status = EXIT_FAILURE;
312     }
313
314   return status;
315 }
316
317
318 static int
319 print_database (db)
320      DB *db;
321 {
322   DBT key;
323   DBT val;
324   int no_more;
325
326   no_more = db->seq (db, &key, &val, R_FIRST);
327   while (!no_more)
328     {
329       printf ("%.*s %.*s\n", (int) key.size, (char *) key.data, (int) val.size,
330               (char *) val.data);
331
332       no_more = db->seq (db, &key, &val, R_NEXT);
333     }
334
335   if (no_more == -1)
336     {
337       error (0, errno, gettext ("while reading database"));
338       return EXIT_FAILURE;
339     }
340
341   return EXIT_SUCCESS;
342 }