/*

 * Copyright (c) 2006-2008 Voltaire, Inc. All rights reserved.

 * Copyright (c) 2002-2006 Mellanox Technologies LTD. All rights reserved.

 * Copyright (c) 1996-2003 Intel Corporation. All rights reserved.

 * Copyright (c) 2010 HNR Consulting. 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.

 *

 */

/*

 * Abstract:

 *    Implementation of service records testing flow..

 *    Top level is osmt_run_service_records_flow:

 *     osmt_register_service

 *     osmt_get_service_by_name

 *     osmt_get_all_services

 *     osmt_delete_service_by_name

 *

 */

#ifndef __WIN__

#include <unistd.h>

#else

#include <time.h>

#endif

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <complib/cl_debug.h>

#include "osmtest.h"

ib_api_status_t

osmt_register_service(IN osmtest_t * const p_osmt,

		      IN ib_net64_t service_id,

		      IN ib_net16_t service_pkey,

		      IN ib_net32_t service_lease,

		      IN uint8_t service_key_lsb, IN char *service_name)

{

	osmv_query_req_t req;

	osmv_user_query_t user;

	osmtest_req_context_t context;

	ib_service_record_t svc_rec;

	osm_log_t *p_log = &p_osmt->log;

	ib_api_status_t status;

	OSM_LOG_ENTER(p_log);

	OSM_LOG(&p_osmt->log, OSM_LOG_INFO,

		"Registering service: name: %s id: 0x%" PRIx64 "\n",

		service_name, cl_ntoh64(service_id));

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	memset(&user, 0, sizeof(user));

	memset(&svc_rec, 0, sizeof(svc_rec));

	/* set the new service record fields */

	svc_rec.service_id = service_id;

	svc_rec.service_pkey = service_pkey;

	svc_rec.service_gid.unicast.prefix = p_osmt->local_port_gid.unicast.prefix;

	svc_rec.service_gid.unicast.interface_id = p_osmt->local_port.port_guid;

	svc_rec.service_lease = service_lease;

	memset(&svc_rec.service_key, 0, 16 * sizeof(uint8_t));

	svc_rec.service_key[0] = service_key_lsb;

	memset(svc_rec.service_name, 0, sizeof(svc_rec.service_name));

	memcpy(svc_rec.service_name, service_name,

	       (strlen(service_name) + 1) * sizeof(char));

	/* prepare the data used for this query */

	/*  sa_mad_data.method = IB_MAD_METHOD_SET; */

	/*  sa_mad_data.sm_key = 0; */

	context.p_osmt = p_osmt;

	req.query_context = &context;

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.p_query_input = &use;;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.sm_key = 0;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	user.method = IB_MAD_METHOD_SET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	if (ib_pkey_is_invalid(service_pkey)) {

		/* if given an invalid service_pkey - don't turn the PKEY compmask on */

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY | IB_SR_COMPMASK_SNAME;

	} else {

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SPKEY |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY | IB_SR_COMPMASK_SNAME;

	}

	user.p_attr = &svc_rec;

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A01: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A02: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		if (status == IB_REMOTE_ERROR) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"Remote error = %s\n",

				ib_get_mad_status_str(osm_madw_get_mad_ptr

						      (context.result.

						       p_result_madw)));

		}

		goto Exit;

	}

Exit:

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_register_service_with_full_key(IN osmtest_t * const p_osmt,

				    IN ib_net64_t service_id,

				    IN ib_net16_t service_pkey,

				    IN ib_net32_t service_lease,

				    IN uint8_t * service_key,

				    IN char *service_name)

{

	osmv_query_req_t req;

	osmv_user_query_t user;

	osmtest_req_context_t context;

	ib_service_record_t svc_rec, *p_rec;

	osm_log_t *p_log = &p_osmt->log;

	ib_api_status_t status;

	uint8_t i, skey[16];

	OSM_LOG_ENTER(p_log);

	OSM_LOG(&p_osmt->log, OSM_LOG_INFO,

		"Registering service: name: %s id: 0x%" PRIx64 "\n",

		service_name, cl_ntoh64(service_id));

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	memset(&user, 0, sizeof(user));

	memset(&svc_rec, 0, sizeof(svc_rec));

	/* set the new service record fields */

	svc_rec.service_id = service_id;

	svc_rec.service_pkey = service_pkey;

	svc_rec.service_gid.unicast.prefix = p_osmt->local_port_gid.unicast.prefix;

	svc_rec.service_gid.unicast.interface_id = p_osmt->local_port.port_guid;

	svc_rec.service_lease = service_lease;

	memset(&svc_rec.service_key, 0, 16 * sizeof(uint8_t));

	memcpy(svc_rec.service_key, service_key, 16 * sizeof(uint8_t));

	memset(svc_rec.service_name, 0, sizeof(svc_rec.service_name));

	memset(skey, 0, 16 * sizeof(uint8_t));

	memcpy(svc_rec.service_name, service_name,

	       (strlen(service_name) + 1) * sizeof(char));

	/* prepare the data used for this query */

	/*  sa_mad_data.method = IB_MAD_METHOD_SET; */

	/*  sa_mad_data.sm_key = 0; */

	context.p_osmt = p_osmt;

	req.query_context = &context;

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.p_query_input = &use;;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.sm_key = 0;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	user.method = IB_MAD_METHOD_SET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	if (ib_pkey_is_invalid(service_pkey)) {

		/* if given an invalid service_pkey - don't turn the PKEY compmask on */

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY | IB_SR_COMPMASK_SNAME;

	} else {

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SPKEY |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY | IB_SR_COMPMASK_SNAME;

	}

	user.p_attr = &svc_rec;

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A03: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A04: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		if (status == IB_REMOTE_ERROR) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"Remote error = %s\n",

				ib_get_mad_status_str(osm_madw_get_mad_ptr

						      (context.result.

						       p_result_madw)));

		}

		goto Exit;

	}

	/*  Check service key on context to see if match */

	p_rec = osmv_get_query_svc_rec(context.result.p_result_madw, 0);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Comparing service key...\n" "return key is:\n");

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

		OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

			"service_key sent[%u] = %u, service_key returned[%u] = %u\n",

			i, service_key[i], i, p_rec->service_key[i]);

	}

	/*  since c15-0.1.14 not supported all key association queries should bring in return zero in service key */

	if (memcmp(skey, p_rec->service_key, 16 * sizeof(uint8_t)) != 0) {

		status = IB_REMOTE_ERROR;

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A33: "

			"Data mismatch in service_key\n");

		goto Exit;

	}

Exit:

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_register_service_with_data(IN osmtest_t * const p_osmt,

				IN ib_net64_t service_id,

				IN ib_net16_t service_pkey,

				IN ib_net32_t service_lease,

				IN uint8_t service_key_lsb,

				IN uint8_t * service_data8,

				IN ib_net16_t * service_data16,

				IN ib_net32_t * service_data32,

				IN ib_net64_t * service_data64,

				IN char *service_name)

{

	osmv_query_req_t req;

	osmv_user_query_t user;

	osmtest_req_context_t context;

	ib_service_record_t svc_rec, *p_rec;

	osm_log_t *p_log = &p_osmt->log;

	ib_api_status_t status;

	/*   ib_service_record_t* p_rec; */

	OSM_LOG_ENTER(p_log);

	OSM_LOG(&p_osmt->log, OSM_LOG_INFO,

		"Registering service: name: %s id: 0x%" PRIx64 "\n",

		service_name, cl_ntoh64(service_id));

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	memset(&user, 0, sizeof(user));

	memset(&svc_rec, 0, sizeof(svc_rec));

	/* set the new service record fields */

	svc_rec.service_id = service_id;

	svc_rec.service_pkey = service_pkey;

	svc_rec.service_gid.unicast.prefix = p_osmt->local_port_gid.unicast.prefix;

	svc_rec.service_gid.unicast.interface_id = p_osmt->local_port.port_guid;

	svc_rec.service_lease = service_lease;

	memset(&svc_rec.service_key, 0, 16 * sizeof(uint8_t));

	svc_rec.service_key[0] = service_key_lsb;

	/*  Copy data to service_data arrays */

	memcpy(svc_rec.service_data8, service_data8, 16 * sizeof(uint8_t));

	memcpy(svc_rec.service_data16, service_data16, 8 * sizeof(ib_net16_t));

	memcpy(svc_rec.service_data32, service_data32, 4 * sizeof(ib_net32_t));

	memcpy(svc_rec.service_data64, service_data64, 2 * sizeof(ib_net64_t));

	memset(svc_rec.service_name, 0, sizeof(svc_rec.service_name));

	memcpy(svc_rec.service_name, service_name,

	       (strlen(service_name) + 1) * sizeof(char));

	/* prepare the data used for this query */

	/*  sa_mad_data.method = IB_MAD_METHOD_SET; */

	/*  sa_mad_data.sm_key = 0; */

	context.p_osmt = p_osmt;

	req.query_context = &context;

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.p_query_input = &use;;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.sm_key = 0;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	user.method = IB_MAD_METHOD_SET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	if (ib_pkey_is_invalid(service_pkey)) {

		/* if given an invalid service_pkey - don't turn the PKEY compmask on */

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY |

		    IB_SR_COMPMASK_SNAME |

		    IB_SR_COMPMASK_SDATA8_0 |

		    IB_SR_COMPMASK_SDATA8_1 |

		    IB_SR_COMPMASK_SDATA16_0 |

		    IB_SR_COMPMASK_SDATA16_1 |

		    IB_SR_COMPMASK_SDATA32_0 |

		    IB_SR_COMPMASK_SDATA32_1 |

		    IB_SR_COMPMASK_SDATA64_0 | IB_SR_COMPMASK_SDATA64_1;

	} else {

		user.comp_mask = IB_SR_COMPMASK_SID |

		    IB_SR_COMPMASK_SGID |

		    IB_SR_COMPMASK_SPKEY |

		    IB_SR_COMPMASK_SLEASE |

		    IB_SR_COMPMASK_SKEY |

		    IB_SR_COMPMASK_SNAME |

		    IB_SR_COMPMASK_SDATA8_0 |

		    IB_SR_COMPMASK_SDATA8_1 |

		    IB_SR_COMPMASK_SDATA16_0 |

		    IB_SR_COMPMASK_SDATA16_1 |

		    IB_SR_COMPMASK_SDATA32_0 |

		    IB_SR_COMPMASK_SDATA32_1 |

		    IB_SR_COMPMASK_SDATA64_0 | IB_SR_COMPMASK_SDATA64_1;

	}

	user.p_attr = &svc_rec;

	/*  Dump to Service Data b4 send */

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Dumping service data b4 send\n");

	osm_dump_service_record(&p_osmt->log, &svc_rec, OSM_LOG_VERBOSE);

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A05: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A06: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		if (status == IB_REMOTE_ERROR) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"Remote error = %s\n",

				ib_get_mad_status_str(osm_madw_get_mad_ptr

						      (context.result.

						       p_result_madw)));

		}

		goto Exit;

	}

	/*  Check data on context to see if match */

	p_rec = osmv_get_query_svc_rec(context.result.p_result_madw, 0);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE, "Comparing service data...\n");

	if (memcmp(service_data8, p_rec->service_data8, 16 * sizeof(uint8_t)) !=

	    0

	    || memcmp(service_data16, p_rec->service_data16,

		      8 * sizeof(uint16_t)) != 0

	    || memcmp(service_data32, p_rec->service_data32,

		      4 * sizeof(uint32_t)) != 0

	    || memcmp(service_data64, p_rec->service_data64,

		      2 * sizeof(uint64_t)) != 0) {

		status = IB_REMOTE_ERROR;

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

			"Data mismatch in service_data8\n");

		goto Exit;

	}

Exit:

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_get_service_by_id_and_name(IN osmtest_t * const p_osmt,

				IN uint32_t rec_num,

				IN ib_net64_t sid,

				IN char *sr_name,

				OUT ib_service_record_t * p_out_rec)

{

	ib_api_status_t status = IB_SUCCESS;

	osmtest_req_context_t context;

	osmv_query_req_t req;

	ib_service_record_t svc_rec, *p_rec;

	uint32_t num_recs = 0;

	osmv_user_query_t user;

	const uint16_t ERR_SIZE = 512;

	char mad_stat_err[ERR_SIZE];

	OSM_LOG_ENTER(&p_osmt->log);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Getting service record: id: 0x%016" PRIx64

		" and name: %s\n", cl_ntoh64(sid), sr_name);

	/*

	 * Do a blocking query for this record in the subnet.

	 * The result is returned in the result field of the caller's

	 * context structure.

	 *

	 * The query structures are locals.

	 */

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	context.p_osmt = p_osmt;

	/* prepare the data used for this query */

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	req.retry_cnt = p_osmt->opt.retry_count;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.query_context = &context;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.sm_key = 0;

	memset(&svc_rec, 0, sizeof(svc_rec));

	memset(&user, 0, sizeof(user));

	/* set the new service record fields */

	memset(svc_rec.service_name, 0, sizeof(svc_rec.service_name));

	memcpy(svc_rec.service_name, sr_name,

	       (strlen(sr_name) + 1) * sizeof(char));

	svc_rec.service_id = sid;

	req.p_query_input = &use;;

	user.method = IB_MAD_METHOD_GET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	user.comp_mask = IB_SR_COMPMASK_SID | IB_SR_COMPMASK_SNAME;

	user.p_attr = &svc_rec;

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A07: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	num_recs = context.result.result_cnt;

	if (status != IB_SUCCESS) {

		/* If the failure is due to IB_SA_MAD_STATUS_NO_RECORDS and rec_num is 0,

		   then this is fine */

		if (status == IB_REMOTE_ERROR)

			strncpy(mad_stat_err,

			        ib_get_mad_status_str(osm_madw_get_mad_ptr

						      (context.result.

						       p_result_madw)),

				ERR_SIZE -1);

		else

			strncpy(mad_stat_err, ib_get_err_str(status),

				ERR_SIZE -1);

		mad_stat_err[ERR_SIZE -1] = '\0';

		if (status == IB_REMOTE_ERROR &&

		    !strcmp(mad_stat_err, "IB_SA_MAD_STATUS_NO_RECORDS") &&

		    rec_num == 0) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"IS EXPECTED ERROR ^^^^\n");

			status = IB_SUCCESS;

		} else {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A08: "

				"Query failed: %s (%s)\n",

				ib_get_err_str(status), mad_stat_err);

			goto Exit;

		}

	}

	if (rec_num && num_recs != rec_num) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

			"Unmatched number of records: expected: %d, received: %d\n",

			rec_num, num_recs);

		status = IB_REMOTE_ERROR;

		goto Exit;

	}

	p_rec = osmv_get_query_svc_rec(context.result.p_result_madw, 0);

	*p_out_rec = *p_rec;

	if (num_recs) {

		OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

			"Found service record: name: %s id: 0x%016" PRIx64 "\n",

			p_rec->service_name, cl_ntoh64(p_rec->service_id));

		osm_dump_service_record(&p_osmt->log, p_rec, OSM_LOG_DEBUG);

	}

Exit:

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Expected and found %d records\n", rec_num);

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_get_service_by_id(IN osmtest_t * const p_osmt,

		       IN uint32_t rec_num,

		       IN ib_net64_t sid, OUT ib_service_record_t * p_out_rec)

{

	ib_api_status_t status = IB_SUCCESS;

	osmtest_req_context_t context;

	osmv_query_req_t req;

	ib_service_record_t svc_rec, *p_rec;

	uint32_t num_recs = 0;

	osmv_user_query_t user;

	OSM_LOG_ENTER(&p_osmt->log);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Getting service record: id: 0x%016" PRIx64 "\n",

		cl_ntoh64(sid));

	/*

	 * Do a blocking query for this record in the subnet.

	 * The result is returned in the result field of the caller's

	 * context structure.

	 *

	 * The query structures are locals.

	 */

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	context.p_osmt = p_osmt;

	/* prepare the data used for this query */

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	req.retry_cnt = p_osmt->opt.retry_count;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.query_context = &context;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.sm_key = 0;

	memset(&svc_rec, 0, sizeof(svc_rec));

	memset(&user, 0, sizeof(user));

	/* set the new service record fields */

	svc_rec.service_id = sid;

	req.p_query_input = &use;;

	user.method = IB_MAD_METHOD_GET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	user.comp_mask = IB_SR_COMPMASK_SID;

	user.p_attr = &svc_rec;

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A09: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	num_recs = context.result.result_cnt;

	if (status != IB_SUCCESS) {

		char mad_stat_err[256];

		/* If the failure is due to IB_SA_MAD_STATUS_NO_RECORDS and rec_num is 0,

		   then this is fine */

		if (status == IB_REMOTE_ERROR)

			strcpy(mad_stat_err,

			       ib_get_mad_status_str(osm_madw_get_mad_ptr

						     (context.result.

						      p_result_madw)));

		else

			strcpy(mad_stat_err, ib_get_err_str(status));

		if (status == IB_REMOTE_ERROR &&

		    !strcmp(mad_stat_err, "IB_SA_MAD_STATUS_NO_RECORDS") &&

		    rec_num == 0) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"IS EXPECTED ERROR ^^^^\n");

			status = IB_SUCCESS;

		} else {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A0A: "

				"Query failed: %s (%s)\n",

				ib_get_err_str(status), mad_stat_err);

			goto Exit;

		}

	}

	if (rec_num && num_recs != rec_num) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A0B: "

			"Unmatched number of records: expected: %d received: %d\n",

			rec_num, num_recs);

		status = IB_REMOTE_ERROR;

		goto Exit;

	}

	p_rec = osmv_get_query_svc_rec(context.result.p_result_madw, 0);

	*p_out_rec = *p_rec;

	if (num_recs) {

		OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

			"Found service record: name: %s id: 0x%016" PRIx64 "\n",

			p_rec->service_name, cl_ntoh64(p_rec->service_id));

		osm_dump_service_record(&p_osmt->log, p_rec, OSM_LOG_DEBUG);

	}

Exit:

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Expected and found %d records\n", rec_num);

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_get_service_by_name_and_key(IN osmtest_t * const p_osmt,

				 IN char *sr_name,

				 IN uint32_t rec_num,

				 IN uint8_t * skey,

				 OUT ib_service_record_t * p_out_rec)

{

	ib_api_status_t status = IB_SUCCESS;

	osmtest_req_context_t context;

	osmv_query_req_t req;

	ib_service_record_t svc_rec, *p_rec;

	uint32_t num_recs = 0, i;

	osmv_user_query_t user;

	const uint16_t ERR_SIZE = 512;

	char mad_stat_err[ERR_SIZE];

	OSM_LOG_ENTER(&p_osmt->log);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Getting service record: name: %s and key: "

		"0x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x\n",

		sr_name, skey[0], skey[1], skey[2], skey[3], skey[4], skey[5],

		skey[6], skey[7], skey[8], skey[9], skey[10], skey[11],

		skey[12], skey[13], skey[14], skey[15]);

	/*

	 * Do a blocking query for this record in the subnet.

	 * The result is returned in the result field of the caller's

	 * context structure.

	 *

	 * The query structures are locals.

	 */

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	context.p_osmt = p_osmt;

	/* prepare the data used for this query */

	req.query_type = OSMV_QUERY_USER_DEFINED;

	req.timeout_ms = p_osmt->opt.transaction_timeout;

	req.retry_cnt = p_osmt->opt.retry_count;

	req.flags = OSM_SA_FLAGS_SYNC;

	req.query_context = &context;

	req.pfn_query_cb = osmtest_query_res_cb;

	req.sm_key = 0;

	memset(&svc_rec, 0, sizeof(svc_rec));

	memset(&user, 0, sizeof(user));

	/* set the new service record fields */

	memset(svc_rec.service_name, 0, sizeof(svc_rec.service_name));

	memcpy(svc_rec.service_name, sr_name,

	       (strlen(sr_name) + 1) * sizeof(char));

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

		svc_rec.service_key[i] = skey[i];

	req.p_query_input = &use;;

	user.method = IB_MAD_METHOD_GET;

	user.attr_id = IB_MAD_ATTR_SERVICE_RECORD;

	user.comp_mask = IB_SR_COMPMASK_SNAME | IB_SR_COMPMASK_SKEY;

	user.p_attr = &svc_rec;

	if (p_osmt->opt.with_grh) {

		req.with_grh = 1;

		memcpy(&req.gid, &p_osmt->sm_port_gid, 16);

	}

	status = osmv_query_sa(p_osmt->h_bind, &req;;

	if (status != IB_SUCCESS) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A0C: "

			"ib_query failed (%s)\n", ib_get_err_str(status));

		goto Exit;

	}

	status = context.result.status;

	num_recs = context.result.result_cnt;

	if (status != IB_SUCCESS) {

		/* If the failure is due to IB_SA_MAD_STATUS_NO_RECORDS and rec_num is 0,

		   then this is fine */

		if (status == IB_REMOTE_ERROR)

			strncpy(mad_stat_err,

			        ib_get_mad_status_str(osm_madw_get_mad_ptr

						      (context.result.

						      p_result_madw)),

				ERR_SIZE -1);

		else

			strncpy(mad_stat_err, ib_get_err_str(status),

				ERR_SIZE -1);

		mad_stat_err[ERR_SIZE -1] = '\0';

		if (status == IB_REMOTE_ERROR &&

		    !strcmp(mad_stat_err, "IB_SA_MAD_STATUS_NO_RECORDS") &&

		    rec_num == 0) {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

				"IS EXPECTED ERROR ^^^^\n");

			status = IB_SUCCESS;

		} else {

			OSM_LOG(&p_osmt->log, OSM_LOG_ERROR, "ERR 4A0D: "

				"Query failed:%s (%s)\n",

				ib_get_err_str(status), mad_stat_err);

			goto Exit;

		}

	}

	if (rec_num && num_recs != rec_num) {

		OSM_LOG(&p_osmt->log, OSM_LOG_ERROR,

			"Unmatched number of records: expected: %d, received: %d\n",

			rec_num, num_recs);

		status = IB_REMOTE_ERROR;

		goto Exit;

	}

	p_rec = osmv_get_query_svc_rec(context.result.p_result_madw, 0);

	*p_out_rec = *p_rec;

	if (num_recs) {

		OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

			"Found service record: name: %s id: 0x%016" PRIx64 "\n",

			sr_name, cl_ntoh64(p_rec->service_id));

		osm_dump_service_record(&p_osmt->log, p_rec, OSM_LOG_DEBUG);

	}

Exit:

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Expected and found %d records\n", rec_num);

	if (context.result.p_result_madw != NULL) {

		osm_mad_pool_put(&p_osmt->mad_pool,

				 context.result.p_result_madw);

		context.result.p_result_madw = NULL;

	}

	OSM_LOG_EXIT(&p_osmt->log);

	return status;

}

ib_api_status_t

osmt_get_service_by_name(IN osmtest_t * const p_osmt,

			 IN char *sr_name,

			 IN uint32_t rec_num,

			 OUT ib_service_record_t * p_out_rec)

{

	ib_api_status_t status = IB_SUCCESS;

	osmtest_req_context_t context;

	osmv_query_req_t req;

	ib_service_record_t *p_rec;

	ib_svc_name_t service_name;

	uint32_t num_recs = 0;

	OSM_LOG_ENTER(&p_osmt->log);

	OSM_LOG(&p_osmt->log, OSM_LOG_VERBOSE,

		"Getting service record: name: %s\n", sr_name);

	/*

	 * Do a blocking query for this record in the subnet.

	 * The result is returned in the result field of the caller's

	 * context structure.

	 *

	 * The query structures are locals.

	 */

	memset(&req, 0, sizeof(req));

	memset(&context, 0, sizeof(context));

	context.p_osmt = p_osmt;