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*64388Seric static char sccsid[] = "@(#)map.c 8.12 (Berkeley) 09/02/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); 430*64388Seric 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 } 446*64388Seric 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) 594*64388Seric { 595*64388Seric if (lockfile(fd, map->map_file, ".db", LOCK_EX)) 596*64388Seric map->map_mflags |= MF_LOCKED; 597*64388Seric } 59860582Seric # else 59964373Seric if (mode == O_RDONLY && fd >= 0) 60064373Seric (void) lockfile(fd, map->map_file, ".db", LOCK_UN); 601*64388Seric else 602*64388Seric 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) 673*64388Seric { 674*64388Seric if (lockfile(fd, map->map_file, ".db", LOCK_EX)) 675*64388Seric map->map_mflags |= MF_LOCKED; 676*64388Seric } 67760582Seric # else 67864373Seric if (mode == O_RDONLY && fd >= 0) 67964373Seric (void) lockfile(fd, map->map_file, ".db", LOCK_UN); 680*64388Seric else 681*64388Seric 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 732*64388Seric fd = db->fd(db); 733*64388Seric if (fd >= 0 && !bitset(MF_LOCKED, map->map_mflags)) 734*64388Seric (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 752*64388Seric 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 { 86460207Seric errno = ENODEV; 86560207Seric return FALSE; 86660207Seric } 86760207Seric 86860089Seric p = strchr(map->map_file, '@'); 86960089Seric if (p != NULL) 87060089Seric { 87160089Seric *p++ = '\0'; 87260089Seric if (*p != '\0') 87360089Seric map->map_domain = p; 87460089Seric } 87560215Seric 87660089Seric if (map->map_domain == NULL) 87760089Seric yp_get_default_domain(&map->map_domain); 87860089Seric 87960089Seric if (*map->map_file == '\0') 88060089Seric map->map_file = "mail.aliases"; 88160089Seric 88260215Seric /* check to see if this map actually exists */ 88360089Seric yperr = yp_match(map->map_domain, map->map_file, "@", 1, 88460089Seric &vp, &vsize); 88560537Seric if (tTd(38, 10)) 88660089Seric printf("nis_map_open: yp_match(%s, %s) => %s\n", 88760089Seric map->map_domain, map->map_file, yperr_string(yperr)); 88860089Seric if (yperr == 0 || yperr == YPERR_KEY || yperr == YPERR_BUSY) 88960089Seric return TRUE; 89060215Seric 89160215Seric if (!bitset(MF_OPTIONAL, map->map_mflags)) 89260215Seric syserr("Cannot bind to domain %s: %s", map->map_domain, 89360215Seric yperr_string(yperr)); 89460215Seric 89560089Seric return FALSE; 89660089Seric } 89760089Seric 89860089Seric 89960089Seric /* 90057208Seric ** NIS_MAP_LOOKUP -- look up a datum in a NIS map 90157208Seric */ 90257208Seric 90357208Seric char * 90460089Seric nis_map_lookup(map, name, av, statp) 90557208Seric MAP *map; 90660089Seric char *name; 90757208Seric char **av; 90859084Seric int *statp; 90957208Seric { 91057208Seric char *vp; 91157642Seric auto int vsize; 91259274Seric int buflen; 91360215Seric int yperr; 91460089Seric char keybuf[MAXNAME + 1]; 91557208Seric 91660537Seric if (tTd(38, 20)) 91760089Seric printf("nis_map_lookup(%s)\n", name); 91860089Seric 91960089Seric buflen = strlen(name); 92060089Seric if (buflen > sizeof keybuf - 1) 92160089Seric buflen = sizeof keybuf - 1; 92260089Seric bcopy(name, keybuf, buflen + 1); 92360207Seric if (!bitset(MF_NOFOLDCASE, map->map_mflags)) 92460089Seric makelower(keybuf); 92563753Seric yperr = YPERR_KEY; 92663753Seric if (bitset(MF_TRY0NULL, map->map_mflags)) 92763753Seric { 92863753Seric yperr = yp_match(map->map_domain, map->map_file, keybuf, buflen, 92963753Seric &vp, &vsize); 93063753Seric if (yperr == 0) 93163753Seric map->map_mflags &= ~MF_TRY1NULL; 93263753Seric } 93363753Seric if (yperr == YPERR_KEY && bitset(MF_TRY1NULL, map->map_mflags)) 93463753Seric { 93559274Seric buflen++; 93663753Seric yperr = yp_match(map->map_domain, map->map_file, keybuf, buflen, 93763753Seric &vp, &vsize); 93863753Seric if (yperr == 0) 93963753Seric map->map_mflags &= ~MF_TRY0NULL; 94063753Seric } 94160089Seric if (yperr != 0) 94260089Seric { 94360089Seric if (yperr != YPERR_KEY && yperr != YPERR_BUSY) 94460215Seric map->map_mflags &= ~(MF_VALID|MF_OPEN); 94557208Seric return NULL; 94660089Seric } 94760207Seric if (bitset(MF_MATCHONLY, map->map_mflags)) 94863753Seric return map_rewrite(map, name, strlen(name), NULL); 94963753Seric else 95063753Seric return map_rewrite(map, vp, vsize, av); 95157208Seric } 95257208Seric 95360089Seric 95460089Seric /* 95560207Seric ** NIS_MAP_STORE 95660089Seric */ 95760089Seric 95860089Seric void 95960089Seric nis_map_store(map, lhs, rhs) 96060089Seric MAP *map; 96160089Seric char *lhs; 96260089Seric char *rhs; 96360089Seric { 96460089Seric /* nothing */ 96560089Seric } 96660089Seric 96760089Seric 96860089Seric /* 96960207Seric ** NIS_MAP_CLOSE 97060089Seric */ 97160089Seric 97260089Seric void 97360089Seric nis_map_close(map) 97460089Seric MAP *map; 97560089Seric { 97660089Seric /* nothing */ 97760089Seric } 97860089Seric 97960089Seric #endif /* NIS */ 98057208Seric /* 98160089Seric ** STAB (Symbol Table) Modules 98260089Seric */ 98360089Seric 98460089Seric 98560089Seric /* 98660207Seric ** STAB_MAP_LOOKUP -- look up alias in symbol table 98760089Seric */ 98860089Seric 98960089Seric char * 99061707Seric stab_map_lookup(map, name, av, pstat) 99160089Seric register MAP *map; 99260089Seric char *name; 99361707Seric char **av; 99461707Seric int *pstat; 99560089Seric { 99660089Seric register STAB *s; 99760089Seric 99860537Seric if (tTd(38, 20)) 99960089Seric printf("stab_lookup(%s)\n", name); 100060089Seric 100160089Seric s = stab(name, ST_ALIAS, ST_FIND); 100260089Seric if (s != NULL) 100360089Seric return (s->s_alias); 100460089Seric return (NULL); 100560089Seric } 100660089Seric 100760089Seric 100860089Seric /* 100960207Seric ** STAB_MAP_STORE -- store in symtab (actually using during init, not rebuild) 101060089Seric */ 101160089Seric 101260089Seric void 101360089Seric stab_map_store(map, lhs, rhs) 101460089Seric register MAP *map; 101560089Seric char *lhs; 101660089Seric char *rhs; 101760089Seric { 101860089Seric register STAB *s; 101960089Seric 102060089Seric s = stab(lhs, ST_ALIAS, ST_ENTER); 102160089Seric s->s_alias = newstr(rhs); 102260089Seric } 102360089Seric 102460089Seric 102560089Seric /* 102660207Seric ** STAB_MAP_OPEN -- initialize (reads data file) 102760207Seric ** 102860207Seric ** This is a wierd case -- it is only intended as a fallback for 102960207Seric ** aliases. For this reason, opens for write (only during a 103060207Seric ** "newaliases") always fails, and opens for read open the 103160207Seric ** actual underlying text file instead of the database. 103260089Seric */ 103360089Seric 103460089Seric bool 103560089Seric stab_map_open(map, mode) 103660089Seric register MAP *map; 103760089Seric int mode; 103860089Seric { 103963835Seric FILE *af; 104064284Seric struct stat st; 104163835Seric 104260537Seric if (tTd(38, 2)) 104360089Seric printf("stab_map_open(%s)\n", map->map_file); 104460089Seric 104560089Seric if (mode != O_RDONLY) 104660207Seric { 104760207Seric errno = ENODEV; 104860089Seric return FALSE; 104960207Seric } 105060089Seric 105163835Seric af = fopen(map->map_file, "r"); 105263835Seric if (af == NULL) 105363835Seric return FALSE; 105463835Seric readaliases(map, af, TRUE); 105564284Seric 105664284Seric if (fstat(fileno(af), &st) >= 0) 105764284Seric map->map_mtime = st.st_mtime; 105863835Seric fclose(af); 105963835Seric 106060089Seric return TRUE; 106160089Seric } 106260089Seric 106360089Seric 106460089Seric /* 106560207Seric ** STAB_MAP_CLOSE -- close symbol table (???) 106660089Seric */ 106760089Seric 106860089Seric void 106960089Seric stab_map_close(map) 107060089Seric MAP *map; 107160089Seric { 107260089Seric /* ignore it */ 107360089Seric } 107460089Seric /* 107560089Seric ** Implicit Modules 107656822Seric ** 107760089Seric ** Tries several types. For back compatibility of aliases. 107856822Seric */ 107956822Seric 108060089Seric 108160089Seric /* 108260207Seric ** IMPL_MAP_LOOKUP -- lookup in best open database 108360089Seric */ 108460089Seric 108560089Seric char * 108660089Seric impl_map_lookup(map, name, av, pstat) 108760089Seric MAP *map; 108860089Seric char *name; 108956822Seric char **av; 109060089Seric int *pstat; 109156822Seric { 109260537Seric if (tTd(38, 20)) 109360089Seric printf("impl_map_lookup(%s)\n", name); 109456822Seric 109560089Seric #ifdef NEWDB 109660207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 109760089Seric return db_map_lookup(map, name, av, pstat); 109860089Seric #endif 109960089Seric #ifdef NDBM 110060207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 110160089Seric return ndbm_map_lookup(map, name, av, pstat); 110260089Seric #endif 110360089Seric return stab_map_lookup(map, name, av, pstat); 110460089Seric } 110560089Seric 110660089Seric /* 110760207Seric ** IMPL_MAP_STORE -- store in open databases 110860089Seric */ 110960089Seric 111060089Seric void 111160089Seric impl_map_store(map, lhs, rhs) 111260089Seric MAP *map; 111360089Seric char *lhs; 111460089Seric char *rhs; 111560089Seric { 111660089Seric #ifdef NEWDB 111760207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 111860089Seric db_map_store(map, lhs, rhs); 111960089Seric #endif 112060089Seric #ifdef NDBM 112160207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 112260089Seric ndbm_map_store(map, lhs, rhs); 112360089Seric #endif 112460089Seric stab_map_store(map, lhs, rhs); 112560089Seric } 112660089Seric 112760089Seric /* 112860089Seric ** IMPL_MAP_OPEN -- implicit database open 112960089Seric */ 113060089Seric 113160089Seric bool 113260089Seric impl_map_open(map, mode) 113360089Seric MAP *map; 113460089Seric int mode; 113560089Seric { 113660089Seric struct stat stb; 113760089Seric 113860537Seric if (tTd(38, 2)) 113960089Seric printf("impl_map_open(%s)\n", map->map_file); 114060089Seric 114160089Seric if (stat(map->map_file, &stb) < 0) 114256822Seric { 114360089Seric /* no alias file at all */ 114460089Seric return FALSE; 114556822Seric } 114656822Seric 114760089Seric #ifdef NEWDB 114860207Seric map->map_mflags |= MF_IMPL_HASH; 114960089Seric if (hash_map_open(map, mode)) 115056822Seric { 115164250Seric #if defined(NDBM) && defined(NIS) 115260561Seric if (mode == O_RDONLY || access("/var/yp/Makefile", R_OK) != 0) 115360207Seric #endif 115460207Seric return TRUE; 115560089Seric } 115660207Seric else 115760207Seric map->map_mflags &= ~MF_IMPL_HASH; 115860089Seric #endif 115960089Seric #ifdef NDBM 116060207Seric map->map_mflags |= MF_IMPL_NDBM; 116160089Seric if (ndbm_map_open(map, mode)) 116260089Seric { 116360089Seric return TRUE; 116460089Seric } 116560207Seric else 116660207Seric map->map_mflags &= ~MF_IMPL_NDBM; 116760089Seric #endif 116856822Seric 116960207Seric #if !defined(NEWDB) && !defined(NDBM) 117060089Seric if (Verbose) 117160089Seric message("WARNING: cannot open alias database %s", map->map_file); 117260207Seric #endif 117360089Seric 117460207Seric return stab_map_open(map, mode); 117556822Seric } 117660089Seric 117760207Seric 117860089Seric /* 117960207Seric ** IMPL_MAP_CLOSE -- close any open database(s) 118060089Seric */ 118160089Seric 118260089Seric void 118360207Seric impl_map_close(map) 118460089Seric MAP *map; 118560089Seric { 118660089Seric #ifdef NEWDB 118760207Seric if (bitset(MF_IMPL_HASH, map->map_mflags)) 118860089Seric { 118960207Seric db_map_close(map); 119060207Seric map->map_mflags &= ~MF_IMPL_HASH; 119160089Seric } 119260089Seric #endif 119360089Seric 119460089Seric #ifdef NDBM 119560207Seric if (bitset(MF_IMPL_NDBM, map->map_mflags)) 119660089Seric { 119760207Seric ndbm_map_close(map); 119860207Seric map->map_mflags &= ~MF_IMPL_NDBM; 119960089Seric } 120060089Seric #endif 120160089Seric } 120260207Seric /* 120360207Seric ** NULL stubs 120460089Seric */ 120560089Seric 120660207Seric bool 120760207Seric null_map_open(map, mode) 120860089Seric MAP *map; 120960207Seric int mode; 121060089Seric { 121160207Seric return TRUE; 121260089Seric } 121360089Seric 121460207Seric void 121560207Seric null_map_close(map) 121660207Seric MAP *map; 121760089Seric { 121860207Seric return; 121960207Seric } 122060089Seric 122160207Seric void 122260207Seric null_map_store(map, key, val) 122360207Seric MAP *map; 122460207Seric char *key; 122560207Seric char *val; 122660089Seric { 122760207Seric return; 122860089Seric } 1229