/*
 * 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.
 */

/*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
/*	  All Rights Reserved  	*/


/*
 * autopush(1) is the command interface to the STREAMS autopush
 * mechanism.  The autopush command can be used to configure autopush
 * information about a STREAMS driver, remove autopush information,
 * and report on current configuration information.  Its use is as
 * follows:
 *
 *	autopush -f file
 *	autopush -r -M major -m minor
 *	autopush -g -M major -m minor
 *
 * The -f option allows autopush information to be set from a file.  The
 * format of the file is as follows:
 *
 * # Comment lines begin with a # in column one.
 * # The fields are separated by white space and are:
 * # major	minor	lastminor	module1 module2 ... module8
 *
 * "lastminor" is used to configure ranges of minor devices, from "minor"
 * to "lastminor" inclusive.  It should be set to zero when not in use.
 * The -r option allows autopush information to be removed for the given
 * major/minor pair.  The -g option allows the configuration information
 * to be printed.  The format of printing is the same as for the file.
 */

/*
 * Use autopush version 1; keep before #include <sys/sad.h>.
 * See <sys/sad.h> for details.
 */
#define	AP_VERSION	1

#include <sys/types.h>
#include <sys/conf.h>
#include <sys/modctl.h>
#include <sys/sad.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <ctype.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <locale.h>
#include <sys/stat.h>
#include <zone.h>

#define	OPTIONS	"M:f:gm:r"	/* command line options for getopt(3C) */
#define	COMMENT	'#'
#define	MINUS	'-'
#define	SLASH	'/'

/*
 * Output format.
 */
#define	OHEADER		"     Major      Minor  Lastminor\tModules\n"
#define	OFORMAT1_ONE	"%10ld %10ld      -    \t"
#define	OFORMAT1_RANGE	"%10ld %10ld %10ld\t"
#define	OFORMAT1_ALL	"%10ld       ALL       -    \t"

#define	AP_ANCHOR	"[anchor]"

#define	Openerr		gettext("%s: ERROR: Could not open %s: ")
#define	Digiterr	gettext("%s: ERROR: argument to %s option must be " \
			    "numeric\n")
#define	Badline		gettext("%s: WARNING: File %s: bad input line %d " \
			    "ignored\n")

static void	usage();
static int	rem_info(), get_info(), set_info();
static int	is_white_space(), parse_line();

static char	*Cmdp;		/* command name */

/*
 * main():
 *	process command line arguments.
 */
int
main(int argc, char *argv[])
{
	int		c;		/* character read by getopt(3C) */
	char		*filenamep;	/* name of configuration file */
	major_t		major;		/* major device number */
	minor_t		minor;		/* minor device number */
	char		*cp;
	int		exitcode;
	ushort_t	minflag = 0;	/* -m option used */
	ushort_t	majflag = 0;	/* -M option used */
	ushort_t	fflag = 0;	/* -f option used */
	ushort_t	rflag = 0;	/* -r option used */
	ushort_t	gflag = 0;	/* -g option used */
	ushort_t	errflag = 0;	/* options usage error */

	(void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN)
#define	TEXT_DOMAIN "SYS_TEST"
#endif
	(void) textdomain(TEXT_DOMAIN);

	/*
	 * Get command name.
	 */
	Cmdp = argv[0];
	for (filenamep = argv[0]; *filenamep; filenamep++)
		if (*filenamep == SLASH)
			Cmdp = filenamep + 1;

	/*
	 * Get options.
	 */
	while (!errflag && ((c = getopt(argc, argv, OPTIONS)) != -1)) {
		switch (c) {
		case 'M':
			if (fflag|majflag)
				errflag++;
			else {
				majflag++;
				for (cp = optarg; *cp; cp++)
					if (!isdigit(*cp)) {
						(void) fprintf(stderr,
						    Digiterr, Cmdp, "-M");
						exit(1);
					}
				major = (major_t)atol(optarg);
			}
			break;

		case 'm':
			if (fflag|minflag)
				errflag++;
			else {
				minflag++;
				for (cp = optarg; *cp; cp++)
					if (!isdigit(*cp)) {
						(void) fprintf(stderr,
						    Digiterr, Cmdp, "-m");
						exit(1);
					}
				minor = (minor_t)atol(optarg);
			}
			break;

		case 'f':
			if (fflag|gflag|rflag|majflag|minflag)
				errflag++;
			else {
				fflag++;
				filenamep = optarg;
			}
			break;

		case 'r':
			if (fflag|gflag|rflag)
				errflag++;
			else
				rflag++;
			break;

		case 'g':
			if (fflag|gflag|rflag)
				errflag++;
			else
				gflag++;
			break;

		default:
			errflag++;
			break;
		} /* switch */
		if (errflag) {
			usage();
			exit(1);
		}
	} /* while */
	if (((gflag || rflag) && (!majflag || !minflag)) || (optind != argc)) {
		usage();
		exit(1);
	}

	if (getzoneid() != GLOBAL_ZONEID) {
		(void) fprintf(stderr, gettext("autopush "
		    "can only be run from the global zone.\n"));
		exit(1);
	}

	if (fflag)
		exitcode = set_info(filenamep);
	else if (rflag)
		exitcode = rem_info(major, minor);
	else if (gflag)
		exitcode = get_info(major, minor);
	else {
		usage();
		exit(1);
	}

	return (exitcode);
}

/*
 * usage():
 *	print out usage statement.
 */
static void
usage()
{
	(void) fprintf(stderr,	gettext("%s: USAGE:\n\t%s -f filename\n"
	    "\t%s -r -M major -m minor\n"
	    "\t%s -g -M major -m minor\n"), Cmdp, Cmdp, Cmdp, Cmdp);
}

/*
 * set_info():
 *	set autopush configuration information.
 *	namep: autopush configuration filename
 */
static int
set_info(char *namep)
{
	int		line;		/* line number of file */
	FILE		*fp;		/* file pointer of config file */
	char		buf[256];	/* input buffer */
	struct strapush push;		/* configuration information */
	int		sadfd;		/* file descriptor to SAD driver */
	int		retcode = 0;	/* return code */
	int		parsecode;	/* return value from parse function */

	if ((sadfd = open(ADMINDEV, O_RDWR)) < 0) {
		(void) fprintf(stderr, Openerr, Cmdp, ADMINDEV);
		perror("");
		return (1);
	}
	if ((fp = fopen(namep, "r")) == NULL) {
		(void) fprintf(stderr, Openerr, Cmdp, namep);
		perror("");
		return (1);
	}
	line = 0;
	while (fgets(buf, sizeof (buf), fp) != NULL) {
		line++;
		if ((buf[0] == COMMENT) || is_white_space(buf))
			continue;
		(void) memset(&push, 0, sizeof (struct strapush));

		parsecode = parse_line(buf, line, namep, &push);
		if (parsecode != 0) {
			retcode = parsecode;
			continue;
		}

		if (push.sap_minor == (minor_t)-1)
			push.sap_cmd = SAP_ALL;
		else if (push.sap_lastminor == 0)
			push.sap_cmd = SAP_ONE;
		else
			push.sap_cmd = SAP_RANGE;

		if (ioctl(sadfd, SAD_SAP, &push) < 0) {
			int error = errno;

			retcode = 1;
			(void) fprintf(stderr,
			    gettext("%s: ERROR: File %s: could not configure "
			    "autopush for line %d\n"), Cmdp, namep, line);
			switch (error) {
			case EPERM:
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "You don't have permission to set autopush "
				    "information\n"), Cmdp);
				break;

			case EINVAL:
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "Invalid major device number or invalid "
				    "module name or too many modules\n"), Cmdp);
				break;

			case ENOSTR:
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "Major device is not a STREAMS "
				    "driver\n"), Cmdp);
				break;

			case EEXIST:
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "Major/minor already configured\n"), Cmdp);
				break;

			case ENOSR:
				(void) fprintf(stderr, gettext("%s: ERROR: Ran "
				    "out of autopush structures\n"), Cmdp);
				break;

			case ERANGE:
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "lastminor must be greater than minor\n"),
				    Cmdp);
				break;

			default:
				(void) fprintf(stderr, gettext("%s: ERROR: "),
				    Cmdp);
				(void) fprintf(stderr, "%s\n", strerror(error));
				break;
			} /* switch */
		} /* if */
	} /* while */
	return (retcode);
}

/*
 * rem_info():
 *	remove autopush configuration information.
 */
static int
rem_info(major_t maj, minor_t min)
{
	struct strapush push;		/* configuration information */
	int		sadfd;		/* file descriptor to SAD driver */
	int		retcode = 0;	/* return code */

	if ((sadfd = open(ADMINDEV, O_RDWR)) < 0) {
		(void) fprintf(stderr, Openerr, Cmdp, ADMINDEV);
		perror("");
		return (1);
	}
	push.sap_cmd = SAP_CLEAR;
	push.sap_minor = min;
	push.sap_major = maj;

	if (ioctl(sadfd, SAD_SAP, &push) < 0) {
		int error = errno;

		retcode = 1;
		(void) fprintf(stderr, gettext("%s: ERROR: Could not remove "
		    "autopush information\n"), Cmdp);
		switch (error) {
		case EPERM:
			(void) fprintf(stderr, gettext("%s: ERROR: You don't "
			    "have permission to remove autopush "
			    "information\n"), Cmdp);
			break;

		case EINVAL:
			if ((min != 0) && (ioctl(sadfd, SAD_GAP, &push) == 0) &&
			    (push.sap_cmd == SAP_ALL))
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "When removing an entry for ALL minors, "
				    "minor must be set to 0\n"), Cmdp);
			else
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "Invalid major device number\n"), Cmdp);
			break;

		case ENODEV:
			(void) fprintf(stderr, gettext("%s: ERROR: Major/minor "
			    "not configured for autopush\n"), Cmdp);
			break;

		case ERANGE:
			(void) fprintf(stderr, gettext("%s: ERROR: minor must "
			    "be set to begining of range when clearing\n"),
			    Cmdp);
			break;

		default:
			(void) fprintf(stderr, gettext("%s: ERROR: "), Cmdp);
			(void) fprintf(stderr, "%s\n", strerror(error));
			break;
		} /* switch */
	}
	return (retcode);
}

/*
 * get_info():
 *	get autopush configuration information.
 */
static int
get_info(major_t maj, minor_t min)
{
	struct strapush push;	/* configuration information */
	int		i;	/* counter */
	int		sadfd;	/* file descriptor to SAD driver */

	if ((sadfd = open(USERDEV, O_RDWR)) < 0) {
		(void) fprintf(stderr, Openerr, Cmdp, USERDEV);
		perror("");
		return (1);
	}
	push.sap_major = maj;
	push.sap_minor = min;

	if (ioctl(sadfd, SAD_GAP, &push) < 0) {
		int error = errno;

		(void) fprintf(stderr, gettext("%s: ERROR: Could not get "
		    "autopush information\n"), Cmdp);
		switch (error) {
		case EINVAL:
			(void) fprintf(stderr, gettext("%s: ERROR: Invalid "
			    "major device number\n"), Cmdp);
			break;

		case ENOSTR:
			(void) fprintf(stderr, gettext("%s: ERROR: Major "
			    "device is not a STREAMS driver\n"), Cmdp);
			break;

		case ENODEV:
			(void) fprintf(stderr, gettext("%s: ERROR: Major/minor "
			    "not configured for autopush\n"), Cmdp);
			break;

		default:
			(void) fprintf(stderr, gettext("%s: ERROR: "), Cmdp);
			(void) fprintf(stderr, "%s\n", strerror(error));
			break;
		} /* switch */
		return (1);
	}
	(void) printf(OHEADER);
	switch (push.sap_cmd) {
	case SAP_ONE:
		(void) printf(OFORMAT1_ONE, push.sap_major, push.sap_minor);
		break;

	case SAP_RANGE:
		(void) printf(OFORMAT1_RANGE, push.sap_major, push.sap_minor,
		    push.sap_lastminor);
		break;

	case SAP_ALL:
		(void) printf(OFORMAT1_ALL, push.sap_major);
		break;

	default:
		(void) fprintf(stderr,
		    gettext("%s: ERROR: Unknown configuration type\n"), Cmdp);
		return (1);
	}

	for (i = 0; i < push.sap_npush; i++) {

		(void) printf("%s", push.sap_list[i]);

		if (push.sap_anchor == (i + 1))
			(void) printf(" %s", AP_ANCHOR);

		if (i < push.sap_npush - 1)
			(void) printf(" ");

	}

	(void) printf("\n");
	return (0);
}

/*
 * is_white_space():
 *	Return 1 if buffer is all white space.
 *	Return 0 otherwise.
 */
static int
is_white_space(char *bufp)
{
	while (*bufp) {
		if (!isspace(*bufp))
			return (0);
		bufp++;
	}
	return (1);
}

/*
 * parse_line():
 *	Parse input line from file and report any errors found.  Fill
 *	strapush structure along the way.  Returns 1 if the line has
 *	errors and 0 if the line is well-formed.  Another hidden
 *	dependency on MAXAPUSH. `linep' is the input buffer, `lineno'
 *	is the current line number, and `namep' is the filename.
 */
static int
parse_line(char *linep, int lineno, char *namep, struct strapush *pushp)
{
	char		*wp;		/* word pointer */
	char		*cp;		/* character pointer */
	int		midx;		/* module index */
	int		npush;		/* number of modules to push */
	char		c;
	major_t		major_num;

	pushp->sap_anchor = 0;		/* by default, no anchor */

	/*
	 * Find the major device number.
	 */
	for (wp = linep; isspace(*wp); wp++)
		;
	for (cp = wp; !isspace(*cp); cp++)
		;
	if (!isspace(*cp)) {
		(void) fprintf(stderr, Badline, Cmdp, namep, lineno);
		return (1);
	}
	c = *cp;
	*cp = '\0';
	if (modctl(MODGETMAJBIND, wp, strlen(wp) + 1, &major_num) != 0) {
		(void) fprintf(stderr, Badline, Cmdp, namep, lineno);
		return (1);
	}
	*cp = c;
	pushp->sap_major = major_num;

	/*
	 * Find the minor device number.  Must handle negative values here.
	 */
	for (wp = cp; isspace(*wp); wp++)
		;
	for (cp = wp; (isdigit(*cp) || (*cp == MINUS)); cp++)
		;
	if (!isspace(*cp)) {
		(void) fprintf(stderr, Badline, Cmdp, namep, lineno);
		return (1);
	}
	pushp->sap_minor = (minor_t)atol(wp);

	/*
	 * Find the lastminor.
	 */
	for (wp = cp; isspace(*wp); wp++)
		;
	for (cp = wp; isdigit(*cp); cp++)
		;
	if (!isspace(*cp)) {
		(void) fprintf(stderr, Badline, Cmdp, namep, lineno);
		return (1);
	}
	pushp->sap_lastminor = (minor_t)atol(wp);

	/*
	 * Read the list of module names.
	 */
	npush = 0;
	while ((npush < MAXAPUSH) && (*cp)) {

		while (isspace(*cp))
			cp++;

		if (strncasecmp(cp, AP_ANCHOR, sizeof (AP_ANCHOR) - 1) == 0) {
			if (pushp->sap_anchor != 0) {
				(void) fprintf(stderr,
				    gettext("%s: ERROR: File %s: more than "
				    "one anchor in line, line %d ignored\n"),
				    Cmdp, namep, lineno);
				return (1);
			}
			if (npush == 0)
				(void) fprintf(stderr,
				    gettext("%s: WARNING: File %s: anchor at "
				    "beginning of stream on line %d ignored\n"),
				    Cmdp, namep, lineno);
			pushp->sap_anchor = npush;
			cp += sizeof (AP_ANCHOR) - 1;
			continue;
		}

		for (midx = 0; !isspace(*cp) && *cp; midx++) {
			if (midx == FMNAMESZ) {
				(void) fprintf(stderr, gettext("%s: ERROR: "
				    "File %s: module name too long, line %d "
				    "ignored\n"), Cmdp, namep, lineno);
				return (1);
			}
			pushp->sap_list[npush][midx] = *cp++;
		}

		if (midx > 0) {
			pushp->sap_list[npush][midx] = '\0';
			npush++;
		}
	}
	pushp->sap_npush = npush;

	/*
	 * We have everything we want from the line.
	 * Now make sure there is no extra garbage on the line.
	 */
	while (isspace(*cp))
		cp++;
	if (*cp) {
		(void) fprintf(stderr,
		    gettext("%s: ERROR: File %s: too many modules, line %d "
		    "ignored\n"), Cmdp, namep, lineno);
		return (1);
	}
	return (0);
}