1 /* $NetBSD: makedbm.c,v 1.25 2011/08/30 21:10:28 joerg Exp $ */
2
3 /*
4 * Copyright (c) 1994 Mats O Jansson <moj@stacken.kth.se>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
17 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
20 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <sys/cdefs.h>
30 #ifndef lint
31 __RCSID("$NetBSD: makedbm.c,v 1.25 2011/08/30 21:10:28 joerg Exp $");
32 #endif
33
34 #include <sys/param.h>
35 #include <sys/stat.h>
36
37 #include <ctype.h>
38 #include <err.h>
39 #include <errno.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <time.h>
44 #include <unistd.h>
45
46 #include <rpc/rpc.h>
47 #include <rpc/xdr.h>
48
49 #include "protos.h"
50 #include "ypdb.h"
51 #include "ypdef.h"
52
53 __dead static void usage(void);
54 static int add_record(DBM *, const char *, const char *, int);
55 static char *file_date(char *);
56 static void list_database(char *);
57 static void create_database(char *, char *, char *, char *, char *,
58 char *, int, int, int);
59
60 int
main(int argc,char * argv[])61 main(int argc, char *argv[])
62 {
63 int aflag, uflag, bflag, lflag, sflag;
64 char *yp_input_file, *yp_output_file;
65 char *yp_master_name, *yp_domain_name;
66 char *infile, *outfile;
67 int ch;
68
69 yp_input_file = yp_output_file = NULL;
70 yp_master_name = yp_domain_name = NULL;
71 aflag = uflag = bflag = lflag = sflag = 0;
72 infile = outfile = NULL;
73
74 while ((ch = getopt(argc, argv, "blsui:o:m:d:")) != -1) {
75 switch (ch) {
76 case 'b':
77 bflag = aflag = 1;
78 break;
79
80 case 'l':
81 lflag = aflag = 1;
82 break;
83
84 case 's':
85 sflag = aflag = 1;
86 break;
87
88 case 'i':
89 yp_input_file = optarg;
90 aflag = 1;
91 break;
92
93 case 'o':
94 yp_output_file = optarg;
95 aflag = 1;
96 break;
97
98 case 'm':
99 yp_master_name = optarg;
100 aflag = 1;
101 break;
102
103 case 'd':
104 yp_domain_name = optarg;
105 aflag = 1;
106 break;
107
108 case 'u':
109 uflag = 1;
110 break;
111
112 default:
113 usage();
114 }
115 }
116 argc -= optind; argv += optind;
117
118 if ((uflag != 0) && (aflag != 0))
119 usage();
120 else {
121 if (uflag != 0) {
122 if (argc == 1)
123 infile = argv[0];
124 else
125 usage();
126 } else {
127 if (argc == 2) {
128 infile = argv[0];
129 outfile = argv[1];
130 } else
131 usage();
132 }
133 }
134
135 if (uflag != 0)
136 list_database(infile);
137 else
138 create_database(infile, outfile,
139 yp_input_file, yp_output_file, yp_master_name,
140 yp_domain_name, bflag, lflag, sflag);
141
142 exit(0);
143 }
144
145 static int
add_record(DBM * db,const char * str1,const char * str2,int check)146 add_record(DBM *db, const char *str1, const char *str2, int check)
147 {
148 datum key, val;
149 int status;
150
151 key.dptr = __UNCONST(str1);
152 key.dsize = strlen(str1);
153
154 if (check) {
155 val = ypdb_fetch(db, key);
156
157 if (val.dptr != NULL)
158 return 0; /* already there */
159 }
160 val.dptr = __UNCONST(str2);
161 val.dsize = strlen(str2);
162 status = ypdb_store(db, key, val, YPDB_INSERT);
163
164 if (status != 0) {
165 warnx("can't store `%s %s'", str1, str2);
166 return -1;
167 }
168 return 0;
169 }
170
171 static char *
file_date(char * filename)172 file_date(char *filename)
173 {
174 struct stat finfo;
175 static char datestr[11];
176
177 memset(datestr, 0, sizeof(datestr));
178
179 if (strcmp(filename, "-") == 0)
180 snprintf(datestr, sizeof(datestr), "%010d",
181 (int)time(NULL));
182 else {
183 if (stat(filename, &finfo) != 0)
184 err(1, "can't stat %s", filename);
185 snprintf(datestr, sizeof(datestr), "%010d",
186 (int)finfo.st_mtime);
187 }
188
189 return datestr;
190 }
191
192 static void
list_database(char * database)193 list_database(char *database)
194 {
195 DBM *db;
196 datum key, val;
197
198 db = ypdb_open(database);
199 if (db == NULL)
200 err(1, "can't open database `%s'", database);
201
202 key = ypdb_firstkey(db);
203
204 while (key.dptr != NULL) {
205 val = ypdb_fetch(db, key);
206 /* workround trailing \0 in aliases.db */
207 if (key.dptr[key.dsize - 1] == '\0')
208 key.dsize--;
209 printf("%*.*s", key.dsize, key.dsize, key.dptr);
210 if (val.dsize > 0) {
211 if (val.dptr[val.dsize - 1] == '\0')
212 val.dsize--;
213 printf(" %*.*s", val.dsize, val.dsize, val.dptr);
214 }
215 printf("\n");
216 key = ypdb_nextkey(db);
217 }
218
219 ypdb_close(db);
220 }
221
222 static void
create_database(char * infile,char * database,char * yp_input_file,char * yp_output_file,char * yp_master_name,char * yp_domain_name,int bflag,int lflag,int sflag)223 create_database(char *infile, char *database, char *yp_input_file,
224 char *yp_output_file, char *yp_master_name,
225 char *yp_domain_name, int bflag, int lflag, int sflag)
226 {
227 FILE *data_file;
228 char myname[MAXHOSTNAMELEN];
229 size_t line_no = 0;
230 size_t len;
231 char *p, *k, *v, *slash;
232 DBM *new_db;
233 static const char template[] = "ypdbXXXXXX";
234 char db_mapname[MAXPATHLEN + 1], db_outfile[MAXPATHLEN + 1];
235 char empty_str[] = "";
236
237 memset(db_mapname, 0, sizeof(db_mapname));
238 memset(db_outfile, 0, sizeof(db_outfile));
239
240 if (strcmp(infile, "-") == 0)
241 data_file = stdin;
242 else {
243 data_file = fopen(infile, "r");
244 if (data_file == NULL)
245 err(1, "can't open `%s'", infile);
246 }
247
248 if (strlen(database) + strlen(YPDB_SUFFIX) > (sizeof(db_outfile) - 1))
249 errx(1, "file name `%s' too long", database);
250
251 snprintf(db_outfile, sizeof(db_outfile), "%s%s", database, YPDB_SUFFIX);
252
253 slash = strrchr(database, '/');
254 if (slash != NULL)
255 slash[1] = '\0'; /* truncate to dir */
256 else
257 *database = '\0'; /* eliminate */
258
259 /* NOTE: database is now directory where map goes ! */
260
261 if (strlen(database) + strlen(template) + strlen(YPDB_SUFFIX) >
262 (sizeof(db_mapname) - 1))
263 errx(1, "directory name `%s' too long", database);
264
265 snprintf(db_mapname, sizeof(db_mapname), "%s%s",
266 database, template);
267
268 new_db = ypdb_mktemp(db_mapname);
269 if (new_db == NULL)
270 err(1, "can't create temp database `%s'", db_mapname);
271
272 for (;
273 (p = fparseln(data_file, &len, &line_no, NULL, FPARSELN_UNESCALL));
274 free(p)) {
275 k = p; /* set start of key */
276 while (*k && isspace((unsigned char)*k)) /* skip leading whitespace */
277 k++;
278
279 if (! *k)
280 continue;
281
282 v = k;
283 while (*v && !isspace((unsigned char)*v)) { /* find leading whitespace */
284 /* convert key to lower case if forcing. */
285 if (lflag && isupper((unsigned char)*v))
286 *v = tolower((unsigned char)*v);
287 v++;
288 }
289 while (*v && isspace((unsigned char)*v)) /* replace space with <NUL> */
290 *v++ = '\0';
291
292 if (add_record(new_db, k, v, TRUE)) { /* save record */
293 bad_record:
294 ypdb_close(new_db);
295 unlink(db_mapname);
296 errx(1, "error adding record for line %lu",
297 (unsigned long)line_no);
298 }
299 }
300
301 if (strcmp(infile, "-") != 0)
302 (void) fclose(data_file);
303
304 if (add_record(new_db, YP_LAST_KEY, file_date(infile), FALSE))
305 goto bad_record;
306
307 if (yp_input_file) {
308 if (add_record(new_db, YP_INPUT_KEY, yp_input_file, FALSE))
309 goto bad_record;
310 }
311
312 if (yp_output_file) {
313 if (add_record(new_db, YP_OUTPUT_KEY, yp_output_file, FALSE))
314 goto bad_record;
315 }
316
317 if (yp_master_name) {
318 if (add_record(new_db, YP_MASTER_KEY, yp_master_name, FALSE))
319 goto bad_record;
320 } else {
321 localhostname(myname, sizeof(myname) - 1);
322 if (add_record(new_db, YP_MASTER_KEY, myname, FALSE))
323 goto bad_record;
324 }
325
326 if (yp_domain_name) {
327 if (add_record(new_db, YP_DOMAIN_KEY, yp_domain_name, FALSE))
328 goto bad_record;
329 }
330
331 if (bflag) {
332 if (add_record(new_db, YP_INTERDOMAIN_KEY, empty_str, FALSE))
333 goto bad_record;
334 }
335
336 if (sflag) {
337 if (add_record(new_db, YP_SECURE_KEY, empty_str, FALSE))
338 goto bad_record;
339 }
340
341 ypdb_close(new_db);
342 if (rename(db_mapname, db_outfile) < 0) {
343 unlink(db_mapname);
344 err(1, "rename `%s' -> `%s'", db_mapname, db_outfile);
345 }
346 }
347
348 static void
usage(void)349 usage(void)
350 {
351
352 fprintf(stderr, "usage: %s -u file\n", getprogname());
353 fprintf(stderr, " %s [-lbs] %s\n", getprogname(),
354 "[-i YP_INPUT_FILE] [-o YP_OUTPUT_FILE]");
355 fprintf(stderr, " %s infile outfile\n",
356 "[-d YP_DOMAIN_NAME] [-m YP_MASTER_NAME]");
357 exit(1);
358 }
359