1/*	$NetBSD: ypmatch.c,v 1.20 2012/03/02 17:27:49 christos Exp $	*/
2
3/*
4 * Copyright (c) 1992, 1993 Theo de Raadt <deraadt@fsa.ca>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
17 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
20 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30#ifndef lint
31__RCSID("$NetBSD: ypmatch.c,v 1.20 2012/03/02 17:27:49 christos Exp $");
32#endif
33
34#include <sys/param.h>
35#include <sys/types.h>
36#include <sys/socket.h>
37#include <ctype.h>
38#include <err.h>
39#include <stdio.h>
40#include <stdlib.h>
41#include <string.h>
42#include <unistd.h>
43
44#include <rpc/rpc.h>
45#include <rpc/xdr.h>
46#include <rpcsvc/yp_prot.h>
47#include <rpcsvc/ypclnt.h>
48
49#include "ypalias_init.h"
50
51static void	usage(void) __attribute__((__noreturn__));
52
53int
54main(int argc, char *argv[])
55{
56	char *domainname, *b_retry_cnt;
57	char *inkey, *outbuf;
58	const char *inmap;
59	int outbuflen, key, null, notrans;
60	int c, r, len;
61	size_t i;
62	int rval;
63	const struct ypalias *ypaliases;
64
65	setprogname(*argv);
66	domainname = b_retry_cnt = NULL;
67	notrans = key = null = 0;
68	ypaliases = ypalias_init();
69	while ((c = getopt(argc, argv, "bd:ktxz")) != -1) {
70		switch (c) {
71		case 'x':
72			for(i = 0; ypaliases[i].alias; i++)
73				printf("Use \"%s\" for \"%s\"\n",
74					ypaliases[i].alias,
75					ypaliases[i].name);
76			return 0;
77
78		case 'b':
79			b_retry_cnt = optarg;
80			break;
81
82		case 'd':
83			domainname = optarg;
84			break;
85
86		case 't':
87			notrans++;
88			break;
89
90		case 'k':
91			key++;
92			break;
93
94		case 'z':
95			null++;
96			break;
97
98		default:
99			usage();
100		}
101	}
102
103	argc -= optind;
104	argv += optind;
105
106	if (argc < 2)
107		usage();
108
109	if (b_retry_cnt != NULL) {
110		char *s;
111		unsigned long l;
112
113		l = strtoul(b_retry_cnt, &s, 10);
114		if (*s != '\0' || l > 0xffff) usage();
115		yp_setbindtries((int)l);
116	}
117
118	if (domainname == NULL)
119		yp_get_default_domain(&domainname);
120
121	inmap = argv[argc - 1];
122	if (notrans == 0) {
123		for (i = 0; ypaliases[i].alias; i++)
124			if (strcmp(inmap, ypaliases[i].alias) == 0)
125				inmap = ypaliases[i].name;
126	}
127
128	rval = 0;
129	for(c = 0; c < (argc - 1); c++) {
130		inkey = argv[c];
131
132		len = strlen(inkey);
133		if (null)
134			len++;
135		r = yp_match(domainname, inmap, inkey, len,
136		    &outbuf, &outbuflen);
137		switch (r) {
138		case 0:
139			if (key)
140				printf("%s: ", inkey);
141			fwrite(outbuf, outbuflen, 1, stdout);
142			putc('\n', stdout);
143			break;
144
145		case YPERR_YPBIND:
146			errx(1, "not running ypbind");
147
148		default:
149			warnx("can't match key %s in map %s.  Reason: %s",
150			    inkey, inmap, yperr_string(r));
151			rval = 1;
152			break;
153		}
154	}
155
156	return rval;
157}
158
159static void
160usage(void)
161{
162
163	(void)fprintf(stderr, "Usage: %s [-ktz] [-b <num-retry>] "
164	    "[-d <domainname>] <key> ... <mapname>\n", getprogname());
165	(void)fprintf(stderr, "       %s -x\n", getprogname());
166	exit(1);
167}
168