xref: /freebsd/usr.sbin/yp_mkdb/yp_mkdb.c (revision 97d92980a96a50750844f420cc225ddf918f0699)
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 #ifndef lint
34 static const char rcsid[] =
35   "$FreeBSD$";
36 #endif /* not lint */
37 
38 #include <err.h>
39 #include <fcntl.h>
40 #include <limits.h>
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <time.h>
45 #include <unistd.h>
46 #include <rpc/rpc.h>
47 #include <rpcsvc/yp.h>
48 #include <sys/param.h>
49 #include <sys/types.h>
50 #include <sys/stat.h>
51 #include "yp_extern.h"
52 #include "ypxfr_extern.h"
53 
54 char *yp_dir = "";	/* No particular default needed. */
55 int _rpcpmstart = 0;
56 int debug = 1;
57 
58 static void usage()
59 {
60 	fprintf(stderr, "%s\n%s\n%s\n%s\n",
61 	"usage: yp_mkdb -c",
62 	"       yp_mkdb -u dbname",
63 	"       yp_mkdb [-c] [-b] [-s] [-f] [-i inputfile] [-o outputfile]",
64 	"               [-d domainname ] [-m mastername] inputfile dbname");
65 	exit(1);
66 }
67 
68 #define PERM_SECURE (S_IRUSR|S_IWUSR)
69 
70 static DB *open_db(path, flags)
71 	char *path;
72 	int flags;
73 {
74 	extern HASHINFO openinfo;
75 
76 	return(dbopen(path, flags, PERM_SECURE, DB_HASH, &openinfo));
77 }
78 
79 static void unwind(map)
80 	char *map;
81 {
82 	DB *dbp;
83 	DBT key, data;
84 
85 	dbp = open_db(map, O_RDONLY);
86 
87 	if (dbp == NULL)
88 		err(1, "open_db(%s) failed", map);
89 
90 	key.data = NULL;
91 	while(yp_next_record(dbp, &key, &data, 1, 1) == YP_TRUE)
92 		printf("%.*s %.*s\n", key.size,key.data,data.size,data.data);
93 
94 	(void)(dbp->close)(dbp);
95 	return;
96 }
97 
98 int main (argc, argv)
99 	int argc;
100 	char *argv[];
101 {
102 	int ch;
103 	int un = 0;
104 	int clear = 0;
105 	int filter_plusminus = 0;
106 	char *infile = NULL;
107 	char *map = NULL;
108 	char *domain = NULL;
109 	char *infilename = NULL;
110 	char *outfilename = NULL;
111 	char *mastername = NULL;
112 	int interdom = 0;
113 	int secure = 0;
114 	DB *dbp;
115 	DBT key, data;
116 	char buf[10240];
117 	char *keybuf, *datbuf;
118 	FILE *ifp;
119 	char hname[MAXHOSTNAMELEN + 2];
120 
121 	while ((ch = getopt(argc, argv, "uhcbsdf:i:o:m:")) != -1) {
122 		switch(ch) {
123 		case 'f':
124 			filter_plusminus++;
125 			break;
126 		case 'u':
127 			un++;
128 			break;
129 		case 'c':
130 			clear++;
131 			break;
132 		case 'b':
133 			interdom++;
134 			break;
135 		case 's':
136 			secure++;
137 			break;
138 		case 'd':
139 			domain = optarg;
140 			break;
141 		case 'i':
142 			infilename = optarg;
143 			break;
144 		case 'o':
145 			outfilename = optarg;
146 			break;
147 		case 'm':
148 			mastername = optarg;
149 			break;
150 		case 'h':
151 		default:
152 			usage();
153 			break;
154 		}
155 	}
156 
157 	argc -= optind;
158 	argv += optind;
159 
160 	if (un) {
161 		map = argv[0];
162 		if (map == NULL)
163 			usage();
164 		unwind(map);
165 		exit(0);
166 
167 	}
168 
169 	infile = argv[0];
170 	map = argv[1];
171 
172 	if (infile == NULL || map == NULL) {
173 		if (clear)
174 			goto doclear;
175 		usage();
176 	}
177 
178 	if (mastername == NULL) {
179 		if (gethostname((char *)&hname, sizeof(hname)) == -1)
180 			err(1, "gethostname() failed");
181 		mastername = (char *)&hname;
182 	}
183 
184 	/*
185 	 * Note that while we can read from stdin, we can't
186 	 * write to stdout; the db library doesn't let you
187 	 * write to a file stream like that.
188 	 */
189 
190 	if (!strcmp(infile, "-")) {
191 		ifp = stdin;
192 	} else {
193 		if ((ifp = fopen(infile, "r")) == NULL)
194 			err(1, "failed to open %s", infile);
195 	}
196 
197 	if ((dbp = open_db(map, O_RDWR|O_EXLOCK|O_EXCL|O_CREAT)) == NULL)
198 		err(1, "open_db(%s) failed", map);
199 
200 	if (interdom) {
201 		key.data = "YP_INTERDOMAIN";
202 		key.size = sizeof("YP_INTERDOMAIN") - 1;
203 		data.data = "";
204 		data.size = 0;
205 		yp_put_record(dbp, &key, &data, 0);
206 	}
207 
208 	if (secure) {
209 		key.data = "YP_SECURE";
210 		key.size = sizeof("YP_SECURE") - 1;
211 		data.data = "";
212 		data.size = 0;
213 		yp_put_record(dbp, &key, &data, 0);
214 	}
215 
216 	key.data = "YP_MASTER_NAME";
217 	key.size = sizeof("YP_MASTER_NAME") - 1;
218 	data.data = mastername;
219 	data.size = strlen(mastername);
220 	yp_put_record(dbp, &key, &data, 0);
221 
222 	key.data = "YP_LAST_MODIFIED";
223 	key.size = sizeof("YP_LAST_MODIFIED") - 1;
224 	snprintf(buf, sizeof(buf), "%lu", time(NULL));
225 	data.data = (char *)&buf;
226 	data.size = strlen(buf);
227 	yp_put_record(dbp, &key, &data, 0);
228 
229 	if (infilename) {
230 		key.data = "YP_INPUT_FILE";
231 		key.size = sizeof("YP_INPUT_FILE") - 1;
232 		data.data = infilename;
233 		data.size = strlen(infilename);
234 		yp_put_record(dbp, &key, &data, 0);
235 	}
236 
237 	if (outfilename) {
238 		key.data = "YP_OUTPUT_FILE";
239 		key.size = sizeof("YP_OUTPUT_FILE") - 1;
240 		data.data = outfilename;
241 		data.size = strlen(outfilename);
242 		yp_put_record(dbp, &key, &data, 0);
243 	}
244 
245 	if (domain) {
246 		key.data = "YP_DOMAIN_NAME";
247 		key.size = sizeof("YP_DOMAIN_NAME") - 1;
248 		data.data = domain;
249 		data.size = strlen(domain);
250 		yp_put_record(dbp, &key, &data, 0);
251 	}
252 
253 	while(fgets((char *)&buf, sizeof(buf), ifp)) {
254 		char *sep = NULL;
255 		int rval;
256 
257 		/* NUL terminate */
258 		if ((sep = strchr(buf, '\n')))
259 			*sep = '\0';
260 
261 		/* handle backslash line continuations */
262 		while(buf[strlen(buf) - 1] == '\\') {
263 			fgets((char *)&buf[strlen(buf) - 1],
264 					sizeof(buf) - strlen(buf), ifp);
265 			if ((sep = strchr(buf, '\n')))
266 				*sep = '\0';
267 		}
268 
269 		/* find the separation between the key and data */
270 		if ((sep = strpbrk(buf, " \t")) == NULL) {
271 			warnx("bad input -- no white space: %s", buf);
272 			continue;
273 		}
274 
275 		/* separate the strings */
276 		keybuf = (char *)&buf;
277 		datbuf = sep + 1;
278 		*sep = '\0';
279 
280 		/* set datbuf to start at first non-whitespace character */
281 		while (*datbuf == ' ' || *datbuf == '\t')
282 			datbuf++;
283 
284 		/* Check for silliness. */
285 		if (filter_plusminus) {
286 			if  (*keybuf == '+' || *keybuf == '-' ||
287 			     *datbuf == '+' || *datbuf == '-') {
288 				warnx("bad character at "
289 				    "start of line: %s", buf);
290 				continue;
291 			}
292 		}
293 
294 		if (strlen(keybuf) > YPMAXRECORD) {
295 			warnx("key too long: %s", keybuf);
296 			continue;
297 		}
298 
299 		if (!strlen(keybuf)) {
300 			warnx("no key -- check source file for blank lines");
301 			continue;
302 		}
303 
304 		if (strlen(datbuf) > YPMAXRECORD) {
305 			warnx("data too long: %s", datbuf);
306 			continue;
307 		}
308 
309 		key.data = keybuf;
310 		key.size = strlen(keybuf);
311 		data.data = datbuf;
312 		data.size = strlen(datbuf);
313 
314 		if ((rval = yp_put_record(dbp, &key, &data, 0)) != YP_TRUE) {
315 			switch(rval) {
316 			case YP_FALSE:
317 				warnx("duplicate key '%s' - skipping", keybuf);
318 				break;
319 			case YP_BADDB:
320 			default:
321 				err(1,"failed to write new record - exiting");
322 				break;
323 			}
324 		}
325 
326 	}
327 
328 	(void)(dbp->close)(dbp);
329 
330 doclear:
331 
332 	if (clear) {
333 		char in = 0;
334 		char *out = NULL;
335 		int stat;
336 		if ((stat = callrpc("localhost",YPPROG,YPVERS,YPPROC_CLEAR,
337 			xdr_void, (void *)&in,
338 			xdr_void, (void *)out)) != RPC_SUCCESS) {
339 			warnx("failed to send 'clear' to local ypserv: %s",
340 				clnt_sperrno((enum clnt_stat) stat));
341 		}
342 	}
343 
344 	exit(0);
345 }
346