/*
 * Copyright (c) 2018-2019 Cavium, Inc.
 * All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *  1. Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *  2. 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.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#include "qlnxr_def.h"
#include "rdma_common.h"
#include "qlnxr_cm.h"

void
qlnxr_inc_sw_gsi_cons(struct qlnxr_qp_hwq_info *info)
{
	info->gsi_cons = (info->gsi_cons + 1) % info->max_wr;
}

void
qlnxr_store_gsi_qp_cq(struct qlnxr_dev *dev,
		struct qlnxr_qp *qp,
		struct ib_qp_init_attr *attrs)
{
	QL_DPRINT12(dev->ha, "enter\n");
		
	dev->gsi_qp_created = 1;
	dev->gsi_sqcq = get_qlnxr_cq((attrs->send_cq));
	dev->gsi_rqcq = get_qlnxr_cq((attrs->recv_cq));
	dev->gsi_qp = qp;

	QL_DPRINT12(dev->ha, "exit\n");

	return;
}

void
qlnxr_ll2_complete_tx_packet(void *cxt,
		uint8_t connection_handle,
		void *cookie,
		dma_addr_t first_frag_addr,
		bool b_last_fragment,
		bool b_last_packet)
{
	struct qlnxr_dev *dev = (struct qlnxr_dev *)cxt;
	struct ecore_roce_ll2_packet *pkt = cookie;
	struct qlnxr_cq *cq = dev->gsi_sqcq;
	struct qlnxr_qp *qp = dev->gsi_qp;
	unsigned long flags;

	QL_DPRINT12(dev->ha, "enter\n");

	qlnx_dma_free_coherent(&dev->ha->cdev, pkt->header.vaddr,
			pkt->header.baddr, pkt->header.len);
	kfree(pkt);

	spin_lock_irqsave(&qp->q_lock, flags);

	qlnxr_inc_sw_gsi_cons(&qp->sq);

	spin_unlock_irqrestore(&qp->q_lock, flags);

	if (cq->ibcq.comp_handler)
		(*cq->ibcq.comp_handler) (&cq->ibcq, cq->ibcq.cq_context);

	QL_DPRINT12(dev->ha, "exit\n");

	return;
}

void
qlnxr_ll2_complete_rx_packet(void *cxt,
		struct ecore_ll2_comp_rx_data *data)
{
	struct qlnxr_dev *dev = (struct qlnxr_dev *)cxt;
	struct qlnxr_cq *cq = dev->gsi_rqcq;
	// struct qlnxr_qp *qp = dev->gsi_qp;
	struct qlnxr_qp *qp = NULL;
	unsigned long flags;
	// uint32_t delay_count = 0, gsi_cons = 0;
	//void * dest_va;

	QL_DPRINT12(dev->ha, "enter\n");

	if (data->u.data_length_error) {
		/* TODO: add statistic */
	}

	if (data->cookie == NULL) {
		QL_DPRINT12(dev->ha, "cookie is NULL, bad sign\n");
	}

	if (data->qp_id == 1) {
		qp = dev->gsi_qp;
	} else {
		/* TODO: This will be needed for UD QP support */
		/* For RoCEv1 this is invalid */
		QL_DPRINT12(dev->ha, "invalid QP\n");
		return;
	}
	/* note: currently only one recv sg is supported */
	QL_DPRINT12(dev->ha, "MAD received on QP : %x\n", data->rx_buf_addr);

	spin_lock_irqsave(&qp->q_lock, flags);

	qp->rqe_wr_id[qp->rq.gsi_cons].rc =
		data->u.data_length_error ? -EINVAL : 0;
	qp->rqe_wr_id[qp->rq.gsi_cons].vlan_id = data->vlan;
	/* note: length stands for data length i.e. GRH is excluded */
	qp->rqe_wr_id[qp->rq.gsi_cons].sg_list[0].length =
		data->length.data_length;
	*((u32 *)&qp->rqe_wr_id[qp->rq.gsi_cons].smac[0]) =
		ntohl(data->opaque_data_0);
	*((u16 *)&qp->rqe_wr_id[qp->rq.gsi_cons].smac[4]) =
		ntohs((u16)data->opaque_data_1);

	qlnxr_inc_sw_gsi_cons(&qp->rq);

	spin_unlock_irqrestore(&qp->q_lock, flags);

	if (cq->ibcq.comp_handler)
		(*cq->ibcq.comp_handler) (&cq->ibcq, cq->ibcq.cq_context);

	QL_DPRINT12(dev->ha, "exit\n");

	return;
}

void qlnxr_ll2_release_rx_packet(void *cxt,
		u8 connection_handle,
		void *cookie,
		dma_addr_t rx_buf_addr,
		bool b_last_packet)
{
	/* Do nothing... */
}

static void
qlnxr_destroy_gsi_cq(struct qlnxr_dev *dev,
		struct ib_qp_init_attr *attrs)
{
	struct ecore_rdma_destroy_cq_in_params iparams;
	struct ecore_rdma_destroy_cq_out_params oparams;
	struct qlnxr_cq *cq;

	QL_DPRINT12(dev->ha, "enter\n");

	cq = get_qlnxr_cq((attrs->send_cq));
	iparams.icid = cq->icid;
	ecore_rdma_destroy_cq(dev->rdma_ctx, &iparams, &oparams);
	ecore_chain_free(&dev->ha->cdev, &cq->pbl);

	cq = get_qlnxr_cq((attrs->recv_cq));
	/* if a dedicated recv_cq was used, delete it too */
	if (iparams.icid != cq->icid) {
		iparams.icid = cq->icid;
		ecore_rdma_destroy_cq(dev->rdma_ctx, &iparams, &oparams);
		ecore_chain_free(&dev->ha->cdev, &cq->pbl);
	}

	QL_DPRINT12(dev->ha, "exit\n");

	return;
}

static inline int
qlnxr_check_gsi_qp_attrs(struct qlnxr_dev *dev,
		struct ib_qp_init_attr *attrs)
{
	QL_DPRINT12(dev->ha, "enter\n");

	if (attrs->cap.max_recv_sge > QLNXR_GSI_MAX_RECV_SGE) {
		QL_DPRINT11(dev->ha,
			"(attrs->cap.max_recv_sge > QLNXR_GSI_MAX_RECV_SGE)\n");
		return -EINVAL;
	}

	if (attrs->cap.max_recv_wr > QLNXR_GSI_MAX_RECV_WR) {
		QL_DPRINT11(dev->ha,
			"(attrs->cap.max_recv_wr > QLNXR_GSI_MAX_RECV_WR)\n");
		return -EINVAL;
	}

	if (attrs->cap.max_send_wr > QLNXR_GSI_MAX_SEND_WR) {
		QL_DPRINT11(dev->ha,
			"(attrs->cap.max_send_wr > QLNXR_GSI_MAX_SEND_WR)\n");
		return -EINVAL;
	}

	QL_DPRINT12(dev->ha, "exit\n");

	return 0;
}

static int
qlnxr_ll2_post_tx(struct qlnxr_dev *dev, struct ecore_roce_ll2_packet *pkt)
{
	enum ecore_ll2_roce_flavor_type roce_flavor;
	struct ecore_ll2_tx_pkt_info ll2_tx_pkt;
	int rc;
	int i;

	QL_DPRINT12(dev->ha, "enter\n");

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

	if (pkt->roce_mode != ROCE_V1) {
		QL_DPRINT11(dev->ha, "roce_mode != ROCE_V1\n");
		return (-1);
	}

	roce_flavor = (pkt->roce_mode == ROCE_V1) ?
		ECORE_LL2_ROCE : ECORE_LL2_RROCE;

	ll2_tx_pkt.num_of_bds = 1 /* hdr */ +  pkt->n_seg;
	ll2_tx_pkt.vlan = 0; /* ??? */
	ll2_tx_pkt.tx_dest = ECORE_LL2_TX_DEST_NW;
	ll2_tx_pkt.ecore_roce_flavor = roce_flavor;
	ll2_tx_pkt.first_frag = pkt->header.baddr;
	ll2_tx_pkt.first_frag_len = pkt->header.len;
	ll2_tx_pkt.cookie = pkt;
	ll2_tx_pkt.enable_ip_cksum = 1; // Only for RoCEv2:IPv4

	/* tx header */
	rc = ecore_ll2_prepare_tx_packet(dev->rdma_ctx,
			dev->gsi_ll2_handle,
			&ll2_tx_pkt,
			1);
	if (rc) {
		QL_DPRINT11(dev->ha, "ecore_ll2_prepare_tx_packet failed\n");

		/* TX failed while posting header - release resources*/
                qlnx_dma_free_coherent(&dev->ha->cdev,
			pkt->header.vaddr,
			pkt->header.baddr,
                        pkt->header.len);

		kfree(pkt);

		return rc;
	}

	/* tx payload */
	for (i = 0; i < pkt->n_seg; i++) {
		rc = ecore_ll2_set_fragment_of_tx_packet(dev->rdma_ctx,
						       dev->gsi_ll2_handle,
						       pkt->payload[i].baddr,
						       pkt->payload[i].len);
		if (rc) {
			/* if failed not much to do here, partial packet has
			 * been posted we can't free memory, will need to wait
			 * for completion
			 */
			QL_DPRINT11(dev->ha,
				"ecore_ll2_set_fragment_of_tx_packet failed\n");
			return rc;
		}
	}
	struct ecore_ll2_stats stats = {0};
	rc = ecore_ll2_get_stats(dev->rdma_ctx, dev->gsi_ll2_handle, &stats);
	if (rc) {
		QL_DPRINT11(dev->ha, "failed to obtain ll2 stats\n");
	}
	QL_DPRINT12(dev->ha, "exit\n");

	return 0;
}

int
qlnxr_ll2_stop(struct qlnxr_dev *dev)
{
	int rc;

	QL_DPRINT12(dev->ha, "enter\n");

	if (dev->gsi_ll2_handle == 0xFF)
		return 0;

	/* remove LL2 MAC address filter */
	rc = qlnx_rdma_ll2_set_mac_filter(dev->rdma_ctx,
			  dev->gsi_ll2_mac_address, NULL);

	rc = ecore_ll2_terminate_connection(dev->rdma_ctx,
			dev->gsi_ll2_handle);

	ecore_ll2_release_connection(dev->rdma_ctx, dev->gsi_ll2_handle);

	dev->gsi_ll2_handle = 0xFF;

	QL_DPRINT12(dev->ha, "exit rc = %d\n", rc);
	return rc;
}

int qlnxr_ll2_start(struct qlnxr_dev *dev,
		   struct ib_qp_init_attr *attrs,
		   struct qlnxr_qp *qp)
{
	struct ecore_ll2_acquire_data data;
	struct ecore_ll2_cbs cbs;
	int rc;

	QL_DPRINT12(dev->ha, "enter\n");

	/* configure and start LL2 */
	cbs.rx_comp_cb = qlnxr_ll2_complete_rx_packet;
	cbs.tx_comp_cb = qlnxr_ll2_complete_tx_packet;
	cbs.rx_release_cb = qlnxr_ll2_release_rx_packet;
	cbs.tx_release_cb = qlnxr_ll2_complete_tx_packet;
	cbs.cookie = dev;
	dev->gsi_ll2_handle = 0xFF;

	memset(&data, 0, sizeof(data));
	data.input.conn_type = ECORE_LL2_TYPE_ROCE;
	data.input.mtu = if_getmtu(dev->ha->ifp);
	data.input.rx_num_desc = 8 * 1024;
	data.input.rx_drop_ttl0_flg = 1;
	data.input.rx_vlan_removal_en = 0;
	data.input.tx_num_desc = 8 * 1024;
	data.input.tx_tc = 0;
	data.input.tx_dest = ECORE_LL2_TX_DEST_NW;
	data.input.ai_err_packet_too_big = ECORE_LL2_DROP_PACKET;
	data.input.ai_err_no_buf = ECORE_LL2_DROP_PACKET;
	data.input.gsi_enable = 1;
	data.p_connection_handle = &dev->gsi_ll2_handle;
	data.cbs = &cbs;

	rc = ecore_ll2_acquire_connection(dev->rdma_ctx, &data);

	if (rc) {
		QL_DPRINT11(dev->ha,
			"ecore_ll2_acquire_connection failed: %d\n",
			rc);
		return rc;
	}

	QL_DPRINT11(dev->ha,
		"ll2 connection acquired successfully\n");
	rc = ecore_ll2_establish_connection(dev->rdma_ctx,
		dev->gsi_ll2_handle);

	if (rc) {
		QL_DPRINT11(dev->ha,
			"ecore_ll2_establish_connection failed\n", rc);
		goto err1;
	}

	QL_DPRINT11(dev->ha,
		"ll2 connection established successfully\n");
	rc = qlnx_rdma_ll2_set_mac_filter(dev->rdma_ctx, NULL,
			dev->ha->primary_mac);
	if (rc) {
		QL_DPRINT11(dev->ha, "qlnx_rdma_ll2_set_mac_filter failed\n", rc);
		goto err2;
	}

	QL_DPRINT12(dev->ha, "exit rc = %d\n", rc);
	return 0;

err2:
	ecore_ll2_terminate_connection(dev->rdma_ctx, dev->gsi_ll2_handle);
err1:
	ecore_ll2_release_connection(dev->rdma_ctx, dev->gsi_ll2_handle);

	QL_DPRINT12(dev->ha, "exit rc = %d\n", rc);
	return rc;
}

struct ib_qp*
qlnxr_create_gsi_qp(struct qlnxr_dev *dev,
		 struct ib_qp_init_attr *attrs,
		 struct qlnxr_qp *qp)
{
	int rc;

	QL_DPRINT12(dev->ha, "enter\n");

	rc = qlnxr_check_gsi_qp_attrs(dev, attrs);

	if (rc) {
		QL_DPRINT11(dev->ha, "qlnxr_check_gsi_qp_attrs failed\n");
		return ERR_PTR(rc);
	}

	rc = qlnxr_ll2_start(dev, attrs, qp);
	if (rc) {
		QL_DPRINT11(dev->ha, "qlnxr_ll2_start failed\n");
		return ERR_PTR(rc);
	}

	/* create QP */
	qp->ibqp.qp_num = 1;
	qp->rq.max_wr = attrs->cap.max_recv_wr;
	qp->sq.max_wr = attrs->cap.max_send_wr;

	qp->rqe_wr_id = kzalloc(qp->rq.max_wr * sizeof(*qp->rqe_wr_id),
				GFP_KERNEL);
	if (!qp->rqe_wr_id) {
		QL_DPRINT11(dev->ha, "(!qp->rqe_wr_id)\n");
		goto err;
	}

	qp->wqe_wr_id = kzalloc(qp->sq.max_wr * sizeof(*qp->wqe_wr_id),
				GFP_KERNEL);
	if (!qp->wqe_wr_id) {
		QL_DPRINT11(dev->ha, "(!qp->wqe_wr_id)\n");
		goto err;
	}

	qlnxr_store_gsi_qp_cq(dev, qp, attrs);
	memcpy(dev->gsi_ll2_mac_address, dev->ha->primary_mac, ETH_ALEN);

	/* the GSI CQ is handled by the driver so remove it from the FW */
	qlnxr_destroy_gsi_cq(dev, attrs);
	dev->gsi_rqcq->cq_type = QLNXR_CQ_TYPE_GSI;
	dev->gsi_rqcq->cq_type = QLNXR_CQ_TYPE_GSI;

	QL_DPRINT12(dev->ha, "exit &qp->ibqp = %p\n", &qp->ibqp);

	return &qp->ibqp;
err:
	kfree(qp->rqe_wr_id);

	rc = qlnxr_ll2_stop(dev);

	QL_DPRINT12(dev->ha, "exit with error\n");

	return ERR_PTR(-ENOMEM);
}

int
qlnxr_destroy_gsi_qp(struct qlnxr_dev *dev)
{
	int rc = 0;

	QL_DPRINT12(dev->ha, "enter\n");

	rc = qlnxr_ll2_stop(dev);

	QL_DPRINT12(dev->ha, "exit rc = %d\n", rc);
	return (rc);
}

static inline bool
qlnxr_get_vlan_id_gsi(struct ib_ah_attr *ah_attr, u16 *vlan_id)
{
	u16 tmp_vlan_id;
	union ib_gid *dgid = &ah_attr->grh.dgid;

	tmp_vlan_id = (dgid->raw[11] << 8) | dgid->raw[12];
	if (tmp_vlan_id < 0x1000) {
		*vlan_id = tmp_vlan_id;
		return true;
	} else {
		*vlan_id = 0;
		return false;
	}
}

#define QLNXR_MAX_UD_HEADER_SIZE	(100)
#define QLNXR_GSI_QPN		(1)
static inline int
qlnxr_gsi_build_header(struct qlnxr_dev *dev,
		struct qlnxr_qp *qp,
		const struct ib_send_wr *swr,
		struct ib_ud_header *udh,
		int *roce_mode)
{
	bool has_vlan = false, has_grh_ipv6 = true;
	struct ib_ah_attr *ah_attr = &get_qlnxr_ah((ud_wr(swr)->ah))->attr;
	struct ib_global_route *grh = &ah_attr->grh;
	union ib_gid sgid;
	int send_size = 0;
	u16 vlan_id = 0;
	u16 ether_type;

	int rc = 0;
	int ip_ver = 0;
	bool has_udp = false;

	int i;

	send_size = 0;
	for (i = 0; i < swr->num_sge; ++i)
		send_size += swr->sg_list[i].length;

	has_vlan = qlnxr_get_vlan_id_gsi(ah_attr, &vlan_id);
	ether_type = ETH_P_ROCE;
	*roce_mode = ROCE_V1;
	if (grh->sgid_index < QLNXR_MAX_SGID)
		sgid = dev->sgid_tbl[grh->sgid_index];
	else
		sgid = dev->sgid_tbl[0];

	rc = ib_ud_header_init(send_size, false /* LRH */, true /* ETH */,
			has_vlan, has_grh_ipv6, ip_ver, has_udp,
			0 /* immediate */, udh);

	if (rc) {
		QL_DPRINT11(dev->ha, "gsi post send: failed to init header\n");
		return rc;
	}

	/* ENET + VLAN headers*/
	memcpy(udh->eth.dmac_h, ah_attr->dmac, ETH_ALEN);
	memcpy(udh->eth.smac_h, dev->ha->primary_mac, ETH_ALEN);
	if (has_vlan) {
		udh->eth.type = htons(ETH_P_8021Q);
		udh->vlan.tag = htons(vlan_id);
		udh->vlan.type = htons(ether_type);
	} else {
		udh->eth.type = htons(ether_type);
	}

	for (int j = 0; j < 4; j++) {
		QL_DPRINT12(dev->ha, "destination mac: %x\n",
				udh->eth.dmac_h[j]);
	}
	for (int j = 0; j < 4; j++) {
		QL_DPRINT12(dev->ha, "source mac: %x\n",
				udh->eth.smac_h[j]);
	}

	QL_DPRINT12(dev->ha, "QP: %p, opcode: %d, wq: %lx, roce: %x, hops:%d,"
			"imm : %d, vlan :%d, AH: %p\n",
			qp, swr->opcode, swr->wr_id, *roce_mode, grh->hop_limit,
			0, has_vlan, get_qlnxr_ah((ud_wr(swr)->ah)));

	if (has_grh_ipv6) {
		/* GRH / IPv6 header */
		udh->grh.traffic_class = grh->traffic_class;
		udh->grh.flow_label = grh->flow_label;
		udh->grh.hop_limit = grh->hop_limit;
		udh->grh.destination_gid = grh->dgid;
		memcpy(&udh->grh.source_gid.raw, &sgid.raw,
		       sizeof(udh->grh.source_gid.raw));
		QL_DPRINT12(dev->ha, "header: tc: %x, flow_label : %x, "
			"hop_limit: %x \n", udh->grh.traffic_class,
			udh->grh.flow_label, udh->grh.hop_limit);
		for (i = 0; i < 16; i++) {
			QL_DPRINT12(dev->ha, "udh dgid = %x\n", udh->grh.destination_gid.raw[i]);
		}
		for (i = 0; i < 16; i++) {
			QL_DPRINT12(dev->ha, "udh sgid = %x\n", udh->grh.source_gid.raw[i]);
		}
		udh->grh.next_header = 0x1b;
	}
#ifdef DEFINE_IB_UD_HEADER_INIT_UDP_PRESENT 
        /* This is for RoCEv2 */
	else {
                /* IPv4 header */
                u32 ipv4_addr;

                udh->ip4.protocol = IPPROTO_UDP;
                udh->ip4.tos = htonl(grh->flow_label);
                udh->ip4.frag_off = htons(IP_DF);
                udh->ip4.ttl = grh->hop_limit;

                ipv4_addr = qedr_get_ipv4_from_gid(sgid.raw);
                udh->ip4.saddr = ipv4_addr;
                ipv4_addr = qedr_get_ipv4_from_gid(grh->dgid.raw);
                udh->ip4.daddr = ipv4_addr;
                /* note: checksum is calculated by the device */
        }
#endif

	/* BTH */
	udh->bth.solicited_event = !!(swr->send_flags & IB_SEND_SOLICITED);
	udh->bth.pkey = QLNXR_ROCE_PKEY_DEFAULT;/* TODO: ib_get_cahced_pkey?! */
	//udh->bth.destination_qpn = htonl(ud_wr(swr)->remote_qpn);
	udh->bth.destination_qpn = OSAL_CPU_TO_BE32(ud_wr(swr)->remote_qpn);
	//udh->bth.psn = htonl((qp->sq_psn++) & ((1 << 24) - 1));
	udh->bth.psn = OSAL_CPU_TO_BE32((qp->sq_psn++) & ((1 << 24) - 1));
	udh->bth.opcode = IB_OPCODE_UD_SEND_ONLY;

	/* DETH */
	//udh->deth.qkey = htonl(0x80010000); /* qp->qkey */ /* TODO: what is?! */
	//udh->deth.source_qpn = htonl(QLNXR_GSI_QPN);
	udh->deth.qkey = OSAL_CPU_TO_BE32(0x80010000); /* qp->qkey */ /* TODO: what is?! */
	udh->deth.source_qpn = OSAL_CPU_TO_BE32(QLNXR_GSI_QPN);
	QL_DPRINT12(dev->ha, "exit\n");
	return 0;
}

static inline int
qlnxr_gsi_build_packet(struct qlnxr_dev *dev,
	struct qlnxr_qp *qp, const struct ib_send_wr *swr,
	struct ecore_roce_ll2_packet **p_packet)
{
	u8 ud_header_buffer[QLNXR_MAX_UD_HEADER_SIZE];
	struct ecore_roce_ll2_packet *packet;
	int roce_mode, header_size;
	struct ib_ud_header udh;
	int i, rc;

	QL_DPRINT12(dev->ha, "enter\n");

	*p_packet = NULL;

	rc = qlnxr_gsi_build_header(dev, qp, swr, &udh, &roce_mode);
	if (rc) {
		QL_DPRINT11(dev->ha,
			"qlnxr_gsi_build_header failed rc = %d\n", rc);
		return rc;
	}

	header_size = ib_ud_header_pack(&udh, &ud_header_buffer);

	packet = kzalloc(sizeof(*packet), GFP_ATOMIC);
	if (!packet) {
		QL_DPRINT11(dev->ha, "packet == NULL\n");
		return -ENOMEM;
	}

	packet->header.vaddr = qlnx_dma_alloc_coherent(&dev->ha->cdev,
					&packet->header.baddr,
					header_size);
	if (!packet->header.vaddr) {
		QL_DPRINT11(dev->ha, "packet->header.vaddr == NULL\n");
		kfree(packet);
		return -ENOMEM;
	}

	if (memcmp(udh.eth.smac_h, udh.eth.dmac_h, ETH_ALEN))
		packet->tx_dest = ECORE_ROCE_LL2_TX_DEST_NW;
	else
		packet->tx_dest = ECORE_ROCE_LL2_TX_DEST_LB;

	packet->roce_mode = roce_mode;
	memcpy(packet->header.vaddr, ud_header_buffer, header_size);
	packet->header.len = header_size;
	packet->n_seg = swr->num_sge;
	qp->wqe_wr_id[qp->sq.prod].bytes_len = IB_GRH_BYTES; //RDMA_GRH_BYTES
	for (i = 0; i < packet->n_seg; i++) {
		packet->payload[i].baddr = swr->sg_list[i].addr;
		packet->payload[i].len = swr->sg_list[i].length;
		qp->wqe_wr_id[qp->sq.prod].bytes_len +=
			packet->payload[i].len;
		QL_DPRINT11(dev->ha, "baddr: %p, len: %d\n",
				packet->payload[i].baddr,
				packet->payload[i].len);
	}

	*p_packet = packet;

	QL_DPRINT12(dev->ha, "exit, packet->n_seg: %d\n", packet->n_seg);
	return 0;
}

int
qlnxr_gsi_post_send(struct ib_qp *ibqp,
		const struct ib_send_wr *wr,
		const struct ib_send_wr **bad_wr)
{
	struct ecore_roce_ll2_packet *pkt = NULL;
	struct qlnxr_qp *qp = get_qlnxr_qp(ibqp);
	struct qlnxr_dev *dev = qp->dev;
	unsigned long flags;
	int rc;

	QL_DPRINT12(dev->ha, "exit\n");

	if (qp->state != ECORE_ROCE_QP_STATE_RTS) {
		QL_DPRINT11(dev->ha,
			"(qp->state != ECORE_ROCE_QP_STATE_RTS)\n");
		*bad_wr = wr;
		return -EINVAL;
	}

	if (wr->num_sge > RDMA_MAX_SGE_PER_SQ_WQE) {
		QL_DPRINT11(dev->ha,
			"(wr->num_sge > RDMA_MAX_SGE_PER_SQ_WQE)\n");
		rc = -EINVAL;
		goto err;
	}

	if (wr->opcode != IB_WR_SEND) {
		QL_DPRINT11(dev->ha, "(wr->opcode > IB_WR_SEND)\n");
		rc = -EINVAL;
		goto err;
	}

	spin_lock_irqsave(&qp->q_lock, flags);

	rc = qlnxr_gsi_build_packet(dev, qp, wr, &pkt);
	if(rc) {
		spin_unlock_irqrestore(&qp->q_lock, flags);
		QL_DPRINT11(dev->ha, "qlnxr_gsi_build_packet failed\n");
		goto err;
	}

	rc = qlnxr_ll2_post_tx(dev, pkt);

	if (!rc) {
		qp->wqe_wr_id[qp->sq.prod].wr_id = wr->wr_id;
		qp->wqe_wr_id[qp->sq.prod].signaled = 
			!!(wr->send_flags & IB_SEND_SIGNALED);
		qp->wqe_wr_id[qp->sq.prod].opcode = IB_WC_SEND;
		qlnxr_inc_sw_prod(&qp->sq);
		QL_DPRINT11(dev->ha, "packet sent over gsi qp\n");
	} else {
		QL_DPRINT11(dev->ha, "qlnxr_ll2_post_tx failed\n");
		rc = -EAGAIN;
		*bad_wr = wr;
	}

	spin_unlock_irqrestore(&qp->q_lock, flags);

	if (wr->next != NULL) {
		*bad_wr = wr->next;
		rc=-EINVAL;
	}

	QL_DPRINT12(dev->ha, "exit\n");
	return rc;

err:
	*bad_wr = wr;
	QL_DPRINT12(dev->ha, "exit error\n");
	return rc;
}

#define	QLNXR_LL2_RX_BUFFER_SIZE	(4 * 1024)
int
qlnxr_gsi_post_recv(struct ib_qp *ibqp,
		const struct ib_recv_wr *wr,
		const struct ib_recv_wr **bad_wr)
{
	struct qlnxr_dev *dev = get_qlnxr_dev((ibqp->device));
	struct qlnxr_qp *qp = get_qlnxr_qp(ibqp);
	unsigned long flags;
	int rc = 0;

	QL_DPRINT12(dev->ha, "enter, wr: %p\n", wr);

	if ((qp->state != ECORE_ROCE_QP_STATE_RTR) &&
	    (qp->state != ECORE_ROCE_QP_STATE_RTS)) {
		*bad_wr = wr;
		QL_DPRINT11(dev->ha, "exit 0\n");
		return -EINVAL;
	}

	spin_lock_irqsave(&qp->q_lock, flags);

	while (wr) {
		if (wr->num_sge > QLNXR_GSI_MAX_RECV_SGE) {
			QL_DPRINT11(dev->ha, "exit 1\n");
			goto err;
		}

		rc = ecore_ll2_post_rx_buffer(dev->rdma_ctx,
				dev->gsi_ll2_handle,
				wr->sg_list[0].addr,
				wr->sg_list[0].length,
				0 /* cookie */,
				1 /* notify_fw */);
		if (rc) {
			QL_DPRINT11(dev->ha, "exit 2\n");
			goto err;
		}

		memset(&qp->rqe_wr_id[qp->rq.prod], 0,
			sizeof(qp->rqe_wr_id[qp->rq.prod]));
		qp->rqe_wr_id[qp->rq.prod].sg_list[0] = wr->sg_list[0];
		qp->rqe_wr_id[qp->rq.prod].wr_id = wr->wr_id;

		qlnxr_inc_sw_prod(&qp->rq);

		wr = wr->next;
	}

	spin_unlock_irqrestore(&qp->q_lock, flags);

	QL_DPRINT12(dev->ha, "exit rc = %d\n", rc);
	return rc;
err:

	spin_unlock_irqrestore(&qp->q_lock, flags);
	*bad_wr = wr;

	QL_DPRINT12(dev->ha, "exit with -ENOMEM\n");
	return -ENOMEM;
}

int
qlnxr_gsi_poll_cq(struct ib_cq *ibcq, int num_entries, struct ib_wc *wc)
{
	struct qlnxr_dev *dev = get_qlnxr_dev((ibcq->device));
	struct qlnxr_cq *cq = get_qlnxr_cq(ibcq);
	struct qlnxr_qp *qp = dev->gsi_qp;
	unsigned long flags;
	int i = 0;

	QL_DPRINT12(dev->ha, "enter\n");

	spin_lock_irqsave(&cq->cq_lock, flags);

	while (i < num_entries && qp->rq.cons != qp->rq.gsi_cons) {
		memset(&wc[i], 0, sizeof(*wc));

		wc[i].qp = &qp->ibqp;
		wc[i].wr_id = qp->rqe_wr_id[qp->rq.cons].wr_id;
		wc[i].opcode = IB_WC_RECV;
		wc[i].pkey_index = 0;
		wc[i].status = (qp->rqe_wr_id[qp->rq.cons].rc)?
			       IB_WC_GENERAL_ERR:IB_WC_SUCCESS;
		/* 0 - currently only one recv sg is supported */
		wc[i].byte_len = qp->rqe_wr_id[qp->rq.cons].sg_list[0].length;
		wc[i].wc_flags |= IB_WC_GRH | IB_WC_IP_CSUM_OK;

		memcpy(&wc[i].smac, qp->rqe_wr_id[qp->rq.cons].smac, ETH_ALEN);
		wc[i].wc_flags |= IB_WC_WITH_SMAC;

		if (qp->rqe_wr_id[qp->rq.cons].vlan_id) {
			wc[i].wc_flags |= IB_WC_WITH_VLAN;
			wc[i].vlan_id = qp->rqe_wr_id[qp->rq.cons].vlan_id;
		}

		qlnxr_inc_sw_cons(&qp->rq);
		i++;
	}

	while (i < num_entries && qp->sq.cons != qp->sq.gsi_cons) {
		memset(&wc[i], 0, sizeof(*wc));

		wc[i].qp = &qp->ibqp;
		wc[i].wr_id = qp->wqe_wr_id[qp->sq.cons].wr_id;
		wc[i].opcode = IB_WC_SEND;
		wc[i].status = IB_WC_SUCCESS;

		qlnxr_inc_sw_cons(&qp->sq);
		i++;
	}

	spin_unlock_irqrestore(&cq->cq_lock, flags);

	QL_DPRINT12(dev->ha, "exit i = %d\n", i);
	return i;
}