535 lines
9.9 KiB
C
535 lines
9.9 KiB
C
/*
|
||
* Copyright (c) 1992 Eric P. Allman.
|
||
* Copyright (c) 1992, 1993
|
||
* The Regents of the University of California. 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 the University of
|
||
* California, Berkeley and its contributors.
|
||
* 4. Neither the name of the University nor the names of its contributors
|
||
* may be used to endorse or promote products derived from this software
|
||
* without specific prior written permission.
|
||
*
|
||
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``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 REGENTS OR CONTRIBUTORS 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 sccsid[] = "@(#)makemap.c 8.19 (Berkeley) 11/18/96";
|
||
#endif /* not lint */
|
||
|
||
#include <stdio.h>
|
||
#include <sysexits.h>
|
||
#include <sys/types.h>
|
||
#include <ctype.h>
|
||
#include <string.h>
|
||
#include <sys/errno.h>
|
||
#ifndef ISC_UNIX
|
||
# include <sys/file.h>
|
||
#endif
|
||
#define NOT_SENDMAIL
|
||
#include "useful.h"
|
||
#include "conf.h"
|
||
|
||
#ifdef NDBM
|
||
#include <ndbm.h>
|
||
#endif
|
||
|
||
#ifdef NEWDB
|
||
#include <db.h>
|
||
#endif
|
||
|
||
enum type { T_DBM, T_BTREE, T_HASH, T_ERR, T_UNKNOWN };
|
||
|
||
union dbent
|
||
{
|
||
#ifdef NDBM
|
||
datum dbm;
|
||
#endif
|
||
#ifdef NEWDB
|
||
DBT db;
|
||
#endif
|
||
struct
|
||
{
|
||
char *data;
|
||
size_t size;
|
||
} xx;
|
||
};
|
||
|
||
#define BUFSIZE 1024
|
||
|
||
main(argc, argv)
|
||
int argc;
|
||
char **argv;
|
||
{
|
||
char *progname;
|
||
bool inclnull = FALSE;
|
||
bool notrunc = FALSE;
|
||
bool allowreplace = FALSE;
|
||
bool allowdups = FALSE;
|
||
bool verbose = FALSE;
|
||
bool foldcase = TRUE;
|
||
int exitstat;
|
||
int opt;
|
||
char *typename;
|
||
char *mapname;
|
||
char *ext;
|
||
char *lext;
|
||
int lineno;
|
||
int st;
|
||
int mode;
|
||
int putflags;
|
||
long dbcachesize = 1024 * 1024;
|
||
enum type type;
|
||
int fd;
|
||
union
|
||
{
|
||
#ifdef NDBM
|
||
DBM *dbm;
|
||
#endif
|
||
#ifdef NEWDB
|
||
DB *db;
|
||
#endif
|
||
void *dbx;
|
||
} dbp;
|
||
union dbent key, val;
|
||
#ifdef NEWDB
|
||
BTREEINFO bti;
|
||
HASHINFO hinfo;
|
||
#endif
|
||
char ibuf[BUFSIZE];
|
||
char fbuf[MAXNAME];
|
||
char lbuf[MAXNAME];
|
||
extern char *optarg;
|
||
extern int optind;
|
||
extern bool lockfile();
|
||
|
||
progname = argv[0];
|
||
|
||
#ifdef FFR_CFLAG
|
||
#define OPTIONS "Nc:dforv"
|
||
#else
|
||
#define OPTIONS "Ndforv"
|
||
#endif
|
||
while ((opt = getopt(argc, argv, OPTIONS)) != EOF)
|
||
{
|
||
switch (opt)
|
||
{
|
||
case 'N':
|
||
inclnull = TRUE;
|
||
break;
|
||
|
||
#ifdef FFR_CFLAG
|
||
case 'c':
|
||
dbcachesize = atol(optarg);
|
||
break;
|
||
#endif
|
||
|
||
case 'd':
|
||
allowdups = TRUE;
|
||
break;
|
||
|
||
case 'f':
|
||
foldcase = FALSE;
|
||
break;
|
||
|
||
case 'o':
|
||
notrunc = TRUE;
|
||
break;
|
||
|
||
case 'r':
|
||
allowreplace = TRUE;
|
||
break;
|
||
|
||
case 'v':
|
||
verbose = TRUE;
|
||
break;
|
||
|
||
default:
|
||
type = T_ERR;
|
||
break;
|
||
}
|
||
}
|
||
|
||
argc -= optind;
|
||
argv += optind;
|
||
if (argc != 2)
|
||
type = T_ERR;
|
||
else
|
||
{
|
||
typename = argv[0];
|
||
mapname = argv[1];
|
||
ext = NULL;
|
||
lext = NULL;
|
||
|
||
if (strcmp(typename, "dbm") == 0)
|
||
{
|
||
type = T_DBM;
|
||
lext = ".dir";
|
||
}
|
||
else if (strcmp(typename, "btree") == 0)
|
||
{
|
||
type = T_BTREE;
|
||
ext = ".db";
|
||
}
|
||
else if (strcmp(typename, "hash") == 0)
|
||
{
|
||
type = T_HASH;
|
||
ext = ".db";
|
||
}
|
||
else
|
||
type = T_UNKNOWN;
|
||
}
|
||
|
||
switch (type)
|
||
{
|
||
case T_ERR:
|
||
#ifdef FFR_CFLAG
|
||
fprintf(stderr, "Usage: %s [-N] [-c cachesize] [-d] [-f] [-o] [-r] [-v] type mapname\n", progname);
|
||
#else
|
||
fprintf(stderr, "Usage: %s [-N] [-d] [-f] [-o] [-r] [-v] type mapname\n", progname);
|
||
#endif
|
||
exit(EX_USAGE);
|
||
|
||
case T_UNKNOWN:
|
||
fprintf(stderr, "%s: Unknown database type %s\n",
|
||
progname, typename);
|
||
exit(EX_USAGE);
|
||
|
||
#ifndef NDBM
|
||
case T_DBM:
|
||
#endif
|
||
#ifndef NEWDB
|
||
case T_BTREE:
|
||
case T_HASH:
|
||
#endif
|
||
fprintf(stderr, "%s: Type %s not supported in this version\n",
|
||
progname, typename);
|
||
exit(EX_UNAVAILABLE);
|
||
|
||
#ifdef NEWDB
|
||
case T_BTREE:
|
||
bzero(&bti, sizeof bti);
|
||
if (allowdups)
|
||
bti.flags |= R_DUP;
|
||
if (allowdups || allowreplace)
|
||
putflags = 0;
|
||
else
|
||
putflags = R_NOOVERWRITE;
|
||
break;
|
||
|
||
case T_HASH:
|
||
bzero(&hinfo, sizeof hinfo);
|
||
if (allowreplace)
|
||
putflags = 0;
|
||
else
|
||
putflags = R_NOOVERWRITE;
|
||
break;
|
||
#endif
|
||
#ifdef NDBM
|
||
case T_DBM:
|
||
if (allowdups)
|
||
{
|
||
fprintf(stderr, "%s: Type %s does not support -d (allow dups)\n",
|
||
progname, typename);
|
||
exit(EX_UNAVAILABLE);
|
||
}
|
||
if (allowreplace)
|
||
putflags = DBM_REPLACE;
|
||
else
|
||
putflags = DBM_INSERT;
|
||
break;
|
||
#endif
|
||
}
|
||
|
||
/*
|
||
** Adjust file names.
|
||
*/
|
||
|
||
if (ext != NULL)
|
||
{
|
||
int el, fl;
|
||
|
||
el = strlen(ext);
|
||
fl = strlen(mapname);
|
||
if (fl < el || strcmp(&mapname[fl - el], ext) != 0)
|
||
{
|
||
strcpy(fbuf, mapname);
|
||
strcat(fbuf, ext);
|
||
mapname = fbuf;
|
||
}
|
||
}
|
||
|
||
strcpy(lbuf, mapname);
|
||
if (lext != NULL)
|
||
strcat(lbuf, lext);
|
||
|
||
/*
|
||
** Create the database.
|
||
*/
|
||
|
||
mode = O_RDWR;
|
||
if (!notrunc)
|
||
mode |= O_CREAT|O_TRUNC;
|
||
#ifdef O_EXLOCK
|
||
mode |= O_EXLOCK;
|
||
#else
|
||
/* pre-lock the database */
|
||
fd = open(lbuf, mode & ~O_TRUNC, 0644);
|
||
if (fd < 0)
|
||
{
|
||
fprintf(stderr, "%s: cannot create type %s map %s\n",
|
||
progname, typename, mapname);
|
||
exit(EX_CANTCREAT);
|
||
}
|
||
(void) lockfile(fd);
|
||
#endif
|
||
switch (type)
|
||
{
|
||
#ifdef NDBM
|
||
case T_DBM:
|
||
dbp.dbm = dbm_open(mapname, mode, 0644);
|
||
break;
|
||
#endif
|
||
|
||
#ifdef NEWDB
|
||
case T_HASH:
|
||
/* tweak some parameters for performance */
|
||
hinfo.nelem = 4096;
|
||
hinfo.cachesize = dbcachesize;
|
||
|
||
dbp.db = dbopen(mapname, mode, 0644, DB_HASH, &hinfo);
|
||
if (dbp.db != NULL)
|
||
{
|
||
# if OLD_NEWDB
|
||
(void) (*dbp.db->sync)(dbp.db);
|
||
# else
|
||
(void) (*dbp.db->sync)(dbp.db, 0);
|
||
# endif
|
||
}
|
||
break;
|
||
|
||
case T_BTREE:
|
||
/* tweak some parameters for performance */
|
||
bti.cachesize = dbcachesize;
|
||
|
||
dbp.db = dbopen(mapname, mode, 0644, DB_BTREE, &bti);
|
||
if (dbp.db != NULL)
|
||
{
|
||
# if OLD_NEWDB
|
||
(void) (*dbp.db->sync)(dbp.db);
|
||
# else
|
||
(void) (*dbp.db->sync)(dbp.db, 0);
|
||
# endif
|
||
}
|
||
break;
|
||
#endif
|
||
|
||
default:
|
||
fprintf(stderr, "%s: internal error: type %d\n", progname, type);
|
||
exit(EX_SOFTWARE);
|
||
}
|
||
|
||
if (dbp.dbx == NULL)
|
||
{
|
||
fprintf(stderr, "%s: cannot open type %s map %s\n",
|
||
progname, typename, mapname);
|
||
exit(EX_CANTCREAT);
|
||
}
|
||
|
||
/*
|
||
** Copy the data
|
||
*/
|
||
|
||
lineno = 0;
|
||
exitstat = EX_OK;
|
||
while (fgets(ibuf, sizeof ibuf, stdin) != NULL)
|
||
{
|
||
register char *p;
|
||
|
||
lineno++;
|
||
|
||
/*
|
||
** Parse the line.
|
||
*/
|
||
|
||
p = strchr(ibuf, '\n');
|
||
if (p != NULL)
|
||
*p = '\0';
|
||
else if (!feof(stdin))
|
||
{
|
||
fprintf(stderr, "%s: %s: line %d: line too long (%d bytes max)\n",
|
||
progname, mapname, lineno, sizeof ibuf);
|
||
continue;
|
||
}
|
||
|
||
if (ibuf[0] == '\0' || ibuf[0] == '#')
|
||
continue;
|
||
if (isspace(ibuf[0]))
|
||
{
|
||
fprintf(stderr, "%s: %s: line %d: syntax error (leading space)\n",
|
||
progname, mapname, lineno);
|
||
continue;
|
||
}
|
||
key.xx.data = ibuf;
|
||
for (p = ibuf; *p != '\0' && !isspace(*p); p++)
|
||
{
|
||
if (foldcase && isupper(*p))
|
||
*p = tolower(*p);
|
||
}
|
||
key.xx.size = p - key.xx.data;
|
||
if (inclnull)
|
||
key.xx.size++;
|
||
if (*p != '\0')
|
||
*p++ = '\0';
|
||
while (isspace(*p))
|
||
p++;
|
||
if (*p == '\0')
|
||
{
|
||
fprintf(stderr, "%s: %s: line %d: no RHS for LHS %s\n",
|
||
progname, mapname, lineno, key.xx.data);
|
||
continue;
|
||
}
|
||
val.xx.data = p;
|
||
val.xx.size = strlen(p);
|
||
if (inclnull)
|
||
val.xx.size++;
|
||
|
||
/*
|
||
** Do the database insert.
|
||
*/
|
||
|
||
if (verbose)
|
||
{
|
||
printf("key=`%s', val=`%s'\n", key.xx.data, val.xx.data);
|
||
}
|
||
|
||
switch (type)
|
||
{
|
||
#ifdef NDBM
|
||
case T_DBM:
|
||
st = dbm_store(dbp.dbm, key.dbm, val.dbm, putflags);
|
||
break;
|
||
#endif
|
||
|
||
#ifdef NEWDB
|
||
case T_BTREE:
|
||
case T_HASH:
|
||
st = (*dbp.db->put)(dbp.db, &key.db, &val.db, putflags);
|
||
break;
|
||
#endif
|
||
}
|
||
|
||
if (st < 0)
|
||
{
|
||
fprintf(stderr, "%s: %s: line %d: key %s: put error\n",
|
||
progname, mapname, lineno, key.xx.data);
|
||
perror(mapname);
|
||
exitstat = EX_IOERR;
|
||
}
|
||
else if (st > 0)
|
||
{
|
||
fprintf(stderr, "%s: %s: line %d: key %s: duplicate key\n",
|
||
progname, mapname, lineno, key.xx.data);
|
||
}
|
||
}
|
||
|
||
/*
|
||
** Now close the database.
|
||
*/
|
||
|
||
switch (type)
|
||
{
|
||
#ifdef NDBM
|
||
case T_DBM:
|
||
dbm_close(dbp.dbm);
|
||
break;
|
||
#endif
|
||
|
||
#ifdef NEWDB
|
||
case T_HASH:
|
||
case T_BTREE:
|
||
if ((*dbp.db->close)(dbp.db) < 0)
|
||
{
|
||
fprintf(stderr, "%s: %s: error on close\n",
|
||
progname, mapname);
|
||
perror(mapname);
|
||
exitstat = EX_IOERR;
|
||
}
|
||
#endif
|
||
}
|
||
|
||
#ifndef O_EXLOCK
|
||
/* release locks */
|
||
close(fd);
|
||
#endif
|
||
|
||
exit (exitstat);
|
||
}
|
||
/*
|
||
** LOCKFILE -- lock a file using flock or (shudder) fcntl locking
|
||
**
|
||
** Parameters:
|
||
** fd -- the file descriptor of the file.
|
||
**
|
||
** Returns:
|
||
** TRUE if the lock was acquired.
|
||
** FALSE otherwise.
|
||
*/
|
||
|
||
bool
|
||
lockfile(fd)
|
||
int fd;
|
||
{
|
||
# if !HASFLOCK
|
||
int action;
|
||
struct flock lfd;
|
||
extern int errno;
|
||
|
||
bzero(&lfd, sizeof lfd);
|
||
lfd.l_type = F_WRLCK;
|
||
action = F_SETLKW;
|
||
|
||
if (fcntl(fd, action, &lfd) >= 0)
|
||
return TRUE;
|
||
|
||
/*
|
||
** On SunOS, if you are testing using -oQ/tmp/mqueue or
|
||
** -oA/tmp/aliases or anything like that, and /tmp is mounted
|
||
** as type "tmp" (that is, served from swap space), the
|
||
** previous fcntl will fail with "Invalid argument" errors.
|
||
** Since this is fairly common during testing, we will assume
|
||
** that this indicates that the lock is successfully grabbed.
|
||
*/
|
||
|
||
if (errno == EINVAL)
|
||
return TRUE;
|
||
|
||
# else /* HASFLOCK */
|
||
|
||
if (flock(fd, LOCK_EX) >= 0)
|
||
return TRUE;
|
||
|
||
# endif
|
||
|
||
return FALSE;
|
||
}
|