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