/*
 * Copyright (c) 2015, Mellanox Technologies. 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.
 */

#include <linux/irq.h>
#ifdef HAVE_NDO_XSK_WAKEUP
#ifdef HAVE_XDP_SOCK_DRV_H
#include <net/xdp_sock_drv.h>
#else
#include <net/xdp_sock.h>
#endif
#endif
#include "en.h"
#include "en/txrx.h"
#include "en/xdp.h"
#ifdef HAVE_NDO_XSK_WAKEUP
#include "en/xsk/rx.h"
#endif
#include "en/xsk/tx.h"
#include "en_accel/ktls_txrx.h"

static inline bool mlx5e_channel_no_affinity_change(struct mlx5e_channel *c)
{
	int current_cpu = smp_processor_id();

	return cpumask_test_cpu(current_cpu, c->aff_mask);
}

static void mlx5e_handle_tx_dim(struct mlx5e_txqsq *sq)
{
	struct dim_sample *sample = &sq->dim_obj.sample;

	if (unlikely(!test_bit(MLX5E_SQ_STATE_AM, &sq->state)))
		return;

	dim_update_sample(sq->cq.event_ctr, sample->pkt_ctr, sample->byte_ctr, sample);
	net_dim(&sq->dim_obj.dim, *sample);
}

static void mlx5e_handle_rx_dim(struct mlx5e_rq *rq)
{
	struct dim_sample *sample = &rq->dim_obj.sample;

	if (unlikely(!test_bit(MLX5E_RQ_STATE_AM, &rq->state)))
		return;

	dim_update_sample(rq->cq.event_ctr, sample->pkt_ctr, sample->byte_ctr, sample);
	net_dim(&rq->dim_obj.dim, *sample);
}

static void mlx5e_rx_dim_cq_rearm(struct mlx5e_priv *priv, struct mlx5e_rq *rq)
{
	mlx5e_handle_rx_dim(rq);
	if (test_bit(MLX5E_RQ_STATE_ENABLED, &rq->state))
		mlx5e_cq_arm(&rq->cq);
}

void mlx5e_trigger_irq(struct mlx5e_icosq *sq)
{
	struct mlx5_wq_cyc *wq = &sq->wq;
	struct mlx5e_tx_wqe *nopwqe;
	u16 pi = mlx5_wq_cyc_ctr2ix(wq, sq->pc);

	sq->db.wqe_info[pi] = (struct mlx5e_icosq_wqe_info) {
		.wqe_type   = MLX5E_ICOSQ_WQE_NOP,
		.num_wqebbs = 1,
	};

	nopwqe = mlx5e_post_nop(wq, sq->sqn, &sq->pc);
	mlx5e_notify_hw(wq, sq->pc, sq->uar_map, &nopwqe->ctrl);
}

#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
static bool mlx5e_napi_xsk_post(struct mlx5e_xdpsq *xsksq, struct mlx5e_rq *xskrq)
{
#ifdef HAVE_NDO_XSK_WAKEUP
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
	bool need_wakeup = xsk_uses_need_wakeup(xskrq->xsk_pool);
#else
	bool need_wakeup = xsk_umem_uses_need_wakeup(xskrq->umem);
#endif /*HAVE_NETDEV_BPF_XSK_BUFF_POOL */

	bool busy_xsk = false, xsk_rx_alloc_err;

	/* If SQ is empty, there are no TX completions to trigger NAPI, so set
	 * need_wakeup. Do it before queuing packets for TX to avoid race
	 * condition with userspace.
	 */
	if (need_wakeup && xsksq->pc == xsksq->cc)
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
		xsk_set_tx_need_wakeup(xsksq->xsk_pool);
#else
		xsk_set_tx_need_wakeup(xsksq->umem);
#endif
	busy_xsk |= mlx5e_xsk_tx(xsksq, MLX5E_TX_XSK_POLL_BUDGET);
	/* If we queued some packets for TX, no need for wakeup anymore. */
	if (need_wakeup && xsksq->pc != xsksq->cc)
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
		xsk_clear_tx_need_wakeup(xsksq->xsk_pool);
#else
		xsk_clear_tx_need_wakeup(xsksq->umem);
#endif

	/* If WQ is empty, RX won't trigger NAPI, so set need_wakeup. Do it
	 * before refilling to avoid race condition with userspace.
	 */
	if (need_wakeup && !mlx5e_rqwq_get_cur_sz(xskrq))
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
		xsk_set_rx_need_wakeup(xskrq->xsk_pool);
#else
		xsk_set_rx_need_wakeup(xskrq->umem);
#endif
	xsk_rx_alloc_err = INDIRECT_CALL_2(xskrq->post_wqes,
					   mlx5e_post_rx_mpwqes,
					   mlx5e_post_rx_wqes,
					   xskrq);
	/* Ask for wakeup if WQ is not full after refill. */
	if (!need_wakeup)
		busy_xsk |= xsk_rx_alloc_err;
	else if (xsk_rx_alloc_err)
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
		xsk_set_rx_need_wakeup(xskrq->xsk_pool);
#else
		xsk_set_rx_need_wakeup(xskrq->umem);
#endif
	else
#ifdef HAVE_NETDEV_BPF_XSK_BUFF_POOL
		xsk_clear_rx_need_wakeup(xskrq->xsk_pool);
#else
		xsk_clear_rx_need_wakeup(xskrq->umem);
#endif

#else
	bool busy_xsk = false;

	busy_xsk |= mlx5e_xsk_tx(xsksq, MLX5E_TX_XSK_POLL_BUDGET);
	busy_xsk |= xskrq->post_wqes(xskrq);
#endif /* HAVE_NDO_XSK_WAKEUP*/
	return busy_xsk;
}
#endif /* HAVE_XSK_ZERO_COPY_SUPPORT */

int mlx5e_napi_poll(struct napi_struct *napi, int budget)
{
	struct mlx5e_channel *c = container_of(napi, struct mlx5e_channel,
					       napi);
	struct mlx5e_ch_stats *ch_stats = c->stats;
#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
       struct mlx5e_xdpsq *xsksq = &c->xsksq;
       struct mlx5e_rq *xskrq = &c->xskrq;
#endif
	struct mlx5e_txqsq __rcu **qos_sqs;
	struct mlx5e_rq *rq = &c->rq;
#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	bool aff_change = false;
	bool busy_xsk = false;
#endif
	bool busy = false;
	int work_done = 0;
	u16 qos_sqs_size = 0;
#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	bool xsk_open;
#endif
	int i;

	rcu_read_lock();

	qos_sqs = rcu_dereference(c->qos_sqs);

#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	xsk_open = test_bit(MLX5E_CHANNEL_STATE_XSK, c->state);
#endif

	ch_stats->poll++;
#ifndef HAVE_NAPI_STATE_MISSED
	clear_bit(MLX5E_CHANNEL_NAPI_SCHED, &c->flags);
#endif

	for (i = 0; i < c->num_tc; i++)
		busy |= mlx5e_poll_tx_cq(&c->sq[i].cq, budget);

	if (unlikely(qos_sqs)) {
		smp_rmb(); /* Pairs with mlx5e_qos_alloc_queues. */
		qos_sqs_size = READ_ONCE(c->qos_sqs_size);

		for (i = 0; i < qos_sqs_size; i++) {
			struct mlx5e_txqsq *sq = rcu_dereference(qos_sqs[i]);

			if (sq)
				busy |= mlx5e_poll_tx_cq(&sq->cq, budget);
		}
	}

	/* budget=0 means we may be in IRQ context, do as little as possible */
	if (unlikely(!budget))
		goto out;

#ifdef HAVE_XDP_SUPPORT
	busy |= mlx5e_poll_xdpsq_cq(&c->xdpsq.cq);

	if (c->xdp)
		busy |= mlx5e_poll_xdpsq_cq(&c->rq_xdpsq.cq);
#endif

#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	if (xsk_open)
		work_done = mlx5e_poll_rx_cq(&xskrq->cq, budget);
#endif

	if (likely(budget - work_done))
		work_done += mlx5e_poll_rx_cq(&rq->cq, budget - work_done);

	busy |= work_done == budget;

	mlx5e_poll_ico_cq(&c->icosq.cq);
	if (mlx5e_poll_ico_cq(&c->async_icosq.cq))
		/* Don't clear the flag if nothing was polled to prevent
		 * queueing more WQEs and overflowing the async ICOSQ.
		 */
		clear_bit(MLX5E_SQ_STATE_PENDING_XSK_TX, &c->async_icosq.state);

	/* Keep after async ICOSQ CQ poll */
	if (unlikely(mlx5e_ktls_rx_pending_resync_list(c, budget)))
		busy |= mlx5e_ktls_rx_handle_resync_list(c, budget);

	busy |= INDIRECT_CALL_2(rq->post_wqes,
				mlx5e_post_rx_mpwqes,
				mlx5e_post_rx_wqes,
				rq);
#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	if (xsk_open) {
		busy |= mlx5e_poll_xdpsq_cq(&xsksq->cq);
		busy_xsk |= mlx5e_napi_xsk_post(xsksq, xskrq);
	}

	busy |= busy_xsk;
#endif

	if (busy) {
		if (likely(mlx5e_channel_no_affinity_change(c))) {
			work_done = budget;
			goto out;
		}
		ch_stats->aff_change++;
#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
		aff_change = true;
#endif
		if (budget && work_done == budget)
			work_done--;
	}

#ifdef HAVE_NAPI_STATE_MISSED
	if (unlikely(!napi_complete_done(napi, work_done)))
		goto out;
#else
 	napi_complete_done(napi, work_done);
 
	/* avoid losing completion event during/after polling cqs */
	if (test_bit(MLX5E_CHANNEL_NAPI_SCHED, &c->flags)) {
		napi_schedule(napi);
		goto out;
	}
#endif

	ch_stats->arm++;

	for (i = 0; i < c->num_tc; i++) {
		mlx5e_handle_tx_dim(&c->sq[i]);
		mlx5e_cq_arm(&c->sq[i].cq);
	}
	if (unlikely(qos_sqs)) {
		for (i = 0; i < qos_sqs_size; i++) {
			struct mlx5e_txqsq *sq = rcu_dereference(qos_sqs[i]);

			if (sq) {
				mlx5e_handle_tx_dim(sq);
				mlx5e_cq_arm(&sq->cq);
			}
		}
	}

	mlx5e_rx_dim_cq_rearm(c->priv, rq);
	mlx5e_cq_arm(&c->icosq.cq);
	mlx5e_cq_arm(&c->async_icosq.cq);
#ifdef HAVE_XDP_SUPPORT
	mlx5e_cq_arm(&c->xdpsq.cq);
#endif

#ifdef HAVE_XSK_ZERO_COPY_SUPPORT
	if (xsk_open) {
		mlx5e_rx_dim_cq_rearm(c->priv, xskrq);
		mlx5e_cq_arm(&xsksq->cq);
	}

	if (unlikely(aff_change && busy_xsk)) {
		mlx5e_trigger_irq(&c->icosq);
		ch_stats->force_irq++;
	}
#endif

out:
	rcu_read_unlock();

	return work_done;
}

void mlx5e_completion_event(struct mlx5_core_cq *mcq, struct mlx5_eqe *eqe)
{
	struct mlx5e_cq *cq = container_of(mcq, struct mlx5e_cq, mcq);

#ifndef HAVE_NAPI_STATE_MISSED
	set_bit(MLX5E_CHANNEL_NAPI_SCHED, cq->ch_flags);
#endif
	napi_schedule(cq->napi);
	cq->event_ctr++;
	cq->ch_stats->events++;
}

void mlx5e_cq_error_event(struct mlx5_core_cq *mcq, enum mlx5_event event)
{
	struct mlx5e_cq *cq = container_of(mcq, struct mlx5e_cq, mcq);
	struct net_device *netdev = cq->netdev;

	netdev_err(netdev, "%s: cqn=0x%.6x event=0x%.2x\n",
		   __func__, mcq->cqn, event);
}
