[linux-block.git] / include / linux / inet_diag.h

/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _INET_DIAG_H_
#define _INET_DIAG_H_ 1

#include <net/netlink.h>
#include <uapi/linux/inet_diag.h>

struct inet_hashinfo;

struct inet_diag_handler {
	struct module	*owner;
	void		(*dump)(struct sk_buff *skb,
				struct netlink_callback *cb,
				const struct inet_diag_req_v2 *r);

	int		(*dump_one)(struct netlink_callback *cb,
				    const struct inet_diag_req_v2 *req);

	void		(*idiag_get_info)(struct sock *sk,
					  struct inet_diag_msg *r,
					  void *info);

	int		(*idiag_get_aux)(struct sock *sk,
					 bool net_admin,
					 struct sk_buff *skb);

	size_t		(*idiag_get_aux_size)(struct sock *sk,
					      bool net_admin);

	int		(*destroy)(struct sk_buff *in_skb,
				   const struct inet_diag_req_v2 *req);

	__u16		idiag_type;
	__u16		idiag_info_size;
};

struct bpf_sk_storage_diag;
struct inet_diag_dump_data {
	struct nlattr *req_nlas[__INET_DIAG_REQ_MAX];
#define inet_diag_nla_bc req_nlas[INET_DIAG_REQ_BYTECODE]
#define inet_diag_nla_bpf_stgs req_nlas[INET_DIAG_REQ_SK_BPF_STORAGES]

	struct bpf_sk_storage_diag *bpf_stg_diag;
};

struct inet_connection_sock;
int inet_sk_diag_fill(struct sock *sk, struct inet_connection_sock *icsk,
		      struct sk_buff *skb, struct netlink_callback *cb,
		      const struct inet_diag_req_v2 *req,
		      u16 nlmsg_flags, bool net_admin);
void inet_diag_dump_icsk(struct inet_hashinfo *h, struct sk_buff *skb,
			 struct netlink_callback *cb,
			 const struct inet_diag_req_v2 *r);
int inet_diag_dump_one_icsk(struct inet_hashinfo *hashinfo,
			    struct netlink_callback *cb,
			    const struct inet_diag_req_v2 *req);

struct sock *inet_diag_find_one_icsk(struct net *net,
				     struct inet_hashinfo *hashinfo,
				     const struct inet_diag_req_v2 *req);

int inet_diag_bc_sk(const struct nlattr *_bc, struct sock *sk);

void inet_diag_msg_common_fill(struct inet_diag_msg *r, struct sock *sk);

static inline size_t inet_diag_msg_attrs_size(void)
{
	return	  nla_total_size(1)  /* INET_DIAG_SHUTDOWN */
		+ nla_total_size(1)  /* INET_DIAG_TOS */
#if IS_ENABLED(CONFIG_IPV6)
		+ nla_total_size(1)  /* INET_DIAG_TCLASS */
		+ nla_total_size(1)  /* INET_DIAG_SKV6ONLY */
#endif
		+ nla_total_size(4)  /* INET_DIAG_MARK */
		+ nla_total_size(4)  /* INET_DIAG_CLASS_ID */
#ifdef CONFIG_SOCK_CGROUP_DATA
		+ nla_total_size_64bit(sizeof(u64))  /* INET_DIAG_CGROUP_ID */
#endif
		+ nla_total_size(sizeof(struct inet_diag_sockopt))
						     /* INET_DIAG_SOCKOPT */
		;
}
int inet_diag_msg_attrs_fill(struct sock *sk, struct sk_buff *skb,
			     struct inet_diag_msg *r, int ext,
			     struct user_namespace *user_ns, bool net_admin);

extern int  inet_diag_register(const struct inet_diag_handler *handler);
extern void inet_diag_unregister(const struct inet_diag_handler *handler);
#endif /* _INET_DIAG_H_ */
Commit	Line	Data
b2441318	1	/* SPDX-License-Identifier: GPL-2.0 */
73c1f4a0 ACM	2	#ifndef _INET_DIAG_H_
73c1f4a0 ACM	3	#define _INET_DIAG_H_ 1
1da177e4	4
83f73c5b	5	#include <net/netlink.h>
607ca46e	6	#include <uapi/linux/inet_diag.h>
2df005b7	7
4f5736c4 ACM	8	struct inet_hashinfo;
	9
	10	struct inet_diag_handler {
db591469	11	struct module *owner;
34160ea3 ED	12	void (dump)(struct sk_buff skb,
34160ea3 ED	13	struct netlink_callback *cb,
0df6d328	14	const struct inet_diag_req_v2 *r);
34160ea3	15
5682d393	16	int (dump_one)(struct netlink_callback cb,
34160ea3 ED	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);
6eb5d2e0	22
b37e8840 ID	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
6eb5d2e0 LC	30	int (destroy)(struct sk_buff in_skb,
	31	const struct inet_diag_req_v2 *req);
	32
34160ea3	33	__u16 idiag_type;
3fd22af8	34	__u16 idiag_info_size;
4f5736c4 ACM	35	};
4f5736c4 ACM	36
085c20ca	37	struct bpf_sk_storage_diag;
0df6d328 MKL	38	struct 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]
085c20ca MKL	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;
0df6d328 MKL	44	};
0df6d328 MKL	45
3c4d05c8 PE	46	struct inet_connection_sock;
3c4d05c8 PE	47	int inet_sk_diag_fill(struct sock sk, struct inet_connection_sock icsk,
5682d393 MKL	48	struct sk_buff skb, struct netlink_callback cb,
	49	const struct inet_diag_req_v2 *req,
	50	u16 nlmsg_flags, bool net_admin);
1942c518	51	void inet_diag_dump_icsk(struct inet_hashinfo h, struct sk_buff skb,
34160ea3	52	struct netlink_callback *cb,
0df6d328	53	const struct inet_diag_req_v2 *r);
1942c518	54	int inet_diag_dump_one_icsk(struct inet_hashinfo *hashinfo,
5682d393	55	struct netlink_callback *cb,
34160ea3	56	const struct inet_diag_req_v2 *req);
1942c518	57
b613f56e LC	58	struct sock inet_diag_find_one_icsk(struct net net,
	59	struct inet_hashinfo *hashinfo,
	60	const struct inet_diag_req_v2 *req);
	61
8d07d151	62	int inet_diag_bc_sk(const struct nlattr _bc, struct sock sk);
b005ab4e	63
c3ec5e5c BD	64	void inet_diag_msg_common_fill(struct inet_diag_msg r, struct sock sk);
c3ec5e5c BD	65
83f73c5b DY	66	static 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 */
6e3a401f DY	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
c1077616 WW	79	+ nla_total_size(sizeof(struct inet_diag_sockopt))
c1077616 WW	80	/* INET_DIAG_SOCKOPT */
6e3a401f	81	;
83f73c5b	82	}
c3ec5e5c BD	83	int inet_diag_msg_attrs_fill(struct sock sk, struct sk_buff skb,
c3ec5e5c BD	84	struct inet_diag_msg *r, int ext,
d545caca	85	struct user_namespace *user_ns, bool net_admin);
c3ec5e5c	86
4f5736c4 ACM	87	extern int inet_diag_register(const struct inet_diag_handler *handler);
4f5736c4 ACM	88	extern void inet_diag_unregister(const struct inet_diag_handler *handler);
73c1f4a0	89	#endif /* _INET_DIAG_H_ */