/*

 * 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 (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.

 */

#include <stdlib.h>

#include <stdio.h>

#include <wchar.h>

#include <widec.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <arpa/inet.h>

#include <netdb.h>

#include <unistd.h>

#include <libintl.h>

#include <limits.h>

#include <string.h>

#include <strings.h>

#include <syslog.h>

#include <errno.h>

#include <netinet/in.h>

#include <sys/socket.h>

#include <arpa/inet.h>

#include <wctype.h>

#include <assert.h>

#include <ima.h>

#include <libsun_ima.h>

#include <sys/iscsi_protocol.h>

#include <sys/scsi/adapters/iscsi_if.h>

#include "cmdparse.h"

#include "sun_ima.h"

#include "iscsiadm.h"

#define	VERSION_STRING_MAX_LEN	10

#define	MAX_LONG_CHAR_LEN 19

#define	MAX_AUTH_METHODS 5

/*

 * Version number:

 *  MAJOR - This should only change when there is an incompatible change made

 *  to the interfaces or the output.

 *

 *  MINOR - This should change whenever there is a new command or new feature

 *  with no incompatible change.

 */

#define	VERSION_STRING_MAJOR	    "1"

#define	VERSION_STRING_MINOR	    "0"

#define	OPTIONSTRING1	"yes|no"

#define	OPTIONSTRING2	"initiator node name"

#define	OPTIONSTRING3	"initiator node alias"

#define	OPTIONSTRING4	"enable|disable"

#define	OPTIONSTRING5	"key=value,..."

#define	OPTIONSTRING6	"none|CRC32"

#define	OPTIONSTRING7	"CHAP name"

#define	OPTIONSTRING8	"<# sessions>|<IP Address>[,<IP Address>]*"

#define	OPTIONSTRING9	"tunable-prop=value"

#define	OPTIONVAL1	"0 to 3600"

#define	OPTIONVAL2	"512 to 2**24 - 1"

#define	OPTIONVAL3	"1 to 65535"

#define	OPTIONVAL4	"<IP address>[:port]"

#define	MAX_ISCSI_NAME_LEN	    223

#define	MAX_ADDRESS_LEN		    255

#define	MIN_CHAP_SECRET_LEN	    12

#define	MAX_CHAP_SECRET_LEN	    16

#define	DEFAULT_ISCSI_PORT	    3260

#define	ISNS_DEFAULT_SERVER_PORT    3205

#define	DEFAULT_RADIUS_PORT	    1812

#define	MAX_CHAP_NAME_LEN	    512

#define	ISCSI_DEFAULT_RX_TIMEOUT_VALUE		"60"

#define	ISCSI_DEFAULT_CONN_DEFAULT_LOGIN_MAX	"180"

#define	ISCSI_DEFAULT_LOGIN_POLLING_DELAY	"60"

/* For listNode */

#define	INF_ERROR		1

#define	INVALID_NODE_NAME	2

#define	IMABOOLPRINT(prop, option)	 \

	if ((option) == PRINT_CONFIGURED_PARAMS) { \

		(void) fprintf(stdout, "%s/%s\n", \

		(prop).defaultValue == IMA_TRUE ? gettext("yes") : \

			gettext("no"), \

		(prop).currentValueValid == IMA_TRUE ? \

			((prop).currentValue == IMA_TRUE ? \

			gettext("yes"): gettext("no")) : "-"); \

	} else if ((option) == PRINT_NEGOTIATED_PARAMS) { \

		(void) fprintf(stdout, "%s\n", \

		(prop).currentValueValid == IMA_TRUE ? \

		(((prop).currentValue == IMA_TRUE) ? gettext("yes") : \

		gettext("no")) : "-"); \

	}

#define	IMAMINMAXPRINT(prop, option) \

	if ((option) == PRINT_CONFIGURED_PARAMS) { \

		(void) fprintf(stdout, "%d/", (prop).defaultValue); \

		if ((prop).currentValueValid == IMA_TRUE) { \

			(void) fprintf(stdout, "%d\n", (prop).currentValue); \

		} else if ((prop).currentValueValid == IMA_FALSE) { \

			(void) fprintf(stdout, "%s\n", "-"); \

		} \

	} else if ((option) == PRINT_NEGOTIATED_PARAMS) { \

		if ((prop).currentValueValid == IMA_TRUE) { \

			(void) fprintf(stdout, "%d\n", (prop).currentValue); \

		} else if ((prop).currentValueValid == IMA_FALSE) { \

			(void) fprintf(stdout, "%s\n", "-"); \

		} \

	}

/* forward declarations */

#define	PARSE_ADDR_OK				0

#define	PARSE_ADDR_MISSING_CLOSING_BRACKET	1

#define	PARSE_ADDR_PORT_OUT_OF_RANGE		2

#define	PARSE_TARGET_OK				0

#define	PARSE_TARGET_INVALID_TPGT		1

#define	PARSE_TARGET_INVALID_ADDR		2

#define	PRINT_CONFIGURED_PARAMS			1

#define	PRINT_NEGOTIATED_PARAMS			2

typedef enum iSCSINameCheckStatus {

	iSCSINameCheckOK,

	iSCSINameLenZero,

	iSCSINameLenExceededMax,

	iSCSINameUnknownType,

	iSCSINameInvalidCharacter,

	iSCSINameIqnFormatError,

	iSCSINameEUIFormatError,

	iSCSINameIqnDateFormatError,

	iSCSINameIqnSubdomainFormatError,

	iSCSINameIqnInvalidYearError,

	iSCSINameIqnInvalidMonthError,

	iSCSINameIqnFQDNError

} iSCSINameCheckStatusType;

/* Utility functions */

iSCSINameCheckStatusType iSCSINameStringProfileCheck(wchar_t *name);

boolean_t isNaturalNumber(char *numberStr, uint32_t upperBound);

static int parseAddress(char *address_port_str, uint16_t defaultPort,

    char *address_str, size_t address_str_len,

    uint16_t *port, boolean_t *isIpv6);

int parseTarget(char *targetStr,

    wchar_t *targetNameStr,

    size_t targetNameStrLen,

    boolean_t *targetAddressSpecified,

    wchar_t *targetAddressStr,

    size_t targetAddressStrLen,

    uint16_t *port,

    boolean_t *tpgtSpecified,

    uint16_t *tpgt,

    boolean_t *isIpv6);

static int chkConnLoginMaxPollingLoginDelay(IMA_OID oid,

    int key, int uintValue);

/* subcommand functions */

static int addFunc(int, char **, int, cmdOptions_t *, void *, int *);

static int listFunc(int, char **, int, cmdOptions_t *, void *, int *);

static int modifyFunc(int, char **, int, cmdOptions_t *, void *, int *);

static int removeFunc(int, char **, int, cmdOptions_t *, void *, int *);

/* helper functions */

static char *getExecBasename(char *);

static int getNodeProps(IMA_NODE_PROPERTIES *);

static int getSecret(char *, int *, int, int);

static int getTargetAddress(int, char *, IMA_TARGET_ADDRESS *);

static int printLoginParameters(char *, IMA_OID, int);

static void printDiscoveryMethod(char *, IMA_UINT32);

static void printTargetLuns(IMA_OID_LIST *);

static void printSendTargets(SUN_IMA_DISC_ADDRESS_KEY_PROPERTIES *);

static void printDigestAlgorithm(SUN_IMA_DIGEST_ALGORITHM_VALUE *, int);

static int setLoginParameter(IMA_OID, int, char *);

static int setLoginParameters(IMA_OID, char *);

static int setTunableParameters(IMA_OID, char *);

static void printLibError(IMA_STATUS);

/* LINTED E_STATIC_UNUSED */

static int sunPluginChk(IMA_OID, boolean_t *);

static int sunInitiatorFind(IMA_OID *);

static int getAuthMethodValue(char *, IMA_AUTHMETHOD *);

static int getLoginParam(char *);

static int getTunableParam(char *);

static void iSCSINameCheckStatusDisplay(iSCSINameCheckStatusType status);

static int modifyIndividualTargetParam(cmdOptions_t *optionList,

    IMA_OID targetOid, int *);

static void listCHAPName(IMA_OID oid);

static int printConfiguredSessions(IMA_OID);

static int printTunableParameters(IMA_OID oid);

/* object functions per subcommand */

static int addAddress(int, int, char *[], int *);

static int addStaticConfig(int, char *[], int *);

static int listDiscovery(int *);

static int listDiscoveryAddress(int, char *[], cmdOptions_t *, int *);

static int listISNSServerAddress(int, char *[], cmdOptions_t *, int *);

static int listNode(int *);

static int listStaticConfig(int, char *[], int *);

static int listTarget(int, char *[], cmdOptions_t *, int *);

static int listTargetParam(int, char *[], cmdOptions_t *, int *);

static int modifyDiscovery(cmdOptions_t *, int *);

static int modifyNodeAuthMethod(IMA_OID, char *, int *);

static int modifyNodeAuthParam(IMA_OID oid, int, char *, int *);

static int modifyNodeRadiusConfig(IMA_OID, char *, int *);

static int modifyNodeRadiusAccess(IMA_OID, char *, int *);

static int modifyNodeRadiusSharedSecret(IMA_OID, int *);

static int modifyNode(cmdOptions_t *, int *);

static int modifyTargetAuthMethod(IMA_OID, char *, int *);

static int modifyTargetAuthParam(IMA_OID oid, int param, char *chapName, int *);

static int modifyTargetParam(cmdOptions_t *, char *, int *);

static int removeAddress(int, int, char *[], int *);

static int removeStaticConfig(int, char *[], int *);

static int removeTargetParam(int, char *[], int *);

static int modifyTargetBidirAuthFlag(IMA_OID, char *, int *);

static int modifyConfiguredSessions(IMA_OID targetOid, char *optarg);

/* LINTED E_STATIC_UNUSED */

static IMA_STATUS getISCSINodeParameter(int paramType,

    IMA_OID *oid,

    void *pProps,

    uint32_t paramIndex);

/* LINTED E_STATIC_UNUSED */

static IMA_STATUS setISCSINodeParameter(int paramType,

    IMA_OID *oid,

    void *pProps,

    uint32_t paramIndex);

/* LINTED E_STATIC_UNUSED */

static IMA_STATUS getDigest(IMA_OID oid, int ioctlCmd,

    SUN_IMA_DIGEST_ALGORITHM_VALUE *algorithm);

IMA_STATUS getNegotiatedDigest(int digestType,

	SUN_IMA_DIGEST_ALGORITHM_VALUE *algorithm,

	SUN_IMA_CONN_PROPERTIES *connProps);

/* globals */

static char *cmdName;

/*

 * Available option letters:

 *

 * befgijklmnoquwxyz

 *

 * DEFGHIJKLMOQUVWXYZ

 */

/*

 * Add new options here

 */

optionTbl_t longOptions[] = {

	{"static", required_arg, 's', OPTIONSTRING4},

	{"sendtargets", required_arg, 't', OPTIONSTRING4},

	{"iSNS", required_arg, 'i', OPTIONSTRING4},

	{"headerdigest", required_arg, 'h', OPTIONSTRING6},

	{"datadigest", required_arg, 'd', OPTIONSTRING6},

	{"login-param", required_arg, 'p', OPTIONSTRING5},

	{"authentication", required_arg, 'a', "CHAP|none"},

	{"bi-directional-authentication", required_arg, 'B', OPTIONSTRING4},

	{"CHAP-secret", no_arg, 'C', NULL},

	{"CHAP-name", required_arg, 'H', OPTIONSTRING7},

	{"node-name", required_arg, 'N', OPTIONSTRING2},

	{"node-alias", required_arg, 'A', OPTIONSTRING3},

	{"radius-server", required_arg, 'r', OPTIONVAL4},

	{"radius-access", required_arg, 'R', OPTIONSTRING4},

	{"radius-shared-secret", no_arg, 'P', NULL},

	{"verbose", no_arg, 'v', NULL},

	{"scsi-target", no_arg, 'S', NULL},

	{"configured-sessions", required_arg, 'c', OPTIONSTRING8},

	{"tunable-param", required_arg, 'T', OPTIONSTRING9},

	{NULL, 0, 0, 0}

};

parameterTbl_t loginParams[] = {

	{"dataseqinorder", DATA_SEQ_IN_ORDER},

	{"defaulttime2retain", DEFAULT_TIME_2_RETAIN},

	{"defaulttime2wait", DEFAULT_TIME_2_WAIT},

	{"firstburstlength", FIRST_BURST_LENGTH},

	{"immediatedata", IMMEDIATE_DATA},

	{"initialr2t", INITIAL_R2T},

	{"maxburstlength", MAX_BURST_LENGTH},

	{"datapduinorder", DATA_PDU_IN_ORDER},

	{"maxoutstandingr2t", MAX_OUTSTANDING_R2T},

	{"maxrecvdataseglen", MAX_RECV_DATA_SEG_LEN},

	{"maxconnections", MAX_CONNECTIONS},

	{"errorrecoverylevel", ERROR_RECOVERY_LEVEL},

	{NULL, 0}

};

parameterTbl_t tunableParams[] = {

	{"recv-login-rsp-timeout", RECV_LOGIN_RSP_TIMEOUT},

	{"conn-login-max", CONN_LOGIN_MAX},

	{"polling-login-delay", POLLING_LOGIN_DELAY},

	{NULL, 0}

};

/*

 * Add new subcommands here

 */

subcommand_t subcommands[] = {

	{"add", ADD, addFunc},

	{"list", LIST, listFunc},

	{"modify", MODIFY, modifyFunc},

	{"remove", REMOVE, removeFunc},

	{NULL, 0, NULL}

};

/*

 * Add objects here

 */

object_t objects[] = {

	{"discovery", DISCOVERY},

	{"discovery-address", DISCOVERY_ADDRESS},

	{"isns-server", ISNS_SERVER_ADDRESS},

	{"initiator-node", NODE},

	{"static-config", STATIC_CONFIG},

	{"target", TARGET},

	{"target-param", TARGET_PARAM},

	{NULL, 0}

};

/*

 * Rules for subcommands and objects

 */

objectRules_t objectRules[] = {

	{TARGET, 0, LIST, 0, ADD|REMOVE|MODIFY, LIST,

	"target-name"},

	{TARGET_PARAM, MODIFY|REMOVE, LIST, 0, ADD, MODIFY,

	"target-name"},

	{DISCOVERY, 0, 0, LIST|MODIFY, ADD|REMOVE, 0, NULL},

	{NODE, 0, 0, MODIFY|LIST, ADD|REMOVE, 0, NULL},

	{STATIC_CONFIG, ADD|REMOVE, LIST, 0, MODIFY, ADD|REMOVE|LIST,

	"target-name,target-address[:port-number][,tpgt]"},

	{DISCOVERY_ADDRESS, ADD|REMOVE, LIST, 0, MODIFY,

	ADD|REMOVE|LIST, "IP-address[:port-number]"},

	{ISNS_SERVER_ADDRESS, ADD|REMOVE, LIST, 0, MODIFY,

	ADD|REMOVE|LIST, "IP-address[:port-number]"},

	{0, 0, 0, 0, 0, NULL}

};

/*

 * list of objects, subcommands, valid short options, required flag and

 * exclusive option string

 *

 * If it's not here, there are no options for that object.

 */

optionRules_t optionRules[] = {

	{DISCOVERY, MODIFY, "sti", B_TRUE, NULL},

	{DISCOVERY_ADDRESS, LIST, "v", B_FALSE, NULL},

	{ISNS_SERVER_ADDRESS, LIST, "v", B_FALSE, NULL},

	{TARGET, LIST, "vS", B_FALSE, NULL},

	{NODE, MODIFY, "NAhdCaRrPHcT", B_TRUE, "CP"},

	{TARGET_PARAM, MODIFY, "ahdBCpcHT", B_TRUE, "C"},

	{TARGET_PARAM, LIST, "v", B_FALSE, NULL},

	{0, 0, 0, 0, 0}

};

static boolean_t

targetNamesEqual(wchar_t *name1, wchar_t *name2)

{

	int i;

	wchar_t wchar1, wchar2;

	if (name1 == NULL || name2 == NULL) {

		return (B_FALSE);

	}

	if (wcslen(name1) != wcslen(name2)) {

		return (B_FALSE);

	}

	/*

	 * Convert names to lower case and compare

	 */

	for (i = 0; i < wcslen(name1); i++) {

		wchar1 = towctrans((wint_t)name1[i], wctrans("tolower"));

		wchar2 = towctrans((wint_t)name2[i], wctrans("tolower"));

		if (wchar1 != wchar2) {

			return (B_FALSE);

		}

	}

	return (B_TRUE);

}

static boolean_t

ipAddressesEqual(IMA_TARGET_ADDRESS addr1, IMA_TARGET_ADDRESS addr2)

{

#define	IPV4_ADDR_BYTES 4

#define	IPV6_ADDR_BYTES 16

	int compSize;

	if (addr1.hostnameIpAddress.id.ipAddress.ipv4Address !=

	    addr2.hostnameIpAddress.id.ipAddress.ipv4Address) {

		return (B_FALSE);

	}

	compSize = IPV6_ADDR_BYTES;

	if (addr1.hostnameIpAddress.id.ipAddress.ipv4Address) {

		compSize = IPV4_ADDR_BYTES;

	}

	if (bcmp(addr1.hostnameIpAddress.id.ipAddress.ipAddress,

	    addr2.hostnameIpAddress.id.ipAddress.ipAddress, compSize) == 0) {

		return (B_TRUE);

	}

	return (B_FALSE);

}

static int

getLoginParam(char *arg)

{

	parameterTbl_t *paramp;

	int len;

	for (paramp = loginParams; paramp->name; paramp++) {

		len = strlen(arg);

		if (len == strlen(paramp->name) &&

		    strncasecmp(arg, paramp->name, len) == 0) {

			return (paramp->val);

		}

	}

	return (-1);

}

static int

getTunableParam(char *arg)

{

	parameterTbl_t *paramp;

	int len;

	for (paramp = tunableParams; paramp->name != NULL; paramp++) {

		len = strlen(arg);

		if (len == strlen(paramp->name) &&

		    strncasecmp(arg, paramp->name, len) == 0) {

			return (paramp->val);

		}

	}

	return (-1);

}

static void

printLibError(IMA_STATUS status)

{

	char *errorString;

	switch (status) {

	case IMA_ERROR_NOT_SUPPORTED:

		errorString =

		gettext("Operation currently not supported");

		break;

	case IMA_ERROR_INSUFFICIENT_MEMORY:

		errorString = gettext("Insufficient memory");

		break;

	case IMA_ERROR_UNEXPECTED_OS_ERROR:

		errorString = gettext("unexpected OS error");

		break;

	case IMA_ERROR_UNKNOWN_ERROR:

		errorString = gettext("Unknown error");

		break;

	case IMA_ERROR_LU_IN_USE:

		errorString = gettext("Logical unit in use");

		break;

	case IMA_ERROR_INVALID_PARAMETER:

		errorString = gettext("Invalid parameter specified");

		break;

	case IMA_ERROR_INVALID_OBJECT_TYPE:

		errorString =

		gettext("Internal library error: Invalid oid type specified");

		break;

	case IMA_ERROR_INCORRECT_OBJECT_TYPE:

		errorString =

		gettext("Internal library error: Incorrect oid type specified");

		break;

	case IMA_ERROR_OBJECT_NOT_FOUND:

		errorString = gettext("Internal library error: Oid not found");

		break;

	case IMA_ERROR_NAME_TOO_LONG:

		errorString = gettext("Name too long");

		break;

	default:

		errorString = gettext("Unknown error");

	}

	(void) fprintf(stderr, "%s: %s\n", cmdName, errorString);

}

/*

 * input:

 *  execFullName - exec name of program (argv[0])

 *

 * Returns:

 *  command name portion of execFullName

 */

static char *

getExecBasename(char *execFullname)

{

	char *lastSlash, *execBasename;

	/* guard against '/' at end of command invocation */

	for (;;) {

		lastSlash = strrchr(execFullname, '/');

		if (lastSlash == NULL) {

			execBasename = execFullname;

			break;

		} else {