1/*
2 * Copyright (c) 2008 Lawrence Livermore National Security
3 * Copyright (c) 2008-2009 Voltaire Inc.  All rights reserved.
4 * Copyright (c) 2009 HNR Consulting.  All rights reserved.
5 * Copyright (c) 2011 Mellanox Technologies LTD.  All rights reserved.
6 *
7 * Produced at Lawrence Livermore National Laboratory.
8 * Written by Ira Weiny <weiny2@llnl.gov>.
9 *
10 * This software is available to you under a choice of one of two
11 * licenses.  You may choose to be licensed under the terms of the GNU
12 * General Public License (GPL) Version 2, available from the file
13 * COPYING in the main directory of this source tree, or the
14 * OpenIB.org BSD license below:
15 *
16 *     Redistribution and use in source and binary forms, with or
17 *     without modification, are permitted provided that the following
18 *     conditions are met:
19 *
20 *      - Redistributions of source code must retain the above
21 *        copyright notice, this list of conditions and the following
22 *        disclaimer.
23 *
24 *      - Redistributions in binary form must reproduce the above
25 *        copyright notice, this list of conditions and the following
26 *        disclaimer in the documentation and/or other materials
27 *        provided with the distribution.
28 *
29 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
30 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
31 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
32 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
33 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
34 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
35 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
36 * SOFTWARE.
37 *
38 */
39
40#include <stdlib.h>
41#include <stdio.h>
42#include <unistd.h>
43#include <string.h>
44
45#define _GNU_SOURCE
46#include <getopt.h>
47
48#include <infiniband/mad.h>
49#include <iba/ib_types.h>
50
51#include "ibdiag_common.h"
52
53struct ibmad_port *srcport;
54/* for local link integrity */
55int error_port = 1;
56
57static uint16_t get_node_type(ib_portid_t * port)
58{
59	uint16_t node_type = IB_NODE_TYPE_CA;
60	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };
61
62	if (smp_query_via(data, port, IB_ATTR_NODE_INFO, 0, 0, srcport))
63		node_type = (uint16_t) mad_get_field(data, 0, IB_NODE_TYPE_F);
64	return node_type;
65}
66
67static uint32_t get_cap_mask(ib_portid_t * port)
68{
69	uint8_t data[IB_SMP_DATA_SIZE] = { 0 };
70	uint32_t cap_mask = 0;
71
72	if (smp_query_via(data, port, IB_ATTR_PORT_INFO, 0, 0, srcport))
73		cap_mask = (uint32_t) mad_get_field(data, 0, IB_PORT_CAPMASK_F);
74	return cap_mask;
75}
76
77static void build_trap145(ib_mad_notice_attr_t * n, ib_portid_t * port)
78{
79	n->generic_type = 0x80 | IB_NOTICE_TYPE_INFO;
80	n->g_or_v.generic.prod_type_lsb = cl_hton16(get_node_type(port));
81	n->g_or_v.generic.trap_num = cl_hton16(145);
82	n->issuer_lid = cl_hton16((uint16_t) port->lid);
83	n->data_details.ntc_145.new_sys_guid = cl_hton64(0x1234567812345678);
84}
85
86static void build_trap144_local(ib_mad_notice_attr_t * n, ib_portid_t * port)
87{
88	n->generic_type = 0x80 | IB_NOTICE_TYPE_INFO;
89	n->g_or_v.generic.prod_type_lsb = cl_hton16(get_node_type(port));
90	n->g_or_v.generic.trap_num = cl_hton16(144);
91	n->issuer_lid = cl_hton16((uint16_t) port->lid);
92	n->data_details.ntc_144.lid = n->issuer_lid;
93	n->data_details.ntc_144.new_cap_mask = cl_hton32(get_cap_mask(port));
94	n->data_details.ntc_144.local_changes =
95	    TRAP_144_MASK_OTHER_LOCAL_CHANGES;
96}
97
98static void build_trap144_nodedesc(ib_mad_notice_attr_t * n, ib_portid_t * port)
99{
100	build_trap144_local(n, port);
101	n->data_details.ntc_144.change_flgs =
102	    TRAP_144_MASK_NODE_DESCRIPTION_CHANGE;
103}
104
105static void build_trap144_linkspeed(ib_mad_notice_attr_t * n,
106				    ib_portid_t * port)
107{
108	build_trap144_local(n, port);
109	n->data_details.ntc_144.change_flgs =
110	    TRAP_144_MASK_LINK_SPEED_ENABLE_CHANGE;
111}
112
113static void build_trap129(ib_mad_notice_attr_t * n, ib_portid_t * port)
114{
115	n->generic_type = 0x80 | IB_NOTICE_TYPE_URGENT;
116	n->g_or_v.generic.prod_type_lsb = cl_hton16(get_node_type(port));
117	n->g_or_v.generic.trap_num = cl_hton16(129);
118	n->issuer_lid = cl_hton16((uint16_t) port->lid);
119	n->data_details.ntc_129_131.lid = n->issuer_lid;
120	n->data_details.ntc_129_131.pad = 0;
121	n->data_details.ntc_129_131.port_num = (uint8_t) error_port;
122}
123
124static void build_trap256_local(ib_mad_notice_attr_t * n, ib_portid_t * port)
125{
126	n->generic_type = 0x80 | IB_NOTICE_TYPE_SECURITY;
127	n->g_or_v.generic.prod_type_lsb = cl_hton16(get_node_type(port));
128	n->g_or_v.generic.trap_num = cl_hton16(256);
129	n->issuer_lid = cl_hton16((uint16_t) port->lid);
130	n->data_details.ntc_256.lid = n->issuer_lid;
131	n->data_details.ntc_256.dr_slid = 0xffff;
132	n->data_details.ntc_256.method = 1;
133	n->data_details.ntc_256.attr_id = cl_ntoh16(0x15);
134	n->data_details.ntc_256.attr_mod = cl_ntoh32(0x12);
135	n->data_details.ntc_256.mkey = cl_ntoh64(0x1234567812345678);
136}
137
138static void build_trap256_lid(ib_mad_notice_attr_t * n, ib_portid_t * port)
139{
140	build_trap256_local(n, port);
141	n->data_details.ntc_256.dr_trunc_hop = 0;
142}
143
144static void build_trap256_dr(ib_mad_notice_attr_t * n, ib_portid_t * port)
145{
146	build_trap256_local(n, port);
147	n->data_details.ntc_256.dr_trunc_hop = 0x80 | 0x4;
148	n->data_details.ntc_256.dr_rtn_path[0] = 5;
149	n->data_details.ntc_256.dr_rtn_path[1] = 6;
150	n->data_details.ntc_256.dr_rtn_path[2] = 7;
151	n->data_details.ntc_256.dr_rtn_path[3] = 8;
152}
153
154static void build_trap257_258(ib_mad_notice_attr_t * n, ib_portid_t * port,
155			      uint16_t trap_num)
156{
157	n->generic_type = 0x80 | IB_NOTICE_TYPE_SECURITY;
158	n->g_or_v.generic.prod_type_lsb = cl_hton16(get_node_type(port));
159	n->g_or_v.generic.trap_num = cl_hton16(trap_num);
160	n->issuer_lid = cl_hton16((uint16_t) port->lid);
161	n->data_details.ntc_257_258.lid1 = cl_hton16(1);
162	n->data_details.ntc_257_258.lid2 = cl_hton16(2);
163	n->data_details.ntc_257_258.key = cl_hton32(0x12345678);
164	n->data_details.ntc_257_258.qp1 = cl_hton32(0x010101);
165	n->data_details.ntc_257_258.qp2 = cl_hton32(0x020202);
166	n->data_details.ntc_257_258.gid1.unicast.prefix = cl_ntoh64(0xf8c0000000000001);
167	n->data_details.ntc_257_258.gid1.unicast.interface_id = cl_ntoh64(0x1111222233334444);
168	n->data_details.ntc_257_258.gid2.unicast.prefix = cl_ntoh64(0xf8c0000000000001);
169	n->data_details.ntc_257_258.gid2.unicast.interface_id = cl_ntoh64(0x5678567812341234);
170}
171
172static void build_trap257(ib_mad_notice_attr_t * n, ib_portid_t * port)
173{
174	build_trap257_258(n, port, 257);
175}
176
177static void build_trap258(ib_mad_notice_attr_t * n, ib_portid_t * port)
178{
179	build_trap257_258(n, port, 258);
180}
181
182static int send_trap(void (*build) (ib_mad_notice_attr_t *, ib_portid_t *))
183{
184	ib_portid_t sm_port;
185	ib_portid_t selfportid;
186	int selfport;
187	ib_rpc_t trap_rpc;
188	ib_mad_notice_attr_t notice;
189
190	if (resolve_self(ibd_ca, ibd_ca_port, &selfportid, &selfport, NULL))
191		IBEXIT("can't resolve self");
192
193	if (resolve_sm_portid(ibd_ca, ibd_ca_port, &sm_port))
194		IBEXIT("can't resolve SM destination port");
195
196	memset(&trap_rpc, 0, sizeof(trap_rpc));
197	trap_rpc.mgtclass = IB_SMI_CLASS;
198	trap_rpc.method = IB_MAD_METHOD_TRAP;
199	trap_rpc.trid = mad_trid();
200	trap_rpc.attr.id = NOTICE;
201	trap_rpc.datasz = IB_SMP_DATA_SIZE;
202	trap_rpc.dataoffs = IB_SMP_DATA_OFFS;
203
204	memset(&notice, 0, sizeof(notice));
205	build(&notice, &selfportid);
206
207	return mad_send_via(&trap_rpc, &sm_port, NULL, &notice, srcport);
208}
209
210typedef struct _trap_def {
211	char *trap_name;
212	void (*build_func) (ib_mad_notice_attr_t *, ib_portid_t *);
213} trap_def_t;
214
215static const trap_def_t traps[] = {
216	{"node_desc_change", build_trap144_nodedesc},
217	{"link_speed_enabled_change", build_trap144_linkspeed},
218	{"local_link_integrity", build_trap129},
219	{"sys_image_guid_change", build_trap145},
220	{"mkey_lid", build_trap256_lid},
221	{"mkey_dr", build_trap256_dr},
222	{"pkey", build_trap257},
223	{"qkey", build_trap258},
224	{NULL, NULL}
225};
226
227int process_send_trap(char *trap_name)
228{
229	int i;
230
231	for (i = 0; traps[i].trap_name; i++)
232		if (strcmp(traps[i].trap_name, trap_name) == 0)
233			return send_trap(traps[i].build_func);
234	ibdiag_show_usage();
235	return 1;
236}
237
238int main(int argc, char **argv)
239{
240	char usage_args[1024];
241	int mgmt_classes[2] = { IB_SMI_CLASS, IB_SMI_DIRECT_CLASS };
242	char *trap_name = NULL;
243	int i, n, rc;
244
245	n = sprintf(usage_args, "[<trap_name>] [<error_port>]\n"
246		    "\nArgument <trap_name> can be one of the following:\n");
247	for (i = 0; traps[i].trap_name; i++) {
248		n += snprintf(usage_args + n, sizeof(usage_args) - n,
249			      "  %s\n", traps[i].trap_name);
250		if (n >= sizeof(usage_args))
251			exit(-1);
252	}
253	snprintf(usage_args + n, sizeof(usage_args) - n,
254		 "\n  default behavior is to send \"%s\"", traps[0].trap_name);
255
256	ibdiag_process_opts(argc, argv, NULL, "DGKL", NULL, NULL,
257			    usage_args, NULL);
258
259	argc -= optind;
260	argv += optind;
261
262	trap_name = argv[0] ? argv[0] : traps[0].trap_name;
263
264	if (argc > 1)
265		error_port = atoi(argv[1]);
266
267	madrpc_show_errors(1);
268
269	srcport = mad_rpc_open_port(ibd_ca, ibd_ca_port, mgmt_classes, 2);
270	if (!srcport)
271		IBEXIT("Failed to open '%s' port '%d'", ibd_ca, ibd_ca_port);
272
273	smp_mkey_set(srcport, ibd_mkey);
274
275	rc = process_send_trap(trap_name);
276	mad_rpc_close_port(srcport);
277	return rc;
278}
279