/*
 * 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>
#include <net/xdp_sock_drv.h>
#include "en.h"
#include "en/txrx.h"
#include "en/xdp.h"
#include "en/xsk/rx.h"
#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 mlx5e_sq_stats *stats = sq->stats;
        struct dim_sample dim_sample = {};

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

        dim_update_sample(sq->cq.event_ctr, stats->packets, stats->bytes, &dim_sample);
        net_dim(sq->dim, &dim_sample);
}

static void mlx5e_handle_rx_dim(struct mlx5e_rq *rq)
{
        struct mlx5e_rq_stats *stats = rq->stats;
        struct dim_sample dim_sample = {};

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

        dim_update_sample(rq->cq.event_ctr, stats->packets, stats->bytes, &dim_sample);
        net_dim(rq->dim, &dim_sample);
}

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);
}

static bool mlx5e_napi_xsk_post(struct mlx5e_xdpsq *xsksq, struct mlx5e_rq *xskrq)
{
        bool need_wakeup = xsk_uses_need_wakeup(xskrq->xsk_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)
                xsk_set_tx_need_wakeup(xsksq->xsk_pool);
        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)
                xsk_clear_tx_need_wakeup(xsksq->xsk_pool);

        /* 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))
                xsk_set_rx_need_wakeup(xskrq->xsk_pool);
        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)
                xsk_set_rx_need_wakeup(xskrq->xsk_pool);
        else
                xsk_clear_rx_need_wakeup(xskrq->xsk_pool);

        return busy_xsk;
}

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

        rcu_read_lock();

        qos_sqs = rcu_dereference(c->qos_sqs);

        xsk_open = test_bit(MLX5E_CHANNEL_STATE_XSK, c->state);

        ch_stats->poll++;

        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;

        if (c->xdpsq)
                busy |= mlx5e_poll_xdpsq_cq(&c->xdpsq->cq);

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

        if (xsk_open)
                work_done = mlx5e_poll_rx_cq(&xskrq->cq, budget);

        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 (aicosq) {
                if (mlx5e_poll_ico_cq(&aicosq->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,
                                  &aicosq->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);
        if (xsk_open) {
                busy |= mlx5e_poll_xdpsq_cq(&xsksq->cq);
                busy_xsk |= mlx5e_napi_xsk_post(xsksq, xskrq);
        }

        busy |= busy_xsk;

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

        if (unlikely(!napi_complete_done(napi, work_done)))
                goto out;

        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_handle_rx_dim(rq);

        mlx5e_cq_arm(&rq->cq);
        mlx5e_cq_arm(&c->icosq.cq);
        if (aicosq) {
                mlx5e_cq_arm(&aicosq->cq);
                if (xsk_open) {
                        mlx5e_handle_rx_dim(xskrq);
                        mlx5e_cq_arm(&xsksq->cq);
                        mlx5e_cq_arm(&xskrq->cq);
                }
        }
        if (c->xdpsq)
                mlx5e_cq_arm(&c->xdpsq->cq);

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

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);

        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);
}