* Copyright (c) 2010, 2016, Oracle and/or its affiliates. All rights reserved.

#include <errno.h>

#include "ibdiag_common.h"

#include <sys/ib/clients/of/sol_uverbs/sol_uverbs_ioctl.h>

 * Local defines for sol_uverbs IOCTLs, used while

#ifndef	UVERBS_NODEDESC_UPDATE_STRING

#define	UVERBS_IOCTL_GET_NODEDESC		('v' << 8) | 0x04

#define	UVERBS_IOCTL_SET_NODEDESC		('v' << 8) | 0x05

#define	UVERBS_NODEDESC_UPDATE_STRING		0x00000001

#define	UVERBS_NODEDESC_UPDATE_HCA_STRING	0x00000002

typedef struct sol_uverbs_nodedesc_s {

	int32_t		uverbs_solaris_abi_version;

	char		node_desc_str[64];

	uint32_t	node_desc_update_flag;

} sol_uverbs_nodedesc_t;

/*

 * Override verbs abi version.

 * If the build system doesn't have the intended

 * header file then override with the intended abi version.

 * These changes can be deleted once the build system has

 * the correct header file.

 */

#if	(IB_USER_VERBS_SOLARIS_ABI_VERSION == 2)

#undef	IB_USER_VERBS_SOLARIS_ABI_VERSION

#define	IB_USER_VERBS_SOLARIS_ABI_VERSION	3

#endif

int	nd_read_info_cnt = 0;

static int

read_nodedesc_ioctl(struct ibv_context *context,

    sol_uverbs_nodedesc_t *nodedesc);

static int

write_nodedesc_ioctl(struct ibv_context *context,

    sol_uverbs_nodedesc_t *nodedesc);

update_read_info_hwnames(struct ibv_device **dev_list, int num_devices)

	int		i;

	for (i = 0; dev_list[i] != 0 && i < num_devices; ++i) {

		dev_guid = (uint64_t)ntohll(

		    ibv_get_device_guid(dev_list[i]));

				nd_read_info_arr[j].ofuv_name_valid =

				    B_TRUE;

	memcpy(nd_read_info_arr[nd_read_info_cnt].nd_string,

	    nd_str, nd_len);

do_driver_read_ioctl(struct ibv_device *device)

	int			rc;

	uint64_t		hca_guid;

	struct ibv_context	*context;

	sol_uverbs_nodedesc_t	*nodedescp;

	/* Get the context for the device */

	context = ibv_open_device(device);

	if (!context) {

		IBERROR("Unable to open the device.\n");

		/* NOTREACHED */

	}

	if (context->device != device) {

		IBERROR("Device not set.\n");

		/* NOTREACHED */

	}

	/* Allocate the memory for node descriptor */

	nodedescp = (sol_uverbs_nodedesc_t *)malloc(

	    sizeof (sol_uverbs_nodedesc_t));

	if (nodedescp == NULL) {

		IBERROR("Memory allocation failed.\n");

		/* NOTREACHED */

	nodedescp->uverbs_solaris_abi_version =

	    IB_USER_VERBS_SOLARIS_ABI_VERSION;

	/* Get the guid for the device */

	hca_guid = (uint64_t)ntohll(ibv_get_device_guid(device));

	if (!hca_guid) {

		IBERROR("ibv_get_device_guid failed.\n");

		/* NOTREACHED */

	/* Read node descriptor */

	rc = read_nodedesc_ioctl(context, nodedescp);

	if (rc != 0) {

		IBERROR("Failed to read node descriptor.\n");

		/* NOTREACHED */

	}

	add_read_info_arr((char *)nodedescp->node_desc_str,

	    hca_guid);

read_nodedesc_exit_1:

	/* release the allocated memory */

	free(nodedescp);

read_nodedesc_exit_2:

	/* Close the device */

	ibv_close_device(context);

do_driver_update_ioctl(struct ibv_device *device, char *node_desc,

    char *hca_desc, uint32_t update_flag)

	int			rc;

	struct ibv_context	*context;

	sol_uverbs_nodedesc_t	*nodedescp;

	char			*desc_str;

	desc_str = (node_desc ? node_desc : hca_desc);

	/* Get context for the device */

	context = ibv_open_device(device);

	if (!context) {

		IBERROR("Unable to open the device.\n");

		/* NOTREACHED */

	if (context->device != device) {

		IBERROR("Device not set.\n");

		/* NOTREACHED */

	/* Allocate the memory for node descriptor */

	nodedescp = (sol_uverbs_nodedesc_t *)malloc(

	    sizeof (sol_uverbs_nodedesc_t));

	if (nodedescp == NULL) {

		IBERROR("Memory allocation failed.\n");

		/* NOTREACHED */

	strncpy(nodedescp->node_desc_str, desc_str, 64);

	nodedescp->node_desc_update_flag = update_flag;

	nodedescp->uverbs_solaris_abi_version =

	    IB_USER_VERBS_SOLARIS_ABI_VERSION;

	rc = write_nodedesc_ioctl(context, nodedescp);

	if (rc != 0)

		IBERROR("Failed to set node descriptor.\n");

	free(nodedescp);

write_nodedesc_exit:

	ibv_close_device(context);

read_nodedesc(struct ibv_device **device_list, int num_devices)

	int i;

	for (i = 0; device_list[i] != 0 && i < num_devices; i++)

		do_driver_read_ioctl(device_list[i]);

update_nodedesc(struct ibv_device **device_list, int num_devices,

    char *cmn_nodedesc, char *hca_nodedesc, uint64_t guid,

	int		i, rc = -1;

	uint64_t	dev_guid;

	boolean_t	matched = B_FALSE;

	if (cmn_nodedesc && hca_nodedesc == NULL) {

		for (i = 0; i < num_devices; i++) {

			rc = do_driver_update_ioctl(device_list[i],

			    cmn_nodedesc, hca_nodedesc, update_flag);

			if (rc != 0)

				continue;

			else

				break;

		}

		if (rc != 0 && i == num_devices)

			IBERROR("Failed to set the node descriptor.\n");

		return (rc);

	}

	if (hca_nodedesc && guid) {

		for (i = 0; i < num_devices; i++) {

			dev_guid = (uint64_t)ntohll(ibv_get_device_guid(

			    device_list[i]));

			if (!dev_guid) {

				continue;

			}

			if (dev_guid == guid) {

				matched = B_TRUE;

				rc = do_driver_update_ioctl(device_list[i],

				    cmn_nodedesc, hca_nodedesc, update_flag);

				break;

			} else {

				continue;

			}

		}

		if (matched == B_FALSE) {

			IBERROR("No guid matched.\n");

			/* NOTREACHED */

		}

		if (rc != 0) {

			IBERROR("Failed to set the node descriptor.\n");

			/* NOTREACHED */

		}

	}

	return (rc);

}

static int

read_nodedesc_ioctl(struct ibv_context *context,

    sol_uverbs_nodedesc_t *nodedesc)

{

	int	ret;

	/*

	 * Use ioctl call to sol_uverbs module.

	 */

	if (!context || !nodedesc) {

	ret = ioctl(context->cmd_fd, UVERBS_IOCTL_GET_NODEDESC, nodedesc);

	if (ret != 0) {

		if (ret == EINVAL)

			IBERROR("ABI version check failed.\n");

		else

			IBERROR("UVERBS_IOCTL_GET_NODEDESC ioctl failed.\n");

		/* NOTREACHED */

	return (0);

}

static int

write_nodedesc_ioctl(struct ibv_context *context,

    sol_uverbs_nodedesc_t *nodedesc)

{

	int	ret;

	/*

	 * Use ioctl call to sol_uverbs module.

	 */

	if (!context || !nodedesc)

		return (-1);

	ret = ioctl(context->cmd_fd, UVERBS_IOCTL_SET_NODEDESC, nodedesc);

	if (ret != 0) {

		if (ret == EINVAL)

			IBERROR("ABI version check failed.\n");

		else

			IBERROR("UVERBS_IOCTL_SET_NODEDESC ioctl failed.\n");

		/* NOTREACHED */

	}

	return (0);

	int			rc;

	char			*nodedesc = NULL;

	char			*hcadesc = NULL;

	uint32_t		update_flag = 0;

	struct utsname		uts_name;

	uint64_t		hca_guid;

	boolean_t		guid_inited = B_FALSE;

	extern int 		ibdebug;

	char			nodename[64];

	struct ibv_device	**device_list = NULL;

	int			num_devices = 0;

				goto free_and_ret_2;

			update_flag |= UVERBS_NODEDESC_UPDATE_STRING;