yp_mkdb.c revision 114601 
1/*
2 * Copyright (c) 1995, 1996
3 *	Bill Paul <wpaul@ctr.columbia.edu>.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *	This product includes software developed by Bill Paul.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 *    may be used to endorse or promote products derived from this software
18 *    without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED.  IN NO EVENT SHALL Bill Paul OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33#include <sys/cdefs.h>
34__FBSDID("$FreeBSD: head/usr.sbin/yp_mkdb/yp_mkdb.c 114601 2003-05-03 21:06:42Z obrien $");
35
36#include <err.h>
37#include <fcntl.h>
38#include <limits.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <string.h>
42#include <time.h>
43#include <unistd.h>
44#include <rpc/rpc.h>
45#include <rpcsvc/yp.h>
46#include <sys/param.h>
47#include <sys/types.h>
48#include <sys/stat.h>
49#include "yp_extern.h"
50#include "ypxfr_extern.h"
51
52char *yp_dir = "";	/* No particular default needed. */
53int _rpcpmstart = 0;
54int debug = 1;
55
56static void
57usage(void)
58{
59	fprintf(stderr, "%s\n%s\n%s\n%s\n",
60	"usage: yp_mkdb -c",
61	"       yp_mkdb -u dbname",
62	"       yp_mkdb [-c] [-b] [-s] [-f] [-i inputfile] [-o outputfile]",
63	"               [-d domainname ] [-m mastername] inputfile dbname");
64	exit(1);
65}
66
67#define PERM_SECURE (S_IRUSR|S_IWUSR)
68
69static DB *
70open_db(char *path, int flags)
71{
72	extern HASHINFO openinfo;
73
74	return(dbopen(path, flags, PERM_SECURE, DB_HASH, &openinfo));
75}
76
77static void
78unwind(char *map)
79{
80	DB *dbp;
81	DBT key, data;
82
83	dbp = open_db(map, O_RDONLY);
84
85	if (dbp == NULL)
86		err(1, "open_db(%s) failed", map);
87
88	key.data = NULL;
89	while (yp_next_record(dbp, &key, &data, 1, 1) == YP_TRUE)
90		printf("%.*s %.*s\n", key.size,key.data,data.size,data.data);
91
92	(void)(dbp->close)(dbp);
93	return;
94}
95
96int
97main(int argc, char *argv[])
98{
99	int ch;
100	int un = 0;
101	int clear = 0;
102	int filter_plusminus = 0;
103	char *infile = NULL;
104	char *map = NULL;
105	char *domain = NULL;
106	char *infilename = NULL;
107	char *outfilename = NULL;
108	char *mastername = NULL;
109	int interdom = 0;
110	int secure = 0;
111	DB *dbp;
112	DBT key, data;
113	char buf[10240];
114	char *keybuf, *datbuf;
115	FILE *ifp;
116	char hname[MAXHOSTNAMELEN + 2];
117
118	while ((ch = getopt(argc, argv, "uhcbsfd:i:o:m:")) != -1) {
119		switch (ch) {
120		case 'f':
121			filter_plusminus++;
122			break;
123		case 'u':
124			un++;
125			break;
126		case 'c':
127			clear++;
128			break;
129		case 'b':
130			interdom++;
131			break;
132		case 's':
133			secure++;
134			break;
135		case 'd':
136			domain = optarg;
137			break;
138		case 'i':
139			infilename = optarg;
140			break;
141		case 'o':
142			outfilename = optarg;
143			break;
144		case 'm':
145			mastername = optarg;
146			break;
147		case 'h':
148		default:
149			usage();
150			break;
151		}
152	}
153
154	argc -= optind;
155	argv += optind;
156
157	if (un) {
158		map = argv[0];
159		if (map == NULL)
160			usage();
161		unwind(map);
162		exit(0);
163
164	}
165
166	infile = argv[0];
167	map = argv[1];
168
169	if (infile == NULL || map == NULL) {
170		if (clear)
171			goto doclear;
172		usage();
173	}
174
175	if (mastername == NULL) {
176		if (gethostname((char *)&hname, sizeof(hname)) == -1)
177			err(1, "gethostname() failed");
178		mastername = (char *)&hname;
179	}
180
181	/*
182	 * Note that while we can read from stdin, we can't
183	 * write to stdout; the db library doesn't let you
184	 * write to a file stream like that.
185	 */
186
187	if (!strcmp(infile, "-")) {
188		ifp = stdin;
189	} else {
190		if ((ifp = fopen(infile, "r")) == NULL)
191			err(1, "failed to open %s", infile);
192	}
193
194	if ((dbp = open_db(map, O_RDWR|O_EXLOCK|O_EXCL|O_CREAT)) == NULL)
195		err(1, "open_db(%s) failed", map);
196
197	if (interdom) {
198		key.data = "YP_INTERDOMAIN";
199		key.size = sizeof("YP_INTERDOMAIN") - 1;
200		data.data = "";
201		data.size = 0;
202		yp_put_record(dbp, &key, &data, 0);
203	}
204
205	if (secure) {
206		key.data = "YP_SECURE";
207		key.size = sizeof("YP_SECURE") - 1;
208		data.data = "";
209		data.size = 0;
210		yp_put_record(dbp, &key, &data, 0);
211	}
212
213	key.data = "YP_MASTER_NAME";
214	key.size = sizeof("YP_MASTER_NAME") - 1;
215	data.data = mastername;
216	data.size = strlen(mastername);
217	yp_put_record(dbp, &key, &data, 0);
218
219	key.data = "YP_LAST_MODIFIED";
220	key.size = sizeof("YP_LAST_MODIFIED") - 1;
221	snprintf(buf, sizeof(buf), "%lu", time(NULL));
222	data.data = (char *)&buf;
223	data.size = strlen(buf);
224	yp_put_record(dbp, &key, &data, 0);
225
226	if (infilename) {
227		key.data = "YP_INPUT_FILE";
228		key.size = sizeof("YP_INPUT_FILE") - 1;
229		data.data = infilename;
230		data.size = strlen(infilename);
231		yp_put_record(dbp, &key, &data, 0);
232	}
233
234	if (outfilename) {
235		key.data = "YP_OUTPUT_FILE";
236		key.size = sizeof("YP_OUTPUT_FILE") - 1;
237		data.data = outfilename;
238		data.size = strlen(outfilename);
239		yp_put_record(dbp, &key, &data, 0);
240	}
241
242	if (domain) {
243		key.data = "YP_DOMAIN_NAME";
244		key.size = sizeof("YP_DOMAIN_NAME") - 1;
245		data.data = domain;
246		data.size = strlen(domain);
247		yp_put_record(dbp, &key, &data, 0);
248	}
249
250	while (fgets((char *)&buf, sizeof(buf), ifp)) {
251		char *sep = NULL;
252		int rval;
253
254		/* NUL terminate */
255		if ((sep = strchr(buf, '\n')))
256			*sep = '\0';
257
258		/* handle backslash line continuations */
259		while (buf[strlen(buf) - 1] == '\\') {
260			fgets((char *)&buf[strlen(buf) - 1],
261					sizeof(buf) - strlen(buf), ifp);
262			if ((sep = strchr(buf, '\n')))
263				*sep = '\0';
264		}
265
266		/* find the separation between the key and data */
267		if ((sep = strpbrk(buf, " \t")) == NULL) {
268			warnx("bad input -- no white space: %s", buf);
269			continue;
270		}
271
272		/* separate the strings */
273		keybuf = (char *)&buf;
274		datbuf = sep + 1;
275		*sep = '\0';
276
277		/* set datbuf to start at first non-whitespace character */
278		while (*datbuf == ' ' || *datbuf == '\t')
279			datbuf++;
280
281		/* Check for silliness. */
282		if (filter_plusminus) {
283			if  (*keybuf == '+' || *keybuf == '-' ||
284			     *datbuf == '+' || *datbuf == '-') {
285				warnx("bad character at "
286				    "start of line: %s", buf);
287				continue;
288			}
289		}
290
291		if (strlen(keybuf) > YPMAXRECORD) {
292			warnx("key too long: %s", keybuf);
293			continue;
294		}
295
296		if (!strlen(keybuf)) {
297			warnx("no key -- check source file for blank lines");
298			continue;
299		}
300
301		if (strlen(datbuf) > YPMAXRECORD) {
302			warnx("data too long: %s", datbuf);
303			continue;
304		}
305
306		key.data = keybuf;
307		key.size = strlen(keybuf);
308		data.data = datbuf;
309		data.size = strlen(datbuf);
310
311		if ((rval = yp_put_record(dbp, &key, &data, 0)) != YP_TRUE) {
312			switch (rval) {
313			case YP_FALSE:
314				warnx("duplicate key '%s' - skipping", keybuf);
315				break;
316			case YP_BADDB:
317			default:
318				err(1,"failed to write new record - exiting");
319				break;
320			}
321		}
322
323	}
324
325	(void)(dbp->close)(dbp);
326
327doclear:
328
329	if (clear) {
330		char in = 0;
331		char *out = NULL;
332		int stat;
333		if ((stat = callrpc("localhost", YPPROG,YPVERS, YPPROC_CLEAR,
334			(xdrproc_t)xdr_void, &in,
335			(xdrproc_t)xdr_void, out)) != RPC_SUCCESS) {
336			warnx("failed to send 'clear' to local ypserv: %s",
337				clnt_sperrno((enum clnt_stat) stat));
338		}
339	}
340
341	exit(0);
342}
343