1 /* $OpenBSD: mkalias.c,v 1.29 2016/09/22 00:09:47 jsg Exp $ */
2
3 /*
4 * Copyright (c) 1997 Mats O Jansson <moj@stacken.kth.se>
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 <ctype.h>
30 #include <fcntl.h>
31 #include <netdb.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <unistd.h>
36 #include <sys/types.h>
37 #include <netinet/in.h>
38 #include <arpa/nameser.h>
39 #include <resolv.h>
40 #include <err.h>
41 #include <time.h>
42 #include "ypdb.h"
43 #include "ypdef.h"
44
45 static void
split_address(char * address,size_t len,char * user,char * host)46 split_address(char *address, size_t len, char *user, char *host)
47 {
48 char *c, *s, *r;
49 size_t i = 0;
50
51 if (memchr(address, '@', len)) {
52 s = user;
53 for (c = address; i < len; i++) {
54 if (*c == '@') {
55 *s = '\0';
56 s = host;
57 } else {
58 *s++ = *c;
59 }
60 c++;
61 }
62 *s = '\0';
63 }
64
65 if ((r = memrchr(address, '!', len))) {
66 s = host;
67 for (c = address; i < len; i++) {
68 if (c == r) {
69 *s = '\0';
70 s = user;
71 } else {
72 *s++ = *c;
73 }
74 c++;
75 }
76 *s = '\0';
77 }
78 }
79
80 static int
check_host(char * address,size_t len,char * host,int dflag,int uflag,int Eflag)81 check_host(char *address, size_t len, char *host, int dflag, int uflag, int Eflag)
82 {
83 union {
84 HEADER hdr;
85 u_char buf[PACKETSZ];
86 } answer;
87 int status;
88
89 if ((dflag && memchr(address, '@', len)) ||
90 (uflag && memchr(address, '!', len)))
91 return(0);
92
93 if ((_res.options & RES_INIT) == 0)
94 res_init();
95
96 status = res_search(host, C_IN, T_AAAA, answer.buf, sizeof(answer.buf));
97
98 if (status == -1)
99 status = res_search(host, C_IN, T_A, answer.buf, sizeof(answer.buf));
100
101 if (status == -1 && Eflag)
102 status = res_search(host, C_IN, T_MX, answer.buf, sizeof(answer.buf));
103
104 return(status == -1);
105 }
106
107 static void
capitalize(char * name,int len)108 capitalize(char *name, int len)
109 {
110 char last = ' ';
111 char *c;
112 int i = 0;
113
114 for (c = name; i < len; i++) {
115 if (*c == '.')
116 last = '.';
117 c++;
118 }
119
120 i = 0;
121 if (last == '.') {
122 for (c = name; i < len; i++) {
123 if (last == '.')
124 *c = (char)toupper(*c);
125 last = *c++;
126 }
127 }
128 }
129
130 static void
usage(void)131 usage(void)
132 {
133 fprintf(stderr,
134 "usage: mkalias [-nv] [-E | -e [-du]] input [output]\n");
135 exit(1);
136 }
137
138 int
main(int argc,char * argv[])139 main(int argc, char *argv[])
140 {
141 int eflag = 0, dflag = 0, nflag = 0;
142 int uflag = 0, vflag = 0, Eflag = 0;
143 int status, ch, fd;
144 char *input = NULL, *output = NULL;
145 DBM *db;
146 datum key, val;
147 DBM *new_db = NULL;
148 static char mapname[] = "ypdbXXXXXXXXXX";
149 char db_mapname[PATH_MAX], db_outfile[PATH_MAX];
150 char db_tempname[PATH_MAX];
151 char user[4096], host[4096]; /* XXX: DB bsize = 4096 in ypdb.c */
152 char myname[HOST_NAME_MAX+1], datestr[11], *slash;
153
154 while ((ch = getopt(argc, argv, "Edensuv")) != -1)
155 switch (ch) {
156 case 'E':
157 eflag = 1; /* Check hostname */
158 Eflag = 1; /* .. even check MX records */
159 break;
160 case 'd':
161 dflag = 1; /* Don't check DNS hostname */
162 break;
163 case 'e':
164 eflag = 1; /* Check hostname */
165 break;
166 case 'n':
167 nflag = 1; /* Capitalize name parts */
168 break;
169 case 's': /* Ignore */
170 break;
171 case 'u':
172 uflag = 1; /* Don't check UUCP hostname */
173 break;
174 case 'v':
175 vflag = 1; /* Verbose */
176 break;
177 default:
178 usage();
179 break;
180 }
181
182 if (optind == argc)
183 usage();
184
185 input = argv[optind++];
186 if (optind < argc)
187 output = argv[optind++];
188 if (optind < argc)
189 usage();
190
191 db = ypdb_open(input, O_RDONLY, 0444);
192 if (db == NULL) {
193 err(1, "Unable to open input database %s", input);
194 /* NOTREACHED */
195 }
196
197 if (output != NULL) {
198 if (strlen(output) + strlen(YPDB_SUFFIX) > PATH_MAX) {
199 errx(1, "%s: file name too long", output);
200 /* NOTREACHED */
201 }
202
203 snprintf(db_outfile, sizeof(db_outfile),
204 "%s%s", output, YPDB_SUFFIX);
205
206 slash = strrchr(output, '/');
207 if (slash != NULL)
208 slash[1] = 0; /* truncate to dir */
209 else
210 *output = 0; /* eliminate */
211
212 /* note: output is now directory where map goes ! */
213
214 if (strlen(output) + strlen(mapname) +
215 strlen(YPDB_SUFFIX) > PATH_MAX) {
216 errx(1, "%s: directory name too long", output);
217 /* NOTREACHED */
218 }
219
220 snprintf(db_tempname, sizeof(db_tempname), "%s%s%s", output,
221 mapname, YPDB_SUFFIX);
222 fd = mkstemps(db_tempname, 3);
223 if (fd == -1)
224 goto fail;
225 close(fd);
226
227 strncpy(db_mapname, db_tempname, strlen(db_tempname) - 3);
228 db_mapname[sizeof(db_mapname) - 1] = '\0';
229
230 new_db = ypdb_open(db_mapname, O_RDWR|O_TRUNC, 0444);
231 if (new_db == NULL) {
232 fail:
233 if (fd != -1)
234 unlink(db_tempname);
235 err(1, "Unable to open output database %s",
236 db_outfile);
237 /* NOTREACHED */
238 }
239 }
240
241 for (key = ypdb_firstkey(db); key.dptr != NULL;
242 key = ypdb_nextkey(db)) {
243 val = ypdb_fetch(db, key);
244
245 if (val.dptr == NULL)
246 continue; /* No value */
247 if (*key.dptr == '@' && key.dsize == 1)
248 continue; /* Sendmail token */
249 if (strncmp(key.dptr, "YP_", 3)==0) /* YP token */
250 continue;
251 if (memchr(val.dptr, ',', val.dsize))
252 continue; /* List... */
253 if (memchr(val.dptr, '|', val.dsize))
254 continue; /* Pipe... */
255
256 if (!(memchr(val.dptr, '@', val.dsize) ||
257 memchr(val.dptr, '!', val.dsize)))
258 continue; /* Skip local users */
259
260 split_address(val.dptr, val.dsize, user, host);
261
262 if (eflag && check_host(val.dptr, val.dsize, host, dflag, uflag, Eflag)) {
263 warnx("Invalid host %s in %*.*s:%*.*s",
264 host, key.dsize, key.dsize, key.dptr,
265 val.dsize, val.dsize, val.dptr);
266 continue;
267 }
268
269 if (nflag)
270 capitalize(key.dptr, key.dsize);
271
272 if (new_db != NULL) {
273 status = ypdb_store(new_db, val, key, YPDB_INSERT);
274 if (status != 0) {
275 warnx("problem storing %*.*s %*.*s",
276 val.dsize, val.dsize, val.dptr,
277 key.dsize, key.dsize, key.dptr);
278 }
279 }
280
281 if (vflag) {
282 printf("%*.*s --> %*.*s\n",
283 val.dsize, val.dsize, val.dptr,
284 key.dsize, key.dsize, key.dptr);
285 }
286
287 }
288
289 if (new_db != NULL) {
290 snprintf(datestr, sizeof datestr, "%010lld",
291 (long long)time(NULL));
292 key.dptr = YP_LAST_KEY;
293 key.dsize = strlen(YP_LAST_KEY);
294 val.dptr = datestr;
295 val.dsize = strlen(datestr);
296 status = ypdb_store(new_db, key, val, YPDB_INSERT);
297 if (status != 0) {
298 warnx("problem storing %*.*s %*.*s",
299 key.dsize, key.dsize, key.dptr,
300 val.dsize, val.dsize, val.dptr);
301 }
302 }
303
304 if (new_db != NULL) {
305 gethostname(myname, sizeof(myname));
306 key.dptr = YP_MASTER_KEY;
307 key.dsize = strlen(YP_MASTER_KEY);
308 val.dptr = myname;
309 val.dsize = strlen(myname);
310 status = ypdb_store(new_db, key, val, YPDB_INSERT);
311 if (status != 0) {
312 warnx("problem storing %*.*s %*.*s",
313 key.dsize, key.dsize, key.dptr,
314 val.dsize, val.dsize, val.dptr);
315 }
316 }
317
318 ypdb_close(db);
319
320 if (new_db != NULL) {
321 ypdb_close(new_db);
322 if (rename(db_tempname, db_outfile) < 0) {
323 err(1, "rename %s -> %s failed", db_tempname,
324 db_outfile);
325 /* NOTREACHED */
326 }
327 }
328 return(0);
329 }
330