/*
 * 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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 *
 * MODULE: dapl_srq.c
 *
 * PURPOSE: Shared Receive Queue
 * Description: Interfaces in this file are completely described in
 *		the DAPL 1.2 API, Chapter 6, section 5
 *
 */

#include "dapl.h"
#include "dapl_adapter_util.h"
#include "dapl_ia_util.h"
#include "dapl_srq_util.h"
#include "dapl_cookie.h"

/*
 * dapl_srq_create
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.1
 *
 * creates an instance of a Shared Receive Queue (SRQ) that is provided
 * to the Consumer as srq_handle.
 *
 * Input:
 * 	ia_handle
 * 	pz_handle
 * 	srq_attr
 *
 * Output:
 * 	srq_handle
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INSUFFICIENT_RESOURCES
 * 	DAT_INVALID_HANDLE
 * 	DAT_INVALID_PARAMETER
 * 	DAT_MODEL_NOT_SUPPORTED
 */

DAT_RETURN
dapl_srq_create(
	IN	DAT_IA_HANDLE ia_handle,
	IN	DAT_PZ_HANDLE pz_handle,
	IN	DAT_SRQ_ATTR *srq_attr,
	OUT	DAT_SRQ_HANDLE *srq_handle)
{
	DAPL_IA		*ia_ptr;
	DAPL_SRQ	*srq_ptr;
	DAT_SRQ_ATTR	srq_attr_limit;
	DAT_RETURN	dat_status;

	ia_ptr = (DAPL_IA *)ia_handle;
	dat_status = DAT_SUCCESS;
	/*
	 * Verify parameters
	 */
	if (DAPL_BAD_HANDLE(ia_ptr, DAPL_MAGIC_IA)) {
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE,
		    DAT_INVALID_HANDLE_IA);
		goto bail;
	}

	if ((pz_handle == NULL) || DAPL_BAD_HANDLE(pz_handle, DAPL_MAGIC_PZ)) {
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE,
		    DAT_INVALID_HANDLE_PZ);
		goto bail;
	}

	if ((srq_attr == NULL) || ((uintptr_t)srq_attr & 3)) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG3);
		goto bail;
	}

	if (srq_handle == NULL) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG4);
		goto bail;
	}

	if (srq_attr->max_recv_dtos == 0 || srq_attr->max_recv_iov == 0 ||
	    srq_attr->low_watermark != DAT_SRQ_LW_DEFAULT) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG3);
		goto bail;
	}

	/* Verify the attributes against the transport */
	(void) dapl_os_memzero(&srq_attr_limit, sizeof (DAT_SRQ_ATTR));
	dat_status = dapls_ib_query_hca(ia_ptr->hca_ptr, NULL, NULL, NULL,
	    &srq_attr_limit);
	if (dat_status != DAT_SUCCESS) {
			goto bail;
	}
	if (srq_attr->max_recv_dtos > srq_attr_limit.max_recv_dtos ||
	    srq_attr->max_recv_iov > srq_attr_limit.max_recv_iov) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG3);
		goto bail;
	}

	/* Allocate SRQ */
	srq_ptr = dapl_srq_alloc(ia_ptr, srq_attr);
	if (srq_ptr == NULL) {
		dat_status = DAT_ERROR(DAT_INSUFFICIENT_RESOURCES,
		    DAT_RESOURCE_MEMORY);
		goto bail;
	}

	/* Take a reference on the PZ handle */
	dapl_os_atomic_inc(&((DAPL_PZ *)pz_handle)->pz_ref_count);

	/*
	 * Fill in the SRQ
	 */
	srq_ptr->param.ia_handle	= ia_handle;
	srq_ptr->param.srq_state	= DAT_SRQ_STATE_OPERATIONAL;
	srq_ptr->param.pz_handle	= pz_handle;
	srq_ptr->param.max_recv_dtos	= srq_attr->max_recv_dtos;
	srq_ptr->param.max_recv_iov	= srq_attr->max_recv_iov;
	srq_ptr->param.low_watermark	= DAT_SRQ_LW_DEFAULT;

	srq_ptr->param.available_dto_count	= DAT_VALUE_UNKNOWN;
	srq_ptr->param.outstanding_dto_count	= 0;

	dat_status = dapls_ib_srq_alloc(ia_ptr, srq_ptr);
	if (dat_status != DAT_SUCCESS) {
		dapl_os_atomic_dec(&((DAPL_PZ *)pz_handle)->pz_ref_count);
		dapl_srq_dealloc(srq_ptr);
		goto bail;
	}
	/* Link it onto the IA */
	dapl_ia_link_srq(ia_ptr, srq_ptr);

	*srq_handle = srq_ptr;
bail:
	return (dat_status);
}

/*
 * dapl_srq_free
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.5
 *
 * destroys an instance of the SRQ. The SRQ cannot be destroyed if it is
 * in use by an EP.
 *
 * Input:
 * 	srq_handle
 *
 * Output:
 * 	none
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INVALID_HANDLE
 * 	DAT_INVALID_STATE
 */

DAT_RETURN
dapl_srq_free(
	IN	DAT_SRQ_HANDLE srq_handle)
{
	DAPL_SRQ	*srq_ptr;
	DAPL_IA		*ia_ptr;
	DAT_SRQ_PARAM	*param;
	DAT_RETURN	dat_status = DAT_SUCCESS;

	if (DAPL_BAD_HANDLE(srq_handle, DAPL_MAGIC_SRQ)) {
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE,
		    DAT_INVALID_HANDLE_SRQ);
		goto bail;
	}

	srq_ptr = (DAPL_SRQ *)srq_handle;
	param = &srq_ptr->param;
	if (0 != srq_ptr->srq_ref_count) {
		dapl_dbg_log(DAPL_DBG_TYPE_EP,
		    "dapl_srq_free: Free SRQ: %p, refcnt %d\n",
		    srq_ptr, srq_ptr->srq_ref_count);
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE,
		    DAT_INVALID_STATE_SRQ_IN_USE);
		goto bail;
	}

	ia_ptr = srq_ptr->header.owner_ia;
	param->srq_state = DAT_SRQ_STATE_ERROR;

	dapls_ib_srq_free(ia_ptr, srq_ptr);

	/* Remove link from the IA */
	dapl_ia_unlink_srq(ia_ptr, srq_ptr);

	dapl_os_assert(param->pz_handle != NULL);
	dapl_os_atomic_dec(&((DAPL_PZ *)param->pz_handle)->pz_ref_count);
	param->pz_handle = NULL;

	dapl_srq_dealloc(srq_ptr);

bail:
	return (dat_status);
}

/*
 * dapl_srq_post_recv
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.8
 *
 * posts the receive buffer that can be used for the incoming message into
 * the local_iov by any connected EP that uses SRQ.
 *
 * Input:
 * 	srq_handle
 * 	num_segments
 * 	local_iov
 *	user_cookie
 *
 * Output:
 * 	none
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INVALID_HANDLE
 * 	DAT_INSUFFICIENT_RESOURCES
 * 	DAT_INVALID_PARAMETER
 * 	DAT_PROTECTION_VIOLATION
 * 	DAT_PRIVILEGES_VIOLATION
 */

DAT_RETURN
dapl_srq_post_recv(
	IN	DAT_SRQ_HANDLE srq_handle,
	IN	DAT_COUNT num_segments,
	IN	DAT_LMR_TRIPLET *local_iov,
	IN	DAT_DTO_COOKIE user_cookie)
{
	DAPL_SRQ 		*srq_ptr;
	DAPL_COOKIE		*cookie;
	DAT_RETURN		dat_status;

	dapl_dbg_log(DAPL_DBG_TYPE_API,
	    "dapl_srq_post_recv (%p, %d, %p, %P)\n",
	    srq_handle,
	    num_segments,
	    local_iov,
	    user_cookie.as_64);

	if (DAPL_BAD_HANDLE(srq_handle, DAPL_MAGIC_SRQ)) {
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE,
		    DAT_INVALID_HANDLE_SRQ);
		goto bail;
	}

	srq_ptr = (DAPL_SRQ *) srq_handle;

	/*
	 * Synchronization ok since this buffer is only used for receive
	 * requests, which aren't allowed to race with each other.
	 */
	dat_status = dapls_dto_cookie_alloc(&srq_ptr->recv_buffer,
	    DAPL_DTO_TYPE_RECV,
	    user_cookie,
	    &cookie);
	if (DAT_SUCCESS != dat_status) {
		goto bail;
	}

	/*
	 * Invoke provider specific routine to post DTO
	 */
	dat_status = dapls_ib_post_srq(srq_ptr, cookie, num_segments,
	    local_iov);

	if (dat_status != DAT_SUCCESS) {
		dapls_cookie_dealloc(&srq_ptr->recv_buffer, cookie);
	} else {
		dapl_os_atomic_inc(&srq_ptr->recv_count);
	}

bail:
	dapl_dbg_log(DAPL_DBG_TYPE_RTN,
	    "dapl_srq_post_recv () returns 0x%x\n", dat_status);

	return (dat_status);
}


/*
 * dapl_srq_query
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.6
 *
 * provides to the Consumer SRQ parameters. The Consumer passes in a pointer
 * to the Consumer-allocated structures for SRQ parameters that the Provider
 * fills.
 *
 * Input:
 * 	srq_handle
 * 	srq_param_mask
 *
 * Output:
 * 	srq_param
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INVALID_HANDLE
 * 	DAT_INVALID_PARAMETER
 */

DAT_RETURN
dapl_srq_query(
	IN	DAT_SRQ_HANDLE srq_handle,
	IN	DAT_SRQ_PARAM_MASK srq_param_mask,
	OUT	DAT_SRQ_PARAM *srq_param)
{
	DAPL_SRQ	    *srq_ptr;
	DAT_RETURN	    dat_status;

	dat_status = DAT_SUCCESS;

	if (srq_param_mask & ~DAT_SRQ_FIELD_ALL) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG2);
		goto bail;
	}

	if (NULL == srq_param) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG3);
		goto bail;
	}


	if (DAPL_BAD_HANDLE(srq_handle, DAPL_MAGIC_SRQ)) {
		dat_status = DAT_ERROR(DAT_INVALID_HANDLE, 0);
		goto bail;
	}

	srq_ptr = (DAPL_SRQ *)srq_handle;
	/* Do a struct copy */
	*srq_param = srq_ptr->param;
	/* update the outstanding dto count */
	srq_param->outstanding_dto_count  = srq_ptr->recv_count;

bail:
	return (dat_status);
}

/*
 * dapl_srq_set_lw
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.4
 *
 * sets the low watermark value for SRQ and arms SRQ for generating an
 * asynchronous event for low watermark. An asynchronous event will be
 * generated when the number of buffers on SRQ is below the low watermark
 * for the first time. This may happen during this call or when an
 * associated EP takes a buffer from the SRQ.
 *
 * Input:
 * 	srq_handle
 * 	low_watermark
 *
 * Output:
 * 	none
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INVALID_HANDLE
 * 	DAT_INVALID_PARAMETER
 * 	DAT_MODEL_NOT_SUPPORTED
 */

/* ARGSUSED */
DAT_RETURN
dapl_srq_set_lw(
	IN	DAT_SRQ_HANDLE srq_handle,
	IN	DAT_COUNT low_watermark)
{
	return (DAT_MODEL_NOT_SUPPORTED);
}

/*
 * dapl_srq_resize
 *
 * uDAPL: User Direct Access Program Library Version 1.2, 6.5.7
 *
 * modifies the size of the queue of SRQ. Resizing of SRQ shall not cause
 * any incoming messages on any of the EPs that use the SRQ to be lost.
 *
 * Input:
 * 	srq_handle
 * 	srq_max_recv_dto
 *
 * Output:
 * 	none
 *
 * Returns:
 * 	DAT_SUCCESS
 * 	DAT_INVALID_HANDLE
 * 	DAT_INVALID_PARAMETER
 * 	DAT_INSUFFICIENT_RESOURCES
 * 	DAT_INVALID_STATE
 */

/* ARGSUSED */
DAT_RETURN
dapl_srq_resize(
	IN	DAT_SRQ_HANDLE srq_handle,
	IN	DAT_COUNT srq_max_recv_dtos)
{
	DAPL_SRQ		*srq_ptr;
	DAT_SRQ_ATTR		srq_attr_limit;
	DAPL_COOKIE_BUFFER	new_cb;
	DAT_RETURN		dat_status;


	srq_ptr = (DAPL_SRQ *)srq_handle;
	dat_status = DAT_SUCCESS;

	if (DAPL_BAD_HANDLE(srq_handle, DAPL_MAGIC_SRQ)) {
		return (DAT_ERROR(DAT_INVALID_HANDLE, DAT_HANDLE_TYPE_SRQ));
	}

	/* can't shrink below the number of outstanding recvs */
	if (srq_max_recv_dtos < srq_ptr->recv_count) {
		return (DAT_ERROR(DAT_INVALID_STATE, 0));
	}

	/*
	 * shrinking SRQs is not supported on tavor return success without
	 * any modification.
	 */
	if (srq_max_recv_dtos <= srq_ptr->param.max_recv_dtos) {
		return (DAT_SUCCESS);
	}

	/* Verify the attributes against the transport */
	(void) dapl_os_memzero(&srq_attr_limit, sizeof (DAT_SRQ_ATTR));
	dat_status = dapls_ib_query_hca(srq_ptr->header.owner_ia->hca_ptr,
	    NULL, NULL, NULL, &srq_attr_limit);
	if (dat_status != DAT_SUCCESS) {
		return (dat_status);
	}

	if (srq_max_recv_dtos > srq_attr_limit.max_recv_dtos) {
		dat_status = DAT_ERROR(DAT_INVALID_PARAMETER, DAT_INVALID_ARG2);
		return (dat_status);
	}

	dat_status = dapls_cb_resize(&srq_ptr->recv_buffer, srq_max_recv_dtos,
	    &new_cb);
	if (dat_status != DAT_SUCCESS) {
		return (dat_status);
	}

	dat_status = dapls_ib_srq_resize(srq_ptr, srq_max_recv_dtos);
	if (dat_status != DAT_SUCCESS) {
		goto bail;
	}

	dapls_cb_free(&srq_ptr->recv_buffer);
	srq_ptr->recv_buffer = new_cb; /* struct copy */
	srq_ptr->param.max_recv_dtos = srq_max_recv_dtos;

	return (DAT_SUCCESS);
bail:
	dapls_cb_free(&new_cb);

	return (dat_status);
}