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