/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <errno.h>
#include <locale.h>
#include <pwd.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <sys/types.h>

#include "allocate.h"

#if !defined(TEXT_DOMAIN)
#define	TEXT_DOMAIN "SUNW_OST_OSCMD"
#endif

extern void audit_allocate_argv(int, int, char *[]);
extern int audit_allocate_record(int);

static void
usage(int func)
{
	char *use[7];

	use[0] = gettext("allocate [-s] [-U uname] [-F] device");
	use[1] = gettext("allocate [-s] [-U uname] -g dev_type");
	use[2] = gettext("deallocate [-s] [-F] device");
	use[3] = gettext("deallocate [-s] -I");
	use[4] = gettext("list_devices [-s] [-U uid] -l [device]");
	use[5] = gettext("list_devices [-s] [-U uid] -n [device]");
	use[6] = gettext("list_devices [-s] [-U uid] -u [device]");

	switch (func) {
		case 0:
			(void) fprintf(stderr, "%s\n%s\n", use[0], use[1]);
			break;
		case 1:
			(void) fprintf(stderr, "%s\n%s\n", use[2], use[3]);
			break;
		case 2:
			(void) fprintf(stderr, "%s\n%s\n%s\n", use[4], use[5],
			    use[6]);
			break;
		default:
			(void) fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n",
				use[0], use[1], use[2], use[3], use[4]);
	}
	exit(1);
}

static void
print_error(int error, char *name)
{
	char *msg;

	switch (error) {
	case SYSERROR:
		msg = gettext("Unknown System error.");
		break;
	case IMPORT_ERR:
		msg = gettext(
		    "User lacks authorization required for this operation.");
		break;
	case NODAENT:
		msg = gettext(
		    "No device allocate file entry for specified device.");
		break;
	case NODMAPENT:
		msg = gettext(
		    "No device maps file entry for specified device.");
		break;
	case DACLCK:
		msg = gettext("Concurrent operations for specified device, "
		    "try later.");
		break;
	case DACACC:
		msg = gettext(
		    "Can't access DAC file for the device specified.");
		break;
	case DEVLST:
		msg = gettext(
		    "Could not use device list for the device specified.");
		break;
	case NALLOCU:
		msg = gettext("Specified device is allocated to another user.");
		break;
	case NOTAUTH:
		msg = gettext("Not authorized for specified operation.");
		break;
	case CNTFRC:
		msg = gettext("Can't force deallocate specified device.");
		break;
	case CNTDEXEC:
		msg = gettext(
		    "Can't exec device-clean program for specified device.");
		break;
	case NO_DEVICE:
		msg = gettext(
		    "Can't find a device of type requested to allocate.");
		break;
	case DSPMISS:
		msg = gettext(
		    "Device special file(s) missing for specified device.");
		break;
	case ALLOCERR:
		msg = gettext("Device specified is in allocate error state.");
		break;
	case CHOWN_PERR:
		msg = gettext("Process lacks privilege required to chown().");
		break;
	case ALLOC:
		msg = gettext("Device already allocated.");
		break;
	case ALLOC_OTHER:
		msg = gettext("Device allocated to another user.");
		break;
	case NALLOC:
		msg = gettext("Device not allocated.");
		break;
	case AUTHERR:
		msg = gettext("Device not allocatable.");
		break;
	case CLEAN_ERR:
		msg = gettext("Unable to clean up the device.");
		break;
	case SETACL_PERR:
		msg = gettext("Process lacks privilege required to set ACL.");
		break;
	case DEVNAME_ERR:
		msg = gettext("Error forming device name.");
		break;
	case DEVNAME_TOOLONG:
		msg = gettext("Device name is too long.");
		break;
	default:
		msg = gettext("Unknown error code.");
		break;
	}

	(void) fprintf(stderr, "%s: %s\n", name, msg);
	(void) fflush(stderr);
}

char *newenv[] = {"PATH=/usr/bin:/usr/sbin",
			NULL,			/* for LC_ALL		*/
			NULL,			/* for LC_COLLATE	*/
			NULL,			/* for LC_CTYPE		*/
			NULL,			/* for LC_MESSAGES	*/
			NULL,			/* for LC_NUMERIC	*/
			NULL,			/* for LC_TIME		*/
			NULL,			/* for LANG		*/
			NULL
};

static char *
getenvent(char *name, char *env[])
{
	for (; *env != NULL; env++) {
		if (strncmp(*env, name, strlen(name)) == 0)
			return (*env);
	}
	return (NULL);
}

int
main(int argc, char *argv[], char *envp[])
{
	char	*name, *env;
	int	func = -1, optflg = 0, error = 0, c;
	uid_t	uid = getuid();
	char	*uname = NULL, *device = NULL;
	struct passwd *pw_ent;
	int env_num = 1;	/* PATH= is 0 entry */

	(void) setlocale(LC_ALL, "");
	(void) textdomain(TEXT_DOMAIN);

	/*
	 * get all enviroment variables
	 * which affect on internationalization.
	 */
	env = getenvent("LC_ALL=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LC_COLLATE=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LC_CTYPE=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LC_MESSAGES=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LC_NUMERIC=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LC_TIME=", envp);
	if (env != NULL)
		newenv[env_num++] = env;
	env = getenvent("LANG=", envp);
	if (env != NULL)
		newenv[env_num] = env;

	if ((name = strrchr(argv[0], '/')) == NULL)
		name = argv[0];
	else
		name++;

	if (strcmp(name, "allocate") == 0)
		func = 0;
	else if (strcmp(name, "deallocate") == 0)
		func = 1;
	else if (strcmp(name, "list_devices") == 0)
		func = 2;
	else {
		usage(ALL);
	}

	audit_allocate_argv(func, argc, argv);

	if (func == 0) {	/* allocate */
		while ((c = getopt(argc, argv, "sU:Fg")) != -1) {
			switch (c) {
			case 's':
				optflg |= SILENT;
				break;
			case 'U':
				optflg |= USERNAME;
				uname = optarg;
				break;
			case 'g':
				optflg |= TYPE;
				break;
			case 'F':
				optflg |= FORCE;
				break;
			case '?':
			default :
				usage(func);
			}
		}

		if ((optflg & TYPE) && (optflg & FORCE))
			usage(func);

		/*
		 * allocate(1) must be supplied with one device argument
		 */
		if ((argc - optind) != 1) {
			usage(func);
		} else {
			device = argv[optind];
		}
	}

	else if (func == 1) {	/* deallocate */
		while ((c = getopt(argc, argv, "sFI")) != -1) {
			switch (c) {
			case 's':
				optflg |= SILENT;
				break;
			case 'F':
				optflg |= FORCE;
				break;
			case 'I':
				optflg |= FORCE_ALL;
				break;
			case '?':
			default :
				usage(func);
			}
		}

		if ((optflg & FORCE) && (optflg & FORCE_ALL))
			usage(func);

		/*
		 * deallocate(1) must be supplied with one device
		 * argument unless the '-I' argument is supplied
		 */
		if (!(optflg & FORCE_ALL)) {
			if ((argc - optind) != 1) {
				usage(func);
			} else {
				device = argv[optind];
			}
		} else {
			if ((argc - optind) >= 1) {
				usage(func);
			}
		}
	}

	else if (func == 2) {	/* list_devices */
		while ((c = getopt(argc, argv, "sU:lnu")) != -1) {
			switch (c) {
			case 's':
				optflg |= SILENT;
				break;
			case 'U':
				optflg |= USERID;
				uid = atoi(optarg);
				break;
			case 'l':
				optflg |= LIST;
				break;
			case 'n':
				optflg |= FREE;
				break;
			case 'u':
				optflg |= CURRENT;
				break;
			case '?':
			default :
				usage(func);
			}
		}

		if (((optflg & LIST) && (optflg & FREE)) ||
		    ((optflg & LIST) && (optflg & CURRENT)) ||
		    ((optflg & FREE) && (optflg & CURRENT)) ||
		    (!(optflg & (LIST | FREE | CURRENT))))
			usage(func);

		/*
		 * list_devices(1) takes an optional device argument
		 */
		if ((argc - optind) == 1) {
			device = argv[optind];
		} else {
			if ((argc - optind) > 1) {
				usage(func);
			}
		}
	}

	if (optflg & USERNAME) {
		if ((pw_ent = getpwnam(uname)) == NULL) {
			(void) fprintf(stderr, gettext(
			    "Invalid user name -- %s -- \n"), uname);
			exit(1);
		}
		uid = pw_ent->pw_uid;
	}

	if (optflg & USERID) {
		if ((pw_ent = getpwuid(uid)) == NULL) {
			(void) fprintf(stderr, gettext(
			    "Invalid user ID -- %d -- \n"), uid);
			exit(1);
		}
		uid = pw_ent->pw_uid;
	}

	if (func == 0) {
		error = allocate(optflg, uid, device);
	} else if (func == 1) {
		error = deallocate(optflg, uid, device);
	} else if (func == 2) {
		error = list_devices(optflg, uid, device);
	}

	(void) audit_allocate_record(error);

	if (error) {
		if (!(optflg & SILENT))
			print_error(error, name);
		exit(error);
	}

	return (0);
}