/*-
 * Copyright (c) 1992 The Regents of the University of California.
 * All rights reserved.
 *
 * %sccs.include.redist.c%
 */

#ifndef lint
static char copyright[] =
"@(#) Copyright (c) 1992 The Regents of the University of California.\n\
 All rights reserved.\n";
#endif /* not lint */

#ifndef lint
static char sccsid[] = "@(#)cap_mkdb.c	5.1 (Berkeley) 10/17/92";
#endif /* not lint */

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

#include <db.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

static void	 db_build __P((char **));
static void	 err __P((int, const char *, ...));
static void	 getnamefield __P((char **, char *));
static void	 usage __P((void));

int docapdbunlink, printnl;
char *capdb, **inputfiles;

/*
 * Mkcapdb creates a capability hash database for quick retrieval of capability
 * records.  The database contains 2 types of entries: records and references
 * marked by the first byte in the data.  A record entry contains the actual
 * capability record whereas a reference contains the name (key) under which
 * the correct record is stored.
 */
int
main(argc, argv)
	int argc;
	char *argv[];
{
	int c, fd;
	char *outname, buf[MAXPATHLEN + 1], **f;

	outname = NULL;
	while ((c = getopt(argc, argv, "f:")) != EOF) {
		switch(c) {
		case 'f':
			outname = optarg;
			break;
		case '?':
		default:
			usage();
		}
	}
	argc -= optind;
	argv += optind;

	if (*argv == NULL)
		usage();

	inputfiles = argv;

	if (outname == NULL)
		outname = *inputfiles;

#define CAPDBNAMEEXTLEN		3	/* ".db" */
	if ((capdb = malloc(strlen(outname) + CAPDBNAMEEXTLEN + 1)) == NULL)
		err(1, "%s", strerror(errno));
	(void)sprintf(capdb, "%s.db", outname);

	/*
	 * We want to avoid the confusion of where the capability record
	 * is being read from.  Since the user probably intends to read the
	 * ascii file, we should make sure that user knows that the
	 * corresponding .db file will override.
	 */
	for (f = inputfiles; *f != NULL; f++) {
		(void)sprintf(buf, "%s.db", *f);
		fd = open(buf, O_RDONLY, 0444);
		if (fd == -1 && errno != ENOENT)
			err(1, "%s: %s", buf, strerror(errno));
		if (fd >= 0) {
			err(0, "Warning -- %s.db will override %s.", *f, *f);
			(void)close(fd);
		}
	}

	db_build(inputfiles);
	exit(0);
}

/*
 * Any changes to these definitions should be made also in the getcap(3)
 * library routines.
 */

#define RECOK	(char)0
#define TCERR	(char)1

#define NBUFSIZ		(8 * 1024)

/*
 * Db_build() builds the name and capabilty databases according to the
 * details above.
 */
void
db_build(inputfiles)
	char **inputfiles;
{
	DB *capdbp;
	DBT key, data;
	recno_t reccnt;
	size_t lastlen, bplen;
	int st, stdb;
	char *cp, *np, *bp, *nf, namebuf[NBUFSIZ];

	if ((capdbp = dbopen(capdb, O_CREAT | O_TRUNC | O_RDWR,
	    DEFFILEMODE, DB_HASH, NULL)) == NULL)
		err(1, "%s: %s", capdb, strerror(errno));
	docapdbunlink = 1;

	lastlen = 0;
	nf = NULL;
	data.data = NULL;
	key.data = NULL;
	for (reccnt = 0; (st = cgetnext(&bp, inputfiles)) > 0;) {
		getnamefield(&nf, bp);
		if ((bplen = strlen(bp)) > lastlen) {
			if ((data.data = realloc(data.data, bplen + 2)) == NULL)
				err(1, "%s", strerror(errno));
			lastlen = bplen;
		}

		/* Store record under name field. */
		if (st == 2)
			((char *)(data.data))[0] = TCERR;
		else
			((char *)(data.data))[0] = RECOK;

		(void)strcpy(&((char *)(data.data))[1], bp);
		data.size = bplen + 2;
		key.data = nf;
		key.size = strlen(nf) + 1;
		if ((stdb =
		    capdbp->put(capdbp, &key, &data, R_NOOVERWRITE)) < 0)
			err(1, "put: %s", strerror(errno));
		if (stdb == 1) {
			err(0, "ignored duplicate: %s", nf);
			continue;
		}
		++reccnt;

		/* Store references for other names. */
		(void)strcpy((char *)(data.data), nf);

		data.size = key.size;
		key.data = namebuf;
		np = namebuf;
		for (cp = nf; *cp != '\0'; *np++ = *cp++)
			if (*cp == ':' || *cp == '|') {
				*np = '\0';
				key.size = strlen(namebuf) + 1;
				if ((stdb = capdbp->put(capdbp, &key, &data,
				    R_NOOVERWRITE)) < 0)
					err(1, "put: %s", strerror(errno));
				if (stdb == 1)
					err(0,
					    "ignored duplicate: %s", namebuf);
				np = namebuf;
				continue;
			}
	}
	if (capdbp->close(capdbp) < 0)
		err(1, "%s", strerror(errno));

	if (st == -1)
		err(1, "%s", strerror(errno));
	if (st == -2)
		err(1, "potential reference loop detected");

	free(data.data);
	free(nf);
	free(bp);

	(void)printf("cap_mkdb: %d capability records\n", reccnt);
}

void
getnamefield(nf, bp)
	char **nf, *bp;
{
	static size_t nfsize;
	size_t newsize;
	char *cp, tmp;

	for (cp = bp; *cp != ':'; cp++);

	tmp = *(cp + 1);
	*(cp + 1) = '\0';

	if ((newsize = cp - bp + 1) > nfsize) {
		if ((*nf = realloc(*nf, newsize)) == NULL)
			err(1, "%s", strerror(errno));
		nfsize = newsize;
	}
	(void)strcpy(*nf, bp);
	*(cp + 1) = tmp;
}

void
usage()
{
	(void)fprintf(stderr,
	    "usage: cap_mkdb [-f outfile] file1 [file2 ...]\n");
	exit(1);
}

#if __STDC__
#include <stdarg.h>
#else
#include <varargs.h>
#endif

void
#if __STDC__
err(int fatal, const char *fmt, ...)
#else
err(fmt, va_alist)
	char *fmt;
	va_dcl
#endif
{
	va_list ap;
#if __STDC__
	va_start(ap, fmt);
#else
	va_start(ap);
#endif

	if (printnl)
		(void)fprintf(stderr, "\n");
	(void)fprintf(stderr, "cap_mkdb: ");
	(void)vfprintf(stderr, fmt, ap);
	va_end(ap);
	(void)fprintf(stderr, "\n");
	if (fatal) {
		if (docapdbunlink)
			(void)unlink(capdb);
		exit(1);
	}
}