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