xref: /linux/fs/smb/server/smb2pdu.c (revision 5d15ab717d503ff10b585a144870648b9a88c616)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *   Copyright (C) 2016 Namjae Jeon <linkinjeon@kernel.org>
4  *   Copyright (C) 2018 Samsung Electronics Co., Ltd.
5  */
6 
7 #include <crypto/utils.h>
8 #include <linux/inetdevice.h>
9 #include <net/addrconf.h>
10 #include <linux/syscalls.h>
11 #include <linux/namei.h>
12 #include <linux/statfs.h>
13 #include <linux/ethtool.h>
14 #include <linux/falloc.h>
15 #include <linux/mount.h>
16 #include <linux/filelock.h>
17 #include <linux/fileattr.h>
18 
19 #include "glob.h"
20 #include "smbfsctl.h"
21 #include "oplock.h"
22 #include "smbacl.h"
23 
24 #include "auth.h"
25 #include "asn1.h"
26 #include "connection.h"
27 #include "transport_ipc.h"
28 #include "transport_rdma.h"
29 #include "vfs.h"
30 #include "vfs_cache.h"
31 #include "misc.h"
32 
33 #include "server.h"
34 #include "smb_common.h"
35 #include "../common/smb2status.h"
36 #include "ksmbd_work.h"
37 #include "mgmt/user_config.h"
38 #include "mgmt/share_config.h"
39 #include "mgmt/tree_connect.h"
40 #include "mgmt/user_session.h"
41 #include "mgmt/ksmbd_ida.h"
42 #include "ndr.h"
43 #include "stats.h"
44 #include "transport_tcp.h"
45 
46 static void __wbuf(struct ksmbd_work *work, void **req, void **rsp)
47 {
48 	if (work->next_smb2_rcv_hdr_off) {
49 		*req = ksmbd_req_buf_next(work);
50 		*rsp = ksmbd_resp_buf_next(work);
51 	} else {
52 		*req = smb_get_msg(work->request_buf);
53 		*rsp = smb_get_msg(work->response_buf);
54 	}
55 }
56 
57 #define WORK_BUFFERS(w, rq, rs)	__wbuf((w), (void **)&(rq), (void **)&(rs))
58 
59 /**
60  * check_session_id() - check for valid session id in smb header
61  * @conn:	connection instance
62  * @id:		session id from smb header
63  *
64  * Return:      1 if valid session id, otherwise 0
65  */
66 static inline bool check_session_id(struct ksmbd_conn *conn, u64 id)
67 {
68 	struct ksmbd_session *sess;
69 
70 	if (id == 0 || id == -1)
71 		return false;
72 
73 	sess = ksmbd_session_lookup_all(conn, id);
74 	if (sess) {
75 		ksmbd_user_session_put(sess);
76 		return true;
77 	}
78 	pr_err("Invalid user session id: %llu\n", id);
79 	return false;
80 }
81 
82 struct channel *lookup_chann_list(struct ksmbd_session *sess, struct ksmbd_conn *conn)
83 {
84 	struct channel *chann;
85 
86 	down_read(&sess->chann_lock);
87 	chann = xa_load(&sess->ksmbd_chann_list, (long)conn);
88 	up_read(&sess->chann_lock);
89 
90 	return chann;
91 }
92 
93 /**
94  * smb2_get_ksmbd_tcon() - get tree connection information using a tree id.
95  * @work:	smb work
96  *
97  * Return:	0 if there is a tree connection matched or these are
98  *		skipable commands, otherwise error
99  */
100 int smb2_get_ksmbd_tcon(struct ksmbd_work *work)
101 {
102 	struct smb2_hdr *req_hdr = ksmbd_req_buf_next(work);
103 	unsigned int cmd = le16_to_cpu(req_hdr->Command);
104 	unsigned int tree_id;
105 
106 	if (cmd == SMB2_TREE_CONNECT_HE ||
107 	    cmd ==  SMB2_CANCEL_HE ||
108 	    cmd ==  SMB2_LOGOFF_HE) {
109 		ksmbd_debug(SMB, "skip to check tree connect request\n");
110 		return 0;
111 	}
112 
113 	if (xa_empty(&work->sess->tree_conns)) {
114 		ksmbd_debug(SMB, "NO tree connected\n");
115 		return -ENOENT;
116 	}
117 
118 	tree_id = le32_to_cpu(req_hdr->Id.SyncId.TreeId);
119 
120 	/*
121 	 * If request is not the first in Compound request,
122 	 * Just validate tree id in header with work->tcon->id.
123 	 */
124 	if (work->next_smb2_rcv_hdr_off) {
125 		if (!work->tcon) {
126 			pr_err("The first operation in the compound does not have tcon\n");
127 			return -EINVAL;
128 		}
129 		if (work->tcon->t_state != TREE_CONNECTED)
130 			return -ENOENT;
131 		if (tree_id != UINT_MAX && work->tcon->id != tree_id) {
132 			pr_err("tree id(%u) is different with id(%u) in first operation\n",
133 					tree_id, work->tcon->id);
134 			return -EINVAL;
135 		}
136 		return 1;
137 	}
138 
139 	work->tcon = ksmbd_tree_conn_lookup(work->sess, tree_id);
140 	if (!work->tcon) {
141 		pr_err("Invalid tid %d\n", tree_id);
142 		return -ENOENT;
143 	}
144 
145 	return 1;
146 }
147 
148 /**
149  * smb2_set_err_rsp() - set error response code on smb response
150  * @work:	smb work containing response buffer
151  */
152 void smb2_set_err_rsp(struct ksmbd_work *work)
153 {
154 	struct smb2_err_rsp *err_rsp;
155 
156 	if (work->next_smb2_rcv_hdr_off)
157 		err_rsp = ksmbd_resp_buf_next(work);
158 	else
159 		err_rsp = smb_get_msg(work->response_buf);
160 
161 	if (err_rsp->hdr.Status != STATUS_STOPPED_ON_SYMLINK) {
162 		int err;
163 
164 		err_rsp->StructureSize = SMB2_ERROR_STRUCTURE_SIZE2_LE;
165 		err_rsp->ErrorContextCount = 0;
166 		err_rsp->Reserved = 0;
167 		err_rsp->ByteCount = 0;
168 		err_rsp->ErrorData[0] = 0;
169 		err = ksmbd_iov_pin_rsp(work, (void *)err_rsp,
170 					__SMB2_HEADER_STRUCTURE_SIZE +
171 						SMB2_ERROR_STRUCTURE_SIZE2);
172 		if (err)
173 			work->send_no_response = 1;
174 	}
175 }
176 
177 /**
178  * is_smb2_neg_cmd() - is it smb2 negotiation command
179  * @work:	smb work containing smb header
180  *
181  * Return:      true if smb2 negotiation command, otherwise false
182  */
183 bool is_smb2_neg_cmd(struct ksmbd_work *work)
184 {
185 	struct smb2_hdr *hdr = smb_get_msg(work->request_buf);
186 
187 	/* is it SMB2 header ? */
188 	if (hdr->ProtocolId != SMB2_PROTO_NUMBER)
189 		return false;
190 
191 	/* make sure it is request not response message */
192 	if (hdr->Flags & SMB2_FLAGS_SERVER_TO_REDIR)
193 		return false;
194 
195 	if (hdr->Command != SMB2_NEGOTIATE)
196 		return false;
197 
198 	return true;
199 }
200 
201 /**
202  * is_smb2_rsp() - is it smb2 response
203  * @work:	smb work containing smb response buffer
204  *
205  * Return:      true if smb2 response, otherwise false
206  */
207 bool is_smb2_rsp(struct ksmbd_work *work)
208 {
209 	struct smb2_hdr *hdr = smb_get_msg(work->response_buf);
210 
211 	/* is it SMB2 header ? */
212 	if (hdr->ProtocolId != SMB2_PROTO_NUMBER)
213 		return false;
214 
215 	/* make sure it is response not request message */
216 	if (!(hdr->Flags & SMB2_FLAGS_SERVER_TO_REDIR))
217 		return false;
218 
219 	return true;
220 }
221 
222 /**
223  * get_smb2_cmd_val() - get smb command code from smb header
224  * @work:	smb work containing smb request buffer
225  *
226  * Return:      smb2 request command value
227  */
228 u16 get_smb2_cmd_val(struct ksmbd_work *work)
229 {
230 	struct smb2_hdr *rcv_hdr;
231 
232 	if (work->next_smb2_rcv_hdr_off)
233 		rcv_hdr = ksmbd_req_buf_next(work);
234 	else
235 		rcv_hdr = smb_get_msg(work->request_buf);
236 	return le16_to_cpu(rcv_hdr->Command);
237 }
238 
239 /**
240  * set_smb2_rsp_status() - set error response code on smb2 header
241  * @work:	smb work containing response buffer
242  * @err:	error response code
243  */
244 void set_smb2_rsp_status(struct ksmbd_work *work, __le32 err)
245 {
246 	struct smb2_hdr *rsp_hdr;
247 
248 	rsp_hdr = smb_get_msg(work->response_buf);
249 	rsp_hdr->Status = err;
250 
251 	work->iov_idx = 0;
252 	work->iov_cnt = 0;
253 	work->next_smb2_rcv_hdr_off = 0;
254 	smb2_set_err_rsp(work);
255 }
256 
257 /**
258  * init_smb2_neg_rsp() - initialize smb2 response for negotiate command
259  * @work:	smb work containing smb request buffer
260  *
261  * smb2 negotiate response is sent in reply of smb1 negotiate command for
262  * dialect auto-negotiation.
263  */
264 int init_smb2_neg_rsp(struct ksmbd_work *work)
265 {
266 	struct smb2_hdr *rsp_hdr;
267 	struct smb2_negotiate_rsp *rsp;
268 	struct ksmbd_conn *conn = work->conn;
269 	int err;
270 
271 	rsp_hdr = smb_get_msg(work->response_buf);
272 	memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2);
273 	rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER;
274 	rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE;
275 	rsp_hdr->CreditRequest = cpu_to_le16(2);
276 	rsp_hdr->Command = SMB2_NEGOTIATE;
277 	rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR);
278 	rsp_hdr->NextCommand = 0;
279 	rsp_hdr->MessageId = 0;
280 	rsp_hdr->Id.SyncId.ProcessId = 0;
281 	rsp_hdr->Id.SyncId.TreeId = 0;
282 	rsp_hdr->SessionId = 0;
283 	memset(rsp_hdr->Signature, 0, 16);
284 
285 	rsp = smb_get_msg(work->response_buf);
286 
287 	WARN_ON(ksmbd_conn_good(conn));
288 
289 	rsp->StructureSize = cpu_to_le16(65);
290 	ksmbd_debug(SMB, "conn->dialect 0x%x\n", conn->dialect);
291 	rsp->DialectRevision = cpu_to_le16(conn->dialect);
292 	/* Not setting conn guid rsp->ServerGUID, as it
293 	 * not used by client for identifying connection
294 	 */
295 	rsp->Capabilities = cpu_to_le32(conn->vals->req_capabilities);
296 	/* Default Max Message Size till SMB2.0, 64K*/
297 	rsp->MaxTransactSize = cpu_to_le32(conn->vals->max_trans_size);
298 	rsp->MaxReadSize = cpu_to_le32(conn->vals->max_read_size);
299 	rsp->MaxWriteSize = cpu_to_le32(conn->vals->max_write_size);
300 
301 	rsp->SystemTime = cpu_to_le64(ksmbd_systime());
302 	rsp->ServerStartTime = 0;
303 
304 	rsp->SecurityBufferOffset = cpu_to_le16(128);
305 	rsp->SecurityBufferLength = cpu_to_le16(AUTH_GSS_LENGTH);
306 	ksmbd_copy_gss_neg_header((char *)(&rsp->hdr) +
307 		le16_to_cpu(rsp->SecurityBufferOffset));
308 	rsp->SecurityMode = SMB2_NEGOTIATE_SIGNING_ENABLED_LE;
309 	if (server_conf.signing == KSMBD_CONFIG_OPT_MANDATORY)
310 		rsp->SecurityMode |= SMB2_NEGOTIATE_SIGNING_REQUIRED_LE;
311 	err = ksmbd_iov_pin_rsp(work, rsp,
312 				sizeof(struct smb2_negotiate_rsp) + AUTH_GSS_LENGTH);
313 	if (err)
314 		return err;
315 	conn->use_spnego = true;
316 
317 	ksmbd_conn_set_need_negotiate(conn);
318 	return 0;
319 }
320 
321 /**
322  * smb2_set_rsp_credits() - set number of credits in response buffer
323  * @work:	smb work containing smb response buffer
324  */
325 int smb2_set_rsp_credits(struct ksmbd_work *work)
326 {
327 	struct smb2_hdr *req_hdr = ksmbd_req_buf_next(work);
328 	struct smb2_hdr *hdr = ksmbd_resp_buf_next(work);
329 	struct ksmbd_conn *conn = work->conn;
330 	unsigned short credits_requested, aux_max;
331 	unsigned short credit_charge, credits_granted = 0;
332 
333 	if (work->send_no_response)
334 		return 0;
335 
336 	hdr->CreditCharge = req_hdr->CreditCharge;
337 
338 	if (conn->total_credits > conn->vals->max_credits) {
339 		hdr->CreditRequest = 0;
340 		pr_err("Total credits overflow: %d\n", conn->total_credits);
341 		return -EINVAL;
342 	}
343 
344 	credit_charge = max_t(unsigned short,
345 			      le16_to_cpu(req_hdr->CreditCharge), 1);
346 	if (credit_charge > conn->total_credits) {
347 		ksmbd_debug(SMB, "Insufficient credits granted, given: %u, granted: %u\n",
348 			    credit_charge, conn->total_credits);
349 		return -EINVAL;
350 	}
351 
352 	conn->total_credits -= credit_charge;
353 	conn->outstanding_credits -= credit_charge;
354 	credits_requested = max_t(unsigned short,
355 				  le16_to_cpu(req_hdr->CreditRequest), 1);
356 
357 	/* according to smb2.credits smbtorture, Windows server
358 	 * 2016 or later grant up to 8192 credits at once.
359 	 *
360 	 * TODO: Need to adjuct CreditRequest value according to
361 	 * current cpu load
362 	 */
363 	if (hdr->Command == SMB2_NEGOTIATE)
364 		aux_max = 1;
365 	else
366 		aux_max = conn->vals->max_credits - conn->total_credits;
367 	credits_granted = min_t(unsigned short, credits_requested, aux_max);
368 
369 	conn->total_credits += credits_granted;
370 	work->credits_granted += credits_granted;
371 
372 	if (!req_hdr->NextCommand) {
373 		/* Update CreditRequest in last request */
374 		hdr->CreditRequest = cpu_to_le16(work->credits_granted);
375 	}
376 	ksmbd_debug(SMB,
377 		    "credits: requested[%d] granted[%d] total_granted[%d]\n",
378 		    credits_requested, credits_granted,
379 		    conn->total_credits);
380 	return 0;
381 }
382 
383 /**
384  * init_chained_smb2_rsp() - initialize smb2 chained response
385  * @work:	smb work containing smb response buffer
386  */
387 static void init_chained_smb2_rsp(struct ksmbd_work *work)
388 {
389 	struct smb2_hdr *req = ksmbd_req_buf_next(work);
390 	struct smb2_hdr *rsp = ksmbd_resp_buf_next(work);
391 	struct smb2_hdr *rsp_hdr;
392 	struct smb2_hdr *rcv_hdr;
393 	int next_hdr_offset = 0;
394 	int len, new_len;
395 
396 	/* Len of this response = updated RFC len - offset of previous cmd
397 	 * in the compound rsp
398 	 */
399 
400 	/* Storing the current local FID which may be needed by subsequent
401 	 * command in the compound request
402 	 */
403 	if (req->Command == SMB2_CREATE && rsp->Status == STATUS_SUCCESS) {
404 		work->compound_fid = ((struct smb2_create_rsp *)rsp)->VolatileFileId;
405 		work->compound_pfid = ((struct smb2_create_rsp *)rsp)->PersistentFileId;
406 		work->compound_sid = le64_to_cpu(rsp->SessionId);
407 	}
408 
409 	len = get_rfc1002_len(work->response_buf) - work->next_smb2_rsp_hdr_off;
410 	next_hdr_offset = le32_to_cpu(req->NextCommand);
411 
412 	new_len = ALIGN(len, 8);
413 	work->iov[work->iov_idx].iov_len += (new_len - len);
414 	inc_rfc1001_len(work->response_buf, new_len - len);
415 	rsp->NextCommand = cpu_to_le32(new_len);
416 
417 	work->next_smb2_rcv_hdr_off += next_hdr_offset;
418 	work->curr_smb2_rsp_hdr_off = work->next_smb2_rsp_hdr_off;
419 	work->next_smb2_rsp_hdr_off += new_len;
420 	ksmbd_debug(SMB,
421 		    "Compound req new_len = %d rcv off = %d rsp off = %d\n",
422 		    new_len, work->next_smb2_rcv_hdr_off,
423 		    work->next_smb2_rsp_hdr_off);
424 
425 	rsp_hdr = ksmbd_resp_buf_next(work);
426 	rcv_hdr = ksmbd_req_buf_next(work);
427 
428 	if (!(rcv_hdr->Flags & SMB2_FLAGS_RELATED_OPERATIONS)) {
429 		ksmbd_debug(SMB, "related flag should be set\n");
430 		work->compound_fid = KSMBD_NO_FID;
431 		work->compound_pfid = KSMBD_NO_FID;
432 	}
433 	memset((char *)rsp_hdr, 0, sizeof(struct smb2_hdr) + 2);
434 	rsp_hdr->ProtocolId = SMB2_PROTO_NUMBER;
435 	rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE;
436 	rsp_hdr->Command = rcv_hdr->Command;
437 
438 	/*
439 	 * Message is response. We don't grant oplock yet.
440 	 */
441 	rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR |
442 				SMB2_FLAGS_RELATED_OPERATIONS);
443 	rsp_hdr->NextCommand = 0;
444 	rsp_hdr->MessageId = rcv_hdr->MessageId;
445 	rsp_hdr->Id.SyncId.ProcessId = rcv_hdr->Id.SyncId.ProcessId;
446 	rsp_hdr->Id.SyncId.TreeId = rcv_hdr->Id.SyncId.TreeId;
447 	rsp_hdr->SessionId = rcv_hdr->SessionId;
448 	memcpy(rsp_hdr->Signature, rcv_hdr->Signature, 16);
449 }
450 
451 /**
452  * is_chained_smb2_message() - check for chained command
453  * @work:	smb work containing smb request buffer
454  *
455  * Return:      true if chained request, otherwise false
456  */
457 bool is_chained_smb2_message(struct ksmbd_work *work)
458 {
459 	struct smb2_hdr *hdr = smb_get_msg(work->request_buf);
460 	unsigned int len, next_cmd;
461 
462 	if (hdr->ProtocolId != SMB2_PROTO_NUMBER)
463 		return false;
464 
465 	hdr = ksmbd_req_buf_next(work);
466 	next_cmd = le32_to_cpu(hdr->NextCommand);
467 	if (next_cmd > 0) {
468 		if ((u64)work->next_smb2_rcv_hdr_off + next_cmd +
469 			__SMB2_HEADER_STRUCTURE_SIZE >
470 		    get_rfc1002_len(work->request_buf)) {
471 			pr_err("next command(%u) offset exceeds smb msg size\n",
472 			       next_cmd);
473 			return false;
474 		}
475 
476 		if ((u64)get_rfc1002_len(work->response_buf) + MAX_CIFS_SMALL_BUFFER_SIZE >
477 		    work->response_sz) {
478 			pr_err("next response offset exceeds response buffer size\n");
479 			return false;
480 		}
481 
482 		ksmbd_debug(SMB, "got SMB2 chained command\n");
483 		init_chained_smb2_rsp(work);
484 		return true;
485 	} else if (work->next_smb2_rcv_hdr_off) {
486 		/*
487 		 * This is last request in chained command,
488 		 * align response to 8 byte
489 		 */
490 		len = ALIGN(get_rfc1002_len(work->response_buf), 8);
491 		len = len - get_rfc1002_len(work->response_buf);
492 		if (len) {
493 			ksmbd_debug(SMB, "padding len %u\n", len);
494 			work->iov[work->iov_idx].iov_len += len;
495 			inc_rfc1001_len(work->response_buf, len);
496 		}
497 		work->curr_smb2_rsp_hdr_off = work->next_smb2_rsp_hdr_off;
498 	}
499 	return false;
500 }
501 
502 /**
503  * init_smb2_rsp_hdr() - initialize smb2 response
504  * @work:	smb work containing smb request buffer
505  *
506  * Return:      0
507  */
508 int init_smb2_rsp_hdr(struct ksmbd_work *work)
509 {
510 	struct smb2_hdr *rsp_hdr = smb_get_msg(work->response_buf);
511 	struct smb2_hdr *rcv_hdr = smb_get_msg(work->request_buf);
512 
513 	memset(rsp_hdr, 0, sizeof(struct smb2_hdr) + 2);
514 	rsp_hdr->ProtocolId = rcv_hdr->ProtocolId;
515 	rsp_hdr->StructureSize = SMB2_HEADER_STRUCTURE_SIZE;
516 	rsp_hdr->Command = rcv_hdr->Command;
517 
518 	/*
519 	 * Message is response. We don't grant oplock yet.
520 	 */
521 	rsp_hdr->Flags = (SMB2_FLAGS_SERVER_TO_REDIR);
522 	rsp_hdr->NextCommand = 0;
523 	rsp_hdr->MessageId = rcv_hdr->MessageId;
524 	rsp_hdr->Id.SyncId.ProcessId = rcv_hdr->Id.SyncId.ProcessId;
525 	rsp_hdr->Id.SyncId.TreeId = rcv_hdr->Id.SyncId.TreeId;
526 	rsp_hdr->SessionId = rcv_hdr->SessionId;
527 	memcpy(rsp_hdr->Signature, rcv_hdr->Signature, 16);
528 
529 	return 0;
530 }
531 
532 /**
533  * smb2_allocate_rsp_buf() - allocate smb2 response buffer
534  * @work:	smb work containing smb request buffer
535  *
536  * Return:      0 on success, otherwise error
537  */
538 int smb2_allocate_rsp_buf(struct ksmbd_work *work)
539 {
540 	struct smb2_hdr *hdr = smb_get_msg(work->request_buf);
541 	size_t small_sz = MAX_CIFS_SMALL_BUFFER_SIZE;
542 	size_t large_sz = small_sz + work->conn->vals->max_trans_size;
543 	size_t sz = small_sz;
544 	int cmd = le16_to_cpu(hdr->Command);
545 
546 	if (cmd == SMB2_IOCTL_HE || cmd == SMB2_QUERY_DIRECTORY_HE)
547 		sz = large_sz;
548 
549 	if (cmd == SMB2_QUERY_INFO_HE) {
550 		struct smb2_query_info_req *req;
551 
552 		if (get_rfc1002_len(work->request_buf) <
553 		    offsetof(struct smb2_query_info_req, OutputBufferLength))
554 			return -EINVAL;
555 
556 		req = smb_get_msg(work->request_buf);
557 		if ((req->InfoType == SMB2_O_INFO_FILE &&
558 		     (req->FileInfoClass == FILE_FULL_EA_INFORMATION ||
559 		     req->FileInfoClass == FILE_ALL_INFORMATION)) ||
560 		    req->InfoType == SMB2_O_INFO_SECURITY)
561 			sz = large_sz;
562 	}
563 
564 	/* allocate large response buf for chained commands */
565 	if (le32_to_cpu(hdr->NextCommand) > 0)
566 		sz = large_sz;
567 
568 	work->response_buf = kvzalloc(sz, KSMBD_DEFAULT_GFP);
569 	if (!work->response_buf)
570 		return -ENOMEM;
571 
572 	work->response_sz = sz;
573 	return 0;
574 }
575 
576 /**
577  * smb2_check_user_session() - check for valid session for a user
578  * @work:	smb work containing smb request buffer
579  *
580  * Return:      0 on success, otherwise error
581  */
582 int smb2_check_user_session(struct ksmbd_work *work)
583 {
584 	struct smb2_hdr *req_hdr = ksmbd_req_buf_next(work);
585 	struct ksmbd_conn *conn = work->conn;
586 	unsigned int cmd = le16_to_cpu(req_hdr->Command);
587 	unsigned long long sess_id;
588 
589 	/*
590 	 * SMB2_ECHO, SMB2_NEGOTIATE, SMB2_SESSION_SETUP command do not
591 	 * require a session id, so no need to validate user session's for
592 	 * these commands.
593 	 */
594 	if (cmd == SMB2_ECHO_HE || cmd == SMB2_NEGOTIATE_HE ||
595 	    cmd == SMB2_SESSION_SETUP_HE)
596 		return 0;
597 
598 	if (!ksmbd_conn_good(conn))
599 		return -EIO;
600 
601 	sess_id = le64_to_cpu(req_hdr->SessionId);
602 
603 	/*
604 	 * If request is not the first in Compound request,
605 	 * Just validate session id in header with work->sess->id.
606 	 */
607 	if (work->next_smb2_rcv_hdr_off) {
608 		if (!work->sess) {
609 			pr_err("The first operation in the compound does not have sess\n");
610 			return -EINVAL;
611 		}
612 		if (sess_id != ULLONG_MAX && work->sess->id != sess_id) {
613 			pr_err("session id(%llu) is different with the first operation(%lld)\n",
614 					sess_id, work->sess->id);
615 			return -EINVAL;
616 		}
617 		return 1;
618 	}
619 
620 	/* Check for validity of user session */
621 	work->sess = ksmbd_session_lookup_all(conn, sess_id);
622 	if (work->sess)
623 		return 1;
624 	ksmbd_debug(SMB, "Invalid user session, Uid %llu\n", sess_id);
625 	return -ENOENT;
626 }
627 
628 /**
629  * smb2_get_name() - get filename string from on the wire smb format
630  * @src:	source buffer
631  * @maxlen:	maxlen of source string
632  * @local_nls:	nls_table pointer
633  *
634  * Return:      matching converted filename on success, otherwise error ptr
635  */
636 static char *
637 smb2_get_name(const char *src, const int maxlen, struct nls_table *local_nls)
638 {
639 	char *name;
640 
641 	name = smb_strndup_from_utf16(src, maxlen, 1, local_nls);
642 	if (IS_ERR(name)) {
643 		pr_err("failed to get name %ld\n", PTR_ERR(name));
644 		return name;
645 	}
646 
647 	if (*name == '\0') {
648 		kfree(name);
649 		return ERR_PTR(-EINVAL);
650 	}
651 
652 	if (*name == '\\') {
653 		pr_err("not allow directory name included leading slash\n");
654 		kfree(name);
655 		return ERR_PTR(-EINVAL);
656 	}
657 
658 	ksmbd_conv_path_to_unix(name);
659 	ksmbd_strip_last_slash(name);
660 	return name;
661 }
662 
663 int setup_async_work(struct ksmbd_work *work, void (*fn)(void **), void **arg)
664 {
665 	struct ksmbd_conn *conn = work->conn;
666 	int id;
667 
668 	id = ksmbd_acquire_async_msg_id(&conn->async_ida);
669 	if (id < 0) {
670 		pr_err("Failed to alloc async message id\n");
671 		return id;
672 	}
673 	work->asynchronous = true;
674 	work->async_id = id;
675 
676 	ksmbd_debug(SMB,
677 		    "Send interim Response to inform async request id : %d\n",
678 		    work->async_id);
679 
680 	work->cancel_fn = fn;
681 	work->cancel_argv = arg;
682 
683 	if (list_empty(&work->async_request_entry)) {
684 		spin_lock(&conn->request_lock);
685 		list_add_tail(&work->async_request_entry, &conn->async_requests);
686 		spin_unlock(&conn->request_lock);
687 	}
688 
689 	return 0;
690 }
691 
692 void release_async_work(struct ksmbd_work *work)
693 {
694 	struct ksmbd_conn *conn = work->conn;
695 
696 	spin_lock(&conn->request_lock);
697 	list_del_init(&work->async_request_entry);
698 	spin_unlock(&conn->request_lock);
699 
700 	work->asynchronous = 0;
701 	work->cancel_fn = NULL;
702 	kfree(work->cancel_argv);
703 	work->cancel_argv = NULL;
704 	if (work->async_id) {
705 		ksmbd_release_id(&conn->async_ida, work->async_id);
706 		work->async_id = 0;
707 	}
708 }
709 
710 void smb2_send_interim_resp(struct ksmbd_work *work, __le32 status)
711 {
712 	struct smb2_hdr *rsp_hdr;
713 	struct ksmbd_work *in_work = ksmbd_alloc_work_struct();
714 
715 	if (!in_work)
716 		return;
717 
718 	if (allocate_interim_rsp_buf(in_work)) {
719 		pr_err("smb_allocate_rsp_buf failed!\n");
720 		ksmbd_free_work_struct(in_work);
721 		return;
722 	}
723 
724 	in_work->conn = work->conn;
725 	memcpy(smb_get_msg(in_work->response_buf), ksmbd_resp_buf_next(work),
726 	       __SMB2_HEADER_STRUCTURE_SIZE);
727 
728 	rsp_hdr = smb_get_msg(in_work->response_buf);
729 	rsp_hdr->Flags |= SMB2_FLAGS_ASYNC_COMMAND;
730 	rsp_hdr->Id.AsyncId = cpu_to_le64(work->async_id);
731 	smb2_set_err_rsp(in_work);
732 	rsp_hdr->Status = status;
733 
734 	ksmbd_conn_write(in_work);
735 	ksmbd_free_work_struct(in_work);
736 }
737 
738 static __le32 smb2_get_reparse_tag_special_file(umode_t mode)
739 {
740 	if (S_ISDIR(mode) || S_ISREG(mode))
741 		return 0;
742 
743 	if (S_ISLNK(mode))
744 		return IO_REPARSE_TAG_LX_SYMLINK_LE;
745 	else if (S_ISFIFO(mode))
746 		return IO_REPARSE_TAG_LX_FIFO_LE;
747 	else if (S_ISSOCK(mode))
748 		return IO_REPARSE_TAG_AF_UNIX_LE;
749 	else if (S_ISCHR(mode))
750 		return IO_REPARSE_TAG_LX_CHR_LE;
751 	else if (S_ISBLK(mode))
752 		return IO_REPARSE_TAG_LX_BLK_LE;
753 
754 	return 0;
755 }
756 
757 /**
758  * smb2_get_dos_mode() - get file mode in dos format from unix mode
759  * @stat:	kstat containing file mode
760  * @attribute:	attribute flags
761  *
762  * Return:      converted dos mode
763  */
764 static int smb2_get_dos_mode(struct kstat *stat, int attribute)
765 {
766 	int attr = 0;
767 
768 	if (S_ISDIR(stat->mode)) {
769 		attr = FILE_ATTRIBUTE_DIRECTORY |
770 			(attribute & (FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM));
771 	} else {
772 		attr = (attribute & 0x00005137) | FILE_ATTRIBUTE_ARCHIVE;
773 		attr &= ~(FILE_ATTRIBUTE_DIRECTORY);
774 		if (S_ISREG(stat->mode) && (server_conf.share_fake_fscaps &
775 				FILE_SUPPORTS_SPARSE_FILES))
776 			attr |= FILE_ATTRIBUTE_SPARSE_FILE;
777 
778 		if (smb2_get_reparse_tag_special_file(stat->mode))
779 			attr |= FILE_ATTRIBUTE_REPARSE_POINT;
780 	}
781 
782 	return attr;
783 }
784 
785 static void build_preauth_ctxt(struct smb2_preauth_neg_context *pneg_ctxt,
786 			       __le16 hash_id)
787 {
788 	pneg_ctxt->ContextType = SMB2_PREAUTH_INTEGRITY_CAPABILITIES;
789 	pneg_ctxt->DataLength = cpu_to_le16(38);
790 	pneg_ctxt->HashAlgorithmCount = cpu_to_le16(1);
791 	pneg_ctxt->Reserved = cpu_to_le32(0);
792 	pneg_ctxt->SaltLength = cpu_to_le16(SMB311_SALT_SIZE);
793 	get_random_bytes(pneg_ctxt->Salt, SMB311_SALT_SIZE);
794 	pneg_ctxt->HashAlgorithms = hash_id;
795 }
796 
797 static void build_encrypt_ctxt(struct smb2_encryption_neg_context *pneg_ctxt,
798 			       __le16 cipher_type)
799 {
800 	pneg_ctxt->ContextType = SMB2_ENCRYPTION_CAPABILITIES;
801 	pneg_ctxt->DataLength = cpu_to_le16(4);
802 	pneg_ctxt->Reserved = cpu_to_le32(0);
803 	pneg_ctxt->CipherCount = cpu_to_le16(1);
804 	pneg_ctxt->Ciphers[0] = cipher_type;
805 }
806 
807 static void build_sign_cap_ctxt(struct smb2_signing_capabilities *pneg_ctxt,
808 				__le16 sign_algo)
809 {
810 	pneg_ctxt->ContextType = SMB2_SIGNING_CAPABILITIES;
811 	pneg_ctxt->DataLength =
812 		cpu_to_le16((sizeof(struct smb2_signing_capabilities) + 2)
813 			- sizeof(struct smb2_neg_context));
814 	pneg_ctxt->Reserved = cpu_to_le32(0);
815 	pneg_ctxt->SigningAlgorithmCount = cpu_to_le16(1);
816 	pneg_ctxt->SigningAlgorithms[0] = sign_algo;
817 }
818 
819 static void build_posix_ctxt(struct smb2_posix_neg_context *pneg_ctxt)
820 {
821 	pneg_ctxt->ContextType = SMB2_POSIX_EXTENSIONS_AVAILABLE;
822 	pneg_ctxt->DataLength = cpu_to_le16(POSIX_CTXT_DATA_LEN);
823 	/* SMB2_CREATE_TAG_POSIX is "0x93AD25509CB411E7B42383DE968BCD7C" */
824 	pneg_ctxt->Name[0] = 0x93;
825 	pneg_ctxt->Name[1] = 0xAD;
826 	pneg_ctxt->Name[2] = 0x25;
827 	pneg_ctxt->Name[3] = 0x50;
828 	pneg_ctxt->Name[4] = 0x9C;
829 	pneg_ctxt->Name[5] = 0xB4;
830 	pneg_ctxt->Name[6] = 0x11;
831 	pneg_ctxt->Name[7] = 0xE7;
832 	pneg_ctxt->Name[8] = 0xB4;
833 	pneg_ctxt->Name[9] = 0x23;
834 	pneg_ctxt->Name[10] = 0x83;
835 	pneg_ctxt->Name[11] = 0xDE;
836 	pneg_ctxt->Name[12] = 0x96;
837 	pneg_ctxt->Name[13] = 0x8B;
838 	pneg_ctxt->Name[14] = 0xCD;
839 	pneg_ctxt->Name[15] = 0x7C;
840 }
841 
842 static unsigned int assemble_neg_contexts(struct ksmbd_conn *conn,
843 				  struct smb2_negotiate_rsp *rsp)
844 {
845 	char * const pneg_ctxt = (char *)rsp +
846 			le32_to_cpu(rsp->NegotiateContextOffset);
847 	int neg_ctxt_cnt = 1;
848 	int ctxt_size;
849 
850 	ksmbd_debug(SMB,
851 		    "assemble SMB2_PREAUTH_INTEGRITY_CAPABILITIES context\n");
852 	build_preauth_ctxt((struct smb2_preauth_neg_context *)pneg_ctxt,
853 			   conn->preauth_info->Preauth_HashId);
854 	ctxt_size = sizeof(struct smb2_preauth_neg_context);
855 
856 	if (conn->cipher_type) {
857 		/* Round to 8 byte boundary */
858 		ctxt_size = round_up(ctxt_size, 8);
859 		ksmbd_debug(SMB,
860 			    "assemble SMB2_ENCRYPTION_CAPABILITIES context\n");
861 		build_encrypt_ctxt((struct smb2_encryption_neg_context *)
862 				   (pneg_ctxt + ctxt_size),
863 				   conn->cipher_type);
864 		neg_ctxt_cnt++;
865 		ctxt_size += sizeof(struct smb2_encryption_neg_context) + 2;
866 	}
867 
868 	/* compression context not yet supported */
869 	WARN_ON(conn->compress_algorithm != SMB3_COMPRESS_NONE);
870 
871 	if (conn->posix_ext_supported) {
872 		ctxt_size = round_up(ctxt_size, 8);
873 		ksmbd_debug(SMB,
874 			    "assemble SMB2_POSIX_EXTENSIONS_AVAILABLE context\n");
875 		build_posix_ctxt((struct smb2_posix_neg_context *)
876 				 (pneg_ctxt + ctxt_size));
877 		neg_ctxt_cnt++;
878 		ctxt_size += sizeof(struct smb2_posix_neg_context);
879 	}
880 
881 	if (conn->signing_negotiated) {
882 		ctxt_size = round_up(ctxt_size, 8);
883 		ksmbd_debug(SMB,
884 			    "assemble SMB2_SIGNING_CAPABILITIES context\n");
885 		build_sign_cap_ctxt((struct smb2_signing_capabilities *)
886 				    (pneg_ctxt + ctxt_size),
887 				    conn->signing_algorithm);
888 		neg_ctxt_cnt++;
889 		ctxt_size += sizeof(struct smb2_signing_capabilities) + 2;
890 	}
891 
892 	rsp->NegotiateContextCount = cpu_to_le16(neg_ctxt_cnt);
893 	return ctxt_size + AUTH_GSS_PADDING;
894 }
895 
896 static __le32 decode_preauth_ctxt(struct ksmbd_conn *conn,
897 				  struct smb2_preauth_neg_context *pneg_ctxt,
898 				  int ctxt_len)
899 {
900 	/*
901 	 * sizeof(smb2_preauth_neg_context) assumes SMB311_SALT_SIZE Salt,
902 	 * which may not be present. Only check for used HashAlgorithms[1].
903 	 */
904 	if (ctxt_len <
905 	    sizeof(struct smb2_neg_context) + MIN_PREAUTH_CTXT_DATA_LEN)
906 		return STATUS_INVALID_PARAMETER;
907 
908 	if (pneg_ctxt->HashAlgorithms != SMB2_PREAUTH_INTEGRITY_SHA512)
909 		return STATUS_SMB_NO_PREAUTH_INTEGRITY_HASH_OVERLAP;
910 
911 	conn->preauth_info->Preauth_HashId = SMB2_PREAUTH_INTEGRITY_SHA512;
912 	return STATUS_SUCCESS;
913 }
914 
915 static void decode_encrypt_ctxt(struct ksmbd_conn *conn,
916 				struct smb2_encryption_neg_context *pneg_ctxt,
917 				int ctxt_len)
918 {
919 	int cph_cnt;
920 	int i, cphs_size;
921 
922 	if (sizeof(struct smb2_encryption_neg_context) > ctxt_len) {
923 		pr_err("Invalid SMB2_ENCRYPTION_CAPABILITIES context size\n");
924 		return;
925 	}
926 
927 	conn->cipher_type = 0;
928 
929 	cph_cnt = le16_to_cpu(pneg_ctxt->CipherCount);
930 	cphs_size = cph_cnt * sizeof(__le16);
931 
932 	if (sizeof(struct smb2_encryption_neg_context) + cphs_size >
933 	    ctxt_len) {
934 		pr_err("Invalid cipher count(%d)\n", cph_cnt);
935 		return;
936 	}
937 
938 	if (server_conf.flags & KSMBD_GLOBAL_FLAG_SMB2_ENCRYPTION_OFF)
939 		return;
940 
941 	for (i = 0; i < cph_cnt; i++) {
942 		if (pneg_ctxt->Ciphers[i] == SMB2_ENCRYPTION_AES128_GCM ||
943 		    pneg_ctxt->Ciphers[i] == SMB2_ENCRYPTION_AES128_CCM ||
944 		    pneg_ctxt->Ciphers[i] == SMB2_ENCRYPTION_AES256_CCM ||
945 		    pneg_ctxt->Ciphers[i] == SMB2_ENCRYPTION_AES256_GCM) {
946 			ksmbd_debug(SMB, "Cipher ID = 0x%x\n",
947 				    pneg_ctxt->Ciphers[i]);
948 			conn->cipher_type = pneg_ctxt->Ciphers[i];
949 			break;
950 		}
951 	}
952 }
953 
954 /**
955  * smb3_encryption_negotiated() - checks if server and client agreed on enabling encryption
956  * @conn:	smb connection
957  *
958  * Return:	true if connection should be encrypted, else false
959  */
960 bool smb3_encryption_negotiated(struct ksmbd_conn *conn)
961 {
962 	if (!conn->ops->generate_encryptionkey)
963 		return false;
964 
965 	/*
966 	 * SMB 3.0 and 3.0.2 dialects use the SMB2_GLOBAL_CAP_ENCRYPTION flag.
967 	 * SMB 3.1.1 uses the cipher_type field.
968 	 */
969 	return (conn->vals->req_capabilities & SMB2_GLOBAL_CAP_ENCRYPTION) ||
970 	    conn->cipher_type;
971 }
972 
973 static void decode_compress_ctxt(struct ksmbd_conn *conn,
974 				 struct smb2_compression_capabilities_context *pneg_ctxt)
975 {
976 	conn->compress_algorithm = SMB3_COMPRESS_NONE;
977 }
978 
979 static void decode_sign_cap_ctxt(struct ksmbd_conn *conn,
980 				 struct smb2_signing_capabilities *pneg_ctxt,
981 				 int ctxt_len)
982 {
983 	int sign_algo_cnt;
984 	int i, sign_alos_size;
985 
986 	if (sizeof(struct smb2_signing_capabilities) > ctxt_len) {
987 		pr_err("Invalid SMB2_SIGNING_CAPABILITIES context length\n");
988 		return;
989 	}
990 
991 	conn->signing_negotiated = false;
992 	sign_algo_cnt = le16_to_cpu(pneg_ctxt->SigningAlgorithmCount);
993 	sign_alos_size = sign_algo_cnt * sizeof(__le16);
994 
995 	if (sizeof(struct smb2_signing_capabilities) + sign_alos_size >
996 	    ctxt_len) {
997 		pr_err("Invalid signing algorithm count(%d)\n", sign_algo_cnt);
998 		return;
999 	}
1000 
1001 	for (i = 0; i < sign_algo_cnt; i++) {
1002 		if (pneg_ctxt->SigningAlgorithms[i] == SIGNING_ALG_HMAC_SHA256_LE ||
1003 		    pneg_ctxt->SigningAlgorithms[i] == SIGNING_ALG_AES_CMAC_LE) {
1004 			ksmbd_debug(SMB, "Signing Algorithm ID = 0x%x\n",
1005 				    pneg_ctxt->SigningAlgorithms[i]);
1006 			conn->signing_negotiated = true;
1007 			conn->signing_algorithm =
1008 				pneg_ctxt->SigningAlgorithms[i];
1009 			break;
1010 		}
1011 	}
1012 }
1013 
1014 static __le32 deassemble_neg_contexts(struct ksmbd_conn *conn,
1015 				      struct smb2_negotiate_req *req,
1016 				      unsigned int len_of_smb)
1017 {
1018 	/* +4 is to account for the RFC1001 len field */
1019 	struct smb2_neg_context *pctx = (struct smb2_neg_context *)req;
1020 	int i = 0, len_of_ctxts;
1021 	unsigned int offset = le32_to_cpu(req->NegotiateContextOffset);
1022 	unsigned int neg_ctxt_cnt = le16_to_cpu(req->NegotiateContextCount);
1023 	__le32 status = STATUS_INVALID_PARAMETER;
1024 
1025 	ksmbd_debug(SMB, "decoding %d negotiate contexts\n", neg_ctxt_cnt);
1026 	if (len_of_smb <= offset) {
1027 		ksmbd_debug(SMB, "Invalid response: negotiate context offset\n");
1028 		return status;
1029 	}
1030 
1031 	len_of_ctxts = len_of_smb - offset;
1032 
1033 	while (i++ < neg_ctxt_cnt) {
1034 		int clen, ctxt_len;
1035 
1036 		if (len_of_ctxts < (int)sizeof(struct smb2_neg_context))
1037 			break;
1038 
1039 		pctx = (struct smb2_neg_context *)((char *)pctx + offset);
1040 		clen = le16_to_cpu(pctx->DataLength);
1041 		ctxt_len = clen + sizeof(struct smb2_neg_context);
1042 
1043 		if (ctxt_len > len_of_ctxts)
1044 			break;
1045 
1046 		if (pctx->ContextType == SMB2_PREAUTH_INTEGRITY_CAPABILITIES) {
1047 			ksmbd_debug(SMB,
1048 				    "deassemble SMB2_PREAUTH_INTEGRITY_CAPABILITIES context\n");
1049 			if (conn->preauth_info->Preauth_HashId)
1050 				break;
1051 
1052 			status = decode_preauth_ctxt(conn,
1053 						     (struct smb2_preauth_neg_context *)pctx,
1054 						     ctxt_len);
1055 			if (status != STATUS_SUCCESS)
1056 				break;
1057 		} else if (pctx->ContextType == SMB2_ENCRYPTION_CAPABILITIES) {
1058 			ksmbd_debug(SMB,
1059 				    "deassemble SMB2_ENCRYPTION_CAPABILITIES context\n");
1060 			if (conn->cipher_type)
1061 				break;
1062 
1063 			decode_encrypt_ctxt(conn,
1064 					    (struct smb2_encryption_neg_context *)pctx,
1065 					    ctxt_len);
1066 		} else if (pctx->ContextType == SMB2_COMPRESSION_CAPABILITIES) {
1067 			ksmbd_debug(SMB,
1068 				    "deassemble SMB2_COMPRESSION_CAPABILITIES context\n");
1069 			if (conn->compress_algorithm)
1070 				break;
1071 
1072 			decode_compress_ctxt(conn,
1073 					     (struct smb2_compression_capabilities_context *)pctx);
1074 		} else if (pctx->ContextType == SMB2_NETNAME_NEGOTIATE_CONTEXT_ID) {
1075 			ksmbd_debug(SMB,
1076 				    "deassemble SMB2_NETNAME_NEGOTIATE_CONTEXT_ID context\n");
1077 		} else if (pctx->ContextType == SMB2_POSIX_EXTENSIONS_AVAILABLE) {
1078 			ksmbd_debug(SMB,
1079 				    "deassemble SMB2_POSIX_EXTENSIONS_AVAILABLE context\n");
1080 			conn->posix_ext_supported = true;
1081 		} else if (pctx->ContextType == SMB2_SIGNING_CAPABILITIES) {
1082 			ksmbd_debug(SMB,
1083 				    "deassemble SMB2_SIGNING_CAPABILITIES context\n");
1084 
1085 			decode_sign_cap_ctxt(conn,
1086 					     (struct smb2_signing_capabilities *)pctx,
1087 					     ctxt_len);
1088 		}
1089 
1090 		/* offsets must be 8 byte aligned */
1091 		offset = (ctxt_len + 7) & ~0x7;
1092 		len_of_ctxts -= offset;
1093 	}
1094 	return status;
1095 }
1096 
1097 /**
1098  * smb2_handle_negotiate() - handler for smb2 negotiate command
1099  * @work:	smb work containing smb request buffer
1100  *
1101  * Return:      0
1102  */
1103 int smb2_handle_negotiate(struct ksmbd_work *work)
1104 {
1105 	struct ksmbd_conn *conn = work->conn;
1106 	struct smb2_negotiate_req *req = smb_get_msg(work->request_buf);
1107 	struct smb2_negotiate_rsp *rsp = smb_get_msg(work->response_buf);
1108 	int rc = 0;
1109 	unsigned int smb2_buf_len, smb2_neg_size, neg_ctxt_len = 0;
1110 	__le32 status;
1111 
1112 	ksmbd_debug(SMB, "Received negotiate request\n");
1113 	conn->need_neg = false;
1114 	if (ksmbd_conn_good(conn)) {
1115 		pr_err("conn->tcp_status is already in CifsGood State\n");
1116 		work->send_no_response = 1;
1117 		return rc;
1118 	}
1119 
1120 	ksmbd_conn_lock(conn);
1121 	smb2_buf_len = get_rfc1002_len(work->request_buf);
1122 	smb2_neg_size = offsetof(struct smb2_negotiate_req, Dialects);
1123 	if (smb2_neg_size > smb2_buf_len) {
1124 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1125 		rc = -EINVAL;
1126 		goto err_out;
1127 	}
1128 
1129 	if (req->DialectCount == 0) {
1130 		pr_err("malformed packet\n");
1131 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1132 		rc = -EINVAL;
1133 		goto err_out;
1134 	}
1135 
1136 	if (conn->dialect == SMB311_PROT_ID) {
1137 		unsigned int nego_ctxt_off = le32_to_cpu(req->NegotiateContextOffset);
1138 
1139 		if (smb2_buf_len < nego_ctxt_off) {
1140 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1141 			rc = -EINVAL;
1142 			goto err_out;
1143 		}
1144 
1145 		if (smb2_neg_size > nego_ctxt_off) {
1146 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1147 			rc = -EINVAL;
1148 			goto err_out;
1149 		}
1150 
1151 		if (smb2_neg_size + le16_to_cpu(req->DialectCount) * sizeof(__le16) >
1152 		    nego_ctxt_off) {
1153 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1154 			rc = -EINVAL;
1155 			goto err_out;
1156 		}
1157 	} else {
1158 		if (smb2_neg_size + le16_to_cpu(req->DialectCount) * sizeof(__le16) >
1159 		    smb2_buf_len) {
1160 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1161 			rc = -EINVAL;
1162 			goto err_out;
1163 		}
1164 	}
1165 
1166 	conn->cli_cap = le32_to_cpu(req->Capabilities);
1167 	switch (conn->dialect) {
1168 	case SMB311_PROT_ID:
1169 		conn->preauth_info =
1170 			kzalloc_obj(struct preauth_integrity_info,
1171 				    KSMBD_DEFAULT_GFP);
1172 		if (!conn->preauth_info) {
1173 			rc = -ENOMEM;
1174 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1175 			goto err_out;
1176 		}
1177 
1178 		status = deassemble_neg_contexts(conn, req,
1179 						 get_rfc1002_len(work->request_buf));
1180 		if (status != STATUS_SUCCESS) {
1181 			pr_err("deassemble_neg_contexts error(0x%x)\n",
1182 			       status);
1183 			rsp->hdr.Status = status;
1184 			rc = -EINVAL;
1185 			kfree(conn->preauth_info);
1186 			conn->preauth_info = NULL;
1187 			goto err_out;
1188 		}
1189 
1190 		rc = init_smb3_11_server(conn);
1191 		if (rc < 0) {
1192 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1193 			kfree(conn->preauth_info);
1194 			conn->preauth_info = NULL;
1195 			goto err_out;
1196 		}
1197 
1198 		ksmbd_gen_preauth_integrity_hash(conn,
1199 						 work->request_buf,
1200 						 conn->preauth_info->Preauth_HashValue);
1201 		rsp->NegotiateContextOffset =
1202 				cpu_to_le32(OFFSET_OF_NEG_CONTEXT);
1203 		neg_ctxt_len = assemble_neg_contexts(conn, rsp);
1204 		break;
1205 	case SMB302_PROT_ID:
1206 		init_smb3_02_server(conn);
1207 		break;
1208 	case SMB30_PROT_ID:
1209 		init_smb3_0_server(conn);
1210 		break;
1211 	case SMB21_PROT_ID:
1212 		init_smb2_1_server(conn);
1213 		break;
1214 	case SMB2X_PROT_ID:
1215 	case BAD_PROT_ID:
1216 	default:
1217 		ksmbd_debug(SMB, "Server dialect :0x%x not supported\n",
1218 			    conn->dialect);
1219 		rsp->hdr.Status = STATUS_NOT_SUPPORTED;
1220 		rc = -EINVAL;
1221 		goto err_out;
1222 	}
1223 	rsp->Capabilities = cpu_to_le32(conn->vals->req_capabilities);
1224 
1225 	/* For stats */
1226 	conn->connection_type = conn->dialect;
1227 
1228 	rsp->MaxTransactSize = cpu_to_le32(conn->vals->max_trans_size);
1229 	rsp->MaxReadSize = cpu_to_le32(conn->vals->max_read_size);
1230 	rsp->MaxWriteSize = cpu_to_le32(conn->vals->max_write_size);
1231 
1232 	memcpy(conn->ClientGUID, req->ClientGUID,
1233 			SMB2_CLIENT_GUID_SIZE);
1234 	conn->cli_sec_mode = le16_to_cpu(req->SecurityMode);
1235 
1236 	rsp->StructureSize = cpu_to_le16(65);
1237 	rsp->DialectRevision = cpu_to_le16(conn->dialect);
1238 	/* Not setting conn guid rsp->ServerGUID, as it
1239 	 * not used by client for identifying server
1240 	 */
1241 	memset(rsp->ServerGUID, 0, SMB2_CLIENT_GUID_SIZE);
1242 
1243 	rsp->SystemTime = cpu_to_le64(ksmbd_systime());
1244 	rsp->ServerStartTime = 0;
1245 	ksmbd_debug(SMB, "negotiate context offset %d, count %d\n",
1246 		    le32_to_cpu(rsp->NegotiateContextOffset),
1247 		    le16_to_cpu(rsp->NegotiateContextCount));
1248 
1249 	rsp->SecurityBufferOffset = cpu_to_le16(128);
1250 	rsp->SecurityBufferLength = cpu_to_le16(AUTH_GSS_LENGTH);
1251 	ksmbd_copy_gss_neg_header((char *)(&rsp->hdr) +
1252 				  le16_to_cpu(rsp->SecurityBufferOffset));
1253 
1254 	rsp->SecurityMode = SMB2_NEGOTIATE_SIGNING_ENABLED_LE;
1255 	conn->use_spnego = true;
1256 
1257 	if ((server_conf.signing == KSMBD_CONFIG_OPT_AUTO ||
1258 	     server_conf.signing == KSMBD_CONFIG_OPT_DISABLED) &&
1259 	    req->SecurityMode & SMB2_NEGOTIATE_SIGNING_REQUIRED_LE)
1260 		conn->sign = true;
1261 	else if (server_conf.signing == KSMBD_CONFIG_OPT_MANDATORY) {
1262 		server_conf.enforced_signing = true;
1263 		rsp->SecurityMode |= SMB2_NEGOTIATE_SIGNING_REQUIRED_LE;
1264 		conn->sign = true;
1265 	}
1266 
1267 	conn->srv_sec_mode = le16_to_cpu(rsp->SecurityMode);
1268 	ksmbd_conn_set_need_setup(conn);
1269 
1270 err_out:
1271 	ksmbd_conn_unlock(conn);
1272 	if (rc)
1273 		rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
1274 
1275 	if (!rc)
1276 		rc = ksmbd_iov_pin_rsp(work, rsp,
1277 				       sizeof(struct smb2_negotiate_rsp) +
1278 					AUTH_GSS_LENGTH + neg_ctxt_len);
1279 	if (rc < 0)
1280 		smb2_set_err_rsp(work);
1281 	return rc;
1282 }
1283 
1284 static int alloc_preauth_hash(struct ksmbd_session *sess,
1285 			      struct ksmbd_conn *conn)
1286 {
1287 	if (sess->Preauth_HashValue)
1288 		return 0;
1289 
1290 	if (!conn->preauth_info)
1291 		return -ENOMEM;
1292 
1293 	sess->Preauth_HashValue = kmemdup(conn->preauth_info->Preauth_HashValue,
1294 					  PREAUTH_HASHVALUE_SIZE, KSMBD_DEFAULT_GFP);
1295 	if (!sess->Preauth_HashValue)
1296 		return -ENOMEM;
1297 
1298 	return 0;
1299 }
1300 
1301 static int generate_preauth_hash(struct ksmbd_work *work)
1302 {
1303 	struct ksmbd_conn *conn = work->conn;
1304 	struct ksmbd_session *sess = work->sess;
1305 	u8 *preauth_hash;
1306 
1307 	if (conn->dialect != SMB311_PROT_ID)
1308 		return 0;
1309 
1310 	if (conn->binding) {
1311 		struct preauth_session *preauth_sess;
1312 
1313 		preauth_sess = ksmbd_preauth_session_lookup(conn, sess->id);
1314 		if (!preauth_sess) {
1315 			preauth_sess = ksmbd_preauth_session_alloc(conn, sess->id);
1316 			if (!preauth_sess)
1317 				return -ENOMEM;
1318 		}
1319 
1320 		preauth_hash = preauth_sess->Preauth_HashValue;
1321 	} else {
1322 		if (!sess->Preauth_HashValue)
1323 			if (alloc_preauth_hash(sess, conn))
1324 				return -ENOMEM;
1325 		preauth_hash = sess->Preauth_HashValue;
1326 	}
1327 
1328 	ksmbd_gen_preauth_integrity_hash(conn, work->request_buf, preauth_hash);
1329 	return 0;
1330 }
1331 
1332 static int decode_negotiation_token(struct ksmbd_conn *conn,
1333 				    struct negotiate_message *negblob,
1334 				    size_t sz)
1335 {
1336 	if (!conn->use_spnego)
1337 		return -EINVAL;
1338 
1339 	if (ksmbd_decode_negTokenInit((char *)negblob, sz, conn)) {
1340 		if (ksmbd_decode_negTokenTarg((char *)negblob, sz, conn)) {
1341 			conn->auth_mechs |= KSMBD_AUTH_NTLMSSP;
1342 			conn->preferred_auth_mech = KSMBD_AUTH_NTLMSSP;
1343 			conn->use_spnego = false;
1344 		}
1345 	}
1346 	return 0;
1347 }
1348 
1349 static int ntlm_negotiate(struct ksmbd_work *work,
1350 			  struct negotiate_message *negblob,
1351 			  size_t negblob_len, struct smb2_sess_setup_rsp *rsp)
1352 {
1353 	struct challenge_message *chgblob;
1354 	unsigned char *spnego_blob = NULL;
1355 	u16 spnego_blob_len;
1356 	char *neg_blob;
1357 	int sz, rc;
1358 
1359 	ksmbd_debug(SMB, "negotiate phase\n");
1360 	rc = ksmbd_decode_ntlmssp_neg_blob(negblob, negblob_len, work->conn);
1361 	if (rc)
1362 		return rc;
1363 
1364 	sz = le16_to_cpu(rsp->SecurityBufferOffset);
1365 	chgblob = (struct challenge_message *)rsp->Buffer;
1366 	memset(chgblob, 0, sizeof(struct challenge_message));
1367 
1368 	if (!work->conn->use_spnego) {
1369 		sz = ksmbd_build_ntlmssp_challenge_blob(chgblob, work->conn);
1370 		if (sz < 0)
1371 			return -ENOMEM;
1372 
1373 		rsp->SecurityBufferLength = cpu_to_le16(sz);
1374 		return 0;
1375 	}
1376 
1377 	sz = sizeof(struct challenge_message);
1378 	sz += (strlen(ksmbd_netbios_name()) * 2 + 1 + 4) * 6;
1379 
1380 	neg_blob = kzalloc(sz, KSMBD_DEFAULT_GFP);
1381 	if (!neg_blob)
1382 		return -ENOMEM;
1383 
1384 	chgblob = (struct challenge_message *)neg_blob;
1385 	sz = ksmbd_build_ntlmssp_challenge_blob(chgblob, work->conn);
1386 	if (sz < 0) {
1387 		rc = -ENOMEM;
1388 		goto out;
1389 	}
1390 
1391 	rc = build_spnego_ntlmssp_neg_blob(&spnego_blob, &spnego_blob_len,
1392 					   neg_blob, sz);
1393 	if (rc) {
1394 		rc = -ENOMEM;
1395 		goto out;
1396 	}
1397 
1398 	memcpy(rsp->Buffer, spnego_blob, spnego_blob_len);
1399 	rsp->SecurityBufferLength = cpu_to_le16(spnego_blob_len);
1400 
1401 out:
1402 	kfree(spnego_blob);
1403 	kfree(neg_blob);
1404 	return rc;
1405 }
1406 
1407 static struct authenticate_message *user_authblob(struct ksmbd_conn *conn,
1408 						  struct smb2_sess_setup_req *req)
1409 {
1410 	int sz;
1411 
1412 	if (conn->use_spnego && conn->mechToken)
1413 		return (struct authenticate_message *)conn->mechToken;
1414 
1415 	sz = le16_to_cpu(req->SecurityBufferOffset);
1416 	return (struct authenticate_message *)((char *)&req->hdr.ProtocolId
1417 					       + sz);
1418 }
1419 
1420 static struct ksmbd_user *session_user(struct ksmbd_conn *conn,
1421 				       struct smb2_sess_setup_req *req)
1422 {
1423 	struct authenticate_message *authblob;
1424 	struct ksmbd_user *user;
1425 	char *name;
1426 	unsigned int name_off, name_len, secbuf_len;
1427 
1428 	if (conn->use_spnego && conn->mechToken)
1429 		secbuf_len = conn->mechTokenLen;
1430 	else
1431 		secbuf_len = le16_to_cpu(req->SecurityBufferLength);
1432 	if (secbuf_len < sizeof(struct authenticate_message)) {
1433 		ksmbd_debug(SMB, "blob len %d too small\n", secbuf_len);
1434 		return NULL;
1435 	}
1436 	authblob = user_authblob(conn, req);
1437 	name_off = le32_to_cpu(authblob->UserName.BufferOffset);
1438 	name_len = le16_to_cpu(authblob->UserName.Length);
1439 
1440 	if (secbuf_len < (u64)name_off + name_len)
1441 		return NULL;
1442 
1443 	name = smb_strndup_from_utf16((const char *)authblob + name_off,
1444 				      name_len,
1445 				      true,
1446 				      conn->local_nls);
1447 	if (IS_ERR(name)) {
1448 		pr_err("cannot allocate memory\n");
1449 		return NULL;
1450 	}
1451 
1452 	ksmbd_debug(SMB, "session setup request for user %s\n", name);
1453 	user = ksmbd_login_user(name);
1454 	kfree(name);
1455 	return user;
1456 }
1457 
1458 static int ntlm_authenticate(struct ksmbd_work *work,
1459 			     struct smb2_sess_setup_req *req,
1460 			     struct smb2_sess_setup_rsp *rsp)
1461 {
1462 	struct ksmbd_conn *conn = work->conn;
1463 	struct ksmbd_session *sess = work->sess;
1464 	struct channel *chann = NULL, *old;
1465 	struct ksmbd_user *user;
1466 	u64 prev_id;
1467 	int sz, rc;
1468 
1469 	ksmbd_debug(SMB, "authenticate phase\n");
1470 	if (conn->use_spnego) {
1471 		unsigned char *spnego_blob;
1472 		u16 spnego_blob_len;
1473 
1474 		rc = build_spnego_ntlmssp_auth_blob(&spnego_blob,
1475 						    &spnego_blob_len,
1476 						    0);
1477 		if (rc)
1478 			return -ENOMEM;
1479 
1480 		memcpy(rsp->Buffer, spnego_blob, spnego_blob_len);
1481 		rsp->SecurityBufferLength = cpu_to_le16(spnego_blob_len);
1482 		kfree(spnego_blob);
1483 	}
1484 
1485 	user = session_user(conn, req);
1486 	if (!user) {
1487 		ksmbd_debug(SMB, "Unknown user name or an error\n");
1488 		return -EPERM;
1489 	}
1490 
1491 	/* Check for previous session */
1492 	prev_id = le64_to_cpu(req->PreviousSessionId);
1493 	if (prev_id && prev_id != sess->id)
1494 		destroy_previous_session(conn, user, prev_id);
1495 
1496 	if (sess->state == SMB2_SESSION_VALID) {
1497 		/*
1498 		 * Reuse session if anonymous try to connect
1499 		 * on reauthetication.
1500 		 */
1501 		if (conn->binding == false && ksmbd_anonymous_user(user)) {
1502 			ksmbd_free_user(user);
1503 			return 0;
1504 		}
1505 
1506 		if (!ksmbd_compare_user(sess->user, user)) {
1507 			ksmbd_free_user(user);
1508 			return -EPERM;
1509 		}
1510 		ksmbd_free_user(user);
1511 	} else {
1512 		sess->user = user;
1513 	}
1514 
1515 	if (conn->binding == false && user_guest(sess->user)) {
1516 		rsp->SessionFlags = SMB2_SESSION_FLAG_IS_GUEST_LE;
1517 	} else {
1518 		struct authenticate_message *authblob;
1519 
1520 		authblob = user_authblob(conn, req);
1521 		if (conn->use_spnego && conn->mechToken)
1522 			sz = conn->mechTokenLen;
1523 		else
1524 			sz = le16_to_cpu(req->SecurityBufferLength);
1525 		rc = ksmbd_decode_ntlmssp_auth_blob(authblob, sz, conn, sess);
1526 		if (rc) {
1527 			set_user_flag(sess->user, KSMBD_USER_FLAG_BAD_PASSWORD);
1528 			ksmbd_debug(SMB, "authentication failed\n");
1529 			return -EPERM;
1530 		}
1531 	}
1532 
1533 	/*
1534 	 * If session state is SMB2_SESSION_VALID, We can assume
1535 	 * that it is reauthentication. And the user/password
1536 	 * has been verified, so return it here.
1537 	 */
1538 	if (sess->state == SMB2_SESSION_VALID) {
1539 		if (conn->binding)
1540 			goto binding_session;
1541 		return 0;
1542 	}
1543 
1544 	if ((rsp->SessionFlags != SMB2_SESSION_FLAG_IS_GUEST_LE &&
1545 	     (conn->sign || server_conf.enforced_signing)) ||
1546 	    (req->SecurityMode & SMB2_NEGOTIATE_SIGNING_REQUIRED))
1547 		sess->sign = true;
1548 
1549 	if (smb3_encryption_negotiated(conn) &&
1550 			!(req->Flags & SMB2_SESSION_REQ_FLAG_BINDING)) {
1551 		conn->ops->generate_encryptionkey(conn, sess);
1552 		sess->enc = true;
1553 		if (server_conf.flags & KSMBD_GLOBAL_FLAG_SMB2_ENCRYPTION)
1554 			rsp->SessionFlags = SMB2_SESSION_FLAG_ENCRYPT_DATA_LE;
1555 		/*
1556 		 * signing is disable if encryption is enable
1557 		 * on this session
1558 		 */
1559 		sess->sign = false;
1560 	}
1561 
1562 binding_session:
1563 	if (conn->dialect >= SMB30_PROT_ID) {
1564 		chann = lookup_chann_list(sess, conn);
1565 		if (!chann) {
1566 			chann = kmalloc_obj(struct channel, KSMBD_DEFAULT_GFP);
1567 			if (!chann)
1568 				return -ENOMEM;
1569 
1570 			chann->conn = conn;
1571 			down_write(&sess->chann_lock);
1572 			old = xa_store(&sess->ksmbd_chann_list, (long)conn, chann,
1573 					KSMBD_DEFAULT_GFP);
1574 			up_write(&sess->chann_lock);
1575 			if (xa_is_err(old)) {
1576 				kfree(chann);
1577 				return xa_err(old);
1578 			}
1579 		}
1580 	}
1581 
1582 	if (conn->ops->generate_signingkey) {
1583 		rc = conn->ops->generate_signingkey(sess, conn);
1584 		if (rc) {
1585 			ksmbd_debug(SMB, "SMB3 signing key generation failed\n");
1586 			return -EINVAL;
1587 		}
1588 	}
1589 
1590 	if (!ksmbd_conn_lookup_dialect(conn)) {
1591 		pr_err("fail to verify the dialect\n");
1592 		return -ENOENT;
1593 	}
1594 	return 0;
1595 }
1596 
1597 #ifdef CONFIG_SMB_SERVER_KERBEROS5
1598 static int krb5_authenticate(struct ksmbd_work *work,
1599 			     struct smb2_sess_setup_req *req,
1600 			     struct smb2_sess_setup_rsp *rsp)
1601 {
1602 	struct ksmbd_conn *conn = work->conn;
1603 	struct ksmbd_session *sess = work->sess;
1604 	char *in_blob, *out_blob;
1605 	struct channel *chann = NULL, *old;
1606 	u64 prev_sess_id;
1607 	int in_len, out_len;
1608 	int retval;
1609 
1610 	in_blob = (char *)&req->hdr.ProtocolId +
1611 		le16_to_cpu(req->SecurityBufferOffset);
1612 	in_len = le16_to_cpu(req->SecurityBufferLength);
1613 	out_blob = (char *)&rsp->hdr.ProtocolId +
1614 		le16_to_cpu(rsp->SecurityBufferOffset);
1615 	out_len = work->response_sz -
1616 		(le16_to_cpu(rsp->SecurityBufferOffset) + 4);
1617 
1618 	retval = ksmbd_krb5_authenticate(sess, in_blob, in_len,
1619 					 out_blob, &out_len);
1620 	if (retval) {
1621 		ksmbd_debug(SMB, "krb5 authentication failed\n");
1622 		return -EINVAL;
1623 	}
1624 
1625 	/* Check previous session */
1626 	prev_sess_id = le64_to_cpu(req->PreviousSessionId);
1627 	if (prev_sess_id && prev_sess_id != sess->id)
1628 		destroy_previous_session(conn, sess->user, prev_sess_id);
1629 
1630 	rsp->SecurityBufferLength = cpu_to_le16(out_len);
1631 
1632 	/*
1633 	 * If session state is SMB2_SESSION_VALID, We can assume
1634 	 * that it is reauthentication. And the user/password
1635 	 * has been verified, so return it here.
1636 	 */
1637 	if (sess->state == SMB2_SESSION_VALID) {
1638 		if (conn->binding)
1639 			goto binding_session;
1640 		return 0;
1641 	}
1642 
1643 	if ((rsp->SessionFlags != SMB2_SESSION_FLAG_IS_GUEST_LE &&
1644 	    (conn->sign || server_conf.enforced_signing)) ||
1645 	    (req->SecurityMode & SMB2_NEGOTIATE_SIGNING_REQUIRED))
1646 		sess->sign = true;
1647 
1648 	if (smb3_encryption_negotiated(conn) &&
1649 	    !(req->Flags & SMB2_SESSION_REQ_FLAG_BINDING)) {
1650 		conn->ops->generate_encryptionkey(conn, sess);
1651 		sess->enc = true;
1652 		if (server_conf.flags & KSMBD_GLOBAL_FLAG_SMB2_ENCRYPTION)
1653 			rsp->SessionFlags = SMB2_SESSION_FLAG_ENCRYPT_DATA_LE;
1654 		sess->sign = false;
1655 	}
1656 
1657 binding_session:
1658 	if (conn->dialect >= SMB30_PROT_ID) {
1659 		chann = lookup_chann_list(sess, conn);
1660 		if (!chann) {
1661 			chann = kmalloc_obj(struct channel, KSMBD_DEFAULT_GFP);
1662 			if (!chann)
1663 				return -ENOMEM;
1664 
1665 			chann->conn = conn;
1666 			down_write(&sess->chann_lock);
1667 			old = xa_store(&sess->ksmbd_chann_list, (long)conn,
1668 					chann, KSMBD_DEFAULT_GFP);
1669 			up_write(&sess->chann_lock);
1670 			if (xa_is_err(old)) {
1671 				kfree(chann);
1672 				return xa_err(old);
1673 			}
1674 		}
1675 	}
1676 
1677 	if (conn->ops->generate_signingkey) {
1678 		retval = conn->ops->generate_signingkey(sess, conn);
1679 		if (retval) {
1680 			ksmbd_debug(SMB, "SMB3 signing key generation failed\n");
1681 			return -EINVAL;
1682 		}
1683 	}
1684 
1685 	if (!ksmbd_conn_lookup_dialect(conn)) {
1686 		pr_err("fail to verify the dialect\n");
1687 		return -ENOENT;
1688 	}
1689 	return 0;
1690 }
1691 #else
1692 static int krb5_authenticate(struct ksmbd_work *work,
1693 			     struct smb2_sess_setup_req *req,
1694 			     struct smb2_sess_setup_rsp *rsp)
1695 {
1696 	return -EOPNOTSUPP;
1697 }
1698 #endif
1699 
1700 int smb2_sess_setup(struct ksmbd_work *work)
1701 {
1702 	struct ksmbd_conn *conn = work->conn;
1703 	struct smb2_sess_setup_req *req;
1704 	struct smb2_sess_setup_rsp *rsp;
1705 	struct ksmbd_session *sess;
1706 	struct negotiate_message *negblob;
1707 	unsigned int negblob_len, negblob_off;
1708 	int rc = 0;
1709 
1710 	ksmbd_debug(SMB, "Received smb2 session setup request\n");
1711 
1712 	if (!ksmbd_conn_need_setup(conn) && !ksmbd_conn_good(conn)) {
1713 		work->send_no_response = 1;
1714 		return rc;
1715 	}
1716 
1717 	WORK_BUFFERS(work, req, rsp);
1718 
1719 	rsp->StructureSize = cpu_to_le16(9);
1720 	rsp->SessionFlags = 0;
1721 	rsp->SecurityBufferOffset = cpu_to_le16(72);
1722 	rsp->SecurityBufferLength = 0;
1723 
1724 	ksmbd_conn_lock(conn);
1725 	if (!req->hdr.SessionId) {
1726 		sess = ksmbd_smb2_session_create();
1727 		if (!sess) {
1728 			rc = -ENOMEM;
1729 			goto out_err;
1730 		}
1731 		rsp->hdr.SessionId = cpu_to_le64(sess->id);
1732 		rc = ksmbd_session_register(conn, sess);
1733 		if (rc)
1734 			goto out_err;
1735 
1736 		conn->binding = false;
1737 	} else if (conn->dialect >= SMB30_PROT_ID &&
1738 		   (server_conf.flags & KSMBD_GLOBAL_FLAG_SMB3_MULTICHANNEL) &&
1739 		   req->Flags & SMB2_SESSION_REQ_FLAG_BINDING) {
1740 		u64 sess_id = le64_to_cpu(req->hdr.SessionId);
1741 
1742 		sess = ksmbd_session_lookup_slowpath(sess_id);
1743 		if (!sess) {
1744 			rc = -ENOENT;
1745 			goto out_err;
1746 		}
1747 
1748 		if (conn->dialect != sess->dialect) {
1749 			rc = -EINVAL;
1750 			goto out_err;
1751 		}
1752 
1753 		if (!(req->hdr.Flags & SMB2_FLAGS_SIGNED)) {
1754 			rc = -EINVAL;
1755 			goto out_err;
1756 		}
1757 
1758 		if (strncmp(conn->ClientGUID, sess->ClientGUID,
1759 			    SMB2_CLIENT_GUID_SIZE)) {
1760 			rc = -ENOENT;
1761 			goto out_err;
1762 		}
1763 
1764 		if (sess->state == SMB2_SESSION_IN_PROGRESS) {
1765 			rc = -EACCES;
1766 			goto out_err;
1767 		}
1768 
1769 		if (sess->state == SMB2_SESSION_EXPIRED) {
1770 			rc = -EFAULT;
1771 			goto out_err;
1772 		}
1773 
1774 		if (ksmbd_conn_need_reconnect(conn)) {
1775 			rc = -EFAULT;
1776 			ksmbd_user_session_put(sess);
1777 			sess = NULL;
1778 			goto out_err;
1779 		}
1780 
1781 		if (is_ksmbd_session_in_connection(conn, sess_id)) {
1782 			rc = -EACCES;
1783 			goto out_err;
1784 		}
1785 
1786 		if (user_guest(sess->user)) {
1787 			rc = -EOPNOTSUPP;
1788 			goto out_err;
1789 		}
1790 
1791 		conn->binding = true;
1792 	} else if ((conn->dialect < SMB30_PROT_ID ||
1793 		    server_conf.flags & KSMBD_GLOBAL_FLAG_SMB3_MULTICHANNEL) &&
1794 		   (req->Flags & SMB2_SESSION_REQ_FLAG_BINDING)) {
1795 		sess = NULL;
1796 		rc = -EACCES;
1797 		goto out_err;
1798 	} else {
1799 		sess = ksmbd_session_lookup(conn,
1800 					    le64_to_cpu(req->hdr.SessionId));
1801 		if (!sess) {
1802 			rc = -ENOENT;
1803 			goto out_err;
1804 		}
1805 
1806 		if (sess->state == SMB2_SESSION_EXPIRED) {
1807 			rc = -EFAULT;
1808 			goto out_err;
1809 		}
1810 
1811 		if (ksmbd_conn_need_reconnect(conn)) {
1812 			rc = -EFAULT;
1813 			ksmbd_user_session_put(sess);
1814 			sess = NULL;
1815 			goto out_err;
1816 		}
1817 
1818 		conn->binding = false;
1819 	}
1820 	work->sess = sess;
1821 
1822 	negblob_off = le16_to_cpu(req->SecurityBufferOffset);
1823 	negblob_len = le16_to_cpu(req->SecurityBufferLength);
1824 	if (negblob_off < offsetof(struct smb2_sess_setup_req, Buffer)) {
1825 		rc = -EINVAL;
1826 		goto out_err;
1827 	}
1828 
1829 	negblob = (struct negotiate_message *)((char *)&req->hdr.ProtocolId +
1830 			negblob_off);
1831 
1832 	if (decode_negotiation_token(conn, negblob, negblob_len) == 0) {
1833 		if (conn->mechToken) {
1834 			negblob = (struct negotiate_message *)conn->mechToken;
1835 			negblob_len = conn->mechTokenLen;
1836 		}
1837 	}
1838 
1839 	if (negblob_len < offsetof(struct negotiate_message, NegotiateFlags)) {
1840 		rc = -EINVAL;
1841 		goto out_err;
1842 	}
1843 
1844 	if (server_conf.auth_mechs & conn->auth_mechs) {
1845 		rc = generate_preauth_hash(work);
1846 		if (rc)
1847 			goto out_err;
1848 
1849 		if (conn->preferred_auth_mech &
1850 				(KSMBD_AUTH_KRB5 | KSMBD_AUTH_MSKRB5)) {
1851 			rc = krb5_authenticate(work, req, rsp);
1852 			if (rc) {
1853 				rc = -EINVAL;
1854 				goto out_err;
1855 			}
1856 
1857 			if (!ksmbd_conn_need_reconnect(conn)) {
1858 				ksmbd_conn_set_good(conn);
1859 				sess->state = SMB2_SESSION_VALID;
1860 			}
1861 		} else if (conn->preferred_auth_mech == KSMBD_AUTH_NTLMSSP) {
1862 			if (negblob->MessageType == NtLmNegotiate) {
1863 				rc = ntlm_negotiate(work, negblob, negblob_len, rsp);
1864 				if (rc)
1865 					goto out_err;
1866 				rsp->hdr.Status =
1867 					STATUS_MORE_PROCESSING_REQUIRED;
1868 			} else if (negblob->MessageType == NtLmAuthenticate) {
1869 				rc = ntlm_authenticate(work, req, rsp);
1870 				if (rc)
1871 					goto out_err;
1872 
1873 				if (!ksmbd_conn_need_reconnect(conn)) {
1874 					ksmbd_conn_set_good(conn);
1875 					sess->state = SMB2_SESSION_VALID;
1876 				}
1877 				if (conn->binding) {
1878 					struct preauth_session *preauth_sess;
1879 
1880 					preauth_sess =
1881 						ksmbd_preauth_session_lookup(conn, sess->id);
1882 					if (preauth_sess) {
1883 						list_del(&preauth_sess->preauth_entry);
1884 						kfree(preauth_sess);
1885 					}
1886 				}
1887 			} else {
1888 				pr_info_ratelimited("Unknown NTLMSSP message type : 0x%x\n",
1889 						le32_to_cpu(negblob->MessageType));
1890 				rc = -EINVAL;
1891 			}
1892 		} else {
1893 			/* TODO: need one more negotiation */
1894 			pr_err("Not support the preferred authentication\n");
1895 			rc = -EINVAL;
1896 		}
1897 	} else {
1898 		pr_err("Not support authentication\n");
1899 		rc = -EINVAL;
1900 	}
1901 
1902 out_err:
1903 	if (rc == -EINVAL)
1904 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
1905 	else if (rc == -ENOENT)
1906 		rsp->hdr.Status = STATUS_USER_SESSION_DELETED;
1907 	else if (rc == -EACCES)
1908 		rsp->hdr.Status = STATUS_REQUEST_NOT_ACCEPTED;
1909 	else if (rc == -EFAULT)
1910 		rsp->hdr.Status = STATUS_NETWORK_SESSION_EXPIRED;
1911 	else if (rc == -ENOMEM)
1912 		rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
1913 	else if (rc == -EOPNOTSUPP)
1914 		rsp->hdr.Status = STATUS_NOT_SUPPORTED;
1915 	else if (rc)
1916 		rsp->hdr.Status = STATUS_LOGON_FAILURE;
1917 
1918 	if (conn->mechToken) {
1919 		kfree(conn->mechToken);
1920 		conn->mechToken = NULL;
1921 	}
1922 
1923 	if (rc < 0) {
1924 		/*
1925 		 * SecurityBufferOffset should be set to zero
1926 		 * in session setup error response.
1927 		 */
1928 		rsp->SecurityBufferOffset = 0;
1929 
1930 		if (sess) {
1931 			bool try_delay = false;
1932 
1933 			/*
1934 			 * To avoid dictionary attacks (repeated session setups rapidly sent) to
1935 			 * connect to server, ksmbd make a delay of a 5 seconds on session setup
1936 			 * failure to make it harder to send enough random connection requests
1937 			 * to break into a server.
1938 			 */
1939 			if (sess->user && sess->user->flags & KSMBD_USER_FLAG_DELAY_SESSION)
1940 				try_delay = true;
1941 
1942 			/*
1943 			 * For binding requests, session belongs to another
1944 			 * connection. Do not expire it.
1945 			 */
1946 			if (!(req->Flags & SMB2_SESSION_REQ_FLAG_BINDING)) {
1947 				sess->last_active = jiffies;
1948 				sess->state = SMB2_SESSION_EXPIRED;
1949 			}
1950 			ksmbd_user_session_put(sess);
1951 			work->sess = NULL;
1952 			if (try_delay) {
1953 				ksmbd_conn_set_need_reconnect(conn);
1954 				ssleep(5);
1955 				ksmbd_conn_set_need_setup(conn);
1956 			}
1957 		}
1958 		smb2_set_err_rsp(work);
1959 		conn->binding = false;
1960 	} else {
1961 		unsigned int iov_len;
1962 
1963 		if (rsp->SecurityBufferLength)
1964 			iov_len = offsetof(struct smb2_sess_setup_rsp, Buffer) +
1965 				le16_to_cpu(rsp->SecurityBufferLength);
1966 		else
1967 			iov_len = sizeof(struct smb2_sess_setup_rsp);
1968 		rc = ksmbd_iov_pin_rsp(work, rsp, iov_len);
1969 		if (rc)
1970 			rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
1971 	}
1972 
1973 	ksmbd_conn_unlock(conn);
1974 	return rc;
1975 }
1976 
1977 /**
1978  * smb2_tree_connect() - handler for smb2 tree connect command
1979  * @work:	smb work containing smb request buffer
1980  *
1981  * Return:      0 on success, otherwise error
1982  */
1983 int smb2_tree_connect(struct ksmbd_work *work)
1984 {
1985 	struct ksmbd_conn *conn = work->conn;
1986 	struct smb2_tree_connect_req *req;
1987 	struct smb2_tree_connect_rsp *rsp;
1988 	struct ksmbd_session *sess = work->sess;
1989 	char *treename = NULL, *name = NULL;
1990 	struct ksmbd_tree_conn_status status;
1991 	struct ksmbd_share_config *share = NULL;
1992 	int rc = -EINVAL;
1993 
1994 	ksmbd_debug(SMB, "Received smb2 tree connect request\n");
1995 
1996 	WORK_BUFFERS(work, req, rsp);
1997 
1998 	treename = smb_strndup_from_utf16((char *)req + le16_to_cpu(req->PathOffset),
1999 					  le16_to_cpu(req->PathLength), true,
2000 					  conn->local_nls);
2001 	if (IS_ERR(treename)) {
2002 		pr_err("treename is NULL\n");
2003 		status.ret = KSMBD_TREE_CONN_STATUS_ERROR;
2004 		goto out_err1;
2005 	}
2006 
2007 	name = ksmbd_extract_sharename(conn->um, treename);
2008 	if (IS_ERR(name)) {
2009 		status.ret = KSMBD_TREE_CONN_STATUS_ERROR;
2010 		goto out_err1;
2011 	}
2012 
2013 	ksmbd_debug(SMB, "tree connect request for tree %s treename %s\n",
2014 		    name, treename);
2015 
2016 	status = ksmbd_tree_conn_connect(work, name);
2017 	if (status.ret == KSMBD_TREE_CONN_STATUS_OK)
2018 		rsp->hdr.Id.SyncId.TreeId = cpu_to_le32(status.tree_conn->id);
2019 	else
2020 		goto out_err1;
2021 
2022 	share = status.tree_conn->share_conf;
2023 	if (test_share_config_flag(share, KSMBD_SHARE_FLAG_PIPE)) {
2024 		ksmbd_debug(SMB, "IPC share path request\n");
2025 		rsp->ShareType = SMB2_SHARE_TYPE_PIPE;
2026 		rsp->MaximalAccess = FILE_READ_DATA_LE | FILE_READ_EA_LE |
2027 			FILE_EXECUTE_LE | FILE_READ_ATTRIBUTES_LE |
2028 			FILE_DELETE_LE | FILE_READ_CONTROL_LE |
2029 			FILE_WRITE_DAC_LE | FILE_WRITE_OWNER_LE |
2030 			FILE_SYNCHRONIZE_LE;
2031 	} else {
2032 		rsp->ShareType = SMB2_SHARE_TYPE_DISK;
2033 		rsp->MaximalAccess = FILE_READ_DATA_LE | FILE_READ_EA_LE |
2034 			FILE_EXECUTE_LE | FILE_READ_ATTRIBUTES_LE;
2035 		if (test_tree_conn_flag(status.tree_conn,
2036 					KSMBD_TREE_CONN_FLAG_WRITABLE)) {
2037 			rsp->MaximalAccess |= FILE_WRITE_DATA_LE |
2038 				FILE_APPEND_DATA_LE | FILE_WRITE_EA_LE |
2039 				FILE_DELETE_LE | FILE_WRITE_ATTRIBUTES_LE |
2040 				FILE_DELETE_CHILD_LE | FILE_READ_CONTROL_LE |
2041 				FILE_WRITE_DAC_LE | FILE_WRITE_OWNER_LE |
2042 				FILE_SYNCHRONIZE_LE;
2043 		}
2044 	}
2045 
2046 	status.tree_conn->maximal_access = le32_to_cpu(rsp->MaximalAccess);
2047 	if (conn->posix_ext_supported)
2048 		status.tree_conn->posix_extensions = true;
2049 
2050 	down_write(&sess->tree_conns_lock);
2051 	status.tree_conn->t_state = TREE_CONNECTED;
2052 	up_write(&sess->tree_conns_lock);
2053 	rsp->StructureSize = cpu_to_le16(16);
2054 out_err1:
2055 	if (server_conf.flags & KSMBD_GLOBAL_FLAG_DURABLE_HANDLE && share &&
2056 	    test_share_config_flag(share,
2057 				   KSMBD_SHARE_FLAG_CONTINUOUS_AVAILABILITY))
2058 		rsp->Capabilities = SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY;
2059 	else
2060 		rsp->Capabilities = 0;
2061 	rsp->Reserved = 0;
2062 	/* default manual caching */
2063 	rsp->ShareFlags = SMB2_SHAREFLAG_MANUAL_CACHING;
2064 
2065 	rc = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_tree_connect_rsp));
2066 	if (rc)
2067 		status.ret = KSMBD_TREE_CONN_STATUS_NOMEM;
2068 
2069 	if (!IS_ERR(treename))
2070 		kfree(treename);
2071 	if (!IS_ERR(name))
2072 		kfree(name);
2073 
2074 	switch (status.ret) {
2075 	case KSMBD_TREE_CONN_STATUS_OK:
2076 		rsp->hdr.Status = STATUS_SUCCESS;
2077 		rc = 0;
2078 		break;
2079 	case -ESTALE:
2080 	case -ENOENT:
2081 	case KSMBD_TREE_CONN_STATUS_NO_SHARE:
2082 		rsp->hdr.Status = STATUS_BAD_NETWORK_NAME;
2083 		break;
2084 	case -ENOMEM:
2085 	case KSMBD_TREE_CONN_STATUS_NOMEM:
2086 		rsp->hdr.Status = STATUS_NO_MEMORY;
2087 		break;
2088 	case KSMBD_TREE_CONN_STATUS_ERROR:
2089 	case KSMBD_TREE_CONN_STATUS_TOO_MANY_CONNS:
2090 	case KSMBD_TREE_CONN_STATUS_TOO_MANY_SESSIONS:
2091 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
2092 		break;
2093 	case -EINVAL:
2094 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
2095 		break;
2096 	default:
2097 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
2098 	}
2099 
2100 	if (status.ret != KSMBD_TREE_CONN_STATUS_OK)
2101 		smb2_set_err_rsp(work);
2102 
2103 	return rc;
2104 }
2105 
2106 /**
2107  * smb2_create_open_flags() - convert smb open flags to unix open flags
2108  * @file_present:	is file already present
2109  * @access:		file access flags
2110  * @disposition:	file disposition flags
2111  * @may_flags:		set with MAY_ flags
2112  * @coptions:		file creation options
2113  * @mode:		file mode
2114  *
2115  * Return:      file open flags
2116  */
2117 static int smb2_create_open_flags(bool file_present, __le32 access,
2118 				  __le32 disposition,
2119 				  int *may_flags,
2120 				  __le32 coptions,
2121 				  umode_t mode)
2122 {
2123 	int oflags = O_NONBLOCK | O_LARGEFILE;
2124 
2125 	if (coptions & FILE_DIRECTORY_FILE_LE || S_ISDIR(mode)) {
2126 		access &= ~FILE_WRITE_DESIRE_ACCESS_LE;
2127 		ksmbd_debug(SMB, "Discard write access to a directory\n");
2128 	}
2129 
2130 	if (access & FILE_READ_DESIRED_ACCESS_LE &&
2131 	    access & FILE_WRITE_DESIRE_ACCESS_LE) {
2132 		oflags |= O_RDWR;
2133 		*may_flags = MAY_OPEN | MAY_READ | MAY_WRITE;
2134 	} else if (access & FILE_WRITE_DESIRE_ACCESS_LE) {
2135 		oflags |= O_WRONLY;
2136 		*may_flags = MAY_OPEN | MAY_WRITE;
2137 	} else {
2138 		oflags |= O_RDONLY;
2139 		*may_flags = MAY_OPEN | MAY_READ;
2140 	}
2141 
2142 	if (access == FILE_READ_ATTRIBUTES_LE || S_ISBLK(mode) || S_ISCHR(mode))
2143 		oflags |= O_PATH;
2144 
2145 	if (file_present) {
2146 		switch (disposition & FILE_CREATE_MASK_LE) {
2147 		case FILE_OPEN_LE:
2148 		case FILE_CREATE_LE:
2149 			break;
2150 		case FILE_SUPERSEDE_LE:
2151 		case FILE_OVERWRITE_LE:
2152 		case FILE_OVERWRITE_IF_LE:
2153 			oflags |= O_TRUNC;
2154 			break;
2155 		default:
2156 			break;
2157 		}
2158 	} else {
2159 		switch (disposition & FILE_CREATE_MASK_LE) {
2160 		case FILE_SUPERSEDE_LE:
2161 		case FILE_CREATE_LE:
2162 		case FILE_OPEN_IF_LE:
2163 		case FILE_OVERWRITE_IF_LE:
2164 			oflags |= O_CREAT;
2165 			break;
2166 		case FILE_OPEN_LE:
2167 		case FILE_OVERWRITE_LE:
2168 			oflags &= ~O_CREAT;
2169 			break;
2170 		default:
2171 			break;
2172 		}
2173 	}
2174 
2175 	return oflags;
2176 }
2177 
2178 /**
2179  * smb2_tree_disconnect() - handler for smb tree connect request
2180  * @work:	smb work containing request buffer
2181  *
2182  * Return:      0 on success, otherwise error
2183  */
2184 int smb2_tree_disconnect(struct ksmbd_work *work)
2185 {
2186 	struct smb2_tree_disconnect_rsp *rsp;
2187 	struct smb2_tree_disconnect_req *req;
2188 	struct ksmbd_session *sess = work->sess;
2189 	struct ksmbd_tree_connect *tcon = work->tcon;
2190 	int err;
2191 
2192 	ksmbd_debug(SMB, "Received smb2 tree disconnect request\n");
2193 
2194 	WORK_BUFFERS(work, req, rsp);
2195 
2196 	if (!tcon) {
2197 		ksmbd_debug(SMB, "Invalid tid %d\n", req->hdr.Id.SyncId.TreeId);
2198 
2199 		rsp->hdr.Status = STATUS_NETWORK_NAME_DELETED;
2200 		err = -ENOENT;
2201 		goto err_out;
2202 	}
2203 
2204 	ksmbd_close_tree_conn_fds(work);
2205 
2206 	down_write(&sess->tree_conns_lock);
2207 	if (tcon->t_state == TREE_DISCONNECTED) {
2208 		up_write(&sess->tree_conns_lock);
2209 		rsp->hdr.Status = STATUS_NETWORK_NAME_DELETED;
2210 		err = -ENOENT;
2211 		goto err_out;
2212 	}
2213 
2214 	tcon->t_state = TREE_DISCONNECTED;
2215 	up_write(&sess->tree_conns_lock);
2216 
2217 	err = ksmbd_tree_conn_disconnect(sess, tcon);
2218 	if (err) {
2219 		rsp->hdr.Status = STATUS_NETWORK_NAME_DELETED;
2220 		goto err_out;
2221 	}
2222 
2223 	rsp->StructureSize = cpu_to_le16(4);
2224 	err = ksmbd_iov_pin_rsp(work, rsp,
2225 				sizeof(struct smb2_tree_disconnect_rsp));
2226 	if (err) {
2227 		rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
2228 		goto err_out;
2229 	}
2230 
2231 	return 0;
2232 
2233 err_out:
2234 	smb2_set_err_rsp(work);
2235 	return err;
2236 
2237 }
2238 
2239 /**
2240  * smb2_session_logoff() - handler for session log off request
2241  * @work:	smb work containing request buffer
2242  *
2243  * Return:      0 on success, otherwise error
2244  */
2245 int smb2_session_logoff(struct ksmbd_work *work)
2246 {
2247 	struct ksmbd_conn *conn = work->conn;
2248 	struct ksmbd_session *sess = work->sess;
2249 	struct smb2_logoff_req *req;
2250 	struct smb2_logoff_rsp *rsp;
2251 	u64 sess_id;
2252 	int err;
2253 
2254 	WORK_BUFFERS(work, req, rsp);
2255 
2256 	ksmbd_debug(SMB, "Received smb2 session logoff request\n");
2257 
2258 	ksmbd_conn_lock(conn);
2259 	if (!ksmbd_conn_good(conn)) {
2260 		ksmbd_conn_unlock(conn);
2261 		rsp->hdr.Status = STATUS_NETWORK_NAME_DELETED;
2262 		smb2_set_err_rsp(work);
2263 		return -ENOENT;
2264 	}
2265 	sess_id = le64_to_cpu(req->hdr.SessionId);
2266 	ksmbd_all_conn_set_status(sess_id, KSMBD_SESS_NEED_RECONNECT);
2267 	ksmbd_conn_unlock(conn);
2268 
2269 	ksmbd_close_session_fds(work);
2270 	ksmbd_conn_wait_idle(conn);
2271 
2272 	if (ksmbd_tree_conn_session_logoff(sess)) {
2273 		ksmbd_debug(SMB, "Invalid tid %d\n", req->hdr.Id.SyncId.TreeId);
2274 		rsp->hdr.Status = STATUS_NETWORK_NAME_DELETED;
2275 		smb2_set_err_rsp(work);
2276 		return -ENOENT;
2277 	}
2278 
2279 	down_write(&conn->session_lock);
2280 	sess->state = SMB2_SESSION_EXPIRED;
2281 	up_write(&conn->session_lock);
2282 
2283 	ksmbd_all_conn_set_status(sess_id, KSMBD_SESS_NEED_SETUP);
2284 
2285 	rsp->StructureSize = cpu_to_le16(4);
2286 	err = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_logoff_rsp));
2287 	if (err) {
2288 		rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
2289 		smb2_set_err_rsp(work);
2290 		return err;
2291 	}
2292 	return 0;
2293 }
2294 
2295 /**
2296  * create_smb2_pipe() - create IPC pipe
2297  * @work:	smb work containing request buffer
2298  *
2299  * Return:      0 on success, otherwise error
2300  */
2301 static noinline int create_smb2_pipe(struct ksmbd_work *work)
2302 {
2303 	struct smb2_create_rsp *rsp;
2304 	struct smb2_create_req *req;
2305 	int id = -1;
2306 	int err;
2307 	char *name;
2308 
2309 	WORK_BUFFERS(work, req, rsp);
2310 
2311 	name = smb_strndup_from_utf16(req->Buffer, le16_to_cpu(req->NameLength),
2312 				      1, work->conn->local_nls);
2313 	if (IS_ERR(name)) {
2314 		rsp->hdr.Status = STATUS_NO_MEMORY;
2315 		err = PTR_ERR(name);
2316 		goto out;
2317 	}
2318 
2319 	id = ksmbd_session_rpc_open(work->sess, name);
2320 	if (id < 0) {
2321 		pr_err("Unable to open RPC pipe: %d\n", id);
2322 		err = id;
2323 		goto out;
2324 	}
2325 
2326 	rsp->hdr.Status = STATUS_SUCCESS;
2327 	rsp->StructureSize = cpu_to_le16(89);
2328 	rsp->OplockLevel = SMB2_OPLOCK_LEVEL_NONE;
2329 	rsp->Flags = 0;
2330 	rsp->CreateAction = cpu_to_le32(FILE_OPENED);
2331 
2332 	rsp->CreationTime = cpu_to_le64(0);
2333 	rsp->LastAccessTime = cpu_to_le64(0);
2334 	rsp->ChangeTime = cpu_to_le64(0);
2335 	rsp->AllocationSize = cpu_to_le64(0);
2336 	rsp->EndofFile = cpu_to_le64(0);
2337 	rsp->FileAttributes = FILE_ATTRIBUTE_NORMAL_LE;
2338 	rsp->Reserved2 = 0;
2339 	rsp->VolatileFileId = id;
2340 	rsp->PersistentFileId = 0;
2341 	rsp->CreateContextsOffset = 0;
2342 	rsp->CreateContextsLength = 0;
2343 
2344 	err = ksmbd_iov_pin_rsp(work, rsp, offsetof(struct smb2_create_rsp, Buffer));
2345 	if (err)
2346 		goto out;
2347 
2348 	kfree(name);
2349 	return 0;
2350 
2351 out:
2352 	switch (err) {
2353 	case -EINVAL:
2354 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
2355 		break;
2356 	case -ENOSPC:
2357 	case -ENOMEM:
2358 		rsp->hdr.Status = STATUS_NO_MEMORY;
2359 		break;
2360 	}
2361 
2362 	if (id >= 0)
2363 		ksmbd_session_rpc_close(work->sess, id);
2364 
2365 	if (!IS_ERR(name))
2366 		kfree(name);
2367 
2368 	smb2_set_err_rsp(work);
2369 	return err;
2370 }
2371 
2372 /**
2373  * smb2_set_ea() - handler for setting extended attributes using set
2374  *		info command
2375  * @eabuf:	set info command buffer
2376  * @buf_len:	set info command buffer length
2377  * @path:	dentry path for get ea
2378  * @get_write:	get write access to a mount
2379  *
2380  * Return:	0 on success, otherwise error
2381  */
2382 static int smb2_set_ea(struct smb2_ea_info *eabuf, unsigned int buf_len,
2383 		       const struct path *path, bool get_write)
2384 {
2385 	struct mnt_idmap *idmap = mnt_idmap(path->mnt);
2386 	char *attr_name = NULL, *value;
2387 	int rc = 0;
2388 	unsigned int next = 0;
2389 
2390 	if (buf_len < sizeof(struct smb2_ea_info) + eabuf->EaNameLength + 1 +
2391 			le16_to_cpu(eabuf->EaValueLength))
2392 		return -EINVAL;
2393 
2394 	attr_name = kmalloc(XATTR_NAME_MAX + 1, KSMBD_DEFAULT_GFP);
2395 	if (!attr_name)
2396 		return -ENOMEM;
2397 
2398 	do {
2399 		if (!eabuf->EaNameLength)
2400 			goto next;
2401 
2402 		ksmbd_debug(SMB,
2403 			    "name : <%s>, name_len : %u, value_len : %u, next : %u\n",
2404 			    eabuf->name, eabuf->EaNameLength,
2405 			    le16_to_cpu(eabuf->EaValueLength),
2406 			    le32_to_cpu(eabuf->NextEntryOffset));
2407 
2408 		if (eabuf->EaNameLength >
2409 		    (XATTR_NAME_MAX - XATTR_USER_PREFIX_LEN)) {
2410 			rc = -EINVAL;
2411 			break;
2412 		}
2413 
2414 		memcpy(attr_name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN);
2415 		memcpy(&attr_name[XATTR_USER_PREFIX_LEN], eabuf->name,
2416 		       eabuf->EaNameLength);
2417 		attr_name[XATTR_USER_PREFIX_LEN + eabuf->EaNameLength] = '\0';
2418 		value = (char *)&eabuf->name + eabuf->EaNameLength + 1;
2419 
2420 		if (!eabuf->EaValueLength) {
2421 			rc = ksmbd_vfs_casexattr_len(idmap,
2422 						     path->dentry,
2423 						     attr_name,
2424 						     XATTR_USER_PREFIX_LEN +
2425 						     eabuf->EaNameLength);
2426 
2427 			/* delete the EA only when it exits */
2428 			if (rc > 0) {
2429 				rc = ksmbd_vfs_remove_xattr(idmap,
2430 							    path,
2431 							    attr_name,
2432 							    get_write);
2433 
2434 				if (rc < 0) {
2435 					ksmbd_debug(SMB,
2436 						    "remove xattr failed(%d)\n",
2437 						    rc);
2438 					break;
2439 				}
2440 			}
2441 
2442 			/* if the EA doesn't exist, just do nothing. */
2443 			rc = 0;
2444 		} else {
2445 			rc = ksmbd_vfs_setxattr(idmap, path, attr_name, value,
2446 						le16_to_cpu(eabuf->EaValueLength),
2447 						0, get_write);
2448 			if (rc < 0) {
2449 				ksmbd_debug(SMB,
2450 					    "ksmbd_vfs_setxattr is failed(%d)\n",
2451 					    rc);
2452 				break;
2453 			}
2454 		}
2455 
2456 next:
2457 		next = le32_to_cpu(eabuf->NextEntryOffset);
2458 		if (next == 0 || buf_len < next)
2459 			break;
2460 		buf_len -= next;
2461 		eabuf = (struct smb2_ea_info *)((char *)eabuf + next);
2462 		if (buf_len < sizeof(struct smb2_ea_info)) {
2463 			rc = -EINVAL;
2464 			break;
2465 		}
2466 
2467 		if (buf_len < sizeof(struct smb2_ea_info) + eabuf->EaNameLength + 1 +
2468 				le16_to_cpu(eabuf->EaValueLength)) {
2469 			rc = -EINVAL;
2470 			break;
2471 		}
2472 	} while (next != 0);
2473 
2474 	kfree(attr_name);
2475 	return rc;
2476 }
2477 
2478 static noinline int smb2_set_stream_name_xattr(const struct path *path,
2479 					       struct ksmbd_file *fp,
2480 					       char *stream_name, int s_type)
2481 {
2482 	struct mnt_idmap *idmap = mnt_idmap(path->mnt);
2483 	size_t xattr_stream_size;
2484 	char *xattr_stream_name;
2485 	int rc;
2486 
2487 	rc = ksmbd_vfs_xattr_stream_name(stream_name,
2488 					 &xattr_stream_name,
2489 					 &xattr_stream_size,
2490 					 s_type);
2491 	if (rc)
2492 		return rc;
2493 
2494 	fp->stream.name = xattr_stream_name;
2495 	fp->stream.size = xattr_stream_size;
2496 
2497 	/* Check if there is stream prefix in xattr space */
2498 	rc = ksmbd_vfs_casexattr_len(idmap,
2499 				     path->dentry,
2500 				     xattr_stream_name,
2501 				     xattr_stream_size);
2502 	if (rc >= 0)
2503 		return 0;
2504 
2505 	if (fp->cdoption == FILE_OPEN_LE) {
2506 		ksmbd_debug(SMB, "XATTR stream name lookup failed: %d\n", rc);
2507 		return -EBADF;
2508 	}
2509 
2510 	rc = ksmbd_vfs_setxattr(idmap, path, xattr_stream_name, NULL, 0, 0, false);
2511 	if (rc < 0)
2512 		pr_err("Failed to store XATTR stream name :%d\n", rc);
2513 	return 0;
2514 }
2515 
2516 static int smb2_remove_smb_xattrs(const struct path *path)
2517 {
2518 	struct mnt_idmap *idmap = mnt_idmap(path->mnt);
2519 	char *name, *xattr_list = NULL;
2520 	ssize_t xattr_list_len;
2521 	int err = 0;
2522 
2523 	xattr_list_len = ksmbd_vfs_listxattr(path->dentry, &xattr_list);
2524 	if (xattr_list_len < 0) {
2525 		goto out;
2526 	} else if (!xattr_list_len) {
2527 		ksmbd_debug(SMB, "empty xattr in the file\n");
2528 		goto out;
2529 	}
2530 
2531 	for (name = xattr_list; name - xattr_list < xattr_list_len;
2532 			name += strlen(name) + 1) {
2533 		ksmbd_debug(SMB, "%s, len %zd\n", name, strlen(name));
2534 
2535 		if (!strncmp(name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN) &&
2536 		    !strncmp(&name[XATTR_USER_PREFIX_LEN], STREAM_PREFIX,
2537 			     STREAM_PREFIX_LEN)) {
2538 			err = ksmbd_vfs_remove_xattr(idmap, path,
2539 						     name, true);
2540 			if (err)
2541 				ksmbd_debug(SMB, "remove xattr failed : %s\n",
2542 					    name);
2543 		}
2544 	}
2545 out:
2546 	kvfree(xattr_list);
2547 	return err;
2548 }
2549 
2550 static int smb2_create_truncate(const struct path *path)
2551 {
2552 	int rc = vfs_truncate(path, 0);
2553 
2554 	if (rc) {
2555 		pr_err("vfs_truncate failed, rc %d\n", rc);
2556 		return rc;
2557 	}
2558 
2559 	rc = smb2_remove_smb_xattrs(path);
2560 	if (rc == -EOPNOTSUPP)
2561 		rc = 0;
2562 	if (rc)
2563 		ksmbd_debug(SMB,
2564 			    "ksmbd_truncate_stream_name_xattr failed, rc %d\n",
2565 			    rc);
2566 	return rc;
2567 }
2568 
2569 static void smb2_new_xattrs(struct ksmbd_tree_connect *tcon, const struct path *path,
2570 			    struct ksmbd_file *fp)
2571 {
2572 	struct xattr_dos_attrib da = {0};
2573 	int rc;
2574 
2575 	if (!test_share_config_flag(tcon->share_conf,
2576 				    KSMBD_SHARE_FLAG_STORE_DOS_ATTRS))
2577 		return;
2578 
2579 	da.version = 4;
2580 	da.attr = le32_to_cpu(fp->f_ci->m_fattr);
2581 	da.itime = da.create_time = fp->create_time;
2582 	da.flags = XATTR_DOSINFO_ATTRIB | XATTR_DOSINFO_CREATE_TIME |
2583 		XATTR_DOSINFO_ITIME;
2584 
2585 	rc = ksmbd_vfs_set_dos_attrib_xattr(mnt_idmap(path->mnt), path, &da, true);
2586 	if (rc)
2587 		ksmbd_debug(SMB, "failed to store file attribute into xattr\n");
2588 }
2589 
2590 static void smb2_update_xattrs(struct ksmbd_tree_connect *tcon,
2591 			       const struct path *path, struct ksmbd_file *fp)
2592 {
2593 	struct xattr_dos_attrib da;
2594 	int rc;
2595 
2596 	fp->f_ci->m_fattr &= ~(FILE_ATTRIBUTE_HIDDEN_LE | FILE_ATTRIBUTE_SYSTEM_LE);
2597 
2598 	/* get FileAttributes from XATTR_NAME_DOS_ATTRIBUTE */
2599 	if (!test_share_config_flag(tcon->share_conf,
2600 				    KSMBD_SHARE_FLAG_STORE_DOS_ATTRS))
2601 		return;
2602 
2603 	rc = ksmbd_vfs_get_dos_attrib_xattr(mnt_idmap(path->mnt),
2604 					    path->dentry, &da);
2605 	if (rc > 0) {
2606 		fp->f_ci->m_fattr = cpu_to_le32(da.attr);
2607 		fp->create_time = da.create_time;
2608 		fp->itime = da.itime;
2609 	}
2610 }
2611 
2612 static int smb2_creat(struct ksmbd_work *work,
2613 		      struct path *path, char *name, int open_flags,
2614 		      umode_t posix_mode, bool is_dir)
2615 {
2616 	struct ksmbd_tree_connect *tcon = work->tcon;
2617 	struct ksmbd_share_config *share = tcon->share_conf;
2618 	umode_t mode;
2619 	int rc;
2620 
2621 	if (!(open_flags & O_CREAT))
2622 		return -EBADF;
2623 
2624 	ksmbd_debug(SMB, "file does not exist, so creating\n");
2625 	if (is_dir == true) {
2626 		ksmbd_debug(SMB, "creating directory\n");
2627 
2628 		mode = share_config_directory_mode(share, posix_mode);
2629 		rc = ksmbd_vfs_mkdir(work, name, mode);
2630 		if (rc)
2631 			return rc;
2632 	} else {
2633 		ksmbd_debug(SMB, "creating regular file\n");
2634 
2635 		mode = share_config_create_mode(share, posix_mode);
2636 		rc = ksmbd_vfs_create(work, name, mode);
2637 		if (rc)
2638 			return rc;
2639 	}
2640 
2641 	rc = ksmbd_vfs_kern_path(work, name, 0, path, 0);
2642 	if (rc) {
2643 		pr_err("cannot get linux path (%s), err = %d\n",
2644 		       name, rc);
2645 		return rc;
2646 	}
2647 	return 0;
2648 }
2649 
2650 static int smb2_create_sd_buffer(struct ksmbd_work *work,
2651 				 struct smb2_create_req *req,
2652 				 const struct path *path)
2653 {
2654 	struct create_context *context;
2655 	struct create_sd_buf_req *sd_buf;
2656 
2657 	if (!req->CreateContextsOffset)
2658 		return -ENOENT;
2659 
2660 	/* Parse SD BUFFER create contexts */
2661 	context = smb2_find_context_vals(req, SMB2_CREATE_SD_BUFFER, 4);
2662 	if (!context)
2663 		return -ENOENT;
2664 	else if (IS_ERR(context))
2665 		return PTR_ERR(context);
2666 
2667 	ksmbd_debug(SMB,
2668 		    "Set ACLs using SMB2_CREATE_SD_BUFFER context\n");
2669 	sd_buf = (struct create_sd_buf_req *)context;
2670 	if (le16_to_cpu(context->DataOffset) +
2671 	    le32_to_cpu(context->DataLength) <
2672 	    sizeof(struct create_sd_buf_req))
2673 		return -EINVAL;
2674 	return set_info_sec(work->conn, work->tcon, path, &sd_buf->ntsd,
2675 			    le32_to_cpu(sd_buf->ccontext.DataLength), true, false);
2676 }
2677 
2678 static void ksmbd_acls_fattr(struct smb_fattr *fattr,
2679 			     struct mnt_idmap *idmap,
2680 			     struct inode *inode)
2681 {
2682 	vfsuid_t vfsuid = i_uid_into_vfsuid(idmap, inode);
2683 	vfsgid_t vfsgid = i_gid_into_vfsgid(idmap, inode);
2684 
2685 	fattr->cf_uid = vfsuid_into_kuid(vfsuid);
2686 	fattr->cf_gid = vfsgid_into_kgid(vfsgid);
2687 	fattr->cf_mode = inode->i_mode;
2688 	fattr->cf_acls = NULL;
2689 	fattr->cf_dacls = NULL;
2690 
2691 	if (IS_ENABLED(CONFIG_FS_POSIX_ACL)) {
2692 		fattr->cf_acls = get_inode_acl(inode, ACL_TYPE_ACCESS);
2693 		if (S_ISDIR(inode->i_mode))
2694 			fattr->cf_dacls = get_inode_acl(inode, ACL_TYPE_DEFAULT);
2695 	}
2696 }
2697 
2698 enum {
2699 	DURABLE_RECONN_V2 = 1,
2700 	DURABLE_RECONN,
2701 	DURABLE_REQ_V2,
2702 	DURABLE_REQ,
2703 };
2704 
2705 struct durable_info {
2706 	struct ksmbd_file *fp;
2707 	unsigned short int type;
2708 	bool persistent;
2709 	bool reconnected;
2710 	unsigned int timeout;
2711 	char *CreateGuid;
2712 };
2713 
2714 static int parse_durable_handle_context(struct ksmbd_work *work,
2715 					struct smb2_create_req *req,
2716 					struct lease_ctx_info *lc,
2717 					struct durable_info *dh_info)
2718 {
2719 	struct ksmbd_conn *conn = work->conn;
2720 	struct create_context *context;
2721 	int dh_idx, err = 0;
2722 	u64 persistent_id = 0;
2723 	int req_op_level;
2724 	static const char * const durable_arr[] = {"DH2C", "DHnC", "DH2Q", "DHnQ"};
2725 
2726 	req_op_level = req->RequestedOplockLevel;
2727 	for (dh_idx = DURABLE_RECONN_V2; dh_idx <= ARRAY_SIZE(durable_arr);
2728 	     dh_idx++) {
2729 		context = smb2_find_context_vals(req, durable_arr[dh_idx - 1], 4);
2730 		if (IS_ERR(context)) {
2731 			err = PTR_ERR(context);
2732 			goto out;
2733 		}
2734 		if (!context)
2735 			continue;
2736 
2737 		switch (dh_idx) {
2738 		case DURABLE_RECONN_V2:
2739 		{
2740 			struct create_durable_handle_reconnect_v2 *recon_v2;
2741 
2742 			if (dh_info->type == DURABLE_RECONN ||
2743 			    dh_info->type == DURABLE_REQ_V2) {
2744 				err = -EINVAL;
2745 				goto out;
2746 			}
2747 
2748 			if (le16_to_cpu(context->DataOffset) +
2749 				le32_to_cpu(context->DataLength) <
2750 			    sizeof(struct create_durable_handle_reconnect_v2)) {
2751 				err = -EINVAL;
2752 				goto out;
2753 			}
2754 
2755 			recon_v2 = (struct create_durable_handle_reconnect_v2 *)context;
2756 			persistent_id = recon_v2->dcontext.Fid.PersistentFileId;
2757 			dh_info->fp = ksmbd_lookup_durable_fd(persistent_id);
2758 			if (!dh_info->fp) {
2759 				ksmbd_debug(SMB, "Failed to get durable handle state\n");
2760 				err = -EBADF;
2761 				goto out;
2762 			}
2763 
2764 			if (memcmp(dh_info->fp->create_guid, recon_v2->dcontext.CreateGuid,
2765 				   SMB2_CREATE_GUID_SIZE)) {
2766 				err = -EBADF;
2767 				ksmbd_put_durable_fd(dh_info->fp);
2768 				goto out;
2769 			}
2770 
2771 			dh_info->type = dh_idx;
2772 			dh_info->reconnected = true;
2773 			ksmbd_debug(SMB,
2774 				"reconnect v2 Persistent-id from reconnect = %llu\n",
2775 					persistent_id);
2776 			break;
2777 		}
2778 		case DURABLE_RECONN:
2779 		{
2780 			create_durable_reconn_t *recon;
2781 
2782 			if (dh_info->type == DURABLE_RECONN_V2 ||
2783 			    dh_info->type == DURABLE_REQ_V2) {
2784 				err = -EINVAL;
2785 				goto out;
2786 			}
2787 
2788 			if (le16_to_cpu(context->DataOffset) +
2789 				le32_to_cpu(context->DataLength) <
2790 			    sizeof(create_durable_reconn_t)) {
2791 				err = -EINVAL;
2792 				goto out;
2793 			}
2794 
2795 			recon = (create_durable_reconn_t *)context;
2796 			persistent_id = recon->Data.Fid.PersistentFileId;
2797 			dh_info->fp = ksmbd_lookup_durable_fd(persistent_id);
2798 			if (!dh_info->fp) {
2799 				ksmbd_debug(SMB, "Failed to get durable handle state\n");
2800 				err = -EBADF;
2801 				goto out;
2802 			}
2803 
2804 			dh_info->type = dh_idx;
2805 			dh_info->reconnected = true;
2806 			ksmbd_debug(SMB, "reconnect Persistent-id from reconnect = %llu\n",
2807 				    persistent_id);
2808 			break;
2809 		}
2810 		case DURABLE_REQ_V2:
2811 		{
2812 			struct create_durable_req_v2 *durable_v2_blob;
2813 
2814 			if (dh_info->type == DURABLE_RECONN ||
2815 			    dh_info->type == DURABLE_RECONN_V2) {
2816 				err = -EINVAL;
2817 				goto out;
2818 			}
2819 
2820 			if (le16_to_cpu(context->DataOffset) +
2821 				le32_to_cpu(context->DataLength) <
2822 			    sizeof(struct create_durable_req_v2)) {
2823 				err = -EINVAL;
2824 				goto out;
2825 			}
2826 
2827 			durable_v2_blob =
2828 				(struct create_durable_req_v2 *)context;
2829 			ksmbd_debug(SMB, "Request for durable v2 open\n");
2830 			dh_info->fp = ksmbd_lookup_fd_cguid(durable_v2_blob->dcontext.CreateGuid);
2831 			if (dh_info->fp) {
2832 				if (!memcmp(conn->ClientGUID, dh_info->fp->client_guid,
2833 					    SMB2_CLIENT_GUID_SIZE)) {
2834 					if (!(req->hdr.Flags & SMB2_FLAGS_REPLAY_OPERATION)) {
2835 						err = -ENOEXEC;
2836 						ksmbd_put_durable_fd(dh_info->fp);
2837 						goto out;
2838 					}
2839 
2840 					if (dh_info->fp->conn) {
2841 						ksmbd_put_durable_fd(dh_info->fp);
2842 						err = -EBADF;
2843 						goto out;
2844 					}
2845 					dh_info->reconnected = true;
2846 					goto out;
2847 				}
2848 				ksmbd_put_durable_fd(dh_info->fp);
2849 				dh_info->fp = NULL;
2850 			}
2851 
2852 			if ((lc && (lc->req_state & SMB2_LEASE_HANDLE_CACHING_LE)) ||
2853 			    req_op_level == SMB2_OPLOCK_LEVEL_BATCH) {
2854 				dh_info->CreateGuid =
2855 					durable_v2_blob->dcontext.CreateGuid;
2856 				dh_info->persistent =
2857 					le32_to_cpu(durable_v2_blob->dcontext.Flags);
2858 				dh_info->timeout =
2859 					le32_to_cpu(durable_v2_blob->dcontext.Timeout);
2860 				dh_info->type = dh_idx;
2861 			}
2862 			break;
2863 		}
2864 		case DURABLE_REQ:
2865 			if (dh_info->type == DURABLE_RECONN)
2866 				goto out;
2867 			if (dh_info->type == DURABLE_RECONN_V2 ||
2868 			    dh_info->type == DURABLE_REQ_V2) {
2869 				err = -EINVAL;
2870 				goto out;
2871 			}
2872 
2873 			if ((lc && (lc->req_state & SMB2_LEASE_HANDLE_CACHING_LE)) ||
2874 			    req_op_level == SMB2_OPLOCK_LEVEL_BATCH) {
2875 				ksmbd_debug(SMB, "Request for durable open\n");
2876 				dh_info->type = dh_idx;
2877 			}
2878 		}
2879 	}
2880 
2881 out:
2882 	return err;
2883 }
2884 
2885 /**
2886  * smb2_open() - handler for smb file open request
2887  * @work:	smb work containing request buffer
2888  *
2889  * Return:      0 on success, otherwise error
2890  */
2891 int smb2_open(struct ksmbd_work *work)
2892 {
2893 	struct ksmbd_conn *conn = work->conn;
2894 	struct ksmbd_session *sess = work->sess;
2895 	struct ksmbd_tree_connect *tcon = work->tcon;
2896 	struct smb2_create_req *req;
2897 	struct smb2_create_rsp *rsp;
2898 	struct path path;
2899 	struct ksmbd_share_config *share = tcon->share_conf;
2900 	struct ksmbd_file *fp = NULL;
2901 	struct file *filp = NULL;
2902 	struct mnt_idmap *idmap = NULL;
2903 	struct kstat stat;
2904 	struct create_context *context;
2905 	struct lease_ctx_info *lc = NULL;
2906 	struct create_ea_buf_req *ea_buf = NULL;
2907 	struct oplock_info *opinfo;
2908 	struct durable_info dh_info = {0};
2909 	__le32 *next_ptr = NULL;
2910 	int req_op_level = 0, open_flags = 0, may_flags = 0, file_info = 0;
2911 	int rc = 0;
2912 	int contxt_cnt = 0, query_disk_id = 0;
2913 	bool maximal_access_ctxt = false, posix_ctxt = false;
2914 	int s_type = 0;
2915 	int next_off = 0;
2916 	char *name = NULL;
2917 	char *stream_name = NULL;
2918 	bool file_present = false, created = false, already_permitted = false;
2919 	int share_ret, need_truncate = 0;
2920 	u64 time;
2921 	umode_t posix_mode = 0;
2922 	__le32 daccess, maximal_access = 0;
2923 	int iov_len = 0;
2924 
2925 	ksmbd_debug(SMB, "Received smb2 create request\n");
2926 
2927 	WORK_BUFFERS(work, req, rsp);
2928 
2929 	if (req->hdr.NextCommand && !work->next_smb2_rcv_hdr_off &&
2930 	    (req->hdr.Flags & SMB2_FLAGS_RELATED_OPERATIONS)) {
2931 		ksmbd_debug(SMB, "invalid flag in chained command\n");
2932 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
2933 		smb2_set_err_rsp(work);
2934 		return -EINVAL;
2935 	}
2936 
2937 	if (test_share_config_flag(share, KSMBD_SHARE_FLAG_PIPE)) {
2938 		ksmbd_debug(SMB, "IPC pipe create request\n");
2939 		return create_smb2_pipe(work);
2940 	}
2941 
2942 	if (req->CreateContextsOffset && tcon->posix_extensions) {
2943 		context = smb2_find_context_vals(req, SMB2_CREATE_TAG_POSIX, 16);
2944 		if (IS_ERR(context)) {
2945 			rc = PTR_ERR(context);
2946 			goto err_out2;
2947 		} else if (context) {
2948 			struct create_posix *posix = (struct create_posix *)context;
2949 
2950 			if (le16_to_cpu(context->DataOffset) +
2951 				le32_to_cpu(context->DataLength) <
2952 			    sizeof(struct create_posix) - 4) {
2953 				rc = -EINVAL;
2954 				goto err_out2;
2955 			}
2956 			ksmbd_debug(SMB, "get posix context\n");
2957 
2958 			posix_mode = le32_to_cpu(posix->Mode);
2959 			posix_ctxt = true;
2960 		}
2961 	}
2962 
2963 	if (req->NameLength) {
2964 		name = smb2_get_name((char *)req + le16_to_cpu(req->NameOffset),
2965 				     le16_to_cpu(req->NameLength),
2966 				     work->conn->local_nls);
2967 		if (IS_ERR(name)) {
2968 			rc = PTR_ERR(name);
2969 			name = NULL;
2970 			goto err_out2;
2971 		}
2972 
2973 		ksmbd_debug(SMB, "converted name = %s\n", name);
2974 
2975 		if (posix_ctxt == false) {
2976 			if (strchr(name, ':')) {
2977 				if (!test_share_config_flag(work->tcon->share_conf,
2978 							KSMBD_SHARE_FLAG_STREAMS)) {
2979 					rc = -EBADF;
2980 					goto err_out2;
2981 				}
2982 				rc = parse_stream_name(name, &stream_name, &s_type);
2983 				if (rc < 0)
2984 					goto err_out2;
2985 			}
2986 
2987 			rc = ksmbd_validate_filename(name);
2988 			if (rc < 0)
2989 				goto err_out2;
2990 		}
2991 
2992 		if (ksmbd_share_veto_filename(share, name)) {
2993 			rc = -ENOENT;
2994 			ksmbd_debug(SMB, "Reject open(), vetoed file: %s\n",
2995 				    name);
2996 			goto err_out2;
2997 		}
2998 	} else {
2999 		name = kstrdup("", KSMBD_DEFAULT_GFP);
3000 		if (!name) {
3001 			rc = -ENOMEM;
3002 			goto err_out2;
3003 		}
3004 	}
3005 
3006 	req_op_level = req->RequestedOplockLevel;
3007 
3008 	if (server_conf.flags & KSMBD_GLOBAL_FLAG_DURABLE_HANDLE &&
3009 	    req->CreateContextsOffset) {
3010 		lc = parse_lease_state(req);
3011 		rc = parse_durable_handle_context(work, req, lc, &dh_info);
3012 		if (rc) {
3013 			ksmbd_debug(SMB, "error parsing durable handle context\n");
3014 			goto err_out2;
3015 		}
3016 
3017 		if (dh_info.reconnected == true) {
3018 			rc = smb2_check_durable_oplock(conn, share, dh_info.fp,
3019 					lc, sess->user, name);
3020 			if (rc)
3021 				goto err_out2;
3022 
3023 			rc = ksmbd_reopen_durable_fd(work, dh_info.fp);
3024 			if (rc)
3025 				goto err_out2;
3026 
3027 			fp = dh_info.fp;
3028 
3029 			if (ksmbd_override_fsids(work)) {
3030 				rc = -ENOMEM;
3031 				goto err_out2;
3032 			}
3033 
3034 			file_info = FILE_OPENED;
3035 
3036 			rc = ksmbd_vfs_getattr(&fp->filp->f_path, &stat);
3037 			if (rc)
3038 				goto err_out2;
3039 
3040 			goto reconnected_fp;
3041 		}
3042 	} else if (req_op_level == SMB2_OPLOCK_LEVEL_LEASE)
3043 		lc = parse_lease_state(req);
3044 
3045 	if (le32_to_cpu(req->ImpersonationLevel) > le32_to_cpu(IL_DELEGATE)) {
3046 		pr_err("Invalid impersonationlevel : 0x%x\n",
3047 		       le32_to_cpu(req->ImpersonationLevel));
3048 		rc = -EIO;
3049 		rsp->hdr.Status = STATUS_BAD_IMPERSONATION_LEVEL;
3050 		goto err_out2;
3051 	}
3052 
3053 	if (req->CreateOptions && !(req->CreateOptions & CREATE_OPTIONS_MASK_LE)) {
3054 		pr_err("Invalid create options : 0x%x\n",
3055 		       le32_to_cpu(req->CreateOptions));
3056 		rc = -EINVAL;
3057 		goto err_out2;
3058 	} else {
3059 		if (req->CreateOptions & FILE_SEQUENTIAL_ONLY_LE &&
3060 		    req->CreateOptions & FILE_RANDOM_ACCESS_LE)
3061 			req->CreateOptions &= ~FILE_SEQUENTIAL_ONLY_LE;
3062 
3063 		if (req->CreateOptions &
3064 		    (FILE_OPEN_BY_FILE_ID_LE | CREATE_TREE_CONNECTION |
3065 		     FILE_RESERVE_OPFILTER_LE)) {
3066 			rc = -EOPNOTSUPP;
3067 			goto err_out2;
3068 		}
3069 
3070 		if (req->CreateOptions & FILE_DIRECTORY_FILE_LE) {
3071 			if (req->CreateOptions & FILE_NON_DIRECTORY_FILE_LE) {
3072 				rc = -EINVAL;
3073 				goto err_out2;
3074 			} else if (req->CreateOptions & FILE_NO_COMPRESSION_LE) {
3075 				req->CreateOptions &= ~FILE_NO_COMPRESSION_LE;
3076 			}
3077 		}
3078 	}
3079 
3080 	if (le32_to_cpu(req->CreateDisposition) >
3081 	    le32_to_cpu(FILE_OVERWRITE_IF_LE)) {
3082 		pr_err("Invalid create disposition : 0x%x\n",
3083 		       le32_to_cpu(req->CreateDisposition));
3084 		rc = -EINVAL;
3085 		goto err_out2;
3086 	}
3087 
3088 	if (!(req->DesiredAccess & DESIRED_ACCESS_MASK)) {
3089 		pr_err("Invalid desired access : 0x%x\n",
3090 		       le32_to_cpu(req->DesiredAccess));
3091 		rc = -EACCES;
3092 		goto err_out2;
3093 	}
3094 
3095 	if (req->FileAttributes && !(req->FileAttributes & FILE_ATTRIBUTE_MASK_LE)) {
3096 		pr_err("Invalid file attribute : 0x%x\n",
3097 		       le32_to_cpu(req->FileAttributes));
3098 		rc = -EINVAL;
3099 		goto err_out2;
3100 	}
3101 
3102 	if (req->CreateContextsOffset) {
3103 		/* Parse non-durable handle create contexts */
3104 		context = smb2_find_context_vals(req, SMB2_CREATE_EA_BUFFER, 4);
3105 		if (IS_ERR(context)) {
3106 			rc = PTR_ERR(context);
3107 			goto err_out2;
3108 		} else if (context) {
3109 			ea_buf = (struct create_ea_buf_req *)context;
3110 			if (le16_to_cpu(context->DataOffset) +
3111 			    le32_to_cpu(context->DataLength) <
3112 			    sizeof(struct create_ea_buf_req)) {
3113 				rc = -EINVAL;
3114 				goto err_out2;
3115 			}
3116 			if (req->CreateOptions & FILE_NO_EA_KNOWLEDGE_LE) {
3117 				rsp->hdr.Status = STATUS_ACCESS_DENIED;
3118 				rc = -EACCES;
3119 				goto err_out2;
3120 			}
3121 		}
3122 
3123 		context = smb2_find_context_vals(req,
3124 						 SMB2_CREATE_QUERY_MAXIMAL_ACCESS_REQUEST, 4);
3125 		if (IS_ERR(context)) {
3126 			rc = PTR_ERR(context);
3127 			goto err_out2;
3128 		} else if (context) {
3129 			ksmbd_debug(SMB,
3130 				    "get query maximal access context\n");
3131 			maximal_access_ctxt = 1;
3132 		}
3133 
3134 		context = smb2_find_context_vals(req,
3135 						 SMB2_CREATE_TIMEWARP_REQUEST, 4);
3136 		if (IS_ERR(context)) {
3137 			rc = PTR_ERR(context);
3138 			goto err_out2;
3139 		} else if (context) {
3140 			ksmbd_debug(SMB, "get timewarp context\n");
3141 			rc = -EBADF;
3142 			goto err_out2;
3143 		}
3144 	}
3145 
3146 	if (ksmbd_override_fsids(work)) {
3147 		rc = -ENOMEM;
3148 		goto err_out2;
3149 	}
3150 
3151 	rc = ksmbd_vfs_kern_path(work, name, LOOKUP_NO_SYMLINKS,
3152 				 &path, 1);
3153 	if (!rc) {
3154 		file_present = true;
3155 
3156 		if (req->CreateOptions & FILE_DELETE_ON_CLOSE_LE) {
3157 			/*
3158 			 * If file exists with under flags, return access
3159 			 * denied error.
3160 			 */
3161 			if (req->CreateDisposition == FILE_OVERWRITE_IF_LE ||
3162 			    req->CreateDisposition == FILE_OPEN_IF_LE) {
3163 				rc = -EACCES;
3164 				goto err_out;
3165 			}
3166 
3167 			if (!test_tree_conn_flag(tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
3168 				ksmbd_debug(SMB,
3169 					    "User does not have write permission\n");
3170 				rc = -EACCES;
3171 				goto err_out;
3172 			}
3173 		} else if (d_is_symlink(path.dentry)) {
3174 			rc = -EACCES;
3175 			goto err_out;
3176 		}
3177 
3178 		idmap = mnt_idmap(path.mnt);
3179 	} else {
3180 		if (rc != -ENOENT)
3181 			goto err_out;
3182 		ksmbd_debug(SMB, "can not get linux path for %s, rc = %d\n",
3183 			    name, rc);
3184 		rc = 0;
3185 	}
3186 
3187 	if (stream_name) {
3188 		if (req->CreateOptions & FILE_DIRECTORY_FILE_LE) {
3189 			if (s_type == DATA_STREAM) {
3190 				rc = -EIO;
3191 				rsp->hdr.Status = STATUS_NOT_A_DIRECTORY;
3192 			}
3193 		} else {
3194 			if (file_present && S_ISDIR(d_inode(path.dentry)->i_mode) &&
3195 			    s_type == DATA_STREAM) {
3196 				rc = -EIO;
3197 				rsp->hdr.Status = STATUS_FILE_IS_A_DIRECTORY;
3198 			}
3199 		}
3200 
3201 		if (req->CreateOptions & FILE_DIRECTORY_FILE_LE &&
3202 		    req->FileAttributes & FILE_ATTRIBUTE_NORMAL_LE) {
3203 			rsp->hdr.Status = STATUS_NOT_A_DIRECTORY;
3204 			rc = -EIO;
3205 		}
3206 
3207 		if (rc < 0)
3208 			goto err_out;
3209 	}
3210 
3211 	if (file_present && req->CreateOptions & FILE_NON_DIRECTORY_FILE_LE &&
3212 	    S_ISDIR(d_inode(path.dentry)->i_mode) &&
3213 	    !(req->CreateOptions & FILE_DELETE_ON_CLOSE_LE)) {
3214 		ksmbd_debug(SMB, "open() argument is a directory: %s, %x\n",
3215 			    name, req->CreateOptions);
3216 		rsp->hdr.Status = STATUS_FILE_IS_A_DIRECTORY;
3217 		rc = -EIO;
3218 		goto err_out;
3219 	}
3220 
3221 	if (file_present && (req->CreateOptions & FILE_DIRECTORY_FILE_LE) &&
3222 	    !(req->CreateDisposition == FILE_CREATE_LE) &&
3223 	    !S_ISDIR(d_inode(path.dentry)->i_mode)) {
3224 		rsp->hdr.Status = STATUS_NOT_A_DIRECTORY;
3225 		rc = -EIO;
3226 		goto err_out;
3227 	}
3228 
3229 	if (!stream_name && file_present &&
3230 	    req->CreateDisposition == FILE_CREATE_LE) {
3231 		rc = -EEXIST;
3232 		goto err_out;
3233 	}
3234 
3235 	daccess = smb_map_generic_desired_access(req->DesiredAccess);
3236 
3237 	if (file_present && !(req->CreateOptions & FILE_DELETE_ON_CLOSE_LE)) {
3238 		rc = smb_check_perm_dacl(conn, &path, &daccess,
3239 					 sess->user->uid);
3240 		if (rc)
3241 			goto err_out;
3242 	}
3243 
3244 	if (daccess & FILE_MAXIMAL_ACCESS_LE) {
3245 		if (!file_present) {
3246 			daccess = cpu_to_le32(GENERIC_ALL_FLAGS);
3247 		} else {
3248 			ksmbd_vfs_query_maximal_access(idmap,
3249 							    path.dentry,
3250 							    &daccess);
3251 			already_permitted = true;
3252 		}
3253 		maximal_access = daccess;
3254 	}
3255 
3256 	open_flags = smb2_create_open_flags(file_present, daccess,
3257 					    req->CreateDisposition,
3258 					    &may_flags,
3259 					    req->CreateOptions,
3260 					    file_present ? d_inode(path.dentry)->i_mode : 0);
3261 
3262 	if (!test_tree_conn_flag(tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
3263 		if (open_flags & (O_CREAT | O_TRUNC)) {
3264 			ksmbd_debug(SMB,
3265 				    "User does not have write permission\n");
3266 			rc = -EACCES;
3267 			goto err_out;
3268 		}
3269 	}
3270 
3271 	/*create file if not present */
3272 	if (!file_present) {
3273 		rc = smb2_creat(work, &path, name, open_flags,
3274 				posix_mode,
3275 				req->CreateOptions & FILE_DIRECTORY_FILE_LE);
3276 		if (rc) {
3277 			if (rc == -ENOENT) {
3278 				rc = -EIO;
3279 				rsp->hdr.Status = STATUS_OBJECT_PATH_NOT_FOUND;
3280 			}
3281 			goto err_out;
3282 		}
3283 
3284 		created = true;
3285 		idmap = mnt_idmap(path.mnt);
3286 		if (ea_buf) {
3287 			if (le32_to_cpu(ea_buf->ccontext.DataLength) <
3288 			    sizeof(struct smb2_ea_info)) {
3289 				rc = -EINVAL;
3290 				goto err_out;
3291 			}
3292 
3293 			rc = smb2_set_ea(&ea_buf->ea,
3294 					 le32_to_cpu(ea_buf->ccontext.DataLength),
3295 					 &path, false);
3296 			if (rc == -EOPNOTSUPP)
3297 				rc = 0;
3298 			else if (rc)
3299 				goto err_out;
3300 		}
3301 	} else if (!already_permitted) {
3302 		/* FILE_READ_ATTRIBUTE is allowed without inode_permission,
3303 		 * because execute(search) permission on a parent directory,
3304 		 * is already granted.
3305 		 */
3306 		if (daccess & ~(FILE_READ_ATTRIBUTES_LE | FILE_READ_CONTROL_LE)) {
3307 			rc = inode_permission(idmap,
3308 					      d_inode(path.dentry),
3309 					      may_flags);
3310 			if (rc)
3311 				goto err_out;
3312 
3313 			if ((daccess & FILE_DELETE_LE) ||
3314 			    (req->CreateOptions & FILE_DELETE_ON_CLOSE_LE)) {
3315 				rc = inode_permission(idmap,
3316 						      d_inode(path.dentry->d_parent),
3317 						      MAY_EXEC | MAY_WRITE);
3318 				if (rc)
3319 					goto err_out;
3320 			}
3321 		}
3322 	}
3323 
3324 	rc = ksmbd_query_inode_status(path.dentry->d_parent);
3325 	if (rc == KSMBD_INODE_STATUS_PENDING_DELETE) {
3326 		rc = -EBUSY;
3327 		goto err_out;
3328 	}
3329 
3330 	rc = 0;
3331 	filp = dentry_open(&path, open_flags, current_cred());
3332 	if (IS_ERR(filp)) {
3333 		rc = PTR_ERR(filp);
3334 		pr_err("dentry open for dir failed, rc %d\n", rc);
3335 		goto err_out;
3336 	}
3337 
3338 	if (file_present) {
3339 		if (!(open_flags & O_TRUNC))
3340 			file_info = FILE_OPENED;
3341 		else
3342 			file_info = FILE_OVERWRITTEN;
3343 
3344 		if ((req->CreateDisposition & FILE_CREATE_MASK_LE) ==
3345 		    FILE_SUPERSEDE_LE)
3346 			file_info = FILE_SUPERSEDED;
3347 	} else if (open_flags & O_CREAT) {
3348 		file_info = FILE_CREATED;
3349 	}
3350 
3351 	ksmbd_vfs_set_fadvise(filp, req->CreateOptions);
3352 
3353 	/* Obtain Volatile-ID */
3354 	fp = ksmbd_open_fd(work, filp);
3355 	if (IS_ERR(fp)) {
3356 		fput(filp);
3357 		rc = PTR_ERR(fp);
3358 		fp = NULL;
3359 		goto err_out;
3360 	}
3361 
3362 	/* Get Persistent-ID */
3363 	ksmbd_open_durable_fd(fp);
3364 	if (!has_file_id(fp->persistent_id)) {
3365 		rc = -ENOMEM;
3366 		goto err_out;
3367 	}
3368 
3369 	fp->cdoption = req->CreateDisposition;
3370 	fp->daccess = daccess;
3371 	fp->saccess = req->ShareAccess;
3372 	fp->coption = req->CreateOptions;
3373 
3374 	/* Set default windows and posix acls if creating new file */
3375 	if (created) {
3376 		int posix_acl_rc;
3377 		struct inode *inode = d_inode(path.dentry);
3378 
3379 		posix_acl_rc = ksmbd_vfs_inherit_posix_acl(idmap,
3380 							   &path,
3381 							   d_inode(path.dentry->d_parent));
3382 		if (posix_acl_rc)
3383 			ksmbd_debug(SMB, "inherit posix acl failed : %d\n", posix_acl_rc);
3384 
3385 		if (test_share_config_flag(work->tcon->share_conf,
3386 					   KSMBD_SHARE_FLAG_ACL_XATTR)) {
3387 			rc = smb_inherit_dacl(conn, &path, sess->user->uid,
3388 					      sess->user->gid);
3389 		}
3390 
3391 		if (rc) {
3392 			rc = smb2_create_sd_buffer(work, req, &path);
3393 			if (rc) {
3394 				if (posix_acl_rc)
3395 					ksmbd_vfs_set_init_posix_acl(idmap,
3396 								     &path);
3397 
3398 				if (test_share_config_flag(work->tcon->share_conf,
3399 							   KSMBD_SHARE_FLAG_ACL_XATTR)) {
3400 					struct smb_fattr fattr;
3401 					struct smb_ntsd *pntsd;
3402 					int pntsd_size;
3403 					size_t scratch_len;
3404 
3405 					ksmbd_acls_fattr(&fattr, idmap, inode);
3406 					scratch_len = smb_acl_sec_desc_scratch_len(&fattr,
3407 							NULL, 0,
3408 							OWNER_SECINFO | GROUP_SECINFO |
3409 							DACL_SECINFO);
3410 					if (!scratch_len || scratch_len == SIZE_MAX) {
3411 						rc = -EFBIG;
3412 						posix_acl_release(fattr.cf_acls);
3413 						posix_acl_release(fattr.cf_dacls);
3414 						goto err_out;
3415 					}
3416 
3417 					pntsd = kvzalloc(scratch_len, KSMBD_DEFAULT_GFP);
3418 					if (!pntsd) {
3419 						rc = -ENOMEM;
3420 						posix_acl_release(fattr.cf_acls);
3421 						posix_acl_release(fattr.cf_dacls);
3422 						goto err_out;
3423 					}
3424 
3425 					rc = build_sec_desc(idmap,
3426 							    pntsd, NULL, 0,
3427 							    OWNER_SECINFO |
3428 							    GROUP_SECINFO |
3429 							    DACL_SECINFO,
3430 							    &pntsd_size, &fattr);
3431 					posix_acl_release(fattr.cf_acls);
3432 					posix_acl_release(fattr.cf_dacls);
3433 					if (rc) {
3434 						kvfree(pntsd);
3435 						goto err_out;
3436 					}
3437 
3438 					rc = ksmbd_vfs_set_sd_xattr(conn,
3439 								    idmap,
3440 								    &path,
3441 								    pntsd,
3442 								    pntsd_size,
3443 								    false);
3444 					kvfree(pntsd);
3445 					if (rc)
3446 						pr_err("failed to store ntacl in xattr : %d\n",
3447 						       rc);
3448 				}
3449 			}
3450 		}
3451 		rc = 0;
3452 	}
3453 
3454 	if (stream_name) {
3455 		rc = smb2_set_stream_name_xattr(&path,
3456 						fp,
3457 						stream_name,
3458 						s_type);
3459 		if (rc)
3460 			goto err_out;
3461 		file_info = FILE_CREATED;
3462 	}
3463 
3464 	fp->attrib_only = !(req->DesiredAccess & ~(FILE_READ_ATTRIBUTES_LE |
3465 			FILE_WRITE_ATTRIBUTES_LE | FILE_SYNCHRONIZE_LE));
3466 
3467 	fp->is_posix_ctxt = posix_ctxt;
3468 
3469 	/* fp should be searchable through ksmbd_inode.m_fp_list
3470 	 * after daccess, saccess, attrib_only, and stream are
3471 	 * initialized.
3472 	 */
3473 	down_write(&fp->f_ci->m_lock);
3474 	list_add(&fp->node, &fp->f_ci->m_fp_list);
3475 	up_write(&fp->f_ci->m_lock);
3476 
3477 	/* Check delete pending among previous fp before oplock break */
3478 	if (ksmbd_inode_pending_delete(fp)) {
3479 		rc = -EBUSY;
3480 		goto err_out;
3481 	}
3482 
3483 	if (file_present || created)
3484 		path_put(&path);
3485 
3486 	if (!S_ISDIR(file_inode(filp)->i_mode) && open_flags & O_TRUNC &&
3487 	    !fp->attrib_only && !stream_name) {
3488 		smb_break_all_oplock(work, fp);
3489 		need_truncate = 1;
3490 	}
3491 
3492 	share_ret = ksmbd_smb_check_shared_mode(fp->filp, fp);
3493 	if (!test_share_config_flag(work->tcon->share_conf, KSMBD_SHARE_FLAG_OPLOCKS) ||
3494 	    (req_op_level == SMB2_OPLOCK_LEVEL_LEASE &&
3495 	     !(conn->vals->req_capabilities & SMB2_GLOBAL_CAP_LEASING))) {
3496 		if (share_ret < 0 && !S_ISDIR(file_inode(fp->filp)->i_mode)) {
3497 			rc = share_ret;
3498 			goto err_out1;
3499 		}
3500 	} else {
3501 		if (req_op_level == SMB2_OPLOCK_LEVEL_LEASE && lc) {
3502 			if (S_ISDIR(file_inode(filp)->i_mode)) {
3503 				lc->req_state &= ~SMB2_LEASE_WRITE_CACHING_LE;
3504 				lc->is_dir = true;
3505 			}
3506 
3507 			/*
3508 			 * Compare parent lease using parent key. If there is no
3509 			 * a lease that has same parent key, Send lease break
3510 			 * notification.
3511 			 */
3512 			smb_send_parent_lease_break_noti(fp, lc);
3513 
3514 			req_op_level = smb2_map_lease_to_oplock(lc->req_state);
3515 			ksmbd_debug(SMB,
3516 				    "lease req for(%s) req oplock state 0x%x, lease state 0x%x\n",
3517 				    name, req_op_level, lc->req_state);
3518 			rc = find_same_lease_key(sess, fp->f_ci, lc);
3519 			if (rc)
3520 				goto err_out1;
3521 		} else if (open_flags == O_RDONLY &&
3522 			   (req_op_level == SMB2_OPLOCK_LEVEL_BATCH ||
3523 			    req_op_level == SMB2_OPLOCK_LEVEL_EXCLUSIVE))
3524 			req_op_level = SMB2_OPLOCK_LEVEL_II;
3525 
3526 		rc = smb_grant_oplock(work, req_op_level,
3527 				      fp->persistent_id, fp,
3528 				      le32_to_cpu(req->hdr.Id.SyncId.TreeId),
3529 				      lc, share_ret);
3530 		if (rc < 0)
3531 			goto err_out1;
3532 	}
3533 
3534 	if (req->CreateOptions & FILE_DELETE_ON_CLOSE_LE)
3535 		ksmbd_fd_set_delete_on_close(fp, file_info);
3536 
3537 	if (need_truncate) {
3538 		rc = smb2_create_truncate(&fp->filp->f_path);
3539 		if (rc)
3540 			goto err_out1;
3541 	}
3542 
3543 	if (req->CreateContextsOffset) {
3544 		struct create_alloc_size_req *az_req;
3545 
3546 		az_req = (struct create_alloc_size_req *)smb2_find_context_vals(req,
3547 					SMB2_CREATE_ALLOCATION_SIZE, 4);
3548 		if (IS_ERR(az_req)) {
3549 			rc = PTR_ERR(az_req);
3550 			goto err_out1;
3551 		} else if (az_req) {
3552 			loff_t alloc_size;
3553 			int err;
3554 
3555 			if (le16_to_cpu(az_req->ccontext.DataOffset) +
3556 			    le32_to_cpu(az_req->ccontext.DataLength) <
3557 			    sizeof(struct create_alloc_size_req)) {
3558 				rc = -EINVAL;
3559 				goto err_out1;
3560 			}
3561 			alloc_size = le64_to_cpu(az_req->AllocationSize);
3562 			ksmbd_debug(SMB,
3563 				    "request smb2 create allocate size : %llu\n",
3564 				    alloc_size);
3565 			smb_break_all_levII_oplock(work, fp, 1);
3566 			err = vfs_fallocate(fp->filp, FALLOC_FL_KEEP_SIZE, 0,
3567 					    alloc_size);
3568 			if (err < 0)
3569 				ksmbd_debug(SMB,
3570 					    "vfs_fallocate is failed : %d\n",
3571 					    err);
3572 		}
3573 
3574 		context = smb2_find_context_vals(req, SMB2_CREATE_QUERY_ON_DISK_ID, 4);
3575 		if (IS_ERR(context)) {
3576 			rc = PTR_ERR(context);
3577 			goto err_out1;
3578 		} else if (context) {
3579 			ksmbd_debug(SMB, "get query on disk id context\n");
3580 			query_disk_id = 1;
3581 		}
3582 
3583 		if (conn->is_aapl == false) {
3584 			context = smb2_find_context_vals(req, SMB2_CREATE_AAPL, 4);
3585 			if (IS_ERR(context)) {
3586 				rc = PTR_ERR(context);
3587 				goto err_out1;
3588 			} else if (context)
3589 				conn->is_aapl = true;
3590 		}
3591 	}
3592 
3593 	rc = ksmbd_vfs_getattr(&path, &stat);
3594 	if (rc)
3595 		goto err_out1;
3596 
3597 	if (stat.result_mask & STATX_BTIME)
3598 		fp->create_time = ksmbd_UnixTimeToNT(stat.btime);
3599 	else
3600 		fp->create_time = ksmbd_UnixTimeToNT(stat.ctime);
3601 	if (req->FileAttributes || fp->f_ci->m_fattr == 0)
3602 		fp->f_ci->m_fattr =
3603 			cpu_to_le32(smb2_get_dos_mode(&stat, le32_to_cpu(req->FileAttributes)));
3604 
3605 	if (!created)
3606 		smb2_update_xattrs(tcon, &path, fp);
3607 	else
3608 		smb2_new_xattrs(tcon, &path, fp);
3609 
3610 	memcpy(fp->client_guid, conn->ClientGUID, SMB2_CLIENT_GUID_SIZE);
3611 
3612 	if (dh_info.type == DURABLE_REQ_V2 || dh_info.type == DURABLE_REQ) {
3613 		if (dh_info.type == DURABLE_REQ_V2 && dh_info.persistent &&
3614 		    test_share_config_flag(work->tcon->share_conf,
3615 					   KSMBD_SHARE_FLAG_CONTINUOUS_AVAILABILITY))
3616 			fp->is_persistent = true;
3617 		else
3618 			fp->is_durable = true;
3619 
3620 		if (dh_info.type == DURABLE_REQ_V2) {
3621 			memcpy(fp->create_guid, dh_info.CreateGuid,
3622 					SMB2_CREATE_GUID_SIZE);
3623 			if (dh_info.timeout)
3624 				fp->durable_timeout =
3625 					min_t(unsigned int, dh_info.timeout,
3626 					      DURABLE_HANDLE_MAX_TIMEOUT);
3627 			else
3628 				fp->durable_timeout = 60;
3629 		}
3630 	}
3631 
3632 reconnected_fp:
3633 	rsp->StructureSize = cpu_to_le16(89);
3634 	opinfo = opinfo_get(fp);
3635 	rsp->OplockLevel = opinfo != NULL ? opinfo->level : 0;
3636 	rsp->Flags = 0;
3637 	rsp->CreateAction = cpu_to_le32(file_info);
3638 	rsp->CreationTime = cpu_to_le64(fp->create_time);
3639 	time = ksmbd_UnixTimeToNT(stat.atime);
3640 	rsp->LastAccessTime = cpu_to_le64(time);
3641 	time = ksmbd_UnixTimeToNT(stat.mtime);
3642 	rsp->LastWriteTime = cpu_to_le64(time);
3643 	time = ksmbd_UnixTimeToNT(stat.ctime);
3644 	rsp->ChangeTime = cpu_to_le64(time);
3645 	rsp->AllocationSize = S_ISDIR(stat.mode) ? 0 :
3646 		cpu_to_le64(stat.blocks << 9);
3647 	rsp->EndofFile = S_ISDIR(stat.mode) ? 0 : cpu_to_le64(stat.size);
3648 	rsp->FileAttributes = fp->f_ci->m_fattr;
3649 
3650 	rsp->Reserved2 = 0;
3651 
3652 	rsp->PersistentFileId = fp->persistent_id;
3653 	rsp->VolatileFileId = fp->volatile_id;
3654 
3655 	rsp->CreateContextsOffset = 0;
3656 	rsp->CreateContextsLength = 0;
3657 	iov_len = offsetof(struct smb2_create_rsp, Buffer);
3658 
3659 	/* If lease is request send lease context response */
3660 	if (opinfo && opinfo->is_lease) {
3661 		struct create_context *lease_ccontext;
3662 
3663 		ksmbd_debug(SMB, "lease granted on(%s) lease state 0x%x\n",
3664 			    name, opinfo->o_lease->state);
3665 		rsp->OplockLevel = SMB2_OPLOCK_LEVEL_LEASE;
3666 
3667 		lease_ccontext = (struct create_context *)rsp->Buffer;
3668 		contxt_cnt++;
3669 		create_lease_buf(rsp->Buffer, opinfo->o_lease);
3670 		le32_add_cpu(&rsp->CreateContextsLength,
3671 			     conn->vals->create_lease_size);
3672 		iov_len += conn->vals->create_lease_size;
3673 		next_ptr = &lease_ccontext->Next;
3674 		next_off = conn->vals->create_lease_size;
3675 	}
3676 	opinfo_put(opinfo);
3677 
3678 	if (maximal_access_ctxt) {
3679 		struct create_context *mxac_ccontext;
3680 
3681 		if (maximal_access == 0)
3682 			ksmbd_vfs_query_maximal_access(idmap,
3683 						       path.dentry,
3684 						       &maximal_access);
3685 		mxac_ccontext = (struct create_context *)(rsp->Buffer +
3686 				le32_to_cpu(rsp->CreateContextsLength));
3687 		contxt_cnt++;
3688 		create_mxac_rsp_buf(rsp->Buffer +
3689 				le32_to_cpu(rsp->CreateContextsLength),
3690 				le32_to_cpu(maximal_access));
3691 		le32_add_cpu(&rsp->CreateContextsLength,
3692 			     conn->vals->create_mxac_size);
3693 		iov_len += conn->vals->create_mxac_size;
3694 		if (next_ptr)
3695 			*next_ptr = cpu_to_le32(next_off);
3696 		next_ptr = &mxac_ccontext->Next;
3697 		next_off = conn->vals->create_mxac_size;
3698 	}
3699 
3700 	if (query_disk_id) {
3701 		struct create_context *disk_id_ccontext;
3702 
3703 		disk_id_ccontext = (struct create_context *)(rsp->Buffer +
3704 				le32_to_cpu(rsp->CreateContextsLength));
3705 		contxt_cnt++;
3706 		create_disk_id_rsp_buf(rsp->Buffer +
3707 				le32_to_cpu(rsp->CreateContextsLength),
3708 				stat.ino, tcon->id);
3709 		le32_add_cpu(&rsp->CreateContextsLength,
3710 			     conn->vals->create_disk_id_size);
3711 		iov_len += conn->vals->create_disk_id_size;
3712 		if (next_ptr)
3713 			*next_ptr = cpu_to_le32(next_off);
3714 		next_ptr = &disk_id_ccontext->Next;
3715 		next_off = conn->vals->create_disk_id_size;
3716 	}
3717 
3718 	if (dh_info.type == DURABLE_REQ || dh_info.type == DURABLE_REQ_V2) {
3719 		struct create_context *durable_ccontext;
3720 
3721 		durable_ccontext = (struct create_context *)(rsp->Buffer +
3722 				le32_to_cpu(rsp->CreateContextsLength));
3723 		contxt_cnt++;
3724 		if (dh_info.type == DURABLE_REQ) {
3725 			create_durable_rsp_buf(rsp->Buffer +
3726 					le32_to_cpu(rsp->CreateContextsLength));
3727 			le32_add_cpu(&rsp->CreateContextsLength,
3728 					conn->vals->create_durable_size);
3729 			iov_len += conn->vals->create_durable_size;
3730 		} else {
3731 			create_durable_v2_rsp_buf(rsp->Buffer +
3732 					le32_to_cpu(rsp->CreateContextsLength),
3733 					fp);
3734 			le32_add_cpu(&rsp->CreateContextsLength,
3735 					conn->vals->create_durable_v2_size);
3736 			iov_len += conn->vals->create_durable_v2_size;
3737 		}
3738 
3739 		if (next_ptr)
3740 			*next_ptr = cpu_to_le32(next_off);
3741 		next_ptr = &durable_ccontext->Next;
3742 		next_off = conn->vals->create_durable_size;
3743 	}
3744 
3745 	if (posix_ctxt) {
3746 		contxt_cnt++;
3747 		create_posix_rsp_buf(rsp->Buffer +
3748 				le32_to_cpu(rsp->CreateContextsLength),
3749 				fp);
3750 		le32_add_cpu(&rsp->CreateContextsLength,
3751 			     conn->vals->create_posix_size);
3752 		iov_len += conn->vals->create_posix_size;
3753 		if (next_ptr)
3754 			*next_ptr = cpu_to_le32(next_off);
3755 	}
3756 
3757 	if (contxt_cnt > 0) {
3758 		rsp->CreateContextsOffset =
3759 			cpu_to_le32(offsetof(struct smb2_create_rsp, Buffer));
3760 	}
3761 
3762 err_out:
3763 	if (rc && (file_present || created))
3764 		path_put(&path);
3765 
3766 err_out1:
3767 	ksmbd_revert_fsids(work);
3768 
3769 err_out2:
3770 	if (!rc) {
3771 		rc = ksmbd_update_fstate(&work->sess->file_table, fp,
3772 					 FP_INITED);
3773 		if (!rc)
3774 			rc = ksmbd_iov_pin_rsp(work, (void *)rsp, iov_len);
3775 	}
3776 	if (rc) {
3777 		if (rc == -EINVAL)
3778 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
3779 		else if (rc == -EOPNOTSUPP)
3780 			rsp->hdr.Status = STATUS_NOT_SUPPORTED;
3781 		else if (rc == -EACCES || rc == -ESTALE || rc == -EXDEV)
3782 			rsp->hdr.Status = STATUS_ACCESS_DENIED;
3783 		else if (rc == -ENOENT)
3784 			rsp->hdr.Status = STATUS_OBJECT_NAME_INVALID;
3785 		else if (rc == -EPERM)
3786 			rsp->hdr.Status = STATUS_SHARING_VIOLATION;
3787 		else if (rc == -EBUSY)
3788 			rsp->hdr.Status = STATUS_DELETE_PENDING;
3789 		else if (rc == -EBADF)
3790 			rsp->hdr.Status = STATUS_OBJECT_NAME_NOT_FOUND;
3791 		else if (rc == -ENOEXEC)
3792 			rsp->hdr.Status = STATUS_DUPLICATE_OBJECTID;
3793 		else if (rc == -ENXIO)
3794 			rsp->hdr.Status = STATUS_NO_SUCH_DEVICE;
3795 		else if (rc == -EEXIST)
3796 			rsp->hdr.Status = STATUS_OBJECT_NAME_COLLISION;
3797 		else if (rc == -EMFILE)
3798 			rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
3799 		if (!rsp->hdr.Status)
3800 			rsp->hdr.Status = STATUS_UNEXPECTED_IO_ERROR;
3801 
3802 		if (fp)
3803 			ksmbd_fd_put(work, fp);
3804 		smb2_set_err_rsp(work);
3805 		ksmbd_debug(SMB, "Error response: %x\n", rsp->hdr.Status);
3806 	}
3807 
3808 	if (dh_info.reconnected) {
3809 		/*
3810 		 * If reconnect succeeded, fp was republished in the
3811 		 * session file table.  On a later error, ksmbd_fd_put()
3812 		 * above drops the session reference; drop the durable
3813 		 * lookup reference through the same session-aware path so
3814 		 * final close removes the volatile id before freeing fp.
3815 		 */
3816 		if (rc && fp == dh_info.fp)
3817 			ksmbd_fd_put(work, dh_info.fp);
3818 		else
3819 			ksmbd_put_durable_fd(dh_info.fp);
3820 	}
3821 
3822 	kfree(name);
3823 	kfree(lc);
3824 
3825 	return rc;
3826 }
3827 
3828 static int readdir_info_level_struct_sz(int info_level)
3829 {
3830 	switch (info_level) {
3831 	case FILE_FULL_DIRECTORY_INFORMATION:
3832 		return sizeof(FILE_FULL_DIRECTORY_INFO);
3833 	case FILE_BOTH_DIRECTORY_INFORMATION:
3834 		return sizeof(FILE_BOTH_DIRECTORY_INFO);
3835 	case FILE_DIRECTORY_INFORMATION:
3836 		return sizeof(FILE_DIRECTORY_INFO);
3837 	case FILE_NAMES_INFORMATION:
3838 		return sizeof(struct file_names_info);
3839 	case FILEID_FULL_DIRECTORY_INFORMATION:
3840 		return sizeof(FILE_ID_FULL_DIR_INFO);
3841 	case FILEID_BOTH_DIRECTORY_INFORMATION:
3842 		return sizeof(struct file_id_both_directory_info);
3843 	case SMB_FIND_FILE_POSIX_INFO:
3844 		return sizeof(struct smb2_posix_info);
3845 	default:
3846 		return -EOPNOTSUPP;
3847 	}
3848 }
3849 
3850 static int dentry_name(struct ksmbd_dir_info *d_info, int info_level)
3851 {
3852 	switch (info_level) {
3853 	case FILE_FULL_DIRECTORY_INFORMATION:
3854 	{
3855 		FILE_FULL_DIRECTORY_INFO *ffdinfo;
3856 
3857 		ffdinfo = (FILE_FULL_DIRECTORY_INFO *)d_info->rptr;
3858 		d_info->rptr += le32_to_cpu(ffdinfo->NextEntryOffset);
3859 		d_info->name = ffdinfo->FileName;
3860 		d_info->name_len = le32_to_cpu(ffdinfo->FileNameLength);
3861 		return 0;
3862 	}
3863 	case FILE_BOTH_DIRECTORY_INFORMATION:
3864 	{
3865 		FILE_BOTH_DIRECTORY_INFO *fbdinfo;
3866 
3867 		fbdinfo = (FILE_BOTH_DIRECTORY_INFO *)d_info->rptr;
3868 		d_info->rptr += le32_to_cpu(fbdinfo->NextEntryOffset);
3869 		d_info->name = fbdinfo->FileName;
3870 		d_info->name_len = le32_to_cpu(fbdinfo->FileNameLength);
3871 		return 0;
3872 	}
3873 	case FILE_DIRECTORY_INFORMATION:
3874 	{
3875 		FILE_DIRECTORY_INFO *fdinfo;
3876 
3877 		fdinfo = (FILE_DIRECTORY_INFO *)d_info->rptr;
3878 		d_info->rptr += le32_to_cpu(fdinfo->NextEntryOffset);
3879 		d_info->name = fdinfo->FileName;
3880 		d_info->name_len = le32_to_cpu(fdinfo->FileNameLength);
3881 		return 0;
3882 	}
3883 	case FILE_NAMES_INFORMATION:
3884 	{
3885 		struct file_names_info *fninfo;
3886 
3887 		fninfo = (struct file_names_info *)d_info->rptr;
3888 		d_info->rptr += le32_to_cpu(fninfo->NextEntryOffset);
3889 		d_info->name = fninfo->FileName;
3890 		d_info->name_len = le32_to_cpu(fninfo->FileNameLength);
3891 		return 0;
3892 	}
3893 	case FILEID_FULL_DIRECTORY_INFORMATION:
3894 	{
3895 		FILE_ID_FULL_DIR_INFO *dinfo;
3896 
3897 		dinfo = (FILE_ID_FULL_DIR_INFO *)d_info->rptr;
3898 		d_info->rptr += le32_to_cpu(dinfo->NextEntryOffset);
3899 		d_info->name = dinfo->FileName;
3900 		d_info->name_len = le32_to_cpu(dinfo->FileNameLength);
3901 		return 0;
3902 	}
3903 	case FILEID_BOTH_DIRECTORY_INFORMATION:
3904 	{
3905 		struct file_id_both_directory_info *fibdinfo;
3906 
3907 		fibdinfo = (struct file_id_both_directory_info *)d_info->rptr;
3908 		d_info->rptr += le32_to_cpu(fibdinfo->NextEntryOffset);
3909 		d_info->name = fibdinfo->FileName;
3910 		d_info->name_len = le32_to_cpu(fibdinfo->FileNameLength);
3911 		return 0;
3912 	}
3913 	case SMB_FIND_FILE_POSIX_INFO:
3914 	{
3915 		struct smb2_posix_info *posix_info;
3916 
3917 		posix_info = (struct smb2_posix_info *)d_info->rptr;
3918 		d_info->rptr += le32_to_cpu(posix_info->NextEntryOffset);
3919 		d_info->name = posix_info->name;
3920 		d_info->name_len = le32_to_cpu(posix_info->name_len);
3921 		return 0;
3922 	}
3923 	default:
3924 		return -EINVAL;
3925 	}
3926 }
3927 
3928 /**
3929  * smb2_populate_readdir_entry() - encode directory entry in smb2 response
3930  * buffer
3931  * @conn:	connection instance
3932  * @info_level:	smb information level
3933  * @d_info:	structure included variables for query dir
3934  * @ksmbd_kstat:	ksmbd wrapper of dirent stat information
3935  *
3936  * if directory has many entries, find first can't read it fully.
3937  * find next might be called multiple times to read remaining dir entries
3938  *
3939  * Return:	0 on success, otherwise error
3940  */
3941 static int smb2_populate_readdir_entry(struct ksmbd_conn *conn, int info_level,
3942 				       struct ksmbd_dir_info *d_info,
3943 				       struct ksmbd_kstat *ksmbd_kstat)
3944 {
3945 	int next_entry_offset = 0;
3946 	char *conv_name;
3947 	int conv_len;
3948 	void *kstat;
3949 	int struct_sz, rc = 0;
3950 
3951 	conv_name = ksmbd_convert_dir_info_name(d_info,
3952 						conn->local_nls,
3953 						&conv_len);
3954 	if (!conv_name)
3955 		return -ENOMEM;
3956 
3957 	/* Somehow the name has only terminating NULL bytes */
3958 	if (conv_len < 0) {
3959 		rc = -EINVAL;
3960 		goto free_conv_name;
3961 	}
3962 
3963 	struct_sz = readdir_info_level_struct_sz(info_level);
3964 	if (struct_sz == -EOPNOTSUPP) {
3965 		rc = -EINVAL;
3966 		goto free_conv_name;
3967 	}
3968 
3969 	struct_sz += conv_len;
3970 	next_entry_offset = ALIGN(struct_sz, KSMBD_DIR_INFO_ALIGNMENT);
3971 	d_info->last_entry_off_align = next_entry_offset - struct_sz;
3972 
3973 	if (next_entry_offset > d_info->out_buf_len) {
3974 		d_info->out_buf_len = 0;
3975 		rc = -ENOSPC;
3976 		goto free_conv_name;
3977 	}
3978 
3979 	kstat = d_info->wptr;
3980 	if (info_level != FILE_NAMES_INFORMATION)
3981 		kstat = ksmbd_vfs_init_kstat(&d_info->wptr, ksmbd_kstat);
3982 
3983 	switch (info_level) {
3984 	case FILE_FULL_DIRECTORY_INFORMATION:
3985 	{
3986 		FILE_FULL_DIRECTORY_INFO *ffdinfo;
3987 
3988 		ffdinfo = (FILE_FULL_DIRECTORY_INFO *)kstat;
3989 		ffdinfo->FileNameLength = cpu_to_le32(conv_len);
3990 		ffdinfo->EaSize =
3991 			smb2_get_reparse_tag_special_file(ksmbd_kstat->kstat->mode);
3992 		if (ffdinfo->EaSize)
3993 			ffdinfo->ExtFileAttributes = FILE_ATTRIBUTE_REPARSE_POINT_LE;
3994 		if (d_info->hide_dot_file && d_info->name[0] == '.')
3995 			ffdinfo->ExtFileAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
3996 		memcpy(ffdinfo->FileName, conv_name, conv_len);
3997 		ffdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
3998 		break;
3999 	}
4000 	case FILE_BOTH_DIRECTORY_INFORMATION:
4001 	{
4002 		FILE_BOTH_DIRECTORY_INFO *fbdinfo;
4003 
4004 		fbdinfo = (FILE_BOTH_DIRECTORY_INFO *)kstat;
4005 		fbdinfo->FileNameLength = cpu_to_le32(conv_len);
4006 		fbdinfo->EaSize =
4007 			smb2_get_reparse_tag_special_file(ksmbd_kstat->kstat->mode);
4008 		if (fbdinfo->EaSize)
4009 			fbdinfo->ExtFileAttributes = FILE_ATTRIBUTE_REPARSE_POINT_LE;
4010 		fbdinfo->ShortNameLength = 0;
4011 		fbdinfo->Reserved = 0;
4012 		if (d_info->hide_dot_file && d_info->name[0] == '.')
4013 			fbdinfo->ExtFileAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
4014 		memcpy(fbdinfo->FileName, conv_name, conv_len);
4015 		fbdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4016 		break;
4017 	}
4018 	case FILE_DIRECTORY_INFORMATION:
4019 	{
4020 		FILE_DIRECTORY_INFO *fdinfo;
4021 
4022 		fdinfo = (FILE_DIRECTORY_INFO *)kstat;
4023 		fdinfo->FileNameLength = cpu_to_le32(conv_len);
4024 		if (d_info->hide_dot_file && d_info->name[0] == '.')
4025 			fdinfo->ExtFileAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
4026 		memcpy(fdinfo->FileName, conv_name, conv_len);
4027 		fdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4028 		break;
4029 	}
4030 	case FILE_NAMES_INFORMATION:
4031 	{
4032 		struct file_names_info *fninfo;
4033 
4034 		fninfo = (struct file_names_info *)kstat;
4035 		fninfo->FileNameLength = cpu_to_le32(conv_len);
4036 		memcpy(fninfo->FileName, conv_name, conv_len);
4037 		fninfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4038 		break;
4039 	}
4040 	case FILEID_FULL_DIRECTORY_INFORMATION:
4041 	{
4042 		FILE_ID_FULL_DIR_INFO *dinfo;
4043 
4044 		dinfo = (FILE_ID_FULL_DIR_INFO *)kstat;
4045 		dinfo->FileNameLength = cpu_to_le32(conv_len);
4046 		dinfo->EaSize =
4047 			smb2_get_reparse_tag_special_file(ksmbd_kstat->kstat->mode);
4048 		if (dinfo->EaSize)
4049 			dinfo->ExtFileAttributes = FILE_ATTRIBUTE_REPARSE_POINT_LE;
4050 		dinfo->Reserved = 0;
4051 		if (conn->is_aapl)
4052 			dinfo->UniqueId = 0;
4053 		else
4054 			dinfo->UniqueId = cpu_to_le64(ksmbd_kstat->kstat->ino);
4055 		if (d_info->hide_dot_file && d_info->name[0] == '.')
4056 			dinfo->ExtFileAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
4057 		memcpy(dinfo->FileName, conv_name, conv_len);
4058 		dinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4059 		break;
4060 	}
4061 	case FILEID_BOTH_DIRECTORY_INFORMATION:
4062 	{
4063 		struct file_id_both_directory_info *fibdinfo;
4064 
4065 		fibdinfo = (struct file_id_both_directory_info *)kstat;
4066 		fibdinfo->FileNameLength = cpu_to_le32(conv_len);
4067 		fibdinfo->EaSize =
4068 			smb2_get_reparse_tag_special_file(ksmbd_kstat->kstat->mode);
4069 		if (fibdinfo->EaSize)
4070 			fibdinfo->ExtFileAttributes = FILE_ATTRIBUTE_REPARSE_POINT_LE;
4071 		if (conn->is_aapl)
4072 			fibdinfo->UniqueId = 0;
4073 		else
4074 			fibdinfo->UniqueId = cpu_to_le64(ksmbd_kstat->kstat->ino);
4075 		fibdinfo->ShortNameLength = 0;
4076 		fibdinfo->Reserved = 0;
4077 		fibdinfo->Reserved2 = cpu_to_le16(0);
4078 		if (d_info->hide_dot_file && d_info->name[0] == '.')
4079 			fibdinfo->ExtFileAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
4080 		memcpy(fibdinfo->FileName, conv_name, conv_len);
4081 		fibdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4082 		break;
4083 	}
4084 	case SMB_FIND_FILE_POSIX_INFO:
4085 	{
4086 		struct smb2_posix_info *posix_info;
4087 		u64 time;
4088 
4089 		posix_info = (struct smb2_posix_info *)kstat;
4090 		posix_info->Ignored = 0;
4091 		posix_info->CreationTime = cpu_to_le64(ksmbd_kstat->create_time);
4092 		time = ksmbd_UnixTimeToNT(ksmbd_kstat->kstat->ctime);
4093 		posix_info->ChangeTime = cpu_to_le64(time);
4094 		time = ksmbd_UnixTimeToNT(ksmbd_kstat->kstat->atime);
4095 		posix_info->LastAccessTime = cpu_to_le64(time);
4096 		time = ksmbd_UnixTimeToNT(ksmbd_kstat->kstat->mtime);
4097 		posix_info->LastWriteTime = cpu_to_le64(time);
4098 		posix_info->EndOfFile = cpu_to_le64(ksmbd_kstat->kstat->size);
4099 		posix_info->AllocationSize = cpu_to_le64(ksmbd_kstat->kstat->blocks << 9);
4100 		posix_info->DeviceId = cpu_to_le32(ksmbd_kstat->kstat->rdev);
4101 		posix_info->HardLinks = cpu_to_le32(ksmbd_kstat->kstat->nlink);
4102 		posix_info->Mode = cpu_to_le32(ksmbd_kstat->kstat->mode & 0777);
4103 		switch (ksmbd_kstat->kstat->mode & S_IFMT) {
4104 		case S_IFDIR:
4105 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_DIR << POSIX_FILETYPE_SHIFT);
4106 			break;
4107 		case S_IFLNK:
4108 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_SYMLINK << POSIX_FILETYPE_SHIFT);
4109 			break;
4110 		case S_IFCHR:
4111 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_CHARDEV << POSIX_FILETYPE_SHIFT);
4112 			break;
4113 		case S_IFBLK:
4114 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_BLKDEV << POSIX_FILETYPE_SHIFT);
4115 			break;
4116 		case S_IFIFO:
4117 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_FIFO << POSIX_FILETYPE_SHIFT);
4118 			break;
4119 		case S_IFSOCK:
4120 			posix_info->Mode |= cpu_to_le32(POSIX_TYPE_SOCKET << POSIX_FILETYPE_SHIFT);
4121 		}
4122 
4123 		posix_info->Inode = cpu_to_le64(ksmbd_kstat->kstat->ino);
4124 		posix_info->DosAttributes =
4125 			S_ISDIR(ksmbd_kstat->kstat->mode) ?
4126 				FILE_ATTRIBUTE_DIRECTORY_LE : FILE_ATTRIBUTE_ARCHIVE_LE;
4127 		if (d_info->hide_dot_file && d_info->name[0] == '.')
4128 			posix_info->DosAttributes |= FILE_ATTRIBUTE_HIDDEN_LE;
4129 		/*
4130 		 * SidBuffer(32) contain two sids(Domain sid(16), UNIX group sid(16)).
4131 		 * UNIX sid(16) = revision(1) + num_subauth(1) + authority(6) +
4132 		 *		  sub_auth(4 * 1(num_subauth)) + RID(4).
4133 		 */
4134 		id_to_sid(from_kuid_munged(&init_user_ns, ksmbd_kstat->kstat->uid),
4135 			  SIDUNIX_USER, (struct smb_sid *)&posix_info->SidBuffer[0]);
4136 		id_to_sid(from_kgid_munged(&init_user_ns, ksmbd_kstat->kstat->gid),
4137 			  SIDUNIX_GROUP, (struct smb_sid *)&posix_info->SidBuffer[16]);
4138 		memcpy(posix_info->name, conv_name, conv_len);
4139 		posix_info->name_len = cpu_to_le32(conv_len);
4140 		posix_info->NextEntryOffset = cpu_to_le32(next_entry_offset);
4141 		break;
4142 	}
4143 
4144 	} /* switch (info_level) */
4145 
4146 	d_info->last_entry_offset = d_info->data_count;
4147 	d_info->data_count += next_entry_offset;
4148 	d_info->out_buf_len -= next_entry_offset;
4149 	d_info->wptr += next_entry_offset;
4150 
4151 	ksmbd_debug(SMB,
4152 		    "info_level : %d, buf_len :%d, next_offset : %d, data_count : %d\n",
4153 		    info_level, d_info->out_buf_len,
4154 		    next_entry_offset, d_info->data_count);
4155 
4156 free_conv_name:
4157 	kfree(conv_name);
4158 	return rc;
4159 }
4160 
4161 struct smb2_query_dir_private {
4162 	struct ksmbd_work	*work;
4163 	char			*search_pattern;
4164 	struct ksmbd_file	*dir_fp;
4165 
4166 	struct ksmbd_dir_info	*d_info;
4167 	int			info_level;
4168 };
4169 
4170 static int process_query_dir_entries(struct smb2_query_dir_private *priv)
4171 {
4172 	struct mnt_idmap	*idmap = file_mnt_idmap(priv->dir_fp->filp);
4173 	struct kstat		kstat;
4174 	struct ksmbd_kstat	ksmbd_kstat;
4175 	int			rc;
4176 	int			i;
4177 
4178 	for (i = 0; i < priv->d_info->num_entry; i++) {
4179 		struct dentry *dent;
4180 
4181 		if (dentry_name(priv->d_info, priv->info_level))
4182 			return -EINVAL;
4183 
4184 		dent = lookup_one_unlocked(idmap,
4185 					   &QSTR_LEN(priv->d_info->name,
4186 						     priv->d_info->name_len),
4187 					   priv->dir_fp->filp->f_path.dentry);
4188 
4189 		if (IS_ERR(dent)) {
4190 			ksmbd_debug(SMB, "Cannot lookup `%s' [%ld]\n",
4191 				    priv->d_info->name,
4192 				    PTR_ERR(dent));
4193 			continue;
4194 		}
4195 		if (unlikely(d_is_negative(dent))) {
4196 			dput(dent);
4197 			ksmbd_debug(SMB, "Negative dentry `%s'\n",
4198 				    priv->d_info->name);
4199 			continue;
4200 		}
4201 
4202 		ksmbd_kstat.kstat = &kstat;
4203 		if (priv->info_level != FILE_NAMES_INFORMATION) {
4204 			rc = ksmbd_vfs_fill_dentry_attrs(priv->work,
4205 							 idmap,
4206 							 dent,
4207 							 &ksmbd_kstat);
4208 			if (rc) {
4209 				dput(dent);
4210 				continue;
4211 			}
4212 		}
4213 
4214 		rc = smb2_populate_readdir_entry(priv->work->conn,
4215 						 priv->info_level,
4216 						 priv->d_info,
4217 						 &ksmbd_kstat);
4218 		dput(dent);
4219 		if (rc)
4220 			return rc;
4221 	}
4222 	return 0;
4223 }
4224 
4225 static int reserve_populate_dentry(struct ksmbd_dir_info *d_info,
4226 				   int info_level)
4227 {
4228 	int struct_sz;
4229 	int conv_len;
4230 	int next_entry_offset;
4231 
4232 	struct_sz = readdir_info_level_struct_sz(info_level);
4233 	if (struct_sz == -EOPNOTSUPP)
4234 		return -EOPNOTSUPP;
4235 
4236 	conv_len = (d_info->name_len + 1) * 2;
4237 	next_entry_offset = ALIGN(struct_sz + conv_len,
4238 				  KSMBD_DIR_INFO_ALIGNMENT);
4239 
4240 	if (next_entry_offset > d_info->out_buf_len) {
4241 		d_info->out_buf_len = 0;
4242 		return -ENOSPC;
4243 	}
4244 
4245 	switch (info_level) {
4246 	case FILE_FULL_DIRECTORY_INFORMATION:
4247 	{
4248 		FILE_FULL_DIRECTORY_INFO *ffdinfo;
4249 
4250 		ffdinfo = (FILE_FULL_DIRECTORY_INFO *)d_info->wptr;
4251 		memcpy(ffdinfo->FileName, d_info->name, d_info->name_len);
4252 		ffdinfo->FileName[d_info->name_len] = 0x00;
4253 		ffdinfo->FileNameLength = cpu_to_le32(d_info->name_len);
4254 		ffdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4255 		break;
4256 	}
4257 	case FILE_BOTH_DIRECTORY_INFORMATION:
4258 	{
4259 		FILE_BOTH_DIRECTORY_INFO *fbdinfo;
4260 
4261 		fbdinfo = (FILE_BOTH_DIRECTORY_INFO *)d_info->wptr;
4262 		memcpy(fbdinfo->FileName, d_info->name, d_info->name_len);
4263 		fbdinfo->FileName[d_info->name_len] = 0x00;
4264 		fbdinfo->FileNameLength = cpu_to_le32(d_info->name_len);
4265 		fbdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4266 		break;
4267 	}
4268 	case FILE_DIRECTORY_INFORMATION:
4269 	{
4270 		FILE_DIRECTORY_INFO *fdinfo;
4271 
4272 		fdinfo = (FILE_DIRECTORY_INFO *)d_info->wptr;
4273 		memcpy(fdinfo->FileName, d_info->name, d_info->name_len);
4274 		fdinfo->FileName[d_info->name_len] = 0x00;
4275 		fdinfo->FileNameLength = cpu_to_le32(d_info->name_len);
4276 		fdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4277 		break;
4278 	}
4279 	case FILE_NAMES_INFORMATION:
4280 	{
4281 		struct file_names_info *fninfo;
4282 
4283 		fninfo = (struct file_names_info *)d_info->wptr;
4284 		memcpy(fninfo->FileName, d_info->name, d_info->name_len);
4285 		fninfo->FileName[d_info->name_len] = 0x00;
4286 		fninfo->FileNameLength = cpu_to_le32(d_info->name_len);
4287 		fninfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4288 		break;
4289 	}
4290 	case FILEID_FULL_DIRECTORY_INFORMATION:
4291 	{
4292 		FILE_ID_FULL_DIR_INFO *dinfo;
4293 
4294 		dinfo = (FILE_ID_FULL_DIR_INFO *)d_info->wptr;
4295 		memcpy(dinfo->FileName, d_info->name, d_info->name_len);
4296 		dinfo->FileName[d_info->name_len] = 0x00;
4297 		dinfo->FileNameLength = cpu_to_le32(d_info->name_len);
4298 		dinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4299 		break;
4300 	}
4301 	case FILEID_BOTH_DIRECTORY_INFORMATION:
4302 	{
4303 		struct file_id_both_directory_info *fibdinfo;
4304 
4305 		fibdinfo = (struct file_id_both_directory_info *)d_info->wptr;
4306 		memcpy(fibdinfo->FileName, d_info->name, d_info->name_len);
4307 		fibdinfo->FileName[d_info->name_len] = 0x00;
4308 		fibdinfo->FileNameLength = cpu_to_le32(d_info->name_len);
4309 		fibdinfo->NextEntryOffset = cpu_to_le32(next_entry_offset);
4310 		break;
4311 	}
4312 	case SMB_FIND_FILE_POSIX_INFO:
4313 	{
4314 		struct smb2_posix_info *posix_info;
4315 
4316 		posix_info = (struct smb2_posix_info *)d_info->wptr;
4317 		memcpy(posix_info->name, d_info->name, d_info->name_len);
4318 		posix_info->name[d_info->name_len] = 0x00;
4319 		posix_info->name_len = cpu_to_le32(d_info->name_len);
4320 		posix_info->NextEntryOffset =
4321 			cpu_to_le32(next_entry_offset);
4322 		break;
4323 	}
4324 	} /* switch (info_level) */
4325 
4326 	d_info->num_entry++;
4327 	d_info->out_buf_len -= next_entry_offset;
4328 	d_info->wptr += next_entry_offset;
4329 	return 0;
4330 }
4331 
4332 static bool __query_dir(struct dir_context *ctx, const char *name, int namlen,
4333 		       loff_t offset, u64 ino, unsigned int d_type)
4334 {
4335 	struct ksmbd_readdir_data	*buf;
4336 	struct smb2_query_dir_private	*priv;
4337 	struct ksmbd_dir_info		*d_info;
4338 	int				rc;
4339 
4340 	buf	= container_of(ctx, struct ksmbd_readdir_data, ctx);
4341 	priv	= buf->private;
4342 	d_info	= priv->d_info;
4343 
4344 	/* dot and dotdot entries are already reserved */
4345 	if (!strcmp(".", name) || !strcmp("..", name))
4346 		return true;
4347 	d_info->num_scan++;
4348 	if (ksmbd_share_veto_filename(priv->work->tcon->share_conf, name))
4349 		return true;
4350 	if (!match_pattern(name, namlen, priv->search_pattern))
4351 		return true;
4352 
4353 	d_info->name		= name;
4354 	d_info->name_len	= namlen;
4355 	rc = reserve_populate_dentry(d_info, priv->info_level);
4356 	if (rc)
4357 		return false;
4358 	if (d_info->flags & SMB2_RETURN_SINGLE_ENTRY)
4359 		d_info->out_buf_len = 0;
4360 	return true;
4361 }
4362 
4363 static int verify_info_level(int info_level)
4364 {
4365 	switch (info_level) {
4366 	case FILE_FULL_DIRECTORY_INFORMATION:
4367 	case FILE_BOTH_DIRECTORY_INFORMATION:
4368 	case FILE_DIRECTORY_INFORMATION:
4369 	case FILE_NAMES_INFORMATION:
4370 	case FILEID_FULL_DIRECTORY_INFORMATION:
4371 	case FILEID_BOTH_DIRECTORY_INFORMATION:
4372 	case SMB_FIND_FILE_POSIX_INFO:
4373 		break;
4374 	default:
4375 		return -EOPNOTSUPP;
4376 	}
4377 
4378 	return 0;
4379 }
4380 
4381 static int smb2_resp_buf_len(struct ksmbd_work *work, unsigned short hdr2_len)
4382 {
4383 	int free_len;
4384 
4385 	free_len = (int)(work->response_sz -
4386 		(get_rfc1002_len(work->response_buf) + 4)) - hdr2_len;
4387 	return free_len;
4388 }
4389 
4390 static int smb2_calc_max_out_buf_len(struct ksmbd_work *work,
4391 				     unsigned short hdr2_len,
4392 				     unsigned int out_buf_len)
4393 {
4394 	int free_len;
4395 
4396 	if (out_buf_len > work->conn->vals->max_trans_size)
4397 		return -EINVAL;
4398 
4399 	free_len = smb2_resp_buf_len(work, hdr2_len);
4400 	if (free_len < 0)
4401 		return -EINVAL;
4402 
4403 	return min_t(int, out_buf_len, free_len);
4404 }
4405 
4406 int smb2_query_dir(struct ksmbd_work *work)
4407 {
4408 	struct ksmbd_conn *conn = work->conn;
4409 	struct smb2_query_directory_req *req;
4410 	struct smb2_query_directory_rsp *rsp;
4411 	struct ksmbd_share_config *share = work->tcon->share_conf;
4412 	struct ksmbd_file *dir_fp = NULL;
4413 	struct ksmbd_dir_info d_info;
4414 	int rc = 0;
4415 	char *srch_ptr = NULL;
4416 	unsigned char srch_flag;
4417 	int buffer_sz;
4418 	struct smb2_query_dir_private query_dir_private = {NULL, };
4419 
4420 	ksmbd_debug(SMB, "Received smb2 query directory request\n");
4421 
4422 	WORK_BUFFERS(work, req, rsp);
4423 
4424 	if (ksmbd_override_fsids(work)) {
4425 		rsp->hdr.Status = STATUS_NO_MEMORY;
4426 		smb2_set_err_rsp(work);
4427 		return -ENOMEM;
4428 	}
4429 
4430 	rc = verify_info_level(req->FileInformationClass);
4431 	if (rc) {
4432 		rc = -EFAULT;
4433 		goto err_out2;
4434 	}
4435 
4436 	dir_fp = ksmbd_lookup_fd_slow(work, req->VolatileFileId, req->PersistentFileId);
4437 	if (!dir_fp) {
4438 		rc = -EBADF;
4439 		goto err_out2;
4440 	}
4441 
4442 	if (!(dir_fp->daccess & FILE_LIST_DIRECTORY_LE) ||
4443 	    inode_permission(file_mnt_idmap(dir_fp->filp),
4444 			     file_inode(dir_fp->filp),
4445 			     MAY_READ | MAY_EXEC)) {
4446 		pr_err("no right to enumerate directory (%pD)\n", dir_fp->filp);
4447 		rc = -EACCES;
4448 		goto err_out2;
4449 	}
4450 
4451 	if (!S_ISDIR(file_inode(dir_fp->filp)->i_mode)) {
4452 		pr_err("can't do query dir for a file\n");
4453 		rc = -EINVAL;
4454 		goto err_out2;
4455 	}
4456 
4457 	srch_flag = req->Flags;
4458 	srch_ptr = smb_strndup_from_utf16((char *)req + le16_to_cpu(req->FileNameOffset),
4459 					  le16_to_cpu(req->FileNameLength), 1,
4460 					  conn->local_nls);
4461 	if (IS_ERR(srch_ptr)) {
4462 		ksmbd_debug(SMB, "Search Pattern not found\n");
4463 		rc = -EINVAL;
4464 		goto err_out2;
4465 	} else {
4466 		ksmbd_debug(SMB, "Search pattern is %s\n", srch_ptr);
4467 	}
4468 
4469 	if (srch_flag & SMB2_REOPEN || srch_flag & SMB2_RESTART_SCANS) {
4470 		ksmbd_debug(SMB, "Restart directory scan\n");
4471 		generic_file_llseek(dir_fp->filp, 0, SEEK_SET);
4472 	}
4473 
4474 	memset(&d_info, 0, sizeof(struct ksmbd_dir_info));
4475 	d_info.wptr = (char *)rsp->Buffer;
4476 	d_info.rptr = (char *)rsp->Buffer;
4477 	d_info.out_buf_len =
4478 		smb2_calc_max_out_buf_len(work,
4479 				offsetof(struct smb2_query_directory_rsp, Buffer),
4480 				le32_to_cpu(req->OutputBufferLength));
4481 	if (d_info.out_buf_len < 0) {
4482 		rc = -EINVAL;
4483 		goto err_out;
4484 	}
4485 	d_info.flags = srch_flag;
4486 
4487 	/*
4488 	 * reserve dot and dotdot entries in head of buffer
4489 	 * in first response
4490 	 */
4491 	rc = ksmbd_populate_dot_dotdot_entries(work, req->FileInformationClass,
4492 					       dir_fp, &d_info, srch_ptr,
4493 					       smb2_populate_readdir_entry);
4494 	if (rc == -ENOSPC)
4495 		rc = 0;
4496 	else if (rc)
4497 		goto err_out;
4498 
4499 	if (test_share_config_flag(share, KSMBD_SHARE_FLAG_HIDE_DOT_FILES))
4500 		d_info.hide_dot_file = true;
4501 
4502 	buffer_sz				= d_info.out_buf_len;
4503 	d_info.rptr				= d_info.wptr;
4504 	query_dir_private.work			= work;
4505 	query_dir_private.search_pattern	= srch_ptr;
4506 	query_dir_private.dir_fp		= dir_fp;
4507 	query_dir_private.d_info		= &d_info;
4508 	query_dir_private.info_level		= req->FileInformationClass;
4509 	dir_fp->readdir_data.private		= &query_dir_private;
4510 	set_ctx_actor(&dir_fp->readdir_data.ctx, __query_dir);
4511 again:
4512 	d_info.num_scan = 0;
4513 	rc = iterate_dir(dir_fp->filp, &dir_fp->readdir_data.ctx);
4514 	/*
4515 	 * num_entry can be 0 if the directory iteration stops before reaching
4516 	 * the end of the directory and no file is matched with the search
4517 	 * pattern.
4518 	 */
4519 	if (rc >= 0 && !d_info.num_entry && d_info.num_scan &&
4520 	    d_info.out_buf_len > 0)
4521 		goto again;
4522 	/*
4523 	 * req->OutputBufferLength is too small to contain even one entry.
4524 	 * In this case, it immediately returns OutputBufferLength 0 to client.
4525 	 */
4526 	if (!d_info.out_buf_len && !d_info.num_entry)
4527 		goto no_buf_len;
4528 	if (rc > 0 || rc == -ENOSPC)
4529 		rc = 0;
4530 	else if (rc)
4531 		goto err_out;
4532 
4533 	d_info.wptr = d_info.rptr;
4534 	d_info.out_buf_len = buffer_sz;
4535 	rc = process_query_dir_entries(&query_dir_private);
4536 	if (rc)
4537 		goto err_out;
4538 
4539 	if (!d_info.data_count && d_info.out_buf_len >= 0) {
4540 		if (srch_flag & SMB2_RETURN_SINGLE_ENTRY && !is_asterisk(srch_ptr)) {
4541 			rsp->hdr.Status = STATUS_NO_SUCH_FILE;
4542 		} else {
4543 			dir_fp->dot_dotdot[0] = dir_fp->dot_dotdot[1] = 0;
4544 			rsp->hdr.Status = STATUS_NO_MORE_FILES;
4545 		}
4546 		rsp->StructureSize = cpu_to_le16(9);
4547 		rsp->OutputBufferOffset = cpu_to_le16(0);
4548 		rsp->OutputBufferLength = cpu_to_le32(0);
4549 		rsp->Buffer[0] = 0;
4550 		rc = ksmbd_iov_pin_rsp(work, (void *)rsp,
4551 				       offsetof(struct smb2_query_directory_rsp, Buffer)
4552 				       + 1);
4553 		if (rc)
4554 			goto err_out;
4555 	} else {
4556 no_buf_len:
4557 		((FILE_DIRECTORY_INFO *)
4558 		((char *)rsp->Buffer + d_info.last_entry_offset))
4559 		->NextEntryOffset = 0;
4560 		if (d_info.data_count >= d_info.last_entry_off_align)
4561 			d_info.data_count -= d_info.last_entry_off_align;
4562 
4563 		rsp->StructureSize = cpu_to_le16(9);
4564 		rsp->OutputBufferOffset = cpu_to_le16(72);
4565 		rsp->OutputBufferLength = cpu_to_le32(d_info.data_count);
4566 		rc = ksmbd_iov_pin_rsp(work, (void *)rsp,
4567 				       offsetof(struct smb2_query_directory_rsp, Buffer) +
4568 				       d_info.data_count);
4569 		if (rc)
4570 			goto err_out;
4571 	}
4572 
4573 	kfree(srch_ptr);
4574 	ksmbd_fd_put(work, dir_fp);
4575 	ksmbd_revert_fsids(work);
4576 	return 0;
4577 
4578 err_out:
4579 	pr_err("error while processing smb2 query dir rc = %d\n", rc);
4580 	kfree(srch_ptr);
4581 
4582 err_out2:
4583 	if (rc == -EINVAL)
4584 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
4585 	else if (rc == -EACCES)
4586 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
4587 	else if (rc == -ENOENT)
4588 		rsp->hdr.Status = STATUS_NO_SUCH_FILE;
4589 	else if (rc == -EBADF)
4590 		rsp->hdr.Status = STATUS_FILE_CLOSED;
4591 	else if (rc == -ENOMEM)
4592 		rsp->hdr.Status = STATUS_NO_MEMORY;
4593 	else if (rc == -EFAULT)
4594 		rsp->hdr.Status = STATUS_INVALID_INFO_CLASS;
4595 	else if (rc == -EIO)
4596 		rsp->hdr.Status = STATUS_FILE_CORRUPT_ERROR;
4597 	if (!rsp->hdr.Status)
4598 		rsp->hdr.Status = STATUS_UNEXPECTED_IO_ERROR;
4599 
4600 	smb2_set_err_rsp(work);
4601 	ksmbd_fd_put(work, dir_fp);
4602 	ksmbd_revert_fsids(work);
4603 	return rc;
4604 }
4605 
4606 /**
4607  * buffer_check_err() - helper function to check buffer errors
4608  * @reqOutputBufferLength:	max buffer length expected in command response
4609  * @rsp:		query info response buffer contains output buffer length
4610  * @rsp_org:		base response buffer pointer in case of chained response
4611  *
4612  * Return:	0 on success, otherwise error
4613  */
4614 static int buffer_check_err(int reqOutputBufferLength,
4615 			    struct smb2_query_info_rsp *rsp,
4616 			    void *rsp_org)
4617 {
4618 	if (reqOutputBufferLength < le32_to_cpu(rsp->OutputBufferLength)) {
4619 		pr_err("Invalid Buffer Size Requested\n");
4620 		rsp->hdr.Status = STATUS_INFO_LENGTH_MISMATCH;
4621 		*(__be32 *)rsp_org = cpu_to_be32(sizeof(struct smb2_hdr));
4622 		return -EINVAL;
4623 	}
4624 	return 0;
4625 }
4626 
4627 static void get_standard_info_pipe(struct smb2_query_info_rsp *rsp,
4628 				   void *rsp_org)
4629 {
4630 	struct smb2_file_standard_info *sinfo;
4631 
4632 	sinfo = (struct smb2_file_standard_info *)rsp->Buffer;
4633 
4634 	sinfo->AllocationSize = cpu_to_le64(4096);
4635 	sinfo->EndOfFile = cpu_to_le64(0);
4636 	sinfo->NumberOfLinks = cpu_to_le32(1);
4637 	sinfo->DeletePending = 1;
4638 	sinfo->Directory = 0;
4639 	rsp->OutputBufferLength =
4640 		cpu_to_le32(sizeof(struct smb2_file_standard_info));
4641 }
4642 
4643 static void get_internal_info_pipe(struct smb2_query_info_rsp *rsp, u64 num,
4644 				   void *rsp_org)
4645 {
4646 	struct smb2_file_internal_info *file_info;
4647 
4648 	file_info = (struct smb2_file_internal_info *)rsp->Buffer;
4649 
4650 	/* any unique number */
4651 	file_info->IndexNumber = cpu_to_le64(num | (1ULL << 63));
4652 	rsp->OutputBufferLength =
4653 		cpu_to_le32(sizeof(struct smb2_file_internal_info));
4654 }
4655 
4656 static int smb2_get_info_file_pipe(struct ksmbd_session *sess,
4657 				   struct smb2_query_info_req *req,
4658 				   struct smb2_query_info_rsp *rsp,
4659 				   void *rsp_org)
4660 {
4661 	u64 id;
4662 	int rc;
4663 
4664 	/*
4665 	 * Windows can sometime send query file info request on
4666 	 * pipe without opening it, checking error condition here
4667 	 */
4668 	id = req->VolatileFileId;
4669 
4670 	lockdep_assert_not_held(&sess->rpc_lock);
4671 
4672 	down_read(&sess->rpc_lock);
4673 	if (!ksmbd_session_rpc_method(sess, id)) {
4674 		up_read(&sess->rpc_lock);
4675 		return -ENOENT;
4676 	}
4677 	up_read(&sess->rpc_lock);
4678 
4679 	ksmbd_debug(SMB, "FileInfoClass %u, FileId 0x%llx\n",
4680 		    req->FileInfoClass, req->VolatileFileId);
4681 
4682 	switch (req->FileInfoClass) {
4683 	case FILE_STANDARD_INFORMATION:
4684 		get_standard_info_pipe(rsp, rsp_org);
4685 		rc = buffer_check_err(le32_to_cpu(req->OutputBufferLength),
4686 				      rsp, rsp_org);
4687 		break;
4688 	case FILE_INTERNAL_INFORMATION:
4689 		get_internal_info_pipe(rsp, id, rsp_org);
4690 		rc = buffer_check_err(le32_to_cpu(req->OutputBufferLength),
4691 				      rsp, rsp_org);
4692 		break;
4693 	default:
4694 		ksmbd_debug(SMB, "smb2_info_file_pipe for %u not supported\n",
4695 			    req->FileInfoClass);
4696 		rc = -EOPNOTSUPP;
4697 	}
4698 	return rc;
4699 }
4700 
4701 /**
4702  * smb2_get_ea() - handler for smb2 get extended attribute command
4703  * @work:	smb work containing query info command buffer
4704  * @fp:		ksmbd_file pointer
4705  * @req:	get extended attribute request
4706  * @rsp:	response buffer pointer
4707  * @rsp_org:	base response buffer pointer in case of chained response
4708  *
4709  * Return:	0 on success, otherwise error
4710  */
4711 static int smb2_get_ea(struct ksmbd_work *work, struct ksmbd_file *fp,
4712 		       struct smb2_query_info_req *req,
4713 		       struct smb2_query_info_rsp *rsp, void *rsp_org)
4714 {
4715 	struct smb2_ea_info *eainfo, *prev_eainfo;
4716 	char *name, *ptr, *xattr_list = NULL, *buf;
4717 	int rc, name_len, value_len, xattr_list_len, idx;
4718 	ssize_t buf_free_len, alignment_bytes, next_offset, rsp_data_cnt = 0;
4719 	struct smb2_ea_info_req *ea_req = NULL;
4720 	const struct path *path;
4721 	struct mnt_idmap *idmap = file_mnt_idmap(fp->filp);
4722 
4723 	if (!(fp->daccess & FILE_READ_EA_LE)) {
4724 		pr_err("Not permitted to read ext attr : 0x%x\n",
4725 		       fp->daccess);
4726 		return -EACCES;
4727 	}
4728 
4729 	path = &fp->filp->f_path;
4730 	/* single EA entry is requested with given user.* name */
4731 	if (req->InputBufferLength) {
4732 		if (le32_to_cpu(req->InputBufferLength) <=
4733 		    sizeof(struct smb2_ea_info_req))
4734 			return -EINVAL;
4735 
4736 		ea_req = (struct smb2_ea_info_req *)((char *)req +
4737 						     le16_to_cpu(req->InputBufferOffset));
4738 
4739 		if (le32_to_cpu(req->InputBufferLength) <
4740 		    offsetof(struct smb2_ea_info_req, name) +
4741 		    ea_req->EaNameLength)
4742 			return -EINVAL;
4743 	} else {
4744 		/* need to send all EAs, if no specific EA is requested*/
4745 		if (le32_to_cpu(req->Flags) & SL_RETURN_SINGLE_ENTRY)
4746 			ksmbd_debug(SMB,
4747 				    "All EAs are requested but need to send single EA entry in rsp flags 0x%x\n",
4748 				    le32_to_cpu(req->Flags));
4749 	}
4750 
4751 	buf_free_len =
4752 		smb2_calc_max_out_buf_len(work,
4753 				offsetof(struct smb2_query_info_rsp, Buffer),
4754 				le32_to_cpu(req->OutputBufferLength));
4755 	if (buf_free_len < 0)
4756 		return -EINVAL;
4757 
4758 	rc = ksmbd_vfs_listxattr(path->dentry, &xattr_list);
4759 	if (rc < 0) {
4760 		rsp->hdr.Status = STATUS_INVALID_HANDLE;
4761 		goto out;
4762 	} else if (!rc) { /* there is no EA in the file */
4763 		ksmbd_debug(SMB, "no ea data in the file\n");
4764 		goto done;
4765 	}
4766 	xattr_list_len = rc;
4767 
4768 	ptr = (char *)rsp->Buffer;
4769 	eainfo = (struct smb2_ea_info *)ptr;
4770 	prev_eainfo = eainfo;
4771 	idx = 0;
4772 
4773 	while (idx < xattr_list_len) {
4774 		name = xattr_list + idx;
4775 		name_len = strlen(name);
4776 
4777 		ksmbd_debug(SMB, "%s, len %d\n", name, name_len);
4778 		idx += name_len + 1;
4779 
4780 		/*
4781 		 * CIFS does not support EA other than user.* namespace,
4782 		 * still keep the framework generic, to list other attrs
4783 		 * in future.
4784 		 */
4785 		if (strncmp(name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN))
4786 			continue;
4787 
4788 		if (!strncmp(&name[XATTR_USER_PREFIX_LEN], STREAM_PREFIX,
4789 			     STREAM_PREFIX_LEN))
4790 			continue;
4791 
4792 		if (req->InputBufferLength &&
4793 		    strncmp(&name[XATTR_USER_PREFIX_LEN], ea_req->name,
4794 			    ea_req->EaNameLength))
4795 			continue;
4796 
4797 		if (!strncmp(&name[XATTR_USER_PREFIX_LEN],
4798 			     DOS_ATTRIBUTE_PREFIX, DOS_ATTRIBUTE_PREFIX_LEN))
4799 			continue;
4800 
4801 		if (!strncmp(name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN))
4802 			name_len -= XATTR_USER_PREFIX_LEN;
4803 
4804 		ptr = eainfo->name + name_len + 1;
4805 		buf_free_len -= (offsetof(struct smb2_ea_info, name) +
4806 				name_len + 1);
4807 		/* bailout if xattr can't fit in buf_free_len */
4808 		value_len = ksmbd_vfs_getxattr(idmap, path->dentry,
4809 					       name, &buf);
4810 		if (value_len <= 0) {
4811 			rc = -ENOENT;
4812 			rsp->hdr.Status = STATUS_INVALID_HANDLE;
4813 			goto out;
4814 		}
4815 
4816 		buf_free_len -= value_len;
4817 		if (buf_free_len < 0) {
4818 			kfree(buf);
4819 			break;
4820 		}
4821 
4822 		memcpy(ptr, buf, value_len);
4823 		kfree(buf);
4824 
4825 		ptr += value_len;
4826 		eainfo->Flags = 0;
4827 		eainfo->EaNameLength = name_len;
4828 
4829 		if (!strncmp(name, XATTR_USER_PREFIX, XATTR_USER_PREFIX_LEN))
4830 			memcpy(eainfo->name, &name[XATTR_USER_PREFIX_LEN],
4831 			       name_len);
4832 		else
4833 			memcpy(eainfo->name, name, name_len);
4834 
4835 		eainfo->name[name_len] = '\0';
4836 		eainfo->EaValueLength = cpu_to_le16(value_len);
4837 		next_offset = offsetof(struct smb2_ea_info, name) +
4838 			name_len + 1 + value_len;
4839 
4840 		/* align next xattr entry at 4 byte bundary */
4841 		alignment_bytes = ((next_offset + 3) & ~3) - next_offset;
4842 		if (alignment_bytes) {
4843 			if (buf_free_len < alignment_bytes)
4844 				break;
4845 			memset(ptr, '\0', alignment_bytes);
4846 			ptr += alignment_bytes;
4847 			next_offset += alignment_bytes;
4848 			buf_free_len -= alignment_bytes;
4849 		}
4850 		eainfo->NextEntryOffset = cpu_to_le32(next_offset);
4851 		prev_eainfo = eainfo;
4852 		eainfo = (struct smb2_ea_info *)ptr;
4853 		rsp_data_cnt += next_offset;
4854 
4855 		if (req->InputBufferLength) {
4856 			ksmbd_debug(SMB, "single entry requested\n");
4857 			break;
4858 		}
4859 	}
4860 
4861 	/* no more ea entries */
4862 	prev_eainfo->NextEntryOffset = 0;
4863 done:
4864 	rc = 0;
4865 	if (rsp_data_cnt == 0)
4866 		rsp->hdr.Status = STATUS_NO_EAS_ON_FILE;
4867 	rsp->OutputBufferLength = cpu_to_le32(rsp_data_cnt);
4868 out:
4869 	kvfree(xattr_list);
4870 	return rc;
4871 }
4872 
4873 static void get_file_access_info(struct smb2_query_info_rsp *rsp,
4874 				 struct ksmbd_file *fp, void *rsp_org)
4875 {
4876 	struct smb2_file_access_info *file_info;
4877 
4878 	file_info = (struct smb2_file_access_info *)rsp->Buffer;
4879 	file_info->AccessFlags = fp->daccess;
4880 	rsp->OutputBufferLength =
4881 		cpu_to_le32(sizeof(struct smb2_file_access_info));
4882 }
4883 
4884 static int get_file_basic_info(struct smb2_query_info_rsp *rsp,
4885 			       struct ksmbd_file *fp, void *rsp_org)
4886 {
4887 	struct file_basic_info *basic_info;
4888 	struct kstat stat;
4889 	u64 time;
4890 	int ret;
4891 
4892 	if (!(fp->daccess & FILE_READ_ATTRIBUTES_LE)) {
4893 		pr_err("no right to read the attributes : 0x%x\n",
4894 		       fp->daccess);
4895 		return -EACCES;
4896 	}
4897 
4898 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
4899 			  AT_STATX_SYNC_AS_STAT);
4900 	if (ret)
4901 		return ret;
4902 
4903 	basic_info = (struct file_basic_info *)rsp->Buffer;
4904 	basic_info->CreationTime = cpu_to_le64(fp->create_time);
4905 	time = ksmbd_UnixTimeToNT(stat.atime);
4906 	basic_info->LastAccessTime = cpu_to_le64(time);
4907 	time = ksmbd_UnixTimeToNT(stat.mtime);
4908 	basic_info->LastWriteTime = cpu_to_le64(time);
4909 	time = ksmbd_UnixTimeToNT(stat.ctime);
4910 	basic_info->ChangeTime = cpu_to_le64(time);
4911 	basic_info->Attributes = fp->f_ci->m_fattr;
4912 	basic_info->Pad = 0;
4913 	rsp->OutputBufferLength =
4914 		cpu_to_le32(sizeof(struct file_basic_info));
4915 	return 0;
4916 }
4917 
4918 static int get_file_standard_info(struct smb2_query_info_rsp *rsp,
4919 				  struct ksmbd_file *fp, void *rsp_org)
4920 {
4921 	struct smb2_file_standard_info *sinfo;
4922 	unsigned int delete_pending;
4923 	struct kstat stat;
4924 	int ret;
4925 
4926 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
4927 			  AT_STATX_SYNC_AS_STAT);
4928 	if (ret)
4929 		return ret;
4930 
4931 	sinfo = (struct smb2_file_standard_info *)rsp->Buffer;
4932 	delete_pending = ksmbd_inode_pending_delete(fp);
4933 
4934 	if (ksmbd_stream_fd(fp) == false) {
4935 		sinfo->AllocationSize = cpu_to_le64(stat.blocks << 9);
4936 		sinfo->EndOfFile = S_ISDIR(stat.mode) ? 0 : cpu_to_le64(stat.size);
4937 	} else {
4938 		sinfo->AllocationSize = cpu_to_le64(fp->stream.size);
4939 		sinfo->EndOfFile = cpu_to_le64(fp->stream.size);
4940 	}
4941 	sinfo->NumberOfLinks = cpu_to_le32(get_nlink(&stat) - delete_pending);
4942 	sinfo->DeletePending = delete_pending;
4943 	sinfo->Directory = S_ISDIR(stat.mode) ? 1 : 0;
4944 	rsp->OutputBufferLength =
4945 		cpu_to_le32(sizeof(struct smb2_file_standard_info));
4946 
4947 	return 0;
4948 }
4949 
4950 static void get_file_alignment_info(struct smb2_query_info_rsp *rsp,
4951 				    void *rsp_org)
4952 {
4953 	struct smb2_file_alignment_info *file_info;
4954 
4955 	file_info = (struct smb2_file_alignment_info *)rsp->Buffer;
4956 	file_info->AlignmentRequirement = 0;
4957 	rsp->OutputBufferLength =
4958 		cpu_to_le32(sizeof(struct smb2_file_alignment_info));
4959 }
4960 
4961 static int get_file_all_info(struct ksmbd_work *work,
4962 			     struct smb2_query_info_rsp *rsp,
4963 			     struct ksmbd_file *fp,
4964 			     void *rsp_org)
4965 {
4966 	struct ksmbd_conn *conn = work->conn;
4967 	struct smb2_file_all_info *file_info;
4968 	unsigned int delete_pending;
4969 	struct kstat stat;
4970 	int conv_len;
4971 	char *filename;
4972 	u64 time;
4973 	int ret, buf_free_len, filename_len;
4974 	struct smb2_query_info_req *req = ksmbd_req_buf_next(work);
4975 
4976 	if (!(fp->daccess & FILE_READ_ATTRIBUTES_LE)) {
4977 		ksmbd_debug(SMB, "no right to read the attributes : 0x%x\n",
4978 			    fp->daccess);
4979 		return -EACCES;
4980 	}
4981 
4982 	filename = convert_to_nt_pathname(work->tcon->share_conf, &fp->filp->f_path);
4983 	if (IS_ERR(filename))
4984 		return PTR_ERR(filename);
4985 
4986 	filename_len = strlen(filename);
4987 	buf_free_len = smb2_calc_max_out_buf_len(work,
4988 			offsetof(struct smb2_query_info_rsp, Buffer) +
4989 			offsetof(struct smb2_file_all_info, FileName),
4990 			le32_to_cpu(req->OutputBufferLength));
4991 	if (buf_free_len < (filename_len + 1) * 2) {
4992 		kfree(filename);
4993 		return -EINVAL;
4994 	}
4995 
4996 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
4997 			  AT_STATX_SYNC_AS_STAT);
4998 	if (ret) {
4999 		kfree(filename);
5000 		return ret;
5001 	}
5002 
5003 	ksmbd_debug(SMB, "filename = %s\n", filename);
5004 	delete_pending = ksmbd_inode_pending_delete(fp);
5005 	file_info = (struct smb2_file_all_info *)rsp->Buffer;
5006 
5007 	file_info->CreationTime = cpu_to_le64(fp->create_time);
5008 	time = ksmbd_UnixTimeToNT(stat.atime);
5009 	file_info->LastAccessTime = cpu_to_le64(time);
5010 	time = ksmbd_UnixTimeToNT(stat.mtime);
5011 	file_info->LastWriteTime = cpu_to_le64(time);
5012 	time = ksmbd_UnixTimeToNT(stat.ctime);
5013 	file_info->ChangeTime = cpu_to_le64(time);
5014 	file_info->Attributes = fp->f_ci->m_fattr;
5015 	file_info->Pad1 = 0;
5016 	if (ksmbd_stream_fd(fp) == false) {
5017 		file_info->AllocationSize =
5018 			cpu_to_le64(stat.blocks << 9);
5019 		file_info->EndOfFile = S_ISDIR(stat.mode) ? 0 : cpu_to_le64(stat.size);
5020 	} else {
5021 		file_info->AllocationSize = cpu_to_le64(fp->stream.size);
5022 		file_info->EndOfFile = cpu_to_le64(fp->stream.size);
5023 	}
5024 	file_info->NumberOfLinks =
5025 			cpu_to_le32(get_nlink(&stat) - delete_pending);
5026 	file_info->DeletePending = delete_pending;
5027 	file_info->Directory = S_ISDIR(stat.mode) ? 1 : 0;
5028 	file_info->Pad2 = 0;
5029 	file_info->IndexNumber = cpu_to_le64(stat.ino);
5030 	file_info->EASize = 0;
5031 	file_info->AccessFlags = fp->daccess;
5032 	if (ksmbd_stream_fd(fp) == false)
5033 		file_info->CurrentByteOffset = cpu_to_le64(fp->filp->f_pos);
5034 	else
5035 		file_info->CurrentByteOffset = cpu_to_le64(fp->stream.pos);
5036 	file_info->Mode = fp->coption;
5037 	file_info->AlignmentRequirement = 0;
5038 	conv_len = smbConvertToUTF16((__le16 *)file_info->FileName, filename,
5039 				     min(filename_len, PATH_MAX),
5040 				     conn->local_nls, 0);
5041 	conv_len *= 2;
5042 	file_info->FileNameLength = cpu_to_le32(conv_len);
5043 	rsp->OutputBufferLength =
5044 		cpu_to_le32(sizeof(struct smb2_file_all_info) + conv_len - 1);
5045 	kfree(filename);
5046 	return 0;
5047 }
5048 
5049 static void get_file_alternate_info(struct ksmbd_work *work,
5050 				    struct smb2_query_info_rsp *rsp,
5051 				    struct ksmbd_file *fp,
5052 				    void *rsp_org)
5053 {
5054 	struct ksmbd_conn *conn = work->conn;
5055 	struct smb2_file_alt_name_info *file_info;
5056 	struct dentry *dentry = fp->filp->f_path.dentry;
5057 	int conv_len;
5058 
5059 	spin_lock(&dentry->d_lock);
5060 	file_info = (struct smb2_file_alt_name_info *)rsp->Buffer;
5061 	conv_len = ksmbd_extract_shortname(conn,
5062 					   dentry->d_name.name,
5063 					   file_info->FileName);
5064 	spin_unlock(&dentry->d_lock);
5065 	file_info->FileNameLength = cpu_to_le32(conv_len);
5066 	rsp->OutputBufferLength =
5067 		cpu_to_le32(struct_size(file_info, FileName, conv_len));
5068 }
5069 
5070 static int get_file_stream_info(struct ksmbd_work *work,
5071 				struct smb2_query_info_rsp *rsp,
5072 				struct ksmbd_file *fp,
5073 				void *rsp_org)
5074 {
5075 	struct ksmbd_conn *conn = work->conn;
5076 	struct smb2_file_stream_info *file_info;
5077 	char *stream_name, *xattr_list = NULL, *stream_buf;
5078 	struct kstat stat;
5079 	const struct path *path = &fp->filp->f_path;
5080 	ssize_t xattr_list_len;
5081 	int nbytes = 0, streamlen, stream_name_len, next, idx = 0;
5082 	int buf_free_len;
5083 	struct smb2_query_info_req *req = ksmbd_req_buf_next(work);
5084 	int ret;
5085 
5086 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
5087 			  AT_STATX_SYNC_AS_STAT);
5088 	if (ret)
5089 		return ret;
5090 
5091 	file_info = (struct smb2_file_stream_info *)rsp->Buffer;
5092 
5093 	buf_free_len =
5094 		smb2_calc_max_out_buf_len(work,
5095 				offsetof(struct smb2_query_info_rsp, Buffer),
5096 				le32_to_cpu(req->OutputBufferLength));
5097 	if (buf_free_len < 0)
5098 		goto out;
5099 
5100 	xattr_list_len = ksmbd_vfs_listxattr(path->dentry, &xattr_list);
5101 	if (xattr_list_len < 0) {
5102 		goto out;
5103 	} else if (!xattr_list_len) {
5104 		ksmbd_debug(SMB, "empty xattr in the file\n");
5105 		goto out;
5106 	}
5107 
5108 	while (idx < xattr_list_len) {
5109 		stream_name = xattr_list + idx;
5110 		streamlen = strlen(stream_name);
5111 		idx += streamlen + 1;
5112 
5113 		ksmbd_debug(SMB, "%s, len %d\n", stream_name, streamlen);
5114 
5115 		if (strncmp(&stream_name[XATTR_USER_PREFIX_LEN],
5116 			    STREAM_PREFIX, STREAM_PREFIX_LEN))
5117 			continue;
5118 
5119 		stream_name_len = streamlen - (XATTR_USER_PREFIX_LEN +
5120 				STREAM_PREFIX_LEN);
5121 		streamlen = stream_name_len;
5122 
5123 		/* plus : size */
5124 		streamlen += 1;
5125 		stream_buf = kmalloc(streamlen + 1, KSMBD_DEFAULT_GFP);
5126 		if (!stream_buf)
5127 			break;
5128 
5129 		streamlen = snprintf(stream_buf, streamlen + 1,
5130 				     ":%s", &stream_name[XATTR_NAME_STREAM_LEN]);
5131 
5132 		next = sizeof(struct smb2_file_stream_info) + streamlen * 2;
5133 		if (next > buf_free_len) {
5134 			kfree(stream_buf);
5135 			break;
5136 		}
5137 
5138 		file_info = (struct smb2_file_stream_info *)&rsp->Buffer[nbytes];
5139 		streamlen  = smbConvertToUTF16((__le16 *)file_info->StreamName,
5140 					       stream_buf, streamlen,
5141 					       conn->local_nls, 0);
5142 		streamlen *= 2;
5143 		kfree(stream_buf);
5144 		file_info->StreamNameLength = cpu_to_le32(streamlen);
5145 		file_info->StreamSize = cpu_to_le64(stream_name_len);
5146 		file_info->StreamAllocationSize = cpu_to_le64(stream_name_len);
5147 
5148 		nbytes += next;
5149 		buf_free_len -= next;
5150 		file_info->NextEntryOffset = cpu_to_le32(next);
5151 	}
5152 
5153 out:
5154 	if (!S_ISDIR(stat.mode) &&
5155 	    buf_free_len >= sizeof(struct smb2_file_stream_info) + 7 * 2) {
5156 		file_info = (struct smb2_file_stream_info *)
5157 			&rsp->Buffer[nbytes];
5158 		streamlen = smbConvertToUTF16((__le16 *)file_info->StreamName,
5159 					      "::$DATA", 7, conn->local_nls, 0);
5160 		streamlen *= 2;
5161 		file_info->StreamNameLength = cpu_to_le32(streamlen);
5162 		file_info->StreamSize = cpu_to_le64(stat.size);
5163 		file_info->StreamAllocationSize = cpu_to_le64(stat.blocks << 9);
5164 		nbytes += sizeof(struct smb2_file_stream_info) + streamlen;
5165 	}
5166 
5167 	/* last entry offset should be 0 */
5168 	file_info->NextEntryOffset = 0;
5169 	kvfree(xattr_list);
5170 
5171 	rsp->OutputBufferLength = cpu_to_le32(nbytes);
5172 
5173 	return 0;
5174 }
5175 
5176 static int get_file_internal_info(struct smb2_query_info_rsp *rsp,
5177 				  struct ksmbd_file *fp, void *rsp_org)
5178 {
5179 	struct smb2_file_internal_info *file_info;
5180 	struct kstat stat;
5181 	int ret;
5182 
5183 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
5184 			  AT_STATX_SYNC_AS_STAT);
5185 	if (ret)
5186 		return ret;
5187 
5188 	file_info = (struct smb2_file_internal_info *)rsp->Buffer;
5189 	file_info->IndexNumber = cpu_to_le64(stat.ino);
5190 	rsp->OutputBufferLength =
5191 		cpu_to_le32(sizeof(struct smb2_file_internal_info));
5192 
5193 	return 0;
5194 }
5195 
5196 static int get_file_network_open_info(struct smb2_query_info_rsp *rsp,
5197 				      struct ksmbd_file *fp, void *rsp_org)
5198 {
5199 	struct smb2_file_network_open_info *file_info;
5200 	struct kstat stat;
5201 	u64 time;
5202 	int ret;
5203 
5204 	if (!(fp->daccess & FILE_READ_ATTRIBUTES_LE)) {
5205 		pr_err("no right to read the attributes : 0x%x\n",
5206 		       fp->daccess);
5207 		return -EACCES;
5208 	}
5209 
5210 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
5211 			  AT_STATX_SYNC_AS_STAT);
5212 	if (ret)
5213 		return ret;
5214 
5215 	file_info = (struct smb2_file_network_open_info *)rsp->Buffer;
5216 
5217 	file_info->CreationTime = cpu_to_le64(fp->create_time);
5218 	time = ksmbd_UnixTimeToNT(stat.atime);
5219 	file_info->LastAccessTime = cpu_to_le64(time);
5220 	time = ksmbd_UnixTimeToNT(stat.mtime);
5221 	file_info->LastWriteTime = cpu_to_le64(time);
5222 	time = ksmbd_UnixTimeToNT(stat.ctime);
5223 	file_info->ChangeTime = cpu_to_le64(time);
5224 	file_info->Attributes = fp->f_ci->m_fattr;
5225 	if (ksmbd_stream_fd(fp) == false) {
5226 		file_info->AllocationSize = cpu_to_le64(stat.blocks << 9);
5227 		file_info->EndOfFile = S_ISDIR(stat.mode) ? 0 : cpu_to_le64(stat.size);
5228 	} else {
5229 		file_info->AllocationSize = cpu_to_le64(fp->stream.size);
5230 		file_info->EndOfFile = cpu_to_le64(fp->stream.size);
5231 	}
5232 	file_info->Reserved = cpu_to_le32(0);
5233 	rsp->OutputBufferLength =
5234 		cpu_to_le32(sizeof(struct smb2_file_network_open_info));
5235 	return 0;
5236 }
5237 
5238 static void get_file_ea_info(struct smb2_query_info_rsp *rsp, void *rsp_org)
5239 {
5240 	struct smb2_file_ea_info *file_info;
5241 
5242 	file_info = (struct smb2_file_ea_info *)rsp->Buffer;
5243 	file_info->EASize = 0;
5244 	rsp->OutputBufferLength =
5245 		cpu_to_le32(sizeof(struct smb2_file_ea_info));
5246 }
5247 
5248 static void get_file_position_info(struct smb2_query_info_rsp *rsp,
5249 				   struct ksmbd_file *fp, void *rsp_org)
5250 {
5251 	struct smb2_file_pos_info *file_info;
5252 
5253 	file_info = (struct smb2_file_pos_info *)rsp->Buffer;
5254 	if (ksmbd_stream_fd(fp) == false)
5255 		file_info->CurrentByteOffset = cpu_to_le64(fp->filp->f_pos);
5256 	else
5257 		file_info->CurrentByteOffset = cpu_to_le64(fp->stream.pos);
5258 
5259 	rsp->OutputBufferLength =
5260 		cpu_to_le32(sizeof(struct smb2_file_pos_info));
5261 }
5262 
5263 static void get_file_mode_info(struct smb2_query_info_rsp *rsp,
5264 			       struct ksmbd_file *fp, void *rsp_org)
5265 {
5266 	struct smb2_file_mode_info *file_info;
5267 
5268 	file_info = (struct smb2_file_mode_info *)rsp->Buffer;
5269 	file_info->Mode = fp->coption & FILE_MODE_INFO_MASK;
5270 	rsp->OutputBufferLength =
5271 		cpu_to_le32(sizeof(struct smb2_file_mode_info));
5272 }
5273 
5274 static int get_file_compression_info(struct smb2_query_info_rsp *rsp,
5275 				     struct ksmbd_file *fp, void *rsp_org)
5276 {
5277 	struct smb2_file_comp_info *file_info;
5278 	struct kstat stat;
5279 	int ret;
5280 
5281 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
5282 			  AT_STATX_SYNC_AS_STAT);
5283 	if (ret)
5284 		return ret;
5285 
5286 	file_info = (struct smb2_file_comp_info *)rsp->Buffer;
5287 	file_info->CompressedFileSize = cpu_to_le64(stat.blocks << 9);
5288 	file_info->CompressionFormat = COMPRESSION_FORMAT_NONE;
5289 	file_info->CompressionUnitShift = 0;
5290 	file_info->ChunkShift = 0;
5291 	file_info->ClusterShift = 0;
5292 	memset(&file_info->Reserved[0], 0, 3);
5293 
5294 	rsp->OutputBufferLength =
5295 		cpu_to_le32(sizeof(struct smb2_file_comp_info));
5296 
5297 	return 0;
5298 }
5299 
5300 static int get_file_attribute_tag_info(struct smb2_query_info_rsp *rsp,
5301 				       struct ksmbd_file *fp, void *rsp_org)
5302 {
5303 	struct smb2_file_attr_tag_info *file_info;
5304 
5305 	if (!(fp->daccess & FILE_READ_ATTRIBUTES_LE)) {
5306 		pr_err("no right to read the attributes : 0x%x\n",
5307 		       fp->daccess);
5308 		return -EACCES;
5309 	}
5310 
5311 	file_info = (struct smb2_file_attr_tag_info *)rsp->Buffer;
5312 	file_info->FileAttributes = fp->f_ci->m_fattr;
5313 	file_info->ReparseTag = 0;
5314 	rsp->OutputBufferLength =
5315 		cpu_to_le32(sizeof(struct smb2_file_attr_tag_info));
5316 	return 0;
5317 }
5318 
5319 static int find_file_posix_info(struct smb2_query_info_rsp *rsp,
5320 				struct ksmbd_file *fp, void *rsp_org)
5321 {
5322 	struct smb311_posix_qinfo *file_info;
5323 	struct inode *inode = file_inode(fp->filp);
5324 	struct mnt_idmap *idmap = file_mnt_idmap(fp->filp);
5325 	vfsuid_t vfsuid = i_uid_into_vfsuid(idmap, inode);
5326 	vfsgid_t vfsgid = i_gid_into_vfsgid(idmap, inode);
5327 	struct kstat stat;
5328 	u64 time;
5329 	int out_buf_len = sizeof(struct smb311_posix_qinfo) + 32;
5330 	int ret;
5331 
5332 	ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
5333 			  AT_STATX_SYNC_AS_STAT);
5334 	if (ret)
5335 		return ret;
5336 
5337 	file_info = (struct smb311_posix_qinfo *)rsp->Buffer;
5338 	file_info->CreationTime = cpu_to_le64(fp->create_time);
5339 	time = ksmbd_UnixTimeToNT(stat.atime);
5340 	file_info->LastAccessTime = cpu_to_le64(time);
5341 	time = ksmbd_UnixTimeToNT(stat.mtime);
5342 	file_info->LastWriteTime = cpu_to_le64(time);
5343 	time = ksmbd_UnixTimeToNT(stat.ctime);
5344 	file_info->ChangeTime = cpu_to_le64(time);
5345 	file_info->DosAttributes = fp->f_ci->m_fattr;
5346 	file_info->Inode = cpu_to_le64(stat.ino);
5347 	if (ksmbd_stream_fd(fp) == false) {
5348 		file_info->EndOfFile = cpu_to_le64(stat.size);
5349 		file_info->AllocationSize = cpu_to_le64(stat.blocks << 9);
5350 	} else {
5351 		file_info->EndOfFile = cpu_to_le64(fp->stream.size);
5352 		file_info->AllocationSize = cpu_to_le64(fp->stream.size);
5353 	}
5354 	file_info->HardLinks = cpu_to_le32(stat.nlink);
5355 	file_info->Mode = cpu_to_le32(stat.mode & 0777);
5356 	switch (stat.mode & S_IFMT) {
5357 	case S_IFDIR:
5358 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_DIR << POSIX_FILETYPE_SHIFT);
5359 		break;
5360 	case S_IFLNK:
5361 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_SYMLINK << POSIX_FILETYPE_SHIFT);
5362 		break;
5363 	case S_IFCHR:
5364 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_CHARDEV << POSIX_FILETYPE_SHIFT);
5365 		break;
5366 	case S_IFBLK:
5367 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_BLKDEV << POSIX_FILETYPE_SHIFT);
5368 		break;
5369 	case S_IFIFO:
5370 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_FIFO << POSIX_FILETYPE_SHIFT);
5371 		break;
5372 	case S_IFSOCK:
5373 		file_info->Mode |= cpu_to_le32(POSIX_TYPE_SOCKET << POSIX_FILETYPE_SHIFT);
5374 	}
5375 
5376 	file_info->DeviceId = cpu_to_le32(stat.rdev);
5377 
5378 	/*
5379 	 * Sids(32) contain two sids(Domain sid(16), UNIX group sid(16)).
5380 	 * UNIX sid(16) = revision(1) + num_subauth(1) + authority(6) +
5381 	 *		  sub_auth(4 * 1(num_subauth)) + RID(4).
5382 	 */
5383 	id_to_sid(from_kuid_munged(&init_user_ns, vfsuid_into_kuid(vfsuid)),
5384 		  SIDUNIX_USER, (struct smb_sid *)&file_info->Sids[0]);
5385 	id_to_sid(from_kgid_munged(&init_user_ns, vfsgid_into_kgid(vfsgid)),
5386 		  SIDUNIX_GROUP, (struct smb_sid *)&file_info->Sids[16]);
5387 
5388 	rsp->OutputBufferLength = cpu_to_le32(out_buf_len);
5389 
5390 	return 0;
5391 }
5392 
5393 static int smb2_get_info_file(struct ksmbd_work *work,
5394 			      struct smb2_query_info_req *req,
5395 			      struct smb2_query_info_rsp *rsp)
5396 {
5397 	struct ksmbd_file *fp;
5398 	int fileinfoclass = 0;
5399 	int rc = 0;
5400 	unsigned int id = KSMBD_NO_FID, pid = KSMBD_NO_FID;
5401 
5402 	if (test_share_config_flag(work->tcon->share_conf,
5403 				   KSMBD_SHARE_FLAG_PIPE)) {
5404 		/* smb2 info file called for pipe */
5405 		rc = smb2_get_info_file_pipe(work->sess, req, rsp,
5406 					       work->response_buf);
5407 		goto iov_pin_out;
5408 	}
5409 
5410 	if (work->next_smb2_rcv_hdr_off) {
5411 		if (!has_file_id(req->VolatileFileId)) {
5412 			ksmbd_debug(SMB, "Compound request set FID = %llu\n",
5413 				    work->compound_fid);
5414 			id = work->compound_fid;
5415 			pid = work->compound_pfid;
5416 		}
5417 	}
5418 
5419 	if (!has_file_id(id)) {
5420 		id = req->VolatileFileId;
5421 		pid = req->PersistentFileId;
5422 	}
5423 
5424 	fp = ksmbd_lookup_fd_slow(work, id, pid);
5425 	if (!fp)
5426 		return -ENOENT;
5427 
5428 	fileinfoclass = req->FileInfoClass;
5429 
5430 	switch (fileinfoclass) {
5431 	case FILE_ACCESS_INFORMATION:
5432 		get_file_access_info(rsp, fp, work->response_buf);
5433 		break;
5434 
5435 	case FILE_BASIC_INFORMATION:
5436 		rc = get_file_basic_info(rsp, fp, work->response_buf);
5437 		break;
5438 
5439 	case FILE_STANDARD_INFORMATION:
5440 		rc = get_file_standard_info(rsp, fp, work->response_buf);
5441 		break;
5442 
5443 	case FILE_ALIGNMENT_INFORMATION:
5444 		get_file_alignment_info(rsp, work->response_buf);
5445 		break;
5446 
5447 	case FILE_ALL_INFORMATION:
5448 		rc = get_file_all_info(work, rsp, fp, work->response_buf);
5449 		break;
5450 
5451 	case FILE_ALTERNATE_NAME_INFORMATION:
5452 		get_file_alternate_info(work, rsp, fp, work->response_buf);
5453 		break;
5454 
5455 	case FILE_STREAM_INFORMATION:
5456 		rc = get_file_stream_info(work, rsp, fp, work->response_buf);
5457 		break;
5458 
5459 	case FILE_INTERNAL_INFORMATION:
5460 		rc = get_file_internal_info(rsp, fp, work->response_buf);
5461 		break;
5462 
5463 	case FILE_NETWORK_OPEN_INFORMATION:
5464 		rc = get_file_network_open_info(rsp, fp, work->response_buf);
5465 		break;
5466 
5467 	case FILE_EA_INFORMATION:
5468 		get_file_ea_info(rsp, work->response_buf);
5469 		break;
5470 
5471 	case FILE_FULL_EA_INFORMATION:
5472 		rc = smb2_get_ea(work, fp, req, rsp, work->response_buf);
5473 		break;
5474 
5475 	case FILE_POSITION_INFORMATION:
5476 		get_file_position_info(rsp, fp, work->response_buf);
5477 		break;
5478 
5479 	case FILE_MODE_INFORMATION:
5480 		get_file_mode_info(rsp, fp, work->response_buf);
5481 		break;
5482 
5483 	case FILE_COMPRESSION_INFORMATION:
5484 		rc = get_file_compression_info(rsp, fp, work->response_buf);
5485 		break;
5486 
5487 	case FILE_ATTRIBUTE_TAG_INFORMATION:
5488 		rc = get_file_attribute_tag_info(rsp, fp, work->response_buf);
5489 		break;
5490 	case SMB_FIND_FILE_POSIX_INFO:
5491 		if (!work->tcon->posix_extensions) {
5492 			pr_err("client doesn't negotiate with SMB3.1.1 POSIX Extensions\n");
5493 			rc = -EOPNOTSUPP;
5494 		} else {
5495 			rc = find_file_posix_info(rsp, fp, work->response_buf);
5496 		}
5497 		break;
5498 	default:
5499 		ksmbd_debug(SMB, "fileinfoclass %d not supported yet\n",
5500 			    fileinfoclass);
5501 		rc = -EOPNOTSUPP;
5502 	}
5503 	if (!rc)
5504 		rc = buffer_check_err(le32_to_cpu(req->OutputBufferLength),
5505 				      rsp, work->response_buf);
5506 	ksmbd_fd_put(work, fp);
5507 
5508 iov_pin_out:
5509 	if (!rc)
5510 		rc = ksmbd_iov_pin_rsp(work, (void *)rsp,
5511 				offsetof(struct smb2_query_info_rsp, Buffer) +
5512 				le32_to_cpu(rsp->OutputBufferLength));
5513 	return rc;
5514 }
5515 
5516 static int smb2_get_info_filesystem(struct ksmbd_work *work,
5517 				    struct smb2_query_info_req *req,
5518 				    struct smb2_query_info_rsp *rsp)
5519 {
5520 	struct ksmbd_conn *conn = work->conn;
5521 	struct ksmbd_share_config *share = work->tcon->share_conf;
5522 	int fsinfoclass = 0;
5523 	struct kstatfs stfs;
5524 	struct path path;
5525 	int rc = 0, len;
5526 
5527 	if (!share->path)
5528 		return -EIO;
5529 
5530 	rc = kern_path(share->path, LOOKUP_NO_SYMLINKS, &path);
5531 	if (rc) {
5532 		pr_err("cannot create vfs path\n");
5533 		return -EIO;
5534 	}
5535 
5536 	rc = vfs_statfs(&path, &stfs);
5537 	if (rc) {
5538 		pr_err("cannot do stat of path %s\n", share->path);
5539 		path_put(&path);
5540 		return -EIO;
5541 	}
5542 
5543 	fsinfoclass = req->FileInfoClass;
5544 
5545 	switch (fsinfoclass) {
5546 	case FS_DEVICE_INFORMATION:
5547 	{
5548 		FILE_SYSTEM_DEVICE_INFO *info;
5549 
5550 		info = (FILE_SYSTEM_DEVICE_INFO *)rsp->Buffer;
5551 
5552 		info->DeviceType = cpu_to_le32(FILE_DEVICE_DISK);
5553 		info->DeviceCharacteristics =
5554 			cpu_to_le32(FILE_DEVICE_IS_MOUNTED);
5555 		if (!test_tree_conn_flag(work->tcon,
5556 					 KSMBD_TREE_CONN_FLAG_WRITABLE))
5557 			info->DeviceCharacteristics |=
5558 				cpu_to_le32(FILE_READ_ONLY_DEVICE);
5559 		rsp->OutputBufferLength = cpu_to_le32(8);
5560 		break;
5561 	}
5562 	case FS_ATTRIBUTE_INFORMATION:
5563 	{
5564 		FILE_SYSTEM_ATTRIBUTE_INFO *info;
5565 		struct file_kattr fa = {};
5566 		size_t sz;
5567 		u32 attrs;
5568 		int err;
5569 
5570 		info = (FILE_SYSTEM_ATTRIBUTE_INFO *)rsp->Buffer;
5571 		attrs = FILE_SUPPORTS_OBJECT_IDS |
5572 			FILE_PERSISTENT_ACLS |
5573 			FILE_UNICODE_ON_DISK |
5574 			FILE_SUPPORTS_BLOCK_REFCOUNTING;
5575 
5576 		err = vfs_fileattr_get(path.dentry, &fa);
5577 		/*
5578 		 * -EINVAL, -EOPNOTSUPP: ntfs-3g and other FUSE
5579 		 * filesystems that lack FS_IOC_FSGETXATTR support.
5580 		 */
5581 		if (err && err != -ENOIOCTLCMD && err != -ENOTTY &&
5582 		    err != -EINVAL && err != -EOPNOTSUPP) {
5583 			path_put(&path);
5584 			return err;
5585 		}
5586 		if (!(fa.fsx_xflags & FS_XFLAG_CASEFOLD))
5587 			attrs |= FILE_CASE_SENSITIVE_SEARCH;
5588 		if (!(fa.fsx_xflags & FS_XFLAG_CASENONPRESERVING))
5589 			attrs |= FILE_CASE_PRESERVED_NAMES;
5590 
5591 		info->Attributes = cpu_to_le32(attrs);
5592 		info->Attributes |= cpu_to_le32(server_conf.share_fake_fscaps);
5593 
5594 		if (test_share_config_flag(work->tcon->share_conf,
5595 		    KSMBD_SHARE_FLAG_STREAMS))
5596 			info->Attributes |= cpu_to_le32(FILE_NAMED_STREAMS);
5597 
5598 		info->MaxPathNameComponentLength = cpu_to_le32(stfs.f_namelen);
5599 		/*
5600 		 * some application(potableapp) can not run on ksmbd share
5601 		 * because only NTFS handle security setting on windows.
5602 		 * So Although local fs(EXT4 or F2fs, etc) is not NTFS,
5603 		 * ksmbd should show share as NTFS. Later, If needed, we can add
5604 		 * fs type(s) parameter to change fs type user wanted.
5605 		 */
5606 		len = smbConvertToUTF16((__le16 *)info->FileSystemName,
5607 					"NTFS", PATH_MAX, conn->local_nls, 0);
5608 		len = len * 2;
5609 		info->FileSystemNameLen = cpu_to_le32(len);
5610 		sz = sizeof(FILE_SYSTEM_ATTRIBUTE_INFO) + len;
5611 		rsp->OutputBufferLength = cpu_to_le32(sz);
5612 		break;
5613 	}
5614 	case FS_VOLUME_INFORMATION:
5615 	{
5616 		struct filesystem_vol_info *info;
5617 		size_t sz;
5618 		unsigned int serial_crc = 0;
5619 
5620 		info = (struct filesystem_vol_info *)(rsp->Buffer);
5621 		info->VolumeCreationTime = 0;
5622 		serial_crc = crc32_le(serial_crc, share->name,
5623 				      strlen(share->name));
5624 		serial_crc = crc32_le(serial_crc, share->path,
5625 				      strlen(share->path));
5626 		serial_crc = crc32_le(serial_crc, ksmbd_netbios_name(),
5627 				      strlen(ksmbd_netbios_name()));
5628 		/* Taking dummy value of serial number*/
5629 		info->VolumeSerialNumber = cpu_to_le32(serial_crc);
5630 		len = smbConvertToUTF16((__le16 *)info->VolumeLabel,
5631 					share->name, PATH_MAX,
5632 					conn->local_nls, 0);
5633 		len = len * 2;
5634 		info->VolumeLabelLength = cpu_to_le32(len);
5635 		info->Reserved = 0;
5636 		info->SupportsObjects = 0;
5637 		sz = sizeof(struct filesystem_vol_info) + len;
5638 		rsp->OutputBufferLength = cpu_to_le32(sz);
5639 		break;
5640 	}
5641 	case FS_SIZE_INFORMATION:
5642 	{
5643 		FILE_SYSTEM_SIZE_INFO *info;
5644 
5645 		info = (FILE_SYSTEM_SIZE_INFO *)(rsp->Buffer);
5646 		info->TotalAllocationUnits = cpu_to_le64(stfs.f_blocks);
5647 		info->AvailableAllocationUnits = cpu_to_le64(stfs.f_bfree);
5648 		info->SectorsPerAllocationUnit = cpu_to_le32(1);
5649 		info->BytesPerSector = cpu_to_le32(stfs.f_bsize);
5650 		rsp->OutputBufferLength = cpu_to_le32(24);
5651 		break;
5652 	}
5653 	case FS_FULL_SIZE_INFORMATION:
5654 	{
5655 		struct smb2_fs_full_size_info *info;
5656 
5657 		info = (struct smb2_fs_full_size_info *)(rsp->Buffer);
5658 		info->TotalAllocationUnits = cpu_to_le64(stfs.f_blocks);
5659 		info->CallerAvailableAllocationUnits =
5660 					cpu_to_le64(stfs.f_bavail);
5661 		info->ActualAvailableAllocationUnits =
5662 					cpu_to_le64(stfs.f_bfree);
5663 		info->SectorsPerAllocationUnit = cpu_to_le32(1);
5664 		info->BytesPerSector = cpu_to_le32(stfs.f_bsize);
5665 		rsp->OutputBufferLength = cpu_to_le32(32);
5666 		break;
5667 	}
5668 	case FS_OBJECT_ID_INFORMATION:
5669 	{
5670 		struct object_id_info *info;
5671 
5672 		info = (struct object_id_info *)(rsp->Buffer);
5673 
5674 		if (path.mnt->mnt_sb->s_uuid_len == 16)
5675 			memcpy(info->objid, path.mnt->mnt_sb->s_uuid.b,
5676 					path.mnt->mnt_sb->s_uuid_len);
5677 		else
5678 			memcpy(info->objid, &stfs.f_fsid, sizeof(stfs.f_fsid));
5679 
5680 		info->extended_info.magic = cpu_to_le32(EXTENDED_INFO_MAGIC);
5681 		info->extended_info.version = cpu_to_le32(1);
5682 		info->extended_info.release = cpu_to_le32(1);
5683 		info->extended_info.rel_date = 0;
5684 		memcpy(info->extended_info.version_string, "1.1.0", strlen("1.1.0"));
5685 		rsp->OutputBufferLength = cpu_to_le32(64);
5686 		break;
5687 	}
5688 	case FS_SECTOR_SIZE_INFORMATION:
5689 	{
5690 		struct smb3_fs_ss_info *info;
5691 		unsigned int sector_size =
5692 			min_t(unsigned int, path.mnt->mnt_sb->s_blocksize, 4096);
5693 
5694 		info = (struct smb3_fs_ss_info *)(rsp->Buffer);
5695 
5696 		info->LogicalBytesPerSector = cpu_to_le32(sector_size);
5697 		info->PhysicalBytesPerSectorForAtomicity =
5698 				cpu_to_le32(sector_size);
5699 		info->PhysicalBytesPerSectorForPerf = cpu_to_le32(sector_size);
5700 		info->FSEffPhysicalBytesPerSectorForAtomicity =
5701 				cpu_to_le32(sector_size);
5702 		info->Flags = cpu_to_le32(SSINFO_FLAGS_ALIGNED_DEVICE |
5703 				    SSINFO_FLAGS_PARTITION_ALIGNED_ON_DEVICE);
5704 		info->ByteOffsetForSectorAlignment = 0;
5705 		info->ByteOffsetForPartitionAlignment = 0;
5706 		rsp->OutputBufferLength = cpu_to_le32(28);
5707 		break;
5708 	}
5709 	case FS_CONTROL_INFORMATION:
5710 	{
5711 		/*
5712 		 * TODO : The current implementation is based on
5713 		 * test result with win7(NTFS) server. It's need to
5714 		 * modify this to get valid Quota values
5715 		 * from Linux kernel
5716 		 */
5717 		struct smb2_fs_control_info *info;
5718 
5719 		info = (struct smb2_fs_control_info *)(rsp->Buffer);
5720 		info->FreeSpaceStartFiltering = 0;
5721 		info->FreeSpaceThreshold = 0;
5722 		info->FreeSpaceStopFiltering = 0;
5723 		info->DefaultQuotaThreshold = cpu_to_le64(SMB2_NO_FID);
5724 		info->DefaultQuotaLimit = cpu_to_le64(SMB2_NO_FID);
5725 		info->Padding = 0;
5726 		rsp->OutputBufferLength = cpu_to_le32(48);
5727 		break;
5728 	}
5729 	case FS_POSIX_INFORMATION:
5730 	{
5731 		FILE_SYSTEM_POSIX_INFO *info;
5732 
5733 		if (!work->tcon->posix_extensions) {
5734 			pr_err("client doesn't negotiate with SMB3.1.1 POSIX Extensions\n");
5735 			path_put(&path);
5736 			return -EOPNOTSUPP;
5737 		} else {
5738 			info = (FILE_SYSTEM_POSIX_INFO *)(rsp->Buffer);
5739 			info->OptimalTransferSize = cpu_to_le32(stfs.f_bsize);
5740 			info->BlockSize = cpu_to_le32(stfs.f_bsize);
5741 			info->TotalBlocks = cpu_to_le64(stfs.f_blocks);
5742 			info->BlocksAvail = cpu_to_le64(stfs.f_bfree);
5743 			info->UserBlocksAvail = cpu_to_le64(stfs.f_bavail);
5744 			info->TotalFileNodes = cpu_to_le64(stfs.f_files);
5745 			info->FreeFileNodes = cpu_to_le64(stfs.f_ffree);
5746 			rsp->OutputBufferLength = cpu_to_le32(56);
5747 		}
5748 		break;
5749 	}
5750 	default:
5751 		path_put(&path);
5752 		return -EOPNOTSUPP;
5753 	}
5754 	rc = buffer_check_err(le32_to_cpu(req->OutputBufferLength),
5755 			      rsp, work->response_buf);
5756 	path_put(&path);
5757 
5758 	if (!rc)
5759 		rc = ksmbd_iov_pin_rsp(work, (void *)rsp,
5760 				offsetof(struct smb2_query_info_rsp, Buffer) +
5761 				le32_to_cpu(rsp->OutputBufferLength));
5762 	return rc;
5763 }
5764 
5765 static int smb2_get_info_sec(struct ksmbd_work *work,
5766 			     struct smb2_query_info_req *req,
5767 			     struct smb2_query_info_rsp *rsp)
5768 {
5769 	struct ksmbd_file *fp;
5770 	struct mnt_idmap *idmap;
5771 	struct smb_ntsd *pntsd = NULL, *ppntsd = NULL;
5772 	struct smb_fattr fattr = {{0}};
5773 	struct inode *inode;
5774 	__u32 secdesclen = 0;
5775 	unsigned int id = KSMBD_NO_FID, pid = KSMBD_NO_FID;
5776 	int addition_info = le32_to_cpu(req->AdditionalInformation);
5777 	int rc = 0, ppntsd_size = 0, max_len;
5778 	size_t scratch_len = 0;
5779 
5780 	if (addition_info & ~(OWNER_SECINFO | GROUP_SECINFO | DACL_SECINFO |
5781 			      PROTECTED_DACL_SECINFO |
5782 			      UNPROTECTED_DACL_SECINFO)) {
5783 		ksmbd_debug(SMB, "Unsupported addition info: 0x%x)\n",
5784 		       addition_info);
5785 
5786 		rsp->hdr.Status = STATUS_NOT_SUPPORTED;
5787 		return -EINVAL;
5788 	}
5789 
5790 	if (work->next_smb2_rcv_hdr_off) {
5791 		if (!has_file_id(req->VolatileFileId)) {
5792 			ksmbd_debug(SMB, "Compound request set FID = %llu\n",
5793 				    work->compound_fid);
5794 			id = work->compound_fid;
5795 			pid = work->compound_pfid;
5796 		}
5797 	}
5798 
5799 	if (!has_file_id(id)) {
5800 		id = req->VolatileFileId;
5801 		pid = req->PersistentFileId;
5802 	}
5803 
5804 	fp = ksmbd_lookup_fd_slow(work, id, pid);
5805 	if (!fp)
5806 		return -ENOENT;
5807 
5808 	idmap = file_mnt_idmap(fp->filp);
5809 	inode = file_inode(fp->filp);
5810 	ksmbd_acls_fattr(&fattr, idmap, inode);
5811 
5812 	if (test_share_config_flag(work->tcon->share_conf,
5813 				   KSMBD_SHARE_FLAG_ACL_XATTR))
5814 		ppntsd_size = ksmbd_vfs_get_sd_xattr(work->conn, idmap,
5815 						     fp->filp->f_path.dentry,
5816 						     &ppntsd);
5817 
5818 	/* Check if sd buffer size exceeds response buffer size */
5819 	max_len = smb2_calc_max_out_buf_len(work,
5820 			offsetof(struct smb2_query_info_rsp, Buffer),
5821 			le32_to_cpu(req->OutputBufferLength));
5822 	if (max_len < 0) {
5823 		rc = -EINVAL;
5824 		goto release_acl;
5825 	}
5826 
5827 	scratch_len = smb_acl_sec_desc_scratch_len(&fattr, ppntsd,
5828 			ppntsd_size, addition_info);
5829 	if (!scratch_len || scratch_len == SIZE_MAX) {
5830 		rc = -EFBIG;
5831 		goto release_acl;
5832 	}
5833 
5834 	pntsd = kvzalloc(scratch_len, KSMBD_DEFAULT_GFP);
5835 	if (!pntsd) {
5836 		rc = -ENOMEM;
5837 		goto release_acl;
5838 	}
5839 
5840 	rc = build_sec_desc(idmap, pntsd, ppntsd, ppntsd_size,
5841 			addition_info, &secdesclen, &fattr);
5842 
5843 release_acl:
5844 	posix_acl_release(fattr.cf_acls);
5845 	posix_acl_release(fattr.cf_dacls);
5846 	kfree(ppntsd);
5847 	ksmbd_fd_put(work, fp);
5848 
5849 	if (!rc && ALIGN(secdesclen, 8) > scratch_len)
5850 		rc = -EFBIG;
5851 	if (rc)
5852 		goto err_out;
5853 
5854 	rsp->OutputBufferLength = cpu_to_le32(secdesclen);
5855 	rc = buffer_check_err(le32_to_cpu(req->OutputBufferLength),
5856 			      rsp, work->response_buf);
5857 	if (rc)
5858 		goto err_out;
5859 
5860 	rc = ksmbd_iov_pin_rsp_read(work, (void *)rsp,
5861 			offsetof(struct smb2_query_info_rsp, Buffer),
5862 			pntsd, secdesclen);
5863 err_out:
5864 	if (rc) {
5865 		rsp->OutputBufferLength = 0;
5866 		kvfree(pntsd);
5867 	}
5868 
5869 	return rc;
5870 }
5871 
5872 /**
5873  * smb2_query_info() - handler for smb2 query info command
5874  * @work:	smb work containing query info request buffer
5875  *
5876  * Return:	0 on success, otherwise error
5877  */
5878 int smb2_query_info(struct ksmbd_work *work)
5879 {
5880 	struct smb2_query_info_req *req;
5881 	struct smb2_query_info_rsp *rsp;
5882 	int rc = 0;
5883 
5884 	ksmbd_debug(SMB, "Received request smb2 query info request\n");
5885 
5886 	WORK_BUFFERS(work, req, rsp);
5887 
5888 	if (ksmbd_override_fsids(work)) {
5889 		rc = -ENOMEM;
5890 		goto err_out;
5891 	}
5892 
5893 	rsp->StructureSize = cpu_to_le16(9);
5894 	rsp->OutputBufferOffset = cpu_to_le16(72);
5895 
5896 	switch (req->InfoType) {
5897 	case SMB2_O_INFO_FILE:
5898 		ksmbd_debug(SMB, "GOT SMB2_O_INFO_FILE\n");
5899 		rc = smb2_get_info_file(work, req, rsp);
5900 		break;
5901 	case SMB2_O_INFO_FILESYSTEM:
5902 		ksmbd_debug(SMB, "GOT SMB2_O_INFO_FILESYSTEM\n");
5903 		rc = smb2_get_info_filesystem(work, req, rsp);
5904 		break;
5905 	case SMB2_O_INFO_SECURITY:
5906 		ksmbd_debug(SMB, "GOT SMB2_O_INFO_SECURITY\n");
5907 		rc = smb2_get_info_sec(work, req, rsp);
5908 		break;
5909 	default:
5910 		ksmbd_debug(SMB, "InfoType %d not supported yet\n",
5911 			    req->InfoType);
5912 		rc = -EOPNOTSUPP;
5913 	}
5914 	ksmbd_revert_fsids(work);
5915 
5916 err_out:
5917 	if (rc < 0) {
5918 		if (rc == -EACCES)
5919 			rsp->hdr.Status = STATUS_ACCESS_DENIED;
5920 		else if (rc == -ENOENT)
5921 			rsp->hdr.Status = STATUS_FILE_CLOSED;
5922 		else if (rc == -EIO)
5923 			rsp->hdr.Status = STATUS_UNEXPECTED_IO_ERROR;
5924 		else if (rc == -ENOMEM)
5925 			rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
5926 		else if (rc == -EINVAL && rsp->hdr.Status == 0)
5927 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
5928 		else if (rc == -EOPNOTSUPP || rsp->hdr.Status == 0)
5929 			rsp->hdr.Status = STATUS_INVALID_INFO_CLASS;
5930 		smb2_set_err_rsp(work);
5931 
5932 		ksmbd_debug(SMB, "error while processing smb2 query rc = %d\n",
5933 			    rc);
5934 		return rc;
5935 	}
5936 	return 0;
5937 }
5938 
5939 /**
5940  * smb2_close_pipe() - handler for closing IPC pipe
5941  * @work:	smb work containing close request buffer
5942  *
5943  * Return:	0
5944  */
5945 static noinline int smb2_close_pipe(struct ksmbd_work *work)
5946 {
5947 	u64 id;
5948 	struct smb2_close_req *req;
5949 	struct smb2_close_rsp *rsp;
5950 
5951 	WORK_BUFFERS(work, req, rsp);
5952 
5953 	id = req->VolatileFileId;
5954 	ksmbd_session_rpc_close(work->sess, id);
5955 
5956 	rsp->StructureSize = cpu_to_le16(60);
5957 	rsp->Flags = 0;
5958 	rsp->Reserved = 0;
5959 	rsp->CreationTime = 0;
5960 	rsp->LastAccessTime = 0;
5961 	rsp->LastWriteTime = 0;
5962 	rsp->ChangeTime = 0;
5963 	rsp->AllocationSize = 0;
5964 	rsp->EndOfFile = 0;
5965 	rsp->Attributes = 0;
5966 
5967 	return ksmbd_iov_pin_rsp(work, (void *)rsp,
5968 				 sizeof(struct smb2_close_rsp));
5969 }
5970 
5971 /**
5972  * smb2_close() - handler for smb2 close file command
5973  * @work:	smb work containing close request buffer
5974  *
5975  * Return:	0 on success, otherwise error
5976  */
5977 int smb2_close(struct ksmbd_work *work)
5978 {
5979 	u64 volatile_id = KSMBD_NO_FID;
5980 	u64 sess_id;
5981 	struct smb2_close_req *req;
5982 	struct smb2_close_rsp *rsp;
5983 	struct ksmbd_conn *conn = work->conn;
5984 	struct ksmbd_file *fp;
5985 	u64 time;
5986 	int err = 0;
5987 
5988 	ksmbd_debug(SMB, "Received smb2 close request\n");
5989 
5990 	WORK_BUFFERS(work, req, rsp);
5991 
5992 	if (test_share_config_flag(work->tcon->share_conf,
5993 				   KSMBD_SHARE_FLAG_PIPE)) {
5994 		ksmbd_debug(SMB, "IPC pipe close request\n");
5995 		return smb2_close_pipe(work);
5996 	}
5997 
5998 	sess_id = le64_to_cpu(req->hdr.SessionId);
5999 	if (req->hdr.Flags & SMB2_FLAGS_RELATED_OPERATIONS)
6000 		sess_id = work->compound_sid;
6001 
6002 	work->compound_sid = 0;
6003 	if (check_session_id(conn, sess_id)) {
6004 		work->compound_sid = sess_id;
6005 	} else {
6006 		rsp->hdr.Status = STATUS_USER_SESSION_DELETED;
6007 		if (req->hdr.Flags & SMB2_FLAGS_RELATED_OPERATIONS)
6008 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
6009 		err = -EBADF;
6010 		goto out;
6011 	}
6012 
6013 	if (work->next_smb2_rcv_hdr_off &&
6014 	    !has_file_id(req->VolatileFileId)) {
6015 		if (!has_file_id(work->compound_fid)) {
6016 			/* file already closed, return FILE_CLOSED */
6017 			ksmbd_debug(SMB, "file already closed\n");
6018 			rsp->hdr.Status = STATUS_FILE_CLOSED;
6019 			err = -EBADF;
6020 			goto out;
6021 		} else {
6022 			ksmbd_debug(SMB,
6023 				    "Compound request set FID = %llu:%llu\n",
6024 				    work->compound_fid,
6025 				    work->compound_pfid);
6026 			volatile_id = work->compound_fid;
6027 
6028 			/* file closed, stored id is not valid anymore */
6029 			work->compound_fid = KSMBD_NO_FID;
6030 			work->compound_pfid = KSMBD_NO_FID;
6031 		}
6032 	} else {
6033 		volatile_id = req->VolatileFileId;
6034 	}
6035 	ksmbd_debug(SMB, "volatile_id = %llu\n", volatile_id);
6036 
6037 	rsp->StructureSize = cpu_to_le16(60);
6038 	rsp->Reserved = 0;
6039 
6040 	if (req->Flags == SMB2_CLOSE_FLAG_POSTQUERY_ATTRIB) {
6041 		struct kstat stat;
6042 		int ret;
6043 
6044 		fp = ksmbd_lookup_fd_fast(work, volatile_id);
6045 		if (!fp) {
6046 			err = -ENOENT;
6047 			goto out;
6048 		}
6049 
6050 		ret = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
6051 				  AT_STATX_SYNC_AS_STAT);
6052 		if (ret) {
6053 			ksmbd_fd_put(work, fp);
6054 			goto out;
6055 		}
6056 
6057 		rsp->Flags = SMB2_CLOSE_FLAG_POSTQUERY_ATTRIB;
6058 		rsp->AllocationSize = S_ISDIR(stat.mode) ? 0 :
6059 			cpu_to_le64(stat.blocks << 9);
6060 		rsp->EndOfFile = cpu_to_le64(stat.size);
6061 		rsp->Attributes = fp->f_ci->m_fattr;
6062 		rsp->CreationTime = cpu_to_le64(fp->create_time);
6063 		time = ksmbd_UnixTimeToNT(stat.atime);
6064 		rsp->LastAccessTime = cpu_to_le64(time);
6065 		time = ksmbd_UnixTimeToNT(stat.mtime);
6066 		rsp->LastWriteTime = cpu_to_le64(time);
6067 		time = ksmbd_UnixTimeToNT(stat.ctime);
6068 		rsp->ChangeTime = cpu_to_le64(time);
6069 		ksmbd_fd_put(work, fp);
6070 	} else {
6071 		rsp->Flags = 0;
6072 		rsp->AllocationSize = 0;
6073 		rsp->EndOfFile = 0;
6074 		rsp->Attributes = 0;
6075 		rsp->CreationTime = 0;
6076 		rsp->LastAccessTime = 0;
6077 		rsp->LastWriteTime = 0;
6078 		rsp->ChangeTime = 0;
6079 	}
6080 
6081 	err = ksmbd_close_fd(work, volatile_id);
6082 out:
6083 	if (!err)
6084 		err = ksmbd_iov_pin_rsp(work, (void *)rsp,
6085 					sizeof(struct smb2_close_rsp));
6086 
6087 	if (err) {
6088 		if (rsp->hdr.Status == 0)
6089 			rsp->hdr.Status = STATUS_FILE_CLOSED;
6090 		smb2_set_err_rsp(work);
6091 	}
6092 
6093 	return err;
6094 }
6095 
6096 /**
6097  * smb2_echo() - handler for smb2 echo(ping) command
6098  * @work:	smb work containing echo request buffer
6099  *
6100  * Return:	0 on success, otherwise error
6101  */
6102 int smb2_echo(struct ksmbd_work *work)
6103 {
6104 	struct smb2_echo_rsp *rsp = smb_get_msg(work->response_buf);
6105 
6106 	ksmbd_debug(SMB, "Received smb2 echo request\n");
6107 
6108 	if (work->next_smb2_rcv_hdr_off)
6109 		rsp = ksmbd_resp_buf_next(work);
6110 
6111 	rsp->StructureSize = cpu_to_le16(4);
6112 	rsp->Reserved = 0;
6113 	return ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_echo_rsp));
6114 }
6115 
6116 static int smb2_rename(struct ksmbd_work *work,
6117 		       struct ksmbd_file *fp,
6118 		       struct smb2_file_rename_info *file_info,
6119 		       struct nls_table *local_nls)
6120 {
6121 	struct ksmbd_share_config *share = fp->tcon->share_conf;
6122 	char *new_name = NULL;
6123 	int rc, flags = 0;
6124 
6125 	ksmbd_debug(SMB, "setting FILE_RENAME_INFO\n");
6126 	new_name = smb2_get_name(file_info->FileName,
6127 				 le32_to_cpu(file_info->FileNameLength),
6128 				 local_nls);
6129 	if (IS_ERR(new_name))
6130 		return PTR_ERR(new_name);
6131 
6132 	if (fp->is_posix_ctxt == false && strchr(new_name, ':')) {
6133 		int s_type;
6134 		char *xattr_stream_name, *stream_name = NULL;
6135 		size_t xattr_stream_size;
6136 		int len;
6137 
6138 		rc = parse_stream_name(new_name, &stream_name, &s_type);
6139 		if (rc < 0)
6140 			goto out;
6141 
6142 		len = strlen(new_name);
6143 		if (len > 0 && new_name[len - 1] != '/') {
6144 			pr_err("not allow base filename in rename\n");
6145 			rc = -ESHARE;
6146 			goto out;
6147 		}
6148 
6149 		rc = ksmbd_vfs_xattr_stream_name(stream_name,
6150 						 &xattr_stream_name,
6151 						 &xattr_stream_size,
6152 						 s_type);
6153 		if (rc)
6154 			goto out;
6155 
6156 		rc = ksmbd_vfs_setxattr(file_mnt_idmap(fp->filp),
6157 					&fp->filp->f_path,
6158 					xattr_stream_name,
6159 					NULL, 0, 0, true);
6160 		if (rc < 0) {
6161 			pr_err("failed to store stream name in xattr: %d\n",
6162 			       rc);
6163 			rc = -EINVAL;
6164 			goto out;
6165 		}
6166 
6167 		goto out;
6168 	}
6169 
6170 	ksmbd_debug(SMB, "new name %s\n", new_name);
6171 	if (ksmbd_share_veto_filename(share, new_name)) {
6172 		rc = -ENOENT;
6173 		ksmbd_debug(SMB, "Can't rename vetoed file: %s\n", new_name);
6174 		goto out;
6175 	}
6176 
6177 	if (!file_info->ReplaceIfExists)
6178 		flags = RENAME_NOREPLACE;
6179 
6180 	rc = ksmbd_vfs_rename(work, &fp->filp->f_path, new_name, flags);
6181 	if (!rc)
6182 		smb_break_all_levII_oplock(work, fp, 0);
6183 out:
6184 	kfree(new_name);
6185 	return rc;
6186 }
6187 
6188 static int smb2_create_link(struct ksmbd_work *work,
6189 			    struct ksmbd_share_config *share,
6190 			    struct smb2_file_link_info *file_info,
6191 			    unsigned int buf_len, struct file *filp,
6192 			    struct nls_table *local_nls)
6193 {
6194 	char *link_name = NULL, *target_name = NULL, *pathname = NULL;
6195 	struct path path;
6196 	int rc;
6197 
6198 	if (buf_len < (u64)sizeof(struct smb2_file_link_info) +
6199 			le32_to_cpu(file_info->FileNameLength))
6200 		return -EINVAL;
6201 
6202 	ksmbd_debug(SMB, "setting FILE_LINK_INFORMATION\n");
6203 	pathname = kmalloc(PATH_MAX, KSMBD_DEFAULT_GFP);
6204 	if (!pathname)
6205 		return -ENOMEM;
6206 
6207 	link_name = smb2_get_name(file_info->FileName,
6208 				  le32_to_cpu(file_info->FileNameLength),
6209 				  local_nls);
6210 	if (IS_ERR(link_name) || S_ISDIR(file_inode(filp)->i_mode)) {
6211 		rc = -EINVAL;
6212 		goto out;
6213 	}
6214 
6215 	ksmbd_debug(SMB, "link name is %s\n", link_name);
6216 	target_name = file_path(filp, pathname, PATH_MAX);
6217 	if (IS_ERR(target_name)) {
6218 		rc = -EINVAL;
6219 		goto out;
6220 	}
6221 
6222 	ksmbd_debug(SMB, "target name is %s\n", target_name);
6223 	rc = ksmbd_vfs_kern_path_start_removing(work, link_name, LOOKUP_NO_SYMLINKS,
6224 						&path, 0);
6225 	if (rc) {
6226 		if (rc != -ENOENT)
6227 			goto out;
6228 	} else {
6229 		if (file_info->ReplaceIfExists) {
6230 			rc = ksmbd_vfs_remove_file(work, &path);
6231 			if (rc) {
6232 				rc = -EINVAL;
6233 				ksmbd_debug(SMB, "cannot delete %s\n",
6234 					    link_name);
6235 			}
6236 		} else {
6237 			rc = -EEXIST;
6238 			ksmbd_debug(SMB, "link already exists\n");
6239 		}
6240 		ksmbd_vfs_kern_path_end_removing(&path);
6241 		if (rc)
6242 			goto out;
6243 	}
6244 	rc = ksmbd_vfs_link(work, target_name, link_name);
6245 	if (rc)
6246 		rc = -EINVAL;
6247 out:
6248 
6249 	if (!IS_ERR(link_name))
6250 		kfree(link_name);
6251 	kfree(pathname);
6252 	return rc;
6253 }
6254 
6255 static int set_file_basic_info(struct ksmbd_file *fp,
6256 			       struct file_basic_info *file_info,
6257 			       struct ksmbd_share_config *share)
6258 {
6259 	struct iattr attrs;
6260 	struct file *filp;
6261 	struct inode *inode;
6262 	struct mnt_idmap *idmap;
6263 	int rc = 0;
6264 
6265 	if (!(fp->daccess & FILE_WRITE_ATTRIBUTES_LE))
6266 		return -EACCES;
6267 
6268 	attrs.ia_valid = 0;
6269 	filp = fp->filp;
6270 	inode = file_inode(filp);
6271 	idmap = file_mnt_idmap(filp);
6272 
6273 	if (file_info->CreationTime)
6274 		fp->create_time = le64_to_cpu(file_info->CreationTime);
6275 
6276 	if (file_info->LastAccessTime) {
6277 		attrs.ia_atime = ksmbd_NTtimeToUnix(file_info->LastAccessTime);
6278 		attrs.ia_valid |= (ATTR_ATIME | ATTR_ATIME_SET);
6279 	}
6280 
6281 	if (file_info->ChangeTime)
6282 		inode_set_ctime_to_ts(inode,
6283 				ksmbd_NTtimeToUnix(file_info->ChangeTime));
6284 
6285 	if (file_info->LastWriteTime) {
6286 		attrs.ia_mtime = ksmbd_NTtimeToUnix(file_info->LastWriteTime);
6287 		attrs.ia_valid |= (ATTR_MTIME | ATTR_MTIME_SET | ATTR_CTIME);
6288 	}
6289 
6290 	if (file_info->Attributes) {
6291 		if (!S_ISDIR(inode->i_mode) &&
6292 		    file_info->Attributes & FILE_ATTRIBUTE_DIRECTORY_LE) {
6293 			pr_err("can't change a file to a directory\n");
6294 			return -EINVAL;
6295 		}
6296 
6297 		if (!(S_ISDIR(inode->i_mode) && file_info->Attributes == FILE_ATTRIBUTE_NORMAL_LE))
6298 			fp->f_ci->m_fattr = file_info->Attributes |
6299 				(fp->f_ci->m_fattr & FILE_ATTRIBUTE_DIRECTORY_LE);
6300 	}
6301 
6302 	if (test_share_config_flag(share, KSMBD_SHARE_FLAG_STORE_DOS_ATTRS) &&
6303 	    (file_info->CreationTime || file_info->Attributes)) {
6304 		struct xattr_dos_attrib da = {0};
6305 
6306 		da.version = 4;
6307 		da.itime = fp->itime;
6308 		da.create_time = fp->create_time;
6309 		da.attr = le32_to_cpu(fp->f_ci->m_fattr);
6310 		da.flags = XATTR_DOSINFO_ATTRIB | XATTR_DOSINFO_CREATE_TIME |
6311 			XATTR_DOSINFO_ITIME;
6312 
6313 		rc = ksmbd_vfs_set_dos_attrib_xattr(idmap, &filp->f_path, &da,
6314 				true);
6315 		if (rc)
6316 			ksmbd_debug(SMB,
6317 				    "failed to restore file attribute in EA\n");
6318 		rc = 0;
6319 	}
6320 
6321 	if (attrs.ia_valid) {
6322 		struct dentry *dentry = filp->f_path.dentry;
6323 		struct inode *inode = d_inode(dentry);
6324 
6325 		if (IS_IMMUTABLE(inode) || IS_APPEND(inode))
6326 			return -EACCES;
6327 
6328 		inode_lock(inode);
6329 		rc = notify_change(idmap, dentry, &attrs, NULL);
6330 		inode_unlock(inode);
6331 	}
6332 	return rc;
6333 }
6334 
6335 static int set_file_allocation_info(struct ksmbd_work *work,
6336 				    struct ksmbd_file *fp,
6337 				    struct smb2_file_alloc_info *file_alloc_info)
6338 {
6339 	/*
6340 	 * TODO : It's working fine only when store dos attributes
6341 	 * is not yes. need to implement a logic which works
6342 	 * properly with any smb.conf option
6343 	 */
6344 
6345 	loff_t alloc_blks;
6346 	struct inode *inode;
6347 	struct kstat stat;
6348 	int rc;
6349 
6350 	if (!(fp->daccess & FILE_WRITE_DATA_LE))
6351 		return -EACCES;
6352 
6353 	if (ksmbd_stream_fd(fp) == true)
6354 		return 0;
6355 
6356 	rc = vfs_getattr(&fp->filp->f_path, &stat, STATX_BASIC_STATS,
6357 			 AT_STATX_SYNC_AS_STAT);
6358 	if (rc)
6359 		return rc;
6360 
6361 	alloc_blks = (le64_to_cpu(file_alloc_info->AllocationSize) + 511) >> 9;
6362 	inode = file_inode(fp->filp);
6363 
6364 	if (alloc_blks > stat.blocks) {
6365 		smb_break_all_levII_oplock(work, fp, 1);
6366 		rc = vfs_fallocate(fp->filp, FALLOC_FL_KEEP_SIZE, 0,
6367 				   alloc_blks * 512);
6368 		if (rc && rc != -EOPNOTSUPP) {
6369 			pr_err("vfs_fallocate is failed : %d\n", rc);
6370 			return rc;
6371 		}
6372 	} else if (alloc_blks < stat.blocks) {
6373 		loff_t size;
6374 
6375 		/*
6376 		 * Allocation size could be smaller than original one
6377 		 * which means allocated blocks in file should be
6378 		 * deallocated. use truncate to cut out it, but inode
6379 		 * size is also updated with truncate offset.
6380 		 * inode size is retained by backup inode size.
6381 		 */
6382 		size = i_size_read(inode);
6383 		rc = ksmbd_vfs_truncate(work, fp, alloc_blks * 512);
6384 		if (rc) {
6385 			pr_err("truncate failed!, err %d\n", rc);
6386 			return rc;
6387 		}
6388 		if (size < alloc_blks * 512)
6389 			i_size_write(inode, size);
6390 	}
6391 	return 0;
6392 }
6393 
6394 static int set_end_of_file_info(struct ksmbd_work *work, struct ksmbd_file *fp,
6395 				struct smb2_file_eof_info *file_eof_info)
6396 {
6397 	loff_t newsize;
6398 	struct inode *inode;
6399 	int rc;
6400 
6401 	if (!(fp->daccess & FILE_WRITE_DATA_LE))
6402 		return -EACCES;
6403 
6404 	newsize = le64_to_cpu(file_eof_info->EndOfFile);
6405 	inode = file_inode(fp->filp);
6406 
6407 	/*
6408 	 * If FILE_END_OF_FILE_INFORMATION of set_info_file is called
6409 	 * on FAT32 shared device, truncate execution time is too long
6410 	 * and network error could cause from windows client. because
6411 	 * truncate of some filesystem like FAT32 fill zero data in
6412 	 * truncated range.
6413 	 */
6414 	if (inode->i_sb->s_magic != MSDOS_SUPER_MAGIC &&
6415 	    ksmbd_stream_fd(fp) == false) {
6416 		ksmbd_debug(SMB, "truncated to newsize %lld\n", newsize);
6417 		rc = ksmbd_vfs_truncate(work, fp, newsize);
6418 		if (rc) {
6419 			ksmbd_debug(SMB, "truncate failed!, err %d\n", rc);
6420 			if (rc != -EAGAIN)
6421 				rc = -EBADF;
6422 			return rc;
6423 		}
6424 	}
6425 	return 0;
6426 }
6427 
6428 static int set_rename_info(struct ksmbd_work *work, struct ksmbd_file *fp,
6429 			   struct smb2_file_rename_info *rename_info,
6430 			   unsigned int buf_len)
6431 {
6432 	if (!(fp->daccess & FILE_DELETE_LE)) {
6433 		pr_err("no right to delete : 0x%x\n", fp->daccess);
6434 		return -EACCES;
6435 	}
6436 
6437 	if (buf_len < (u64)sizeof(struct smb2_file_rename_info) +
6438 			le32_to_cpu(rename_info->FileNameLength))
6439 		return -EINVAL;
6440 
6441 	if (!le32_to_cpu(rename_info->FileNameLength))
6442 		return -EINVAL;
6443 
6444 	return smb2_rename(work, fp, rename_info, work->conn->local_nls);
6445 }
6446 
6447 static int set_file_disposition_info(struct ksmbd_file *fp,
6448 				     struct smb2_file_disposition_info *file_info)
6449 {
6450 	struct inode *inode;
6451 
6452 	if (!(fp->daccess & FILE_DELETE_LE)) {
6453 		pr_err("no right to delete : 0x%x\n", fp->daccess);
6454 		return -EACCES;
6455 	}
6456 
6457 	inode = file_inode(fp->filp);
6458 	if (file_info->DeletePending) {
6459 		if (S_ISDIR(inode->i_mode) &&
6460 		    ksmbd_vfs_empty_dir(fp) == -ENOTEMPTY)
6461 			return -EBUSY;
6462 		ksmbd_set_inode_pending_delete(fp);
6463 	} else {
6464 		ksmbd_clear_inode_pending_delete(fp);
6465 	}
6466 	return 0;
6467 }
6468 
6469 static int set_file_position_info(struct ksmbd_file *fp,
6470 				  struct smb2_file_pos_info *file_info)
6471 {
6472 	loff_t current_byte_offset;
6473 	unsigned long sector_size;
6474 	struct inode *inode;
6475 
6476 	inode = file_inode(fp->filp);
6477 	current_byte_offset = le64_to_cpu(file_info->CurrentByteOffset);
6478 	sector_size = inode->i_sb->s_blocksize;
6479 
6480 	if (current_byte_offset < 0 ||
6481 	    (fp->coption == FILE_NO_INTERMEDIATE_BUFFERING_LE &&
6482 	     current_byte_offset & (sector_size - 1))) {
6483 		pr_err("CurrentByteOffset is not valid : %llu\n",
6484 		       current_byte_offset);
6485 		return -EINVAL;
6486 	}
6487 
6488 	if (ksmbd_stream_fd(fp) == false)
6489 		fp->filp->f_pos = current_byte_offset;
6490 	else {
6491 		if (current_byte_offset > XATTR_SIZE_MAX)
6492 			current_byte_offset = XATTR_SIZE_MAX;
6493 		fp->stream.pos = current_byte_offset;
6494 	}
6495 	return 0;
6496 }
6497 
6498 static int set_file_mode_info(struct ksmbd_file *fp,
6499 			      struct smb2_file_mode_info *file_info)
6500 {
6501 	__le32 mode;
6502 
6503 	mode = file_info->Mode;
6504 
6505 	if ((mode & ~FILE_MODE_INFO_MASK)) {
6506 		pr_err("Mode is not valid : 0x%x\n", le32_to_cpu(mode));
6507 		return -EINVAL;
6508 	}
6509 
6510 	/*
6511 	 * TODO : need to implement consideration for
6512 	 * FILE_SYNCHRONOUS_IO_ALERT and FILE_SYNCHRONOUS_IO_NONALERT
6513 	 */
6514 	ksmbd_vfs_set_fadvise(fp->filp, mode);
6515 	fp->coption = mode;
6516 	return 0;
6517 }
6518 
6519 /**
6520  * smb2_set_info_file() - handler for smb2 set info command
6521  * @work:	smb work containing set info command buffer
6522  * @fp:		ksmbd_file pointer
6523  * @req:	request buffer pointer
6524  * @share:	ksmbd_share_config pointer
6525  *
6526  * Return:	0 on success, otherwise error
6527  */
6528 static int smb2_set_info_file(struct ksmbd_work *work, struct ksmbd_file *fp,
6529 			      struct smb2_set_info_req *req,
6530 			      struct ksmbd_share_config *share)
6531 {
6532 	unsigned int buf_len = le32_to_cpu(req->BufferLength);
6533 	char *buffer = (char *)req + le16_to_cpu(req->BufferOffset);
6534 
6535 	switch (req->FileInfoClass) {
6536 	case FILE_BASIC_INFORMATION:
6537 	{
6538 		if (buf_len < sizeof(struct file_basic_info))
6539 			return -EMSGSIZE;
6540 
6541 		return set_file_basic_info(fp, (struct file_basic_info *)buffer, share);
6542 	}
6543 	case FILE_ALLOCATION_INFORMATION:
6544 	{
6545 		if (buf_len < sizeof(struct smb2_file_alloc_info))
6546 			return -EMSGSIZE;
6547 
6548 		return set_file_allocation_info(work, fp,
6549 						(struct smb2_file_alloc_info *)buffer);
6550 	}
6551 	case FILE_END_OF_FILE_INFORMATION:
6552 	{
6553 		if (buf_len < sizeof(struct smb2_file_eof_info))
6554 			return -EMSGSIZE;
6555 
6556 		return set_end_of_file_info(work, fp,
6557 					    (struct smb2_file_eof_info *)buffer);
6558 	}
6559 	case FILE_RENAME_INFORMATION:
6560 	{
6561 		if (buf_len < sizeof(struct smb2_file_rename_info))
6562 			return -EMSGSIZE;
6563 
6564 		return set_rename_info(work, fp,
6565 				       (struct smb2_file_rename_info *)buffer,
6566 				       buf_len);
6567 	}
6568 	case FILE_LINK_INFORMATION:
6569 	{
6570 		if (buf_len < sizeof(struct smb2_file_link_info))
6571 			return -EMSGSIZE;
6572 
6573 		return smb2_create_link(work, work->tcon->share_conf,
6574 					(struct smb2_file_link_info *)buffer,
6575 					buf_len, fp->filp,
6576 					work->conn->local_nls);
6577 	}
6578 	case FILE_DISPOSITION_INFORMATION:
6579 	{
6580 		if (buf_len < sizeof(struct smb2_file_disposition_info))
6581 			return -EMSGSIZE;
6582 
6583 		return set_file_disposition_info(fp,
6584 						 (struct smb2_file_disposition_info *)buffer);
6585 	}
6586 	case FILE_FULL_EA_INFORMATION:
6587 	{
6588 		if (!(fp->daccess & FILE_WRITE_EA_LE)) {
6589 			pr_err("Not permitted to write ext  attr: 0x%x\n",
6590 			       fp->daccess);
6591 			return -EACCES;
6592 		}
6593 
6594 		if (buf_len < sizeof(struct smb2_ea_info))
6595 			return -EMSGSIZE;
6596 
6597 		return smb2_set_ea((struct smb2_ea_info *)buffer,
6598 				   buf_len, &fp->filp->f_path, true);
6599 	}
6600 	case FILE_POSITION_INFORMATION:
6601 	{
6602 		if (buf_len < sizeof(struct smb2_file_pos_info))
6603 			return -EMSGSIZE;
6604 
6605 		return set_file_position_info(fp, (struct smb2_file_pos_info *)buffer);
6606 	}
6607 	case FILE_MODE_INFORMATION:
6608 	{
6609 		if (buf_len < sizeof(struct smb2_file_mode_info))
6610 			return -EMSGSIZE;
6611 
6612 		return set_file_mode_info(fp, (struct smb2_file_mode_info *)buffer);
6613 	}
6614 	}
6615 
6616 	pr_err("Unimplemented Fileinfoclass :%d\n", req->FileInfoClass);
6617 	return -EOPNOTSUPP;
6618 }
6619 
6620 static int smb2_set_info_sec(struct ksmbd_file *fp, int addition_info,
6621 			     char *buffer, int buf_len)
6622 {
6623 	struct smb_ntsd *pntsd = (struct smb_ntsd *)buffer;
6624 
6625 	fp->saccess |= FILE_SHARE_DELETE_LE;
6626 
6627 	return set_info_sec(fp->conn, fp->tcon, &fp->filp->f_path, pntsd,
6628 			buf_len, false, true);
6629 }
6630 
6631 /**
6632  * smb2_set_info() - handler for smb2 set info command handler
6633  * @work:	smb work containing set info request buffer
6634  *
6635  * Return:	0 on success, otherwise error
6636  */
6637 int smb2_set_info(struct ksmbd_work *work)
6638 {
6639 	struct smb2_set_info_req *req;
6640 	struct smb2_set_info_rsp *rsp;
6641 	struct ksmbd_file *fp = NULL;
6642 	int rc = 0;
6643 	unsigned int id = KSMBD_NO_FID, pid = KSMBD_NO_FID;
6644 
6645 	ksmbd_debug(SMB, "Received smb2 set info request\n");
6646 
6647 	if (work->next_smb2_rcv_hdr_off) {
6648 		req = ksmbd_req_buf_next(work);
6649 		rsp = ksmbd_resp_buf_next(work);
6650 		if (!has_file_id(req->VolatileFileId)) {
6651 			ksmbd_debug(SMB, "Compound request set FID = %llu\n",
6652 				    work->compound_fid);
6653 			id = work->compound_fid;
6654 			pid = work->compound_pfid;
6655 		}
6656 	} else {
6657 		req = smb_get_msg(work->request_buf);
6658 		rsp = smb_get_msg(work->response_buf);
6659 	}
6660 
6661 	if (!test_tree_conn_flag(work->tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
6662 		ksmbd_debug(SMB, "User does not have write permission\n");
6663 		pr_err("User does not have write permission\n");
6664 		rc = -EACCES;
6665 		goto err_out;
6666 	}
6667 
6668 	if (!has_file_id(id)) {
6669 		id = req->VolatileFileId;
6670 		pid = req->PersistentFileId;
6671 	}
6672 
6673 	fp = ksmbd_lookup_fd_slow(work, id, pid);
6674 	if (!fp) {
6675 		ksmbd_debug(SMB, "Invalid id for close: %u\n", id);
6676 		rc = -ENOENT;
6677 		goto err_out;
6678 	}
6679 
6680 	switch (req->InfoType) {
6681 	case SMB2_O_INFO_FILE:
6682 		ksmbd_debug(SMB, "GOT SMB2_O_INFO_FILE\n");
6683 		rc = smb2_set_info_file(work, fp, req, work->tcon->share_conf);
6684 		break;
6685 	case SMB2_O_INFO_SECURITY:
6686 		ksmbd_debug(SMB, "GOT SMB2_O_INFO_SECURITY\n");
6687 		if (ksmbd_override_fsids(work)) {
6688 			rc = -ENOMEM;
6689 			goto err_out;
6690 		}
6691 		rc = smb2_set_info_sec(fp,
6692 				       le32_to_cpu(req->AdditionalInformation),
6693 				       (char *)req + le16_to_cpu(req->BufferOffset),
6694 				       le32_to_cpu(req->BufferLength));
6695 		ksmbd_revert_fsids(work);
6696 		break;
6697 	default:
6698 		rc = -EOPNOTSUPP;
6699 	}
6700 
6701 	if (rc < 0)
6702 		goto err_out;
6703 
6704 	rsp->StructureSize = cpu_to_le16(2);
6705 	rc = ksmbd_iov_pin_rsp(work, (void *)rsp,
6706 			       sizeof(struct smb2_set_info_rsp));
6707 	if (rc)
6708 		goto err_out;
6709 	ksmbd_fd_put(work, fp);
6710 	return 0;
6711 
6712 err_out:
6713 	if (rc == -EACCES || rc == -EPERM || rc == -EXDEV)
6714 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
6715 	else if (rc == -EINVAL)
6716 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
6717 	else if (rc == -EMSGSIZE)
6718 		rsp->hdr.Status = STATUS_INFO_LENGTH_MISMATCH;
6719 	else if (rc == -ESHARE)
6720 		rsp->hdr.Status = STATUS_SHARING_VIOLATION;
6721 	else if (rc == -ENOENT)
6722 		rsp->hdr.Status = STATUS_OBJECT_NAME_INVALID;
6723 	else if (rc == -EBUSY || rc == -ENOTEMPTY)
6724 		rsp->hdr.Status = STATUS_DIRECTORY_NOT_EMPTY;
6725 	else if (rc == -EAGAIN)
6726 		rsp->hdr.Status = STATUS_FILE_LOCK_CONFLICT;
6727 	else if (rc == -EBADF || rc == -ESTALE)
6728 		rsp->hdr.Status = STATUS_INVALID_HANDLE;
6729 	else if (rc == -EEXIST)
6730 		rsp->hdr.Status = STATUS_OBJECT_NAME_COLLISION;
6731 	else if (rsp->hdr.Status == 0 || rc == -EOPNOTSUPP)
6732 		rsp->hdr.Status = STATUS_INVALID_INFO_CLASS;
6733 	smb2_set_err_rsp(work);
6734 	ksmbd_fd_put(work, fp);
6735 	ksmbd_debug(SMB, "error while processing smb2 query rc = %d\n", rc);
6736 	return rc;
6737 }
6738 
6739 /**
6740  * smb2_read_pipe() - handler for smb2 read from IPC pipe
6741  * @work:	smb work containing read IPC pipe command buffer
6742  *
6743  * Return:	0 on success, otherwise error
6744  */
6745 static noinline int smb2_read_pipe(struct ksmbd_work *work)
6746 {
6747 	int nbytes = 0, err;
6748 	u64 id;
6749 	struct ksmbd_rpc_command *rpc_resp;
6750 	struct smb2_read_req *req;
6751 	struct smb2_read_rsp *rsp;
6752 
6753 	WORK_BUFFERS(work, req, rsp);
6754 
6755 	id = req->VolatileFileId;
6756 
6757 	rpc_resp = ksmbd_rpc_read(work->sess, id);
6758 	if (rpc_resp) {
6759 		void *aux_payload_buf;
6760 
6761 		if (rpc_resp->flags != KSMBD_RPC_OK) {
6762 			err = -EINVAL;
6763 			goto out;
6764 		}
6765 
6766 		aux_payload_buf =
6767 			kvmalloc(rpc_resp->payload_sz, KSMBD_DEFAULT_GFP);
6768 		if (!aux_payload_buf) {
6769 			err = -ENOMEM;
6770 			goto out;
6771 		}
6772 
6773 		memcpy(aux_payload_buf, rpc_resp->payload, rpc_resp->payload_sz);
6774 
6775 		nbytes = rpc_resp->payload_sz;
6776 		err = ksmbd_iov_pin_rsp_read(work, (void *)rsp,
6777 					     offsetof(struct smb2_read_rsp, Buffer),
6778 					     aux_payload_buf, nbytes);
6779 		if (err) {
6780 			kvfree(aux_payload_buf);
6781 			goto out;
6782 		}
6783 		kvfree(rpc_resp);
6784 	} else {
6785 		err = ksmbd_iov_pin_rsp(work, (void *)rsp,
6786 					offsetof(struct smb2_read_rsp, Buffer));
6787 		if (err)
6788 			goto out;
6789 	}
6790 
6791 	rsp->StructureSize = cpu_to_le16(17);
6792 	rsp->DataOffset = 80;
6793 	rsp->Reserved = 0;
6794 	rsp->DataLength = cpu_to_le32(nbytes);
6795 	rsp->DataRemaining = 0;
6796 	rsp->Flags = 0;
6797 	return 0;
6798 
6799 out:
6800 	rsp->hdr.Status = STATUS_UNEXPECTED_IO_ERROR;
6801 	smb2_set_err_rsp(work);
6802 	kvfree(rpc_resp);
6803 	return err;
6804 }
6805 
6806 static int smb2_set_remote_key_for_rdma(struct ksmbd_work *work,
6807 					struct smbdirect_buffer_descriptor_v1 *desc,
6808 					__le32 Channel,
6809 					__le16 ChannelInfoLength)
6810 {
6811 	unsigned int i, ch_count;
6812 
6813 	if (work->conn->dialect == SMB30_PROT_ID &&
6814 	    Channel != SMB2_CHANNEL_RDMA_V1)
6815 		return -EINVAL;
6816 
6817 	ch_count = le16_to_cpu(ChannelInfoLength) / sizeof(*desc);
6818 	if (ksmbd_debug_types & KSMBD_DEBUG_RDMA) {
6819 		for (i = 0; i < ch_count; i++) {
6820 			pr_info("RDMA r/w request %#x: token %#x, length %#x\n",
6821 				i,
6822 				le32_to_cpu(desc[i].token),
6823 				le32_to_cpu(desc[i].length));
6824 		}
6825 	}
6826 	if (!ch_count)
6827 		return -EINVAL;
6828 
6829 	work->need_invalidate_rkey =
6830 		(Channel == SMB2_CHANNEL_RDMA_V1_INVALIDATE);
6831 	if (Channel == SMB2_CHANNEL_RDMA_V1_INVALIDATE)
6832 		work->remote_key = le32_to_cpu(desc->token);
6833 	return 0;
6834 }
6835 
6836 static ssize_t smb2_read_rdma_channel(struct ksmbd_work *work,
6837 				      struct smb2_read_req *req, void *data_buf,
6838 				      size_t length)
6839 {
6840 	int err;
6841 
6842 	err = ksmbd_conn_rdma_write(work->conn, data_buf, length,
6843 				    (struct smbdirect_buffer_descriptor_v1 *)
6844 				    ((char *)req + le16_to_cpu(req->ReadChannelInfoOffset)),
6845 				    le16_to_cpu(req->ReadChannelInfoLength));
6846 	if (err)
6847 		return err;
6848 
6849 	return length;
6850 }
6851 
6852 /**
6853  * smb2_read() - handler for smb2 read from file
6854  * @work:	smb work containing read command buffer
6855  *
6856  * Return:	0 on success, otherwise error
6857  */
6858 int smb2_read(struct ksmbd_work *work)
6859 {
6860 	struct ksmbd_conn *conn = work->conn;
6861 	struct smb2_read_req *req;
6862 	struct smb2_read_rsp *rsp;
6863 	struct ksmbd_file *fp = NULL;
6864 	loff_t offset;
6865 	size_t length, mincount;
6866 	ssize_t nbytes = 0, remain_bytes = 0;
6867 	int err = 0;
6868 	bool is_rdma_channel = false;
6869 	unsigned int max_read_size = conn->vals->max_read_size;
6870 	unsigned int id = KSMBD_NO_FID, pid = KSMBD_NO_FID;
6871 	void *aux_payload_buf;
6872 
6873 	ksmbd_debug(SMB, "Received smb2 read request\n");
6874 
6875 	if (test_share_config_flag(work->tcon->share_conf,
6876 				   KSMBD_SHARE_FLAG_PIPE)) {
6877 		ksmbd_debug(SMB, "IPC pipe read request\n");
6878 		return smb2_read_pipe(work);
6879 	}
6880 
6881 	if (work->next_smb2_rcv_hdr_off) {
6882 		req = ksmbd_req_buf_next(work);
6883 		rsp = ksmbd_resp_buf_next(work);
6884 		if (!has_file_id(req->VolatileFileId)) {
6885 			ksmbd_debug(SMB, "Compound request set FID = %llu\n",
6886 					work->compound_fid);
6887 			id = work->compound_fid;
6888 			pid = work->compound_pfid;
6889 		}
6890 	} else {
6891 		req = smb_get_msg(work->request_buf);
6892 		rsp = smb_get_msg(work->response_buf);
6893 	}
6894 
6895 	if (!has_file_id(id)) {
6896 		id = req->VolatileFileId;
6897 		pid = req->PersistentFileId;
6898 	}
6899 
6900 	if (req->Channel == SMB2_CHANNEL_RDMA_V1_INVALIDATE ||
6901 	    req->Channel == SMB2_CHANNEL_RDMA_V1) {
6902 		is_rdma_channel = true;
6903 		max_read_size = get_smbd_max_read_write_size(work->conn->transport);
6904 		if (max_read_size == 0) {
6905 			err = -EINVAL;
6906 			goto out;
6907 		}
6908 	}
6909 
6910 	if (is_rdma_channel == true) {
6911 		unsigned int ch_offset = le16_to_cpu(req->ReadChannelInfoOffset);
6912 
6913 		if (ch_offset < offsetof(struct smb2_read_req, Buffer)) {
6914 			err = -EINVAL;
6915 			goto out;
6916 		}
6917 		err = smb2_set_remote_key_for_rdma(work,
6918 						   (struct smbdirect_buffer_descriptor_v1 *)
6919 						   ((char *)req + ch_offset),
6920 						   req->Channel,
6921 						   req->ReadChannelInfoLength);
6922 		if (err)
6923 			goto out;
6924 	}
6925 
6926 	fp = ksmbd_lookup_fd_slow(work, id, pid);
6927 	if (!fp) {
6928 		err = -ENOENT;
6929 		goto out;
6930 	}
6931 
6932 	if (!(fp->daccess & (FILE_READ_DATA_LE | FILE_READ_ATTRIBUTES_LE))) {
6933 		pr_err("Not permitted to read : 0x%x\n", fp->daccess);
6934 		err = -EACCES;
6935 		goto out;
6936 	}
6937 
6938 	offset = le64_to_cpu(req->Offset);
6939 	if (offset < 0) {
6940 		err = -EINVAL;
6941 		goto out;
6942 	}
6943 	length = le32_to_cpu(req->Length);
6944 	mincount = le32_to_cpu(req->MinimumCount);
6945 
6946 	if (length > max_read_size) {
6947 		ksmbd_debug(SMB, "limiting read size to max size(%u)\n",
6948 			    max_read_size);
6949 		err = -EINVAL;
6950 		goto out;
6951 	}
6952 
6953 	ksmbd_debug(SMB, "filename %pD, offset %lld, len %zu\n",
6954 		    fp->filp, offset, length);
6955 
6956 	aux_payload_buf = kvzalloc(ALIGN(length, 8), KSMBD_DEFAULT_GFP);
6957 	if (!aux_payload_buf) {
6958 		err = -ENOMEM;
6959 		goto out;
6960 	}
6961 
6962 	nbytes = ksmbd_vfs_read(work, fp, length, &offset, aux_payload_buf);
6963 	if (nbytes < 0) {
6964 		kvfree(aux_payload_buf);
6965 		err = nbytes;
6966 		goto out;
6967 	}
6968 
6969 	if ((nbytes == 0 && length != 0) || nbytes < mincount) {
6970 		kvfree(aux_payload_buf);
6971 		rsp->hdr.Status = STATUS_END_OF_FILE;
6972 		smb2_set_err_rsp(work);
6973 		ksmbd_fd_put(work, fp);
6974 		return -ENODATA;
6975 	}
6976 
6977 	ksmbd_debug(SMB, "nbytes %zu, offset %lld mincount %zu\n",
6978 		    nbytes, offset, mincount);
6979 
6980 	if (is_rdma_channel == true) {
6981 		/* write data to the client using rdma channel */
6982 		remain_bytes = smb2_read_rdma_channel(work, req,
6983 						      aux_payload_buf,
6984 						      nbytes);
6985 		kvfree(aux_payload_buf);
6986 		aux_payload_buf = NULL;
6987 		nbytes = 0;
6988 		if (remain_bytes < 0) {
6989 			err = (int)remain_bytes;
6990 			goto out;
6991 		}
6992 	}
6993 
6994 	rsp->StructureSize = cpu_to_le16(17);
6995 	rsp->DataOffset = 80;
6996 	rsp->Reserved = 0;
6997 	rsp->DataLength = cpu_to_le32(nbytes);
6998 	rsp->DataRemaining = cpu_to_le32(remain_bytes);
6999 	rsp->Flags = 0;
7000 	err = ksmbd_iov_pin_rsp_read(work, (void *)rsp,
7001 				     offsetof(struct smb2_read_rsp, Buffer),
7002 				     aux_payload_buf, nbytes);
7003 	if (err) {
7004 		kvfree(aux_payload_buf);
7005 		goto out;
7006 	}
7007 	ksmbd_fd_put(work, fp);
7008 	return 0;
7009 
7010 out:
7011 	if (err) {
7012 		if (err == -EISDIR)
7013 			rsp->hdr.Status = STATUS_INVALID_DEVICE_REQUEST;
7014 		else if (err == -EAGAIN)
7015 			rsp->hdr.Status = STATUS_FILE_LOCK_CONFLICT;
7016 		else if (err == -ENOENT)
7017 			rsp->hdr.Status = STATUS_FILE_CLOSED;
7018 		else if (err == -EACCES)
7019 			rsp->hdr.Status = STATUS_ACCESS_DENIED;
7020 		else if (err == -ESHARE)
7021 			rsp->hdr.Status = STATUS_SHARING_VIOLATION;
7022 		else if (err == -EINVAL)
7023 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7024 		else
7025 			rsp->hdr.Status = STATUS_INVALID_HANDLE;
7026 
7027 		smb2_set_err_rsp(work);
7028 	}
7029 	ksmbd_fd_put(work, fp);
7030 	return err;
7031 }
7032 
7033 /**
7034  * smb2_write_pipe() - handler for smb2 write on IPC pipe
7035  * @work:	smb work containing write IPC pipe command buffer
7036  *
7037  * Return:	0 on success, otherwise error
7038  */
7039 static noinline int smb2_write_pipe(struct ksmbd_work *work)
7040 {
7041 	struct smb2_write_req *req;
7042 	struct smb2_write_rsp *rsp;
7043 	struct ksmbd_rpc_command *rpc_resp;
7044 	u64 id = 0;
7045 	int err = 0, ret = 0;
7046 	char *data_buf;
7047 	size_t length;
7048 
7049 	WORK_BUFFERS(work, req, rsp);
7050 
7051 	length = le32_to_cpu(req->Length);
7052 	id = req->VolatileFileId;
7053 
7054 	if ((u64)le16_to_cpu(req->DataOffset) + length >
7055 	    get_rfc1002_len(work->request_buf)) {
7056 		pr_err("invalid write data offset %u, smb_len %u\n",
7057 		       le16_to_cpu(req->DataOffset),
7058 		       get_rfc1002_len(work->request_buf));
7059 		err = -EINVAL;
7060 		goto out;
7061 	}
7062 
7063 	data_buf = (char *)(((char *)&req->hdr.ProtocolId) +
7064 			   le16_to_cpu(req->DataOffset));
7065 
7066 	rpc_resp = ksmbd_rpc_write(work->sess, id, data_buf, length);
7067 	if (rpc_resp) {
7068 		if (rpc_resp->flags == KSMBD_RPC_ENOTIMPLEMENTED) {
7069 			rsp->hdr.Status = STATUS_NOT_SUPPORTED;
7070 			kvfree(rpc_resp);
7071 			smb2_set_err_rsp(work);
7072 			return -EOPNOTSUPP;
7073 		}
7074 		if (rpc_resp->flags != KSMBD_RPC_OK) {
7075 			rsp->hdr.Status = STATUS_INVALID_HANDLE;
7076 			smb2_set_err_rsp(work);
7077 			kvfree(rpc_resp);
7078 			return ret;
7079 		}
7080 		kvfree(rpc_resp);
7081 	}
7082 
7083 	rsp->StructureSize = cpu_to_le16(17);
7084 	rsp->DataOffset = 0;
7085 	rsp->Reserved = 0;
7086 	rsp->DataLength = cpu_to_le32(length);
7087 	rsp->DataRemaining = 0;
7088 	rsp->Reserved2 = 0;
7089 	err = ksmbd_iov_pin_rsp(work, (void *)rsp,
7090 				offsetof(struct smb2_write_rsp, Buffer));
7091 out:
7092 	if (err) {
7093 		rsp->hdr.Status = STATUS_INVALID_HANDLE;
7094 		smb2_set_err_rsp(work);
7095 	}
7096 
7097 	return err;
7098 }
7099 
7100 static ssize_t smb2_write_rdma_channel(struct ksmbd_work *work,
7101 				       struct smb2_write_req *req,
7102 				       struct ksmbd_file *fp,
7103 				       loff_t offset, size_t length, bool sync)
7104 {
7105 	char *data_buf;
7106 	int ret;
7107 	ssize_t nbytes;
7108 
7109 	data_buf = kvzalloc(length, KSMBD_DEFAULT_GFP);
7110 	if (!data_buf)
7111 		return -ENOMEM;
7112 
7113 	ret = ksmbd_conn_rdma_read(work->conn, data_buf, length,
7114 				   (struct smbdirect_buffer_descriptor_v1 *)
7115 				   ((char *)req + le16_to_cpu(req->WriteChannelInfoOffset)),
7116 				   le16_to_cpu(req->WriteChannelInfoLength));
7117 	if (ret < 0) {
7118 		kvfree(data_buf);
7119 		return ret;
7120 	}
7121 
7122 	ret = ksmbd_vfs_write(work, fp, data_buf, length, &offset, sync, &nbytes);
7123 	kvfree(data_buf);
7124 	if (ret < 0)
7125 		return ret;
7126 
7127 	return nbytes;
7128 }
7129 
7130 /**
7131  * smb2_write() - handler for smb2 write from file
7132  * @work:	smb work containing write command buffer
7133  *
7134  * Return:	0 on success, otherwise error
7135  */
7136 int smb2_write(struct ksmbd_work *work)
7137 {
7138 	struct smb2_write_req *req;
7139 	struct smb2_write_rsp *rsp;
7140 	struct ksmbd_file *fp = NULL;
7141 	loff_t offset;
7142 	size_t length;
7143 	ssize_t nbytes;
7144 	char *data_buf;
7145 	bool writethrough = false, is_rdma_channel = false;
7146 	int err = 0;
7147 	unsigned int max_write_size = work->conn->vals->max_write_size;
7148 
7149 	ksmbd_debug(SMB, "Received smb2 write request\n");
7150 
7151 	WORK_BUFFERS(work, req, rsp);
7152 
7153 	if (test_share_config_flag(work->tcon->share_conf, KSMBD_SHARE_FLAG_PIPE)) {
7154 		ksmbd_debug(SMB, "IPC pipe write request\n");
7155 		return smb2_write_pipe(work);
7156 	}
7157 
7158 	offset = le64_to_cpu(req->Offset);
7159 	if (offset < 0)
7160 		return -EINVAL;
7161 	length = le32_to_cpu(req->Length);
7162 
7163 	if (req->Channel == SMB2_CHANNEL_RDMA_V1 ||
7164 	    req->Channel == SMB2_CHANNEL_RDMA_V1_INVALIDATE) {
7165 		is_rdma_channel = true;
7166 		max_write_size = get_smbd_max_read_write_size(work->conn->transport);
7167 		if (max_write_size == 0) {
7168 			err = -EINVAL;
7169 			goto out;
7170 		}
7171 		length = le32_to_cpu(req->RemainingBytes);
7172 	}
7173 
7174 	if (is_rdma_channel == true) {
7175 		unsigned int ch_offset = le16_to_cpu(req->WriteChannelInfoOffset);
7176 
7177 		if (req->Length != 0 || req->DataOffset != 0 ||
7178 		    ch_offset < offsetof(struct smb2_write_req, Buffer)) {
7179 			err = -EINVAL;
7180 			goto out;
7181 		}
7182 		err = smb2_set_remote_key_for_rdma(work,
7183 						   (struct smbdirect_buffer_descriptor_v1 *)
7184 						   ((char *)req + ch_offset),
7185 						   req->Channel,
7186 						   req->WriteChannelInfoLength);
7187 		if (err)
7188 			goto out;
7189 	}
7190 
7191 	if (!test_tree_conn_flag(work->tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
7192 		ksmbd_debug(SMB, "User does not have write permission\n");
7193 		err = -EACCES;
7194 		goto out;
7195 	}
7196 
7197 	fp = ksmbd_lookup_fd_slow(work, req->VolatileFileId, req->PersistentFileId);
7198 	if (!fp) {
7199 		err = -ENOENT;
7200 		goto out;
7201 	}
7202 
7203 	if (!(fp->daccess & (FILE_WRITE_DATA_LE | FILE_READ_ATTRIBUTES_LE))) {
7204 		pr_err("Not permitted to write : 0x%x\n", fp->daccess);
7205 		err = -EACCES;
7206 		goto out;
7207 	}
7208 
7209 	if (length > max_write_size) {
7210 		ksmbd_debug(SMB, "limiting write size to max size(%u)\n",
7211 			    max_write_size);
7212 		err = -EINVAL;
7213 		goto out;
7214 	}
7215 
7216 	ksmbd_debug(SMB, "flags %u\n", le32_to_cpu(req->Flags));
7217 	if (le32_to_cpu(req->Flags) & SMB2_WRITEFLAG_WRITE_THROUGH)
7218 		writethrough = true;
7219 
7220 	if (is_rdma_channel == false) {
7221 		if (le16_to_cpu(req->DataOffset) <
7222 		    offsetof(struct smb2_write_req, Buffer)) {
7223 			err = -EINVAL;
7224 			goto out;
7225 		}
7226 
7227 		data_buf = (char *)(((char *)&req->hdr.ProtocolId) +
7228 				    le16_to_cpu(req->DataOffset));
7229 
7230 		ksmbd_debug(SMB, "filename %pD, offset %lld, len %zu\n",
7231 			    fp->filp, offset, length);
7232 		err = ksmbd_vfs_write(work, fp, data_buf, length, &offset,
7233 				      writethrough, &nbytes);
7234 		if (err < 0)
7235 			goto out;
7236 	} else {
7237 		/* read data from the client using rdma channel, and
7238 		 * write the data.
7239 		 */
7240 		nbytes = smb2_write_rdma_channel(work, req, fp, offset, length,
7241 						 writethrough);
7242 		if (nbytes < 0) {
7243 			err = (int)nbytes;
7244 			goto out;
7245 		}
7246 	}
7247 
7248 	rsp->StructureSize = cpu_to_le16(17);
7249 	rsp->DataOffset = 0;
7250 	rsp->Reserved = 0;
7251 	rsp->DataLength = cpu_to_le32(nbytes);
7252 	rsp->DataRemaining = 0;
7253 	rsp->Reserved2 = 0;
7254 	err = ksmbd_iov_pin_rsp(work, rsp, offsetof(struct smb2_write_rsp, Buffer));
7255 	if (err)
7256 		goto out;
7257 	ksmbd_fd_put(work, fp);
7258 	return 0;
7259 
7260 out:
7261 	if (err == -EAGAIN)
7262 		rsp->hdr.Status = STATUS_FILE_LOCK_CONFLICT;
7263 	else if (err == -ENOSPC || err == -EFBIG)
7264 		rsp->hdr.Status = STATUS_DISK_FULL;
7265 	else if (err == -ENOENT)
7266 		rsp->hdr.Status = STATUS_FILE_CLOSED;
7267 	else if (err == -EACCES)
7268 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
7269 	else if (err == -ESHARE)
7270 		rsp->hdr.Status = STATUS_SHARING_VIOLATION;
7271 	else if (err == -EINVAL)
7272 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7273 	else
7274 		rsp->hdr.Status = STATUS_INVALID_HANDLE;
7275 
7276 	smb2_set_err_rsp(work);
7277 	ksmbd_fd_put(work, fp);
7278 	return err;
7279 }
7280 
7281 /**
7282  * smb2_flush() - handler for smb2 flush file - fsync
7283  * @work:	smb work containing flush command buffer
7284  *
7285  * Return:	0 on success, otherwise error
7286  */
7287 int smb2_flush(struct ksmbd_work *work)
7288 {
7289 	struct smb2_flush_req *req;
7290 	struct smb2_flush_rsp *rsp;
7291 	int err;
7292 
7293 	WORK_BUFFERS(work, req, rsp);
7294 
7295 	ksmbd_debug(SMB, "Received smb2 flush request(fid : %llu)\n", req->VolatileFileId);
7296 
7297 	err = ksmbd_vfs_fsync(work, req->VolatileFileId, req->PersistentFileId);
7298 	if (err)
7299 		goto out;
7300 
7301 	rsp->StructureSize = cpu_to_le16(4);
7302 	rsp->Reserved = 0;
7303 	return ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_flush_rsp));
7304 
7305 out:
7306 	rsp->hdr.Status = STATUS_INVALID_HANDLE;
7307 	smb2_set_err_rsp(work);
7308 	return err;
7309 }
7310 
7311 /**
7312  * smb2_cancel() - handler for smb2 cancel command
7313  * @work:	smb work containing cancel command buffer
7314  *
7315  * Return:	0 on success, otherwise error
7316  */
7317 int smb2_cancel(struct ksmbd_work *work)
7318 {
7319 	struct ksmbd_conn *conn = work->conn;
7320 	struct smb2_hdr *hdr = smb_get_msg(work->request_buf);
7321 	struct smb2_hdr *chdr;
7322 	struct ksmbd_work *iter;
7323 	struct list_head *command_list;
7324 
7325 	if (work->next_smb2_rcv_hdr_off)
7326 		hdr = ksmbd_resp_buf_next(work);
7327 
7328 	ksmbd_debug(SMB, "smb2 cancel called on mid %llu, async flags 0x%x\n",
7329 		    hdr->MessageId, hdr->Flags);
7330 
7331 	if (hdr->Flags & SMB2_FLAGS_ASYNC_COMMAND) {
7332 		command_list = &conn->async_requests;
7333 
7334 		spin_lock(&conn->request_lock);
7335 		list_for_each_entry(iter, command_list,
7336 				    async_request_entry) {
7337 			chdr = smb_get_msg(iter->request_buf);
7338 
7339 			if (iter->async_id !=
7340 			    le64_to_cpu(hdr->Id.AsyncId))
7341 				continue;
7342 
7343 			/*
7344 			 * A cancelled deferred byte-range lock frees its
7345 			 * file_lock and takes the smb2_lock() early-exit that
7346 			 * skips release_async_work(), so the work stays on
7347 			 * conn->async_requests with a live cancel_fn pointing
7348 			 * at the freed file_lock.  Re-firing it on a second
7349 			 * SMB2_CANCEL is a use-after-free.
7350 			 */
7351 			if (iter->state == KSMBD_WORK_CANCELLED)
7352 				break;
7353 
7354 			ksmbd_debug(SMB,
7355 				    "smb2 with AsyncId %llu cancelled command = 0x%x\n",
7356 				    le64_to_cpu(hdr->Id.AsyncId),
7357 				    le16_to_cpu(chdr->Command));
7358 			iter->state = KSMBD_WORK_CANCELLED;
7359 			if (iter->cancel_fn)
7360 				iter->cancel_fn(iter->cancel_argv);
7361 			break;
7362 		}
7363 		spin_unlock(&conn->request_lock);
7364 	} else {
7365 		command_list = &conn->requests;
7366 
7367 		spin_lock(&conn->request_lock);
7368 		list_for_each_entry(iter, command_list, request_entry) {
7369 			chdr = smb_get_msg(iter->request_buf);
7370 
7371 			if (chdr->MessageId != hdr->MessageId ||
7372 			    iter == work)
7373 				continue;
7374 
7375 			ksmbd_debug(SMB,
7376 				    "smb2 with mid %llu cancelled command = 0x%x\n",
7377 				    le64_to_cpu(hdr->MessageId),
7378 				    le16_to_cpu(chdr->Command));
7379 			iter->state = KSMBD_WORK_CANCELLED;
7380 			break;
7381 		}
7382 		spin_unlock(&conn->request_lock);
7383 	}
7384 
7385 	/* For SMB2_CANCEL command itself send no response*/
7386 	work->send_no_response = 1;
7387 	return 0;
7388 }
7389 
7390 struct file_lock *smb_flock_init(struct file *f)
7391 {
7392 	struct file_lock *fl;
7393 
7394 	fl = locks_alloc_lock();
7395 	if (!fl)
7396 		goto out;
7397 
7398 	locks_init_lock(fl);
7399 
7400 	fl->c.flc_owner = f;
7401 	fl->c.flc_pid = current->tgid;
7402 	fl->c.flc_file = f;
7403 	fl->c.flc_flags = FL_POSIX;
7404 	fl->fl_ops = NULL;
7405 	fl->fl_lmops = NULL;
7406 
7407 out:
7408 	return fl;
7409 }
7410 
7411 static int smb2_set_flock_flags(struct file_lock *flock, int flags)
7412 {
7413 	int cmd = -EINVAL;
7414 
7415 	/* Checking for wrong flag combination during lock request*/
7416 	switch (flags) {
7417 	case SMB2_LOCKFLAG_SHARED:
7418 		ksmbd_debug(SMB, "received shared request\n");
7419 		cmd = F_SETLKW;
7420 		flock->c.flc_type = F_RDLCK;
7421 		flock->c.flc_flags |= FL_SLEEP;
7422 		break;
7423 	case SMB2_LOCKFLAG_EXCLUSIVE:
7424 		ksmbd_debug(SMB, "received exclusive request\n");
7425 		cmd = F_SETLKW;
7426 		flock->c.flc_type = F_WRLCK;
7427 		flock->c.flc_flags |= FL_SLEEP;
7428 		break;
7429 	case SMB2_LOCKFLAG_SHARED | SMB2_LOCKFLAG_FAIL_IMMEDIATELY:
7430 		ksmbd_debug(SMB,
7431 			    "received shared & fail immediately request\n");
7432 		cmd = F_SETLK;
7433 		flock->c.flc_type = F_RDLCK;
7434 		break;
7435 	case SMB2_LOCKFLAG_EXCLUSIVE | SMB2_LOCKFLAG_FAIL_IMMEDIATELY:
7436 		ksmbd_debug(SMB,
7437 			    "received exclusive & fail immediately request\n");
7438 		cmd = F_SETLK;
7439 		flock->c.flc_type = F_WRLCK;
7440 		break;
7441 	case SMB2_LOCKFLAG_UNLOCK:
7442 		ksmbd_debug(SMB, "received unlock request\n");
7443 		flock->c.flc_type = F_UNLCK;
7444 		cmd = F_SETLK;
7445 		break;
7446 	}
7447 
7448 	return cmd;
7449 }
7450 
7451 static struct ksmbd_lock *smb2_lock_init(struct file_lock *flock,
7452 					 unsigned int cmd, int flags,
7453 					 struct list_head *lock_list)
7454 {
7455 	struct ksmbd_lock *lock;
7456 
7457 	lock = kzalloc_obj(struct ksmbd_lock, KSMBD_DEFAULT_GFP);
7458 	if (!lock)
7459 		return NULL;
7460 
7461 	lock->cmd = cmd;
7462 	lock->fl = flock;
7463 	lock->start = flock->fl_start;
7464 	lock->end = flock->fl_end;
7465 	lock->flags = flags;
7466 	if (lock->start == lock->end)
7467 		lock->zero_len = 1;
7468 	INIT_LIST_HEAD(&lock->clist);
7469 	INIT_LIST_HEAD(&lock->flist);
7470 	INIT_LIST_HEAD(&lock->llist);
7471 	list_add_tail(&lock->llist, lock_list);
7472 
7473 	return lock;
7474 }
7475 
7476 static void smb2_remove_blocked_lock(void **argv)
7477 {
7478 	struct file_lock *flock = (struct file_lock *)argv[0];
7479 
7480 	ksmbd_vfs_posix_lock_unblock(flock);
7481 	locks_wake_up(flock);
7482 }
7483 
7484 static inline bool lock_defer_pending(struct file_lock *fl)
7485 {
7486 	/* check pending lock waiters */
7487 	return waitqueue_active(&fl->c.flc_wait);
7488 }
7489 
7490 /**
7491  * smb2_lock() - handler for smb2 file lock command
7492  * @work:	smb work containing lock command buffer
7493  *
7494  * Return:	0 on success, otherwise error
7495  */
7496 int smb2_lock(struct ksmbd_work *work)
7497 {
7498 	struct smb2_lock_req *req;
7499 	struct smb2_lock_rsp *rsp;
7500 	struct smb2_lock_element *lock_ele;
7501 	struct ksmbd_file *fp = NULL;
7502 	struct file_lock *flock = NULL;
7503 	struct file *filp = NULL;
7504 	int lock_count;
7505 	int flags = 0;
7506 	int cmd = 0;
7507 	int err = -EIO, i, rc = 0;
7508 	u64 lock_start, lock_length;
7509 	struct ksmbd_lock *smb_lock = NULL, *cmp_lock, *tmp, *tmp2;
7510 	struct ksmbd_conn *conn;
7511 	int nolock = 0;
7512 	LIST_HEAD(lock_list);
7513 	LIST_HEAD(rollback_list);
7514 	int prior_lock = 0, bkt;
7515 
7516 	WORK_BUFFERS(work, req, rsp);
7517 
7518 	ksmbd_debug(SMB, "Received smb2 lock request\n");
7519 	fp = ksmbd_lookup_fd_slow(work, req->VolatileFileId, req->PersistentFileId);
7520 	if (!fp) {
7521 		ksmbd_debug(SMB, "Invalid file id for lock : %llu\n", req->VolatileFileId);
7522 		err = -ENOENT;
7523 		goto out2;
7524 	}
7525 
7526 	filp = fp->filp;
7527 	lock_count = le16_to_cpu(req->LockCount);
7528 	lock_ele = req->locks;
7529 
7530 	ksmbd_debug(SMB, "lock count is %d\n", lock_count);
7531 	/*
7532 	 * Cap lock_count at 64. The MS-SMB2 spec defines Open.LockSequenceArray
7533 	 * as exactly 64 entries so 64 is the intended ceiling. No real workload
7534 	 * comes close to this in a single request.
7535 	 */
7536 	if (!lock_count || lock_count > 64) {
7537 		err = -EINVAL;
7538 		goto out2;
7539 	}
7540 
7541 	for (i = 0; i < lock_count; i++) {
7542 		flags = le32_to_cpu(lock_ele[i].Flags);
7543 
7544 		flock = smb_flock_init(filp);
7545 		if (!flock)
7546 			goto out;
7547 
7548 		cmd = smb2_set_flock_flags(flock, flags);
7549 
7550 		lock_start = le64_to_cpu(lock_ele[i].Offset);
7551 		lock_length = le64_to_cpu(lock_ele[i].Length);
7552 		if (lock_start > U64_MAX - lock_length) {
7553 			pr_err("Invalid lock range requested\n");
7554 			rsp->hdr.Status = STATUS_INVALID_LOCK_RANGE;
7555 			locks_free_lock(flock);
7556 			goto out;
7557 		}
7558 
7559 		if (lock_start > OFFSET_MAX)
7560 			flock->fl_start = OFFSET_MAX;
7561 		else
7562 			flock->fl_start = lock_start;
7563 
7564 		lock_length = le64_to_cpu(lock_ele[i].Length);
7565 		if (lock_length > OFFSET_MAX - flock->fl_start)
7566 			lock_length = OFFSET_MAX - flock->fl_start;
7567 
7568 		flock->fl_end = flock->fl_start + lock_length;
7569 
7570 		if (flock->fl_end < flock->fl_start) {
7571 			ksmbd_debug(SMB,
7572 				    "the end offset(%llx) is smaller than the start offset(%llx)\n",
7573 				    flock->fl_end, flock->fl_start);
7574 			rsp->hdr.Status = STATUS_INVALID_LOCK_RANGE;
7575 			locks_free_lock(flock);
7576 			goto out;
7577 		}
7578 
7579 		/* Check conflict locks in one request */
7580 		list_for_each_entry(cmp_lock, &lock_list, llist) {
7581 			if (cmp_lock->fl->fl_start <= flock->fl_start &&
7582 			    cmp_lock->fl->fl_end >= flock->fl_end) {
7583 				if (cmp_lock->fl->c.flc_type != F_UNLCK &&
7584 				    flock->c.flc_type != F_UNLCK) {
7585 					pr_err("conflict two locks in one request\n");
7586 					err = -EINVAL;
7587 					locks_free_lock(flock);
7588 					goto out;
7589 				}
7590 			}
7591 		}
7592 
7593 		smb_lock = smb2_lock_init(flock, cmd, flags, &lock_list);
7594 		if (!smb_lock) {
7595 			err = -EINVAL;
7596 			locks_free_lock(flock);
7597 			goto out;
7598 		}
7599 	}
7600 
7601 	list_for_each_entry_safe(smb_lock, tmp, &lock_list, llist) {
7602 		if (smb_lock->cmd < 0) {
7603 			err = -EINVAL;
7604 			goto out;
7605 		}
7606 
7607 		if (!(smb_lock->flags & SMB2_LOCKFLAG_MASK)) {
7608 			err = -EINVAL;
7609 			goto out;
7610 		}
7611 
7612 		if ((prior_lock & (SMB2_LOCKFLAG_EXCLUSIVE | SMB2_LOCKFLAG_SHARED) &&
7613 		     smb_lock->flags & SMB2_LOCKFLAG_UNLOCK) ||
7614 		    (prior_lock == SMB2_LOCKFLAG_UNLOCK &&
7615 		     !(smb_lock->flags & SMB2_LOCKFLAG_UNLOCK))) {
7616 			err = -EINVAL;
7617 			goto out;
7618 		}
7619 
7620 		prior_lock = smb_lock->flags;
7621 
7622 		if (!(smb_lock->flags & SMB2_LOCKFLAG_UNLOCK) &&
7623 		    !(smb_lock->flags & SMB2_LOCKFLAG_FAIL_IMMEDIATELY))
7624 			goto no_check_cl;
7625 
7626 		nolock = 1;
7627 		/* check locks in connection list */
7628 		down_read(&conn_list_lock);
7629 		hash_for_each(conn_list, bkt, conn, hlist) {
7630 			spin_lock(&conn->llist_lock);
7631 			list_for_each_entry_safe(cmp_lock, tmp2, &conn->lock_list, clist) {
7632 				if (file_inode(cmp_lock->fl->c.flc_file) !=
7633 				    file_inode(smb_lock->fl->c.flc_file))
7634 					continue;
7635 
7636 				if (lock_is_unlock(smb_lock->fl)) {
7637 					if (cmp_lock->fl->c.flc_file == smb_lock->fl->c.flc_file &&
7638 					    cmp_lock->start == smb_lock->start &&
7639 					    cmp_lock->end == smb_lock->end &&
7640 					    !lock_defer_pending(cmp_lock->fl)) {
7641 						nolock = 0;
7642 						list_del(&cmp_lock->flist);
7643 						list_del(&cmp_lock->clist);
7644 						spin_unlock(&conn->llist_lock);
7645 						up_read(&conn_list_lock);
7646 
7647 						locks_free_lock(cmp_lock->fl);
7648 						kfree(cmp_lock);
7649 						goto out_check_cl;
7650 					}
7651 					continue;
7652 				}
7653 
7654 				if (cmp_lock->fl->c.flc_file == smb_lock->fl->c.flc_file) {
7655 					if (smb_lock->flags & SMB2_LOCKFLAG_SHARED)
7656 						continue;
7657 				} else {
7658 					if (cmp_lock->flags & SMB2_LOCKFLAG_SHARED)
7659 						continue;
7660 				}
7661 
7662 				/* check zero byte lock range */
7663 				if (cmp_lock->zero_len && !smb_lock->zero_len &&
7664 				    cmp_lock->start > smb_lock->start &&
7665 				    cmp_lock->start < smb_lock->end) {
7666 					spin_unlock(&conn->llist_lock);
7667 					up_read(&conn_list_lock);
7668 					pr_err("previous lock conflict with zero byte lock range\n");
7669 					goto out;
7670 				}
7671 
7672 				if (smb_lock->zero_len && !cmp_lock->zero_len &&
7673 				    smb_lock->start > cmp_lock->start &&
7674 				    smb_lock->start < cmp_lock->end) {
7675 					spin_unlock(&conn->llist_lock);
7676 					up_read(&conn_list_lock);
7677 					pr_err("current lock conflict with zero byte lock range\n");
7678 					goto out;
7679 				}
7680 
7681 				if (((cmp_lock->start <= smb_lock->start &&
7682 				      cmp_lock->end > smb_lock->start) ||
7683 				     (cmp_lock->start < smb_lock->end &&
7684 				      cmp_lock->end >= smb_lock->end)) &&
7685 				    !cmp_lock->zero_len && !smb_lock->zero_len) {
7686 					spin_unlock(&conn->llist_lock);
7687 					up_read(&conn_list_lock);
7688 					pr_err("Not allow lock operation on exclusive lock range\n");
7689 					goto out;
7690 				}
7691 			}
7692 			spin_unlock(&conn->llist_lock);
7693 		}
7694 		up_read(&conn_list_lock);
7695 out_check_cl:
7696 		if (lock_is_unlock(smb_lock->fl) && nolock) {
7697 			pr_err("Try to unlock nolocked range\n");
7698 			rsp->hdr.Status = STATUS_RANGE_NOT_LOCKED;
7699 			goto out;
7700 		}
7701 
7702 no_check_cl:
7703 		flock = smb_lock->fl;
7704 		list_del(&smb_lock->llist);
7705 
7706 		if (smb_lock->zero_len) {
7707 			err = 0;
7708 			goto skip;
7709 		}
7710 retry:
7711 		rc = vfs_lock_file(filp, smb_lock->cmd, flock, NULL);
7712 skip:
7713 		if (smb_lock->flags & SMB2_LOCKFLAG_UNLOCK) {
7714 			locks_free_lock(flock);
7715 			kfree(smb_lock);
7716 			if (!rc) {
7717 				ksmbd_debug(SMB, "File unlocked\n");
7718 			} else if (rc == -ENOENT) {
7719 				rsp->hdr.Status = STATUS_NOT_LOCKED;
7720 				err = rc;
7721 				goto out;
7722 			}
7723 		} else {
7724 			if (rc == FILE_LOCK_DEFERRED) {
7725 				void **argv;
7726 
7727 				ksmbd_debug(SMB,
7728 					    "would have to wait for getting lock\n");
7729 				list_add(&smb_lock->llist, &rollback_list);
7730 
7731 				argv = kmalloc(sizeof(void *), KSMBD_DEFAULT_GFP);
7732 				if (!argv) {
7733 					err = -ENOMEM;
7734 					goto out;
7735 				}
7736 				argv[0] = flock;
7737 
7738 				rc = setup_async_work(work,
7739 						      smb2_remove_blocked_lock,
7740 						      argv);
7741 				if (rc) {
7742 					kfree(argv);
7743 					err = -ENOMEM;
7744 					goto out;
7745 				}
7746 				spin_lock(&fp->f_lock);
7747 				list_add(&work->fp_entry, &fp->blocked_works);
7748 				spin_unlock(&fp->f_lock);
7749 
7750 				smb2_send_interim_resp(work, STATUS_PENDING);
7751 
7752 				ksmbd_vfs_posix_lock_wait(flock);
7753 
7754 				spin_lock(&fp->f_lock);
7755 				list_del(&work->fp_entry);
7756 				spin_unlock(&fp->f_lock);
7757 
7758 				if (work->state != KSMBD_WORK_ACTIVE) {
7759 					list_del(&smb_lock->llist);
7760 					locks_free_lock(flock);
7761 
7762 					if (work->state == KSMBD_WORK_CANCELLED) {
7763 						rsp->hdr.Status =
7764 							STATUS_CANCELLED;
7765 						kfree(smb_lock);
7766 						smb2_send_interim_resp(work,
7767 								       STATUS_CANCELLED);
7768 						work->send_no_response = 1;
7769 						goto out;
7770 					}
7771 
7772 					rsp->hdr.Status =
7773 						STATUS_RANGE_NOT_LOCKED;
7774 					kfree(smb_lock);
7775 					goto out2;
7776 				}
7777 
7778 				list_del(&smb_lock->llist);
7779 				release_async_work(work);
7780 				goto retry;
7781 			} else if (!rc) {
7782 				list_add(&smb_lock->llist, &rollback_list);
7783 				spin_lock(&work->conn->llist_lock);
7784 				list_add_tail(&smb_lock->clist,
7785 					      &work->conn->lock_list);
7786 				list_add_tail(&smb_lock->flist,
7787 					      &fp->lock_list);
7788 				spin_unlock(&work->conn->llist_lock);
7789 				ksmbd_debug(SMB, "successful in taking lock\n");
7790 			} else {
7791 				locks_free_lock(flock);
7792 				kfree(smb_lock);
7793 				err = rc;
7794 				goto out;
7795 			}
7796 		}
7797 	}
7798 
7799 	if (atomic_read(&fp->f_ci->op_count) > 1)
7800 		smb_break_all_oplock(work, fp);
7801 
7802 	rsp->StructureSize = cpu_to_le16(4);
7803 	ksmbd_debug(SMB, "successful in taking lock\n");
7804 	rsp->hdr.Status = STATUS_SUCCESS;
7805 	rsp->Reserved = 0;
7806 	err = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_lock_rsp));
7807 	if (err)
7808 		goto out;
7809 
7810 	ksmbd_fd_put(work, fp);
7811 	return 0;
7812 
7813 out:
7814 	list_for_each_entry_safe(smb_lock, tmp, &lock_list, llist) {
7815 		locks_free_lock(smb_lock->fl);
7816 		list_del(&smb_lock->llist);
7817 		kfree(smb_lock);
7818 	}
7819 
7820 	list_for_each_entry_safe(smb_lock, tmp, &rollback_list, llist) {
7821 		struct file_lock *rlock = NULL;
7822 
7823 		rlock = smb_flock_init(filp);
7824 		if (rlock) {
7825 			rlock->c.flc_type = F_UNLCK;
7826 			rlock->fl_start = smb_lock->start;
7827 			rlock->fl_end = smb_lock->end;
7828 
7829 			rc = vfs_lock_file(filp, F_SETLK, rlock, NULL);
7830 			if (rc)
7831 				pr_err("rollback unlock fail : %d\n", rc);
7832 		} else {
7833 			pr_err("rollback unlock alloc failed\n");
7834 		}
7835 
7836 		list_del(&smb_lock->llist);
7837 		spin_lock(&work->conn->llist_lock);
7838 		if (!list_empty(&smb_lock->flist))
7839 			list_del(&smb_lock->flist);
7840 		list_del(&smb_lock->clist);
7841 		spin_unlock(&work->conn->llist_lock);
7842 
7843 		locks_free_lock(smb_lock->fl);
7844 		if (rlock)
7845 			locks_free_lock(rlock);
7846 		kfree(smb_lock);
7847 	}
7848 out2:
7849 	ksmbd_debug(SMB, "failed in taking lock(flags : %x), err : %d\n", flags, err);
7850 
7851 	if (!rsp->hdr.Status) {
7852 		if (err == -EINVAL)
7853 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7854 		else if (err == -ENOMEM)
7855 			rsp->hdr.Status = STATUS_INSUFFICIENT_RESOURCES;
7856 		else if (err == -ENOENT)
7857 			rsp->hdr.Status = STATUS_FILE_CLOSED;
7858 		else
7859 			rsp->hdr.Status = STATUS_LOCK_NOT_GRANTED;
7860 	}
7861 
7862 	smb2_set_err_rsp(work);
7863 	ksmbd_fd_put(work, fp);
7864 	return err;
7865 }
7866 
7867 static int fsctl_copychunk(struct ksmbd_work *work,
7868 			   struct copychunk_ioctl_req *ci_req,
7869 			   unsigned int cnt_code,
7870 			   unsigned int input_count,
7871 			   unsigned long long volatile_id,
7872 			   unsigned long long persistent_id,
7873 			   struct smb2_ioctl_rsp *rsp)
7874 {
7875 	struct copychunk_ioctl_rsp *ci_rsp;
7876 	struct ksmbd_file *src_fp = NULL, *dst_fp = NULL;
7877 	struct srv_copychunk *chunks;
7878 	unsigned int i, chunk_count, chunk_count_written = 0;
7879 	unsigned int chunk_size_written = 0;
7880 	loff_t total_size_written = 0;
7881 	int ret = 0;
7882 
7883 	ci_rsp = (struct copychunk_ioctl_rsp *)&rsp->Buffer[0];
7884 
7885 	rsp->VolatileFileId = volatile_id;
7886 	rsp->PersistentFileId = persistent_id;
7887 	ci_rsp->ChunksWritten =
7888 		cpu_to_le32(ksmbd_server_side_copy_max_chunk_count());
7889 	ci_rsp->ChunkBytesWritten =
7890 		cpu_to_le32(ksmbd_server_side_copy_max_chunk_size());
7891 	ci_rsp->TotalBytesWritten =
7892 		cpu_to_le32(ksmbd_server_side_copy_max_total_size());
7893 
7894 	chunk_count = le32_to_cpu(ci_req->ChunkCount);
7895 	if (chunk_count == 0)
7896 		goto out;
7897 	total_size_written = 0;
7898 
7899 	/* verify the SRV_COPYCHUNK_COPY packet */
7900 	if (chunk_count > ksmbd_server_side_copy_max_chunk_count() ||
7901 	    input_count < struct_size(ci_req, Chunks, chunk_count)) {
7902 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7903 		return -EINVAL;
7904 	}
7905 
7906 	chunks = &ci_req->Chunks[0];
7907 	for (i = 0; i < chunk_count; i++) {
7908 		if (le32_to_cpu(chunks[i].Length) == 0 ||
7909 		    le32_to_cpu(chunks[i].Length) > ksmbd_server_side_copy_max_chunk_size())
7910 			break;
7911 		total_size_written += le32_to_cpu(chunks[i].Length);
7912 	}
7913 
7914 	if (i < chunk_count ||
7915 	    total_size_written > ksmbd_server_side_copy_max_total_size()) {
7916 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7917 		return -EINVAL;
7918 	}
7919 
7920 	src_fp = ksmbd_lookup_foreign_fd(work,
7921 					 le64_to_cpu(ci_req->SourceKeyU64[0]));
7922 	dst_fp = ksmbd_lookup_fd_slow(work, volatile_id, persistent_id);
7923 	ret = -EINVAL;
7924 	if (!src_fp ||
7925 	    src_fp->persistent_id != le64_to_cpu(ci_req->SourceKeyU64[1])) {
7926 		rsp->hdr.Status = STATUS_OBJECT_NAME_NOT_FOUND;
7927 		goto out;
7928 	}
7929 
7930 	if (!dst_fp) {
7931 		rsp->hdr.Status = STATUS_FILE_CLOSED;
7932 		goto out;
7933 	}
7934 
7935 	/*
7936 	 * FILE_READ_DATA should only be included in
7937 	 * the FSCTL_COPYCHUNK case
7938 	 */
7939 	if (cnt_code == FSCTL_COPYCHUNK &&
7940 	    !(dst_fp->daccess & (FILE_READ_DATA_LE | FILE_GENERIC_READ_LE))) {
7941 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
7942 		goto out;
7943 	}
7944 
7945 	ret = ksmbd_vfs_copy_file_ranges(work, src_fp, dst_fp,
7946 					 chunks, chunk_count,
7947 					 &chunk_count_written,
7948 					 &chunk_size_written,
7949 					 &total_size_written);
7950 	if (ret < 0) {
7951 		if (ret == -EACCES)
7952 			rsp->hdr.Status = STATUS_ACCESS_DENIED;
7953 		if (ret == -EAGAIN)
7954 			rsp->hdr.Status = STATUS_FILE_LOCK_CONFLICT;
7955 		else if (ret == -EBADF)
7956 			rsp->hdr.Status = STATUS_INVALID_HANDLE;
7957 		else if (ret == -EFBIG || ret == -ENOSPC)
7958 			rsp->hdr.Status = STATUS_DISK_FULL;
7959 		else if (ret == -EINVAL)
7960 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
7961 		else if (ret == -EISDIR)
7962 			rsp->hdr.Status = STATUS_FILE_IS_A_DIRECTORY;
7963 		else if (ret == -E2BIG)
7964 			rsp->hdr.Status = STATUS_INVALID_VIEW_SIZE;
7965 		else
7966 			rsp->hdr.Status = STATUS_UNEXPECTED_IO_ERROR;
7967 	}
7968 
7969 	ci_rsp->ChunksWritten = cpu_to_le32(chunk_count_written);
7970 	ci_rsp->ChunkBytesWritten = cpu_to_le32(chunk_size_written);
7971 	ci_rsp->TotalBytesWritten = cpu_to_le32(total_size_written);
7972 out:
7973 	ksmbd_fd_put(work, src_fp);
7974 	ksmbd_fd_put(work, dst_fp);
7975 	return ret;
7976 }
7977 
7978 static __be32 idev_ipv4_address(struct in_device *idev)
7979 {
7980 	__be32 addr = 0;
7981 
7982 	struct in_ifaddr *ifa;
7983 
7984 	rcu_read_lock();
7985 	in_dev_for_each_ifa_rcu(ifa, idev) {
7986 		if (ifa->ifa_flags & IFA_F_SECONDARY)
7987 			continue;
7988 
7989 		addr = ifa->ifa_address;
7990 		break;
7991 	}
7992 	rcu_read_unlock();
7993 	return addr;
7994 }
7995 
7996 static int fsctl_query_iface_info_ioctl(struct ksmbd_conn *conn,
7997 					struct smb2_ioctl_rsp *rsp,
7998 					unsigned int out_buf_len)
7999 {
8000 	struct network_interface_info_ioctl_rsp *nii_rsp = NULL;
8001 	int nbytes = 0;
8002 	struct net_device *netdev;
8003 	struct sockaddr_storage_rsp *sockaddr_storage;
8004 	unsigned int flags;
8005 	unsigned long long speed;
8006 
8007 	rtnl_lock();
8008 	for_each_netdev(&init_net, netdev) {
8009 		bool ipv4_set = false;
8010 
8011 		if (netdev->type == ARPHRD_LOOPBACK)
8012 			continue;
8013 
8014 		if (!ksmbd_find_netdev_name_iface_list(netdev->name))
8015 			continue;
8016 
8017 		flags = netif_get_flags(netdev);
8018 		if (!(flags & IFF_RUNNING))
8019 			continue;
8020 ipv6_retry:
8021 		if (out_buf_len <
8022 		    nbytes + sizeof(struct network_interface_info_ioctl_rsp)) {
8023 			rtnl_unlock();
8024 			return -ENOSPC;
8025 		}
8026 
8027 		nii_rsp = (struct network_interface_info_ioctl_rsp *)
8028 				&rsp->Buffer[nbytes];
8029 		nii_rsp->IfIndex = cpu_to_le32(netdev->ifindex);
8030 
8031 		nii_rsp->Capability = 0;
8032 		if (netdev->real_num_tx_queues > 1)
8033 			nii_rsp->Capability |= RSS_CAPABLE;
8034 		if (ksmbd_rdma_capable_netdev(netdev))
8035 			nii_rsp->Capability |= RDMA_CAPABLE;
8036 
8037 		nii_rsp->Next = cpu_to_le32(152);
8038 		nii_rsp->Reserved = 0;
8039 
8040 		if (netdev->ethtool_ops->get_link_ksettings) {
8041 			struct ethtool_link_ksettings cmd;
8042 
8043 			netdev->ethtool_ops->get_link_ksettings(netdev, &cmd);
8044 			speed = cmd.base.speed;
8045 		} else {
8046 			ksmbd_debug(SMB, "%s %s\n", netdev->name,
8047 				    "speed is unknown, defaulting to 1Gb/sec");
8048 			speed = SPEED_1000;
8049 		}
8050 
8051 		speed *= 1000000;
8052 		nii_rsp->LinkSpeed = cpu_to_le64(speed);
8053 
8054 		sockaddr_storage = (struct sockaddr_storage_rsp *)
8055 					nii_rsp->SockAddr_Storage;
8056 		memset(sockaddr_storage, 0, 128);
8057 
8058 		if (!ipv4_set) {
8059 			struct in_device *idev;
8060 
8061 			sockaddr_storage->Family = INTERNETWORK;
8062 			sockaddr_storage->addr4.Port = 0;
8063 
8064 			idev = __in_dev_get_rtnl(netdev);
8065 			if (!idev)
8066 				continue;
8067 			sockaddr_storage->addr4.IPv4Address =
8068 						idev_ipv4_address(idev);
8069 			nbytes += sizeof(struct network_interface_info_ioctl_rsp);
8070 			ipv4_set = true;
8071 			goto ipv6_retry;
8072 		} else {
8073 			struct inet6_dev *idev6;
8074 			struct inet6_ifaddr *ifa;
8075 			__u8 *ipv6_addr = sockaddr_storage->addr6.IPv6Address;
8076 
8077 			sockaddr_storage->Family = INTERNETWORKV6;
8078 			sockaddr_storage->addr6.Port = 0;
8079 			sockaddr_storage->addr6.FlowInfo = 0;
8080 
8081 			idev6 = __in6_dev_get(netdev);
8082 			if (!idev6)
8083 				continue;
8084 
8085 			list_for_each_entry(ifa, &idev6->addr_list, if_list) {
8086 				if (ifa->flags & (IFA_F_TENTATIVE |
8087 							IFA_F_DEPRECATED))
8088 					continue;
8089 				memcpy(ipv6_addr, ifa->addr.s6_addr, 16);
8090 				break;
8091 			}
8092 			sockaddr_storage->addr6.ScopeId = 0;
8093 			nbytes += sizeof(struct network_interface_info_ioctl_rsp);
8094 		}
8095 	}
8096 	rtnl_unlock();
8097 
8098 	/* zero if this is last one */
8099 	if (nii_rsp)
8100 		nii_rsp->Next = 0;
8101 
8102 	rsp->PersistentFileId = SMB2_NO_FID;
8103 	rsp->VolatileFileId = SMB2_NO_FID;
8104 	return nbytes;
8105 }
8106 
8107 static int fsctl_validate_negotiate_info(struct ksmbd_conn *conn,
8108 					 struct validate_negotiate_info_req *neg_req,
8109 					 struct validate_negotiate_info_rsp *neg_rsp,
8110 					 unsigned int in_buf_len)
8111 {
8112 	int ret = 0;
8113 	int dialect;
8114 
8115 	if (in_buf_len < offsetof(struct validate_negotiate_info_req, Dialects) +
8116 			le16_to_cpu(neg_req->DialectCount) * sizeof(__le16))
8117 		return -EINVAL;
8118 
8119 	dialect = ksmbd_lookup_dialect_by_id(neg_req->Dialects,
8120 					     neg_req->DialectCount);
8121 	if (dialect == BAD_PROT_ID || dialect != conn->dialect) {
8122 		ret = -EINVAL;
8123 		goto err_out;
8124 	}
8125 
8126 	if (strncmp(neg_req->Guid, conn->ClientGUID, SMB2_CLIENT_GUID_SIZE)) {
8127 		ret = -EINVAL;
8128 		goto err_out;
8129 	}
8130 
8131 	if (le16_to_cpu(neg_req->SecurityMode) != conn->cli_sec_mode) {
8132 		ret = -EINVAL;
8133 		goto err_out;
8134 	}
8135 
8136 	if (le32_to_cpu(neg_req->Capabilities) != conn->cli_cap) {
8137 		ret = -EINVAL;
8138 		goto err_out;
8139 	}
8140 
8141 	neg_rsp->Capabilities = cpu_to_le32(conn->vals->req_capabilities);
8142 	memset(neg_rsp->Guid, 0, SMB2_CLIENT_GUID_SIZE);
8143 	neg_rsp->SecurityMode = cpu_to_le16(conn->srv_sec_mode);
8144 	neg_rsp->Dialect = cpu_to_le16(conn->dialect);
8145 err_out:
8146 	return ret;
8147 }
8148 
8149 static int fsctl_query_allocated_ranges(struct ksmbd_work *work, u64 id,
8150 					struct file_allocated_range_buffer *qar_req,
8151 					struct file_allocated_range_buffer *qar_rsp,
8152 					unsigned int in_count, unsigned int *out_count)
8153 {
8154 	struct ksmbd_file *fp;
8155 	loff_t start, length;
8156 	int ret = 0;
8157 
8158 	*out_count = 0;
8159 	if (in_count == 0)
8160 		return -EINVAL;
8161 
8162 	start = le64_to_cpu(qar_req->file_offset);
8163 	length = le64_to_cpu(qar_req->length);
8164 
8165 	if (start < 0 || length < 0)
8166 		return -EINVAL;
8167 
8168 	fp = ksmbd_lookup_fd_fast(work, id);
8169 	if (!fp)
8170 		return -ENOENT;
8171 
8172 	ret = ksmbd_vfs_fqar_lseek(fp, start, length,
8173 				   qar_rsp, in_count, out_count);
8174 	if (ret && ret != -E2BIG)
8175 		*out_count = 0;
8176 
8177 	ksmbd_fd_put(work, fp);
8178 	return ret;
8179 }
8180 
8181 static int fsctl_pipe_transceive(struct ksmbd_work *work, u64 id,
8182 				 unsigned int out_buf_len,
8183 				 struct smb2_ioctl_req *req,
8184 				 struct smb2_ioctl_rsp *rsp)
8185 {
8186 	struct ksmbd_rpc_command *rpc_resp;
8187 	char *data_buf = (char *)req + le32_to_cpu(req->InputOffset);
8188 	int nbytes = 0;
8189 
8190 	rpc_resp = ksmbd_rpc_ioctl(work->sess, id, data_buf,
8191 				   le32_to_cpu(req->InputCount));
8192 	if (rpc_resp) {
8193 		if (rpc_resp->flags == KSMBD_RPC_SOME_NOT_MAPPED) {
8194 			/*
8195 			 * set STATUS_SOME_NOT_MAPPED response
8196 			 * for unknown domain sid.
8197 			 */
8198 			rsp->hdr.Status = STATUS_SOME_NOT_MAPPED;
8199 		} else if (rpc_resp->flags == KSMBD_RPC_ENOTIMPLEMENTED) {
8200 			rsp->hdr.Status = STATUS_NOT_SUPPORTED;
8201 			goto out;
8202 		} else if (rpc_resp->flags != KSMBD_RPC_OK) {
8203 			rsp->hdr.Status = STATUS_INVALID_PARAMETER;
8204 			goto out;
8205 		}
8206 
8207 		nbytes = rpc_resp->payload_sz;
8208 		if (rpc_resp->payload_sz > out_buf_len) {
8209 			rsp->hdr.Status = STATUS_BUFFER_OVERFLOW;
8210 			nbytes = out_buf_len;
8211 		}
8212 
8213 		if (!rpc_resp->payload_sz) {
8214 			rsp->hdr.Status =
8215 				STATUS_UNEXPECTED_IO_ERROR;
8216 			goto out;
8217 		}
8218 
8219 		memcpy((char *)rsp->Buffer, rpc_resp->payload, nbytes);
8220 	}
8221 out:
8222 	kvfree(rpc_resp);
8223 	return nbytes;
8224 }
8225 
8226 static inline int fsctl_set_sparse(struct ksmbd_work *work, u64 id,
8227 				   struct file_sparse *sparse)
8228 {
8229 	struct ksmbd_file *fp;
8230 	struct mnt_idmap *idmap;
8231 	int ret = 0;
8232 	__le32 old_fattr;
8233 
8234 	if (!test_tree_conn_flag(work->tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
8235 		ksmbd_debug(SMB, "User does not have write permission\n");
8236 		return -EACCES;
8237 	}
8238 
8239 	fp = ksmbd_lookup_fd_fast(work, id);
8240 	if (!fp)
8241 		return -ENOENT;
8242 
8243 	if (!(fp->daccess & (FILE_WRITE_DATA_LE | FILE_WRITE_ATTRIBUTES_LE))) {
8244 		ret = -EACCES;
8245 		goto out;
8246 	}
8247 
8248 	idmap = file_mnt_idmap(fp->filp);
8249 
8250 	old_fattr = fp->f_ci->m_fattr;
8251 	if (sparse->SetSparse)
8252 		fp->f_ci->m_fattr |= FILE_ATTRIBUTE_SPARSE_FILE_LE;
8253 	else
8254 		fp->f_ci->m_fattr &= ~FILE_ATTRIBUTE_SPARSE_FILE_LE;
8255 
8256 	if (fp->f_ci->m_fattr != old_fattr &&
8257 	    test_share_config_flag(work->tcon->share_conf,
8258 				   KSMBD_SHARE_FLAG_STORE_DOS_ATTRS)) {
8259 		struct xattr_dos_attrib da;
8260 
8261 		ret = ksmbd_vfs_get_dos_attrib_xattr(idmap,
8262 						     fp->filp->f_path.dentry, &da);
8263 		if (ret <= 0)
8264 			goto out;
8265 
8266 		da.attr = le32_to_cpu(fp->f_ci->m_fattr);
8267 		ret = ksmbd_vfs_set_dos_attrib_xattr(idmap,
8268 						     &fp->filp->f_path,
8269 						     &da, true);
8270 		if (ret)
8271 			fp->f_ci->m_fattr = old_fattr;
8272 	}
8273 
8274 out:
8275 	ksmbd_fd_put(work, fp);
8276 	return ret;
8277 }
8278 
8279 static int fsctl_request_resume_key(struct ksmbd_work *work,
8280 				    struct smb2_ioctl_req *req,
8281 				    struct resume_key_ioctl_rsp *key_rsp)
8282 {
8283 	struct ksmbd_file *fp;
8284 
8285 	fp = ksmbd_lookup_fd_slow(work, req->VolatileFileId, req->PersistentFileId);
8286 	if (!fp)
8287 		return -ENOENT;
8288 
8289 	memset(key_rsp, 0, sizeof(*key_rsp));
8290 	key_rsp->ResumeKeyU64[0] = req->VolatileFileId;
8291 	key_rsp->ResumeKeyU64[1] = req->PersistentFileId;
8292 	ksmbd_fd_put(work, fp);
8293 
8294 	return 0;
8295 }
8296 
8297 /**
8298  * smb2_ioctl() - handler for smb2 ioctl command
8299  * @work:	smb work containing ioctl command buffer
8300  *
8301  * Return:	0 on success, otherwise error
8302  */
8303 int smb2_ioctl(struct ksmbd_work *work)
8304 {
8305 	struct smb2_ioctl_req *req;
8306 	struct smb2_ioctl_rsp *rsp;
8307 	unsigned int cnt_code, nbytes = 0, out_buf_len, in_buf_len;
8308 	u64 id = KSMBD_NO_FID;
8309 	struct ksmbd_conn *conn = work->conn;
8310 	int ret = 0;
8311 	char *buffer;
8312 
8313 	ksmbd_debug(SMB, "Received smb2 ioctl request\n");
8314 
8315 	if (work->next_smb2_rcv_hdr_off) {
8316 		req = ksmbd_req_buf_next(work);
8317 		rsp = ksmbd_resp_buf_next(work);
8318 		if (!has_file_id(req->VolatileFileId)) {
8319 			ksmbd_debug(SMB, "Compound request set FID = %llu\n",
8320 				    work->compound_fid);
8321 			id = work->compound_fid;
8322 		}
8323 	} else {
8324 		req = smb_get_msg(work->request_buf);
8325 		rsp = smb_get_msg(work->response_buf);
8326 	}
8327 
8328 	if (!has_file_id(id))
8329 		id = req->VolatileFileId;
8330 
8331 	if (req->Flags != cpu_to_le32(SMB2_0_IOCTL_IS_FSCTL)) {
8332 		ret = -EOPNOTSUPP;
8333 		goto out;
8334 	}
8335 
8336 	buffer = (char *)req + le32_to_cpu(req->InputOffset);
8337 
8338 	cnt_code = le32_to_cpu(req->CtlCode);
8339 	ret = smb2_calc_max_out_buf_len(work,
8340 			offsetof(struct smb2_ioctl_rsp, Buffer),
8341 			le32_to_cpu(req->MaxOutputResponse));
8342 	if (ret < 0) {
8343 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
8344 		goto out;
8345 	}
8346 	out_buf_len = (unsigned int)ret;
8347 	in_buf_len = le32_to_cpu(req->InputCount);
8348 
8349 	switch (cnt_code) {
8350 	case FSCTL_DFS_GET_REFERRALS:
8351 	case FSCTL_DFS_GET_REFERRALS_EX:
8352 		/* Not support DFS yet */
8353 		ret = -EOPNOTSUPP;
8354 		rsp->hdr.Status = STATUS_FS_DRIVER_REQUIRED;
8355 		goto out2;
8356 	case FSCTL_CREATE_OR_GET_OBJECT_ID:
8357 	{
8358 		struct file_object_buf_type1_ioctl_rsp *obj_buf;
8359 
8360 		nbytes = sizeof(struct file_object_buf_type1_ioctl_rsp);
8361 		obj_buf = (struct file_object_buf_type1_ioctl_rsp *)
8362 			&rsp->Buffer[0];
8363 
8364 		/*
8365 		 * TODO: This is dummy implementation to pass smbtorture
8366 		 * Need to check correct response later
8367 		 */
8368 		memset(obj_buf->ObjectId, 0x0, 16);
8369 		memset(obj_buf->BirthVolumeId, 0x0, 16);
8370 		memset(obj_buf->BirthObjectId, 0x0, 16);
8371 		memset(obj_buf->DomainId, 0x0, 16);
8372 
8373 		break;
8374 	}
8375 	case FSCTL_PIPE_TRANSCEIVE:
8376 		out_buf_len = min_t(u32, KSMBD_IPC_MAX_PAYLOAD, out_buf_len);
8377 		nbytes = fsctl_pipe_transceive(work, id, out_buf_len, req, rsp);
8378 		break;
8379 	case FSCTL_VALIDATE_NEGOTIATE_INFO:
8380 		if (conn->dialect < SMB30_PROT_ID) {
8381 			ret = -EOPNOTSUPP;
8382 			goto out;
8383 		}
8384 
8385 		if (in_buf_len < offsetof(struct validate_negotiate_info_req,
8386 					  Dialects)) {
8387 			ret = -EINVAL;
8388 			goto out;
8389 		}
8390 
8391 		if (out_buf_len < sizeof(struct validate_negotiate_info_rsp)) {
8392 			ret = -EINVAL;
8393 			goto out;
8394 		}
8395 
8396 		ret = fsctl_validate_negotiate_info(conn,
8397 			(struct validate_negotiate_info_req *)buffer,
8398 			(struct validate_negotiate_info_rsp *)&rsp->Buffer[0],
8399 			in_buf_len);
8400 		if (ret < 0)
8401 			goto out;
8402 
8403 		nbytes = sizeof(struct validate_negotiate_info_rsp);
8404 		rsp->PersistentFileId = SMB2_NO_FID;
8405 		rsp->VolatileFileId = SMB2_NO_FID;
8406 		break;
8407 	case FSCTL_QUERY_NETWORK_INTERFACE_INFO:
8408 		ret = fsctl_query_iface_info_ioctl(conn, rsp, out_buf_len);
8409 		if (ret < 0)
8410 			goto out;
8411 		nbytes = ret;
8412 		break;
8413 	case FSCTL_REQUEST_RESUME_KEY:
8414 		if (out_buf_len < sizeof(struct resume_key_ioctl_rsp)) {
8415 			ret = -EINVAL;
8416 			goto out;
8417 		}
8418 
8419 		ret = fsctl_request_resume_key(work, req,
8420 					       (struct resume_key_ioctl_rsp *)&rsp->Buffer[0]);
8421 		if (ret < 0)
8422 			goto out;
8423 		rsp->PersistentFileId = req->PersistentFileId;
8424 		rsp->VolatileFileId = req->VolatileFileId;
8425 		nbytes = sizeof(struct resume_key_ioctl_rsp);
8426 		break;
8427 	case FSCTL_COPYCHUNK:
8428 	case FSCTL_COPYCHUNK_WRITE:
8429 		if (!test_tree_conn_flag(work->tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
8430 			ksmbd_debug(SMB,
8431 				    "User does not have write permission\n");
8432 			ret = -EACCES;
8433 			goto out;
8434 		}
8435 
8436 		if (in_buf_len <= sizeof(struct copychunk_ioctl_req)) {
8437 			ret = -EINVAL;
8438 			goto out;
8439 		}
8440 
8441 		if (out_buf_len < sizeof(struct copychunk_ioctl_rsp)) {
8442 			ret = -EINVAL;
8443 			goto out;
8444 		}
8445 
8446 		nbytes = sizeof(struct copychunk_ioctl_rsp);
8447 		rsp->VolatileFileId = req->VolatileFileId;
8448 		rsp->PersistentFileId = req->PersistentFileId;
8449 		fsctl_copychunk(work,
8450 				(struct copychunk_ioctl_req *)buffer,
8451 				le32_to_cpu(req->CtlCode),
8452 				le32_to_cpu(req->InputCount),
8453 				req->VolatileFileId,
8454 				req->PersistentFileId,
8455 				rsp);
8456 		break;
8457 	case FSCTL_SET_SPARSE:
8458 		if (in_buf_len < sizeof(struct file_sparse)) {
8459 			ret = -EINVAL;
8460 			goto out;
8461 		}
8462 
8463 		ret = fsctl_set_sparse(work, id, (struct file_sparse *)buffer);
8464 		if (ret < 0)
8465 			goto out;
8466 		break;
8467 	case FSCTL_SET_ZERO_DATA:
8468 	{
8469 		struct file_zero_data_information *zero_data;
8470 		struct ksmbd_file *fp;
8471 		loff_t off, len, bfz;
8472 
8473 		if (!test_tree_conn_flag(work->tcon, KSMBD_TREE_CONN_FLAG_WRITABLE)) {
8474 			ksmbd_debug(SMB,
8475 				    "User does not have write permission\n");
8476 			ret = -EACCES;
8477 			goto out;
8478 		}
8479 
8480 		if (in_buf_len < sizeof(struct file_zero_data_information)) {
8481 			ret = -EINVAL;
8482 			goto out;
8483 		}
8484 
8485 		zero_data =
8486 			(struct file_zero_data_information *)buffer;
8487 
8488 		off = le64_to_cpu(zero_data->FileOffset);
8489 		bfz = le64_to_cpu(zero_data->BeyondFinalZero);
8490 		if (off < 0 || bfz < 0 || off > bfz) {
8491 			ret = -EINVAL;
8492 			goto out;
8493 		}
8494 
8495 		len = bfz - off;
8496 		if (len) {
8497 			fp = ksmbd_lookup_fd_fast(work, id);
8498 			if (!fp) {
8499 				ret = -ENOENT;
8500 				goto out;
8501 			}
8502 
8503 			ret = ksmbd_vfs_zero_data(work, fp, off, len);
8504 			ksmbd_fd_put(work, fp);
8505 			if (ret < 0)
8506 				goto out;
8507 		}
8508 		break;
8509 	}
8510 	case FSCTL_QUERY_ALLOCATED_RANGES:
8511 		if (in_buf_len < sizeof(struct file_allocated_range_buffer)) {
8512 			ret = -EINVAL;
8513 			goto out;
8514 		}
8515 
8516 		ret = fsctl_query_allocated_ranges(work, id,
8517 			(struct file_allocated_range_buffer *)buffer,
8518 			(struct file_allocated_range_buffer *)&rsp->Buffer[0],
8519 			out_buf_len /
8520 			sizeof(struct file_allocated_range_buffer), &nbytes);
8521 		if (ret == -E2BIG) {
8522 			rsp->hdr.Status = STATUS_BUFFER_OVERFLOW;
8523 		} else if (ret < 0) {
8524 			nbytes = 0;
8525 			goto out;
8526 		}
8527 
8528 		nbytes *= sizeof(struct file_allocated_range_buffer);
8529 		break;
8530 	case FSCTL_GET_REPARSE_POINT:
8531 	{
8532 		struct reparse_data_buffer *reparse_ptr;
8533 		struct ksmbd_file *fp;
8534 
8535 		reparse_ptr = (struct reparse_data_buffer *)&rsp->Buffer[0];
8536 		fp = ksmbd_lookup_fd_fast(work, id);
8537 		if (!fp) {
8538 			pr_err("not found fp!!\n");
8539 			ret = -ENOENT;
8540 			goto out;
8541 		}
8542 
8543 		reparse_ptr->ReparseTag =
8544 			smb2_get_reparse_tag_special_file(file_inode(fp->filp)->i_mode);
8545 		reparse_ptr->ReparseDataLength = 0;
8546 		ksmbd_fd_put(work, fp);
8547 		nbytes = sizeof(struct reparse_data_buffer);
8548 		break;
8549 	}
8550 	case FSCTL_DUPLICATE_EXTENTS_TO_FILE:
8551 	{
8552 		struct ksmbd_file *fp_in, *fp_out = NULL;
8553 		struct duplicate_extents_to_file *dup_ext;
8554 		loff_t src_off, dst_off, length, cloned;
8555 
8556 		if (in_buf_len < sizeof(struct duplicate_extents_to_file)) {
8557 			ret = -EINVAL;
8558 			goto out;
8559 		}
8560 
8561 		dup_ext = (struct duplicate_extents_to_file *)buffer;
8562 
8563 		fp_in = ksmbd_lookup_fd_slow(work, dup_ext->VolatileFileHandle,
8564 					     dup_ext->PersistentFileHandle);
8565 		if (!fp_in) {
8566 			pr_err("not found file handle in duplicate extent to file\n");
8567 			ret = -ENOENT;
8568 			goto out;
8569 		}
8570 
8571 		fp_out = ksmbd_lookup_fd_fast(work, id);
8572 		if (!fp_out) {
8573 			pr_err("not found fp\n");
8574 			ret = -ENOENT;
8575 			goto dup_ext_out;
8576 		}
8577 
8578 		src_off = le64_to_cpu(dup_ext->SourceFileOffset);
8579 		dst_off = le64_to_cpu(dup_ext->TargetFileOffset);
8580 		length = le64_to_cpu(dup_ext->ByteCount);
8581 		/*
8582 		 * XXX: It is not clear if FSCTL_DUPLICATE_EXTENTS_TO_FILE
8583 		 * should fall back to vfs_copy_file_range().  This could be
8584 		 * beneficial when re-exporting nfs/smb mount, but note that
8585 		 * this can result in partial copy that returns an error status.
8586 		 * If/when FSCTL_DUPLICATE_EXTENTS_TO_FILE_EX is implemented,
8587 		 * fall back to vfs_copy_file_range(), should be avoided when
8588 		 * the flag DUPLICATE_EXTENTS_DATA_EX_SOURCE_ATOMIC is set.
8589 		 */
8590 		cloned = vfs_clone_file_range(fp_in->filp, src_off,
8591 					      fp_out->filp, dst_off, length, 0);
8592 		if (cloned == -EXDEV || cloned == -EOPNOTSUPP) {
8593 			ret = -EOPNOTSUPP;
8594 			goto dup_ext_out;
8595 		} else if (cloned != length) {
8596 			cloned = vfs_copy_file_range(fp_in->filp, src_off,
8597 						     fp_out->filp, dst_off,
8598 						     length, 0);
8599 			if (cloned != length) {
8600 				if (cloned < 0)
8601 					ret = cloned;
8602 				else
8603 					ret = -EINVAL;
8604 			}
8605 		}
8606 
8607 dup_ext_out:
8608 		ksmbd_fd_put(work, fp_in);
8609 		ksmbd_fd_put(work, fp_out);
8610 		if (ret < 0)
8611 			goto out;
8612 		break;
8613 	}
8614 	default:
8615 		ksmbd_debug(SMB, "not implemented yet ioctl command 0x%x\n",
8616 			    cnt_code);
8617 		ret = -EOPNOTSUPP;
8618 		goto out;
8619 	}
8620 
8621 	rsp->CtlCode = cpu_to_le32(cnt_code);
8622 	rsp->InputCount = cpu_to_le32(0);
8623 	rsp->InputOffset = cpu_to_le32(112);
8624 	rsp->OutputOffset = cpu_to_le32(112);
8625 	rsp->OutputCount = cpu_to_le32(nbytes);
8626 	rsp->StructureSize = cpu_to_le16(49);
8627 	rsp->Reserved = cpu_to_le16(0);
8628 	rsp->Flags = cpu_to_le32(0);
8629 	rsp->Reserved2 = cpu_to_le32(0);
8630 	ret = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_ioctl_rsp) + nbytes);
8631 	if (!ret)
8632 		return ret;
8633 
8634 out:
8635 	if (ret == -EACCES)
8636 		rsp->hdr.Status = STATUS_ACCESS_DENIED;
8637 	else if (ret == -ENOENT)
8638 		rsp->hdr.Status = STATUS_OBJECT_NAME_NOT_FOUND;
8639 	else if (ret == -EOPNOTSUPP)
8640 		rsp->hdr.Status = STATUS_NOT_SUPPORTED;
8641 	else if (ret == -ENOSPC)
8642 		rsp->hdr.Status = STATUS_BUFFER_TOO_SMALL;
8643 	else if (ret < 0 || rsp->hdr.Status == 0)
8644 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
8645 
8646 out2:
8647 	smb2_set_err_rsp(work);
8648 	return ret;
8649 }
8650 
8651 /**
8652  * smb20_oplock_break_ack() - handler for smb2.0 oplock break command
8653  * @work:	smb work containing oplock break command buffer
8654  *
8655  * Return:	0
8656  */
8657 static void smb20_oplock_break_ack(struct ksmbd_work *work)
8658 {
8659 	struct smb2_oplock_break *req;
8660 	struct smb2_oplock_break *rsp;
8661 	struct ksmbd_file *fp;
8662 	struct oplock_info *opinfo = NULL;
8663 	__le32 err = 0;
8664 	int ret = 0;
8665 	u64 volatile_id, persistent_id;
8666 	char req_oplevel = 0, rsp_oplevel = 0;
8667 	unsigned int oplock_change_type;
8668 
8669 	WORK_BUFFERS(work, req, rsp);
8670 
8671 	volatile_id = req->VolatileFid;
8672 	persistent_id = req->PersistentFid;
8673 	req_oplevel = req->OplockLevel;
8674 	ksmbd_debug(OPLOCK, "v_id %llu, p_id %llu request oplock level %d\n",
8675 		    volatile_id, persistent_id, req_oplevel);
8676 
8677 	fp = ksmbd_lookup_fd_slow(work, volatile_id, persistent_id);
8678 	if (!fp) {
8679 		rsp->hdr.Status = STATUS_FILE_CLOSED;
8680 		smb2_set_err_rsp(work);
8681 		return;
8682 	}
8683 
8684 	opinfo = opinfo_get(fp);
8685 	if (!opinfo) {
8686 		pr_err("unexpected null oplock_info\n");
8687 		rsp->hdr.Status = STATUS_INVALID_OPLOCK_PROTOCOL;
8688 		smb2_set_err_rsp(work);
8689 		ksmbd_fd_put(work, fp);
8690 		return;
8691 	}
8692 
8693 	if (opinfo->level == SMB2_OPLOCK_LEVEL_NONE) {
8694 		rsp->hdr.Status = STATUS_INVALID_OPLOCK_PROTOCOL;
8695 		goto err_out;
8696 	}
8697 
8698 	if (opinfo->op_state == OPLOCK_STATE_NONE) {
8699 		ksmbd_debug(SMB, "unexpected oplock state 0x%x\n", opinfo->op_state);
8700 		rsp->hdr.Status = STATUS_UNSUCCESSFUL;
8701 		goto err_out;
8702 	}
8703 
8704 	if ((opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE ||
8705 	     opinfo->level == SMB2_OPLOCK_LEVEL_BATCH) &&
8706 	    (req_oplevel != SMB2_OPLOCK_LEVEL_II &&
8707 	     req_oplevel != SMB2_OPLOCK_LEVEL_NONE)) {
8708 		err = STATUS_INVALID_OPLOCK_PROTOCOL;
8709 		oplock_change_type = OPLOCK_WRITE_TO_NONE;
8710 	} else if (opinfo->level == SMB2_OPLOCK_LEVEL_II &&
8711 		   req_oplevel != SMB2_OPLOCK_LEVEL_NONE) {
8712 		err = STATUS_INVALID_OPLOCK_PROTOCOL;
8713 		oplock_change_type = OPLOCK_READ_TO_NONE;
8714 	} else if (req_oplevel == SMB2_OPLOCK_LEVEL_II ||
8715 		   req_oplevel == SMB2_OPLOCK_LEVEL_NONE) {
8716 		err = STATUS_INVALID_DEVICE_STATE;
8717 		if ((opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE ||
8718 		     opinfo->level == SMB2_OPLOCK_LEVEL_BATCH) &&
8719 		    req_oplevel == SMB2_OPLOCK_LEVEL_II) {
8720 			oplock_change_type = OPLOCK_WRITE_TO_READ;
8721 		} else if ((opinfo->level == SMB2_OPLOCK_LEVEL_EXCLUSIVE ||
8722 			    opinfo->level == SMB2_OPLOCK_LEVEL_BATCH) &&
8723 			   req_oplevel == SMB2_OPLOCK_LEVEL_NONE) {
8724 			oplock_change_type = OPLOCK_WRITE_TO_NONE;
8725 		} else if (opinfo->level == SMB2_OPLOCK_LEVEL_II &&
8726 			   req_oplevel == SMB2_OPLOCK_LEVEL_NONE) {
8727 			oplock_change_type = OPLOCK_READ_TO_NONE;
8728 		} else {
8729 			oplock_change_type = 0;
8730 		}
8731 	} else {
8732 		oplock_change_type = 0;
8733 	}
8734 
8735 	switch (oplock_change_type) {
8736 	case OPLOCK_WRITE_TO_READ:
8737 		ret = opinfo_write_to_read(opinfo);
8738 		rsp_oplevel = SMB2_OPLOCK_LEVEL_II;
8739 		break;
8740 	case OPLOCK_WRITE_TO_NONE:
8741 		ret = opinfo_write_to_none(opinfo);
8742 		rsp_oplevel = SMB2_OPLOCK_LEVEL_NONE;
8743 		break;
8744 	case OPLOCK_READ_TO_NONE:
8745 		ret = opinfo_read_to_none(opinfo);
8746 		rsp_oplevel = SMB2_OPLOCK_LEVEL_NONE;
8747 		break;
8748 	default:
8749 		pr_err("unknown oplock change 0x%x -> 0x%x\n",
8750 		       opinfo->level, rsp_oplevel);
8751 	}
8752 
8753 	if (ret < 0) {
8754 		rsp->hdr.Status = err;
8755 		goto err_out;
8756 	}
8757 
8758 	rsp->StructureSize = cpu_to_le16(24);
8759 	rsp->OplockLevel = rsp_oplevel;
8760 	rsp->Reserved = 0;
8761 	rsp->Reserved2 = 0;
8762 	rsp->VolatileFid = volatile_id;
8763 	rsp->PersistentFid = persistent_id;
8764 	ret = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_oplock_break));
8765 	if (ret) {
8766 err_out:
8767 		smb2_set_err_rsp(work);
8768 	}
8769 
8770 	opinfo->op_state = OPLOCK_STATE_NONE;
8771 	wake_up_interruptible_all(&opinfo->oplock_q);
8772 	opinfo_put(opinfo);
8773 	ksmbd_fd_put(work, fp);
8774 }
8775 
8776 static int check_lease_state(struct lease *lease, __le32 req_state)
8777 {
8778 	if ((lease->new_state ==
8779 	     (SMB2_LEASE_READ_CACHING_LE | SMB2_LEASE_HANDLE_CACHING_LE)) &&
8780 	    !(req_state & SMB2_LEASE_WRITE_CACHING_LE)) {
8781 		lease->new_state = req_state;
8782 		return 0;
8783 	}
8784 
8785 	if (lease->new_state == req_state)
8786 		return 0;
8787 
8788 	return 1;
8789 }
8790 
8791 /**
8792  * smb21_lease_break_ack() - handler for smb2.1 lease break command
8793  * @work:	smb work containing lease break command buffer
8794  *
8795  * Return:	0
8796  */
8797 static void smb21_lease_break_ack(struct ksmbd_work *work)
8798 {
8799 	struct ksmbd_conn *conn = work->conn;
8800 	struct smb2_lease_ack *req;
8801 	struct smb2_lease_ack *rsp;
8802 	struct oplock_info *opinfo;
8803 	__le32 err = 0;
8804 	int ret = 0;
8805 	unsigned int lease_change_type;
8806 	__le32 lease_state;
8807 	struct lease *lease;
8808 
8809 	WORK_BUFFERS(work, req, rsp);
8810 
8811 	ksmbd_debug(OPLOCK, "smb21 lease break, lease state(0x%x)\n",
8812 		    le32_to_cpu(req->LeaseState));
8813 	opinfo = lookup_lease_in_table(conn, req->LeaseKey);
8814 	if (!opinfo) {
8815 		ksmbd_debug(OPLOCK, "file not opened\n");
8816 		smb2_set_err_rsp(work);
8817 		rsp->hdr.Status = STATUS_UNSUCCESSFUL;
8818 		return;
8819 	}
8820 	lease = opinfo->o_lease;
8821 
8822 	if (opinfo->op_state == OPLOCK_STATE_NONE) {
8823 		pr_err("unexpected lease break state 0x%x\n",
8824 		       opinfo->op_state);
8825 		rsp->hdr.Status = STATUS_UNSUCCESSFUL;
8826 		goto err_out;
8827 	}
8828 
8829 	if (check_lease_state(lease, req->LeaseState)) {
8830 		rsp->hdr.Status = STATUS_REQUEST_NOT_ACCEPTED;
8831 		ksmbd_debug(OPLOCK,
8832 			    "req lease state: 0x%x, expected state: 0x%x\n",
8833 			    req->LeaseState, lease->new_state);
8834 		goto err_out;
8835 	}
8836 
8837 	if (!atomic_read(&opinfo->breaking_cnt)) {
8838 		rsp->hdr.Status = STATUS_UNSUCCESSFUL;
8839 		goto err_out;
8840 	}
8841 
8842 	/* check for bad lease state */
8843 	if (req->LeaseState &
8844 	    (~(SMB2_LEASE_READ_CACHING_LE | SMB2_LEASE_HANDLE_CACHING_LE))) {
8845 		err = STATUS_INVALID_OPLOCK_PROTOCOL;
8846 		if (lease->state & SMB2_LEASE_WRITE_CACHING_LE)
8847 			lease_change_type = OPLOCK_WRITE_TO_NONE;
8848 		else
8849 			lease_change_type = OPLOCK_READ_TO_NONE;
8850 		ksmbd_debug(OPLOCK, "handle bad lease state 0x%x -> 0x%x\n",
8851 			    le32_to_cpu(lease->state),
8852 			    le32_to_cpu(req->LeaseState));
8853 	} else if (lease->state == SMB2_LEASE_READ_CACHING_LE &&
8854 		   req->LeaseState != SMB2_LEASE_NONE_LE) {
8855 		err = STATUS_INVALID_OPLOCK_PROTOCOL;
8856 		lease_change_type = OPLOCK_READ_TO_NONE;
8857 		ksmbd_debug(OPLOCK, "handle bad lease state 0x%x -> 0x%x\n",
8858 			    le32_to_cpu(lease->state),
8859 			    le32_to_cpu(req->LeaseState));
8860 	} else {
8861 		/* valid lease state changes */
8862 		err = STATUS_INVALID_DEVICE_STATE;
8863 		if (req->LeaseState == SMB2_LEASE_NONE_LE) {
8864 			if (lease->state & SMB2_LEASE_WRITE_CACHING_LE)
8865 				lease_change_type = OPLOCK_WRITE_TO_NONE;
8866 			else
8867 				lease_change_type = OPLOCK_READ_TO_NONE;
8868 		} else if (req->LeaseState & SMB2_LEASE_READ_CACHING_LE) {
8869 			if (lease->state & SMB2_LEASE_WRITE_CACHING_LE)
8870 				lease_change_type = OPLOCK_WRITE_TO_READ;
8871 			else
8872 				lease_change_type = OPLOCK_READ_HANDLE_TO_READ;
8873 		} else {
8874 			lease_change_type = 0;
8875 		}
8876 	}
8877 
8878 	switch (lease_change_type) {
8879 	case OPLOCK_WRITE_TO_READ:
8880 		ret = opinfo_write_to_read(opinfo);
8881 		break;
8882 	case OPLOCK_READ_HANDLE_TO_READ:
8883 		ret = opinfo_read_handle_to_read(opinfo);
8884 		break;
8885 	case OPLOCK_WRITE_TO_NONE:
8886 		ret = opinfo_write_to_none(opinfo);
8887 		break;
8888 	case OPLOCK_READ_TO_NONE:
8889 		ret = opinfo_read_to_none(opinfo);
8890 		break;
8891 	default:
8892 		ksmbd_debug(OPLOCK, "unknown lease change 0x%x -> 0x%x\n",
8893 			    le32_to_cpu(lease->state),
8894 			    le32_to_cpu(req->LeaseState));
8895 	}
8896 
8897 	if (ret < 0) {
8898 		rsp->hdr.Status = err;
8899 		goto err_out;
8900 	}
8901 
8902 	lease_state = lease->state;
8903 
8904 	rsp->StructureSize = cpu_to_le16(36);
8905 	rsp->Reserved = 0;
8906 	rsp->Flags = 0;
8907 	memcpy(rsp->LeaseKey, req->LeaseKey, 16);
8908 	rsp->LeaseState = lease_state;
8909 	rsp->LeaseDuration = 0;
8910 	ret = ksmbd_iov_pin_rsp(work, rsp, sizeof(struct smb2_lease_ack));
8911 	if (ret) {
8912 err_out:
8913 		smb2_set_err_rsp(work);
8914 	}
8915 
8916 	opinfo->op_state = OPLOCK_STATE_NONE;
8917 	wake_up_interruptible_all(&opinfo->oplock_q);
8918 	atomic_dec(&opinfo->breaking_cnt);
8919 	wake_up_interruptible_all(&opinfo->oplock_brk);
8920 	opinfo_put(opinfo);
8921 }
8922 
8923 /**
8924  * smb2_oplock_break() - dispatcher for smb2.0 and 2.1 oplock/lease break
8925  * @work:	smb work containing oplock/lease break command buffer
8926  *
8927  * Return:	0 on success, otherwise error
8928  */
8929 int smb2_oplock_break(struct ksmbd_work *work)
8930 {
8931 	struct smb2_oplock_break *req;
8932 	struct smb2_oplock_break *rsp;
8933 
8934 	ksmbd_debug(SMB, "Received smb2 oplock break acknowledgment request\n");
8935 
8936 	WORK_BUFFERS(work, req, rsp);
8937 
8938 	switch (le16_to_cpu(req->StructureSize)) {
8939 	case OP_BREAK_STRUCT_SIZE_20:
8940 		smb20_oplock_break_ack(work);
8941 		break;
8942 	case OP_BREAK_STRUCT_SIZE_21:
8943 		smb21_lease_break_ack(work);
8944 		break;
8945 	default:
8946 		ksmbd_debug(OPLOCK, "invalid break cmd %d\n",
8947 			    le16_to_cpu(req->StructureSize));
8948 		rsp->hdr.Status = STATUS_INVALID_PARAMETER;
8949 		smb2_set_err_rsp(work);
8950 		return -EINVAL;
8951 	}
8952 
8953 	return 0;
8954 }
8955 
8956 /**
8957  * smb2_notify() - handler for smb2 notify request
8958  * @work:   smb work containing notify command buffer
8959  *
8960  * Return:      0 on success, otherwise error
8961  */
8962 int smb2_notify(struct ksmbd_work *work)
8963 {
8964 	struct smb2_change_notify_req *req;
8965 	struct smb2_change_notify_rsp *rsp;
8966 
8967 	ksmbd_debug(SMB, "Received smb2 notify\n");
8968 
8969 	WORK_BUFFERS(work, req, rsp);
8970 
8971 	if (work->next_smb2_rcv_hdr_off && req->hdr.NextCommand) {
8972 		rsp->hdr.Status = STATUS_INTERNAL_ERROR;
8973 		smb2_set_err_rsp(work);
8974 		return -EIO;
8975 	}
8976 
8977 	smb2_set_err_rsp(work);
8978 	rsp->hdr.Status = STATUS_NOT_IMPLEMENTED;
8979 	return -EOPNOTSUPP;
8980 }
8981 
8982 /**
8983  * smb2_is_sign_req() - handler for checking packet signing status
8984  * @work:	smb work containing notify command buffer
8985  * @command:	SMB2 command id
8986  *
8987  * Return:	true if packed is signed, false otherwise
8988  */
8989 bool smb2_is_sign_req(struct ksmbd_work *work, unsigned int command)
8990 {
8991 	struct smb2_hdr *rcv_hdr2 = smb_get_msg(work->request_buf);
8992 
8993 	if ((rcv_hdr2->Flags & SMB2_FLAGS_SIGNED) &&
8994 	    command != SMB2_NEGOTIATE_HE &&
8995 	    command != SMB2_SESSION_SETUP_HE &&
8996 	    command != SMB2_OPLOCK_BREAK_HE)
8997 		return true;
8998 
8999 	return false;
9000 }
9001 
9002 /**
9003  * smb2_check_sign_req() - handler for req packet sign processing
9004  * @work:   smb work containing notify command buffer
9005  *
9006  * Return:	1 on success, 0 otherwise
9007  */
9008 int smb2_check_sign_req(struct ksmbd_work *work)
9009 {
9010 	struct smb2_hdr *hdr;
9011 	char signature_req[SMB2_SIGNATURE_SIZE];
9012 	char signature[SMB2_HMACSHA256_SIZE];
9013 	struct kvec iov[1];
9014 	size_t len;
9015 
9016 	hdr = smb_get_msg(work->request_buf);
9017 	if (work->next_smb2_rcv_hdr_off)
9018 		hdr = ksmbd_req_buf_next(work);
9019 
9020 	if (!hdr->NextCommand && !work->next_smb2_rcv_hdr_off)
9021 		len = get_rfc1002_len(work->request_buf);
9022 	else if (hdr->NextCommand)
9023 		len = le32_to_cpu(hdr->NextCommand);
9024 	else
9025 		len = get_rfc1002_len(work->request_buf) -
9026 			work->next_smb2_rcv_hdr_off;
9027 
9028 	memcpy(signature_req, hdr->Signature, SMB2_SIGNATURE_SIZE);
9029 	memset(hdr->Signature, 0, SMB2_SIGNATURE_SIZE);
9030 
9031 	iov[0].iov_base = (char *)&hdr->ProtocolId;
9032 	iov[0].iov_len = len;
9033 
9034 	ksmbd_sign_smb2_pdu(work->conn, work->sess->sess_key, iov, 1,
9035 			    signature);
9036 
9037 	if (crypto_memneq(signature, signature_req, SMB2_SIGNATURE_SIZE)) {
9038 		pr_err("bad smb2 signature\n");
9039 		return 0;
9040 	}
9041 
9042 	return 1;
9043 }
9044 
9045 /**
9046  * smb2_set_sign_rsp() - handler for rsp packet sign processing
9047  * @work:   smb work containing notify command buffer
9048  *
9049  */
9050 void smb2_set_sign_rsp(struct ksmbd_work *work)
9051 {
9052 	struct smb2_hdr *hdr;
9053 	char signature[SMB2_HMACSHA256_SIZE];
9054 	struct kvec *iov;
9055 	int n_vec = 1;
9056 
9057 	hdr = ksmbd_resp_buf_curr(work);
9058 	hdr->Flags |= SMB2_FLAGS_SIGNED;
9059 	memset(hdr->Signature, 0, SMB2_SIGNATURE_SIZE);
9060 
9061 	if (hdr->Command == SMB2_READ) {
9062 		iov = &work->iov[work->iov_idx - 1];
9063 		n_vec++;
9064 	} else {
9065 		iov = &work->iov[work->iov_idx];
9066 	}
9067 
9068 	ksmbd_sign_smb2_pdu(work->conn, work->sess->sess_key, iov, n_vec,
9069 			    signature);
9070 	memcpy(hdr->Signature, signature, SMB2_SIGNATURE_SIZE);
9071 }
9072 
9073 /**
9074  * smb3_check_sign_req() - handler for req packet sign processing
9075  * @work:   smb work containing notify command buffer
9076  *
9077  * Return:	1 on success, 0 otherwise
9078  */
9079 int smb3_check_sign_req(struct ksmbd_work *work)
9080 {
9081 	struct ksmbd_conn *conn = work->conn;
9082 	char *signing_key;
9083 	struct smb2_hdr *hdr;
9084 	struct channel *chann;
9085 	char signature_req[SMB2_SIGNATURE_SIZE];
9086 	char signature[SMB2_CMACAES_SIZE];
9087 	struct kvec iov[1];
9088 	size_t len;
9089 
9090 	hdr = smb_get_msg(work->request_buf);
9091 	if (work->next_smb2_rcv_hdr_off)
9092 		hdr = ksmbd_req_buf_next(work);
9093 
9094 	if (!hdr->NextCommand && !work->next_smb2_rcv_hdr_off)
9095 		len = get_rfc1002_len(work->request_buf);
9096 	else if (hdr->NextCommand)
9097 		len = le32_to_cpu(hdr->NextCommand);
9098 	else
9099 		len = get_rfc1002_len(work->request_buf) -
9100 			work->next_smb2_rcv_hdr_off;
9101 
9102 	if (le16_to_cpu(hdr->Command) == SMB2_SESSION_SETUP_HE) {
9103 		signing_key = work->sess->smb3signingkey;
9104 	} else {
9105 		chann = lookup_chann_list(work->sess, conn);
9106 		if (!chann) {
9107 			return 0;
9108 		}
9109 		signing_key = chann->smb3signingkey;
9110 	}
9111 
9112 	if (!signing_key) {
9113 		pr_err("SMB3 signing key is not generated\n");
9114 		return 0;
9115 	}
9116 
9117 	memcpy(signature_req, hdr->Signature, SMB2_SIGNATURE_SIZE);
9118 	memset(hdr->Signature, 0, SMB2_SIGNATURE_SIZE);
9119 	iov[0].iov_base = (char *)&hdr->ProtocolId;
9120 	iov[0].iov_len = len;
9121 
9122 	ksmbd_sign_smb3_pdu(conn, signing_key, iov, 1, signature);
9123 
9124 	if (crypto_memneq(signature, signature_req, SMB2_SIGNATURE_SIZE)) {
9125 		pr_err("bad smb2 signature\n");
9126 		return 0;
9127 	}
9128 
9129 	return 1;
9130 }
9131 
9132 /**
9133  * smb3_set_sign_rsp() - handler for rsp packet sign processing
9134  * @work:   smb work containing notify command buffer
9135  *
9136  */
9137 void smb3_set_sign_rsp(struct ksmbd_work *work)
9138 {
9139 	struct ksmbd_conn *conn = work->conn;
9140 	struct smb2_hdr *hdr;
9141 	struct channel *chann;
9142 	char signature[SMB2_CMACAES_SIZE];
9143 	struct kvec *iov;
9144 	int n_vec = 1;
9145 	char *signing_key;
9146 
9147 	hdr = ksmbd_resp_buf_curr(work);
9148 
9149 	if (conn->binding == false &&
9150 	    le16_to_cpu(hdr->Command) == SMB2_SESSION_SETUP_HE) {
9151 		signing_key = work->sess->smb3signingkey;
9152 	} else {
9153 		chann = lookup_chann_list(work->sess, work->conn);
9154 		if (!chann) {
9155 			return;
9156 		}
9157 		signing_key = chann->smb3signingkey;
9158 	}
9159 
9160 	if (!signing_key)
9161 		return;
9162 
9163 	hdr->Flags |= SMB2_FLAGS_SIGNED;
9164 	memset(hdr->Signature, 0, SMB2_SIGNATURE_SIZE);
9165 
9166 	if (hdr->Command == SMB2_READ) {
9167 		iov = &work->iov[work->iov_idx - 1];
9168 		n_vec++;
9169 	} else {
9170 		iov = &work->iov[work->iov_idx];
9171 	}
9172 
9173 	ksmbd_sign_smb3_pdu(conn, signing_key, iov, n_vec, signature);
9174 	memcpy(hdr->Signature, signature, SMB2_SIGNATURE_SIZE);
9175 }
9176 
9177 /**
9178  * smb3_preauth_hash_rsp() - handler for computing preauth hash on response
9179  * @work:   smb work containing response buffer
9180  *
9181  */
9182 void smb3_preauth_hash_rsp(struct ksmbd_work *work)
9183 {
9184 	struct ksmbd_conn *conn = work->conn;
9185 	struct ksmbd_session *sess = work->sess;
9186 	struct smb2_hdr *req, *rsp;
9187 
9188 	if (conn->dialect != SMB311_PROT_ID)
9189 		return;
9190 
9191 	WORK_BUFFERS(work, req, rsp);
9192 
9193 	if (le16_to_cpu(req->Command) == SMB2_NEGOTIATE_HE &&
9194 	    conn->preauth_info)
9195 		ksmbd_gen_preauth_integrity_hash(conn, work->response_buf,
9196 						 conn->preauth_info->Preauth_HashValue);
9197 
9198 	if (le16_to_cpu(rsp->Command) == SMB2_SESSION_SETUP_HE && sess) {
9199 		__u8 *hash_value;
9200 
9201 		if (conn->binding) {
9202 			struct preauth_session *preauth_sess;
9203 
9204 			preauth_sess = ksmbd_preauth_session_lookup(conn, sess->id);
9205 			if (!preauth_sess)
9206 				return;
9207 			hash_value = preauth_sess->Preauth_HashValue;
9208 		} else {
9209 			hash_value = sess->Preauth_HashValue;
9210 			if (!hash_value)
9211 				return;
9212 		}
9213 		ksmbd_gen_preauth_integrity_hash(conn, work->response_buf,
9214 						 hash_value);
9215 	}
9216 }
9217 
9218 static void fill_transform_hdr(void *tr_buf, char *old_buf, __le16 cipher_type)
9219 {
9220 	struct smb2_transform_hdr *tr_hdr = tr_buf + 4;
9221 	struct smb2_hdr *hdr = smb_get_msg(old_buf);
9222 	unsigned int orig_len = get_rfc1002_len(old_buf);
9223 
9224 	/* tr_buf must be cleared by the caller */
9225 	tr_hdr->ProtocolId = SMB2_TRANSFORM_PROTO_NUM;
9226 	tr_hdr->OriginalMessageSize = cpu_to_le32(orig_len);
9227 	tr_hdr->Flags = cpu_to_le16(TRANSFORM_FLAG_ENCRYPTED);
9228 	if (cipher_type == SMB2_ENCRYPTION_AES128_GCM ||
9229 	    cipher_type == SMB2_ENCRYPTION_AES256_GCM)
9230 		get_random_bytes(&tr_hdr->Nonce, SMB3_AES_GCM_NONCE);
9231 	else
9232 		get_random_bytes(&tr_hdr->Nonce, SMB3_AES_CCM_NONCE);
9233 	memcpy(&tr_hdr->SessionId, &hdr->SessionId, 8);
9234 	inc_rfc1001_len(tr_buf, sizeof(struct smb2_transform_hdr));
9235 	inc_rfc1001_len(tr_buf, orig_len);
9236 }
9237 
9238 int smb3_encrypt_resp(struct ksmbd_work *work)
9239 {
9240 	struct kvec *iov = work->iov;
9241 	int rc = -ENOMEM;
9242 	void *tr_buf;
9243 
9244 	tr_buf = kzalloc(sizeof(struct smb2_transform_hdr) + 4, KSMBD_DEFAULT_GFP);
9245 	if (!tr_buf)
9246 		return rc;
9247 
9248 	/* fill transform header */
9249 	fill_transform_hdr(tr_buf, work->response_buf, work->conn->cipher_type);
9250 
9251 	iov[0].iov_base = tr_buf;
9252 	iov[0].iov_len = sizeof(struct smb2_transform_hdr) + 4;
9253 	work->tr_buf = tr_buf;
9254 
9255 	return ksmbd_crypt_message(work, iov, work->iov_idx + 1, 1);
9256 }
9257 
9258 bool smb3_is_transform_hdr(void *buf)
9259 {
9260 	struct smb2_transform_hdr *trhdr = smb_get_msg(buf);
9261 
9262 	return trhdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM;
9263 }
9264 
9265 int smb3_decrypt_req(struct ksmbd_work *work)
9266 {
9267 	struct ksmbd_session *sess;
9268 	char *buf = work->request_buf;
9269 	unsigned int pdu_length = get_rfc1002_len(buf);
9270 	struct kvec iov[2];
9271 	int buf_data_size = pdu_length - sizeof(struct smb2_transform_hdr);
9272 	struct smb2_transform_hdr *tr_hdr = smb_get_msg(buf);
9273 	int rc = 0;
9274 
9275 	if (pdu_length < sizeof(struct smb2_transform_hdr) ||
9276 	    buf_data_size < sizeof(struct smb2_hdr)) {
9277 		pr_err("Transform message is too small (%u)\n",
9278 		       pdu_length);
9279 		return -ECONNABORTED;
9280 	}
9281 
9282 	if (buf_data_size < le32_to_cpu(tr_hdr->OriginalMessageSize)) {
9283 		pr_err("Transform message is broken\n");
9284 		return -ECONNABORTED;
9285 	}
9286 
9287 	sess = ksmbd_session_lookup_all(work->conn, le64_to_cpu(tr_hdr->SessionId));
9288 	if (!sess) {
9289 		pr_err("invalid session id(%llx) in transform header\n",
9290 		       le64_to_cpu(tr_hdr->SessionId));
9291 		return -ECONNABORTED;
9292 	}
9293 	ksmbd_user_session_put(sess);
9294 
9295 	iov[0].iov_base = buf;
9296 	iov[0].iov_len = sizeof(struct smb2_transform_hdr) + 4;
9297 	iov[1].iov_base = buf + sizeof(struct smb2_transform_hdr) + 4;
9298 	iov[1].iov_len = buf_data_size;
9299 	rc = ksmbd_crypt_message(work, iov, 2, 0);
9300 	if (rc)
9301 		return rc;
9302 
9303 	memmove(buf + 4, iov[1].iov_base, buf_data_size);
9304 	*(__be32 *)buf = cpu_to_be32(buf_data_size);
9305 
9306 	return rc;
9307 }
9308 
9309 bool smb3_11_final_sess_setup_resp(struct ksmbd_work *work)
9310 {
9311 	struct ksmbd_conn *conn = work->conn;
9312 	struct ksmbd_session *sess = work->sess;
9313 	struct smb2_hdr *rsp = smb_get_msg(work->response_buf);
9314 
9315 	if (conn->dialect < SMB30_PROT_ID)
9316 		return false;
9317 
9318 	if (work->next_smb2_rcv_hdr_off)
9319 		rsp = ksmbd_resp_buf_next(work);
9320 
9321 	if (le16_to_cpu(rsp->Command) == SMB2_SESSION_SETUP_HE &&
9322 	    sess->user && !user_guest(sess->user) &&
9323 	    rsp->Status == STATUS_SUCCESS)
9324 		return true;
9325 	return false;
9326 }
9327