linux/fs/cifs/smb2pdu.c
<<
>>
Prefs
   1// SPDX-License-Identifier: LGPL-2.1
   2/*
   3 *   fs/cifs/smb2pdu.c
   4 *
   5 *   Copyright (C) International Business Machines  Corp., 2009, 2013
   6 *                 Etersoft, 2012
   7 *   Author(s): Steve French (sfrench@us.ibm.com)
   8 *              Pavel Shilovsky (pshilovsky@samba.org) 2012
   9 *
  10 *   Contains the routines for constructing the SMB2 PDUs themselves
  11 *
  12 */
  13
  14 /* SMB2 PDU handling routines here - except for leftovers (eg session setup) */
  15 /* Note that there are handle based routines which must be                   */
  16 /* treated slightly differently for reconnection purposes since we never     */
  17 /* want to reuse a stale file handle and only the caller knows the file info */
  18
  19#include <linux/fs.h>
  20#include <linux/kernel.h>
  21#include <linux/vfs.h>
  22#include <linux/task_io_accounting_ops.h>
  23#include <linux/uaccess.h>
  24#include <linux/uuid.h>
  25#include <linux/pagemap.h>
  26#include <linux/xattr.h>
  27#include "smb2pdu.h"
  28#include "cifsglob.h"
  29#include "cifsacl.h"
  30#include "cifsproto.h"
  31#include "smb2proto.h"
  32#include "cifs_unicode.h"
  33#include "cifs_debug.h"
  34#include "ntlmssp.h"
  35#include "smb2status.h"
  36#include "smb2glob.h"
  37#include "cifspdu.h"
  38#include "cifs_spnego.h"
  39#include "smbdirect.h"
  40#include "trace.h"
  41#ifdef CONFIG_CIFS_DFS_UPCALL
  42#include "dfs_cache.h"
  43#endif
  44
  45/*
  46 *  The following table defines the expected "StructureSize" of SMB2 requests
  47 *  in order by SMB2 command.  This is similar to "wct" in SMB/CIFS requests.
  48 *
  49 *  Note that commands are defined in smb2pdu.h in le16 but the array below is
  50 *  indexed by command in host byte order.
  51 */
  52static const int smb2_req_struct_sizes[NUMBER_OF_SMB2_COMMANDS] = {
  53        /* SMB2_NEGOTIATE */ 36,
  54        /* SMB2_SESSION_SETUP */ 25,
  55        /* SMB2_LOGOFF */ 4,
  56        /* SMB2_TREE_CONNECT */ 9,
  57        /* SMB2_TREE_DISCONNECT */ 4,
  58        /* SMB2_CREATE */ 57,
  59        /* SMB2_CLOSE */ 24,
  60        /* SMB2_FLUSH */ 24,
  61        /* SMB2_READ */ 49,
  62        /* SMB2_WRITE */ 49,
  63        /* SMB2_LOCK */ 48,
  64        /* SMB2_IOCTL */ 57,
  65        /* SMB2_CANCEL */ 4,
  66        /* SMB2_ECHO */ 4,
  67        /* SMB2_QUERY_DIRECTORY */ 33,
  68        /* SMB2_CHANGE_NOTIFY */ 32,
  69        /* SMB2_QUERY_INFO */ 41,
  70        /* SMB2_SET_INFO */ 33,
  71        /* SMB2_OPLOCK_BREAK */ 24 /* BB this is 36 for LEASE_BREAK variant */
  72};
  73
  74int smb3_encryption_required(const struct cifs_tcon *tcon)
  75{
  76        if (!tcon || !tcon->ses)
  77                return 0;
  78        if ((tcon->ses->session_flags & SMB2_SESSION_FLAG_ENCRYPT_DATA) ||
  79            (tcon->share_flags & SHI1005_FLAGS_ENCRYPT_DATA))
  80                return 1;
  81        if (tcon->seal &&
  82            (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_ENCRYPTION))
  83                return 1;
  84        return 0;
  85}
  86
  87static void
  88smb2_hdr_assemble(struct smb2_sync_hdr *shdr, __le16 smb2_cmd,
  89                  const struct cifs_tcon *tcon,
  90                  struct TCP_Server_Info *server)
  91{
  92        shdr->ProtocolId = SMB2_PROTO_NUMBER;
  93        shdr->StructureSize = cpu_to_le16(64);
  94        shdr->Command = smb2_cmd;
  95        if (server) {
  96                spin_lock(&server->req_lock);
  97                /* Request up to 10 credits but don't go over the limit. */
  98                if (server->credits >= server->max_credits)
  99                        shdr->CreditRequest = cpu_to_le16(0);
 100                else
 101                        shdr->CreditRequest = cpu_to_le16(
 102                                min_t(int, server->max_credits -
 103                                                server->credits, 10));
 104                spin_unlock(&server->req_lock);
 105        } else {
 106                shdr->CreditRequest = cpu_to_le16(2);
 107        }
 108        shdr->ProcessId = cpu_to_le32((__u16)current->tgid);
 109
 110        if (!tcon)
 111                goto out;
 112
 113        /* GLOBAL_CAP_LARGE_MTU will only be set if dialect > SMB2.02 */
 114        /* See sections 2.2.4 and 3.2.4.1.5 of MS-SMB2 */
 115        if (server && (server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
 116                shdr->CreditCharge = cpu_to_le16(1);
 117        /* else CreditCharge MBZ */
 118
 119        shdr->TreeId = tcon->tid;
 120        /* Uid is not converted */
 121        if (tcon->ses)
 122                shdr->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                shdr->Flags |= SMB2_FLAGS_DFS_OPERATIONS; */
 136
 137        if (server && server->sign && !smb3_encryption_required(tcon))
 138                shdr->Flags |= SMB2_FLAGS_SIGNED;
 139out:
 140        return;
 141}
 142
 143static int
 144smb2_reconnect(__le16 smb2_command, struct cifs_tcon *tcon,
 145               struct TCP_Server_Info *server)
 146{
 147        int rc;
 148        struct nls_table *nls_codepage;
 149        struct cifs_ses *ses;
 150        int retries;
 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 0;
 159
 160        if (smb2_command == SMB2_TREE_CONNECT)
 161                return 0;
 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) || !server)
 179                return -EIO;
 180
 181        ses = tcon->ses;
 182        retries = server->nr_targets;
 183
 184        /*
 185         * Give demultiplex thread up to 10 seconds to each target available for
 186         * reconnect -- should be greater than cifs socket timeout which is 7
 187         * seconds.
 188         */
 189        while (server->tcpStatus == CifsNeedReconnect) {
 190                /*
 191                 * Return to caller for TREE_DISCONNECT and LOGOFF and CLOSE
 192                 * here since they are implicitly done when session drops.
 193                 */
 194                switch (smb2_command) {
 195                /*
 196                 * BB Should we keep oplock break and add flush to exceptions?
 197                 */
 198                case SMB2_TREE_DISCONNECT:
 199                case SMB2_CANCEL:
 200                case SMB2_CLOSE:
 201                case SMB2_OPLOCK_BREAK:
 202                        return -EAGAIN;
 203                }
 204
 205                rc = wait_event_interruptible_timeout(server->response_q,
 206                                                      (server->tcpStatus != CifsNeedReconnect),
 207                                                      10 * HZ);
 208                if (rc < 0) {
 209                        cifs_dbg(FYI, "%s: aborting reconnect due to a received signal by the process\n",
 210                                 __func__);
 211                        return -ERESTARTSYS;
 212                }
 213
 214                /* are we still trying to reconnect? */
 215                if (server->tcpStatus != CifsNeedReconnect)
 216                        break;
 217
 218                if (retries && --retries)
 219                        continue;
 220
 221                /*
 222                 * on "soft" mounts we wait once. Hard mounts keep
 223                 * retrying until process is killed or server comes
 224                 * back on-line
 225                 */
 226                if (!tcon->retry) {
 227                        cifs_dbg(FYI, "gave up waiting on reconnect in smb_init\n");
 228                        return -EHOSTDOWN;
 229                }
 230                retries = server->nr_targets;
 231        }
 232
 233        if (!tcon->ses->need_reconnect && !tcon->need_reconnect)
 234                return 0;
 235
 236        nls_codepage = load_nls_default();
 237
 238        /*
 239         * need to prevent multiple threads trying to simultaneously reconnect
 240         * the same SMB session
 241         */
 242        mutex_lock(&tcon->ses->session_mutex);
 243
 244        /*
 245         * Recheck after acquire mutex. If another thread is negotiating
 246         * and the server never sends an answer the socket will be closed
 247         * and tcpStatus set to reconnect.
 248         */
 249        if (server->tcpStatus == CifsNeedReconnect) {
 250                rc = -EHOSTDOWN;
 251                mutex_unlock(&tcon->ses->session_mutex);
 252                goto out;
 253        }
 254
 255        /*
 256         * If we are reconnecting an extra channel, bind
 257         */
 258        if (server->is_channel) {
 259                ses->binding = true;
 260                ses->binding_chan = cifs_ses_find_chan(ses, server);
 261        }
 262
 263        rc = cifs_negotiate_protocol(0, tcon->ses);
 264        if (!rc && tcon->ses->need_reconnect) {
 265                rc = cifs_setup_session(0, tcon->ses, nls_codepage);
 266                if ((rc == -EACCES) && !tcon->retry) {
 267                        rc = -EHOSTDOWN;
 268                        ses->binding = false;
 269                        ses->binding_chan = NULL;
 270                        mutex_unlock(&tcon->ses->session_mutex);
 271                        goto failed;
 272                }
 273        }
 274        /*
 275         * End of channel binding
 276         */
 277        ses->binding = false;
 278        ses->binding_chan = NULL;
 279
 280        if (rc || !tcon->need_reconnect) {
 281                mutex_unlock(&tcon->ses->session_mutex);
 282                goto out;
 283        }
 284
 285        cifs_mark_open_files_invalid(tcon);
 286        if (tcon->use_persistent)
 287                tcon->need_reopen_files = true;
 288
 289        rc = cifs_tree_connect(0, tcon, nls_codepage);
 290        mutex_unlock(&tcon->ses->session_mutex);
 291
 292        cifs_dbg(FYI, "reconnect tcon rc = %d\n", rc);
 293        if (rc) {
 294                /* If sess reconnected but tcon didn't, something strange ... */
 295                pr_warn_once("reconnect tcon failed rc = %d\n", rc);
 296                goto out;
 297        }
 298
 299        if (smb2_command != SMB2_INTERNAL_CMD)
 300                mod_delayed_work(cifsiod_wq, &server->reconnect, 0);
 301
 302        atomic_inc(&tconInfoReconnectCount);
 303out:
 304        /*
 305         * Check if handle based operation so we know whether we can continue
 306         * or not without returning to caller to reset file handle.
 307         */
 308        /*
 309         * BB Is flush done by server on drop of tcp session? Should we special
 310         * case it and skip above?
 311         */
 312        switch (smb2_command) {
 313        case SMB2_FLUSH:
 314        case SMB2_READ:
 315        case SMB2_WRITE:
 316        case SMB2_LOCK:
 317        case SMB2_IOCTL:
 318        case SMB2_QUERY_DIRECTORY:
 319        case SMB2_CHANGE_NOTIFY:
 320        case SMB2_QUERY_INFO:
 321        case SMB2_SET_INFO:
 322                rc = -EAGAIN;
 323        }
 324failed:
 325        unload_nls(nls_codepage);
 326        return rc;
 327}
 328
 329static void
 330fill_small_buf(__le16 smb2_command, struct cifs_tcon *tcon,
 331               struct TCP_Server_Info *server,
 332               void *buf,
 333               unsigned int *total_len)
 334{
 335        struct smb2_sync_pdu *spdu = (struct smb2_sync_pdu *)buf;
 336        /* lookup word count ie StructureSize from table */
 337        __u16 parmsize = smb2_req_struct_sizes[le16_to_cpu(smb2_command)];
 338
 339        /*
 340         * smaller than SMALL_BUFFER_SIZE but bigger than fixed area of
 341         * largest operations (Create)
 342         */
 343        memset(buf, 0, 256);
 344
 345        smb2_hdr_assemble(&spdu->sync_hdr, smb2_command, tcon, server);
 346        spdu->StructureSize2 = cpu_to_le16(parmsize);
 347
 348        *total_len = parmsize + sizeof(struct smb2_sync_hdr);
 349}
 350
 351/*
 352 * Allocate and return pointer to an SMB request hdr, and set basic
 353 * SMB information in the SMB header. If the return code is zero, this
 354 * function must have filled in request_buf pointer.
 355 */
 356static int __smb2_plain_req_init(__le16 smb2_command, struct cifs_tcon *tcon,
 357                                 struct TCP_Server_Info *server,
 358                                 void **request_buf, unsigned int *total_len)
 359{
 360        /* BB eventually switch this to SMB2 specific small buf size */
 361        if (smb2_command == SMB2_SET_INFO)
 362                *request_buf = cifs_buf_get();
 363        else
 364                *request_buf = cifs_small_buf_get();
 365        if (*request_buf == NULL) {
 366                /* BB should we add a retry in here if not a writepage? */
 367                return -ENOMEM;
 368        }
 369
 370        fill_small_buf(smb2_command, tcon, server,
 371                       (struct smb2_sync_hdr *)(*request_buf),
 372                       total_len);
 373
 374        if (tcon != NULL) {
 375                uint16_t com_code = le16_to_cpu(smb2_command);
 376                cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_sent[com_code]);
 377                cifs_stats_inc(&tcon->num_smbs_sent);
 378        }
 379
 380        return 0;
 381}
 382
 383static int smb2_plain_req_init(__le16 smb2_command, struct cifs_tcon *tcon,
 384                               struct TCP_Server_Info *server,
 385                               void **request_buf, unsigned int *total_len)
 386{
 387        int rc;
 388
 389        rc = smb2_reconnect(smb2_command, tcon, server);
 390        if (rc)
 391                return rc;
 392
 393        return __smb2_plain_req_init(smb2_command, tcon, server, request_buf,
 394                                     total_len);
 395}
 396
 397static int smb2_ioctl_req_init(u32 opcode, struct cifs_tcon *tcon,
 398                               struct TCP_Server_Info *server,
 399                               void **request_buf, unsigned int *total_len)
 400{
 401        /* Skip reconnect only for FSCTL_VALIDATE_NEGOTIATE_INFO IOCTLs */
 402        if (opcode == FSCTL_VALIDATE_NEGOTIATE_INFO) {
 403                return __smb2_plain_req_init(SMB2_IOCTL, tcon, server,
 404                                             request_buf, total_len);
 405        }
 406        return smb2_plain_req_init(SMB2_IOCTL, tcon, server,
 407                                   request_buf, total_len);
 408}
 409
 410/* For explanation of negotiate contexts see MS-SMB2 section 2.2.3.1 */
 411
 412static void
 413build_preauth_ctxt(struct smb2_preauth_neg_context *pneg_ctxt)
 414{
 415        pneg_ctxt->ContextType = SMB2_PREAUTH_INTEGRITY_CAPABILITIES;
 416        pneg_ctxt->DataLength = cpu_to_le16(38);
 417        pneg_ctxt->HashAlgorithmCount = cpu_to_le16(1);
 418        pneg_ctxt->SaltLength = cpu_to_le16(SMB311_LINUX_CLIENT_SALT_SIZE);
 419        get_random_bytes(pneg_ctxt->Salt, SMB311_LINUX_CLIENT_SALT_SIZE);
 420        pneg_ctxt->HashAlgorithms = SMB2_PREAUTH_INTEGRITY_SHA512;
 421}
 422
 423static void
 424build_compression_ctxt(struct smb2_compression_capabilities_context *pneg_ctxt)
 425{
 426        pneg_ctxt->ContextType = SMB2_COMPRESSION_CAPABILITIES;
 427        pneg_ctxt->DataLength =
 428                cpu_to_le16(sizeof(struct smb2_compression_capabilities_context)
 429                          - sizeof(struct smb2_neg_context));
 430        pneg_ctxt->CompressionAlgorithmCount = cpu_to_le16(3);
 431        pneg_ctxt->CompressionAlgorithms[0] = SMB3_COMPRESS_LZ77;
 432        pneg_ctxt->CompressionAlgorithms[1] = SMB3_COMPRESS_LZ77_HUFF;
 433        pneg_ctxt->CompressionAlgorithms[2] = SMB3_COMPRESS_LZNT1;
 434}
 435
 436static unsigned int
 437build_signing_ctxt(struct smb2_signing_capabilities *pneg_ctxt)
 438{
 439        unsigned int ctxt_len = sizeof(struct smb2_signing_capabilities);
 440        unsigned short num_algs = 1; /* number of signing algorithms sent */
 441
 442        pneg_ctxt->ContextType = SMB2_SIGNING_CAPABILITIES;
 443        /*
 444         * Context Data length must be rounded to multiple of 8 for some servers
 445         */
 446        pneg_ctxt->DataLength = cpu_to_le16(DIV_ROUND_UP(
 447                                sizeof(struct smb2_signing_capabilities) -
 448                                sizeof(struct smb2_neg_context) +
 449                                (num_algs * 2 /* sizeof u16 */), 8) * 8);
 450        pneg_ctxt->SigningAlgorithmCount = cpu_to_le16(num_algs);
 451        pneg_ctxt->SigningAlgorithms[0] = cpu_to_le16(SIGNING_ALG_AES_CMAC);
 452
 453        ctxt_len += 2 /* sizeof le16 */ * num_algs;
 454        ctxt_len = DIV_ROUND_UP(ctxt_len, 8) * 8;
 455        return ctxt_len;
 456        /* TBD add SIGNING_ALG_AES_GMAC and/or SIGNING_ALG_HMAC_SHA256 */
 457}
 458
 459static void
 460build_encrypt_ctxt(struct smb2_encryption_neg_context *pneg_ctxt)
 461{
 462        pneg_ctxt->ContextType = SMB2_ENCRYPTION_CAPABILITIES;
 463        if (require_gcm_256) {
 464                pneg_ctxt->DataLength = cpu_to_le16(4); /* Cipher Count + 1 cipher */
 465                pneg_ctxt->CipherCount = cpu_to_le16(1);
 466                pneg_ctxt->Ciphers[0] = SMB2_ENCRYPTION_AES256_GCM;
 467        } else if (enable_gcm_256) {
 468                pneg_ctxt->DataLength = cpu_to_le16(8); /* Cipher Count + 3 ciphers */
 469                pneg_ctxt->CipherCount = cpu_to_le16(3);
 470                pneg_ctxt->Ciphers[0] = SMB2_ENCRYPTION_AES128_GCM;
 471                pneg_ctxt->Ciphers[1] = SMB2_ENCRYPTION_AES256_GCM;
 472                pneg_ctxt->Ciphers[2] = SMB2_ENCRYPTION_AES128_CCM;
 473        } else {
 474                pneg_ctxt->DataLength = cpu_to_le16(6); /* Cipher Count + 2 ciphers */
 475                pneg_ctxt->CipherCount = cpu_to_le16(2);
 476                pneg_ctxt->Ciphers[0] = SMB2_ENCRYPTION_AES128_GCM;
 477                pneg_ctxt->Ciphers[1] = SMB2_ENCRYPTION_AES128_CCM;
 478        }
 479}
 480
 481static unsigned int
 482build_netname_ctxt(struct smb2_netname_neg_context *pneg_ctxt, char *hostname)
 483{
 484        struct nls_table *cp = load_nls_default();
 485
 486        pneg_ctxt->ContextType = SMB2_NETNAME_NEGOTIATE_CONTEXT_ID;
 487
 488        /* copy up to max of first 100 bytes of server name to NetName field */
 489        pneg_ctxt->DataLength = cpu_to_le16(2 * cifs_strtoUTF16(pneg_ctxt->NetName, hostname, 100, cp));
 490        /* context size is DataLength + minimal smb2_neg_context */
 491        return DIV_ROUND_UP(le16_to_cpu(pneg_ctxt->DataLength) +
 492                        sizeof(struct smb2_neg_context), 8) * 8;
 493}
 494
 495static void
 496build_posix_ctxt(struct smb2_posix_neg_context *pneg_ctxt)
 497{
 498        pneg_ctxt->ContextType = SMB2_POSIX_EXTENSIONS_AVAILABLE;
 499        pneg_ctxt->DataLength = cpu_to_le16(POSIX_CTXT_DATA_LEN);
 500        /* SMB2_CREATE_TAG_POSIX is "0x93AD25509CB411E7B42383DE968BCD7C" */
 501        pneg_ctxt->Name[0] = 0x93;
 502        pneg_ctxt->Name[1] = 0xAD;
 503        pneg_ctxt->Name[2] = 0x25;
 504        pneg_ctxt->Name[3] = 0x50;
 505        pneg_ctxt->Name[4] = 0x9C;
 506        pneg_ctxt->Name[5] = 0xB4;
 507        pneg_ctxt->Name[6] = 0x11;
 508        pneg_ctxt->Name[7] = 0xE7;
 509        pneg_ctxt->Name[8] = 0xB4;
 510        pneg_ctxt->Name[9] = 0x23;
 511        pneg_ctxt->Name[10] = 0x83;
 512        pneg_ctxt->Name[11] = 0xDE;
 513        pneg_ctxt->Name[12] = 0x96;
 514        pneg_ctxt->Name[13] = 0x8B;
 515        pneg_ctxt->Name[14] = 0xCD;
 516        pneg_ctxt->Name[15] = 0x7C;
 517}
 518
 519static void
 520assemble_neg_contexts(struct smb2_negotiate_req *req,
 521                      struct TCP_Server_Info *server, unsigned int *total_len)
 522{
 523        char *pneg_ctxt;
 524        unsigned int ctxt_len, neg_context_count;
 525
 526        if (*total_len > 200) {
 527                /* In case length corrupted don't want to overrun smb buffer */
 528                cifs_server_dbg(VFS, "Bad frame length assembling neg contexts\n");
 529                return;
 530        }
 531
 532        /*
 533         * round up total_len of fixed part of SMB3 negotiate request to 8
 534         * byte boundary before adding negotiate contexts
 535         */
 536        *total_len = roundup(*total_len, 8);
 537
 538        pneg_ctxt = (*total_len) + (char *)req;
 539        req->NegotiateContextOffset = cpu_to_le32(*total_len);
 540
 541        build_preauth_ctxt((struct smb2_preauth_neg_context *)pneg_ctxt);
 542        ctxt_len = DIV_ROUND_UP(sizeof(struct smb2_preauth_neg_context), 8) * 8;
 543        *total_len += ctxt_len;
 544        pneg_ctxt += ctxt_len;
 545
 546        build_encrypt_ctxt((struct smb2_encryption_neg_context *)pneg_ctxt);
 547        ctxt_len = DIV_ROUND_UP(sizeof(struct smb2_encryption_neg_context), 8) * 8;
 548        *total_len += ctxt_len;
 549        pneg_ctxt += ctxt_len;
 550
 551        ctxt_len = build_netname_ctxt((struct smb2_netname_neg_context *)pneg_ctxt,
 552                                        server->hostname);
 553        *total_len += ctxt_len;
 554        pneg_ctxt += ctxt_len;
 555
 556        build_posix_ctxt((struct smb2_posix_neg_context *)pneg_ctxt);
 557        *total_len += sizeof(struct smb2_posix_neg_context);
 558        pneg_ctxt += sizeof(struct smb2_posix_neg_context);
 559
 560        neg_context_count = 4;
 561
 562        if (server->compress_algorithm) {
 563                build_compression_ctxt((struct smb2_compression_capabilities_context *)
 564                                pneg_ctxt);
 565                ctxt_len = DIV_ROUND_UP(
 566                        sizeof(struct smb2_compression_capabilities_context),
 567                                8) * 8;
 568                *total_len += ctxt_len;
 569                pneg_ctxt += ctxt_len;
 570                neg_context_count++;
 571        }
 572
 573        if (enable_negotiate_signing) {
 574                ctxt_len = build_signing_ctxt((struct smb2_signing_capabilities *)
 575                                pneg_ctxt);
 576                *total_len += ctxt_len;
 577                pneg_ctxt += ctxt_len;
 578                neg_context_count++;
 579        }
 580
 581        /* check for and add transport_capabilities and signing capabilities */
 582        req->NegotiateContextCount = cpu_to_le16(neg_context_count);
 583
 584}
 585
 586static void decode_preauth_context(struct smb2_preauth_neg_context *ctxt)
 587{
 588        unsigned int len = le16_to_cpu(ctxt->DataLength);
 589
 590        /* If invalid preauth context warn but use what we requested, SHA-512 */
 591        if (len < MIN_PREAUTH_CTXT_DATA_LEN) {
 592                pr_warn_once("server sent bad preauth context\n");
 593                return;
 594        } else if (len < MIN_PREAUTH_CTXT_DATA_LEN + le16_to_cpu(ctxt->SaltLength)) {
 595                pr_warn_once("server sent invalid SaltLength\n");
 596                return;
 597        }
 598        if (le16_to_cpu(ctxt->HashAlgorithmCount) != 1)
 599                pr_warn_once("Invalid SMB3 hash algorithm count\n");
 600        if (ctxt->HashAlgorithms != SMB2_PREAUTH_INTEGRITY_SHA512)
 601                pr_warn_once("unknown SMB3 hash algorithm\n");
 602}
 603
 604static void decode_compress_ctx(struct TCP_Server_Info *server,
 605                         struct smb2_compression_capabilities_context *ctxt)
 606{
 607        unsigned int len = le16_to_cpu(ctxt->DataLength);
 608
 609        /* sizeof compress context is a one element compression capbility struct */
 610        if (len < 10) {
 611                pr_warn_once("server sent bad compression cntxt\n");
 612                return;
 613        }
 614        if (le16_to_cpu(ctxt->CompressionAlgorithmCount) != 1) {
 615                pr_warn_once("Invalid SMB3 compress algorithm count\n");
 616                return;
 617        }
 618        if (le16_to_cpu(ctxt->CompressionAlgorithms[0]) > 3) {
 619                pr_warn_once("unknown compression algorithm\n");
 620                return;
 621        }
 622        server->compress_algorithm = ctxt->CompressionAlgorithms[0];
 623}
 624
 625static int decode_encrypt_ctx(struct TCP_Server_Info *server,
 626                              struct smb2_encryption_neg_context *ctxt)
 627{
 628        unsigned int len = le16_to_cpu(ctxt->DataLength);
 629
 630        cifs_dbg(FYI, "decode SMB3.11 encryption neg context of len %d\n", len);
 631        if (len < MIN_ENCRYPT_CTXT_DATA_LEN) {
 632                pr_warn_once("server sent bad crypto ctxt len\n");
 633                return -EINVAL;
 634        }
 635
 636        if (le16_to_cpu(ctxt->CipherCount) != 1) {
 637                pr_warn_once("Invalid SMB3.11 cipher count\n");
 638                return -EINVAL;
 639        }
 640        cifs_dbg(FYI, "SMB311 cipher type:%d\n", le16_to_cpu(ctxt->Ciphers[0]));
 641        if (require_gcm_256) {
 642                if (ctxt->Ciphers[0] != SMB2_ENCRYPTION_AES256_GCM) {
 643                        cifs_dbg(VFS, "Server does not support requested encryption type (AES256 GCM)\n");
 644                        return -EOPNOTSUPP;
 645                }
 646        } else if (ctxt->Ciphers[0] == 0) {
 647                /*
 648                 * e.g. if server only supported AES256_CCM (very unlikely)
 649                 * or server supported no encryption types or had all disabled.
 650                 * Since GLOBAL_CAP_ENCRYPTION will be not set, in the case
 651                 * in which mount requested encryption ("seal") checks later
 652                 * on during tree connection will return proper rc, but if
 653                 * seal not requested by client, since server is allowed to
 654                 * return 0 to indicate no supported cipher, we can't fail here
 655                 */
 656                server->cipher_type = 0;
 657                server->capabilities &= ~SMB2_GLOBAL_CAP_ENCRYPTION;
 658                pr_warn_once("Server does not support requested encryption types\n");
 659                return 0;
 660        } else if ((ctxt->Ciphers[0] != SMB2_ENCRYPTION_AES128_CCM) &&
 661                   (ctxt->Ciphers[0] != SMB2_ENCRYPTION_AES128_GCM) &&
 662                   (ctxt->Ciphers[0] != SMB2_ENCRYPTION_AES256_GCM)) {
 663                /* server returned a cipher we didn't ask for */
 664                pr_warn_once("Invalid SMB3.11 cipher returned\n");
 665                return -EINVAL;
 666        }
 667        server->cipher_type = ctxt->Ciphers[0];
 668        server->capabilities |= SMB2_GLOBAL_CAP_ENCRYPTION;
 669        return 0;
 670}
 671
 672static void decode_signing_ctx(struct TCP_Server_Info *server,
 673                               struct smb2_signing_capabilities *pctxt)
 674{
 675        unsigned int len = le16_to_cpu(pctxt->DataLength);
 676
 677        if ((len < 4) || (len > 16)) {
 678                pr_warn_once("server sent bad signing negcontext\n");
 679                return;
 680        }
 681        if (le16_to_cpu(pctxt->SigningAlgorithmCount) != 1) {
 682                pr_warn_once("Invalid signing algorithm count\n");
 683                return;
 684        }
 685        if (le16_to_cpu(pctxt->SigningAlgorithms[0]) > 2) {
 686                pr_warn_once("unknown signing algorithm\n");
 687                return;
 688        }
 689
 690        server->signing_negotiated = true;
 691        server->signing_algorithm = le16_to_cpu(pctxt->SigningAlgorithms[0]);
 692        cifs_dbg(FYI, "signing algorithm %d chosen\n",
 693                     server->signing_algorithm);
 694}
 695
 696
 697static int smb311_decode_neg_context(struct smb2_negotiate_rsp *rsp,
 698                                     struct TCP_Server_Info *server,
 699                                     unsigned int len_of_smb)
 700{
 701        struct smb2_neg_context *pctx;
 702        unsigned int offset = le32_to_cpu(rsp->NegotiateContextOffset);
 703        unsigned int ctxt_cnt = le16_to_cpu(rsp->NegotiateContextCount);
 704        unsigned int len_of_ctxts, i;
 705        int rc = 0;
 706
 707        cifs_dbg(FYI, "decoding %d negotiate contexts\n", ctxt_cnt);
 708        if (len_of_smb <= offset) {
 709                cifs_server_dbg(VFS, "Invalid response: negotiate context offset\n");
 710                return -EINVAL;
 711        }
 712
 713        len_of_ctxts = len_of_smb - offset;
 714
 715        for (i = 0; i < ctxt_cnt; i++) {
 716                int clen;
 717                /* check that offset is not beyond end of SMB */
 718                if (len_of_ctxts == 0)
 719                        break;
 720
 721                if (len_of_ctxts < sizeof(struct smb2_neg_context))
 722                        break;
 723
 724                pctx = (struct smb2_neg_context *)(offset + (char *)rsp);
 725                clen = le16_to_cpu(pctx->DataLength);
 726                if (clen > len_of_ctxts)
 727                        break;
 728
 729                if (pctx->ContextType == SMB2_PREAUTH_INTEGRITY_CAPABILITIES)
 730                        decode_preauth_context(
 731                                (struct smb2_preauth_neg_context *)pctx);
 732                else if (pctx->ContextType == SMB2_ENCRYPTION_CAPABILITIES)
 733                        rc = decode_encrypt_ctx(server,
 734                                (struct smb2_encryption_neg_context *)pctx);
 735                else if (pctx->ContextType == SMB2_COMPRESSION_CAPABILITIES)
 736                        decode_compress_ctx(server,
 737                                (struct smb2_compression_capabilities_context *)pctx);
 738                else if (pctx->ContextType == SMB2_POSIX_EXTENSIONS_AVAILABLE)
 739                        server->posix_ext_supported = true;
 740                else if (pctx->ContextType == SMB2_SIGNING_CAPABILITIES)
 741                        decode_signing_ctx(server,
 742                                (struct smb2_signing_capabilities *)pctx);
 743                else
 744                        cifs_server_dbg(VFS, "unknown negcontext of type %d ignored\n",
 745                                le16_to_cpu(pctx->ContextType));
 746
 747                if (rc)
 748                        break;
 749                /* offsets must be 8 byte aligned */
 750                clen = (clen + 7) & ~0x7;
 751                offset += clen + sizeof(struct smb2_neg_context);
 752                len_of_ctxts -= clen;
 753        }
 754        return rc;
 755}
 756
 757static struct create_posix *
 758create_posix_buf(umode_t mode)
 759{
 760        struct create_posix *buf;
 761
 762        buf = kzalloc(sizeof(struct create_posix),
 763                        GFP_KERNEL);
 764        if (!buf)
 765                return NULL;
 766
 767        buf->ccontext.DataOffset =
 768                cpu_to_le16(offsetof(struct create_posix, Mode));
 769        buf->ccontext.DataLength = cpu_to_le32(4);
 770        buf->ccontext.NameOffset =
 771                cpu_to_le16(offsetof(struct create_posix, Name));
 772        buf->ccontext.NameLength = cpu_to_le16(16);
 773
 774        /* SMB2_CREATE_TAG_POSIX is "0x93AD25509CB411E7B42383DE968BCD7C" */
 775        buf->Name[0] = 0x93;
 776        buf->Name[1] = 0xAD;
 777        buf->Name[2] = 0x25;
 778        buf->Name[3] = 0x50;
 779        buf->Name[4] = 0x9C;
 780        buf->Name[5] = 0xB4;
 781        buf->Name[6] = 0x11;
 782        buf->Name[7] = 0xE7;
 783        buf->Name[8] = 0xB4;
 784        buf->Name[9] = 0x23;
 785        buf->Name[10] = 0x83;
 786        buf->Name[11] = 0xDE;
 787        buf->Name[12] = 0x96;
 788        buf->Name[13] = 0x8B;
 789        buf->Name[14] = 0xCD;
 790        buf->Name[15] = 0x7C;
 791        buf->Mode = cpu_to_le32(mode);
 792        cifs_dbg(FYI, "mode on posix create 0%o\n", mode);
 793        return buf;
 794}
 795
 796static int
 797add_posix_context(struct kvec *iov, unsigned int *num_iovec, umode_t mode)
 798{
 799        struct smb2_create_req *req = iov[0].iov_base;
 800        unsigned int num = *num_iovec;
 801
 802        iov[num].iov_base = create_posix_buf(mode);
 803        if (mode == ACL_NO_MODE)
 804                cifs_dbg(FYI, "Invalid mode\n");
 805        if (iov[num].iov_base == NULL)
 806                return -ENOMEM;
 807        iov[num].iov_len = sizeof(struct create_posix);
 808        if (!req->CreateContextsOffset)
 809                req->CreateContextsOffset = cpu_to_le32(
 810                                sizeof(struct smb2_create_req) +
 811                                iov[num - 1].iov_len);
 812        le32_add_cpu(&req->CreateContextsLength, sizeof(struct create_posix));
 813        *num_iovec = num + 1;
 814        return 0;
 815}
 816
 817
 818/*
 819 *
 820 *      SMB2 Worker functions follow:
 821 *
 822 *      The general structure of the worker functions is:
 823 *      1) Call smb2_init (assembles SMB2 header)
 824 *      2) Initialize SMB2 command specific fields in fixed length area of SMB
 825 *      3) Call smb_sendrcv2 (sends request on socket and waits for response)
 826 *      4) Decode SMB2 command specific fields in the fixed length area
 827 *      5) Decode variable length data area (if any for this SMB2 command type)
 828 *      6) Call free smb buffer
 829 *      7) return
 830 *
 831 */
 832
 833int
 834SMB2_negotiate(const unsigned int xid, struct cifs_ses *ses)
 835{
 836        struct smb_rqst rqst;
 837        struct smb2_negotiate_req *req;
 838        struct smb2_negotiate_rsp *rsp;
 839        struct kvec iov[1];
 840        struct kvec rsp_iov;
 841        int rc = 0;
 842        int resp_buftype;
 843        struct TCP_Server_Info *server = cifs_ses_server(ses);
 844        int blob_offset, blob_length;
 845        char *security_blob;
 846        int flags = CIFS_NEG_OP;
 847        unsigned int total_len;
 848
 849        cifs_dbg(FYI, "Negotiate protocol\n");
 850
 851        if (!server) {
 852                WARN(1, "%s: server is NULL!\n", __func__);
 853                return -EIO;
 854        }
 855
 856        rc = smb2_plain_req_init(SMB2_NEGOTIATE, NULL, server,
 857                                 (void **) &req, &total_len);
 858        if (rc)
 859                return rc;
 860
 861        req->sync_hdr.SessionId = 0;
 862
 863        memset(server->preauth_sha_hash, 0, SMB2_PREAUTH_HASH_SIZE);
 864        memset(ses->preauth_sha_hash, 0, SMB2_PREAUTH_HASH_SIZE);
 865
 866        if (strcmp(server->vals->version_string,
 867                   SMB3ANY_VERSION_STRING) == 0) {
 868                req->Dialects[0] = cpu_to_le16(SMB30_PROT_ID);
 869                req->Dialects[1] = cpu_to_le16(SMB302_PROT_ID);
 870                req->Dialects[2] = cpu_to_le16(SMB311_PROT_ID);
 871                req->DialectCount = cpu_to_le16(3);
 872                total_len += 6;
 873        } else if (strcmp(server->vals->version_string,
 874                   SMBDEFAULT_VERSION_STRING) == 0) {
 875                req->Dialects[0] = cpu_to_le16(SMB21_PROT_ID);
 876                req->Dialects[1] = cpu_to_le16(SMB30_PROT_ID);
 877                req->Dialects[2] = cpu_to_le16(SMB302_PROT_ID);
 878                req->Dialects[3] = cpu_to_le16(SMB311_PROT_ID);
 879                req->DialectCount = cpu_to_le16(4);
 880                total_len += 8;
 881        } else {
 882                /* otherwise send specific dialect */
 883                req->Dialects[0] = cpu_to_le16(server->vals->protocol_id);
 884                req->DialectCount = cpu_to_le16(1);
 885                total_len += 2;
 886        }
 887
 888        /* only one of SMB2 signing flags may be set in SMB2 request */
 889        if (ses->sign)
 890                req->SecurityMode = cpu_to_le16(SMB2_NEGOTIATE_SIGNING_REQUIRED);
 891        else if (global_secflags & CIFSSEC_MAY_SIGN)
 892                req->SecurityMode = cpu_to_le16(SMB2_NEGOTIATE_SIGNING_ENABLED);
 893        else
 894                req->SecurityMode = 0;
 895
 896        req->Capabilities = cpu_to_le32(server->vals->req_capabilities);
 897        if (ses->chan_max > 1)
 898                req->Capabilities |= cpu_to_le32(SMB2_GLOBAL_CAP_MULTI_CHANNEL);
 899
 900        /* ClientGUID must be zero for SMB2.02 dialect */
 901        if (server->vals->protocol_id == SMB20_PROT_ID)
 902                memset(req->ClientGUID, 0, SMB2_CLIENT_GUID_SIZE);
 903        else {
 904                memcpy(req->ClientGUID, server->client_guid,
 905                        SMB2_CLIENT_GUID_SIZE);
 906                if ((server->vals->protocol_id == SMB311_PROT_ID) ||
 907                    (strcmp(server->vals->version_string,
 908                     SMB3ANY_VERSION_STRING) == 0) ||
 909                    (strcmp(server->vals->version_string,
 910                     SMBDEFAULT_VERSION_STRING) == 0))
 911                        assemble_neg_contexts(req, server, &total_len);
 912        }
 913        iov[0].iov_base = (char *)req;
 914        iov[0].iov_len = total_len;
 915
 916        memset(&rqst, 0, sizeof(struct smb_rqst));
 917        rqst.rq_iov = iov;
 918        rqst.rq_nvec = 1;
 919
 920        rc = cifs_send_recv(xid, ses, server,
 921                            &rqst, &resp_buftype, flags, &rsp_iov);
 922        cifs_small_buf_release(req);
 923        rsp = (struct smb2_negotiate_rsp *)rsp_iov.iov_base;
 924        /*
 925         * No tcon so can't do
 926         * cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_fail[SMB2...]);
 927         */
 928        if (rc == -EOPNOTSUPP) {
 929                cifs_server_dbg(VFS, "Dialect not supported by server. Consider  specifying vers=1.0 or vers=2.0 on mount for accessing older servers\n");
 930                goto neg_exit;
 931        } else if (rc != 0)
 932                goto neg_exit;
 933
 934        if (strcmp(server->vals->version_string,
 935                   SMB3ANY_VERSION_STRING) == 0) {
 936                if (rsp->DialectRevision == cpu_to_le16(SMB20_PROT_ID)) {
 937                        cifs_server_dbg(VFS,
 938                                "SMB2 dialect returned but not requested\n");
 939                        return -EIO;
 940                } else if (rsp->DialectRevision == cpu_to_le16(SMB21_PROT_ID)) {
 941                        cifs_server_dbg(VFS,
 942                                "SMB2.1 dialect returned but not requested\n");
 943                        return -EIO;
 944                } else if (rsp->DialectRevision == cpu_to_le16(SMB311_PROT_ID)) {
 945                        /* ops set to 3.0 by default for default so update */
 946                        server->ops = &smb311_operations;
 947                        server->vals = &smb311_values;
 948                }
 949        } else if (strcmp(server->vals->version_string,
 950                   SMBDEFAULT_VERSION_STRING) == 0) {
 951                if (rsp->DialectRevision == cpu_to_le16(SMB20_PROT_ID)) {
 952                        cifs_server_dbg(VFS,
 953                                "SMB2 dialect returned but not requested\n");
 954                        return -EIO;
 955                } else if (rsp->DialectRevision == cpu_to_le16(SMB21_PROT_ID)) {
 956                        /* ops set to 3.0 by default for default so update */
 957                        server->ops = &smb21_operations;
 958                        server->vals = &smb21_values;
 959                } else if (rsp->DialectRevision == cpu_to_le16(SMB311_PROT_ID)) {
 960                        server->ops = &smb311_operations;
 961                        server->vals = &smb311_values;
 962                }
 963        } else if (le16_to_cpu(rsp->DialectRevision) !=
 964                                server->vals->protocol_id) {
 965                /* if requested single dialect ensure returned dialect matched */
 966                cifs_server_dbg(VFS, "Invalid 0x%x dialect returned: not requested\n",
 967                                le16_to_cpu(rsp->DialectRevision));
 968                return -EIO;
 969        }
 970
 971        cifs_dbg(FYI, "mode 0x%x\n", rsp->SecurityMode);
 972
 973        if (rsp->DialectRevision == cpu_to_le16(SMB20_PROT_ID))
 974                cifs_dbg(FYI, "negotiated smb2.0 dialect\n");
 975        else if (rsp->DialectRevision == cpu_to_le16(SMB21_PROT_ID))
 976                cifs_dbg(FYI, "negotiated smb2.1 dialect\n");
 977        else if (rsp->DialectRevision == cpu_to_le16(SMB30_PROT_ID))
 978                cifs_dbg(FYI, "negotiated smb3.0 dialect\n");
 979        else if (rsp->DialectRevision == cpu_to_le16(SMB302_PROT_ID))
 980                cifs_dbg(FYI, "negotiated smb3.02 dialect\n");
 981        else if (rsp->DialectRevision == cpu_to_le16(SMB311_PROT_ID))
 982                cifs_dbg(FYI, "negotiated smb3.1.1 dialect\n");
 983        else {
 984                cifs_server_dbg(VFS, "Invalid dialect returned by server 0x%x\n",
 985                                le16_to_cpu(rsp->DialectRevision));
 986                rc = -EIO;
 987                goto neg_exit;
 988        }
 989        server->dialect = le16_to_cpu(rsp->DialectRevision);
 990
 991        /*
 992         * Keep a copy of the hash after negprot. This hash will be
 993         * the starting hash value for all sessions made from this
 994         * server.
 995         */
 996        memcpy(server->preauth_sha_hash, ses->preauth_sha_hash,
 997               SMB2_PREAUTH_HASH_SIZE);
 998
 999        /* SMB2 only has an extended negflavor */
1000        server->negflavor = CIFS_NEGFLAVOR_EXTENDED;
1001        /* set it to the maximum buffer size value we can send with 1 credit */
1002        server->maxBuf = min_t(unsigned int, le32_to_cpu(rsp->MaxTransactSize),
1003                               SMB2_MAX_BUFFER_SIZE);
1004        server->max_read = le32_to_cpu(rsp->MaxReadSize);
1005        server->max_write = le32_to_cpu(rsp->MaxWriteSize);
1006        server->sec_mode = le16_to_cpu(rsp->SecurityMode);
1007        if ((server->sec_mode & SMB2_SEC_MODE_FLAGS_ALL) != server->sec_mode)
1008                cifs_dbg(FYI, "Server returned unexpected security mode 0x%x\n",
1009                                server->sec_mode);
1010        server->capabilities = le32_to_cpu(rsp->Capabilities);
1011        /* Internal types */
1012        server->capabilities |= SMB2_NT_FIND | SMB2_LARGE_FILES;
1013
1014        /*
1015         * SMB3.0 supports only 1 cipher and doesn't have a encryption neg context
1016         * Set the cipher type manually.
1017         */
1018        if (server->dialect == SMB30_PROT_ID && (server->capabilities & SMB2_GLOBAL_CAP_ENCRYPTION))
1019                server->cipher_type = SMB2_ENCRYPTION_AES128_CCM;
1020
1021        security_blob = smb2_get_data_area_len(&blob_offset, &blob_length,
1022                                               (struct smb2_sync_hdr *)rsp);
1023        /*
1024         * See MS-SMB2 section 2.2.4: if no blob, client picks default which
1025         * for us will be
1026         *      ses->sectype = RawNTLMSSP;
1027         * but for time being this is our only auth choice so doesn't matter.
1028         * We just found a server which sets blob length to zero expecting raw.
1029         */
1030        if (blob_length == 0) {
1031                cifs_dbg(FYI, "missing security blob on negprot\n");
1032                server->sec_ntlmssp = true;
1033        }
1034
1035        rc = cifs_enable_signing(server, ses->sign);
1036        if (rc)
1037                goto neg_exit;
1038        if (blob_length) {
1039                rc = decode_negTokenInit(security_blob, blob_length, server);
1040                if (rc == 1)
1041                        rc = 0;
1042                else if (rc == 0)
1043                        rc = -EIO;
1044        }
1045
1046        if (rsp->DialectRevision == cpu_to_le16(SMB311_PROT_ID)) {
1047                if (rsp->NegotiateContextCount)
1048                        rc = smb311_decode_neg_context(rsp, server,
1049                                                       rsp_iov.iov_len);
1050                else
1051                        cifs_server_dbg(VFS, "Missing expected negotiate contexts\n");
1052        }
1053neg_exit:
1054        free_rsp_buf(resp_buftype, rsp);
1055        return rc;
1056}
1057
1058int smb3_validate_negotiate(const unsigned int xid, struct cifs_tcon *tcon)
1059{
1060        int rc;
1061        struct validate_negotiate_info_req *pneg_inbuf;
1062        struct validate_negotiate_info_rsp *pneg_rsp = NULL;
1063        u32 rsplen;
1064        u32 inbuflen; /* max of 4 dialects */
1065        struct TCP_Server_Info *server = tcon->ses->server;
1066
1067        cifs_dbg(FYI, "validate negotiate\n");
1068
1069        /* In SMB3.11 preauth integrity supersedes validate negotiate */
1070        if (server->dialect == SMB311_PROT_ID)
1071                return 0;
1072
1073        /*
1074         * validation ioctl must be signed, so no point sending this if we
1075         * can not sign it (ie are not known user).  Even if signing is not
1076         * required (enabled but not negotiated), in those cases we selectively
1077         * sign just this, the first and only signed request on a connection.
1078         * Having validation of negotiate info  helps reduce attack vectors.
1079         */
1080        if (tcon->ses->session_flags & SMB2_SESSION_FLAG_IS_GUEST)
1081                return 0; /* validation requires signing */
1082
1083        if (tcon->ses->user_name == NULL) {
1084                cifs_dbg(FYI, "Can't validate negotiate: null user mount\n");
1085                return 0; /* validation requires signing */
1086        }
1087
1088        if (tcon->ses->session_flags & SMB2_SESSION_FLAG_IS_NULL)
1089                cifs_tcon_dbg(VFS, "Unexpected null user (anonymous) auth flag sent by server\n");
1090
1091        pneg_inbuf = kmalloc(sizeof(*pneg_inbuf), GFP_NOFS);
1092        if (!pneg_inbuf)
1093                return -ENOMEM;
1094
1095        pneg_inbuf->Capabilities =
1096                        cpu_to_le32(server->vals->req_capabilities);
1097        if (tcon->ses->chan_max > 1)
1098                pneg_inbuf->Capabilities |= cpu_to_le32(SMB2_GLOBAL_CAP_MULTI_CHANNEL);
1099
1100        memcpy(pneg_inbuf->Guid, server->client_guid,
1101                                        SMB2_CLIENT_GUID_SIZE);
1102
1103        if (tcon->ses->sign)
1104                pneg_inbuf->SecurityMode =
1105                        cpu_to_le16(SMB2_NEGOTIATE_SIGNING_REQUIRED);
1106        else if (global_secflags & CIFSSEC_MAY_SIGN)
1107                pneg_inbuf->SecurityMode =
1108                        cpu_to_le16(SMB2_NEGOTIATE_SIGNING_ENABLED);
1109        else
1110                pneg_inbuf->SecurityMode = 0;
1111
1112
1113        if (strcmp(server->vals->version_string,
1114                SMB3ANY_VERSION_STRING) == 0) {
1115                pneg_inbuf->Dialects[0] = cpu_to_le16(SMB30_PROT_ID);
1116                pneg_inbuf->Dialects[1] = cpu_to_le16(SMB302_PROT_ID);
1117                pneg_inbuf->Dialects[2] = cpu_to_le16(SMB311_PROT_ID);
1118                pneg_inbuf->DialectCount = cpu_to_le16(3);
1119                /* SMB 2.1 not included so subtract one dialect from len */
1120                inbuflen = sizeof(*pneg_inbuf) -
1121                                (sizeof(pneg_inbuf->Dialects[0]));
1122        } else if (strcmp(server->vals->version_string,
1123                SMBDEFAULT_VERSION_STRING) == 0) {
1124                pneg_inbuf->Dialects[0] = cpu_to_le16(SMB21_PROT_ID);
1125                pneg_inbuf->Dialects[1] = cpu_to_le16(SMB30_PROT_ID);
1126                pneg_inbuf->Dialects[2] = cpu_to_le16(SMB302_PROT_ID);
1127                pneg_inbuf->Dialects[3] = cpu_to_le16(SMB311_PROT_ID);
1128                pneg_inbuf->DialectCount = cpu_to_le16(4);
1129                /* structure is big enough for 4 dialects */
1130                inbuflen = sizeof(*pneg_inbuf);
1131        } else {
1132                /* otherwise specific dialect was requested */
1133                pneg_inbuf->Dialects[0] =
1134                        cpu_to_le16(server->vals->protocol_id);
1135                pneg_inbuf->DialectCount = cpu_to_le16(1);
1136                /* structure is big enough for 3 dialects, sending only 1 */
1137                inbuflen = sizeof(*pneg_inbuf) -
1138                                sizeof(pneg_inbuf->Dialects[0]) * 2;
1139        }
1140
1141        rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
1142                FSCTL_VALIDATE_NEGOTIATE_INFO, true /* is_fsctl */,
1143                (char *)pneg_inbuf, inbuflen, CIFSMaxBufSize,
1144                (char **)&pneg_rsp, &rsplen);
1145        if (rc == -EOPNOTSUPP) {
1146                /*
1147                 * Old Windows versions or Netapp SMB server can return
1148                 * not supported error. Client should accept it.
1149                 */
1150                cifs_tcon_dbg(VFS, "Server does not support validate negotiate\n");
1151                rc = 0;
1152                goto out_free_inbuf;
1153        } else if (rc != 0) {
1154                cifs_tcon_dbg(VFS, "validate protocol negotiate failed: %d\n",
1155                              rc);
1156                rc = -EIO;
1157                goto out_free_inbuf;
1158        }
1159
1160        rc = -EIO;
1161        if (rsplen != sizeof(*pneg_rsp)) {
1162                cifs_tcon_dbg(VFS, "Invalid protocol negotiate response size: %d\n",
1163                              rsplen);
1164
1165                /* relax check since Mac returns max bufsize allowed on ioctl */
1166                if (rsplen > CIFSMaxBufSize || rsplen < sizeof(*pneg_rsp))
1167                        goto out_free_rsp;
1168        }
1169
1170        /* check validate negotiate info response matches what we got earlier */
1171        if (pneg_rsp->Dialect != cpu_to_le16(server->dialect))
1172                goto vneg_out;
1173
1174        if (pneg_rsp->SecurityMode != cpu_to_le16(server->sec_mode))
1175                goto vneg_out;
1176
1177        /* do not validate server guid because not saved at negprot time yet */
1178
1179        if ((le32_to_cpu(pneg_rsp->Capabilities) | SMB2_NT_FIND |
1180              SMB2_LARGE_FILES) != server->capabilities)
1181                goto vneg_out;
1182
1183        /* validate negotiate successful */
1184        rc = 0;
1185        cifs_dbg(FYI, "validate negotiate info successful\n");
1186        goto out_free_rsp;
1187
1188vneg_out:
1189        cifs_tcon_dbg(VFS, "protocol revalidation - security settings mismatch\n");
1190out_free_rsp:
1191        kfree(pneg_rsp);
1192out_free_inbuf:
1193        kfree(pneg_inbuf);
1194        return rc;
1195}
1196
1197enum securityEnum
1198smb2_select_sectype(struct TCP_Server_Info *server, enum securityEnum requested)
1199{
1200        switch (requested) {
1201        case Kerberos:
1202        case RawNTLMSSP:
1203                return requested;
1204        case NTLMv2:
1205                return RawNTLMSSP;
1206        case Unspecified:
1207                if (server->sec_ntlmssp &&
1208                        (global_secflags & CIFSSEC_MAY_NTLMSSP))
1209                        return RawNTLMSSP;
1210                if ((server->sec_kerberos || server->sec_mskerberos) &&
1211                        (global_secflags & CIFSSEC_MAY_KRB5))
1212                        return Kerberos;
1213                fallthrough;
1214        default:
1215                return Unspecified;
1216        }
1217}
1218
1219struct SMB2_sess_data {
1220        unsigned int xid;
1221        struct cifs_ses *ses;
1222        struct nls_table *nls_cp;
1223        void (*func)(struct SMB2_sess_data *);
1224        int result;
1225        u64 previous_session;
1226
1227        /* we will send the SMB in three pieces:
1228         * a fixed length beginning part, an optional
1229         * SPNEGO blob (which can be zero length), and a
1230         * last part which will include the strings
1231         * and rest of bcc area. This allows us to avoid
1232         * a large buffer 17K allocation
1233         */
1234        int buf0_type;
1235        struct kvec iov[2];
1236};
1237
1238static int
1239SMB2_sess_alloc_buffer(struct SMB2_sess_data *sess_data)
1240{
1241        int rc;
1242        struct cifs_ses *ses = sess_data->ses;
1243        struct smb2_sess_setup_req *req;
1244        struct TCP_Server_Info *server = cifs_ses_server(ses);
1245        unsigned int total_len;
1246
1247        rc = smb2_plain_req_init(SMB2_SESSION_SETUP, NULL, server,
1248                                 (void **) &req,
1249                                 &total_len);
1250        if (rc)
1251                return rc;
1252
1253        if (sess_data->ses->binding) {
1254                req->sync_hdr.SessionId = sess_data->ses->Suid;
1255                req->sync_hdr.Flags |= SMB2_FLAGS_SIGNED;
1256                req->PreviousSessionId = 0;
1257                req->Flags = SMB2_SESSION_REQ_FLAG_BINDING;
1258        } else {
1259                /* First session, not a reauthenticate */
1260                req->sync_hdr.SessionId = 0;
1261                /*
1262                 * if reconnect, we need to send previous sess id
1263                 * otherwise it is 0
1264                 */
1265                req->PreviousSessionId = sess_data->previous_session;
1266                req->Flags = 0; /* MBZ */
1267        }
1268
1269        /* enough to enable echos and oplocks and one max size write */
1270        req->sync_hdr.CreditRequest = cpu_to_le16(130);
1271
1272        /* only one of SMB2 signing flags may be set in SMB2 request */
1273        if (server->sign)
1274                req->SecurityMode = SMB2_NEGOTIATE_SIGNING_REQUIRED;
1275        else if (global_secflags & CIFSSEC_MAY_SIGN) /* one flag unlike MUST_ */
1276                req->SecurityMode = SMB2_NEGOTIATE_SIGNING_ENABLED;
1277        else
1278                req->SecurityMode = 0;
1279
1280#ifdef CONFIG_CIFS_DFS_UPCALL
1281        req->Capabilities = cpu_to_le32(SMB2_GLOBAL_CAP_DFS);
1282#else
1283        req->Capabilities = 0;
1284#endif /* DFS_UPCALL */
1285
1286        req->Channel = 0; /* MBZ */
1287
1288        sess_data->iov[0].iov_base = (char *)req;
1289        /* 1 for pad */
1290        sess_data->iov[0].iov_len = total_len - 1;
1291        /*
1292         * This variable will be used to clear the buffer
1293         * allocated above in case of any error in the calling function.
1294         */
1295        sess_data->buf0_type = CIFS_SMALL_BUFFER;
1296
1297        return 0;
1298}
1299
1300static void
1301SMB2_sess_free_buffer(struct SMB2_sess_data *sess_data)
1302{
1303        free_rsp_buf(sess_data->buf0_type, sess_data->iov[0].iov_base);
1304        sess_data->buf0_type = CIFS_NO_BUFFER;
1305}
1306
1307static int
1308SMB2_sess_sendreceive(struct SMB2_sess_data *sess_data)
1309{
1310        int rc;
1311        struct smb_rqst rqst;
1312        struct smb2_sess_setup_req *req = sess_data->iov[0].iov_base;
1313        struct kvec rsp_iov = { NULL, 0 };
1314
1315        /* Testing shows that buffer offset must be at location of Buffer[0] */
1316        req->SecurityBufferOffset =
1317                cpu_to_le16(sizeof(struct smb2_sess_setup_req) - 1 /* pad */);
1318        req->SecurityBufferLength = cpu_to_le16(sess_data->iov[1].iov_len);
1319
1320        memset(&rqst, 0, sizeof(struct smb_rqst));
1321        rqst.rq_iov = sess_data->iov;
1322        rqst.rq_nvec = 2;
1323
1324        /* BB add code to build os and lm fields */
1325        rc = cifs_send_recv(sess_data->xid, sess_data->ses,
1326                            cifs_ses_server(sess_data->ses),
1327                            &rqst,
1328                            &sess_data->buf0_type,
1329                            CIFS_LOG_ERROR | CIFS_SESS_OP, &rsp_iov);
1330        cifs_small_buf_release(sess_data->iov[0].iov_base);
1331        memcpy(&sess_data->iov[0], &rsp_iov, sizeof(struct kvec));
1332
1333        return rc;
1334}
1335
1336static int
1337SMB2_sess_establish_session(struct SMB2_sess_data *sess_data)
1338{
1339        int rc = 0;
1340        struct cifs_ses *ses = sess_data->ses;
1341        struct TCP_Server_Info *server = cifs_ses_server(ses);
1342
1343        mutex_lock(&server->srv_mutex);
1344        if (server->ops->generate_signingkey) {
1345                rc = server->ops->generate_signingkey(ses);
1346                if (rc) {
1347                        cifs_dbg(FYI,
1348                                "SMB3 session key generation failed\n");
1349                        mutex_unlock(&server->srv_mutex);
1350                        return rc;
1351                }
1352        }
1353        if (!server->session_estab) {
1354                server->sequence_number = 0x2;
1355                server->session_estab = true;
1356        }
1357        mutex_unlock(&server->srv_mutex);
1358
1359        cifs_dbg(FYI, "SMB2/3 session established successfully\n");
1360        /* keep existing ses state if binding */
1361        if (!ses->binding) {
1362                spin_lock(&GlobalMid_Lock);
1363                ses->status = CifsGood;
1364                ses->need_reconnect = false;
1365                spin_unlock(&GlobalMid_Lock);
1366        }
1367
1368        return rc;
1369}
1370
1371#ifdef CONFIG_CIFS_UPCALL
1372static void
1373SMB2_auth_kerberos(struct SMB2_sess_data *sess_data)
1374{
1375        int rc;
1376        struct cifs_ses *ses = sess_data->ses;
1377        struct cifs_spnego_msg *msg;
1378        struct key *spnego_key = NULL;
1379        struct smb2_sess_setup_rsp *rsp = NULL;
1380
1381        rc = SMB2_sess_alloc_buffer(sess_data);
1382        if (rc)
1383                goto out;
1384
1385        spnego_key = cifs_get_spnego_key(ses);
1386        if (IS_ERR(spnego_key)) {
1387                rc = PTR_ERR(spnego_key);
1388                if (rc == -ENOKEY)
1389                        cifs_dbg(VFS, "Verify user has a krb5 ticket and keyutils is installed\n");
1390                spnego_key = NULL;
1391                goto out;
1392        }
1393
1394        msg = spnego_key->payload.data[0];
1395        /*
1396         * check version field to make sure that cifs.upcall is
1397         * sending us a response in an expected form
1398         */
1399        if (msg->version != CIFS_SPNEGO_UPCALL_VERSION) {
1400                cifs_dbg(VFS, "bad cifs.upcall version. Expected %d got %d\n",
1401                         CIFS_SPNEGO_UPCALL_VERSION, msg->version);
1402                rc = -EKEYREJECTED;
1403                goto out_put_spnego_key;
1404        }
1405
1406        /* keep session key if binding */
1407        if (!ses->binding) {
1408                ses->auth_key.response = kmemdup(msg->data, msg->sesskey_len,
1409                                                 GFP_KERNEL);
1410                if (!ses->auth_key.response) {
1411                        cifs_dbg(VFS, "Kerberos can't allocate (%u bytes) memory\n",
1412                                 msg->sesskey_len);
1413                        rc = -ENOMEM;
1414                        goto out_put_spnego_key;
1415                }
1416                ses->auth_key.len = msg->sesskey_len;
1417        }
1418
1419        sess_data->iov[1].iov_base = msg->data + msg->sesskey_len;
1420        sess_data->iov[1].iov_len = msg->secblob_len;
1421
1422        rc = SMB2_sess_sendreceive(sess_data);
1423        if (rc)
1424                goto out_put_spnego_key;
1425
1426        rsp = (struct smb2_sess_setup_rsp *)sess_data->iov[0].iov_base;
1427        /* keep session id and flags if binding */
1428        if (!ses->binding) {
1429                ses->Suid = rsp->sync_hdr.SessionId;
1430                ses->session_flags = le16_to_cpu(rsp->SessionFlags);
1431        }
1432
1433        rc = SMB2_sess_establish_session(sess_data);
1434out_put_spnego_key:
1435        key_invalidate(spnego_key);
1436        key_put(spnego_key);
1437out:
1438        sess_data->result = rc;
1439        sess_data->func = NULL;
1440        SMB2_sess_free_buffer(sess_data);
1441}
1442#else
1443static void
1444SMB2_auth_kerberos(struct SMB2_sess_data *sess_data)
1445{
1446        cifs_dbg(VFS, "Kerberos negotiated but upcall support disabled!\n");
1447        sess_data->result = -EOPNOTSUPP;
1448        sess_data->func = NULL;
1449}
1450#endif
1451
1452static void
1453SMB2_sess_auth_rawntlmssp_authenticate(struct SMB2_sess_data *sess_data);
1454
1455static void
1456SMB2_sess_auth_rawntlmssp_negotiate(struct SMB2_sess_data *sess_data)
1457{
1458        int rc;
1459        struct cifs_ses *ses = sess_data->ses;
1460        struct smb2_sess_setup_rsp *rsp = NULL;
1461        char *ntlmssp_blob = NULL;
1462        bool use_spnego = false; /* else use raw ntlmssp */
1463        u16 blob_length = 0;
1464
1465        /*
1466         * If memory allocation is successful, caller of this function
1467         * frees it.
1468         */
1469        ses->ntlmssp = kmalloc(sizeof(struct ntlmssp_auth), GFP_KERNEL);
1470        if (!ses->ntlmssp) {
1471                rc = -ENOMEM;
1472                goto out_err;
1473        }
1474        ses->ntlmssp->sesskey_per_smbsess = true;
1475
1476        rc = SMB2_sess_alloc_buffer(sess_data);
1477        if (rc)
1478                goto out_err;
1479
1480        ntlmssp_blob = kmalloc(sizeof(struct _NEGOTIATE_MESSAGE),
1481                               GFP_KERNEL);
1482        if (ntlmssp_blob == NULL) {
1483                rc = -ENOMEM;
1484                goto out;
1485        }
1486
1487        build_ntlmssp_negotiate_blob(ntlmssp_blob, ses);
1488        if (use_spnego) {
1489                /* BB eventually need to add this */
1490                cifs_dbg(VFS, "spnego not supported for SMB2 yet\n");
1491                rc = -EOPNOTSUPP;
1492                goto out;
1493        } else {
1494                blob_length = sizeof(struct _NEGOTIATE_MESSAGE);
1495                /* with raw NTLMSSP we don't encapsulate in SPNEGO */
1496        }
1497        sess_data->iov[1].iov_base = ntlmssp_blob;
1498        sess_data->iov[1].iov_len = blob_length;
1499
1500        rc = SMB2_sess_sendreceive(sess_data);
1501        rsp = (struct smb2_sess_setup_rsp *)sess_data->iov[0].iov_base;
1502
1503        /* If true, rc here is expected and not an error */
1504        if (sess_data->buf0_type != CIFS_NO_BUFFER &&
1505                rsp->sync_hdr.Status == STATUS_MORE_PROCESSING_REQUIRED)
1506                rc = 0;
1507
1508        if (rc)
1509                goto out;
1510
1511        if (offsetof(struct smb2_sess_setup_rsp, Buffer) !=
1512                        le16_to_cpu(rsp->SecurityBufferOffset)) {
1513                cifs_dbg(VFS, "Invalid security buffer offset %d\n",
1514                        le16_to_cpu(rsp->SecurityBufferOffset));
1515                rc = -EIO;
1516                goto out;
1517        }
1518        rc = decode_ntlmssp_challenge(rsp->Buffer,
1519                        le16_to_cpu(rsp->SecurityBufferLength), ses);
1520        if (rc)
1521                goto out;
1522
1523        cifs_dbg(FYI, "rawntlmssp session setup challenge phase\n");
1524
1525        /* keep existing ses id and flags if binding */
1526        if (!ses->binding) {
1527                ses->Suid = rsp->sync_hdr.SessionId;
1528                ses->session_flags = le16_to_cpu(rsp->SessionFlags);
1529        }
1530
1531out:
1532        kfree(ntlmssp_blob);
1533        SMB2_sess_free_buffer(sess_data);
1534        if (!rc) {
1535                sess_data->result = 0;
1536                sess_data->func = SMB2_sess_auth_rawntlmssp_authenticate;
1537                return;
1538        }
1539out_err:
1540        kfree(ses->ntlmssp);
1541        ses->ntlmssp = NULL;
1542        sess_data->result = rc;
1543        sess_data->func = NULL;
1544}
1545
1546static void
1547SMB2_sess_auth_rawntlmssp_authenticate(struct SMB2_sess_data *sess_data)
1548{
1549        int rc;
1550        struct cifs_ses *ses = sess_data->ses;
1551        struct smb2_sess_setup_req *req;
1552        struct smb2_sess_setup_rsp *rsp = NULL;
1553        unsigned char *ntlmssp_blob = NULL;
1554        bool use_spnego = false; /* else use raw ntlmssp */
1555        u16 blob_length = 0;
1556
1557        rc = SMB2_sess_alloc_buffer(sess_data);
1558        if (rc)
1559                goto out;
1560
1561        req = (struct smb2_sess_setup_req *) sess_data->iov[0].iov_base;
1562        req->sync_hdr.SessionId = ses->Suid;
1563
1564        rc = build_ntlmssp_auth_blob(&ntlmssp_blob, &blob_length, ses,
1565                                        sess_data->nls_cp);
1566        if (rc) {
1567                cifs_dbg(FYI, "build_ntlmssp_auth_blob failed %d\n", rc);
1568                goto out;
1569        }
1570
1571        if (use_spnego) {
1572                /* BB eventually need to add this */
1573                cifs_dbg(VFS, "spnego not supported for SMB2 yet\n");
1574                rc = -EOPNOTSUPP;
1575                goto out;
1576        }
1577        sess_data->iov[1].iov_base = ntlmssp_blob;
1578        sess_data->iov[1].iov_len = blob_length;
1579
1580        rc = SMB2_sess_sendreceive(sess_data);
1581        if (rc)
1582                goto out;
1583
1584        rsp = (struct smb2_sess_setup_rsp *)sess_data->iov[0].iov_base;
1585
1586        /* keep existing ses id and flags if binding */
1587        if (!ses->binding) {
1588                ses->Suid = rsp->sync_hdr.SessionId;
1589                ses->session_flags = le16_to_cpu(rsp->SessionFlags);
1590        }
1591
1592        rc = SMB2_sess_establish_session(sess_data);
1593#ifdef CONFIG_CIFS_DEBUG_DUMP_KEYS
1594        if (ses->server->dialect < SMB30_PROT_ID) {
1595                cifs_dbg(VFS, "%s: dumping generated SMB2 session keys\n", __func__);
1596                /*
1597                 * The session id is opaque in terms of endianness, so we can't
1598                 * print it as a long long. we dump it as we got it on the wire
1599                 */
1600                cifs_dbg(VFS, "Session Id    %*ph\n", (int)sizeof(ses->Suid),
1601                         &ses->Suid);
1602                cifs_dbg(VFS, "Session Key   %*ph\n",
1603                         SMB2_NTLMV2_SESSKEY_SIZE, ses->auth_key.response);
1604                cifs_dbg(VFS, "Signing Key   %*ph\n",
1605                         SMB3_SIGN_KEY_SIZE, ses->auth_key.response);
1606        }
1607#endif
1608out:
1609        kfree(ntlmssp_blob);
1610        SMB2_sess_free_buffer(sess_data);
1611        kfree(ses->ntlmssp);
1612        ses->ntlmssp = NULL;
1613        sess_data->result = rc;
1614        sess_data->func = NULL;
1615}
1616
1617static int
1618SMB2_select_sec(struct cifs_ses *ses, struct SMB2_sess_data *sess_data)
1619{
1620        int type;
1621
1622        type = smb2_select_sectype(cifs_ses_server(ses), ses->sectype);
1623        cifs_dbg(FYI, "sess setup type %d\n", type);
1624        if (type == Unspecified) {
1625                cifs_dbg(VFS, "Unable to select appropriate authentication method!\n");
1626                return -EINVAL;
1627        }
1628
1629        switch (type) {
1630        case Kerberos:
1631                sess_data->func = SMB2_auth_kerberos;
1632                break;
1633        case RawNTLMSSP:
1634                sess_data->func = SMB2_sess_auth_rawntlmssp_negotiate;
1635                break;
1636        default:
1637                cifs_dbg(VFS, "secType %d not supported!\n", type);
1638                return -EOPNOTSUPP;
1639        }
1640
1641        return 0;
1642}
1643
1644int
1645SMB2_sess_setup(const unsigned int xid, struct cifs_ses *ses,
1646                const struct nls_table *nls_cp)
1647{
1648        int rc = 0;
1649        struct TCP_Server_Info *server = cifs_ses_server(ses);
1650        struct SMB2_sess_data *sess_data;
1651
1652        cifs_dbg(FYI, "Session Setup\n");
1653
1654        if (!server) {
1655                WARN(1, "%s: server is NULL!\n", __func__);
1656                return -EIO;
1657        }
1658
1659        sess_data = kzalloc(sizeof(struct SMB2_sess_data), GFP_KERNEL);
1660        if (!sess_data)
1661                return -ENOMEM;
1662
1663        rc = SMB2_select_sec(ses, sess_data);
1664        if (rc)
1665                goto out;
1666        sess_data->xid = xid;
1667        sess_data->ses = ses;
1668        sess_data->buf0_type = CIFS_NO_BUFFER;
1669        sess_data->nls_cp = (struct nls_table *) nls_cp;
1670        sess_data->previous_session = ses->Suid;
1671
1672        /*
1673         * Initialize the session hash with the server one.
1674         */
1675        memcpy(ses->preauth_sha_hash, server->preauth_sha_hash,
1676               SMB2_PREAUTH_HASH_SIZE);
1677
1678        while (sess_data->func)
1679                sess_data->func(sess_data);
1680
1681        if ((ses->session_flags & SMB2_SESSION_FLAG_IS_GUEST) && (ses->sign))
1682                cifs_server_dbg(VFS, "signing requested but authenticated as guest\n");
1683        rc = sess_data->result;
1684out:
1685        kfree(sess_data);
1686        return rc;
1687}
1688
1689int
1690SMB2_logoff(const unsigned int xid, struct cifs_ses *ses)
1691{
1692        struct smb_rqst rqst;
1693        struct smb2_logoff_req *req; /* response is also trivial struct */
1694        int rc = 0;
1695        struct TCP_Server_Info *server;
1696        int flags = 0;
1697        unsigned int total_len;
1698        struct kvec iov[1];
1699        struct kvec rsp_iov;
1700        int resp_buf_type;
1701
1702        cifs_dbg(FYI, "disconnect session %p\n", ses);
1703
1704        if (ses && (ses->server))
1705                server = ses->server;
1706        else
1707                return -EIO;
1708
1709        /* no need to send SMB logoff if uid already closed due to reconnect */
1710        if (ses->need_reconnect)
1711                goto smb2_session_already_dead;
1712
1713        rc = smb2_plain_req_init(SMB2_LOGOFF, NULL, ses->server,
1714                                 (void **) &req, &total_len);
1715        if (rc)
1716                return rc;
1717
1718         /* since no tcon, smb2_init can not do this, so do here */
1719        req->sync_hdr.SessionId = ses->Suid;
1720
1721        if (ses->session_flags & SMB2_SESSION_FLAG_ENCRYPT_DATA)
1722                flags |= CIFS_TRANSFORM_REQ;
1723        else if (server->sign)
1724                req->sync_hdr.Flags |= SMB2_FLAGS_SIGNED;
1725
1726        flags |= CIFS_NO_RSP_BUF;
1727
1728        iov[0].iov_base = (char *)req;
1729        iov[0].iov_len = total_len;
1730
1731        memset(&rqst, 0, sizeof(struct smb_rqst));
1732        rqst.rq_iov = iov;
1733        rqst.rq_nvec = 1;
1734
1735        rc = cifs_send_recv(xid, ses, ses->server,
1736                            &rqst, &resp_buf_type, flags, &rsp_iov);
1737        cifs_small_buf_release(req);
1738        /*
1739         * No tcon so can't do
1740         * cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_fail[SMB2...]);
1741         */
1742
1743smb2_session_already_dead:
1744        return rc;
1745}
1746
1747static inline void cifs_stats_fail_inc(struct cifs_tcon *tcon, uint16_t code)
1748{
1749        cifs_stats_inc(&tcon->stats.smb2_stats.smb2_com_failed[code]);
1750}
1751
1752#define MAX_SHARENAME_LENGTH (255 /* server */ + 80 /* share */ + 1 /* NULL */)
1753
1754/* These are similar values to what Windows uses */
1755static inline void init_copy_chunk_defaults(struct cifs_tcon *tcon)
1756{
1757        tcon->max_chunks = 256;
1758        tcon->max_bytes_chunk = 1048576;
1759        tcon->max_bytes_copy = 16777216;
1760}
1761
1762int
1763SMB2_tcon(const unsigned int xid, struct cifs_ses *ses, const char *tree,
1764          struct cifs_tcon *tcon, const struct nls_table *cp)
1765{
1766        struct smb_rqst rqst;
1767        struct smb2_tree_connect_req *req;
1768        struct smb2_tree_connect_rsp *rsp = NULL;
1769        struct kvec iov[2];
1770        struct kvec rsp_iov = { NULL, 0 };
1771        int rc = 0;
1772        int resp_buftype;
1773        int unc_path_len;
1774        __le16 *unc_path = NULL;
1775        int flags = 0;
1776        unsigned int total_len;
1777        struct TCP_Server_Info *server;
1778
1779        /* always use master channel */
1780        server = ses->server;
1781
1782        cifs_dbg(FYI, "TCON\n");
1783
1784        if (!server || !tree)
1785                return -EIO;
1786
1787        unc_path = kmalloc(MAX_SHARENAME_LENGTH * 2, GFP_KERNEL);
1788        if (unc_path == NULL)
1789                return -ENOMEM;
1790
1791        unc_path_len = cifs_strtoUTF16(unc_path, tree, strlen(tree), cp) + 1;
1792        unc_path_len *= 2;
1793        if (unc_path_len < 2) {
1794                kfree(unc_path);
1795                return -EINVAL;
1796        }
1797
1798        /* SMB2 TREE_CONNECT request must be called with TreeId == 0 */
1799        tcon->tid = 0;
1800        atomic_set(&tcon->num_remote_opens, 0);
1801        rc = smb2_plain_req_init(SMB2_TREE_CONNECT, tcon, server,
1802                                 (void **) &req, &total_len);
1803        if (rc) {
1804                kfree(unc_path);
1805                return rc;
1806        }
1807
1808        if (smb3_encryption_required(tcon))
1809                flags |= CIFS_TRANSFORM_REQ;
1810
1811        iov[0].iov_base = (char *)req;
1812        /* 1 for pad */
1813        iov[0].iov_len = total_len - 1;
1814
1815        /* Testing shows that buffer offset must be at location of Buffer[0] */
1816        req->PathOffset = cpu_to_le16(sizeof(struct smb2_tree_connect_req)
1817                        - 1 /* pad */);
1818        req->PathLength = cpu_to_le16(unc_path_len - 2);
1819        iov[1].iov_base = unc_path;
1820        iov[1].iov_len = unc_path_len;
1821
1822        /*
1823         * 3.11 tcon req must be signed if not encrypted. See MS-SMB2 3.2.4.1.1
1824         * unless it is guest or anonymous user. See MS-SMB2 3.2.5.3.1
1825         * (Samba servers don't always set the flag so also check if null user)
1826         */
1827        if ((server->dialect == SMB311_PROT_ID) &&
1828            !smb3_encryption_required(tcon) &&
1829            !(ses->session_flags &
1830                    (SMB2_SESSION_FLAG_IS_GUEST|SMB2_SESSION_FLAG_IS_NULL)) &&
1831            ((ses->user_name != NULL) || (ses->sectype == Kerberos)))
1832                req->sync_hdr.Flags |= SMB2_FLAGS_SIGNED;
1833
1834        memset(&rqst, 0, sizeof(struct smb_rqst));
1835        rqst.rq_iov = iov;
1836        rqst.rq_nvec = 2;
1837
1838        /* Need 64 for max size write so ask for more in case not there yet */
1839        req->sync_hdr.CreditRequest = cpu_to_le16(64);
1840
1841        rc = cifs_send_recv(xid, ses, server,
1842                            &rqst, &resp_buftype, flags, &rsp_iov);
1843        cifs_small_buf_release(req);
1844        rsp = (struct smb2_tree_connect_rsp *)rsp_iov.iov_base;
1845        trace_smb3_tcon(xid, tcon->tid, ses->Suid, tree, rc);
1846        if (rc != 0) {
1847                cifs_stats_fail_inc(tcon, SMB2_TREE_CONNECT_HE);
1848                tcon->need_reconnect = true;
1849                goto tcon_error_exit;
1850        }
1851
1852        switch (rsp->ShareType) {
1853        case SMB2_SHARE_TYPE_DISK:
1854                cifs_dbg(FYI, "connection to disk share\n");
1855                break;
1856        case SMB2_SHARE_TYPE_PIPE:
1857                tcon->pipe = true;
1858                cifs_dbg(FYI, "connection to pipe share\n");
1859                break;
1860        case SMB2_SHARE_TYPE_PRINT:
1861                tcon->print = true;
1862                cifs_dbg(FYI, "connection to printer\n");
1863                break;
1864        default:
1865                cifs_server_dbg(VFS, "unknown share type %d\n", rsp->ShareType);
1866                rc = -EOPNOTSUPP;
1867                goto tcon_error_exit;
1868        }
1869
1870        tcon->share_flags = le32_to_cpu(rsp->ShareFlags);
1871        tcon->capabilities = rsp->Capabilities; /* we keep caps little endian */
1872        tcon->maximal_access = le32_to_cpu(rsp->MaximalAccess);
1873        tcon->tidStatus = CifsGood;
1874        tcon->need_reconnect = false;
1875        tcon->tid = rsp->sync_hdr.TreeId;
1876        strlcpy(tcon->treeName, tree, sizeof(tcon->treeName));
1877
1878        if ((rsp->Capabilities & SMB2_SHARE_CAP_DFS) &&
1879            ((tcon->share_flags & SHI1005_FLAGS_DFS) == 0))
1880                cifs_tcon_dbg(VFS, "DFS capability contradicts DFS flag\n");
1881
1882        if (tcon->seal &&
1883            !(server->capabilities & SMB2_GLOBAL_CAP_ENCRYPTION))
1884                cifs_tcon_dbg(VFS, "Encryption is requested but not supported\n");
1885
1886        init_copy_chunk_defaults(tcon);
1887        if (server->ops->validate_negotiate)
1888                rc = server->ops->validate_negotiate(xid, tcon);
1889tcon_exit:
1890
1891        free_rsp_buf(resp_buftype, rsp);
1892        kfree(unc_path);
1893        return rc;
1894
1895tcon_error_exit:
1896        if (rsp && rsp->sync_hdr.Status == STATUS_BAD_NETWORK_NAME) {
1897                cifs_tcon_dbg(VFS, "BAD_NETWORK_NAME: %s\n", tree);
1898        }
1899        goto tcon_exit;
1900}
1901
1902int
1903SMB2_tdis(const unsigned int xid, struct cifs_tcon *tcon)
1904{
1905        struct smb_rqst rqst;
1906        struct smb2_tree_disconnect_req *req; /* response is trivial */
1907        int rc = 0;
1908        struct cifs_ses *ses = tcon->ses;
1909        int flags = 0;
1910        unsigned int total_len;
1911        struct kvec iov[1];
1912        struct kvec rsp_iov;
1913        int resp_buf_type;
1914
1915        cifs_dbg(FYI, "Tree Disconnect\n");
1916
1917        if (!ses || !(ses->server))
1918                return -EIO;
1919
1920        if ((tcon->need_reconnect) || (tcon->ses->need_reconnect))
1921                return 0;
1922
1923        close_cached_dir_lease(&tcon->crfid);
1924
1925        rc = smb2_plain_req_init(SMB2_TREE_DISCONNECT, tcon, ses->server,
1926                                 (void **) &req,
1927                                 &total_len);
1928        if (rc)
1929                return rc;
1930
1931        if (smb3_encryption_required(tcon))
1932                flags |= CIFS_TRANSFORM_REQ;
1933
1934        flags |= CIFS_NO_RSP_BUF;
1935
1936        iov[0].iov_base = (char *)req;
1937        iov[0].iov_len = total_len;
1938
1939        memset(&rqst, 0, sizeof(struct smb_rqst));
1940        rqst.rq_iov = iov;
1941        rqst.rq_nvec = 1;
1942
1943        rc = cifs_send_recv(xid, ses, ses->server,
1944                            &rqst, &resp_buf_type, flags, &rsp_iov);
1945        cifs_small_buf_release(req);
1946        if (rc)
1947                cifs_stats_fail_inc(tcon, SMB2_TREE_DISCONNECT_HE);
1948
1949        return rc;
1950}
1951
1952
1953static struct create_durable *
1954create_durable_buf(void)
1955{
1956        struct create_durable *buf;
1957
1958        buf = kzalloc(sizeof(struct create_durable), GFP_KERNEL);
1959        if (!buf)
1960                return NULL;
1961
1962        buf->ccontext.DataOffset = cpu_to_le16(offsetof
1963                                        (struct create_durable, Data));
1964        buf->ccontext.DataLength = cpu_to_le32(16);
1965        buf->ccontext.NameOffset = cpu_to_le16(offsetof
1966                                (struct create_durable, Name));
1967        buf->ccontext.NameLength = cpu_to_le16(4);
1968        /* SMB2_CREATE_DURABLE_HANDLE_REQUEST is "DHnQ" */
1969        buf->Name[0] = 'D';
1970        buf->Name[1] = 'H';
1971        buf->Name[2] = 'n';
1972        buf->Name[3] = 'Q';
1973        return buf;
1974}
1975
1976static struct create_durable *
1977create_reconnect_durable_buf(struct cifs_fid *fid)
1978{
1979        struct create_durable *buf;
1980
1981        buf = kzalloc(sizeof(struct create_durable), GFP_KERNEL);
1982        if (!buf)
1983                return NULL;
1984
1985        buf->ccontext.DataOffset = cpu_to_le16(offsetof
1986                                        (struct create_durable, Data));
1987        buf->ccontext.DataLength = cpu_to_le32(16);
1988        buf->ccontext.NameOffset = cpu_to_le16(offsetof
1989                                (struct create_durable, Name));
1990        buf->ccontext.NameLength = cpu_to_le16(4);
1991        buf->Data.Fid.PersistentFileId = fid->persistent_fid;
1992        buf->Data.Fid.VolatileFileId = fid->volatile_fid;
1993        /* SMB2_CREATE_DURABLE_HANDLE_RECONNECT is "DHnC" */
1994        buf->Name[0] = 'D';
1995        buf->Name[1] = 'H';
1996        buf->Name[2] = 'n';
1997        buf->Name[3] = 'C';
1998        return buf;
1999}
2000
2001static void
2002parse_query_id_ctxt(struct create_context *cc, struct smb2_file_all_info *buf)
2003{
2004        struct create_on_disk_id *pdisk_id = (struct create_on_disk_id *)cc;
2005
2006        cifs_dbg(FYI, "parse query id context 0x%llx 0x%llx\n",
2007                pdisk_id->DiskFileId, pdisk_id->VolumeId);
2008        buf->IndexNumber = pdisk_id->DiskFileId;
2009}
2010
2011static void
2012parse_posix_ctxt(struct create_context *cc, struct smb2_file_all_info *info,
2013                 struct create_posix_rsp *posix)
2014{
2015        int sid_len;
2016        u8 *beg = (u8 *)cc + le16_to_cpu(cc->DataOffset);
2017        u8 *end = beg + le32_to_cpu(cc->DataLength);
2018        u8 *sid;
2019
2020        memset(posix, 0, sizeof(*posix));
2021
2022        posix->nlink = le32_to_cpu(*(__le32 *)(beg + 0));
2023        posix->reparse_tag = le32_to_cpu(*(__le32 *)(beg + 4));
2024        posix->mode = le32_to_cpu(*(__le32 *)(beg + 8));
2025
2026        sid = beg + 12;
2027        sid_len = posix_info_sid_size(sid, end);
2028        if (sid_len < 0) {
2029                cifs_dbg(VFS, "bad owner sid in posix create response\n");
2030                return;
2031        }
2032        memcpy(&posix->owner, sid, sid_len);
2033
2034        sid = sid + sid_len;
2035        sid_len = posix_info_sid_size(sid, end);
2036        if (sid_len < 0) {
2037                cifs_dbg(VFS, "bad group sid in posix create response\n");
2038                return;
2039        }
2040        memcpy(&posix->group, sid, sid_len);
2041
2042        cifs_dbg(FYI, "nlink=%d mode=%o reparse_tag=%x\n",
2043                 posix->nlink, posix->mode, posix->reparse_tag);
2044}
2045
2046void
2047smb2_parse_contexts(struct TCP_Server_Info *server,
2048                    struct smb2_create_rsp *rsp,
2049                    unsigned int *epoch, char *lease_key, __u8 *oplock,
2050                    struct smb2_file_all_info *buf,
2051                    struct create_posix_rsp *posix)
2052{
2053        char *data_offset;
2054        struct create_context *cc;
2055        unsigned int next;
2056        unsigned int remaining;
2057        char *name;
2058        static const char smb3_create_tag_posix[] = {
2059                0x93, 0xAD, 0x25, 0x50, 0x9C,
2060                0xB4, 0x11, 0xE7, 0xB4, 0x23, 0x83,
2061                0xDE, 0x96, 0x8B, 0xCD, 0x7C
2062        };
2063
2064        *oplock = 0;
2065        data_offset = (char *)rsp + le32_to_cpu(rsp->CreateContextsOffset);
2066        remaining = le32_to_cpu(rsp->CreateContextsLength);
2067        cc = (struct create_context *)data_offset;
2068
2069        /* Initialize inode number to 0 in case no valid data in qfid context */
2070        if (buf)
2071                buf->IndexNumber = 0;
2072
2073        while (remaining >= sizeof(struct create_context)) {
2074                name = le16_to_cpu(cc->NameOffset) + (char *)cc;
2075                if (le16_to_cpu(cc->NameLength) == 4 &&
2076                    strncmp(name, SMB2_CREATE_REQUEST_LEASE, 4) == 0)
2077                        *oplock = server->ops->parse_lease_buf(cc, epoch,
2078                                                           lease_key);
2079                else if (buf && (le16_to_cpu(cc->NameLength) == 4) &&
2080                    strncmp(name, SMB2_CREATE_QUERY_ON_DISK_ID, 4) == 0)
2081                        parse_query_id_ctxt(cc, buf);
2082                else if ((le16_to_cpu(cc->NameLength) == 16)) {
2083                        if (posix &&
2084                            memcmp(name, smb3_create_tag_posix, 16) == 0)
2085                                parse_posix_ctxt(cc, buf, posix);
2086                }
2087                /* else {
2088                        cifs_dbg(FYI, "Context not matched with len %d\n",
2089                                le16_to_cpu(cc->NameLength));
2090                        cifs_dump_mem("Cctxt name: ", name, 4);
2091                } */
2092
2093                next = le32_to_cpu(cc->Next);
2094                if (!next)
2095                        break;
2096                remaining -= next;
2097                cc = (struct create_context *)((char *)cc + next);
2098        }
2099
2100        if (rsp->OplockLevel != SMB2_OPLOCK_LEVEL_LEASE)
2101                *oplock = rsp->OplockLevel;
2102
2103        return;
2104}
2105
2106static int
2107add_lease_context(struct TCP_Server_Info *server, struct kvec *iov,
2108                  unsigned int *num_iovec, u8 *lease_key, __u8 *oplock)
2109{
2110        struct smb2_create_req *req = iov[0].iov_base;
2111        unsigned int num = *num_iovec;
2112
2113        iov[num].iov_base = server->ops->create_lease_buf(lease_key, *oplock);
2114        if (iov[num].iov_base == NULL)
2115                return -ENOMEM;
2116        iov[num].iov_len = server->vals->create_lease_size;
2117        req->RequestedOplockLevel = SMB2_OPLOCK_LEVEL_LEASE;
2118        if (!req->CreateContextsOffset)
2119                req->CreateContextsOffset = cpu_to_le32(
2120                                sizeof(struct smb2_create_req) +
2121                                iov[num - 1].iov_len);
2122        le32_add_cpu(&req->CreateContextsLength,
2123                     server->vals->create_lease_size);
2124        *num_iovec = num + 1;
2125        return 0;
2126}
2127
2128static struct create_durable_v2 *
2129create_durable_v2_buf(struct cifs_open_parms *oparms)
2130{
2131        struct cifs_fid *pfid = oparms->fid;
2132        struct create_durable_v2 *buf;
2133
2134        buf = kzalloc(sizeof(struct create_durable_v2), GFP_KERNEL);
2135        if (!buf)
2136                return NULL;
2137
2138        buf->ccontext.DataOffset = cpu_to_le16(offsetof
2139                                        (struct create_durable_v2, dcontext));
2140        buf->ccontext.DataLength = cpu_to_le32(sizeof(struct durable_context_v2));
2141        buf->ccontext.NameOffset = cpu_to_le16(offsetof
2142                                (struct create_durable_v2, Name));
2143        buf->ccontext.NameLength = cpu_to_le16(4);
2144
2145        /*
2146         * NB: Handle timeout defaults to 0, which allows server to choose
2147         * (most servers default to 120 seconds) and most clients default to 0.
2148         * This can be overridden at mount ("handletimeout=") if the user wants
2149         * a different persistent (or resilient) handle timeout for all opens
2150         * opens on a particular SMB3 mount.
2151         */
2152        buf->dcontext.Timeout = cpu_to_le32(oparms->tcon->handle_timeout);
2153        buf->dcontext.Flags = cpu_to_le32(SMB2_DHANDLE_FLAG_PERSISTENT);
2154        generate_random_uuid(buf->dcontext.CreateGuid);
2155        memcpy(pfid->create_guid, buf->dcontext.CreateGuid, 16);
2156
2157        /* SMB2_CREATE_DURABLE_HANDLE_REQUEST is "DH2Q" */
2158        buf->Name[0] = 'D';
2159        buf->Name[1] = 'H';
2160        buf->Name[2] = '2';
2161        buf->Name[3] = 'Q';
2162        return buf;
2163}
2164
2165static struct create_durable_handle_reconnect_v2 *
2166create_reconnect_durable_v2_buf(struct cifs_fid *fid)
2167{
2168        struct create_durable_handle_reconnect_v2 *buf;
2169
2170        buf = kzalloc(sizeof(struct create_durable_handle_reconnect_v2),
2171                        GFP_KERNEL);
2172        if (!buf)
2173                return NULL;
2174
2175        buf->ccontext.DataOffset =
2176                cpu_to_le16(offsetof(struct create_durable_handle_reconnect_v2,
2177                                     dcontext));
2178        buf->ccontext.DataLength =
2179                cpu_to_le32(sizeof(struct durable_reconnect_context_v2));
2180        buf->ccontext.NameOffset =
2181                cpu_to_le16(offsetof(struct create_durable_handle_reconnect_v2,
2182                            Name));
2183        buf->ccontext.NameLength = cpu_to_le16(4);
2184
2185        buf->dcontext.Fid.PersistentFileId = fid->persistent_fid;
2186        buf->dcontext.Fid.VolatileFileId = fid->volatile_fid;
2187        buf->dcontext.Flags = cpu_to_le32(SMB2_DHANDLE_FLAG_PERSISTENT);
2188        memcpy(buf->dcontext.CreateGuid, fid->create_guid, 16);
2189
2190        /* SMB2_CREATE_DURABLE_HANDLE_RECONNECT_V2 is "DH2C" */
2191        buf->Name[0] = 'D';
2192        buf->Name[1] = 'H';
2193        buf->Name[2] = '2';
2194        buf->Name[3] = 'C';
2195        return buf;
2196}
2197
2198static int
2199add_durable_v2_context(struct kvec *iov, unsigned int *num_iovec,
2200                    struct cifs_open_parms *oparms)
2201{
2202        struct smb2_create_req *req = iov[0].iov_base;
2203        unsigned int num = *num_iovec;
2204
2205        iov[num].iov_base = create_durable_v2_buf(oparms);
2206        if (iov[num].iov_base == NULL)
2207                return -ENOMEM;
2208        iov[num].iov_len = sizeof(struct create_durable_v2);
2209        if (!req->CreateContextsOffset)
2210                req->CreateContextsOffset =
2211                        cpu_to_le32(sizeof(struct smb2_create_req) +
2212                                                                iov[1].iov_len);
2213        le32_add_cpu(&req->CreateContextsLength, sizeof(struct create_durable_v2));
2214        *num_iovec = num + 1;
2215        return 0;
2216}
2217
2218static int
2219add_durable_reconnect_v2_context(struct kvec *iov, unsigned int *num_iovec,
2220                    struct cifs_open_parms *oparms)
2221{
2222        struct smb2_create_req *req = iov[0].iov_base;
2223        unsigned int num = *num_iovec;
2224
2225        /* indicate that we don't need to relock the file */
2226        oparms->reconnect = false;
2227
2228        iov[num].iov_base = create_reconnect_durable_v2_buf(oparms->fid);
2229        if (iov[num].iov_base == NULL)
2230                return -ENOMEM;
2231        iov[num].iov_len = sizeof(struct create_durable_handle_reconnect_v2);
2232        if (!req->CreateContextsOffset)
2233                req->CreateContextsOffset =
2234                        cpu_to_le32(sizeof(struct smb2_create_req) +
2235                                                                iov[1].iov_len);
2236        le32_add_cpu(&req->CreateContextsLength,
2237                        sizeof(struct create_durable_handle_reconnect_v2));
2238        *num_iovec = num + 1;
2239        return 0;
2240}
2241
2242static int
2243add_durable_context(struct kvec *iov, unsigned int *num_iovec,
2244                    struct cifs_open_parms *oparms, bool use_persistent)
2245{
2246        struct smb2_create_req *req = iov[0].iov_base;
2247        unsigned int num = *num_iovec;
2248
2249        if (use_persistent) {
2250                if (oparms->reconnect)
2251                        return add_durable_reconnect_v2_context(iov, num_iovec,
2252                                                                oparms);
2253                else
2254                        return add_durable_v2_context(iov, num_iovec, oparms);
2255        }
2256
2257        if (oparms->reconnect) {
2258                iov[num].iov_base = create_reconnect_durable_buf(oparms->fid);
2259                /* indicate that we don't need to relock the file */
2260                oparms->reconnect = false;
2261        } else
2262                iov[num].iov_base = create_durable_buf();
2263        if (iov[num].iov_base == NULL)
2264                return -ENOMEM;
2265        iov[num].iov_len = sizeof(struct create_durable);
2266        if (!req->CreateContextsOffset)
2267                req->CreateContextsOffset =
2268                        cpu_to_le32(sizeof(struct smb2_create_req) +
2269                                                                iov[1].iov_len);
2270        le32_add_cpu(&req->CreateContextsLength, sizeof(struct create_durable));
2271        *num_iovec = num + 1;
2272        return 0;
2273}
2274
2275/* See MS-SMB2 2.2.13.2.7 */
2276static struct crt_twarp_ctxt *
2277create_twarp_buf(__u64 timewarp)
2278{
2279        struct crt_twarp_ctxt *buf;
2280
2281        buf = kzalloc(sizeof(struct crt_twarp_ctxt), GFP_KERNEL);
2282        if (!buf)
2283                return NULL;
2284
2285        buf->ccontext.DataOffset = cpu_to_le16(offsetof
2286                                        (struct crt_twarp_ctxt, Timestamp));
2287        buf->ccontext.DataLength = cpu_to_le32(8);
2288        buf->ccontext.NameOffset = cpu_to_le16(offsetof
2289                                (struct crt_twarp_ctxt, Name));
2290        buf->ccontext.NameLength = cpu_to_le16(4);
2291        /* SMB2_CREATE_TIMEWARP_TOKEN is "TWrp" */
2292        buf->Name[0] = 'T';
2293        buf->Name[1] = 'W';
2294        buf->Name[2] = 'r';
2295        buf->Name[3] = 'p';
2296        buf->Timestamp = cpu_to_le64(timewarp);
2297        return buf;
2298}
2299
2300/* See MS-SMB2 2.2.13.2.7 */
2301static int
2302add_twarp_context(struct kvec *iov, unsigned int *num_iovec, __u64 timewarp)
2303{
2304        struct smb2_create_req *req = iov[0].iov_base;
2305        unsigned int num = *num_iovec;
2306
2307        iov[num].iov_base = create_twarp_buf(timewarp);
2308        if (iov[num].iov_base == NULL)
2309                return -ENOMEM;
2310        iov[num].iov_len = sizeof(struct crt_twarp_ctxt);
2311        if (!req->CreateContextsOffset)
2312                req->CreateContextsOffset = cpu_to_le32(
2313                                sizeof(struct smb2_create_req) +
2314                                iov[num - 1].iov_len);
2315        le32_add_cpu(&req->CreateContextsLength, sizeof(struct crt_twarp_ctxt));
2316        *num_iovec = num + 1;
2317        return 0;
2318}
2319
2320/* See See http://technet.microsoft.com/en-us/library/hh509017(v=ws.10).aspx */
2321static void setup_owner_group_sids(char *buf)
2322{
2323        struct owner_group_sids *sids = (struct owner_group_sids *)buf;
2324
2325        /* Populate the user ownership fields S-1-5-88-1 */
2326        sids->owner.Revision = 1;
2327        sids->owner.NumAuth = 3;
2328        sids->owner.Authority[5] = 5;
2329        sids->owner.SubAuthorities[0] = cpu_to_le32(88);
2330        sids->owner.SubAuthorities[1] = cpu_to_le32(1);
2331        sids->owner.SubAuthorities[2] = cpu_to_le32(current_fsuid().val);
2332
2333        /* Populate the group ownership fields S-1-5-88-2 */
2334        sids->group.Revision = 1;
2335        sids->group.NumAuth = 3;
2336        sids->group.Authority[5] = 5;
2337        sids->group.SubAuthorities[0] = cpu_to_le32(88);
2338        sids->group.SubAuthorities[1] = cpu_to_le32(2);
2339        sids->group.SubAuthorities[2] = cpu_to_le32(current_fsgid().val);
2340
2341        cifs_dbg(FYI, "owner S-1-5-88-1-%d, group S-1-5-88-2-%d\n", current_fsuid().val, current_fsgid().val);
2342}
2343
2344/* See MS-SMB2 2.2.13.2.2 and MS-DTYP 2.4.6 */
2345static struct crt_sd_ctxt *
2346create_sd_buf(umode_t mode, bool set_owner, unsigned int *len)
2347{
2348        struct crt_sd_ctxt *buf;
2349        __u8 *ptr, *aclptr;
2350        unsigned int acelen, acl_size, ace_count;
2351        unsigned int owner_offset = 0;
2352        unsigned int group_offset = 0;
2353        struct smb3_acl acl;
2354
2355        *len = roundup(sizeof(struct crt_sd_ctxt) + (sizeof(struct cifs_ace) * 4), 8);
2356
2357        if (set_owner) {
2358                /* sizeof(struct owner_group_sids) is already multiple of 8 so no need to round */
2359                *len += sizeof(struct owner_group_sids);
2360        }
2361
2362        buf = kzalloc(*len, GFP_KERNEL);
2363        if (buf == NULL)
2364                return buf;
2365
2366        ptr = (__u8 *)&buf[1];
2367        if (set_owner) {
2368                /* offset fields are from beginning of security descriptor not of create context */
2369                owner_offset = ptr - (__u8 *)&buf->sd;
2370                buf->sd.OffsetOwner = cpu_to_le32(owner_offset);
2371                group_offset = owner_offset + offsetof(struct owner_group_sids, group);
2372                buf->sd.OffsetGroup = cpu_to_le32(group_offset);
2373
2374                setup_owner_group_sids(ptr);
2375                ptr += sizeof(struct owner_group_sids);
2376        } else {
2377                buf->sd.OffsetOwner = 0;
2378                buf->sd.OffsetGroup = 0;
2379        }
2380
2381        buf->ccontext.DataOffset = cpu_to_le16(offsetof(struct crt_sd_ctxt, sd));
2382        buf->ccontext.NameOffset = cpu_to_le16(offsetof(struct crt_sd_ctxt, Name));
2383        buf->ccontext.NameLength = cpu_to_le16(4);
2384        /* SMB2_CREATE_SD_BUFFER_TOKEN is "SecD" */
2385        buf->Name[0] = 'S';
2386        buf->Name[1] = 'e';
2387        buf->Name[2] = 'c';
2388        buf->Name[3] = 'D';
2389        buf->sd.Revision = 1;  /* Must be one see MS-DTYP 2.4.6 */
2390
2391        /*
2392         * ACL is "self relative" ie ACL is stored in contiguous block of memory
2393         * and "DP" ie the DACL is present
2394         */
2395        buf->sd.Control = cpu_to_le16(ACL_CONTROL_SR | ACL_CONTROL_DP);
2396
2397        /* offset owner, group and Sbz1 and SACL are all zero */
2398        buf->sd.OffsetDacl = cpu_to_le32(ptr - (__u8 *)&buf->sd);
2399        /* Ship the ACL for now. we will copy it into buf later. */
2400        aclptr = ptr;
2401        ptr += sizeof(struct cifs_acl);
2402
2403        /* create one ACE to hold the mode embedded in reserved special SID */
2404        acelen = setup_special_mode_ACE((struct cifs_ace *)ptr, (__u64)mode);
2405        ptr += acelen;
2406        acl_size = acelen + sizeof(struct smb3_acl);
2407        ace_count = 1;
2408
2409        if (set_owner) {
2410                /* we do not need to reallocate buffer to add the two more ACEs. plenty of space */
2411                acelen = setup_special_user_owner_ACE((struct cifs_ace *)ptr);
2412                ptr += acelen;
2413                acl_size += acelen;
2414                ace_count += 1;
2415        }
2416
2417        /* and one more ACE to allow access for authenticated users */
2418        acelen = setup_authusers_ACE((struct cifs_ace *)ptr);
2419        ptr += acelen;
2420        acl_size += acelen;
2421        ace_count += 1;
2422
2423        acl.AclRevision = ACL_REVISION; /* See 2.4.4.1 of MS-DTYP */
2424        acl.AclSize = cpu_to_le16(acl_size);
2425        acl.AceCount = cpu_to_le16(ace_count);
2426        memcpy(aclptr, &acl, sizeof(struct cifs_acl));
2427
2428        buf->ccontext.DataLength = cpu_to_le32(ptr - (__u8 *)&buf->sd);
2429        *len = roundup(ptr - (__u8 *)buf, 8);
2430
2431        return buf;
2432}
2433
2434static int
2435add_sd_context(struct kvec *iov, unsigned int *num_iovec, umode_t mode, bool set_owner)
2436{
2437        struct smb2_create_req *req = iov[0].iov_base;
2438        unsigned int num = *num_iovec;
2439        unsigned int len = 0;
2440
2441        iov[num].iov_base = create_sd_buf(mode, set_owner, &len);
2442        if (iov[num].iov_base == NULL)
2443                return -ENOMEM;
2444        iov[num].iov_len = len;
2445        if (!req->CreateContextsOffset)
2446                req->CreateContextsOffset = cpu_to_le32(
2447                                sizeof(struct smb2_create_req) +
2448                                iov[num - 1].iov_len);
2449        le32_add_cpu(&req->CreateContextsLength, len);
2450        *num_iovec = num + 1;
2451        return 0;
2452}
2453
2454static struct crt_query_id_ctxt *
2455create_query_id_buf(void)
2456{
2457        struct crt_query_id_ctxt *buf;
2458
2459        buf = kzalloc(sizeof(struct crt_query_id_ctxt), GFP_KERNEL);
2460        if (!buf)
2461                return NULL;
2462
2463        buf->ccontext.DataOffset = cpu_to_le16(0);
2464        buf->ccontext.DataLength = cpu_to_le32(0);
2465        buf->ccontext.NameOffset = cpu_to_le16(offsetof
2466                                (struct crt_query_id_ctxt, Name));
2467        buf->ccontext.NameLength = cpu_to_le16(4);
2468        /* SMB2_CREATE_QUERY_ON_DISK_ID is "QFid" */
2469        buf->Name[0] = 'Q';
2470        buf->Name[1] = 'F';
2471        buf->Name[2] = 'i';
2472        buf->Name[3] = 'd';
2473        return buf;
2474}
2475
2476/* See MS-SMB2 2.2.13.2.9 */
2477static int
2478add_query_id_context(struct kvec *iov, unsigned int *num_iovec)
2479{
2480        struct smb2_create_req *req = iov[0].iov_base;
2481        unsigned int num = *num_iovec;
2482
2483        iov[num].iov_base = create_query_id_buf();
2484        if (iov[num].iov_base == NULL)
2485                return -ENOMEM;
2486        iov[num].iov_len = sizeof(struct crt_query_id_ctxt);
2487        if (!req->CreateContextsOffset)
2488                req->CreateContextsOffset = cpu_to_le32(
2489                                sizeof(struct smb2_create_req) +
2490                                iov[num - 1].iov_len);
2491        le32_add_cpu(&req->CreateContextsLength, sizeof(struct crt_query_id_ctxt));
2492        *num_iovec = num + 1;
2493        return 0;
2494}
2495
2496static int
2497alloc_path_with_tree_prefix(__le16 **out_path, int *out_size, int *out_len,
2498                            const char *treename, const __le16 *path)
2499{
2500        int treename_len, path_len;
2501        struct nls_table *cp;
2502        const __le16 sep[] = {cpu_to_le16('\\'), cpu_to_le16(0x0000)};
2503
2504        /*
2505         * skip leading "\\"
2506         */
2507        treename_len = strlen(treename);
2508        if (treename_len < 2 || !(treename[0] == '\\' && treename[1] == '\\'))
2509                return -EINVAL;
2510
2511        treename += 2;
2512        treename_len -= 2;
2513
2514        path_len = UniStrnlen((wchar_t *)path, PATH_MAX);
2515
2516        /*
2517         * make room for one path separator between the treename and
2518         * path
2519         */
2520        *out_len = treename_len + 1 + path_len;
2521
2522        /*
2523         * final path needs to be null-terminated UTF16 with a
2524         * size aligned to 8
2525         */
2526
2527        *out_size = roundup((*out_len+1)*2, 8);
2528        *out_path = kzalloc(*out_size, GFP_KERNEL);
2529        if (!*out_path)
2530                return -ENOMEM;
2531
2532        cp = load_nls_default();
2533        cifs_strtoUTF16(*out_path, treename, treename_len, cp);
2534        UniStrcat(*out_path, sep);
2535        UniStrcat(*out_path, path);
2536        unload_nls(cp);
2537
2538        return 0;
2539}
2540
2541int smb311_posix_mkdir(const unsigned int xid, struct inode *inode,
2542                               umode_t mode, struct cifs_tcon *tcon,
2543                               const char *full_path,
2544                               struct cifs_sb_info *cifs_sb)
2545{
2546        struct smb_rqst rqst;
2547        struct smb2_create_req *req;
2548        struct smb2_create_rsp *rsp = NULL;
2549        struct cifs_ses *ses = tcon->ses;
2550        struct kvec iov[3]; /* make sure at least one for each open context */
2551        struct kvec rsp_iov = {NULL, 0};
2552        int resp_buftype;
2553        int uni_path_len;
2554        __le16 *copy_path = NULL;
2555        int copy_size;
2556        int rc = 0;
2557        unsigned int n_iov = 2;
2558        __u32 file_attributes = 0;
2559        char *pc_buf = NULL;
2560        int flags = 0;
2561        unsigned int total_len;
2562        __le16 *utf16_path = NULL;
2563        struct TCP_Server_Info *server = cifs_pick_channel(ses);
2564
2565        cifs_dbg(FYI, "mkdir\n");
2566
2567        /* resource #1: path allocation */
2568        utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
2569        if (!utf16_path)
2570                return -ENOMEM;
2571
2572        if (!ses || !server) {
2573                rc = -EIO;
2574                goto err_free_path;
2575        }
2576
2577        /* resource #2: request */
2578        rc = smb2_plain_req_init(SMB2_CREATE, tcon, server,
2579                                 (void **) &req, &total_len);
2580        if (rc)
2581                goto err_free_path;
2582
2583
2584        if (smb3_encryption_required(tcon))
2585                flags |= CIFS_TRANSFORM_REQ;
2586
2587        req->ImpersonationLevel = IL_IMPERSONATION;
2588        req->DesiredAccess = cpu_to_le32(FILE_WRITE_ATTRIBUTES);
2589        /* File attributes ignored on open (used in create though) */
2590        req->FileAttributes = cpu_to_le32(file_attributes);
2591        req->ShareAccess = FILE_SHARE_ALL_LE;
2592        req->CreateDisposition = cpu_to_le32(FILE_CREATE);
2593        req->CreateOptions = cpu_to_le32(CREATE_NOT_FILE);
2594
2595        iov[0].iov_base = (char *)req;
2596        /* -1 since last byte is buf[0] which is sent below (path) */
2597        iov[0].iov_len = total_len - 1;
2598
2599        req->NameOffset = cpu_to_le16(sizeof(struct smb2_create_req));
2600
2601        /* [MS-SMB2] 2.2.13 NameOffset:
2602         * If SMB2_FLAGS_DFS_OPERATIONS is set in the Flags field of
2603         * the SMB2 header, the file name includes a prefix that will
2604         * be processed during DFS name normalization as specified in
2605         * section 3.3.5.9. Otherwise, the file name is relative to
2606         * the share that is identified by the TreeId in the SMB2
2607         * header.
2608         */
2609        if (tcon->share_flags & SHI1005_FLAGS_DFS) {
2610                int name_len;
2611
2612                req->sync_hdr.Flags |= SMB2_FLAGS_DFS_OPERATIONS;
2613                rc = alloc_path_with_tree_prefix(&copy_path, &copy_size,
2614                                                 &name_len,
2615                                                 tcon->treeName, utf16_path);
2616                if (rc)
2617                        goto err_free_req;
2618
2619                req->NameLength = cpu_to_le16(name_len * 2);
2620                uni_path_len = copy_size;
2621                /* free before overwriting resource */
2622                kfree(utf16_path);
2623                utf16_path = copy_path;
2624        } else {
2625                uni_path_len = (2 * UniStrnlen((wchar_t *)utf16_path, PATH_MAX)) + 2;
2626                /* MUST set path len (NameLength) to 0 opening root of share */
2627                req->NameLength = cpu_to_le16(uni_path_len - 2);
2628                if (uni_path_len % 8 != 0) {
2629                        copy_size = roundup(uni_path_len, 8);
2630                        copy_path = kzalloc(copy_size, GFP_KERNEL);
2631                        if (!copy_path) {
2632                                rc = -ENOMEM;
2633                                goto err_free_req;
2634                        }
2635                        memcpy((char *)copy_path, (const char *)utf16_path,
2636                               uni_path_len);
2637                        uni_path_len = copy_size;
2638                        /* free before overwriting resource */
2639                        kfree(utf16_path);
2640                        utf16_path = copy_path;
2641                }
2642        }
2643
2644        iov[1].iov_len = uni_path_len;
2645        iov[1].iov_base = utf16_path;
2646        req->RequestedOplockLevel = SMB2_OPLOCK_LEVEL_NONE;
2647
2648        if (tcon->posix_extensions) {
2649                /* resource #3: posix buf */
2650                rc = add_posix_context(iov, &n_iov, mode);
2651                if (rc)
2652                        goto err_free_req;
2653                pc_buf = iov[n_iov-1].iov_base;
2654        }
2655
2656
2657        memset(&rqst, 0, sizeof(struct smb_rqst));
2658        rqst.rq_iov = iov;
2659        rqst.rq_nvec = n_iov;
2660
2661        /* no need to inc num_remote_opens because we close it just below */
2662        trace_smb3_posix_mkdir_enter(xid, tcon->tid, ses->Suid, CREATE_NOT_FILE,
2663                                    FILE_WRITE_ATTRIBUTES);
2664        /* resource #4: response buffer */
2665        rc = cifs_send_recv(xid, ses, server,
2666                            &rqst, &resp_buftype, flags, &rsp_iov);
2667        if (rc) {
2668                cifs_stats_fail_inc(tcon, SMB2_CREATE_HE);
2669                trace_smb3_posix_mkdir_err(xid, tcon->tid, ses->Suid,
2670                                           CREATE_NOT_FILE,
2671                                           FILE_WRITE_ATTRIBUTES, rc);
2672                goto err_free_rsp_buf;
2673        }
2674
2675        rsp = (struct smb2_create_rsp *)rsp_iov.iov_base;
2676        trace_smb3_posix_mkdir_done(xid, rsp->PersistentFileId, tcon->tid,
2677                                    ses->Suid, CREATE_NOT_FILE,
2678                                    FILE_WRITE_ATTRIBUTES);
2679
2680        SMB2_close(xid, tcon, rsp->PersistentFileId, rsp->VolatileFileId);
2681
2682        /* Eventually save off posix specific response info and timestaps */
2683
2684err_free_rsp_buf:
2685        free_rsp_buf(resp_buftype, rsp);
2686        kfree(pc_buf);
2687err_free_req:
2688        cifs_small_buf_release(req);
2689err_free_path:
2690        kfree(utf16_path);
2691        return rc;
2692}
2693
2694int
2695SMB2_open_init(struct cifs_tcon *tcon, struct TCP_Server_Info *server,
2696               struct smb_rqst *rqst, __u8 *oplock,
2697               struct cifs_open_parms *oparms, __le16 *path)
2698{
2699        struct smb2_create_req *req;
2700        unsigned int n_iov = 2;
2701        __u32 file_attributes = 0;
2702        int copy_size;
2703        int uni_path_len;
2704        unsigned int total_len;
2705        struct kvec *iov = rqst->rq_iov;
2706        __le16 *copy_path;
2707        int rc;
2708
2709        rc = smb2_plain_req_init(SMB2_CREATE, tcon, server,
2710                                 (void **) &req, &total_len);
2711        if (rc)
2712                return rc;
2713
2714        iov[0].iov_base = (char *)req;
2715        /* -1 since last byte is buf[0] which is sent below (path) */
2716        iov[0].iov_len = total_len - 1;
2717
2718        if (oparms->create_options & CREATE_OPTION_READONLY)
2719                file_attributes |= ATTR_READONLY;
2720        if (oparms->create_options & CREATE_OPTION_SPECIAL)
2721                file_attributes |= ATTR_SYSTEM;
2722
2723        req->ImpersonationLevel = IL_IMPERSONATION;
2724        req->DesiredAccess = cpu_to_le32(oparms->desired_access);
2725        /* File attributes ignored on open (used in create though) */
2726        req->FileAttributes = cpu_to_le32(file_attributes);
2727        req->ShareAccess = FILE_SHARE_ALL_LE;
2728
2729        req->CreateDisposition = cpu_to_le32(oparms->disposition);
2730        req->CreateOptions = cpu_to_le32(oparms->create_options & CREATE_OPTIONS_MASK);
2731        req->NameOffset = cpu_to_le16(sizeof(struct smb2_create_req));
2732
2733        /* [MS-SMB2] 2.2.13 NameOffset:
2734         * If SMB2_FLAGS_DFS_OPERATIONS is set in the Flags field of
2735         * the SMB2 header, the file name includes a prefix that will
2736         * be processed during DFS name normalization as specified in
2737         * section 3.3.5.9. Otherwise, the file name is relative to
2738         * the share that is identified by the TreeId in the SMB2
2739         * header.
2740         */
2741        if (tcon->share_flags & SHI1005_FLAGS_DFS) {
2742                int name_len;
2743
2744                req->sync_hdr.Flags |= SMB2_FLAGS_DFS_OPERATIONS;
2745                rc = alloc_path_with_tree_prefix(&copy_path, &copy_size,
2746                                                 &name_len,
2747                                                 tcon->treeName, path);
2748                if (rc)
2749                        return rc;
2750                req->NameLength = cpu_to_le16(name_len * 2);
2751                uni_path_len = copy_size;
2752                path = copy_path;
2753        } else {
2754                uni_path_len = (2 * UniStrnlen((wchar_t *)path, PATH_MAX)) + 2;
2755                /* MUST set path len (NameLength) to 0 opening root of share */
2756                req->NameLength = cpu_to_le16(uni_path_len - 2);
2757                copy_size = uni_path_len;
2758                if (copy_size % 8 != 0)
2759                        copy_size = roundup(copy_size, 8);
2760                copy_path = kzalloc(copy_size, GFP_KERNEL);
2761                if (!copy_path)
2762                        return -ENOMEM;
2763                memcpy((char *)copy_path, (const char *)path,
2764                       uni_path_len);
2765                uni_path_len = copy_size;
2766                path = copy_path;
2767        }
2768
2769        iov[1].iov_len = uni_path_len;
2770        iov[1].iov_base = path;
2771
2772        if ((!server->oplocks) || (tcon->no_lease))
2773                *oplock = SMB2_OPLOCK_LEVEL_NONE;
2774
2775        if (!(server->capabilities & SMB2_GLOBAL_CAP_LEASING) ||
2776            *oplock == SMB2_OPLOCK_LEVEL_NONE)
2777                req->RequestedOplockLevel = *oplock;
2778        else if (!(server->capabilities & SMB2_GLOBAL_CAP_DIRECTORY_LEASING) &&
2779                  (oparms->create_options & CREATE_NOT_FILE))
2780                req->RequestedOplockLevel = *oplock; /* no srv lease support */
2781        else {
2782                rc = add_lease_context(server, iov, &n_iov,
2783                                       oparms->fid->lease_key, oplock);
2784                if (rc)
2785                        return rc;
2786        }
2787
2788        if (*oplock == SMB2_OPLOCK_LEVEL_BATCH) {
2789                /* need to set Next field of lease context if we request it */
2790                if (server->capabilities & SMB2_GLOBAL_CAP_LEASING) {
2791                        struct create_context *ccontext =
2792                            (struct create_context *)iov[n_iov-1].iov_base;
2793                        ccontext->Next =
2794                                cpu_to_le32(server->vals->create_lease_size);
2795                }
2796
2797                rc = add_durable_context(iov, &n_iov, oparms,
2798                                        tcon->use_persistent);
2799                if (rc)
2800                        return rc;
2801        }
2802
2803        if (tcon->posix_extensions) {
2804                if (n_iov > 2) {
2805                        struct create_context *ccontext =
2806                            (struct create_context *)iov[n_iov-1].iov_base;
2807                        ccontext->Next =
2808                                cpu_to_le32(iov[n_iov-1].iov_len);
2809                }
2810
2811                rc = add_posix_context(iov, &n_iov, oparms->mode);
2812                if (rc)
2813                        return rc;
2814        }
2815
2816        if (tcon->snapshot_time) {
2817                cifs_dbg(FYI, "adding snapshot context\n");
2818                if (n_iov > 2) {
2819                        struct create_context *ccontext =
2820                            (struct create_context *)iov[n_iov-1].iov_base;
2821                        ccontext->Next =
2822                                cpu_to_le32(iov[n_iov-1].iov_len);
2823                }
2824
2825                rc = add_twarp_context(iov, &n_iov, tcon->snapshot_time);
2826                if (rc)
2827                        return rc;
2828        }
2829
2830        if ((oparms->disposition != FILE_OPEN) && (oparms->cifs_sb)) {
2831                bool set_mode;
2832                bool set_owner;
2833
2834                if ((oparms->cifs_sb->mnt_cifs_flags & CIFS_MOUNT_MODE_FROM_SID) &&
2835                    (oparms->mode != ACL_NO_MODE))
2836                        set_mode = true;
2837                else {
2838                        set_mode = false;
2839                        oparms->mode = ACL_NO_MODE;
2840                }
2841
2842                if (oparms->cifs_sb->mnt_cifs_flags & CIFS_MOUNT_UID_FROM_ACL)
2843                        set_owner = true;
2844                else
2845                        set_owner = false;
2846
2847                if (set_owner | set_mode) {
2848                        if (n_iov > 2) {
2849                                struct create_context *ccontext =
2850                                    (struct create_context *)iov[n_iov-1].iov_base;
2851                                ccontext->Next = cpu_to_le32(iov[n_iov-1].iov_len);
2852                        }
2853
2854                        cifs_dbg(FYI, "add sd with mode 0x%x\n", oparms->mode);
2855                        rc = add_sd_context(iov, &n_iov, oparms->mode, set_owner);
2856                        if (rc)
2857                                return rc;
2858                }
2859        }
2860
2861        if (n_iov > 2) {
2862                struct create_context *ccontext =
2863                        (struct create_context *)iov[n_iov-1].iov_base;
2864                ccontext->Next = cpu_to_le32(iov[n_iov-1].iov_len);
2865        }
2866        add_query_id_context(iov, &n_iov);
2867
2868        rqst->rq_nvec = n_iov;
2869        return 0;
2870}
2871
2872/* rq_iov[0] is the request and is released by cifs_small_buf_release().
2873 * All other vectors are freed by kfree().
2874 */
2875void
2876SMB2_open_free(struct smb_rqst *rqst)
2877{
2878        int i;
2879
2880        if (rqst && rqst->rq_iov) {
2881                cifs_small_buf_release(rqst->rq_iov[0].iov_base);
2882                for (i = 1; i < rqst->rq_nvec; i++)
2883                        if (rqst->rq_iov[i].iov_base != smb2_padding)
2884                                kfree(rqst->rq_iov[i].iov_base);
2885        }
2886}
2887
2888int
2889SMB2_open(const unsigned int xid, struct cifs_open_parms *oparms, __le16 *path,
2890          __u8 *oplock, struct smb2_file_all_info *buf,
2891          struct create_posix_rsp *posix,
2892          struct kvec *err_iov, int *buftype)
2893{
2894        struct smb_rqst rqst;
2895        struct smb2_create_rsp *rsp = NULL;
2896        struct cifs_tcon *tcon = oparms->tcon;
2897        struct cifs_ses *ses = tcon->ses;
2898        struct TCP_Server_Info *server = cifs_pick_channel(ses);
2899        struct kvec iov[SMB2_CREATE_IOV_SIZE];
2900        struct kvec rsp_iov = {NULL, 0};
2901        int resp_buftype = CIFS_NO_BUFFER;
2902        int rc = 0;
2903        int flags = 0;
2904
2905        cifs_dbg(FYI, "create/open\n");
2906        if (!ses || !server)
2907                return -EIO;
2908
2909        if (smb3_encryption_required(tcon))
2910                flags |= CIFS_TRANSFORM_REQ;
2911
2912        memset(&rqst, 0, sizeof(struct smb_rqst));
2913        memset(&iov, 0, sizeof(iov));
2914        rqst.rq_iov = iov;
2915        rqst.rq_nvec = SMB2_CREATE_IOV_SIZE;
2916
2917        rc = SMB2_open_init(tcon, server,
2918                            &rqst, oplock, oparms, path);
2919        if (rc)
2920                goto creat_exit;
2921
2922        trace_smb3_open_enter(xid, tcon->tid, tcon->ses->Suid,
2923                oparms->create_options, oparms->desired_access);
2924
2925        rc = cifs_send_recv(xid, ses, server,
2926                            &rqst, &resp_buftype, flags,
2927                            &rsp_iov);
2928        rsp = (struct smb2_create_rsp *)rsp_iov.iov_base;
2929
2930        if (rc != 0) {
2931                cifs_stats_fail_inc(tcon, SMB2_CREATE_HE);
2932                if (err_iov && rsp) {
2933                        *err_iov = rsp_iov;
2934                        *buftype = resp_buftype;
2935                        resp_buftype = CIFS_NO_BUFFER;
2936                        rsp = NULL;
2937                }
2938                trace_smb3_open_err(xid, tcon->tid, ses->Suid,
2939                                    oparms->create_options, oparms->desired_access, rc);
2940                if (rc == -EREMCHG) {
2941                        pr_warn_once("server share %s deleted\n",
2942                                     tcon->treeName);
2943                        tcon->need_reconnect = true;
2944                }
2945                goto creat_exit;
2946        } else
2947                trace_smb3_open_done(xid, rsp->PersistentFileId, tcon->tid,
2948                                     ses->Suid, oparms->create_options,
2949                                     oparms->desired_access);
2950
2951        atomic_inc(&tcon->num_remote_opens);
2952        oparms->fid->persistent_fid = rsp->PersistentFileId;
2953        oparms->fid->volatile_fid = rsp->VolatileFileId;
2954        oparms->fid->access = oparms->desired_access;
2955#ifdef CONFIG_CIFS_DEBUG2
2956        oparms->fid->mid = le64_to_cpu(rsp->sync_hdr.MessageId);
2957#endif /* CIFS_DEBUG2 */
2958
2959        if (buf) {
2960                buf->CreationTime = rsp->CreationTime;
2961                buf->LastAccessTime = rsp->LastAccessTime;
2962                buf->LastWriteTime = rsp->LastWriteTime;
2963                buf->ChangeTime = rsp->ChangeTime;
2964                buf->AllocationSize = rsp->AllocationSize;
2965                buf->EndOfFile = rsp->EndofFile;
2966                buf->Attributes = rsp->FileAttributes;
2967                buf->NumberOfLinks = cpu_to_le32(1);
2968                buf->DeletePending = 0;
2969        }
2970
2971
2972        smb2_parse_contexts(server, rsp, &oparms->fid->epoch,
2973                            oparms->fid->lease_key, oplock, buf, posix);
2974creat_exit:
2975        SMB2_open_free(&rqst);
2976        free_rsp_buf(resp_buftype, rsp);
2977        return rc;
2978}
2979
2980int
2981SMB2_ioctl_init(struct cifs_tcon *tcon, struct TCP_Server_Info *server,
2982                struct smb_rqst *rqst,
2983                u64 persistent_fid, u64 volatile_fid, u32 opcode,
2984                bool is_fsctl, char *in_data, u32 indatalen,
2985                __u32 max_response_size)
2986{
2987        struct smb2_ioctl_req *req;
2988        struct kvec *iov = rqst->rq_iov;
2989        unsigned int total_len;
2990        int rc;
2991        char *in_data_buf;
2992
2993        rc = smb2_ioctl_req_init(opcode, tcon, server,
2994                                 (void **) &req, &total_len);
2995        if (rc)
2996                return rc;
2997
2998        if (indatalen) {
2999                /*
3000                 * indatalen is usually small at a couple of bytes max, so
3001                 * just allocate through generic pool
3002                 */
3003                in_data_buf = kmemdup(in_data, indatalen, GFP_NOFS);
3004                if (!in_data_buf) {
3005                        cifs_small_buf_release(req);
3006                        return -ENOMEM;
3007                }
3008        }
3009
3010        req->CtlCode = cpu_to_le32(opcode);
3011        req->PersistentFileId = persistent_fid;
3012        req->VolatileFileId = volatile_fid;
3013
3014        iov[0].iov_base = (char *)req;
3015        /*
3016         * If no input data, the size of ioctl struct in
3017         * protocol spec still includes a 1 byte data buffer,
3018         * but if input data passed to ioctl, we do not
3019         * want to double count this, so we do not send
3020         * the dummy one byte of data in iovec[0] if sending
3021         * input data (in iovec[1]).
3022         */
3023        if (indatalen) {
3024                req->InputCount = cpu_to_le32(indatalen);
3025                /* do not set InputOffset if no input data */
3026                req->InputOffset =
3027                       cpu_to_le32(offsetof(struct smb2_ioctl_req, Buffer));
3028                rqst->rq_nvec = 2;
3029                iov[0].iov_len = total_len - 1;
3030                iov[1].iov_base = in_data_buf;
3031                iov[1].iov_len = indatalen;
3032        } else {
3033                rqst->rq_nvec = 1;
3034                iov[0].iov_len = total_len;
3035        }
3036
3037        req->OutputOffset = 0;
3038        req->OutputCount = 0; /* MBZ */
3039
3040        /*
3041         * In most cases max_response_size is set to 16K (CIFSMaxBufSize)
3042         * We Could increase default MaxOutputResponse, but that could require
3043         * more credits. Windows typically sets this smaller, but for some
3044         * ioctls it may be useful to allow server to send more. No point
3045         * limiting what the server can send as long as fits in one credit
3046         * We can not handle more than CIFS_MAX_BUF_SIZE yet but may want
3047         * to increase this limit up in the future.
3048         * Note that for snapshot queries that servers like Azure expect that
3049         * the first query be minimal size (and just used to get the number/size
3050         * of previous versions) so response size must be specified as EXACTLY
3051         * sizeof(struct snapshot_array) which is 16 when rounded up to multiple
3052         * of eight bytes.  Currently that is the only case where we set max
3053         * response size smaller.
3054         */
3055        req->MaxOutputResponse = cpu_to_le32(max_response_size);
3056        req->sync_hdr.CreditCharge =
3057                cpu_to_le16(DIV_ROUND_UP(max(indatalen, max_response_size),
3058                                         SMB2_MAX_BUFFER_SIZE));
3059        if (is_fsctl)
3060                req->Flags = cpu_to_le32(SMB2_0_IOCTL_IS_FSCTL);
3061        else
3062                req->Flags = 0;
3063
3064        /* validate negotiate request must be signed - see MS-SMB2 3.2.5.5 */
3065        if (opcode == FSCTL_VALIDATE_NEGOTIATE_INFO)
3066                req->sync_hdr.Flags |= SMB2_FLAGS_SIGNED;
3067
3068        return 0;
3069}
3070
3071void
3072SMB2_ioctl_free(struct smb_rqst *rqst)
3073{
3074        int i;
3075        if (rqst && rqst->rq_iov) {
3076                cifs_small_buf_release(rqst->rq_iov[0].iov_base); /* request */
3077                for (i = 1; i < rqst->rq_nvec; i++)
3078                        if (rqst->rq_iov[i].iov_base != smb2_padding)
3079                                kfree(rqst->rq_iov[i].iov_base);
3080        }
3081}
3082
3083
3084/*
3085 *      SMB2 IOCTL is used for both IOCTLs and FSCTLs
3086 */
3087int
3088SMB2_ioctl(const unsigned int xid, struct cifs_tcon *tcon, u64 persistent_fid,
3089           u64 volatile_fid, u32 opcode, bool is_fsctl,
3090           char *in_data, u32 indatalen, u32 max_out_data_len,
3091           char **out_data, u32 *plen /* returned data len */)
3092{
3093        struct smb_rqst rqst;
3094        struct smb2_ioctl_rsp *rsp = NULL;
3095        struct cifs_ses *ses;
3096        struct TCP_Server_Info *server;
3097        struct kvec iov[SMB2_IOCTL_IOV_SIZE];
3098        struct kvec rsp_iov = {NULL, 0};
3099        int resp_buftype = CIFS_NO_BUFFER;
3100        int rc = 0;
3101        int flags = 0;
3102
3103        cifs_dbg(FYI, "SMB2 IOCTL\n");
3104
3105        if (out_data != NULL)
3106                *out_data = NULL;
3107
3108        /* zero out returned data len, in case of error */
3109        if (plen)
3110                *plen = 0;
3111
3112        if (!tcon)
3113                return -EIO;
3114
3115        ses = tcon->ses;
3116        if (!ses)
3117                return -EIO;
3118
3119        server = cifs_pick_channel(ses);
3120        if (!server)
3121                return -EIO;
3122
3123        if (smb3_encryption_required(tcon))
3124                flags |= CIFS_TRANSFORM_REQ;
3125
3126        memset(&rqst, 0, sizeof(struct smb_rqst));
3127        memset(&iov, 0, sizeof(iov));
3128        rqst.rq_iov = iov;
3129        rqst.rq_nvec = SMB2_IOCTL_IOV_SIZE;
3130
3131        rc = SMB2_ioctl_init(tcon, server,
3132                             &rqst, persistent_fid, volatile_fid, opcode,
3133                             is_fsctl, in_data, indatalen, max_out_data_len);
3134        if (rc)
3135                goto ioctl_exit;
3136
3137        rc = cifs_send_recv(xid, ses, server,
3138                            &rqst, &resp_buftype, flags,
3139                            &rsp_iov);
3140        rsp = (struct smb2_ioctl_rsp *)rsp_iov.iov_base;
3141
3142        if (rc != 0)
3143                trace_smb3_fsctl_err(xid, persistent_fid, tcon->tid,
3144                                ses->Suid, 0, opcode, rc);
3145
3146        if ((rc != 0) && (rc != -EINVAL) && (rc != -E2BIG)) {
3147                cifs_stats_fail_inc(tcon, SMB2_IOCTL_HE);
3148                goto ioctl_exit;
3149        } else if (rc == -EINVAL) {
3150                if ((opcode != FSCTL_SRV_COPYCHUNK_WRITE) &&
3151                    (opcode != FSCTL_SRV_COPYCHUNK)) {
3152                        cifs_stats_fail_inc(tcon, SMB2_IOCTL_HE);
3153                        goto ioctl_exit;
3154                }
3155        } else if (rc == -E2BIG) {
3156                if (opcode != FSCTL_QUERY_ALLOCATED_RANGES) {
3157                        cifs_stats_fail_inc(tcon, SMB2_IOCTL_HE);
3158                        goto ioctl_exit;
3159                }
3160        }
3161
3162        /* check if caller wants to look at return data or just return rc */
3163        if ((plen == NULL) || (out_data == NULL))
3164                goto ioctl_exit;
3165
3166        *plen = le32_to_cpu(rsp->OutputCount);
3167
3168        /* We check for obvious errors in the output buffer length and offset */
3169        if (*plen == 0)
3170                goto ioctl_exit; /* server returned no data */
3171        else if (*plen > rsp_iov.iov_len || *plen > 0xFF00) {
3172                cifs_tcon_dbg(VFS, "srv returned invalid ioctl length: %d\n", *plen);
3173                *plen = 0;
3174                rc = -EIO;
3175                goto ioctl_exit;
3176        }
3177
3178        if (rsp_iov.iov_len - *plen < le32_to_cpu(rsp->OutputOffset)) {
3179                cifs_tcon_dbg(VFS, "Malformed ioctl resp: len %d offset %d\n", *plen,
3180                        le32_to_cpu(rsp->OutputOffset));
3181                *plen = 0;
3182                rc = -EIO;
3183                goto ioctl_exit;
3184        }
3185
3186        *out_data = kmemdup((char *)rsp + le32_to_cpu(rsp->OutputOffset),
3187                            *plen, GFP_KERNEL);
3188        if (*out_data == NULL) {
3189                rc = -ENOMEM;
3190                goto ioctl_exit;
3191        }
3192
3193ioctl_exit:
3194        SMB2_ioctl_free(&rqst);
3195        free_rsp_buf(resp_buftype, rsp);
3196        return rc;
3197}
3198
3199/*
3200 *   Individual callers to ioctl worker function follow
3201 */
3202
3203int
3204SMB2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
3205                     u64 persistent_fid, u64 volatile_fid)
3206{
3207        int rc;
3208        struct  compress_ioctl fsctl_input;
3209        char *ret_data = NULL;
3210
3211        fsctl_input.CompressionState =
3212                        cpu_to_le16(COMPRESSION_FORMAT_DEFAULT);
3213
3214        rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
3215                        FSCTL_SET_COMPRESSION, true /* is_fsctl */,
3216                        (char *)&fsctl_input /* data input */,
3217                        2 /* in data len */, CIFSMaxBufSize /* max out data */,
3218                        &ret_data /* out data */, NULL);
3219
3220        cifs_dbg(FYI, "set compression rc %d\n", rc);
3221
3222        return rc;
3223}
3224
3225int
3226SMB2_close_init(struct cifs_tcon *tcon, struct TCP_Server_Info *server,
3227                struct smb_rqst *rqst,
3228                u64 persistent_fid, u64 volatile_fid, bool query_attrs)
3229{
3230        struct smb2_close_req *req;
3231        struct kvec *iov = rqst->rq_iov;
3232        unsigned int total_len;
3233        int rc;
3234
3235        rc = smb2_plain_req_init(SMB2_CLOSE, tcon, server,
3236                                 (void **) &req, &total_len);
3237        if (rc)
3238                return rc;
3239
3240        req->PersistentFileId = persistent_fid;
3241        req->VolatileFileId = volatile_fid;
3242        if (query_attrs)
3243                req->Flags = SMB2_CLOSE_FLAG_POSTQUERY_ATTRIB;
3244        else
3245                req->Flags = 0;
3246        iov[0].iov_base = (char *)req;
3247        iov[0].iov_len = total_len;
3248
3249        return 0;
3250}
3251
3252void
3253SMB2_close_free(struct smb_rqst *rqst)
3254{
3255        if (rqst && rqst->rq_iov)
3256                cifs_small_buf_release(rqst->rq_iov[0].iov_base); /* request */
3257}
3258
3259int
3260__SMB2_close(const unsigned int xid, struct cifs_tcon *tcon,
3261             u64 persistent_fid, u64 volatile_fid,
3262             struct smb2_file_network_open_info *pbuf)
3263{
3264        struct smb_rqst rqst;
3265        struct smb2_close_rsp *rsp = NULL;
3266        struct cifs_ses *ses = tcon->ses;
3267        struct TCP_Server_Info *server = cifs_pick_channel(ses);
3268        struct kvec iov[1];
3269        struct kvec rsp_iov;
3270        int resp_buftype = CIFS_NO_BUFFER;
3271        int rc = 0;
3272        int flags = 0;
3273        bool query_attrs = false;
3274
3275        cifs_dbg(FYI, "Close\n");
3276
3277        if (!ses || !server)
3278                return -EIO;
3279
3280        if (smb3_encryption_required(tcon))
3281                flags |= CIFS_TRANSFORM_REQ;
3282
3283        memset(&rqst, 0, sizeof(struct smb_rqst));
3284        memset(&iov, 0, sizeof(iov));
3285        rqst.rq_iov = iov;
3286        rqst.rq_nvec = 1;
3287
3288        /* check if need to ask server to return timestamps in close response */
3289        if (pbuf)
3290                query_attrs = true;
3291
3292        trace_smb3_close_enter(xid, persistent_fid, tcon->tid, ses->Suid);
3293        rc = SMB2_close_init(tcon, server,
3294                             &rqst, persistent_fid, volatile_fid,
3295                             query_attrs);
3296        if (rc)
3297                goto close_exit;
3298
3299        rc = cifs_send_recv(xid, ses, server,
3300                            &rqst, &resp_buftype, flags, &rsp_iov);
3301        rsp = (struct smb2_close_rsp *)rsp_iov.iov_base;
3302
3303        if (rc != 0) {
3304                cifs_stats_fail_inc(tcon, SMB2_CLOSE_HE);
3305                trace_smb3_close_err(xid, persistent_fid, tcon->tid, ses->Suid,
3306                                     rc);
3307                goto close_exit;
3308        } else {
3309                trace_smb3_close_done(xid, persistent_fid, tcon->tid,
3310                                      ses->Suid);
3311                /*
3312                 * Note that have to subtract 4 since struct network_open_info
3313                 * has a final 4 byte pad that close response does not have
3314                 */
3315                if (pbuf)
3316                        memcpy(pbuf, (char *)&rsp->CreationTime, sizeof(*pbuf) - 4);
3317        }
3318
3319        atomic_dec(&tcon->num_remote_opens);
3320close_exit:
3321        SMB2_close_free(&rqst);
3322        free_rsp_buf(resp_buftype, rsp);
3323
3324        /* retry close in a worker thread if this one is interrupted */
3325        if (is_interrupt_error(rc)) {
3326                int tmp_rc;
3327
3328                tmp_rc = smb2_handle_cancelled_close(tcon, persistent_fid,
3329                                                     volatile_fid);
3330                if (tmp_rc)
3331                        cifs_dbg(VFS, "handle cancelled close fid 0x%llx returned error %d\n",
3332                                 persistent_fid, tmp_rc);
3333        }
3334        return rc;
3335}
3336
3337int
3338SMB2_close(const unsigned int xid, struct cifs_tcon *tcon,
3339                u64 persistent_fid, u64 volatile_fid)
3340{
3341        return __SMB2_close(xid, tcon, persistent_fid, volatile_fid, NULL);
3342}
3343
3344int
3345smb2_validate_iov(unsigned int offset, unsigned int buffer_length,
3346                  struct kvec *iov, unsigned int min_buf_size)
3347{
3348        unsigned int smb_len = iov->iov_len;
3349        char *end_of_smb = smb_len + (char *)iov->iov_base;
3350        char *begin_of_buf = offset + (char *)iov->iov_base;
3351        char *end_of_buf = begin_of_buf + buffer_length;
3352
3353
3354        if (buffer_length < min_buf_size) {
3355                cifs_dbg(VFS, "buffer length %d smaller than minimum size %d\n",
3356                         buffer_length, min_buf_size);
3357                return -EINVAL;
3358        }
3359
3360        /* check if beyond RFC1001 maximum length */
3361        if ((smb_len > 0x7FFFFF) || (buffer_length > 0x7FFFFF)) {
3362                cifs_dbg(VFS, "buffer length %d or smb length %d too large\n",
3363                         buffer_length, smb_len);
3364                return -EINVAL;
3365        }
3366
3367        if ((begin_of_buf > end_of_smb) || (end_of_buf > end_of_smb)) {
3368                cifs_dbg(VFS, "Invalid server response, bad offset to data\n");
3369                return -EINVAL;
3370        }
3371
3372        return 0;
3373}
3374
3375/*
3376 * If SMB buffer fields are valid, copy into temporary buffer to hold result.
3377 * Caller must free buffer.
3378 */
3379int
3380smb2_validate_and_copy_iov(unsigned int offset, unsigned int buffer_length,
3381                           struct kvec *iov, unsigned int minbufsize,
3382                           char *data)
3383{
3384        char *begin_of_buf = offset + (char *)iov->iov_base;
3385        int rc;
3386
3387        if (!data)
3388                return -EINVAL;
3389
3390        rc = smb2_validate_iov(offset, buffer_length, iov, minbufsize);
3391        if (rc)
3392                return rc;
3393
3394        memcpy(data, begin_of_buf, buffer_length);
3395
3396        return 0;
3397}
3398
3399int
3400SMB2_query_info_init(struct cifs_tcon *tcon, struct TCP_Server_Info *server,
3401                     struct smb_rqst *rqst,
3402                     u64 persistent_fid, u64 volatile_fid,
3403                     u8 info_class, u8 info_type, u32 additional_info,
3404                     size_t output_len, size_t input_len, void *input)
3405{
3406        struct smb2_query_info_req *req;
3407        struct kvec *iov = rqst->rq_iov;
3408        unsigned int total_len;
3409        int rc;
3410
3411        rc = smb2_plain_req_init(SMB2_QUERY_INFO, tcon, server,
3412                                 (void **) &req, &total_len);
3413        if (rc)
3414                return rc;
3415
3416        req->InfoType = info_type;
3417        req->FileInfoClass = info_class;
3418        req->PersistentFileId = persistent_fid;
3419        req->VolatileFileId = volatile_fid;
3420        req->AdditionalInformation = cpu_to_le32(additional_info);
3421
3422        req->OutputBufferLength = cpu_to_le32(output_len);
3423        if (input_len) {
3424                req->InputBufferLength = cpu_to_le32(input_len);
3425                /* total_len for smb query request never close to le16 max */
3426                req->InputBufferOffset = cpu_to_le16(total_len - 1);
3427                memcpy(req->Buffer, input, input_len);
3428        }
3429
3430        iov[0].iov_base = (char *)req;
3431        /* 1 for Buffer */
3432        iov[0].iov_len = total_len - 1 + input_len;
3433        return 0;
3434}
3435
3436void
3437SMB2_query_info_free(struct smb_rqst *rqst)
3438{
3439        if (rqst && rqst->rq_iov)
3440                cifs_small_buf_release(rqst->rq_iov[0].iov_base); /* request */
3441}
3442
3443static int
3444query_info(const unsigned int xid, struct cifs_tcon *tcon,
3445           u64 persistent_fid, u64 volatile_fid, u8 info_class, u8 info_type,
3446           u32 additional_info, size_t output_len, size_t min_len, void **data,
3447                u32 *dlen)
3448{
3449        struct smb_rqst rqst;
3450        struct smb2_query_info_rsp *rsp = NULL;
3451        struct kvec iov[1];
3452        struct kvec rsp_iov;
3453        int rc = 0;
3454        int resp_buftype = CIFS_NO_BUFFER;
3455        struct cifs_ses *ses = tcon->ses;
3456        struct TCP_Server_Info *server;
3457        int flags = 0;
3458        bool allocated = false;
3459
3460        cifs_dbg(FYI, "Query Info\n");
3461
3462        if (!ses)
3463                return -EIO;
3464        server = cifs_pick_channel(ses);
3465        if (!server)
3466                return -EIO;
3467
3468        if (smb3_encryption_required(tcon))
3469                flags |= CIFS_TRANSFORM_REQ;
3470
3471        memset(&rqst, 0, sizeof(struct smb_rqst));
3472        memset(&iov, 0, sizeof(iov));
3473        rqst.rq_iov = iov;
3474        rqst.rq_nvec = 1;
3475
3476        rc = SMB2_query_info_init(tcon, server,
3477                                  &rqst, persistent_fid, volatile_fid,
3478                                  info_class, info_type, additional_info,
3479                                  output_len, 0, NULL);
3480        if (rc)
3481                goto qinf_exit;
3482
3483        trace_smb3_query_info_enter(xid, persistent_fid, tcon->tid,
3484                                    ses->Suid, info_class, (__u32)info_type);
3485
3486        rc = cifs_send_recv(xid, ses, server,
3487                            &rqst, &resp_buftype, flags, &rsp_iov);
3488        rsp = (struct smb2_query_info_rsp *)rsp_iov.iov_base;
3489
3490        if (rc) {
3491                cifs_stats_fail_inc(tcon, SMB2_QUERY_INFO_HE);
3492                trace_smb3_query_info_err(xid, persistent_fid, tcon->tid,
3493                                ses->Suid, info_class, (__u32)info_type, rc);
3494                goto qinf_exit;
3495        }
3496
3497        trace_smb3_query_info_done(xid, persistent_fid, tcon->tid,
3498                                ses->Suid, info_class, (__u32)info_type);
3499
3500        if (dlen) {
3501                *dlen = le32_to_cpu(rsp->OutputBufferLength);
3502                if (!*data) {
3503                        *data = kmalloc(*dlen, GFP_KERNEL);
3504                        if (!*data) {
3505                                cifs_tcon_dbg(VFS,
3506                                        "Error %d allocating memory for acl\n",
3507                                        rc);
3508                                *dlen = 0;
3509                                rc = -ENOMEM;
3510                                goto qinf_exit;
3511                        }
3512                        allocated = true;
3513                }
3514        }
3515
3516        rc = smb2_validate_and_copy_iov(le16_to_cpu(rsp->OutputBufferOffset),
3517                                        le32_to_cpu(rsp->OutputBufferLength),
3518                                        &rsp_iov, min_len, *data);
3519        if (rc && allocated) {
3520                kfree(*data);
3521                *data = NULL;
3522                *dlen = 0;
3523        }
3524
3525qinf_exit:
3526        SMB2_query_info_free(&rqst);
3527        free_rsp_buf(resp_buftype, rsp);
3528        return rc;
3529}
3530
3531int SMB2_query_info(const unsigned int xid, struct cifs_tcon *tcon,
3532        u64 persistent_fid, u64 volatile_fid, struct smb2_file_all_info *data)
3533{
3534        return query_info(xid, tcon, persistent_fid, volatile_fid,
3535                          FILE_ALL_INFORMATION, SMB2_O_INFO_FILE, 0,
3536                          sizeof(struct smb2_file_all_info) + PATH_MAX * 2,
3537                          sizeof(struct smb2_file_all_info), (void **)&data,
3538                          NULL);
3539}
3540
3541#if 0
3542/* currently unused, as now we are doing compounding instead (see smb311_posix_query_path_info) */
3543int
3544SMB311_posix_query_info(const unsigned int xid, struct cifs_tcon *tcon,
3545                u64 persistent_fid, u64 volatile_fid, struct smb311_posix_qinfo *data, u32 *plen)
3546{
3547        size_t output_len = sizeof(struct smb311_posix_qinfo *) +
3548                        (sizeof(struct cifs_sid) * 2) + (PATH_MAX * 2);
3549        *plen = 0;
3550
3551        return query_info(xid, tcon, persistent_fid, volatile_fid,
3552                          SMB_FIND_FILE_POSIX_INFO, SMB2_O_INFO_FILE, 0,
3553                          output_len, sizeof(struct smb311_posix_qinfo), (void **)&data, plen);
3554        /* Note caller must free "data" (passed in above). It may be allocated in query_info call */
3555}
3556#endif
3557
3558int
3559SMB2_query_acl(const unsigned int xid, struct cifs_tcon *tcon,
3560               u64 persistent_fid, u64 volatile_fid,
3561               void **data, u32 *plen, u32 extra_info)
3562{
3563        __u32 additional_info = OWNER_SECINFO | GROUP_SECINFO | DACL_SECINFO |
3564                                extra_info;
3565        *plen = 0;
3566
3567        return query_info(xid, tcon, persistent_fid, volatile_fid,
3568                          0, SMB2_O_INFO_SECURITY, additional_info,
3569                          SMB2_MAX_BUFFER_SIZE, MIN_SEC_DESC_LEN, data, plen);
3570}
3571
3572int
3573SMB2_get_srv_num(const unsigned int xid, struct cifs_tcon *tcon,
3574                 u64 persistent_fid, u64 volatile_fid, __le64 *uniqueid)
3575{
3576        return query_info(xid, tcon, persistent_fid, volatile_fid,
3577                          FILE_INTERNAL_INFORMATION, SMB2_O_INFO_FILE, 0,
3578                          sizeof(struct smb2_file_internal_info),
3579                          sizeof(struct smb2_file_internal_info),
3580                          (void **)&uniqueid, NULL);
3581}
3582
3583/*
3584 * CHANGE_NOTIFY Request is sent to get notifications on changes to a directory
3585 * See MS-SMB2 2.2.35 and 2.2.36
3586 */
3587
3588static int
3589SMB2_notify_init(const unsigned int xid, struct smb_rqst *rqst,
3590                 struct cifs_tcon *tcon, struct TCP_Server_Info *server,
3591                 u64 persistent_fid, u64 volatile_fid,
3592                 u32 completion_filter, bool watch_tree)
3593{
3594        struct smb2_change_notify_req *req;
3595        struct kvec *iov = rqst->rq_iov;
3596        unsigned int total_len;
3597        int rc;
3598
3599        rc = smb2_plain_req_init(SMB2_CHANGE_NOTIFY, tcon, server,
3600                                 (void **) &req, &total_len);
3601        if (rc)
3602                return rc;
3603
3604        req->PersistentFileId = persistent_fid;
3605        req->VolatileFileId = volatile_fid;
3606        /* See note 354 of MS-SMB2, 64K max */
3607        req->OutputBufferLength =
3608                cpu_to_le32(SMB2_MAX_BUFFER_SIZE - MAX_SMB2_HDR_SIZE);
3609        req->CompletionFilter = cpu_to_le32(completion_filter);
3610        if (watch_tree)
3611                req->Flags = cpu_to_le16(SMB2_WATCH_TREE);
3612        else
3613                req->Flags = 0;
3614
3615        iov[0].iov_base = (char *)req;
3616        iov[0].iov_len = total_len;
3617
3618        return 0;
3619}
3620
3621int
3622SMB2_change_notify(const unsigned int xid, struct cifs_tcon *tcon,
3623                u64 persistent_fid, u64 volatile_fid, bool watch_tree,
3624                u32 completion_filter)
3625{
3626        struct cifs_ses *ses = tcon->ses;
3627        struct TCP_Server_Info *server = cifs_pick_channel(ses);
3628        struct smb_rqst rqst;
3629        struct kvec iov[1];
3630        struct kvec rsp_iov = {NULL, 0};
3631        int resp_buftype = CIFS_NO_BUFFER;
3632        int flags = 0;
3633        int rc = 0;
3634
3635        cifs_dbg(FYI, "change notify\n");
3636        if (!ses || !server)
3637                return -EIO;
3638
3639        if (smb3_encryption_required(tcon))
3640                flags |= CIFS_TRANSFORM_REQ;
3641
3642        memset(&rqst, 0, sizeof(struct smb_rqst));
3643        memset(&iov, 0, sizeof(iov));
3644        rqst.rq_iov = iov;
3645        rqst.rq_nvec = 1;
3646
3647        rc = SMB2_notify_init(xid, &rqst, tcon, server,
3648                              persistent_fid, volatile_fid,
3649                              completion_filter, watch_tree);
3650        if (rc)
3651                goto cnotify_exit;
3652
3653        trace_smb3_notify_enter(xid, persistent_fid, tcon->tid, ses->Suid,
3654                                (u8)watch_tree, completion_filter);
3655        rc = cifs_send_recv(xid, ses, server,
3656                            &rqst, &resp_buftype, flags, &rsp_iov);
3657
3658        if (rc != 0) {
3659                cifs_stats_fail_inc(tcon, SMB2_CHANGE_NOTIFY_HE);
3660                trace_smb3_notify_err(xid, persistent_fid, tcon->tid, ses->Suid,
3661                                (u8)watch_tree, completion_filter, rc);
3662        } else
3663                trace_smb3_notify_done(xid, persistent_fid, tcon->tid,
3664                                ses->Suid, (u8)watch_tree, completion_filter);
3665
3666 cnotify_exit:
3667        if (rqst.rq_iov)
3668                cifs_small_buf_release(rqst.rq_iov[0].iov_base); /* request */
3669        free_rsp_buf(resp_buftype, rsp_iov.iov_base);
3670        return rc;
3671}
3672
3673
3674
3675/*
3676 * This is a no-op for now. We're not really interested in the reply, but
3677 * rather in the fact that the server sent one and that server->lstrp
3678 * gets updated.
3679 *
3680 * FIXME: maybe we should consider checking that the reply matches request?
3681 */
3682static void
3683smb2_echo_callback(struct mid_q_entry *mid)
3684{
3685        struct TCP_Server_Info *server = mid->callback_data;
3686        struct smb2_echo_rsp *rsp = (struct smb2_echo_rsp *)mid->resp_buf;
3687        struct cifs_credits credits = { .value = 0, .instance = 0 };
3688
3689        if (mid->mid_state == MID_RESPONSE_RECEIVED
3690            || mid->mid_state == MID_RESPONSE_MALFORMED) {
3691                credits.value = le16_to_cpu(rsp->sync_hdr.CreditRequest);
3692                credits.instance = server->reconnect_instance;
3693        }
3694
3695        DeleteMidQEntry(mid);
3696        add_credits(server, &credits, CIFS_ECHO_OP);
3697}
3698
3699void smb2_reconnect_server(struct work_struct *work)
3700{
3701        struct TCP_Server_Info *server = container_of(work,
3702                                        struct TCP_Server_Info, reconnect.work);
3703        struct cifs_ses *ses;
3704        struct cifs_tcon *tcon, *tcon2;
3705        struct list_head tmp_list;
3706        int tcon_exist = false;
3707        int rc;
3708        int resched = false;
3709
3710
3711        /* Prevent simultaneous reconnects that can corrupt tcon->rlist list */
3712        mutex_lock(&server->reconnect_mutex);
3713
3714        INIT_LIST_HEAD(&tmp_list);
3715        cifs_dbg(FYI, "Need negotiate, reconnecting tcons\n");
3716
3717        spin_lock(&cifs_tcp_ses_lock);
3718        list_for_each_entry(ses, &server->smb_ses_list, smb_ses_list) {
3719                list_for_each_entry(tcon, &ses->tcon_list, tcon_list) {
3720                        if (tcon->need_reconnect || tcon->need_reopen_files) {
3721                                tcon->tc_count++;
3722                                list_add_tail(&tcon->rlist, &tmp_list);
3723                                tcon_exist = true;
3724                        }
3725                }
3726                /*
3727                 * IPC has the same lifetime as its session and uses its
3728                 * refcount.
3729                 */
3730                if (ses->tcon_ipc && ses->tcon_ipc->need_reconnect) {
3731                        list_add_tail(&ses->tcon_ipc->rlist, &tmp_list);
3732                        tcon_exist = true;
3733                        ses->ses_count++;
3734                }
3735        }
3736        /*
3737         * Get the reference to server struct to be sure that the last call of
3738         * cifs_put_tcon() in the loop below won't release the server pointer.
3739         */
3740        if (tcon_exist)
3741                server->srv_count++;
3742
3743        spin_unlock(&cifs_tcp_ses_lock);
3744
3745        list_for_each_entry_safe(tcon, tcon2, &tmp_list, rlist) {
3746                rc = smb2_reconnect(SMB2_INTERNAL_CMD, tcon, server);
3747                if (!rc)
3748                        cifs_reopen_persistent_handles(tcon);
3749                else
3750                        resched = true;
3751                list_del_init(&tcon->rlist);
3752                if (tcon->ipc)
3753                        cifs_put_smb_ses(tcon->ses);
3754                else
3755                        cifs_put_tcon(tcon);
3756        }
3757
3758        cifs_dbg(FYI, "Reconnecting tcons finished\n");
3759        if (resched)
3760                queue_delayed_work(cifsiod_wq, &server->reconnect, 2 * HZ);
3761        mutex_unlock(&server->reconnect_mutex);
3762
3763        /* now we can safely release srv struct */
3764        if (tcon_exist)
3765                cifs_put_tcp_session(server, 1);
3766}
3767
3768int
3769SMB2_echo(struct TCP_Server_Info *server)
3770{
3771        struct smb2_echo_req *req;
3772        int rc = 0;
3773        struct kvec iov[1];
3774        struct smb_rqst rqst = { .rq_iov = iov,
3775                                 .rq_nvec = 1 };
3776        unsigned int total_len;
3777
3778        cifs_dbg(FYI, "In echo request\n");
3779
3780        if (server->tcpStatus == CifsNeedNegotiate) {
3781                /* No need to send echo on newly established connections */
3782                mod_delayed_work(cifsiod_wq, &server->reconnect, 0);
3783                return rc;
3784        }
3785
3786        rc = smb2_plain_req_init(SMB2_ECHO, NULL, server,
3787                                 (void **)&req, &total_len);
3788        if (rc)
3789                return rc;
3790
3791        req->sync_hdr.CreditRequest = cpu_to_le16(1);
3792
3793        iov[0].iov_len = total_len;
3794        iov[0].iov_base = (char *)req;
3795
3796        rc = cifs_call_async(server, &rqst, NULL, smb2_echo_callback, NULL,
3797                             server, CIFS_ECHO_OP, NULL);
3798        if (rc)
3799                cifs_dbg(FYI, "Echo request failed: %d\n", rc);
3800
3801        cifs_small_buf_release(req);
3802        return rc;
3803}
3804
3805void
3806SMB2_flush_free(struct smb_rqst *rqst)
3807{
3808        if (rqst && rqst->rq_iov)
3809                cifs_small_buf_release(rqst->rq_iov[0].iov_base); /* request */
3810}
3811
3812int
3813SMB2_flush_init(const unsigned int xid, struct smb_rqst *rqst,
3814                struct cifs_tcon *tcon, struct TCP_Server_Info *server,
3815                u64 persistent_fid, u64 volatile_fid)
3816{
3817        struct smb2_flush_req *req;
3818        struct kvec *iov = rqst->rq_iov;
3819        unsigned int total_len;
3820        int rc;
3821
3822        rc = smb2_plain_req_init(SMB2_FLUSH, tcon, server,
3823                                 (void **) &req, &total_len);
3824        if (rc)
3825                return rc;
3826
3827        req->PersistentFileId = persistent_fid;
3828        req->VolatileFileId = volatile_fid;
3829
3830        iov[0].iov_base = (char *)req;
3831        iov[0].iov_len = total_len;
3832
3833        return 0;
3834}
3835
3836int
3837SMB2_flush(const unsigned int xid, struct cifs_tcon *tcon, u64 persistent_fid,
3838           u64 volatile_fid)
3839{
3840        struct cifs_ses *ses = tcon->ses;
3841        struct smb_rqst rqst;
3842        struct kvec iov[1];
3843        struct kvec rsp_iov = {NULL, 0};
3844        struct TCP_Server_Info *server = cifs_pick_channel(ses);
3845        int resp_buftype = CIFS_NO_BUFFER;
3846        int flags = 0;
3847        int rc = 0;
3848
3849        cifs_dbg(FYI, "flush\n");
3850        if (!ses || !(ses->server))
3851                return -EIO;
3852
3853        if (smb3_encryption_required(tcon))
3854                flags |= CIFS_TRANSFORM_REQ;
3855
3856        memset(&rqst, 0, sizeof(struct smb_rqst));
3857        memset(&iov, 0, sizeof(iov));
3858        rqst.rq_iov = iov;
3859        rqst.rq_nvec = 1;
3860
3861        rc = SMB2_flush_init(xid, &rqst, tcon, server,
3862                             persistent_fid, volatile_fid);
3863        if (rc)
3864                goto flush_exit;
3865
3866        trace_smb3_flush_enter(xid, persistent_fid, tcon->tid, ses->Suid);
3867        rc = cifs_send_recv(xid, ses, server,
3868                            &rqst, &resp_buftype, flags, &rsp_iov);
3869
3870        if (rc != 0) {
3871                cifs_stats_fail_inc(tcon, SMB2_FLUSH_HE);
3872                trace_smb3_flush_err(xid, persistent_fid, tcon->tid, ses->Suid,
3873                                     rc);
3874        } else
3875                trace_smb3_flush_done(xid, persistent_fid, tcon->tid,
3876                                      ses->Suid);
3877
3878 flush_exit:
3879        SMB2_flush_free(&rqst);
3880        free_rsp_buf(resp_buftype, rsp_iov.iov_base);
3881        return rc;
3882}
3883
3884/*
3885 * To form a chain of read requests, any read requests after the first should
3886 * have the end_of_chain boolean set to true.
3887 */
3888static int
3889smb2_new_read_req(void **buf, unsigned int *total_len,
3890        struct cifs_io_parms *io_parms, struct cifs_readdata *rdata,
3891        unsigned int remaining_bytes, int request_type)
3892{
3893        int rc = -EACCES;
3894        struct smb2_read_plain_req *req = NULL;
3895        struct smb2_sync_hdr *shdr;
3896        struct TCP_Server_Info *server = io_parms->server;
3897
3898        rc = smb2_plain_req_init(SMB2_READ, io_parms->tcon, server,
3899                                 (void **) &req, total_len);
3900        if (rc)
3901                return rc;
3902
3903        if (server == NULL)
3904                return -ECONNABORTED;
3905
3906        shdr = &req->sync_hdr;
3907        shdr->ProcessId = cpu_to_le32(io_parms->pid);
3908
3909        req->PersistentFileId = io_parms->persistent_fid;
3910        req->VolatileFileId = io_parms->volatile_fid;
3911        req->ReadChannelInfoOffset = 0; /* reserved */
3912        req->ReadChannelInfoLength = 0; /* reserved */
3913        req->Channel = 0; /* reserved */
3914        req->MinimumCount = 0;
3915        req->Length = cpu_to_le32(io_parms->length);
3916        req->Offset = cpu_to_le64(io_parms->offset);
3917
3918        trace_smb3_read_enter(0 /* xid */,
3919                        io_parms->persistent_fid,
3920                        io_parms->tcon->tid, io_parms->tcon->ses->Suid,
3921                        io_parms->offset, io_parms->length);
3922#ifdef CONFIG_CIFS_SMB_DIRECT
3923        /*
3924         * If we want to do a RDMA write, fill in and append
3925         * smbd_buffer_descriptor_v1 to the end of read request
3926         */
3927        if (server->rdma && rdata && !server->sign &&
3928                rdata->bytes >= server->smbd_conn->rdma_readwrite_threshold) {
3929
3930                struct smbd_buffer_descriptor_v1 *v1;
3931                bool need_invalidate = server->dialect == SMB30_PROT_ID;
3932
3933                rdata->mr = smbd_register_mr(
3934                                server->smbd_conn, rdata->pages,
3935                                rdata->nr_pages, rdata->page_offset,
3936                                rdata->tailsz, true, need_invalidate);
3937                if (!rdata->mr)
3938                        return -EAGAIN;
3939
3940                req->Channel = SMB2_CHANNEL_RDMA_V1_INVALIDATE;
3941                if (need_invalidate)
3942                        req->Channel = SMB2_CHANNEL_RDMA_V1;
3943                req->ReadChannelInfoOffset =
3944                        cpu_to_le16(offsetof(struct smb2_read_plain_req, Buffer));
3945                req->ReadChannelInfoLength =
3946                        cpu_to_le16(sizeof(struct smbd_buffer_descriptor_v1));
3947                v1 = (struct smbd_buffer_descriptor_v1 *) &req->Buffer[0];
3948                v1->offset = cpu_to_le64(rdata->mr->mr->iova);
3949                v1->token = cpu_to_le32(rdata->mr->mr->rkey);
3950                v1->length = cpu_to_le32(rdata->mr->mr->length);
3951
3952                *total_len += sizeof(*v1) - 1;
3953        }
3954#endif
3955        if (request_type & CHAINED_REQUEST) {
3956                if (!(request_type & END_OF_CHAIN)) {
3957                        /* next 8-byte aligned request */
3958                        *total_len = DIV_ROUND_UP(*total_len, 8) * 8;
3959                        shdr->NextCommand = cpu_to_le32(*total_len);
3960                } else /* END_OF_CHAIN */
3961                        shdr->NextCommand = 0;
3962                if (request_type & RELATED_REQUEST) {
3963                        shdr->Flags |= SMB2_FLAGS_RELATED_OPERATIONS;
3964                        /*
3965                         * Related requests use info from previous read request
3966                         * in chain.
3967                         */
3968                        shdr->SessionId = 0xFFFFFFFFFFFFFFFF;
3969                        shdr->TreeId = 0xFFFFFFFF;
3970                        req->PersistentFileId = 0xFFFFFFFFFFFFFFFF;
3971                        req->VolatileFileId = 0xFFFFFFFFFFFFFFFF;
3972                }
3973        }
3974        if (remaining_bytes > io_parms->length)
3975                req->RemainingBytes = cpu_to_le32(remaining_bytes);
3976        else
3977                req->RemainingBytes = 0;
3978
3979        *buf = req;
3980        return rc;
3981}
3982
3983static void
3984smb2_readv_callback(struct mid_q_entry *mid)
3985{
3986        struct cifs_readdata *rdata = mid->callback_data;
3987        struct cifs_tcon *tcon = tlink_tcon(rdata->cfile->tlink);
3988        struct TCP_Server_Info *server = rdata->server;
3989        struct smb2_sync_hdr *shdr =
3990                                (struct smb2_sync_hdr *)rdata->iov[0].iov_base;
3991        struct cifs_credits credits = { .value = 0, .instance = 0 };
3992        struct smb_rqst rqst = { .rq_iov = &rdata->iov[1],
3993                                 .rq_nvec = 1,
3994                                 .rq_pages = rdata->pages,
3995                                 .rq_offset = rdata->page_offset,
3996                                 .rq_npages = rdata->nr_pages,
3997                                 .rq_pagesz = rdata->pagesz,
3998                                 .rq_tailsz = rdata->tailsz };
3999
4000        WARN_ONCE(rdata->server != mid->server,
4001                  "rdata server %p != mid server %p",
4002                  rdata->server, mid->server);
4003
4004        cifs_dbg(FYI, "%s: mid=%llu state=%d result=%d bytes=%u\n",
4005                 __func__, mid->mid, mid->mid_state, rdata->result,
4006                 rdata->bytes);
4007
4008        switch (mid->mid_state) {
4009        case MID_RESPONSE_RECEIVED:
4010                credits.value = le16_to_cpu(shdr->CreditRequest);
4011                credits.instance = server->reconnect_instance;
4012                /* result already set, check signature */
4013                if (server->sign && !mid->decrypted) {
4014                        int rc;
4015
4016                        rc = smb2_verify_signature(&rqst, server);
4017                        if (rc)
4018                                cifs_tcon_dbg(VFS, "SMB signature verification returned error = %d\n",
4019                                         rc);
4020                }
4021                /* FIXME: should this be counted toward the initiating task? */
4022                task_io_account_read(rdata->got_bytes);
4023                cifs_stats_bytes_read(tcon, rdata->got_bytes);
4024                break;
4025        case MID_REQUEST_SUBMITTED:
4026        case MID_RETRY_NEEDED:
4027                rdata->result = -EAGAIN;
4028                if (server->sign && rdata->got_bytes)
4029                        /* reset bytes number since we can not check a sign */
4030                        rdata->got_bytes = 0;
4031                /* FIXME: should this be counted toward the initiating task? */
4032                task_io_account_read(rdata->got_bytes);
4033                cifs_stats_bytes_read(tcon, rdata->got_bytes);
4034                break;
4035        case MID_RESPONSE_MALFORMED:
4036                credits.value = le16_to_cpu(shdr->CreditRequest);
4037                credits.instance = server->reconnect_instance;
4038                fallthrough;
4039        default:
4040                rdata->result = -EIO;
4041        }
4042#ifdef CONFIG_CIFS_SMB_DIRECT
4043        /*
4044         * If this rdata has a memmory registered, the MR can be freed
4045         * MR needs to be freed as soon as I/O finishes to prevent deadlock
4046         * because they have limited number and are used for future I/Os
4047         */
4048        if (rdata->mr) {
4049                smbd_deregister_mr(rdata->mr);
4050                rdata->mr = NULL;
4051        }
4052#endif
4053        if (rdata->result && rdata->result != -ENODATA) {
4054                cifs_stats_fail_inc(tcon, SMB2_READ_HE);
4055                trace_smb3_read_err(0 /* xid */,
4056                                    rdata->cfile->fid.persistent_fid,
4057                                    tcon->tid, tcon->ses->Suid, rdata->offset,
4058                                    rdata->bytes, rdata->result);
4059        } else
4060                trace_smb3_read_done(0 /* xid */,
4061                                     rdata->cfile->fid.persistent_fid,
4062                                     tcon->tid, tcon->ses->Suid,
4063                                     rdata->offset, rdata->got_bytes);
4064
4065        queue_work(cifsiod_wq, &rdata->work);
4066        DeleteMidQEntry(mid);
4067        add_credits(server, &credits, 0);
4068}
4069
4070/* smb2_async_readv - send an async read, and set up mid to handle result */
4071int
4072smb2_async_readv(struct cifs_readdata *rdata)
4073{
4074        int rc, flags = 0;
4075        char *buf;
4076        struct smb2_sync_hdr *shdr;
4077        struct cifs_io_parms io_parms;
4078        struct smb_rqst rqst = { .rq_iov = rdata->iov,
4079                                 .rq_nvec = 1 };
4080        struct TCP_Server_Info *server;
4081        struct cifs_tcon *tcon = tlink_tcon(rdata->cfile->tlink);
4082        unsigned int total_len;
4083
4084        cifs_dbg(FYI, "%s: offset=%llu bytes=%u\n",
4085                 __func__, rdata->offset, rdata->bytes);
4086
4087        if (!rdata->server)
4088                rdata->server = cifs_pick_channel(tcon->ses);
4089
4090        io_parms.tcon = tlink_tcon(rdata->cfile->tlink);
4091        io_parms.server = server = rdata->server;
4092        io_parms.offset = rdata->offset;
4093        io_parms.length = rdata->bytes;
4094        io_parms.persistent_fid = rdata->cfile->fid.persistent_fid;
4095        io_parms.volatile_fid = rdata->cfile->fid.volatile_fid;
4096        io_parms.pid = rdata->pid;
4097
4098        rc = smb2_new_read_req(
4099                (void **) &buf, &total_len, &io_parms, rdata, 0, 0);
4100        if (rc)
4101                return rc;
4102
4103        if (smb3_encryption_required(io_parms.tcon))
4104                flags |= CIFS_TRANSFORM_REQ;
4105
4106        rdata->iov[0].iov_base = buf;
4107        rdata->iov[0].iov_len = total_len;
4108
4109        shdr = (struct smb2_sync_hdr *)buf;
4110
4111        if (rdata->credits.value > 0) {
4112                shdr->CreditCharge = cpu_to_le16(DIV_ROUND_UP(rdata->bytes,
4113                                                SMB2_MAX_BUFFER_SIZE));
4114                shdr->CreditRequest = cpu_to_le16(le16_to_cpu(shdr->CreditCharge) + 8);
4115
4116                rc = adjust_credits(server, &rdata->credits, rdata->bytes);
4117                if (rc)
4118                        goto async_readv_out;
4119
4120                flags |= CIFS_HAS_CREDITS;
4121        }
4122
4123        kref_get(&rdata->refcount);
4124        rc = cifs_call_async(server, &rqst,
4125                             cifs_readv_receive, smb2_readv_callback,
4126                             smb3_handle_read_data, rdata, flags,
4127                             &rdata->credits);
4128        if (rc) {
4129                kref_put(&rdata->refcount, cifs_readdata_release);
4130                cifs_stats_fail_inc(io_parms.tcon, SMB2_READ_HE);
4131                trace_smb3_read_err(0 /* xid */, io_parms.persistent_fid,
4132                                    io_parms.tcon->tid,
4133                                    io_parms.tcon->ses->Suid,
4134                                    io_parms.offset, io_parms.length, rc);
4135        }
4136
4137async_readv_out:
4138        cifs_small_buf_release(buf);
4139        return rc;
4140}
4141
4142int
4143SMB2_read(const unsigned int xid, struct cifs_io_parms *io_parms,
4144          unsigned int *nbytes, char **buf, int *buf_type)
4145{
4146        struct smb_rqst rqst;
4147        int resp_buftype, rc;
4148        struct smb2_read_plain_req *req = NULL;
4149        struct smb2_read_rsp *rsp = NULL;
4150        struct kvec iov[1];
4151        struct kvec rsp_iov;
4152        unsigned int total_len;
4153        int flags = CIFS_LOG_ERROR;
4154        struct cifs_ses *ses = io_parms->tcon->ses;
4155
4156        if (!io_parms->server)
4157                io_parms->server = cifs_pick_channel(io_parms->tcon->ses);
4158
4159        *nbytes = 0;
4160        rc = smb2_new_read_req((void **)&req, &total_len, io_parms, NULL, 0, 0);
4161        if (rc)
4162                return rc;
4163
4164        if (smb3_encryption_required(io_parms->tcon))
4165                flags |= CIFS_TRANSFORM_REQ;
4166
4167        iov[0].iov_base = (char *)req;
4168        iov[0].iov_len = total_len;
4169
4170        memset(&rqst, 0, sizeof(struct smb_rqst));
4171        rqst.rq_iov = iov;
4172        rqst.rq_nvec = 1;
4173
4174        rc = cifs_send_recv(xid, ses, io_parms->server,
4175                            &rqst, &resp_buftype, flags, &rsp_iov);
4176        rsp = (struct smb2_read_rsp *)rsp_iov.iov_base;
4177
4178        if (rc) {
4179                if (rc != -ENODATA) {
4180                        cifs_stats_fail_inc(io_parms->tcon, SMB2_READ_HE);
4181                        cifs_dbg(VFS, "Send error in read = %d\n", rc);
4182                        trace_smb3_read_err(xid, req->PersistentFileId,
4183                                            io_parms->tcon->tid, ses->Suid,
4184                                            io_parms->offset, io_parms->length,
4185                                            rc);
4186                } else
4187                        trace_smb3_read_done(xid, req->PersistentFileId,
4188                                    io_parms->tcon->tid, ses->Suid,
4189                                    io_parms->offset, 0);
4190                free_rsp_buf(resp_buftype, rsp_iov.iov_base);
4191                cifs_small_buf_release(req);
4192                return rc == -ENODATA ? 0 : rc;
4193        } else
4194                trace_smb3_read_done(xid, req->PersistentFileId,
4195                                    io_parms->tcon->tid, ses->Suid,
4196                                    io_parms->offset, io_parms->length);
4197
4198        cifs_small_buf_release(req);
4199
4200        *nbytes = le32_to_cpu(rsp->DataLength);
4201        if ((*nbytes > CIFS_MAX_MSGSIZE) ||
4202            (*nbytes > io_parms->length)) {
4203                cifs_dbg(FYI, "bad length %d for count %d\n",
4204                         *nbytes, io_parms->length);
4205                rc = -EIO;
4206                *nbytes = 0;
4207        }
4208
4209        if (*buf) {
4210                memcpy(*buf, (char *)rsp + rsp->DataOffset, *nbytes);
4211                free_rsp_buf(resp_buftype, rsp_iov.iov_base);
4212        } else if (resp_buftype != CIFS_NO_BUFFER) {
4213                *buf = rsp_iov.iov_base;
4214                if (resp_buftype == CIFS_SMALL_BUFFER)
4215                        *buf_type = CIFS_SMALL_BUFFER;
4216                else if (resp_buftype == CIFS_LARGE_BUFFER)
4217                        *buf_type = CIFS_LARGE_BUFFER;
4218        }
4219        return rc;
4220}
4221
4222/*
4223 * Check the mid_state and signature on received buffer (if any), and queue the
4224 * workqueue completion task.
4225 */
4226static void
4227smb2_writev_callback(struct mid_q_entry *mid)
4228{
4229        struct cifs_writedata *wdata = mid->callback_data;
4230        struct cifs_tcon *tcon = tlink_tcon(wdata->cfile->tlink);
4231        struct TCP_Server_Info *server = wdata->server;
4232        unsigned int written;
4233        struct smb2_write_rsp *rsp = (struct smb2_write_rsp *)mid->resp_buf;
4234        struct cifs_credits credits = { .value = 0, .instance = 0 };
4235
4236        WARN_ONCE(wdata->server != mid->server,
4237                  "wdata server %p != mid server %p",
4238                  wdata->server, mid->server);
4239
4240        switch (mid->mid_state) {
4241        case MID_RESPONSE_RECEIVED:
4242                credits.value = le16_to_cpu(rsp->sync_hdr.CreditRequest);
4243                credits.instance = server->reconnect_instance;
4244                wdata->result = smb2_check_receive(mid, server, 0);
4245                if (wdata->result != 0)
4246                        break;
4247
4248                written = le32_to_cpu(rsp->DataLength);
4249                /*
4250                 * Mask off high 16 bits when bytes written as returned
4251                 * by the server is greater than bytes requested by the
4252                 * client. OS/2 servers are known to set incorrect
4253                 * CountHigh values.
4254                 */
4255                if (written > wdata->bytes)
4256                        written &= 0xFFFF;
4257
4258                if (written < wdata->bytes)
4259                        wdata->result = -ENOSPC;
4260                else
4261                        wdata->bytes = written;
4262                break;
4263        case MID_REQUEST_SUBMITTED:
4264        case MID_RETRY_NEEDED:
4265                wdata->result = -EAGAIN;
4266                break;
4267        case MID_RESPONSE_MALFORMED:
4268                credits.value = le16_to_cpu(rsp->sync_hdr.CreditRequest);
4269                credits.instance = server->reconnect_instance;
4270                fallthrough;
4271        default:
4272                wdata->result = -EIO;
4273                break;
4274        }
4275#ifdef CONFIG_CIFS_SMB_DIRECT
4276        /*
4277         * If this wdata has a memory registered, the MR can be freed
4278         * The number of MRs available is limited, it's important to recover
4279         * used MR as soon as I/O is finished. Hold MR longer in the later
4280         * I/O process can possibly result in I/O deadlock due to lack of MR
4281         * to send request on I/O retry
4282         */
4283        if (wdata->mr) {
4284                smbd_deregister_mr(wdata->mr);
4285                wdata->mr = NULL;
4286        }
4287#endif
4288        if (wdata->result) {
4289                cifs_stats_fail_inc(tcon, SMB2_WRITE_HE);
4290                trace_smb3_write_err(0 /* no xid */,
4291                                     wdata->cfile->fid.persistent_fid,
4292                                     tcon->tid, tcon->ses->Suid, wdata->offset,
4293                                     wdata->bytes, wdata->result);
4294                if (wdata->result == -ENOSPC)
4295                        pr_warn_once("Out of space writing to %s\n",
4296                                     tcon->treeName);
4297        } else
4298                trace_smb3_write_done(0 /* no xid */,
4299                                      wdata->cfile->fid.persistent_fid,
4300                                      tcon->tid, tcon->ses->Suid,
4301                                      wdata->offset, wdata->bytes);
4302
4303        queue_work(cifsiod_wq, &wdata->work);
4304        DeleteMidQEntry(mid);
4305        add_credits(server, &credits, 0);
4306}
4307
4308/* smb2_async_writev - send an async write, and set up mid to handle result */
4309int
4310smb2_async_writev(struct cifs_writedata *wdata,
4311                  void (*release)(struct kref *kref))
4312{
4313        int rc = -EACCES, flags = 0;
4314        struct smb2_write_req *req = NULL;
4315        struct smb2_sync_hdr *shdr;
4316        struct cifs_tcon *tcon = tlink_tcon(wdata->cfile->tlink);
4317        struct TCP_Server_Info *server = wdata->server;
4318        struct kvec iov[1];
4319        struct smb_rqst rqst = { };
4320        unsigned int total_len;
4321
4322        if (!wdata->server)
4323                server = wdata->server = cifs_pick_channel(tcon->ses);
4324
4325        rc = smb2_plain_req_init(SMB2_WRITE, tcon, server,
4326                                 (void **) &req, &total_len);
4327        if (rc)
4328                return rc;
4329
4330        if (smb3_encryption_required(tcon))
4331                flags |= CIFS_TRANSFORM_REQ;
4332
4333        shdr = (struct smb2_sync_hdr *)req;
4334        shdr->ProcessId = cpu_to_le32(wdata->cfile->pid);
4335
4336        req->PersistentFileId = wdata->cfile->fid.persistent_fid;
4337        req->VolatileFileId = wdata->cfile->fid.volatile_fid;
4338        req->WriteChannelInfoOffset = 0;
4339        req->WriteChannelInfoLength = 0;
4340        req->Channel = 0;
4341        req->Offset = cpu_to_le64(wdata->offset);
4342        req->DataOffset = cpu_to_le16(
4343                                offsetof(struct smb2_write_req, Buffer));
4344        req->RemainingBytes = 0;
4345
4346        trace_smb3_write_enter(0 /* xid */, wdata->cfile->fid.persistent_fid,
4347                tcon->tid, tcon->ses->Suid, wdata->offset, wdata->bytes);
4348#ifdef CONFIG_CIFS_SMB_DIRECT
4349        /*
4350         * If we want to do a server RDMA read, fill in and append
4351         * smbd_buffer_descriptor_v1 to the end of write request
4352         */
4353        if (server->rdma && !server->sign && wdata->bytes >=
4354                server->smbd_conn->rdma_readwrite_threshold) {
4355
4356                struct smbd_buffer_descriptor_v1 *v1;
4357                bool need_invalidate = server->dialect == SMB30_PROT_ID;
4358
4359                wdata->mr = smbd_register_mr(
4360                                server->smbd_conn, wdata->pages,
4361                                wdata->nr_pages, wdata->page_offset,
4362                                wdata->tailsz, false, need_invalidate);
4363                if (!wdata->mr) {
4364                        rc = -EAGAIN;
4365                        goto async_writev_out;
4366                }
4367                req->Length = 0;
4368                req->DataOffset = 0;
4369                if (wdata->nr_pages > 1)
4370                        req->RemainingBytes =
4371                                cpu_to_le32(
4372                                        (wdata->nr_pages - 1) * wdata->pagesz -
4373                                        wdata->page_offset + wdata->tailsz
4374                                );
4375                else
4376                        req->RemainingBytes = cpu_to_le32(wdata->tailsz);
4377                req->Channel = SMB2_CHANNEL_RDMA_V1_I