usr.sbin/sendmail/makemap/makemap.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.1 (Berkeley) 6/7/93";
#endif /* not lint */

#include <stdio.h>
#include <sysexits.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 = FALSE;
        int exitstat;
        int opt;
        char *typename;
        char *mapname;
        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];
        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 = TRUE;
                        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];

                if (strcmp(typename, "dbm") == 0)
                        type = T_DBM;
                else if (strcmp(typename, "btree") == 0)
                        type = T_BTREE;
                else if (strcmp(typename, "hash") == 0)
                        type = T_HASH;
                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);
        }

        /*
        **  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 != '\0')
                        *p = '\0';
                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);
}