1 /* $OpenBSD: mkalias.c,v 1.26 2013/12/05 14:18:54 jca 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/param.h> 37 #include <sys/types.h> 38 #include <netinet/in.h> 39 #include <arpa/nameser.h> 40 #include <resolv.h> 41 #include <err.h> 42 #include "ypdb.h" 43 #include "ypdef.h" 44 45 static void 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 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 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 131 usage(void) 132 { 133 fprintf(stderr, 134 "usage: mkalias [-nv] [-E | -e [-du]] input [output]\n"); 135 exit(1); 136 } 137 138 int 139 main(int argc, char *argv[]) 140 { 141 int eflag = 0, dflag = 0, nflag = 0, sflag = 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[MAXPATHLEN], db_outfile[MAXPATHLEN]; 150 char db_tempname[MAXPATHLEN]; 151 char user[4096], host[4096]; /* XXX: DB bsize = 4096 in ypdb.c */ 152 char myname[MAXHOSTNAMELEN], datestr[11], *slash; 153 154 while ((ch = getopt(argc, argv, "Edensuv")) != -1) 155 switch (ch) { 156 case 'E': 157 eflag++; /* Check hostname */ 158 Eflag++; /* .. even check MX records */ 159 break; 160 case 'd': 161 dflag++; /* Don't check DNS hostname */ 162 break; 163 case 'e': 164 eflag++; /* Check hostname */ 165 break; 166 case 'n': 167 nflag++; /* Capitalize name parts */ 168 break; 169 case 's': 170 sflag++; /* Don't know... */ 171 break; 172 case 'u': 173 uflag++; /* Don't check UUCP hostname */ 174 break; 175 case 'v': 176 vflag++; /* Verbose */ 177 break; 178 default: 179 usage(); 180 break; 181 } 182 183 if (optind == argc) 184 usage(); 185 186 input = argv[optind++]; 187 if (optind < argc) 188 output = argv[optind++]; 189 if (optind < argc) 190 usage(); 191 192 db = ypdb_open(input, O_RDONLY, 0444); 193 if (db == NULL) { 194 err(1, "Unable to open input database %s", input); 195 /* NOTREACHED */ 196 } 197 198 if (output != NULL) { 199 if (strlen(output) + strlen(YPDB_SUFFIX) > MAXPATHLEN) { 200 errx(1, "%s: file name too long", output); 201 /* NOTREACHED */ 202 } 203 204 snprintf(db_outfile, sizeof(db_outfile), 205 "%s%s", output, YPDB_SUFFIX); 206 207 slash = strrchr(output, '/'); 208 if (slash != NULL) 209 slash[1] = 0; /* truncate to dir */ 210 else 211 *output = 0; /* eliminate */ 212 213 /* note: output is now directory where map goes ! */ 214 215 if (strlen(output) + strlen(mapname) + 216 strlen(YPDB_SUFFIX) > MAXPATHLEN) { 217 errx(1, "%s: directory name too long", output); 218 /* NOTREACHED */ 219 } 220 221 snprintf(db_tempname, sizeof(db_tempname), "%s%s%s", output, 222 mapname, YPDB_SUFFIX); 223 fd = mkstemps(db_tempname, 3); 224 if (fd == -1) 225 goto fail; 226 close(fd); 227 228 strncpy(db_mapname, db_tempname, strlen(db_tempname) - 3); 229 db_mapname[sizeof(db_mapname) - 1] = '\0'; 230 231 new_db = ypdb_open(db_mapname, O_RDWR|O_TRUNC, 0444); 232 if (new_db == NULL) { 233 fail: 234 if (fd != -1) 235 unlink(db_tempname); 236 err(1, "Unable to open output database %s", 237 db_outfile); 238 /* NOTREACHED */ 239 } 240 } 241 242 for (key = ypdb_firstkey(db); key.dptr != NULL; 243 key = ypdb_nextkey(db)) { 244 val = ypdb_fetch(db, key); 245 246 if (val.dptr == NULL) 247 continue; /* No value */ 248 if (*key.dptr == '@' && key.dsize == 1) 249 continue; /* Sendmail token */ 250 if (strncmp(key.dptr, "YP_", 3)==0) /* YP token */ 251 continue; 252 if (memchr(val.dptr, ',', val.dsize)) 253 continue; /* List... */ 254 if (memchr(val.dptr, '|', val.dsize)) 255 continue; /* Pipe... */ 256 257 if (!(memchr(val.dptr, '@', val.dsize) || 258 memchr(val.dptr, '!', val.dsize))) 259 continue; /* Skip local users */ 260 261 split_address(val.dptr, val.dsize, user, host); 262 263 if (eflag && check_host(val.dptr, val.dsize, host, dflag, uflag, Eflag)) { 264 warnx("Invalid host %s in %*.*s:%*.*s", 265 host, key.dsize, key.dsize, key.dptr, 266 val.dsize, val.dsize, val.dptr); 267 continue; 268 } 269 270 if (nflag) 271 capitalize(key.dptr, key.dsize); 272 273 if (new_db != NULL) { 274 status = ypdb_store(new_db, val, key, YPDB_INSERT); 275 if (status != 0) { 276 warnx("problem storing %*.*s %*.*s", 277 val.dsize, val.dsize, val.dptr, 278 key.dsize, key.dsize, key.dptr); 279 } 280 } 281 282 if (vflag) { 283 printf("%*.*s --> %*.*s\n", 284 val.dsize, val.dsize, val.dptr, 285 key.dsize, key.dsize, key.dptr); 286 } 287 288 } 289 290 if (new_db != NULL) { 291 snprintf(datestr, sizeof datestr, "%010lld", 292 (long long)time(NULL)); 293 key.dptr = YP_LAST_KEY; 294 key.dsize = strlen(YP_LAST_KEY); 295 val.dptr = datestr; 296 val.dsize = strlen(datestr); 297 status = ypdb_store(new_db, key, val, YPDB_INSERT); 298 if (status != 0) { 299 warnx("problem storing %*.*s %*.*s", 300 key.dsize, key.dsize, key.dptr, 301 val.dsize, val.dsize, val.dptr); 302 } 303 } 304 305 if (new_db != NULL) { 306 gethostname(myname, sizeof(myname)); 307 key.dptr = YP_MASTER_KEY; 308 key.dsize = strlen(YP_MASTER_KEY); 309 val.dptr = myname; 310 val.dsize = strlen(myname); 311 status = ypdb_store(new_db, key, val, YPDB_INSERT); 312 if (status != 0) { 313 warnx("problem storing %*.*s %*.*s", 314 key.dsize, key.dsize, key.dptr, 315 val.dsize, val.dsize, val.dptr); 316 } 317 } 318 319 ypdb_close(db); 320 321 if (new_db != NULL) { 322 ypdb_close(new_db); 323 if (rename(db_tempname, db_outfile) < 0) { 324 err(1, "rename %s -> %s failed", db_tempname, 325 db_outfile); 326 /* NOTREACHED */ 327 } 328 } 329 return(0); 330 } 331