/*
 **************************************************************************
 * Copyright (c) 2019, 2021, The Linux Foundation. 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.
 **************************************************************************
 */

#include "nss_core.h"
#include <nss_qvpn.h>
#include "nss_qvpn_stats.h"
#include "nss_qvpn_strings.h"

/*
 * Declare atomic notifier data structure for statistics.
 */
ATOMIC_NOTIFIER_HEAD(nss_qvpn_stats_notifier);

/*
 * Spinlock to protect qvpn statistics update/read
 */
DEFINE_SPINLOCK(nss_qvpn_stats_lock);

uint64_t nss_qvpn_stats[NSS_MAX_NET_INTERFACES][NSS_STATS_NODE_MAX]; /* to store the qvpn statistics */

/*
 * nss_qvpn_stats_iface_type()
 *	Return a string for each interface type.
 */
static const char *nss_qvpn_stats_iface_type(enum nss_dynamic_interface_type type)
{
	switch (type) {
	case NSS_DYNAMIC_INTERFACE_TYPE_QVPN_INNER:
		return "qvpn_inner";

	case NSS_DYNAMIC_INTERFACE_TYPE_QVPN_OUTER:
		return "qvpn_outer";

	default:
		return "invalid_interface";
	}
}

/*
 * nss_qvpn_stats_read()
 *	Read qvpn node statiistics.
 */
static ssize_t nss_qvpn_stats_read(struct file *fp, char __user *ubuf, size_t sz, loff_t *ppos)
{
	/*
	 * Max output lines = #stats +
	 * few blank lines for banner printing + Number of Extra outputlines
	 * for future reference to add new stats
	 */
	uint32_t max_output_lines = NSS_STATS_NODE_MAX + NSS_STATS_EXTRA_OUTPUT_LINES;
	size_t size_al = NSS_STATS_MAX_STR_LENGTH * max_output_lines;
	struct nss_ctx_instance *nss_ctx = nss_qvpn_get_context();
	enum nss_dynamic_interface_type type;
	unsigned long *ifmap;
	uint64_t *stats_shadow;
	ssize_t bytes_read = 0;
	size_t size_wr = 0;
	uint32_t if_num;
	int32_t i;
	int count;
	char *lbuf;

	ifmap = nss_qvpn_ifmap_get();
	count = bitmap_weight(ifmap, NSS_MAX_NET_INTERFACES);
	if (count) {
		size_al = size_al * count;
	}

	lbuf = vzalloc(size_al);
	if (unlikely(!lbuf)) {
		nss_warning("Could not allocate memory for local statistics buffer");
		return -ENOMEM;
	}

	stats_shadow = vzalloc(NSS_STATS_NODE_MAX * 8);
	if (unlikely(!stats_shadow)) {
		nss_warning("Could not allocate memory for local shadow buffer");
		vfree(lbuf);
		return -ENOMEM;
	}

	/*
	 * Common node stats for each QVPN dynamic interface.
	 */
	size_wr += nss_stats_banner(lbuf, size_wr, size_al, "qvpn stats", NSS_STATS_SINGLE_CORE);
	for_each_set_bit(if_num, ifmap, NSS_MAX_NET_INTERFACES) {

		type = nss_dynamic_interface_get_type(nss_ctx, if_num);
		if ((type != NSS_DYNAMIC_INTERFACE_TYPE_QVPN_INNER) &&
			(type != NSS_DYNAMIC_INTERFACE_TYPE_QVPN_OUTER)) {
			continue;
		}

		spin_lock_bh(&nss_qvpn_stats_lock);
		for (i = 0; i < NSS_STATS_NODE_MAX; i++) {
			stats_shadow[i] = nss_qvpn_stats[if_num][i];
		}
		spin_unlock_bh(&nss_qvpn_stats_lock);

		size_wr += scnprintf(lbuf + size_wr, size_al - size_wr, "\n%s if_num:%03u\n",
					nss_qvpn_stats_iface_type(type), if_num);
		size_wr += nss_stats_print("qvpn", NULL, NSS_STATS_SINGLE_INSTANCE, nss_qvpn_strings_stats,
						stats_shadow, NSS_STATS_NODE_MAX, lbuf, size_wr, size_al);
	}

	bytes_read = simple_read_from_buffer(ubuf, sz, ppos, lbuf, strlen(lbuf));
	vfree(lbuf);
	vfree(stats_shadow);
	return bytes_read;
}

/*
 * nss_qvpn_stats_ops
 */
NSS_STATS_DECLARE_FILE_OPERATIONS(qvpn)

/*
 * nss_qvpn_stats_tunnel_sync
 *	Update qvpn interface statistics.
 */
void nss_qvpn_stats_tunnel_sync(struct nss_ctx_instance *nss_ctx, struct nss_cmn_msg *ncm)
{
	struct nss_qvpn_msg *ndcm = (struct nss_qvpn_msg *)ncm;
	struct nss_qvpn_stats_sync_msg  *msg_stats = &ndcm->msg.stats;

	spin_lock_bh(&nss_qvpn_stats_lock);

	/*
	 * Update common node stats
	 */
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_PKTS] += msg_stats->node_stats.rx_packets;
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_BYTES] += msg_stats->node_stats.rx_bytes;
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_QUEUE_0_DROPPED] += msg_stats->node_stats.rx_dropped[0];
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_QUEUE_1_DROPPED] += msg_stats->node_stats.rx_dropped[1];
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_QUEUE_2_DROPPED] += msg_stats->node_stats.rx_dropped[2];
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_RX_QUEUE_3_DROPPED] += msg_stats->node_stats.rx_dropped[3];

	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_TX_PKTS] += msg_stats->node_stats.tx_packets;
	nss_qvpn_stats[ncm->interface][NSS_STATS_NODE_TX_BYTES] += msg_stats->node_stats.tx_bytes;

	spin_unlock_bh(&nss_qvpn_stats_lock);
}

/*
 * nss_qvpn_stats_notify()
 *	Sends notifications to all the registered modules.
 *
 * Leverage NSS-FW statistics timing to update Netlink.
 */
void nss_qvpn_stats_notify(struct nss_ctx_instance *nss_ctx, uint32_t if_num)
{
	struct nss_qvpn_stats_notification qvpn_stats;

	spin_lock_bh(&nss_qvpn_stats_lock);
	qvpn_stats.core_id = nss_ctx->id;
	qvpn_stats.if_num = if_num;
	memcpy(qvpn_stats.stats_ctx, nss_qvpn_stats[if_num], sizeof(qvpn_stats.stats_ctx));
	spin_unlock_bh(&nss_qvpn_stats_lock);

	atomic_notifier_call_chain(&nss_qvpn_stats_notifier, NSS_STATS_EVENT_NOTIFY, &qvpn_stats);
}

/*
 * nss_qvpn_stats_unregister_notifier()
 *	Deregisters statistics notifier.
 */
int nss_qvpn_stats_unregister_notifier(struct notifier_block *nb)
{
	return atomic_notifier_chain_unregister(&nss_qvpn_stats_notifier, nb);
}
EXPORT_SYMBOL(nss_qvpn_stats_unregister_notifier);

/*
 * nss_qvpn_stats_register_notifier()
 *	Registers statistics notifier.
 */
int nss_qvpn_stats_register_notifier(struct notifier_block *nb)
{
	return atomic_notifier_chain_register(&nss_qvpn_stats_notifier, nb);
}
EXPORT_SYMBOL(nss_qvpn_stats_register_notifier);

/*
 * nss_qvpn_stats_dentry_create()
 *	Create QVPN statistics debug entry.
 */
void nss_qvpn_stats_dentry_create(void)
{
	nss_stats_create_dentry("qvpn", &nss_qvpn_stats_ops);
}