/*	$NetBSD: makedbm.c,v 1.10 1997/11/13 18:35:57 thorpej 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.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by Mats O Jansson
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * 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.10 1997/11/13 18:35:57 thorpej Exp $");
#endif

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

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

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

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

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

int	main __P((int, char *[]));
void	usage __P((void));
int	add_record __P((DBM *, char *, char *, int));
char	*file_date __P((char *));
void	list_database __P((char *));
void	create_database __P((char *, char *, char *, char *,
	    char *, char *, int, int, int));

int
main(argc, argv)
	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);
}

int
add_record(db, str1, str2, check)
	DBM *db;
	char *str1, *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 0;	/* already there */
	}
	val.dptr = 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;
}

char *
file_date(filename)
	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;
}

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

	db = ypdb_open(database, O_RDONLY, 0444);
	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);
}

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

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

	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) > MAXPATHLEN)
		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(mapname) +
	    strlen(YPDB_SUFFIX) > MAXPATHLEN)
		errx(1, "directory name `%s' too long", database);

	snprintf(db_tempname, sizeof(db_tempname), "%s%s",
	    database, mapname);
	mktemp(db_tempname);
	snprintf(db_mapname, sizeof(db_mapname), "%s%s",
	    db_tempname, YPDB_SUFFIX);

	new_db = ypdb_open(db_tempname, O_RDWR | O_CREAT | O_EXCL, 0644);
	if (new_db == NULL)
		err(1, "can't create temp database `%s'", db_tempname);

	while ((p = read_line(data_file, &len, &line_no)) != NULL) {
		while (*p && isspace(*p))	/* skip leading whitespace */
			p++;

		if (! *p)
			continue;

		k = p;				/* save start of key */
		while (*p && !isspace(*p)) {	/* find whitespace */
				/* Convert to lower case if forcing. */
			if (lflag && isupper(*p))
				*p = tolower(*p);
			p++;
		}
		if (! *p)
			v = "";			/* no value found - use "" */
		else {
			while (isspace(*p))	/* replace space with <NUL> */
				*p++ = '\0';
			v = p;			/* save start of value */
		}

		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 %d", 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);
	}
}

void
usage()
{

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