/* SPDX-License-Identifier: GPL-2.0 */
/******************************************************************************
 *
 * Copyright (c) 2020 Intel Corporation
 *
 *****************************************************************************/
#ifndef _PON_QOS_TC_QOS_
#define _PON_QOS_TC_QOS_

#include <net/pkt_cls.h>
#include <linux/list.h>
#include <linux/radix-tree.h>
#include <linux/types.h>

#define PON_QOS_TC_MAX_Q 8
#define PON_QOS_SCH_MAX_LEVEL 3

enum pon_qos_qdisc_type {
	PON_QOS_QDISC_NONE,
	PON_QOS_QDISC_PRIO,
	PON_QOS_QDISC_DRR,
	PON_QOS_QDISC_TBF,
	PON_QOS_QDISC_RED,
};

struct pon_qos_qdisc_ops {
	int (*offload)(struct net_device *dev, void *params);

	int (*unoffload)(struct net_device *dev, void *params);
};

struct pon_qos_q_data {
	u32 qid;
	int tc;
	atomic_t ref_cnt;
	int arbi;
	int p_w;
};

struct pon_qos_qdisc {
	struct net_device *dev;
	enum pon_qos_qdisc_type type;
	u32 parent;
	u32 handle;
	int use_cnt;

	/* HW specific settings */
	u32 sch_id;
	int inst;
	s32 port;
	s32 deq_idx;
	int epn;
	u16 def_q;
	bool ds;

	unsigned int num_q;
	struct pon_qos_q_data qids[PON_QOS_TC_MAX_Q];

	/* TODO: for multi-stage schedulers */
	unsigned int num_children;
	struct pon_qos_qdisc *children[PON_QOS_TC_MAX_Q];

	union {
		struct {
			u8 bands;
			u8 priomap[TC_PRIO_MAX + 1];
		} prio;

		struct {
			u8 num_tc;
			u8 priomap[TC_PRIO_MAX + 1];
		} mqprio;

		struct {
			int classes;
		} drr;
	};

	/* TODO: abstrac the hw specific part */
	struct pon_qos_qdisc_ops *ops;
};

struct pon_qos_port {
	struct net_device *dev;
	struct pon_qos_qdisc root_qdisc;
	unsigned int sch_num;
	struct radix_tree_root qdiscs;
	struct list_head list;
};

struct pon_qos_qmap_tc {
	int flags;
	u32 handle;
	struct net_device *indev;
	int tc;
};

int pon_qos_get_queue_by_handle(struct net_device *dev,
				u32 handle,
				struct pon_qos_q_data **qid);

struct pon_qos_port *pon_qos_port_get(struct net_device *dev);
struct pon_qos_port *pon_qos_port_alloc(struct net_device *dev);
int pon_qos_port_delete(struct pon_qos_port *port);
int pon_qos_alloc_qdisc(struct pon_qos_qdisc **qdisc);
void pon_qos_free_qdisc(struct pon_qos_qdisc *qdisc);
struct pon_qos_qdisc *pon_qos_qdisc_find(struct pon_qos_port *port, u32 handle);

int pon_qos_queue_add(struct pon_qos_qdisc *sch, int arbi, int prio_w, int idx);
int pon_qos_queue_del(struct pon_qos_qdisc *sch, int idx);

int pon_qos_qos_init(void);
int pon_qos_qos_destroy(void);

int pon_qos_add_sched(struct pon_qos_qdisc *sch, int prio);
int pon_qos_add_staged_sched(struct pon_qos_qdisc *psch,
			     struct pon_qos_qdisc *csch,
			     int prio);
int pon_qos_sched_del(struct pon_qos_qdisc *sch);
int pon_qos_add_child_qdisc(struct net_device *dev,
			    struct pon_qos_port *port,
			    enum pon_qos_qdisc_type type,
			    u32 parent,
			    u32 hanle);

int pon_qos_fill_port_data(struct pon_qos_qdisc *sch);
int pon_qos_get_port_info(struct pon_qos_qdisc *sch);

int pon_qos_tc_qdisc_unlink(struct pon_qos_port *p, struct pon_qos_qdisc *sch);
int pon_qos_qdisc_tree_del(struct pon_qos_port *p, struct pon_qos_qdisc *root);
int pon_qos_tc_sched_status(struct pon_qos_port *p, struct pon_qos_qdisc *root);

int pon_qos_update_qmap(struct net_device *dev,
			struct pon_qos_qmap_tc *q_tc, bool en);

int pon_qos_ports_cleanup(void);

#endif