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, unsigned int size);
118 int (*writev)(struct ksmbd_transport *t, struct kvec *iovs, int niov,
119 int size, bool need_invalidate_rkey,
120 unsigned int remote_key);
121 int (*rdma_read)(struct ksmbd_transport *t,
122 void *buf, unsigned int len,
123 struct smb2_buffer_desc_v1 *desc,
124 unsigned int desc_len);
125 int (*rdma_write)(struct ksmbd_transport *t,
126 void *buf, unsigned int len,
127 struct smb2_buffer_desc_v1 *desc,
128 unsigned int desc_len);
131 struct ksmbd_transport {
132 struct ksmbd_conn *conn;
133 struct ksmbd_transport_ops *ops;
134 struct task_struct *handler;
137 #define KSMBD_TCP_RECV_TIMEOUT (7 * HZ)
138 #define KSMBD_TCP_SEND_TIMEOUT (5 * HZ)
139 #define KSMBD_TCP_PEER_SOCKADDR(c) ((struct sockaddr *)&((c)->peer_addr))
141 extern struct list_head conn_list;
142 extern rwlock_t conn_list_lock;
144 bool ksmbd_conn_alive(struct ksmbd_conn *conn);
145 void ksmbd_conn_wait_idle(struct ksmbd_conn *conn);
146 struct ksmbd_conn *ksmbd_conn_alloc(void);
147 void ksmbd_conn_free(struct ksmbd_conn *conn);
148 bool ksmbd_conn_lookup_dialect(struct ksmbd_conn *c);
149 int ksmbd_conn_write(struct ksmbd_work *work);
150 int ksmbd_conn_rdma_read(struct ksmbd_conn *conn,
151 void *buf, unsigned int buflen,
152 struct smb2_buffer_desc_v1 *desc,
153 unsigned int desc_len);
154 int ksmbd_conn_rdma_write(struct ksmbd_conn *conn,
155 void *buf, unsigned int buflen,
156 struct smb2_buffer_desc_v1 *desc,
157 unsigned int desc_len);
158 void ksmbd_conn_enqueue_request(struct ksmbd_work *work);
159 int ksmbd_conn_try_dequeue_request(struct ksmbd_work *work);
160 void ksmbd_conn_init_server_callbacks(struct ksmbd_conn_ops *ops);
161 int ksmbd_conn_handler_loop(void *p);
162 int ksmbd_conn_transport_init(void);
163 void ksmbd_conn_transport_destroy(void);
168 * This is a hack. We will move status to a proper place once we land
169 * a multi-sessions support.
171 static inline bool ksmbd_conn_good(struct ksmbd_work *work)
173 return work->conn->status == KSMBD_SESS_GOOD;
176 static inline bool ksmbd_conn_need_negotiate(struct ksmbd_work *work)
178 return work->conn->status == KSMBD_SESS_NEED_NEGOTIATE;
181 static inline bool ksmbd_conn_need_reconnect(struct ksmbd_work *work)
183 return work->conn->status == KSMBD_SESS_NEED_RECONNECT;
186 static inline bool ksmbd_conn_exiting(struct ksmbd_work *work)
188 return work->conn->status == KSMBD_SESS_EXITING;
191 static inline void ksmbd_conn_set_good(struct ksmbd_work *work)
193 work->conn->status = KSMBD_SESS_GOOD;
196 static inline void ksmbd_conn_set_need_negotiate(struct ksmbd_work *work)
198 work->conn->status = KSMBD_SESS_NEED_NEGOTIATE;
201 static inline void ksmbd_conn_set_need_reconnect(struct ksmbd_work *work)
203 work->conn->status = KSMBD_SESS_NEED_RECONNECT;
206 static inline void ksmbd_conn_set_exiting(struct ksmbd_work *work)
208 work->conn->status = KSMBD_SESS_EXITING;
210 #endif /* __CONNECTION_H__ */