OSDN Git Service

smb3: add dynamic tracepoints for simple fallocate and zero range
[uclinux-h8/linux.git] / fs / cifs / smb2ops.c
1 /*
2  *  SMB2 version specific operations
3  *
4  *  Copyright (c) 2012, Jeff Layton <jlayton@redhat.com>
5  *
6  *  This library is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License v2 as published
8  *  by the Free Software Foundation.
9  *
10  *  This library is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
13  *  the GNU Lesser General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Lesser General Public License
16  *  along with this library; if not, write to the Free Software
17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18  */
19
20 #include <linux/pagemap.h>
21 #include <linux/vfs.h>
22 #include <linux/falloc.h>
23 #include <linux/scatterlist.h>
24 #include <linux/uuid.h>
25 #include <crypto/aead.h>
26 #include "cifsglob.h"
27 #include "smb2pdu.h"
28 #include "smb2proto.h"
29 #include "cifsproto.h"
30 #include "cifs_debug.h"
31 #include "cifs_unicode.h"
32 #include "smb2status.h"
33 #include "smb2glob.h"
34 #include "cifs_ioctl.h"
35 #include "smbdirect.h"
36
37 /* Change credits for different ops and return the total number of credits */
38 static int
39 change_conf(struct TCP_Server_Info *server)
40 {
41         server->credits += server->echo_credits + server->oplock_credits;
42         server->oplock_credits = server->echo_credits = 0;
43         switch (server->credits) {
44         case 0:
45                 return 0;
46         case 1:
47                 server->echoes = false;
48                 server->oplocks = false;
49                 break;
50         case 2:
51                 server->echoes = true;
52                 server->oplocks = false;
53                 server->echo_credits = 1;
54                 break;
55         default:
56                 server->echoes = true;
57                 if (enable_oplocks) {
58                         server->oplocks = true;
59                         server->oplock_credits = 1;
60                 } else
61                         server->oplocks = false;
62
63                 server->echo_credits = 1;
64         }
65         server->credits -= server->echo_credits + server->oplock_credits;
66         return server->credits + server->echo_credits + server->oplock_credits;
67 }
68
69 static void
70 smb2_add_credits(struct TCP_Server_Info *server,
71                  const struct cifs_credits *credits, const int optype)
72 {
73         int *val, rc = -1;
74         unsigned int add = credits->value;
75         unsigned int instance = credits->instance;
76         bool reconnect_detected = false;
77
78         spin_lock(&server->req_lock);
79         val = server->ops->get_credits_field(server, optype);
80
81         /* eg found case where write overlapping reconnect messed up credits */
82         if (((optype & CIFS_OP_MASK) == CIFS_NEG_OP) && (*val != 0))
83                 trace_smb3_reconnect_with_invalid_credits(server->CurrentMid,
84                         server->hostname, *val);
85         if ((instance == 0) || (instance == server->reconnect_instance))
86                 *val += add;
87         else
88                 reconnect_detected = true;
89
90         if (*val > 65000) {
91                 *val = 65000; /* Don't get near 64K credits, avoid srv bugs */
92                 printk_once(KERN_WARNING "server overflowed SMB3 credits\n");
93         }
94         server->in_flight--;
95         if (server->in_flight == 0 && (optype & CIFS_OP_MASK) != CIFS_NEG_OP)
96                 rc = change_conf(server);
97         /*
98          * Sometimes server returns 0 credits on oplock break ack - we need to
99          * rebalance credits in this case.
100          */
101         else if (server->in_flight > 0 && server->oplock_credits == 0 &&
102                  server->oplocks) {
103                 if (server->credits > 1) {
104                         server->credits--;
105                         server->oplock_credits++;
106                 }
107         }
108         spin_unlock(&server->req_lock);
109         wake_up(&server->request_q);
110
111         if (reconnect_detected)
112                 cifs_dbg(FYI, "trying to put %d credits from the old server instance %d\n",
113                          add, instance);
114
115         if (server->tcpStatus == CifsNeedReconnect
116             || server->tcpStatus == CifsExiting)
117                 return;
118
119         switch (rc) {
120         case -1:
121                 /* change_conf hasn't been executed */
122                 break;
123         case 0:
124                 cifs_dbg(VFS, "Possible client or server bug - zero credits\n");
125                 break;
126         case 1:
127                 cifs_dbg(VFS, "disabling echoes and oplocks\n");
128                 break;
129         case 2:
130                 cifs_dbg(FYI, "disabling oplocks\n");
131                 break;
132         default:
133                 cifs_dbg(FYI, "add %u credits total=%d\n", add, rc);
134         }
135 }
136
137 static void
138 smb2_set_credits(struct TCP_Server_Info *server, const int val)
139 {
140         spin_lock(&server->req_lock);
141         server->credits = val;
142         if (val == 1)
143                 server->reconnect_instance++;
144         spin_unlock(&server->req_lock);
145         /* don't log while holding the lock */
146         if (val == 1)
147                 cifs_dbg(FYI, "set credits to 1 due to smb2 reconnect\n");
148 }
149
150 static int *
151 smb2_get_credits_field(struct TCP_Server_Info *server, const int optype)
152 {
153         switch (optype) {
154         case CIFS_ECHO_OP:
155                 return &server->echo_credits;
156         case CIFS_OBREAK_OP:
157                 return &server->oplock_credits;
158         default:
159                 return &server->credits;
160         }
161 }
162
163 static unsigned int
164 smb2_get_credits(struct mid_q_entry *mid)
165 {
166         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)mid->resp_buf;
167
168         if (mid->mid_state == MID_RESPONSE_RECEIVED
169             || mid->mid_state == MID_RESPONSE_MALFORMED)
170                 return le16_to_cpu(shdr->CreditRequest);
171
172         return 0;
173 }
174
175 static int
176 smb2_wait_mtu_credits(struct TCP_Server_Info *server, unsigned int size,
177                       unsigned int *num, struct cifs_credits *credits)
178 {
179         int rc = 0;
180         unsigned int scredits;
181
182         spin_lock(&server->req_lock);
183         while (1) {
184                 if (server->credits <= 0) {
185                         spin_unlock(&server->req_lock);
186                         cifs_num_waiters_inc(server);
187                         rc = wait_event_killable(server->request_q,
188                                 has_credits(server, &server->credits, 1));
189                         cifs_num_waiters_dec(server);
190                         if (rc)
191                                 return rc;
192                         spin_lock(&server->req_lock);
193                 } else {
194                         if (server->tcpStatus == CifsExiting) {
195                                 spin_unlock(&server->req_lock);
196                                 return -ENOENT;
197                         }
198
199                         scredits = server->credits;
200                         /* can deadlock with reopen */
201                         if (scredits <= 8) {
202                                 *num = SMB2_MAX_BUFFER_SIZE;
203                                 credits->value = 0;
204                                 credits->instance = 0;
205                                 break;
206                         }
207
208                         /* leave some credits for reopen and other ops */
209                         scredits -= 8;
210                         *num = min_t(unsigned int, size,
211                                      scredits * SMB2_MAX_BUFFER_SIZE);
212
213                         credits->value =
214                                 DIV_ROUND_UP(*num, SMB2_MAX_BUFFER_SIZE);
215                         credits->instance = server->reconnect_instance;
216                         server->credits -= credits->value;
217                         server->in_flight++;
218                         break;
219                 }
220         }
221         spin_unlock(&server->req_lock);
222         return rc;
223 }
224
225 static int
226 smb2_adjust_credits(struct TCP_Server_Info *server,
227                     struct cifs_credits *credits,
228                     const unsigned int payload_size)
229 {
230         int new_val = DIV_ROUND_UP(payload_size, SMB2_MAX_BUFFER_SIZE);
231
232         if (!credits->value || credits->value == new_val)
233                 return 0;
234
235         if (credits->value < new_val) {
236                 WARN_ONCE(1, "request has less credits (%d) than required (%d)",
237                           credits->value, new_val);
238                 return -ENOTSUPP;
239         }
240
241         spin_lock(&server->req_lock);
242
243         if (server->reconnect_instance != credits->instance) {
244                 spin_unlock(&server->req_lock);
245                 cifs_dbg(VFS, "trying to return %d credits to old session\n",
246                          credits->value - new_val);
247                 return -EAGAIN;
248         }
249
250         server->credits += credits->value - new_val;
251         spin_unlock(&server->req_lock);
252         wake_up(&server->request_q);
253         credits->value = new_val;
254         return 0;
255 }
256
257 static __u64
258 smb2_get_next_mid(struct TCP_Server_Info *server)
259 {
260         __u64 mid;
261         /* for SMB2 we need the current value */
262         spin_lock(&GlobalMid_Lock);
263         mid = server->CurrentMid++;
264         spin_unlock(&GlobalMid_Lock);
265         return mid;
266 }
267
268 static void
269 smb2_revert_current_mid(struct TCP_Server_Info *server, const unsigned int val)
270 {
271         spin_lock(&GlobalMid_Lock);
272         if (server->CurrentMid >= val)
273                 server->CurrentMid -= val;
274         spin_unlock(&GlobalMid_Lock);
275 }
276
277 static struct mid_q_entry *
278 smb2_find_mid(struct TCP_Server_Info *server, char *buf)
279 {
280         struct mid_q_entry *mid;
281         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
282         __u64 wire_mid = le64_to_cpu(shdr->MessageId);
283
284         if (shdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM) {
285                 cifs_dbg(VFS, "encrypted frame parsing not supported yet");
286                 return NULL;
287         }
288
289         spin_lock(&GlobalMid_Lock);
290         list_for_each_entry(mid, &server->pending_mid_q, qhead) {
291                 if ((mid->mid == wire_mid) &&
292                     (mid->mid_state == MID_REQUEST_SUBMITTED) &&
293                     (mid->command == shdr->Command)) {
294                         kref_get(&mid->refcount);
295                         spin_unlock(&GlobalMid_Lock);
296                         return mid;
297                 }
298         }
299         spin_unlock(&GlobalMid_Lock);
300         return NULL;
301 }
302
303 static void
304 smb2_dump_detail(void *buf, struct TCP_Server_Info *server)
305 {
306 #ifdef CONFIG_CIFS_DEBUG2
307         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
308
309         cifs_dbg(VFS, "Cmd: %d Err: 0x%x Flags: 0x%x Mid: %llu Pid: %d\n",
310                  shdr->Command, shdr->Status, shdr->Flags, shdr->MessageId,
311                  shdr->ProcessId);
312         cifs_dbg(VFS, "smb buf %p len %u\n", buf,
313                  server->ops->calc_smb_size(buf, server));
314 #endif
315 }
316
317 static bool
318 smb2_need_neg(struct TCP_Server_Info *server)
319 {
320         return server->max_read == 0;
321 }
322
323 static int
324 smb2_negotiate(const unsigned int xid, struct cifs_ses *ses)
325 {
326         int rc;
327         ses->server->CurrentMid = 0;
328         rc = SMB2_negotiate(xid, ses);
329         /* BB we probably don't need to retry with modern servers */
330         if (rc == -EAGAIN)
331                 rc = -EHOSTDOWN;
332         return rc;
333 }
334
335 static unsigned int
336 smb2_negotiate_wsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
337 {
338         struct TCP_Server_Info *server = tcon->ses->server;
339         unsigned int wsize;
340
341         /* start with specified wsize, or default */
342         wsize = volume_info->wsize ? volume_info->wsize : CIFS_DEFAULT_IOSIZE;
343         wsize = min_t(unsigned int, wsize, server->max_write);
344 #ifdef CONFIG_CIFS_SMB_DIRECT
345         if (server->rdma) {
346                 if (server->sign)
347                         wsize = min_t(unsigned int,
348                                 wsize, server->smbd_conn->max_fragmented_send_size);
349                 else
350                         wsize = min_t(unsigned int,
351                                 wsize, server->smbd_conn->max_readwrite_size);
352         }
353 #endif
354         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
355                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
356
357         return wsize;
358 }
359
360 static unsigned int
361 smb3_negotiate_wsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
362 {
363         struct TCP_Server_Info *server = tcon->ses->server;
364         unsigned int wsize;
365
366         /* start with specified wsize, or default */
367         wsize = volume_info->wsize ? volume_info->wsize : SMB3_DEFAULT_IOSIZE;
368         wsize = min_t(unsigned int, wsize, server->max_write);
369 #ifdef CONFIG_CIFS_SMB_DIRECT
370         if (server->rdma) {
371                 if (server->sign)
372                         wsize = min_t(unsigned int,
373                                 wsize, server->smbd_conn->max_fragmented_send_size);
374                 else
375                         wsize = min_t(unsigned int,
376                                 wsize, server->smbd_conn->max_readwrite_size);
377         }
378 #endif
379         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
380                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
381
382         return wsize;
383 }
384
385 static unsigned int
386 smb2_negotiate_rsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
387 {
388         struct TCP_Server_Info *server = tcon->ses->server;
389         unsigned int rsize;
390
391         /* start with specified rsize, or default */
392         rsize = volume_info->rsize ? volume_info->rsize : CIFS_DEFAULT_IOSIZE;
393         rsize = min_t(unsigned int, rsize, server->max_read);
394 #ifdef CONFIG_CIFS_SMB_DIRECT
395         if (server->rdma) {
396                 if (server->sign)
397                         rsize = min_t(unsigned int,
398                                 rsize, server->smbd_conn->max_fragmented_recv_size);
399                 else
400                         rsize = min_t(unsigned int,
401                                 rsize, server->smbd_conn->max_readwrite_size);
402         }
403 #endif
404
405         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
406                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
407
408         return rsize;
409 }
410
411 static unsigned int
412 smb3_negotiate_rsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
413 {
414         struct TCP_Server_Info *server = tcon->ses->server;
415         unsigned int rsize;
416
417         /* start with specified rsize, or default */
418         rsize = volume_info->rsize ? volume_info->rsize : SMB3_DEFAULT_IOSIZE;
419         rsize = min_t(unsigned int, rsize, server->max_read);
420 #ifdef CONFIG_CIFS_SMB_DIRECT
421         if (server->rdma) {
422                 if (server->sign)
423                         rsize = min_t(unsigned int,
424                                 rsize, server->smbd_conn->max_fragmented_recv_size);
425                 else
426                         rsize = min_t(unsigned int,
427                                 rsize, server->smbd_conn->max_readwrite_size);
428         }
429 #endif
430
431         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
432                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
433
434         return rsize;
435 }
436
437 static int
438 parse_server_interfaces(struct network_interface_info_ioctl_rsp *buf,
439                         size_t buf_len,
440                         struct cifs_server_iface **iface_list,
441                         size_t *iface_count)
442 {
443         struct network_interface_info_ioctl_rsp *p;
444         struct sockaddr_in *addr4;
445         struct sockaddr_in6 *addr6;
446         struct iface_info_ipv4 *p4;
447         struct iface_info_ipv6 *p6;
448         struct cifs_server_iface *info;
449         ssize_t bytes_left;
450         size_t next = 0;
451         int nb_iface = 0;
452         int rc = 0;
453
454         *iface_list = NULL;
455         *iface_count = 0;
456
457         /*
458          * Fist pass: count and sanity check
459          */
460
461         bytes_left = buf_len;
462         p = buf;
463         while (bytes_left >= sizeof(*p)) {
464                 nb_iface++;
465                 next = le32_to_cpu(p->Next);
466                 if (!next) {
467                         bytes_left -= sizeof(*p);
468                         break;
469                 }
470                 p = (struct network_interface_info_ioctl_rsp *)((u8 *)p+next);
471                 bytes_left -= next;
472         }
473
474         if (!nb_iface) {
475                 cifs_dbg(VFS, "%s: malformed interface info\n", __func__);
476                 rc = -EINVAL;
477                 goto out;
478         }
479
480         if (bytes_left || p->Next)
481                 cifs_dbg(VFS, "%s: incomplete interface info\n", __func__);
482
483
484         /*
485          * Second pass: extract info to internal structure
486          */
487
488         *iface_list = kcalloc(nb_iface, sizeof(**iface_list), GFP_KERNEL);
489         if (!*iface_list) {
490                 rc = -ENOMEM;
491                 goto out;
492         }
493
494         info = *iface_list;
495         bytes_left = buf_len;
496         p = buf;
497         while (bytes_left >= sizeof(*p)) {
498                 info->speed = le64_to_cpu(p->LinkSpeed);
499                 info->rdma_capable = le32_to_cpu(p->Capability & RDMA_CAPABLE);
500                 info->rss_capable = le32_to_cpu(p->Capability & RSS_CAPABLE);
501
502                 cifs_dbg(FYI, "%s: adding iface %zu\n", __func__, *iface_count);
503                 cifs_dbg(FYI, "%s: speed %zu bps\n", __func__, info->speed);
504                 cifs_dbg(FYI, "%s: capabilities 0x%08x\n", __func__,
505                          le32_to_cpu(p->Capability));
506
507                 switch (p->Family) {
508                 /*
509                  * The kernel and wire socket structures have the same
510                  * layout and use network byte order but make the
511                  * conversion explicit in case either one changes.
512                  */
513                 case INTERNETWORK:
514                         addr4 = (struct sockaddr_in *)&info->sockaddr;
515                         p4 = (struct iface_info_ipv4 *)p->Buffer;
516                         addr4->sin_family = AF_INET;
517                         memcpy(&addr4->sin_addr, &p4->IPv4Address, 4);
518
519                         /* [MS-SMB2] 2.2.32.5.1.1 Clients MUST ignore these */
520                         addr4->sin_port = cpu_to_be16(CIFS_PORT);
521
522                         cifs_dbg(FYI, "%s: ipv4 %pI4\n", __func__,
523                                  &addr4->sin_addr);
524                         break;
525                 case INTERNETWORKV6:
526                         addr6 = (struct sockaddr_in6 *)&info->sockaddr;
527                         p6 = (struct iface_info_ipv6 *)p->Buffer;
528                         addr6->sin6_family = AF_INET6;
529                         memcpy(&addr6->sin6_addr, &p6->IPv6Address, 16);
530
531                         /* [MS-SMB2] 2.2.32.5.1.2 Clients MUST ignore these */
532                         addr6->sin6_flowinfo = 0;
533                         addr6->sin6_scope_id = 0;
534                         addr6->sin6_port = cpu_to_be16(CIFS_PORT);
535
536                         cifs_dbg(FYI, "%s: ipv6 %pI6\n", __func__,
537                                  &addr6->sin6_addr);
538                         break;
539                 default:
540                         cifs_dbg(VFS,
541                                  "%s: skipping unsupported socket family\n",
542                                  __func__);
543                         goto next_iface;
544                 }
545
546                 (*iface_count)++;
547                 info++;
548 next_iface:
549                 next = le32_to_cpu(p->Next);
550                 if (!next)
551                         break;
552                 p = (struct network_interface_info_ioctl_rsp *)((u8 *)p+next);
553                 bytes_left -= next;
554         }
555
556         if (!*iface_count) {
557                 rc = -EINVAL;
558                 goto out;
559         }
560
561 out:
562         if (rc) {
563                 kfree(*iface_list);
564                 *iface_count = 0;
565                 *iface_list = NULL;
566         }
567         return rc;
568 }
569
570
571 static int
572 SMB3_request_interfaces(const unsigned int xid, struct cifs_tcon *tcon)
573 {
574         int rc;
575         unsigned int ret_data_len = 0;
576         struct network_interface_info_ioctl_rsp *out_buf = NULL;
577         struct cifs_server_iface *iface_list;
578         size_t iface_count;
579         struct cifs_ses *ses = tcon->ses;
580
581         rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
582                         FSCTL_QUERY_NETWORK_INTERFACE_INFO, true /* is_fsctl */,
583                         NULL /* no data input */, 0 /* no data input */,
584                         (char **)&out_buf, &ret_data_len);
585         if (rc == -EOPNOTSUPP) {
586                 cifs_dbg(FYI,
587                          "server does not support query network interfaces\n");
588                 goto out;
589         } else if (rc != 0) {
590                 cifs_dbg(VFS, "error %d on ioctl to get interface list\n", rc);
591                 goto out;
592         }
593
594         rc = parse_server_interfaces(out_buf, ret_data_len,
595                                      &iface_list, &iface_count);
596         if (rc)
597                 goto out;
598
599         spin_lock(&ses->iface_lock);
600         kfree(ses->iface_list);
601         ses->iface_list = iface_list;
602         ses->iface_count = iface_count;
603         ses->iface_last_update = jiffies;
604         spin_unlock(&ses->iface_lock);
605
606 out:
607         kfree(out_buf);
608         return rc;
609 }
610
611 static void
612 smb2_close_cached_fid(struct kref *ref)
613 {
614         struct cached_fid *cfid = container_of(ref, struct cached_fid,
615                                                refcount);
616
617         if (cfid->is_valid) {
618                 cifs_dbg(FYI, "clear cached root file handle\n");
619                 SMB2_close(0, cfid->tcon, cfid->fid->persistent_fid,
620                            cfid->fid->volatile_fid);
621                 cfid->is_valid = false;
622                 cfid->file_all_info_is_valid = false;
623         }
624 }
625
626 void close_shroot(struct cached_fid *cfid)
627 {
628         mutex_lock(&cfid->fid_mutex);
629         kref_put(&cfid->refcount, smb2_close_cached_fid);
630         mutex_unlock(&cfid->fid_mutex);
631 }
632
633 void
634 smb2_cached_lease_break(struct work_struct *work)
635 {
636         struct cached_fid *cfid = container_of(work,
637                                 struct cached_fid, lease_break);
638
639         close_shroot(cfid);
640 }
641
642 /*
643  * Open the directory at the root of a share
644  */
645 int open_shroot(unsigned int xid, struct cifs_tcon *tcon, struct cifs_fid *pfid)
646 {
647         struct cifs_ses *ses = tcon->ses;
648         struct TCP_Server_Info *server = ses->server;
649         struct cifs_open_parms oparms;
650         struct smb2_create_rsp *o_rsp = NULL;
651         struct smb2_query_info_rsp *qi_rsp = NULL;
652         int resp_buftype[2];
653         struct smb_rqst rqst[2];
654         struct kvec rsp_iov[2];
655         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
656         struct kvec qi_iov[1];
657         int rc, flags = 0;
658         __le16 utf16_path = 0; /* Null - since an open of top of share */
659         u8 oplock = SMB2_OPLOCK_LEVEL_II;
660
661         mutex_lock(&tcon->crfid.fid_mutex);
662         if (tcon->crfid.is_valid) {
663                 cifs_dbg(FYI, "found a cached root file handle\n");
664                 memcpy(pfid, tcon->crfid.fid, sizeof(struct cifs_fid));
665                 kref_get(&tcon->crfid.refcount);
666                 mutex_unlock(&tcon->crfid.fid_mutex);
667                 return 0;
668         }
669
670         if (smb3_encryption_required(tcon))
671                 flags |= CIFS_TRANSFORM_REQ;
672
673         memset(rqst, 0, sizeof(rqst));
674         resp_buftype[0] = resp_buftype[1] = CIFS_NO_BUFFER;
675         memset(rsp_iov, 0, sizeof(rsp_iov));
676
677         /* Open */
678         memset(&open_iov, 0, sizeof(open_iov));
679         rqst[0].rq_iov = open_iov;
680         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
681
682         oparms.tcon = tcon;
683         oparms.create_options = 0;
684         oparms.desired_access = FILE_READ_ATTRIBUTES;
685         oparms.disposition = FILE_OPEN;
686         oparms.fid = pfid;
687         oparms.reconnect = false;
688
689         rc = SMB2_open_init(tcon, &rqst[0], &oplock, &oparms, &utf16_path);
690         if (rc)
691                 goto oshr_exit;
692         smb2_set_next_command(tcon, &rqst[0]);
693
694         memset(&qi_iov, 0, sizeof(qi_iov));
695         rqst[1].rq_iov = qi_iov;
696         rqst[1].rq_nvec = 1;
697
698         rc = SMB2_query_info_init(tcon, &rqst[1], COMPOUND_FID,
699                                   COMPOUND_FID, FILE_ALL_INFORMATION,
700                                   SMB2_O_INFO_FILE, 0,
701                                   sizeof(struct smb2_file_all_info) +
702                                   PATH_MAX * 2, 0, NULL);
703         if (rc)
704                 goto oshr_exit;
705
706         smb2_set_related(&rqst[1]);
707
708         rc = compound_send_recv(xid, ses, flags, 2, rqst,
709                                 resp_buftype, rsp_iov);
710         if (rc)
711                 goto oshr_exit;
712
713         o_rsp = (struct smb2_create_rsp *)rsp_iov[0].iov_base;
714         oparms.fid->persistent_fid = o_rsp->PersistentFileId;
715         oparms.fid->volatile_fid = o_rsp->VolatileFileId;
716 #ifdef CONFIG_CIFS_DEBUG2
717         oparms.fid->mid = le64_to_cpu(o_rsp->sync_hdr.MessageId);
718 #endif /* CIFS_DEBUG2 */
719
720         if (o_rsp->OplockLevel == SMB2_OPLOCK_LEVEL_LEASE)
721                 oplock = smb2_parse_lease_state(server, o_rsp,
722                                                 &oparms.fid->epoch,
723                                                 oparms.fid->lease_key);
724         else
725                 goto oshr_exit;
726
727
728         memcpy(tcon->crfid.fid, pfid, sizeof(struct cifs_fid));
729         tcon->crfid.tcon = tcon;
730         tcon->crfid.is_valid = true;
731         kref_init(&tcon->crfid.refcount);
732         kref_get(&tcon->crfid.refcount);
733
734
735         qi_rsp = (struct smb2_query_info_rsp *)rsp_iov[1].iov_base;
736         if (le32_to_cpu(qi_rsp->OutputBufferLength) < sizeof(struct smb2_file_all_info))
737                 goto oshr_exit;
738         rc = smb2_validate_and_copy_iov(
739                                 le16_to_cpu(qi_rsp->OutputBufferOffset),
740                                 sizeof(struct smb2_file_all_info),
741                                 &rsp_iov[1], sizeof(struct smb2_file_all_info),
742                                 (char *)&tcon->crfid.file_all_info);
743         if (rc)
744                 goto oshr_exit;
745         tcon->crfid.file_all_info_is_valid = 1;
746
747  oshr_exit:
748         mutex_unlock(&tcon->crfid.fid_mutex);
749         SMB2_open_free(&rqst[0]);
750         SMB2_query_info_free(&rqst[1]);
751         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
752         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
753         return rc;
754 }
755
756 static void
757 smb3_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
758 {
759         int rc;
760         __le16 srch_path = 0; /* Null - open root of share */
761         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
762         struct cifs_open_parms oparms;
763         struct cifs_fid fid;
764         bool no_cached_open = tcon->nohandlecache;
765
766         oparms.tcon = tcon;
767         oparms.desired_access = FILE_READ_ATTRIBUTES;
768         oparms.disposition = FILE_OPEN;
769         oparms.create_options = 0;
770         oparms.fid = &fid;
771         oparms.reconnect = false;
772
773         if (no_cached_open)
774                 rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL,
775                                NULL);
776         else
777                 rc = open_shroot(xid, tcon, &fid);
778
779         if (rc)
780                 return;
781
782         SMB3_request_interfaces(xid, tcon);
783
784         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
785                         FS_ATTRIBUTE_INFORMATION);
786         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
787                         FS_DEVICE_INFORMATION);
788         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
789                         FS_VOLUME_INFORMATION);
790         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
791                         FS_SECTOR_SIZE_INFORMATION); /* SMB3 specific */
792         if (no_cached_open)
793                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
794         else
795                 close_shroot(&tcon->crfid);
796
797         return;
798 }
799
800 static void
801 smb2_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
802 {
803         int rc;
804         __le16 srch_path = 0; /* Null - open root of share */
805         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
806         struct cifs_open_parms oparms;
807         struct cifs_fid fid;
808
809         oparms.tcon = tcon;
810         oparms.desired_access = FILE_READ_ATTRIBUTES;
811         oparms.disposition = FILE_OPEN;
812         oparms.create_options = 0;
813         oparms.fid = &fid;
814         oparms.reconnect = false;
815
816         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL, NULL);
817         if (rc)
818                 return;
819
820         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
821                         FS_ATTRIBUTE_INFORMATION);
822         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
823                         FS_DEVICE_INFORMATION);
824         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
825         return;
826 }
827
828 static int
829 smb2_is_path_accessible(const unsigned int xid, struct cifs_tcon *tcon,
830                         struct cifs_sb_info *cifs_sb, const char *full_path)
831 {
832         int rc;
833         __le16 *utf16_path;
834         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
835         struct cifs_open_parms oparms;
836         struct cifs_fid fid;
837
838         if ((*full_path == 0) && tcon->crfid.is_valid)
839                 return 0;
840
841         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
842         if (!utf16_path)
843                 return -ENOMEM;
844
845         oparms.tcon = tcon;
846         oparms.desired_access = FILE_READ_ATTRIBUTES;
847         oparms.disposition = FILE_OPEN;
848         if (backup_cred(cifs_sb))
849                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
850         else
851                 oparms.create_options = 0;
852         oparms.fid = &fid;
853         oparms.reconnect = false;
854
855         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL);
856         if (rc) {
857                 kfree(utf16_path);
858                 return rc;
859         }
860
861         rc = SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
862         kfree(utf16_path);
863         return rc;
864 }
865
866 static int
867 smb2_get_srv_inum(const unsigned int xid, struct cifs_tcon *tcon,
868                   struct cifs_sb_info *cifs_sb, const char *full_path,
869                   u64 *uniqueid, FILE_ALL_INFO *data)
870 {
871         *uniqueid = le64_to_cpu(data->IndexNumber);
872         return 0;
873 }
874
875 static int
876 smb2_query_file_info(const unsigned int xid, struct cifs_tcon *tcon,
877                      struct cifs_fid *fid, FILE_ALL_INFO *data)
878 {
879         int rc;
880         struct smb2_file_all_info *smb2_data;
881
882         smb2_data = kzalloc(sizeof(struct smb2_file_all_info) + PATH_MAX * 2,
883                             GFP_KERNEL);
884         if (smb2_data == NULL)
885                 return -ENOMEM;
886
887         rc = SMB2_query_info(xid, tcon, fid->persistent_fid, fid->volatile_fid,
888                              smb2_data);
889         if (!rc)
890                 move_smb2_info_to_cifs(data, smb2_data);
891         kfree(smb2_data);
892         return rc;
893 }
894
895 #ifdef CONFIG_CIFS_XATTR
896 static ssize_t
897 move_smb2_ea_to_cifs(char *dst, size_t dst_size,
898                      struct smb2_file_full_ea_info *src, size_t src_size,
899                      const unsigned char *ea_name)
900 {
901         int rc = 0;
902         unsigned int ea_name_len = ea_name ? strlen(ea_name) : 0;
903         char *name, *value;
904         size_t buf_size = dst_size;
905         size_t name_len, value_len, user_name_len;
906
907         while (src_size > 0) {
908                 name = &src->ea_data[0];
909                 name_len = (size_t)src->ea_name_length;
910                 value = &src->ea_data[src->ea_name_length + 1];
911                 value_len = (size_t)le16_to_cpu(src->ea_value_length);
912
913                 if (name_len == 0) {
914                         break;
915                 }
916
917                 if (src_size < 8 + name_len + 1 + value_len) {
918                         cifs_dbg(FYI, "EA entry goes beyond length of list\n");
919                         rc = -EIO;
920                         goto out;
921                 }
922
923                 if (ea_name) {
924                         if (ea_name_len == name_len &&
925                             memcmp(ea_name, name, name_len) == 0) {
926                                 rc = value_len;
927                                 if (dst_size == 0)
928                                         goto out;
929                                 if (dst_size < value_len) {
930                                         rc = -ERANGE;
931                                         goto out;
932                                 }
933                                 memcpy(dst, value, value_len);
934                                 goto out;
935                         }
936                 } else {
937                         /* 'user.' plus a terminating null */
938                         user_name_len = 5 + 1 + name_len;
939
940                         if (buf_size == 0) {
941                                 /* skip copy - calc size only */
942                                 rc += user_name_len;
943                         } else if (dst_size >= user_name_len) {
944                                 dst_size -= user_name_len;
945                                 memcpy(dst, "user.", 5);
946                                 dst += 5;
947                                 memcpy(dst, src->ea_data, name_len);
948                                 dst += name_len;
949                                 *dst = 0;
950                                 ++dst;
951                                 rc += user_name_len;
952                         } else {
953                                 /* stop before overrun buffer */
954                                 rc = -ERANGE;
955                                 break;
956                         }
957                 }
958
959                 if (!src->next_entry_offset)
960                         break;
961
962                 if (src_size < le32_to_cpu(src->next_entry_offset)) {
963                         /* stop before overrun buffer */
964                         rc = -ERANGE;
965                         break;
966                 }
967                 src_size -= le32_to_cpu(src->next_entry_offset);
968                 src = (void *)((char *)src +
969                                le32_to_cpu(src->next_entry_offset));
970         }
971
972         /* didn't find the named attribute */
973         if (ea_name)
974                 rc = -ENODATA;
975
976 out:
977         return (ssize_t)rc;
978 }
979
980 static ssize_t
981 smb2_query_eas(const unsigned int xid, struct cifs_tcon *tcon,
982                const unsigned char *path, const unsigned char *ea_name,
983                char *ea_data, size_t buf_size,
984                struct cifs_sb_info *cifs_sb)
985 {
986         int rc;
987         __le16 *utf16_path;
988         struct kvec rsp_iov = {NULL, 0};
989         int buftype = CIFS_NO_BUFFER;
990         struct smb2_query_info_rsp *rsp;
991         struct smb2_file_full_ea_info *info = NULL;
992
993         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
994         if (!utf16_path)
995                 return -ENOMEM;
996
997         rc = smb2_query_info_compound(xid, tcon, utf16_path,
998                                       FILE_READ_EA,
999                                       FILE_FULL_EA_INFORMATION,
1000                                       SMB2_O_INFO_FILE,
1001                                       CIFSMaxBufSize -
1002                                       MAX_SMB2_CREATE_RESPONSE_SIZE -
1003                                       MAX_SMB2_CLOSE_RESPONSE_SIZE,
1004                                       &rsp_iov, &buftype, cifs_sb);
1005         if (rc) {
1006                 /*
1007                  * If ea_name is NULL (listxattr) and there are no EAs,
1008                  * return 0 as it's not an error. Otherwise, the specified
1009                  * ea_name was not found.
1010                  */
1011                 if (!ea_name && rc == -ENODATA)
1012                         rc = 0;
1013                 goto qeas_exit;
1014         }
1015
1016         rsp = (struct smb2_query_info_rsp *)rsp_iov.iov_base;
1017         rc = smb2_validate_iov(le16_to_cpu(rsp->OutputBufferOffset),
1018                                le32_to_cpu(rsp->OutputBufferLength),
1019                                &rsp_iov,
1020                                sizeof(struct smb2_file_full_ea_info));
1021         if (rc)
1022                 goto qeas_exit;
1023
1024         info = (struct smb2_file_full_ea_info *)(
1025                         le16_to_cpu(rsp->OutputBufferOffset) + (char *)rsp);
1026         rc = move_smb2_ea_to_cifs(ea_data, buf_size, info,
1027                         le32_to_cpu(rsp->OutputBufferLength), ea_name);
1028
1029  qeas_exit:
1030         kfree(utf16_path);
1031         free_rsp_buf(buftype, rsp_iov.iov_base);
1032         return rc;
1033 }
1034
1035
1036 static int
1037 smb2_set_ea(const unsigned int xid, struct cifs_tcon *tcon,
1038             const char *path, const char *ea_name, const void *ea_value,
1039             const __u16 ea_value_len, const struct nls_table *nls_codepage,
1040             struct cifs_sb_info *cifs_sb)
1041 {
1042         struct cifs_ses *ses = tcon->ses;
1043         __le16 *utf16_path = NULL;
1044         int ea_name_len = strlen(ea_name);
1045         int flags = 0;
1046         int len;
1047         struct smb_rqst rqst[3];
1048         int resp_buftype[3];
1049         struct kvec rsp_iov[3];
1050         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
1051         struct cifs_open_parms oparms;
1052         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1053         struct cifs_fid fid;
1054         struct kvec si_iov[SMB2_SET_INFO_IOV_SIZE];
1055         unsigned int size[1];
1056         void *data[1];
1057         struct smb2_file_full_ea_info *ea = NULL;
1058         struct kvec close_iov[1];
1059         int rc;
1060
1061         if (smb3_encryption_required(tcon))
1062                 flags |= CIFS_TRANSFORM_REQ;
1063
1064         if (ea_name_len > 255)
1065                 return -EINVAL;
1066
1067         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1068         if (!utf16_path)
1069                 return -ENOMEM;
1070
1071         memset(rqst, 0, sizeof(rqst));
1072         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
1073         memset(rsp_iov, 0, sizeof(rsp_iov));
1074
1075         if (ses->server->ops->query_all_EAs) {
1076                 if (!ea_value) {
1077                         rc = ses->server->ops->query_all_EAs(xid, tcon, path,
1078                                                              ea_name, NULL, 0,
1079                                                              cifs_sb);
1080                         if (rc == -ENODATA)
1081                                 goto sea_exit;
1082                 }
1083         }
1084
1085         /* Open */
1086         memset(&open_iov, 0, sizeof(open_iov));
1087         rqst[0].rq_iov = open_iov;
1088         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
1089
1090         memset(&oparms, 0, sizeof(oparms));
1091         oparms.tcon = tcon;
1092         oparms.desired_access = FILE_WRITE_EA;
1093         oparms.disposition = FILE_OPEN;
1094         if (backup_cred(cifs_sb))
1095                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
1096         else
1097                 oparms.create_options = 0;
1098         oparms.fid = &fid;
1099         oparms.reconnect = false;
1100
1101         rc = SMB2_open_init(tcon, &rqst[0], &oplock, &oparms, utf16_path);
1102         if (rc)
1103                 goto sea_exit;
1104         smb2_set_next_command(tcon, &rqst[0]);
1105
1106
1107         /* Set Info */
1108         memset(&si_iov, 0, sizeof(si_iov));
1109         rqst[1].rq_iov = si_iov;
1110         rqst[1].rq_nvec = 1;
1111
1112         len = sizeof(ea) + ea_name_len + ea_value_len + 1;
1113         ea = kzalloc(len, GFP_KERNEL);
1114         if (ea == NULL) {
1115                 rc = -ENOMEM;
1116                 goto sea_exit;
1117         }
1118
1119         ea->ea_name_length = ea_name_len;
1120         ea->ea_value_length = cpu_to_le16(ea_value_len);
1121         memcpy(ea->ea_data, ea_name, ea_name_len + 1);
1122         memcpy(ea->ea_data + ea_name_len + 1, ea_value, ea_value_len);
1123
1124         size[0] = len;
1125         data[0] = ea;
1126
1127         rc = SMB2_set_info_init(tcon, &rqst[1], COMPOUND_FID,
1128                                 COMPOUND_FID, current->tgid,
1129                                 FILE_FULL_EA_INFORMATION,
1130                                 SMB2_O_INFO_FILE, 0, data, size);
1131         smb2_set_next_command(tcon, &rqst[1]);
1132         smb2_set_related(&rqst[1]);
1133
1134
1135         /* Close */
1136         memset(&close_iov, 0, sizeof(close_iov));
1137         rqst[2].rq_iov = close_iov;
1138         rqst[2].rq_nvec = 1;
1139         rc = SMB2_close_init(tcon, &rqst[2], COMPOUND_FID, COMPOUND_FID);
1140         smb2_set_related(&rqst[2]);
1141
1142         rc = compound_send_recv(xid, ses, flags, 3, rqst,
1143                                 resp_buftype, rsp_iov);
1144
1145  sea_exit:
1146         kfree(ea);
1147         kfree(utf16_path);
1148         SMB2_open_free(&rqst[0]);
1149         SMB2_set_info_free(&rqst[1]);
1150         SMB2_close_free(&rqst[2]);
1151         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
1152         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
1153         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
1154         return rc;
1155 }
1156 #endif
1157
1158 static bool
1159 smb2_can_echo(struct TCP_Server_Info *server)
1160 {
1161         return server->echoes;
1162 }
1163
1164 static void
1165 smb2_clear_stats(struct cifs_tcon *tcon)
1166 {
1167         int i;
1168         for (i = 0; i < NUMBER_OF_SMB2_COMMANDS; i++) {
1169                 atomic_set(&tcon->stats.smb2_stats.smb2_com_sent[i], 0);
1170                 atomic_set(&tcon->stats.smb2_stats.smb2_com_failed[i], 0);
1171         }
1172 }
1173
1174 static void
1175 smb2_dump_share_caps(struct seq_file *m, struct cifs_tcon *tcon)
1176 {
1177         seq_puts(m, "\n\tShare Capabilities:");
1178         if (tcon->capabilities & SMB2_SHARE_CAP_DFS)
1179                 seq_puts(m, " DFS,");
1180         if (tcon->capabilities & SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY)
1181                 seq_puts(m, " CONTINUOUS AVAILABILITY,");
1182         if (tcon->capabilities & SMB2_SHARE_CAP_SCALEOUT)
1183                 seq_puts(m, " SCALEOUT,");
1184         if (tcon->capabilities & SMB2_SHARE_CAP_CLUSTER)
1185                 seq_puts(m, " CLUSTER,");
1186         if (tcon->capabilities & SMB2_SHARE_CAP_ASYMMETRIC)
1187                 seq_puts(m, " ASYMMETRIC,");
1188         if (tcon->capabilities == 0)
1189                 seq_puts(m, " None");
1190         if (tcon->ss_flags & SSINFO_FLAGS_ALIGNED_DEVICE)
1191                 seq_puts(m, " Aligned,");
1192         if (tcon->ss_flags & SSINFO_FLAGS_PARTITION_ALIGNED_ON_DEVICE)
1193                 seq_puts(m, " Partition Aligned,");
1194         if (tcon->ss_flags & SSINFO_FLAGS_NO_SEEK_PENALTY)
1195                 seq_puts(m, " SSD,");
1196         if (tcon->ss_flags & SSINFO_FLAGS_TRIM_ENABLED)
1197                 seq_puts(m, " TRIM-support,");
1198
1199         seq_printf(m, "\tShare Flags: 0x%x", tcon->share_flags);
1200         seq_printf(m, "\n\ttid: 0x%x", tcon->tid);
1201         if (tcon->perf_sector_size)
1202                 seq_printf(m, "\tOptimal sector size: 0x%x",
1203                            tcon->perf_sector_size);
1204         seq_printf(m, "\tMaximal Access: 0x%x", tcon->maximal_access);
1205 }
1206
1207 static void
1208 smb2_print_stats(struct seq_file *m, struct cifs_tcon *tcon)
1209 {
1210         atomic_t *sent = tcon->stats.smb2_stats.smb2_com_sent;
1211         atomic_t *failed = tcon->stats.smb2_stats.smb2_com_failed;
1212
1213         /*
1214          *  Can't display SMB2_NEGOTIATE, SESSION_SETUP, LOGOFF, CANCEL and ECHO
1215          *  totals (requests sent) since those SMBs are per-session not per tcon
1216          */
1217         seq_printf(m, "\nBytes read: %llu  Bytes written: %llu",
1218                    (long long)(tcon->bytes_read),
1219                    (long long)(tcon->bytes_written));
1220         seq_printf(m, "\nOpen files: %d total (local), %d open on server",
1221                    atomic_read(&tcon->num_local_opens),
1222                    atomic_read(&tcon->num_remote_opens));
1223         seq_printf(m, "\nTreeConnects: %d total %d failed",
1224                    atomic_read(&sent[SMB2_TREE_CONNECT_HE]),
1225                    atomic_read(&failed[SMB2_TREE_CONNECT_HE]));
1226         seq_printf(m, "\nTreeDisconnects: %d total %d failed",
1227                    atomic_read(&sent[SMB2_TREE_DISCONNECT_HE]),
1228                    atomic_read(&failed[SMB2_TREE_DISCONNECT_HE]));
1229         seq_printf(m, "\nCreates: %d total %d failed",
1230                    atomic_read(&sent[SMB2_CREATE_HE]),
1231                    atomic_read(&failed[SMB2_CREATE_HE]));
1232         seq_printf(m, "\nCloses: %d total %d failed",
1233                    atomic_read(&sent[SMB2_CLOSE_HE]),
1234                    atomic_read(&failed[SMB2_CLOSE_HE]));
1235         seq_printf(m, "\nFlushes: %d total %d failed",
1236                    atomic_read(&sent[SMB2_FLUSH_HE]),
1237                    atomic_read(&failed[SMB2_FLUSH_HE]));
1238         seq_printf(m, "\nReads: %d total %d failed",
1239                    atomic_read(&sent[SMB2_READ_HE]),
1240                    atomic_read(&failed[SMB2_READ_HE]));
1241         seq_printf(m, "\nWrites: %d total %d failed",
1242                    atomic_read(&sent[SMB2_WRITE_HE]),
1243                    atomic_read(&failed[SMB2_WRITE_HE]));
1244         seq_printf(m, "\nLocks: %d total %d failed",
1245                    atomic_read(&sent[SMB2_LOCK_HE]),
1246                    atomic_read(&failed[SMB2_LOCK_HE]));
1247         seq_printf(m, "\nIOCTLs: %d total %d failed",
1248                    atomic_read(&sent[SMB2_IOCTL_HE]),
1249                    atomic_read(&failed[SMB2_IOCTL_HE]));
1250         seq_printf(m, "\nQueryDirectories: %d total %d failed",
1251                    atomic_read(&sent[SMB2_QUERY_DIRECTORY_HE]),
1252                    atomic_read(&failed[SMB2_QUERY_DIRECTORY_HE]));
1253         seq_printf(m, "\nChangeNotifies: %d total %d failed",
1254                    atomic_read(&sent[SMB2_CHANGE_NOTIFY_HE]),
1255                    atomic_read(&failed[SMB2_CHANGE_NOTIFY_HE]));
1256         seq_printf(m, "\nQueryInfos: %d total %d failed",
1257                    atomic_read(&sent[SMB2_QUERY_INFO_HE]),
1258                    atomic_read(&failed[SMB2_QUERY_INFO_HE]));
1259         seq_printf(m, "\nSetInfos: %d total %d failed",
1260                    atomic_read(&sent[SMB2_SET_INFO_HE]),
1261                    atomic_read(&failed[SMB2_SET_INFO_HE]));
1262         seq_printf(m, "\nOplockBreaks: %d sent %d failed",
1263                    atomic_read(&sent[SMB2_OPLOCK_BREAK_HE]),
1264                    atomic_read(&failed[SMB2_OPLOCK_BREAK_HE]));
1265 }
1266
1267 static void
1268 smb2_set_fid(struct cifsFileInfo *cfile, struct cifs_fid *fid, __u32 oplock)
1269 {
1270         struct cifsInodeInfo *cinode = CIFS_I(d_inode(cfile->dentry));
1271         struct TCP_Server_Info *server = tlink_tcon(cfile->tlink)->ses->server;
1272
1273         cfile->fid.persistent_fid = fid->persistent_fid;
1274         cfile->fid.volatile_fid = fid->volatile_fid;
1275 #ifdef CONFIG_CIFS_DEBUG2
1276         cfile->fid.mid = fid->mid;
1277 #endif /* CIFS_DEBUG2 */
1278         server->ops->set_oplock_level(cinode, oplock, fid->epoch,
1279                                       &fid->purge_cache);
1280         cinode->can_cache_brlcks = CIFS_CACHE_WRITE(cinode);
1281         memcpy(cfile->fid.create_guid, fid->create_guid, 16);
1282 }
1283
1284 static void
1285 smb2_close_file(const unsigned int xid, struct cifs_tcon *tcon,
1286                 struct cifs_fid *fid)
1287 {
1288         SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1289 }
1290
1291 static int
1292 SMB2_request_res_key(const unsigned int xid, struct cifs_tcon *tcon,
1293                      u64 persistent_fid, u64 volatile_fid,
1294                      struct copychunk_ioctl *pcchunk)
1295 {
1296         int rc;
1297         unsigned int ret_data_len;
1298         struct resume_key_req *res_key;
1299
1300         rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
1301                         FSCTL_SRV_REQUEST_RESUME_KEY, true /* is_fsctl */,
1302                         NULL, 0 /* no input */,
1303                         (char **)&res_key, &ret_data_len);
1304
1305         if (rc) {
1306                 cifs_dbg(VFS, "refcpy ioctl error %d getting resume key\n", rc);
1307                 goto req_res_key_exit;
1308         }
1309         if (ret_data_len < sizeof(struct resume_key_req)) {
1310                 cifs_dbg(VFS, "Invalid refcopy resume key length\n");
1311                 rc = -EINVAL;
1312                 goto req_res_key_exit;
1313         }
1314         memcpy(pcchunk->SourceKey, res_key->ResumeKey, COPY_CHUNK_RES_KEY_SIZE);
1315
1316 req_res_key_exit:
1317         kfree(res_key);
1318         return rc;
1319 }
1320
1321 static int
1322 smb2_ioctl_query_info(const unsigned int xid,
1323                       struct cifs_tcon *tcon,
1324                       __le16 *path, int is_dir,
1325                       unsigned long p)
1326 {
1327         struct cifs_ses *ses = tcon->ses;
1328         char __user *arg = (char __user *)p;
1329         struct smb_query_info qi;
1330         struct smb_query_info __user *pqi;
1331         int rc = 0;
1332         int flags = 0;
1333         struct smb2_query_info_rsp *rsp = NULL;
1334         void *buffer = NULL;
1335         struct smb_rqst rqst[3];
1336         int resp_buftype[3];
1337         struct kvec rsp_iov[3];
1338         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
1339         struct cifs_open_parms oparms;
1340         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1341         struct cifs_fid fid;
1342         struct kvec qi_iov[1];
1343         struct kvec close_iov[1];
1344
1345         memset(rqst, 0, sizeof(rqst));
1346         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
1347         memset(rsp_iov, 0, sizeof(rsp_iov));
1348
1349         if (copy_from_user(&qi, arg, sizeof(struct smb_query_info)))
1350                 return -EFAULT;
1351
1352         if (qi.output_buffer_length > 1024)
1353                 return -EINVAL;
1354
1355         if (!ses || !(ses->server))
1356                 return -EIO;
1357
1358         if (smb3_encryption_required(tcon))
1359                 flags |= CIFS_TRANSFORM_REQ;
1360
1361         buffer = kmalloc(qi.output_buffer_length, GFP_KERNEL);
1362         if (buffer == NULL)
1363                 return -ENOMEM;
1364
1365         if (copy_from_user(buffer, arg + sizeof(struct smb_query_info),
1366                            qi.output_buffer_length)) {
1367                 rc = -EFAULT;
1368                 goto iqinf_exit;
1369         }
1370
1371         /* Open */
1372         memset(&open_iov, 0, sizeof(open_iov));
1373         rqst[0].rq_iov = open_iov;
1374         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
1375
1376         memset(&oparms, 0, sizeof(oparms));
1377         oparms.tcon = tcon;
1378         oparms.desired_access = FILE_READ_ATTRIBUTES | READ_CONTROL;
1379         oparms.disposition = FILE_OPEN;
1380         if (is_dir)
1381                 oparms.create_options = CREATE_NOT_FILE;
1382         else
1383                 oparms.create_options = CREATE_NOT_DIR;
1384         oparms.fid = &fid;
1385         oparms.reconnect = false;
1386
1387         rc = SMB2_open_init(tcon, &rqst[0], &oplock, &oparms, path);
1388         if (rc)
1389                 goto iqinf_exit;
1390         smb2_set_next_command(tcon, &rqst[0]);
1391
1392         /* Query */
1393         memset(&qi_iov, 0, sizeof(qi_iov));
1394         rqst[1].rq_iov = qi_iov;
1395         rqst[1].rq_nvec = 1;
1396
1397         rc = SMB2_query_info_init(tcon, &rqst[1], COMPOUND_FID, COMPOUND_FID,
1398                                   qi.file_info_class, qi.info_type,
1399                                   qi.additional_information,
1400                                   qi.input_buffer_length,
1401                                   qi.output_buffer_length, buffer);
1402         if (rc)
1403                 goto iqinf_exit;
1404         smb2_set_next_command(tcon, &rqst[1]);
1405         smb2_set_related(&rqst[1]);
1406
1407         /* Close */
1408         memset(&close_iov, 0, sizeof(close_iov));
1409         rqst[2].rq_iov = close_iov;
1410         rqst[2].rq_nvec = 1;
1411
1412         rc = SMB2_close_init(tcon, &rqst[2], COMPOUND_FID, COMPOUND_FID);
1413         if (rc)
1414                 goto iqinf_exit;
1415         smb2_set_related(&rqst[2]);
1416
1417         rc = compound_send_recv(xid, ses, flags, 3, rqst,
1418                                 resp_buftype, rsp_iov);
1419         if (rc)
1420                 goto iqinf_exit;
1421         pqi = (struct smb_query_info __user *)arg;
1422         rsp = (struct smb2_query_info_rsp *)rsp_iov[1].iov_base;
1423         if (le32_to_cpu(rsp->OutputBufferLength) < qi.input_buffer_length)
1424                 qi.input_buffer_length = le32_to_cpu(rsp->OutputBufferLength);
1425         if (copy_to_user(&pqi->input_buffer_length, &qi.input_buffer_length,
1426                          sizeof(qi.input_buffer_length))) {
1427                 rc = -EFAULT;
1428                 goto iqinf_exit;
1429         }
1430         if (copy_to_user(pqi + 1, rsp->Buffer, qi.input_buffer_length)) {
1431                 rc = -EFAULT;
1432                 goto iqinf_exit;
1433         }
1434
1435  iqinf_exit:
1436         kfree(buffer);
1437         SMB2_open_free(&rqst[0]);
1438         SMB2_query_info_free(&rqst[1]);
1439         SMB2_close_free(&rqst[2]);
1440         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
1441         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
1442         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
1443         return rc;
1444 }
1445
1446 static ssize_t
1447 smb2_copychunk_range(const unsigned int xid,
1448                         struct cifsFileInfo *srcfile,
1449                         struct cifsFileInfo *trgtfile, u64 src_off,
1450                         u64 len, u64 dest_off)
1451 {
1452         int rc;
1453         unsigned int ret_data_len;
1454         struct copychunk_ioctl *pcchunk;
1455         struct copychunk_ioctl_rsp *retbuf = NULL;
1456         struct cifs_tcon *tcon;
1457         int chunks_copied = 0;
1458         bool chunk_sizes_updated = false;
1459         ssize_t bytes_written, total_bytes_written = 0;
1460
1461         pcchunk = kmalloc(sizeof(struct copychunk_ioctl), GFP_KERNEL);
1462
1463         if (pcchunk == NULL)
1464                 return -ENOMEM;
1465
1466         cifs_dbg(FYI, "in smb2_copychunk_range - about to call request res key\n");
1467         /* Request a key from the server to identify the source of the copy */
1468         rc = SMB2_request_res_key(xid, tlink_tcon(srcfile->tlink),
1469                                 srcfile->fid.persistent_fid,
1470                                 srcfile->fid.volatile_fid, pcchunk);
1471
1472         /* Note: request_res_key sets res_key null only if rc !=0 */
1473         if (rc)
1474                 goto cchunk_out;
1475
1476         /* For now array only one chunk long, will make more flexible later */
1477         pcchunk->ChunkCount = cpu_to_le32(1);
1478         pcchunk->Reserved = 0;
1479         pcchunk->Reserved2 = 0;
1480
1481         tcon = tlink_tcon(trgtfile->tlink);
1482
1483         while (len > 0) {
1484                 pcchunk->SourceOffset = cpu_to_le64(src_off);
1485                 pcchunk->TargetOffset = cpu_to_le64(dest_off);
1486                 pcchunk->Length =
1487                         cpu_to_le32(min_t(u32, len, tcon->max_bytes_chunk));
1488
1489                 /* Request server copy to target from src identified by key */
1490                 rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
1491                         trgtfile->fid.volatile_fid, FSCTL_SRV_COPYCHUNK_WRITE,
1492                         true /* is_fsctl */, (char *)pcchunk,
1493                         sizeof(struct copychunk_ioctl), (char **)&retbuf,
1494                         &ret_data_len);
1495                 if (rc == 0) {
1496                         if (ret_data_len !=
1497                                         sizeof(struct copychunk_ioctl_rsp)) {
1498                                 cifs_dbg(VFS, "invalid cchunk response size\n");
1499                                 rc = -EIO;
1500                                 goto cchunk_out;
1501                         }
1502                         if (retbuf->TotalBytesWritten == 0) {
1503                                 cifs_dbg(FYI, "no bytes copied\n");
1504                                 rc = -EIO;
1505                                 goto cchunk_out;
1506                         }
1507                         /*
1508                          * Check if server claimed to write more than we asked
1509                          */
1510                         if (le32_to_cpu(retbuf->TotalBytesWritten) >
1511                             le32_to_cpu(pcchunk->Length)) {
1512                                 cifs_dbg(VFS, "invalid copy chunk response\n");
1513                                 rc = -EIO;
1514                                 goto cchunk_out;
1515                         }
1516                         if (le32_to_cpu(retbuf->ChunksWritten) != 1) {
1517                                 cifs_dbg(VFS, "invalid num chunks written\n");
1518                                 rc = -EIO;
1519                                 goto cchunk_out;
1520                         }
1521                         chunks_copied++;
1522
1523                         bytes_written = le32_to_cpu(retbuf->TotalBytesWritten);
1524                         src_off += bytes_written;
1525                         dest_off += bytes_written;
1526                         len -= bytes_written;
1527                         total_bytes_written += bytes_written;
1528
1529                         cifs_dbg(FYI, "Chunks %d PartialChunk %d Total %zu\n",
1530                                 le32_to_cpu(retbuf->ChunksWritten),
1531                                 le32_to_cpu(retbuf->ChunkBytesWritten),
1532                                 bytes_written);
1533                 } else if (rc == -EINVAL) {
1534                         if (ret_data_len != sizeof(struct copychunk_ioctl_rsp))
1535                                 goto cchunk_out;
1536
1537                         cifs_dbg(FYI, "MaxChunks %d BytesChunk %d MaxCopy %d\n",
1538                                 le32_to_cpu(retbuf->ChunksWritten),
1539                                 le32_to_cpu(retbuf->ChunkBytesWritten),
1540                                 le32_to_cpu(retbuf->TotalBytesWritten));
1541
1542                         /*
1543                          * Check if this is the first request using these sizes,
1544                          * (ie check if copy succeed once with original sizes
1545                          * and check if the server gave us different sizes after
1546                          * we already updated max sizes on previous request).
1547                          * if not then why is the server returning an error now
1548                          */
1549                         if ((chunks_copied != 0) || chunk_sizes_updated)
1550                                 goto cchunk_out;
1551
1552                         /* Check that server is not asking us to grow size */
1553                         if (le32_to_cpu(retbuf->ChunkBytesWritten) <
1554                                         tcon->max_bytes_chunk)
1555                                 tcon->max_bytes_chunk =
1556                                         le32_to_cpu(retbuf->ChunkBytesWritten);
1557                         else
1558                                 goto cchunk_out; /* server gave us bogus size */
1559
1560                         /* No need to change MaxChunks since already set to 1 */
1561                         chunk_sizes_updated = true;
1562                 } else
1563                         goto cchunk_out;
1564         }
1565
1566 cchunk_out:
1567         kfree(pcchunk);
1568         kfree(retbuf);
1569         if (rc)
1570                 return rc;
1571         else
1572                 return total_bytes_written;
1573 }
1574
1575 static int
1576 smb2_flush_file(const unsigned int xid, struct cifs_tcon *tcon,
1577                 struct cifs_fid *fid)
1578 {
1579         return SMB2_flush(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1580 }
1581
1582 static unsigned int
1583 smb2_read_data_offset(char *buf)
1584 {
1585         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1586         return rsp->DataOffset;
1587 }
1588
1589 static unsigned int
1590 smb2_read_data_length(char *buf, bool in_remaining)
1591 {
1592         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1593
1594         if (in_remaining)
1595                 return le32_to_cpu(rsp->DataRemaining);
1596
1597         return le32_to_cpu(rsp->DataLength);
1598 }
1599
1600
1601 static int
1602 smb2_sync_read(const unsigned int xid, struct cifs_fid *pfid,
1603                struct cifs_io_parms *parms, unsigned int *bytes_read,
1604                char **buf, int *buf_type)
1605 {
1606         parms->persistent_fid = pfid->persistent_fid;
1607         parms->volatile_fid = pfid->volatile_fid;
1608         return SMB2_read(xid, parms, bytes_read, buf, buf_type);
1609 }
1610
1611 static int
1612 smb2_sync_write(const unsigned int xid, struct cifs_fid *pfid,
1613                 struct cifs_io_parms *parms, unsigned int *written,
1614                 struct kvec *iov, unsigned long nr_segs)
1615 {
1616
1617         parms->persistent_fid = pfid->persistent_fid;
1618         parms->volatile_fid = pfid->volatile_fid;
1619         return SMB2_write(xid, parms, written, iov, nr_segs);
1620 }
1621
1622 /* Set or clear the SPARSE_FILE attribute based on value passed in setsparse */
1623 static bool smb2_set_sparse(const unsigned int xid, struct cifs_tcon *tcon,
1624                 struct cifsFileInfo *cfile, struct inode *inode, __u8 setsparse)
1625 {
1626         struct cifsInodeInfo *cifsi;
1627         int rc;
1628
1629         cifsi = CIFS_I(inode);
1630
1631         /* if file already sparse don't bother setting sparse again */
1632         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && setsparse)
1633                 return true; /* already sparse */
1634
1635         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && !setsparse)
1636                 return true; /* already not sparse */
1637
1638         /*
1639          * Can't check for sparse support on share the usual way via the
1640          * FS attribute info (FILE_SUPPORTS_SPARSE_FILES) on the share
1641          * since Samba server doesn't set the flag on the share, yet
1642          * supports the set sparse FSCTL and returns sparse correctly
1643          * in the file attributes. If we fail setting sparse though we
1644          * mark that server does not support sparse files for this share
1645          * to avoid repeatedly sending the unsupported fsctl to server
1646          * if the file is repeatedly extended.
1647          */
1648         if (tcon->broken_sparse_sup)
1649                 return false;
1650
1651         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1652                         cfile->fid.volatile_fid, FSCTL_SET_SPARSE,
1653                         true /* is_fctl */,
1654                         &setsparse, 1, NULL, NULL);
1655         if (rc) {
1656                 tcon->broken_sparse_sup = true;
1657                 cifs_dbg(FYI, "set sparse rc = %d\n", rc);
1658                 return false;
1659         }
1660
1661         if (setsparse)
1662                 cifsi->cifsAttrs |= FILE_ATTRIBUTE_SPARSE_FILE;
1663         else
1664                 cifsi->cifsAttrs &= (~FILE_ATTRIBUTE_SPARSE_FILE);
1665
1666         return true;
1667 }
1668
1669 static int
1670 smb2_set_file_size(const unsigned int xid, struct cifs_tcon *tcon,
1671                    struct cifsFileInfo *cfile, __u64 size, bool set_alloc)
1672 {
1673         __le64 eof = cpu_to_le64(size);
1674         struct inode *inode;
1675
1676         /*
1677          * If extending file more than one page make sparse. Many Linux fs
1678          * make files sparse by default when extending via ftruncate
1679          */
1680         inode = d_inode(cfile->dentry);
1681
1682         if (!set_alloc && (size > inode->i_size + 8192)) {
1683                 __u8 set_sparse = 1;
1684
1685                 /* whether set sparse succeeds or not, extend the file */
1686                 smb2_set_sparse(xid, tcon, cfile, inode, set_sparse);
1687         }
1688
1689         return SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
1690                             cfile->fid.volatile_fid, cfile->pid, &eof);
1691 }
1692
1693 static int
1694 smb2_duplicate_extents(const unsigned int xid,
1695                         struct cifsFileInfo *srcfile,
1696                         struct cifsFileInfo *trgtfile, u64 src_off,
1697                         u64 len, u64 dest_off)
1698 {
1699         int rc;
1700         unsigned int ret_data_len;
1701         struct duplicate_extents_to_file dup_ext_buf;
1702         struct cifs_tcon *tcon = tlink_tcon(trgtfile->tlink);
1703
1704         /* server fileays advertise duplicate extent support with this flag */
1705         if ((le32_to_cpu(tcon->fsAttrInfo.Attributes) &
1706              FILE_SUPPORTS_BLOCK_REFCOUNTING) == 0)
1707                 return -EOPNOTSUPP;
1708
1709         dup_ext_buf.VolatileFileHandle = srcfile->fid.volatile_fid;
1710         dup_ext_buf.PersistentFileHandle = srcfile->fid.persistent_fid;
1711         dup_ext_buf.SourceFileOffset = cpu_to_le64(src_off);
1712         dup_ext_buf.TargetFileOffset = cpu_to_le64(dest_off);
1713         dup_ext_buf.ByteCount = cpu_to_le64(len);
1714         cifs_dbg(FYI, "duplicate extents: src off %lld dst off %lld len %lld",
1715                 src_off, dest_off, len);
1716
1717         rc = smb2_set_file_size(xid, tcon, trgtfile, dest_off + len, false);
1718         if (rc)
1719                 goto duplicate_extents_out;
1720
1721         rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
1722                         trgtfile->fid.volatile_fid,
1723                         FSCTL_DUPLICATE_EXTENTS_TO_FILE,
1724                         true /* is_fsctl */,
1725                         (char *)&dup_ext_buf,
1726                         sizeof(struct duplicate_extents_to_file),
1727                         NULL,
1728                         &ret_data_len);
1729
1730         if (ret_data_len > 0)
1731                 cifs_dbg(FYI, "non-zero response length in duplicate extents");
1732
1733 duplicate_extents_out:
1734         return rc;
1735 }
1736
1737 static int
1738 smb2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
1739                    struct cifsFileInfo *cfile)
1740 {
1741         return SMB2_set_compression(xid, tcon, cfile->fid.persistent_fid,
1742                             cfile->fid.volatile_fid);
1743 }
1744
1745 static int
1746 smb3_set_integrity(const unsigned int xid, struct cifs_tcon *tcon,
1747                    struct cifsFileInfo *cfile)
1748 {
1749         struct fsctl_set_integrity_information_req integr_info;
1750         unsigned int ret_data_len;
1751
1752         integr_info.ChecksumAlgorithm = cpu_to_le16(CHECKSUM_TYPE_UNCHANGED);
1753         integr_info.Flags = 0;
1754         integr_info.Reserved = 0;
1755
1756         return SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1757                         cfile->fid.volatile_fid,
1758                         FSCTL_SET_INTEGRITY_INFORMATION,
1759                         true /* is_fsctl */,
1760                         (char *)&integr_info,
1761                         sizeof(struct fsctl_set_integrity_information_req),
1762                         NULL,
1763                         &ret_data_len);
1764
1765 }
1766
1767 /* GMT Token is @GMT-YYYY.MM.DD-HH.MM.SS Unicode which is 48 bytes + null */
1768 #define GMT_TOKEN_SIZE 50
1769
1770 /*
1771  * Input buffer contains (empty) struct smb_snapshot array with size filled in
1772  * For output see struct SRV_SNAPSHOT_ARRAY in MS-SMB2 section 2.2.32.2
1773  */
1774 static int
1775 smb3_enum_snapshots(const unsigned int xid, struct cifs_tcon *tcon,
1776                    struct cifsFileInfo *cfile, void __user *ioc_buf)
1777 {
1778         char *retbuf = NULL;
1779         unsigned int ret_data_len = 0;
1780         int rc;
1781         struct smb_snapshot_array snapshot_in;
1782
1783         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1784                         cfile->fid.volatile_fid,
1785                         FSCTL_SRV_ENUMERATE_SNAPSHOTS,
1786                         true /* is_fsctl */,
1787                         NULL, 0 /* no input data */,
1788                         (char **)&retbuf,
1789                         &ret_data_len);
1790         cifs_dbg(FYI, "enum snaphots ioctl returned %d and ret buflen is %d\n",
1791                         rc, ret_data_len);
1792         if (rc)
1793                 return rc;
1794
1795         if (ret_data_len && (ioc_buf != NULL) && (retbuf != NULL)) {
1796                 /* Fixup buffer */
1797                 if (copy_from_user(&snapshot_in, ioc_buf,
1798                     sizeof(struct smb_snapshot_array))) {
1799                         rc = -EFAULT;
1800                         kfree(retbuf);
1801                         return rc;
1802                 }
1803
1804                 /*
1805                  * Check for min size, ie not large enough to fit even one GMT
1806                  * token (snapshot).  On the first ioctl some users may pass in
1807                  * smaller size (or zero) to simply get the size of the array
1808                  * so the user space caller can allocate sufficient memory
1809                  * and retry the ioctl again with larger array size sufficient
1810                  * to hold all of the snapshot GMT tokens on the second try.
1811                  */
1812                 if (snapshot_in.snapshot_array_size < GMT_TOKEN_SIZE)
1813                         ret_data_len = sizeof(struct smb_snapshot_array);
1814
1815                 /*
1816                  * We return struct SRV_SNAPSHOT_ARRAY, followed by
1817                  * the snapshot array (of 50 byte GMT tokens) each
1818                  * representing an available previous version of the data
1819                  */
1820                 if (ret_data_len > (snapshot_in.snapshot_array_size +
1821                                         sizeof(struct smb_snapshot_array)))
1822                         ret_data_len = snapshot_in.snapshot_array_size +
1823                                         sizeof(struct smb_snapshot_array);
1824
1825                 if (copy_to_user(ioc_buf, retbuf, ret_data_len))
1826                         rc = -EFAULT;
1827         }
1828
1829         kfree(retbuf);
1830         return rc;
1831 }
1832
1833 static int
1834 smb2_query_dir_first(const unsigned int xid, struct cifs_tcon *tcon,
1835                      const char *path, struct cifs_sb_info *cifs_sb,
1836                      struct cifs_fid *fid, __u16 search_flags,
1837                      struct cifs_search_info *srch_inf)
1838 {
1839         __le16 *utf16_path;
1840         int rc;
1841         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1842         struct cifs_open_parms oparms;
1843
1844         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1845         if (!utf16_path)
1846                 return -ENOMEM;
1847
1848         oparms.tcon = tcon;
1849         oparms.desired_access = FILE_READ_ATTRIBUTES | FILE_READ_DATA;
1850         oparms.disposition = FILE_OPEN;
1851         if (backup_cred(cifs_sb))
1852                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
1853         else
1854                 oparms.create_options = 0;
1855         oparms.fid = fid;
1856         oparms.reconnect = false;
1857
1858         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL);
1859         kfree(utf16_path);
1860         if (rc) {
1861                 cifs_dbg(FYI, "open dir failed rc=%d\n", rc);
1862                 return rc;
1863         }
1864
1865         srch_inf->entries_in_buffer = 0;
1866         srch_inf->index_of_last_entry = 2;
1867
1868         rc = SMB2_query_directory(xid, tcon, fid->persistent_fid,
1869                                   fid->volatile_fid, 0, srch_inf);
1870         if (rc) {
1871                 cifs_dbg(FYI, "query directory failed rc=%d\n", rc);
1872                 SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1873         }
1874         return rc;
1875 }
1876
1877 static int
1878 smb2_query_dir_next(const unsigned int xid, struct cifs_tcon *tcon,
1879                     struct cifs_fid *fid, __u16 search_flags,
1880                     struct cifs_search_info *srch_inf)
1881 {
1882         return SMB2_query_directory(xid, tcon, fid->persistent_fid,
1883                                     fid->volatile_fid, 0, srch_inf);
1884 }
1885
1886 static int
1887 smb2_close_dir(const unsigned int xid, struct cifs_tcon *tcon,
1888                struct cifs_fid *fid)
1889 {
1890         return SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1891 }
1892
1893 /*
1894 * If we negotiate SMB2 protocol and get STATUS_PENDING - update
1895 * the number of credits and return true. Otherwise - return false.
1896 */
1897 static bool
1898 smb2_is_status_pending(char *buf, struct TCP_Server_Info *server)
1899 {
1900         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
1901
1902         if (shdr->Status != STATUS_PENDING)
1903                 return false;
1904
1905         if (shdr->CreditRequest) {
1906                 spin_lock(&server->req_lock);
1907                 server->credits += le16_to_cpu(shdr->CreditRequest);
1908                 spin_unlock(&server->req_lock);
1909                 wake_up(&server->request_q);
1910         }
1911
1912         return true;
1913 }
1914
1915 static bool
1916 smb2_is_session_expired(char *buf)
1917 {
1918         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
1919
1920         if (shdr->Status != STATUS_NETWORK_SESSION_EXPIRED &&
1921             shdr->Status != STATUS_USER_SESSION_DELETED)
1922                 return false;
1923
1924         trace_smb3_ses_expired(shdr->TreeId, shdr->SessionId,
1925                                le16_to_cpu(shdr->Command),
1926                                le64_to_cpu(shdr->MessageId));
1927         cifs_dbg(FYI, "Session expired or deleted\n");
1928
1929         return true;
1930 }
1931
1932 static int
1933 smb2_oplock_response(struct cifs_tcon *tcon, struct cifs_fid *fid,
1934                      struct cifsInodeInfo *cinode)
1935 {
1936         if (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_LEASING)
1937                 return SMB2_lease_break(0, tcon, cinode->lease_key,
1938                                         smb2_get_lease_state(cinode));
1939
1940         return SMB2_oplock_break(0, tcon, fid->persistent_fid,
1941                                  fid->volatile_fid,
1942                                  CIFS_CACHE_READ(cinode) ? 1 : 0);
1943 }
1944
1945 void
1946 smb2_set_related(struct smb_rqst *rqst)
1947 {
1948         struct smb2_sync_hdr *shdr;
1949
1950         shdr = (struct smb2_sync_hdr *)(rqst->rq_iov[0].iov_base);
1951         shdr->Flags |= SMB2_FLAGS_RELATED_OPERATIONS;
1952 }
1953
1954 char smb2_padding[7] = {0, 0, 0, 0, 0, 0, 0};
1955
1956 void
1957 smb2_set_next_command(struct cifs_tcon *tcon, struct smb_rqst *rqst)
1958 {
1959         struct smb2_sync_hdr *shdr;
1960         struct cifs_ses *ses = tcon->ses;
1961         struct TCP_Server_Info *server = ses->server;
1962         unsigned long len = smb_rqst_len(server, rqst);
1963         int i, num_padding;
1964
1965         /* SMB headers in a compound are 8 byte aligned. */
1966
1967         /* No padding needed */
1968         if (!(len & 7))
1969                 goto finished;
1970
1971         num_padding = 8 - (len & 7);
1972         if (!smb3_encryption_required(tcon)) {
1973                 /*
1974                  * If we do not have encryption then we can just add an extra
1975                  * iov for the padding.
1976                  */
1977                 rqst->rq_iov[rqst->rq_nvec].iov_base = smb2_padding;
1978                 rqst->rq_iov[rqst->rq_nvec].iov_len = num_padding;
1979                 rqst->rq_nvec++;
1980                 len += num_padding;
1981         } else {
1982                 /*
1983                  * We can not add a small padding iov for the encryption case
1984                  * because the encryption framework can not handle the padding
1985                  * iovs.
1986                  * We have to flatten this into a single buffer and add
1987                  * the padding to it.
1988                  */
1989                 for (i = 1; i < rqst->rq_nvec; i++) {
1990                         memcpy(rqst->rq_iov[0].iov_base +
1991                                rqst->rq_iov[0].iov_len,
1992                                rqst->rq_iov[i].iov_base,
1993                                rqst->rq_iov[i].iov_len);
1994                         rqst->rq_iov[0].iov_len += rqst->rq_iov[i].iov_len;
1995                 }
1996                 memset(rqst->rq_iov[0].iov_base + rqst->rq_iov[0].iov_len,
1997                        0, num_padding);
1998                 rqst->rq_iov[0].iov_len += num_padding;
1999                 len += num_padding;
2000                 rqst->rq_nvec = 1;
2001         }
2002
2003  finished:
2004         shdr = (struct smb2_sync_hdr *)(rqst->rq_iov[0].iov_base);
2005         shdr->NextCommand = cpu_to_le32(len);
2006 }
2007
2008 /*
2009  * Passes the query info response back to the caller on success.
2010  * Caller need to free this with free_rsp_buf().
2011  */
2012 int
2013 smb2_query_info_compound(const unsigned int xid, struct cifs_tcon *tcon,
2014                          __le16 *utf16_path, u32 desired_access,
2015                          u32 class, u32 type, u32 output_len,
2016                          struct kvec *rsp, int *buftype,
2017                          struct cifs_sb_info *cifs_sb)
2018 {
2019         struct cifs_ses *ses = tcon->ses;
2020         int flags = 0;
2021         struct smb_rqst rqst[3];
2022         int resp_buftype[3];
2023         struct kvec rsp_iov[3];
2024         struct kvec open_iov[SMB2_CREATE_IOV_SIZE];
2025         struct kvec qi_iov[1];
2026         struct kvec close_iov[1];
2027         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2028         struct cifs_open_parms oparms;
2029         struct cifs_fid fid;
2030         int rc;
2031
2032         if (smb3_encryption_required(tcon))
2033                 flags |= CIFS_TRANSFORM_REQ;
2034
2035         memset(rqst, 0, sizeof(rqst));
2036         resp_buftype[0] = resp_buftype[1] = resp_buftype[2] = CIFS_NO_BUFFER;
2037         memset(rsp_iov, 0, sizeof(rsp_iov));
2038
2039         memset(&open_iov, 0, sizeof(open_iov));
2040         rqst[0].rq_iov = open_iov;
2041         rqst[0].rq_nvec = SMB2_CREATE_IOV_SIZE;
2042
2043         oparms.tcon = tcon;
2044         oparms.desired_access = desired_access;
2045         oparms.disposition = FILE_OPEN;
2046         if (cifs_sb && backup_cred(cifs_sb))
2047                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
2048         else
2049                 oparms.create_options = 0;
2050         oparms.fid = &fid;
2051         oparms.reconnect = false;
2052
2053         rc = SMB2_open_init(tcon, &rqst[0], &oplock, &oparms, utf16_path);
2054         if (rc)
2055                 goto qic_exit;
2056         smb2_set_next_command(tcon, &rqst[0]);
2057
2058         memset(&qi_iov, 0, sizeof(qi_iov));
2059         rqst[1].rq_iov = qi_iov;
2060         rqst[1].rq_nvec = 1;
2061
2062         rc = SMB2_query_info_init(tcon, &rqst[1], COMPOUND_FID, COMPOUND_FID,
2063                                   class, type, 0,
2064                                   output_len, 0,
2065                                   NULL);
2066         if (rc)
2067                 goto qic_exit;
2068         smb2_set_next_command(tcon, &rqst[1]);
2069         smb2_set_related(&rqst[1]);
2070
2071         memset(&close_iov, 0, sizeof(close_iov));
2072         rqst[2].rq_iov = close_iov;
2073         rqst[2].rq_nvec = 1;
2074
2075         rc = SMB2_close_init(tcon, &rqst[2], COMPOUND_FID, COMPOUND_FID);
2076         if (rc)
2077                 goto qic_exit;
2078         smb2_set_related(&rqst[2]);
2079
2080         rc = compound_send_recv(xid, ses, flags, 3, rqst,
2081                                 resp_buftype, rsp_iov);
2082         if (rc) {
2083                 free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
2084                 goto qic_exit;
2085         }
2086         *rsp = rsp_iov[1];
2087         *buftype = resp_buftype[1];
2088
2089  qic_exit:
2090         SMB2_open_free(&rqst[0]);
2091         SMB2_query_info_free(&rqst[1]);
2092         SMB2_close_free(&rqst[2]);
2093         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
2094         free_rsp_buf(resp_buftype[2], rsp_iov[2].iov_base);
2095         return rc;
2096 }
2097
2098 static int
2099 smb2_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
2100              struct kstatfs *buf)
2101 {
2102         struct smb2_query_info_rsp *rsp;
2103         struct smb2_fs_full_size_info *info = NULL;
2104         __le16 utf16_path = 0; /* Null - open root of share */
2105         struct kvec rsp_iov = {NULL, 0};
2106         int buftype = CIFS_NO_BUFFER;
2107         int rc;
2108
2109
2110         rc = smb2_query_info_compound(xid, tcon, &utf16_path,
2111                                       FILE_READ_ATTRIBUTES,
2112                                       FS_FULL_SIZE_INFORMATION,
2113                                       SMB2_O_INFO_FILESYSTEM,
2114                                       sizeof(struct smb2_fs_full_size_info),
2115                                       &rsp_iov, &buftype, NULL);
2116         if (rc)
2117                 goto qfs_exit;
2118
2119         rsp = (struct smb2_query_info_rsp *)rsp_iov.iov_base;
2120         buf->f_type = SMB2_MAGIC_NUMBER;
2121         info = (struct smb2_fs_full_size_info *)(
2122                 le16_to_cpu(rsp->OutputBufferOffset) + (char *)rsp);
2123         rc = smb2_validate_iov(le16_to_cpu(rsp->OutputBufferOffset),
2124                                le32_to_cpu(rsp->OutputBufferLength),
2125                                &rsp_iov,
2126                                sizeof(struct smb2_fs_full_size_info));
2127         if (!rc)
2128                 smb2_copy_fs_info_to_kstatfs(info, buf);
2129
2130 qfs_exit:
2131         free_rsp_buf(buftype, rsp_iov.iov_base);
2132         return rc;
2133 }
2134
2135 static int
2136 smb311_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
2137              struct kstatfs *buf)
2138 {
2139         int rc;
2140         __le16 srch_path = 0; /* Null - open root of share */
2141         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2142         struct cifs_open_parms oparms;
2143         struct cifs_fid fid;
2144
2145         if (!tcon->posix_extensions)
2146                 return smb2_queryfs(xid, tcon, buf);
2147
2148         oparms.tcon = tcon;
2149         oparms.desired_access = FILE_READ_ATTRIBUTES;
2150         oparms.disposition = FILE_OPEN;
2151         oparms.create_options = 0;
2152         oparms.fid = &fid;
2153         oparms.reconnect = false;
2154
2155         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL, NULL);
2156         if (rc)
2157                 return rc;
2158
2159         rc = SMB311_posix_qfs_info(xid, tcon, fid.persistent_fid,
2160                                    fid.volatile_fid, buf);
2161         buf->f_type = SMB2_MAGIC_NUMBER;
2162         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
2163         return rc;
2164 }
2165
2166 static bool
2167 smb2_compare_fids(struct cifsFileInfo *ob1, struct cifsFileInfo *ob2)
2168 {
2169         return ob1->fid.persistent_fid == ob2->fid.persistent_fid &&
2170                ob1->fid.volatile_fid == ob2->fid.volatile_fid;
2171 }
2172
2173 static int
2174 smb2_mand_lock(const unsigned int xid, struct cifsFileInfo *cfile, __u64 offset,
2175                __u64 length, __u32 type, int lock, int unlock, bool wait)
2176 {
2177         if (unlock && !lock)
2178                 type = SMB2_LOCKFLAG_UNLOCK;
2179         return SMB2_lock(xid, tlink_tcon(cfile->tlink),
2180                          cfile->fid.persistent_fid, cfile->fid.volatile_fid,
2181                          current->tgid, length, offset, type, wait);
2182 }
2183
2184 static void
2185 smb2_get_lease_key(struct inode *inode, struct cifs_fid *fid)
2186 {
2187         memcpy(fid->lease_key, CIFS_I(inode)->lease_key, SMB2_LEASE_KEY_SIZE);
2188 }
2189
2190 static void
2191 smb2_set_lease_key(struct inode *inode, struct cifs_fid *fid)
2192 {
2193         memcpy(CIFS_I(inode)->lease_key, fid->lease_key, SMB2_LEASE_KEY_SIZE);
2194 }
2195
2196 static void
2197 smb2_new_lease_key(struct cifs_fid *fid)
2198 {
2199         generate_random_uuid(fid->lease_key);
2200 }
2201
2202 static int
2203 smb2_get_dfs_refer(const unsigned int xid, struct cifs_ses *ses,
2204                    const char *search_name,
2205                    struct dfs_info3_param **target_nodes,
2206                    unsigned int *num_of_nodes,
2207                    const struct nls_table *nls_codepage, int remap)
2208 {
2209         int rc;
2210         __le16 *utf16_path = NULL;
2211         int utf16_path_len = 0;
2212         struct cifs_tcon *tcon;
2213         struct fsctl_get_dfs_referral_req *dfs_req = NULL;
2214         struct get_dfs_referral_rsp *dfs_rsp = NULL;
2215         u32 dfs_req_size = 0, dfs_rsp_size = 0;
2216
2217         cifs_dbg(FYI, "smb2_get_dfs_refer path <%s>\n", search_name);
2218
2219         /*
2220          * Try to use the IPC tcon, otherwise just use any
2221          */
2222         tcon = ses->tcon_ipc;
2223         if (tcon == NULL) {
2224                 spin_lock(&cifs_tcp_ses_lock);
2225                 tcon = list_first_entry_or_null(&ses->tcon_list,
2226                                                 struct cifs_tcon,
2227                                                 tcon_list);
2228                 if (tcon)
2229                         tcon->tc_count++;
2230                 spin_unlock(&cifs_tcp_ses_lock);
2231         }
2232
2233         if (tcon == NULL) {
2234                 cifs_dbg(VFS, "session %p has no tcon available for a dfs referral request\n",
2235                          ses);
2236                 rc = -ENOTCONN;
2237                 goto out;
2238         }
2239
2240         utf16_path = cifs_strndup_to_utf16(search_name, PATH_MAX,
2241                                            &utf16_path_len,
2242                                            nls_codepage, remap);
2243         if (!utf16_path) {
2244                 rc = -ENOMEM;
2245                 goto out;
2246         }
2247
2248         dfs_req_size = sizeof(*dfs_req) + utf16_path_len;
2249         dfs_req = kzalloc(dfs_req_size, GFP_KERNEL);
2250         if (!dfs_req) {
2251                 rc = -ENOMEM;
2252                 goto out;
2253         }
2254
2255         /* Highest DFS referral version understood */
2256         dfs_req->MaxReferralLevel = DFS_VERSION;
2257
2258         /* Path to resolve in an UTF-16 null-terminated string */
2259         memcpy(dfs_req->RequestFileName, utf16_path, utf16_path_len);
2260
2261         do {
2262                 rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
2263                                 FSCTL_DFS_GET_REFERRALS,
2264                                 true /* is_fsctl */,
2265                                 (char *)dfs_req, dfs_req_size,
2266                                 (char **)&dfs_rsp, &dfs_rsp_size);
2267         } while (rc == -EAGAIN);
2268
2269         if (rc) {
2270                 if ((rc != -ENOENT) && (rc != -EOPNOTSUPP))
2271                         cifs_dbg(VFS, "ioctl error in smb2_get_dfs_refer rc=%d\n", rc);
2272                 goto out;
2273         }
2274
2275         rc = parse_dfs_referrals(dfs_rsp, dfs_rsp_size,
2276                                  num_of_nodes, target_nodes,
2277                                  nls_codepage, remap, search_name,
2278                                  true /* is_unicode */);
2279         if (rc) {
2280                 cifs_dbg(VFS, "parse error in smb2_get_dfs_refer rc=%d\n", rc);
2281                 goto out;
2282         }
2283
2284  out:
2285         if (tcon && !tcon->ipc) {
2286                 /* ipc tcons are not refcounted */
2287                 spin_lock(&cifs_tcp_ses_lock);
2288                 tcon->tc_count--;
2289                 spin_unlock(&cifs_tcp_ses_lock);
2290         }
2291         kfree(utf16_path);
2292         kfree(dfs_req);
2293         kfree(dfs_rsp);
2294         return rc;
2295 }
2296 #define SMB2_SYMLINK_STRUCT_SIZE \
2297         (sizeof(struct smb2_err_rsp) - 1 + sizeof(struct smb2_symlink_err_rsp))
2298
2299 static int
2300 smb2_query_symlink(const unsigned int xid, struct cifs_tcon *tcon,
2301                    const char *full_path, char **target_path,
2302                    struct cifs_sb_info *cifs_sb)
2303 {
2304         int rc;
2305         __le16 *utf16_path;
2306         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2307         struct cifs_open_parms oparms;
2308         struct cifs_fid fid;
2309         struct kvec err_iov = {NULL, 0};
2310         struct smb2_err_rsp *err_buf = NULL;
2311         int resp_buftype;
2312         struct smb2_symlink_err_rsp *symlink;
2313         unsigned int sub_len;
2314         unsigned int sub_offset;
2315         unsigned int print_len;
2316         unsigned int print_offset;
2317
2318         cifs_dbg(FYI, "%s: path: %s\n", __func__, full_path);
2319
2320         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
2321         if (!utf16_path)
2322                 return -ENOMEM;
2323
2324         oparms.tcon = tcon;
2325         oparms.desired_access = FILE_READ_ATTRIBUTES;
2326         oparms.disposition = FILE_OPEN;
2327         if (backup_cred(cifs_sb))
2328                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
2329         else
2330                 oparms.create_options = 0;
2331         oparms.fid = &fid;
2332         oparms.reconnect = false;
2333
2334         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, &err_iov,
2335                        &resp_buftype);
2336         if (!rc || !err_iov.iov_base) {
2337                 rc = -ENOENT;
2338                 goto free_path;
2339         }
2340
2341         err_buf = err_iov.iov_base;
2342         if (le32_to_cpu(err_buf->ByteCount) < sizeof(struct smb2_symlink_err_rsp) ||
2343             err_iov.iov_len < SMB2_SYMLINK_STRUCT_SIZE) {
2344                 rc = -ENOENT;
2345                 goto querty_exit;
2346         }
2347
2348         /* open must fail on symlink - reset rc */
2349         rc = 0;
2350         symlink = (struct smb2_symlink_err_rsp *)err_buf->ErrorData;
2351         sub_len = le16_to_cpu(symlink->SubstituteNameLength);
2352         sub_offset = le16_to_cpu(symlink->SubstituteNameOffset);
2353         print_len = le16_to_cpu(symlink->PrintNameLength);
2354         print_offset = le16_to_cpu(symlink->PrintNameOffset);
2355
2356         if (err_iov.iov_len < SMB2_SYMLINK_STRUCT_SIZE + sub_offset + sub_len) {
2357                 rc = -ENOENT;
2358                 goto querty_exit;
2359         }
2360
2361         if (err_iov.iov_len <
2362             SMB2_SYMLINK_STRUCT_SIZE + print_offset + print_len) {
2363                 rc = -ENOENT;
2364                 goto querty_exit;
2365         }
2366
2367         *target_path = cifs_strndup_from_utf16(
2368                                 (char *)symlink->PathBuffer + sub_offset,
2369                                 sub_len, true, cifs_sb->local_nls);
2370         if (!(*target_path)) {
2371                 rc = -ENOMEM;
2372                 goto querty_exit;
2373         }
2374         convert_delimiter(*target_path, '/');
2375         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
2376
2377  querty_exit:
2378         free_rsp_buf(resp_buftype, err_buf);
2379  free_path:
2380         kfree(utf16_path);
2381         return rc;
2382 }
2383
2384 #ifdef CONFIG_CIFS_ACL
2385 static struct cifs_ntsd *
2386 get_smb2_acl_by_fid(struct cifs_sb_info *cifs_sb,
2387                 const struct cifs_fid *cifsfid, u32 *pacllen)
2388 {
2389         struct cifs_ntsd *pntsd = NULL;
2390         unsigned int xid;
2391         int rc = -EOPNOTSUPP;
2392         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
2393
2394         if (IS_ERR(tlink))
2395                 return ERR_CAST(tlink);
2396
2397         xid = get_xid();
2398         cifs_dbg(FYI, "trying to get acl\n");
2399
2400         rc = SMB2_query_acl(xid, tlink_tcon(tlink), cifsfid->persistent_fid,
2401                             cifsfid->volatile_fid, (void **)&pntsd, pacllen);
2402         free_xid(xid);
2403
2404         cifs_put_tlink(tlink);
2405
2406         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
2407         if (rc)
2408                 return ERR_PTR(rc);
2409         return pntsd;
2410
2411 }
2412
2413 static struct cifs_ntsd *
2414 get_smb2_acl_by_path(struct cifs_sb_info *cifs_sb,
2415                 const char *path, u32 *pacllen)
2416 {
2417         struct cifs_ntsd *pntsd = NULL;
2418         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2419         unsigned int xid;
2420         int rc;
2421         struct cifs_tcon *tcon;
2422         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
2423         struct cifs_fid fid;
2424         struct cifs_open_parms oparms;
2425         __le16 *utf16_path;
2426
2427         cifs_dbg(FYI, "get smb3 acl for path %s\n", path);
2428         if (IS_ERR(tlink))
2429                 return ERR_CAST(tlink);
2430
2431         tcon = tlink_tcon(tlink);
2432         xid = get_xid();
2433
2434         if (backup_cred(cifs_sb))
2435                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
2436         else
2437                 oparms.create_options = 0;
2438
2439         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
2440         if (!utf16_path) {
2441                 rc = -ENOMEM;
2442                 free_xid(xid);
2443                 return ERR_PTR(rc);
2444         }
2445
2446         oparms.tcon = tcon;
2447         oparms.desired_access = READ_CONTROL;
2448         oparms.disposition = FILE_OPEN;
2449         oparms.fid = &fid;
2450         oparms.reconnect = false;
2451
2452         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL);
2453         kfree(utf16_path);
2454         if (!rc) {
2455                 rc = SMB2_query_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
2456                             fid.volatile_fid, (void **)&pntsd, pacllen);
2457                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
2458         }
2459
2460         cifs_put_tlink(tlink);
2461         free_xid(xid);
2462
2463         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
2464         if (rc)
2465                 return ERR_PTR(rc);
2466         return pntsd;
2467 }
2468
2469 #ifdef CONFIG_CIFS_ACL
2470 static int
2471 set_smb2_acl(struct cifs_ntsd *pnntsd, __u32 acllen,
2472                 struct inode *inode, const char *path, int aclflag)
2473 {
2474         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
2475         unsigned int xid;
2476         int rc, access_flags = 0;
2477         struct cifs_tcon *tcon;
2478         struct cifs_sb_info *cifs_sb = CIFS_SB(inode->i_sb);
2479         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
2480         struct cifs_fid fid;
2481         struct cifs_open_parms oparms;
2482         __le16 *utf16_path;
2483
2484         cifs_dbg(FYI, "set smb3 acl for path %s\n", path);
2485         if (IS_ERR(tlink))
2486                 return PTR_ERR(tlink);
2487
2488         tcon = tlink_tcon(tlink);
2489         xid = get_xid();
2490
2491         if (backup_cred(cifs_sb))
2492                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
2493         else
2494                 oparms.create_options = 0;
2495
2496         if (aclflag == CIFS_ACL_OWNER || aclflag == CIFS_ACL_GROUP)
2497                 access_flags = WRITE_OWNER;
2498         else
2499                 access_flags = WRITE_DAC;
2500
2501         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
2502         if (!utf16_path) {
2503                 rc = -ENOMEM;
2504                 free_xid(xid);
2505                 return rc;
2506         }
2507
2508         oparms.tcon = tcon;
2509         oparms.desired_access = access_flags;
2510         oparms.disposition = FILE_OPEN;
2511         oparms.path = path;
2512         oparms.fid = &fid;
2513         oparms.reconnect = false;
2514
2515         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL, NULL);
2516         kfree(utf16_path);
2517         if (!rc) {
2518                 rc = SMB2_set_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
2519                             fid.volatile_fid, pnntsd, acllen, aclflag);
2520                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
2521         }
2522
2523         cifs_put_tlink(tlink);
2524         free_xid(xid);
2525         return rc;
2526 }
2527 #endif /* CIFS_ACL */
2528
2529 /* Retrieve an ACL from the server */
2530 static struct cifs_ntsd *
2531 get_smb2_acl(struct cifs_sb_info *cifs_sb,
2532                                       struct inode *inode, const char *path,
2533                                       u32 *pacllen)
2534 {
2535         struct cifs_ntsd *pntsd = NULL;
2536         struct cifsFileInfo *open_file = NULL;
2537
2538         if (inode)
2539                 open_file = find_readable_file(CIFS_I(inode), true);
2540         if (!open_file)
2541                 return get_smb2_acl_by_path(cifs_sb, path, pacllen);
2542
2543         pntsd = get_smb2_acl_by_fid(cifs_sb, &open_file->fid, pacllen);
2544         cifsFileInfo_put(open_file);
2545         return pntsd;
2546 }
2547 #endif
2548
2549 static long smb3_zero_range(struct file *file, struct cifs_tcon *tcon,
2550                             loff_t offset, loff_t len, bool keep_size)
2551 {
2552         struct cifs_ses *ses = tcon->ses;
2553         struct inode *inode;
2554         struct cifsInodeInfo *cifsi;
2555         struct cifsFileInfo *cfile = file->private_data;
2556         struct file_zero_data_information fsctl_buf;
2557         struct smb_rqst rqst[2];
2558         int resp_buftype[2];
2559         struct kvec rsp_iov[2];
2560         struct kvec io_iov[SMB2_IOCTL_IOV_SIZE];
2561         struct kvec si_iov[1];
2562         unsigned int size[1];
2563         void *data[1];
2564         long rc;
2565         unsigned int xid;
2566         int num = 0, flags = 0;
2567         __le64 eof;
2568
2569         xid = get_xid();
2570
2571         inode = d_inode(cfile->dentry);
2572         cifsi = CIFS_I(inode);
2573
2574         trace_smb3_zero_enter(xid, cfile->fid.persistent_fid, tcon->tid,
2575                               ses->Suid, offset, len);
2576
2577
2578         /* if file not oplocked can't be sure whether asking to extend size */
2579         if (!CIFS_CACHE_READ(cifsi))
2580                 if (keep_size == false) {
2581                         rc = -EOPNOTSUPP;
2582                         trace_smb3_zero_err(xid, cfile->fid.persistent_fid,
2583                                 tcon->tid, ses->Suid, offset, len, rc);
2584                         free_xid(xid);
2585                         return rc;
2586                 }
2587
2588         /*
2589          * Must check if file sparse since fallocate -z (zero range) assumes
2590          * non-sparse allocation
2591          */
2592         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE)) {
2593                 rc = -EOPNOTSUPP;
2594                 trace_smb3_zero_err(xid, cfile->fid.persistent_fid, tcon->tid,
2595                               ses->Suid, offset, len, rc);
2596                 free_xid(xid);
2597                 return rc;
2598         }
2599
2600         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
2601
2602         fsctl_buf.FileOffset = cpu_to_le64(offset);
2603         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
2604
2605         if (smb3_encryption_required(tcon))
2606                 flags |= CIFS_TRANSFORM_REQ;
2607
2608         memset(rqst, 0, sizeof(rqst));
2609         resp_buftype[0] = resp_buftype[1] = CIFS_NO_BUFFER;
2610         memset(rsp_iov, 0, sizeof(rsp_iov));
2611
2612
2613         memset(&io_iov, 0, sizeof(io_iov));
2614         rqst[num].rq_iov = io_iov;
2615         rqst[num].rq_nvec = SMB2_IOCTL_IOV_SIZE;
2616         rc = SMB2_ioctl_init(tcon, &rqst[num++], cfile->fid.persistent_fid,
2617                              cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
2618                              true /* is_fctl */, (char *)&fsctl_buf,
2619                              sizeof(struct file_zero_data_information));
2620         if (rc)
2621                 goto zero_range_exit;
2622
2623         /*
2624          * do we also need to change the size of the file?
2625          */
2626         if (keep_size == false && i_size_read(inode) < offset + len) {
2627                 smb2_set_next_command(tcon, &rqst[0]);
2628
2629                 memset(&si_iov, 0, sizeof(si_iov));
2630                 rqst[num].rq_iov = si_iov;
2631                 rqst[num].rq_nvec = 1;
2632
2633                 eof = cpu_to_le64(offset + len);
2634                 size[0] = 8; /* sizeof __le64 */
2635                 data[0] = &eof;
2636
2637                 rc = SMB2_set_info_init(tcon, &rqst[num++],
2638                                         cfile->fid.persistent_fid,
2639                                         cfile->fid.persistent_fid,
2640                                         current->tgid,
2641                                         FILE_END_OF_FILE_INFORMATION,
2642                                         SMB2_O_INFO_FILE, 0, data, size);
2643                 smb2_set_related(&rqst[1]);
2644         }
2645
2646         rc = compound_send_recv(xid, ses, flags, num, rqst,
2647                                 resp_buftype, rsp_iov);
2648
2649  zero_range_exit:
2650         SMB2_ioctl_free(&rqst[0]);
2651         SMB2_set_info_free(&rqst[1]);
2652         free_rsp_buf(resp_buftype[0], rsp_iov[0].iov_base);
2653         free_rsp_buf(resp_buftype[1], rsp_iov[1].iov_base);
2654         free_xid(xid);
2655         if (rc)
2656                 trace_smb3_zero_err(xid, cfile->fid.persistent_fid, tcon->tid,
2657                               ses->Suid, offset, len, rc);
2658         else
2659                 trace_smb3_zero_done(xid, cfile->fid.persistent_fid, tcon->tid,
2660                               ses->Suid, offset, len);
2661         return rc;
2662 }
2663
2664 static long smb3_punch_hole(struct file *file, struct cifs_tcon *tcon,
2665                             loff_t offset, loff_t len)
2666 {
2667         struct inode *inode;
2668         struct cifsInodeInfo *cifsi;
2669         struct cifsFileInfo *cfile = file->private_data;
2670         struct file_zero_data_information fsctl_buf;
2671         long rc;
2672         unsigned int xid;
2673         __u8 set_sparse = 1;
2674
2675         xid = get_xid();
2676
2677         inode = d_inode(cfile->dentry);
2678         cifsi = CIFS_I(inode);
2679
2680         /* Need to make file sparse, if not already, before freeing range. */
2681         /* Consider adding equivalent for compressed since it could also work */
2682         if (!smb2_set_sparse(xid, tcon, cfile, inode, set_sparse)) {
2683                 rc = -EOPNOTSUPP;
2684                 free_xid(xid);
2685                 return rc;
2686         }
2687
2688         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
2689
2690         fsctl_buf.FileOffset = cpu_to_le64(offset);
2691         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
2692
2693         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
2694                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
2695                         true /* is_fctl */, (char *)&fsctl_buf,
2696                         sizeof(struct file_zero_data_information), NULL, NULL);
2697         free_xid(xid);
2698         return rc;
2699 }
2700
2701 static long smb3_simple_falloc(struct file *file, struct cifs_tcon *tcon,
2702                             loff_t off, loff_t len, bool keep_size)
2703 {
2704         struct inode *inode;
2705         struct cifsInodeInfo *cifsi;
2706         struct cifsFileInfo *cfile = file->private_data;
2707         long rc = -EOPNOTSUPP;
2708         unsigned int xid;
2709
2710         xid = get_xid();
2711
2712         inode = d_inode(cfile->dentry);
2713         cifsi = CIFS_I(inode);
2714
2715         trace_smb3_falloc_enter(xid, cfile->fid.persistent_fid, tcon->tid,
2716                                 tcon->ses->Suid, off, len);
2717         /* if file not oplocked can't be sure whether asking to extend size */
2718         if (!CIFS_CACHE_READ(cifsi))
2719                 if (keep_size == false) {
2720                         trace_smb3_falloc_err(xid, cfile->fid.persistent_fid,
2721                                 tcon->tid, tcon->ses->Suid, off, len, rc);
2722                         free_xid(xid);
2723                         return rc;
2724                 }
2725
2726         /*
2727          * Files are non-sparse by default so falloc may be a no-op
2728          * Must check if file sparse. If not sparse, and not extending
2729          * then no need to do anything since file already allocated
2730          */
2731         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) == 0) {
2732                 if (keep_size == true)
2733                         rc = 0;
2734                 /* check if extending file */
2735                 else if (i_size_read(inode) >= off + len)
2736                         /* not extending file and already not sparse */
2737                         rc = 0;
2738                 /* BB: in future add else clause to extend file */
2739                 else
2740                         rc = -EOPNOTSUPP;
2741                 if (rc)
2742                         trace_smb3_falloc_err(xid, cfile->fid.persistent_fid,
2743                                 tcon->tid, tcon->ses->Suid, off, len, rc);
2744                 else
2745                         trace_smb3_falloc_done(xid, cfile->fid.persistent_fid,
2746                                 tcon->tid, tcon->ses->Suid, off, len);
2747                 free_xid(xid);
2748                 return rc;
2749         }
2750
2751         if ((keep_size == true) || (i_size_read(inode) >= off + len)) {
2752                 /*
2753                  * Check if falloc starts within first few pages of file
2754                  * and ends within a few pages of the end of file to
2755                  * ensure that most of file is being forced to be
2756                  * fallocated now. If so then setting whole file sparse
2757                  * ie potentially making a few extra pages at the beginning
2758                  * or end of the file non-sparse via set_sparse is harmless.
2759                  */
2760                 if ((off > 8192) || (off + len + 8192 < i_size_read(inode))) {
2761                         rc = -EOPNOTSUPP;
2762                         trace_smb3_falloc_err(xid, cfile->fid.persistent_fid,
2763                                 tcon->tid, tcon->ses->Suid, off, len, rc);
2764                         free_xid(xid);
2765                         return rc;
2766                 }
2767
2768                 rc = smb2_set_sparse(xid, tcon, cfile, inode, false);
2769         }
2770         /* BB: else ... in future add code to extend file and set sparse */
2771
2772         if (rc)
2773                 trace_smb3_falloc_err(xid, cfile->fid.persistent_fid, tcon->tid,
2774                                 tcon->ses->Suid, off, len, rc);
2775         else
2776                 trace_smb3_falloc_done(xid, cfile->fid.persistent_fid, tcon->tid,
2777                                 tcon->ses->Suid, off, len);
2778
2779         free_xid(xid);
2780         return rc;
2781 }
2782
2783
2784 static long smb3_fallocate(struct file *file, struct cifs_tcon *tcon, int mode,
2785                            loff_t off, loff_t len)
2786 {
2787         /* KEEP_SIZE already checked for by do_fallocate */
2788         if (mode & FALLOC_FL_PUNCH_HOLE)
2789                 return smb3_punch_hole(file, tcon, off, len);
2790         else if (mode & FALLOC_FL_ZERO_RANGE) {
2791                 if (mode & FALLOC_FL_KEEP_SIZE)
2792                         return smb3_zero_range(file, tcon, off, len, true);
2793                 return smb3_zero_range(file, tcon, off, len, false);
2794         } else if (mode == FALLOC_FL_KEEP_SIZE)
2795                 return smb3_simple_falloc(file, tcon, off, len, true);
2796         else if (mode == 0)
2797                 return smb3_simple_falloc(file, tcon, off, len, false);
2798
2799         return -EOPNOTSUPP;
2800 }
2801
2802 static void
2803 smb2_downgrade_oplock(struct TCP_Server_Info *server,
2804                         struct cifsInodeInfo *cinode, bool set_level2)
2805 {
2806         if (set_level2)
2807                 server->ops->set_oplock_level(cinode, SMB2_OPLOCK_LEVEL_II,
2808                                                 0, NULL);
2809         else
2810                 server->ops->set_oplock_level(cinode, 0, 0, NULL);
2811 }
2812
2813 static void
2814 smb21_downgrade_oplock(struct TCP_Server_Info *server,
2815                        struct cifsInodeInfo *cinode, bool set_level2)
2816 {
2817         server->ops->set_oplock_level(cinode,
2818                                       set_level2 ? SMB2_LEASE_READ_CACHING_HE :
2819                                       0, 0, NULL);
2820 }
2821
2822 static void
2823 smb2_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
2824                       unsigned int epoch, bool *purge_cache)
2825 {
2826         oplock &= 0xFF;
2827         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
2828                 return;
2829         if (oplock == SMB2_OPLOCK_LEVEL_BATCH) {
2830                 cinode->oplock = CIFS_CACHE_RHW_FLG;
2831                 cifs_dbg(FYI, "Batch Oplock granted on inode %p\n",
2832                          &cinode->vfs_inode);
2833         } else if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE) {
2834                 cinode->oplock = CIFS_CACHE_RW_FLG;
2835                 cifs_dbg(FYI, "Exclusive Oplock granted on inode %p\n",
2836                          &cinode->vfs_inode);
2837         } else if (oplock == SMB2_OPLOCK_LEVEL_II) {
2838                 cinode->oplock = CIFS_CACHE_READ_FLG;
2839                 cifs_dbg(FYI, "Level II Oplock granted on inode %p\n",
2840                          &cinode->vfs_inode);
2841         } else
2842                 cinode->oplock = 0;
2843 }
2844
2845 static void
2846 smb21_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
2847                        unsigned int epoch, bool *purge_cache)
2848 {
2849         char message[5] = {0};
2850
2851         oplock &= 0xFF;
2852         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
2853                 return;
2854
2855         cinode->oplock = 0;
2856         if (oplock & SMB2_LEASE_READ_CACHING_HE) {
2857                 cinode->oplock |= CIFS_CACHE_READ_FLG;
2858                 strcat(message, "R");
2859         }
2860         if (oplock & SMB2_LEASE_HANDLE_CACHING_HE) {
2861                 cinode->oplock |= CIFS_CACHE_HANDLE_FLG;
2862                 strcat(message, "H");
2863         }
2864         if (oplock & SMB2_LEASE_WRITE_CACHING_HE) {
2865                 cinode->oplock |= CIFS_CACHE_WRITE_FLG;
2866                 strcat(message, "W");
2867         }
2868         if (!cinode->oplock)
2869                 strcat(message, "None");
2870         cifs_dbg(FYI, "%s Lease granted on inode %p\n", message,
2871                  &cinode->vfs_inode);
2872 }
2873
2874 static void
2875 smb3_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
2876                       unsigned int epoch, bool *purge_cache)
2877 {
2878         unsigned int old_oplock = cinode->oplock;
2879
2880         smb21_set_oplock_level(cinode, oplock, epoch, purge_cache);
2881
2882         if (purge_cache) {
2883                 *purge_cache = false;
2884                 if (old_oplock == CIFS_CACHE_READ_FLG) {
2885                         if (cinode->oplock == CIFS_CACHE_READ_FLG &&
2886                             (epoch - cinode->epoch > 0))
2887                                 *purge_cache = true;
2888                         else if (cinode->oplock == CIFS_CACHE_RH_FLG &&
2889                                  (epoch - cinode->epoch > 1))
2890                                 *purge_cache = true;
2891                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
2892                                  (epoch - cinode->epoch > 1))
2893                                 *purge_cache = true;
2894                         else if (cinode->oplock == 0 &&
2895                                  (epoch - cinode->epoch > 0))
2896                                 *purge_cache = true;
2897                 } else if (old_oplock == CIFS_CACHE_RH_FLG) {
2898                         if (cinode->oplock == CIFS_CACHE_RH_FLG &&
2899                             (epoch - cinode->epoch > 0))
2900                                 *purge_cache = true;
2901                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
2902                                  (epoch - cinode->epoch > 1))
2903                                 *purge_cache = true;
2904                 }
2905                 cinode->epoch = epoch;
2906         }
2907 }
2908
2909 static bool
2910 smb2_is_read_op(__u32 oplock)
2911 {
2912         return oplock == SMB2_OPLOCK_LEVEL_II;
2913 }
2914
2915 static bool
2916 smb21_is_read_op(__u32 oplock)
2917 {
2918         return (oplock & SMB2_LEASE_READ_CACHING_HE) &&
2919                !(oplock & SMB2_LEASE_WRITE_CACHING_HE);
2920 }
2921
2922 static __le32
2923 map_oplock_to_lease(u8 oplock)
2924 {
2925         if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE)
2926                 return SMB2_LEASE_WRITE_CACHING | SMB2_LEASE_READ_CACHING;
2927         else if (oplock == SMB2_OPLOCK_LEVEL_II)
2928                 return SMB2_LEASE_READ_CACHING;
2929         else if (oplock == SMB2_OPLOCK_LEVEL_BATCH)
2930                 return SMB2_LEASE_HANDLE_CACHING | SMB2_LEASE_READ_CACHING |
2931                        SMB2_LEASE_WRITE_CACHING;
2932         return 0;
2933 }
2934
2935 static char *
2936 smb2_create_lease_buf(u8 *lease_key, u8 oplock)
2937 {
2938         struct create_lease *buf;
2939
2940         buf = kzalloc(sizeof(struct create_lease), GFP_KERNEL);
2941         if (!buf)
2942                 return NULL;
2943
2944         memcpy(&buf->lcontext.LeaseKey, lease_key, SMB2_LEASE_KEY_SIZE);
2945         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
2946
2947         buf->ccontext.DataOffset = cpu_to_le16(offsetof
2948                                         (struct create_lease, lcontext));
2949         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context));
2950         buf->ccontext.NameOffset = cpu_to_le16(offsetof
2951                                 (struct create_lease, Name));
2952         buf->ccontext.NameLength = cpu_to_le16(4);
2953         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
2954         buf->Name[0] = 'R';
2955         buf->Name[1] = 'q';
2956         buf->Name[2] = 'L';
2957         buf->Name[3] = 's';
2958         return (char *)buf;
2959 }
2960
2961 static char *
2962 smb3_create_lease_buf(u8 *lease_key, u8 oplock)
2963 {
2964         struct create_lease_v2 *buf;
2965
2966         buf = kzalloc(sizeof(struct create_lease_v2), GFP_KERNEL);
2967         if (!buf)
2968                 return NULL;
2969
2970         memcpy(&buf->lcontext.LeaseKey, lease_key, SMB2_LEASE_KEY_SIZE);
2971         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
2972
2973         buf->ccontext.DataOffset = cpu_to_le16(offsetof
2974                                         (struct create_lease_v2, lcontext));
2975         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2));
2976         buf->ccontext.NameOffset = cpu_to_le16(offsetof
2977                                 (struct create_lease_v2, Name));
2978         buf->ccontext.NameLength = cpu_to_le16(4);
2979         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
2980         buf->Name[0] = 'R';
2981         buf->Name[1] = 'q';
2982         buf->Name[2] = 'L';
2983         buf->Name[3] = 's';
2984         return (char *)buf;
2985 }
2986
2987 static __u8
2988 smb2_parse_lease_buf(void *buf, unsigned int *epoch, char *lease_key)
2989 {
2990         struct create_lease *lc = (struct create_lease *)buf;
2991
2992         *epoch = 0; /* not used */
2993         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
2994                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
2995         return le32_to_cpu(lc->lcontext.LeaseState);
2996 }
2997
2998 static __u8
2999 smb3_parse_lease_buf(void *buf, unsigned int *epoch, char *lease_key)
3000 {
3001         struct create_lease_v2 *lc = (struct create_lease_v2 *)buf;
3002
3003         *epoch = le16_to_cpu(lc->lcontext.Epoch);
3004         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
3005                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
3006         if (lease_key)
3007                 memcpy(lease_key, &lc->lcontext.LeaseKey, SMB2_LEASE_KEY_SIZE);
3008         return le32_to_cpu(lc->lcontext.LeaseState);
3009 }
3010
3011 static unsigned int
3012 smb2_wp_retry_size(struct inode *inode)
3013 {
3014         return min_t(unsigned int, CIFS_SB(inode->i_sb)->wsize,
3015                      SMB2_MAX_BUFFER_SIZE);
3016 }
3017
3018 static bool
3019 smb2_dir_needs_close(struct cifsFileInfo *cfile)
3020 {
3021         return !cfile->invalidHandle;
3022 }
3023
3024 static void
3025 fill_transform_hdr(struct smb2_transform_hdr *tr_hdr, unsigned int orig_len,
3026                    struct smb_rqst *old_rq)
3027 {
3028         struct smb2_sync_hdr *shdr =
3029                         (struct smb2_sync_hdr *)old_rq->rq_iov[0].iov_base;
3030
3031         memset(tr_hdr, 0, sizeof(struct smb2_transform_hdr));
3032         tr_hdr->ProtocolId = SMB2_TRANSFORM_PROTO_NUM;
3033         tr_hdr->OriginalMessageSize = cpu_to_le32(orig_len);
3034         tr_hdr->Flags = cpu_to_le16(0x01);
3035         get_random_bytes(&tr_hdr->Nonce, SMB3_AES128CMM_NONCE);
3036         memcpy(&tr_hdr->SessionId, &shdr->SessionId, 8);
3037 }
3038
3039 /* We can not use the normal sg_set_buf() as we will sometimes pass a
3040  * stack object as buf.
3041  */
3042 static inline void smb2_sg_set_buf(struct scatterlist *sg, const void *buf,
3043                                    unsigned int buflen)
3044 {
3045         sg_set_page(sg, virt_to_page(buf), buflen, offset_in_page(buf));
3046 }
3047
3048 /* Assumes the first rqst has a transform header as the first iov.
3049  * I.e.
3050  * rqst[0].rq_iov[0]  is transform header
3051  * rqst[0].rq_iov[1+] data to be encrypted/decrypted
3052  * rqst[1+].rq_iov[0+] data to be encrypted/decrypted
3053  */
3054 static struct scatterlist *
3055 init_sg(int num_rqst, struct smb_rqst *rqst, u8 *sign)
3056 {
3057         unsigned int sg_len;
3058         struct scatterlist *sg;
3059         unsigned int i;
3060         unsigned int j;
3061         unsigned int idx = 0;
3062         int skip;
3063
3064         sg_len = 1;
3065         for (i = 0; i < num_rqst; i++)
3066                 sg_len += rqst[i].rq_nvec + rqst[i].rq_npages;
3067
3068         sg = kmalloc_array(sg_len, sizeof(struct scatterlist), GFP_KERNEL);
3069         if (!sg)
3070                 return NULL;
3071
3072         sg_init_table(sg, sg_len);
3073         for (i = 0; i < num_rqst; i++) {
3074                 for (j = 0; j < rqst[i].rq_nvec; j++) {
3075                         /*
3076                          * The first rqst has a transform header where the
3077                          * first 20 bytes are not part of the encrypted blob
3078                          */
3079                         skip = (i == 0) && (j == 0) ? 20 : 0;
3080                         smb2_sg_set_buf(&sg[idx++],
3081                                         rqst[i].rq_iov[j].iov_base + skip,
3082                                         rqst[i].rq_iov[j].iov_len - skip);
3083                         }
3084
3085                 for (j = 0; j < rqst[i].rq_npages; j++) {
3086                         unsigned int len, offset;
3087
3088                         rqst_page_get_length(&rqst[i], j, &len, &offset);
3089                         sg_set_page(&sg[idx++], rqst[i].rq_pages[j], len, offset);
3090                 }
3091         }
3092         smb2_sg_set_buf(&sg[idx], sign, SMB2_SIGNATURE_SIZE);
3093         return sg;
3094 }
3095
3096 static int
3097 smb2_get_enc_key(struct TCP_Server_Info *server, __u64 ses_id, int enc, u8 *key)
3098 {
3099         struct cifs_ses *ses;
3100         u8 *ses_enc_key;
3101
3102         spin_lock(&cifs_tcp_ses_lock);
3103         list_for_each_entry(ses, &server->smb_ses_list, smb_ses_list) {
3104                 if (ses->Suid != ses_id)
3105                         continue;
3106                 ses_enc_key = enc ? ses->smb3encryptionkey :
3107                                                         ses->smb3decryptionkey;
3108                 memcpy(key, ses_enc_key, SMB3_SIGN_KEY_SIZE);
3109                 spin_unlock(&cifs_tcp_ses_lock);
3110                 return 0;
3111         }
3112         spin_unlock(&cifs_tcp_ses_lock);
3113
3114         return 1;
3115 }
3116 /*
3117  * Encrypt or decrypt @rqst message. @rqst[0] has the following format:
3118  * iov[0]   - transform header (associate data),
3119  * iov[1-N] - SMB2 header and pages - data to encrypt.
3120  * On success return encrypted data in iov[1-N] and pages, leave iov[0]
3121  * untouched.
3122  */
3123 static int
3124 crypt_message(struct TCP_Server_Info *server, int num_rqst,
3125               struct smb_rqst *rqst, int enc)
3126 {
3127         struct smb2_transform_hdr *tr_hdr =
3128                 (struct smb2_transform_hdr *)rqst[0].rq_iov[0].iov_base;
3129         unsigned int assoc_data_len = sizeof(struct smb2_transform_hdr) - 20;
3130         int rc = 0;
3131         struct scatterlist *sg;
3132         u8 sign[SMB2_SIGNATURE_SIZE] = {};
3133         u8 key[SMB3_SIGN_KEY_SIZE];
3134         struct aead_request *req;
3135         char *iv;
3136         unsigned int iv_len;
3137         DECLARE_CRYPTO_WAIT(wait);
3138         struct crypto_aead *tfm;
3139         unsigned int crypt_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
3140
3141         rc = smb2_get_enc_key(server, tr_hdr->SessionId, enc, key);
3142         if (rc) {
3143                 cifs_dbg(VFS, "%s: Could not get %scryption key\n", __func__,
3144                          enc ? "en" : "de");
3145                 return 0;
3146         }
3147
3148         rc = smb3_crypto_aead_allocate(server);
3149         if (rc) {
3150                 cifs_dbg(VFS, "%s: crypto alloc failed\n", __func__);
3151                 return rc;
3152         }
3153
3154         tfm = enc ? server->secmech.ccmaesencrypt :
3155                                                 server->secmech.ccmaesdecrypt;
3156         rc = crypto_aead_setkey(tfm, key, SMB3_SIGN_KEY_SIZE);
3157         if (rc) {
3158                 cifs_dbg(VFS, "%s: Failed to set aead key %d\n", __func__, rc);
3159                 return rc;
3160         }
3161
3162         rc = crypto_aead_setauthsize(tfm, SMB2_SIGNATURE_SIZE);
3163         if (rc) {
3164                 cifs_dbg(VFS, "%s: Failed to set authsize %d\n", __func__, rc);
3165                 return rc;
3166         }
3167
3168         req = aead_request_alloc(tfm, GFP_KERNEL);
3169         if (!req) {
3170                 cifs_dbg(VFS, "%s: Failed to alloc aead request", __func__);
3171                 return -ENOMEM;
3172         }
3173
3174         if (!enc) {
3175                 memcpy(sign, &tr_hdr->Signature, SMB2_SIGNATURE_SIZE);
3176                 crypt_len += SMB2_SIGNATURE_SIZE;
3177         }
3178
3179         sg = init_sg(num_rqst, rqst, sign);
3180         if (!sg) {
3181                 cifs_dbg(VFS, "%s: Failed to init sg", __func__);
3182                 rc = -ENOMEM;
3183                 goto free_req;
3184         }
3185
3186         iv_len = crypto_aead_ivsize(tfm);
3187         iv = kzalloc(iv_len, GFP_KERNEL);
3188         if (!iv) {
3189                 cifs_dbg(VFS, "%s: Failed to alloc IV", __func__);
3190                 rc = -ENOMEM;
3191                 goto free_sg;
3192         }
3193         iv[0] = 3;
3194         memcpy(iv + 1, (char *)tr_hdr->Nonce, SMB3_AES128CMM_NONCE);
3195
3196         aead_request_set_crypt(req, sg, sg, crypt_len, iv);
3197         aead_request_set_ad(req, assoc_data_len);
3198
3199         aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
3200                                   crypto_req_done, &wait);
3201
3202         rc = crypto_wait_req(enc ? crypto_aead_encrypt(req)
3203                                 : crypto_aead_decrypt(req), &wait);
3204
3205         if (!rc && enc)
3206                 memcpy(&tr_hdr->Signature, sign, SMB2_SIGNATURE_SIZE);
3207
3208         kfree(iv);
3209 free_sg:
3210         kfree(sg);
3211 free_req:
3212         kfree(req);
3213         return rc;
3214 }
3215
3216 void
3217 smb3_free_compound_rqst(int num_rqst, struct smb_rqst *rqst)
3218 {
3219         int i, j;
3220
3221         for (i = 0; i < num_rqst; i++) {
3222                 if (rqst[i].rq_pages) {
3223                         for (j = rqst[i].rq_npages - 1; j >= 0; j--)
3224                                 put_page(rqst[i].rq_pages[j]);
3225                         kfree(rqst[i].rq_pages);
3226                 }
3227         }
3228 }
3229
3230 /*
3231  * This function will initialize new_rq and encrypt the content.
3232  * The first entry, new_rq[0], only contains a single iov which contains
3233  * a smb2_transform_hdr and is pre-allocated by the caller.
3234  * This function then populates new_rq[1+] with the content from olq_rq[0+].
3235  *
3236  * The end result is an array of smb_rqst structures where the first structure
3237  * only contains a single iov for the transform header which we then can pass
3238  * to crypt_message().
3239  *
3240  * new_rq[0].rq_iov[0] :  smb2_transform_hdr pre-allocated by the caller
3241  * new_rq[1+].rq_iov[*] == old_rq[0+].rq_iov[*] : SMB2/3 requests
3242  */
3243 static int
3244 smb3_init_transform_rq(struct TCP_Server_Info *server, int num_rqst,
3245                        struct smb_rqst *new_rq, struct smb_rqst *old_rq)
3246 {
3247         struct page **pages;
3248         struct smb2_transform_hdr *tr_hdr = new_rq[0].rq_iov[0].iov_base;
3249         unsigned int npages;
3250         unsigned int orig_len = 0;
3251         int i, j;
3252         int rc = -ENOMEM;
3253
3254         for (i = 1; i < num_rqst; i++) {
3255                 npages = old_rq[i - 1].rq_npages;
3256                 pages = kmalloc_array(npages, sizeof(struct page *),
3257                                       GFP_KERNEL);
3258                 if (!pages)
3259                         goto err_free;
3260
3261                 new_rq[i].rq_pages = pages;
3262                 new_rq[i].rq_npages = npages;
3263                 new_rq[i].rq_offset = old_rq[i - 1].rq_offset;
3264                 new_rq[i].rq_pagesz = old_rq[i - 1].rq_pagesz;
3265                 new_rq[i].rq_tailsz = old_rq[i - 1].rq_tailsz;
3266                 new_rq[i].rq_iov = old_rq[i - 1].rq_iov;
3267                 new_rq[i].rq_nvec = old_rq[i - 1].rq_nvec;
3268
3269                 orig_len += smb_rqst_len(server, &old_rq[i - 1]);
3270
3271                 for (j = 0; j < npages; j++) {
3272                         pages[j] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
3273                         if (!pages[j])
3274                                 goto err_free;
3275                 }
3276
3277                 /* copy pages form the old */
3278                 for (j = 0; j < npages; j++) {
3279                         char *dst, *src;
3280                         unsigned int offset, len;
3281
3282                         rqst_page_get_length(&new_rq[i], j, &len, &offset);
3283
3284                         dst = (char *) kmap(new_rq[i].rq_pages[j]) + offset;
3285                         src = (char *) kmap(old_rq[i - 1].rq_pages[j]) + offset;
3286
3287                         memcpy(dst, src, len);
3288                         kunmap(new_rq[i].rq_pages[j]);
3289                         kunmap(old_rq[i - 1].rq_pages[j]);
3290                 }
3291         }
3292
3293         /* fill the 1st iov with a transform header */
3294         fill_transform_hdr(tr_hdr, orig_len, old_rq);
3295
3296         rc = crypt_message(server, num_rqst, new_rq, 1);
3297         cifs_dbg(FYI, "encrypt message returned %d", rc);
3298         if (rc)
3299                 goto err_free;
3300
3301         return rc;
3302
3303 err_free:
3304         smb3_free_compound_rqst(num_rqst - 1, &new_rq[1]);
3305         return rc;
3306 }
3307
3308 static int
3309 smb3_is_transform_hdr(void *buf)
3310 {
3311         struct smb2_transform_hdr *trhdr = buf;
3312
3313         return trhdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM;
3314 }
3315
3316 static int
3317 decrypt_raw_data(struct TCP_Server_Info *server, char *buf,
3318                  unsigned int buf_data_size, struct page **pages,
3319                  unsigned int npages, unsigned int page_data_size)
3320 {
3321         struct kvec iov[2];
3322         struct smb_rqst rqst = {NULL};
3323         int rc;
3324
3325         iov[0].iov_base = buf;
3326         iov[0].iov_len = sizeof(struct smb2_transform_hdr);
3327         iov[1].iov_base = buf + sizeof(struct smb2_transform_hdr);
3328         iov[1].iov_len = buf_data_size;
3329
3330         rqst.rq_iov = iov;
3331         rqst.rq_nvec = 2;
3332         rqst.rq_pages = pages;
3333         rqst.rq_npages = npages;
3334         rqst.rq_pagesz = PAGE_SIZE;
3335         rqst.rq_tailsz = (page_data_size % PAGE_SIZE) ? : PAGE_SIZE;
3336
3337         rc = crypt_message(server, 1, &rqst, 0);
3338         cifs_dbg(FYI, "decrypt message returned %d\n", rc);
3339
3340         if (rc)
3341                 return rc;
3342
3343         memmove(buf, iov[1].iov_base, buf_data_size);
3344
3345         server->total_read = buf_data_size + page_data_size;
3346
3347         return rc;
3348 }
3349
3350 static int
3351 read_data_into_pages(struct TCP_Server_Info *server, struct page **pages,
3352                      unsigned int npages, unsigned int len)
3353 {
3354         int i;
3355         int length;
3356
3357         for (i = 0; i < npages; i++) {
3358                 struct page *page = pages[i];
3359                 size_t n;
3360
3361                 n = len;
3362                 if (len >= PAGE_SIZE) {
3363                         /* enough data to fill the page */
3364                         n = PAGE_SIZE;
3365                         len -= n;
3366                 } else {
3367                         zero_user(page, len, PAGE_SIZE - len);
3368                         len = 0;
3369                 }
3370                 length = cifs_read_page_from_socket(server, page, 0, n);
3371                 if (length < 0)
3372                         return length;
3373                 server->total_read += length;
3374         }
3375
3376         return 0;
3377 }
3378
3379 static int
3380 init_read_bvec(struct page **pages, unsigned int npages, unsigned int data_size,
3381                unsigned int cur_off, struct bio_vec **page_vec)
3382 {
3383         struct bio_vec *bvec;
3384         int i;
3385
3386         bvec = kcalloc(npages, sizeof(struct bio_vec), GFP_KERNEL);
3387         if (!bvec)
3388                 return -ENOMEM;
3389
3390         for (i = 0; i < npages; i++) {
3391                 bvec[i].bv_page = pages[i];
3392                 bvec[i].bv_offset = (i == 0) ? cur_off : 0;
3393                 bvec[i].bv_len = min_t(unsigned int, PAGE_SIZE, data_size);
3394                 data_size -= bvec[i].bv_len;
3395         }
3396
3397         if (data_size != 0) {
3398                 cifs_dbg(VFS, "%s: something went wrong\n", __func__);
3399                 kfree(bvec);
3400                 return -EIO;
3401         }
3402
3403         *page_vec = bvec;
3404         return 0;
3405 }
3406
3407 static int
3408 handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid,
3409                  char *buf, unsigned int buf_len, struct page **pages,
3410                  unsigned int npages, unsigned int page_data_size)
3411 {
3412         unsigned int data_offset;
3413         unsigned int data_len;
3414         unsigned int cur_off;
3415         unsigned int cur_page_idx;
3416         unsigned int pad_len;
3417         struct cifs_readdata *rdata = mid->callback_data;
3418         struct smb2_sync_hdr *shdr = (struct smb2_sync_hdr *)buf;
3419         struct bio_vec *bvec = NULL;
3420         struct iov_iter iter;
3421         struct kvec iov;
3422         int length;
3423         bool use_rdma_mr = false;
3424
3425         if (shdr->Command != SMB2_READ) {
3426                 cifs_dbg(VFS, "only big read responses are supported\n");
3427                 return -ENOTSUPP;
3428         }
3429
3430         if (server->ops->is_session_expired &&
3431             server->ops->is_session_expired(buf)) {
3432                 cifs_reconnect(server);
3433                 wake_up(&server->response_q);
3434                 return -1;
3435         }
3436
3437         if (server->ops->is_status_pending &&
3438                         server->ops->is_status_pending(buf, server))
3439                 return -1;
3440
3441         /* set up first two iov to get credits */
3442         rdata->iov[0].iov_base = buf;
3443         rdata->iov[0].iov_len = 0;
3444         rdata->iov[1].iov_base = buf;
3445         rdata->iov[1].iov_len =
3446                 min_t(unsigned int, buf_len, server->vals->read_rsp_size);
3447         cifs_dbg(FYI, "0: iov_base=%p iov_len=%zu\n",
3448                  rdata->iov[0].iov_base, rdata->iov[0].iov_len);
3449         cifs_dbg(FYI, "1: iov_base=%p iov_len=%zu\n",
3450                  rdata->iov[1].iov_base, rdata->iov[1].iov_len);
3451
3452         rdata->result = server->ops->map_error(buf, true);
3453         if (rdata->result != 0) {
3454                 cifs_dbg(FYI, "%s: server returned error %d\n",
3455                          __func__, rdata->result);
3456                 /* normal error on read response */
3457                 dequeue_mid(mid, false);
3458                 return 0;
3459         }
3460
3461         data_offset = server->ops->read_data_offset(buf);
3462 #ifdef CONFIG_CIFS_SMB_DIRECT
3463         use_rdma_mr = rdata->mr;
3464 #endif
3465         data_len = server->ops->read_data_length(buf, use_rdma_mr);
3466
3467         if (data_offset < server->vals->read_rsp_size) {
3468                 /*
3469                  * win2k8 sometimes sends an offset of 0 when the read
3470                  * is beyond the EOF. Treat it as if the data starts just after
3471                  * the header.
3472                  */
3473                 cifs_dbg(FYI, "%s: data offset (%u) inside read response header\n",
3474                          __func__, data_offset);
3475                 data_offset = server->vals->read_rsp_size;
3476         } else if (data_offset > MAX_CIFS_SMALL_BUFFER_SIZE) {
3477                 /* data_offset is beyond the end of smallbuf */
3478                 cifs_dbg(FYI, "%s: data offset (%u) beyond end of smallbuf\n",
3479                          __func__, data_offset);
3480                 rdata->result = -EIO;
3481                 dequeue_mid(mid, rdata->result);
3482                 return 0;
3483         }
3484
3485         pad_len = data_offset - server->vals->read_rsp_size;
3486
3487         if (buf_len <= data_offset) {
3488                 /* read response payload is in pages */
3489                 cur_page_idx = pad_len / PAGE_SIZE;
3490                 cur_off = pad_len % PAGE_SIZE;
3491
3492                 if (cur_page_idx != 0) {
3493                         /* data offset is beyond the 1st page of response */
3494                         cifs_dbg(FYI, "%s: data offset (%u) beyond 1st page of response\n",
3495                                  __func__, data_offset);
3496                         rdata->result = -EIO;
3497                         dequeue_mid(mid, rdata->result);
3498                         return 0;
3499                 }
3500
3501                 if (data_len > page_data_size - pad_len) {
3502                         /* data_len is corrupt -- discard frame */
3503                         rdata->result = -EIO;
3504                         dequeue_mid(mid, rdata->result);
3505                         return 0;
3506                 }
3507
3508                 rdata->result = init_read_bvec(pages, npages, page_data_size,
3509                                                cur_off, &bvec);
3510                 if (rdata->result != 0) {
3511                         dequeue_mid(mid, rdata->result);
3512                         return 0;
3513                 }
3514
3515                 iov_iter_bvec(&iter, WRITE, bvec, npages, data_len);
3516         } else if (buf_len >= data_offset + data_len) {
3517                 /* read response payload is in buf */
3518                 WARN_ONCE(npages > 0, "read data can be either in buf or in pages");
3519                 iov.iov_base = buf + data_offset;
3520                 iov.iov_len = data_len;
3521                 iov_iter_kvec(&iter, WRITE, &iov, 1, data_len);
3522         } else {
3523                 /* read response payload cannot be in both buf and pages */
3524                 WARN_ONCE(1, "buf can not contain only a part of read data");
3525                 rdata->result = -EIO;
3526                 dequeue_mid(mid, rdata->result);
3527                 return 0;
3528         }
3529
3530         length = rdata->copy_into_pages(server, rdata, &iter);
3531
3532         kfree(bvec);
3533
3534         if (length < 0)
3535                 return length;
3536
3537         dequeue_mid(mid, false);
3538         return length;
3539 }
3540
3541 static int
3542 receive_encrypted_read(struct TCP_Server_Info *server, struct mid_q_entry **mid)
3543 {
3544         char *buf = server->smallbuf;
3545         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
3546         unsigned int npages;
3547         struct page **pages;
3548         unsigned int len;
3549         unsigned int buflen = server->pdu_size;
3550         int rc;
3551         int i = 0;
3552
3553         len = min_t(unsigned int, buflen, server->vals->read_rsp_size +
3554                 sizeof(struct smb2_transform_hdr)) - HEADER_SIZE(server) + 1;
3555
3556         rc = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1, len);
3557         if (rc < 0)
3558                 return rc;
3559         server->total_read += rc;
3560
3561         len = le32_to_cpu(tr_hdr->OriginalMessageSize) -
3562                 server->vals->read_rsp_size;
3563         npages = DIV_ROUND_UP(len, PAGE_SIZE);
3564
3565         pages = kmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
3566         if (!pages) {
3567                 rc = -ENOMEM;
3568                 goto discard_data;
3569         }
3570
3571         for (; i < npages; i++) {
3572                 pages[i] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
3573                 if (!pages[i]) {
3574                         rc = -ENOMEM;
3575                         goto discard_data;
3576                 }
3577         }
3578
3579         /* read read data into pages */
3580         rc = read_data_into_pages(server, pages, npages, len);
3581         if (rc)
3582                 goto free_pages;
3583
3584         rc = cifs_discard_remaining_data(server);
3585         if (rc)
3586                 goto free_pages;
3587
3588         rc = decrypt_raw_data(server, buf, server->vals->read_rsp_size,
3589                               pages, npages, len);
3590         if (rc)
3591                 goto free_pages;
3592
3593         *mid = smb2_find_mid(server, buf);
3594         if (*mid == NULL)
3595                 cifs_dbg(FYI, "mid not found\n");
3596         else {
3597                 cifs_dbg(FYI, "mid found\n");
3598                 (*mid)->decrypted = true;
3599                 rc = handle_read_data(server, *mid, buf,
3600                                       server->vals->read_rsp_size,
3601                                       pages, npages, len);
3602         }
3603
3604 free_pages:
3605         for (i = i - 1; i >= 0; i--)
3606                 put_page(pages[i]);
3607         kfree(pages);
3608         return rc;
3609 discard_data:
3610         cifs_discard_remaining_data(server);
3611         goto free_pages;
3612 }
3613
3614 static int
3615 receive_encrypted_standard(struct TCP_Server_Info *server,
3616                            struct mid_q_entry **mids, char **bufs,
3617                            int *num_mids)
3618 {
3619         int ret, length;
3620         char *buf = server->smallbuf;
3621         char *tmpbuf;
3622         struct smb2_sync_hdr *shdr;
3623         unsigned int pdu_length = server->pdu_size;
3624         unsigned int buf_size;
3625         struct mid_q_entry *mid_entry;
3626         int next_is_large;
3627         char *next_buffer = NULL;
3628
3629         *num_mids = 0;
3630
3631         /* switch to large buffer if too big for a small one */
3632         if (pdu_length > MAX_CIFS_SMALL_BUFFER_SIZE) {
3633                 server->large_buf = true;
3634                 memcpy(server->bigbuf, buf, server->total_read);
3635                 buf = server->bigbuf;
3636         }
3637
3638         /* now read the rest */
3639         length = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1,
3640                                 pdu_length - HEADER_SIZE(server) + 1);
3641         if (length < 0)
3642                 return length;
3643         server->total_read += length;
3644
3645         buf_size = pdu_length - sizeof(struct smb2_transform_hdr);
3646         length = decrypt_raw_data(server, buf, buf_size, NULL, 0, 0);
3647         if (length)
3648                 return length;
3649
3650         next_is_large = server->large_buf;
3651  one_more:
3652         shdr = (struct smb2_sync_hdr *)buf;
3653         if (shdr->NextCommand) {
3654                 if (next_is_large) {
3655                         tmpbuf = server->bigbuf;
3656                         next_buffer = (char *)cifs_buf_get();
3657                 } else {
3658                         tmpbuf = server->smallbuf;
3659                         next_buffer = (char *)cifs_small_buf_get();
3660                 }
3661                 memcpy(next_buffer,
3662                        tmpbuf + le32_to_cpu(shdr->NextCommand),
3663                        pdu_length - le32_to_cpu(shdr->NextCommand));
3664         }
3665
3666         mid_entry = smb2_find_mid(server, buf);
3667         if (mid_entry == NULL)
3668                 cifs_dbg(FYI, "mid not found\n");
3669         else {
3670                 cifs_dbg(FYI, "mid found\n");
3671                 mid_entry->decrypted = true;
3672                 mid_entry->resp_buf_size = server->pdu_size;
3673         }
3674
3675         if (*num_mids >= MAX_COMPOUND) {
3676                 cifs_dbg(VFS, "too many PDUs in compound\n");
3677                 return -1;
3678         }
3679         bufs[*num_mids] = buf;
3680         mids[(*num_mids)++] = mid_entry;
3681
3682         if (mid_entry && mid_entry->handle)
3683                 ret = mid_entry->handle(server, mid_entry);
3684         else
3685                 ret = cifs_handle_standard(server, mid_entry);
3686
3687         if (ret == 0 && shdr->NextCommand) {
3688                 pdu_length -= le32_to_cpu(shdr->NextCommand);
3689                 server->large_buf = next_is_large;
3690                 if (next_is_large)
3691                         server->bigbuf = next_buffer;
3692                 else
3693                         server->smallbuf = next_buffer;
3694
3695                 buf += le32_to_cpu(shdr->NextCommand);
3696                 goto one_more;
3697         }
3698
3699         return ret;
3700 }
3701
3702 static int
3703 smb3_receive_transform(struct TCP_Server_Info *server,
3704                        struct mid_q_entry **mids, char **bufs, int *num_mids)
3705 {
3706         char *buf = server->smallbuf;
3707         unsigned int pdu_length = server->pdu_size;
3708         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
3709         unsigned int orig_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
3710
3711         if (pdu_length < sizeof(struct smb2_transform_hdr) +
3712                                                 sizeof(struct smb2_sync_hdr)) {
3713                 cifs_dbg(VFS, "Transform message is too small (%u)\n",
3714                          pdu_length);
3715                 cifs_reconnect(server);
3716                 wake_up(&server->response_q);
3717                 return -ECONNABORTED;
3718         }
3719
3720         if (pdu_length < orig_len + sizeof(struct smb2_transform_hdr)) {
3721                 cifs_dbg(VFS, "Transform message is broken\n");
3722                 cifs_reconnect(server);
3723                 wake_up(&server->response_q);
3724                 return -ECONNABORTED;
3725         }
3726
3727         /* TODO: add support for compounds containing READ. */
3728         if (pdu_length > CIFSMaxBufSize + MAX_HEADER_SIZE(server)) {
3729                 *num_mids = 1;
3730                 return receive_encrypted_read(server, &mids[0]);
3731         }
3732
3733         return receive_encrypted_standard(server, mids, bufs, num_mids);
3734 }
3735
3736 int
3737 smb3_handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid)
3738 {
3739         char *buf = server->large_buf ? server->bigbuf : server->smallbuf;
3740
3741         return handle_read_data(server, mid, buf, server->pdu_size,
3742                                 NULL, 0, 0);
3743 }
3744
3745 static int
3746 smb2_next_header(char *buf)
3747 {
3748         struct smb2_sync_hdr *hdr = (struct smb2_sync_hdr *)buf;
3749         struct smb2_transform_hdr *t_hdr = (struct smb2_transform_hdr *)buf;
3750
3751         if (hdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM)
3752                 return sizeof(struct smb2_transform_hdr) +
3753                   le32_to_cpu(t_hdr->OriginalMessageSize);
3754
3755         return le32_to_cpu(hdr->NextCommand);
3756 }
3757
3758 struct smb_version_operations smb20_operations = {
3759         .compare_fids = smb2_compare_fids,
3760         .setup_request = smb2_setup_request,
3761         .setup_async_request = smb2_setup_async_request,
3762         .check_receive = smb2_check_receive,
3763         .add_credits = smb2_add_credits,
3764         .set_credits = smb2_set_credits,
3765         .get_credits_field = smb2_get_credits_field,
3766         .get_credits = smb2_get_credits,
3767         .wait_mtu_credits = cifs_wait_mtu_credits,
3768         .get_next_mid = smb2_get_next_mid,
3769         .revert_current_mid = smb2_revert_current_mid,
3770         .read_data_offset = smb2_read_data_offset,
3771         .read_data_length = smb2_read_data_length,
3772         .map_error = map_smb2_to_linux_error,
3773         .find_mid = smb2_find_mid,
3774         .check_message = smb2_check_message,
3775         .dump_detail = smb2_dump_detail,
3776         .clear_stats = smb2_clear_stats,
3777         .print_stats = smb2_print_stats,
3778         .is_oplock_break = smb2_is_valid_oplock_break,
3779         .handle_cancelled_mid = smb2_handle_cancelled_mid,
3780         .downgrade_oplock = smb2_downgrade_oplock,
3781         .need_neg = smb2_need_neg,
3782         .negotiate = smb2_negotiate,
3783         .negotiate_wsize = smb2_negotiate_wsize,
3784         .negotiate_rsize = smb2_negotiate_rsize,
3785         .sess_setup = SMB2_sess_setup,
3786         .logoff = SMB2_logoff,
3787         .tree_connect = SMB2_tcon,
3788         .tree_disconnect = SMB2_tdis,
3789         .qfs_tcon = smb2_qfs_tcon,
3790         .is_path_accessible = smb2_is_path_accessible,
3791         .can_echo = smb2_can_echo,
3792         .echo = SMB2_echo,
3793         .query_path_info = smb2_query_path_info,
3794         .get_srv_inum = smb2_get_srv_inum,
3795         .query_file_info = smb2_query_file_info,
3796         .set_path_size = smb2_set_path_size,
3797         .set_file_size = smb2_set_file_size,
3798         .set_file_info = smb2_set_file_info,
3799         .set_compression = smb2_set_compression,
3800         .mkdir = smb2_mkdir,
3801         .mkdir_setinfo = smb2_mkdir_setinfo,
3802         .rmdir = smb2_rmdir,
3803         .unlink = smb2_unlink,
3804         .rename = smb2_rename_path,
3805         .create_hardlink = smb2_create_hardlink,
3806         .query_symlink = smb2_query_symlink,
3807         .query_mf_symlink = smb3_query_mf_symlink,
3808         .create_mf_symlink = smb3_create_mf_symlink,
3809         .open = smb2_open_file,
3810         .set_fid = smb2_set_fid,
3811         .close = smb2_close_file,
3812         .flush = smb2_flush_file,
3813         .async_readv = smb2_async_readv,
3814         .async_writev = smb2_async_writev,
3815         .sync_read = smb2_sync_read,
3816         .sync_write = smb2_sync_write,
3817         .query_dir_first = smb2_query_dir_first,
3818         .query_dir_next = smb2_query_dir_next,
3819         .close_dir = smb2_close_dir,
3820         .calc_smb_size = smb2_calc_size,
3821         .is_status_pending = smb2_is_status_pending,
3822         .is_session_expired = smb2_is_session_expired,
3823         .oplock_response = smb2_oplock_response,
3824         .queryfs = smb2_queryfs,
3825         .mand_lock = smb2_mand_lock,
3826         .mand_unlock_range = smb2_unlock_range,
3827         .push_mand_locks = smb2_push_mandatory_locks,
3828         .get_lease_key = smb2_get_lease_key,
3829         .set_lease_key = smb2_set_lease_key,
3830         .new_lease_key = smb2_new_lease_key,
3831         .calc_signature = smb2_calc_signature,
3832         .is_read_op = smb2_is_read_op,
3833         .set_oplock_level = smb2_set_oplock_level,
3834         .create_lease_buf = smb2_create_lease_buf,
3835         .parse_lease_buf = smb2_parse_lease_buf,
3836         .copychunk_range = smb2_copychunk_range,
3837         .wp_retry_size = smb2_wp_retry_size,
3838         .dir_needs_close = smb2_dir_needs_close,
3839         .get_dfs_refer = smb2_get_dfs_refer,
3840         .select_sectype = smb2_select_sectype,
3841 #ifdef CONFIG_CIFS_XATTR
3842         .query_all_EAs = smb2_query_eas,
3843         .set_EA = smb2_set_ea,
3844 #endif /* CIFS_XATTR */
3845 #ifdef CONFIG_CIFS_ACL
3846         .get_acl = get_smb2_acl,
3847         .get_acl_by_fid = get_smb2_acl_by_fid,
3848         .set_acl = set_smb2_acl,
3849 #endif /* CIFS_ACL */
3850         .next_header = smb2_next_header,
3851         .ioctl_query_info = smb2_ioctl_query_info,
3852 };
3853
3854 struct smb_version_operations smb21_operations = {
3855         .compare_fids = smb2_compare_fids,
3856         .setup_request = smb2_setup_request,
3857         .setup_async_request = smb2_setup_async_request,
3858         .check_receive = smb2_check_receive,
3859         .add_credits = smb2_add_credits,
3860         .set_credits = smb2_set_credits,
3861         .get_credits_field = smb2_get_credits_field,
3862         .get_credits = smb2_get_credits,
3863         .wait_mtu_credits = smb2_wait_mtu_credits,
3864         .adjust_credits = smb2_adjust_credits,
3865         .get_next_mid = smb2_get_next_mid,
3866         .revert_current_mid = smb2_revert_current_mid,
3867         .read_data_offset = smb2_read_data_offset,
3868         .read_data_length = smb2_read_data_length,
3869         .map_error = map_smb2_to_linux_error,
3870         .find_mid = smb2_find_mid,
3871         .check_message = smb2_check_message,
3872         .dump_detail = smb2_dump_detail,
3873         .clear_stats = smb2_clear_stats,
3874         .print_stats = smb2_print_stats,
3875         .is_oplock_break = smb2_is_valid_oplock_break,
3876         .handle_cancelled_mid = smb2_handle_cancelled_mid,
3877         .downgrade_oplock = smb21_downgrade_oplock,
3878         .need_neg = smb2_need_neg,
3879         .negotiate = smb2_negotiate,
3880         .negotiate_wsize = smb2_negotiate_wsize,
3881         .negotiate_rsize = smb2_negotiate_rsize,
3882         .sess_setup = SMB2_sess_setup,
3883         .logoff = SMB2_logoff,
3884         .tree_connect = SMB2_tcon,
3885         .tree_disconnect = SMB2_tdis,
3886         .qfs_tcon = smb2_qfs_tcon,
3887         .is_path_accessible = smb2_is_path_accessible,
3888         .can_echo = smb2_can_echo,
3889         .echo = SMB2_echo,
3890         .query_path_info = smb2_query_path_info,
3891         .get_srv_inum = smb2_get_srv_inum,
3892         .query_file_info = smb2_query_file_info,
3893         .set_path_size = smb2_set_path_size,
3894         .set_file_size = smb2_set_file_size,
3895         .set_file_info = smb2_set_file_info,
3896         .set_compression = smb2_set_compression,
3897         .mkdir = smb2_mkdir,
3898         .mkdir_setinfo = smb2_mkdir_setinfo,
3899         .rmdir = smb2_rmdir,
3900         .unlink = smb2_unlink,
3901         .rename = smb2_rename_path,
3902         .create_hardlink = smb2_create_hardlink,
3903         .query_symlink = smb2_query_symlink,
3904         .query_mf_symlink = smb3_query_mf_symlink,
3905         .create_mf_symlink = smb3_create_mf_symlink,
3906         .open = smb2_open_file,
3907         .set_fid = smb2_set_fid,
3908         .close = smb2_close_file,
3909         .flush = smb2_flush_file,
3910         .async_readv = smb2_async_readv,
3911         .async_writev = smb2_async_writev,
3912         .sync_read = smb2_sync_read,
3913         .sync_write = smb2_sync_write,
3914         .query_dir_first = smb2_query_dir_first,
3915         .query_dir_next = smb2_query_dir_next,
3916         .close_dir = smb2_close_dir,
3917         .calc_smb_size = smb2_calc_size,
3918         .is_status_pending = smb2_is_status_pending,
3919         .is_session_expired = smb2_is_session_expired,
3920         .oplock_response = smb2_oplock_response,
3921         .queryfs = smb2_queryfs,
3922         .mand_lock = smb2_mand_lock,
3923         .mand_unlock_range = smb2_unlock_range,
3924         .push_mand_locks = smb2_push_mandatory_locks,
3925         .get_lease_key = smb2_get_lease_key,
3926         .set_lease_key = smb2_set_lease_key,
3927         .new_lease_key = smb2_new_lease_key,
3928         .calc_signature = smb2_calc_signature,
3929         .is_read_op = smb21_is_read_op,
3930         .set_oplock_level = smb21_set_oplock_level,
3931         .create_lease_buf = smb2_create_lease_buf,
3932         .parse_lease_buf = smb2_parse_lease_buf,
3933         .copychunk_range = smb2_copychunk_range,
3934         .wp_retry_size = smb2_wp_retry_size,
3935         .dir_needs_close = smb2_dir_needs_close,
3936         .enum_snapshots = smb3_enum_snapshots,
3937         .get_dfs_refer = smb2_get_dfs_refer,
3938         .select_sectype = smb2_select_sectype,
3939 #ifdef CONFIG_CIFS_XATTR
3940         .query_all_EAs = smb2_query_eas,
3941         .set_EA = smb2_set_ea,
3942 #endif /* CIFS_XATTR */
3943 #ifdef CONFIG_CIFS_ACL
3944         .get_acl = get_smb2_acl,
3945         .get_acl_by_fid = get_smb2_acl_by_fid,
3946         .set_acl = set_smb2_acl,
3947 #endif /* CIFS_ACL */
3948         .next_header = smb2_next_header,
3949         .ioctl_query_info = smb2_ioctl_query_info,
3950 };
3951
3952 struct smb_version_operations smb30_operations = {
3953         .compare_fids = smb2_compare_fids,
3954         .setup_request = smb2_setup_request,
3955         .setup_async_request = smb2_setup_async_request,
3956         .check_receive = smb2_check_receive,
3957         .add_credits = smb2_add_credits,
3958         .set_credits = smb2_set_credits,
3959         .get_credits_field = smb2_get_credits_field,
3960         .get_credits = smb2_get_credits,
3961         .wait_mtu_credits = smb2_wait_mtu_credits,
3962         .adjust_credits = smb2_adjust_credits,
3963         .get_next_mid = smb2_get_next_mid,
3964         .revert_current_mid = smb2_revert_current_mid,
3965         .read_data_offset = smb2_read_data_offset,
3966         .read_data_length = smb2_read_data_length,
3967         .map_error = map_smb2_to_linux_error,
3968         .find_mid = smb2_find_mid,
3969         .check_message = smb2_check_message,
3970         .dump_detail = smb2_dump_detail,
3971         .clear_stats = smb2_clear_stats,
3972         .print_stats = smb2_print_stats,
3973         .dump_share_caps = smb2_dump_share_caps,
3974         .is_oplock_break = smb2_is_valid_oplock_break,
3975         .handle_cancelled_mid = smb2_handle_cancelled_mid,
3976         .downgrade_oplock = smb21_downgrade_oplock,
3977         .need_neg = smb2_need_neg,
3978         .negotiate = smb2_negotiate,
3979         .negotiate_wsize = smb3_negotiate_wsize,
3980         .negotiate_rsize = smb3_negotiate_rsize,
3981         .sess_setup = SMB2_sess_setup,
3982         .logoff = SMB2_logoff,
3983         .tree_connect = SMB2_tcon,
3984         .tree_disconnect = SMB2_tdis,
3985         .qfs_tcon = smb3_qfs_tcon,
3986         .is_path_accessible = smb2_is_path_accessible,
3987         .can_echo = smb2_can_echo,
3988         .echo = SMB2_echo,
3989         .query_path_info = smb2_query_path_info,
3990         .get_srv_inum = smb2_get_srv_inum,
3991         .query_file_info = smb2_query_file_info,
3992         .set_path_size = smb2_set_path_size,
3993         .set_file_size = smb2_set_file_size,
3994         .set_file_info = smb2_set_file_info,
3995         .set_compression = smb2_set_compression,
3996         .mkdir = smb2_mkdir,
3997         .mkdir_setinfo = smb2_mkdir_setinfo,
3998         .rmdir = smb2_rmdir,
3999         .unlink = smb2_unlink,
4000         .rename = smb2_rename_path,
4001         .create_hardlink = smb2_create_hardlink,
4002         .query_symlink = smb2_query_symlink,
4003         .query_mf_symlink = smb3_query_mf_symlink,
4004         .create_mf_symlink = smb3_create_mf_symlink,
4005         .open = smb2_open_file,
4006         .set_fid = smb2_set_fid,
4007         .close = smb2_close_file,
4008         .flush = smb2_flush_file,
4009         .async_readv = smb2_async_readv,
4010         .async_writev = smb2_async_writev,
4011         .sync_read = smb2_sync_read,
4012         .sync_write = smb2_sync_write,
4013         .query_dir_first = smb2_query_dir_first,
4014         .query_dir_next = smb2_query_dir_next,
4015         .close_dir = smb2_close_dir,
4016         .calc_smb_size = smb2_calc_size,
4017         .is_status_pending = smb2_is_status_pending,
4018         .is_session_expired = smb2_is_session_expired,
4019         .oplock_response = smb2_oplock_response,
4020         .queryfs = smb2_queryfs,
4021         .mand_lock = smb2_mand_lock,
4022         .mand_unlock_range = smb2_unlock_range,
4023         .push_mand_locks = smb2_push_mandatory_locks,
4024         .get_lease_key = smb2_get_lease_key,
4025         .set_lease_key = smb2_set_lease_key,
4026         .new_lease_key = smb2_new_lease_key,
4027         .generate_signingkey = generate_smb30signingkey,
4028         .calc_signature = smb3_calc_signature,
4029         .set_integrity  = smb3_set_integrity,
4030         .is_read_op = smb21_is_read_op,
4031         .set_oplock_level = smb3_set_oplock_level,
4032         .create_lease_buf = smb3_create_lease_buf,
4033         .parse_lease_buf = smb3_parse_lease_buf,
4034         .copychunk_range = smb2_copychunk_range,
4035         .duplicate_extents = smb2_duplicate_extents,
4036         .validate_negotiate = smb3_validate_negotiate,
4037         .wp_retry_size = smb2_wp_retry_size,
4038         .dir_needs_close = smb2_dir_needs_close,
4039         .fallocate = smb3_fallocate,
4040         .enum_snapshots = smb3_enum_snapshots,
4041         .init_transform_rq = smb3_init_transform_rq,
4042         .is_transform_hdr = smb3_is_transform_hdr,
4043         .receive_transform = smb3_receive_transform,
4044         .get_dfs_refer = smb2_get_dfs_refer,
4045         .select_sectype = smb2_select_sectype,
4046 #ifdef CONFIG_CIFS_XATTR
4047         .query_all_EAs = smb2_query_eas,
4048         .set_EA = smb2_set_ea,
4049 #endif /* CIFS_XATTR */
4050 #ifdef CONFIG_CIFS_ACL
4051         .get_acl = get_smb2_acl,
4052         .get_acl_by_fid = get_smb2_acl_by_fid,
4053         .set_acl = set_smb2_acl,
4054 #endif /* CIFS_ACL */
4055         .next_header = smb2_next_header,
4056         .ioctl_query_info = smb2_ioctl_query_info,
4057 };
4058
4059 struct smb_version_operations smb311_operations = {
4060         .compare_fids = smb2_compare_fids,
4061         .setup_request = smb2_setup_request,
4062         .setup_async_request = smb2_setup_async_request,
4063         .check_receive = smb2_check_receive,
4064         .add_credits = smb2_add_credits,
4065         .set_credits = smb2_set_credits,
4066         .get_credits_field = smb2_get_credits_field,
4067         .get_credits = smb2_get_credits,
4068         .wait_mtu_credits = smb2_wait_mtu_credits,
4069         .adjust_credits = smb2_adjust_credits,
4070         .get_next_mid = smb2_get_next_mid,
4071         .revert_current_mid = smb2_revert_current_mid,
4072         .read_data_offset = smb2_read_data_offset,
4073         .read_data_length = smb2_read_data_length,
4074         .map_error = map_smb2_to_linux_error,
4075         .find_mid = smb2_find_mid,
4076         .check_message = smb2_check_message,
4077         .dump_detail = smb2_dump_detail,
4078         .clear_stats = smb2_clear_stats,
4079         .print_stats = smb2_print_stats,
4080         .dump_share_caps = smb2_dump_share_caps,
4081         .is_oplock_break = smb2_is_valid_oplock_break,
4082         .handle_cancelled_mid = smb2_handle_cancelled_mid,
4083         .downgrade_oplock = smb21_downgrade_oplock,
4084         .need_neg = smb2_need_neg,
4085         .negotiate = smb2_negotiate,
4086         .negotiate_wsize = smb3_negotiate_wsize,
4087         .negotiate_rsize = smb3_negotiate_rsize,
4088         .sess_setup = SMB2_sess_setup,
4089         .logoff = SMB2_logoff,
4090         .tree_connect = SMB2_tcon,
4091         .tree_disconnect = SMB2_tdis,
4092         .qfs_tcon = smb3_qfs_tcon,
4093         .is_path_accessible = smb2_is_path_accessible,
4094         .can_echo = smb2_can_echo,
4095         .echo = SMB2_echo,
4096         .query_path_info = smb2_query_path_info,
4097         .get_srv_inum = smb2_get_srv_inum,
4098         .query_file_info = smb2_query_file_info,
4099         .set_path_size = smb2_set_path_size,
4100         .set_file_size = smb2_set_file_size,
4101         .set_file_info = smb2_set_file_info,
4102         .set_compression = smb2_set_compression,
4103         .mkdir = smb2_mkdir,
4104         .mkdir_setinfo = smb2_mkdir_setinfo,
4105         .posix_mkdir = smb311_posix_mkdir,
4106         .rmdir = smb2_rmdir,
4107         .unlink = smb2_unlink,
4108         .rename = smb2_rename_path,
4109         .create_hardlink = smb2_create_hardlink,
4110         .query_symlink = smb2_query_symlink,
4111         .query_mf_symlink = smb3_query_mf_symlink,
4112         .create_mf_symlink = smb3_create_mf_symlink,
4113         .open = smb2_open_file,
4114         .set_fid = smb2_set_fid,
4115         .close = smb2_close_file,
4116         .flush = smb2_flush_file,
4117         .async_readv = smb2_async_readv,
4118         .async_writev = smb2_async_writev,
4119         .sync_read = smb2_sync_read,
4120         .sync_write = smb2_sync_write,
4121         .query_dir_first = smb2_query_dir_first,
4122         .query_dir_next = smb2_query_dir_next,
4123         .close_dir = smb2_close_dir,
4124         .calc_smb_size = smb2_calc_size,
4125         .is_status_pending = smb2_is_status_pending,
4126         .is_session_expired = smb2_is_session_expired,
4127         .oplock_response = smb2_oplock_response,
4128         .queryfs = smb311_queryfs,
4129         .mand_lock = smb2_mand_lock,
4130         .mand_unlock_range = smb2_unlock_range,
4131         .push_mand_locks = smb2_push_mandatory_locks,
4132         .get_lease_key = smb2_get_lease_key,
4133         .set_lease_key = smb2_set_lease_key,
4134         .new_lease_key = smb2_new_lease_key,
4135         .generate_signingkey = generate_smb311signingkey,
4136         .calc_signature = smb3_calc_signature,
4137         .set_integrity  = smb3_set_integrity,
4138         .is_read_op = smb21_is_read_op,
4139         .set_oplock_level = smb3_set_oplock_level,
4140         .create_lease_buf = smb3_create_lease_buf,
4141         .parse_lease_buf = smb3_parse_lease_buf,
4142         .copychunk_range = smb2_copychunk_range,
4143         .duplicate_extents = smb2_duplicate_extents,
4144 /*      .validate_negotiate = smb3_validate_negotiate, */ /* not used in 3.11 */
4145         .wp_retry_size = smb2_wp_retry_size,
4146         .dir_needs_close = smb2_dir_needs_close,
4147         .fallocate = smb3_fallocate,
4148         .enum_snapshots = smb3_enum_snapshots,
4149         .init_transform_rq = smb3_init_transform_rq,
4150         .is_transform_hdr = smb3_is_transform_hdr,
4151         .receive_transform = smb3_receive_transform,
4152         .get_dfs_refer = smb2_get_dfs_refer,
4153         .select_sectype = smb2_select_sectype,
4154 #ifdef CONFIG_CIFS_XATTR
4155         .query_all_EAs = smb2_query_eas,
4156         .set_EA = smb2_set_ea,
4157 #endif /* CIFS_XATTR */
4158 #ifdef CONFIG_CIFS_ACL
4159         .get_acl = get_smb2_acl,
4160         .get_acl_by_fid = get_smb2_acl_by_fid,
4161         .set_acl = set_smb2_acl,
4162 #endif /* CIFS_ACL */
4163         .next_header = smb2_next_header,
4164         .ioctl_query_info = smb2_ioctl_query_info,
4165 };
4166
4167 struct smb_version_values smb20_values = {
4168         .version_string = SMB20_VERSION_STRING,
4169         .protocol_id = SMB20_PROT_ID,
4170         .req_capabilities = 0, /* MBZ */
4171         .large_lock_type = 0,
4172         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4173         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4174         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4175         .header_size = sizeof(struct smb2_sync_hdr),
4176         .header_preamble_size = 0,
4177         .max_header_size = MAX_SMB2_HDR_SIZE,
4178         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4179         .lock_cmd = SMB2_LOCK,
4180         .cap_unix = 0,
4181         .cap_nt_find = SMB2_NT_FIND,
4182         .cap_large_files = SMB2_LARGE_FILES,
4183         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4184         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4185         .create_lease_size = sizeof(struct create_lease),
4186 };
4187
4188 struct smb_version_values smb21_values = {
4189         .version_string = SMB21_VERSION_STRING,
4190         .protocol_id = SMB21_PROT_ID,
4191         .req_capabilities = 0, /* MBZ on negotiate req until SMB3 dialect */
4192         .large_lock_type = 0,
4193         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4194         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4195         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4196         .header_size = sizeof(struct smb2_sync_hdr),
4197         .header_preamble_size = 0,
4198         .max_header_size = MAX_SMB2_HDR_SIZE,
4199         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4200         .lock_cmd = SMB2_LOCK,
4201         .cap_unix = 0,
4202         .cap_nt_find = SMB2_NT_FIND,
4203         .cap_large_files = SMB2_LARGE_FILES,
4204         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4205         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4206         .create_lease_size = sizeof(struct create_lease),
4207 };
4208
4209 struct smb_version_values smb3any_values = {
4210         .version_string = SMB3ANY_VERSION_STRING,
4211         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
4212         .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,
4213         .large_lock_type = 0,
4214         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4215         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4216         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4217         .header_size = sizeof(struct smb2_sync_hdr),
4218         .header_preamble_size = 0,
4219         .max_header_size = MAX_SMB2_HDR_SIZE,
4220         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4221         .lock_cmd = SMB2_LOCK,
4222         .cap_unix = 0,
4223         .cap_nt_find = SMB2_NT_FIND,
4224         .cap_large_files = SMB2_LARGE_FILES,
4225         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4226         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4227         .create_lease_size = sizeof(struct create_lease_v2),
4228 };
4229
4230 struct smb_version_values smbdefault_values = {
4231         .version_string = SMBDEFAULT_VERSION_STRING,
4232         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
4233         .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,
4234         .large_lock_type = 0,
4235         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4236         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4237         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4238         .header_size = sizeof(struct smb2_sync_hdr),
4239         .header_preamble_size = 0,
4240         .max_header_size = MAX_SMB2_HDR_SIZE,
4241         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4242         .lock_cmd = SMB2_LOCK,
4243         .cap_unix = 0,
4244         .cap_nt_find = SMB2_NT_FIND,
4245         .cap_large_files = SMB2_LARGE_FILES,
4246         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4247         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4248         .create_lease_size = sizeof(struct create_lease_v2),
4249 };
4250
4251 struct smb_version_values smb30_values = {
4252         .version_string = SMB30_VERSION_STRING,
4253         .protocol_id = SMB30_PROT_ID,
4254         .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,
4255         .large_lock_type = 0,
4256         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4257         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4258         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4259         .header_size = sizeof(struct smb2_sync_hdr),
4260         .header_preamble_size = 0,
4261         .max_header_size = MAX_SMB2_HDR_SIZE,
4262         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4263         .lock_cmd = SMB2_LOCK,
4264         .cap_unix = 0,
4265         .cap_nt_find = SMB2_NT_FIND,
4266         .cap_large_files = SMB2_LARGE_FILES,
4267         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4268         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4269         .create_lease_size = sizeof(struct create_lease_v2),
4270 };
4271
4272 struct smb_version_values smb302_values = {
4273         .version_string = SMB302_VERSION_STRING,
4274         .protocol_id = SMB302_PROT_ID,
4275         .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,
4276         .large_lock_type = 0,
4277         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4278         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4279         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4280         .header_size = sizeof(struct smb2_sync_hdr),
4281         .header_preamble_size = 0,
4282         .max_header_size = MAX_SMB2_HDR_SIZE,
4283         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4284         .lock_cmd = SMB2_LOCK,
4285         .cap_unix = 0,
4286         .cap_nt_find = SMB2_NT_FIND,
4287         .cap_large_files = SMB2_LARGE_FILES,
4288         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4289         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4290         .create_lease_size = sizeof(struct create_lease_v2),
4291 };
4292
4293 struct smb_version_values smb311_values = {
4294         .version_string = SMB311_VERSION_STRING,
4295         .protocol_id = SMB311_PROT_ID,
4296         .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,
4297         .large_lock_type = 0,
4298         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
4299         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
4300         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
4301         .header_size = sizeof(struct smb2_sync_hdr),
4302         .header_preamble_size = 0,
4303         .max_header_size = MAX_SMB2_HDR_SIZE,
4304         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
4305         .lock_cmd = SMB2_LOCK,
4306         .cap_unix = 0,
4307         .cap_nt_find = SMB2_NT_FIND,
4308         .cap_large_files = SMB2_LARGE_FILES,
4309         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
4310         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
4311         .create_lease_size = sizeof(struct create_lease_v2),
4312 };