Merge branch 'vendor/LESS' into less_update
[dragonfly.git] / usr.sbin / yp_mkdb / yp_mkdb.c
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  * $FreeBSD: src/usr.sbin/yp_mkdb/yp_mkdb.c,v 1.12.2.1 2002/02/15 00:46:59 des Exp $
33  * $DragonFly: src/usr.sbin/yp_mkdb/yp_mkdb.c,v 1.4 2005/11/24 22:23:02 swildner Exp $
34  */
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
52 const char *yp_dir = "";        /* No particular default needed. */
53 int _rpcpmstart = 0;
54 int debug = 1;
55
56 static void
57 usage(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
69 static DB *
70 open_db(char *path, int flags)
71 {
72         extern HASHINFO openinfo;
73
74         return(dbopen(path, flags, PERM_SECURE, DB_HASH, &openinfo));
75 }
76
77 static void
78 unwind(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         dbp->close(dbp);
93         return;
94 }
95
96 int
97 main(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         dbp->close(dbp);
326
327 doclear:
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, (void *)&in,
335                         (xdrproc_t)xdr_void, (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 }