/*
 * Copyright (c) 2014-2019 The Linux Foundation. All rights reserved.
 * Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved.
 *
 * Permission to use, copy, modify, and/or distribute this software for
 * any purpose with or without fee is hereby granted, provided that the
 * above copyright notice and this permission notice appear in all
 * copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
 * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
 * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
 * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */

/*========================================================================

   \file  epping_tx.c

   \brief WLAN End Point Ping test tool implementation

   ========================================================================*/

/*--------------------------------------------------------------------------
   Include Files
   ------------------------------------------------------------------------*/
#include <cds_api.h>
#include <cds_sched.h>
#include <linux/etherdevice.h>
#include <linux/firmware.h>
#include <wni_api.h>
#include <wlan_ptt_sock_svc.h>
#include <linux/wireless.h>
#include <net/cfg80211.h>
#include <linux/rtnetlink.h>
#include <linux/semaphore.h>
#include <linux/ctype.h>
#include "epping_main.h"
#include "epping_internal.h"
#include "epping_test.h"

#define TX_RETRY_TIMEOUT_IN_MS 1

static bool enb_tx_dump;

void epping_tx_dup_pkt(epping_adapter_t *adapter,
		       HTC_ENDPOINT_ID eid, qdf_nbuf_t skb)
{
	struct epping_cookie *cookie = NULL;
	int skb_len, ret;
	qdf_nbuf_t new_skb;

	cookie = epping_alloc_cookie(adapter->pEpping_ctx);
	if (!cookie) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: epping_alloc_cookie returns no resource\n",
			   __func__);
		return;
	}
	new_skb = qdf_nbuf_copy(skb);
	if (!new_skb) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: qdf_nbuf_copy returns no resource\n", __func__);
		epping_free_cookie(adapter->pEpping_ctx, cookie);
		return;
	}
	SET_HTC_PACKET_INFO_TX(&cookie->HtcPkt,
			       cookie, qdf_nbuf_data(skb),
			       qdf_nbuf_len(new_skb), eid, 0);
	SET_HTC_PACKET_NET_BUF_CONTEXT(&cookie->HtcPkt, new_skb);
	skb_len = (int)qdf_nbuf_len(new_skb);
	/* send the packet */
	ret = htc_send_pkt(adapter->pEpping_ctx->HTCHandle, &cookie->HtcPkt);
	if (ret != QDF_STATUS_SUCCESS) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: htc_send_pkt failed, ret = %d\n", __func__, ret);
		epping_free_cookie(adapter->pEpping_ctx, cookie);
		qdf_nbuf_free(new_skb);
		return;
	}
	adapter->stats.tx_bytes += skb_len;
	++adapter->stats.tx_packets;
	if (((adapter->stats.tx_packets +
	      adapter->stats.tx_dropped) % EPPING_STATS_LOG_COUNT) == 0 &&
	    (adapter->stats.tx_packets || adapter->stats.tx_dropped)) {
		epping_log_stats(adapter, __func__);
	}
}

static int epping_tx_send_int(qdf_nbuf_t skb, epping_adapter_t *adapter)
{
	EPPING_HEADER *eppingHdr = (EPPING_HEADER *) qdf_nbuf_data(skb);
	HTC_ENDPOINT_ID eid = ENDPOINT_UNUSED;
	struct epping_cookie *cookie = NULL;
	uint8_t ac = 0;
	QDF_STATUS ret = QDF_STATUS_SUCCESS;
	int skb_len;
	EPPING_HEADER tmpHdr = *eppingHdr;

	/* allocate resource for this packet */
	cookie = epping_alloc_cookie(adapter->pEpping_ctx);
	/* no resource */
	if (!cookie) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: epping_alloc_cookie returns no resource\n",
			   __func__);
		return A_ERROR;
	}

	if (enb_tx_dump)
		epping_hex_dump((void *)eppingHdr, skb->len, __func__);
	/*
	 * a quirk of linux, the payload of the frame is 32-bit aligned and thus
	 * the addition of the HTC header will mis-align the start of the HTC
	 * frame, so we add some padding which will be stripped off in the target
	 */
	if (EPPING_ALIGNMENT_PAD > 0) {
		A_NETBUF_PUSH(skb, EPPING_ALIGNMENT_PAD);
	}
	/* prepare ep/HTC information */
	ac = eppingHdr->StreamNo_h;
	eid = adapter->pEpping_ctx->EppingEndpoint[ac];
	if (eid < 0 || eid >= EPPING_MAX_NUM_EPIDS) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: invalid eid = %d, ac = %d\n", __func__, eid,
			   ac);
		return A_ERROR;
	}
	if (tmpHdr.Cmd_h == EPPING_CMD_RESET_RECV_CNT ||
	    tmpHdr.Cmd_h == EPPING_CMD_CONT_RX_START) {
		epping_set_kperf_flag(adapter, eid, tmpHdr.CmdBuffer_t[0]);
	}
	SET_HTC_PACKET_INFO_TX(&cookie->HtcPkt,
			       cookie, qdf_nbuf_data(skb), qdf_nbuf_len(skb),
			       eid, 0);
	SET_HTC_PACKET_NET_BUF_CONTEXT(&cookie->HtcPkt, skb);
	skb_len = skb->len;
	/* send the packet */
	ret = htc_send_pkt(adapter->pEpping_ctx->HTCHandle, &cookie->HtcPkt);
	epping_log_packet(adapter, &tmpHdr, ret, __func__);
	if (ret != QDF_STATUS_SUCCESS) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: htc_send_pkt failed, status = %d\n", __func__,
			   ret);
		epping_free_cookie(adapter->pEpping_ctx, cookie);
		return A_ERROR;
	}
	adapter->stats.tx_bytes += skb_len;
	++adapter->stats.tx_packets;
	if (((adapter->stats.tx_packets +
	      adapter->stats.tx_dropped) % EPPING_STATS_LOG_COUNT) == 0 &&
	    (adapter->stats.tx_packets || adapter->stats.tx_dropped)) {
		epping_log_stats(adapter, __func__);
	}

	return 0;
}

void epping_tx_timer_expire(epping_adapter_t *adapter)
{
	qdf_nbuf_t nodrop_skb;

	EPPING_LOG(QDF_TRACE_LEVEL_INFO, "%s: queue len: %d\n", __func__,
		   qdf_nbuf_queue_len(&adapter->nodrop_queue));

	if (!qdf_nbuf_queue_len(&adapter->nodrop_queue)) {
		/* nodrop queue is empty so no need to arm timer */
		adapter->epping_timer_state = EPPING_TX_TIMER_STOPPED;
		return;
	}

	/* try to flush nodrop queue */
	while ((nodrop_skb = qdf_nbuf_queue_remove(&adapter->nodrop_queue))) {
		htc_set_nodrop_pkt(adapter->pEpping_ctx->HTCHandle, true);
		if (epping_tx_send_int(nodrop_skb, adapter)) {
			EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
				   "%s: nodrop: %pK xmit fail in timer\n",
				   __func__, nodrop_skb);
			/* fail to xmit so put the nodrop packet to the nodrop queue */
			qdf_nbuf_queue_insert_head(&adapter->nodrop_queue,
						   nodrop_skb);
			break;
		} else {
			htc_set_nodrop_pkt(adapter->pEpping_ctx->HTCHandle, false);
			EPPING_LOG(QDF_TRACE_LEVEL_INFO,
				   "%s: nodrop: %pK xmit ok in timer\n",
				   __func__, nodrop_skb);
		}
	}

	/* if nodrop queue is not empty, continue to arm timer */
	if (nodrop_skb) {
		qdf_spin_lock_bh(&adapter->data_lock);
		/* if nodrop queue is not empty, continue to arm timer */
		if (adapter->epping_timer_state != EPPING_TX_TIMER_RUNNING) {
			adapter->epping_timer_state = EPPING_TX_TIMER_RUNNING;
			qdf_timer_mod(&adapter->epping_timer,
					      TX_RETRY_TIMEOUT_IN_MS);
		}
		qdf_spin_unlock_bh(&adapter->data_lock);
	} else {
		adapter->epping_timer_state = EPPING_TX_TIMER_STOPPED;
	}
}

int epping_tx_send(qdf_nbuf_t skb, epping_adapter_t *adapter)
{
	qdf_nbuf_t nodrop_skb;
	EPPING_HEADER *eppingHdr;
	uint8_t ac = 0;

	eppingHdr = (EPPING_HEADER *) qdf_nbuf_data(skb);

	if (!IS_EPPING_PACKET(eppingHdr)) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: Received non endpoint ping packets\n", __func__);
		/* no packet to send, cleanup */
		qdf_nbuf_free(skb);
		return -ENOMEM;
	}

	/* the stream ID is mapped to an access class */
	ac = eppingHdr->StreamNo_h;
	/* hard coded two ep ids */
	if (ac != 0 && ac != 1) {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: ac %d is not mapped to mboxping service\n",
			   __func__, ac);
		qdf_nbuf_free(skb);
		return -ENOMEM;
	}

	/*
	 * some EPPING packets cannot be dropped no matter what access class
	 * it was sent on. A special care has been taken:
	 * 1. when there is no TX resource, queue the control packets to
	 *    a special queue
	 * 2. when there is TX resource, send the queued control packets first
	 *    and then other packets
	 * 3. a timer launches to check if there is queued control packets and
	 *    flush them
	 */

	/* check the nodrop queue first */
	while ((nodrop_skb = qdf_nbuf_queue_remove(&adapter->nodrop_queue))) {
		htc_set_nodrop_pkt(adapter->pEpping_ctx->HTCHandle, true);
		if (epping_tx_send_int(nodrop_skb, adapter)) {
			EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
				   "%s: nodrop: %pK xmit fail\n", __func__,
				   nodrop_skb);
			/* fail to xmit so put the nodrop packet to the nodrop queue */
			qdf_nbuf_queue_insert_head(&adapter->nodrop_queue,
						   nodrop_skb);
			/* no cookie so free the current skb */
			goto tx_fail;
		} else {
			htc_set_nodrop_pkt(adapter->pEpping_ctx->HTCHandle, false);
			EPPING_LOG(QDF_TRACE_LEVEL_INFO,
				   "%s: nodrop: %pK xmit ok\n", __func__,
				   nodrop_skb);
		}
	}

	/* send the original packet */
	if (epping_tx_send_int(skb, adapter))
		goto tx_fail;

	return 0;

tx_fail:
	if (!IS_EPING_PACKET_NO_DROP(eppingHdr)) {
		/* allow to drop the skb so drop it */
		qdf_nbuf_free(skb);
		++adapter->stats.tx_dropped;
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: Tx skb %pK dropped, stats.tx_dropped = %ld\n",
			   __func__, skb, adapter->stats.tx_dropped);
		return -ENOMEM;
	} else {
		EPPING_LOG(QDF_TRACE_LEVEL_FATAL,
			   "%s: nodrop: %pK queued\n", __func__, skb);
		qdf_nbuf_queue_add(&adapter->nodrop_queue, skb);
		qdf_spin_lock_bh(&adapter->data_lock);
		if (adapter->epping_timer_state != EPPING_TX_TIMER_RUNNING) {
			adapter->epping_timer_state = EPPING_TX_TIMER_RUNNING;
			qdf_timer_mod(&adapter->epping_timer,
					      TX_RETRY_TIMEOUT_IN_MS);
		}
		qdf_spin_unlock_bh(&adapter->data_lock);
	}

	return 0;
}

#ifdef HIF_SDIO
enum htc_send_full_action epping_tx_queue_full(void *Context,
						HTC_PACKET *pPacket)
{
	/*
	 * Call netif_stop_queue frequently will impact the mboxping tx t-put.
	 * Return HTC_SEND_FULL_KEEP directly in epping_tx_queue_full to avoid.
	 */
	return HTC_SEND_FULL_KEEP;
}
#endif /* HIF_SDIO */
void epping_tx_complete(void *ctx, HTC_PACKET *htc_pkt)
{
	epping_context_t *pEpping_ctx = (epping_context_t *) ctx;
	epping_adapter_t *adapter = pEpping_ctx->epping_adapter;
	struct net_device *dev = adapter->dev;
	QDF_STATUS status;
	HTC_ENDPOINT_ID eid;
	qdf_nbuf_t pktSkb;
	struct epping_cookie *cookie;
	A_BOOL flushing = false;
	qdf_nbuf_queue_t skb_queue;

	if (!htc_pkt)
		return;

	qdf_nbuf_queue_init(&skb_queue);

	qdf_spin_lock_bh(&adapter->data_lock);

	status = htc_pkt->Status;
	eid = htc_pkt->Endpoint;
	pktSkb = GET_HTC_PACKET_NET_BUF_CONTEXT(htc_pkt);
	cookie = htc_pkt->pPktContext;

	if (!pktSkb) {
		EPPING_LOG(QDF_TRACE_LEVEL_ERROR,
			   "%s: NULL skb from hc packet", __func__);
		QDF_BUG(0);
	} else {
		if (htc_pkt->pBuffer != qdf_nbuf_data(pktSkb)) {
			EPPING_LOG(QDF_TRACE_LEVEL_ERROR,
				   "%s: htc_pkt buffer not equal to skb->data",
				   __func__);
			QDF_BUG(0);
		}
		/* add this to the list, use faster non-lock API */
		qdf_nbuf_queue_add(&skb_queue, pktSkb);

		if (QDF_IS_STATUS_SUCCESS(status)) {
			if (htc_pkt->ActualLength !=
				qdf_nbuf_len(pktSkb)) {
				EPPING_LOG(QDF_TRACE_LEVEL_ERROR,
					   "%s: htc_pkt length not equal to skb->len",
					   __func__);
				QDF_BUG(0);
			}
		}
	}

	EPPING_LOG(QDF_TRACE_LEVEL_INFO,
		   "%s skb=%pK data=%pK len=0x%x eid=%d ",
		   __func__, pktSkb, htc_pkt->pBuffer,
		   htc_pkt->ActualLength, eid);

	if (QDF_IS_STATUS_ERROR(status)) {
		if (status == QDF_STATUS_E_CANCELED) {
			/* a packet was flushed  */
			flushing = true;
		}
		if (status != QDF_STATUS_E_RESOURCES) {
			EPPING_LOG(QDF_TRACE_LEVEL_ERROR,
				   "%s() -TX ERROR, status: 0x%x",
				   __func__, status);
		}
	} else {
		EPPING_LOG(QDF_TRACE_LEVEL_INFO, "%s: OK\n", __func__);
		flushing = false;
	}

	epping_free_cookie(adapter->pEpping_ctx, cookie);
	qdf_spin_unlock_bh(&adapter->data_lock);

	/* free all skbs in our local list */
	while (qdf_nbuf_queue_len(&skb_queue)) {
		/* use non-lock version */
		pktSkb = qdf_nbuf_queue_remove(&skb_queue);
		if (!pktSkb)
			break;
		qdf_nbuf_tx_free(pktSkb, QDF_NBUF_PKT_ERROR);
		pEpping_ctx->total_tx_acks++;
	}

	if (!flushing) {
		netif_wake_queue(dev);
	}
}