156822Seric /* 256822Seric * Copyright (c) 1992 Eric P. Allman. 362526Sbostic * Copyright (c) 1992, 1993 462526Sbostic * The Regents of the University of California. All rights reserved. 556822Seric * 656822Seric * %sccs.include.redist.c% 756822Seric */ 856822Seric 956822Seric #ifndef lint 10*64650Seric static char sccsid[] = "@(#)map.c 8.14 (Berkeley) 09/25/93"; 1156822Seric #endif /* not lint */ 1256822Seric 1356822Seric #include "sendmail.h" 1456822Seric 1560089Seric #ifdef NDBM 1656822Seric #include <ndbm.h> 1756822Seric #endif 1860089Seric #ifdef NEWDB 1956822Seric #include <db.h> 2056822Seric #endif 2160089Seric #ifdef NIS 2257208Seric #include <rpcsvc/ypclnt.h> 2357208Seric #endif 2456822Seric 2556822Seric /* 2660089Seric ** MAP.C -- implementations for various map classes. 2756822Seric ** 2860089Seric ** Each map class implements a series of functions: 2960089Seric ** 3060089Seric ** bool map_parse(MAP *map, char *args) 3160089Seric ** Parse the arguments from the config file. Return TRUE 3260089Seric ** if they were ok, FALSE otherwise. Fill in map with the 3360089Seric ** values. 3460089Seric ** 3560222Seric ** char *map_lookup(MAP *map, char *key, char **args, int *pstat) 3660222Seric ** Look up the key in the given map. If found, do any 3760222Seric ** rewriting the map wants (including "args" if desired) 3860089Seric ** and return the value. Set *pstat to the appropriate status 3960222Seric ** on error and return NULL. Args will be NULL if called 4060222Seric ** from the alias routines, although this should probably 4160222Seric ** not be relied upon. It is suggested you call map_rewrite 4260222Seric ** to return the results -- it takes care of null termination 4360222Seric ** and uses a dynamically expanded buffer as needed. 4460089Seric ** 4560089Seric ** void map_store(MAP *map, char *key, char *value) 4660089Seric ** Store the key:value pair in the map. 4760089Seric ** 4860089Seric ** bool map_open(MAP *map, int mode) 4960222Seric ** Open the map for the indicated mode. Mode should 5060222Seric ** be either O_RDONLY or O_RDWR. Return TRUE if it 5160222Seric ** was opened successfully, FALSE otherwise. If the open 5260222Seric ** failed an the MF_OPTIONAL flag is not set, it should 5360222Seric ** also print an error. If the MF_ALIAS bit is set 5460222Seric ** and this map class understands the @:@ convention, it 5560222Seric ** should call aliaswait() before returning. 5660089Seric ** 5760089Seric ** void map_close(MAP *map) 5860089Seric ** Close the map. 5960089Seric */ 6060089Seric 6160089Seric #define DBMMODE 0644 6260089Seric /* 6360089Seric ** MAP_PARSEARGS -- parse config line arguments for database lookup 6460089Seric ** 6560089Seric ** This is a generic version of the map_parse method. 6660089Seric ** 6756822Seric ** Parameters: 6860089Seric ** map -- the map being initialized. 6960089Seric ** ap -- a pointer to the args on the config line. 7056822Seric ** 7156822Seric ** Returns: 7260089Seric ** TRUE -- if everything parsed OK. 7356822Seric ** FALSE -- otherwise. 7456822Seric ** 7556822Seric ** Side Effects: 7660089Seric ** null terminates the filename; stores it in map 7756822Seric */ 7856822Seric 7956822Seric bool 8060089Seric map_parseargs(map, ap) 8156822Seric MAP *map; 8260089Seric char *ap; 8356822Seric { 8460089Seric register char *p = ap; 8556822Seric 8663753Seric map->map_mflags |= MF_TRY0NULL | MF_TRY1NULL; 8760089Seric for (;;) 8860089Seric { 8960089Seric while (isascii(*p) && isspace(*p)) 9060089Seric p++; 9160089Seric if (*p != '-') 9260089Seric break; 9360089Seric switch (*++p) 9460089Seric { 9560089Seric case 'N': 9660207Seric map->map_mflags |= MF_INCLNULL; 9763753Seric map->map_mflags &= ~MF_TRY0NULL; 9860089Seric break; 9960089Seric 10063753Seric case 'O': 10163753Seric map->map_mflags &= ~MF_TRY1NULL; 10263753Seric break; 10363753Seric 10460089Seric case 'o': 10560207Seric map->map_mflags |= MF_OPTIONAL; 10660089Seric break; 10760089Seric 10860089Seric case 'f': 10960207Seric map->map_mflags |= MF_NOFOLDCASE; 11060089Seric break; 11160089Seric 11260089Seric case 'm': 11360207Seric map->map_mflags |= MF_MATCHONLY; 11460089Seric break; 11560089Seric 11660089Seric case 'a': 11760089Seric map->map_app = ++p; 11860089Seric break; 11960089Seric } 12060089Seric while (*p != '\0' && !(isascii(*p) && isspace(*p))) 12160089Seric p++; 12260089Seric if (*p != '\0') 12360089Seric *p++ = '\0'; 12460089Seric } 12560089Seric if (map->map_app != NULL) 12660089Seric map->map_app = newstr(map->map_app); 12760089Seric 12860089Seric if (*p != '\0') 12960089Seric { 13060089Seric map->map_file = p; 13160089Seric while (*p != '\0' && !(isascii(*p) && isspace(*p))) 13260089Seric p++; 13360089Seric if (*p != '\0') 13460089Seric *p++ = '\0'; 13560089Seric map->map_file = newstr(map->map_file); 13660089Seric } 13760089Seric 13860089Seric while (*p != '\0' && isascii(*p) && isspace(*p)) 13960089Seric p++; 14060089Seric if (*p != '\0') 14160089Seric map->map_rebuild = newstr(p); 14260089Seric 14356822Seric if (map->map_file == NULL) 14457208Seric { 14560089Seric syserr("No file name for %s map %s", 14660089Seric map->map_class->map_cname, map->map_mname); 14756822Seric return FALSE; 14857208Seric } 14960089Seric return TRUE; 15060089Seric } 15160089Seric /* 15260089Seric ** MAP_REWRITE -- rewrite a database key, interpolating %n indications. 15360089Seric ** 15460089Seric ** It also adds the map_app string. It can be used as a utility 15560089Seric ** in the map_lookup method. 15660089Seric ** 15760089Seric ** Parameters: 15860089Seric ** map -- the map that causes this. 15960089Seric ** s -- the string to rewrite, NOT necessarily null terminated. 16060089Seric ** slen -- the length of s. 16160089Seric ** av -- arguments to interpolate into buf. 16260089Seric ** 16360089Seric ** Returns: 16460089Seric ** Pointer to rewritten result. 16560089Seric ** 16660089Seric ** Side Effects: 16760089Seric ** none. 16860089Seric */ 16960089Seric 17060492Seric struct rwbuf 17160492Seric { 17260492Seric int rwb_len; /* size of buffer */ 17360492Seric char *rwb_buf; /* ptr to buffer */ 17460492Seric }; 17560492Seric 17660492Seric struct rwbuf RwBufs[2]; /* buffers for rewriting output */ 17760492Seric 17860089Seric char * 17960089Seric map_rewrite(map, s, slen, av) 18060089Seric register MAP *map; 18160089Seric register char *s; 18260089Seric int slen; 18360089Seric char **av; 18460089Seric { 18560089Seric register char *bp; 18660089Seric register char c; 18760089Seric char **avp; 18860089Seric register char *ap; 18960492Seric register struct rwbuf *rwb; 19060089Seric int i; 19160089Seric int len; 19260089Seric 19360537Seric if (tTd(39, 1)) 19460089Seric { 19560256Seric printf("map_rewrite(%.*s), av =", slen, s); 19660256Seric if (av == NULL) 19760256Seric printf(" (nullv)"); 19860256Seric else 19960256Seric { 20060256Seric for (avp = av; *avp != NULL; avp++) 20160256Seric printf("\n\t%s", *avp); 20260256Seric } 20360256Seric printf("\n"); 20460089Seric } 20560089Seric 20660492Seric rwb = RwBufs; 20760492Seric if (av == NULL) 20860492Seric rwb++; 20960492Seric 21060089Seric /* count expected size of output (can safely overestimate) */ 21160089Seric i = len = slen; 21260089Seric if (av != NULL) 21360089Seric { 21460089Seric bp = s; 21560089Seric for (i = slen; --i >= 0 && (c = *bp++) != 0; ) 21660089Seric { 21760089Seric if (c != '%') 21860089Seric continue; 21960089Seric if (--i < 0) 22060089Seric break; 22160089Seric c = *bp++; 22260089Seric if (!(isascii(c) && isdigit(c))) 22360089Seric continue; 22463937Seric for (avp = av; --c >= '0' && *avp != NULL; avp++) 22560089Seric continue; 22660089Seric if (*avp == NULL) 22760089Seric continue; 22860089Seric len += strlen(*avp); 22960089Seric } 23060089Seric } 23160089Seric if (map->map_app != NULL) 23260089Seric len += strlen(map->map_app); 23360492Seric if (rwb->rwb_len < ++len) 23460089Seric { 23560089Seric /* need to malloc additional space */ 23660492Seric rwb->rwb_len = len; 23760492Seric if (rwb->rwb_buf != NULL) 23860492Seric free(rwb->rwb_buf); 23960492Seric rwb->rwb_buf = xalloc(rwb->rwb_len); 24060089Seric } 24160089Seric 24260492Seric bp = rwb->rwb_buf; 24360089Seric if (av == NULL) 24460089Seric { 24560089Seric bcopy(s, bp, slen); 24660089Seric bp += slen; 24760089Seric } 24860089Seric else 24960089Seric { 25060089Seric while (--slen >= 0 && (c = *s++) != '\0') 25160089Seric { 25260089Seric if (c != '%') 25360089Seric { 25460089Seric pushc: 25560089Seric *bp++ = c; 25660089Seric continue; 25760089Seric } 25860089Seric if (--slen < 0 || (c = *s++) == '\0') 25960089Seric c = '%'; 26060089Seric if (c == '%') 26160089Seric goto pushc; 26260089Seric if (!(isascii(c) && isdigit(c))) 26360089Seric { 26460089Seric *bp++ = '%'; 26560089Seric goto pushc; 26660089Seric } 26763937Seric for (avp = av; --c >= '0' && *avp != NULL; avp++) 26860089Seric continue; 26960089Seric if (*avp == NULL) 27060089Seric continue; 27160089Seric 27260089Seric /* transliterate argument into output string */ 27360089Seric for (ap = *avp; (c = *ap++) != '\0'; ) 27460089Seric *bp++ = c; 27560089Seric } 27660089Seric } 27760089Seric if (map->map_app != NULL) 27860089Seric strcpy(bp, map->map_app); 27960089Seric else 28060089Seric *bp = '\0'; 28160537Seric if (tTd(39, 1)) 28260492Seric printf("map_rewrite => %s\n", rwb->rwb_buf); 28360492Seric return rwb->rwb_buf; 28460089Seric } 28560089Seric /* 28660537Seric ** INITMAPS -- initialize for aliasing 28760537Seric ** 28860537Seric ** Parameters: 28960537Seric ** rebuild -- if TRUE, this rebuilds the cached versions. 29060537Seric ** e -- current envelope. 29160537Seric ** 29260537Seric ** Returns: 29360537Seric ** none. 29460537Seric ** 29560537Seric ** Side Effects: 29660537Seric ** initializes aliases: 29760537Seric ** if NDBM: opens the database. 29860537Seric ** if ~NDBM: reads the aliases into the symbol table. 29960537Seric */ 30060537Seric 30160537Seric initmaps(rebuild, e) 30260537Seric bool rebuild; 30360537Seric register ENVELOPE *e; 30460537Seric { 30560537Seric extern void map_init(); 30660537Seric 30760537Seric CurEnv = e; 30860537Seric stabapply(map_init, rebuild); 30960537Seric } 31060537Seric 31160537Seric void 31260537Seric map_init(s, rebuild) 31360537Seric register STAB *s; 31460537Seric int rebuild; 31560537Seric { 31660537Seric register MAP *map; 31760537Seric 31860537Seric /* has to be a map */ 31960537Seric if (s->s_type != ST_MAP) 32060537Seric return; 32160537Seric 32260537Seric map = &s->s_map; 32360537Seric if (!bitset(MF_VALID, map->map_mflags)) 32460537Seric return; 32560537Seric 32660537Seric if (tTd(38, 2)) 32760537Seric printf("map_init(%s:%s)\n", 32860537Seric map->map_class->map_cname, map->map_file); 32960537Seric 33060537Seric /* if already open, close it (for nested open) */ 33160537Seric if (bitset(MF_OPEN, map->map_mflags)) 33260537Seric { 33360537Seric map->map_class->map_close(map); 33460537Seric map->map_mflags &= ~(MF_OPEN|MF_WRITABLE); 33560537Seric } 33660537Seric 33760537Seric if (rebuild) 33860537Seric { 33960926Seric if (bitset(MF_ALIAS, map->map_mflags) && 34060926Seric bitset(MCF_REBUILDABLE, map->map_class->map_cflags)) 34160537Seric rebuildaliases(map, FALSE); 34260537Seric } 34360537Seric else 34460537Seric { 34560537Seric if (map->map_class->map_open(map, O_RDONLY)) 34660537Seric { 34760537Seric if (tTd(38, 4)) 34860537Seric printf("%s:%s: valid\n", 34960537Seric map->map_class->map_cname, 35060537Seric map->map_file); 35160537Seric map->map_mflags |= MF_OPEN; 35260537Seric } 35360537Seric else if (tTd(38, 4)) 35460537Seric printf("%s:%s: invalid: %s\n", 35560537Seric map->map_class->map_cname, 35660537Seric map->map_file, 35760537Seric errstring(errno)); 35860537Seric } 35960537Seric } 36060537Seric /* 36160089Seric ** NDBM modules 36260089Seric */ 36360089Seric 36460089Seric #ifdef NDBM 36560089Seric 36660089Seric /* 36760089Seric ** DBM_MAP_OPEN -- DBM-style map open 36860089Seric */ 36960089Seric 37060089Seric bool 37160089Seric ndbm_map_open(map, mode) 37260089Seric MAP *map; 37360089Seric int mode; 37460089Seric { 37564284Seric register DBM *dbm; 37664284Seric struct stat st; 37760089Seric 37860537Seric if (tTd(38, 2)) 37960089Seric printf("ndbm_map_open(%s, %d)\n", map->map_file, mode); 38060089Seric 38160207Seric if (mode == O_RDWR) 38260207Seric mode |= O_CREAT|O_TRUNC; 38360207Seric 38460089Seric /* open the database */ 38560089Seric dbm = dbm_open(map->map_file, mode, DBMMODE); 38656822Seric if (dbm == NULL) 38756822Seric { 38860207Seric if (!bitset(MF_OPTIONAL, map->map_mflags)) 38956836Seric syserr("Cannot open DBM database %s", map->map_file); 39056822Seric return FALSE; 39156822Seric } 39260089Seric map->map_db1 = (void *) dbm; 39360207Seric if (mode == O_RDONLY && bitset(MF_ALIAS, map->map_mflags)) 39460604Seric aliaswait(map, ".pag"); 39564284Seric if (fstat(dbm_dirfno(dbm), &st) >= 0) 39664284Seric map->map_mtime = st.st_mtime; 39756822Seric return TRUE; 39856822Seric } 39960089Seric 40060089Seric 40160089Seric /* 40256822Seric ** DBM_MAP_LOOKUP -- look up a datum in a DBM-type map 40356822Seric */ 40456822Seric 40556822Seric char * 40660089Seric ndbm_map_lookup(map, name, av, statp) 40756822Seric MAP *map; 40860089Seric char *name; 40956822Seric char **av; 41059084Seric int *statp; 41156822Seric { 41256822Seric datum key, val; 41364373Seric int fd; 41460089Seric char keybuf[MAXNAME + 1]; 41556822Seric 41660537Seric if (tTd(38, 20)) 41760089Seric printf("ndbm_map_lookup(%s)\n", name); 41860089Seric 41960089Seric key.dptr = name; 42060089Seric key.dsize = strlen(name); 42160207Seric if (!bitset(MF_NOFOLDCASE, map->map_mflags)) 42257014Seric { 42360089Seric if (key.dsize > sizeof keybuf - 1) 42460089Seric key.dsize = sizeof keybuf - 1; 42560089Seric bcopy(key.dptr, keybuf, key.dsize + 1); 42660089Seric makelower(keybuf); 42760089Seric key.dptr = keybuf; 42857014Seric } 42964373Seric fd = dbm_dirfno((DBM *) map->map_db1); 43064388Seric if (fd >= 0 && !bitset(MF_LOCKED, map->map_mflags)) 43164373Seric (void) lockfile(fd, map->map_file, ".dir", LOCK_SH); 43263753Seric val.dptr = NULL; 43363753Seric if (bitset(MF_TRY0NULL, map->map_mflags)) 43463753Seric { 43563753Seric val = dbm_fetch((DBM *) map->map_db1, key); 43663753Seric if (val.dptr != NULL) 43763753Seric map->map_mflags &= ~MF_TRY1NULL; 43863753Seric } 43963753Seric if (val.dptr == NULL && bitset(MF_TRY1NULL, map->map_mflags)) 44063753Seric { 44156822Seric key.dsize++; 44263753Seric val = dbm_fetch((DBM *) map->map_db1, key); 44363753Seric if (val.dptr != NULL) 44463753Seric map->map_mflags &= ~MF_TRY0NULL; 44563753Seric } 44664388Seric if (fd >= 0 && !bitset(MF_LOCKED, map->map_mflags)) 44764373Seric (void) lockfile(fd, map->map_file, ".dir", LOCK_UN); 44856822Seric if (val.dptr == NULL) 44956822Seric return NULL; 45060207Seric if (bitset(MF_MATCHONLY, map->map_mflags)) 45163753Seric return map_rewrite(map, name, strlen(name), NULL); 45263753Seric else 45363753Seric return map_rewrite(map, val.dptr, val.dsize, av); 45456822Seric } 45556822Seric 45656822Seric 45756822Seric /* 45860089Seric ** DBM_MAP_STORE -- store a datum in the database 45956822Seric */ 46056822Seric 46160089Seric void 46260089Seric ndbm_map_store(map, lhs, rhs) 46360089Seric register MAP *map; 46460089Seric char *lhs; 46560089Seric char *rhs; 46660089Seric { 46760089Seric datum key; 46860089Seric datum data; 46960089Seric int stat; 47060089Seric 47160537Seric if (tTd(38, 12)) 47260089Seric printf("ndbm_map_store(%s, %s)\n", lhs, rhs); 47360089Seric 47460089Seric key.dsize = strlen(lhs); 47560089Seric key.dptr = lhs; 47660089Seric 47760089Seric data.dsize = strlen(rhs); 47860089Seric data.dptr = rhs; 47960089Seric 48060207Seric if (bitset(MF_INCLNULL, map->map_mflags)) 48160089Seric { 48260089Seric key.dsize++; 48360089Seric data.dsize++; 48460089Seric } 48560089Seric 48660089Seric stat = dbm_store((DBM *) map->map_db1, key, data, DBM_INSERT); 48760089Seric if (stat > 0) 48860089Seric { 48960089Seric usrerr("050 Warning: duplicate alias name %s", lhs); 49060089Seric stat = dbm_store((DBM *) map->map_db1, key, data, DBM_REPLACE); 49160089Seric } 49260089Seric if (stat != 0) 49360089Seric syserr("readaliases: dbm put (%s)", lhs); 49460089Seric } 49560089Seric 49660089Seric 49760089Seric /* 49860207Seric ** NDBM_MAP_CLOSE -- close the database 49960089Seric */ 50060089Seric 50160089Seric void 50260089Seric ndbm_map_close(map) 50360089Seric register MAP *map; 50460089Seric { 50560207Seric if (bitset(MF_WRITABLE, map->map_mflags)) 50660089Seric { 50764250Seric #ifdef NIS 50864075Seric bool inclnull; 50960089Seric char buf[200]; 51060089Seric 51164075Seric inclnull = bitset(MF_INCLNULL, map->map_mflags); 51264075Seric map->map_mflags &= ~MF_INCLNULL; 51364075Seric 51460089Seric (void) sprintf(buf, "%010ld", curtime()); 51560089Seric ndbm_map_store(map, "YP_LAST_MODIFIED", buf); 51660089Seric 51760089Seric (void) myhostname(buf, sizeof buf); 51860089Seric ndbm_map_store(map, "YP_MASTER_NAME", buf); 51964075Seric 52064075Seric if (inclnull) 52164075Seric map->map_mflags |= MF_INCLNULL; 52260089Seric #endif 52360089Seric 52460089Seric /* write out the distinguished alias */ 52560089Seric ndbm_map_store(map, "@", "@"); 52660089Seric } 52760089Seric dbm_close((DBM *) map->map_db1); 52860089Seric } 52960089Seric 53060089Seric #endif 53160089Seric /* 53260582Seric ** NEWDB (Hash and BTree) Modules 53360089Seric */ 53460089Seric 53560089Seric #ifdef NEWDB 53660089Seric 53760089Seric /* 53860582Seric ** BT_MAP_OPEN, HASH_MAP_OPEN -- database open primitives. 53960582Seric ** 54060582Seric ** These do rather bizarre locking. If you can lock on open, 54160582Seric ** do that to avoid the condition of opening a database that 54260582Seric ** is being rebuilt. If you don't, we'll try to fake it, but 54360582Seric ** there will be a race condition. If opening for read-only, 54460582Seric ** we immediately release the lock to avoid freezing things up. 54560582Seric ** We really ought to hold the lock, but guarantee that we won't 54660582Seric ** be pokey about it. That's hard to do. 54760089Seric */ 54860089Seric 54956822Seric bool 55060089Seric bt_map_open(map, mode) 55156822Seric MAP *map; 55260089Seric int mode; 55356822Seric { 55456822Seric DB *db; 55560228Seric int i; 55660582Seric int omode; 55764373Seric int fd; 55864284Seric struct stat st; 55960089Seric char buf[MAXNAME]; 56056822Seric 56160537Seric if (tTd(38, 2)) 56260089Seric printf("bt_map_open(%s, %d)\n", map->map_file, mode); 56360089Seric 56460582Seric omode = mode; 56560582Seric if (omode == O_RDWR) 56660582Seric { 56760582Seric omode |= O_CREAT|O_TRUNC; 56864035Seric #if defined(O_EXLOCK) && defined(HASFLOCK) 56960582Seric omode |= O_EXLOCK; 57060582Seric # if !defined(OLD_NEWDB) 57160582Seric } 57260582Seric else 57360582Seric { 57460582Seric omode |= O_SHLOCK; 57560582Seric # endif 57660582Seric #endif 57760582Seric } 57860207Seric 57960228Seric (void) strcpy(buf, map->map_file); 58060228Seric i = strlen(buf); 58160228Seric if (i < 3 || strcmp(&buf[i - 3], ".db") != 0) 58260228Seric (void) strcat(buf, ".db"); 58360582Seric db = dbopen(buf, omode, DBMMODE, DB_BTREE, NULL); 58456822Seric if (db == NULL) 58556822Seric { 58660207Seric if (!bitset(MF_OPTIONAL, map->map_mflags)) 58756836Seric syserr("Cannot open BTREE database %s", map->map_file); 58856822Seric return FALSE; 58956822Seric } 59064035Seric #if !defined(OLD_NEWDB) && defined(HASFLOCK) 59164373Seric fd = db->fd(db); 59260582Seric # if !defined(O_EXLOCK) 59364373Seric if (mode == O_RDWR && fd >= 0) 59464388Seric { 59564388Seric if (lockfile(fd, map->map_file, ".db", LOCK_EX)) 59664388Seric map->map_mflags |= MF_LOCKED; 59764388Seric } 59860582Seric # else 59964373Seric if (mode == O_RDONLY && fd >= 0) 60064373Seric (void) lockfile(fd, map->map_file, ".db", LOCK_UN); 60164388Seric else 60264388Seric map->map_mflags |= MF_LOCKED; 60360582Seric # endif 60460582Seric #endif 60560585Seric 60660585Seric /* try to make sure that at least the database header is on disk */ 60760585Seric if (mode == O_RDWR) 60864373Seric #ifdef OLD_NEWDB 60964373Seric (void) db->sync(db); 61064373Seric #else 61160585Seric (void) db->sync(db, 0); 61260585Seric 61364373Seric if (fd >= 0 && fstat(fd, &st) >= 0) 61464284Seric map->map_mtime = st.st_mtime; 61564284Seric #endif 61664284Seric 61760089Seric map->map_db2 = (void *) db; 61860207Seric if (mode == O_RDONLY && bitset(MF_ALIAS, map->map_mflags)) 61960207Seric aliaswait(map, ".db"); 62056822Seric return TRUE; 62156822Seric } 62256822Seric 62356822Seric 62456822Seric /* 62556822Seric ** HASH_MAP_INIT -- HASH-style map initialization 62656822Seric */ 62756822Seric 62856822Seric bool 62960089Seric hash_map_open(map, mode) 63056822Seric MAP *map; 63160089Seric int mode; 63256822Seric { 63356822Seric DB *db; 63460228Seric int i; 63560582Seric int omode; 63664373Seric int fd; 63764284Seric struct stat st; 63860089Seric char buf[MAXNAME]; 63956822Seric 64060537Seric if (tTd(38, 2)) 64160089Seric printf("hash_map_open(%s, %d)\n", map->map_file, mode); 64260089Seric 64360582Seric omode = mode; 64460582Seric if (omode == O_RDWR) 64560582Seric { 64660582Seric omode |= O_CREAT|O_TRUNC; 64764035Seric #if defined(O_EXLOCK) && defined(HASFLOCK) 64860582Seric omode |= O_EXLOCK; 64960582Seric # if !defined(OLD_NEWDB) 65060582Seric } 65160582Seric else 65260582Seric { 65360582Seric omode |= O_SHLOCK; 65460582Seric # endif 65560582Seric #endif 65660582Seric } 65760207Seric 65860228Seric (void) strcpy(buf, map->map_file); 65960228Seric i = strlen(buf); 66060228Seric if (i < 3 || strcmp(&buf[i - 3], ".db") != 0) 66160228Seric (void) strcat(buf, ".db"); 66260582Seric db = dbopen(buf, omode, DBMMODE, DB_HASH, NULL); 66356822Seric if (db == NULL) 66456822Seric { 66560207Seric if (!bitset(MF_OPTIONAL, map->map_mflags)) 66656836Seric syserr("Cannot open HASH database %s", map->map_file); 66756822Seric return FALSE; 66856822Seric } 66964035Seric #if !defined(OLD_NEWDB) && defined(HASFLOCK) 67064373Seric fd = db->fd(db); 67160582Seric # if !defined(O_EXLOCK) 67264373Seric if (mode == O_RDWR && fd >= 0) 67364388Seric { 67464388Seric if (lockfile(fd, map->map_file, ".db", LOCK_EX)) 67564388Seric map->map_mflags |= MF_LOCKED; 67664388Seric } 67760582Seric # else 67864373Seric if (mode == O_RDONLY && fd >= 0) 67964373Seric (void) lockfile(fd, map->map_file, ".db", LOCK_UN); 68064388Seric else 68164388Seric map->map_mflags |= MF_LOCKED; 68260582Seric # endif 68360582Seric #endif 68460585Seric 68560585Seric /* try to make sure that at least the database header is on disk */ 68660585Seric if (mode == O_RDWR) 68764373Seric #ifdef OLD_NEWDB 68864373Seric (void) db->sync(db); 68964373Seric #else 69060585Seric (void) db->sync(db, 0); 69160585Seric 69264373Seric if (fd >= 0 && fstat(fd, &st) >= 0) 69364284Seric map->map_mtime = st.st_mtime; 69464284Seric #endif 69564284Seric 69660089Seric map->map_db2 = (void *) db; 69760207Seric if (mode == O_RDONLY && bitset(MF_ALIAS, map->map_mflags)) 69860207Seric aliaswait(map, ".db"); 69956822Seric return TRUE; 70056822Seric } 70156822Seric 70256822Seric 70356822Seric /* 70456822Seric ** DB_MAP_LOOKUP -- look up a datum in a BTREE- or HASH-type map 70556822Seric */ 70656822Seric 70756822Seric char * 70860089Seric db_map_lookup(map, name, av, statp) 70956822Seric MAP *map; 71060089Seric char *name; 71156822Seric char **av; 71259084Seric int *statp; 71356822Seric { 71456822Seric DBT key, val; 71560422Seric register DB *db = (DB *) map->map_db2; 71660422Seric int st; 71760422Seric int saveerrno; 71864373Seric int fd; 71960089Seric char keybuf[MAXNAME + 1]; 72056822Seric 72160537Seric if (tTd(38, 20)) 72260089Seric printf("db_map_lookup(%s)\n", name); 72360089Seric 72460089Seric key.size = strlen(name); 72560089Seric if (key.size > sizeof keybuf - 1) 72660089Seric key.size = sizeof keybuf - 1; 72760089Seric key.data = keybuf; 72860089Seric bcopy(name, keybuf, key.size + 1); 72960207Seric if (!bitset(MF_NOFOLDCASE, map->map_mflags)) 73060089Seric makelower(keybuf); 73160422Seric #ifndef OLD_NEWDB 73264388Seric fd = db->fd(db); 73364388Seric if (fd >= 0 && !bitset(MF_LOCKED, map->map_mflags)) 73464388Seric (void) lockfile(db->fd(db), map->map_file, ".db", LOCK_SH); 73560422Seric #endif 73663753Seric st = 1; 73763753Seric if (bitset(MF_TRY0NULL, map->map_mflags)) 73863753Seric { 73963753Seric st = db->get(db, &key, &val, 0); 74063753Seric if (st == 0) 74163753Seric map->map_mflags &= ~MF_TRY1NULL; 74263753Seric } 74363753Seric if (st != 0 && bitset(MF_TRY1NULL, map->map_mflags)) 74463753Seric { 74563753Seric key.size++; 74663753Seric st = db->get(db, &key, &val, 0); 74763753Seric if (st == 0) 74863753Seric map->map_mflags &= ~MF_TRY0NULL; 74963753Seric } 75060422Seric saveerrno = errno; 75160422Seric #ifndef OLD_NEWDB 75264388Seric if (fd >= 0 && !bitset(MF_LOCKED, map->map_mflags)) 75364373Seric (void) lockfile(fd, map->map_file, ".db", LOCK_UN); 75460422Seric #endif 75560422Seric if (st != 0) 75660422Seric { 75760422Seric errno = saveerrno; 75860422Seric if (st < 0) 75960422Seric syserr("db_map_lookup: get (%s)", name); 76056822Seric return NULL; 76160422Seric } 76260207Seric if (bitset(MF_MATCHONLY, map->map_mflags)) 76363753Seric return map_rewrite(map, name, strlen(name), NULL); 76463753Seric else 76563753Seric return map_rewrite(map, val.data, val.size, av); 76656822Seric } 76756822Seric 76860089Seric 76960089Seric /* 77060089Seric ** DB_MAP_STORE -- store a datum in the NEWDB database 77156822Seric */ 77256822Seric 77360089Seric void 77460089Seric db_map_store(map, lhs, rhs) 77560089Seric register MAP *map; 77660089Seric char *lhs; 77760089Seric char *rhs; 77856822Seric { 77960089Seric int stat; 78060089Seric DBT key; 78160089Seric DBT data; 78260089Seric register DB *db = map->map_db2; 78356822Seric 78460537Seric if (tTd(38, 20)) 78560089Seric printf("db_map_store(%s, %s)\n", lhs, rhs); 78660089Seric 78760089Seric key.size = strlen(lhs); 78860089Seric key.data = lhs; 78960089Seric 79060089Seric data.size = strlen(rhs); 79160089Seric data.data = rhs; 79260089Seric 79360207Seric if (bitset(MF_INCLNULL, map->map_mflags)) 79456822Seric { 79560089Seric key.size++; 79660089Seric data.size++; 79760089Seric } 79856836Seric 79960089Seric stat = db->put(db, &key, &data, R_NOOVERWRITE); 80060089Seric if (stat > 0) 80160089Seric { 80260089Seric usrerr("050 Warning: duplicate alias name %s", lhs); 80360089Seric stat = db->put(db, &key, &data, 0); 80460089Seric } 80560089Seric if (stat != 0) 80660089Seric syserr("readaliases: db put (%s)", lhs); 80760089Seric } 80856836Seric 80956847Seric 81060089Seric /* 81160089Seric ** DB_MAP_CLOSE -- add distinguished entries and close the database 81260089Seric */ 81360089Seric 81460089Seric void 81560089Seric db_map_close(map) 81660089Seric MAP *map; 81760089Seric { 81860089Seric register DB *db = map->map_db2; 81960089Seric 82060537Seric if (tTd(38, 9)) 82160207Seric printf("db_map_close(%s, %x)\n", map->map_file, map->map_mflags); 82260089Seric 82360207Seric if (bitset(MF_WRITABLE, map->map_mflags)) 82458804Seric { 82560089Seric /* write out the distinguished alias */ 82660089Seric db_map_store(map, "@", "@"); 82758804Seric } 82858963Seric 82960089Seric if (db->close(db) != 0) 83060089Seric syserr("readaliases: db close failure"); 83156822Seric } 83257208Seric 83360089Seric #endif 83460089Seric /* 83560089Seric ** NIS Modules 83660089Seric */ 83760089Seric 83860089Seric # ifdef NIS 83960089Seric 84064369Seric # ifndef YPERR_BUSY 84164369Seric # define YPERR_BUSY 16 84264369Seric # endif 84364369Seric 84457208Seric /* 84560089Seric ** NIS_MAP_OPEN -- open DBM map 84657208Seric */ 84757208Seric 84857208Seric bool 84960089Seric nis_map_open(map, mode) 85057208Seric MAP *map; 85160089Seric int mode; 85257208Seric { 85357216Seric int yperr; 85460215Seric register char *p; 85560215Seric auto char *vp; 85660215Seric auto int vsize; 85757216Seric char *master; 85857216Seric 85960537Seric if (tTd(38, 2)) 86060089Seric printf("nis_map_open(%s)\n", map->map_file); 86160089Seric 86260207Seric if (mode != O_RDONLY) 86360207Seric { 864*64650Seric /* issue a pseudo-error message */ 865*64650Seric #ifdef ENOSYS 866*64650Seric errno = ENOSYS; 867*64650Seric #else 868*64650Seric # ifdef EFTYPE 869*64650Seric errno = EFTYPE; 870*64650Seric # else 871*64650Seric errno = ENXIO; 872*64650Seric # endif 873*64650Seric #endif 87460207Seric return FALSE; 87560207Seric } 87660207Seric 87760089Seric p = strchr(map->map_file, '@'); 87860089Seric if (p != NULL) 87960089Seric { 88060089Seric *p++ = '\0'; 88160089Seric if (*p != '\0') 88260089Seric map->map_domain = p; 88360089Seric } 88460215Seric 88560089Seric if (map->map_domain == NULL) 88660089Seric yp_get_default_domain(&map->map_domain); 88760089Seric 88860089Seric if (*map->map_file == '\0') 88960089Seric map->map_file = "mail.aliases"; 89060089Seric 89160215Seric /* check to see if this map actually exists */ 89260089Seric yperr = yp_match(map->map_domain, map->map_file, "@", 1, 89360089Seric &vp, &vsize); 89460537Seric if (tTd(38, 10)) 89560089Seric printf("nis_map_open: yp_match(%s, %s) => %s\n", 89660089Seric map->map_domain, map->map_file, yperr_string(yperr)); 89760089Seric if (yperr == 0 || yperr == YPERR_KEY || yperr == YPERR_BUSY) 89860089Seric return TRUE; 89960215Seric 90060215Seric if (!bitset(MF_OPTIONAL, map->map_mflags)) 90160215Seric syserr("Cannot bind to domain %s: %s", map->map_domain, 90260215Seric yperr_string(yperr)); 90360215Seric 90460089Seric return FALSE; 90560089Seric } 90660089Seric 90760089Seric 90860089Seric /* 90957208Seric ** NIS_MAP_LOOKUP -- look up a datum in a NIS map 91057208Seric */ 91157208Seric 91257208Seric char * 91360089Seric nis_map_lookup(map, name, av, statp) 91457208Seric MAP *map; 91560089Seric char *name; 91657208Seric char **av; 91759084Seric int *statp; 91857208Seric { 91957208Seric char *vp; 92057642Seric auto int vsize; 92159274Seric int buflen; 92260215Seric int yperr; 92360089Seric char keybuf[MAXNAME + 1]; 92457208Seric 92560537Seric if (tTd(38, 20)) 92660089Seric printf("nis_map_lookup(%s)\n", name); 92760089Seric 92860089Seric buflen = strlen(name); 92960089Seric if (buflen > sizeof keybuf - 1) 93060089Seric buflen = sizeof keybuf - 1; 93160089Seric bcopy(name, keybuf, buflen + 1); 93260207Seric if (!bitset(MF_NOFOLDCASE, map->map_mflags)) 93360089Seric makelower(keybuf); 93463753Seric yperr = YPERR_KEY; 93563753Seric if (bitset(MF_TRY0NULL, map->map_mflags)) 93663753Seric { 93763753Seric yperr = yp_match(map->map_domain, map->map_file, keybuf, buflen, 93863753Seric &vp, &vsize); 93963753Seric if (yperr == 0) 94063753Seric map->map_mflags &= ~MF_TRY1NULL; 94163753Seric } 94263753Seric if (yperr == YPERR_KEY && bitset(MF_TRY1NULL, map->map_mflags)) 94363753Seric { 94459274Seric buflen++; 94563753Seric yperr = yp_match(map->map_domain, map->map_file, keybuf, buflen, 94663753Seric &vp, &vsize); 94763753Seric if (yperr == 0) 94863753Seric map->map_mflags &= ~MF_TRY0NULL; 94963753Seric } 95060089Seric if (yperr != 0) 95160089Seric { 95260089Seric if (yperr != YPERR_KEY && yperr != YPERR_BUSY) 95360215Seric map->map_mflags &= ~(MF_VALID|MF_OPEN); 95457208Seric return NULL; 95560089Seric } 95660207Seric if (bitset(MF_MATCHONLY, map->map_mflags)) 95763753Seric return map_rewrite(map, name, strlen(name), NULL); 95863753Seric else 95963753Seric return map_rewrite(map, vp, vsize, av); 96057208Seric } 96157208Seric 96260089Seric 96360089Seric /* 96460207Seric ** NIS_MAP_STORE 96560089Seric */ 96660089Seric 96760089Seric void 96860089Seric nis_map_store(map, lhs, rhs) 96960089Seric MAP *map; 97060089Seric char *lhs; 97160089Seric char *rhs; 97260089Seric { 97360089Seric /* nothing */ 97460089Seric } 97560089Seric 97660089Seric 97760089Seric /* 97860207Seric ** NIS_MAP_CLOSE 97960089Seric */ 98060089Seric 98160089Seric void 98260089Seric nis_map_close(map) 98360089Seric MAP *map; 98460089Seric { 98560089Seric /* nothing */ 98660089Seric } 98760089Seric 98860089Seric #endif /* NIS */ 98957208Seric /* 99060089Seric ** STAB (Symbol Table) Modules 99160089Seric */ 99260089Seric 99360089Seric 99460089Seric /* 99560207Seric ** STAB_MAP_LOOKUP -- look up alias in symbol table 99660089Seric */ 99760089Seric 99860089Seric char * 99961707Seric stab_map_lookup(map, name, av, pstat) 100060089Seric register MAP *map; 100160089Seric char *name; 100261707Seric char **av; 100361707Seric int *pstat; 100460089Seric { 100560089Seric register STAB *s; 100660089Seric 100760537Seric if (tTd(38, 20)) 100860089Seric printf("stab_lookup(%s)\n", name); 100960089Seric 101060089Seric s = stab(name, ST_ALIAS, ST_FIND); 101160089Seric if (s != NULL) 101260089Seric return (s->s_alias); 101360089Seric return (NULL); 101460089Seric } 101560089Seric 101660089Seric 101760089Seric /* 101860207Seric ** STAB_MAP_STORE -- store in symtab (actually using during init, not rebuild) 101960089Seric */ 102060089Seric 102160089Seric void 102260089Seric stab_map_store(map, lhs, rhs) 102360089Seric register MAP *map; 102460089Seric char *lhs; 102560089Seric char *rhs; 102660089Seric { 102760089Seric register STAB *s; 102860089Seric 102960089Seric s = stab(lhs, ST_ALIAS, ST_ENTER); 103060089Seric s->s_alias = newstr(rhs); 103160089Seric } 103260089Seric 103360089Seric 103460089Seric /* 103560207Seric ** STAB_MAP_OPEN -- initialize (reads data file) 103660207Seric ** 103760207Seric ** This is a wierd case -- it is only intended as a fallback for 103860207Seric ** aliases. For this reason, opens for write (only during a 103960207Seric ** "newaliases") always fails, and opens for read open the 104060207Seric ** actual underlying text file instead of the database. 104160089Seric */ 104260089Seric 104360089Seric bool 104460089Seric stab_map_open(map, mode) 104560089Seric register MAP *map; 104660089Seric int mode; 104760089Seric { 104863835Seric FILE *af; 104964284Seric struct stat st; 105063835Seric 105160537Seric if (tTd(38, 2)) 105260089Seric printf("stab_map_open(%s)\n", map->map_file); 105360089Seric 105460089Seric if (mode != O_RDONLY) 105560207Seric { 105660207Seric errno = ENODEV; 105760089Seric return FALSE; 105860207Seric } 105960089Seric 106063835Seric af = fopen(map->map_file, "r"); 106163835Seric if (af == NULL) 106263835Seric return FALSE; 106363835Seric readaliases(map, af, TRUE); 106464284Seric 106564284Seric if (fstat(fileno(af), &st) >= 0) 106664284Seric map->map_mtime = st.st_mtime; 106763835Seric fclose(af); 106863835Seric 106960089Seric return TRUE; 107060089Seric } 107160089Seric 107260089Seric 107360089Seric /* 107464642Seric ** STAB_MAP_CLOSE -- close symbol table. 107564642Seric ** 107664642Seric ** Since this is in memory, there is nothing to do. 107760089Seric */ 107860089Seric 107960089Seric void 108060089Seric stab_map_close(map) 108160089Seric MAP *map; 108260089Seric { 108360089Seric /* ignore it */ 108460089Seric } 108560089Seric /* 108660089Seric ** Implicit Modules 108756822Seric ** 108860089Seric ** Tries several types. For back compatibility of aliases. 108956822Seric */ 109056822Seric 109160089Seric 109260089Seric /* 109360207Seric ** IMPL_MAP_LOOKUP -- lookup in best open database 109460089Seric */ 109560089Seric 109660089Seric char * 109760089Seric impl_map_lookup(map, name, av, pstat) 109860089Seric MAP *map; 109960089Seric char *name; 110056822Seric char **av; 110160089Seric int *pstat; 110256822Seric { 110360537Seric if (tTd(38, 20)) 110460089Seric printf("impl_map_lookup(%s)\n", name); 110556822Seric 110660089Seric #ifdef NEWDB 110760207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 110860089Seric return db_map_lookup(map, name, av, pstat); 110960089Seric #endif 111060089Seric #ifdef NDBM 111160207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 111260089Seric return ndbm_map_lookup(map, name, av, pstat); 111360089Seric #endif 111460089Seric return stab_map_lookup(map, name, av, pstat); 111560089Seric } 111660089Seric 111760089Seric /* 111860207Seric ** IMPL_MAP_STORE -- store in open databases 111960089Seric */ 112060089Seric 112160089Seric void 112260089Seric impl_map_store(map, lhs, rhs) 112360089Seric MAP *map; 112460089Seric char *lhs; 112560089Seric char *rhs; 112660089Seric { 112760089Seric #ifdef NEWDB 112860207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 112960089Seric db_map_store(map, lhs, rhs); 113060089Seric #endif 113160089Seric #ifdef NDBM 113260207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 113360089Seric ndbm_map_store(map, lhs, rhs); 113460089Seric #endif 113560089Seric stab_map_store(map, lhs, rhs); 113660089Seric } 113760089Seric 113860089Seric /* 113960089Seric ** IMPL_MAP_OPEN -- implicit database open 114060089Seric */ 114160089Seric 114260089Seric bool 114360089Seric impl_map_open(map, mode) 114460089Seric MAP *map; 114560089Seric int mode; 114660089Seric { 114760089Seric struct stat stb; 114860089Seric 114960537Seric if (tTd(38, 2)) 115060089Seric printf("impl_map_open(%s)\n", map->map_file); 115160089Seric 115260089Seric if (stat(map->map_file, &stb) < 0) 115356822Seric { 115460089Seric /* no alias file at all */ 115560089Seric return FALSE; 115656822Seric } 115756822Seric 115860089Seric #ifdef NEWDB 115960207Seric map->map_mflags |= MF_IMPL_HASH; 116060089Seric if (hash_map_open(map, mode)) 116156822Seric { 116264250Seric #if defined(NDBM) && defined(NIS) 116360561Seric if (mode == O_RDONLY || access("/var/yp/Makefile", R_OK) != 0) 116460207Seric #endif 116560207Seric return TRUE; 116660089Seric } 116760207Seric else 116860207Seric map->map_mflags &= ~MF_IMPL_HASH; 116960089Seric #endif 117060089Seric #ifdef NDBM 117160207Seric map->map_mflags |= MF_IMPL_NDBM; 117260089Seric if (ndbm_map_open(map, mode)) 117360089Seric { 117460089Seric return TRUE; 117560089Seric } 117660207Seric else 117760207Seric map->map_mflags &= ~MF_IMPL_NDBM; 117860089Seric #endif 117956822Seric 1180*64650Seric #if defined(NEWDB) || defined(NDBM) 118160089Seric if (Verbose) 118260089Seric message("WARNING: cannot open alias database %s", map->map_file); 118360207Seric #endif 118460089Seric 118560207Seric return stab_map_open(map, mode); 118656822Seric } 118760089Seric 118860207Seric 118960089Seric /* 119060207Seric ** IMPL_MAP_CLOSE -- close any open database(s) 119160089Seric */ 119260089Seric 119360089Seric void 119460207Seric impl_map_close(map) 119560089Seric MAP *map; 119660089Seric { 119760089Seric #ifdef NEWDB 119860207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 119960089Seric { 120060207Seric db_map_close(map); 120160207Seric map->map_mflags &= ~MF_IMPL_HASH; 120260089Seric } 120360089Seric #endif 120460089Seric 120560089Seric #ifdef NDBM 120660207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 120760089Seric { 120860207Seric ndbm_map_close(map); 120960207Seric map->map_mflags &= ~MF_IMPL_NDBM; 121060089Seric } 121160089Seric #endif 121260089Seric } 121360207Seric /* 121460207Seric ** NULL stubs 121560089Seric */ 121660089Seric 121760207Seric bool 121860207Seric null_map_open(map, mode) 121960089Seric MAP *map; 122060207Seric int mode; 122160089Seric { 122260207Seric return TRUE; 122360089Seric } 122460089Seric 122560207Seric void 122660207Seric null_map_close(map) 122760207Seric MAP *map; 122860089Seric { 122960207Seric return; 123060207Seric } 123160089Seric 123260207Seric void 123360207Seric null_map_store(map, key, val) 123460207Seric MAP *map; 123560207Seric char *key; 123660207Seric char *val; 123760089Seric { 123860207Seric return; 123960089Seric } 1240