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 bool is_anonymous;
45
46 int state;
47 __u8 *Preauth_HashValue;
48
49 char sess_key[CIFS_KEY_SIZE];
50
51 struct hlist_node hlist;
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 rwlock_t tree_conns_lock;
64
65 atomic_t refcnt;
66 struct rw_semaphore rpc_lock;
67 };
68
test_session_flag(struct ksmbd_session * sess,int bit)69 static inline int test_session_flag(struct ksmbd_session *sess, int bit)
70 {
71 return sess->flags & bit;
72 }
73
set_session_flag(struct ksmbd_session * sess,int bit)74 static inline void set_session_flag(struct ksmbd_session *sess, int bit)
75 {
76 sess->flags |= bit;
77 }
78
clear_session_flag(struct ksmbd_session * sess,int bit)79 static inline void clear_session_flag(struct ksmbd_session *sess, int bit)
80 {
81 sess->flags &= ~bit;
82 }
83
84 struct ksmbd_session *ksmbd_smb2_session_create(void);
85
86 void ksmbd_session_destroy(struct ksmbd_session *sess);
87
88 struct ksmbd_session *ksmbd_session_lookup_slowpath(unsigned long long id);
89 struct ksmbd_session *ksmbd_session_lookup(struct ksmbd_conn *conn,
90 unsigned long long id);
91 bool is_ksmbd_session_in_connection(struct ksmbd_conn *conn,
92 unsigned long long id);
93 int ksmbd_session_register(struct ksmbd_conn *conn,
94 struct ksmbd_session *sess);
95 void ksmbd_sessions_deregister(struct ksmbd_conn *conn);
96 struct ksmbd_session *__session_lookup(unsigned long long id);
97 struct ksmbd_session *ksmbd_session_lookup_all(struct ksmbd_conn *conn,
98 unsigned long long id);
99 void destroy_previous_session(struct ksmbd_conn *conn,
100 struct ksmbd_user *user, u64 id);
101 struct preauth_session *ksmbd_preauth_session_alloc(struct ksmbd_conn *conn,
102 u64 sess_id);
103 struct preauth_session *ksmbd_preauth_session_lookup(struct ksmbd_conn *conn,
104 unsigned long long id);
105
106 int ksmbd_acquire_tree_conn_id(struct ksmbd_session *sess);
107 void ksmbd_release_tree_conn_id(struct ksmbd_session *sess, int id);
108
109 int ksmbd_session_rpc_open(struct ksmbd_session *sess, char *rpc_name);
110 void ksmbd_session_rpc_close(struct ksmbd_session *sess, int id);
111 int ksmbd_session_rpc_method(struct ksmbd_session *sess, int id);
112 void ksmbd_user_session_get(struct ksmbd_session *sess);
113 void ksmbd_user_session_put(struct ksmbd_session *sess);
114 #endif /* __USER_SESSION_MANAGEMENT_H__ */
115