/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * tnctl.c -
 *          Trusted Network control utility
 */
#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <locale.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/param.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <libtsnet.h>
#include <zone.h>
#include <nss_dbdefs.h>

static void process_rh(const char *);
static void process_rhl(const char *);
static void process_mlp(const char *);
static void process_tp(const char *);
static void process_tpl(const char *);
static void process_tnzone(const char *);
static void usage(void);
static void translate_inet_addr(tsol_rhent_t *, int *, char [], int);

static boolean_t verbose_mode;
static boolean_t delete_mode;
static boolean_t flush_mode;

int
main(int argc, char **argv)
{
	extern char *optarg;
	int chr;

	/* Don't do anything if labeling is not active. */
	if (!is_system_labeled())
		return (0);

	/* set the locale for only the messages system (all else is clean) */
	(void) setlocale(LC_ALL, "");
#ifndef TEXT_DOMAIN		/* Should be defined by cc -D */
#define	TEXT_DOMAIN	"SYS_TEST"	/* Use this only if it weren't */
#endif

	(void) textdomain(TEXT_DOMAIN);

	while ((chr = getopt(argc, argv, "dfh:H:m:t:T:vz:")) != EOF) {
		switch (chr) {
		case 'd':
			delete_mode = B_TRUE;
			break;
		case 'f':
			flush_mode = B_TRUE;
			break;
		case 'h':
			process_rh(optarg);
			break;
		case 'H':
			process_rhl(optarg);
			break;
		case 'm':
			process_mlp(optarg);
			break;
		case 't':
			process_tp(optarg);
			break;
		case 'T':
			process_tpl(optarg);
			break;
		case 'v':
			verbose_mode = B_TRUE;
			break;
		case 'z':
			process_tnzone(optarg);
			break;
		case '?':
			usage();
		}
	}
	return (0);
}

static void
print_error(int linenum, int err, const char *errstr)
{
	if (linenum > 0)
		(void) fprintf(stderr, gettext("line %1$d: %2$s:\n"), linenum,
		    tsol_strerror(err, errno));
	else
		(void) fprintf(stderr, gettext("tnctl: parsing error: %s\n"),
		    tsol_strerror(err, errno));
	(void) fprintf(stderr, "%.32s\n", errstr);
}

/*
 * Produce ascii format of address and prefix length
 */
static void
translate_inet_addr(tsol_rhent_t *rhentp, int *alen, char abuf[], int abuflen)
{
	void *aptr;
	tsol_rhent_t rhent;
	struct in6_addr ipv6addr;
	char tmpbuf[20];

	(void) snprintf(tmpbuf, sizeof (tmpbuf), "/%d", rhentp->rh_prefix);

	if (rhentp->rh_address.ta_family == AF_INET6) {
		aptr = &(rhentp->rh_address.ta_addr_v6);
		(void) inet_ntop(rhentp->rh_address.ta_family, aptr, abuf,
		    abuflen);
		if (rhentp->rh_prefix != 128) {
			if (strlcat(abuf, tmpbuf, abuflen) >= abuflen)
				(void) fprintf(stderr, gettext(
				    "tnctl: buffer overflow detected: %s\n"),
				    abuf);
		}
		*alen = strlen(abuf);
	} else {
		aptr = &(rhentp->rh_address.ta_addr_v4);
		(void) inet_ntop(rhentp->rh_address.ta_family, aptr, abuf,
		    abuflen);
		if (rhentp->rh_prefix != 32) {
			if (strlcat(abuf, tmpbuf, abuflen) >= abuflen)
				(void) fprintf(stderr, gettext(
				    "tnctl: buffer overflow detected: %s\n"),
				    abuf);
		}
		*alen = strlen(abuf);
	}
}

/*
 * Load remote host entries from the designated file.
 */
static void
process_rhl(const char *file)
{
	boolean_t	error = B_FALSE;
	boolean_t	success = B_FALSE;
	tsol_rhent_t	*rhentp = NULL;
	FILE		*fp;
	int alen;
	/* abuf holds: <numeric-ip-addr>'/'<prefix-length>'\0' */
	char abuf[INET6_ADDRSTRLEN+5];

	if ((fp = fopen(file, "r")) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: failed to open %1$s: %2$s\n"),
		    file, strerror(errno));
		exit(1);
	}

	tsol_setrhent(1);
	while (rhentp = tsol_fgetrhent(fp, &error)) {
		/* First time through the loop, flush it all */
		if (!success && flush_mode)
			(void) tnrh(TNDB_FLUSH, NULL);
		success = B_TRUE;

		if (verbose_mode)
			(void) printf("loading rh entry...\n");

		if (tnrh(TNDB_LOAD, rhentp) != 0) {
			(void) fclose(fp);
			if (errno == EFAULT) {
				perror("tnrh");
			} else {
				translate_inet_addr(rhentp, &alen, abuf,
				    sizeof (abuf));
				(void) fprintf(stderr,
				    gettext("tnctl: load of remote-host entry "
				    "%1$s into kernel cache failed: %2$s\n"),
				    abuf, strerror(errno));
			}
			tsol_endrhent();
			exit(1);
		}
		tsol_freerhent(rhentp);
	}
	if (!success) {
		(void) fprintf(stderr,
		    gettext("tnctl: No valid tnrhdb entries found in %s\n"),
		    file);
	}
	(void) fclose(fp);
	tsol_endrhent();

	if (error)
		exit(1);
}

/*
 * The argument can be either a host name, an address
 * in tnrhdb address format, or a complete tnrhdb entry.
 */
static void
process_rh(const char *hostname)
{
	tsol_rhstr_t rhstr;
	tsol_rhent_t rhent;
	tsol_rhent_t *rhentp;
	int err;
	int alen;
	char *errstr;
	/* abuf holds: <numeric-ip-addr>'/'<prefix-length>'\0' */
	char abuf[INET6_ADDRSTRLEN+5];
	const char *cp;
	char *cp1;
	char *cp2;
	void *aptr;
	char buf[NSS_BUFLEN_TSOL_RH];
	struct in6_addr ipv6addr;

	/* was a template name provided on the command line? */
	if ((cp = strrchr(hostname, ':')) != NULL && cp != hostname &&
	    cp[-1] != '\\') {
		/* use common tnrhdb line conversion function */
		(void) str_to_rhstr(hostname, strlen(hostname), &rhstr, buf,
		    sizeof (buf));
		rhentp = rhstr_to_ent(&rhstr, &err, &errstr);
		if (rhentp == NULL) {
			print_error(0, err, errstr);
			exit(1);
		}
	} else {
		char *hostname_p;
		char *prefix_p;
		struct hostent *hp;

		/* Check for a subnet prefix length */
		if ((prefix_p = strchr(hostname, '/')) != NULL) {
			cp1 = prefix_p + 1;
			errno = 0;
			rhent.rh_prefix = strtol(cp1, &cp2, 0);
			if (*cp2 != '\0' || errno != 0 || rhent.rh_prefix < 0) {
				(void) fprintf(stderr, gettext("tnct: invalid "
				    "prefix length: %s\n"), cp);
				exit(2);
			}
		} else {
			rhent.rh_prefix = -1;
		}

		/* Strip any backslashes from numeric address */
		hostname_p = malloc(strlen(hostname)+1);
		if (hostname_p == NULL) {
			perror("tnctl");
			exit(2);
		}
		cp1 = hostname_p;
		while (*hostname != '\0' && *hostname != '/') {
			*cp1 = *hostname++;
			if (*cp1 != '\\')
				cp1++;
		}
		*cp1 = '\0';

		/* Convert address or hostname to binary af_inet6 format */
		hp = getipnodebyname(hostname_p, AF_INET6,
		    AI_ALL | AI_ADDRCONFIG | AI_V4MAPPED, &err);
		if (hp == NULL) {
			(void) fprintf(stderr, gettext("tnctl: unknown host "
			    "or invalid literal address: %s\n"), hostname_p);
			if (err == TRY_AGAIN)
				(void) fprintf(stderr,
				    gettext("\t(try again later)\n"));
			exit(2);
		}
		free(hostname_p);
		(void) memcpy(&ipv6addr, hp->h_addr, hp->h_length);

		/* if ipv4 address, convert to af_inet format */
		if (IN6_IS_ADDR_V4MAPPED(&ipv6addr)) {
			rhent.rh_address.ta_family = AF_INET;
			IN6_V4MAPPED_TO_INADDR(&ipv6addr,
			    &rhent.rh_address.ta_addr_v4);
			if (rhent.rh_prefix == -1)
				rhent.rh_prefix = 32;
		} else {
			rhent.rh_address.ta_family = AF_INET6;
			rhent.rh_address.ta_addr_v6 = ipv6addr;
			if (rhent.rh_prefix == -1)
				rhent.rh_prefix = 128;
		}
		rhent.rh_template[0] = '\0';
		rhentp = &rhent;
	}

	/* produce ascii format of address and prefix length */
	translate_inet_addr(rhentp, &alen, abuf, sizeof (abuf));

	/*
	 * look up the entry from ldap or tnrhdb if this is a load
	 * request and a template name was not provided.
	 */
	if (!delete_mode &&
	    rhentp->rh_template[0] == '\0' &&
	    (rhentp = tsol_getrhbyaddr(abuf, alen+1,
	    rhent.rh_address.ta_family)) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: database lookup failed for %s\n"),
		    abuf);
		exit(1);
	}

	if (verbose_mode)
		(void) printf("%s rh entry %s\n", delete_mode ? "deleting" :
		    "loading", abuf);

	/* update the tnrhdb entry in the kernel */
	if (tnrh(delete_mode ? TNDB_DELETE : TNDB_LOAD, rhentp) != 0) {
		if (errno == EFAULT)
			perror("tnrh");
		else if (errno == ENOENT)
			(void) fprintf(stderr,
			    gettext("tnctl: %1$s of remote-host kernel cache "
			    "entry %2$s failed: no such entry\n"),
			    delete_mode ? gettext("delete") : gettext("load"),
			    abuf);
		else
			(void) fprintf(stderr,
			    gettext("tnctl: %1$s of remote-host kernel cache "
			    "entry %2$s failed: %3$s\n"),
			    delete_mode ? gettext("delete") : gettext("load"),
			    abuf, strerror(errno));
		exit(1);
	}
	if (rhentp != &rhent)
		tsol_freerhent(rhentp);
}

static void
handle_mlps(zoneid_t zoneid, tsol_mlp_t *mlp, int flags, int cmd)
{
	tsol_mlpent_t tsme;

	tsme.tsme_zoneid = zoneid;
	tsme.tsme_flags = flags;
	while (!TSOL_MLP_END(mlp)) {
		tsme.tsme_mlp = *mlp;
		if (tnmlp(cmd, &tsme) != 0) {
			/*
			 * Usage of ?: here is ugly, but helps with
			 * localization.
			 */
			(void) fprintf(stderr,
			    flags & TSOL_MEF_SHARED ?
			    gettext("tnctl: cannot set "
			    "shared MLP on %1$d-%2$d/%3$d: %4$s\n") :
			    gettext("tnctl: cannot set "
			    "zone-specific MLP on %1$d-%2$d/%3$d: %4$s\n"),
			    mlp->mlp_port, mlp->mlp_port_upper, mlp->mlp_ipp,
			    strerror(errno));
			exit(1);
		}
		mlp++;
	}
}

/*
 * This reads the configuration for the global zone out of tnzonecfg
 * and sets it in the kernel.  The non-global zones are configured
 * by zoneadmd.
 */
static void
process_tnzone(const char *file)
{
	tsol_zcent_t *zc;
	tsol_mlpent_t tsme;
	int err;
	char *errstr;
	FILE *fp;
	char line[2048], *cp;
	int linenum, errors;

	if ((fp = fopen(file, "r")) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: failed to open %s: %s\n"), file,
		    strerror(errno));
		exit(1);
	}

	linenum = errors = 0;
	zc = NULL;
	while (fgets(line, sizeof (line), fp) != NULL) {
		if ((cp = strchr(line, '\n')) != NULL)
			*cp = '\0';

		linenum++;
		if ((zc = tsol_sgetzcent(line, &err, &errstr)) == NULL) {
			if (err == LTSNET_EMPTY)
				continue;
			if (errors == 0) {
				int errtmp = errno;

				(void) fprintf(stderr, gettext("tnctl: errors "
				    "parsing %s:\n"), file);
				errno = errtmp;
			}
			print_error(linenum, err, errstr);
			errors++;
			continue;
		}

		if (strcasecmp(zc->zc_name, "global") == 0)
			break;
		tsol_freezcent(zc);
	}
	(void) fclose(fp);

	if (zc == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: cannot find global zone in %s\n"), file);
		exit(1);
	}

	tsme.tsme_zoneid = GLOBAL_ZONEID;
	tsme.tsme_flags = 0;
	if (flush_mode)
		(void) tnmlp(TNDB_FLUSH, &tsme);

	handle_mlps(GLOBAL_ZONEID, zc->zc_private_mlp, 0, TNDB_LOAD);
	handle_mlps(GLOBAL_ZONEID, zc->zc_shared_mlp, TSOL_MEF_SHARED,
	    TNDB_LOAD);

	tsol_freezcent(zc);
}

static void
process_tpl(const char *file)
{
	FILE		*fp;
	boolean_t	error = B_FALSE;
	boolean_t	success = B_FALSE;
	tsol_tpent_t	*tpentp;

	if ((fp = fopen(file, "r")) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: failed to open %s: %s\n"), file,
		    strerror(errno));
		exit(1);
	}

	tsol_settpent(1);
	while (tpentp = tsol_fgettpent(fp, &error)) {
		/* First time through the loop, flush it all */
		if (!success && flush_mode)
			(void) tnrhtp(TNDB_FLUSH, NULL);

		success = B_TRUE;

		if (verbose_mode)
			(void) printf("tnctl: loading rhtp entry ...\n");

		if (tnrhtp(TNDB_LOAD, tpentp) != 0) {
			(void) fclose(fp);
			if (errno == EFAULT)
				perror("tnrhtp");
			else
				(void) fprintf(stderr, gettext("tnctl: load "
				    "of remote-host template %1$s into kernel "
				    "cache failed: %2$s\n"), tpentp->name,
				    strerror(errno));
			tsol_endtpent();
			exit(1);
		}
		tsol_freetpent(tpentp);
	}
	if (!success) {
		(void) fprintf(stderr,
		    gettext("tnctl: No valid tnrhtp entries found in %s\n"),
		    file);
	}
	(void) fclose(fp);
	tsol_endtpent();

	if (error)
		exit(1);
}

static void
process_tp(const char *template)
{
	tsol_tpstr_t tpstr;
	tsol_tpent_t tpent;
	tsol_tpent_t *tpentp;
	int err;
	char *errstr;
	char buf[NSS_BUFLEN_TSOL_TP];

	if (strchr(template, ':') != NULL) {
		(void) str_to_tpstr(template, strlen(template), &tpstr, buf,
		    sizeof (buf));
		tpentp = tpstr_to_ent(&tpstr, &err, &errstr);
		if (tpentp == NULL) {
			print_error(0, err, errstr);
			exit(1);
		}
	} else if (delete_mode) {
		(void) memset(&tpent, 0, sizeof (tpent));
		tpentp = &tpent;
		(void) strlcpy(tpentp->name, template, sizeof (tpentp->name));
	} else if ((tpentp = tsol_gettpbyname(template)) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: template %s not found\n"), template);
		exit(1);
	}

	if (verbose_mode)
		(void) printf("%s rhtp entry ...\n", delete_mode ? "deleting" :
		    "loading");

	if (tnrhtp(delete_mode ? TNDB_DELETE : TNDB_LOAD, tpentp) != 0) {
		if (errno == EFAULT)
			perror("tnrhtp");
		else if (errno == ENOENT)
			(void) fprintf(stderr,
			    gettext("tnctl: %1$s of remote-host template "
			    "kernel cache entry %2$s failed: no such "
			    "entry\n"),
			    delete_mode ? gettext("delete") : gettext("load"),
			    tpentp->name);
		else
			(void) fprintf(stderr,
			    gettext("tnctl: %1$s of remote-host template "
			    "kernel cache entry %2$s failed: %3$s\n"),
			    delete_mode ? gettext("delete") : gettext("load"),
			    tpentp->name, strerror(errno));
		exit(1);
	}
	if (tpentp != &tpent)
		tsol_freetpent(tpentp);
}

static void
process_mlp(const char *str)
{
	const char *cp;
	char zonename[ZONENAME_MAX];
	zoneid_t zoneid;
	tsol_zcent_t *zc;
	int err;
	char *errstr;
	char *sbuf;

	if ((cp = strchr(str, ':')) == NULL) {
		if (!delete_mode) {
			(void) fprintf(stderr,
			    gettext("tnctl: need MLP list to insert\n"));
			exit(2);
		}
		(void) strlcpy(zonename, str, sizeof (zonename));
	} else if (cp - str >= ZONENAME_MAX) {
		(void) fprintf(stderr, gettext("tnctl: illegal zone name\n"));
		exit(2);
	} else {
		(void) memcpy(zonename, str, cp - str);
		zonename[cp - str] = '\0';
		str = cp + 1;
	}

	if ((zoneid = getzoneidbyname(zonename)) == -1) {
		(void) fprintf(stderr, gettext("tninfo: zone '%s' unknown\n"),
		    zonename);
		exit(1);
	}

	sbuf = malloc(strlen(zonename) + sizeof (":ADMIN_LOW:0:") +
	    strlen(str));
	if (sbuf == NULL) {
		perror("malloc");
		exit(1);
	}
	/* LINTED: sprintf is known not to be unbounded here */
	(void) sprintf(sbuf, "%s:ADMIN_LOW:0:%s", zonename, str);
	if ((zc = tsol_sgetzcent(sbuf, &err, &errstr)) == NULL) {
		(void) fprintf(stderr,
		    gettext("tnctl: unable to parse MLPs\n"));
		exit(1);
	}
	handle_mlps(zoneid, zc->zc_private_mlp, 0,
	    delete_mode ? TNDB_DELETE : TNDB_LOAD);
	handle_mlps(zoneid, zc->zc_shared_mlp, TSOL_MEF_SHARED,
	    delete_mode ? TNDB_DELETE : TNDB_LOAD);
	tsol_freezcent(zc);
}

static void
usage(void)
{
	(void) fprintf(stderr, gettext("usage: tnctl [-dfv] "
	    "[-h host[/prefix][:tmpl]] [-m zone:priv:share]\n\t"
	    "[-t tmpl[:key=val[;key=val]]] [-[HTz] file]\n"));

	exit(1);
}