1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3 * Copyright (C) 2018 Samsung Electronics Co., Ltd.
4 */
5
6 #ifndef __USER_SESSION_MANAGEMENT_H__
7 #define __USER_SESSION_MANAGEMENT_H__
8
9 #include <linux/hashtable.h>
10 #include <linux/xarray.h>
11
12 #include "../smb_common.h"
13 #include "../ntlmssp.h"
14
15 #define CIFDS_SESSION_FLAG_SMB2 BIT(1)
16
17 #define PREAUTH_HASHVALUE_SIZE 64
18
19 struct ksmbd_file_table;
20
21 struct channel {
22 __u8 smb3signingkey[SMB3_SIGN_KEY_SIZE];
23 struct ksmbd_conn *conn;
24 };
25
26 struct preauth_session {
27 __u8 Preauth_HashValue[PREAUTH_HASHVALUE_SIZE];
28 u64 id;
29 struct list_head preauth_entry;
30 };
31
32 struct ksmbd_session {
33 u64 id;
34
35 __u16 dialect;
36 char ClientGUID[SMB2_CLIENT_GUID_SIZE];
37
38 struct ksmbd_user *user;
39 unsigned int sequence_number;
40 unsigned int flags;
41
42 bool sign;
43 bool enc;
44
45 int state;
46 __u8 *Preauth_HashValue;
47
48 char sess_key[CIFS_KEY_SIZE];
49
50 struct hlist_node hlist;
51 struct rw_semaphore chann_lock;
52 struct xarray ksmbd_chann_list;
53 struct xarray tree_conns;
54 struct ida tree_conn_ida;
55 struct xarray rpc_handle_list;
56
57 __u8 smb3encryptionkey[SMB3_ENC_DEC_KEY_SIZE];
58 __u8 smb3decryptionkey[SMB3_ENC_DEC_KEY_SIZE];
59 __u8 smb3signingkey[SMB3_SIGN_KEY_SIZE];
60
61 struct ksmbd_file_table file_table;
62 unsigned long last_active;
63 struct rw_semaphore tree_conns_lock;
64
65 #ifdef CONFIG_PROC_FS
66 struct proc_dir_entry *proc_entry;
67 #endif
68 atomic_t refcnt;
69 struct rw_semaphore rpc_lock;
70 };
71
test_session_flag(struct ksmbd_session * sess,int bit)72 static inline int test_session_flag(struct ksmbd_session *sess, int bit)
73 {
74 return sess->flags & bit;
75 }
76
set_session_flag(struct ksmbd_session * sess,int bit)77 static inline void set_session_flag(struct ksmbd_session *sess, int bit)
78 {
79 sess->flags |= bit;
80 }
81
clear_session_flag(struct ksmbd_session * sess,int bit)82 static inline void clear_session_flag(struct ksmbd_session *sess, int bit)
83 {
84 sess->flags &= ~bit;
85 }
86
87 struct ksmbd_session *ksmbd_smb2_session_create(void);
88
89 void ksmbd_session_destroy(struct ksmbd_session *sess);
90
91 struct ksmbd_session *ksmbd_session_lookup_slowpath(unsigned long long id);
92 struct ksmbd_session *ksmbd_session_lookup(struct ksmbd_conn *conn,
93 unsigned long long id);
94 bool is_ksmbd_session_in_connection(struct ksmbd_conn *conn,
95 unsigned long long id);
96 int ksmbd_session_register(struct ksmbd_conn *conn,
97 struct ksmbd_session *sess);
98 void ksmbd_sessions_deregister(struct ksmbd_conn *conn);
99 struct ksmbd_session *__session_lookup(unsigned long long id);
100 struct ksmbd_session *ksmbd_session_lookup_all(struct ksmbd_conn *conn,
101 unsigned long long id);
102 void destroy_previous_session(struct ksmbd_conn *conn,
103 struct ksmbd_user *user, u64 id);
104 struct preauth_session *ksmbd_preauth_session_alloc(struct ksmbd_conn *conn,
105 u64 sess_id);
106 struct preauth_session *ksmbd_preauth_session_lookup(struct ksmbd_conn *conn,
107 unsigned long long id);
108
109 int ksmbd_acquire_tree_conn_id(struct ksmbd_session *sess);
110 void ksmbd_release_tree_conn_id(struct ksmbd_session *sess, int id);
111
112 int ksmbd_session_rpc_open(struct ksmbd_session *sess, char *rpc_name);
113 void ksmbd_session_rpc_close(struct ksmbd_session *sess, int id);
114 int ksmbd_session_rpc_method(struct ksmbd_session *sess, int id);
115 void ksmbd_user_session_get(struct ksmbd_session *sess);
116 void ksmbd_user_session_put(struct ksmbd_session *sess);
117 int create_proc_sessions(void);
118 #endif /* __USER_SESSION_MANAGEMENT_H__ */
119