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