/* $OpenBSD: mkalias.c,v 1.6 2001/02/05 14:47:17 deraadt Exp $ */ /* * Copyright (c) 1997 Mats O Jansson * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * 3. All advertising materials mentioning features or use of this software * must display the following acknowledgement: * This product includes software developed by Mats O Jansson * 4. The name of the author may not be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #ifndef LINT static char rcsid[] = "$OpenBSD: mkalias.c,v 1.6 2001/02/05 14:47:17 deraadt Exp $"; #endif #include #include #include #include #include #include #include #include #include #include #include #include "ypdb.h" #include "ypdef.h" extern char *__progname; /* from crt0.o */ void split_address(address, len, user, host) char *address; int len; char *user, *host; { char *c,*s,*r; int i = 0; if (index(address,'@')) { s = user; for(c = address; i < len; i++) { if (*c == '@') { *s = '\0'; s = host; } else { *s++ = *c; } c++; } *s = '\0'; } if (r = rindex(address,'!')) { s = host; for(c = address; i < len; i++) { if (c == r) { *s = '\0'; s = user; } else { *s++ = *c; } c++; } *s = '\0'; } } int check_host(address, host, dflag, uflag, Eflag) char *address, *host; int dflag, uflag, Eflag; { char answer[PACKETSZ]; int status; if ((dflag && index(address,'@')) || (uflag && index(address,'!'))) return(0); if ((_res.options & RES_INIT) == 0) res_init(); status = res_search(host, C_IN, T_AAAA, answer, sizeof(answer)); if (status == -1) status = res_search(host, C_IN, T_A, answer, sizeof(answer)); if ((status == -1) && Eflag) status = res_search(host, C_IN, T_MX, answer, sizeof(answer)); return(status == -1); } void capitalize(name,len) char *name; int len; { char last = ' '; char *c; int i = 0; for(c = name; i < len; i++) { if (*c == '.') last = '.'; c++; } i = 0; if (last == '.') { for(c = name; i < len; i++) { if (last == '.') { *c = toupper(*c); } last = *c++; } } } int main (argc,argv) int argc; char *argv[]; { int usage = 0; int eflag = 0; int dflag = 0; int nflag = 0; int sflag = 0; int uflag = 0; int vflag = 0; int Eflag = 0; int ch, fd; char *input = NULL; char *output = NULL; DBM *db; datum key,val; char *slash; DBM *new_db = NULL; static char mapname[] = "ypdbXXXXXXXXXX"; char db_mapname[MAXPATHLEN],db_outfile[MAXPATHLEN], db_tempname[MAXPATHLEN]; int status; char user[4096],host[4096]; /* XXX: DB bsize = 4096 in ypdb.c */ char datestr[11]; char myname[MAXHOSTNAMELEN]; while ((ch = getopt(argc, argv, "Edensuv")) != -1) switch(ch) { case 'E': eflag++; /* Check hostname */ Eflag++; /* .. even check MX records */ break; case 'd': dflag++; /* Don't check DNS hostname */ break; case 'e': eflag++; /* Check hostname */ break; case 'n': nflag++; /* Capitalize name parts */ break; case 's': sflag++; /* Don't know... */ break; case 'u': uflag++; /* Don't check UUCP hostname */ break; case 'v': vflag++; /* Verbose */ break; default: usage++; break; } if (optind == argc) { usage++; } else { input = argv[optind++]; if (optind < argc) output = argv[optind++]; if (optind < argc) usage++; } if (usage) { fprintf(stderr, "usage: %s [-v] [-e|-E [-d] [-u]] [-n] input [output]\n", __progname); exit(1); } db = ypdb_open(input, O_RDONLY, 0444); if (db == NULL) { fprintf(stderr, "%s: Unable to open input database %s\n", __progname, input); exit(1); } if (output != NULL) { if (strlen(output) + strlen(YPDB_SUFFIX) > MAXPATHLEN) { fprintf(stderr,"%s: %s: file name too long\n", __progname, output); } snprintf(db_outfile, sizeof(db_outfile), "%s%s", output, YPDB_SUFFIX); slash = strrchr(output, '/'); if (slash != NULL) slash[1] = 0; /* truncate to dir */ else *output = 0; /* elminate */ /* note: output is now directory where map goes ! */ if (strlen(output) + strlen(mapname) + strlen(YPDB_SUFFIX) > MAXPATHLEN) { fprintf(stderr,"%s: %s: directory name too long\n", __progname, output); exit(1); } snprintf(db_tempname, sizeof(db_tempname), "%s%s", output, mapname); fd = mkstemp(db_tempname); if (fd == -1) goto fail; close(fd); snprintf(db_mapname, sizeof(db_mapname), "%s%s", db_tempname, YPDB_SUFFIX); new_db = ypdb_open(db_tempname, O_RDWR|O_CREAT, 0444); if (new_db == NULL) { fail: if (fd != -1) unlink(db_tempname); fprintf(stderr, "%s: Unable to open output database %s\n", __progname, db_outfile); exit(1); } } for (key = ypdb_firstkey(db); key.dptr != NULL; key = ypdb_nextkey(db)) { val = ypdb_fetch(db,key); if (val.dptr == NULL) continue; /* No value */ if ((*key.dptr == '@') && (key.dsize == 1)) continue; /* Sendmail token */ if (strncmp(key.dptr, "YP_", 3)==0) /* YP token */ continue; if (index(val.dptr,',')) continue; /* List... */ if (index(val.dptr,'|')) continue; /* Pipe... */ if (!((index(val.dptr,'@')) || (index(val.dptr,'!')))) continue; /* Skip local users */ split_address(val.dptr,val.dsize,user,host); if (eflag && check_host(val.dptr, host, dflag, uflag, Eflag)) { printf("Invalid host %s in %*.*s:%*.*s\n", host, key.dsize, key.dsize, key.dptr, val.dsize, val.dsize, val.dptr); continue; } if (nflag) { capitalize(key.dptr,key.dsize); } if (new_db != NULL) { status = ypdb_store(new_db, val, key, YPDB_INSERT); if (status != 0) { printf("%s: problem storing %*.*s %*.*s\n", __progname, val.dsize, val.dsize, val.dptr, key.dsize, key.dsize, key.dptr); } } if (vflag) { printf("%*.*s --> %*.*s\n", val.dsize, val.dsize, val.dptr, key.dsize, key.dsize, key.dptr); } } if (new_db != NULL) { snprintf(datestr, sizeof datestr, "%010u", time(NULL)); key.dptr = YP_LAST_KEY; key.dsize = strlen(YP_LAST_KEY); val.dptr = datestr; val.dsize = strlen(datestr); status = ypdb_store(new_db, key, val, YPDB_INSERT); if (status != 0) { printf("%s: problem storing %*.*s %*.*s\n", __progname, key.dsize, key.dsize, key.dptr, val.dsize, val.dsize, val.dptr); } } if (new_db != NULL) { gethostname(myname, sizeof(myname)); key.dptr = YP_MASTER_KEY; key.dsize = strlen(YP_MASTER_KEY); val.dptr = myname; val.dsize = strlen(myname); status = ypdb_store(new_db, key, val, YPDB_INSERT); if (status != 0) { printf("%s: problem storing %*.*s %*.*s\n", __progname, key.dsize, key.dsize, key.dptr, val.dsize, val.dsize, val.dptr); } } ypdb_close(db); if (new_db != NULL) { ypdb_close(new_db); if (rename(db_mapname,db_outfile) < 0) { perror("rename"); fprintf(stderr,"rename %s -> %s failed!\n", db_mapname, db_outfile); exit(1); } } return(0); }