mkalias.c revision 1.8
1 /* $NetBSD: mkalias.c,v 1.8 1999/06/07 03:06:09 mrg Exp $ */ 2 3 /* 4 * Copyright (c) 1997 Mats O Jansson <moj (at) 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 * 3. All advertising materials mentioning features or use of this software 16 * must display the following acknowledgement: 17 * This product includes software developed by Mats O Jansson 18 * 4. The name of the author may not be used to endorse or promote products 19 * derived from this software without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS 22 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 23 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY 25 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #include <sys/cdefs.h> 35 #ifndef lint 36 __RCSID("$NetBSD: mkalias.c,v 1.8 1999/06/07 03:06:09 mrg Exp $"); 37 #endif 38 39 #include <sys/types.h> 40 #include <sys/param.h> 41 #include <netinet/in.h> 42 #include <arpa/nameser.h> 43 44 #include <ctype.h> 45 #include <err.h> 46 #include <fcntl.h> 47 #include <netdb.h> 48 #include <resolv.h> 49 #include <stdio.h> 50 #include <stdlib.h> 51 #include <string.h> 52 #include <time.h> 53 #include <unistd.h> 54 55 #include <rpc/rpc.h> 56 57 #include "protos.h" 58 #include "ypdb.h" 59 #include "ypdef.h" 60 61 void capitalize __P((char *, int)); 62 int check_host __P((char *, char *, int, int, int)); 63 int main __P((int, char *[])); 64 void split_address __P((char *, int, char *, char *)); 65 void usage __P((void)); 66 67 extern char *__progname; /* from crt0.o */ 68 69 void 70 split_address(address, len, user, host) 71 char *address; 72 int len; 73 char *user, *host; 74 { 75 char *c, *s, *r; 76 int i = 0; 77 78 if ((strchr(address, '@')) != NULL) { 79 80 s = user; 81 82 for(c = address; i < len; i++) { 83 if (*c == '@') { 84 *s = '\0'; 85 s = host; 86 } else 87 *s++ = *c; 88 c++; 89 } 90 *s = '\0'; 91 } 92 93 if ((r = strrchr(address, '!')) != NULL) { 94 95 s = host; 96 97 for(c = address; i < len; i++) { 98 if (c == r) { 99 *s = '\0'; 100 s = user; 101 } else 102 *s++ = *c; 103 c++; 104 } 105 *s = '\0'; 106 } 107 } 108 109 int 110 check_host(address, host, dflag, uflag, Eflag) 111 char *address, *host; 112 int dflag, uflag, Eflag; 113 { 114 char answer[PACKETSZ]; 115 int status; 116 117 if ((dflag && strchr(address, '@')) || 118 (uflag && strchr(address, '!'))) 119 return(0); 120 121 if ((_res.options & RES_INIT) == 0) 122 res_init(); 123 124 status = res_search(host, C_IN, T_AAAA, answer, sizeof(answer)); 125 126 if (status == -1) 127 status = res_search(host, C_IN, T_A, answer, sizeof(answer)); 128 129 if ((status == -1) && Eflag) 130 status = res_search(host, C_IN, T_MX, answer, sizeof(answer)); 131 132 return(status == -1); 133 } 134 135 void 136 capitalize(name, len) 137 char *name; 138 int len; 139 { 140 char last = ' '; 141 char *c; 142 int i = 0; 143 144 for(c = name; i < len; i++) { 145 if (*c == '.') last = '.'; 146 c++; 147 } 148 149 i = 0; 150 if (last == '.') { 151 for(c = name; i < len; i++) { 152 if (last == '.') 153 *c = toupper(*c); 154 last = *c++; 155 } 156 } 157 } 158 159 int 160 main(argc, argv) 161 int argc; 162 char *argv[]; 163 { 164 int eflag = 0; 165 int dflag = 0; 166 int nflag = 0; 167 int sflag = 0; 168 int uflag = 0; 169 int vflag = 0; 170 int Eflag = 0; 171 int ch; 172 char *input = NULL; 173 char *output = NULL; 174 DBM *db; 175 datum key, val; 176 char *slash; 177 DBM *new_db = NULL; 178 static char mapname[] = "ypdbXXXXXXXXXX"; 179 char db_mapname[MAXPATHLEN], db_outfile[MAXPATHLEN], 180 db_tempname[MAXPATHLEN]; 181 int status; 182 char user[4096], host[4096]; /* XXX: DB bsize = 4096 in ypdb.c */ 183 char datestr[11]; 184 char myname[MAXHOSTNAMELEN]; 185 186 while ((ch = getopt(argc, argv, "Edensuv")) != -1) { 187 switch(ch) { 188 case 'd': 189 dflag++; /* Don't check DNS hostname */ 190 break; 191 192 case 'e': 193 eflag++; /* Check hostname */ 194 break; 195 196 case 'E': 197 eflag++; /* Check hostname */ 198 Eflag++; /* .. even check MX records */ 199 break; 200 201 case 'n': 202 nflag++; /* Capitalize name parts */ 203 break; 204 205 case 's': 206 sflag++; /* Don't know... */ 207 break; 208 209 case 'u': 210 uflag++; /* Don't check UUCP hostname */ 211 break; 212 213 case 'v': 214 vflag++; /* Verbose */ 215 break; 216 217 default: 218 usage(); 219 } 220 } 221 222 if (optind == argc) 223 usage(); 224 225 input = argv[optind++]; 226 if (optind < argc) 227 output = argv[optind++]; 228 if (optind < argc) 229 usage(); 230 231 db = ypdb_open(input, O_RDONLY, 0444); 232 if (db == NULL) 233 err(1, "Unable to open input database `%s'", input); 234 235 if (output != NULL) { 236 if (strlen(output) + strlen(YPDB_SUFFIX) > MAXPATHLEN) 237 warnx("file name `%s' too long", output); 238 snprintf(db_outfile, sizeof(db_outfile), 239 "%s%s", output, YPDB_SUFFIX); 240 241 slash = strrchr(output, '/'); 242 if (slash != NULL) 243 slash[1] = 0; /* truncate to dir */ 244 else 245 *output = 0; /* elminate */ 246 247 /* note: output is now directory where map goes ! */ 248 249 if (strlen(output) + strlen(mapname) 250 + strlen(YPDB_SUFFIX) > MAXPATHLEN) 251 errx(1, "Directory name `%s' too long", output); 252 253 snprintf(db_tempname, sizeof(db_tempname), "%s%s", output, 254 mapname); 255 mktemp(db_tempname); /* OK */ 256 snprintf(db_mapname, sizeof(db_mapname), "%s%s", db_tempname, 257 YPDB_SUFFIX); 258 259 new_db = ypdb_open(db_tempname, O_RDWR|O_CREAT, 0444); 260 if (new_db == NULL) 261 err(1, "Unable to open output database `%s'", 262 db_outfile); 263 } 264 265 for (key = ypdb_firstkey(db); 266 key.dptr != NULL; 267 key = ypdb_nextkey(db)) { 268 269 val = ypdb_fetch(db, key); 270 271 if (val.dptr == NULL) 272 continue; /* No value */ 273 if ((*key.dptr == '@') && (key.dsize == 1)) 274 continue; /* Sendmail token */ 275 if (strncmp(key.dptr, "YP_", 3)==0) /* YP token */ 276 continue; 277 if (strchr(val.dptr, ',')) /* List... */ 278 continue; 279 if (strchr(val.dptr, '|')) /* Pipe... */ 280 continue; 281 282 if (!((strchr(val.dptr, '@')) || 283 (strchr(val.dptr, '!')))) 284 continue; /* Skip local users */ 285 286 split_address(val.dptr, val.dsize, user, host); 287 288 if (eflag && check_host(val.dptr, host, dflag, uflag, Eflag)) { 289 printf("Invalid host %s in %*.*s:%*.*s\n", host, 290 key.dsize, key.dsize, key.dptr, 291 val.dsize, val.dsize, val.dptr); 292 continue; 293 } 294 295 if (nflag) 296 capitalize(key.dptr, key.dsize); 297 298 if (new_db != NULL) { 299 status = ypdb_store(new_db, val, key, YPDB_INSERT); 300 if (status != 0) { 301 printf("%s: problem storing %*.*s %*.*s\n", 302 __progname, 303 val.dsize, val.dsize, val.dptr, 304 key.dsize, key.dsize, key.dptr); 305 } 306 } 307 308 if (vflag) { 309 printf("%*.*s --> %*.*s\n", 310 val.dsize, val.dsize, val.dptr, 311 key.dsize, key.dsize, key.dptr); 312 } 313 314 } 315 316 if (new_db != NULL) { 317 sprintf(datestr, "%010d", (int)time(NULL)); 318 key.dptr = YP_LAST_KEY; 319 key.dsize = strlen(YP_LAST_KEY); 320 val.dptr = datestr; 321 val.dsize = strlen(datestr); 322 status = ypdb_store(new_db, key, val, YPDB_INSERT); 323 if (status != 0) 324 warnx("problem storing %*.*s %*.*s", 325 key.dsize, key.dsize, key.dptr, 326 val.dsize, val.dsize, val.dptr); 327 } 328 329 if (new_db != NULL) { 330 localhostname(myname, sizeof(myname) - 1); 331 key.dptr = YP_MASTER_KEY; 332 key.dsize = strlen(YP_MASTER_KEY); 333 val.dptr = myname; 334 val.dsize = strlen(myname); 335 status = ypdb_store(new_db, key, val, YPDB_INSERT); 336 if (status != 0) 337 warnx("problem storing %*.*s %*.*s", 338 key.dsize, key.dsize, key.dptr, 339 val.dsize, val.dsize, val.dptr); 340 } 341 342 ypdb_close(db); 343 344 if (new_db != NULL) { 345 ypdb_close(new_db); 346 if (rename(db_mapname, db_outfile) < 0) 347 err(1, "rename `%s' to `%s' failed", db_mapname, 348 db_outfile); 349 } 350 351 exit(0); 352 } 353 354 void 355 usage() 356 { 357 fprintf(stderr, 358 "usage: %s [-e|-E [-d] [-u]] [-n] [-v] input [output]\n", 359 __progname); 360 exit(1); 361 } 362
Indexes created Thu Sep 25 16:09:42 GMT 2025