]> git.ipfire.org Git - thirdparty/kernel/linux.git/blame - fs/cifs/smb2pdu.c
projects / thirdparty / kernel / linux.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
CIFS: Indicate reconnect with ECONNABORTED error code
[thirdparty/kernel/linux.git] / fs / cifs / smb2pdu.c
CommitLineData
ec2e4523
PS		 1/*
2 * fs/cifs/smb2pdu.c
3 *
2b80d049 4 * Copyright (C) International Business Machines Corp., 2009, 2013
ec2e4523
PS		 5 * Etersoft, 2012
6 * Author(s): Steve French (sfrench@us.ibm.com)
7 * Pavel Shilovsky (pshilovsky@samba.org) 2012
8 *
9 * Contains the routines for constructing the SMB2 PDUs themselves
10 *
11 * This library is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU Lesser General Public License as published
13 * by the Free Software Foundation; either version 2.1 of the License, or
14 * (at your option) any later version.
15 *
16 * This library is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
19 * the GNU Lesser General Public License for more details.
20 *
21 * You should have received a copy of the GNU Lesser General Public License
22 * along with this library; if not, write to the Free Software
23 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 */
25
26 /* SMB2 PDU handling routines here - except for leftovers (eg session setup) */
27 /* Note that there are handle based routines which must be */
28 /* treated slightly differently for reconnection purposes since we never */
29 /* want to reuse a stale file handle and only the caller knows the file info */
30
31#include <linux/fs.h>
32#include <linux/kernel.h>
33#include <linux/vfs.h>
09a4707e 34#include <linux/task_io_accounting_ops.h>
ec2e4523 35#include <linux/uaccess.h>
33319141 36#include <linux/pagemap.h>
ec2e4523
PS		 37#include <linux/xattr.h>
38#include "smb2pdu.h"
39#include "cifsglob.h"
40#include "cifsacl.h"
41#include "cifsproto.h"
42#include "smb2proto.h"
43#include "cifs_unicode.h"
44#include "cifs_debug.h"
45#include "ntlmssp.h"
46#include "smb2status.h"
09a4707e 47#include "smb2glob.h"
d324f08d 48#include "cifspdu.h"
ec2e4523
PS		 49
50/*
51 * The following table defines the expected "StructureSize" of SMB2 requests
52 * in order by SMB2 command. This is similar to "wct" in SMB/CIFS requests.
53 *
54 * Note that commands are defined in smb2pdu.h in le16 but the array below is
55 * indexed by command in host byte order.
56 */
57static const int smb2_req_struct_sizes[NUMBER_OF_SMB2_COMMANDS] = {
58 /* SMB2_NEGOTIATE */ 36,
59 /* SMB2_SESSION_SETUP */ 25,
60 /* SMB2_LOGOFF */ 4,
61 /* SMB2_TREE_CONNECT */ 9,
62 /* SMB2_TREE_DISCONNECT */ 4,
63 /* SMB2_CREATE */ 57,
64 /* SMB2_CLOSE */ 24,
65 /* SMB2_FLUSH */ 24,
66 /* SMB2_READ */ 49,
67 /* SMB2_WRITE */ 49,
68 /* SMB2_LOCK */ 48,
69 /* SMB2_IOCTL */ 57,
70 /* SMB2_CANCEL */ 4,
71 /* SMB2_ECHO */ 4,
72 /* SMB2_QUERY_DIRECTORY */ 33,
73 /* SMB2_CHANGE_NOTIFY */ 32,
74 /* SMB2_QUERY_INFO */ 41,
75 /* SMB2_SET_INFO */ 33,
76 /* SMB2_OPLOCK_BREAK */ 24 /* BB this is 36 for LEASE_BREAK variant */
77};
78
79
80static void
81smb2_hdr_assemble(struct smb2_hdr *hdr, __le16 smb2_cmd /* command */ ,
82 const struct cifs_tcon *tcon)
83{
84 struct smb2_pdu *pdu = (struct smb2_pdu *)hdr;
85 char *temp = (char *)hdr;
86 /* lookup word count ie StructureSize from table */
87 __u16 parmsize = smb2_req_struct_sizes[le16_to_cpu(smb2_cmd)];
88
89 /*
90 * smaller than SMALL_BUFFER_SIZE but bigger than fixed area of
91 * largest operations (Create)
92 */
93 memset(temp, 0, 256);
94
95 /* Note this is only network field converted to big endian */
96 hdr->smb2_buf_length = cpu_to_be32(parmsize + sizeof(struct smb2_hdr)
97 - 4 /* RFC 1001 length field itself not counted */);
98
99 hdr->ProtocolId[0] = 0xFE;
100 hdr->ProtocolId[1] = 'S';
101 hdr->ProtocolId[2] = 'M';
102 hdr->ProtocolId[3] = 'B';
103 hdr->StructureSize = cpu_to_le16(64);
104 hdr->Command = smb2_cmd;
105 hdr->CreditRequest = cpu_to_le16(2); /* BB make this dynamic */
106 hdr->ProcessId = cpu_to_le32((__u16)current->tgid);
107
108 if (!tcon)
109 goto out;
110
2b80d049
SF		 111 /* GLOBAL_CAP_LARGE_MTU will only be set if dialect > SMB2.02 */
112 /* See sections 2.2.4 and 3.2.4.1.5 of MS-SMB2 */
84ceeb96
SF		 113 if ((tcon->ses) &&
114 (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
2b80d049
SF		 115 hdr->CreditCharge = cpu_to_le16(1);
116 /* else CreditCharge MBZ */
117
ec2e4523
PS		 118 hdr->TreeId = tcon->tid;
119 /* Uid is not converted */
120 if (tcon->ses)
121 hdr->SessionId = tcon->ses->Suid;
f87ab88b
SF		 122
123 /*
124 * If we would set SMB2_FLAGS_DFS_OPERATIONS on open we also would have
125 * to pass the path on the Open SMB prefixed by \\server\share.
126 * Not sure when we would need to do the augmented path (if ever) and
127 * setting this flag breaks the SMB2 open operation since it is
128 * illegal to send an empty path name (without \\server\share prefix)
129 * when the DFS flag is set in the SMB open header. We could
130 * consider setting the flag on all operations other than open
131 * but it is safer to net set it for now.
132 */
133/* if (tcon->share_flags & SHI1005_FLAGS_DFS)
134 hdr->Flags |= SMB2_FLAGS_DFS_OPERATIONS; */
135
38d77c50 136 if (tcon->ses && tcon->ses->server && tcon->ses->server->sign)
3c1bf7e4 137 hdr->Flags |= SMB2_FLAGS_SIGNED;
ec2e4523
PS		 138out:
139 pdu->StructureSize2 = cpu_to_le16(parmsize);
140 return;
141}
142
143static int
144smb2_reconnect(__le16 smb2_command, struct cifs_tcon *tcon)
145{
146 int rc = 0;
aa24d1e9
PS		 147 struct nls_table *nls_codepage;
148 struct cifs_ses *ses;
149 struct TCP_Server_Info *server;
150
151 /*
152 * SMB2s NegProt, SessSetup, Logoff do not have tcon yet so
153 * check for tcp and smb session status done differently
154 * for those three - in the calling routine.
155 */
156 if (tcon == NULL)
157 return rc;
158
159 if (smb2_command == SMB2_TREE_CONNECT)
160 return rc;
161
162 if (tcon->tidStatus == CifsExiting) {
163 /*
164 * only tree disconnect, open, and write,
165 * (and ulogoff which does not have tcon)
166 * are allowed as we start force umount.
167 */
168 if ((smb2_command != SMB2_WRITE) &&
169 (smb2_command != SMB2_CREATE) &&
170 (smb2_command != SMB2_TREE_DISCONNECT)) {
f96637be
JP		 171 cifs_dbg(FYI, "can not send cmd %d while umounting\n",
172 smb2_command);
aa24d1e9
PS		 173 return -ENODEV;
174 }
175 }
176 if ((!tcon->ses) || (tcon->ses->status == CifsExiting) ||
177 (!tcon->ses->server))
178 return -EIO;
179
180 ses = tcon->ses;
181 server = ses->server;
182
183 /*
184 * Give demultiplex thread up to 10 seconds to reconnect, should be
185 * greater than cifs socket timeout which is 7 seconds
186 */
187 while (server->tcpStatus == CifsNeedReconnect) {
188 /*
189 * Return to caller for TREE_DISCONNECT and LOGOFF and CLOSE
190 * here since they are implicitly done when session drops.
191 */
192 switch (smb2_command) {
193 /*
194 * BB Should we keep oplock break and add flush to exceptions?
195 */
196 case SMB2_TREE_DISCONNECT:
197 case SMB2_CANCEL:
198 case SMB2_CLOSE:
199 case SMB2_OPLOCK_BREAK:
200 return -EAGAIN;
201 }
202
203 wait_event_interruptible_timeout(server->response_q,
204 (server->tcpStatus != CifsNeedReconnect), 10 * HZ);
205
206 /* are we still trying to reconnect? */
207 if (server->tcpStatus != CifsNeedReconnect)
208 break;
209
210 /*
211 * on "soft" mounts we wait once. Hard mounts keep
212 * retrying until process is killed or server comes
213 * back on-line
214 */
215 if (!tcon->retry) {
f96637be 216 cifs_dbg(FYI, "gave up waiting on reconnect in smb_init\n");
aa24d1e9
PS		 217 return -EHOSTDOWN;
218 }
219 }
220
221 if (!tcon->ses->need_reconnect && !tcon->need_reconnect)
222 return rc;
223
224 nls_codepage = load_nls_default();
225
226 /*
227 * need to prevent multiple threads trying to simultaneously reconnect
228 * the same SMB session
229 */
230 mutex_lock(&tcon->ses->session_mutex);
231 rc = cifs_negotiate_protocol(0, tcon->ses);
232 if (!rc && tcon->ses->need_reconnect)
233 rc = cifs_setup_session(0, tcon->ses, nls_codepage);
234
235 if (rc || !tcon->need_reconnect) {
236 mutex_unlock(&tcon->ses->session_mutex);
237 goto out;
238 }
239
240 cifs_mark_open_files_invalid(tcon);
241 rc = SMB2_tcon(0, tcon->ses, tcon->treeName, tcon, nls_codepage);
242 mutex_unlock(&tcon->ses->session_mutex);
f96637be 243 cifs_dbg(FYI, "reconnect tcon rc = %d\n", rc);
aa24d1e9
PS		 244 if (rc)
245 goto out;
246 atomic_inc(&tconInfoReconnectCount);
aa24d1e9
PS		 247out:
248 /*
249 * Check if handle based operation so we know whether we can continue
250 * or not without returning to caller to reset file handle.
251 */
252 /*
253 * BB Is flush done by server on drop of tcp session? Should we special
254 * case it and skip above?
255 */
256 switch (smb2_command) {
257 case SMB2_FLUSH:
258 case SMB2_READ:
259 case SMB2_WRITE:
260 case SMB2_LOCK:
261 case SMB2_IOCTL:
262 case SMB2_QUERY_DIRECTORY:
263 case SMB2_CHANGE_NOTIFY:
264 case SMB2_QUERY_INFO:
265 case SMB2_SET_INFO:
266 return -EAGAIN;
267 }
268 unload_nls(nls_codepage);
ec2e4523
PS		 269 return rc;
270}
271
272/*
273 * Allocate and return pointer to an SMB request hdr, and set basic
274 * SMB information in the SMB header. If the return code is zero, this
275 * function must have filled in request_buf pointer.
276 */
277static int
278small_smb2_init(__le16 smb2_command, struct cifs_tcon *tcon,
279 void **request_buf)
280{
281 int rc = 0;
282
283 rc = smb2_reconnect(smb2_command, tcon);
284 if (rc)
285 return rc;
286
287 /* BB eventually switch this to SMB2 specific small buf size */
288 *request_buf = cifs_small_buf_get();
289 if (*request_buf == NULL) {
290 /* BB should we add a retry in here if not a writepage? */
291 return -ENOMEM;
292 }
293
294 smb2_hdr_assemble((struct smb2_hdr *) *request_buf, smb2_command, tcon);
295
296 if (tcon != NULL) {
297#ifdef CONFIG_CIFS_STATS2
ec2e4523
PS		 298 uint16_t com_code = le16_to_cpu(smb2_command);
299 cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_sent[com_code]);
ec2e4523
PS		 300#endif
301 cifs_stats_inc(&tcon->num_smbs_sent);
302 }
303
304 return rc;
305}
306
ec2e4523
PS		 307/*
308 *
309 * SMB2 Worker functions follow:
310 *
311 * The general structure of the worker functions is:
312 * 1) Call smb2_init (assembles SMB2 header)
313 * 2) Initialize SMB2 command specific fields in fixed length area of SMB
314 * 3) Call smb_sendrcv2 (sends request on socket and waits for response)
315 * 4) Decode SMB2 command specific fields in the fixed length area
316 * 5) Decode variable length data area (if any for this SMB2 command type)
317 * 6) Call free smb buffer
318 * 7) return
319 *
320 */
321
322int
323SMB2_negotiate(const unsigned int xid, struct cifs_ses *ses)
324{
325 struct smb2_negotiate_req *req;
326 struct smb2_negotiate_rsp *rsp;
327 struct kvec iov[1];
328 int rc = 0;
329 int resp_buftype;
3534b850 330 struct TCP_Server_Info *server = ses->server;
ec2e4523
PS		 331 int blob_offset, blob_length;
332 char *security_blob;
333 int flags = CIFS_NEG_OP;
334
f96637be 335 cifs_dbg(FYI, "Negotiate protocol\n");
ec2e4523 336
3534b850
JL		 337 if (!server) {
338 WARN(1, "%s: server is NULL!\n", __func__);
339 return -EIO;
ec2e4523
PS		 340 }
341
342 rc = small_smb2_init(SMB2_NEGOTIATE, NULL, (void **) &req);
343 if (rc)
344 return rc;
345
ec2e4523
PS		 346 req->hdr.SessionId = 0;
347
e4aa25e7 348 req->Dialects[0] = cpu_to_le16(ses->server->vals->protocol_id);
ec2e4523 349
e4aa25e7
SF		 350 req->DialectCount = cpu_to_le16(1); /* One vers= at a time for now */
351 inc_rfc1001_len(req, 2);
ec2e4523
PS		 352
353 /* only one of SMB2 signing flags may be set in SMB2 request */
38d77c50 354 if (ses->sign)
9cd2e62c 355 req->SecurityMode = cpu_to_le16(SMB2_NEGOTIATE_SIGNING_REQUIRED);
38d77c50 356 else if (global_secflags & CIFSSEC_MAY_SIGN)
9cd2e62c 357 req->SecurityMode = cpu_to_le16(SMB2_NEGOTIATE_SIGNING_ENABLED);
38d77c50
JL		 358 else
359 req->SecurityMode = 0;
ec2e4523 360
e4aa25e7 361 req->Capabilities = cpu_to_le32(ses->server->vals->req_capabilities);
ec2e4523 362
3c5f9be1
SF		 363 /* ClientGUID must be zero for SMB2.02 dialect */
364 if (ses->server->vals->protocol_id == SMB20_PROT_ID)
365 memset(req->ClientGUID, 0, SMB2_CLIENT_GUID_SIZE);
366 else
367 memcpy(req->ClientGUID, server->client_guid,
368 SMB2_CLIENT_GUID_SIZE);
b8c32dbb 369
ec2e4523
PS		 370 iov[0].iov_base = (char *)req;
371 /* 4 for rfc1002 length field */
372 iov[0].iov_len = get_rfc1002_length(req) + 4;
373
374 rc = SendReceive2(xid, ses, iov, 1, &resp_buftype, flags);
375
376 rsp = (struct smb2_negotiate_rsp *)iov[0].iov_base;
377 /*
378 * No tcon so can't do
379 * cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_fail[SMB2...]);
380 */
381 if (rc != 0)
382 goto neg_exit;
383
f96637be 384 cifs_dbg(FYI, "mode 0x%x\n", rsp->SecurityMode);
ec2e4523 385
e4aa25e7
SF		 386 /* BB we may eventually want to match the negotiated vs. requested
387 dialect, even though we are only requesting one at a time */
388 if (rsp->DialectRevision == cpu_to_le16(SMB20_PROT_ID))
f96637be 389 cifs_dbg(FYI, "negotiated smb2.0 dialect\n");
e4aa25e7 390 else if (rsp->DialectRevision == cpu_to_le16(SMB21_PROT_ID))
f96637be 391 cifs_dbg(FYI, "negotiated smb2.1 dialect\n");
e4aa25e7 392 else if (rsp->DialectRevision == cpu_to_le16(SMB30_PROT_ID))
f96637be 393 cifs_dbg(FYI, "negotiated smb3.0 dialect\n");
20b6d8b4
SF		 394 else if (rsp->DialectRevision == cpu_to_le16(SMB302_PROT_ID))
395 cifs_dbg(FYI, "negotiated smb3.02 dialect\n");
ec2e4523 396 else {
f96637be
JP		 397 cifs_dbg(VFS, "Illegal dialect returned by server %d\n",
398 le16_to_cpu(rsp->DialectRevision));
ec2e4523
PS		 399 rc = -EIO;
400 goto neg_exit;
401 }
402 server->dialect = le16_to_cpu(rsp->DialectRevision);
403
e598d1d8
JL		 404 /* SMB2 only has an extended negflavor */
405 server->negflavor = CIFS_NEGFLAVOR_EXTENDED;
2365c4ea
PS		 406 /* set it to the maximum buffer size value we can send with 1 credit */
407 server->maxBuf = min_t(unsigned int, le32_to_cpu(rsp->MaxTransactSize),
408 SMB2_MAX_BUFFER_SIZE);
ec2e4523
PS		 409 server->max_read = le32_to_cpu(rsp->MaxReadSize);
410 server->max_write = le32_to_cpu(rsp->MaxWriteSize);
411 /* BB Do we need to validate the SecurityMode? */
412 server->sec_mode = le16_to_cpu(rsp->SecurityMode);
413 server->capabilities = le32_to_cpu(rsp->Capabilities);
29e20f9c
PS		 414 /* Internal types */
415 server->capabilities |= SMB2_NT_FIND | SMB2_LARGE_FILES;
ec2e4523
PS		 416
417 security_blob = smb2_get_data_area_len(&blob_offset, &blob_length,
418 &rsp->hdr);
5d875cc9
SF		 419 /*
420 * See MS-SMB2 section 2.2.4: if no blob, client picks default which
421 * for us will be
422 * ses->sectype = RawNTLMSSP;
423 * but for time being this is our only auth choice so doesn't matter.
424 * We just found a server which sets blob length to zero expecting raw.
425 */
426 if (blob_length == 0)
427 cifs_dbg(FYI, "missing security blob on negprot\n");
3c1bf7e4 428
38d77c50 429 rc = cifs_enable_signing(server, ses->sign);
ec2e4523 430#ifdef CONFIG_SMB2_ASN1 /* BB REMOVEME when updated asn1.c ready */
9ddec561
JL		 431 if (rc)
432 goto neg_exit;
5d875cc9
SF		 433 if (blob_length)
434 rc = decode_neg_token_init(security_blob, blob_length,
ec2e4523
PS		 435 &server->sec_type);
436 if (rc == 1)
437 rc = 0;
438 else if (rc == 0) {
439 rc = -EIO;
440 goto neg_exit;
441 }
442#endif
443
444neg_exit:
445 free_rsp_buf(resp_buftype, rsp);
446 return rc;
447}
5478f9ba 448
ff1c038a
SF		 449int smb3_validate_negotiate(const unsigned int xid, struct cifs_tcon *tcon)
450{
451 int rc = 0;
452 struct validate_negotiate_info_req vneg_inbuf;
453 struct validate_negotiate_info_rsp *pneg_rsp;
454 u32 rsplen;
455
456 cifs_dbg(FYI, "validate negotiate\n");
457
458 /*
459 * validation ioctl must be signed, so no point sending this if we
460 * can not sign it. We could eventually change this to selectively
461 * sign just this, the first and only signed request on a connection.
462 * This is good enough for now since a user who wants better security
463 * would also enable signing on the mount. Having validation of
464 * negotiate info for signed connections helps reduce attack vectors
465 */
466 if (tcon->ses->server->sign == false)
467 return 0; /* validation requires signing */
468
469 vneg_inbuf.Capabilities =
470 cpu_to_le32(tcon->ses->server->vals->req_capabilities);
39552ea8
SP		 471 memcpy(vneg_inbuf.Guid, tcon->ses->server->client_guid,
472 SMB2_CLIENT_GUID_SIZE);
ff1c038a
SF		 473
474 if (tcon->ses->sign)
475 vneg_inbuf.SecurityMode =
476 cpu_to_le16(SMB2_NEGOTIATE_SIGNING_REQUIRED);
477 else if (global_secflags & CIFSSEC_MAY_SIGN)
478 vneg_inbuf.SecurityMode =
479 cpu_to_le16(SMB2_NEGOTIATE_SIGNING_ENABLED);
480 else
481 vneg_inbuf.SecurityMode = 0;
482
483 vneg_inbuf.DialectCount = cpu_to_le16(1);
484 vneg_inbuf.Dialects[0] =
485 cpu_to_le16(tcon->ses->server->vals->protocol_id);
486
487 rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
488 FSCTL_VALIDATE_NEGOTIATE_INFO, true /* is_fsctl */,
489 (char *)&vneg_inbuf, sizeof(struct validate_negotiate_info_req),
490 (char **)&pneg_rsp, &rsplen);
491
492 if (rc != 0) {
493 cifs_dbg(VFS, "validate protocol negotiate failed: %d\n", rc);
494 return -EIO;
495 }
496
497 if (rsplen != sizeof(struct validate_negotiate_info_rsp)) {
498 cifs_dbg(VFS, "invalid size of protocol negotiate response\n");
499 return -EIO;
500 }
501
502 /* check validate negotiate info response matches what we got earlier */
503 if (pneg_rsp->Dialect !=
504 cpu_to_le16(tcon->ses->server->vals->protocol_id))
505 goto vneg_out;
506
507 if (pneg_rsp->SecurityMode != cpu_to_le16(tcon->ses->server->sec_mode))
508 goto vneg_out;
509
510 /* do not validate server guid because not saved at negprot time yet */
511
512 if ((le32_to_cpu(pneg_rsp->Capabilities) | SMB2_NT_FIND |
513 SMB2_LARGE_FILES) != tcon->ses->server->capabilities)
514 goto vneg_out;
515
516 /* validate negotiate successful */
517 cifs_dbg(FYI, "validate negotiate info successful\n");
518 return 0;
519
520vneg_out:
521 cifs_dbg(VFS, "protocol revalidation - security settings mismatch\n");
522 return -EIO;
523}
524
5478f9ba
PS		 525int
526SMB2_sess_setup(const unsigned int xid, struct cifs_ses *ses,
527 const struct nls_table *nls_cp)
528{
529 struct smb2_sess_setup_req *req;
530 struct smb2_sess_setup_rsp *rsp = NULL;
531 struct kvec iov[2];
532 int rc = 0;
533 int resp_buftype;
534 __le32 phase = NtLmNegotiate; /* NTLMSSP, if needed, is multistage */
3534b850 535 struct TCP_Server_Info *server = ses->server;
5478f9ba
PS		 536 u16 blob_length = 0;
537 char *security_blob;
538 char *ntlmssp_blob = NULL;
539 bool use_spnego = false; /* else use raw ntlmssp */
540
f96637be 541 cifs_dbg(FYI, "Session Setup\n");
5478f9ba 542
3534b850
JL		 543 if (!server) {
544 WARN(1, "%s: server is NULL!\n", __func__);
545 return -EIO;
5478f9ba
PS		 546 }
547
d4e63bd6
SP		 548 /*
549 * If we are here due to reconnect, free per-smb session key
550 * in case signing was required.
551 */
552 kfree(ses->auth_key.response);
553 ses->auth_key.response = NULL;
554
5478f9ba
PS		 555 /*
556 * If memory allocation is successful, caller of this function
557 * frees it.
558 */
559 ses->ntlmssp = kmalloc(sizeof(struct ntlmssp_auth), GFP_KERNEL);
560 if (!ses->ntlmssp)
561 return -ENOMEM;
5c234aa5 562 ses->ntlmssp->sesskey_per_smbsess = true;
5478f9ba 563
3f618223
JL		 564 /* FIXME: allow for other auth types besides NTLMSSP (e.g. krb5) */
565 ses->sectype = RawNTLMSSP;
5478f9ba
PS		 566
567ssetup_ntlmssp_authenticate:
568 if (phase == NtLmChallenge)
569 phase = NtLmAuthenticate; /* if ntlmssp, now final phase */
570
571 rc = small_smb2_init(SMB2_SESSION_SETUP, NULL, (void **) &req);
572 if (rc)
573 return rc;
574
5478f9ba
PS		 575 req->hdr.SessionId = 0; /* First session, not a reauthenticate */
576 req->VcNumber = 0; /* MBZ */
577 /* to enable echos and oplocks */
578 req->hdr.CreditRequest = cpu_to_le16(3);
579
580 /* only one of SMB2 signing flags may be set in SMB2 request */
38d77c50
JL		 581 if (server->sign)
582 req->SecurityMode = SMB2_NEGOTIATE_SIGNING_REQUIRED;
583 else if (global_secflags & CIFSSEC_MAY_SIGN) /* one flag unlike MUST_ */
584 req->SecurityMode = SMB2_NEGOTIATE_SIGNING_ENABLED;
585 else
586 req->SecurityMode = 0;
587
5478f9ba
PS		 588 req->Capabilities = 0;
589 req->Channel = 0; /* MBZ */
590
591 iov[0].iov_base = (char *)req;
592 /* 4 for rfc1002 length field and 1 for pad */
593 iov[0].iov_len = get_rfc1002_length(req) + 4 - 1;
594 if (phase == NtLmNegotiate) {
595 ntlmssp_blob = kmalloc(sizeof(struct _NEGOTIATE_MESSAGE),
596 GFP_KERNEL);
597 if (ntlmssp_blob == NULL) {
598 rc = -ENOMEM;
599 goto ssetup_exit;
600 }
601 build_ntlmssp_negotiate_blob(ntlmssp_blob, ses);
602 if (use_spnego) {
603 /* blob_length = build_spnego_ntlmssp_blob(
604 &security_blob,
605 sizeof(struct _NEGOTIATE_MESSAGE),
606 ntlmssp_blob); */
607 /* BB eventually need to add this */
f96637be 608 cifs_dbg(VFS, "spnego not supported for SMB2 yet\n");
5478f9ba
PS		 609 rc = -EOPNOTSUPP;
610 kfree(ntlmssp_blob);
611 goto ssetup_exit;
612 } else {
613 blob_length = sizeof(struct _NEGOTIATE_MESSAGE);
614 /* with raw NTLMSSP we don't encapsulate in SPNEGO */
615 security_blob = ntlmssp_blob;
616 }
617 } else if (phase == NtLmAuthenticate) {
618 req->hdr.SessionId = ses->Suid;
619 ntlmssp_blob = kzalloc(sizeof(struct _NEGOTIATE_MESSAGE) + 500,
620 GFP_KERNEL);
621 if (ntlmssp_blob == NULL) {
5478f9ba
PS		 622 rc = -ENOMEM;
623 goto ssetup_exit;
624 }
625 rc = build_ntlmssp_auth_blob(ntlmssp_blob, &blob_length, ses,
626 nls_cp);
627 if (rc) {
f96637be
JP		 628 cifs_dbg(FYI, "build_ntlmssp_auth_blob failed %d\n",
629 rc);
5478f9ba
PS		 630 goto ssetup_exit; /* BB double check error handling */
631 }
632 if (use_spnego) {
633 /* blob_length = build_spnego_ntlmssp_blob(
634 &security_blob,
635 blob_length,
636 ntlmssp_blob); */
f96637be 637 cifs_dbg(VFS, "spnego not supported for SMB2 yet\n");
5478f9ba
PS		 638 rc = -EOPNOTSUPP;
639 kfree(ntlmssp_blob);
640 goto ssetup_exit;
641 } else {
642 security_blob = ntlmssp_blob;
643 }
644 } else {
f96637be 645 cifs_dbg(VFS, "illegal ntlmssp phase\n");
5478f9ba
PS		 646 rc = -EIO;
647 goto ssetup_exit;
648 }
649
650 /* Testing shows that buffer offset must be at location of Buffer[0] */
651 req->SecurityBufferOffset =
652 cpu_to_le16(sizeof(struct smb2_sess_setup_req) -
653 1 /* pad */ - 4 /* rfc1001 len */);
654 req->SecurityBufferLength = cpu_to_le16(blob_length);
655 iov[1].iov_base = security_blob;
656 iov[1].iov_len = blob_length;
657
658 inc_rfc1001_len(req, blob_length - 1 /* pad */);
659
660 /* BB add code to build os and lm fields */
661
6d8b59d7
SF		 662 rc = SendReceive2(xid, ses, iov, 2, &resp_buftype,
663 CIFS_LOG_ERROR | CIFS_NEG_OP);
5478f9ba
PS		 664
665 kfree(security_blob);
666 rsp = (struct smb2_sess_setup_rsp *)iov[0].iov_base;
4ca3a99c
PS		 667 if (resp_buftype != CIFS_NO_BUFFER &&
668 rsp->hdr.Status == STATUS_MORE_PROCESSING_REQUIRED) {
5478f9ba 669 if (phase != NtLmNegotiate) {
f96637be 670 cifs_dbg(VFS, "Unexpected more processing error\n");
5478f9ba
PS		 671 goto ssetup_exit;
672 }
673 if (offsetof(struct smb2_sess_setup_rsp, Buffer) - 4 !=
4ca3a99c 674 le16_to_cpu(rsp->SecurityBufferOffset)) {
f96637be
JP		 675 cifs_dbg(VFS, "Invalid security buffer offset %d\n",
676 le16_to_cpu(rsp->SecurityBufferOffset));
5478f9ba
PS		 677 rc = -EIO;
678 goto ssetup_exit;
679 }
680
681 /* NTLMSSP Negotiate sent now processing challenge (response) */
682 phase = NtLmChallenge; /* process ntlmssp challenge */
683 rc = 0; /* MORE_PROCESSING is not an error here but expected */
684 ses->Suid = rsp->hdr.SessionId;
685 rc = decode_ntlmssp_challenge(rsp->Buffer,
686 le16_to_cpu(rsp->SecurityBufferLength), ses);
687 }
688
689 /*
690 * BB eventually add code for SPNEGO decoding of NtlmChallenge blob,
691 * but at least the raw NTLMSSP case works.
692 */
693 /*
694 * No tcon so can't do
695 * cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_fail[SMB2...]);
696 */
697 if (rc != 0)
698 goto ssetup_exit;
699
5478f9ba 700 ses->session_flags = le16_to_cpu(rsp->SessionFlags);
0cbaa53c
SF		 701 if (ses->session_flags & SMB2_SESSION_FLAG_ENCRYPT_DATA)
702 cifs_dbg(VFS, "SMB3 encryption not supported yet\n");
5478f9ba
PS		 703ssetup_exit:
704 free_rsp_buf(resp_buftype, rsp);
705
706 /* if ntlmssp, and negotiate succeeded, proceed to authenticate phase */
707 if ((phase == NtLmChallenge) && (rc == 0))
708 goto ssetup_ntlmssp_authenticate;
d4e63bd6
SP		 709
710 if (!rc) {
711 mutex_lock(&server->srv_mutex);
32811d24
SP		 712 if (server->sign && server->ops->generate_signingkey) {
713 rc = server->ops->generate_signingkey(ses);
714 kfree(ses->auth_key.response);
715 ses->auth_key.response = NULL;
716 if (rc) {
717 cifs_dbg(FYI,
718 "SMB3 session key generation failed\n");
719 mutex_unlock(&server->srv_mutex);
720 goto keygen_exit;
721 }
722 }
d4e63bd6
SP		 723 if (!server->session_estab) {
724 server->sequence_number = 0x2;
725 server->session_estab = true;
d4e63bd6
SP		 726 }
727 mutex_unlock(&server->srv_mutex);
728
729 cifs_dbg(FYI, "SMB2/3 session established successfully\n");
730 spin_lock(&GlobalMid_Lock);
731 ses->status = CifsGood;
732 ses->need_reconnect = false;
733 spin_unlock(&GlobalMid_Lock);
734 }
735
32811d24 736keygen_exit:
d4e63bd6
SP		 737 if (!server->sign) {
738 kfree(ses->auth_key.response);
739 ses->auth_key.response = NULL;
740 }
741 kfree(ses->ntlmssp);
742
5478f9ba
PS		 743 return rc;
744}
745
746int
747SMB2_logoff(const unsigned int xid, struct cifs_ses *ses)
748{
749 struct smb2_logoff_req *req; /* response is also trivial struct */
750 int rc = 0;
751 struct TCP_Server_Info *server;
752
f96637be 753 cifs_dbg(FYI, "disconnect session %p\n", ses);
5478f9ba
PS		 754
755 if (ses && (ses->server))
756 server = ses->server;
757 else
758 return -EIO;
759
eb4c7df6
SP		 760 /* no need to send SMB logoff if uid already closed due to reconnect */
761 if (ses->need_reconnect)
762 goto smb2_session_already_dead;
763
5478f9ba
PS		 764 rc = small_smb2_init(SMB2_LOGOFF, NULL, (void **) &req);
765 if (rc)
766 return rc;
767
768 /* since no tcon, smb2_init can not do this, so do here */
769 req->hdr.SessionId = ses->Suid;
38d77c50 770 if (server->sign)
3c1bf7e4 771 req->hdr.Flags |= SMB2_FLAGS_SIGNED;
5478f9ba
PS		 772
773 rc = SendReceiveNoRsp(xid, ses, (char *) &req->hdr, 0);
774 /*
775 * No tcon so can't do
776 * cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_fail[SMB2...]);
777 */
eb4c7df6
SP		 778
779smb2_session_already_dead:
5478f9ba
PS		 780 return rc;
781}
faaf946a
PS		 782
783static inline void cifs_stats_fail_inc(struct cifs_tcon *tcon, uint16_t code)
784{
d60622eb 785 cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_failed[code]);
faaf946a
PS		 786}
787
788#define MAX_SHARENAME_LENGTH (255 /* server */ + 80 /* share */ + 1 /* NULL */)
789
de9f68df
SF		 790/* These are similar values to what Windows uses */
791static inline void init_copy_chunk_defaults(struct cifs_tcon *tcon)
792{
793 tcon->max_chunks = 256;
794 tcon->max_bytes_chunk = 1048576;
795 tcon->max_bytes_copy = 16777216;
796}
797
faaf946a
PS		 798int
799SMB2_tcon(const unsigned int xid, struct cifs_ses *ses, const char *tree,
800 struct cifs_tcon *tcon, const struct nls_table *cp)
801{
802 struct smb2_tree_connect_req *req;
803 struct smb2_tree_connect_rsp *rsp = NULL;
804 struct kvec iov[2];
805 int rc = 0;
806 int resp_buftype;
807 int unc_path_len;
808 struct TCP_Server_Info *server;
809 __le16 *unc_path = NULL;
810
f96637be 811 cifs_dbg(FYI, "TCON\n");
faaf946a
PS		 812
813 if ((ses->server) && tree)
814 server = ses->server;
815 else
816 return -EIO;
817
818 if (tcon && tcon->bad_network_name)
819 return -ENOENT;
820
821 unc_path = kmalloc(MAX_SHARENAME_LENGTH * 2, GFP_KERNEL);
822 if (unc_path == NULL)
823 return -ENOMEM;
824
825 unc_path_len = cifs_strtoUTF16(unc_path, tree, strlen(tree), cp) + 1;
826 unc_path_len *= 2;
827 if (unc_path_len < 2) {
828 kfree(unc_path);
829 return -EINVAL;
830 }
831
832 rc = small_smb2_init(SMB2_TREE_CONNECT, tcon, (void **) &req);
833 if (rc) {
834 kfree(unc_path);
835 return rc;
836 }
837
838 if (tcon == NULL) {
839 /* since no tcon, smb2_init can not do this, so do here */
840 req->hdr.SessionId = ses->Suid;
841 /* if (ses->server->sec_mode & SECMODE_SIGN_REQUIRED)
842 req->hdr.Flags |= SMB2_FLAGS_SIGNED; */
843 }
844
845 iov[0].iov_base = (char *)req;
846 /* 4 for rfc1002 length field and 1 for pad */
847 iov[0].iov_len = get_rfc1002_length(req) + 4 - 1;
848
849 /* Testing shows that buffer offset must be at location of Buffer[0] */
850 req->PathOffset = cpu_to_le16(sizeof(struct smb2_tree_connect_req)
851 - 1 /* pad */ - 4 /* do not count rfc1001 len field */);
852 req->PathLength = cpu_to_le16(unc_path_len - 2);
853 iov[1].iov_base = unc_path;
854 iov[1].iov_len = unc_path_len;
855
856 inc_rfc1001_len(req, unc_path_len - 1 /* pad */);
857
858 rc = SendReceive2(xid, ses, iov, 2, &resp_buftype, 0);
859 rsp = (struct smb2_tree_connect_rsp *)iov[0].iov_base;
860
861 if (rc != 0) {
862 if (tcon) {
863 cifs_stats_fail_inc(tcon, SMB2_TREE_CONNECT_HE);
864 tcon->need_reconnect = true;
865 }
866 goto tcon_error_exit;
867 }
868
faaf946a
PS		 869 if (tcon == NULL) {
870 ses->ipc_tid = rsp->hdr.TreeId;
871 goto tcon_exit;
872 }
873
874 if (rsp->ShareType & SMB2_SHARE_TYPE_DISK)
f96637be 875 cifs_dbg(FYI, "connection to disk share\n");
faaf946a
PS		 876 else if (rsp->ShareType & SMB2_SHARE_TYPE_PIPE) {
877 tcon->ipc = true;
f96637be 878 cifs_dbg(FYI, "connection to pipe share\n");
faaf946a
PS		 879 } else if (rsp->ShareType & SMB2_SHARE_TYPE_PRINT) {
880 tcon->print = true;
f96637be 881 cifs_dbg(FYI, "connection to printer\n");
faaf946a 882 } else {
f96637be 883 cifs_dbg(VFS, "unknown share type %d\n", rsp->ShareType);
faaf946a
PS		 884 rc = -EOPNOTSUPP;
885 goto tcon_error_exit;
886 }
887
888 tcon->share_flags = le32_to_cpu(rsp->ShareFlags);
769ee6a4 889 tcon->capabilities = rsp->Capabilities; /* we keep caps little endian */
faaf946a
PS		 890 tcon->maximal_access = le32_to_cpu(rsp->MaximalAccess);
891 tcon->tidStatus = CifsGood;
892 tcon->need_reconnect = false;
893 tcon->tid = rsp->hdr.TreeId;
46b51d08 894 strlcpy(tcon->treeName, tree, sizeof(tcon->treeName));
faaf946a
PS		 895
896 if ((rsp->Capabilities & SMB2_SHARE_CAP_DFS) &&
897 ((tcon->share_flags & SHI1005_FLAGS_DFS) == 0))
f96637be 898 cifs_dbg(VFS, "DFS capability contradicts DFS flag\n");
de9f68df 899 init_copy_chunk_defaults(tcon);
ff1c038a
SF		 900 if (tcon->ses->server->ops->validate_negotiate)
901 rc = tcon->ses->server->ops->validate_negotiate(xid, tcon);
faaf946a
PS		 902tcon_exit:
903 free_rsp_buf(resp_buftype, rsp);
904 kfree(unc_path);
905 return rc;
906
907tcon_error_exit:
908 if (rsp->hdr.Status == STATUS_BAD_NETWORK_NAME) {
f96637be 909 cifs_dbg(VFS, "BAD_NETWORK_NAME: %s\n", tree);
faaf946a
PS		 910 tcon->bad_network_name = true;
911 }
912 goto tcon_exit;
913}
914
915int
916SMB2_tdis(const unsigned int xid, struct cifs_tcon *tcon)
917{
918 struct smb2_tree_disconnect_req *req; /* response is trivial */
919 int rc = 0;
920 struct TCP_Server_Info *server;
921 struct cifs_ses *ses = tcon->ses;
922
f96637be 923 cifs_dbg(FYI, "Tree Disconnect\n");
faaf946a
PS		 924
925 if (ses && (ses->server))
926 server = ses->server;
927 else
928 return -EIO;
929
930 if ((tcon->need_reconnect) || (tcon->ses->need_reconnect))
931 return 0;
932
933 rc = small_smb2_init(SMB2_TREE_DISCONNECT, tcon, (void **) &req);
934 if (rc)
935 return rc;
936
937 rc = SendReceiveNoRsp(xid, ses, (char *)&req->hdr, 0);
938 if (rc)
939 cifs_stats_fail_inc(tcon, SMB2_TREE_DISCONNECT_HE);
940
941 return rc;
942}
2503a0db 943
b8c32dbb 944
63eb3def
PS		 945static struct create_durable *
946create_durable_buf(void)
947{
948 struct create_durable *buf;
949
950 buf = kzalloc(sizeof(struct create_durable), GFP_KERNEL);
951 if (!buf)
952 return NULL;
953
954 buf->ccontext.DataOffset = cpu_to_le16(offsetof
9cbc0b73 955 (struct create_durable, Data));
63eb3def
PS		 956 buf->ccontext.DataLength = cpu_to_le32(16);
957 buf->ccontext.NameOffset = cpu_to_le16(offsetof
958 (struct create_durable, Name));
959 buf->ccontext.NameLength = cpu_to_le16(4);
12197a7f 960 /* SMB2_CREATE_DURABLE_HANDLE_REQUEST is "DHnQ" */
63eb3def
PS		 961 buf->Name[0] = 'D';
962 buf->Name[1] = 'H';
963 buf->Name[2] = 'n';
964 buf->Name[3] = 'Q';
965 return buf;
966}
967
9cbc0b73
PS		 968static struct create_durable *
969create_reconnect_durable_buf(struct cifs_fid *fid)
970{
971 struct create_durable *buf;
972
973 buf = kzalloc(sizeof(struct create_durable), GFP_KERNEL);
974 if (!buf)
975 return NULL;
976
977 buf->ccontext.DataOffset = cpu_to_le16(offsetof
978 (struct create_durable, Data));
979 buf->ccontext.DataLength = cpu_to_le32(16);
980 buf->ccontext.NameOffset = cpu_to_le16(offsetof
981 (struct create_durable, Name));
982 buf->ccontext.NameLength = cpu_to_le16(4);
983 buf->Data.Fid.PersistentFileId = fid->persistent_fid;
984 buf->Data.Fid.VolatileFileId = fid->volatile_fid;
12197a7f 985 /* SMB2_CREATE_DURABLE_HANDLE_RECONNECT is "DHnC" */
9cbc0b73
PS		 986 buf->Name[0] = 'D';
987 buf->Name[1] = 'H';
988 buf->Name[2] = 'n';
989 buf->Name[3] = 'C';
990 return buf;
991}
992
b8c32dbb 993static __u8
42873b0a
PS		 994parse_lease_state(struct TCP_Server_Info *server, struct smb2_create_rsp *rsp,
995 unsigned int *epoch)
b8c32dbb
PS		 996{
997 char *data_offset;
b5c7cde3 998 struct create_context *cc;
fd554396
PS		 999 unsigned int next = 0;
1000 char *name;
b8c32dbb 1001
fd554396 1002 data_offset = (char *)rsp + 4 + le32_to_cpu(rsp->CreateContextsOffset);
b5c7cde3 1003 cc = (struct create_context *)data_offset;
b8c32dbb 1004 do {
b5c7cde3
PS		 1005 cc = (struct create_context *)((char *)cc + next);
1006 name = le16_to_cpu(cc->NameOffset) + (char *)cc;
1007 if (le16_to_cpu(cc->NameLength) != 4 ||
b8c32dbb 1008 strncmp(name, "RqLs", 4)) {
b5c7cde3 1009 next = le32_to_cpu(cc->Next);
b8c32dbb
PS		 1010 continue;
1011 }
42873b0a 1012 return server->ops->parse_lease_buf(cc, epoch);
fd554396 1013 } while (next != 0);
b8c32dbb 1014
b5c7cde3 1015 return 0;
b8c32dbb
PS		 1016}
1017
d22cbfec 1018static int
a41a28bd
PS		 1019add_lease_context(struct TCP_Server_Info *server, struct kvec *iov,
1020 unsigned int *num_iovec, __u8 *oplock)
d22cbfec
PS		 1021{
1022 struct smb2_create_req *req = iov[0].iov_base;
1023 unsigned int num = *num_iovec;
1024
a41a28bd 1025 iov[num].iov_base = server->ops->create_lease_buf(oplock+1, *oplock);
d22cbfec
PS		 1026 if (iov[num].iov_base == NULL)
1027 return -ENOMEM;
a41a28bd 1028 iov[num].iov_len = server->vals->create_lease_size;
d22cbfec
PS		 1029 req->RequestedOplockLevel = SMB2_OPLOCK_LEVEL_LEASE;
1030 if (!req->CreateContextsOffset)
1031 req->CreateContextsOffset = cpu_to_le32(
1032 sizeof(struct smb2_create_req) - 4 +
1033 iov[num - 1].iov_len);
a41a28bd
PS		 1034 le32_add_cpu(&req->CreateContextsLength,
1035 server->vals->create_lease_size);
1036 inc_rfc1001_len(&req->hdr, server->vals->create_lease_size);
d22cbfec
PS		 1037 *num_iovec = num + 1;
1038 return 0;
1039}
1040
63eb3def 1041static int
9cbc0b73
PS		 1042add_durable_context(struct kvec *iov, unsigned int *num_iovec,
1043 struct cifs_open_parms *oparms)
63eb3def
PS		 1044{
1045 struct smb2_create_req *req = iov[0].iov_base;
1046 unsigned int num = *num_iovec;
1047
9cbc0b73
PS		 1048 if (oparms->reconnect) {
1049 iov[num].iov_base = create_reconnect_durable_buf(oparms->fid);
1050 /* indicate that we don't need to relock the file */
1051 oparms->reconnect = false;
1052 } else
1053 iov[num].iov_base = create_durable_buf();
63eb3def
PS		 1054 if (iov[num].iov_base == NULL)
1055 return -ENOMEM;
1056 iov[num].iov_len = sizeof(struct create_durable);
1057 if (!req->CreateContextsOffset)
1058 req->CreateContextsOffset =
1059 cpu_to_le32(sizeof(struct smb2_create_req) - 4 +
1060 iov[1].iov_len);
31f92e9a 1061 le32_add_cpu(&req->CreateContextsLength, sizeof(struct create_durable));
63eb3def
PS		 1062 inc_rfc1001_len(&req->hdr, sizeof(struct create_durable));
1063 *num_iovec = num + 1;
1064 return 0;
1065}
1066
2503a0db 1067int
064f6047 1068SMB2_open(const unsigned int xid, struct cifs_open_parms *oparms, __le16 *path,
b42bf888
PS		 1069 __u8 *oplock, struct smb2_file_all_info *buf,
1070 struct smb2_err_rsp **err_buf)
2503a0db
PS		 1071{
1072 struct smb2_create_req *req;
1073 struct smb2_create_rsp *rsp;
1074 struct TCP_Server_Info *server;
064f6047 1075 struct cifs_tcon *tcon = oparms->tcon;
2503a0db 1076 struct cifs_ses *ses = tcon->ses;
63eb3def 1077 struct kvec iov[4];
2503a0db
PS		 1078 int resp_buftype;
1079 int uni_path_len;
b8c32dbb
PS		 1080 __le16 *copy_path = NULL;
1081 int copy_size;
2503a0db 1082 int rc = 0;
59aa3718 1083 unsigned int num_iovecs = 2;
ca81983f 1084 __u32 file_attributes = 0;
663a9621 1085 char *dhc_buf = NULL, *lc_buf = NULL;
2503a0db 1086
f96637be 1087 cifs_dbg(FYI, "create/open\n");
2503a0db
PS		 1088
1089 if (ses && (ses->server))
1090 server = ses->server;
1091 else
1092 return -EIO;
1093
1094 rc = small_smb2_init(SMB2_CREATE, tcon, (void **) &req);
1095 if (rc)
1096 return rc;
1097
064f6047 1098 if (oparms->create_options & CREATE_OPTION_READONLY)
ca81983f
PS		 1099 file_attributes |= ATTR_READONLY;
1100
2503a0db 1101 req->ImpersonationLevel = IL_IMPERSONATION;
064f6047 1102 req->DesiredAccess = cpu_to_le32(oparms->desired_access);
2503a0db
PS		 1103 /* File attributes ignored on open (used in create though) */
1104 req->FileAttributes = cpu_to_le32(file_attributes);
1105 req->ShareAccess = FILE_SHARE_ALL_LE;
064f6047
PS		 1106 req->CreateDisposition = cpu_to_le32(oparms->disposition);
1107 req->CreateOptions = cpu_to_le32(oparms->create_options & CREATE_OPTIONS_MASK);
2503a0db 1108 uni_path_len = (2 * UniStrnlen((wchar_t *)path, PATH_MAX)) + 2;
59aa3718
PS		 1109 /* do not count rfc1001 len field */
1110 req->NameOffset = cpu_to_le16(sizeof(struct smb2_create_req) - 4);
2503a0db
PS		 1111
1112 iov[0].iov_base = (char *)req;
1113 /* 4 for rfc1002 length field */
1114 iov[0].iov_len = get_rfc1002_length(req) + 4;
1115
1116 /* MUST set path len (NameLength) to 0 opening root of share */
59aa3718
PS		 1117 req->NameLength = cpu_to_le16(uni_path_len - 2);
1118 /* -1 since last byte is buf[0] which is sent below (path) */
1119 iov[0].iov_len--;
1120 if (uni_path_len % 8 != 0) {
1121 copy_size = uni_path_len / 8 * 8;
1122 if (copy_size < uni_path_len)
1123 copy_size += 8;
1124
1125 copy_path = kzalloc(copy_size, GFP_KERNEL);
1126 if (!copy_path)
1127 return -ENOMEM;
1128 memcpy((char *)copy_path, (const char *)path,
1129 uni_path_len);
1130 uni_path_len = copy_size;
1131 path = copy_path;
2503a0db
PS		 1132 }
1133
59aa3718
PS		 1134 iov[1].iov_len = uni_path_len;
1135 iov[1].iov_base = path;
1136 /* -1 since last byte is buf[0] which was counted in smb2_buf_len */
1137 inc_rfc1001_len(req, uni_path_len - 1);
1138
b8c32dbb
PS		 1139 if (!server->oplocks)
1140 *oplock = SMB2_OPLOCK_LEVEL_NONE;
1141
a41a28bd 1142 if (!(server->capabilities & SMB2_GLOBAL_CAP_LEASING) ||
b8c32dbb
PS		 1143 *oplock == SMB2_OPLOCK_LEVEL_NONE)
1144 req->RequestedOplockLevel = *oplock;
1145 else {
a41a28bd 1146 rc = add_lease_context(server, iov, &num_iovecs, oplock);
d22cbfec 1147 if (rc) {
b8c32dbb
PS		 1148 cifs_small_buf_release(req);
1149 kfree(copy_path);
d22cbfec 1150 return rc;
b8c32dbb 1151 }
663a9621 1152 lc_buf = iov[num_iovecs-1].iov_base;
b8c32dbb
PS		 1153 }
1154
63eb3def
PS		 1155 if (*oplock == SMB2_OPLOCK_LEVEL_BATCH) {
1156 /* need to set Next field of lease context if we request it */
a41a28bd 1157 if (server->capabilities & SMB2_GLOBAL_CAP_LEASING) {
63eb3def
PS		 1158 struct create_context *ccontext =
1159 (struct create_context *)iov[num_iovecs-1].iov_base;
1c46943f 1160 ccontext->Next =
a41a28bd 1161 cpu_to_le32(server->vals->create_lease_size);
63eb3def 1162 }
9cbc0b73 1163 rc = add_durable_context(iov, &num_iovecs, oparms);
63eb3def
PS		 1164 if (rc) {
1165 cifs_small_buf_release(req);
1166 kfree(copy_path);
663a9621 1167 kfree(lc_buf);
63eb3def
PS		 1168 return rc;
1169 }
663a9621 1170 dhc_buf = iov[num_iovecs-1].iov_base;
63eb3def
PS		 1171 }
1172
2503a0db
PS		 1173 rc = SendReceive2(xid, ses, iov, num_iovecs, &resp_buftype, 0);
1174 rsp = (struct smb2_create_rsp *)iov[0].iov_base;
1175
1176 if (rc != 0) {
1177 cifs_stats_fail_inc(tcon, SMB2_CREATE_HE);
b42bf888
PS		 1178 if (err_buf)
1179 *err_buf = kmemdup(rsp, get_rfc1002_length(rsp) + 4,
1180 GFP_KERNEL);
2503a0db
PS		 1181 goto creat_exit;
1182 }
1183
064f6047
PS		 1184 oparms->fid->persistent_fid = rsp->PersistentFileId;
1185 oparms->fid->volatile_fid = rsp->VolatileFileId;
f0df737e
PS		 1186
1187 if (buf) {
1188 memcpy(buf, &rsp->CreationTime, 32);
1189 buf->AllocationSize = rsp->AllocationSize;
1190 buf->EndOfFile = rsp->EndofFile;
1191 buf->Attributes = rsp->FileAttributes;
1192 buf->NumberOfLinks = cpu_to_le32(1);
1193 buf->DeletePending = 0;
1194 }
2e44b288 1195
b8c32dbb 1196 if (rsp->OplockLevel == SMB2_OPLOCK_LEVEL_LEASE)
42873b0a 1197 *oplock = parse_lease_state(server, rsp, &oparms->fid->epoch);
b8c32dbb
PS		 1198 else
1199 *oplock = rsp->OplockLevel;
2503a0db 1200creat_exit:
b8c32dbb 1201 kfree(copy_path);
663a9621
PS		 1202 kfree(lc_buf);
1203 kfree(dhc_buf);
2503a0db
PS		 1204 free_rsp_buf(resp_buftype, rsp);
1205 return rc;
1206}
1207
4a72dafa
SF		 1208/*
1209 * SMB2 IOCTL is used for both IOCTLs and FSCTLs
1210 */
1211int
1212SMB2_ioctl(const unsigned int xid, struct cifs_tcon *tcon, u64 persistent_fid,
1213 u64 volatile_fid, u32 opcode, bool is_fsctl, char *in_data,
1214 u32 indatalen, char **out_data, u32 *plen /* returned data len */)
1215{
1216 struct smb2_ioctl_req *req;
1217 struct smb2_ioctl_rsp *rsp;
1218 struct TCP_Server_Info *server;
1219 struct cifs_ses *ses = tcon->ses;
1220 struct kvec iov[2];
1221 int resp_buftype;
1222 int num_iovecs;
1223 int rc = 0;
1224
1225 cifs_dbg(FYI, "SMB2 IOCTL\n");
1226
7ff8d45c 1227 *out_data = NULL;
4a72dafa
SF		 1228 /* zero out returned data len, in case of error */
1229 if (plen)
1230 *plen = 0;
1231
1232 if (ses && (ses->server))
1233 server = ses->server;
1234 else
1235 return -EIO;
1236
1237 rc = small_smb2_init(SMB2_IOCTL, tcon, (void **) &req);
1238 if (rc)
1239 return rc;
1240
1241 req->CtlCode = cpu_to_le32(opcode);
1242 req->PersistentFileId = persistent_fid;
1243 req->VolatileFileId = volatile_fid;
1244
1245 if (indatalen) {
1246 req->InputCount = cpu_to_le32(indatalen);
1247 /* do not set InputOffset if no input data */
1248 req->InputOffset =
1249 cpu_to_le32(offsetof(struct smb2_ioctl_req, Buffer) - 4);
1250 iov[1].iov_base = in_data;
1251 iov[1].iov_len = indatalen;
1252 num_iovecs = 2;
1253 } else
1254 num_iovecs = 1;
1255
1256 req->OutputOffset = 0;
1257 req->OutputCount = 0; /* MBZ */
1258
1259 /*
1260 * Could increase MaxOutputResponse, but that would require more
1261 * than one credit. Windows typically sets this smaller, but for some
1262 * ioctls it may be useful to allow server to send more. No point
1263 * limiting what the server can send as long as fits in one credit
1264 */
1265 req->MaxOutputResponse = cpu_to_le32(0xFF00); /* < 64K uses 1 credit */
1266
1267 if (is_fsctl)
1268 req->Flags = cpu_to_le32(SMB2_0_IOCTL_IS_FSCTL);
1269 else
1270 req->Flags = 0;
1271
1272 iov[0].iov_base = (char *)req;
4a72dafa 1273
7ff8d45c
SF		 1274 /*
1275 * If no input data, the size of ioctl struct in
1276 * protocol spec still includes a 1 byte data buffer,
1277 * but if input data passed to ioctl, we do not
1278 * want to double count this, so we do not send
1279 * the dummy one byte of data in iovec[0] if sending
1280 * input data (in iovec[1]). We also must add 4 bytes
1281 * in first iovec to allow for rfc1002 length field.
1282 */
1283
1284 if (indatalen) {
1285 iov[0].iov_len = get_rfc1002_length(req) + 4 - 1;
1286 inc_rfc1001_len(req, indatalen - 1);
1287 } else
1288 iov[0].iov_len = get_rfc1002_length(req) + 4;
1289
4a72dafa
SF		 1290
1291 rc = SendReceive2(xid, ses, iov, num_iovecs, &resp_buftype, 0);
1292 rsp = (struct smb2_ioctl_rsp *)iov[0].iov_base;
1293
9bf0c9cd 1294 if ((rc != 0) && (rc != -EINVAL)) {
4a72dafa
SF		 1295 if (tcon)
1296 cifs_stats_fail_inc(tcon, SMB2_IOCTL_HE);
1297 goto ioctl_exit;
9bf0c9cd
SF		 1298 } else if (rc == -EINVAL) {
1299 if ((opcode != FSCTL_SRV_COPYCHUNK_WRITE) &&
1300 (opcode != FSCTL_SRV_COPYCHUNK)) {
1301 if (tcon)
1302 cifs_stats_fail_inc(tcon, SMB2_IOCTL_HE);
1303 goto ioctl_exit;
1304 }
4a72dafa
SF		 1305 }
1306
1307 /* check if caller wants to look at return data or just return rc */
1308 if ((plen == NULL) || (out_data == NULL))
1309 goto ioctl_exit;
1310
1311 *plen = le32_to_cpu(rsp->OutputCount);
1312
1313 /* We check for obvious errors in the output buffer length and offset */
1314 if (*plen == 0)
1315 goto ioctl_exit; /* server returned no data */
1316 else if (*plen > 0xFF00) {
1317 cifs_dbg(VFS, "srv returned invalid ioctl length: %d\n", *plen);
1318 *plen = 0;
1319 rc = -EIO;
1320 goto ioctl_exit;
1321 }
1322
1323 if (get_rfc1002_length(rsp) < le32_to_cpu(rsp->OutputOffset) + *plen) {
1324 cifs_dbg(VFS, "Malformed ioctl resp: len %d offset %d\n", *plen,
1325 le32_to_cpu(rsp->OutputOffset));
1326 *plen = 0;
1327 rc = -EIO;
1328 goto ioctl_exit;
1329 }
1330
1331 *out_data = kmalloc(*plen, GFP_KERNEL);
1332 if (*out_data == NULL) {
1333 rc = -ENOMEM;
1334 goto ioctl_exit;
1335 }
1336
1337 memcpy(*out_data, rsp->hdr.ProtocolId + le32_to_cpu(rsp->OutputOffset),
1338 *plen);
1339ioctl_exit:
1340 free_rsp_buf(resp_buftype, rsp);
1341 return rc;
1342}
1343
64a5cfa6
SF		 1344/*
1345 * Individual callers to ioctl worker function follow
1346 */
1347
1348int
1349SMB2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
1350 u64 persistent_fid, u64 volatile_fid)
1351{
1352 int rc;
64a5cfa6
SF		 1353 struct compress_ioctl fsctl_input;
1354 char *ret_data = NULL;
1355
1356 fsctl_input.CompressionState =
1357 __constant_cpu_to_le16(COMPRESSION_FORMAT_DEFAULT);
1358
1359 rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
1360 FSCTL_SET_COMPRESSION, true /* is_fsctl */,
1361 (char *)&fsctl_input /* data input */,
1362 2 /* in data len */, &ret_data /* out data */, NULL);
1363
1364 cifs_dbg(FYI, "set compression rc %d\n", rc);
64a5cfa6
SF		 1365
1366 return rc;
1367}
1368
2503a0db
PS		 1369int
1370SMB2_close(const unsigned int xid, struct cifs_tcon *tcon,
1371 u64 persistent_fid, u64 volatile_fid)
1372{
1373 struct smb2_close_req *req;
1374 struct smb2_close_rsp *rsp;
1375 struct TCP_Server_Info *server;
1376 struct cifs_ses *ses = tcon->ses;
1377 struct kvec iov[1];
1378 int resp_buftype;
1379 int rc = 0;
1380
f96637be 1381 cifs_dbg(FYI, "Close\n");
2503a0db
PS		 1382
1383 if (ses && (ses->server))
1384 server = ses->server;
1385 else
1386 return -EIO;
1387
1388 rc = small_smb2_init(SMB2_CLOSE, tcon, (void **) &req);
1389 if (rc)
1390 return rc;
1391
1392 req->PersistentFileId = persistent_fid;
1393 req->VolatileFileId = volatile_fid;
1394
1395 iov[0].iov_base = (char *)req;
1396 /* 4 for rfc1002 length field */
1397 iov[0].iov_len = get_rfc1002_length(req) + 4;
1398
1399 rc = SendReceive2(xid, ses, iov, 1, &resp_buftype, 0);
1400 rsp = (struct smb2_close_rsp *)iov[0].iov_base;
1401
1402 if (rc != 0) {
1403 if (tcon)
1404 cifs_stats_fail_inc(tcon, SMB2_CLOSE_HE);
1405 goto close_exit;
1406 }
1407
2503a0db
PS		 1408 /* BB FIXME - decode close response, update inode for caching */
1409
1410close_exit:
1411 free_rsp_buf(resp_buftype, rsp);
1412 return rc;
1413}
be4cb9e3
PS		 1414
1415static int
1416validate_buf(unsigned int offset, unsigned int buffer_length,
1417 struct smb2_hdr *hdr, unsigned int min_buf_size)
1418
1419{
1420 unsigned int smb_len = be32_to_cpu(hdr->smb2_buf_length);
1421 char *end_of_smb = smb_len + 4 /* RFC1001 length field */ + (char *)hdr;
1422 char *begin_of_buf = 4 /* RFC1001 len field */ + offset + (char *)hdr;
1423 char *end_of_buf = begin_of_buf + buffer_length;
1424
1425
1426 if (buffer_length < min_buf_size) {
f96637be
JP		 1427 cifs_dbg(VFS, "buffer length %d smaller than minimum size %d\n",
1428 buffer_length, min_buf_size);
be4cb9e3
PS		 1429 return -EINVAL;
1430 }
1431
1432 /* check if beyond RFC1001 maximum length */
1433 if ((smb_len > 0x7FFFFF) || (buffer_length > 0x7FFFFF)) {
f96637be
JP		 1434 cifs_dbg(VFS, "buffer length %d or smb length %d too large\n",
1435 buffer_length, smb_len);
be4cb9e3
PS		 1436 return -EINVAL;
1437 }
1438
1439 if ((begin_of_buf > end_of_smb) || (end_of_buf > end_of_smb)) {
f96637be 1440 cifs_dbg(VFS, "illegal server response, bad offset to data\n");
be4cb9e3
PS		 1441 return -EINVAL;
1442 }
1443
1444 return 0;
1445}
1446
1447/*
1448 * If SMB buffer fields are valid, copy into temporary buffer to hold result.
1449 * Caller must free buffer.
1450 */
1451static int
1452validate_and_copy_buf(unsigned int offset, unsigned int buffer_length,
1453 struct smb2_hdr *hdr, unsigned int minbufsize,
1454 char *data)
1455
1456{
1457 char *begin_of_buf = 4 /* RFC1001 len field */ + offset + (char *)hdr;
1458 int rc;
1459
1460 if (!data)
1461 return -EINVAL;
1462
1463 rc = validate_buf(offset, buffer_length, hdr, minbufsize);
1464 if (rc)
1465 return rc;
1466
1467 memcpy(data, begin_of_buf, buffer_length);
1468
1469 return 0;
1470}
1471
f0df737e
PS		 1472static int
1473query_info(const unsigned int xid, struct cifs_tcon *tcon,
1474 u64 persistent_fid, u64 volatile_fid, u8 info_class,
1475 size_t output_len, size_t min_len, void *data)
be4cb9e3
PS		 1476{
1477 struct smb2_query_info_req *req;
1478 struct smb2_query_info_rsp *rsp = NULL;
1479 struct kvec iov[2];
1480 int rc = 0;
1481 int resp_buftype;
1482 struct TCP_Server_Info *server;
1483 struct cifs_ses *ses = tcon->ses;
1484
f96637be 1485 cifs_dbg(FYI, "Query Info\n");
be4cb9e3
PS		 1486
1487 if (ses && (ses->server))
1488 server = ses->server;
1489 else
1490 return -EIO;
1491
1492 rc = small_smb2_init(SMB2_QUERY_INFO, tcon, (void **) &req);
1493 if (rc)
1494 return rc;
1495
1496 req->InfoType = SMB2_O_INFO_FILE;
f0df737e 1497 req->FileInfoClass = info_class;
be4cb9e3
PS		 1498 req->PersistentFileId = persistent_fid;
1499 req->VolatileFileId = volatile_fid;
1500 /* 4 for rfc1002 length field and 1 for Buffer */
1501 req->InputBufferOffset =
1502 cpu_to_le16(sizeof(struct smb2_query_info_req) - 1 - 4);
f0df737e 1503 req->OutputBufferLength = cpu_to_le32(output_len);
be4cb9e3
PS		 1504
1505 iov[0].iov_base = (char *)req;
1506 /* 4 for rfc1002 length field */
1507 iov[0].iov_len = get_rfc1002_length(req) + 4;
1508
1509 rc = SendReceive2(xid, ses, iov, 1, &resp_buftype, 0);
e5d04887
PS		 1510 rsp = (struct smb2_query_info_rsp *)iov[0].iov_base;
1511
be4cb9e3
PS		 1512 if (rc) {
1513 cifs_stats_fail_inc(tcon, SMB2_QUERY_INFO_HE);
1514 goto qinf_exit;
1515 }
1516
be4cb9e3
PS		 1517 rc = validate_and_copy_buf(le16_to_cpu(rsp->OutputBufferOffset),
1518 le32_to_cpu(rsp->OutputBufferLength),
f0df737e 1519 &rsp->hdr, min_len, data);
be4cb9e3
PS		 1520
1521qinf_exit:
1522 free_rsp_buf(resp_buftype, rsp);
1523 return rc;
1524}
9094fad1 1525
f0df737e
PS		 1526int
1527SMB2_query_info(const unsigned int xid, struct cifs_tcon *tcon,
1528 u64 persistent_fid, u64 volatile_fid,
1529 struct smb2_file_all_info *data)
1530{
1531 return query_info(xid, tcon, persistent_fid, volatile_fid,
1532 FILE_ALL_INFORMATION,
1533 sizeof(struct smb2_file_all_info) + MAX_NAME * 2,
1534 sizeof(struct smb2_file_all_info), data);
1535}
1536
1537int
1538SMB2_get_srv_num(const unsigned int xid, struct cifs_tcon *tcon,
1539 u64 persistent_fid, u64 volatile_fid, __le64 *uniqueid)
1540{
1541 return query_info(xid, tcon, persistent_fid, volatile_fid,
1542 FILE_INTERNAL_INFORMATION,
1543 sizeof(struct smb2_file_internal_info),
1544 sizeof(struct smb2_file_internal_info), uniqueid);
1545}
1546
9094fad1
PS		 1547/*
1548 * This is a no-op for now. We're not really interested in the reply, but
1549 * rather in the fact that the server sent one and that server->lstrp
1550 * gets updated.
1551 *
1552 * FIXME: maybe we should consider checking that the reply matches request?
1553 */
1554static void
1555smb2_echo_callback(struct mid_q_entry *mid)
1556{
1557 struct TCP_Server_Info *server = mid->callback_data;
1558 struct smb2_echo_rsp *smb2 = (struct smb2_echo_rsp *)mid->resp_buf;
1559 unsigned int credits_received = 1;
1560
1561 if (mid->mid_state == MID_RESPONSE_RECEIVED)
1562 credits_received = le16_to_cpu(smb2->hdr.CreditRequest);
1563
1564 DeleteMidQEntry(mid);
1565 add_credits(server, credits_received, CIFS_ECHO_OP);
1566}
1567
1568int
1569SMB2_echo(struct TCP_Server_Info *server)
1570{
1571 struct smb2_echo_req *req;
1572 int rc = 0;
1573 struct kvec iov;
fec344e3
JL		 1574 struct smb_rqst rqst = { .rq_iov = &iov,
1575 .rq_nvec = 1 };
9094fad1 1576
f96637be 1577 cifs_dbg(FYI, "In echo request\n");
9094fad1
PS		 1578
1579 rc = small_smb2_init(SMB2_ECHO, NULL, (void **)&req);
1580 if (rc)
1581 return rc;
1582
1583 req->hdr.CreditRequest = cpu_to_le16(1);
1584
1585 iov.iov_base = (char *)req;
1586 /* 4 for rfc1002 length field */
1587 iov.iov_len = get_rfc1002_length(req) + 4;
1588
fec344e3 1589 rc = cifs_call_async(server, &rqst, NULL, smb2_echo_callback, server,
9094fad1
PS		 1590 CIFS_ECHO_OP);
1591 if (rc)
f96637be 1592 cifs_dbg(FYI, "Echo request failed: %d\n", rc);
9094fad1
PS		 1593
1594 cifs_small_buf_release(req);
1595 return rc;
1596}
7a5cfb19
PS		 1597
1598int
1599SMB2_flush(const unsigned int xid, struct cifs_tcon *tcon, u64 persistent_fid,
1600 u64 volatile_fid)
1601{
1602 struct smb2_flush_req *req;
1603 struct TCP_Server_Info *server;
1604 struct cifs_ses *ses = tcon->ses;
1605 struct kvec iov[1];
1606 int resp_buftype;
1607 int rc = 0;
1608
f96637be 1609 cifs_dbg(FYI, "Flush\n");
7a5cfb19
PS		 1610
1611 if (ses && (ses->server))
1612 server = ses->server;
1613 else
1614 return -EIO;
1615
1616 rc = small_smb2_init(SMB2_FLUSH, tcon, (void **) &req);
1617 if (rc)
1618 return rc;
1619
1620 req->PersistentFileId = persistent_fid;
1621 req->VolatileFileId = volatile_fid;
1622
1623 iov[0].iov_base = (char *)req;
1624 /* 4 for rfc1002 length field */
1625 iov[0].iov_len = get_rfc1002_length(req) + 4;
1626
1627 rc = SendReceive2(xid, ses, iov, 1, &resp_buftype, 0);
1628
1629 if ((rc != 0) && tcon)
1630 cifs_stats_fail_inc(tcon, SMB2_FLUSH_HE);
1631
1632 free_rsp_buf(resp_buftype, iov[0].iov_base);
1633 return rc;
1634}
09a4707e
PS		 1635
1636/*
1637 * To form a chain of read requests, any read requests after the first should
1638 * have the end_of_chain boolean set to true.
1639 */
1640static int
1641smb2_new_read_req(struct kvec *iov, struct cifs_io_parms *io_parms,
1642 unsigned int remaining_bytes, int request_type)
1643{
1644 int rc = -EACCES;
1645 struct smb2_read_req *req = NULL;
1646
1647 rc = small_smb2_init(SMB2_READ, io_parms->tcon, (void **) &req);
1648 if (rc)
1649 return rc;
1650 if (io_parms->tcon->ses->server == NULL)
1651 return -ECONNABORTED;
1652
1653 req->hdr.ProcessId = cpu_to_le32(io_parms->pid);
1654
1655 req->PersistentFileId = io_parms->persistent_fid;
1656 req->VolatileFileId = io_parms->volatile_fid;
1657 req->ReadChannelInfoOffset = 0; /* reserved */
1658 req->ReadChannelInfoLength = 0; /* reserved */
1659 req->Channel = 0; /* reserved */
1660 req->MinimumCount = 0;
1661 req->Length = cpu_to_le32(io_parms->length);
1662 req->Offset = cpu_to_le64(io_parms->offset);
1663
1664 if (request_type & CHAINED_REQUEST) {
1665 if (!(request_type & END_OF_CHAIN)) {
1666 /* 4 for rfc1002 length field */
1667 req->hdr.NextCommand =
1668 cpu_to_le32(get_rfc1002_length(req) + 4);
1669 } else /* END_OF_CHAIN */
1670 req->hdr.NextCommand = 0;
1671 if (request_type & RELATED_REQUEST) {
1672 req->hdr.Flags |= SMB2_FLAGS_RELATED_OPERATIONS;
1673 /*
1674 * Related requests use info from previous read request
1675 * in chain.
1676 */
1677 req->hdr.SessionId = 0xFFFFFFFF;
1678 req->hdr.TreeId = 0xFFFFFFFF;
1679 req->PersistentFileId = 0xFFFFFFFF;
1680 req->VolatileFileId = 0xFFFFFFFF;
1681 }
1682 }
1683 if (remaining_bytes > io_parms->length)
1684 req->RemainingBytes = cpu_to_le32(remaining_bytes);
1685 else
1686 req->RemainingBytes = 0;
1687
1688 iov[0].iov_base = (char *)req;
1689 /* 4 for rfc1002 length field */
1690 iov[0].iov_len = get_rfc1002_length(req) + 4;
1691 return rc;
1692}
1693
1694static void
1695smb2_readv_callback(struct mid_q_entry *mid)
1696{
1697 struct cifs_readdata *rdata = mid->callback_data;
1698 struct cifs_tcon *tcon = tlink_tcon(rdata->cfile->tlink);
1699 struct TCP_Server_Info *server = tcon->ses->server;
5819575e 1700 struct smb2_hdr *buf = (struct smb2_hdr *)rdata->iov.iov_base;
09a4707e 1701 unsigned int credits_received = 1;
5819575e 1702 struct smb_rqst rqst = { .rq_iov = &rdata->iov,
8321fec4
JL		 1703 .rq_nvec = 1,
1704 .rq_pages = rdata->pages,
1705 .rq_npages = rdata->nr_pages,
1706 .rq_pagesz = rdata->pagesz,
1707 .rq_tailsz = rdata->tailsz };
09a4707e 1708
f96637be
JP		 1709 cifs_dbg(FYI, "%s: mid=%llu state=%d result=%d bytes=%u\n",
1710 __func__, mid->mid, mid->mid_state, rdata->result,
1711 rdata->bytes);
09a4707e
PS		 1712
1713 switch (mid->mid_state) {
1714 case MID_RESPONSE_RECEIVED:
1715 credits_received = le16_to_cpu(buf->CreditRequest);
1716 /* result already set, check signature */
38d77c50 1717 if (server->sign) {
3c1bf7e4
PS		 1718 int rc;
1719
0b688cfc 1720 rc = smb2_verify_signature(&rqst, server);
3c1bf7e4 1721 if (rc)
f96637be
JP		 1722 cifs_dbg(VFS, "SMB signature verification returned error = %d\n",
1723 rc);
3c1bf7e4 1724 }
09a4707e
PS		 1725 /* FIXME: should this be counted toward the initiating task? */
1726 task_io_account_read(rdata->bytes);
1727 cifs_stats_bytes_read(tcon, rdata->bytes);
1728 break;
1729 case MID_REQUEST_SUBMITTED:
1730 case MID_RETRY_NEEDED:
1731 rdata->result = -EAGAIN;
1732 break;
1733 default:
1734 if (rdata->result != -ENODATA)
1735 rdata->result = -EIO;
1736 }
1737
1738 if (rdata->result)
1739 cifs_stats_fail_inc(tcon, SMB2_READ_HE);
1740
1741 queue_work(cifsiod_wq, &rdata->work);
1742 DeleteMidQEntry(mid);
1743 add_credits(server, credits_received, 0);
1744}
1745
1746/* smb2_async_readv - send an async write, and set up mid to handle result */
1747int
1748smb2_async_readv(struct cifs_readdata *rdata)
1749{
bed9da02 1750 int rc, flags = 0;
09a4707e
PS		 1751 struct smb2_hdr *buf;
1752 struct cifs_io_parms io_parms;
5819575e 1753 struct smb_rqst rqst = { .rq_iov = &rdata->iov,
fec344e3 1754 .rq_nvec = 1 };
bed9da02 1755 struct TCP_Server_Info *server;
09a4707e 1756
f96637be
JP		 1757 cifs_dbg(FYI, "%s: offset=%llu bytes=%u\n",
1758 __func__, rdata->offset, rdata->bytes);
09a4707e
PS		 1759
1760 io_parms.tcon = tlink_tcon(rdata->cfile->tlink);
1761 io_parms.offset = rdata->offset;
1762 io_parms.length = rdata->bytes;
1763 io_parms.persistent_fid = rdata->cfile->fid.persistent_fid;
1764 io_parms.volatile_fid = rdata->cfile->fid.volatile_fid;
1765 io_parms.pid = rdata->pid;
bed9da02
PS		 1766
1767 server = io_parms.tcon->ses->server;
1768
5819575e 1769 rc = smb2_new_read_req(&rdata->iov, &io_parms, 0, 0);
bed9da02
PS		 1770 if (rc) {
1771 if (rc == -EAGAIN && rdata->credits) {
1772 /* credits was reset by reconnect */
1773 rdata->credits = 0;
1774 /* reduce in_flight value since we won't send the req */
1775 spin_lock(&server->req_lock);
1776 server->in_flight--;
1777 spin_unlock(&server->req_lock);
1778 }
09a4707e 1779 return rc;
bed9da02 1780 }
09a4707e 1781
5819575e 1782 buf = (struct smb2_hdr *)rdata->iov.iov_base;
09a4707e 1783 /* 4 for rfc1002 length field */
5819575e 1784 rdata->iov.iov_len = get_rfc1002_length(rdata->iov.iov_base) + 4;
09a4707e 1785
bed9da02
PS		 1786 if (rdata->credits) {
1787 buf->CreditCharge = cpu_to_le16(DIV_ROUND_UP(rdata->bytes,
1788 SMB2_MAX_BUFFER_SIZE));
1789 spin_lock(&server->req_lock);
1790 server->credits += rdata->credits -
1791 le16_to_cpu(buf->CreditCharge);
1792 spin_unlock(&server->req_lock);
1793 wake_up(&server->request_q);
1794 flags = CIFS_HAS_CREDITS;
1795 }
1796
09a4707e 1797 kref_get(&rdata->refcount);
fec344e3 1798 rc = cifs_call_async(io_parms.tcon->ses->server, &rqst,
09a4707e 1799 cifs_readv_receive, smb2_readv_callback,
bed9da02 1800 rdata, flags);
e5d04887 1801 if (rc) {
09a4707e 1802 kref_put(&rdata->refcount, cifs_readdata_release);
e5d04887
PS		 1803 cifs_stats_fail_inc(io_parms.tcon, SMB2_READ_HE);
1804 }
09a4707e
PS		 1805
1806 cifs_small_buf_release(buf);
1807 return rc;
1808}
33319141 1809
d8e05039
PS		 1810int
1811SMB2_read(const unsigned int xid, struct cifs_io_parms *io_parms,
1812 unsigned int *nbytes, char **buf, int *buf_type)
1813{
1814 int resp_buftype, rc = -EACCES;
1815 struct smb2_read_rsp *rsp = NULL;
1816 struct kvec iov[1];
1817
1818 *nbytes = 0;
1819 rc = smb2_new_read_req(iov, io_parms, 0, 0);
1820 if (rc)
1821 return rc;
1822
1823 rc = SendReceive2(xid, io_parms->tcon->ses, iov, 1,
1824 &resp_buftype, CIFS_LOG_ERROR);
1825
1826 rsp = (struct smb2_read_rsp *)iov[0].iov_base;
1827
1828 if (rsp->hdr.Status == STATUS_END_OF_FILE) {
1829 free_rsp_buf(resp_buftype, iov[0].iov_base);
1830 return 0;
1831 }
1832
1833 if (rc) {
1834 cifs_stats_fail_inc(io_parms->tcon, SMB2_READ_HE);
f96637be 1835 cifs_dbg(VFS, "Send error in read = %d\n", rc);
d8e05039
PS		 1836 } else {
1837 *nbytes = le32_to_cpu(rsp->DataLength);
1838 if ((*nbytes > CIFS_MAX_MSGSIZE) ||
1839 (*nbytes > io_parms->length)) {
f96637be
JP		 1840 cifs_dbg(FYI, "bad length %d for count %d\n",
1841 *nbytes, io_parms->length);
d8e05039
PS		 1842 rc = -EIO;
1843 *nbytes = 0;
1844 }
1845 }
1846
1847 if (*buf) {
1848 memcpy(*buf, (char *)rsp->hdr.ProtocolId + rsp->DataOffset,
1849 *nbytes);
1850 free_rsp_buf(resp_buftype, iov[0].iov_base);
1851 } else if (resp_buftype != CIFS_NO_BUFFER) {
1852 *buf = iov[0].iov_base;
1853 if (resp_buftype == CIFS_SMALL_BUFFER)
1854 *buf_type = CIFS_SMALL_BUFFER;
1855 else if (resp_buftype == CIFS_LARGE_BUFFER)
1856 *buf_type = CIFS_LARGE_BUFFER;
1857 }
1858 return rc;
1859}
1860
33319141
PS		 1861/*
1862 * Check the mid_state and signature on received buffer (if any), and queue the
1863 * workqueue completion task.
1864 */
1865static void
1866smb2_writev_callback(struct mid_q_entry *mid)
1867{
1868 struct cifs_writedata *wdata = mid->callback_data;
1869 struct cifs_tcon *tcon = tlink_tcon(wdata->cfile->tlink);
1870 unsigned int written;
1871 struct smb2_write_rsp *rsp = (struct smb2_write_rsp *)mid->resp_buf;
1872 unsigned int credits_received = 1;
1873
1874 switch (mid->mid_state) {
1875 case MID_RESPONSE_RECEIVED:
1876 credits_received = le16_to_cpu(rsp->hdr.CreditRequest);
1877 wdata->result = smb2_check_receive(mid, tcon->ses->server, 0);
1878 if (wdata->result != 0)
1879 break;
1880
1881 written = le32_to_cpu(rsp->DataLength);
1882 /*
1883 * Mask off high 16 bits when bytes written as returned
1884 * by the server is greater than bytes requested by the
1885 * client. OS/2 servers are known to set incorrect
1886 * CountHigh values.
1887 */
1888 if (written > wdata->bytes)
1889 written &= 0xFFFF;
1890
1891 if (written < wdata->bytes)
1892 wdata->result = -ENOSPC;
1893 else
1894 wdata->bytes = written;
1895 break;
1896 case MID_REQUEST_SUBMITTED:
1897 case MID_RETRY_NEEDED:
1898 wdata->result = -EAGAIN;
1899 break;
1900 default:
1901 wdata->result = -EIO;
1902 break;
1903 }
1904
1905 if (wdata->result)
1906 cifs_stats_fail_inc(tcon, SMB2_WRITE_HE);
1907
1908 queue_work(cifsiod_wq, &wdata->work);
1909 DeleteMidQEntry(mid);
1910 add_credits(tcon->ses->server, credits_received, 0);
1911}
1912
1913/* smb2_async_writev - send an async write, and set up mid to handle result */
1914int
4a5c80d7
SF		 1915smb2_async_writev(struct cifs_writedata *wdata,
1916 void (*release)(struct kref *kref))
33319141 1917{
cb7e9eab 1918 int rc = -EACCES, flags = 0;
33319141
PS		 1919 struct smb2_write_req *req = NULL;
1920 struct cifs_tcon *tcon = tlink_tcon(wdata->cfile->tlink);
cb7e9eab 1921 struct TCP_Server_Info *server = tcon->ses->server;
eddb079d 1922 struct kvec iov;
fec344e3 1923 struct smb_rqst rqst;
33319141
PS		 1924
1925 rc = small_smb2_init(SMB2_WRITE, tcon, (void **) &req);
cb7e9eab
PS		 1926 if (rc) {
1927 if (rc == -EAGAIN && wdata->credits) {
1928 /* credits was reset by reconnect */
1929 wdata->credits = 0;
1930 /* reduce in_flight value since we won't send the req */
1931 spin_lock(&server->req_lock);
1932 server->in_flight--;
1933 spin_unlock(&server->req_lock);
1934 }
33319141 1935 goto async_writev_out;
cb7e9eab 1936 }
33319141 1937
33319141
PS		 1938 req->hdr.ProcessId = cpu_to_le32(wdata->cfile->pid);
1939
1940 req->PersistentFileId = wdata->cfile->fid.persistent_fid;
1941 req->VolatileFileId = wdata->cfile->fid.volatile_fid;
1942 req->WriteChannelInfoOffset = 0;
1943 req->WriteChannelInfoLength = 0;
1944 req->Channel = 0;
1945 req->Offset = cpu_to_le64(wdata->offset);
1946 /* 4 for rfc1002 length field */
1947 req->DataOffset = cpu_to_le16(
1948 offsetof(struct smb2_write_req, Buffer) - 4);
1949 req->RemainingBytes = 0;
1950
1951 /* 4 for rfc1002 length field and 1 for Buffer */
eddb079d
JL		 1952 iov.iov_len = get_rfc1002_length(req) + 4 - 1;
1953 iov.iov_base = req;
33319141 1954
eddb079d
JL		 1955 rqst.rq_iov = &iov;
1956 rqst.rq_nvec = 1;
1957 rqst.rq_pages = wdata->pages;
1958 rqst.rq_npages = wdata->nr_pages;
1959 rqst.rq_pagesz = wdata->pagesz;
1960 rqst.rq_tailsz = wdata->tailsz;
33319141 1961
f96637be
JP		 1962 cifs_dbg(FYI, "async write at %llu %u bytes\n",
1963 wdata->offset, wdata->bytes);
33319141
PS		 1964
1965 req->Length = cpu_to_le32(wdata->bytes);
1966
1967 inc_rfc1001_len(&req->hdr, wdata->bytes - 1 /* Buffer */);
1968
cb7e9eab
PS		 1969 if (wdata->credits) {
1970 req->hdr.CreditCharge = cpu_to_le16(DIV_ROUND_UP(wdata->bytes,
1971 SMB2_MAX_BUFFER_SIZE));
1972 spin_lock(&server->req_lock);
1973 server->credits += wdata->credits -
1974 le16_to_cpu(req->hdr.CreditCharge);
1975 spin_unlock(&server->req_lock);
1976 wake_up(&server->request_q);
1977 flags = CIFS_HAS_CREDITS;
1978 }
1979
33319141 1980 kref_get(&wdata->refcount);
cb7e9eab
PS		 1981 rc = cifs_call_async(server, &rqst, NULL, smb2_writev_callback, wdata,
1982 flags);
33319141 1983
e5d04887 1984 if (rc) {
4a5c80d7 1985 kref_put(&wdata->refcount, release);
e5d04887
PS		 1986 cifs_stats_fail_inc(tcon, SMB2_WRITE_HE);
1987 }
33319141 1988
33319141
PS		 1989async_writev_out:
1990 cifs_small_buf_release(req);
33319141
PS		 1991 return rc;
1992}
009d3443
PS		 1993
1994/*
1995 * SMB2_write function gets iov pointer to kvec array with n_vec as a length.
1996 * The length field from io_parms must be at least 1 and indicates a number of
1997 * elements with data to write that begins with position 1 in iov array. All
1998 * data length is specified by count.
1999 */
2000int
2001SMB2_write(const unsigned int xid, struct cifs_io_parms *io_parms,
2002 unsigned int *nbytes, struct kvec *iov, int n_vec)
2003{
2004 int rc = 0;
2005 struct smb2_write_req *req = NULL;
2006 struct smb2_write_rsp *rsp = NULL;
2007 int resp_buftype;
2008 *nbytes = 0;
2009
2010 if (n_vec < 1)
2011 return rc;
2012
2013 rc = small_smb2_init(SMB2_WRITE, io_parms->tcon, (void **) &req);
2014 if (rc)
2015 return rc;
2016
2017 if (io_parms->tcon->ses->server == NULL)
2018 return -ECONNABORTED;
2019
2020 req->hdr.ProcessId = cpu_to_le32(io_parms->pid);
2021
2022 req->PersistentFileId = io_parms->persistent_fid;
2023 req->VolatileFileId = io_parms->volatile_fid;
2024 req->WriteChannelInfoOffset = 0;
2025 req->WriteChannelInfoLength = 0;
2026 req->Channel = 0;
2027 req->Length = cpu_to_le32(io_parms->length);
2028 req->Offset = cpu_to_le64(io_parms->offset);
2029 /* 4 for rfc1002 length field */
2030 req->DataOffset = cpu_to_le16(
2031 offsetof(struct smb2_write_req, Buffer) - 4);
2032 req->RemainingBytes = 0;
2033
2034 iov[0].iov_base = (char *)req;
2035 /* 4 for rfc1002 length field and 1 for Buffer */
2036 iov[0].iov_len = get_rfc1002_length(req) + 4 - 1;
2037
2038 /* length of entire message including data to be written */
2039 inc_rfc1001_len(req, io_parms->length - 1 /* Buffer */);
2040
2041 rc = SendReceive2(xid, io_parms->tcon->ses, iov, n_vec + 1,
2042 &resp_buftype, 0);
e5d04887 2043 rsp = (struct smb2_write_rsp *)iov[0].iov_base;
009d3443
PS		 2044
2045 if (rc) {
2046 cifs_stats_fail_inc(io_parms->tcon, SMB2_WRITE_HE);
f96637be 2047 cifs_dbg(VFS, "Send error in write = %d\n", rc);
e5d04887 2048 } else
009d3443 2049 *nbytes = le32_to_cpu(rsp->DataLength);
e5d04887
PS		 2050
2051 free_rsp_buf(resp_buftype, rsp);
009d3443
PS		 2052 return rc;
2053}
35143eb5 2054
d324f08d
PS		 2055static unsigned int
2056num_entries(char *bufstart, char *end_of_buf, char **lastentry, size_t size)
2057{
2058 int len;
2059 unsigned int entrycount = 0;
2060 unsigned int next_offset = 0;
2061 FILE_DIRECTORY_INFO *entryptr;
2062
2063 if (bufstart == NULL)
2064 return 0;
2065
2066 entryptr = (FILE_DIRECTORY_INFO *)bufstart;
2067
2068 while (1) {
2069 entryptr = (FILE_DIRECTORY_INFO *)
2070 ((char *)entryptr + next_offset);
2071
2072 if ((char *)entryptr + size > end_of_buf) {
f96637be 2073 cifs_dbg(VFS, "malformed search entry would overflow\n");
d324f08d
PS		 2074 break;
2075 }
2076
2077 len = le32_to_cpu(entryptr->FileNameLength);
2078 if ((char *)entryptr + len + size > end_of_buf) {
f96637be
JP		 2079 cifs_dbg(VFS, "directory entry name would overflow frame end of buf %p\n",
2080 end_of_buf);
d324f08d
PS		 2081 break;
2082 }
2083
2084 *lastentry = (char *)entryptr;
2085 entrycount++;
2086
2087 next_offset = le32_to_cpu(entryptr->NextEntryOffset);
2088 if (!next_offset)
2089 break;
2090 }
2091
2092 return entrycount;
2093}
2094
2095/*
2096 * Readdir/FindFirst
2097 */
2098int
2099SMB2_query_directory(const unsigned int xid, struct cifs_tcon *tcon,
2100 u64 persistent_fid, u64 volatile_fid, int index,
2101 struct cifs_search_info *srch_inf)
2102{
2103 struct smb2_query_directory_req *req;
2104 struct smb2_query_directory_rsp *rsp = NULL;
2105 struct kvec iov[2];
2106 int rc = 0;
2107 int len;
2108 int resp_buftype;
2109 unsigned char *bufptr;
2110 struct TCP_Server_Info *server;
2111 struct cifs_ses *ses = tcon->ses;
2112 __le16 asteriks = cpu_to_le16('*');
2113 char *end_of_smb;
2114 unsigned int output_size = CIFSMaxBufSize;
2115 size_t info_buf_size;
2116
2117 if (ses && (ses->server))
2118 server = ses->server;
2119 else
2120 return -EIO;
2121
2122 rc = small_smb2_init(SMB2_QUERY_DIRECTORY, tcon, (void **) &req);
2123 if (rc)
2124 return rc;
2125
2126 switch (srch_inf->info_level) {
2127 case SMB_FIND_FILE_DIRECTORY_INFO:
2128 req->FileInformationClass = FILE_DIRECTORY_INFORMATION;
2129 info_buf_size = sizeof(FILE_DIRECTORY_INFO) - 1;
2130 break;
2131 case SMB_FIND_FILE_ID_FULL_DIR_INFO:
2132 req->FileInformationClass = FILEID_FULL_DIRECTORY_INFORMATION;
2133 info_buf_size = sizeof(SEARCH_ID_FULL_DIR_INFO) - 1;
2134 break;
2135 default:
f96637be
JP		 2136 cifs_dbg(VFS, "info level %u isn't supported\n",
2137 srch_inf->info_level);
d324f08d
PS		 2138 rc = -EINVAL;
2139 goto qdir_exit;
2140 }
2141
2142 req->FileIndex = cpu_to_le32(index);
2143 req->PersistentFileId = persistent_fid;
2144 req->VolatileFileId = volatile_fid;
2145
2146 len = 0x2;
2147 bufptr = req->Buffer;
2148 memcpy(bufptr, &asteriks, len);
2149
2150 req->FileNameOffset =
2151 cpu_to_le16(sizeof(struct smb2_query_directory_req) - 1 - 4);
2152 req->FileNameLength = cpu_to_le16(len);
2153 /*
2154 * BB could be 30 bytes or so longer if we used SMB2 specific
2155 * buffer lengths, but this is safe and close enough.
2156 */
2157 output_size = min_t(unsigned int, output_size, server->maxBuf);
2158 output_size = min_t(unsigned int, output_size, 2 << 15);
2159 req->OutputBufferLength = cpu_to_le32(output_size);
2160
2161 iov[0].iov_base = (char *)req;
2162 /* 4 for RFC1001 length and 1 for Buffer */
2163 iov[0].iov_len = get_rfc1002_length(req) + 4 - 1;
2164
2165 iov[1].iov_base = (char *)(req->Buffer);
2166 iov[1].iov_len = len;
2167
2168 inc_rfc1001_len(req, len - 1 /* Buffer */);
2169
2170 rc = SendReceive2(xid, ses, iov, 2, &resp_buftype, 0);
e5d04887
PS		 2171 rsp = (struct smb2_query_directory_rsp *)iov[0].iov_base;
2172
d324f08d
PS		 2173 if (rc) {
2174 cifs_stats_fail_inc(tcon, SMB2_QUERY_DIRECTORY_HE);
2175 goto qdir_exit;
2176 }
d324f08d
PS		 2177
2178 rc = validate_buf(le16_to_cpu(rsp->OutputBufferOffset),
2179 le32_to_cpu(rsp->OutputBufferLength), &rsp->hdr,
2180 info_buf_size);
2181 if (rc)
2182 goto qdir_exit;
2183
2184 srch_inf->unicode = true;
2185
2186 if (srch_inf->ntwrk_buf_start) {
2187 if (srch_inf->smallBuf)
2188 cifs_small_buf_release(srch_inf->ntwrk_buf_start);
2189 else
2190 cifs_buf_release(srch_inf->ntwrk_buf_start);
2191 }
2192 srch_inf->ntwrk_buf_start = (char *)rsp;
2193 srch_inf->srch_entries_start = srch_inf->last_entry = 4 /* rfclen */ +
2194 (char *)&rsp->hdr + le16_to_cpu(rsp->OutputBufferOffset);
2195 /* 4 for rfc1002 length field */
2196 end_of_smb = get_rfc1002_length(rsp) + 4 + (char *)&rsp->hdr;
2197 srch_inf->entries_in_buffer =
2198 num_entries(srch_inf->srch_entries_start, end_of_smb,
2199 &srch_inf->last_entry, info_buf_size);
2200 srch_inf->index_of_last_entry += srch_inf->entries_in_buffer;
f96637be
JP		 2201 cifs_dbg(FYI, "num entries %d last_index %lld srch start %p srch end %p\n",
2202 srch_inf->entries_in_buffer, srch_inf->index_of_last_entry,
2203 srch_inf->srch_entries_start, srch_inf->last_entry);
d324f08d
PS		 2204 if (resp_buftype == CIFS_LARGE_BUFFER)
2205 srch_inf->smallBuf = false;
2206 else if (resp_buftype == CIFS_SMALL_BUFFER)
2207 srch_inf->smallBuf = true;
2208 else
f96637be 2209 cifs_dbg(VFS, "illegal search buffer type\n");
d324f08d
PS		 2210
2211 if (rsp->hdr.Status == STATUS_NO_MORE_FILES)
2212 srch_inf->endOfSearch = 1;
2213 else
2214 srch_inf->endOfSearch = 0;
2215
2216 return rc;
2217
2218qdir_exit:
2219 free_rsp_buf(resp_buftype, rsp);
2220 return rc;
2221}
2222
35143eb5
PS		 2223static int
2224send_set_info(const unsigned int xid, struct cifs_tcon *tcon,
c839ff24 2225 u64 persistent_fid, u64 volatile_fid, u32 pid, int info_class,
35143eb5
PS		 2226 unsigned int num, void **data, unsigned int *size)
2227{
2228 struct smb2_set_info_req *req;
2229 struct smb2_set_info_rsp *rsp = NULL;
2230 struct kvec *iov;
2231 int rc = 0;
2232 int resp_buftype;
2233 unsigned int i;
2234 struct TCP_Server_Info *server;
2235 struct cifs_ses *ses = tcon->ses;
2236
2237 if (ses && (ses->server))
2238 server = ses->server;
2239 else
2240 return -EIO;
2241
2242 if (!num)
2243 return -EINVAL;
2244
2245 iov = kmalloc(sizeof(struct kvec) * num, GFP_KERNEL);
2246 if (!iov)
2247 return -ENOMEM;
2248
2249 rc = small_smb2_init(SMB2_SET_INFO, tcon, (void **) &req);
2250 if (rc) {
2251 kfree(iov);
2252 return rc;
2253 }
2254
c839ff24
PS		 2255 req->hdr.ProcessId = cpu_to_le32(pid);
2256
35143eb5
PS		 2257 req->InfoType = SMB2_O_INFO_FILE;
2258 req->FileInfoClass = info_class;
2259 req->PersistentFileId = persistent_fid;
2260 req->VolatileFileId = volatile_fid;
2261
2262 /* 4 for RFC1001 length and 1 for Buffer */
2263 req->BufferOffset =
2264 cpu_to_le16(sizeof(struct smb2_set_info_req) - 1 - 4);
2265 req->BufferLength = cpu_to_le32(*size);
2266
2267 inc_rfc1001_len(req, *size - 1 /* Buffer */);
2268
2269 memcpy(req->Buffer, *data, *size);
2270
2271 iov[0].iov_base = (char *)req;
2272 /* 4 for RFC1001 length */
2273 iov[0].iov_len = get_rfc1002_length(req) + 4;
2274
2275 for (i = 1; i < num; i++) {
2276 inc_rfc1001_len(req, size[i]);
2277 le32_add_cpu(&req->BufferLength, size[i]);
2278 iov[i].iov_base = (char *)data[i];
2279 iov[i].iov_len = size[i];
2280 }
2281
2282 rc = SendReceive2(xid, ses, iov, num, &resp_buftype, 0);
2283 rsp = (struct smb2_set_info_rsp *)iov[0].iov_base;
2284
7d3fb24b 2285 if (rc != 0)
35143eb5 2286 cifs_stats_fail_inc(tcon, SMB2_SET_INFO_HE);
7d3fb24b 2287
35143eb5
PS		 2288 free_rsp_buf(resp_buftype, rsp);
2289 kfree(iov);
2290 return rc;
2291}
2292
2293int
2294SMB2_rename(const unsigned int xid, struct cifs_tcon *tcon,
2295 u64 persistent_fid, u64 volatile_fid, __le16 *target_file)
2296{
2297 struct smb2_file_rename_info info;
2298 void **data;
2299 unsigned int size[2];
2300 int rc;
2301 int len = (2 * UniStrnlen((wchar_t *)target_file, PATH_MAX));
2302
2303 data = kmalloc(sizeof(void *) * 2, GFP_KERNEL);
2304 if (!data)
2305 return -ENOMEM;
2306
2307 info.ReplaceIfExists = 1; /* 1 = replace existing target with new */
2308 /* 0 = fail if target already exists */
2309 info.RootDirectory = 0; /* MBZ for network ops (why does spec say?) */
2310 info.FileNameLength = cpu_to_le32(len);
2311
2312 data[0] = &info;
2313 size[0] = sizeof(struct smb2_file_rename_info);
2314
2315 data[1] = target_file;
2316 size[1] = len + 2 /* null */;
2317
2318 rc = send_set_info(xid, tcon, persistent_fid, volatile_fid,
c839ff24
PS		 2319 current->tgid, FILE_RENAME_INFORMATION, 2, data,
2320 size);
35143eb5
PS		 2321 kfree(data);
2322 return rc;
2323}
568798cc
PS		 2324
2325int
2326SMB2_set_hardlink(const unsigned int xid, struct cifs_tcon *tcon,
2327 u64 persistent_fid, u64 volatile_fid, __le16 *target_file)
2328{
2329 struct smb2_file_link_info info;
2330 void **data;
2331 unsigned int size[2];
2332 int rc;
2333 int len = (2 * UniStrnlen((wchar_t *)target_file, PATH_MAX));
2334
2335 data = kmalloc(sizeof(void *) * 2, GFP_KERNEL);
2336 if (!data)
2337 return -ENOMEM;
2338
2339 info.ReplaceIfExists = 0; /* 1 = replace existing link with new */
2340 /* 0 = fail if link already exists */
2341 info.RootDirectory = 0; /* MBZ for network ops (why does spec say?) */
2342 info.FileNameLength = cpu_to_le32(len);
2343
2344 data[0] = &info;
2345 size[0] = sizeof(struct smb2_file_link_info);
2346
2347 data[1] = target_file;
2348 size[1] = len + 2 /* null */;
2349
2350 rc = send_set_info(xid, tcon, persistent_fid, volatile_fid,
c839ff24 2351 current->tgid, FILE_LINK_INFORMATION, 2, data, size);
568798cc
PS		 2352 kfree(data);
2353 return rc;
2354}
c839ff24
PS		 2355
2356int
2357SMB2_set_eof(const unsigned int xid, struct cifs_tcon *tcon, u64 persistent_fid,
2358 u64 volatile_fid, u32 pid, __le64 *eof)
2359{
2360 struct smb2_file_eof_info info;
2361 void *data;
2362 unsigned int size;
2363
2364 info.EndOfFile = *eof;
2365
2366 data = &info;
2367 size = sizeof(struct smb2_file_eof_info);
2368
2369 return send_set_info(xid, tcon, persistent_fid, volatile_fid, pid,
2370 FILE_END_OF_FILE_INFORMATION, 1, &data, &size);
2371}
1feeaac7
PS		 2372
2373int
2374SMB2_set_info(const unsigned int xid, struct cifs_tcon *tcon,
2375 u64 persistent_fid, u64 volatile_fid, FILE_BASIC_INFO *buf)
2376{
2377 unsigned int size;
2378 size = sizeof(FILE_BASIC_INFO);
2379 return send_set_info(xid, tcon, persistent_fid, volatile_fid,
2380 current->tgid, FILE_BASIC_INFORMATION, 1,
2381 (void **)&buf, &size);
2382}
983c88a4
PS		 2383
2384int
2385SMB2_oplock_break(const unsigned int xid, struct cifs_tcon *tcon,
2386 const u64 persistent_fid, const u64 volatile_fid,
2387 __u8 oplock_level)
2388{
2389 int rc;
2390 struct smb2_oplock_break *req = NULL;
2391
f96637be 2392 cifs_dbg(FYI, "SMB2_oplock_break\n");
983c88a4
PS		 2393 rc = small_smb2_init(SMB2_OPLOCK_BREAK, tcon, (void **) &req);
2394
2395 if (rc)
2396 return rc;
2397
2398 req->VolatileFid = volatile_fid;
2399 req->PersistentFid = persistent_fid;
2400 req->OplockLevel = oplock_level;
2401 req->hdr.CreditRequest = cpu_to_le16(1);
2402
2403 rc = SendReceiveNoRsp(xid, tcon->ses, (char *) req, CIFS_OBREAK_OP);
2404 /* SMB2 buffer freed by function above */
2405
2406 if (rc) {
2407 cifs_stats_fail_inc(tcon, SMB2_OPLOCK_BREAK_HE);
f96637be 2408 cifs_dbg(FYI, "Send error in Oplock Break = %d\n", rc);
983c88a4
PS		 2409 }
2410
2411 return rc;
2412}
6fc05c25
PS		 2413
2414static void
2415copy_fs_info_to_kstatfs(struct smb2_fs_full_size_info *pfs_inf,
2416 struct kstatfs *kst)
2417{
2418 kst->f_bsize = le32_to_cpu(pfs_inf->BytesPerSector) *
2419 le32_to_cpu(pfs_inf->SectorsPerAllocationUnit);
2420 kst->f_blocks = le64_to_cpu(pfs_inf->TotalAllocationUnits);
2421 kst->f_bfree = le64_to_cpu(pfs_inf->ActualAvailableAllocationUnits);
2422 kst->f_bavail = le64_to_cpu(pfs_inf->CallerAvailableAllocationUnits);
2423 return;
2424}
2425
2426static int
2427build_qfs_info_req(struct kvec *iov, struct cifs_tcon *tcon, int level,
2428 int outbuf_len, u64 persistent_fid, u64 volatile_fid)
2429{
2430 int rc;
2431 struct smb2_query_info_req *req;
2432
f96637be 2433 cifs_dbg(FYI, "Query FSInfo level %d\n", level);
6fc05c25
PS		 2434
2435 if ((tcon->ses == NULL) || (tcon->ses->server == NULL))
2436 return -EIO;
2437
2438 rc = small_smb2_init(SMB2_QUERY_INFO, tcon, (void **) &req);
2439 if (rc)
2440 return rc;
2441
2442 req->InfoType = SMB2_O_INFO_FILESYSTEM;
2443 req->FileInfoClass = level;
2444 req->PersistentFileId = persistent_fid;
2445 req->VolatileFileId = volatile_fid;
2446 /* 4 for rfc1002 length field and 1 for pad */
2447 req->InputBufferOffset =
2448 cpu_to_le16(sizeof(struct smb2_query_info_req) - 1 - 4);
2449 req->OutputBufferLength = cpu_to_le32(
2450 outbuf_len + sizeof(struct smb2_query_info_rsp) - 1 - 4);
2451
2452 iov->iov_base = (char *)req;
2453 /* 4 for rfc1002 length field */
2454 iov->iov_len = get_rfc1002_length(req) + 4;
2455 return 0;
2456}
2457
2458int
2459SMB2_QFS_info(const unsigned int xid, struct cifs_tcon *tcon,
2460 u64 persistent_fid, u64 volatile_fid, struct kstatfs *fsdata)
2461{
2462 struct smb2_query_info_rsp *rsp = NULL;
2463 struct kvec iov;
2464 int rc = 0;
2465 int resp_buftype;
2466 struct cifs_ses *ses = tcon->ses;
2467 struct smb2_fs_full_size_info *info = NULL;
2468
2469 rc = build_qfs_info_req(&iov, tcon, FS_FULL_SIZE_INFORMATION,
2470 sizeof(struct smb2_fs_full_size_info),
2471 persistent_fid, volatile_fid);
2472 if (rc)
2473 return rc;
2474
2475 rc = SendReceive2(xid, ses, &iov, 1, &resp_buftype, 0);
2476 if (rc) {
2477 cifs_stats_fail_inc(tcon, SMB2_QUERY_INFO_HE);
34f62640 2478 goto qfsinf_exit;
6fc05c25
PS		 2479 }
2480 rsp = (struct smb2_query_info_rsp *)iov.iov_base;
2481
2482 info = (struct smb2_fs_full_size_info *)(4 /* RFC1001 len */ +
2483 le16_to_cpu(rsp->OutputBufferOffset) + (char *)&rsp->hdr);
2484 rc = validate_buf(le16_to_cpu(rsp->OutputBufferOffset),
2485 le32_to_cpu(rsp->OutputBufferLength), &rsp->hdr,
2486 sizeof(struct smb2_fs_full_size_info));
2487 if (!rc)
2488 copy_fs_info_to_kstatfs(info, fsdata);
2489
34f62640
SF		 2490qfsinf_exit:
2491 free_rsp_buf(resp_buftype, iov.iov_base);
2492 return rc;
2493}
2494
2495int
2496SMB2_QFS_attr(const unsigned int xid, struct cifs_tcon *tcon,
2167114c 2497 u64 persistent_fid, u64 volatile_fid, int level)
34f62640
SF		 2498{
2499 struct smb2_query_info_rsp *rsp = NULL;
2500 struct kvec iov;
2501 int rc = 0;
2167114c 2502 int resp_buftype, max_len, min_len;
34f62640
SF		 2503 struct cifs_ses *ses = tcon->ses;
2504 unsigned int rsp_len, offset;
2505
2167114c
SF		 2506 if (level == FS_DEVICE_INFORMATION) {
2507 max_len = sizeof(FILE_SYSTEM_DEVICE_INFO);
2508 min_len = sizeof(FILE_SYSTEM_DEVICE_INFO);
2509 } else if (level == FS_ATTRIBUTE_INFORMATION) {
2510 max_len = sizeof(FILE_SYSTEM_ATTRIBUTE_INFO);
2511 min_len = MIN_FS_ATTR_INFO_SIZE;
af6a12ea
SF		 2512 } else if (level == FS_SECTOR_SIZE_INFORMATION) {
2513 max_len = sizeof(struct smb3_fs_ss_info);
2514 min_len = sizeof(struct smb3_fs_ss_info);
2167114c 2515 } else {
af6a12ea 2516 cifs_dbg(FYI, "Invalid qfsinfo level %d\n", level);
2167114c
SF		 2517 return -EINVAL;
2518 }
2519
2520 rc = build_qfs_info_req(&iov, tcon, level, max_len,
34f62640
SF		 2521 persistent_fid, volatile_fid);
2522 if (rc)
2523 return rc;
2524
2525 rc = SendReceive2(xid, ses, &iov, 1, &resp_buftype, 0);
2526 if (rc) {
2527 cifs_stats_fail_inc(tcon, SMB2_QUERY_INFO_HE);
2528 goto qfsattr_exit;
2529 }
2530 rsp = (struct smb2_query_info_rsp *)iov.iov_base;
2531
2532 rsp_len = le32_to_cpu(rsp->OutputBufferLength);
2533 offset = le16_to_cpu(rsp->OutputBufferOffset);
2167114c
SF		 2534 rc = validate_buf(offset, rsp_len, &rsp->hdr, min_len);
2535 if (rc)
2536 goto qfsattr_exit;
2537
2538 if (level == FS_ATTRIBUTE_INFORMATION)
34f62640
SF		 2539 memcpy(&tcon->fsAttrInfo, 4 /* RFC1001 len */ + offset
2540 + (char *)&rsp->hdr, min_t(unsigned int,
2167114c
SF		 2541 rsp_len, max_len));
2542 else if (level == FS_DEVICE_INFORMATION)
2543 memcpy(&tcon->fsDevInfo, 4 /* RFC1001 len */ + offset
2544 + (char *)&rsp->hdr, sizeof(FILE_SYSTEM_DEVICE_INFO));
af6a12ea
SF		 2545 else if (level == FS_SECTOR_SIZE_INFORMATION) {
2546 struct smb3_fs_ss_info *ss_info = (struct smb3_fs_ss_info *)
2547 (4 /* RFC1001 len */ + offset + (char *)&rsp->hdr);
2548 tcon->ss_flags = le32_to_cpu(ss_info->Flags);
2549 tcon->perf_sector_size =
2550 le32_to_cpu(ss_info->PhysicalBytesPerSectorForPerf);
2551 }
34f62640
SF		 2552
2553qfsattr_exit:
6fc05c25
PS		 2554 free_rsp_buf(resp_buftype, iov.iov_base);
2555 return rc;
2556}
f7ba7fe6
PS		 2557
2558int
2559smb2_lockv(const unsigned int xid, struct cifs_tcon *tcon,
2560 const __u64 persist_fid, const __u64 volatile_fid, const __u32 pid,
2561 const __u32 num_lock, struct smb2_lock_element *buf)
2562{
2563 int rc = 0;
2564 struct smb2_lock_req *req = NULL;
2565 struct kvec iov[2];
2566 int resp_buf_type;
2567 unsigned int count;
2568
f96637be 2569 cifs_dbg(FYI, "smb2_lockv num lock %d\n", num_lock);
f7ba7fe6
PS		 2570
2571 rc = small_smb2_init(SMB2_LOCK, tcon, (void **) &req);
2572 if (rc)
2573 return rc;
2574
2575 req->hdr.ProcessId = cpu_to_le32(pid);
2576 req->LockCount = cpu_to_le16(num_lock);
2577
2578 req->PersistentFileId = persist_fid;
2579 req->VolatileFileId = volatile_fid;
2580
2581 count = num_lock * sizeof(struct smb2_lock_element);
2582 inc_rfc1001_len(req, count - sizeof(struct smb2_lock_element));
2583
2584 iov[0].iov_base = (char *)req;
2585 /* 4 for rfc1002 length field and count for all locks */
2586 iov[0].iov_len = get_rfc1002_length(req) + 4 - count;
2587 iov[1].iov_base = (char *)buf;
2588 iov[1].iov_len = count;
2589
2590 cifs_stats_inc(&tcon->stats.cifs_stats.num_locks);
2591 rc = SendReceive2(xid, tcon->ses, iov, 2, &resp_buf_type, CIFS_NO_RESP);
2592 if (rc) {
f96637be 2593 cifs_dbg(FYI, "Send error in smb2_lockv = %d\n", rc);
f7ba7fe6
PS		 2594 cifs_stats_fail_inc(tcon, SMB2_LOCK_HE);
2595 }
2596
2597 return rc;
2598}
2599
2600int
2601SMB2_lock(const unsigned int xid, struct cifs_tcon *tcon,
2602 const __u64 persist_fid, const __u64 volatile_fid, const __u32 pid,
2603 const __u64 length, const __u64 offset, const __u32 lock_flags,
2604 const bool wait)
2605{
2606 struct smb2_lock_element lock;
2607
2608 lock.Offset = cpu_to_le64(offset);
2609 lock.Length = cpu_to_le64(length);
2610 lock.Flags = cpu_to_le32(lock_flags);
2611 if (!wait && lock_flags != SMB2_LOCKFLAG_UNLOCK)
2612 lock.Flags |= cpu_to_le32(SMB2_LOCKFLAG_FAIL_IMMEDIATELY);
2613
2614 return smb2_lockv(xid, tcon, persist_fid, volatile_fid, pid, 1, &lock);
2615}
0822f514
PS		 2616
2617int
2618SMB2_lease_break(const unsigned int xid, struct cifs_tcon *tcon,
2619 __u8 *lease_key, const __le32 lease_state)
2620{
2621 int rc;
2622 struct smb2_lease_ack *req = NULL;
2623
f96637be 2624 cifs_dbg(FYI, "SMB2_lease_break\n");
0822f514
PS		 2625 rc = small_smb2_init(SMB2_OPLOCK_BREAK, tcon, (void **) &req);
2626
2627 if (rc)
2628 return rc;
2629
2630 req->hdr.CreditRequest = cpu_to_le16(1);
2631 req->StructureSize = cpu_to_le16(36);
2632 inc_rfc1001_len(req, 12);
2633
2634 memcpy(req->LeaseKey, lease_key, 16);
2635 req->LeaseState = lease_state;
2636
2637 rc = SendReceiveNoRsp(xid, tcon->ses, (char *) req, CIFS_OBREAK_OP);
2638 /* SMB2 buffer freed by function above */
2639
2640 if (rc) {
2641 cifs_stats_fail_inc(tcon, SMB2_OPLOCK_BREAK_HE);
f96637be 2642 cifs_dbg(FYI, "Send error in Lease Break = %d\n", rc);
0822f514
PS		 2643 }
2644
2645 return rc;
2646}
A mirror of Linus' kernel repository