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