/*      $NetBSD: makedbm.c,v 1.25 2011/08/30 21:10:28 joerg Exp $       */

/*
 * Copyright (c) 1994 Mats O Jansson <moj@stacken.kth.se>
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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.
 */

#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: makedbm.c,v 1.25 2011/08/30 21:10:28 joerg Exp $");
#endif

#include <sys/param.h>
#include <sys/stat.h>

#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include <rpc/rpc.h>
#include <rpc/xdr.h>

#include "protos.h"
#include "ypdb.h"
#include "ypdef.h"

__dead static void      usage(void);
static int      add_record(DBM *, const char *, const char *, int);
static char     *file_date(char *);
static void     list_database(char *);
static void     create_database(char *, char *, char *, char *, char *,
                                char *, int, int, int);

int
main(int argc, char *argv[])
{
        int aflag, uflag, bflag, lflag, sflag;
        char *yp_input_file, *yp_output_file;
        char *yp_master_name, *yp_domain_name;
        char *infile, *outfile;
        int ch;

        yp_input_file = yp_output_file = NULL;
        yp_master_name = yp_domain_name = NULL;
        aflag = uflag = bflag = lflag = sflag = 0;
        infile = outfile = NULL;

        while ((ch = getopt(argc, argv, "blsui:o:m:d:")) != -1) {
                switch (ch) {
                case 'b':
                        bflag = aflag = 1;
                        break;

                case 'l':
                        lflag = aflag = 1;
                        break;

                case 's':
                        sflag = aflag = 1;
                        break;

                case 'i':
                        yp_input_file = optarg;
                        aflag = 1;
                        break;

                case 'o':
                        yp_output_file = optarg;
                        aflag = 1;
                        break;

                case 'm':
                        yp_master_name = optarg;
                        aflag = 1;
                        break;

                case 'd':
                        yp_domain_name = optarg;
                        aflag = 1;
                        break;

                case 'u':
                        uflag = 1;
                        break;

                default:
                        usage();
                }
        }
        argc -= optind; argv += optind;

        if ((uflag != 0) && (aflag != 0))
                usage();
        else {
                if (uflag != 0) {
                        if (argc == 1)
                                infile = argv[0];
                        else
                                usage();
                } else {
                        if (argc == 2) {
                                infile = argv[0];
                                outfile = argv[1];
                        } else
                                usage();
                }
        }

        if (uflag != 0)
                list_database(infile);
        else
                create_database(infile, outfile,
                    yp_input_file, yp_output_file, yp_master_name,
                    yp_domain_name, bflag, lflag, sflag);

        exit(0);
}

static int
add_record(DBM *db, const char *str1, const char *str2, int check)
{
        datum key, val;
        int status;

        key.dptr = __UNCONST(str1);
        key.dsize = strlen(str1);

        if (check) {
                val = ypdb_fetch(db, key);

                if (val.dptr != NULL)
                        return 0;       /* already there */
        }
        val.dptr = __UNCONST(str2);
        val.dsize = strlen(str2);
        status = ypdb_store(db, key, val, YPDB_INSERT);

        if (status != 0) {
                warnx("can't store `%s %s'", str1, str2);
                return -1;
        }
        return 0;
}

static char *
file_date(char *filename)
{
        struct stat finfo;
        static char datestr[11];

        memset(datestr, 0, sizeof(datestr));

        if (strcmp(filename, "-") == 0)
                snprintf(datestr, sizeof(datestr), "%010d",
                    (int)time(NULL));
        else {
                if (stat(filename, &finfo) != 0)
                        err(1, "can't stat %s", filename);
                snprintf(datestr, sizeof(datestr), "%010d",
                    (int)finfo.st_mtime);
        }

        return datestr;
}

static void
list_database(char *database)
{
        DBM *db;
        datum key, val;

        db = ypdb_open(database);
        if (db == NULL)
                err(1, "can't open database `%s'", database);

        key = ypdb_firstkey(db);

        while (key.dptr != NULL) {
                val = ypdb_fetch(db, key);
                                /* workround trailing \0 in aliases.db */
                if (key.dptr[key.dsize - 1] == '\0')
                        key.dsize--;
                printf("%*.*s", key.dsize, key.dsize, key.dptr);
                if (val.dsize > 0) {
                        if (val.dptr[val.dsize - 1] == '\0')
                                val.dsize--;
                        printf(" %*.*s", val.dsize, val.dsize, val.dptr);
                }
                printf("\n");
                key = ypdb_nextkey(db);
        }

        ypdb_close(db);
}

static void
create_database(char *infile, char *database, char *yp_input_file,
                char *yp_output_file, char *yp_master_name,
                char *yp_domain_name, int bflag, int lflag, int sflag)
{
        FILE *data_file;
        char myname[MAXHOSTNAMELEN];
        size_t line_no = 0;
        size_t len;
        char *p, *k, *v, *slash;
        DBM *new_db;
        static const char template[] = "ypdbXXXXXX";
        char db_mapname[MAXPATHLEN + 1], db_outfile[MAXPATHLEN + 1];
        char empty_str[] = "";

        memset(db_mapname, 0, sizeof(db_mapname));
        memset(db_outfile, 0, sizeof(db_outfile));

        if (strcmp(infile, "-") == 0)
                data_file = stdin;
        else {
                data_file = fopen(infile, "r");
                if (data_file == NULL)
                        err(1, "can't open `%s'", infile);
        }

        if (strlen(database) + strlen(YPDB_SUFFIX) > (sizeof(db_outfile) - 1))
                errx(1, "file name `%s' too long", database);

        snprintf(db_outfile, sizeof(db_outfile), "%s%s", database, YPDB_SUFFIX);

        slash = strrchr(database, '/');
        if (slash != NULL)
                slash[1] = '\0';        /* truncate to dir */
        else
                *database = '\0';       /* eliminate */

        /* NOTE: database is now directory where map goes ! */

        if (strlen(database) + strlen(template) + strlen(YPDB_SUFFIX) >
            (sizeof(db_mapname) - 1))
                errx(1, "directory name `%s' too long", database);

        snprintf(db_mapname, sizeof(db_mapname), "%s%s",
            database, template);

        new_db = ypdb_mktemp(db_mapname);
        if (new_db == NULL)
                err(1, "can't create temp database `%s'", db_mapname);

        for (;
            (p = fparseln(data_file, &len, &line_no, NULL, FPARSELN_UNESCALL));
            free(p)) {
                k = p;                          /* set start of key */
                while (*k && isspace((unsigned char)*k)) /* skip leading whitespace */
                        k++;

                if (! *k)
                        continue;

                v = k;
                while (*v && !isspace((unsigned char)*v)) {     /* find leading whitespace */
                                /* convert key to lower case if forcing. */
                        if (lflag && isupper((unsigned char)*v))
                                *v = tolower((unsigned char)*v);
                        v++;
                }
                while (*v && isspace((unsigned char)*v))        /* replace space with <NUL> */
                        *v++ = '\0';

                if (add_record(new_db, k, v, TRUE)) {    /* save record */
bad_record:
                        ypdb_close(new_db);
                        unlink(db_mapname);
                        errx(1, "error adding record for line %lu",
                            (unsigned long)line_no);
                }
        }

        if (strcmp(infile, "-") != 0)
                (void) fclose(data_file);

        if (add_record(new_db, YP_LAST_KEY, file_date(infile), FALSE))
                goto bad_record;

        if (yp_input_file) {
                if (add_record(new_db, YP_INPUT_KEY, yp_input_file, FALSE))
                        goto bad_record;
        }

        if (yp_output_file) {
                if (add_record(new_db, YP_OUTPUT_KEY, yp_output_file, FALSE))
                        goto bad_record;
        }

        if (yp_master_name) {
                if (add_record(new_db, YP_MASTER_KEY, yp_master_name, FALSE))
                        goto bad_record;
        } else {
                localhostname(myname, sizeof(myname) - 1);
                if (add_record(new_db, YP_MASTER_KEY, myname, FALSE))
                        goto bad_record;
        }

        if (yp_domain_name) {
                if (add_record(new_db, YP_DOMAIN_KEY, yp_domain_name, FALSE))
                        goto bad_record;
        }

        if (bflag) {
                if (add_record(new_db, YP_INTERDOMAIN_KEY, empty_str, FALSE))
                        goto bad_record;
        }

        if (sflag) {
                if (add_record(new_db, YP_SECURE_KEY, empty_str, FALSE))
                        goto bad_record;
        }

        ypdb_close(new_db);
        if (rename(db_mapname, db_outfile) < 0) {
                unlink(db_mapname);
                err(1, "rename `%s' -> `%s'", db_mapname, db_outfile);
        }
}

static void
usage(void)
{

        fprintf(stderr, "usage: %s -u file\n", getprogname());
        fprintf(stderr, "       %s [-lbs] %s\n", getprogname(),
            "[-i YP_INPUT_FILE] [-o YP_OUTPUT_FILE]");
        fprintf(stderr, "          %s infile outfile\n",
            "[-d YP_DOMAIN_NAME] [-m YP_MASTER_NAME]");
        exit(1);
}