/*
 * 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 (c) 2010, Oracle and/or its affiliates. All rights reserved.
 */

#include <sys/types.h>
#include <sys/kmem.h>
#include <sys/conf.h>
#include <sys/ddi.h>
#include <sys/sunddi.h>
#include <sys/ksynch.h>

#include <sys/ib/clients/eoib/eib_impl.h>

/*
 * Declarations private to this file
 */
static int eib_ctl_setup_cq(eib_t *, eib_vnic_t *);
static int eib_ctl_setup_ud_channel(eib_t *, eib_vnic_t *);
static void eib_ctl_comp_intr(ibt_cq_hdl_t, void *);
static void eib_ctl_rx_comp(eib_vnic_t *, eib_wqe_t *);
static void eib_ctl_tx_comp(eib_vnic_t *, eib_wqe_t *);
static void eib_ctl_err_comp(eib_vnic_t *, eib_wqe_t *, ibt_wc_t *);
static void eib_rb_ctl_setup_cq(eib_t *, eib_vnic_t *);
static void eib_rb_ctl_setup_ud_channel(eib_t *, eib_vnic_t *);

int
eib_ctl_create_qp(eib_t *ss, eib_vnic_t *vnic, int *err)
{
	eib_chan_t *chan = NULL;

	/*
	 * Allocate a eib_chan_t to store stuff about this vnic's ctl qp
	 * and initialize it with default admin qp pkey parameters. We'll
	 * re-associate this with the pkey we receive from the gw once we
	 * receive the login ack.
	 */
	vnic->vn_ctl_chan = eib_chan_init();

	chan = vnic->vn_ctl_chan;
	chan->ch_pkey = ss->ei_admin_chan->ch_pkey;
	chan->ch_pkey_ix = ss->ei_admin_chan->ch_pkey_ix;
	chan->ch_vnic_inst = vnic->vn_instance;

	/*
	 * Setup a combined CQ and completion handler
	 */
	if (eib_ctl_setup_cq(ss, vnic) != EIB_E_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_create_qp: "
		    "eib_ctl_setup_cq() failed");
		*err = ENOMEM;
		goto ctl_create_qp_fail;
	}

	/*
	 * Setup UD channel
	 */
	if (eib_ctl_setup_ud_channel(ss, vnic) != EIB_E_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_create_qp: "
		    "eib_ctl_setup_ud_channel() failed");
		*err = ENOMEM;
		goto ctl_create_qp_fail;
	}

	return (EIB_E_SUCCESS);

ctl_create_qp_fail:
	eib_rb_ctl_create_qp(ss, vnic);
	return (EIB_E_FAILURE);
}

/*ARGSUSED*/
uint_t
eib_ctl_comp_handler(caddr_t arg1, caddr_t arg2)
{
	eib_vnic_t *vnic = (eib_vnic_t *)(void *)arg1;
	eib_chan_t *chan = vnic->vn_ctl_chan;
	eib_t *ss = vnic->vn_ss;
	ibt_wc_t *wc;
	eib_wqe_t *wqe;
	ibt_status_t ret;
	uint_t polled;
	int i;

	/*
	 * Re-arm the notification callback before we start polling
	 * the completion queue.  There's nothing much we can do if the
	 * enable_cq_notify fails - we issue a warning and move on.
	 */
	ret = ibt_enable_cq_notify(chan->ch_cq_hdl, IBT_NEXT_COMPLETION);
	if (ret != IBT_SUCCESS) {
		EIB_DPRINTF_WARN(ss->ei_instance, "eib_ctl_comp_handler: "
		    "ibt_enable_cq_notify() failed, ret=%d", ret);
	}

	/*
	 * Handle tx and rx completions
	 */
	while ((ret = ibt_poll_cq(chan->ch_cq_hdl, chan->ch_wc, chan->ch_cq_sz,
	    &polled)) == IBT_SUCCESS) {
		for (wc = chan->ch_wc, i = 0; i < polled; i++, wc++) {
			wqe = (eib_wqe_t *)(uintptr_t)wc->wc_id;
			if (wc->wc_status != IBT_WC_SUCCESS) {
				eib_ctl_err_comp(vnic, wqe, wc);
			} else if (EIB_WQE_TYPE(wqe->qe_info) == EIB_WQE_RX) {
				eib_ctl_rx_comp(vnic, wqe);
			} else {
				eib_ctl_tx_comp(vnic, wqe);
			}
		}
	}

	return (DDI_INTR_CLAIMED);
}

void
eib_rb_ctl_create_qp(eib_t *ss, eib_vnic_t *vnic)
{
	eib_rb_ctl_setup_ud_channel(ss, vnic);

	eib_rb_ctl_setup_cq(ss, vnic);

	eib_chan_fini(vnic->vn_ctl_chan);
	vnic->vn_ctl_chan = NULL;
}

static int
eib_ctl_setup_cq(eib_t *ss, eib_vnic_t *vnic)
{
	eib_chan_t *chan = vnic->vn_ctl_chan;
	ibt_cq_attr_t cq_attr;
	ibt_status_t ret;
	uint_t sz;
	int rv;

	/*
	 * Allocate a completion queue for sending vhub table request
	 * and vhub-update/vnic-alive messages and responses from the
	 * gateway
	 */
	cq_attr.cq_sched = NULL;
	cq_attr.cq_flags = IBT_CQ_NO_FLAGS;
	if (ss->ei_hca_attrs->hca_max_cq_sz < EIB_CTL_CQ_SIZE)
		cq_attr.cq_size = ss->ei_hca_attrs->hca_max_cq_sz;
	else
		cq_attr.cq_size = EIB_CTL_CQ_SIZE;

	ret = ibt_alloc_cq(ss->ei_hca_hdl, &cq_attr, &chan->ch_cq_hdl, &sz);
	if (ret != IBT_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_setup_cq: "
		    "ibt_alloc_cq(cq_sz=0x%lx) failed, ret=%d",
		    cq_attr.cq_size, ret);
		goto ctl_setup_cq_fail;
	}

	/*
	 * Set up other parameters for collecting completion information
	 */
	chan->ch_cq_sz = sz;
	chan->ch_wc = kmem_zalloc(sizeof (ibt_wc_t) * sz, KM_SLEEP);

	/*
	 * Allocate soft interrupt for this vnic's control channel cq
	 * handler and set up the IBTL cq handler.
	 */
	if ((rv = ddi_intr_add_softint(ss->ei_dip, &vnic->vn_ctl_si_hdl,
	    EIB_SOFTPRI_CTL, eib_ctl_comp_handler, vnic)) != DDI_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_setup_cq: "
		    "ddi_intr_add_softint() failed for vnic %d ctl qp, ret=%d",
		    vnic->vn_instance, rv);
		goto ctl_setup_cq_fail;
	}

	/*
	 * Now, set up this vnic's control channel completion queue handler
	 */
	ibt_set_cq_handler(chan->ch_cq_hdl, eib_ctl_comp_intr, vnic);

	ret = ibt_enable_cq_notify(chan->ch_cq_hdl, IBT_NEXT_COMPLETION);
	if (ret != IBT_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_setup_cq: "
		    "ibt_enable_cq_notify() failed, ret=%d", ret);
		goto ctl_setup_cq_fail;
	}

	return (EIB_E_SUCCESS);

ctl_setup_cq_fail:
	eib_rb_ctl_setup_cq(ss, vnic);
	return (EIB_E_FAILURE);
}

static int
eib_ctl_setup_ud_channel(eib_t *ss, eib_vnic_t *vnic)
{
	eib_chan_t *chan = vnic->vn_ctl_chan;
	ibt_ud_chan_alloc_args_t alloc_attr;
	ibt_ud_chan_query_attr_t query_attr;
	ibt_status_t ret;

	bzero(&alloc_attr, sizeof (ibt_ud_chan_alloc_args_t));
	bzero(&query_attr, sizeof (ibt_ud_chan_query_attr_t));

	alloc_attr.ud_flags = IBT_ALL_SIGNALED;
	alloc_attr.ud_hca_port_num = ss->ei_props->ep_port_num;
	alloc_attr.ud_pkey_ix = chan->ch_pkey_ix;
	alloc_attr.ud_sizes.cs_sq = EIB_CTL_MAX_SWQE;
	alloc_attr.ud_sizes.cs_rq = EIB_CTL_MAX_RWQE;
	alloc_attr.ud_sizes.cs_sq_sgl = 1;
	alloc_attr.ud_sizes.cs_rq_sgl = 1;
	alloc_attr.ud_sizes.cs_inline = 0;

	alloc_attr.ud_qkey = EIB_FIP_QKEY;
	alloc_attr.ud_scq = chan->ch_cq_hdl;
	alloc_attr.ud_rcq = chan->ch_cq_hdl;
	alloc_attr.ud_pd = ss->ei_pd_hdl;

	ret = ibt_alloc_ud_channel(ss->ei_hca_hdl, IBT_ACHAN_NO_FLAGS,
	    &alloc_attr, &chan->ch_chan, NULL);
	if (ret != IBT_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_setup_ud_channel: "
		    "ibt_alloc_ud_channel(port=0x%x, pkey_ix=0x%x) "
		    "failed, ret=%d", alloc_attr.ud_hca_port_num,
		    chan->ch_pkey_ix, ret);
		goto ctl_setup_ud_channel_fail;
	}

	ret = ibt_query_ud_channel(chan->ch_chan, &query_attr);
	if (ret != IBT_SUCCESS) {
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_setup_ud_channel: "
		    "ibt_query_ud_channel() failed, ret=%d", ret);
		goto ctl_setup_ud_channel_fail;
	}

	chan->ch_qpn = query_attr.ud_qpn;
	chan->ch_max_swqes = query_attr.ud_chan_sizes.cs_sq;
	chan->ch_max_rwqes = query_attr.ud_chan_sizes.cs_rq;
	chan->ch_lwm_rwqes = chan->ch_max_rwqes >> 2;
	chan->ch_rwqe_bktsz = chan->ch_max_rwqes;
	chan->ch_ip_hdr_align = 0;
	chan->ch_alloc_mp = B_FALSE;
	chan->ch_tear_down = B_FALSE;

	return (EIB_E_SUCCESS);

ctl_setup_ud_channel_fail:
	eib_rb_ctl_setup_ud_channel(ss, vnic);
	return (EIB_E_FAILURE);
}

static void
eib_ctl_comp_intr(ibt_cq_hdl_t cq_hdl, void *arg)
{
	eib_vnic_t *vnic = arg;
	eib_t *ss = vnic->vn_ss;
	eib_chan_t *chan = vnic->vn_ctl_chan;

	if (cq_hdl != chan->ch_cq_hdl) {
		EIB_DPRINTF_DEBUG(ss->ei_instance, "eib_ctl_comp_intr: "
		    "cq_hdl(0x%llx) != chan->ch_cq_hdl(0x%llx), "
		    "ignoring completion", cq_hdl, chan->ch_cq_hdl);
		return;
	}

	ASSERT(vnic->vn_ctl_si_hdl != NULL);

	(void) ddi_intr_trigger_softint(vnic->vn_ctl_si_hdl, NULL);
}

static void
eib_ctl_rx_comp(eib_vnic_t *vnic, eib_wqe_t *wqe)
{
	eib_t *ss = vnic->vn_ss;
	eib_chan_t *chan = vnic->vn_ctl_chan;
	uint8_t *pkt = (uint8_t *)(uintptr_t)(wqe->qe_sgl.ds_va);
	ibt_status_t ret;

	/*
	 * Skip the GRH and parse the message in the packet
	 */
	(void) eib_fip_parse_ctl_pkt(pkt + EIB_GRH_SZ, vnic);

	/*
	 * Try to repost the rwqe.  For control channels, we take the shortcut
	 * and not go through eib_chan_post_recv(), since we know that the
	 * qe_info flag, qe_chan and qe_vinst are all already set correctly; we
	 * just took this out of the rx queue, so the ch_rx_posted will be ok
	 * if we just posted it back. And there are no mblk allocation or
	 * buffer alignment restrictions for this channel as well.
	 */
	if (chan->ch_tear_down) {
		eib_rsrc_return_rwqe(ss, wqe, chan);
	} else {
		ret = ibt_post_recv(chan->ch_chan, &(wqe->qe_wr.recv), 1, NULL);
		if (ret != IBT_SUCCESS) {
			EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_rx_comp: "
			    "ibt_post_recv() failed, ret=%d", ret);
			eib_rsrc_return_rwqe(ss, wqe, chan);
		}
	}
}

static void
eib_ctl_tx_comp(eib_vnic_t *vnic, eib_wqe_t *wqe)
{
	eib_rsrc_return_swqe(vnic->vn_ss, wqe, vnic->vn_ctl_chan);
}

static void
eib_ctl_err_comp(eib_vnic_t *vnic, eib_wqe_t *wqe, ibt_wc_t *wc)
{
	eib_t *ss = vnic->vn_ss;

	/*
	 * Currently, all we do is report
	 */
	switch (wc->wc_status) {
	case IBT_WC_WR_FLUSHED_ERR:
		break;

	case IBT_WC_LOCAL_CHAN_OP_ERR:
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_err_comp: "
		    "IBT_WC_LOCAL_CHAN_OP_ERR seen, wqe_info=0x%lx ",
		    wqe->qe_info);
		break;

	case IBT_WC_LOCAL_PROTECT_ERR:
		EIB_DPRINTF_ERR(ss->ei_instance, "eib_ctl_err_comp: "
		    "IBT_WC_LOCAL_PROTECT_ERR seen, wqe_info=0x%lx ",
		    wqe->qe_info);
		break;
	}

	/*
	 * When a wc indicates error, we do not attempt to repost but
	 * simply return it to the wqe pool.
	 */
	if (EIB_WQE_TYPE(wqe->qe_info) == EIB_WQE_RX)
		eib_rsrc_return_rwqe(ss, wqe, vnic->vn_ctl_chan);
	else
		eib_rsrc_return_swqe(ss, wqe, vnic->vn_ctl_chan);
}

/*ARGSUSED*/
static void
eib_rb_ctl_setup_cq(eib_t *ss, eib_vnic_t *vnic)
{
	eib_chan_t *chan = vnic->vn_ctl_chan;
	ibt_status_t ret;

	if (chan == NULL)
		return;

	/*
	 * Reset any completion handler we may have set up
	 */
	if (chan->ch_cq_hdl)
		ibt_set_cq_handler(chan->ch_cq_hdl, NULL, NULL);

	/*
	 * Remove any softint we may have allocated for this cq
	 */
	if (vnic->vn_ctl_si_hdl) {
		(void) ddi_intr_remove_softint(vnic->vn_ctl_si_hdl);
		vnic->vn_ctl_si_hdl = NULL;
	}

	/*
	 * Release any work completion buffers we may have allocated
	 */
	if (chan->ch_wc && chan->ch_cq_sz)
		kmem_free(chan->ch_wc, sizeof (ibt_wc_t) * chan->ch_cq_sz);

	chan->ch_cq_sz = 0;
	chan->ch_wc = NULL;

	/*
	 * Free any completion queue we may have allocated
	 */
	if (chan->ch_cq_hdl) {
		ret = ibt_free_cq(chan->ch_cq_hdl);
		if (ret != IBT_SUCCESS) {
			EIB_DPRINTF_WARN(ss->ei_instance,
			    "eib_rb_ctl_setup_cq: "
			    "ibt_free_cq() failed, ret=%d", ret);
		}
		chan->ch_cq_hdl = NULL;
	}
}

/*ARGSUSED*/
static void
eib_rb_ctl_setup_ud_channel(eib_t *ss, eib_vnic_t *vnic)
{
	eib_chan_t *chan = vnic->vn_ctl_chan;
	ibt_status_t ret;

	if (chan == NULL)
		return;

	if (chan->ch_chan) {
		/*
		 * We're trying to tear down this UD channel. Make sure that
		 * we don't attempt to refill (repost) at any point from now on.
		 */
		chan->ch_tear_down = B_TRUE;
		if ((ret = ibt_flush_channel(chan->ch_chan)) != IBT_SUCCESS) {
			EIB_DPRINTF_WARN(ss->ei_instance,
			    "eib_rb_ctl_setup_ud_channel: "
			    "ibt_flush_channel() failed, ret=%d", ret);
		}

		/*
		 * Wait until all posted tx wqes on this channel are back with
		 * the wqe pool.
		 */
		mutex_enter(&chan->ch_tx_lock);
		while (chan->ch_tx_posted > 0)
			cv_wait(&chan->ch_tx_cv, &chan->ch_tx_lock);
		mutex_exit(&chan->ch_tx_lock);

		/*
		 * Wait until all posted rx wqes on this channel are back with
		 * the wqe pool.
		 */
		mutex_enter(&chan->ch_rx_lock);
		while (chan->ch_rx_posted > 0)
			cv_wait(&chan->ch_rx_cv, &chan->ch_rx_lock);
		mutex_exit(&chan->ch_rx_lock);

		/*
		 * Now we're ready to free this channel
		 */
		if ((ret = ibt_free_channel(chan->ch_chan)) != IBT_SUCCESS) {
			EIB_DPRINTF_WARN(ss->ei_instance,
			    "eib_rb_ctl_setup_ud_channel: "
			    "ibt_free_channel() failed, ret=%d", ret);
		}

		chan->ch_alloc_mp = B_FALSE;
		chan->ch_ip_hdr_align = 0;
		chan->ch_rwqe_bktsz = 0;
		chan->ch_lwm_rwqes = 0;
		chan->ch_max_rwqes = 0;
		chan->ch_max_swqes = 0;
		chan->ch_qpn = 0;
		chan->ch_chan = NULL;
	}
}