include/net/psample.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef __NET_PSAMPLE_H
   3 #define __NET_PSAMPLE_H
   4
   5 #include <uapi/linux/psample.h>
   6 #include <linux/list.h>
   7
   8 struct psample_group {
   9         struct list_head list;
  10         struct net *net;
  11         u32 group_num;
  12         u32 refcount;
  13         u32 seq;
  14         struct rcu_head rcu;
  15 };
  16
  17 struct psample_metadata {
  18         u32 trunc_size;
  19         int in_ifindex;
  20         int out_ifindex;
  21         u16 out_tc;
  22         u64 out_tc_occ; /* bytes */
  23         u64 latency;    /* nanoseconds */
  24         u8 out_tc_valid:1,
  25            out_tc_occ_valid:1,
  26            latency_valid:1,
  27            rate_as_probability:1,
  28            unused:4;
  29         const u8 *user_cookie;
  30         u32 user_cookie_len;
  31 };
  32
  33 struct psample_group *psample_group_get(struct net *net, u32 group_num);
  34 void psample_group_take(struct psample_group *group);
  35 void psample_group_put(struct psample_group *group);
  36
  37 struct sk_buff;
  38
  39 #if IS_ENABLED(CONFIG_PSAMPLE)
  40
  41 void psample_sample_packet(struct psample_group *group,
  42                            const struct sk_buff *skb, u32 sample_rate,
  43                            const struct psample_metadata *md);
  44
  45 #else
  46
  47 static inline void psample_sample_packet(struct psample_group *group,
  48                                          const struct sk_buff *skb,
  49                                          u32 sample_rate,
  50                                          const struct psample_metadata *md)
  51 {
  52 }
  53
  54 #endif
  55
  56 #endif /* __NET_PSAMPLE_H */