/*      $OpenBSD: makedbm.c,v 1.33 2015/02/09 23:00:15 deraadt Exp $ */

/*
 * Copyright (c) 1994-97 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 <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <ctype.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include "ypdb.h"
#include "ypdef.h"
#include "db.h"

extern char *__progname;                /* from crt0.o */

/*
 * Read one line
 */

static int
read_line(FILE *fp, char *buf, int size)
{
        int     done = 0;

        do {
                while (fgets(buf, size, fp)) {
                        int len = strlen(buf);

                        done += len;
                        if (len > 1 && buf[len-2] == '\\' &&
                            buf[len-1] == '\n') {
                                int ch;

                                buf += len - 2;
                                size -= len - 2;
                                *buf = '\n'; buf[1] = '\0';

                                /* Skip leading white space on next line */
                                while ((ch = getc(fp)) != EOF &&
                                    isascii(ch) && isspace(ch))
                                        ;
                                (void) ungetc(ch, fp);
                        } else {
                                return done;
                        }
                }
        } while (size > 0 && !feof(fp));
        return done;
}

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

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

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

                if (val.dptr != NULL)
                        return;         /* already there */
        }

        val.dptr = str2;
        val.dsize = strlen(str2);
        status = ypdb_store(db, key, val, YPDB_INSERT);

        if (status != 0) {
                printf("%s: problem storing %s %s\n",
                    __progname, str1, str2);
                exit(1);
        }
}

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

        if (strcmp(filename,"-") == 0) {
                snprintf(datestr, sizeof datestr, "%010lld",
                    (long long)time(0));
        } else {
                status = stat(filename, &finfo);
                if (status < 0) {
                        fprintf(stderr, "%s: can't stat %s\n", __progname, filename);
                        exit(1);
                }
                snprintf(datestr, sizeof datestr, "%010lld",
                    (long long)finfo.st_mtime);
        }
        return datestr;
}

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

        db = ypdb_open(database, O_RDONLY, 0444);

        if (db == NULL) {
                if (Uflag != 0)
                        if (db_hash_list_database(database))
                                return;

                fprintf(stderr, "%s: can't open database %s: %s\n", __progname,
                    database, strerror(errno));
                exit(1);
        }

        key = ypdb_firstkey(db);

        while (key.dptr != NULL) {
                val = ypdb_fetch(db, key);
                printf("%*.*s %*.*s\n",
                    key.dsize, key.dsize, key.dptr,
                    val.dsize, val.dsize, val.dptr);
                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    data_line[4096]; /* XXX: DB bsize = 4096 in ypdb.c */
        char    myname[HOST_NAME_MAX+1];
        int     line_no = 0, len;
        char    *p, *k, *v, *slash;
        DBM     *new_db;
        static  char mapname[] = "ypdbXXXXXXXXXX";
        char    db_mapname[PATH_MAX], db_outfile[PATH_MAX];
        char    db_tempname[PATH_MAX], empty_str[] = "";

        if (strcmp(infile,"-") == 0) {
                data_file = stdin;
        } else {
                data_file = fopen(infile, "r");
                if (errno != 0) {
                        (void)fprintf(stderr,"%s: ", __progname);
                        perror(infile);
                        exit(1);
                }
        }

        if (strlen(database) + strlen(YPDB_SUFFIX) > PATH_MAX) {
                fprintf(stderr,"%s: %s: file name too long\n",
                    __progname, database);
                exit(1);
        }
        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(mapname) +
            strlen(YPDB_SUFFIX) > PATH_MAX) {
                fprintf(stderr,"%s: %s: directory name too long\n",
                    __progname, database);
                exit(1);
        }

        snprintf(db_mapname, sizeof(db_mapname), "%s%s%s",
            database, mapname, YPDB_SUFFIX);
        mkstemps(db_mapname, sizeof(YPDB_SUFFIX));
        strlcpy(db_tempname, db_mapname, sizeof db_tempname);
        db_tempname[strlen(db_tempname) - sizeof(YPDB_SUFFIX) + 1] = '\0';

        new_db = ypdb_open(db_tempname, O_RDWR|O_CREAT, 0444);
        if (new_db == NULL) {
                fprintf(stderr, "%s: Unable to open output database %s\n",
                    __progname, db_outfile);
                exit(1);
        }

        while (read_line(data_file, data_line, sizeof(data_line))) {
                line_no++;
                len =  strlen(data_line);

                /* Check if we have the whole line */

                if (data_line[len-1] != '\n') {
                        fprintf(stderr, "line %d in \"%s\" is too long",
                            line_no, infile);
                } else {
                        data_line[len-1] = '\0';
                }

                p = (char *) &data_line;

                k = p;                                   /* save start of key */
                while (!isspace((unsigned char)*p)) {           /* find first "space" */
                        if (lflag && isupper((unsigned char)*p))        /* if force lower case */
                                *p = (char)tolower((unsigned char)*p);   /* fix it */
                        p++;
                }
                while (isspace((unsigned char)*p)) {            /* replace space with <NUL> */
                        *p = '\0';
                        p++;
                }

                v = p;                          /* save start of value */
                while (*p != '\0')              /* find end of string */
                        p++;

                add_record(new_db, k, v, TRUE); /* save record */

        }

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

        add_record(new_db, YP_LAST_KEY, file_date(infile), FALSE);

        if (yp_input_file)
                add_record(new_db, YP_INPUT_KEY, yp_input_file, FALSE);

        if (yp_output_file)
                add_record(new_db, YP_OUTPUT_KEY, yp_output_file, FALSE);

        if (yp_master_name)
                add_record(new_db, YP_MASTER_KEY, yp_master_name, FALSE);
        else {
                gethostname(myname, sizeof(myname));
                add_record(new_db, YP_MASTER_KEY, myname, FALSE);
        }

        if (yp_domain_name)
                add_record(new_db, YP_DOMAIN_KEY, yp_domain_name, FALSE);
        if (bflag)
                add_record(new_db, YP_INTERDOMAIN_KEY, empty_str, FALSE);
        if (sflag)
                add_record(new_db, YP_SECURE_KEY, empty_str, FALSE);

        ypdb_close(new_db);
        if (rename(db_mapname, db_outfile) < 0) {
                perror("rename");
                fprintf(stderr,"rename %s -> %s failed!\n", db_mapname,
                    db_outfile);
                exit(1);
        }

}

static void
usage(void)
{
        fprintf(stderr,"usage: makedbm [-blsUu] [-d yp_domain_name] "
            "[-i yp_input_file]\n"
            "\t[-m yp_master_name] [-o yp_output_file] infile outfile\n");
        exit(1);
}

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

        extern int optind;
        extern char *optarg;

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

        while ((ch = getopt(argc, argv, "Ublsui:o:m:d:")) != -1)
                switch (ch) {
                case 'U':
                        uflag = 1;
                        Uflag = 1;
                        break;
                case 'b':
                        bflag = 1;
                        aflag = 1;
                        break;
                case 'l':
                        lflag = 1;
                        aflag = 1;
                        break;
                case 's':
                        sflag = 1;
                        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();
                        break;
                }

        if ((uflag != 0) && (aflag != 0))
                usage();

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

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

        return(0);
}