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