smb/server/connection.c

28  * @conn:	connection instance to be cleand up
30  * During the thread termination, the corresponding conn instance
31  * resources(sock/memory) are released and finally the conn object is freed.
33 void ksmbd_conn_free(struct ksmbd_conn *conn)
36 	list_del(&conn->conns_list);
39 	xa_destroy(&conn->sessions);
40 	kvfree(conn->request_buf);
41 	kfree(conn->preauth_info);
42 	kfree(conn);
52 	struct ksmbd_conn *conn;
54 	conn = kzalloc(sizeof(struct ksmbd_conn), GFP_KERNEL);
55 	if (!conn)
58 	conn->need_neg = true;
59 	ksmbd_conn_set_new(conn);
60 	conn->local_nls = load_nls("utf8");
61 	if (!conn->local_nls)
62 		conn->local_nls = load_nls_default();
64 		conn->um = utf8_load(UNICODE_AGE(12, 1, 0));
66 		conn->um = ERR_PTR(-EOPNOTSUPP);
67 	if (IS_ERR(conn->um))
68 		conn->um = NULL;
69 	atomic_set(&conn->req_running, 0);
70 	atomic_set(&conn->r_count, 0);
71 	conn->total_credits = 1;
72 	conn->outstanding_credits = 0;
74 	init_waitqueue_head(&conn->req_running_q);
75 	init_waitqueue_head(&conn->r_count_q);
76 	INIT_LIST_HEAD(&conn->conns_list);
77 	INIT_LIST_HEAD(&conn->requests);
78 	INIT_LIST_HEAD(&conn->async_requests);
79 	spin_lock_init(&conn->request_lock);
80 	spin_lock_init(&conn->credits_lock);
81 	ida_init(&conn->async_ida);
82 	xa_init(&conn->sessions);
84 	spin_lock_init(&conn->llist_lock);
85 	INIT_LIST_HEAD(&conn->lock_list);
87 	init_rwsem(&conn->session_lock);
90 	list_add(&conn->conns_list, &conn_list);
92 	return conn;
114 	struct ksmbd_conn *conn = work->conn;
117 	if (conn->ops->get_cmd_val(work) != SMB2_CANCEL_HE)
118 		requests_queue = &conn->requests;
121 		atomic_inc(&conn->req_running);
122 		spin_lock(&conn->request_lock);
124 		spin_unlock(&conn->request_lock);
130 	struct ksmbd_conn *conn = work->conn;
136 	atomic_dec(&conn->req_running);
137 	spin_lock(&conn->request_lock);
139 	spin_unlock(&conn->request_lock);
143 	wake_up_all(&conn->req_running_q);
146 void ksmbd_conn_lock(struct ksmbd_conn *conn)
148 	mutex_lock(&conn->srv_mutex);
151 void ksmbd_conn_unlock(struct ksmbd_conn *conn)
153 	mutex_unlock(&conn->srv_mutex);
158 	struct ksmbd_conn *conn;
161 	list_for_each_entry(conn, &conn_list, conns_list) {
162 		if (conn->binding || xa_load(&conn->sessions, sess_id))
163 			WRITE_ONCE(conn->status, status);
168 void ksmbd_conn_wait_idle(struct ksmbd_conn *conn, u64 sess_id)
170 	wait_event(conn->req_running_q, atomic_read(&conn->req_running) < 2);
175 	struct ksmbd_conn *conn = work->conn;
189 	ksmbd_conn_lock(conn);
190 	sent = conn->transport->ops->writev(conn->transport, work->iov,
195 	ksmbd_conn_unlock(conn);
205 int ksmbd_conn_rdma_read(struct ksmbd_conn *conn,
212 	if (conn->transport->ops->rdma_read)
213 		ret = conn->transport->ops->rdma_read(conn->transport,
219 int ksmbd_conn_rdma_write(struct ksmbd_conn *conn,
226 	if (conn->transport->ops->rdma_write)
227 		ret = conn->transport->ops->rdma_write(conn->transport,
233 bool ksmbd_conn_alive(struct ksmbd_conn *conn)
238 	if (ksmbd_conn_exiting(conn))
244 	if (atomic_read(&conn->stats.open_files_count) > 0)
253 	    time_after(jiffies, conn->last_active + server_conf.deadtime)) {
274 	struct ksmbd_conn *conn = (struct ksmbd_conn *)p;
275 	struct ksmbd_transport *t = conn->transport;
280 	mutex_init(&conn->srv_mutex);
286 	conn->last_active = jiffies;
288 	while (ksmbd_conn_alive(conn)) {
292 		kvfree(conn->request_buf);
293 		conn->request_buf = NULL;
302 		if (ksmbd_conn_good(conn))
304 				SMB3_MAX_MSGSIZE + conn->vals->max_write_size;
311 					READ_ONCE(conn->status));
327 		conn->request_buf = kvmalloc(size, GFP_KERNEL);
328 		if (!conn->request_buf)
331 		memcpy(conn->request_buf, hdr_buf, sizeof(hdr_buf));
337 		size = t->ops->read(t, conn->request_buf + 4, pdu_size, 2);
349 		if (!ksmbd_smb_request(conn))
352 		if (((struct smb2_hdr *)smb2_get_msg(conn->request_buf))->ProtocolId ==
363 		if (default_conn_ops.process_fn(conn)) {
370 	ksmbd_conn_set_releasing(conn);
372 	wait_event(conn->r_count_q, atomic_read(&conn->r_count) == 0);
375 		utf8_unload(conn->um);
376 	unload_nls(conn->local_nls);
378 		default_conn_ops.terminate_fn(conn);
413 	struct ksmbd_conn *conn;
418 	list_for_each_entry(conn, &conn_list, conns_list) {
419 		t = conn->transport;
420 		ksmbd_conn_set_exiting(conn);