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 
69 static inline int test_session_flag(struct ksmbd_session *sess, int bit)
70 {
71 	return sess->flags & bit;
72 }
73 
74 static inline void set_session_flag(struct ksmbd_session *sess, int bit)
75 {
76 	sess->flags |= bit;
77 }
78 
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