/*

 * Copyright (c) 2004-2009 Voltaire Inc.  All rights reserved.

 * Copyright (c) 2011 Mellanox Technologies LTD.  All rights reserved.

 *

 * This software is available to you under a choice of one of two

 * licenses.  You may choose to be licensed under the terms of the GNU

 * General Public License (GPL) Version 2, available from the file

 * COPYING in the main directory of this source tree, or the

 * OpenIB.org BSD license below:

 *

 *     Redistribution and use in source and binary forms, with or

 *     without modification, are permitted provided that the following

 *     conditions are met:

 *

 *      - Redistributions of source code must retain the above

 *        copyright notice, this list of conditions and the following

 *        disclaimer.

 *

 *      - Redistributions in binary form must reproduce the above

 *        copyright notice, this list of conditions and the following

 *        disclaimer in the documentation and/or other materials

 *        provided with the distribution.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF

 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS

 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN

 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN

 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

 * SOFTWARE.

 *

 */

#if HAVE_CONFIG_H

#  include <config.h>

#endif				/* HAVE_CONFIG_H */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <netinet/in.h>

#define __STDC_FORMAT_MACROS

#include <inttypes.h>

#include <infiniband/umad.h>

#include <infiniband/mad.h>

#include <complib/cl_nodenamemap.h>

#include "ibdiag_common.h"

static struct ibmad_port *srcport;

static op_fn_t node_desc, node_info, port_info, switch_info, pkey_table,

    sl2vl_table, vlarb_table, guid_info, mlnx_ext_port_info, port_info_extended;

static const match_rec_t match_tbl[] = {

	{"NodeInfo", "NI", node_info, 0, ""},

	{"NodeDesc", "ND", node_desc, 0, ""},

	{"PortInfo", "PI", port_info, 1, ""},

	{"PortInfoExtended", "PIE", port_info_extended, 1, ""},

	{"SwitchInfo", "SI", switch_info, 0, ""},

	{"PKeyTable", "PKeys", pkey_table, 1, ""},

	{"SL2VLTable", "SL2VL", sl2vl_table, 1, ""},

	{"VLArbitration", "VLArb", vlarb_table, 1, ""},

	{"GUIDInfo", "GI", guid_info, 0, ""},

	{"MlnxExtPortInfo", "MEPI", mlnx_ext_port_info, 1, ""},

	{}

};

static char *node_name_map_file = NULL;

static nn_map_t *node_name_map = NULL;

static int extended_speeds = 0;

/*******************************************/

static const char *node_desc(ib_portid_t *dest, char **argv, int argc)

{

	int node_type, l;

	uint64_t node_guid;

	char nd[IB_SMP_DATA_SIZE] = { 0 };

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	char dots[128];

	char *nodename = NULL;

	if (!smp_query_via(data, dest, IB_ATTR_NODE_INFO, 0, 0, srcport))

		return "node info query failed";

	mad_decode_field(data, IB_NODE_TYPE_F, &node_type);

	mad_decode_field(data, IB_NODE_GUID_F, &node_guid);

	if (!smp_query_via(nd, dest, IB_ATTR_NODE_DESC, 0, 0, srcport))

		return "node desc query failed";

	nodename = remap_node_name(node_name_map, node_guid, nd);

	l = strlen(nodename);

	if (l < 32) {

		memset(dots, '.', 32 - l);

		dots[32 - l] = '\0';

	} else {

		dots[0] = '.';

		dots[1] = '\0';

	}

	printf("Node Description:%s%s\n", dots, nodename);

	free(nodename);

	return NULL;

}

static const char *node_info(ib_portid_t * dest, char **argv, int argc)

{

	char buf[2048];

	char data[IB_SMP_DATA_SIZE] = { 0 };

	if (!smp_query_via(data, dest, IB_ATTR_NODE_INFO, 0, 0, srcport))

		return "node info query failed";

	mad_dump_nodeinfo(buf, sizeof buf, data, sizeof data);

	printf("# Node info: %s\n%s", portid2str(dest), buf);

	return NULL;

}

static const char *port_info_extended(ib_portid_t *dest, char **argv, int argc)

{

	char buf[2048];

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	int portnum = 0;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	if (!is_port_info_extended_supported(dest, portnum, srcport))

		return "port info extended not supported";

	if (!smp_query_via(data, dest, IB_ATTR_PORT_INFO_EXT, portnum, 0,

			   srcport))

		return "port info extended query failed";

	mad_dump_portinfo_ext(buf, sizeof buf, data, sizeof data);

	printf("# Port info Extended: %s port %d\n%s", portid2str(dest),

	       portnum, buf);

	return NULL;

}

static const char *port_info(ib_portid_t *dest, char **argv, int argc)

{

	char data[IB_SMP_DATA_SIZE] = { 0 };

	int portnum = 0, orig_portnum;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	orig_portnum = portnum;

	if (extended_speeds)

		portnum |= 1 << 31;

	if (!smp_query_via(data, dest, IB_ATTR_PORT_INFO, portnum, 0, srcport))

		return "port info query failed";

	printf("# Port info: %s port %d\n", portid2str(dest), orig_portnum);

	dump_portinfo(data, 0);

	return NULL;

}

static const char *mlnx_ext_port_info(ib_portid_t *dest, char **argv, int argc)

{

	char buf[2300];

	char data[IB_SMP_DATA_SIZE];

	int portnum = 0;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	if (!smp_query_via(data, dest, IB_ATTR_MLNX_EXT_PORT_INFO, portnum, 0, srcport))

		return "Mellanox ext port info query failed";

	mad_dump_mlnx_ext_port_info(buf, sizeof buf, data, sizeof data);

	printf("# MLNX ext Port info: %s port %d\n%s", portid2str(dest), portnum, buf);

	return NULL;

}

static const char *switch_info(ib_portid_t *dest, char **argv, int argc)

{

	char buf[2048];

	char data[IB_SMP_DATA_SIZE] = { 0 };

	if (!smp_query_via(data, dest, IB_ATTR_SWITCH_INFO, 0, 0, srcport))

		return "switch info query failed";

	mad_dump_switchinfo(buf, sizeof buf, data, sizeof data);

	printf("# Switch info: %s\n%s", portid2str(dest), buf);

	return NULL;

}

static const char *pkey_table(ib_portid_t *dest, char **argv, int argc)

{

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	int i, j, k;

	__be16 *p;

	unsigned mod;

	int n, t, phy_ports;

	int portnum = 0;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	/* Get the partition capacity */

	if (!smp_query_via(data, dest, IB_ATTR_NODE_INFO, 0, 0, srcport))

		return "node info query failed";

	mad_decode_field(data, IB_NODE_TYPE_F, &t);

	mad_decode_field(data, IB_NODE_NPORTS_F, &phy_ports);

	if (portnum > phy_ports)

		return "invalid port number";

	if ((t == IB_NODE_SWITCH) && (portnum != 0)) {

		if (!smp_query_via(data, dest, IB_ATTR_SWITCH_INFO, 0, 0,

				   srcport))

			return "switch info failed";

		mad_decode_field(data, IB_SW_PARTITION_ENFORCE_CAP_F, &n);

	} else

		mad_decode_field(data, IB_NODE_PARTITION_CAP_F, &n);

	for (i = 0; i < (n + 31) / 32; i++) {

		mod = i | (portnum << 16);

		if (!smp_query_via(data, dest, IB_ATTR_PKEY_TBL, mod, 0,

				   srcport))

			return "pkey table query failed";

		if (i + 1 == (n + 31) / 32)

			k = ((n + 7 - i * 32) / 8) * 8;

		else

			k = 32;

		p = (__be16 *) data;

		for (j = 0; j < k; j += 8, p += 8) {

			printf

			    ("%4u: 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x 0x%04x\n",

			     (i * 32) + j, ntohs(p[0]), ntohs(p[1]),

			     ntohs(p[2]), ntohs(p[3]), ntohs(p[4]), ntohs(p[5]),

			     ntohs(p[6]), ntohs(p[7]));

		}

	}

	printf("%d pkeys capacity for this port\n", n);

	return NULL;

}

static const char *sl2vl_dump_table_entry(ib_portid_t *dest, int in, int out)

{

	char buf[2048];

	char data[IB_SMP_DATA_SIZE] = { 0 };

	int portnum = (in << 8) | out;

	if (!smp_query_via(data, dest, IB_ATTR_SLVL_TABLE, portnum, 0, srcport))

		return "slvl query failed";

	mad_dump_sltovl(buf, sizeof buf, data, sizeof data);

	printf("ports: in %2d, out %2d: ", in, out);

	printf("%s", buf);

	return NULL;

}

static const char *sl2vl_table(ib_portid_t *dest, char **argv, int argc)

{

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	int type, num_ports, portnum = 0;

	int i;

	const char *ret;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	if (!smp_query_via(data, dest, IB_ATTR_NODE_INFO, 0, 0, srcport))

		return "node info query failed";

	mad_decode_field(data, IB_NODE_TYPE_F, &type);

	mad_decode_field(data, IB_NODE_NPORTS_F, &num_ports);

	if (portnum > num_ports)

		return "invalid port number";

	printf("# SL2VL table: %s\n", portid2str(dest));

	printf("#                 SL: |");

	for (i = 0; i < 16; i++)

		printf("%2d|", i);

	printf("\n");

	if (type != IB_NODE_SWITCH)

		return sl2vl_dump_table_entry(dest, 0, 0);

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

		ret = sl2vl_dump_table_entry(dest, i, portnum);

		if (ret)

			return ret;

	}

	return NULL;

}

static const char *vlarb_dump_table_entry(ib_portid_t *dest, int portnum,

					  int offset, unsigned cap)

{

	char buf[2048];

	char data[IB_SMP_DATA_SIZE] = { 0 };

	if (!smp_query_via(data, dest, IB_ATTR_VL_ARBITRATION,

			   (offset << 16) | portnum, 0, srcport))

		return "vl arb query failed";

	mad_dump_vlarbitration(buf, sizeof(buf), data, cap * 2);

	printf("%s", buf);

	return NULL;

}

static const char *vlarb_dump_table(ib_portid_t *dest, int portnum,

				    const char *name, int offset, int cap)

{

	const char *ret;

	printf("# %s priority VL Arbitration Table:", name);

	ret = vlarb_dump_table_entry(dest, portnum, offset,

				     cap < 32 ? cap : 32);

	if (!ret && cap > 32)

		ret = vlarb_dump_table_entry(dest, portnum, offset + 1,

					     cap - 32);

	return ret;

}

static const char *vlarb_table(ib_portid_t *dest, char **argv, int argc)

{

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	int portnum = 0;

	int type, enhsp0, lowcap, highcap;

	const char *ret = NULL;

	if (argc > 0)

		portnum = strtol(argv[0], NULL, 0);

	/* port number of 0 could mean SP0 or port MAD arrives on */

	if (portnum == 0) {

		if (!smp_query_via(data, dest, IB_ATTR_NODE_INFO, 0, 0,

				   srcport))

			return "node info query failed";

		mad_decode_field(data, IB_NODE_TYPE_F, &type);

		if (type == IB_NODE_SWITCH) {

			memset(data, 0, sizeof(data));

			if (!smp_query_via(data, dest, IB_ATTR_SWITCH_INFO, 0,

					   0, srcport))

				return "switch info query failed";

			mad_decode_field(data, IB_SW_ENHANCED_PORT0_F, &enhsp0);

			if (!enhsp0) {

				printf

				    ("# No VLArbitration tables (BSP0): %s port %d\n",

				     portid2str(dest), 0);

				return NULL;

			}

			memset(data, 0, sizeof(data));

		}

	}

	if (!smp_query_via(data, dest, IB_ATTR_PORT_INFO, portnum, 0, srcport))

		return "port info query failed";

	mad_decode_field(data, IB_PORT_VL_ARBITRATION_LOW_CAP_F, &lowcap);

	mad_decode_field(data, IB_PORT_VL_ARBITRATION_HIGH_CAP_F, &highcap);

	printf("# VLArbitration tables: %s port %d LowCap %d HighCap %d\n",

	       portid2str(dest), portnum, lowcap, highcap);

	if (lowcap > 0)

		ret = vlarb_dump_table(dest, portnum, "Low", 1, lowcap);

	if (!ret && highcap > 0)

		ret = vlarb_dump_table(dest, portnum, "High", 3, highcap);

	return ret;

}

static const char *guid_info(ib_portid_t *dest, char **argv, int argc)

{

	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };

	int i, j, k;

	__be64 *p;

	unsigned mod;

	int n;

	/* Get the guid capacity */

	if (!smp_query_via(data, dest, IB_ATTR_PORT_INFO, 0, 0, srcport))

		return "port info failed";

	mad_decode_field(data, IB_PORT_GUID_CAP_F, &n);

	for (i = 0; i < (n + 7) / 8; i++) {

		mod = i;

		if (!smp_query_via(data, dest, IB_ATTR_GUID_INFO, mod, 0,

				   srcport))

			return "guid info query failed";

		if (i + 1 == (n + 7) / 8)

			k = ((n + 1 - i * 8) / 2) * 2;

		else

			k = 8;

		p = (__be64 *) data;

		for (j = 0; j < k; j += 2, p += 2) {

			printf("%4u: 0x%016" PRIx64 " 0x%016" PRIx64 "\n",

			       (i * 8) + j, be64toh(p[0]), be64toh(p[1]));

		}

	}

	printf("%d guids capacity for this port\n", n);

	return NULL;

}

static int process_opt(void *context, int ch)

{

	switch (ch) {

	case 1:

		node_name_map_file = strdup(optarg);

		if (node_name_map_file == NULL)

			IBEXIT("out of memory, strdup for node_name_map_file name failed");

		break;

	case 'c':

		ibd_dest_type = IB_DEST_DRSLID;

		break;

	case 'x':

		extended_speeds = 1;

		break;

	default:

		return -1;

	}

	return 0;

}

int main(int argc, char **argv)

{

	char usage_args[1024];

	int mgmt_classes[3] =

	    { IB_SMI_CLASS, IB_SMI_DIRECT_CLASS, IB_SA_CLASS };

	ib_portid_t portid = { 0 };

	const char *err;

	op_fn_t *fn;

	const match_rec_t *r;

	int n;

	const struct ibdiag_opt opts[] = {

		{"combined", 'c', 0, NULL,

		 "use Combined route address argument"},

		{"node-name-map", 1, 1, "<file>", "node name map file"},

		{"extended", 'x', 0, NULL, "use extended speeds"},

		{}

	};

	const char *usage_examples[] = {

		"portinfo 3 1\t\t\t\t# portinfo by lid, with port modifier",

		"-G switchinfo 0x2C9000100D051 1\t# switchinfo by guid",

		"-D nodeinfo 0\t\t\t\t# nodeinfo by direct route",

		"-c nodeinfo 6 0,12\t\t\t# nodeinfo by combined route",

		NULL

	};

	n = sprintf(usage_args, "<op> <dest dr_path|lid|guid> [op params]\n"

		    "\nSupported ops (and aliases, case insensitive):\n");

	for (r = match_tbl; r->name; r++) {

		n += snprintf(usage_args + n, sizeof(usage_args) - n,

			      "  %s (%s) <addr>%s\n", r->name,

			      r->alias ? r->alias : "",

			      r->opt_portnum ? " [<portnum>]" : "");

		if (n >= sizeof(usage_args))

			exit(-1);

	}

	ibdiag_process_opts(argc, argv, NULL, NULL, opts, process_opt,

			    usage_args, usage_examples);

	argc -= optind;

	argv += optind;

	if (argc < 2)

		ibdiag_show_usage();

	if (!(fn = match_op(match_tbl, argv[0])))

		IBEXIT("operation '%s' not supported", argv[0]);

	srcport = mad_rpc_open_port(ibd_ca, ibd_ca_port, mgmt_classes, 3);

	if (!srcport)

		IBEXIT("Failed to open '%s' port '%d'", ibd_ca, ibd_ca_port);

	smp_mkey_set(srcport, ibd_mkey);

	node_name_map = open_node_name_map(node_name_map_file);

	if (ibd_dest_type != IB_DEST_DRSLID) {

		if (resolve_portid_str(ibd_ca, ibd_ca_port, &portid, argv[1],

				       ibd_dest_type, ibd_sm_id, srcport) < 0)

			IBEXIT("can't resolve destination port %s", argv[1]);

		if ((err = fn(&portid, argv + 2, argc - 2)))

			IBEXIT("operation %s: %s", argv[0], err);

	} else {

		char concat[64];

		memset(concat, 0, 64);

		snprintf(concat, sizeof(concat), "%s %s", argv[1], argv[2]);

		if (resolve_portid_str(ibd_ca, ibd_ca_port, &portid, concat,

				       ibd_dest_type, ibd_sm_id, srcport) < 0)

			IBEXIT("can't resolve destination port %s", concat);

		if ((err = fn(&portid, argv + 3, argc - 3)))

			IBEXIT("operation %s: %s", argv[0], err);

	}

	close_node_name_map(node_name_map);

	mad_rpc_close_port(srcport);

	exit(0);

}