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