cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

act_api.h (9384B)


      1/* SPDX-License-Identifier: GPL-2.0 */
      2#ifndef __NET_ACT_API_H
      3#define __NET_ACT_API_H
      4
      5/*
      6 * Public action API for classifiers/qdiscs
      7*/
      8
      9#include <linux/refcount.h>
     10#include <net/flow_offload.h>
     11#include <net/sch_generic.h>
     12#include <net/pkt_sched.h>
     13#include <net/net_namespace.h>
     14#include <net/netns/generic.h>
     15
     16struct tcf_idrinfo {
     17	struct mutex	lock;
     18	struct idr	action_idr;
     19	struct net	*net;
     20};
     21
     22struct tc_action_ops;
     23
     24struct tc_action {
     25	const struct tc_action_ops	*ops;
     26	__u32				type; /* for backward compat(TCA_OLD_COMPAT) */
     27	struct tcf_idrinfo		*idrinfo;
     28
     29	u32				tcfa_index;
     30	refcount_t			tcfa_refcnt;
     31	atomic_t			tcfa_bindcnt;
     32	int				tcfa_action;
     33	struct tcf_t			tcfa_tm;
     34	struct gnet_stats_basic_sync	tcfa_bstats;
     35	struct gnet_stats_basic_sync	tcfa_bstats_hw;
     36	struct gnet_stats_queue		tcfa_qstats;
     37	struct net_rate_estimator __rcu *tcfa_rate_est;
     38	spinlock_t			tcfa_lock;
     39	struct gnet_stats_basic_sync __percpu *cpu_bstats;
     40	struct gnet_stats_basic_sync __percpu *cpu_bstats_hw;
     41	struct gnet_stats_queue __percpu *cpu_qstats;
     42	struct tc_cookie	__rcu *act_cookie;
     43	struct tcf_chain	__rcu *goto_chain;
     44	u32			tcfa_flags;
     45	u8			hw_stats;
     46	u8			used_hw_stats;
     47	bool			used_hw_stats_valid;
     48	u32			in_hw_count;
     49};
     50#define tcf_index	common.tcfa_index
     51#define tcf_refcnt	common.tcfa_refcnt
     52#define tcf_bindcnt	common.tcfa_bindcnt
     53#define tcf_action	common.tcfa_action
     54#define tcf_tm		common.tcfa_tm
     55#define tcf_bstats	common.tcfa_bstats
     56#define tcf_qstats	common.tcfa_qstats
     57#define tcf_rate_est	common.tcfa_rate_est
     58#define tcf_lock	common.tcfa_lock
     59
     60#define TCA_ACT_HW_STATS_ANY (TCA_ACT_HW_STATS_IMMEDIATE | \
     61			      TCA_ACT_HW_STATS_DELAYED)
     62
     63/* Reserve 16 bits for user-space. See TCA_ACT_FLAGS_NO_PERCPU_STATS. */
     64#define TCA_ACT_FLAGS_USER_BITS 16
     65#define TCA_ACT_FLAGS_USER_MASK 0xffff
     66#define TCA_ACT_FLAGS_POLICE	(1U << TCA_ACT_FLAGS_USER_BITS)
     67#define TCA_ACT_FLAGS_BIND	(1U << (TCA_ACT_FLAGS_USER_BITS + 1))
     68#define TCA_ACT_FLAGS_REPLACE	(1U << (TCA_ACT_FLAGS_USER_BITS + 2))
     69#define TCA_ACT_FLAGS_NO_RTNL	(1U << (TCA_ACT_FLAGS_USER_BITS + 3))
     70
     71/* Update lastuse only if needed, to avoid dirtying a cache line.
     72 * We use a temp variable to avoid fetching jiffies twice.
     73 */
     74static inline void tcf_lastuse_update(struct tcf_t *tm)
     75{
     76	unsigned long now = jiffies;
     77
     78	if (tm->lastuse != now)
     79		tm->lastuse = now;
     80	if (unlikely(!tm->firstuse))
     81		tm->firstuse = now;
     82}
     83
     84static inline void tcf_tm_dump(struct tcf_t *dtm, const struct tcf_t *stm)
     85{
     86	dtm->install = jiffies_to_clock_t(jiffies - stm->install);
     87	dtm->lastuse = jiffies_to_clock_t(jiffies - stm->lastuse);
     88	dtm->firstuse = stm->firstuse ?
     89		jiffies_to_clock_t(jiffies - stm->firstuse) : 0;
     90	dtm->expires = jiffies_to_clock_t(stm->expires);
     91}
     92
     93static inline enum flow_action_hw_stats tc_act_hw_stats(u8 hw_stats)
     94{
     95	if (WARN_ON_ONCE(hw_stats > TCA_ACT_HW_STATS_ANY))
     96		return FLOW_ACTION_HW_STATS_DONT_CARE;
     97	else if (!hw_stats)
     98		return FLOW_ACTION_HW_STATS_DISABLED;
     99
    100	return hw_stats;
    101}
    102
    103#ifdef CONFIG_NET_CLS_ACT
    104
    105#define ACT_P_CREATED 1
    106#define ACT_P_DELETED 1
    107
    108typedef void (*tc_action_priv_destructor)(void *priv);
    109
    110struct tc_action_ops {
    111	struct list_head head;
    112	char    kind[IFNAMSIZ];
    113	enum tca_id  id; /* identifier should match kind */
    114	size_t	size;
    115	struct module		*owner;
    116	int     (*act)(struct sk_buff *, const struct tc_action *,
    117		       struct tcf_result *); /* called under RCU BH lock*/
    118	int     (*dump)(struct sk_buff *, struct tc_action *, int, int);
    119	void	(*cleanup)(struct tc_action *);
    120	int     (*lookup)(struct net *net, struct tc_action **a, u32 index);
    121	int     (*init)(struct net *net, struct nlattr *nla,
    122			struct nlattr *est, struct tc_action **act,
    123			struct tcf_proto *tp,
    124			u32 flags, struct netlink_ext_ack *extack);
    125	int     (*walk)(struct net *, struct sk_buff *,
    126			struct netlink_callback *, int,
    127			const struct tc_action_ops *,
    128			struct netlink_ext_ack *);
    129	void	(*stats_update)(struct tc_action *, u64, u64, u64, u64, bool);
    130	size_t  (*get_fill_size)(const struct tc_action *act);
    131	struct net_device *(*get_dev)(const struct tc_action *a,
    132				      tc_action_priv_destructor *destructor);
    133	struct psample_group *
    134	(*get_psample_group)(const struct tc_action *a,
    135			     tc_action_priv_destructor *destructor);
    136	int     (*offload_act_setup)(struct tc_action *act, void *entry_data,
    137				     u32 *index_inc, bool bind,
    138				     struct netlink_ext_ack *extack);
    139};
    140
    141struct tc_action_net {
    142	struct tcf_idrinfo *idrinfo;
    143	const struct tc_action_ops *ops;
    144};
    145
    146static inline
    147int tc_action_net_init(struct net *net, struct tc_action_net *tn,
    148		       const struct tc_action_ops *ops)
    149{
    150	int err = 0;
    151
    152	tn->idrinfo = kmalloc(sizeof(*tn->idrinfo), GFP_KERNEL);
    153	if (!tn->idrinfo)
    154		return -ENOMEM;
    155	tn->ops = ops;
    156	tn->idrinfo->net = net;
    157	mutex_init(&tn->idrinfo->lock);
    158	idr_init(&tn->idrinfo->action_idr);
    159	return err;
    160}
    161
    162void tcf_idrinfo_destroy(const struct tc_action_ops *ops,
    163			 struct tcf_idrinfo *idrinfo);
    164
    165static inline void tc_action_net_exit(struct list_head *net_list,
    166				      unsigned int id)
    167{
    168	struct net *net;
    169
    170	rtnl_lock();
    171	list_for_each_entry(net, net_list, exit_list) {
    172		struct tc_action_net *tn = net_generic(net, id);
    173
    174		tcf_idrinfo_destroy(tn->ops, tn->idrinfo);
    175		kfree(tn->idrinfo);
    176	}
    177	rtnl_unlock();
    178}
    179
    180int tcf_generic_walker(struct tc_action_net *tn, struct sk_buff *skb,
    181		       struct netlink_callback *cb, int type,
    182		       const struct tc_action_ops *ops,
    183		       struct netlink_ext_ack *extack);
    184int tcf_idr_search(struct tc_action_net *tn, struct tc_action **a, u32 index);
    185int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est,
    186		   struct tc_action **a, const struct tc_action_ops *ops,
    187		   int bind, bool cpustats, u32 flags);
    188int tcf_idr_create_from_flags(struct tc_action_net *tn, u32 index,
    189			      struct nlattr *est, struct tc_action **a,
    190			      const struct tc_action_ops *ops, int bind,
    191			      u32 flags);
    192void tcf_idr_insert_many(struct tc_action *actions[]);
    193void tcf_idr_cleanup(struct tc_action_net *tn, u32 index);
    194int tcf_idr_check_alloc(struct tc_action_net *tn, u32 *index,
    195			struct tc_action **a, int bind);
    196int tcf_idr_release(struct tc_action *a, bool bind);
    197
    198int tcf_register_action(struct tc_action_ops *a, struct pernet_operations *ops);
    199int tcf_unregister_action(struct tc_action_ops *a,
    200			  struct pernet_operations *ops);
    201int tcf_action_destroy(struct tc_action *actions[], int bind);
    202int tcf_action_exec(struct sk_buff *skb, struct tc_action **actions,
    203		    int nr_actions, struct tcf_result *res);
    204int tcf_action_init(struct net *net, struct tcf_proto *tp, struct nlattr *nla,
    205		    struct nlattr *est,
    206		    struct tc_action *actions[], int init_res[], size_t *attr_size,
    207		    u32 flags, u32 fl_flags, struct netlink_ext_ack *extack);
    208struct tc_action_ops *tc_action_load_ops(struct nlattr *nla, bool police,
    209					 bool rtnl_held,
    210					 struct netlink_ext_ack *extack);
    211struct tc_action *tcf_action_init_1(struct net *net, struct tcf_proto *tp,
    212				    struct nlattr *nla, struct nlattr *est,
    213				    struct tc_action_ops *a_o, int *init_res,
    214				    u32 flags, struct netlink_ext_ack *extack);
    215int tcf_action_dump(struct sk_buff *skb, struct tc_action *actions[], int bind,
    216		    int ref, bool terse);
    217int tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int, int);
    218int tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int, int);
    219
    220static inline void tcf_action_update_bstats(struct tc_action *a,
    221					    struct sk_buff *skb)
    222{
    223	if (likely(a->cpu_bstats)) {
    224		bstats_update(this_cpu_ptr(a->cpu_bstats), skb);
    225		return;
    226	}
    227	spin_lock(&a->tcfa_lock);
    228	bstats_update(&a->tcfa_bstats, skb);
    229	spin_unlock(&a->tcfa_lock);
    230}
    231
    232static inline void tcf_action_inc_drop_qstats(struct tc_action *a)
    233{
    234	if (likely(a->cpu_qstats)) {
    235		qstats_drop_inc(this_cpu_ptr(a->cpu_qstats));
    236		return;
    237	}
    238	spin_lock(&a->tcfa_lock);
    239	qstats_drop_inc(&a->tcfa_qstats);
    240	spin_unlock(&a->tcfa_lock);
    241}
    242
    243static inline void tcf_action_inc_overlimit_qstats(struct tc_action *a)
    244{
    245	if (likely(a->cpu_qstats)) {
    246		qstats_overlimit_inc(this_cpu_ptr(a->cpu_qstats));
    247		return;
    248	}
    249	spin_lock(&a->tcfa_lock);
    250	qstats_overlimit_inc(&a->tcfa_qstats);
    251	spin_unlock(&a->tcfa_lock);
    252}
    253
    254void tcf_action_update_stats(struct tc_action *a, u64 bytes, u64 packets,
    255			     u64 drops, bool hw);
    256int tcf_action_copy_stats(struct sk_buff *, struct tc_action *, int);
    257
    258int tcf_action_update_hw_stats(struct tc_action *action);
    259int tcf_action_reoffload_cb(flow_indr_block_bind_cb_t *cb,
    260			    void *cb_priv, bool add);
    261int tcf_action_check_ctrlact(int action, struct tcf_proto *tp,
    262			     struct tcf_chain **handle,
    263			     struct netlink_ext_ack *newchain);
    264struct tcf_chain *tcf_action_set_ctrlact(struct tc_action *a, int action,
    265					 struct tcf_chain *newchain);
    266
    267#ifdef CONFIG_INET
    268DECLARE_STATIC_KEY_FALSE(tcf_frag_xmit_count);
    269#endif
    270
    271int tcf_dev_queue_xmit(struct sk_buff *skb, int (*xmit)(struct sk_buff *skb));
    272
    273#else /* !CONFIG_NET_CLS_ACT */
    274
    275static inline int tcf_action_reoffload_cb(flow_indr_block_bind_cb_t *cb,
    276					  void *cb_priv, bool add) {
    277	return 0;
    278}
    279
    280#endif /* CONFIG_NET_CLS_ACT */
    281
    282static inline void tcf_action_stats_update(struct tc_action *a, u64 bytes,
    283					   u64 packets, u64 drops,
    284					   u64 lastuse, bool hw)
    285{
    286#ifdef CONFIG_NET_CLS_ACT
    287	if (!a->ops->stats_update)
    288		return;
    289
    290	a->ops->stats_update(a, bytes, packets, drops, lastuse, hw);
    291#endif
    292}
    293
    294
    295#endif