1 /* SPDX-License-Identifier: GPL-2.0-or-later */
3 * Copyright (C) 2018 Samsung Electronics Co., Ltd.
6 #ifndef __KSMBD_CONNECTION_H__
7 #define __KSMBD_CONNECTION_H__
9 #include <linux/list.h>
13 #include <net/inet_connection_sock.h>
14 #include <net/request_sock.h>
15 #include <linux/kthread.h>
16 #include <linux/nls.h>
17 #include <linux/unicode.h>
19 #include "smb_common.h"
20 #include "ksmbd_work.h"
22 #define KSMBD_SOCKET_BACKLOG 16
28 KSMBD_SESS_NEED_RECONNECT,
29 KSMBD_SESS_NEED_NEGOTIATE
33 atomic_t open_files_count;
34 atomic64_t request_served;
37 struct ksmbd_transport;
40 struct smb_version_values *vals;
41 struct smb_version_ops *ops;
42 struct smb_version_cmds *cmds;
43 unsigned int max_cmds;
44 struct mutex srv_mutex;
48 struct ksmbd_transport *transport;
49 struct nls_table *local_nls;
50 struct unicode_map *um;
51 struct list_head conns_list;
52 /* smb session 1 per user */
53 struct xarray sessions;
54 unsigned long last_active;
55 /* How many request are running currently */
57 /* References which are made for this Server object*/
59 unsigned int total_credits;
60 unsigned int outstanding_credits;
61 spinlock_t credits_lock;
62 wait_queue_head_t req_running_q;
63 wait_queue_head_t r_count_q;
64 /* Lock to protect requests list*/
65 spinlock_t request_lock;
66 struct list_head requests;
67 struct list_head async_requests;
69 struct ksmbd_stats stats;
70 char ClientGUID[SMB2_CLIENT_GUID_SIZE];
71 struct ntlmssp_auth ntlmssp;
73 spinlock_t llist_lock;
74 struct list_head lock_list;
76 struct preauth_integrity_info *preauth_info;
79 unsigned int auth_mechs;
80 unsigned int preferred_auth_mech;
85 /* dialect index that server chose */
90 struct ksmbd_conn_ops *conn_ops;
92 /* Preauth Session Table */
93 struct list_head preauth_sess_table;
95 struct sockaddr_storage peer_addr;
97 /* Identifier for async message */
101 __le16 compress_algorithm;
102 bool posix_ext_supported;
103 bool signing_negotiated;
104 __le16 signing_algorithm;
108 struct ksmbd_conn_ops {
109 int (*process_fn)(struct ksmbd_conn *conn);
110 int (*terminate_fn)(struct ksmbd_conn *conn);
113 struct ksmbd_transport_ops {
114 int (*prepare)(struct ksmbd_transport *t);
115 void (*disconnect)(struct ksmbd_transport *t);
116 void (*shutdown)(struct ksmbd_transport *t);
117 int (*read)(struct ksmbd_transport *t, char *buf,
118 unsigned int size, int max_retries);
119 int (*writev)(struct ksmbd_transport *t, struct kvec *iovs, int niov,
120 int size, bool need_invalidate_rkey,
121 unsigned int remote_key);
122 int (*rdma_read)(struct ksmbd_transport *t,
123 void *buf, unsigned int len,
124 struct smb2_buffer_desc_v1 *desc,
125 unsigned int desc_len);
126 int (*rdma_write)(struct ksmbd_transport *t,
127 void *buf, unsigned int len,
128 struct smb2_buffer_desc_v1 *desc,
129 unsigned int desc_len);
132 struct ksmbd_transport {
133 struct ksmbd_conn *conn;
134 struct ksmbd_transport_ops *ops;
135 struct task_struct *handler;
138 #define KSMBD_TCP_RECV_TIMEOUT (7 * HZ)
139 #define KSMBD_TCP_SEND_TIMEOUT (5 * HZ)
140 #define KSMBD_TCP_PEER_SOCKADDR(c) ((struct sockaddr *)&((c)->peer_addr))
142 extern struct list_head conn_list;
143 extern rwlock_t conn_list_lock;
145 bool ksmbd_conn_alive(struct ksmbd_conn *conn);
146 void ksmbd_conn_wait_idle(struct ksmbd_conn *conn);
147 struct ksmbd_conn *ksmbd_conn_alloc(void);
148 void ksmbd_conn_free(struct ksmbd_conn *conn);
149 bool ksmbd_conn_lookup_dialect(struct ksmbd_conn *c);
150 int ksmbd_conn_write(struct ksmbd_work *work);
151 int ksmbd_conn_rdma_read(struct ksmbd_conn *conn,
152 void *buf, unsigned int buflen,
153 struct smb2_buffer_desc_v1 *desc,
154 unsigned int desc_len);
155 int ksmbd_conn_rdma_write(struct ksmbd_conn *conn,
156 void *buf, unsigned int buflen,
157 struct smb2_buffer_desc_v1 *desc,
158 unsigned int desc_len);
159 void ksmbd_conn_enqueue_request(struct ksmbd_work *work);
160 int ksmbd_conn_try_dequeue_request(struct ksmbd_work *work);
161 void ksmbd_conn_init_server_callbacks(struct ksmbd_conn_ops *ops);
162 int ksmbd_conn_handler_loop(void *p);
163 int ksmbd_conn_transport_init(void);
164 void ksmbd_conn_transport_destroy(void);
169 * This is a hack. We will move status to a proper place once we land
170 * a multi-sessions support.
172 static inline bool ksmbd_conn_good(struct ksmbd_work *work)
174 return work->conn->status == KSMBD_SESS_GOOD;
177 static inline bool ksmbd_conn_need_negotiate(struct ksmbd_work *work)
179 return work->conn->status == KSMBD_SESS_NEED_NEGOTIATE;
182 static inline bool ksmbd_conn_need_reconnect(struct ksmbd_work *work)
184 return work->conn->status == KSMBD_SESS_NEED_RECONNECT;
187 static inline bool ksmbd_conn_exiting(struct ksmbd_work *work)
189 return work->conn->status == KSMBD_SESS_EXITING;
192 static inline void ksmbd_conn_set_good(struct ksmbd_work *work)
194 work->conn->status = KSMBD_SESS_GOOD;
197 static inline void ksmbd_conn_set_need_negotiate(struct ksmbd_work *work)
199 work->conn->status = KSMBD_SESS_NEED_NEGOTIATE;
202 static inline void ksmbd_conn_set_need_reconnect(struct ksmbd_work *work)
204 work->conn->status = KSMBD_SESS_NEED_RECONNECT;
207 static inline void ksmbd_conn_set_exiting(struct ksmbd_work *work)
209 work->conn->status = KSMBD_SESS_EXITING;
211 #endif /* __CONNECTION_H__ */