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

#include <sys/bootprops.h>
#include <sys/bootconf.h>
#include <sys/modctl.h>
#include <sys/mman.h>
#include <sys/kmem.h>
#include <sys/ddi.h>
#include <sys/sunddi.h>
#include <sys/types.h>
#include <sys/obpdefs.h>
#include <sys/promif.h>
#include <sys/iscsi_protocol.h>

#define	ISCSI_OBP_MAX_CHAP_USER_LEN	16
#define	ISCSI_OBP_MIN_CHAP_LEN		12
#define	ISCSI_OBP_MAX_CHAP_LEN		16

#define	OBP_GET_KEY_STATUS_OK		0
#define	OBP_GET_KEY_STATUS_NOT_EXIST	-3

ib_boot_prop_t boot_property;
extern ib_boot_prop_t *iscsiboot_prop;
static int inet_aton(char *ipstr, uchar_t *ip);
static boolean_t parse_lun_num(uchar_t *str_num, uchar_t *hex_num);
static void generate_iscsi_initiator_id(void);

static int
isdigit(int ch)
{
	return (ch >= '0' && ch <= '9');
}

static boolean_t
iscsiboot_tgt_prop_read(void)
{
	int		proplen;
	boolean_t	set		= B_FALSE;
	char		iscsi_target_ip[INET6_ADDRSTRLEN];
	uchar_t		iscsi_target_name[ISCSI_MAX_NAME_LEN];
	uchar_t		iscsi_par[8];
	char		chap_user[ISCSI_OBP_MAX_CHAP_USER_LEN]	= {0};
	char		chap_password[ISCSI_OBP_MAX_CHAP_LEN]	= {0};
	uchar_t		iscsi_port[8];
	uchar_t		iscsi_lun[8];
	uchar_t		iscsi_tpgt[8];
	long		iscsi_tpgtl;
	long		port;
	int		ret		= 0;
	int		status		= 0;
	int		chap_user_len	= 0;
	int		chap_pwd_len	= 0;

	/* Get iscsi target IP address */
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_TARGET_IP);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_TARGET_IP,
		    iscsi_target_ip) > 0) {
			if (inet_aton(iscsi_target_ip,
			    (uchar_t *)&boot_property.boot_tgt.tgt_ip_u) ==
			    0) {
				boot_property.boot_tgt.sin_family = AF_INET;
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		return (B_FALSE);
	}

	/* Get iscsi target port number */
	set = B_FALSE;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_PORT);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_PORT,
		    iscsi_port) > 0) {
			if (ddi_strtol((const char *)iscsi_port, NULL,
			    10, &port) == 0) {
				boot_property.boot_tgt.tgt_port =
				    (unsigned int)port;
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		boot_property.boot_tgt.tgt_port = 3260;
	}

	/* Get iscsi target LUN number */
	set = B_FALSE;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_LUN);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_LUN,
		    iscsi_lun) > 0) {
			if (parse_lun_num(iscsi_lun,
			    (uchar_t *)
			    (&boot_property.boot_tgt.tgt_boot_lun[0]))
			    == B_TRUE) {
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		bzero((void *)boot_property.boot_tgt.tgt_boot_lun, 8);
	}

	/* Get iscsi target portal group tag */
	set = B_FALSE;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_TPGT);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_TPGT,
		    iscsi_tpgt) > 0) {
			if (ddi_strtol((const char *)iscsi_tpgt, NULL, 10,
			    &iscsi_tpgtl) == 0) {
				boot_property.boot_tgt.tgt_tpgt =
				    (uint16_t)iscsi_tpgtl;
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		boot_property.boot_tgt.tgt_tpgt = 1;
	}

	/* Get iscsi target node name */
	set = B_FALSE;
	boot_property.boot_tgt.tgt_name = NULL;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_TARGET_NAME);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_TARGET_NAME,
		    iscsi_target_name) > 0) {
			boot_property.boot_tgt.tgt_name =
			    (uchar_t *)kmem_zalloc(proplen + 1, KM_SLEEP);
			boot_property.boot_tgt.tgt_name_len = proplen + 1;
			(void) snprintf((char *)boot_property.boot_tgt.tgt_name,
			    proplen + 1, "%s", iscsi_target_name);
			set = B_TRUE;
		}
	}
	if (set != B_TRUE) {
		if (boot_property.boot_tgt.tgt_name != NULL) {
			kmem_free(boot_property.boot_tgt.tgt_name,
			    boot_property.boot_tgt.tgt_name_len);
			boot_property.boot_tgt.tgt_name = NULL;
			boot_property.boot_tgt.tgt_name_len = 0;
		}
		return (B_FALSE);
	}

	/* Get iscsi target boot partition */
	set = B_FALSE;
	boot_property.boot_tgt.tgt_boot_par = NULL;
	boot_property.boot_tgt.tgt_boot_par_len = 0;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_PAR);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_PAR, iscsi_par) > 0) {
			boot_property.boot_tgt.tgt_boot_par =
			    (uchar_t *)kmem_zalloc(proplen + 1, KM_SLEEP);
			boot_property.boot_tgt.tgt_boot_par_len = proplen + 1;
			(void) snprintf(
			    (char *)boot_property.boot_tgt.tgt_boot_par,
			    proplen + 1, "%s", iscsi_par);
			set = B_TRUE;
		}
	}
	if (set != B_TRUE) {
		boot_property.boot_tgt.tgt_boot_par =
		    (uchar_t *)kmem_zalloc(2, KM_SLEEP);
		boot_property.boot_tgt.tgt_boot_par_len = 2;
		boot_property.boot_tgt.tgt_boot_par[0] = 'a';
	}

	/* Get CHAP name and secret */
	ret = prom_get_security_key(BP_CHAP_USER, chap_user,
	    ISCSI_OBP_MAX_CHAP_USER_LEN, &chap_user_len, &status);
	if (ret != 0) {
		return (B_FALSE);
	}
	if (status == OBP_GET_KEY_STATUS_NOT_EXIST) {
		/* No chap name */
		return (B_TRUE);
	}
	if (status != OBP_GET_KEY_STATUS_OK ||
	    chap_user_len > ISCSI_OBP_MAX_CHAP_USER_LEN ||
	    chap_user_len <= 0) {
		return (B_FALSE);
	}

	ret = prom_get_security_key(BP_CHAP_PASSWORD, chap_password,
	    ISCSI_OBP_MAX_CHAP_LEN, &chap_pwd_len, &status);
	if (ret != 0) {
		return (B_FALSE);
	}

	if (status == OBP_GET_KEY_STATUS_NOT_EXIST) {
		/* No chap secret */
		return (B_TRUE);
	}
	if (status != OBP_GET_KEY_STATUS_OK ||
	    chap_pwd_len > ISCSI_OBP_MAX_CHAP_LEN ||
	    chap_pwd_len <= 0) {
		return (B_FALSE);
	}

	boot_property.boot_init.ini_chap_name =
	    (uchar_t *)kmem_zalloc(chap_user_len + 1, KM_SLEEP);
	boot_property.boot_init.ini_chap_name_len = chap_user_len + 1;
	(void) memcpy(boot_property.boot_init.ini_chap_name, chap_user,
	    chap_user_len);

	boot_property.boot_init.ini_chap_sec =
	    (uchar_t *)kmem_zalloc(chap_pwd_len + 1, KM_SLEEP);
	boot_property.boot_init.ini_chap_sec_len = chap_pwd_len + 1;
	(void) memcpy(boot_property.boot_init.ini_chap_sec, chap_password,
	    chap_pwd_len);

	return (B_TRUE);
}

static boolean_t
iscsiboot_init_prop_read(void)
{
	int	proplen;
	uchar_t	iscsi_initiator_id[ISCSI_MAX_NAME_LEN];
	boolean_t	set = B_FALSE;

	/* Get initiator node name */
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_INITIATOR_ID);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_INITIATOR_ID,
		    iscsi_initiator_id) > 0) {
			boot_property.boot_init.ini_name =
			    (uchar_t *)kmem_zalloc(proplen + 1, KM_SLEEP);
			boot_property.boot_init.ini_name_len = proplen + 1;
			(void) snprintf(
			    (char *)boot_property.boot_init.ini_name,
			    proplen + 1, "%s", iscsi_initiator_id);
			set = B_TRUE;
		}
	}
	if (set != B_TRUE) {
		generate_iscsi_initiator_id();
	}
	return (B_TRUE);
}

static boolean_t
iscsiboot_nic_prop_read(void)
{
	int	proplen;
	char	host_ip[INET6_ADDRSTRLEN];
	char	router_ip[INET6_ADDRSTRLEN];
	char	subnet_mask[INET6_ADDRSTRLEN];
	uchar_t	iscsi_network_path[MAXPATHLEN];
	char	host_mac[6];
	uchar_t	hex_netmask[4];
	pnode_t	nodeid;
	boolean_t	set = B_FALSE;

	/* Get host IP address */
	proplen = BOP_GETPROPLEN(bootops, BP_HOST_IP);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_HOST_IP,
		    host_ip) > 0) {
			if (inet_aton(host_ip,
			    (uchar_t *)&boot_property.boot_nic.nic_ip_u) ==
			    0) {
				boot_property.boot_nic.sin_family = AF_INET;
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		return (B_FALSE);
	}

	/* Get router IP address */
	proplen = BOP_GETPROPLEN(bootops, BP_ROUTER_IP);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ROUTER_IP,
		    router_ip) > 0) {
			(void) inet_aton(router_ip,
			    (uchar_t *)&boot_property.boot_nic.nic_gw_u);
		}
	}

	/* Get host netmask */
	set = B_FALSE;
	proplen = BOP_GETPROPLEN(bootops, BP_SUBNET_MASK);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_SUBNET_MASK,
		    subnet_mask) > 0) {
			if (inet_aton(subnet_mask, hex_netmask) == 0) {
				int i = 0;
				uint32_t tmp = *((uint32_t *)hex_netmask);
				while (tmp) {
					i ++;
					tmp = tmp << 1;
				}
				boot_property.boot_nic.sub_mask_prefix = i;
				set = B_TRUE;
			}
		}
	}
	if (set != B_TRUE) {
		boot_property.boot_nic.sub_mask_prefix = 24;
	}

	/* Get iscsi boot NIC path in OBP */
	set = B_FALSE;
	proplen = BOP_GETPROPLEN(bootops, BP_ISCSI_NETWORK_BOOTPATH);
	if (proplen > 0) {
		if (BOP_GETPROP(bootops, BP_ISCSI_NETWORK_BOOTPATH,
		    iscsi_network_path) > 0) {
			nodeid = prom_finddevice((char *)iscsi_network_path);
			proplen = prom_getproplen(nodeid, BP_LOCAL_MAC_ADDRESS);
			if (proplen > 0) {
				if (prom_getprop(nodeid, BP_LOCAL_MAC_ADDRESS,
				    host_mac) > 0) {
					(void) memcpy(
					    boot_property.boot_nic.nic_mac,
					    host_mac, 6);
					set = B_TRUE;
				}
			}
		}
	}
	if (set != B_TRUE) {
		return (B_FALSE);
	}

	return (B_TRUE);
}

/*
 * Manully construct iscsiboot_prop table based on
 * OBP '/chosen' properties related to iscsi boot
 */
void
ld_ib_prop()
{
	if (iscsiboot_prop != NULL)
		return;

	if ((iscsiboot_tgt_prop_read() == B_TRUE) &&
	    (iscsiboot_init_prop_read() == B_TRUE) &&
	    (iscsiboot_nic_prop_read() == B_TRUE)) {
		iscsiboot_prop = &boot_property;
	} else {
		iscsi_boot_prop_free();
	}
}

static boolean_t
parse_lun_num(uchar_t *str_num, uchar_t *hex_num)
{
	char *p, *buf;
	uint16_t *conv_num = (uint16_t *)hex_num;
	long tmp;
	int i = 0;

	if ((str_num == NULL) || (hex_num == NULL)) {
		return (B_FALSE);
	}
	bzero((void *)hex_num, 8);
	buf = (char *)str_num;

	for (i = 0; i < 4; i++) {
		p = NULL;
		p = strchr((const char *)buf, '-');
		if (p != NULL) {
			*p = '\0';
		}
		if (ddi_strtol((const char *)buf, NULL, 16, &tmp) != 0) {
			return (B_FALSE);
		}
		conv_num[i] = (uint16_t)tmp;
		if (p != NULL) {
			buf = p + 1;
		} else {
			break;
		}
	}

	return (B_TRUE);
}

static void
generate_iscsi_initiator_id(void)
{
	boot_property.boot_init.ini_name_len = 38;
	boot_property.boot_init.ini_name =
	    (uchar_t *)kmem_zalloc(boot_property.boot_init.ini_name_len,
	    KM_SLEEP);
	(void) snprintf((char *)boot_property.boot_init.ini_name,
	    38, "iqn.1986-03.com.sun:boot.%02x%02x%02x%02x%02x%02x",
	    boot_property.boot_nic.nic_mac[0],
	    boot_property.boot_nic.nic_mac[1],
	    boot_property.boot_nic.nic_mac[2],
	    boot_property.boot_nic.nic_mac[3],
	    boot_property.boot_nic.nic_mac[4],
	    boot_property.boot_nic.nic_mac[5]);
}


/* We only deal with a.b.c.d decimal format. ip points to 4 byte storage */
static int
inet_aton(char *ipstr, uchar_t *ip)
{
	int i = 0;
	uchar_t val[4] = {0};
	char c = *ipstr;

	for (;;) {
		if (!isdigit(c))
			return (-1);
		for (;;) {
			if (!isdigit(c))
				break;
			val[i] = val[i] * 10 + (c - '0');
			c = *++ipstr;
		}
		i++;
		if (i == 4)
			break;
		if (c != '.')
			return (-1);
		c = *++ipstr;
	}
	if (c != 0)
		return (-1);
	bcopy(val, ip, 4);
	return (0);
}