NetBSD/usr.sbin/sendmail/makemap/makemap.c

374 lines
7.0 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.6 (Berkeley) 11/22/93";
#endif /* not lint */
#include <stdio.h>
#include <sysexits.h>
#include <sys/types.h>
#include <sys/file.h>
#include <ctype.h>
#include <string.h>
#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;
int size;
} xx;
};
#define BUFSIZE 1024
main(argc, argv)
int argc;
char **argv;
{
char *progname;
bool inclnull = FALSE;
bool notrunc = FALSE;
bool allowreplace = FALSE;
bool verbose = FALSE;
bool foldcase = TRUE;
int exitstat;
int opt;
char *typename;
char *mapname;
char *ext;
int lineno;
int st;
int mode;
enum type type;
union
{
#ifdef NDBM
DBM *dbm;
#endif
#ifdef NEWDB
DB *db;
#endif
void *dbx;
} dbp;
union dbent key, val;
char ibuf[BUFSIZE];
char fbuf[MAXNAME];
extern char *optarg;
extern int optind;
progname = argv[0];
while ((opt = getopt(argc, argv, "Nforv")) != EOF)
{
switch (opt)
{
case 'N':
inclnull = 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;
if (strcmp(typename, "dbm") == 0)
{
type = T_DBM;
}
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:
fprintf(stderr, "Usage: %s [-N] [-o] [-v] type mapname\n", progname);
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);
}
/*
** 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;
}
}
/*
** Create the database.
*/
mode = O_RDWR;
if (!notrunc)
mode |= O_CREAT|O_TRUNC;
switch (type)
{
#ifdef NDBM
case T_DBM:
dbp.dbm = dbm_open(mapname, mode, 0644);
break;
#endif
#ifdef NEWDB
case T_HASH:
dbp.db = dbopen(mapname, mode, 0644, DB_HASH, NULL);
break;
case T_BTREE:
dbp.db = dbopen(mapname, mode, 0644, DB_BTREE, NULL);
break;
#endif
default:
fprintf(stderr, "%s: internal error: type %d\n", progname, type);
exit(EX_SOFTWARE);
}
if (dbp.dbx == NULL)
{
fprintf(stderr, "%s: cannot create 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,
allowreplace ? DBM_REPLACE : DBM_INSERT);
break;
#endif
#ifdef NEWDB
case T_BTREE:
case T_HASH:
st = (*dbp.db->put)(dbp.db, &key.db, &val.db,
allowreplace ? 0 : R_NOOVERWRITE);
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
}
exit (exitstat);
}