fs/cifs/smb2ops.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  *  SMB2 version specific operations
   4  *
   5  *  Copyright (c) 2012, Jeff Layton <[email protected]>
   6  */
   7
   8 #include <linux/pagemap.h>
   9 #include <linux/vfs.h>
  10 #include <linux/falloc.h>
  11 #include <linux/scatterlist.h>
  12 #include <linux/uuid.h>
  13 #include <linux/sort.h>
  14 #include <crypto/aead.h>
  15 #include <linux/fiemap.h>
  16 #include "cifsfs.h"
  17 #include "cifsglob.h"
  18 #include "smb2pdu.h"
  19 #include "smb2proto.h"
  20 #include "cifsproto.h"
  21 #include "cifs_debug.h"
  22 #include "cifs_unicode.h"
  23 #include "smb2status.h"
  24 #include "smb2glob.h"
  25 #include "cifs_ioctl.h"
  26 #include "smbdirect.h"
  27 #include "fs_context.h"
  28
  29 /* Change credits for different ops and return the total number of credits */
  30 static int
  31 change_conf(struct TCP_Server_Info *server)
  32 {
  33         server->credits += server->echo_credits + server->oplock_credits;
  34         server->oplock_credits = server->echo_credits = 0;
  35         switch (server->credits) {
  36         case 0:
  37                 return 0;
  38         case 1:
  39                 server->echoes = false;
  40                 server->oplocks = false;
  41                 break;
  42         case 2:
  43                 server->echoes = true;
  44                 server->oplocks = false;
  45                 server->echo_credits = 1;
  46                 break;
  47         default:
  48                 server->echoes = true;
  49                 if (enable_oplocks) {
  50                         server->oplocks = true;
  51                         server->oplock_credits = 1;
  52                 } else
  53                         server->oplocks = false;
  54
  55                 server->echo_credits = 1;
  56         }
  57         server->credits -= server->echo_credits + server->oplock_credits;
  58         return server->credits + server->echo_credits + server->oplock_credits;
  59 }
  60
  61 static void
  62 smb2_add_credits(struct TCP_Server_Info *server,
  63                  const struct cifs_credits *credits, const int optype)
  64 {
  65         int *val, rc = -1;
  66         int scredits, in_flight;
  67         unsigned int add = credits->value;
  68         unsigned int instance = credits->instance;
  69         bool reconnect_detected = false;
  70         bool reconnect_with_invalid_credits = false;
  71
  72         spin_lock(&server->req_lock);
  73         val = server->ops->get_credits_field(server, optype);
  74
  75         /* eg found case where write overlapping reconnect messed up credits */
  76         if (((optype & CIFS_OP_MASK) == CIFS_NEG_OP) && (*val != 0))
  77                 reconnect_with_invalid_credits = true;
  78
  79         if ((instance == 0) || (instance == server->reconnect_instance))
  80                 *val += add;
  81         else
  82                 reconnect_detected = true;
  83
  84         if (*val > 65000) {
  85                 *val = 65000; /* Don't get near 64K credits, avoid srv bugs */
  86                 pr_warn_once("server overflowed SMB3 credits\n");
  87         }
  88         server->in_flight--;
  89         if (server->in_flight == 0 &&
  90            ((optype & CIFS_OP_MASK) != CIFS_NEG_OP) &&
  91            ((optype & CIFS_OP_MASK) != CIFS_SESS_OP))
  92                 rc = change_conf(server);
  93         /*
  94          * Sometimes server returns 0 credits on oplock break ack - we need to
  95          * rebalance credits in this case.
  96          */
  97         else if (server->in_flight > 0 && server->oplock_credits == 0 &&
  98                  server->oplocks) {
  99                 if (server->credits > 1) {
 100                         server->credits--;
 101                         server->oplock_credits++;
 102                 }
 103         }
 104         scredits = *val;
 105         in_flight = server->in_flight;
 106         spin_unlock(&server->req_lock);
 107         wake_up(&server->request_q);
 108
 109         if (reconnect_detected) {
 110                 trace_smb3_reconnect_detected(server->CurrentMid,
 111                         server->conn_id, server->hostname, scredits, add, in_flight);
 112
 113                 cifs_dbg(FYI, "trying to put %d credits from the old server instance %d\n",
 114                          add, instance);
 115         }
 116
 117         if (reconnect_with_invalid_credits) {
 118                 trace_smb3_reconnect_with_invalid_credits(server->CurrentMid,
 119                         server->conn_id, server->hostname, scredits, add, in_flight);
 120                 cifs_dbg(FYI, "Negotiate operation when server credits is non-zero. Optype: %d, server credits: %d, credits added: %d\n",
 121                          optype, scredits, add);
 122         }
 123
 124         if (server->tcpStatus == CifsNeedReconnect
 125             || server->tcpStatus == CifsExiting)
 126                 return;
 127
 128         switch (rc) {
 129         case -1:
 130                 /* change_conf hasn't been executed */
 131                 break;
 132         case 0:
 133                 cifs_server_dbg(VFS, "Possible client or server bug - zero credits\n");
 134                 break;
 135         case 1:
 136                 cifs_server_dbg(VFS, "disabling echoes and oplocks\n");
 137                 break;
 138         case 2:
 139                 cifs_dbg(FYI, "disabling oplocks\n");
 140                 break;
 141         default:
 142                 /* change_conf rebalanced credits for different types */
 143                 break;
 144         }
 145
 146         trace_smb3_add_credits(server->CurrentMid,
 147                         server->conn_id, server->hostname, scredits, add, in_flight);
 148         cifs_dbg(FYI, "%s: added %u credits total=%d\n", __func__, add, scredits);
 149 }
 150
 151 static void
 152 smb2_set_credits(struct TCP_Server_Info *server, const int val)
 153 {
 154         int scredits, in_flight;
 155
 156         spin_lock(&server->req_lock);
 157         server->credits = val;
 158         if (val == 1)
 159                 server->reconnect_instance++;
 160         scredits = server->credits;
 161         in_flight = server->in_flight;
 162         spin_unlock(&server->req_lock);
 163
 164         trace_smb3_set_credits(server->CurrentMid,
 165                         server->conn_id, server->hostname, scredits, val, in_flight);
 166         cifs_dbg(FYI, "%s: set %u credits\n", __func__, val);
 167
 168         /* don't log while holding the lock */
 169         if (val == 1)
 170                 cifs_dbg(FYI, "set credits to 1 due to smb2 reconnect\n");
 171 }
 172
 173 static int *
 174 smb2_get_credits_field(struct TCP_Server_Info *server, const int optype)
 175 {
 176         switch (optype) {
 177         case CIFS_ECHO_OP:
 178                 return &server->echo_credits;
 179         case CIFS_OBREAK_OP:
 180                 return &server->oplock_credits;
 181         default:
 182                 return &server->credits;
 183         }
 184 }
 185
 186 static unsigned int
 187 smb2_get_credits(struct mid_q_entry *mid)
 188 {
 189         return mid->credits_received;
 190 }
 191
 192 static int
 193 smb2_wait_mtu_credits(struct TCP_Server_Info *server, unsigned int size,
 194                       unsigned int *num, struct cifs_credits *credits)
 195 {
 196         int rc = 0;
 197         unsigned int scredits, in_flight;
 198
 199         spin_lock(&server->req_lock);
 200         while (1) {
 201                 if (server->credits <= 0) {
 202                         spin_unlock(&server->req_lock);
 203                         cifs_num_waiters_inc(server);
 204                         rc = wait_event_killable(server->request_q,
 205                                 has_credits(server, &server->credits, 1));
 206                         cifs_num_waiters_dec(server);
 207                         if (rc)
 208                                 return rc;
 209                         spin_lock(&server->req_lock);
 210                 } else {
 211                         if (server->tcpStatus == CifsExiting) {
 212                                 spin_unlock(&server->req_lock);
 213                                 return -ENOENT;
 214                         }
 215
 216                         scredits = server->credits;
 217                         /* can deadlock with reopen */
 218                         if (scredits <= 8) {
 219                                 *num = SMB2_MAX_BUFFER_SIZE;
 220                                 credits->value = 0;
 221                                 credits->instance = 0;
 222                                 break;
 223                         }
 224
 225                         /* leave some credits for reopen and other ops */
 226                         scredits -= 8;
 227                         *num = min_t(unsigned int, size,
 228                                      scredits * SMB2_MAX_BUFFER_SIZE);
 229
 230                         credits->value =
 231                                 DIV_ROUND_UP(*num, SMB2_MAX_BUFFER_SIZE);
 232                         credits->instance = server->reconnect_instance;
 233                         server->credits -= credits->value;
 234                         server->in_flight++;
 235                         if (server->in_flight > server->max_in_flight)
 236                                 server->max_in_flight = server->in_flight;
 237                         break;
 238                 }
 239         }
 240         scredits = server->credits;
 241         in_flight = server->in_flight;
 242         spin_unlock(&server->req_lock);
 243
 244         trace_smb3_add_credits(server->CurrentMid,
 245                         server->conn_id, server->hostname, scredits, -(credits->value), in_flight);
 246         cifs_dbg(FYI, "%s: removed %u credits total=%d\n",
 247                         __func__, credits->value, scredits);
 248
 249         return rc;
 250 }
 251
 252 static int
 253 smb2_adjust_credits(struct TCP_Server_Info *server,
 254                     struct cifs_credits *credits,
 255                     const unsigned int payload_size)
 256 {
 257         int new_val = DIV_ROUND_UP(payload_size, SMB2_MAX_BUFFER_SIZE);
 258         int scredits, in_flight;
 259
 260         if (!credits->value || credits->value == new_val)
 261                 return 0;
 262
 263         if (credits->value < new_val) {
 264                 trace_smb3_too_many_credits(server->CurrentMid,
 265                                 server->conn_id, server->hostname, 0, credits->value - new_val, 0);
 266                 cifs_server_dbg(VFS, "request has less credits (%d) than required (%d)",
 267                                 credits->value, new_val);
 268
 269                 return -ENOTSUPP;
 270         }
 271
 272         spin_lock(&server->req_lock);
 273
 274         if (server->reconnect_instance != credits->instance) {
 275                 scredits = server->credits;
 276                 in_flight = server->in_flight;
 277                 spin_unlock(&server->req_lock);
 278
 279                 trace_smb3_reconnect_detected(server->CurrentMid,
 280                         server->conn_id, server->hostname, scredits,
 281                         credits->value - new_val, in_flight);
 282                 cifs_server_dbg(VFS, "trying to return %d credits to old session\n",
 283                          credits->value - new_val);
 284                 return -EAGAIN;
 285         }
 286
 287         server->credits += credits->value - new_val;
 288         scredits = server->credits;
 289         in_flight = server->in_flight;
 290         spin_unlock(&server->req_lock);
 291         wake_up(&server->request_q);
 292
 293         trace_smb3_add_credits(server->CurrentMid,
 294                         server->conn_id, server->hostname, scredits,
 295                         credits->value - new_val, in_flight);
 296         cifs_dbg(FYI, "%s: adjust added %u credits total=%d\n",
 297                         __func__, credits->value - new_val, scredits);
 298
 299         credits->value = new_val;
 300
 301         return 0;
 302 }
 303
 304 static __u64
 305 smb2_get_next_mid(struct TCP_Server_Info *server)
 306 {
 307         __u64 mid;
 308         /* for SMB2 we need the current value */
 309         spin_lock(&GlobalMid_Lock);
 310         mid = server->CurrentMid++;
 311         spin_unlock(&GlobalMid_Lock);
 312         return mid;
 313 }
 314
 315 static void
 316 smb2_revert_current_mid(struct TCP_Server_Info *server, const unsigned int val)
 317 {
 318         spin_lock(&GlobalMid_Lock);
 319         if (server->CurrentMid >= val)
 320                 server->CurrentMid -= val;
 321         spin_unlock(&GlobalMid_Lock);
 322 }
 323
 324 static struct mid_q_entry *
 325 __smb2_find_mid(struct TCP_Server_Info *server, char *buf, bool dequeue)
 326 {
 327         struct mid_q_entry *mid;
 328         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
 329         __u64 wire_mid = le64_to_cpu(shdr->MessageId);
 330
 331         if (shdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM) {
 332                 cifs_server_dbg(VFS, "Encrypted frame parsing not supported yet\n");
 333                 return NULL;
 334         }
 335
 336         spin_lock(&GlobalMid_Lock);
 337         list_for_each_entry(mid, &server->pending_mid_q, qhead) {
 338                 if ((mid->mid == wire_mid) &&
 339                     (mid->mid_state == MID_REQUEST_SUBMITTED) &&
 340                     (mid->command == shdr->Command)) {
 341                         kref_get(&mid->refcount);
 342                         if (dequeue) {
 343                                 list_del_init(&mid->qhead);
 344                                 mid->mid_flags |= MID_DELETED;
 345                         }
 346                         spin_unlock(&GlobalMid_Lock);
 347                         return mid;
 348                 }
 349         }
 350         spin_unlock(&GlobalMid_Lock);
 351         return NULL;
 352 }
 353
 354 static struct mid_q_entry *
 355 smb2_find_mid(struct TCP_Server_Info *server, char *buf)
 356 {
 357         return __smb2_find_mid(server, buf, false);
 358 }
 359
 360 static struct mid_q_entry *
 361 smb2_find_dequeue_mid(struct TCP_Server_Info *server, char *buf)
 362 {
 363         return __smb2_find_mid(server, buf, true);
 364 }
 365
 366 static void
 367 smb2_dump_detail(void *buf, struct TCP_Server_Info *server)
 368 {
 369 #ifdef CONFIG_CIFS_DEBUG2
 370         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
 371
 372         cifs_server_dbg(VFS, "Cmd: %d Err: 0x%x Flags: 0x%x Mid: %llu Pid: %d\n",
 373                  shdr->Command, shdr->Status, shdr->Flags, shdr->MessageId,
 374                  shdr->ProcessId);
 375         cifs_server_dbg(VFS, "smb buf %p len %u\n", buf,
 376                  server->ops->calc_smb_size(buf, server));
 377 #endif
 378 }
 379
 380 static bool
 381 smb2_need_neg(struct TCP_Server_Info *server)
 382 {
 383         return server->max_read == 0;
 384 }
 385
 386 static int
 387 smb2_negotiate(const unsigned int xid, struct cifs_ses *ses)
 388 {
 389         int rc;
 390
 391         cifs_ses_server(ses)->CurrentMid = 0;
 392         rc = SMB2_negotiate(xid, ses);
 393         /* BB we probably don't need to retry with modern servers */
 394         if (rc == -EAGAIN)
 395                 rc = -EHOSTDOWN;
 396         return rc;
 397 }
 398
 399 static unsigned int
 400 smb2_negotiate_wsize(struct cifs_tcon *tcon, struct smb3_fs_context *ctx)
 401 {
 402         struct TCP_Server_Info *server = tcon->ses->server;
 403         unsigned int wsize;
 404
 405         /* start with specified wsize, or default */
 406         wsize = ctx->wsize ? ctx->wsize : CIFS_DEFAULT_IOSIZE;
 407         wsize = min_t(unsigned int, wsize, server->max_write);
 408         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
 409                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
 410
 411         return wsize;
 412 }
 413
 414 static unsigned int
 415 smb3_negotiate_wsize(struct cifs_tcon *tcon, struct smb3_fs_context *ctx)
 416 {
 417         struct TCP_Server_Info *server = tcon->ses->server;
 418         unsigned int wsize;
 419
 420         /* start with specified wsize, or default */
 421         wsize = ctx->wsize ? ctx->wsize : SMB3_DEFAULT_IOSIZE;
 422         wsize = min_t(unsigned int, wsize, server->max_write);
 423 #ifdef CONFIG_CIFS_SMB_DIRECT
 424         if (server->rdma) {
 425                 if (server->sign)
 426                         /*
 427                          * Account for SMB2 data transfer packet header and
 428                          * possible encryption header
 429                          */
 430                         wsize = min_t(unsigned int,
 431                                 wsize,
 432                                 server->smbd_conn->max_fragmented_send_size -
 433                                         SMB2_READWRITE_PDU_HEADER_SIZE -
 434                                         sizeof(struct smb2_transform_hdr));
 435                 else
 436                         wsize = min_t(unsigned int,
 437                                 wsize, server->smbd_conn->max_readwrite_size);
 438         }
 439 #endif
 440         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
 441                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
 442
 443         return wsize;
 444 }
 445
 446 static unsigned int
 447 smb2_negotiate_rsize(struct cifs_tcon *tcon, struct smb3_fs_context *ctx)
 448 {
 449         struct TCP_Server_Info *server = tcon->ses->server;
 450         unsigned int rsize;
 451
 452         /* start with specified rsize, or default */
 453         rsize = ctx->rsize ? ctx->rsize : CIFS_DEFAULT_IOSIZE;
 454         rsize = min_t(unsigned int, rsize, server->max_read);
 455
 456         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
 457                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
 458
 459         return rsize;
 460 }
 461
 462 static unsigned int
 463 smb3_negotiate_rsize(struct cifs_tcon *tcon, struct smb3_fs_context *ctx)
 464 {
 465         struct TCP_Server_Info *server = tcon->ses->server;
 466         unsigned int rsize;
 467
 468         /* start with specified rsize, or default */
 469         rsize = ctx->rsize ? ctx->rsize : SMB3_DEFAULT_IOSIZE;
 470         rsize = min_t(unsigned int, rsize, server->max_read);
 471 #ifdef CONFIG_CIFS_SMB_DIRECT
 472         if (server->rdma) {
 473                 if (server->sign)
 474                         /*
 475                          * Account for SMB2 data transfer packet header and
 476                          * possible encryption header
 477                          */
 478                         rsize = min_t(unsigned int,
 479                                 rsize,
 480                                 server->smbd_conn->max_fragmented_recv_size -
 481                                         SMB2_READWRITE_PDU_HEADER_SIZE -
 482                                         sizeof(struct smb2_transform_hdr));
 483                 else
 484                         rsize = min_t(unsigned int,
 485                                 rsize, server->smbd_conn->max_readwrite_size);
 486         }
 487 #endif
 488
 489         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
 490                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
 491
 492         return rsize;
 493 }
 494
 495 static int
 496 parse_server_interfaces(struct network_interface_info_ioctl_rsp *buf,
 497                         size_t buf_len,
 498                         struct cifs_server_iface **iface_list,
 499                         size_t *iface_count)
 500 {
 501         struct network_interface_info_ioctl_rsp *p;
 502         struct sockaddr_in *addr4;
 503         struct sockaddr_in6 *addr6;
 504         struct iface_info_ipv4 *p4;
 505         struct iface_info_ipv6 *p6;
 506         struct cifs_server_iface *info;
 507         ssize_t bytes_left;
 508         size_t next = 0;
 509         int nb_iface = 0;
 510         int rc = 0;
 511
 512         *iface_list = NULL;
 513         *iface_count = 0;
 514
 515         /*
 516          * Fist pass: count and sanity check
 517          */
 518
 519         bytes_left = buf_len;
 520         p = buf;
 521         while (bytes_left >= sizeof(*p)) {
 522                 nb_iface++;
 523                 next = le32_to_cpu(p->Next);
 524                 if (!next) {
 525                         bytes_left -= sizeof(*p);
 526                         break;
 527                 }
 528                 p = (struct network_interface_info_ioctl_rsp *)((u8 *)p+next);
 529                 bytes_left -= next;
 530         }
 531
 532         if (!nb_iface) {
 533                 cifs_dbg(VFS, "%s: malformed interface info\n", __func__);
 534                 rc = -EINVAL;
 535                 goto out;
 536         }
 537
 538         /* Azure rounds the buffer size up 8, to a 16 byte boundary */
 539         if ((bytes_left > 8) || p->Next)
 540                 cifs_dbg(VFS, "%s: incomplete interface info\n", __func__);
 541
 542
 543         /*
 544          * Second pass: extract info to internal structure
 545          */
 546
 547         *iface_list = kcalloc(nb_iface, sizeof(**iface_list), GFP_KERNEL);
 548         if (!*iface_list) {
 549                 rc = -ENOMEM;
 550                 goto out;
 551         }
 552
 553         info = *iface_list;
 554         bytes_left = buf_len;
 555         p = buf;
 556         while (bytes_left >= sizeof(*p)) {
 557                 info->speed = le64_to_cpu(p->LinkSpeed);
 558                 info->rdma_capable = le32_to_cpu(p->Capability & RDMA_CAPABLE);
 559                 info->rss_capable = le32_to_cpu(p->Capability & RSS_CAPABLE);
 560
 561                 cifs_dbg(FYI, "%s: adding iface %zu\n", __func__, *iface_count);
 562                 cifs_dbg(FYI, "%s: speed %zu bps\n", __func__, info->speed);
 563                 cifs_dbg(FYI, "%s: capabilities 0x%08x\n", __func__,
 564                          le32_to_cpu(p->Capability));
 565
 566                 switch (p->Family) {
 567                 /*
 568                  * The kernel and wire socket structures have the same
 569                  * layout and use network byte order but make the
 570                  * conversion explicit in case either one changes.
 571                  */
 572                 case INTERNETWORK:
 573                         addr4 = (struct sockaddr_in *)&info->sockaddr;
 574                         p4 = (struct iface_info_ipv4 *)p->Buffer;
 575                         addr4->sin_family = AF_INET;
 576                         memcpy(&addr4->sin_addr, &p4->IPv4Address, 4);
 577
 578                         /* [MS-SMB2] 2.2.32.5.1.1 Clients MUST ignore these */
 579                         addr4->sin_port = cpu_to_be16(CIFS_PORT);
 580
 581                         cifs_dbg(FYI, "%s: ipv4 %pI4\n", __func__,
 582                                  &addr4->sin_addr);
 583                         break;
 584                 case INTERNETWORKV6:
 585                         addr6 = (struct sockaddr_in6 *)&info->sockaddr;
 586                         p6 = (struct iface_info_ipv6 *)p->Buffer;
 587                         addr6->sin6_family = AF_INET6;
 588                         memcpy(&addr6->sin6_addr, &p6->IPv6Address, 16);
 589
 590                         /* [MS-SMB2] 2.2.32.5.1.2 Clients MUST ignore these */
 591                         addr6->sin6_flowinfo = 0;
 592                         addr6->sin6_scope_id = 0;
 593                         addr6->sin6_port = cpu_to_be16(CIFS_PORT);
 594
 595                         cifs_dbg(FYI, "%s: ipv6 %pI6\n", __func__,
 596                                  &addr6->sin6_addr);
 597                         break;
 598                 default:
 599                         cifs_dbg(VFS,
 600                                  "%s: skipping unsupported socket family\n",
 601                                  __func__);
 602                         goto next_iface;
 603                 }
 604
 605                 (*iface_count)++;
 606                 info++;
 607 next_iface:
 608                 next = le32_to_cpu(p->Next);
 609                 if (!next)
 610                         break;
 611                 p = (struct network_interface_info_ioctl_rsp *)((u8 *)p+next);
 612                 bytes_left -= next;
 613         }
 614
 615         if (!*iface_count) {
 616                 rc = -EINVAL;
 617                 goto out;
 618         }
 619
 620 out:
 621         if (rc) {
 622                 kfree(*iface_list);
 623                 *iface_count = 0;
 624                 *iface_list = NULL;
 625         }
 626         return rc;
 627 }
 628
 629 static int compare_iface(const void *ia, const void *ib)
 630 {
 631         const struct cifs_server_iface *a = (struct cifs_server_iface *)ia;
 632         const struct cifs_server_iface *b = (struct cifs_server_iface *)ib;
 633
 634         return a->speed == b->speed ? 0 : (a->speed > b->speed ? -1 : 1);
 635 }
 636
 637 static int
 638 SMB3_request_interfaces(const unsigned int xid, struct cifs_tcon *tcon)
 639 {
 640         int rc;
 641         unsigned int ret_data_len = 0;
 642         struct network_interface_info_ioctl_rsp *out_buf = NULL;
 643         struct cifs_server_iface *iface_list;
 644         size_t iface_count;
 645         struct cifs_ses *ses = tcon->ses;
 646
 647         rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
 648                         FSCTL_QUERY_NETWORK_INTERFACE_INFO, true /* is_fsctl */,
 649                         NULL /* no data input */, 0 /* no data input */,
 650                         CIFSMaxBufSize, (char **)&out_buf, &ret_data_len);
 651         if (rc == -EOPNOTSUPP) {
 652                 cifs_dbg(FYI,
 653                          "server does not support query network interfaces\n");
 654                 goto out;
 655         } else if (rc != 0) {
 656                 cifs_tcon_dbg(VFS, "error %d on ioctl to get interface list\n", rc);
 657                 goto out;
 658         }
 659
 660         rc = parse_server_interfaces(out_buf, ret_data_len,
 661                                      &iface_list, &iface_count);
 662         if (rc)
 663                 goto out;
 664
 665         /* sort interfaces from fastest to slowest */
 666         sort(iface_list, iface_count, sizeof(*iface_list), compare_iface, NULL);
 667
 668         spin_lock(&ses->iface_lock);
 669         kfree(ses->iface_list);
 670         ses->iface_list = iface_list;
 671         ses->iface_count = iface_count;
 672         ses->iface_last_update = jiffies;
 673         spin_unlock(&ses->iface_lock);
 674
 675 out:
 676         kfree(out_buf);
 677         return rc;
 678 }
 679
 680 static void
 681 smb2_close_cached_fid(struct kref *ref)
 682 {
 683         struct cached_fid *cfid = container_of(ref, struct cached_fid,
 684                                                refcount);
 685
 686         if (cfid->is_valid) {
 687                 cifs_dbg(FYI, "clear cached root file handle\n");
 688                 SMB2_close(0, cfid->tcon, cfid->fid->persistent_fid,
 689                            cfid->fid->volatile_fid);
 690                 cfid->is_valid = false;
 691                 cfid->file_all_info_is_valid = false;
 692                 cfid->has_lease = false;
 693         }
 694 }
 695
 696 void close_shroot(struct cached_fid *cfid)
 697 {
 698         mutex_lock(&cfid->fid_mutex);
 699         kref_put(&cfid->refcount, smb2_close_cached_fid);
 700         mutex_unlock(&cfid->fid_mutex);
 701 }
 702
 703 void close_shroot_lease_locked(struct cached_fid *cfid)
 704 {
 705         if (cfid->has_lease) {
 706                 cfid->has_lease = false;
 707                 kref_put(&cfid->refcount, smb2_close_cached_fid);
 708         }
 709 }
 710
 711 void close_shroot_lease(struct cached_fid *cfid)
 712 {
 713         mutex_lock(&cfid->fid_mutex);
 714         close_shroot_lease_locked(cfid);
 715         mutex_unlock(&cfid->fid_mutex);
 716 }
 717
 718 void
 719 smb2_cached_lease_break(struct work_struct *work)
 720 {
 721         struct cached_fid *cfid = container_of(work,
 722                                 struct cached_fid, lease_break);
 723
 724         close_shroot_lease(cfid);
 725 }
 726
 727 /*
 728  * Open the directory at the root of a share
 729  */
 730 int open_shroot(unsigned int xid, struct cifs_tcon *tcon,
 731                 struct cifs_sb_info *cifs_sb,
 732                 struct cached_fid **cfid)
 733 {
 734         struct cifs_ses *ses = tcon->ses;
 735         struct TCP_Server_Info *server = ses->server;
 736         struct cifs_open_parms oparms;
 737         struct smb2_create_rsp *o_rsp = NULL;
 738         struct smb2_query_info_rsp *qi_rsp = NULL;
 739         int resp_buftype[2];
 740         struct smb_rqst rqst[2];
 741         struct kvec rsp_iov[2];
 742         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
 743         struct kvec qi_iov[1];
 744         int rc, flags = 0;
 745         __le16 utf16_path = 0; /* Null - since an open of top of share */
 746         u8 oplock = SMB2_OPLOCK_LEVEL_II;
 747         struct cifs_fid *pfid;
 748
 749         mutex_lock(&tcon->crfid.fid_mutex);
 750         if (tcon->crfid.is_valid) {
 751                 cifs_dbg(FYI, "found a cached root file handle\n");
 752                 *cfid = &tcon->crfid;
 753                 kref_get(&tcon->crfid.refcount);
 754                 mutex_unlock(&tcon->crfid.fid_mutex);
 755                 return 0;
 756         }
 757
 758         /*
 759          * We do not hold the lock for the open because in case
 760          * SMB2_open needs to reconnect, it will end up calling
 761          * cifs_mark_open_files_invalid() which takes the lock again
 762          * thus causing a deadlock
 763          */
 764
 765         mutex_unlock(&tcon->crfid.fid_mutex);
 766
 767         if (smb3_encryption_required(tcon))
 768                 flags |= CIFS_TRANSFORM_REQ;
 769
 770         if (!server->ops->new_lease_key)
 771                 return -EIO;
 772
 773         pfid = tcon->crfid.fid;
 774         server->ops->new_lease_key(pfid);
 775
 776         memset(rqst, 0, sizeof(rqst));
 777         resp_buftype[0] = resp_buftype[1] = CIFS_NO_BUFFER;
 778         memset(rsp_iov, 0, sizeof(rsp_iov));
 779
 780         /* Open */
 781         memset(&open_iov, 0, sizeof(open_iov));
 782         rqst[0].rq_iov = open_iov;
 783         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
 784
 785         oparms.tcon = tcon;
 786         oparms.create_options = cifs_create_options(cifs_sb, 0);
 787         oparms.desired_access = FILE_READ_ATTRIBUTES;
 788         oparms.disposition = FILE_OPEN;
 789         oparms.fid = pfid;
 790         oparms.reconnect = false;
 791
 792         rc = SMB2_open_init(tcon, server,
 793                             &rqst[0], &oplock, &oparms, &utf16_path);
 794         if (rc)
 795                 goto oshr_free;
 796         smb2_set_next_command(tcon, &rqst[0]);
 797
 798         memset(&qi_iov, 0, sizeof(qi_iov));
 799         rqst[1].rq_iov = qi_iov;
 800         rqst[1].rq_nvec = 1;
 801
 802         rc = SMB2_query_info_init(tcon, server,
 803                                   &rqst[1], COMPOUND_FID,
 804                                   COMPOUND_FID, FILE_ALL_INFORMATION,
 805                                   SMB2_O_INFO_FILE, 0,
 806                                   sizeof(struct smb2_file_all_info) +
 807                                   PATH_MAX * 2, 0, NULL);
 808         if (rc)
 809                 goto oshr_free;
 810
 811         smb2_set_related(&rqst[1]);
 812
 813         rc = compound_send_recv(xid, ses, server,
 814                                 flags, 2, rqst,
 815                                 resp_buftype, rsp_iov);
 816         mutex_lock(&tcon->crfid.fid_mutex);
 817
 818         /*
 819          * Now we need to check again as the cached root might have
 820          * been successfully re-opened from a concurrent process
 821          */
 822
 823         if (tcon->crfid.is_valid) {
 824                 /* work was already done */
 825
 826                 /* stash fids for close() later */
 827                 struct cifs_fid fid = {
 828                         .persistent_fid = pfid->persistent_fid,
 829                         .volatile_fid = pfid->volatile_fid,
 830                 };
 831
 832                 /*
 833                  * caller expects this func to set pfid to a valid
 834                  * cached root, so we copy the existing one and get a
 835                  * reference.
 836                  */
 837                 memcpy(pfid, tcon->crfid.fid, sizeof(*pfid));
 838                 kref_get(&tcon->crfid.refcount);
 839
 840                 mutex_unlock(&tcon->crfid.fid_mutex);
 841
 842                 if (rc == 0) {
 843                         /* close extra handle outside of crit sec */
 844                         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
 845                 }
 846                 rc = 0;
 847                 goto oshr_free;
 848         }
 849
 850         /* Cached root is still invalid, continue normaly */
 851
 852         if (rc) {
 853                 if (rc == -EREMCHG) {
 854                         tcon->need_reconnect = true;
 855                         pr_warn_once("server share %s deleted\n",
 856                                      tcon->treeName);
 857                 }
 858                 goto oshr_exit;
 859         }
 860
 861         atomic_inc(&tcon->num_remote_opens);
 862
 863         o_rsp = (struct smb2_create_rsp *)rsp_iov[0].iov_base;
 864         oparms.fid->persistent_fid = o_rsp->PersistentFileId;
 865         oparms.fid->volatile_fid = o_rsp->VolatileFileId;
 866 #ifdef CONFIG_CIFS_DEBUG2
 867         oparms.fid->mid = le64_to_cpu(o_rsp->sync_hdr.MessageId);
 868 #endif /* CIFS_DEBUG2 */
 869
 870         memcpy(tcon->crfid.fid, pfid, sizeof(struct cifs_fid));
 871         tcon->crfid.tcon = tcon;
 872         tcon->crfid.is_valid = true;
 873         kref_init(&tcon->crfid.refcount);
 874
 875         /* BB TBD check to see if oplock level check can be removed below */
 876         if (o_rsp->OplockLevel == SMB2_OPLOCK_LEVEL_LEASE) {
 877                 kref_get(&tcon->crfid.refcount);
 878                 tcon->crfid.has_lease = true;
 879                 smb2_parse_contexts(server, o_rsp,
 880                                 &oparms.fid->epoch,
 881                                     oparms.fid->lease_key, &oplock,
 882                                     NULL, NULL);
 883         } else
 884                 goto oshr_exit;
 885
 886         qi_rsp = (struct smb2_query_info_rsp *)rsp_iov[1].iov_base;
 887         if (le32_to_cpu(qi_rsp->OutputBufferLength) < sizeof(struct smb2_file_all_info))
 888                 goto oshr_exit;
 889         if (!smb2_validate_and_copy_iov(
 890                                 le16_to_cpu(qi_rsp->OutputBufferOffset),
 891                                 sizeof(struct smb2_file_all_info),
 892                                 &rsp_iov[1], sizeof(struct smb2_file_all_info),
 893                                 (char *)&tcon->crfid.file_all_info))
 894                 tcon->crfid.file_all_info_is_valid = true;
 895
 896 oshr_exit:
 897         mutex_unlock(&tcon->crfid.fid_mutex);
 898 oshr_free:
 899         SMB2_open_free(&rqst[0]);
 900         SMB2_query_info_free(&rqst[1]);
 901         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
 902         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
 903         if (rc == 0)
 904                 *cfid = &tcon->crfid;
 905         return rc;
 906 }
 907
 908 static void
 909 smb3_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon,
 910               struct cifs_sb_info *cifs_sb)
 911 {
 912         int rc;
 913         __le16 srch_path = 0; /* Null - open root of share */
 914         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
 915         struct cifs_open_parms oparms;
 916         struct cifs_fid fid;
 917         bool no_cached_open = tcon->nohandlecache;
 918         struct cached_fid *cfid = NULL;
 919
 920         oparms.tcon = tcon;
 921         oparms.desired_access = FILE_READ_ATTRIBUTES;
 922         oparms.disposition = FILE_OPEN;
 923         oparms.create_options = cifs_create_options(cifs_sb, 0);
 924         oparms.fid = &fid;
 925         oparms.reconnect = false;
 926
 927         if (no_cached_open) {
 928                 rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL,
 929                                NULL, NULL);
 930         } else {
 931                 rc = open_shroot(xid, tcon, cifs_sb, &cfid);
 932                 if (rc == 0)
 933                         memcpy(&fid, cfid->fid, sizeof(struct cifs_fid));
 934         }
 935         if (rc)
 936                 return;
 937
 938         SMB3_request_interfaces(xid, tcon);
 939
 940         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 941                         FS_ATTRIBUTE_INFORMATION);
 942         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 943                         FS_DEVICE_INFORMATION);
 944         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 945                         FS_VOLUME_INFORMATION);
 946         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 947                         FS_SECTOR_SIZE_INFORMATION); /* SMB3 specific */
 948         if (no_cached_open)
 949                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
 950         else
 951                 close_shroot(cfid);
 952 }
 953
 954 static void
 955 smb2_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon,
 956               struct cifs_sb_info *cifs_sb)
 957 {
 958         int rc;
 959         __le16 srch_path = 0; /* Null - open root of share */
 960         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
 961         struct cifs_open_parms oparms;
 962         struct cifs_fid fid;
 963
 964         oparms.tcon = tcon;
 965         oparms.desired_access = FILE_READ_ATTRIBUTES;
 966         oparms.disposition = FILE_OPEN;
 967         oparms.create_options = cifs_create_options(cifs_sb, 0);
 968         oparms.fid = &fid;
 969         oparms.reconnect = false;
 970
 971         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL,
 972                        NULL, NULL);
 973         if (rc)
 974                 return;
 975
 976         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 977                         FS_ATTRIBUTE_INFORMATION);
 978         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
 979                         FS_DEVICE_INFORMATION);
 980         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
 981 }
 982
 983 static int
 984 smb2_is_path_accessible(const unsigned int xid, struct cifs_tcon *tcon,
 985                         struct cifs_sb_info *cifs_sb, const char *full_path)
 986 {
 987         int rc;
 988         __le16 *utf16_path;
 989         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
 990         struct cifs_open_parms oparms;
 991         struct cifs_fid fid;
 992
 993         if ((*full_path == 0) && tcon->crfid.is_valid)
 994                 return 0;
 995
 996         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
 997         if (!utf16_path)
 998                 return -ENOMEM;
 999
1000         oparms.tcon = tcon;
1001         oparms.desired_access = FILE_READ_ATTRIBUTES;
1002         oparms.disposition = FILE_OPEN;
1003         oparms.create_options = cifs_create_options(cifs_sb, 0);
1004         oparms.fid = &fid;
1005         oparms.reconnect = false;
1006
1007         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL,
1008                        NULL);
1009         if (rc) {
1010                 kfree(utf16_path);
1011                 return rc;
1012         }
1013
1014         rc = SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
1015         kfree(utf16_path);
1016         return rc;
1017 }
1018
1019 static int
1020 smb2_get_srv_inum(const unsigned int xid, struct cifs_tcon *tcon,
1021                   struct cifs_sb_info *cifs_sb, const char *full_path,
1022                   u64 *uniqueid, FILE_ALL_INFO *data)
1023 {
1024         *uniqueid = le64_to_cpu(data->IndexNumber);
1025         return 0;
1026 }
1027
1028 static int
1029 smb2_query_file_info(const unsigned int xid, struct cifs_tcon *tcon,
1030                      struct cifs_fid *fid, FILE_ALL_INFO *data)
1031 {
1032         int rc;
1033         struct smb2_file_all_info *smb2_data;
1034
1035         smb2_data = kzalloc(sizeof(struct smb2_file_all_info) + PATH_MAX * 2,
1036                             GFP_KERNEL);
1037         if (smb2_data == NULL)
1038                 return -ENOMEM;
1039
1040         rc = SMB2_query_info(xid, tcon, fid->persistent_fid, fid->volatile_fid,
1041                              smb2_data);
1042         if (!rc)
1043                 move_smb2_info_to_cifs(data, smb2_data);
1044         kfree(smb2_data);
1045         return rc;
1046 }
1047
1048 #ifdef CONFIG_CIFS_XATTR
1049 static ssize_t
1050 move_smb2_ea_to_cifs(char *dst, size_t dst_size,
1051                      struct smb2_file_full_ea_info *src, size_t src_size,
1052                      const unsigned char *ea_name)
1053 {
1054         int rc = 0;
1055         unsigned int ea_name_len = ea_name ? strlen(ea_name) : 0;
1056         char *name, *value;
1057         size_t buf_size = dst_size;
1058         size_t name_len, value_len, user_name_len;
1059
1060         while (src_size > 0) {
1061                 name = &src->ea_data[0];
1062                 name_len = (size_t)src->ea_name_length;
1063                 value = &src->ea_data[src->ea_name_length + 1];
1064                 value_len = (size_t)le16_to_cpu(src->ea_value_length);
1065
1066                 if (name_len == 0)
1067                         break;
1068
1069                 if (src_size < 8 + name_len + 1 + value_len) {
1070                         cifs_dbg(FYI, "EA entry goes beyond length of list\n");
1071                         rc = -EIO;
1072                         goto out;
1073                 }
1074
1075                 if (ea_name) {
1076                         if (ea_name_len == name_len &&
1077                             memcmp(ea_name, name, name_len) == 0) {
1078                                 rc = value_len;
1079                                 if (dst_size == 0)
1080                                         goto out;
1081                                 if (dst_size < value_len) {
1082                                         rc = -ERANGE;
1083                                         goto out;
1084                                 }
1085                                 memcpy(dst, value, value_len);
1086                                 goto out;
1087                         }
1088                 } else {
1089                         /* 'user.' plus a terminating null */
1090                         user_name_len = 5 + 1 + name_len;
1091
1092                         if (buf_size == 0) {
1093                                 /* skip copy - calc size only */
1094                                 rc += user_name_len;
1095                         } else if (dst_size >= user_name_len) {
1096                                 dst_size -= user_name_len;
1097                                 memcpy(dst, "user.", 5);
1098                                 dst += 5;
1099                                 memcpy(dst, src->ea_data, name_len);
1100                                 dst += name_len;
1101                                 *dst = 0;
1102                                 ++dst;
1103                                 rc += user_name_len;
1104                         } else {
1105                                 /* stop before overrun buffer */
1106                                 rc = -ERANGE;
1107                                 break;
1108                         }
1109                 }
1110
1111                 if (!src->next_entry_offset)
1112                         break;
1113
1114                 if (src_size < le32_to_cpu(src->next_entry_offset)) {
1115                         /* stop before overrun buffer */
1116                         rc = -ERANGE;
1117                         break;
1118                 }
1119                 src_size -= le32_to_cpu(src->next_entry_offset);
1120                 src = (void *)((char *)src +
1121                                le32_to_cpu(src->next_entry_offset));
1122         }
1123
1124         /* didn't find the named attribute */
1125         if (ea_name)
1126                 rc = -ENODATA;
1127
1128 out:
1129         return (ssize_t)rc;
1130 }
1131
1132 static ssize_t
1133 smb2_query_eas(const unsigned int xid, struct cifs_tcon *tcon,
1134                const unsigned char *path, const unsigned char *ea_name,
1135                char *ea_data, size_t buf_size,
1136                struct cifs_sb_info *cifs_sb)
1137 {
1138         int rc;
1139         __le16 *utf16_path;
1140         struct kvec rsp_iov = {NULL, 0};
1141         int buftype = CIFS_NO_BUFFER;
1142         struct smb2_query_info_rsp *rsp;
1143         struct smb2_file_full_ea_info *info = NULL;
1144
1145         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1146         if (!utf16_path)
1147                 return -ENOMEM;
1148
1149         rc = smb2_query_info_compound(xid, tcon, utf16_path,
1150                                       FILE_READ_EA,
1151                                       FILE_FULL_EA_INFORMATION,
1152                                       SMB2_O_INFO_FILE,
1153                                       CIFSMaxBufSize -
1154                                       MAX_SMB2_CREATE_RESPONSE_SIZE -
1155                                       MAX_SMB2_CLOSE_RESPONSE_SIZE,
1156                                       &rsp_iov, &buftype, cifs_sb);
1157         if (rc) {
1158                 /*
1159                  * If ea_name is NULL (listxattr) and there are no EAs,
1160                  * return 0 as it's not an error. Otherwise, the specified
1161                  * ea_name was not found.
1162                  */
1163                 if (!ea_name && rc == -ENODATA)
1164                         rc = 0;
1165                 goto qeas_exit;
1166         }
1167
1168         rsp = (struct smb2_query_info_rsp *)rsp_iov.iov_base;
1169         rc = smb2_validate_iov(le16_to_cpu(rsp->OutputBufferOffset),
1170                                le32_to_cpu(rsp->OutputBufferLength),
1171                                &rsp_iov,
1172                                sizeof(struct smb2_file_full_ea_info));
1173         if (rc)
1174                 goto qeas_exit;
1175
1176         info = (struct smb2_file_full_ea_info *)(
1177                         le16_to_cpu(rsp->OutputBufferOffset) + (char *)rsp);
1178         rc = move_smb2_ea_to_cifs(ea_data, buf_size, info,
1179                         le32_to_cpu(rsp->OutputBufferLength), ea_name);
1180
1181  qeas_exit:
1182         kfree(utf16_path);
1183         free_rsp_buf(buftype, rsp_iov.iov_base);
1184         return rc;
1185 }
1186
1187
1188 static int
1189 smb2_set_ea(const unsigned int xid, struct cifs_tcon *tcon,
1190             const char *path, const char *ea_name, const void *ea_value,
1191             const __u16 ea_value_len, const struct nls_table *nls_codepage,
1192             struct cifs_sb_info *cifs_sb)
1193 {
1194         struct cifs_ses *ses = tcon->ses;
1195         struct TCP_Server_Info *server = cifs_pick_channel(ses);
1196         __le16 *utf16_path = NULL;
1197         int ea_name_len = strlen(ea_name);
1198         int flags = 0;
1199         int len;
1200         struct smb_rqst rqst[3];
1201         int resp_buftype[3];
1202         struct kvec rsp_iov[3];
1203         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
1204         struct cifs_open_parms oparms;
1205         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1206         struct cifs_fid fid;
1207         struct kvec si_iov[SMB2_SET_INFO_IOV_SIZE];
1208         unsigned int size[1];
1209         void *data[1];
1210         struct smb2_file_full_ea_info *ea = NULL;
1211         struct kvec close_iov[1];
1212         struct smb2_query_info_rsp *rsp;
1213         int rc, used_len = 0;
1214
1215         if (smb3_encryption_required(tcon))
1216                 flags |= CIFS_TRANSFORM_REQ;
1217
1218         if (ea_name_len > 255)
1219                 return -EINVAL;
1220
1221         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1222         if (!utf16_path)
1223                 return -ENOMEM;
1224
1225         memset(rqst, 0, sizeof(rqst));
1226         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
1227         memset(rsp_iov, 0, sizeof(rsp_iov));
1228
1229         if (ses->server->ops->query_all_EAs) {
1230                 if (!ea_value) {
1231                         rc = ses->server->ops->query_all_EAs(xid, tcon, path,
1232                                                              ea_name, NULL, 0,
1233                                                              cifs_sb);
1234                         if (rc == -ENODATA)
1235                                 goto sea_exit;
1236                 } else {
1237                         /* If we are adding a attribute we should first check
1238                          * if there will be enough space available to store
1239                          * the new EA. If not we should not add it since we
1240                          * would not be able to even read the EAs back.
1241                          */
1242                         rc = smb2_query_info_compound(xid, tcon, utf16_path,
1243                                       FILE_READ_EA,
1244                                       FILE_FULL_EA_INFORMATION,
1245                                       SMB2_O_INFO_FILE,
1246                                       CIFSMaxBufSize -
1247                                       MAX_SMB2_CREATE_RESPONSE_SIZE -
1248                                       MAX_SMB2_CLOSE_RESPONSE_SIZE,
1249                                       &rsp_iov[1], &resp_buftype[1], cifs_sb);
1250                         if (rc == 0) {
1251                                 rsp = (struct smb2_query_info_rsp *)rsp_iov[1].iov_base;
1252                                 used_len = le32_to_cpu(rsp->OutputBufferLength);
1253                         }
1254                         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
1255                         resp_buftype[1] = CIFS_NO_BUFFER;
1256                         memset(&rsp_iov[1], 0, sizeof(rsp_iov[1]));
1257                         rc = 0;
1258
1259                         /* Use a fudge factor of 256 bytes in case we collide
1260                          * with a different set_EAs command.
1261                          */
1262                         if(CIFSMaxBufSize - MAX_SMB2_CREATE_RESPONSE_SIZE -
1263                            MAX_SMB2_CLOSE_RESPONSE_SIZE - 256 <
1264                            used_len + ea_name_len + ea_value_len + 1) {
1265                                 rc = -ENOSPC;
1266                                 goto sea_exit;
1267                         }
1268                 }
1269         }
1270
1271         /* Open */
1272         memset(&open_iov, 0, sizeof(open_iov));
1273         rqst[0].rq_iov = open_iov;
1274         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
1275
1276         memset(&oparms, 0, sizeof(oparms));
1277         oparms.tcon = tcon;
1278         oparms.desired_access = FILE_WRITE_EA;
1279         oparms.disposition = FILE_OPEN;
1280         oparms.create_options = cifs_create_options(cifs_sb, 0);
1281         oparms.fid = &fid;
1282         oparms.reconnect = false;
1283
1284         rc = SMB2_open_init(tcon, server,
1285                             &rqst[0], &oplock, &oparms, utf16_path);
1286         if (rc)
1287                 goto sea_exit;
1288         smb2_set_next_command(tcon, &rqst[0]);
1289
1290
1291         /* Set Info */
1292         memset(&si_iov, 0, sizeof(si_iov));
1293         rqst[1].rq_iov = si_iov;
1294         rqst[1].rq_nvec = 1;
1295
1296         len = sizeof(*ea) + ea_name_len + ea_value_len + 1;
1297         ea = kzalloc(len, GFP_KERNEL);
1298         if (ea == NULL) {
1299                 rc = -ENOMEM;
1300                 goto sea_exit;
1301         }
1302
1303         ea->ea_name_length = ea_name_len;
1304         ea->ea_value_length = cpu_to_le16(ea_value_len);
1305         memcpy(ea->ea_data, ea_name, ea_name_len + 1);
1306         memcpy(ea->ea_data + ea_name_len + 1, ea_value, ea_value_len);
1307
1308         size[0] = len;
1309         data[0] = ea;
1310
1311         rc = SMB2_set_info_init(tcon, server,
1312                                 &rqst[1], COMPOUND_FID,
1313                                 COMPOUND_FID, current->tgid,
1314                                 FILE_FULL_EA_INFORMATION,
1315                                 SMB2_O_INFO_FILE, 0, data, size);
1316         smb2_set_next_command(tcon, &rqst[1]);
1317         smb2_set_related(&rqst[1]);
1318
1319
1320         /* Close */
1321         memset(&close_iov, 0, sizeof(close_iov));
1322         rqst[2].rq_iov = close_iov;
1323         rqst[2].rq_nvec = 1;
1324         rc = SMB2_close_init(tcon, server,
1325                              &rqst[2], COMPOUND_FID, COMPOUND_FID, false);
1326         smb2_set_related(&rqst[2]);
1327
1328         rc = compound_send_recv(xid, ses, server,
1329                                 flags, 3, rqst,
1330                                 resp_buftype, rsp_iov);
1331         /* no need to bump num_remote_opens because handle immediately closed */
1332
1333  sea_exit:
1334         kfree(ea);
1335         kfree(utf16_path);
1336         SMB2_open_free(&rqst[0]);
1337         SMB2_set_info_free(&rqst[1]);
1338         SMB2_close_free(&rqst[2]);
1339         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
1340         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
1341         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
1342         return rc;
1343 }
1344 #endif
1345
1346 static bool
1347 smb2_can_echo(struct TCP_Server_Info *server)
1348 {
1349         return server->echoes;
1350 }
1351
1352 static void
1353 smb2_clear_stats(struct cifs_tcon *tcon)
1354 {
1355         int i;
1356
1357         for (i = 0; i < NUMBER_OF_SMB2_COMMANDS; i++) {
1358                 atomic_set(&tcon->stats.smb2_stats.smb2_com_sent[i], 0);
1359                 atomic_set(&tcon->stats.smb2_stats.smb2_com_failed[i], 0);
1360         }
1361 }
1362
1363 static void
1364 smb2_dump_share_caps(struct seq_file *m, struct cifs_tcon *tcon)
1365 {
1366         seq_puts(m, "\n\tShare Capabilities:");
1367         if (tcon->capabilities & SMB2_SHARE_CAP_DFS)
1368                 seq_puts(m, " DFS,");
1369         if (tcon->capabilities & SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY)
1370                 seq_puts(m, " CONTINUOUS AVAILABILITY,");
1371         if (tcon->capabilities & SMB2_SHARE_CAP_SCALEOUT)
1372                 seq_puts(m, " SCALEOUT,");
1373         if (tcon->capabilities & SMB2_SHARE_CAP_CLUSTER)
1374                 seq_puts(m, " CLUSTER,");
1375         if (tcon->capabilities & SMB2_SHARE_CAP_ASYMMETRIC)
1376                 seq_puts(m, " ASYMMETRIC,");
1377         if (tcon->capabilities == 0)
1378                 seq_puts(m, " None");
1379         if (tcon->ss_flags & SSINFO_FLAGS_ALIGNED_DEVICE)
1380                 seq_puts(m, " Aligned,");
1381         if (tcon->ss_flags & SSINFO_FLAGS_PARTITION_ALIGNED_ON_DEVICE)
1382                 seq_puts(m, " Partition Aligned,");
1383         if (tcon->ss_flags & SSINFO_FLAGS_NO_SEEK_PENALTY)
1384                 seq_puts(m, " SSD,");
1385         if (tcon->ss_flags & SSINFO_FLAGS_TRIM_ENABLED)
1386                 seq_puts(m, " TRIM-support,");
1387
1388         seq_printf(m, "\tShare Flags: 0x%x", tcon->share_flags);
1389         seq_printf(m, "\n\ttid: 0x%x", tcon->tid);
1390         if (tcon->perf_sector_size)
1391                 seq_printf(m, "\tOptimal sector size: 0x%x",
1392                            tcon->perf_sector_size);
1393         seq_printf(m, "\tMaximal Access: 0x%x", tcon->maximal_access);
1394 }
1395
1396 static void
1397 smb2_print_stats(struct seq_file *m, struct cifs_tcon *tcon)
1398 {
1399         atomic_t *sent = tcon->stats.smb2_stats.smb2_com_sent;
1400         atomic_t *failed = tcon->stats.smb2_stats.smb2_com_failed;
1401
1402         /*
1403          *  Can't display SMB2_NEGOTIATE, SESSION_SETUP, LOGOFF, CANCEL and ECHO
1404          *  totals (requests sent) since those SMBs are per-session not per tcon
1405          */
1406         seq_printf(m, "\nBytes read: %llu  Bytes written: %llu",
1407                    (long long)(tcon->bytes_read),
1408                    (long long)(tcon->bytes_written));
1409         seq_printf(m, "\nOpen files: %d total (local), %d open on server",
1410                    atomic_read(&tcon->num_local_opens),
1411                    atomic_read(&tcon->num_remote_opens));
1412         seq_printf(m, "\nTreeConnects: %d total %d failed",
1413                    atomic_read(&sent[SMB2_TREE_CONNECT_HE]),
1414                    atomic_read(&failed[SMB2_TREE_CONNECT_HE]));
1415         seq_printf(m, "\nTreeDisconnects: %d total %d failed",
1416                    atomic_read(&sent[SMB2_TREE_DISCONNECT_HE]),
1417                    atomic_read(&failed[SMB2_TREE_DISCONNECT_HE]));
1418         seq_printf(m, "\nCreates: %d total %d failed",
1419                    atomic_read(&sent[SMB2_CREATE_HE]),
1420                    atomic_read(&failed[SMB2_CREATE_HE]));
1421         seq_printf(m, "\nCloses: %d total %d failed",
1422                    atomic_read(&sent[SMB2_CLOSE_HE]),
1423                    atomic_read(&failed[SMB2_CLOSE_HE]));
1424         seq_printf(m, "\nFlushes: %d total %d failed",
1425                    atomic_read(&sent[SMB2_FLUSH_HE]),
1426                    atomic_read(&failed[SMB2_FLUSH_HE]));
1427         seq_printf(m, "\nReads: %d total %d failed",
1428                    atomic_read(&sent[SMB2_READ_HE]),
1429                    atomic_read(&failed[SMB2_READ_HE]));
1430         seq_printf(m, "\nWrites: %d total %d failed",
1431                    atomic_read(&sent[SMB2_WRITE_HE]),
1432                    atomic_read(&failed[SMB2_WRITE_HE]));
1433         seq_printf(m, "\nLocks: %d total %d failed",
1434                    atomic_read(&sent[SMB2_LOCK_HE]),
1435                    atomic_read(&failed[SMB2_LOCK_HE]));
1436         seq_printf(m, "\nIOCTLs: %d total %d failed",
1437                    atomic_read(&sent[SMB2_IOCTL_HE]),
1438                    atomic_read(&failed[SMB2_IOCTL_HE]));
1439         seq_printf(m, "\nQueryDirectories: %d total %d failed",
1440                    atomic_read(&sent[SMB2_QUERY_DIRECTORY_HE]),
1441                    atomic_read(&failed[SMB2_QUERY_DIRECTORY_HE]));
1442         seq_printf(m, "\nChangeNotifies: %d total %d failed",
1443                    atomic_read(&sent[SMB2_CHANGE_NOTIFY_HE]),
1444                    atomic_read(&failed[SMB2_CHANGE_NOTIFY_HE]));
1445         seq_printf(m, "\nQueryInfos: %d total %d failed",
1446                    atomic_read(&sent[SMB2_QUERY_INFO_HE]),
1447                    atomic_read(&failed[SMB2_QUERY_INFO_HE]));
1448         seq_printf(m, "\nSetInfos: %d total %d failed",
1449                    atomic_read(&sent[SMB2_SET_INFO_HE]),
1450                    atomic_read(&failed[SMB2_SET_INFO_HE]));
1451         seq_printf(m, "\nOplockBreaks: %d sent %d failed",
1452                    atomic_read(&sent[SMB2_OPLOCK_BREAK_HE]),
1453                    atomic_read(&failed[SMB2_OPLOCK_BREAK_HE]));
1454 }
1455
1456 static void
1457 smb2_set_fid(struct cifsFileInfo *cfile, struct cifs_fid *fid, __u32 oplock)
1458 {
1459         struct cifsInodeInfo *cinode = CIFS_I(d_inode(cfile->dentry));
1460         struct TCP_Server_Info *server = tlink_tcon(cfile->tlink)->ses->server;
1461
1462         cfile->fid.persistent_fid = fid->persistent_fid;
1463         cfile->fid.volatile_fid = fid->volatile_fid;
1464         cfile->fid.access = fid->access;
1465 #ifdef CONFIG_CIFS_DEBUG2
1466         cfile->fid.mid = fid->mid;
1467 #endif /* CIFS_DEBUG2 */
1468         server->ops->set_oplock_level(cinode, oplock, fid->epoch,
1469                                       &fid->purge_cache);
1470         cinode->can_cache_brlcks = CIFS_CACHE_WRITE(cinode);
1471         memcpy(cfile->fid.create_guid, fid->create_guid, 16);
1472 }
1473
1474 static void
1475 smb2_close_file(const unsigned int xid, struct cifs_tcon *tcon,
1476                 struct cifs_fid *fid)
1477 {
1478         SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1479 }
1480
1481 static void
1482 smb2_close_getattr(const unsigned int xid, struct cifs_tcon *tcon,
1483                    struct cifsFileInfo *cfile)
1484 {
1485         struct smb2_file_network_open_info file_inf;
1486         struct inode *inode;
1487         int rc;
1488
1489         rc = __SMB2_close(xid, tcon, cfile->fid.persistent_fid,
1490                    cfile->fid.volatile_fid, &file_inf);
1491         if (rc)
1492                 return;
1493
1494         inode = d_inode(cfile->dentry);
1495
1496         spin_lock(&inode->i_lock);
1497         CIFS_I(inode)->time = jiffies;
1498
1499         /* Creation time should not need to be updated on close */
1500         if (file_inf.LastWriteTime)
1501                 inode->i_mtime = cifs_NTtimeToUnix(file_inf.LastWriteTime);
1502         if (file_inf.ChangeTime)
1503                 inode->i_ctime = cifs_NTtimeToUnix(file_inf.ChangeTime);
1504         if (file_inf.LastAccessTime)
1505                 inode->i_atime = cifs_NTtimeToUnix(file_inf.LastAccessTime);
1506
1507         /*
1508          * i_blocks is not related to (i_size / i_blksize),
1509          * but instead 512 byte (2**9) size is required for
1510          * calculating num blocks.
1511          */
1512         if (le64_to_cpu(file_inf.AllocationSize) > 4096)
1513                 inode->i_blocks =
1514                         (512 - 1 + le64_to_cpu(file_inf.AllocationSize)) >> 9;
1515
1516         /* End of file and Attributes should not have to be updated on close */
1517         spin_unlock(&inode->i_lock);
1518 }
1519
1520 static int
1521 SMB2_request_res_key(const unsigned int xid, struct cifs_tcon *tcon,
1522                      u64 persistent_fid, u64 volatile_fid,
1523                      struct copychunk_ioctl *pcchunk)
1524 {
1525         int rc;
1526         unsigned int ret_data_len;
1527         struct resume_key_req *res_key;
1528
1529         rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
1530                         FSCTL_SRV_REQUEST_RESUME_KEY, true /* is_fsctl */,
1531                         NULL, 0 /* no input */, CIFSMaxBufSize,
1532                         (char **)&res_key, &ret_data_len);
1533
1534         if (rc) {
1535                 cifs_tcon_dbg(VFS, "refcpy ioctl error %d getting resume key\n", rc);
1536                 goto req_res_key_exit;
1537         }
1538         if (ret_data_len < sizeof(struct resume_key_req)) {
1539                 cifs_tcon_dbg(VFS, "Invalid refcopy resume key length\n");
1540                 rc = -EINVAL;
1541                 goto req_res_key_exit;
1542         }
1543         memcpy(pcchunk->SourceKey, res_key->ResumeKey, COPY_CHUNK_RES_KEY_SIZE);
1544
1545 req_res_key_exit:
1546         kfree(res_key);
1547         return rc;
1548 }
1549
1550 struct iqi_vars {
1551         struct smb_rqst rqst[3];
1552         struct kvec rsp_iov[3];
1553         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
1554         struct kvec qi_iov[1];
1555         struct kvec io_iov[SMB2_IOCTL_IOV_SIZE];
1556         struct kvec si_iov[SMB2_SET_INFO_IOV_SIZE];
1557         struct kvec close_iov[1];
1558 };
1559
1560 static int
1561 smb2_ioctl_query_info(const unsigned int xid,
1562                       struct cifs_tcon *tcon,
1563                       struct cifs_sb_info *cifs_sb,
1564                       __le16 *path, int is_dir,
1565                       unsigned long p)
1566 {
1567         struct iqi_vars *vars;
1568         struct smb_rqst *rqst;
1569         struct kvec *rsp_iov;
1570         struct cifs_ses *ses = tcon->ses;
1571         struct TCP_Server_Info *server = cifs_pick_channel(ses);
1572         char __user *arg = (char __user *)p;
1573         struct smb_query_info qi;
1574         struct smb_query_info __user *pqi;
1575         int rc = 0;
1576         int flags = 0;
1577         struct smb2_query_info_rsp *qi_rsp = NULL;
1578         struct smb2_ioctl_rsp *io_rsp = NULL;
1579         void *buffer = NULL;
1580         int resp_buftype[3];
1581         struct cifs_open_parms oparms;
1582         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1583         struct cifs_fid fid;
1584         unsigned int size[2];
1585         void *data[2];
1586         int create_options = is_dir ? CREATE_NOT_FILE : CREATE_NOT_DIR;
1587
1588         vars = kzalloc(sizeof(*vars), GFP_ATOMIC);
1589         if (vars == NULL)
1590                 return -ENOMEM;
1591         rqst = &vars->rqst[0];
1592         rsp_iov = &vars->rsp_iov[0];
1593
1594         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
1595
1596         if (copy_from_user(&qi, arg, sizeof(struct smb_query_info)))
1597                 goto e_fault;
1598
1599         if (qi.output_buffer_length > 1024) {
1600                 kfree(vars);
1601                 return -EINVAL;
1602         }
1603
1604         if (!ses || !server) {
1605                 kfree(vars);
1606                 return -EIO;
1607         }
1608
1609         if (smb3_encryption_required(tcon))
1610                 flags |= CIFS_TRANSFORM_REQ;
1611
1612         buffer = memdup_user(arg + sizeof(struct smb_query_info),
1613                              qi.output_buffer_length);
1614         if (IS_ERR(buffer)) {
1615                 kfree(vars);
1616                 return PTR_ERR(buffer);
1617         }
1618
1619         /* Open */
1620         rqst[0].rq_iov = &vars->open_iov[0];
1621         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
1622
1623         memset(&oparms, 0, sizeof(oparms));
1624         oparms.tcon = tcon;
1625         oparms.disposition = FILE_OPEN;
1626         oparms.create_options = cifs_create_options(cifs_sb, create_options);
1627         oparms.fid = &fid;
1628         oparms.reconnect = false;
1629
1630         if (qi.flags & PASSTHRU_FSCTL) {
1631                 switch (qi.info_type & FSCTL_DEVICE_ACCESS_MASK) {
1632                 case FSCTL_DEVICE_ACCESS_FILE_READ_WRITE_ACCESS:
1633                         oparms.desired_access = FILE_READ_DATA | FILE_WRITE_DATA | FILE_READ_ATTRIBUTES | SYNCHRONIZE;
1634                         break;
1635                 case FSCTL_DEVICE_ACCESS_FILE_ANY_ACCESS:
1636                         oparms.desired_access = GENERIC_ALL;
1637                         break;
1638                 case FSCTL_DEVICE_ACCESS_FILE_READ_ACCESS:
1639                         oparms.desired_access = GENERIC_READ;
1640                         break;
1641                 case FSCTL_DEVICE_ACCESS_FILE_WRITE_ACCESS:
1642                         oparms.desired_access = GENERIC_WRITE;
1643                         break;
1644                 }
1645         } else if (qi.flags & PASSTHRU_SET_INFO) {
1646                 oparms.desired_access = GENERIC_WRITE;
1647         } else {
1648                 oparms.desired_access = FILE_READ_ATTRIBUTES | READ_CONTROL;
1649         }
1650
1651         rc = SMB2_open_init(tcon, server,
1652                             &rqst[0], &oplock, &oparms, path);
1653         if (rc)
1654                 goto iqinf_exit;
1655         smb2_set_next_command(tcon, &rqst[0]);
1656
1657         /* Query */
1658         if (qi.flags & PASSTHRU_FSCTL) {
1659                 /* Can eventually relax perm check since server enforces too */
1660                 if (!capable(CAP_SYS_ADMIN))
1661                         rc = -EPERM;
1662                 else  {
1663                         rqst[1].rq_iov = &vars->io_iov[0];
1664                         rqst[1].rq_nvec = SMB2_IOCTL_IOV_SIZE;
1665
1666                         rc = SMB2_ioctl_init(tcon, server,
1667                                              &rqst[1],
1668                                              COMPOUND_FID, COMPOUND_FID,
1669                                              qi.info_type, true, buffer,
1670                                              qi.output_buffer_length,
1671                                              CIFSMaxBufSize -
1672                                              MAX_SMB2_CREATE_RESPONSE_SIZE -
1673                                              MAX_SMB2_CLOSE_RESPONSE_SIZE);
1674                 }
1675         } else if (qi.flags == PASSTHRU_SET_INFO) {
1676                 /* Can eventually relax perm check since server enforces too */
1677                 if (!capable(CAP_SYS_ADMIN))
1678                         rc = -EPERM;
1679                 else  {
1680                         rqst[1].rq_iov = &vars->si_iov[0];
1681                         rqst[1].rq_nvec = 1;
1682
1683                         size[0] = 8;
1684                         data[0] = buffer;
1685
1686                         rc = SMB2_set_info_init(tcon, server,
1687                                         &rqst[1],
1688                                         COMPOUND_FID, COMPOUND_FID,
1689                                         current->tgid,
1690                                         FILE_END_OF_FILE_INFORMATION,
1691                                         SMB2_O_INFO_FILE, 0, data, size);
1692                 }
1693         } else if (qi.flags == PASSTHRU_QUERY_INFO) {
1694                 rqst[1].rq_iov = &vars->qi_iov[0];
1695                 rqst[1].rq_nvec = 1;
1696
1697                 rc = SMB2_query_info_init(tcon, server,
1698                                   &rqst[1], COMPOUND_FID,
1699                                   COMPOUND_FID, qi.file_info_class,
1700                                   qi.info_type, qi.additional_information,
1701                                   qi.input_buffer_length,
1702                                   qi.output_buffer_length, buffer);
1703         } else { /* unknown flags */
1704                 cifs_tcon_dbg(VFS, "Invalid passthru query flags: 0x%x\n",
1705                               qi.flags);
1706                 rc = -EINVAL;
1707         }
1708
1709         if (rc)
1710                 goto iqinf_exit;
1711         smb2_set_next_command(tcon, &rqst[1]);
1712         smb2_set_related(&rqst[1]);
1713
1714         /* Close */
1715         rqst[2].rq_iov = &vars->close_iov[0];
1716         rqst[2].rq_nvec = 1;
1717
1718         rc = SMB2_close_init(tcon, server,
1719                              &rqst[2], COMPOUND_FID, COMPOUND_FID, false);
1720         if (rc)
1721                 goto iqinf_exit;
1722         smb2_set_related(&rqst[2]);
1723
1724         rc = compound_send_recv(xid, ses, server,
1725                                 flags, 3, rqst,
1726                                 resp_buftype, rsp_iov);
1727         if (rc)
1728                 goto iqinf_exit;
1729
1730         /* No need to bump num_remote_opens since handle immediately closed */
1731         if (qi.flags & PASSTHRU_FSCTL) {
1732                 pqi = (struct smb_query_info __user *)arg;
1733                 io_rsp = (struct smb2_ioctl_rsp *)rsp_iov[1].iov_base;
1734                 if (le32_to_cpu(io_rsp->OutputCount) < qi.input_buffer_length)
1735                         qi.input_buffer_length = le32_to_cpu(io_rsp->OutputCount);
1736                 if (qi.input_buffer_length > 0 &&
1737                     le32_to_cpu(io_rsp->OutputOffset) + qi.input_buffer_length
1738                     > rsp_iov[1].iov_len)
1739                         goto e_fault;
1740
1741                 if (copy_to_user(&pqi->input_buffer_length,
1742                                  &qi.input_buffer_length,
1743                                  sizeof(qi.input_buffer_length)))
1744                         goto e_fault;
1745
1746                 if (copy_to_user((void __user *)pqi + sizeof(struct smb_query_info),
1747                                  (const void *)io_rsp + le32_to_cpu(io_rsp->OutputOffset),
1748                                  qi.input_buffer_length))
1749                         goto e_fault;
1750         } else {
1751                 pqi = (struct smb_query_info __user *)arg;
1752                 qi_rsp = (struct smb2_query_info_rsp *)rsp_iov[1].iov_base;
1753                 if (le32_to_cpu(qi_rsp->OutputBufferLength) < qi.input_buffer_length)
1754                         qi.input_buffer_length = le32_to_cpu(qi_rsp->OutputBufferLength);
1755                 if (copy_to_user(&pqi->input_buffer_length,
1756                                  &qi.input_buffer_length,
1757                                  sizeof(qi.input_buffer_length)))
1758                         goto e_fault;
1759
1760                 if (copy_to_user(pqi + 1, qi_rsp->Buffer,
1761                                  qi.input_buffer_length))
1762                         goto e_fault;
1763         }
1764
1765  iqinf_exit:
1766         kfree(vars);
1767         kfree(buffer);
1768         SMB2_open_free(&rqst[0]);
1769         if (qi.flags & PASSTHRU_FSCTL)
1770                 SMB2_ioctl_free(&rqst[1]);
1771         else
1772                 SMB2_query_info_free(&rqst[1]);
1773
1774         SMB2_close_free(&rqst[2]);
1775         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
1776         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
1777         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
1778         return rc;
1779
1780 e_fault:
1781         rc = -EFAULT;
1782         goto iqinf_exit;
1783 }
1784
1785 static ssize_t
1786 smb2_copychunk_range(const unsigned int xid,
1787                         struct cifsFileInfo *srcfile,
1788                         struct cifsFileInfo *trgtfile, u64 src_off,
1789                         u64 len, u64 dest_off)
1790 {
1791         int rc;
1792         unsigned int ret_data_len;
1793         struct copychunk_ioctl *pcchunk;
1794         struct copychunk_ioctl_rsp *retbuf = NULL;
1795         struct cifs_tcon *tcon;
1796         int chunks_copied = 0;
1797         bool chunk_sizes_updated = false;
1798         ssize_t bytes_written, total_bytes_written = 0;
1799
1800         pcchunk = kmalloc(sizeof(struct copychunk_ioctl), GFP_KERNEL);
1801
1802         if (pcchunk == NULL)
1803                 return -ENOMEM;
1804
1805         cifs_dbg(FYI, "%s: about to call request res key\n", __func__);
1806         /* Request a key from the server to identify the source of the copy */
1807         rc = SMB2_request_res_key(xid, tlink_tcon(srcfile->tlink),
1808                                 srcfile->fid.persistent_fid,
1809                                 srcfile->fid.volatile_fid, pcchunk);
1810
1811         /* Note: request_res_key sets res_key null only if rc !=0 */
1812         if (rc)
1813                 goto cchunk_out;
1814
1815         /* For now array only one chunk long, will make more flexible later */
1816         pcchunk->ChunkCount = cpu_to_le32(1);
1817         pcchunk->Reserved = 0;
1818         pcchunk->Reserved2 = 0;
1819
1820         tcon = tlink_tcon(trgtfile->tlink);
1821
1822         while (len > 0) {
1823                 pcchunk->SourceOffset = cpu_to_le64(src_off);
1824                 pcchunk->TargetOffset = cpu_to_le64(dest_off);
1825                 pcchunk->Length =
1826                         cpu_to_le32(min_t(u32, len, tcon->max_bytes_chunk));
1827
1828                 /* Request server copy to target from src identified by key */
1829                 rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
1830                         trgtfile->fid.volatile_fid, FSCTL_SRV_COPYCHUNK_WRITE,
1831                         true /* is_fsctl */, (char *)pcchunk,
1832                         sizeof(struct copychunk_ioctl), CIFSMaxBufSize,
1833                         (char **)&retbuf, &ret_data_len);
1834                 if (rc == 0) {
1835                         if (ret_data_len !=
1836                                         sizeof(struct copychunk_ioctl_rsp)) {
1837                                 cifs_tcon_dbg(VFS, "Invalid cchunk response size\n");
1838                                 rc = -EIO;
1839                                 goto cchunk_out;
1840                         }
1841                         if (retbuf->TotalBytesWritten == 0) {
1842                                 cifs_dbg(FYI, "no bytes copied\n");
1843                                 rc = -EIO;
1844                                 goto cchunk_out;
1845                         }
1846                         /*
1847                          * Check if server claimed to write more than we asked
1848                          */
1849                         if (le32_to_cpu(retbuf->TotalBytesWritten) >
1850                             le32_to_cpu(pcchunk->Length)) {
1851                                 cifs_tcon_dbg(VFS, "Invalid copy chunk response\n");
1852                                 rc = -EIO;
1853                                 goto cchunk_out;
1854                         }
1855                         if (le32_to_cpu(retbuf->ChunksWritten) != 1) {
1856                                 cifs_tcon_dbg(VFS, "Invalid num chunks written\n");
1857                                 rc = -EIO;
1858                                 goto cchunk_out;
1859                         }
1860                         chunks_copied++;
1861
1862                         bytes_written = le32_to_cpu(retbuf->TotalBytesWritten);
1863                         src_off += bytes_written;
1864                         dest_off += bytes_written;
1865                         len -= bytes_written;
1866                         total_bytes_written += bytes_written;
1867
1868                         cifs_dbg(FYI, "Chunks %d PartialChunk %d Total %zu\n",
1869                                 le32_to_cpu(retbuf->ChunksWritten),
1870                                 le32_to_cpu(retbuf->ChunkBytesWritten),
1871                                 bytes_written);
1872                 } else if (rc == -EINVAL) {
1873                         if (ret_data_len != sizeof(struct copychunk_ioctl_rsp))
1874                                 goto cchunk_out;
1875
1876                         cifs_dbg(FYI, "MaxChunks %d BytesChunk %d MaxCopy %d\n",
1877                                 le32_to_cpu(retbuf->ChunksWritten),
1878                                 le32_to_cpu(retbuf->ChunkBytesWritten),
1879                                 le32_to_cpu(retbuf->TotalBytesWritten));
1880
1881                         /*
1882                          * Check if this is the first request using these sizes,
1883                          * (ie check if copy succeed once with original sizes
1884                          * and check if the server gave us different sizes after
1885                          * we already updated max sizes on previous request).
1886                          * if not then why is the server returning an error now
1887                          */
1888                         if ((chunks_copied != 0) || chunk_sizes_updated)
1889                                 goto cchunk_out;
1890
1891                         /* Check that server is not asking us to grow size */
1892                         if (le32_to_cpu(retbuf->ChunkBytesWritten) <
1893                                         tcon->max_bytes_chunk)
1894                                 tcon->max_bytes_chunk =
1895                                         le32_to_cpu(retbuf->ChunkBytesWritten);
1896                         else
1897                                 goto cchunk_out; /* server gave us bogus size */
1898
1899                         /* No need to change MaxChunks since already set to 1 */
1900                         chunk_sizes_updated = true;
1901                 } else
1902                         goto cchunk_out;
1903         }
1904
1905 cchunk_out:
1906         kfree(pcchunk);
1907         kfree(retbuf);
1908         if (rc)
1909                 return rc;
1910         else
1911                 return total_bytes_written;
1912 }
1913
1914 static int
1915 smb2_flush_file(const unsigned int xid, struct cifs_tcon *tcon,
1916                 struct cifs_fid *fid)
1917 {
1918         return SMB2_flush(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1919 }
1920
1921 static unsigned int
1922 smb2_read_data_offset(char *buf)
1923 {
1924         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1925
1926         return rsp->DataOffset;
1927 }
1928
1929 static unsigned int
1930 smb2_read_data_length(char *buf, bool in_remaining)
1931 {
1932         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1933
1934         if (in_remaining)
1935                 return le32_to_cpu(rsp->DataRemaining);
1936
1937         return le32_to_cpu(rsp->DataLength);
1938 }
1939
1940
1941 static int
1942 smb2_sync_read(const unsigned int xid, struct cifs_fid *pfid,
1943                struct cifs_io_parms *parms, unsigned int *bytes_read,
1944                char **buf, int *buf_type)
1945 {
1946         parms->persistent_fid = pfid->persistent_fid;
1947         parms->volatile_fid = pfid->volatile_fid;
1948         return SMB2_read(xid, parms, bytes_read, buf, buf_type);
1949 }
1950
1951 static int
1952 smb2_sync_write(const unsigned int xid, struct cifs_fid *pfid,
1953                 struct cifs_io_parms *parms, unsigned int *written,
1954                 struct kvec *iov, unsigned long nr_segs)
1955 {
1956
1957         parms->persistent_fid = pfid->persistent_fid;
1958         parms->volatile_fid = pfid->volatile_fid;
1959         return SMB2_write(xid, parms, written, iov, nr_segs);
1960 }
1961
1962 /* Set or clear the SPARSE_FILE attribute based on value passed in setsparse */
1963 static bool smb2_set_sparse(const unsigned int xid, struct cifs_tcon *tcon,
1964                 struct cifsFileInfo *cfile, struct inode *inode, __u8 setsparse)
1965 {
1966         struct cifsInodeInfo *cifsi;
1967         int rc;
1968
1969         cifsi = CIFS_I(inode);
1970
1971         /* if file already sparse don't bother setting sparse again */
1972         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && setsparse)
1973                 return true; /* already sparse */
1974
1975         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && !setsparse)
1976                 return true; /* already not sparse */
1977
1978         /*
1979          * Can't check for sparse support on share the usual way via the
1980          * FS attribute info (FILE_SUPPORTS_SPARSE_FILES) on the share
1981          * since Samba server doesn't set the flag on the share, yet
1982          * supports the set sparse FSCTL and returns sparse correctly
1983          * in the file attributes. If we fail setting sparse though we
1984          * mark that server does not support sparse files for this share
1985          * to avoid repeatedly sending the unsupported fsctl to server
1986          * if the file is repeatedly extended.
1987          */
1988         if (tcon->broken_sparse_sup)
1989                 return false;
1990
1991         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1992                         cfile->fid.volatile_fid, FSCTL_SET_SPARSE,
1993                         true /* is_fctl */,
1994                         &setsparse, 1, CIFSMaxBufSize, NULL, NULL);
1995         if (rc) {
1996                 tcon->broken_sparse_sup = true;
1997                 cifs_dbg(FYI, "set sparse rc = %d\n", rc);
1998                 return false;
1999         }
2000
2001         if (setsparse)
2002                 cifsi->cifsAttrs |= FILE_ATTRIBUTE_SPARSE_FILE;
2003         else
2004                 cifsi->cifsAttrs &= (~FILE_ATTRIBUTE_SPARSE_FILE);
2005
2006         return true;
2007 }
2008
2009 static int
2010 smb2_set_file_size(const unsigned int xid, struct cifs_tcon *tcon,
2011                    struct cifsFileInfo *cfile, __u64 size, bool set_alloc)
2012 {
2013         __le64 eof = cpu_to_le64(size);
2014         struct inode *inode;
2015
2016         /*
2017          * If extending file more than one page make sparse. Many Linux fs
2018          * make files sparse by default when extending via ftruncate
2019          */
2020         inode = d_inode(cfile->dentry);
2021
2022         if (!set_alloc && (size > inode->i_size + 8192)) {
2023                 __u8 set_sparse = 1;
2024
2025                 /* whether set sparse succeeds or not, extend the file */
2026                 smb2_set_sparse(xid, tcon, cfile, inode, set_sparse);
2027         }
2028
2029         return SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
2030                             cfile->fid.volatile_fid, cfile->pid, &eof);
2031 }
2032
2033 static int
2034 smb2_duplicate_extents(const unsigned int xid,
2035                         struct cifsFileInfo *srcfile,
2036                         struct cifsFileInfo *trgtfile, u64 src_off,
2037                         u64 len, u64 dest_off)
2038 {
2039         int rc;
2040         unsigned int ret_data_len;
2041         struct duplicate_extents_to_file dup_ext_buf;
2042         struct cifs_tcon *tcon = tlink_tcon(trgtfile->tlink);
2043
2044         /* server fileays advertise duplicate extent support with this flag */
2045         if ((le32_to_cpu(tcon->fsAttrInfo.Attributes) &
2046              FILE_SUPPORTS_BLOCK_REFCOUNTING) == 0)
2047                 return -EOPNOTSUPP;
2048
2049         dup_ext_buf.VolatileFileHandle = srcfile->fid.volatile_fid;
2050         dup_ext_buf.PersistentFileHandle = srcfile->fid.persistent_fid;
2051         dup_ext_buf.SourceFileOffset = cpu_to_le64(src_off);
2052         dup_ext_buf.TargetFileOffset = cpu_to_le64(dest_off);
2053         dup_ext_buf.ByteCount = cpu_to_le64(len);
2054         cifs_dbg(FYI, "Duplicate extents: src off %lld dst off %lld len %lld\n",
2055                 src_off, dest_off, len);
2056
2057         rc = smb2_set_file_size(xid, tcon, trgtfile, dest_off + len, false);
2058         if (rc)
2059                 goto duplicate_extents_out;
2060
2061         rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
2062                         trgtfile->fid.volatile_fid,
2063                         FSCTL_DUPLICATE_EXTENTS_TO_FILE,
2064                         true /* is_fsctl */,
2065                         (char *)&dup_ext_buf,
2066                         sizeof(struct duplicate_extents_to_file),
2067                         CIFSMaxBufSize, NULL,
2068                         &ret_data_len);
2069
2070         if (ret_data_len > 0)
2071                 cifs_dbg(FYI, "Non-zero response length in duplicate extents\n");
2072
2073 duplicate_extents_out:
2074         return rc;
2075 }
2076
2077 static int
2078 smb2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
2079                    struct cifsFileInfo *cfile)
2080 {
2081         return SMB2_set_compression(xid, tcon, cfile->fid.persistent_fid,
2082                             cfile->fid.volatile_fid);
2083 }
2084
2085 static int
2086 smb3_set_integrity(const unsigned int xid, struct cifs_tcon *tcon,
2087                    struct cifsFileInfo *cfile)
2088 {
2089         struct fsctl_set_integrity_information_req integr_info;
2090         unsigned int ret_data_len;
2091
2092         integr_info.ChecksumAlgorithm = cpu_to_le16(CHECKSUM_TYPE_UNCHANGED);
2093         integr_info.Flags = 0;
2094         integr_info.Reserved = 0;
2095
2096         return SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
2097                         cfile->fid.volatile_fid,
2098                         FSCTL_SET_INTEGRITY_INFORMATION,
2099                         true /* is_fsctl */,
2100                         (char *)&integr_info,
2101                         sizeof(struct fsctl_set_integrity_information_req),
2102                         CIFSMaxBufSize, NULL,
2103                         &ret_data_len);
2104
2105 }
2106
2107 /* GMT Token is @GMT-YYYY.MM.DD-HH.MM.SS Unicode which is 48 bytes + null */
2108 #define GMT_TOKEN_SIZE 50
2109
2110 #define MIN_SNAPSHOT_ARRAY_SIZE 16 /* See MS-SMB2 section 3.3.5.15.1 */
2111
2112 /*
2113  * Input buffer contains (empty) struct smb_snapshot array with size filled in
2114  * For output see struct SRV_SNAPSHOT_ARRAY in MS-SMB2 section 2.2.32.2
2115  */
2116 static int
2117 smb3_enum_snapshots(const unsigned int xid, struct cifs_tcon *tcon,
2118                    struct cifsFileInfo *cfile, void __user *ioc_buf)
2119 {
2120         char *retbuf = NULL;
2121         unsigned int ret_data_len = 0;
2122         int rc;
2123         u32 max_response_size;
2124         struct smb_snapshot_array snapshot_in;
2125
2126         /*
2127          * On the first query to enumerate the list of snapshots available
2128          * for this volume the buffer begins with 0 (number of snapshots
2129          * which can be returned is zero since at that point we do not know
2130          * how big the buffer needs to be). On the second query,
2131          * it (ret_data_len) is set to number of snapshots so we can
2132          * know to set the maximum response size larger (see below).
2133          */
2134         if (get_user(ret_data_len, (unsigned int __user *)ioc_buf))
2135                 return -EFAULT;
2136
2137         /*
2138          * Note that for snapshot queries that servers like Azure expect that
2139          * the first query be minimal size (and just used to get the number/size
2140          * of previous versions) so response size must be specified as EXACTLY
2141          * sizeof(struct snapshot_array) which is 16 when rounded up to multiple
2142          * of eight bytes.
2143          */
2144         if (ret_data_len == 0)
2145                 max_response_size = MIN_SNAPSHOT_ARRAY_SIZE;
2146         else
2147                 max_response_size = CIFSMaxBufSize;
2148
2149         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
2150                         cfile->fid.volatile_fid,
2151                         FSCTL_SRV_ENUMERATE_SNAPSHOTS,
2152                         true /* is_fsctl */,
2153                         NULL, 0 /* no input data */, max_response_size,
2154                         (char **)&retbuf,
2155                         &ret_data_len);
2156         cifs_dbg(FYI, "enum snaphots ioctl returned %d and ret buflen is %d\n",
2157                         rc, ret_data_len);
2158         if (rc)
2159                 return rc;
2160
2161         if (ret_data_len && (ioc_buf != NULL) && (retbuf != NULL)) {
2162                 /* Fixup buffer */
2163                 if (copy_from_user(&snapshot_in, ioc_buf,
2164                     sizeof(struct smb_snapshot_array))) {
2165                         rc = -EFAULT;
2166                         kfree(retbuf);
2167                         return rc;
2168                 }
2169
2170                 /*
2171                  * Check for min size, ie not large enough to fit even one GMT
2172                  * token (snapshot).  On the first ioctl some users may pass in
2173                  * smaller size (or zero) to simply get the size of the array
2174                  * so the user space caller can allocate sufficient memory
2175                  * and retry the ioctl again with larger array size sufficient
2176                  * to hold all of the snapshot GMT tokens on the second try.
2177                  */
2178                 if (snapshot_in.snapshot_array_size < GMT_TOKEN_SIZE)
2179                         ret_data_len = sizeof(struct smb_snapshot_array);
2180
2181                 /*
2182                  * We return struct SRV_SNAPSHOT_ARRAY, followed by
2183                  * the snapshot array (of 50 byte GMT tokens) each
2184                  * representing an available previous version of the data
2185                  */
2186                 if (ret_data_len > (snapshot_in.snapshot_array_size +
2187                                         sizeof(struct smb_snapshot_array)))
2188                         ret_data_len = snapshot_in.snapshot_array_size +
2189                                         sizeof(struct smb_snapshot_array);
2190
2191                 if (copy_to_user(ioc_buf, retbuf, ret_data_len))
2192                         rc = -EFAULT;
2193         }
2194
2195         kfree(retbuf);
2196         return rc;
2197 }
2198
2199
2200
2201 static int
2202 smb3_notify(const unsigned int xid, struct file *pfile,
2203             void __user *ioc_buf)
2204 {
2205         struct smb3_notify notify;
2206         struct dentry *dentry = pfile->f_path.dentry;
2207         struct inode *inode = file_inode(pfile);
2208         struct cifs_sb_info *cifs_sb;
2209         struct cifs_open_parms oparms;
2210         struct cifs_fid fid;
2211         struct cifs_tcon *tcon;
2212         unsigned char *path = NULL;
2213         __le16 *utf16_path = NULL;
2214         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2215         int rc = 0;
2216
2217         path = build_path_from_dentry(dentry);
2218         if (path == NULL)
2219                 return -ENOMEM;
2220
2221         cifs_sb = CIFS_SB(inode->i_sb);
2222
2223         utf16_path = cifs_convert_path_to_utf16(path + 1, cifs_sb);
2224         if (utf16_path == NULL) {
2225                 rc = -ENOMEM;
2226                 goto notify_exit;
2227         }
2228
2229         if (copy_from_user(&notify, ioc_buf, sizeof(struct smb3_notify))) {
2230                 rc = -EFAULT;
2231                 goto notify_exit;
2232         }
2233
2234         tcon = cifs_sb_master_tcon(cifs_sb);
2235         oparms.tcon = tcon;
2236         oparms.desired_access = FILE_READ_ATTRIBUTES | FILE_READ_DATA;
2237         oparms.disposition = FILE_OPEN;
2238         oparms.create_options = cifs_create_options(cifs_sb, 0);
2239         oparms.fid = &fid;
2240         oparms.reconnect = false;
2241
2242         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL,
2243                        NULL);
2244         if (rc)
2245                 goto notify_exit;
2246
2247         rc = SMB2_change_notify(xid, tcon, fid.persistent_fid, fid.volatile_fid,
2248                                 notify.watch_tree, notify.completion_filter);
2249
2250         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
2251
2252         cifs_dbg(FYI, "change notify for path %s rc %d\n", path, rc);
2253
2254 notify_exit:
2255         kfree(path);
2256         kfree(utf16_path);
2257         return rc;
2258 }
2259
2260 static int
2261 smb2_query_dir_first(const unsigned int xid, struct cifs_tcon *tcon,
2262                      const char *path, struct cifs_sb_info *cifs_sb,
2263                      struct cifs_fid *fid, __u16 search_flags,
2264                      struct cifs_search_info *srch_inf)
2265 {
2266         __le16 *utf16_path;
2267         struct smb_rqst rqst[2];
2268         struct kvec rsp_iov[2];
2269         int resp_buftype[2];
2270         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
2271         struct kvec qd_iov[SMB2_QUERY_DIRECTORY_IOV_SIZE];
2272         int rc, flags = 0;
2273         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2274         struct cifs_open_parms oparms;
2275         struct smb2_query_directory_rsp *qd_rsp = NULL;
2276         struct smb2_create_rsp *op_rsp = NULL;
2277         struct TCP_Server_Info *server = cifs_pick_channel(tcon->ses);
2278
2279         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
2280         if (!utf16_path)
2281                 return -ENOMEM;
2282
2283         if (smb3_encryption_required(tcon))
2284                 flags |= CIFS_TRANSFORM_REQ;
2285
2286         memset(rqst, 0, sizeof(rqst));
2287         resp_buftype[0] = resp_buftype[1] = CIFS_NO_BUFFER;
2288         memset(rsp_iov, 0, sizeof(rsp_iov));
2289
2290         /* Open */
2291         memset(&open_iov, 0, sizeof(open_iov));
2292         rqst[0].rq_iov = open_iov;
2293         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
2294
2295         oparms.tcon = tcon;
2296         oparms.desired_access = FILE_READ_ATTRIBUTES | FILE_READ_DATA;
2297         oparms.disposition = FILE_OPEN;
2298         oparms.create_options = cifs_create_options(cifs_sb, 0);
2299         oparms.fid = fid;
2300         oparms.reconnect = false;
2301
2302         rc = SMB2_open_init(tcon, server,
2303                             &rqst[0], &oplock, &oparms, utf16_path);
2304         if (rc)
2305                 goto qdf_free;
2306         smb2_set_next_command(tcon, &rqst[0]);
2307
2308         /* Query directory */
2309         srch_inf->entries_in_buffer = 0;
2310         srch_inf->index_of_last_entry = 2;
2311
2312         memset(&qd_iov, 0, sizeof(qd_iov));
2313         rqst[1].rq_iov = qd_iov;
2314         rqst[1].rq_nvec = SMB2_QUERY_DIRECTORY_IOV_SIZE;
2315
2316         rc = SMB2_query_directory_init(xid, tcon, server,
2317                                        &rqst[1],
2318                                        COMPOUND_FID, COMPOUND_FID,
2319                                        0, srch_inf->info_level);
2320         if (rc)
2321                 goto qdf_free;
2322
2323         smb2_set_related(&rqst[1]);
2324
2325         rc = compound_send_recv(xid, tcon->ses, server,
2326                                 flags, 2, rqst,
2327                                 resp_buftype, rsp_iov);
2328
2329         /* If the open failed there is nothing to do */
2330         op_rsp = (struct smb2_create_rsp *)rsp_iov[0].iov_base;
2331         if (op_rsp == NULL || op_rsp->sync_hdr.Status != STATUS_SUCCESS) {
2332                 cifs_dbg(FYI, "query_dir_first: open failed rc=%d\n", rc);
2333                 goto qdf_free;
2334         }
2335         fid->persistent_fid = op_rsp->PersistentFileId;
2336         fid->volatile_fid = op_rsp->VolatileFileId;
2337
2338         /* Anything else than ENODATA means a genuine error */
2339         if (rc && rc != -ENODATA) {
2340                 SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
2341                 cifs_dbg(FYI, "query_dir_first: query directory failed rc=%d\n", rc);
2342                 trace_smb3_query_dir_err(xid, fid->persistent_fid,
2343                                          tcon->tid, tcon->ses->Suid, 0, 0, rc);
2344                 goto qdf_free;
2345         }
2346
2347         atomic_inc(&tcon->num_remote_opens);
2348
2349         qd_rsp = (struct smb2_query_directory_rsp *)rsp_iov[1].iov_base;
2350         if (qd_rsp->sync_hdr.Status == STATUS_NO_MORE_FILES) {
2351                 trace_smb3_query_dir_done(xid, fid->persistent_fid,
2352                                           tcon->tid, tcon->ses->Suid, 0, 0);
2353                 srch_inf->endOfSearch = true;
2354                 rc = 0;
2355                 goto qdf_free;
2356         }
2357
2358         rc = smb2_parse_query_directory(tcon, &rsp_iov[1], resp_buftype[1],
2359                                         srch_inf);
2360         if (rc) {
2361                 trace_smb3_query_dir_err(xid, fid->persistent_fid, tcon->tid,
2362                         tcon->ses->Suid, 0, 0, rc);
2363                 goto qdf_free;
2364         }
2365         resp_buftype[1] = CIFS_NO_BUFFER;
2366
2367         trace_smb3_query_dir_done(xid, fid->persistent_fid, tcon->tid,
2368                         tcon->ses->Suid, 0, srch_inf->entries_in_buffer);
2369
2370  qdf_free:
2371         kfree(utf16_path);
2372         SMB2_open_free(&rqst[0]);
2373         SMB2_query_directory_free(&rqst[1]);
2374         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
2375         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
2376         return rc;
2377 }
2378
2379 static int
2380 smb2_query_dir_next(const unsigned int xid, struct cifs_tcon *tcon,
2381                     struct cifs_fid *fid, __u16 search_flags,
2382                     struct cifs_search_info *srch_inf)
2383 {
2384         return SMB2_query_directory(xid, tcon, fid->persistent_fid,
2385                                     fid->volatile_fid, 0, srch_inf);
2386 }
2387
2388 static int
2389 smb2_close_dir(const unsigned int xid, struct cifs_tcon *tcon,
2390                struct cifs_fid *fid)
2391 {
2392         return SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
2393 }
2394
2395 /*
2396  * If we negotiate SMB2 protocol and get STATUS_PENDING - update
2397  * the number of credits and return true. Otherwise - return false.
2398  */
2399 static bool
2400 smb2_is_status_pending(char *buf, struct TCP_Server_Info *server)
2401 {
2402         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
2403         int scredits, in_flight;
2404
2405         if (shdr->Status != STATUS_PENDING)
2406                 return false;
2407
2408         if (shdr->CreditRequest) {
2409                 spin_lock(&server->req_lock);
2410                 server->credits += le16_to_cpu(shdr->CreditRequest);
2411                 scredits = server->credits;
2412                 in_flight = server->in_flight;
2413                 spin_unlock(&server->req_lock);
2414                 wake_up(&server->request_q);
2415
2416                 trace_smb3_add_credits(server->CurrentMid,
2417                                 server->conn_id, server->hostname, scredits,
2418                                 le16_to_cpu(shdr->CreditRequest), in_flight);
2419                 cifs_dbg(FYI, "%s: status pending add %u credits total=%d\n",
2420                                 __func__, le16_to_cpu(shdr->CreditRequest), scredits);
2421         }
2422
2423         return true;
2424 }
2425
2426 static bool
2427 smb2_is_session_expired(char *buf)
2428 {
2429         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
2430
2431         if (shdr->Status != STATUS_NETWORK_SESSION_EXPIRED &&
2432             shdr->Status != STATUS_USER_SESSION_DELETED)
2433                 return false;
2434
2435         trace_smb3_ses_expired(shdr->TreeId, shdr->SessionId,
2436                                le16_to_cpu(shdr->Command),
2437                                le64_to_cpu(shdr->MessageId));
2438         cifs_dbg(FYI, "Session expired or deleted\n");
2439
2440         return true;
2441 }
2442
2443 static bool
2444 smb2_is_status_io_timeout(char *buf)
2445 {
2446         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
2447
2448         if (shdr->Status == STATUS_IO_TIMEOUT)
2449                 return true;
2450         else
2451                 return false;
2452 }
2453
2454 static void
2455 smb2_is_network_name_deleted(char *buf, struct TCP_Server_Info *server)
2456 {
2457         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
2458         struct list_head *tmp, *tmp1;
2459         struct cifs_ses *ses;
2460         struct cifs_tcon *tcon;
2461
2462         if (shdr->Status != STATUS_NETWORK_NAME_DELETED)
2463                 return;
2464
2465         spin_lock(&cifs_tcp_ses_lock);
2466         list_for_each(tmp, &server->smb_ses_list) {
2467                 ses = list_entry(tmp, struct cifs_ses, smb_ses_list);
2468                 list_for_each(tmp1, &ses->tcon_list) {
2469                         tcon = list_entry(tmp1, struct cifs_tcon, tcon_list);
2470                         if (tcon->tid == shdr->TreeId) {
2471                                 tcon->need_reconnect = true;
2472                                 spin_unlock(&cifs_tcp_ses_lock);
2473                                 pr_warn_once("Server share %s deleted.\n",
2474                                              tcon->treeName);
2475                                 return;
2476                         }
2477                 }
2478         }
2479         spin_unlock(&cifs_tcp_ses_lock);
2480 }
2481
2482 static int
2483 smb2_oplock_response(struct cifs_tcon *tcon, struct cifs_fid *fid,
2484                      struct cifsInodeInfo *cinode)
2485 {
2486         if (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_LEASING)
2487                 return SMB2_lease_break(0, tcon, cinode->lease_key,
2488                                         smb2_get_lease_state(cinode));
2489
2490         return SMB2_oplock_break(0, tcon, fid->persistent_fid,
2491                                  fid->volatile_fid,
2492                                  CIFS_CACHE_READ(cinode) ? 1 : 0);
2493 }
2494
2495 void
2496 smb2_set_related(struct smb_rqst *rqst)
2497 {
2498         struct smb2_sync_hdr *shdr;
2499
2500         shdr = (struct smb2_sync_hdr *)(rqst->rq_iov[0].iov_base);
2501         if (shdr == NULL) {
2502                 cifs_dbg(FYI, "shdr NULL in smb2_set_related\n");
2503                 return;
2504         }
2505         shdr->Flags |= SMB2_FLAGS_RELATED_OPERATIONS;
2506 }
2507
2508 char smb2_padding[7] = {0, 0, 0, 0, 0, 0, 0};
2509
2510 void
2511 smb2_set_next_command(struct cifs_tcon *tcon, struct smb_rqst *rqst)
2512 {
2513         struct smb2_sync_hdr *shdr;
2514         struct cifs_ses *ses = tcon->ses;
2515         struct TCP_Server_Info *server = ses->server;
2516         unsigned long len = smb_rqst_len(server, rqst);
2517         int i, num_padding;
2518
2519         shdr = (struct smb2_sync_hdr *)(rqst->rq_iov[0].iov_base);
2520         if (shdr == NULL) {
2521                 cifs_dbg(FYI, "shdr NULL in smb2_set_next_command\n");
2522                 return;
2523         }
2524
2525         /* SMB headers in a compound are 8 byte aligned. */
2526
2527         /* No padding needed */
2528         if (!(len & 7))
2529                 goto finished;
2530
2531         num_padding = 8 - (len & 7);
2532         if (!smb3_encryption_required(tcon)) {
2533                 /*
2534                  * If we do not have encryption then we can just add an extra
2535                  * iov for the padding.
2536                  */
2537                 rqst->rq_iov[rqst->rq_nvec].iov_base = smb2_padding;
2538                 rqst->rq_iov[rqst->rq_nvec].iov_len = num_padding;
2539                 rqst->rq_nvec++;
2540                 len += num_padding;
2541         } else {
2542                 /*
2543                  * We can not add a small padding iov for the encryption case
2544                  * because the encryption framework can not handle the padding
2545                  * iovs.
2546                  * We have to flatten this into a single buffer and add
2547                  * the padding to it.
2548                  */
2549                 for (i = 1; i < rqst->rq_nvec; i++) {
2550                         memcpy(rqst->rq_iov[0].iov_base +
2551                                rqst->rq_iov[0].iov_len,
2552                                rqst->rq_iov[i].iov_base,
2553                                rqst->rq_iov[i].iov_len);
2554                         rqst->rq_iov[0].iov_len += rqst->rq_iov[i].iov_len;
2555                 }
2556                 memset(rqst->rq_iov[0].iov_base + rqst->rq_iov[0].iov_len,
2557                        0, num_padding);
2558                 rqst->rq_iov[0].iov_len += num_padding;
2559                 len += num_padding;
2560                 rqst->rq_nvec = 1;
2561         }
2562
2563  finished:
2564         shdr->NextCommand = cpu_to_le32(len);
2565 }
2566
2567 /*
2568  * Passes the query info response back to the caller on success.
2569  * Caller need to free this with free_rsp_buf().
2570  */
2571 int
2572 smb2_query_info_compound(const unsigned int xid, struct cifs_tcon *tcon,
2573                          __le16 *utf16_path, u32 desired_access,
2574                          u32 class, u32 type, u32 output_len,
2575                          struct kvec *rsp, int *buftype,
2576                          struct cifs_sb_info *cifs_sb)
2577 {
2578         struct cifs_ses *ses = tcon->ses;
2579         struct TCP_Server_Info *server = cifs_pick_channel(ses);
2580         int flags = 0;
2581         struct smb_rqst rqst[3];
2582         int resp_buftype[3];
2583         struct kvec rsp_iov[3];
2584         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
2585         struct kvec qi_iov[1];
2586         struct kvec close_iov[1];
2587         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2588         struct cifs_open_parms oparms;
2589         struct cifs_fid fid;
2590         int rc;
2591
2592         if (smb3_encryption_required(tcon))
2593                 flags |= CIFS_TRANSFORM_REQ;
2594
2595         memset(rqst, 0, sizeof(rqst));
2596         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
2597         memset(rsp_iov, 0, sizeof(rsp_iov));
2598
2599         memset(&open_iov, 0, sizeof(open_iov));
2600         rqst[0].rq_iov = open_iov;
2601         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
2602
2603         oparms.tcon = tcon;
2604         oparms.desired_access = desired_access;
2605         oparms.disposition = FILE_OPEN;
2606         oparms.create_options = cifs_create_options(cifs_sb, 0);
2607         oparms.fid = &fid;
2608         oparms.reconnect = false;
2609
2610         rc = SMB2_open_init(tcon, server,
2611                             &rqst[0], &oplock, &oparms, utf16_path);
2612         if (rc)
2613                 goto qic_exit;
2614         smb2_set_next_command(tcon, &rqst[0]);
2615
2616         memset(&qi_iov, 0, sizeof(qi_iov));
2617         rqst[1].rq_iov = qi_iov;
2618         rqst[1].rq_nvec = 1;
2619
2620         rc = SMB2_query_info_init(tcon, server,
2621                                   &rqst[1], COMPOUND_FID, COMPOUND_FID,
2622                                   class, type, 0,
2623                                   output_len, 0,
2624                                   NULL);
2625         if (rc)
2626                 goto qic_exit;
2627         smb2_set_next_command(tcon, &rqst[1]);
2628         smb2_set_related(&rqst[1]);
2629
2630         memset(&close_iov, 0, sizeof(close_iov));
2631         rqst[2].rq_iov = close_iov;
2632         rqst[2].rq_nvec = 1;
2633
2634         rc = SMB2_close_init(tcon, server,
2635                              &rqst[2], COMPOUND_FID, COMPOUND_FID, false);
2636         if (rc)
2637                 goto qic_exit;
2638         smb2_set_related(&rqst[2]);
2639
2640         rc = compound_send_recv(xid, ses, server,
2641                                 flags, 3, rqst,
2642                                 resp_buftype, rsp_iov);
2643         if (rc) {
2644                 free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
2645                 if (rc == -EREMCHG) {
2646                         tcon->need_reconnect = true;
2647                         pr_warn_once("server share %s deleted\n",
2648                                      tcon->treeName);
2649                 }
2650                 goto qic_exit;
2651         }
2652         *rsp = rsp_iov[1];
2653         *buftype = resp_buftype[1];
2654
2655  qic_exit:
2656         SMB2_open_free(&rqst[0]);
2657         SMB2_query_info_free(&rqst[1]);
2658         SMB2_close_free(&rqst[2]);
2659         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
2660         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
2661         return rc;
2662 }
2663
2664 static int
2665 smb2_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
2666              struct cifs_sb_info *cifs_sb, struct kstatfs *buf)
2667 {
2668         struct smb2_query_info_rsp *rsp;
2669         struct smb2_fs_full_size_info *info = NULL;
2670         __le16 utf16_path = 0; /* Null - open root of share */
2671         struct kvec rsp_iov = {NULL, 0};
2672         int buftype = CIFS_NO_BUFFER;
2673         int rc;
2674
2675
2676         rc = smb2_query_info_compound(xid, tcon, &utf16_path,
2677                                       FILE_READ_ATTRIBUTES,
2678                                       FS_FULL_SIZE_INFORMATION,
2679                                       SMB2_O_INFO_FILESYSTEM,
2680                                       sizeof(struct smb2_fs_full_size_info),
2681                                       &rsp_iov, &buftype, cifs_sb);
2682         if (rc)
2683                 goto qfs_exit;
2684
2685         rsp = (struct smb2_query_info_rsp *)rsp_iov.iov_base;
2686         buf->f_type = SMB2_MAGIC_NUMBER;
2687         info = (struct smb2_fs_full_size_info *)(
2688                 le16_to_cpu(rsp->OutputBufferOffset) + (char *)rsp);
2689         rc = smb2_validate_iov(le16_to_cpu(rsp->OutputBufferOffset),
2690                                le32_to_cpu(rsp->OutputBufferLength),
2691                                &rsp_iov,
2692                                sizeof(struct smb2_fs_full_size_info));
2693         if (!rc)
2694                 smb2_copy_fs_info_to_kstatfs(info, buf);
2695
2696 qfs_exit:
2697         free_rsp_buf(buftype, rsp_iov.iov_base);
2698         return rc;
2699 }
2700
2701 static int
2702 smb311_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
2703                struct cifs_sb_info *cifs_sb, struct kstatfs *buf)
2704 {
2705         int rc;
2706         __le16 srch_path = 0; /* Null - open root of share */
2707         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2708         struct cifs_open_parms oparms;
2709         struct cifs_fid fid;
2710
2711         if (!tcon->posix_extensions)
2712                 return smb2_queryfs(xid, tcon, cifs_sb, buf);
2713
2714         oparms.tcon = tcon;
2715         oparms.desired_access = FILE_READ_ATTRIBUTES;
2716         oparms.disposition = FILE_OPEN;
2717         oparms.create_options = cifs_create_options(cifs_sb, 0);
2718         oparms.fid = &fid;
2719         oparms.reconnect = false;
2720
2721         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL,
2722                        NULL, NULL);
2723         if (rc)
2724                 return rc;
2725
2726         rc = SMB311_posix_qfs_info(xid, tcon, fid.persistent_fid,
2727                                    fid.volatile_fid, buf);
2728         buf->f_type = SMB2_MAGIC_NUMBER;
2729         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
2730         return rc;
2731 }
2732
2733 static bool
2734 smb2_compare_fids(struct cifsFileInfo *ob1, struct cifsFileInfo *ob2)
2735 {
2736         return ob1->fid.persistent_fid == ob2->fid.persistent_fid &&
2737                ob1->fid.volatile_fid == ob2->fid.volatile_fid;
2738 }
2739
2740 static int
2741 smb2_mand_lock(const unsigned int xid, struct cifsFileInfo *cfile, __u64 offset,
2742                __u64 length, __u32 type, int lock, int unlock, bool wait)
2743 {
2744         if (unlock && !lock)
2745                 type = SMB2_LOCKFLAG_UNLOCK;
2746         return SMB2_lock(xid, tlink_tcon(cfile->tlink),
2747                          cfile->fid.persistent_fid, cfile->fid.volatile_fid,
2748                          current->tgid, length, offset, type, wait);
2749 }
2750
2751 static void
2752 smb2_get_lease_key(struct inode *inode, struct cifs_fid *fid)
2753 {
2754         memcpy(fid->lease_key, CIFS_I(inode)->lease_key, SMB2_LEASE_KEY_SIZE);
2755 }
2756
2757 static void
2758 smb2_set_lease_key(struct inode *inode, struct cifs_fid *fid)
2759 {
2760         memcpy(CIFS_I(inode)->lease_key, fid->lease_key, SMB2_LEASE_KEY_SIZE);
2761 }
2762
2763 static void
2764 smb2_new_lease_key(struct cifs_fid *fid)
2765 {
2766         generate_random_uuid(fid->lease_key);
2767 }
2768
2769 static int
2770 smb2_get_dfs_refer(const unsigned int xid, struct cifs_ses *ses,
2771                    const char *search_name,
2772                    struct dfs_info3_param **target_nodes,
2773                    unsigned int *num_of_nodes,
2774                    const struct nls_table *nls_codepage, int remap)
2775 {
2776         int rc;
2777         __le16 *utf16_path = NULL;
2778         int utf16_path_len = 0;
2779         struct cifs_tcon *tcon;
2780         struct fsctl_get_dfs_referral_req *dfs_req = NULL;
2781         struct get_dfs_referral_rsp *dfs_rsp = NULL;
2782         u32 dfs_req_size = 0, dfs_rsp_size = 0;
2783
2784         cifs_dbg(FYI, "%s: path: %s\n", __func__, search_name);
2785
2786         /*
2787          * Try to use the IPC tcon, otherwise just use any
2788          */
2789         tcon = ses->tcon_ipc;
2790         if (tcon == NULL) {
2791                 spin_lock(&cifs_tcp_ses_lock);
2792                 tcon = list_first_entry_or_null(&ses->tcon_list,
2793                                                 struct cifs_tcon,
2794                                                 tcon_list);
2795                 if (tcon)
2796                         tcon->tc_count++;
2797                 spin_unlock(&cifs_tcp_ses_lock);
2798         }
2799
2800         if (tcon == NULL) {
2801                 cifs_dbg(VFS, "session %p has no tcon available for a dfs referral request\n",
2802                          ses);
2803                 rc = -ENOTCONN;
2804                 goto out;
2805         }
2806
2807         utf16_path = cifs_strndup_to_utf16(search_name, PATH_MAX,
2808                                            &utf16_path_len,
2809                                            nls_codepage, remap);
2810         if (!utf16_path) {
2811                 rc = -ENOMEM;
2812                 goto out;
2813         }
2814
2815         dfs_req_size = sizeof(*dfs_req) + utf16_path_len;
2816         dfs_req = kzalloc(dfs_req_size, GFP_KERNEL);
2817         if (!dfs_req) {
2818                 rc = -ENOMEM;
2819                 goto out;
2820         }
2821
2822         /* Highest DFS referral version understood */
2823         dfs_req->MaxReferralLevel = DFS_VERSION;
2824
2825         /* Path to resolve in an UTF-16 null-terminated string */
2826         memcpy(dfs_req->RequestFileName, utf16_path, utf16_path_len);
2827
2828         do {
2829                 rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
2830                                 FSCTL_DFS_GET_REFERRALS,
2831                                 true /* is_fsctl */,
2832                                 (char *)dfs_req, dfs_req_size, CIFSMaxBufSize,
2833                                 (char **)&dfs_rsp, &dfs_rsp_size);
2834         } while (rc == -EAGAIN);
2835
2836         if (rc) {
2837                 if ((rc != -ENOENT) && (rc != -EOPNOTSUPP))
2838                         cifs_tcon_dbg(VFS, "ioctl error in %s rc=%d\n", __func__, rc);
2839                 goto out;
2840         }
2841
2842         rc = parse_dfs_referrals(dfs_rsp, dfs_rsp_size,
2843                                  num_of_nodes, target_nodes,
2844                                  nls_codepage, remap, search_name,
2845                                  true /* is_unicode */);
2846         if (rc) {
2847                 cifs_tcon_dbg(VFS, "parse error in %s rc=%d\n", __func__, rc);
2848                 goto out;
2849         }
2850
2851  out:
2852         if (tcon && !tcon->ipc) {
2853                 /* ipc tcons are not refcounted */
2854                 spin_lock(&cifs_tcp_ses_lock);
2855                 tcon->tc_count--;
2856                 spin_unlock(&cifs_tcp_ses_lock);
2857         }
2858         kfree(utf16_path);
2859         kfree(dfs_req);
2860         kfree(dfs_rsp);
2861         return rc;
2862 }
2863
2864 static int
2865 parse_reparse_posix(struct reparse_posix_data *symlink_buf,
2866                       u32 plen, char **target_path,
2867                       struct cifs_sb_info *cifs_sb)
2868 {
2869         unsigned int len;
2870
2871         /* See MS-FSCC 2.1.2.6 for the 'NFS' style reparse tags */
2872         len = le16_to_cpu(symlink_buf->ReparseDataLength);
2873
2874         if (le64_to_cpu(symlink_buf->InodeType) != NFS_SPECFILE_LNK) {
2875                 cifs_dbg(VFS, "%lld not a supported symlink type\n",
2876                         le64_to_cpu(symlink_buf->InodeType));
2877                 return -EOPNOTSUPP;
2878         }
2879
2880         *target_path = cifs_strndup_from_utf16(
2881                                 symlink_buf->PathBuffer,
2882                                 len, true, cifs_sb->local_nls);
2883         if (!(*target_path))
2884                 return -ENOMEM;
2885
2886         convert_delimiter(*target_path, '/');
2887         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
2888
2889         return 0;
2890 }
2891
2892 static int
2893 parse_reparse_symlink(struct reparse_symlink_data_buffer *symlink_buf,
2894                       u32 plen, char **target_path,
2895                       struct cifs_sb_info *cifs_sb)
2896 {
2897         unsigned int sub_len;
2898         unsigned int sub_offset;
2899
2900         /* We handle Symbolic Link reparse tag here. See: MS-FSCC 2.1.2.4 */
2901
2902         sub_offset = le16_to_cpu(symlink_buf->SubstituteNameOffset);
2903         sub_len = le16_to_cpu(symlink_buf->SubstituteNameLength);
2904         if (sub_offset + 20 > plen ||
2905             sub_offset + sub_len + 20 > plen) {
2906                 cifs_dbg(VFS, "srv returned malformed symlink buffer\n");
2907                 return -EIO;
2908         }
2909
2910         *target_path = cifs_strndup_from_utf16(
2911                                 symlink_buf->PathBuffer + sub_offset,
2912                                 sub_len, true, cifs_sb->local_nls);
2913         if (!(*target_path))
2914                 return -ENOMEM;
2915
2916         convert_delimiter(*target_path, '/');
2917         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
2918
2919         return 0;
2920 }
2921
2922 static int
2923 parse_reparse_point(struct reparse_data_buffer *buf,
2924                     u32 plen, char **target_path,
2925                     struct cifs_sb_info *cifs_sb)
2926 {
2927         if (plen < sizeof(struct reparse_data_buffer)) {
2928                 cifs_dbg(VFS, "reparse buffer is too small. Must be at least 8 bytes but was %d\n",
2929                          plen);
2930                 return -EIO;
2931         }
2932
2933         if (plen < le16_to_cpu(buf->ReparseDataLength) +
2934             sizeof(struct reparse_data_buffer)) {
2935                 cifs_dbg(VFS, "srv returned invalid reparse buf length: %d\n",
2936                          plen);
2937                 return -EIO;
2938         }
2939
2940         /* See MS-FSCC 2.1.2 */
2941         switch (le32_to_cpu(buf->ReparseTag)) {
2942         case IO_REPARSE_TAG_NFS:
2943                 return parse_reparse_posix(
2944                         (struct reparse_posix_data *)buf,
2945                         plen, target_path, cifs_sb);
2946         case IO_REPARSE_TAG_SYMLINK:
2947                 return parse_reparse_symlink(
2948                         (struct reparse_symlink_data_buffer *)buf,
2949                         plen, target_path, cifs_sb);
2950         default:
2951                 cifs_dbg(VFS, "srv returned unknown symlink buffer tag:0x%08x\n",
2952                          le32_to_cpu(buf->ReparseTag));
2953                 return -EOPNOTSUPP;
2954         }
2955 }
2956
2957 #define SMB2_SYMLINK_STRUCT_SIZE \
2958         (sizeof(struct smb2_err_rsp) - 1 + sizeof(struct smb2_symlink_err_rsp))
2959
2960 static int
2961 smb2_query_symlink(const unsigned int xid, struct cifs_tcon *tcon,
2962                    struct cifs_sb_info *cifs_sb, const char *full_path,
2963                    char **target_path, bool is_reparse_point)
2964 {
2965         int rc;
2966         __le16 *utf16_path = NULL;
2967         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2968         struct cifs_open_parms oparms;
2969         struct cifs_fid fid;
2970         struct kvec err_iov = {NULL, 0};
2971         struct smb2_err_rsp *err_buf = NULL;
2972         struct smb2_symlink_err_rsp *symlink;
2973         struct TCP_Server_Info *server = cifs_pick_channel(tcon->ses);
2974         unsigned int sub_len;
2975         unsigned int sub_offset;
2976         unsigned int print_len;
2977         unsigned int print_offset;
2978         int flags = 0;
2979         struct smb_rqst rqst[3];
2980         int resp_buftype[3];
2981         struct kvec rsp_iov[3];
2982         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
2983         struct kvec io_iov[SMB2_IOCTL_IOV_SIZE];
2984         struct kvec close_iov[1];
2985         struct smb2_create_rsp *create_rsp;
2986         struct smb2_ioctl_rsp *ioctl_rsp;
2987         struct reparse_data_buffer *reparse_buf;
2988         int create_options = is_reparse_point ? OPEN_REPARSE_POINT : 0;
2989         u32 plen;
2990
2991         cifs_dbg(FYI, "%s: path: %s\n", __func__, full_path);
2992
2993         *target_path = NULL;
2994
2995         if (smb3_encryption_required(tcon))
2996                 flags |= CIFS_TRANSFORM_REQ;
2997
2998         memset(rqst, 0, sizeof(rqst));
2999         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
3000         memset(rsp_iov, 0, sizeof(rsp_iov));
3001
3002         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
3003         if (!utf16_path)
3004                 return -ENOMEM;
3005
3006         /* Open */
3007         memset(&open_iov, 0, sizeof(open_iov));
3008         rqst[0].rq_iov = open_iov;
3009         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
3010
3011         memset(&oparms, 0, sizeof(oparms));
3012         oparms.tcon = tcon;
3013         oparms.desired_access = FILE_READ_ATTRIBUTES;
3014         oparms.disposition = FILE_OPEN;
3015         oparms.create_options = cifs_create_options(cifs_sb, create_options);
3016         oparms.fid = &fid;
3017         oparms.reconnect = false;
3018
3019         rc = SMB2_open_init(tcon, server,
3020                             &rqst[0], &oplock, &oparms, utf16_path);
3021         if (rc)
3022                 goto querty_exit;
3023         smb2_set_next_command(tcon, &rqst[0]);
3024
3025
3026         /* IOCTL */
3027         memset(&io_iov, 0, sizeof(io_iov));
3028         rqst[1].rq_iov = io_iov;
3029         rqst[1].rq_nvec = SMB2_IOCTL_IOV_SIZE;
3030
3031         rc = SMB2_ioctl_init(tcon, server,
3032                              &rqst[1], fid.persistent_fid,
3033                              fid.volatile_fid, FSCTL_GET_REPARSE_POINT,
3034                              true /* is_fctl */, NULL, 0,
3035                              CIFSMaxBufSize -
3036                              MAX_SMB2_CREATE_RESPONSE_SIZE -
3037                              MAX_SMB2_CLOSE_RESPONSE_SIZE);
3038         if (rc)
3039                 goto querty_exit;
3040
3041         smb2_set_next_command(tcon, &rqst[1]);
3042         smb2_set_related(&rqst[1]);
3043
3044
3045         /* Close */
3046         memset(&close_iov, 0, sizeof(close_iov));
3047         rqst[2].rq_iov = close_iov;
3048         rqst[2].rq_nvec = 1;
3049
3050         rc = SMB2_close_init(tcon, server,
3051                              &rqst[2], COMPOUND_FID, COMPOUND_FID, false);
3052         if (rc)
3053                 goto querty_exit;
3054
3055         smb2_set_related(&rqst[2]);
3056
3057         rc = compound_send_recv(xid, tcon->ses, server,
3058                                 flags, 3, rqst,
3059                                 resp_buftype, rsp_iov);
3060
3061         create_rsp = rsp_iov[0].iov_base;
3062         if (create_rsp && create_rsp->sync_hdr.Status)
3063                 err_iov = rsp_iov[0];
3064         ioctl_rsp = rsp_iov[1].iov_base;
3065
3066         /*
3067          * Open was successful and we got an ioctl response.
3068          */
3069         if ((rc == 0) && (is_reparse_point)) {
3070                 /* See MS-FSCC 2.3.23 */
3071
3072                 reparse_buf = (struct reparse_data_buffer *)
3073                         ((char *)ioctl_rsp +
3074                          le32_to_cpu(ioctl_rsp->OutputOffset));
3075                 plen = le32_to_cpu(ioctl_rsp->OutputCount);
3076
3077                 if (plen + le32_to_cpu(ioctl_rsp->OutputOffset) >
3078                     rsp_iov[1].iov_len) {
3079                         cifs_tcon_dbg(VFS, "srv returned invalid ioctl len: %d\n",
3080                                  plen);
3081                         rc = -EIO;
3082                         goto querty_exit;
3083                 }
3084
3085                 rc = parse_reparse_point(reparse_buf, plen, target_path,
3086                                          cifs_sb);
3087                 goto querty_exit;
3088         }
3089
3090         if (!rc || !err_iov.iov_base) {
3091                 rc = -ENOENT;
3092                 goto querty_exit;
3093         }
3094
3095         err_buf = err_iov.iov_base;
3096         if (le32_to_cpu(err_buf->ByteCount) < sizeof(struct smb2_symlink_err_rsp) ||
3097             err_iov.iov_len < SMB2_SYMLINK_STRUCT_SIZE) {
3098                 rc = -EINVAL;
3099                 goto querty_exit;
3100         }
3101
3102         symlink = (struct smb2_symlink_err_rsp *)err_buf->ErrorData;
3103         if (le32_to_cpu(symlink->SymLinkErrorTag) != SYMLINK_ERROR_TAG ||
3104             le32_to_cpu(symlink->ReparseTag) != IO_REPARSE_TAG_SYMLINK) {
3105                 rc = -EINVAL;
3106                 goto querty_exit;
3107         }
3108
3109         /* open must fail on symlink - reset rc */
3110         rc = 0;
3111         sub_len = le16_to_cpu(symlink->SubstituteNameLength);
3112         sub_offset = le16_to_cpu(symlink->SubstituteNameOffset);
3113         print_len = le16_to_cpu(symlink->PrintNameLength);
3114         print_offset = le16_to_cpu(symlink->PrintNameOffset);
3115
3116         if (err_iov.iov_len < SMB2_SYMLINK_STRUCT_SIZE + sub_offset + sub_len) {
3117                 rc = -EINVAL;
3118                 goto querty_exit;
3119         }
3120
3121         if (err_iov.iov_len <
3122             SMB2_SYMLINK_STRUCT_SIZE + print_offset + print_len) {
3123                 rc = -EINVAL;
3124                 goto querty_exit;
3125         }
3126
3127         *target_path = cifs_strndup_from_utf16(
3128                                 (char *)symlink->PathBuffer + sub_offset,
3129                                 sub_len, true, cifs_sb->local_nls);
3130         if (!(*target_path)) {
3131                 rc = -ENOMEM;
3132                 goto querty_exit;
3133         }
3134         convert_delimiter(*target_path, '/');
3135         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
3136
3137  querty_exit:
3138         cifs_dbg(FYI, "query symlink rc %d\n", rc);
3139         kfree(utf16_path);
3140         SMB2_open_free(&rqst[0]);
3141         SMB2_ioctl_free(&rqst[1]);
3142         SMB2_close_free(&rqst[2]);
3143         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
3144         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
3145         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
3146         return rc;
3147 }
3148
3149 int
3150 smb2_query_reparse_tag(const unsigned int xid, struct cifs_tcon *tcon,
3151                    struct cifs_sb_info *cifs_sb, const char *full_path,
3152                    __u32 *tag)
3153 {
3154         int rc;
3155         __le16 *utf16_path = NULL;
3156         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
3157         struct cifs_open_parms oparms;
3158         struct cifs_fid fid;
3159         struct TCP_Server_Info *server = cifs_pick_channel(tcon->ses);
3160         int flags = 0;
3161         struct smb_rqst rqst[3];
3162         int resp_buftype[3];
3163         struct kvec rsp_iov[3];
3164         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
3165         struct kvec io_iov[SMB2_IOCTL_IOV_SIZE];
3166         struct kvec close_iov[1];
3167         struct smb2_ioctl_rsp *ioctl_rsp;
3168         struct reparse_data_buffer *reparse_buf;
3169         u32 plen;
3170
3171         cifs_dbg(FYI, "%s: path: %s\n", __func__, full_path);
3172
3173         if (smb3_encryption_required(tcon))
3174                 flags |= CIFS_TRANSFORM_REQ;
3175
3176         memset(rqst, 0, sizeof(rqst));
3177         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
3178         memset(rsp_iov, 0, sizeof(rsp_iov));
3179
3180         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
3181         if (!utf16_path)
3182                 return -ENOMEM;
3183
3184         /*
3185          * setup smb2open - TODO add optimization to call cifs_get_readable_path
3186          * to see if there is a handle already open that we can use
3187          */
3188         memset(&open_iov, 0, sizeof(open_iov));
3189         rqst[0].rq_iov = open_iov;
3190         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
3191
3192         memset(&oparms, 0, sizeof(oparms));
3193         oparms.tcon = tcon;
3194         oparms.desired_access = FILE_READ_ATTRIBUTES;
3195         oparms.disposition = FILE_OPEN;
3196         oparms.create_options = cifs_create_options(cifs_sb, OPEN_REPARSE_POINT);
3197         oparms.fid = &fid;
3198         oparms.reconnect = false;
3199
3200         rc = SMB2_open_init(tcon, server,
3201                             &rqst[0], &oplock, &oparms, utf16_path);
3202         if (rc)
3203                 goto query_rp_exit;
3204         smb2_set_next_command(tcon, &rqst[0]);
3205
3206
3207         /* IOCTL */
3208         memset(&io_iov, 0, sizeof(io_iov));
3209         rqst[1].rq_iov = io_iov;
3210         rqst[1].rq_nvec = SMB2_IOCTL_IOV_SIZE;
3211
3212         rc = SMB2_ioctl_init(tcon, server,
3213                              &rqst[1], COMPOUND_FID,
3214                              COMPOUND_FID, FSCTL_GET_REPARSE_POINT,
3215                              true /* is_fctl */, NULL, 0,
3216                              CIFSMaxBufSize -
3217                              MAX_SMB2_CREATE_RESPONSE_SIZE -
3218                              MAX_SMB2_CLOSE_RESPONSE_SIZE);
3219         if (rc)
3220                 goto query_rp_exit;
3221
3222         smb2_set_next_command(tcon, &rqst[1]);
3223         smb2_set_related(&rqst[1]);
3224
3225
3226         /* Close */
3227         memset(&close_iov, 0, sizeof(close_iov));
3228         rqst[2].rq_iov = close_iov;
3229         rqst[2].rq_nvec = 1;
3230
3231         rc = SMB2_close_init(tcon, server,
3232                              &rqst[2], COMPOUND_FID, COMPOUND_FID, false);
3233         if (rc)
3234                 goto query_rp_exit;
3235
3236         smb2_set_related(&rqst[2]);
3237
3238         rc = compound_send_recv(xid, tcon->ses, server,
3239                                 flags, 3, rqst,
3240                                 resp_buftype, rsp_iov);
3241
3242         ioctl_rsp = rsp_iov[1].iov_base;
3243
3244         /*
3245          * Open was successful and we got an ioctl response.
3246          */
3247         if (rc == 0) {
3248                 /* See MS-FSCC 2.3.23 */
3249
3250                 reparse_buf = (struct reparse_data_buffer *)
3251                         ((char *)ioctl_rsp +
3252                          le32_to_cpu(ioctl_rsp->OutputOffset));
3253                 plen = le32_to_cpu(ioctl_rsp->OutputCount);
3254
3255                 if (plen + le32_to_cpu(ioctl_rsp->OutputOffset) >
3256                     rsp_iov[1].iov_len) {
3257                         cifs_tcon_dbg(FYI, "srv returned invalid ioctl len: %d\n",
3258                                  plen);
3259                         rc = -EIO;
3260                         goto query_rp_exit;
3261                 }
3262                 *tag = le32_to_cpu(reparse_buf->ReparseTag);
3263         }
3264
3265  query_rp_exit:
3266         kfree(utf16_path);
3267         SMB2_open_free(&rqst[0]);
3268         SMB2_ioctl_free(&rqst[1]);
3269         SMB2_close_free(&rqst[2]);
3270         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
3271         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
3272         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
3273         return rc;
3274 }
3275
3276 static struct cifs_ntsd *
3277 get_smb2_acl_by_fid(struct cifs_sb_info *cifs_sb,
3278                     const struct cifs_fid *cifsfid, u32 *pacllen, u32 info)
3279 {
3280         struct cifs_ntsd *pntsd = NULL;
3281         unsigned int xid;
3282         int rc = -EOPNOTSUPP;
3283         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
3284
3285         if (IS_ERR(tlink))
3286                 return ERR_CAST(tlink);
3287
3288         xid = get_xid();
3289         cifs_dbg(FYI, "trying to get acl\n");
3290
3291         rc = SMB2_query_acl(xid, tlink_tcon(tlink), cifsfid->persistent_fid,
3292                             cifsfid->volatile_fid, (void **)&pntsd, pacllen,
3293                             info);
3294         free_xid(xid);
3295
3296         cifs_put_tlink(tlink);
3297
3298         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
3299         if (rc)
3300                 return ERR_PTR(rc);
3301         return pntsd;
3302
3303 }
3304
3305 static struct cifs_ntsd *
3306 get_smb2_acl_by_path(struct cifs_sb_info *cifs_sb,
3307                      const char *path, u32 *pacllen, u32 info)
3308 {
3309         struct cifs_ntsd *pntsd = NULL;
3310         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
3311         unsigned int xid;
3312         int rc;
3313         struct cifs_tcon *tcon;
3314         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
3315         struct cifs_fid fid;
3316         struct cifs_open_parms oparms;
3317         __le16 *utf16_path;
3318
3319         cifs_dbg(FYI, "get smb3 acl for path %s\n", path);
3320         if (IS_ERR(tlink))
3321                 return ERR_CAST(tlink);
3322
3323         tcon = tlink_tcon(tlink);
3324         xid = get_xid();
3325
3326         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
3327         if (!utf16_path) {
3328                 rc = -ENOMEM;
3329                 free_xid(xid);
3330                 return ERR_PTR(rc);
3331         }
3332
3333         oparms.tcon = tcon;
3334         oparms.desired_access = READ_CONTROL;
3335         oparms.disposition = FILE_OPEN;
3336         /*
3337          * When querying an ACL, even if the file is a symlink we want to open
3338          * the source not the target, and so the protocol requires that the
3339          * client specify this flag when opening a reparse point
3340          */
3341         oparms.create_options = cifs_create_options(cifs_sb, 0) | OPEN_REPARSE_POINT;
3342         oparms.fid = &fid;
3343         oparms.reconnect = false;
3344
3345         if (info & SACL_SECINFO)
3346                 oparms.desired_access |= SYSTEM_SECURITY;
3347
3348         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL,
3349                        NULL);
3350         kfree(utf16_path);
3351         if (!rc) {
3352                 rc = SMB2_query_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
3353                                     fid.volatile_fid, (void **)&pntsd, pacllen,
3354                                     info);
3355                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
3356         }
3357
3358         cifs_put_tlink(tlink);
3359         free_xid(xid);
3360
3361         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
3362         if (rc)
3363                 return ERR_PTR(rc);
3364         return pntsd;
3365 }
3366
3367 static int
3368 set_smb2_acl(struct cifs_ntsd *pnntsd, __u32 acllen,
3369                 struct inode *inode, const char *path, int aclflag)
3370 {
3371         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
3372         unsigned int xid;
3373         int rc, access_flags = 0;
3374         struct cifs_tcon *tcon;
3375         struct cifs_sb_info *cifs_sb = CIFS_SB(inode->i_sb);
3376         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
3377         struct cifs_fid fid;
3378         struct cifs_open_parms oparms;
3379         __le16 *utf16_path;
3380
3381         cifs_dbg(FYI, "set smb3 acl for path %s\n", path);
3382         if (IS_ERR(tlink))
3383                 return PTR_ERR(tlink);
3384
3385         tcon = tlink_tcon(tlink);
3386         xid = get_xid();
3387
3388         if (aclflag & CIFS_ACL_OWNER || aclflag & CIFS_ACL_GROUP)
3389                 access_flags |= WRITE_OWNER;
3390         if (aclflag & CIFS_ACL_SACL)
3391                 access_flags |= SYSTEM_SECURITY;
3392         if (aclflag & CIFS_ACL_DACL)
3393                 access_flags |= WRITE_DAC;
3394
3395         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
3396         if (!utf16_path) {
3397                 rc = -ENOMEM;
3398                 free_xid(xid);
3399                 return rc;
3400         }
3401
3402         oparms.tcon = tcon;
3403         oparms.desired_access = access_flags;
3404         oparms.create_options = cifs_create_options(cifs_sb, 0);
3405         oparms.disposition = FILE_OPEN;
3406         oparms.path = path;
3407         oparms.fid = &fid;
3408         oparms.reconnect = false;
3409
3410         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL,
3411                        NULL, NULL);
3412         kfree(utf16_path);
3413         if (!rc) {
3414                 rc = SMB2_set_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
3415                             fid.volatile_fid, pnntsd, acllen, aclflag);
3416                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
3417         }
3418
3419         cifs_put_tlink(tlink);
3420         free_xid(xid);
3421         return rc;
3422 }
3423
3424 /* Retrieve an ACL from the server */
3425 static struct cifs_ntsd *
3426 get_smb2_acl(struct cifs_sb_info *cifs_sb,
3427              struct inode *inode, const char *path,
3428              u32 *pacllen, u32 info)
3429 {
3430         struct cifs_ntsd *pntsd = NULL;
3431         struct cifsFileInfo *open_file = NULL;
3432
3433         if (inode && !(info & SACL_SECINFO))
3434                 open_file = find_readable_file(CIFS_I(inode), true);
3435         if (!open_file || (info & SACL_SECINFO))
3436                 return get_smb2_acl_by_path(cifs_sb, path, pacllen, info);
3437
3438         pntsd = get_smb2_acl_by_fid(cifs_sb, &open_file->fid, pacllen, info);
3439         cifsFileInfo_put(open_file);
3440         return pntsd;
3441 }
3442
3443 static long smb3_zero_range(struct file *file, struct cifs_tcon *tcon,
3444                             loff_t offset, loff_t len, bool keep_size)
3445 {
3446         struct cifs_ses *ses = tcon->ses;
3447         struct inode *inode;
3448         struct cifsInodeInfo *cifsi;
3449         struct cifsFileInfo *cfile = file->private_data;
3450         struct file_zero_data_information fsctl_buf;
3451         long rc;
3452         unsigned int xid;
3453         __le64 eof;
3454
3455         xid = get_xid();
3456
3457         inode = d_inode(cfile->dentry);
3458         cifsi = CIFS_I(inode);
3459
3460         trace_smb3_zero_enter(xid, cfile->fid.persistent_fid, tcon->tid,
3461                               ses->Suid, offset, len);
3462
3463         /*
3464          * We zero the range through ioctl, so we need remove the page caches
3465          * first, otherwise the data may be inconsistent with the server.
3466          */
3467         truncate_pagecache_range(inode, offset, offset + len - 1);
3468
3469         /* if file not oplocked can't be sure whether asking to extend size */
3470         if (!CIFS_CACHE_READ(cifsi))
3471                 if (keep_size == false) {
3472                         rc = -EOPNOTSUPP;
3473                         trace_smb3_zero_err(xid, cfile->fid.persistent_fid,
3474                                 tcon->tid, ses->Suid, offset, len, rc);
3475                         free_xid(xid);
3476                         return rc;
3477                 }
3478
3479         cifs_dbg(FYI, "Offset %lld len %lld\n", offset, len);
3480
3481         fsctl_buf.FileOffset = cpu_to_le64(offset);
3482         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
3483
3484         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
3485                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA, true,
3486                         (char *)&fsctl_buf,
3487                         sizeof(struct file_zero_data_information),
3488                         0, NULL, NULL);
3489         if (rc)
3490                 goto zero_range_exit;
3491
3492         /*
3493          * do we also need to change the size of the file?
3494          */
3495         if (keep_size == false && i_size_read(inode) < offset + len) {
3496                 eof = cpu_to_le64(offset + len);
3497                 rc = SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
3498                                   cfile->fid.volatile_fid, cfile->pid, &eof);
3499         }
3500
3501  zero_range_exit:
3502         free_xid(xid);
3503         if (rc)
3504                 trace_smb3_zero_err(xid, cfile->fid.persistent_fid, tcon->tid,
3505                               ses->Suid, offset, len, rc);
3506         else
3507                 trace_smb3_zero_done(xid, cfile->fid.persistent_fid, tcon->tid,
3508                               ses->Suid, offset, len);
3509         return rc;
3510 }
3511
3512 static long smb3_punch_hole(struct file *file, struct cifs_tcon *tcon,
3513                             loff_t offset, loff_t len)
3514 {
3515         struct inode *inode;
3516         struct cifsFileInfo *cfile = file->private_data;
3517         struct file_zero_data_information fsctl_buf;
3518         long rc;
3519         unsigned int xid;
3520         __u8 set_sparse = 1;
3521
3522         xid = get_xid();
3523
3524         inode = d_inode(cfile->dentry);
3525
3526         /* Need to make file sparse, if not already, before freeing range. */
3527         /* Consider adding equivalent for compressed since it could also work */
3528         if (!smb2_set_sparse(xid, tcon, cfile, inode, set_sparse)) {
3529                 rc = -EOPNOTSUPP;
3530                 free_xid(xid);
3531                 return rc;
3532         }
3533
3534         /*
3535          * We implement the punch hole through ioctl, so we need remove the page
3536          * caches first, otherwise the data may be inconsistent with the server.
3537          */
3538         truncate_pagecache_range(inode, offset, offset + len - 1);
3539
3540         cifs_dbg(FYI, "Offset %lld len %lld\n", offset, len);
3541
3542         fsctl_buf.FileOffset = cpu_to_le64(offset);
3543         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
3544
3545         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
3546                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
3547                         true /* is_fctl */, (char *)&fsctl_buf,
3548                         sizeof(struct file_zero_data_information),
3549                         CIFSMaxBufSize, NULL, NULL);
3550         free_xid(xid);
3551         return rc;
3552 }
3553
3554 static long smb3_simple_falloc(struct file *file, struct cifs_tcon *tcon,
3555                             loff_t off, loff_t len, bool keep_size)
3556 {
3557         struct inode *inode;
3558         struct cifsInodeInfo *cifsi;
3559         struct cifsFileInfo *cfile = file->private_data;
3560         long rc = -EOPNOTSUPP;
3561         unsigned int xid;
3562         __le64 eof;
3563
3564         xid = get_xid();
3565
3566         inode = d_inode(cfile->dentry);
3567         cifsi = CIFS_I(inode);
3568
3569         trace_smb3_falloc_enter(xid, cfile->fid.persistent_fid, tcon->tid,
3570                                 tcon->ses->Suid, off, len);
3571         /* if file not oplocked can't be sure whether asking to extend size */
3572         if (!CIFS_CACHE_READ(cifsi))
3573                 if (keep_size == false) {
3574                         trace_smb3_falloc_err(xid, cfile->fid.persistent_fid,
3575                                 tcon->tid, tcon->ses->Suid, off, len, rc);
3576                         free_xid(xid);
3577                         return rc;
3578                 }
3579
3580         /*
3581          * Extending the file
3582          */
3583         if ((keep_size == false) && i_size_read(inode) < off + len) {
3584                 rc = inode_newsize_ok(inode, off + len);
3585                 if (rc)
3586                         goto out;
3587
3588                 if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) == 0)
3589                         smb2_set_sparse(xid, tcon, cfile, inode, false);
3590
3591                 eof = cpu_to_le64(off + len);
3592                 rc = SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
3593                                   cfile->fid.volatile_fid, cfile->pid, &eof);
3594                 if (rc == 0) {
3595                         cifsi->server_eof = off + len;
3596                         cifs_setsize(inode, off + len);
3597                         cifs_truncate_page(inode->i_mapping, inode->i_size);
3598                         truncate_setsize(inode, off + len);
3599                 }
3600                 goto out;
3601         }
3602
3603         /*
3604          * Files are non-sparse by default so falloc may be a no-op
3605          * Must check if file sparse. If not sparse, and since we are not
3606          * extending then no need to do anything since file already allocated
3607          */
3608         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) == 0) {
3609                 rc = 0;
3610                 goto out;
3611         }
3612
3613         if ((keep_size == true) || (i_size_read(inode) >= off + len)) {
3614                 /*
3615                  * Check if falloc starts within first few pages of file
3616                  * and ends within a few pages of the end of file to
3617                  * ensure that most of file is being forced to be
3618                  * fallocated now. If so then setting whole file sparse
3619                  * ie potentially making a few extra pages at the beginning
3620                  * or end of the file non-sparse via set_sparse is harmless.
3621                  */
3622                 if ((off > 8192) || (off + len + 8192 < i_size_read(inode))) {
3623                         rc = -EOPNOTSUPP;
3624                         goto out;
3625                 }
3626         }
3627
3628         smb2_set_sparse(xid, tcon, cfile, inode, false);
3629         rc = 0;
3630
3631 out:
3632         if (rc)
3633                 trace_smb3_falloc_err(xid, cfile->fid.persistent_fid, tcon->tid,
3634                                 tcon->ses->Suid, off, len, rc);
3635         else
3636                 trace_smb3_falloc_done(xid, cfile->fid.persistent_fid, tcon->tid,
3637                                 tcon->ses->Suid, off, len);
3638
3639         free_xid(xid);
3640         return rc;
3641 }
3642
3643 static loff_t smb3_llseek(struct file *file, struct cifs_tcon *tcon, loff_t offset, int whence)
3644 {
3645         struct cifsFileInfo *wrcfile, *cfile = file->private_data;
3646         struct cifsInodeInfo *cifsi;
3647         struct inode *inode;
3648         int rc = 0;
3649         struct file_allocated_range_buffer in_data, *out_data = NULL;
3650         u32 out_data_len;
3651         unsigned int xid;
3652
3653         if (whence != SEEK_HOLE && whence != SEEK_DATA)
3654                 return generic_file_llseek(file, offset, whence);
3655
3656         inode = d_inode(cfile->dentry);
3657         cifsi = CIFS_I(inode);
3658
3659         if (offset < 0 || offset >= i_size_read(inode))
3660                 return -ENXIO;
3661
3662         xid = get_xid();
3663         /*
3664          * We need to be sure that all dirty pages are written as they
3665          * might fill holes on the server.
3666          * Note that we also MUST flush any written pages since at least
3667          * some servers (Windows2016) will not reflect recent writes in
3668          * QUERY_ALLOCATED_RANGES until SMB2_flush is called.
3669          */
3670         wrcfile = find_writable_file(cifsi, FIND_WR_ANY);
3671         if (wrcfile) {
3672                 filemap_write_and_wait(inode->i_mapping);
3673                 smb2_flush_file(xid, tcon, &wrcfile->fid);
3674                 cifsFileInfo_put(wrcfile);
3675         }
3676
3677         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE)) {
3678                 if (whence == SEEK_HOLE)
3679                         offset = i_size_read(inode);
3680                 goto lseek_exit;
3681         }
3682
3683         in_data.file_offset = cpu_to_le64(offset);
3684         in_data.length = cpu_to_le64(i_size_read(inode));
3685
3686         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
3687                         cfile->fid.volatile_fid,
3688                         FSCTL_QUERY_ALLOCATED_RANGES, true,
3689                         (char *)&in_data, sizeof(in_data),
3690                         sizeof(struct file_allocated_range_buffer),
3691                         (char **)&out_data, &out_data_len);
3692         if (rc == -E2BIG)
3693                 rc = 0;
3694         if (rc)
3695                 goto lseek_exit;
3696
3697         if (whence == SEEK_HOLE && out_data_len == 0)
3698                 goto lseek_exit;
3699
3700         if (whence == SEEK_DATA && out_data_len == 0) {
3701                 rc = -ENXIO;
3702                 goto lseek_exit;
3703         }
3704
3705         if (out_data_len < sizeof(struct file_allocated_range_buffer)) {
3706                 rc = -EINVAL;
3707                 goto lseek_exit;
3708         }
3709         if (whence == SEEK_DATA) {
3710                 offset = le64_to_cpu(out_data->file_offset);
3711                 goto lseek_exit;
3712         }
3713         if (offset < le64_to_cpu(out_data->file_offset))
3714                 goto lseek_exit;
3715
3716         offset = le64_to_cpu(out_data->file_offset) + le64_to_cpu(out_data->length);
3717
3718  lseek_exit:
3719         free_xid(xid);
3720         kfree(out_data);
3721         if (!rc)
3722                 return vfs_setpos(file, offset, inode->i_sb->s_maxbytes);
3723         else
3724                 return rc;
3725 }
3726
3727 static int smb3_fiemap(struct cifs_tcon *tcon,
3728                        struct cifsFileInfo *cfile,
3729                        struct fiemap_extent_info *fei, u64 start, u64 len)
3730 {
3731         unsigned int xid;
3732         struct file_allocated_range_buffer in_data, *out_data;
3733         u32 out_data_len;
3734         int i, num, rc, flags, last_blob;
3735         u64 next;
3736
3737         rc = fiemap_prep(d_inode(cfile->dentry), fei, start, &len, 0);
3738         if (rc)
3739                 return rc;
3740
3741         xid = get_xid();
3742  again:
3743         in_data.file_offset = cpu_to_le64(start);
3744         in_data.length = cpu_to_le64(len);
3745
3746         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
3747                         cfile->fid.volatile_fid,
3748                         FSCTL_QUERY_ALLOCATED_RANGES, true,
3749                         (char *)&in_data, sizeof(in_data),
3750                         1024 * sizeof(struct file_allocated_range_buffer),
3751                         (char **)&out_data, &out_data_len);
3752         if (rc == -E2BIG) {
3753                 last_blob = 0;
3754                 rc = 0;
3755         } else
3756                 last_blob = 1;
3757         if (rc)
3758                 goto out;
3759
3760         if (out_data_len && out_data_len < sizeof(struct file_allocated_range_buffer)) {
3761                 rc = -EINVAL;
3762                 goto out;
3763         }
3764         if (out_data_len % sizeof(struct file_allocated_range_buffer)) {
3765                 rc = -EINVAL;
3766                 goto out;
3767         }
3768
3769         num = out_data_len / sizeof(struct file_allocated_range_buffer);
3770         for (i = 0; i < num; i++) {
3771                 flags = 0;
3772                 if (i == num - 1 && last_blob)
3773                         flags |= FIEMAP_EXTENT_LAST;
3774
3775                 rc = fiemap_fill_next_extent(fei,
3776                                 le64_to_cpu(out_data[i].file_offset),
3777                                 le64_to_cpu(out_data[i].file_offset),
3778                                 le64_to_cpu(out_data[i].length),
3779                                 flags);
3780                 if (rc < 0)
3781                         goto out;
3782                 if (rc == 1) {
3783                         rc = 0;
3784                         goto out;
3785                 }
3786         }
3787
3788         if (!last_blob) {
3789                 next = le64_to_cpu(out_data[num - 1].file_offset) +
3790                   le64_to_cpu(out_data[num - 1].length);
3791                 len = len - (next - start);
3792                 start = next;
3793                 goto again;
3794         }
3795
3796  out:
3797         free_xid(xid);
3798         kfree(out_data);
3799         return rc;
3800 }
3801
3802 static long smb3_fallocate(struct file *file, struct cifs_tcon *tcon, int mode,
3803                            loff_t off, loff_t len)
3804 {
3805         /* KEEP_SIZE already checked for by do_fallocate */
3806         if (mode & FALLOC_FL_PUNCH_HOLE)
3807                 return smb3_punch_hole(file, tcon, off, len);
3808         else if (mode & FALLOC_FL_ZERO_RANGE) {
3809                 if (mode & FALLOC_FL_KEEP_SIZE)
3810                         return smb3_zero_range(file, tcon, off, len, true);
3811                 return smb3_zero_range(file, tcon, off, len, false);
3812         } else if (mode == FALLOC_FL_KEEP_SIZE)
3813                 return smb3_simple_falloc(file, tcon, off, len, true);
3814         else if (mode == 0)
3815                 return smb3_simple_falloc(file, tcon, off, len, false);
3816
3817         return -EOPNOTSUPP;
3818 }
3819
3820 static void
3821 smb2_downgrade_oplock(struct TCP_Server_Info *server,
3822                       struct cifsInodeInfo *cinode, __u32 oplock,
3823                       unsigned int epoch, bool *purge_cache)
3824 {
3825         server->ops->set_oplock_level(cinode, oplock, 0, NULL);
3826 }
3827
3828 static void
3829 smb21_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
3830                        unsigned int epoch, bool *purge_cache);
3831
3832 static void
3833 smb3_downgrade_oplock(struct TCP_Server_Info *server,
3834                        struct cifsInodeInfo *cinode, __u32 oplock,
3835                        unsigned int epoch, bool *purge_cache)
3836 {
3837         unsigned int old_state = cinode->oplock;
3838         unsigned int old_epoch = cinode->epoch;
3839         unsigned int new_state;
3840
3841         if (epoch > old_epoch) {
3842                 smb21_set_oplock_level(cinode, oplock, 0, NULL);
3843                 cinode->epoch = epoch;
3844         }
3845
3846         new_state = cinode->oplock;
3847         *purge_cache = false;
3848
3849         if ((old_state & CIFS_CACHE_READ_FLG) != 0 &&
3850             (new_state & CIFS_CACHE_READ_FLG) == 0)
3851                 *purge_cache = true;
3852         else if (old_state == new_state && (epoch - old_epoch > 1))
3853                 *purge_cache = true;
3854 }
3855
3856 static void
3857 smb2_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
3858                       unsigned int epoch, bool *purge_cache)
3859 {
3860         oplock &= 0xFF;
3861         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
3862                 return;
3863         if (oplock == SMB2_OPLOCK_LEVEL_BATCH) {
3864                 cinode->oplock = CIFS_CACHE_RHW_FLG;
3865                 cifs_dbg(FYI, "Batch Oplock granted on inode %p\n",
3866                          &cinode->vfs_inode);
3867         } else if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE) {
3868                 cinode->oplock = CIFS_CACHE_RW_FLG;
3869                 cifs_dbg(FYI, "Exclusive Oplock granted on inode %p\n",
3870                          &cinode->vfs_inode);
3871         } else if (oplock == SMB2_OPLOCK_LEVEL_II) {
3872                 cinode->oplock = CIFS_CACHE_READ_FLG;
3873                 cifs_dbg(FYI, "Level II Oplock granted on inode %p\n",
3874                          &cinode->vfs_inode);
3875         } else
3876                 cinode->oplock = 0;
3877 }
3878
3879 static void
3880 smb21_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
3881                        unsigned int epoch, bool *purge_cache)
3882 {
3883         char message[5] = {0};
3884         unsigned int new_oplock = 0;
3885
3886         oplock &= 0xFF;
3887         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
3888                 return;
3889
3890         /* Check if the server granted an oplock rather than a lease */
3891         if (oplock & SMB2_OPLOCK_LEVEL_EXCLUSIVE)
3892                 return smb2_set_oplock_level(cinode, oplock, epoch,
3893                                              purge_cache);
3894
3895         if (oplock & SMB2_LEASE_READ_CACHING_HE) {
3896                 new_oplock |= CIFS_CACHE_READ_FLG;
3897                 strcat(message, "R");
3898         }
3899         if (oplock & SMB2_LEASE_HANDLE_CACHING_HE) {
3900                 new_oplock |= CIFS_CACHE_HANDLE_FLG;
3901                 strcat(message, "H");
3902         }
3903         if (oplock & SMB2_LEASE_WRITE_CACHING_HE) {
3904                 new_oplock |= CIFS_CACHE_WRITE_FLG;
3905                 strcat(message, "W");
3906         }
3907         if (!new_oplock)
3908                 strncpy(message, "None", sizeof(message));
3909
3910         cinode->oplock = new_oplock;
3911         cifs_dbg(FYI, "%s Lease granted on inode %p\n", message,
3912                  &cinode->vfs_inode);
3913 }
3914
3915 static void
3916 smb3_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
3917                       unsigned int epoch, bool *purge_cache)
3918 {
3919         unsigned int old_oplock = cinode->oplock;
3920
3921         smb21_set_oplock_level(cinode, oplock, epoch, purge_cache);
3922
3923         if (purge_cache) {
3924                 *purge_cache = false;
3925                 if (old_oplock == CIFS_CACHE_READ_FLG) {
3926                         if (cinode->oplock == CIFS_CACHE_READ_FLG &&
3927                             (epoch - cinode->epoch > 0))
3928                                 *purge_cache = true;
3929                         else if (cinode->oplock == CIFS_CACHE_RH_FLG &&
3930                                  (epoch - cinode->epoch > 1))
3931                                 *purge_cache = true;
3932                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
3933                                  (epoch - cinode->epoch > 1))
3934                                 *purge_cache = true;
3935                         else if (cinode->oplock == 0 &&
3936                                  (epoch - cinode->epoch > 0))
3937                                 *purge_cache = true;
3938                 } else if (old_oplock == CIFS_CACHE_RH_FLG) {
3939                         if (cinode->oplock == CIFS_CACHE_RH_FLG &&
3940                             (epoch - cinode->epoch > 0))
3941                                 *purge_cache = true;
3942                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
3943                                  (epoch - cinode->epoch > 1))
3944                                 *purge_cache = true;
3945                 }
3946                 cinode->epoch = epoch;
3947         }
3948 }
3949
3950 static bool
3951 smb2_is_read_op(__u32 oplock)
3952 {
3953         return oplock == SMB2_OPLOCK_LEVEL_II;
3954 }
3955
3956 static bool
3957 smb21_is_read_op(__u32 oplock)
3958 {
3959         return (oplock & SMB2_LEASE_READ_CACHING_HE) &&
3960                !(oplock & SMB2_LEASE_WRITE_CACHING_HE);
3961 }
3962
3963 static __le32
3964 map_oplock_to_lease(u8 oplock)
3965 {
3966         if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE)
3967                 return SMB2_LEASE_WRITE_CACHING | SMB2_LEASE_READ_CACHING;
3968         else if (oplock == SMB2_OPLOCK_LEVEL_II)
3969                 return SMB2_LEASE_READ_CACHING;
3970         else if (oplock == SMB2_OPLOCK_LEVEL_BATCH)
3971                 return SMB2_LEASE_HANDLE_CACHING | SMB2_LEASE_READ_CACHING |
3972                        SMB2_LEASE_WRITE_CACHING;
3973         return 0;
3974 }
3975
3976 static char *
3977 smb2_create_lease_buf(u8 *lease_key, u8 oplock)
3978 {
3979         struct create_lease *buf;
3980
3981         buf = kzalloc(sizeof(struct create_lease), GFP_KERNEL);
3982         if (!buf)
3983                 return NULL;
3984
3985         memcpy(&buf->lcontext.LeaseKey, lease_key, SMB2_LEASE_KEY_SIZE);
3986         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
3987
3988         buf->ccontext.DataOffset = cpu_to_le16(offsetof
3989                                         (struct create_lease, lcontext));
3990         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context));
3991         buf->ccontext.NameOffset = cpu_to_le16(offsetof
3992                                 (struct create_lease, Name));
3993         buf->ccontext.NameLength = cpu_to_le16(4);
3994         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
3995         buf->Name[0] = 'R';
3996         buf->Name[1] = 'q';
3997         buf->Name[2] = 'L';
3998         buf->Name[3] = 's';
3999         return (char *)buf;
4000 }
4001
4002 static char *
4003 smb3_create_lease_buf(u8 *lease_key, u8 oplock)
4004 {
4005         struct create_lease_v2 *buf;
4006
4007         buf = kzalloc(sizeof(struct create_lease_v2), GFP_KERNEL);
4008         if (!buf)
4009                 return NULL;
4010
4011         memcpy(&buf->lcontext.LeaseKey, lease_key, SMB2_LEASE_KEY_SIZE);
4012         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
4013
4014         buf->ccontext.DataOffset = cpu_to_le16(offsetof
4015                                         (struct create_lease_v2, lcontext));
4016         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2));
4017         buf->ccontext.NameOffset = cpu_to_le16(offsetof
4018                                 (struct create_lease_v2, Name));
4019         buf->ccontext.NameLength = cpu_to_le16(4);
4020         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
4021         buf->Name[0] = 'R';
4022         buf->Name[1] = 'q';
4023         buf->Name[2] = 'L';
4024         buf->Name[3] = 's';
4025         return (char *)buf;
4026 }
4027
4028 static __u8
4029 smb2_parse_lease_buf(void *buf, unsigned int *epoch, char *lease_key)
4030 {
4031         struct create_lease *lc = (struct create_lease *)buf;
4032
4033         *epoch = 0; /* not used */
4034         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
4035                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
4036         return le32_to_cpu(lc->lcontext.LeaseState);
4037 }
4038
4039 static __u8
4040 smb3_parse_lease_buf(void *buf, unsigned int *epoch, char *lease_key)
4041 {
4042         struct create_lease_v2 *lc = (struct create_lease_v2 *)buf;
4043
4044         *epoch = le16_to_cpu(lc->lcontext.Epoch);
4045         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
4046                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
4047         if (lease_key)
4048                 memcpy(lease_key, &lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE);
4049         return le32_to_cpu(lc->lcontext.LeaseState);
4050 }
4051
4052 static unsigned int
4053 smb2_wp_retry_size(struct inode *inode)
4054 {
4055         return min_t(unsigned int, CIFS_SB(inode->i_sb)->ctx->wsize,
4056                      SMB2_MAX_BUFFER_SIZE);
4057 }
4058
4059 static bool
4060 smb2_dir_needs_close(struct cifsFileInfo *cfile)
4061 {
4062         return !cfile->invalidHandle;
4063 }
4064
4065 static void
4066 fill_transform_hdr(struct smb2_transform_hdr *tr_hdr, unsigned int orig_len,
4067                    struct smb_rqst *old_rq, __le16 cipher_type)
4068 {
4069         struct smb2_sync_hdr *shdr =
4070                         (struct smb2_sync_hdr *)old_rq->rq_iov[0].iov_base;
4071
4072         memset(tr_hdr, 0, sizeof(struct smb2_transform_hdr));
4073         tr_hdr->ProtocolId = SMB2_TRANSFORM_PROTO_NUM;
4074         tr_hdr->OriginalMessageSize = cpu_to_le32(orig_len);
4075         tr_hdr->Flags = cpu_to_le16(0x01);
4076         if ((cipher_type == SMB2_ENCRYPTION_AES128_GCM) ||
4077             (cipher_type == SMB2_ENCRYPTION_AES256_GCM))
4078                 get_random_bytes(&tr_hdr->Nonce, SMB3_AES_GCM_NONCE);
4079         else
4080                 get_random_bytes(&tr_hdr->Nonce, SMB3_AES_CCM_NONCE);
4081         memcpy(&tr_hdr->SessionId, &shdr->SessionId, 8);
4082 }
4083
4084 /* We can not use the normal sg_set_buf() as we will sometimes pass a
4085  * stack object as buf.
4086  */
4087 static inline void smb2_sg_set_buf(struct scatterlist *sg, const void *buf,
4088                                    unsigned int buflen)
4089 {
4090         void *addr;
4091         /*
4092          * VMAP_STACK (at least) puts stack into the vmalloc address space
4093          */
4094         if (is_vmalloc_addr(buf))
4095                 addr = vmalloc_to_page(buf);
4096         else
4097                 addr = virt_to_page(buf);
4098         sg_set_page(sg, addr, buflen, offset_in_page(buf));
4099 }
4100
4101 /* Assumes the first rqst has a transform header as the first iov.
4102  * I.e.
4103  * rqst[0].rq_iov[0]  is transform header
4104  * rqst[0].rq_iov[1+] data to be encrypted/decrypted
4105  * rqst[1+].rq_iov[0+] data to be encrypted/decrypted
4106  */
4107 static struct scatterlist *
4108 init_sg(int num_rqst, struct smb_rqst *rqst, u8 *sign)
4109 {
4110         unsigned int sg_len;
4111         struct scatterlist *sg;
4112         unsigned int i;
4113         unsigned int j;
4114         unsigned int idx = 0;
4115         int skip;
4116
4117         sg_len = 1;
4118         for (i = 0; i < num_rqst; i++)
4119                 sg_len += rqst[i].rq_nvec + rqst[i].rq_npages;
4120
4121         sg = kmalloc_array(sg_len, sizeof(struct scatterlist), GFP_KERNEL);
4122         if (!sg)
4123                 return NULL;
4124
4125         sg_init_table(sg, sg_len);
4126         for (i = 0; i < num_rqst; i++) {
4127                 for (j = 0; j < rqst[i].rq_nvec; j++) {
4128                         /*
4129                          * The first rqst has a transform header where the
4130                          * first 20 bytes are not part of the encrypted blob
4131                          */
4132                         skip = (i == 0) && (j == 0) ? 20 : 0;
4133                         smb2_sg_set_buf(&sg[idx++],
4134                                         rqst[i].rq_iov[j].iov_base + skip,
4135                                         rqst[i].rq_iov[j].iov_len - skip);
4136                         }
4137
4138                 for (j = 0; j < rqst[i].rq_npages; j++) {
4139                         unsigned int len, offset;
4140
4141                         rqst_page_get_length(&rqst[i], j, &len, &offset);
4142                         sg_set_page(&sg[idx++], rqst[i].rq_pages[j], len, offset);
4143                 }
4144         }
4145         smb2_sg_set_buf(&sg[idx], sign, SMB2_SIGNATURE_SIZE);
4146         return sg;
4147 }
4148
4149 static int
4150 smb2_get_enc_key(struct TCP_Server_Info *server, __u64 ses_id, int enc, u8 *key)
4151 {
4152         struct cifs_ses *ses;
4153         u8 *ses_enc_key;
4154
4155         spin_lock(&cifs_tcp_ses_lock);
4156         list_for_each_entry(server, &cifs_tcp_ses_list, tcp_ses_list) {
4157                 list_for_each_entry(ses, &server->smb_ses_list, smb_ses_list) {
4158                         if (ses->Suid == ses_id) {
4159                                 ses_enc_key = enc ? ses->smb3encryptionkey :
4160                                         ses->smb3decryptionkey;
4161                                 memcpy(key, ses_enc_key, SMB3_SIGN_KEY_SIZE);
4162                                 spin_unlock(&cifs_tcp_ses_lock);
4163                                 return 0;
4164                         }
4165                 }
4166         }
4167         spin_unlock(&cifs_tcp_ses_lock);
4168
4169         return 1;
4170 }
4171 /*
4172  * Encrypt or decrypt @rqst message. @rqst[0] has the following format:
4173  * iov[0]   - transform header (associate data),
4174  * iov[1-N] - SMB2 header and pages - data to encrypt.
4175  * On success return encrypted data in iov[1-N] and pages, leave iov[0]
4176  * untouched.
4177  */
4178 static int
4179 crypt_message(struct TCP_Server_Info *server, int num_rqst,
4180               struct smb_rqst *rqst, int enc)
4181 {
4182         struct smb2_transform_hdr *tr_hdr =
4183                 (struct smb2_transform_hdr *)rqst[0].rq_iov[0].iov_base;
4184         unsigned int assoc_data_len = sizeof(struct smb2_transform_hdr) - 20;
4185         int rc = 0;
4186         struct scatterlist *sg;
4187         u8 sign[SMB2_SIGNATURE_SIZE] = {};
4188         u8 key[SMB3_SIGN_KEY_SIZE];
4189         struct aead_request *req;
4190         char *iv;
4191         unsigned int iv_len;
4192         DECLARE_CRYPTO_WAIT(wait);
4193         struct crypto_aead *tfm;
4194         unsigned int crypt_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
4195
4196         rc = smb2_get_enc_key(server, tr_hdr->SessionId, enc, key);
4197         if (rc) {
4198                 cifs_server_dbg(VFS, "%s: Could not get %scryption key\n", __func__,
4199                          enc ? "en" : "de");
4200                 return rc;
4201         }
4202
4203         rc = smb3_crypto_aead_allocate(server);
4204         if (rc) {
4205                 cifs_server_dbg(VFS, "%s: crypto alloc failed\n", __func__);
4206                 return rc;
4207         }
4208
4209         tfm = enc ? server->secmech.ccmaesencrypt :
4210                                                 server->secmech.ccmaesdecrypt;
4211
4212         if (server->cipher_type == SMB2_ENCRYPTION_AES256_GCM)
4213                 rc = crypto_aead_setkey(tfm, key, SMB3_GCM256_CRYPTKEY_SIZE);
4214         else
4215                 rc = crypto_aead_setkey(tfm, key, SMB3_SIGN_KEY_SIZE);
4216
4217         if (rc) {
4218                 cifs_server_dbg(VFS, "%s: Failed to set aead key %d\n", __func__, rc);
4219                 return rc;
4220         }
4221
4222         rc = crypto_aead_setauthsize(tfm, SMB2_SIGNATURE_SIZE);
4223         if (rc) {
4224                 cifs_server_dbg(VFS, "%s: Failed to set authsize %d\n", __func__, rc);
4225                 return rc;
4226         }
4227
4228         req = aead_request_alloc(tfm, GFP_KERNEL);
4229         if (!req) {
4230                 cifs_server_dbg(VFS, "%s: Failed to alloc aead request\n", __func__);
4231                 return -ENOMEM;
4232         }
4233
4234         if (!enc) {
4235                 memcpy(sign, &tr_hdr->Signature, SMB2_SIGNATURE_SIZE);
4236                 crypt_len += SMB2_SIGNATURE_SIZE;
4237         }
4238
4239         sg = init_sg(num_rqst, rqst, sign);
4240         if (!sg) {
4241                 cifs_server_dbg(VFS, "%s: Failed to init sg\n", __func__);
4242                 rc = -ENOMEM;
4243                 goto free_req;
4244         }
4245
4246         iv_len = crypto_aead_ivsize(tfm);
4247         iv = kzalloc(iv_len, GFP_KERNEL);
4248         if (!iv) {
4249                 cifs_server_dbg(VFS, "%s: Failed to alloc iv\n", __func__);
4250                 rc = -ENOMEM;
4251                 goto free_sg;
4252         }
4253
4254         if ((server->cipher_type == SMB2_ENCRYPTION_AES128_GCM) ||
4255             (server->cipher_type == SMB2_ENCRYPTION_AES256_GCM))
4256                 memcpy(iv, (char *)tr_hdr->Nonce, SMB3_AES_GCM_NONCE);
4257         else {
4258                 iv[0] = 3;
4259                 memcpy(iv + 1, (char *)tr_hdr->Nonce, SMB3_AES_CCM_NONCE);
4260         }
4261
4262         aead_request_set_crypt(req, sg, sg, crypt_len, iv);
4263         aead_request_set_ad(req, assoc_data_len);
4264
4265         aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
4266                                   crypto_req_done, &wait);
4267
4268         rc = crypto_wait_req(enc ? crypto_aead_encrypt(req)
4269                                 : crypto_aead_decrypt(req), &wait);
4270
4271         if (!rc && enc)
4272                 memcpy(&tr_hdr->Signature, sign, SMB2_SIGNATURE_SIZE);
4273
4274         kfree(iv);
4275 free_sg:
4276         kfree(sg);
4277 free_req:
4278         kfree(req);
4279         return rc;
4280 }
4281
4282 void
4283 smb3_free_compound_rqst(int num_rqst, struct smb_rqst *rqst)
4284 {
4285         int i, j;
4286
4287         for (i = 0; i < num_rqst; i++) {
4288                 if (rqst[i].rq_pages) {
4289                         for (j = rqst[i].rq_npages - 1; j >= 0; j--)
4290                                 put_page(rqst[i].rq_pages[j]);
4291                         kfree(rqst[i].rq_pages);
4292                 }
4293         }
4294 }
4295
4296 /*
4297  * This function will initialize new_rq and encrypt the content.
4298  * The first entry, new_rq[0], only contains a single iov which contains
4299  * a smb2_transform_hdr and is pre-allocated by the caller.
4300  * This function then populates new_rq[1+] with the content from olq_rq[0+].
4301  *
4302  * The end result is an array of smb_rqst structures where the first structure
4303  * only contains a single iov for the transform header which we then can pass
4304  * to crypt_message().
4305  *
4306  * new_rq[0].rq_iov[0] :  smb2_transform_hdr pre-allocated by the caller
4307  * new_rq[1+].rq_iov[*] == old_rq[0+].rq_iov[*] : SMB2/3 requests
4308  */
4309 static int
4310 smb3_init_transform_rq(struct TCP_Server_Info *server, int num_rqst,
4311                        struct smb_rqst *new_rq, struct smb_rqst *old_rq)
4312 {
4313         struct page **pages;
4314         struct smb2_transform_hdr *tr_hdr = new_rq[0].rq_iov[0].iov_base;
4315         unsigned int npages;
4316         unsigned int orig_len = 0;
4317         int i, j;
4318         int rc = -ENOMEM;
4319
4320         for (i = 1; i < num_rqst; i++) {
4321                 npages = old_rq[i - 1].rq_npages;
4322                 pages = kmalloc_array(npages, sizeof(struct page *),
4323                                       GFP_KERNEL);
4324                 if (!pages)
4325                         goto err_free;
4326
4327                 new_rq[i].rq_pages = pages;
4328                 new_rq[i].rq_npages = npages;
4329                 new_rq[i].rq_offset = old_rq[i - 1].rq_offset;
4330                 new_rq[i].rq_pagesz = old_rq[i - 1].rq_pagesz;
4331                 new_rq[i].rq_tailsz = old_rq[i - 1].rq_tailsz;
4332                 new_rq[i].rq_iov = old_rq[i - 1].rq_iov;
4333                 new_rq[i].rq_nvec = old_rq[i - 1].rq_nvec;
4334
4335                 orig_len += smb_rqst_len(server, &old_rq[i - 1]);
4336
4337                 for (j = 0; j < npages; j++) {
4338                         pages[j] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
4339                         if (!pages[j])
4340                                 goto err_free;
4341                 }
4342
4343                 /* copy pages form the old */
4344                 for (j = 0; j < npages; j++) {
4345                         char *dst, *src;
4346                         unsigned int offset, len;
4347
4348                         rqst_page_get_length(&new_rq[i], j, &len, &offset);
4349
4350                         dst = (char *) kmap(new_rq[i].rq_pages[j]) + offset;
4351                         src = (char *) kmap(old_rq[i - 1].rq_pages[j]) + offset;
4352
4353                         memcpy(dst, src, len);
4354                         kunmap(new_rq[i].rq_pages[j]);
4355                         kunmap(old_rq[i - 1].rq_pages[j]);
4356                 }
4357         }
4358
4359         /* fill the 1st iov with a transform header */
4360         fill_transform_hdr(tr_hdr, orig_len, old_rq, server->cipher_type);
4361
4362         rc = crypt_message(server, num_rqst, new_rq, 1);
4363         cifs_dbg(FYI, "Encrypt message returned %d\n", rc);
4364         if (rc)
4365                 goto err_free;
4366
4367         return rc;
4368
4369 err_free:
4370         smb3_free_compound_rqst(num_rqst - 1, &new_rq[1]);
4371         return rc;
4372 }
4373
4374 static int
4375 smb3_is_transform_hdr(void *buf)
4376 {
4377         struct smb2_transform_hdr *trhdr = buf;
4378
4379         return trhdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM;
4380 }
4381
4382 static int
4383 decrypt_raw_data(struct TCP_Server_Info *server, char *buf,
4384                  unsigned int buf_data_size, struct page **pages,
4385                  unsigned int npages, unsigned int page_data_size,
4386                  bool is_offloaded)
4387 {
4388         struct kvec iov[2];
4389         struct smb_rqst rqst = {NULL};
4390         int rc;
4391
4392         iov[0].iov_base = buf;
4393         iov[0].iov_len = sizeof(struct smb2_transform_hdr);
4394         iov[1].iov_base = buf + sizeof(struct smb2_transform_hdr);
4395         iov[1].iov_len = buf_data_size;
4396
4397         rqst.rq_iov = iov;
4398         rqst.rq_nvec = 2;
4399         rqst.rq_pages = pages;
4400         rqst.rq_npages = npages;
4401         rqst.rq_pagesz = PAGE_SIZE;
4402         rqst.rq_tailsz = (page_data_size % PAGE_SIZE) ? : PAGE_SIZE;
4403
4404         rc = crypt_message(server, 1, &rqst, 0);
4405         cifs_dbg(FYI, "Decrypt message returned %d\n", rc);
4406
4407         if (rc)
4408                 return rc;
4409
4410         memmove(buf, iov[1].iov_base, buf_data_size);
4411
4412         if (!is_offloaded)
4413                 server->total_read = buf_data_size + page_data_size;
4414
4415         return rc;
4416 }
4417
4418 static int
4419 read_data_into_pages(struct TCP_Server_Info *server, struct page **pages,
4420                      unsigned int npages, unsigned int len)
4421 {
4422         int i;
4423         int length;
4424
4425         for (i = 0; i < npages; i++) {
4426                 struct page *page = pages[i];
4427                 size_t n;
4428
4429                 n = len;
4430                 if (len >= PAGE_SIZE) {
4431                         /* enough data to fill the page */
4432                         n = PAGE_SIZE;
4433                         len -= n;
4434                 } else {
4435                         zero_user(page, len, PAGE_SIZE - len);
4436                         len = 0;
4437                 }
4438                 length = cifs_read_page_from_socket(server, page, 0, n);
4439                 if (length < 0)
4440                         return length;
4441                 server->total_read += length;
4442         }
4443
4444         return 0;
4445 }
4446
4447 static int
4448 init_read_bvec(struct page **pages, unsigned int npages, unsigned int data_size,
4449                unsigned int cur_off, struct bio_vec **page_vec)
4450 {
4451         struct bio_vec *bvec;
4452         int i;
4453
4454         bvec = kcalloc(npages, sizeof(struct bio_vec), GFP_KERNEL);
4455         if (!bvec)
4456                 return -ENOMEM;
4457
4458         for (i = 0; i < npages; i++) {
4459                 bvec[i].bv_page = pages[i];
4460                 bvec[i].bv_offset = (i == 0) ? cur_off : 0;
4461                 bvec[i].bv_len = min_t(unsigned int, PAGE_SIZE, data_size);
4462                 data_size -= bvec[i].bv_len;
4463         }
4464
4465         if (data_size != 0) {
4466                 cifs_dbg(VFS, "%s: something went wrong\n", __func__);
4467                 kfree(bvec);
4468                 return -EIO;
4469         }
4470
4471         *page_vec = bvec;
4472         return 0;
4473 }
4474
4475 static int
4476 handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid,
4477                  char *buf, unsigned int buf_len, struct page **pages,
4478                  unsigned int npages, unsigned int page_data_size,
4479                  bool is_offloaded)
4480 {
4481         unsigned int data_offset;
4482         unsigned int data_len;
4483         unsigned int cur_off;
4484         unsigned int cur_page_idx;
4485         unsigned int pad_len;
4486         struct cifs_readdata *rdata = mid->callback_data;
4487         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
4488         struct bio_vec *bvec = NULL;
4489         struct iov_iter iter;
4490         struct kvec iov;
4491         int length;
4492         bool use_rdma_mr = false;
4493
4494         if (shdr->Command != SMB2_READ) {
4495                 cifs_server_dbg(VFS, "only big read responses are supported\n");
4496                 return -ENOTSUPP;
4497         }
4498
4499         if (server->ops->is_session_expired &&
4500             server->ops->is_session_expired(buf)) {
4501                 if (!is_offloaded)
4502                         cifs_reconnect(server);
4503                 return -1;
4504         }
4505
4506         if (server->ops->is_status_pending &&
4507                         server->ops->is_status_pending(buf, server))
4508                 return -1;
4509
4510         /* set up first two iov to get credits */
4511         rdata->iov[0].iov_base = buf;
4512         rdata->iov[0].iov_len = 0;
4513         rdata->iov[1].iov_base = buf;
4514         rdata->iov[1].iov_len =
4515                 min_t(unsigned int, buf_len, server->vals->read_rsp_size);
4516         cifs_dbg(FYI, "0: iov_base=%p iov_len=%zu\n",
4517                  rdata->iov[0].iov_base, rdata->iov[0].iov_len);
4518         cifs_dbg(FYI, "1: iov_base=%p iov_len=%zu\n",
4519                  rdata->iov[1].iov_base, rdata->iov[1].iov_len);
4520
4521         rdata->result = server->ops->map_error(buf, true);
4522         if (rdata->result != 0) {
4523                 cifs_dbg(FYI, "%s: server returned error %d\n",
4524                          __func__, rdata->result);
4525                 /* normal error on read response */
4526                 if (is_offloaded)
4527                         mid->mid_state = MID_RESPONSE_RECEIVED;
4528                 else
4529                         dequeue_mid(mid, false);
4530                 return 0;
4531         }
4532
4533         data_offset = server->ops->read_data_offset(buf);
4534 #ifdef CONFIG_CIFS_SMB_DIRECT
4535         use_rdma_mr = rdata->mr;
4536 #endif
4537         data_len = server->ops->read_data_length(buf, use_rdma_mr);
4538
4539         if (data_offset < server->vals->read_rsp_size) {
4540                 /*
4541                  * win2k8 sometimes sends an offset of 0 when the read
4542                  * is beyond the EOF. Treat it as if the data starts just after
4543                  * the header.
4544                  */
4545                 cifs_dbg(FYI, "%s: data offset (%u) inside read response header\n",
4546                          __func__, data_offset);
4547                 data_offset = server->vals->read_rsp_size;
4548         } else if (data_offset > MAX_CIFS_SMALL_BUFFER_SIZE) {
4549                 /* data_offset is beyond the end of smallbuf */
4550                 cifs_dbg(FYI, "%s: data offset (%u) beyond end of smallbuf\n",
4551                          __func__, data_offset);
4552                 rdata->result = -EIO;
4553                 if (is_offloaded)
4554                         mid->mid_state = MID_RESPONSE_MALFORMED;
4555                 else
4556                         dequeue_mid(mid, rdata->result);
4557                 return 0;
4558         }
4559
4560         pad_len = data_offset - server->vals->read_rsp_size;
4561
4562         if (buf_len <= data_offset) {
4563                 /* read response payload is in pages */
4564                 cur_page_idx = pad_len / PAGE_SIZE;
4565                 cur_off = pad_len % PAGE_SIZE;
4566
4567                 if (cur_page_idx != 0) {
4568                         /* data offset is beyond the 1st page of response */
4569                         cifs_dbg(FYI, "%s: data offset (%u) beyond 1st page of response\n",
4570                                  __func__, data_offset);
4571                         rdata->result = -EIO;
4572                         if (is_offloaded)
4573                                 mid->mid_state = MID_RESPONSE_MALFORMED;
4574                         else
4575                                 dequeue_mid(mid, rdata->result);
4576                         return 0;
4577                 }
4578
4579                 if (data_len > page_data_size - pad_len) {
4580                         /* data_len is corrupt -- discard frame */
4581                         rdata->result = -EIO;
4582                         if (is_offloaded)
4583                                 mid->mid_state = MID_RESPONSE_MALFORMED;
4584                         else
4585                                 dequeue_mid(mid, rdata->result);
4586                         return 0;
4587                 }
4588
4589                 rdata->result = init_read_bvec(pages, npages, page_data_size,
4590                                                cur_off, &bvec);
4591                 if (rdata->result != 0) {
4592                         if (is_offloaded)
4593                                 mid->mid_state = MID_RESPONSE_MALFORMED;
4594                         else
4595                                 dequeue_mid(mid, rdata->result);
4596                         return 0;
4597                 }
4598
4599                 iov_iter_bvec(&iter, WRITE, bvec, npages, data_len);
4600         } else if (buf_len >= data_offset + data_len) {
4601                 /* read response payload is in buf */
4602                 WARN_ONCE(npages > 0, "read data can be either in buf or in pages");
4603                 iov.iov_base = buf + data_offset;
4604                 iov.iov_len = data_len;
4605                 iov_iter_kvec(&iter, WRITE, &iov, 1, data_len);
4606         } else {
4607                 /* read response payload cannot be in both buf and pages */
4608                 WARN_ONCE(1, "buf can not contain only a part of read data");
4609                 rdata->result = -EIO;
4610                 if (is_offloaded)
4611                         mid->mid_state = MID_RESPONSE_MALFORMED;
4612                 else
4613                         dequeue_mid(mid, rdata->result);
4614                 return 0;
4615         }
4616
4617         length = rdata->copy_into_pages(server, rdata, &iter);
4618
4619         kfree(bvec);
4620
4621         if (length < 0)
4622                 return length;
4623
4624         if (is_offloaded)
4625                 mid->mid_state = MID_RESPONSE_RECEIVED;
4626         else
4627                 dequeue_mid(mid, false);
4628         return length;
4629 }
4630
4631 struct smb2_decrypt_work {
4632         struct work_struct decrypt;
4633         struct TCP_Server_Info *server;
4634         struct page **ppages;
4635         char *buf;
4636         unsigned int npages;
4637         unsigned int len;
4638 };
4639
4640
4641 static void smb2_decrypt_offload(struct work_struct *work)
4642 {
4643         struct smb2_decrypt_work *dw = container_of(work,
4644                                 struct smb2_decrypt_work, decrypt);
4645         int i, rc;
4646         struct mid_q_entry *mid;
4647
4648         rc = decrypt_raw_data(dw->server, dw->buf, dw->server->vals->read_rsp_size,
4649                               dw->ppages, dw->npages, dw->len, true);
4650         if (rc) {
4651                 cifs_dbg(VFS, "error decrypting rc=%d\n", rc);
4652                 goto free_pages;
4653         }
4654
4655         dw->server->lstrp = jiffies;
4656         mid = smb2_find_dequeue_mid(dw->server, dw->buf);
4657         if (mid == NULL)
4658                 cifs_dbg(FYI, "mid not found\n");
4659         else {
4660                 mid->decrypted = true;
4661                 rc = handle_read_data(dw->server, mid, dw->buf,
4662                                       dw->server->vals->read_rsp_size,
4663                                       dw->ppages, dw->npages, dw->len,
4664                                       true);
4665                 if (rc >= 0) {
4666 #ifdef CONFIG_CIFS_STATS2
4667                         mid->when_received = jiffies;
4668 #endif
4669                         if (dw->server->ops->is_network_name_deleted)
4670                                 dw->server->ops->is_network_name_deleted(dw->buf,
4671                                                                          dw->server);
4672
4673                         mid->callback(mid);
4674                 } else {
4675                         spin_lock(&GlobalMid_Lock);
4676                         if (dw->server->tcpStatus == CifsNeedReconnect) {
4677                                 mid->mid_state = MID_RETRY_NEEDED;
4678                                 spin_unlock(&GlobalMid_Lock);
4679                                 mid->callback(mid);
4680                         } else {
4681                                 mid->mid_state = MID_REQUEST_SUBMITTED;
4682                                 mid->mid_flags &= ~(MID_DELETED);
4683                                 list_add_tail(&mid->qhead,
4684                                         &dw->server->pending_mid_q);
4685                                 spin_unlock(&GlobalMid_Lock);
4686                         }
4687                 }
4688                 cifs_mid_q_entry_release(mid);
4689         }
4690
4691 free_pages:
4692         for (i = dw->npages-1; i >= 0; i--)
4693                 put_page(dw->ppages[i]);
4694
4695         kfree(dw->ppages);
4696         cifs_small_buf_release(dw->buf);
4697         kfree(dw);
4698 }
4699
4700
4701 static int
4702 receive_encrypted_read(struct TCP_Server_Info *server, struct mid_q_entry **mid,
4703                        int *num_mids)
4704 {
4705         char *buf = server->smallbuf;
4706         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
4707         unsigned int npages;
4708         struct page **pages;
4709         unsigned int len;
4710         unsigned int buflen = server->pdu_size;
4711         int rc;
4712         int i = 0;
4713         struct smb2_decrypt_work *dw;
4714
4715         *num_mids = 1;
4716         len = min_t(unsigned int, buflen, server->vals->read_rsp_size +
4717                 sizeof(struct smb2_transform_hdr)) - HEADER_SIZE(server) + 1;
4718
4719         rc = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1, len);
4720         if (rc < 0)
4721                 return rc;
4722         server->total_read += rc;
4723
4724         len = le32_to_cpu(tr_hdr->OriginalMessageSize) -
4725                 server->vals->read_rsp_size;
4726         npages = DIV_ROUND_UP(len, PAGE_SIZE);
4727
4728         pages = kmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
4729         if (!pages) {
4730                 rc = -ENOMEM;
4731                 goto discard_data;
4732         }
4733
4734         for (; i < npages; i++) {
4735                 pages[i] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
4736                 if (!pages[i]) {
4737                         rc = -ENOMEM;
4738                         goto discard_data;
4739                 }
4740         }
4741
4742         /* read read data into pages */
4743         rc = read_data_into_pages(server, pages, npages, len);
4744         if (rc)
4745                 goto free_pages;
4746
4747         rc = cifs_discard_remaining_data(server);
4748         if (rc)
4749                 goto free_pages;
4750
4751         /*
4752          * For large reads, offload to different thread for better performance,
4753          * use more cores decrypting which can be expensive
4754          */
4755
4756         if ((server->min_offload) && (server->in_flight > 1) &&
4757             (server->pdu_size >= server->min_offload)) {
4758                 dw = kmalloc(sizeof(struct smb2_decrypt_work), GFP_KERNEL);
4759                 if (dw == NULL)
4760                         goto non_offloaded_decrypt;
4761
4762                 dw->buf = server->smallbuf;
4763                 server->smallbuf = (char *)cifs_small_buf_get();
4764
4765                 INIT_WORK(&dw->decrypt, smb2_decrypt_offload);
4766
4767                 dw->npages = npages;
4768                 dw->server = server;
4769                 dw->ppages = pages;
4770                 dw->len = len;
4771                 queue_work(decrypt_wq, &dw->decrypt);
4772                 *num_mids = 0; /* worker thread takes care of finding mid */
4773                 return -1;
4774         }
4775
4776 non_offloaded_decrypt:
4777         rc = decrypt_raw_data(server, buf, server->vals->read_rsp_size,
4778                               pages, npages, len, false);
4779         if (rc)
4780                 goto free_pages;
4781
4782         *mid = smb2_find_mid(server, buf);
4783         if (*mid == NULL)
4784                 cifs_dbg(FYI, "mid not found\n");
4785         else {
4786                 cifs_dbg(FYI, "mid found\n");
4787                 (*mid)->decrypted = true;
4788                 rc = handle_read_data(server, *mid, buf,
4789                                       server->vals->read_rsp_size,
4790                                       pages, npages, len, false);
4791                 if (rc >= 0) {
4792                         if (server->ops->is_network_name_deleted) {
4793                                 server->ops->is_network_name_deleted(buf,
4794                                                                 server);
4795                         }
4796                 }
4797         }
4798
4799 free_pages:
4800         for (i = i - 1; i >= 0; i--)
4801                 put_page(pages[i]);
4802         kfree(pages);
4803         return rc;
4804 discard_data:
4805         cifs_discard_remaining_data(server);
4806         goto free_pages;
4807 }
4808
4809 static int
4810 receive_encrypted_standard(struct TCP_Server_Info *server,
4811                            struct mid_q_entry **mids, char **bufs,
4812                            int *num_mids)
4813 {
4814         int ret, length;
4815         char *buf = server->smallbuf;
4816         struct smb2_sync_hdr *shdr;
4817         unsigned int pdu_length = server->pdu_size;
4818         unsigned int buf_size;
4819         struct mid_q_entry *mid_entry;
4820         int next_is_large;
4821         char *next_buffer = NULL;
4822
4823         *num_mids = 0;
4824
4825         /* switch to large buffer if too big for a small one */
4826         if (pdu_length > MAX_CIFS_SMALL_BUFFER_SIZE) {
4827                 server->large_buf = true;
4828                 memcpy(server->bigbuf, buf, server->total_read);
4829                 buf = server->bigbuf;
4830         }
4831
4832         /* now read the rest */
4833         length = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1,
4834                                 pdu_length - HEADER_SIZE(server) + 1);
4835         if (length < 0)
4836                 return length;
4837         server->total_read += length;
4838
4839         buf_size = pdu_length - sizeof(struct smb2_transform_hdr);
4840         length = decrypt_raw_data(server, buf, buf_size, NULL, 0, 0, false);
4841         if (length)
4842                 return length;
4843
4844         next_is_large = server->large_buf;
4845 one_more:
4846         shdr = (struct smb2_sync_hdr *)buf;
4847         if (shdr->NextCommand) {
4848                 if (next_is_large)
4849                         next_buffer = (char *)cifs_buf_get();
4850                 else
4851                         next_buffer = (char *)cifs_small_buf_get();
4852                 memcpy(next_buffer,
4853                        buf + le32_to_cpu(shdr->NextCommand),
4854                        pdu_length - le32_to_cpu(shdr->NextCommand));
4855         }
4856
4857         mid_entry = smb2_find_mid(server, buf);
4858         if (mid_entry == NULL)
4859                 cifs_dbg(FYI, "mid not found\n");
4860         else {
4861                 cifs_dbg(FYI, "mid found\n");
4862                 mid_entry->decrypted = true;
4863                 mid_entry->resp_buf_size = server->pdu_size;
4864         }
4865
4866         if (*num_mids >= MAX_COMPOUND) {
4867                 cifs_server_dbg(VFS, "too many PDUs in compound\n");
4868                 return -1;
4869         }
4870         bufs[*num_mids] = buf;
4871         mids[(*num_mids)++] = mid_entry;
4872
4873         if (mid_entry && mid_entry->handle)
4874                 ret = mid_entry->handle(server, mid_entry);
4875         else
4876                 ret = cifs_handle_standard(server, mid_entry);
4877
4878         if (ret == 0 && shdr->NextCommand) {
4879                 pdu_length -= le32_to_cpu(shdr->NextCommand);
4880                 server->large_buf = next_is_large;
4881                 if (next_is_large)
4882                         server->bigbuf = buf = next_buffer;
4883                 else
4884                         server->smallbuf = buf = next_buffer;
4885                 goto one_more;
4886         } else if (ret != 0) {
4887                 /*
4888                  * ret != 0 here means that we didn't get to handle_mid() thus
4889                  * server->smallbuf and server->bigbuf are still valid. We need
4890                  * to free next_buffer because it is not going to be used
4891                  * anywhere.
4892                  */
4893                 if (next_is_large)
4894                         free_rsp_buf(CIFS_LARGE_BUFFER, next_buffer);
4895                 else
4896                         free_rsp_buf(CIFS_SMALL_BUFFER, next_buffer);
4897         }
4898
4899         return ret;
4900 }
4901
4902 static int
4903 smb3_receive_transform(struct TCP_Server_Info *server,
4904                        struct mid_q_entry **mids, char **bufs, int *num_mids)
4905 {
4906         char *buf = server->smallbuf;
4907         unsigned int pdu_length = server->pdu_size;
4908         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
4909         unsigned int orig_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
4910
4911         if (pdu_length < sizeof(struct smb2_transform_hdr) +
4912                                                 sizeof(struct smb2_sync_hdr)) {
4913                 cifs_server_dbg(VFS, "Transform message is too small (%u)\n",
4914                          pdu_length);
4915                 cifs_reconnect(server);
4916                 return -ECONNABORTED;
4917         }
4918
4919         if (pdu_length < orig_len + sizeof(struct smb2_transform_hdr)) {
4920                 cifs_server_dbg(VFS, "Transform message is broken\n");
4921                 cifs_reconnect(server);
4922                 return -ECONNABORTED;
4923         }
4924
4925         /* TODO: add support for compounds containing READ. */
4926         if (pdu_length > CIFSMaxBufSize + MAX_HEADER_SIZE(server)) {
4927                 return receive_encrypted_read(server, &mids[0], num_mids);
4928         }
4929
4930         return receive_encrypted_standard(server, mids, bufs, num_mids);
4931 }
4932
4933 int
4934 smb3_handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid)
4935 {
4936         char *buf = server->large_buf ? server->bigbuf : server->smallbuf;
4937
4938         return handle_read_data(server, mid, buf, server->pdu_size,
4939                                 NULL, 0, 0, false);
4940 }
4941
4942 static int
4943 smb2_next_header(char *buf)
4944 {
4945         struct smb2_sync_hdr *hdr = (struct smb2_sync_hdr *)buf;
4946         struct smb2_transform_hdr *t_hdr = (struct smb2_transform_hdr *)buf;
4947
4948         if (hdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM)
4949                 return sizeof(struct smb2_transform_hdr) +
4950                   le32_to_cpu(t_hdr->OriginalMessageSize);
4951
4952         return le32_to_cpu(hdr->NextCommand);
4953 }
4954
4955 static int
4956 smb2_make_node(unsigned int xid, struct inode *inode,
4957                struct dentry *dentry, struct cifs_tcon *tcon,
4958                char *full_path, umode_t mode, dev_t dev)
4959 {
4960         struct cifs_sb_info *cifs_sb = CIFS_SB(inode->i_sb);
4961         int rc = -EPERM;
4962         FILE_ALL_INFO *buf = NULL;
4963         struct cifs_io_parms io_parms = {0};
4964         __u32 oplock = 0;
4965         struct cifs_fid fid;
4966         struct cifs_open_parms oparms;
4967         unsigned int bytes_written;
4968         struct win_dev *pdev;
4969         struct kvec iov[2];
4970
4971         /*
4972          * Check if mounted with mount parm 'sfu' mount parm.
4973          * SFU emulation should work with all servers, but only
4974          * supports block and char device (no socket & fifo),
4975          * and was used by default in earlier versions of Windows
4976          */
4977         if (!(cifs_sb->mnt_cifs_flags & CIFS_MOUNT_UNX_EMUL))
4978                 goto out;
4979
4980         /*
4981          * TODO: Add ability to create instead via reparse point. Windows (e.g.
4982          * their current NFS server) uses this approach to expose special files
4983          * over SMB2/SMB3 and Samba will do this with SMB3.1.1 POSIX Extensions
4984          */
4985
4986         if (!S_ISCHR(mode) && !S_ISBLK(mode))
4987                 goto out;
4988
4989         cifs_dbg(FYI, "sfu compat create special file\n");
4990
4991         buf = kmalloc(sizeof(FILE_ALL_INFO), GFP_KERNEL);
4992         if (buf == NULL) {
4993                 rc = -ENOMEM;
4994                 goto out;
4995         }
4996
4997         oparms.tcon = tcon;
4998         oparms.cifs_sb = cifs_sb;
4999         oparms.desired_access = GENERIC_WRITE;
5000         oparms.create_options = cifs_create_options(cifs_sb, CREATE_NOT_DIR |
5001                                                     CREATE_OPTION_SPECIAL);
5002         oparms.disposition = FILE_CREATE;
5003         oparms.path = full_path;
5004         oparms.fid = &fid;
5005         oparms.reconnect = false;
5006
5007         if (tcon->ses->server->oplocks)
5008                 oplock = REQ_OPLOCK;
5009         else
5010                 oplock = 0;
5011         rc = tcon->ses->server->ops->open(xid, &oparms, &oplock, buf);
5012         if (rc)
5013                 goto out;
5014
5015         /*
5016          * BB Do not bother to decode buf since no local inode yet to put
5017          * timestamps in, but we can reuse it safely.
5018          */
5019
5020         pdev = (struct win_dev *)buf;
5021         io_parms.pid = current->tgid;
5022         io_parms.tcon = tcon;
5023         io_parms.offset = 0;
5024         io_parms.length = sizeof(struct win_dev);
5025         iov[1].iov_base = buf;
5026         iov[1].iov_len = sizeof(struct win_dev);
5027         if (S_ISCHR(mode)) {
5028                 memcpy(pdev->type, "IntxCHR", 8);
5029                 pdev->major = cpu_to_le64(MAJOR(dev));
5030                 pdev->minor = cpu_to_le64(MINOR(dev));
5031                 rc = tcon->ses->server->ops->sync_write(xid, &fid, &io_parms,
5032                                                         &bytes_written, iov, 1);
5033         } else if (S_ISBLK(mode)) {
5034                 memcpy(pdev->type, "IntxBLK", 8);
5035                 pdev->major = cpu_to_le64(MAJOR(dev));
5036                 pdev->minor = cpu_to_le64(MINOR(dev));
5037                 rc = tcon->ses->server->ops->sync_write(xid, &fid, &io_parms,
5038                                                         &bytes_written, iov, 1);
5039         }
5040         tcon->ses->server->ops->close(xid, tcon, &fid);
5041         d_drop(dentry);
5042
5043         /* FIXME: add code here to set EAs */
5044 out:
5045         kfree(buf);
5046         return rc;
5047 }
5048
5049
5050 struct smb_version_operations smb20_operations = {
5051         .compare_fids = smb2_compare_fids,
5052         .setup_request = smb2_setup_request,
5053         .setup_async_request = smb2_setup_async_request,
5054         .check_receive = smb2_check_receive,
5055         .add_credits = smb2_add_credits,
5056         .set_credits = smb2_set_credits,
5057         .get_credits_field = smb2_get_credits_field,
5058         .get_credits = smb2_get_credits,
5059         .wait_mtu_credits = cifs_wait_mtu_credits,
5060         .get_next_mid = smb2_get_next_mid,
5061         .revert_current_mid = smb2_revert_current_mid,
5062         .read_data_offset = smb2_read_data_offset,
5063         .read_data_length = smb2_read_data_length,
5064         .map_error = map_smb2_to_linux_error,
5065         .find_mid = smb2_find_mid,
5066         .check_message = smb2_check_message,
5067         .dump_detail = smb2_dump_detail,
5068         .clear_stats = smb2_clear_stats,
5069         .print_stats = smb2_print_stats,
5070         .is_oplock_break = smb2_is_valid_oplock_break,
5071         .handle_cancelled_mid = smb2_handle_cancelled_mid,
5072         .downgrade_oplock = smb2_downgrade_oplock,
5073         .need_neg = smb2_need_neg,
5074         .negotiate = smb2_negotiate,
5075         .negotiate_wsize = smb2_negotiate_wsize,
5076         .negotiate_rsize = smb2_negotiate_rsize,
5077         .sess_setup = SMB2_sess_setup,
5078         .logoff = SMB2_logoff,
5079         .tree_connect = SMB2_tcon,
5080         .tree_disconnect = SMB2_tdis,
5081         .qfs_tcon = smb2_qfs_tcon,
5082         .is_path_accessible = smb2_is_path_accessible,
5083         .can_echo = smb2_can_echo,
5084         .echo = SMB2_echo,
5085         .query_path_info = smb2_query_path_info,
5086         .get_srv_inum = smb2_get_srv_inum,
5087         .query_file_info = smb2_query_file_info,
5088         .set_path_size = smb2_set_path_size,
5089         .set_file_size = smb2_set_file_size,
5090         .set_file_info = smb2_set_file_info,
5091         .set_compression = smb2_set_compression,
5092         .mkdir = smb2_mkdir,
5093         .mkdir_setinfo = smb2_mkdir_setinfo,
5094         .rmdir = smb2_rmdir,
5095         .unlink = smb2_unlink,
5096         .rename = smb2_rename_path,
5097         .create_hardlink = smb2_create_hardlink,
5098         .query_symlink = smb2_query_symlink,
5099         .query_mf_symlink = smb3_query_mf_symlink,
5100         .create_mf_symlink = smb3_create_mf_symlink,
5101         .open = smb2_open_file,
5102         .set_fid = smb2_set_fid,
5103         .close = smb2_close_file,
5104         .flush = smb2_flush_file,
5105         .async_readv = smb2_async_readv,
5106         .async_writev = smb2_async_writev,
5107         .sync_read = smb2_sync_read,
5108         .sync_write = smb2_sync_write,
5109         .query_dir_first = smb2_query_dir_first,
5110         .query_dir_next = smb2_query_dir_next,
5111         .close_dir = smb2_close_dir,
5112         .calc_smb_size = smb2_calc_size,
5113         .is_status_pending = smb2_is_status_pending,
5114         .is_session_expired = smb2_is_session_expired,
5115         .oplock_response = smb2_oplock_response,
5116         .queryfs = smb2_queryfs,
5117         .mand_lock = smb2_mand_lock,
5118         .mand_unlock_range = smb2_unlock_range,
5119         .push_mand_locks = smb2_push_mandatory_locks,
5120         .get_lease_key = smb2_get_lease_key,
5121         .set_lease_key = smb2_set_lease_key,
5122         .new_lease_key = smb2_new_lease_key,
5123         .calc_signature = smb2_calc_signature,
5124         .is_read_op = smb2_is_read_op,
5125         .set_oplock_level = smb2_set_oplock_level,
5126         .create_lease_buf = smb2_create_lease_buf,
5127         .parse_lease_buf = smb2_parse_lease_buf,
5128         .copychunk_range = smb2_copychunk_range,
5129         .wp_retry_size = smb2_wp_retry_size,
5130         .dir_needs_close = smb2_dir_needs_close,
5131         .get_dfs_refer = smb2_get_dfs_refer,
5132         .select_sectype = smb2_select_sectype,
5133 #ifdef CONFIG_CIFS_XATTR
5134         .query_all_EAs = smb2_query_eas,
5135         .set_EA = smb2_set_ea,
5136 #endif /* CIFS_XATTR */
5137         .get_acl = get_smb2_acl,
5138         .get_acl_by_fid = get_smb2_acl_by_fid,
5139         .set_acl = set_smb2_acl,
5140         .next_header = smb2_next_header,
5141         .ioctl_query_info = smb2_ioctl_query_info,
5142         .make_node = smb2_make_node,
5143         .fiemap = smb3_fiemap,
5144         .llseek = smb3_llseek,
5145         .is_status_io_timeout = smb2_is_status_io_timeout,
5146         .is_network_name_deleted = smb2_is_network_name_deleted,
5147 };
5148
5149 struct smb_version_operations smb21_operations = {
5150         .compare_fids = smb2_compare_fids,
5151         .setup_request = smb2_setup_request,
5152         .setup_async_request = smb2_setup_async_request,
5153         .check_receive = smb2_check_receive,
5154         .add_credits = smb2_add_credits,
5155         .set_credits = smb2_set_credits,
5156         .get_credits_field = smb2_get_credits_field,
5157         .get_credits = smb2_get_credits,
5158         .wait_mtu_credits = smb2_wait_mtu_credits,
5159         .adjust_credits = smb2_adjust_credits,
5160         .get_next_mid = smb2_get_next_mid,
5161         .revert_current_mid = smb2_revert_current_mid,
5162         .read_data_offset = smb2_read_data_offset,
5163         .read_data_length = smb2_read_data_length,
5164         .map_error = map_smb2_to_linux_error,
5165         .find_mid = smb2_find_mid,
5166         .check_message = smb2_check_message,
5167         .dump_detail = smb2_dump_detail,
5168         .clear_stats = smb2_clear_stats,
5169         .print_stats = smb2_print_stats,
5170         .is_oplock_break = smb2_is_valid_oplock_break,
5171         .handle_cancelled_mid = smb2_handle_cancelled_mid,
5172         .downgrade_oplock = smb2_downgrade_oplock,
5173         .need_neg = smb2_need_neg,
5174         .negotiate = smb2_negotiate,
5175         .negotiate_wsize = smb2_negotiate_wsize,
5176         .negotiate_rsize = smb2_negotiate_rsize,
5177         .sess_setup = SMB2_sess_setup,
5178         .logoff = SMB2_logoff,
5179         .tree_connect = SMB2_tcon,
5180         .tree_disconnect = SMB2_tdis,
5181         .qfs_tcon = smb2_qfs_tcon,
5182         .is_path_accessible = smb2_is_path_accessible,
5183         .can_echo = smb2_can_echo,
5184         .echo = SMB2_echo,
5185         .query_path_info = smb2_query_path_info,
5186         .get_srv_inum = smb2_get_srv_inum,
5187         .query_file_info = smb2_query_file_info,
5188         .set_path_size = smb2_set_path_size,
5189         .set_file_size = smb2_set_file_size,
5190         .set_file_info = smb2_set_file_info,
5191         .set_compression = smb2_set_compression,
5192         .mkdir = smb2_mkdir,
5193         .mkdir_setinfo = smb2_mkdir_setinfo,
5194         .rmdir = smb2_rmdir,
5195         .unlink = smb2_unlink,
5196         .rename = smb2_rename_path,
5197         .create_hardlink = smb2_create_hardlink,
5198         .query_symlink = smb2_query_symlink,
5199         .query_mf_symlink = smb3_query_mf_symlink,
5200         .create_mf_symlink = smb3_create_mf_symlink,
5201         .open = smb2_open_file,
5202         .set_fid = smb2_set_fid,
5203         .close = smb2_close_file,
5204         .flush = smb2_flush_file,
5205         .async_readv = smb2_async_readv,
5206         .async_writev = smb2_async_writev,
5207         .sync_read = smb2_sync_read,
5208         .sync_write = smb2_sync_write,
5209         .query_dir_first = smb2_query_dir_first,
5210         .query_dir_next = smb2_query_dir_next,
5211         .close_dir = smb2_close_dir,
5212         .calc_smb_size = smb2_calc_size,
5213         .is_status_pending = smb2_is_status_pending,
5214         .is_session_expired = smb2_is_session_expired,
5215         .oplock_response = smb2_oplock_response,
5216         .queryfs = smb2_queryfs,
5217         .mand_lock = smb2_mand_lock,
5218         .mand_unlock_range = smb2_unlock_range,
5219         .push_mand_locks = smb2_push_mandatory_locks,
5220         .get_lease_key = smb2_get_lease_key,
5221         .set_lease_key = smb2_set_lease_key,
5222         .new_lease_key = smb2_new_lease_key,
5223         .calc_signature = smb2_calc_signature,
5224         .is_read_op = smb21_is_read_op,
5225         .set_oplock_level = smb21_set_oplock_level,
5226         .create_lease_buf = smb2_create_lease_buf,
5227         .parse_lease_buf = smb2_parse_lease_buf,
5228         .copychunk_range = smb2_copychunk_range,
5229         .wp_retry_size = smb2_wp_retry_size,
5230         .dir_needs_close = smb2_dir_needs_close,
5231         .enum_snapshots = smb3_enum_snapshots,
5232         .notify = smb3_notify,
5233         .get_dfs_refer = smb2_get_dfs_refer,
5234         .select_sectype = smb2_select_sectype,
5235 #ifdef CONFIG_CIFS_XATTR
5236         .query_all_EAs = smb2_query_eas,
5237         .set_EA = smb2_set_ea,
5238 #endif /* CIFS_XATTR */
5239         .get_acl = get_smb2_acl,
5240         .get_acl_by_fid = get_smb2_acl_by_fid,
5241         .set_acl = set_smb2_acl,
5242         .next_header = smb2_next_header,
5243         .ioctl_query_info = smb2_ioctl_query_info,
5244         .make_node = smb2_make_node,
5245         .fiemap = smb3_fiemap,
5246         .llseek = smb3_llseek,
5247         .is_status_io_timeout = smb2_is_status_io_timeout,
5248         .is_network_name_deleted = smb2_is_network_name_deleted,
5249 };
5250
5251 struct smb_version_operations smb30_operations = {
5252         .compare_fids = smb2_compare_fids,
5253         .setup_request = smb2_setup_request,
5254         .setup_async_request = smb2_setup_async_request,
5255         .check_receive = smb2_check_receive,
5256         .add_credits = smb2_add_credits,
5257         .set_credits = smb2_set_credits,
5258         .get_credits_field = smb2_get_credits_field,
5259         .get_credits = smb2_get_credits,
5260         .wait_mtu_credits = smb2_wait_mtu_credits,
5261         .adjust_credits = smb2_adjust_credits,
5262         .get_next_mid = smb2_get_next_mid,
5263         .revert_current_mid = smb2_revert_current_mid,
5264         .read_data_offset = smb2_read_data_offset,
5265         .read_data_length = smb2_read_data_length,
5266         .map_error = map_smb2_to_linux_error,
5267         .find_mid = smb2_find_mid,
5268         .check_message = smb2_check_message,
5269         .dump_detail = smb2_dump_detail,
5270         .clear_stats = smb2_clear_stats,
5271         .print_stats = smb2_print_stats,
5272         .dump_share_caps = smb2_dump_share_caps,
5273         .is_oplock_break = smb2_is_valid_oplock_break,
5274         .handle_cancelled_mid = smb2_handle_cancelled_mid,
5275         .downgrade_oplock = smb3_downgrade_oplock,
5276         .need_neg = smb2_need_neg,
5277         .negotiate = smb2_negotiate,
5278         .negotiate_wsize = smb3_negotiate_wsize,
5279         .negotiate_rsize = smb3_negotiate_rsize,
5280         .sess_setup = SMB2_sess_setup,
5281         .logoff = SMB2_logoff,
5282         .tree_connect = SMB2_tcon,
5283         .tree_disconnect = SMB2_tdis,
5284         .qfs_tcon = smb3_qfs_tcon,
5285         .is_path_accessible = smb2_is_path_accessible,
5286         .can_echo = smb2_can_echo,
5287         .echo = SMB2_echo,
5288         .query_path_info = smb2_query_path_info,
5289         /* WSL tags introduced long after smb2.1, enable for SMB3, 3.11 only */
5290         .query_reparse_tag = smb2_query_reparse_tag,
5291         .get_srv_inum = smb2_get_srv_inum,
5292         .query_file_info = smb2_query_file_info,
5293         .set_path_size = smb2_set_path_size,
5294         .set_file_size = smb2_set_file_size,
5295         .set_file_info = smb2_set_file_info,
5296         .set_compression = smb2_set_compression,
5297         .mkdir = smb2_mkdir,
5298         .mkdir_setinfo = smb2_mkdir_setinfo,
5299         .rmdir = smb2_rmdir,
5300         .unlink = smb2_unlink,
5301         .rename = smb2_rename_path,
5302         .create_hardlink = smb2_create_hardlink,
5303         .query_symlink = smb2_query_symlink,
5304         .query_mf_symlink = smb3_query_mf_symlink,
5305         .create_mf_symlink = smb3_create_mf_symlink,
5306         .open = smb2_open_file,
5307         .set_fid = smb2_set_fid,
5308         .close = smb2_close_file,
5309         .close_getattr = smb2_close_getattr,
5310         .flush = smb2_flush_file,
5311         .async_readv = smb2_async_readv,
5312         .async_writev = smb2_async_writev,
5313         .sync_read = smb2_sync_read,
5314         .sync_write = smb2_sync_write,
5315         .query_dir_first = smb2_query_dir_first,
5316         .query_dir_next = smb2_query_dir_next,
5317         .close_dir = smb2_close_dir,
5318         .calc_smb_size = smb2_calc_size,
5319         .is_status_pending = smb2_is_status_pending,
5320         .is_session_expired = smb2_is_session_expired,
5321         .oplock_response = smb2_oplock_response,
5322         .queryfs = smb2_queryfs,
5323         .mand_lock = smb2_mand_lock,
5324         .mand_unlock_range = smb2_unlock_range,
5325         .push_mand_locks = smb2_push_mandatory_locks,
5326         .get_lease_key = smb2_get_lease_key,
5327         .set_lease_key = smb2_set_lease_key,
5328         .new_lease_key = smb2_new_lease_key,
5329         .generate_signingkey = generate_smb30signingkey,
5330         .calc_signature = smb3_calc_signature,
5331         .set_integrity  = smb3_set_integrity,
5332         .is_read_op = smb21_is_read_op,
5333         .set_oplock_level = smb3_set_oplock_level,
5334         .create_lease_buf = smb3_create_lease_buf,
5335         .parse_lease_buf = smb3_parse_lease_buf,
5336         .copychunk_range = smb2_copychunk_range,
5337         .duplicate_extents = smb2_duplicate_extents,
5338         .validate_negotiate = smb3_validate_negotiate,
5339         .wp_retry_size = smb2_wp_retry_size,
5340         .dir_needs_close = smb2_dir_needs_close,
5341         .fallocate = smb3_fallocate,
5342         .enum_snapshots = smb3_enum_snapshots,
5343         .notify = smb3_notify,
5344         .init_transform_rq = smb3_init_transform_rq,
5345         .is_transform_hdr = smb3_is_transform_hdr,
5346         .receive_transform = smb3_receive_transform,
5347         .get_dfs_refer = smb2_get_dfs_refer,
5348         .select_sectype = smb2_select_sectype,
5349 #ifdef CONFIG_CIFS_XATTR
5350         .query_all_EAs = smb2_query_eas,
5351         .set_EA = smb2_set_ea,
5352 #endif /* CIFS_XATTR */
5353         .get_acl = get_smb2_acl,
5354         .get_acl_by_fid = get_smb2_acl_by_fid,
5355         .set_acl = set_smb2_acl,
5356         .next_header = smb2_next_header,
5357         .ioctl_query_info = smb2_ioctl_query_info,
5358         .make_node = smb2_make_node,
5359         .fiemap = smb3_fiemap,
5360         .llseek = smb3_llseek,
5361         .is_status_io_timeout = smb2_is_status_io_timeout,
5362         .is_network_name_deleted = smb2_is_network_name_deleted,
5363 };
5364
5365 struct smb_version_operations smb311_operations = {
5366         .compare_fids = smb2_compare_fids,
5367         .setup_request = smb2_setup_request,
5368         .setup_async_request = smb2_setup_async_request,
5369         .check_receive = smb2_check_receive,
5370         .add_credits = smb2_add_credits,
5371         .set_credits = smb2_set_credits,
5372         .get_credits_field = smb2_get_credits_field,
5373         .get_credits = smb2_get_credits,
5374         .wait_mtu_credits = smb2_wait_mtu_credits,
5375         .adjust_credits = smb2_adjust_credits,
5376         .get_next_mid = smb2_get_next_mid,
5377         .revert_current_mid = smb2_revert_current_mid,
5378         .read_data_offset = smb2_read_data_offset,
5379         .read_data_length = smb2_read_data_length,
5380         .map_error = map_smb2_to_linux_error,
5381         .find_mid = smb2_find_mid,
5382         .check_message = smb2_check_message,
5383         .dump_detail = smb2_dump_detail,
5384         .clear_stats = smb2_clear_stats,
5385         .print_stats = smb2_print_stats,
5386         .dump_share_caps = smb2_dump_share_caps,
5387         .is_oplock_break = smb2_is_valid_oplock_break,
5388         .handle_cancelled_mid = smb2_handle_cancelled_mid,
5389         .downgrade_oplock = smb3_downgrade_oplock,
5390         .need_neg = smb2_need_neg,
5391         .negotiate = smb2_negotiate,
5392         .negotiate_wsize = smb3_negotiate_wsize,
5393         .negotiate_rsize = smb3_negotiate_rsize,
5394         .sess_setup = SMB2_sess_setup,
5395         .logoff = SMB2_logoff,
5396         .tree_connect = SMB2_tcon,
5397         .tree_disconnect = SMB2_tdis,
5398         .qfs_tcon = smb3_qfs_tcon,
5399         .is_path_accessible = smb2_is_path_accessible,
5400         .can_echo = smb2_can_echo,
5401         .echo = SMB2_echo,
5402         .query_path_info = smb2_query_path_info,
5403         .query_reparse_tag = smb2_query_reparse_tag,
5404         .get_srv_inum = smb2_get_srv_inum,
5405         .query_file_info = smb2_query_file_info,
5406         .set_path_size = smb2_set_path_size,
5407         .set_file_size = smb2_set_file_size,
5408         .set_file_info = smb2_set_file_info,
5409         .set_compression = smb2_set_compression,
5410         .mkdir = smb2_mkdir,
5411         .mkdir_setinfo = smb2_mkdir_setinfo,
5412         .posix_mkdir = smb311_posix_mkdir,
5413         .rmdir = smb2_rmdir,
5414         .unlink = smb2_unlink,
5415         .rename = smb2_rename_path,
5416         .create_hardlink = smb2_create_hardlink,
5417         .query_symlink = smb2_query_symlink,
5418         .query_mf_symlink = smb3_query_mf_symlink,
5419         .create_mf_symlink = smb3_create_mf_symlink,
5420         .open = smb2_open_file,
5421         .set_fid = smb2_set_fid,
5422         .close = smb2_close_file,
5423         .close_getattr = smb2_close_getattr,
5424         .flush = smb2_flush_file,
5425         .async_readv = smb2_async_readv,
5426         .async_writev = smb2_async_writev,
5427         .sync_read = smb2_sync_read,
5428         .sync_write = smb2_sync_write,
5429         .query_dir_first = smb2_query_dir_first,
5430         .query_dir_next = smb2_query_dir_next,
5431         .close_dir = smb2_close_dir,
5432         .calc_smb_size = smb2_calc_size,
5433         .is_status_pending = smb2_is_status_pending,
5434         .is_session_expired = smb2_is_session_expired,
5435         .oplock_response = smb2_oplock_response,
5436         .queryfs = smb311_queryfs,
5437         .mand_lock = smb2_mand_lock,
5438         .mand_unlock_range = smb2_unlock_range,
5439         .push_mand_locks = smb2_push_mandatory_locks,
5440         .get_lease_key = smb2_get_lease_key,
5441         .set_lease_key = smb2_set_lease_key,
5442         .new_lease_key = smb2_new_lease_key,
5443         .generate_signingkey = generate_smb311signingkey,
5444         .calc_signature = smb3_calc_signature,
5445         .set_integrity  = smb3_set_integrity,
5446         .is_read_op = smb21_is_read_op,
5447         .set_oplock_level = smb3_set_oplock_level,
5448         .create_lease_buf = smb3_create_lease_buf,
5449         .parse_lease_buf = smb3_parse_lease_buf,
5450         .copychunk_range = smb2_copychunk_range,
5451         .duplicate_extents = smb2_duplicate_extents,
5452 /*      .validate_negotiate = smb3_validate_negotiate, */ /* not used in 3.11 */
5453         .wp_retry_size = smb2_wp_retry_size,
5454         .dir_needs_close = smb2_dir_needs_close,
5455         .fallocate = smb3_fallocate,
5456         .enum_snapshots = smb3_enum_snapshots,
5457         .notify = smb3_notify,
5458         .init_transform_rq = smb3_init_transform_rq,
5459         .is_transform_hdr = smb3_is_transform_hdr,
5460         .receive_transform = smb3_receive_transform,
5461         .get_dfs_refer = smb2_get_dfs_refer,
5462         .select_sectype = smb2_select_sectype,
5463 #ifdef CONFIG_CIFS_XATTR
5464         .query_all_EAs = smb2_query_eas,
5465         .set_EA = smb2_set_ea,
5466 #endif /* CIFS_XATTR */
5467         .get_acl = get_smb2_acl,
5468         .get_acl_by_fid = get_smb2_acl_by_fid,
5469         .set_acl = set_smb2_acl,
5470         .next_header = smb2_next_header,
5471         .ioctl_query_info = smb2_ioctl_query_info,
5472         .make_node = smb2_make_node,
5473         .fiemap = smb3_fiemap,
5474         .llseek = smb3_llseek,
5475         .is_status_io_timeout = smb2_is_status_io_timeout,
5476         .is_network_name_deleted = smb2_is_network_name_deleted,
5477 };
5478
5479 struct smb_version_values smb20_values = {
5480         .version_string = SMB20_VERSION_STRING,
5481         .protocol_id = SMB20_PROT_ID,
5482         .req_capabilities = 0, /* MBZ */
5483         .large_lock_type = 0,
5484         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5485         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5486         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5487         .header_size = sizeof(struct smb2_sync_hdr),
5488         .header_preamble_size = 0,
5489         .max_header_size = MAX_SMB2_HDR_SIZE,
5490         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5491         .lock_cmd = SMB2_LOCK,
5492         .cap_unix = 0,
5493         .cap_nt_find = SMB2_NT_FIND,
5494         .cap_large_files = SMB2_LARGE_FILES,
5495         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5496         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5497         .create_lease_size = sizeof(struct create_lease),
5498 };
5499
5500 struct smb_version_values smb21_values = {
5501         .version_string = SMB21_VERSION_STRING,
5502         .protocol_id = SMB21_PROT_ID,
5503         .req_capabilities = 0, /* MBZ on negotiate req until SMB3 dialect */
5504         .large_lock_type = 0,
5505         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5506         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5507         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5508         .header_size = sizeof(struct smb2_sync_hdr),
5509         .header_preamble_size = 0,
5510         .max_header_size = MAX_SMB2_HDR_SIZE,
5511         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5512         .lock_cmd = SMB2_LOCK,
5513         .cap_unix = 0,
5514         .cap_nt_find = SMB2_NT_FIND,
5515         .cap_large_files = SMB2_LARGE_FILES,
5516         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5517         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5518         .create_lease_size = sizeof(struct create_lease),
5519 };
5520
5521 struct smb_version_values smb3any_values = {
5522         .version_string = SMB3ANY_VERSION_STRING,
5523         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
5524         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION | SMB2_GLOBAL_CAP_DIRECTORY_LEASING,
5525         .large_lock_type = 0,
5526         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5527         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5528         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5529         .header_size = sizeof(struct smb2_sync_hdr),
5530         .header_preamble_size = 0,
5531         .max_header_size = MAX_SMB2_HDR_SIZE,
5532         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5533         .lock_cmd = SMB2_LOCK,
5534         .cap_unix = 0,
5535         .cap_nt_find = SMB2_NT_FIND,
5536         .cap_large_files = SMB2_LARGE_FILES,
5537         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5538         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5539         .create_lease_size = sizeof(struct create_lease_v2),
5540 };
5541
5542 struct smb_version_values smbdefault_values = {
5543         .version_string = SMBDEFAULT_VERSION_STRING,
5544         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
5545         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION | SMB2_GLOBAL_CAP_DIRECTORY_LEASING,
5546         .large_lock_type = 0,
5547         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5548         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5549         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5550         .header_size = sizeof(struct smb2_sync_hdr),
5551         .header_preamble_size = 0,
5552         .max_header_size = MAX_SMB2_HDR_SIZE,
5553         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5554         .lock_cmd = SMB2_LOCK,
5555         .cap_unix = 0,
5556         .cap_nt_find = SMB2_NT_FIND,
5557         .cap_large_files = SMB2_LARGE_FILES,
5558         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5559         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5560         .create_lease_size = sizeof(struct create_lease_v2),
5561 };
5562
5563 struct smb_version_values smb30_values = {
5564         .version_string = SMB30_VERSION_STRING,
5565         .protocol_id = SMB30_PROT_ID,
5566         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION | SMB2_GLOBAL_CAP_DIRECTORY_LEASING,
5567         .large_lock_type = 0,
5568         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5569         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5570         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5571         .header_size = sizeof(struct smb2_sync_hdr),
5572         .header_preamble_size = 0,
5573         .max_header_size = MAX_SMB2_HDR_SIZE,
5574         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5575         .lock_cmd = SMB2_LOCK,
5576         .cap_unix = 0,
5577         .cap_nt_find = SMB2_NT_FIND,
5578         .cap_large_files = SMB2_LARGE_FILES,
5579         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5580         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5581         .create_lease_size = sizeof(struct create_lease_v2),
5582 };
5583
5584 struct smb_version_values smb302_values = {
5585         .version_string = SMB302_VERSION_STRING,
5586         .protocol_id = SMB302_PROT_ID,
5587         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION | SMB2_GLOBAL_CAP_DIRECTORY_LEASING,
5588         .large_lock_type = 0,
5589         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5590         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5591         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5592         .header_size = sizeof(struct smb2_sync_hdr),
5593         .header_preamble_size = 0,
5594         .max_header_size = MAX_SMB2_HDR_SIZE,
5595         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5596         .lock_cmd = SMB2_LOCK,
5597         .cap_unix = 0,
5598         .cap_nt_find = SMB2_NT_FIND,
5599         .cap_large_files = SMB2_LARGE_FILES,
5600         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5601         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5602         .create_lease_size = sizeof(struct create_lease_v2),
5603 };
5604
5605 struct smb_version_values smb311_values = {
5606         .version_string = SMB311_VERSION_STRING,
5607         .protocol_id = SMB311_PROT_ID,
5608         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION | SMB2_GLOBAL_CAP_DIRECTORY_LEASING,
5609         .large_lock_type = 0,
5610         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
5611         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
5612         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
5613         .header_size = sizeof(struct smb2_sync_hdr),
5614         .header_preamble_size = 0,
5615         .max_header_size = MAX_SMB2_HDR_SIZE,
5616         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
5617         .lock_cmd = SMB2_LOCK,
5618         .cap_unix = 0,
5619         .cap_nt_find = SMB2_NT_FIND,
5620         .cap_large_files = SMB2_LARGE_FILES,
5621         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
5622         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
5623         .create_lease_size = sizeof(struct create_lease_v2),
5624 };