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