1/* SPDX-License-Identifier: GPL-2.0-or-later */
2/*
3 * RDMA Transport Layer
4 *
5 * Copyright (c) 2014 - 2018 ProfitBricks GmbH. All rights reserved.
6 * Copyright (c) 2018 - 2019 1&1 IONOS Cloud GmbH. All rights reserved.
7 * Copyright (c) 2019 - 2020 1&1 IONOS SE. All rights reserved.
8 */
9
10#ifndef RTRS_SRV_H
11#define RTRS_SRV_H
12
13#include <linux/device.h>
14#include <linux/refcount.h>
15#include <linux/percpu.h>
16#include "rtrs-pri.h"
17
18/*
19 * enum rtrs_srv_state - Server states.
20 */
21enum rtrs_srv_state {
22	RTRS_SRV_CONNECTING,
23	RTRS_SRV_CONNECTED,
24	RTRS_SRV_CLOSING,
25	RTRS_SRV_CLOSED,
26};
27
28/* stats for Read and write operation.
29 * see Documentation/ABI/testing/sysfs-class-rtrs-server for details
30 */
31struct rtrs_srv_stats_rdma_stats {
32	struct {
33		u64 cnt;
34		u64 size_total;
35	} dir[2];
36};
37
38struct rtrs_srv_stats {
39	struct kobject					kobj_stats;
40	struct rtrs_srv_stats_rdma_stats __percpu	*rdma_stats;
41	struct rtrs_srv_path				*srv_path;
42};
43
44struct rtrs_srv_con {
45	struct rtrs_con		c;
46	struct list_head	rsp_wr_wait_list;
47	spinlock_t		rsp_wr_wait_lock;
48};
49
50/* IO context in rtrs_srv, each io has one */
51struct rtrs_srv_op {
52	struct rtrs_srv_con		*con;
53	u32				msg_id;
54	u8				dir;
55	struct rtrs_msg_rdma_read	*rd_msg;
56	struct ib_rdma_wr		tx_wr;
57	struct ib_sge			tx_sg;
58	struct list_head		wait_list;
59	int				status;
60};
61
62/*
63 * server side memory region context, when always_invalidate=Y, we need
64 * queue_depth of memory region to invalidate each memory region.
65 */
66struct rtrs_srv_mr {
67	struct ib_mr	*mr;
68	struct sg_table	sgt;
69	struct ib_cqe	inv_cqe;	/* only for always_invalidate=true */
70	u32		msg_id;		/* only for always_invalidate=true */
71	u32		msg_off;	/* only for always_invalidate=true */
72	struct rtrs_iu	*iu;		/* send buffer for new rkey msg */
73};
74
75struct rtrs_srv_path {
76	struct rtrs_path	s;
77	struct rtrs_srv_sess	*srv;
78	struct work_struct	close_work;
79	enum rtrs_srv_state	state;
80	spinlock_t		state_lock;
81	int			cur_cq_vector;
82	struct rtrs_srv_op	**ops_ids;
83	struct percpu_ref       ids_inflight_ref;
84	struct completion       complete_done;
85	struct rtrs_srv_mr	*mrs;
86	unsigned int		mrs_num;
87	dma_addr_t		*dma_addr;
88	bool			established;
89	unsigned int		mem_bits;
90	struct kobject		kobj;
91	struct rtrs_srv_stats	*stats;
92};
93
94static inline struct rtrs_srv_path *to_srv_path(struct rtrs_path *s)
95{
96	return container_of(s, struct rtrs_srv_path, s);
97}
98
99struct rtrs_srv_sess {
100	struct list_head	paths_list;
101	int			paths_up;
102	struct mutex		paths_ev_mutex;
103	size_t			paths_num;
104	struct mutex		paths_mutex;
105	uuid_t			paths_uuid;
106	refcount_t		refcount;
107	struct rtrs_srv_ctx	*ctx;
108	struct list_head	ctx_list;
109	void			*priv;
110	size_t			queue_depth;
111	struct page		**chunks;
112	struct device		dev;
113	unsigned int		dev_ref;
114	struct kobject		*kobj_paths;
115};
116
117struct rtrs_srv_ctx {
118	struct rtrs_srv_ops ops;
119	struct rdma_cm_id *cm_id_ip;
120	struct rdma_cm_id *cm_id_ib;
121	struct mutex srv_mutex;
122	struct list_head srv_list;
123};
124
125struct rtrs_srv_ib_ctx {
126	struct rtrs_srv_ctx	*srv_ctx;
127	u16			port;
128	struct mutex            ib_dev_mutex;
129	int			ib_dev_count;
130};
131
132extern const struct class rtrs_dev_class;
133
134void close_path(struct rtrs_srv_path *srv_path);
135
136static inline void rtrs_srv_update_rdma_stats(struct rtrs_srv_stats *s,
137					      size_t size, int d)
138{
139	this_cpu_inc(s->rdma_stats->dir[d].cnt);
140	this_cpu_add(s->rdma_stats->dir[d].size_total, size);
141}
142
143/* functions which are implemented in rtrs-srv-stats.c */
144int rtrs_srv_reset_rdma_stats(struct rtrs_srv_stats *stats, bool enable);
145ssize_t rtrs_srv_stats_rdma_to_str(struct rtrs_srv_stats *stats, char *page);
146int rtrs_srv_reset_all_stats(struct rtrs_srv_stats *stats, bool enable);
147ssize_t rtrs_srv_reset_all_help(struct rtrs_srv_stats *stats,
148				 char *page, size_t len);
149
150/* functions which are implemented in rtrs-srv-sysfs.c */
151int rtrs_srv_create_path_files(struct rtrs_srv_path *srv_path);
152void rtrs_srv_destroy_path_files(struct rtrs_srv_path *srv_path);
153
154#endif /* RTRS_SRV_H */
155