1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _INET_DIAG_H_ 3 #define _INET_DIAG_H_ 1 4 5 #include <net/netlink.h> 6 #include <uapi/linux/inet_diag.h> 7 8 struct inet_hashinfo; 9 10 struct inet_diag_handler { 11 void (*dump)(struct sk_buff *skb, 12 struct netlink_callback *cb, 13 const struct inet_diag_req_v2 *r); 14 15 int (*dump_one)(struct netlink_callback *cb, 16 const struct inet_diag_req_v2 *req); 17 18 void (*idiag_get_info)(struct sock *sk, 19 struct inet_diag_msg *r, 20 void *info); 21 22 int (*idiag_get_aux)(struct sock *sk, 23 bool net_admin, 24 struct sk_buff *skb); 25 26 size_t (*idiag_get_aux_size)(struct sock *sk, 27 bool net_admin); 28 29 int (*destroy)(struct sk_buff *in_skb, 30 const struct inet_diag_req_v2 *req); 31 32 __u16 idiag_type; 33 __u16 idiag_info_size; 34 }; 35 36 struct bpf_sk_storage_diag; 37 struct inet_diag_dump_data { 38 struct nlattr *req_nlas[__INET_DIAG_REQ_MAX]; 39 #define inet_diag_nla_bc req_nlas[INET_DIAG_REQ_BYTECODE] 40 #define inet_diag_nla_bpf_stgs req_nlas[INET_DIAG_REQ_SK_BPF_STORAGES] 41 42 struct bpf_sk_storage_diag *bpf_stg_diag; 43 }; 44 45 struct inet_connection_sock; 46 int inet_sk_diag_fill(struct sock *sk, struct inet_connection_sock *icsk, 47 struct sk_buff *skb, struct netlink_callback *cb, 48 const struct inet_diag_req_v2 *req, 49 u16 nlmsg_flags, bool net_admin); 50 void inet_diag_dump_icsk(struct inet_hashinfo *h, struct sk_buff *skb, 51 struct netlink_callback *cb, 52 const struct inet_diag_req_v2 *r); 53 int inet_diag_dump_one_icsk(struct inet_hashinfo *hashinfo, 54 struct netlink_callback *cb, 55 const struct inet_diag_req_v2 *req); 56 57 struct sock *inet_diag_find_one_icsk(struct net *net, 58 struct inet_hashinfo *hashinfo, 59 const struct inet_diag_req_v2 *req); 60 61 int inet_diag_bc_sk(const struct nlattr *_bc, struct sock *sk); 62 63 void inet_diag_msg_common_fill(struct inet_diag_msg *r, struct sock *sk); 64 65 static inline size_t inet_diag_msg_attrs_size(void) 66 { 67 return nla_total_size(1) /* INET_DIAG_SHUTDOWN */ 68 + nla_total_size(1) /* INET_DIAG_TOS */ 69 #if IS_ENABLED(CONFIG_IPV6) 70 + nla_total_size(1) /* INET_DIAG_TCLASS */ 71 + nla_total_size(1) /* INET_DIAG_SKV6ONLY */ 72 #endif 73 + nla_total_size(4) /* INET_DIAG_MARK */ 74 + nla_total_size(4) /* INET_DIAG_CLASS_ID */ 75 #ifdef CONFIG_SOCK_CGROUP_DATA 76 + nla_total_size_64bit(sizeof(u64)) /* INET_DIAG_CGROUP_ID */ 77 #endif 78 + nla_total_size(sizeof(struct inet_diag_sockopt)) 79 /* INET_DIAG_SOCKOPT */ 80 ; 81 } 82 int inet_diag_msg_attrs_fill(struct sock *sk, struct sk_buff *skb, 83 struct inet_diag_msg *r, int ext, 84 struct user_namespace *user_ns, bool net_admin); 85 86 extern int inet_diag_register(const struct inet_diag_handler *handler); 87 extern void inet_diag_unregister(const struct inet_diag_handler *handler); 88 #endif /* _INET_DIAG_H_ */ 89