1 /* 2 * Copyright (c) 1995 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 * $Id$ 33 */ 34 #include <stdio.h> 35 #include <stdlib.h> 36 #include <unistd.h> 37 #include <string.h> 38 #include <syslog.h> 39 #include <errno.h> 40 #include <sys/types.h> 41 #include <sys/param.h> 42 #include <sys/socket.h> 43 #include <netinet/in.h> 44 #include <arpa/inet.h> 45 #include <rpc/rpc.h> 46 #include <rpc/clnt.h> 47 #include <rpcsvc/yp.h> 48 struct dom_binding {}; 49 #include <rpcsvc/ypclnt.h> 50 #include <rpcsvc/ypxfrd.h> 51 #include "ypxfr_extern.h" 52 53 #ifndef lint 54 static const char rcsid[] = "$Id$"; 55 #endif 56 57 char *progname = "ypxfr"; 58 char *yp_dir = _PATH_YP; 59 int _rpcpmstart = 0; 60 int ypxfr_use_yplib = 0; /* Assume the worst. */ 61 int ypxfr_clear = 1; 62 int ypxfr_prognum = 0; 63 struct sockaddr_in ypxfr_callback_addr; 64 struct yppushresp_xfr ypxfr_resp; 65 DB *dbp; 66 67 static void ypxfr_exit(retval, temp) 68 ypxfrstat retval; 69 char *temp; 70 { 71 CLIENT *clnt; 72 int sock = RPC_ANYSOCK; 73 struct timeval timeout; 74 75 /* Clean up no matter what happened previously. */ 76 if (temp != NULL) { 77 if (dbp != NULL) 78 (void)(dbp->close)(dbp); 79 if (unlink(temp) == -1) { 80 yp_error("failed to unlink %s",strerror(errno)); 81 } 82 } 83 84 if (_rpcpmstart) { 85 timeout.tv_sec = 20; 86 timeout.tv_usec = 0; 87 88 if ((clnt = clntudp_create(&ypxfr_callback_addr, ypxfr_prognum, 89 1, timeout, &sock)) == NULL) { 90 yp_error("%s", clnt_spcreateerror("failed to \ 91 establish callback handle")); 92 exit(1); 93 } 94 95 ypxfr_resp.status = retval; 96 97 if (yppushproc_xfrresp_1(&ypxfr_resp, clnt) == NULL) { 98 yp_error("%s", clnt_sperror(clnt, "callback failed")); 99 clnt_destroy(clnt); 100 exit(1); 101 } 102 clnt_destroy(clnt); 103 } else { 104 yp_error("Exiting: %s", ypxfrerr_string(retval)); 105 } 106 107 exit(0); 108 } 109 110 static void usage() 111 { 112 if (_rpcpmstart) { 113 ypxfr_exit(YPXFR_BADARGS,NULL); 114 } else { 115 fprintf(stderr,"usage: %s [-f] [-c] [-d target domain] \ 116 [-h source host] [-s source domain]\n", progname); 117 fprintf(stderr,"\t [-p path] [-C taskid program-number \ 118 ipaddr port] mapname\n"); 119 exit(1); 120 } 121 } 122 123 int ypxfr_foreach(status, key, keylen, val, vallen, data) 124 int status; 125 char *key; 126 int keylen; 127 char *val; 128 int vallen; 129 char *data; 130 { 131 DBT dbkey, dbval; 132 133 if (status != YP_TRUE) 134 return (status); 135 136 dbkey.data = key; 137 dbkey.size = keylen; 138 dbval.data = val; 139 dbval.size = vallen; 140 141 if (yp_put_record(dbp, &dbkey, &dbval, 0) != YP_TRUE) 142 return(yp_errno); 143 144 return (0); 145 } 146 147 main(argc,argv) 148 int argc; 149 char *argv[]; 150 { 151 int ch; 152 int ypxfr_force = 0; 153 char *ypxfr_dest_domain = NULL; 154 char *ypxfr_source_host = NULL; 155 char *ypxfr_source_domain = NULL; 156 char *ypxfr_local_domain = NULL; 157 char *ypxfr_master = NULL; 158 unsigned long ypxfr_order = -1, ypxfr_skew_check = -1; 159 char *ypxfr_mapname = NULL; 160 int ypxfr_args = 0; 161 char ypxfr_temp_map[MAXPATHLEN + 2]; 162 char tempmap[MAXPATHLEN + 2]; 163 char buf[MAXPATHLEN + 2]; 164 DBT key, data; 165 int remoteport; 166 int interdom = 0; 167 int secure = 0; 168 169 debug = 1; 170 171 if (!isatty(fileno(stderr))) { 172 openlog(progname, LOG_PID, LOG_DAEMON); 173 _rpcpmstart = 1; 174 } 175 176 if (argc < 2) 177 usage(); 178 179 while ((ch = getopt(argc, argv, "fcd:h:s:p:C:")) != EOF) { 180 int my_optind; 181 switch(ch) { 182 case 'f': 183 ypxfr_force++; 184 ypxfr_args++; 185 break; 186 case 'c': 187 ypxfr_clear = 0; 188 ypxfr_args++; 189 break; 190 case 'd': 191 ypxfr_dest_domain = optarg; 192 ypxfr_args += 2; 193 break; 194 case 'h': 195 ypxfr_source_host = optarg; 196 ypxfr_args += 2; 197 break; 198 case 's': 199 ypxfr_source_domain = optarg; 200 ypxfr_args += 2; 201 break; 202 case 'p': 203 yp_dir = optarg; 204 ypxfr_args += 2; 205 break; 206 case 'C': 207 /* 208 * Whoever decided that the -C flag should take 209 * four arguments is a twit. 210 */ 211 my_optind = optind - 1; 212 if (argv[my_optind] == NULL || !strlen(argv[my_optind])) { 213 yp_error("transaction ID not specified"); 214 usage(); 215 } 216 ypxfr_resp.transid = atol(argv[my_optind]); 217 my_optind++; 218 if (argv[my_optind] == NULL || !strlen(argv[my_optind])) { 219 yp_error("RPC program number not specified"); 220 usage(); 221 } 222 ypxfr_prognum = atol(argv[my_optind]); 223 my_optind++; 224 if (argv[my_optind] == NULL || !strlen(argv[my_optind])) { 225 yp_error("address not specified"); 226 usage(); 227 } 228 if (!inet_aton(argv[my_optind], &ypxfr_callback_addr.sin_addr)) { 229 yp_error("failed to convert '%s' to IP addr", 230 argv[my_optind]); 231 exit(1); 232 } 233 my_optind++; 234 if (argv[my_optind] == NULL || !strlen(argv[my_optind])) { 235 yp_error("port not specified"); 236 usage(); 237 } 238 ypxfr_callback_addr.sin_port = htons((u_short)atoi(argv[my_optind])); 239 ypxfr_args += 5; 240 break; 241 default: 242 usage(); 243 break; 244 } 245 } 246 247 ypxfr_mapname = argv[ypxfr_args + 1]; 248 249 if (ypxfr_mapname == NULL) { 250 yp_error("no map name specified"); 251 usage(); 252 } 253 254 /* Always the case. */ 255 ypxfr_callback_addr.sin_family = AF_INET; 256 257 /* Determine if local NIS client facilities are turned on. */ 258 if (!yp_get_default_domain(&ypxfr_local_domain) && 259 _yp_check(&ypxfr_local_domain)) 260 ypxfr_use_yplib = 1; 261 262 /* 263 * If no destination domain is specified, assume that the 264 * local default domain is to be used and try to obtain it. 265 * Fails if NIS client facilities are turned off. 266 */ 267 if (ypxfr_dest_domain == NULL) { 268 if (ypxfr_use_yplib) { 269 yp_get_default_domain(&ypxfr_dest_domain); 270 } else { 271 yp_error("no destination domain specified and \ 272 the local domain name isn't set"); 273 ypxfr_exit(YPXFR_BADARGS,NULL); 274 } 275 } 276 277 /* 278 * If a source domain is not specified, assume it to 279 * be the same as the destination domain. 280 */ 281 if (ypxfr_source_domain == NULL) { 282 ypxfr_source_domain = ypxfr_dest_domain; 283 } 284 285 /* 286 * If the source host is not specified, assume it to be the 287 * master for the specified map. If local NIS client facilities 288 * are turned on, we can figure this out using yp_master(). 289 * If not, we have to see if a local copy of the map exists 290 * and extract its YP_MASTER_NAME record. If _that_ fails, 291 * we are stuck and must ask the user for more information. 292 */ 293 if (ypxfr_source_host == NULL) { 294 if (!ypxfr_use_yplib) { 295 /* 296 * Double whammy: NIS isn't turned on and the user 297 * didn't specify a source host. 298 */ 299 char *dptr; 300 key.data = "YP_MASTER_NAME"; 301 key.size = sizeof("YP_MASTER_NAME") - 1; 302 303 if (yp_get_record(ypxfr_dest_domain, ypxfr_mapname, 304 &key, &data, 1) != YP_TRUE) { 305 yp_error("no source host specified"); 306 ypxfr_exit(YPXFR_BADARGS,NULL); 307 } 308 dptr = data.data; 309 dptr[data.size] = '\0'; 310 ypxfr_master = ypxfr_source_host = strdup(dptr); 311 } 312 } else { 313 if (ypxfr_use_yplib) 314 ypxfr_use_yplib = 0; 315 } 316 317 if (ypxfr_master == NULL) { 318 if ((ypxfr_master = ypxfr_get_master(ypxfr_source_domain, 319 ypxfr_mapname, 320 ypxfr_source_host, 321 ypxfr_use_yplib)) == NULL) { 322 yp_error("failed to find master of %s in domain %s: %s", 323 ypxfr_mapname, ypxfr_source_domain, 324 ypxfrerr_string(yp_errno)); 325 ypxfr_exit(YPXFR_MADDR,NULL); 326 } 327 } 328 329 /* 330 * If we got here and ypxfr_source_host is still undefined, 331 * it means we had to resort to using yp_master() to find the 332 * master server for the map. The source host and master should 333 * be identical. 334 */ 335 if (ypxfr_source_host == NULL) 336 ypxfr_source_host = ypxfr_master; 337 338 /* 339 * Don't talk to ypservs on unprivileged ports. 340 */ 341 remoteport = getrpcport(ypxfr_source_host, YPPROG, YPVERS, IPPROTO_UDP); 342 if (remoteport >= IPPORT_RESERVED) { 343 yp_error("ypserv on %s not running on reserved port", 344 ypxfr_source_host); 345 ypxfr_exit(YPXFR_REFUSED, NULL); 346 } 347 348 if ((ypxfr_order = ypxfr_get_order(ypxfr_source_domain, 349 ypxfr_mapname, 350 ypxfr_master, 0)) == 0) { 351 yp_error("failed to get order number of %s: %s", 352 ypxfr_mapname, yp_errno == YPXFR_SUCC ? 353 "map has order 0" : ypxfrerr_string(yp_errno)); 354 ypxfr_exit(YPXFR_YPERR,NULL); 355 } 356 357 if (ypxfr_match(ypxfr_master, ypxfr_source_domain, ypxfr_mapname, 358 "YP_INTERDOMAIN", sizeof("YP_INTERDOMAIN") - 1)) 359 interdom++; 360 361 if (ypxfr_match(ypxfr_master, ypxfr_source_domain, ypxfr_mapname, 362 "YP_SECURE", sizeof("YP_SECURE") - 1)) 363 secure++; 364 365 key.data = "YP_LAST_MODIFIED"; 366 key.size = sizeof("YP_LAST_MODIFIED") - 1; 367 368 /* The order number is immaterial when the 'force' flag is set. */ 369 370 if (!ypxfr_force) { 371 int ignore = 0; 372 if (yp_get_record(ypxfr_dest_domain,ypxfr_mapname,&key,&data,1) != YP_TRUE) { 373 switch(yp_errno) { 374 case YP_NOKEY: 375 ypxfr_exit(YPXFR_FORCE,NULL); 376 break; 377 case YP_NOMAP: 378 /* 379 * If the map doesn't exist, we're 380 * creating it. Ignore the error. 381 */ 382 ignore++; 383 break; 384 case YP_BADDB: 385 default: 386 ypxfr_exit(YPXFR_DBM,NULL); 387 break; 388 } 389 } 390 if (!ignore && ypxfr_order <= atoi(data.data)) 391 ypxfr_exit(YPXFR_AGE, NULL); 392 393 } 394 395 /* Construct a temporary map file name */ 396 snprintf(tempmap, sizeof(tempmap), "%s.%d",ypxfr_mapname, getpid()); 397 snprintf(ypxfr_temp_map, sizeof(ypxfr_temp_map), "%s/%s/%s", yp_dir, 398 ypxfr_dest_domain, tempmap); 399 400 if ((remoteport = getrpcport(ypxfr_source_host, YPXFRD_FREEBSD_PROG, 401 YPXFRD_FREEBSD_VERS, IPPROTO_TCP))) { 402 403 /* Don't talk to rpc.ypxfrds on unprovileged ports. */ 404 if (remoteport >= IPPORT_RESERVED) { 405 yp_error("rpc.ypxfrd on %s not using privileged port", 406 ypxfr_source_host); 407 ypxfr_exit(YPXFR_REFUSED, NULL); 408 } 409 410 /* Try to send using ypxfrd. If it fails, use old method. */ 411 if (!ypxfrd_get_map(ypxfr_source_host, ypxfr_mapname, 412 ypxfr_source_domain, ypxfr_temp_map)) 413 goto leave; 414 } 415 416 /* Open the temporary map read/write. */ 417 if ((dbp = yp_open_db_rw(ypxfr_dest_domain, tempmap, 0)) == NULL) { 418 yp_error("failed to open temporary map file"); 419 ypxfr_exit(YPXFR_DBM,NULL); 420 } 421 422 /* 423 * Fill in the keys we already know, such as the order number, 424 * master name, input file name (we actually make up a bogus 425 * name for that) and output file name. 426 */ 427 snprintf(buf, sizeof(buf), "%d", ypxfr_order); 428 data.data = buf; 429 data.size = strlen(buf); 430 431 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 432 yp_error("failed to write order number to database"); 433 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 434 } 435 436 key.data = "YP_MASTER_NAME"; 437 key.size = sizeof("YP_MASTER_NAME") - 1; 438 data.data = ypxfr_master; 439 data.size = strlen(ypxfr_master); 440 441 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 442 yp_error("failed to write master name to database"); 443 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 444 } 445 446 key.data = "YP_DOMAIN_NAME"; 447 key.size = sizeof("YP_DOMAIN_NAME") - 1; 448 data.data = ypxfr_dest_domain; 449 data.size = strlen(ypxfr_dest_domain); 450 451 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 452 yp_error("failed to write domain name to database"); 453 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 454 } 455 456 snprintf (buf, sizeof(buf), "%s:%s", ypxfr_source_host, ypxfr_mapname); 457 458 key.data = "YP_INPUT_NAME"; 459 key.size = sizeof("YP_INPUT_NAME") - 1; 460 data.data = &buf; 461 data.size = strlen(buf); 462 463 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 464 yp_error("failed to write input name to database"); 465 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 466 467 } 468 469 snprintf(buf, sizeof(buf), "%s/%s/%s", yp_dir, ypxfr_dest_domain, 470 ypxfr_mapname); 471 472 key.data = "YP_OUTPUT_NAME"; 473 key.size = sizeof("YP_OUTPUT_NAME") - 1; 474 data.data = &buf; 475 data.size = strlen(buf); 476 477 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 478 yp_error("failed to write output name to database"); 479 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 480 } 481 482 if (interdom) { 483 key.data = "YP_INTERDOMAIN"; 484 key.size = sizeof("YP_INTERDOMAIN") - 1; 485 data.data = ""; 486 data.size = 0; 487 488 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 489 yp_error("failed to add interdomain flag to database"); 490 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 491 } 492 } 493 494 if (secure) { 495 key.data = "YP_SECURE"; 496 key.size = sizeof("YP_SECURE") - 1; 497 data.data = ""; 498 data.size = 0; 499 500 if (yp_put_record(dbp, &key, &data, 0) != YP_TRUE) { 501 yp_error("failed to add secure flag to database"); 502 ypxfr_exit(YPXFR_DBM,&ypxfr_temp_map); 503 } 504 } 505 506 /* Now suck over the contents of the map from the master. */ 507 508 if (ypxfr_get_map(ypxfr_mapname,ypxfr_source_domain, 509 ypxfr_source_host, ypxfr_foreach)){ 510 yp_error("failed to retrieve map from source host"); 511 ypxfr_exit(YPXFR_YPERR,&ypxfr_temp_map); 512 } 513 514 (void)(dbp->close)(dbp); 515 dbp = NULL; /* <- yes, it seems this is necessary. */ 516 517 leave: 518 519 snprintf(buf, sizeof(buf), "%s/%s/%s", yp_dir, ypxfr_dest_domain, 520 ypxfr_mapname); 521 522 /* Peek at the order number again and check for skew. */ 523 if ((ypxfr_skew_check = ypxfr_get_order(ypxfr_source_domain, 524 ypxfr_mapname, 525 ypxfr_master, 0)) == 0) { 526 yp_error("failed to get order number of %s: %s", 527 ypxfr_mapname, yp_errno == YPXFR_SUCC ? 528 "map has order 0" : ypxfrerr_string(yp_errno)); 529 ypxfr_exit(YPXFR_YPERR,&ypxfr_temp_map); 530 } 531 532 if (ypxfr_order != ypxfr_skew_check) 533 ypxfr_exit(YPXFR_SKEW,&ypxfr_temp_map); 534 535 /* 536 * Send a YPPROC_CLEAR to the local ypserv. 537 */ 538 if (ypxfr_clear) { 539 char in = 0; 540 char *out = NULL; 541 int stat; 542 if ((stat = callrpc("localhost",YPPROG,YPVERS,YPPROC_CLEAR, 543 xdr_void, (void *)&in, 544 xdr_void, (void *)out)) != RPC_SUCCESS) { 545 yp_error("failed to send 'clear' to local ypserv: %s", 546 clnt_sperrno((enum clnt_stat) stat)); 547 ypxfr_exit(YPXFR_CLEAR, &ypxfr_temp_map); 548 } 549 } 550 551 /* 552 * Put the new map in place immediately. I'm not sure if the 553 * kernel does an unlink() and rename() atomically in the event 554 * that we move a new copy of a map over the top of an existing 555 * one, but there's less chance of a race condition happening 556 * than if we were to do the unlink() ourselves. 557 */ 558 if (rename(ypxfr_temp_map, buf) == -1) { 559 yp_error("rename(%s,%s) failed: %s", ypxfr_temp_map, buf, 560 strerror(errno)); 561 ypxfr_exit(YPXFR_FILE,NULL); 562 } 563 564 ypxfr_exit(YPXFR_SUCC,NULL); 565 566 return(1); 567 } 568