]> asedeno.scripts.mit.edu Git - linux.git/blob - fs/cifs/smb2ops.c
smb3: fix redundant opens on root
[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 static int
38 change_conf(struct TCP_Server_Info *server)
39 {
40         server->credits += server->echo_credits + server->oplock_credits;
41         server->oplock_credits = server->echo_credits = 0;
42         switch (server->credits) {
43         case 0:
44                 return -1;
45         case 1:
46                 server->echoes = false;
47                 server->oplocks = false;
48                 cifs_dbg(VFS, "disabling echoes and oplocks\n");
49                 break;
50         case 2:
51                 server->echoes = true;
52                 server->oplocks = false;
53                 server->echo_credits = 1;
54                 cifs_dbg(FYI, "disabling oplocks\n");
55                 break;
56         default:
57                 server->echoes = true;
58                 if (enable_oplocks) {
59                         server->oplocks = true;
60                         server->oplock_credits = 1;
61                 } else
62                         server->oplocks = false;
63
64                 server->echo_credits = 1;
65         }
66         server->credits -= server->echo_credits + server->oplock_credits;
67         return 0;
68 }
69
70 static void
71 smb2_add_credits(struct TCP_Server_Info *server, const unsigned int add,
72                  const int optype)
73 {
74         int *val, rc = 0;
75         spin_lock(&server->req_lock);
76         val = server->ops->get_credits_field(server, optype);
77         *val += add;
78         if (*val > 65000) {
79                 *val = 65000; /* Don't get near 64K credits, avoid srv bugs */
80                 printk_once(KERN_WARNING "server overflowed SMB3 credits\n");
81         }
82         server->in_flight--;
83         if (server->in_flight == 0 && (optype & CIFS_OP_MASK) != CIFS_NEG_OP)
84                 rc = change_conf(server);
85         /*
86          * Sometimes server returns 0 credits on oplock break ack - we need to
87          * rebalance credits in this case.
88          */
89         else if (server->in_flight > 0 && server->oplock_credits == 0 &&
90                  server->oplocks) {
91                 if (server->credits > 1) {
92                         server->credits--;
93                         server->oplock_credits++;
94                 }
95         }
96         spin_unlock(&server->req_lock);
97         wake_up(&server->request_q);
98         if (rc)
99                 cifs_reconnect(server);
100 }
101
102 static void
103 smb2_set_credits(struct TCP_Server_Info *server, const int val)
104 {
105         spin_lock(&server->req_lock);
106         server->credits = val;
107         spin_unlock(&server->req_lock);
108 }
109
110 static int *
111 smb2_get_credits_field(struct TCP_Server_Info *server, const int optype)
112 {
113         switch (optype) {
114         case CIFS_ECHO_OP:
115                 return &server->echo_credits;
116         case CIFS_OBREAK_OP:
117                 return &server->oplock_credits;
118         default:
119                 return &server->credits;
120         }
121 }
122
123 static unsigned int
124 smb2_get_credits(struct mid_q_entry *mid)
125 {
126         struct smb2_sync_hdr *shdr = get_sync_hdr(mid->resp_buf);
127
128         return le16_to_cpu(shdr->CreditRequest);
129 }
130
131 static int
132 smb2_wait_mtu_credits(struct TCP_Server_Info *server, unsigned int size,
133                       unsigned int *num, unsigned int *credits)
134 {
135         int rc = 0;
136         unsigned int scredits;
137
138         spin_lock(&server->req_lock);
139         while (1) {
140                 if (server->credits <= 0) {
141                         spin_unlock(&server->req_lock);
142                         cifs_num_waiters_inc(server);
143                         rc = wait_event_killable(server->request_q,
144                                         has_credits(server, &server->credits));
145                         cifs_num_waiters_dec(server);
146                         if (rc)
147                                 return rc;
148                         spin_lock(&server->req_lock);
149                 } else {
150                         if (server->tcpStatus == CifsExiting) {
151                                 spin_unlock(&server->req_lock);
152                                 return -ENOENT;
153                         }
154
155                         scredits = server->credits;
156                         /* can deadlock with reopen */
157                         if (scredits == 1) {
158                                 *num = SMB2_MAX_BUFFER_SIZE;
159                                 *credits = 0;
160                                 break;
161                         }
162
163                         /* leave one credit for a possible reopen */
164                         scredits--;
165                         *num = min_t(unsigned int, size,
166                                      scredits * SMB2_MAX_BUFFER_SIZE);
167
168                         *credits = DIV_ROUND_UP(*num, SMB2_MAX_BUFFER_SIZE);
169                         server->credits -= *credits;
170                         server->in_flight++;
171                         break;
172                 }
173         }
174         spin_unlock(&server->req_lock);
175         return rc;
176 }
177
178 static __u64
179 smb2_get_next_mid(struct TCP_Server_Info *server)
180 {
181         __u64 mid;
182         /* for SMB2 we need the current value */
183         spin_lock(&GlobalMid_Lock);
184         mid = server->CurrentMid++;
185         spin_unlock(&GlobalMid_Lock);
186         return mid;
187 }
188
189 static struct mid_q_entry *
190 smb2_find_mid(struct TCP_Server_Info *server, char *buf)
191 {
192         struct mid_q_entry *mid;
193         struct smb2_sync_hdr *shdr = get_sync_hdr(buf);
194         __u64 wire_mid = le64_to_cpu(shdr->MessageId);
195
196         if (shdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM) {
197                 cifs_dbg(VFS, "encrypted frame parsing not supported yet");
198                 return NULL;
199         }
200
201         spin_lock(&GlobalMid_Lock);
202         list_for_each_entry(mid, &server->pending_mid_q, qhead) {
203                 if ((mid->mid == wire_mid) &&
204                     (mid->mid_state == MID_REQUEST_SUBMITTED) &&
205                     (mid->command == shdr->Command)) {
206                         spin_unlock(&GlobalMid_Lock);
207                         return mid;
208                 }
209         }
210         spin_unlock(&GlobalMid_Lock);
211         return NULL;
212 }
213
214 static void
215 smb2_dump_detail(void *buf)
216 {
217 #ifdef CONFIG_CIFS_DEBUG2
218         struct smb2_sync_hdr *shdr = get_sync_hdr(buf);
219
220         cifs_dbg(VFS, "Cmd: %d Err: 0x%x Flags: 0x%x Mid: %llu Pid: %d\n",
221                  shdr->Command, shdr->Status, shdr->Flags, shdr->MessageId,
222                  shdr->ProcessId);
223         cifs_dbg(VFS, "smb buf %p len %u\n", buf, smb2_calc_size(buf));
224 #endif
225 }
226
227 static bool
228 smb2_need_neg(struct TCP_Server_Info *server)
229 {
230         return server->max_read == 0;
231 }
232
233 static int
234 smb2_negotiate(const unsigned int xid, struct cifs_ses *ses)
235 {
236         int rc;
237         ses->server->CurrentMid = 0;
238         rc = SMB2_negotiate(xid, ses);
239         /* BB we probably don't need to retry with modern servers */
240         if (rc == -EAGAIN)
241                 rc = -EHOSTDOWN;
242         return rc;
243 }
244
245 static unsigned int
246 smb2_negotiate_wsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
247 {
248         struct TCP_Server_Info *server = tcon->ses->server;
249         unsigned int wsize;
250
251         /* start with specified wsize, or default */
252         wsize = volume_info->wsize ? volume_info->wsize : CIFS_DEFAULT_IOSIZE;
253         wsize = min_t(unsigned int, wsize, server->max_write);
254 #ifdef CONFIG_CIFS_SMB_DIRECT
255         if (server->rdma) {
256                 if (server->sign)
257                         wsize = min_t(unsigned int,
258                                 wsize, server->smbd_conn->max_fragmented_send_size);
259                 else
260                         wsize = min_t(unsigned int,
261                                 wsize, server->smbd_conn->max_readwrite_size);
262         }
263 #endif
264         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
265                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
266
267         return wsize;
268 }
269
270 static unsigned int
271 smb2_negotiate_rsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
272 {
273         struct TCP_Server_Info *server = tcon->ses->server;
274         unsigned int rsize;
275
276         /* start with specified rsize, or default */
277         rsize = volume_info->rsize ? volume_info->rsize : CIFS_DEFAULT_IOSIZE;
278         rsize = min_t(unsigned int, rsize, server->max_read);
279 #ifdef CONFIG_CIFS_SMB_DIRECT
280         if (server->rdma) {
281                 if (server->sign)
282                         rsize = min_t(unsigned int,
283                                 rsize, server->smbd_conn->max_fragmented_recv_size);
284                 else
285                         rsize = min_t(unsigned int,
286                                 rsize, server->smbd_conn->max_readwrite_size);
287         }
288 #endif
289
290         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
291                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
292
293         return rsize;
294 }
295
296 #ifdef CONFIG_CIFS_STATS2
297 static int
298 SMB3_request_interfaces(const unsigned int xid, struct cifs_tcon *tcon)
299 {
300         int rc;
301         unsigned int ret_data_len = 0;
302         struct network_interface_info_ioctl_rsp *out_buf;
303
304         rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
305                         FSCTL_QUERY_NETWORK_INTERFACE_INFO, true /* is_fsctl */,
306                         NULL /* no data input */, 0 /* no data input */,
307                         (char **)&out_buf, &ret_data_len);
308         if (rc != 0)
309                 cifs_dbg(VFS, "error %d on ioctl to get interface list\n", rc);
310         else if (ret_data_len < sizeof(struct network_interface_info_ioctl_rsp)) {
311                 cifs_dbg(VFS, "server returned bad net interface info buf\n");
312                 rc = -EINVAL;
313         } else {
314                 /* Dump info on first interface */
315                 cifs_dbg(FYI, "Adapter Capability 0x%x\t",
316                         le32_to_cpu(out_buf->Capability));
317                 cifs_dbg(FYI, "Link Speed %lld\n",
318                         le64_to_cpu(out_buf->LinkSpeed));
319         }
320         kfree(out_buf);
321         return rc;
322 }
323 #endif /* STATS2 */
324
325 /*
326  * Open the directory at the root of a share
327  */
328 int open_shroot(unsigned int xid, struct cifs_tcon *tcon, struct cifs_fid *pfid)
329 {
330         struct cifs_open_parms oparams;
331         int rc;
332         __le16 srch_path = 0; /* Null - since an open of top of share */
333         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
334
335         mutex_lock(&tcon->prfid_mutex);
336         if (tcon->valid_root_fid) {
337                 cifs_dbg(FYI, "found a cached root file handle\n");
338                 memcpy(pfid, tcon->prfid, sizeof(struct cifs_fid));
339                 mutex_unlock(&tcon->prfid_mutex);
340                 return 0;
341         }
342
343         oparams.tcon = tcon;
344         oparams.create_options = 0;
345         oparams.desired_access = FILE_READ_ATTRIBUTES;
346         oparams.disposition = FILE_OPEN;
347         oparams.fid = pfid;
348         oparams.reconnect = false;
349
350         rc = SMB2_open(xid, &oparams, &srch_path, &oplock, NULL, NULL);
351         if (rc == 0) {
352                 memcpy(tcon->prfid, pfid, sizeof(struct cifs_fid));
353                 tcon->valid_root_fid = true;
354         }
355         mutex_unlock(&tcon->prfid_mutex);
356         return rc;
357 }
358
359 static void
360 smb3_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
361 {
362         int rc;
363         __le16 srch_path = 0; /* Null - open root of share */
364         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
365         struct cifs_open_parms oparms;
366         struct cifs_fid fid;
367         bool no_cached_open = tcon->nohandlecache;
368
369         oparms.tcon = tcon;
370         oparms.desired_access = FILE_READ_ATTRIBUTES;
371         oparms.disposition = FILE_OPEN;
372         oparms.create_options = 0;
373         oparms.fid = &fid;
374         oparms.reconnect = false;
375
376         if (no_cached_open)
377                 rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
378         else
379                 rc = open_shroot(xid, tcon, &fid);
380
381         if (rc)
382                 return;
383
384 #ifdef CONFIG_CIFS_STATS2
385         SMB3_request_interfaces(xid, tcon);
386 #endif /* STATS2 */
387
388         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
389                         FS_ATTRIBUTE_INFORMATION);
390         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
391                         FS_DEVICE_INFORMATION);
392         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
393                         FS_SECTOR_SIZE_INFORMATION); /* SMB3 specific */
394         if (no_cached_open)
395                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
396         return;
397 }
398
399 static void
400 smb2_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
401 {
402         int rc;
403         __le16 srch_path = 0; /* Null - open root of share */
404         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
405         struct cifs_open_parms oparms;
406         struct cifs_fid fid;
407
408         oparms.tcon = tcon;
409         oparms.desired_access = FILE_READ_ATTRIBUTES;
410         oparms.disposition = FILE_OPEN;
411         oparms.create_options = 0;
412         oparms.fid = &fid;
413         oparms.reconnect = false;
414
415         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
416         if (rc)
417                 return;
418
419         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
420                         FS_ATTRIBUTE_INFORMATION);
421         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
422                         FS_DEVICE_INFORMATION);
423         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
424         return;
425 }
426
427 static int
428 smb2_is_path_accessible(const unsigned int xid, struct cifs_tcon *tcon,
429                         struct cifs_sb_info *cifs_sb, const char *full_path)
430 {
431         int rc;
432         __le16 *utf16_path;
433         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
434         struct cifs_open_parms oparms;
435         struct cifs_fid fid;
436
437         if ((*full_path == 0) && tcon->valid_root_fid)
438                 return 0;
439
440         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
441         if (!utf16_path)
442                 return -ENOMEM;
443
444         oparms.tcon = tcon;
445         oparms.desired_access = FILE_READ_ATTRIBUTES;
446         oparms.disposition = FILE_OPEN;
447         oparms.create_options = 0;
448         oparms.fid = &fid;
449         oparms.reconnect = false;
450
451         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
452         if (rc) {
453                 kfree(utf16_path);
454                 return rc;
455         }
456
457         rc = SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
458         kfree(utf16_path);
459         return rc;
460 }
461
462 static int
463 smb2_get_srv_inum(const unsigned int xid, struct cifs_tcon *tcon,
464                   struct cifs_sb_info *cifs_sb, const char *full_path,
465                   u64 *uniqueid, FILE_ALL_INFO *data)
466 {
467         *uniqueid = le64_to_cpu(data->IndexNumber);
468         return 0;
469 }
470
471 static int
472 smb2_query_file_info(const unsigned int xid, struct cifs_tcon *tcon,
473                      struct cifs_fid *fid, FILE_ALL_INFO *data)
474 {
475         int rc;
476         struct smb2_file_all_info *smb2_data;
477
478         smb2_data = kzalloc(sizeof(struct smb2_file_all_info) + PATH_MAX * 2,
479                             GFP_KERNEL);
480         if (smb2_data == NULL)
481                 return -ENOMEM;
482
483         rc = SMB2_query_info(xid, tcon, fid->persistent_fid, fid->volatile_fid,
484                              smb2_data);
485         if (!rc)
486                 move_smb2_info_to_cifs(data, smb2_data);
487         kfree(smb2_data);
488         return rc;
489 }
490
491 #ifdef CONFIG_CIFS_XATTR
492 static ssize_t
493 move_smb2_ea_to_cifs(char *dst, size_t dst_size,
494                      struct smb2_file_full_ea_info *src, size_t src_size,
495                      const unsigned char *ea_name)
496 {
497         int rc = 0;
498         unsigned int ea_name_len = ea_name ? strlen(ea_name) : 0;
499         char *name, *value;
500         size_t name_len, value_len, user_name_len;
501
502         while (src_size > 0) {
503                 name = &src->ea_data[0];
504                 name_len = (size_t)src->ea_name_length;
505                 value = &src->ea_data[src->ea_name_length + 1];
506                 value_len = (size_t)le16_to_cpu(src->ea_value_length);
507
508                 if (name_len == 0) {
509                         break;
510                 }
511
512                 if (src_size < 8 + name_len + 1 + value_len) {
513                         cifs_dbg(FYI, "EA entry goes beyond length of list\n");
514                         rc = -EIO;
515                         goto out;
516                 }
517
518                 if (ea_name) {
519                         if (ea_name_len == name_len &&
520                             memcmp(ea_name, name, name_len) == 0) {
521                                 rc = value_len;
522                                 if (dst_size == 0)
523                                         goto out;
524                                 if (dst_size < value_len) {
525                                         rc = -ERANGE;
526                                         goto out;
527                                 }
528                                 memcpy(dst, value, value_len);
529                                 goto out;
530                         }
531                 } else {
532                         /* 'user.' plus a terminating null */
533                         user_name_len = 5 + 1 + name_len;
534
535                         rc += user_name_len;
536
537                         if (dst_size >= user_name_len) {
538                                 dst_size -= user_name_len;
539                                 memcpy(dst, "user.", 5);
540                                 dst += 5;
541                                 memcpy(dst, src->ea_data, name_len);
542                                 dst += name_len;
543                                 *dst = 0;
544                                 ++dst;
545                         } else if (dst_size == 0) {
546                                 /* skip copy - calc size only */
547                         } else {
548                                 /* stop before overrun buffer */
549                                 rc = -ERANGE;
550                                 break;
551                         }
552                 }
553
554                 if (!src->next_entry_offset)
555                         break;
556
557                 if (src_size < le32_to_cpu(src->next_entry_offset)) {
558                         /* stop before overrun buffer */
559                         rc = -ERANGE;
560                         break;
561                 }
562                 src_size -= le32_to_cpu(src->next_entry_offset);
563                 src = (void *)((char *)src +
564                                le32_to_cpu(src->next_entry_offset));
565         }
566
567         /* didn't find the named attribute */
568         if (ea_name)
569                 rc = -ENODATA;
570
571 out:
572         return (ssize_t)rc;
573 }
574
575 static ssize_t
576 smb2_query_eas(const unsigned int xid, struct cifs_tcon *tcon,
577                const unsigned char *path, const unsigned char *ea_name,
578                char *ea_data, size_t buf_size,
579                struct cifs_sb_info *cifs_sb)
580 {
581         int rc;
582         __le16 *utf16_path;
583         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
584         struct cifs_open_parms oparms;
585         struct cifs_fid fid;
586         struct smb2_file_full_ea_info *smb2_data;
587         int ea_buf_size = SMB2_MIN_EA_BUF;
588
589         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
590         if (!utf16_path)
591                 return -ENOMEM;
592
593         oparms.tcon = tcon;
594         oparms.desired_access = FILE_READ_EA;
595         oparms.disposition = FILE_OPEN;
596         oparms.create_options = 0;
597         oparms.fid = &fid;
598         oparms.reconnect = false;
599
600         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
601         kfree(utf16_path);
602         if (rc) {
603                 cifs_dbg(FYI, "open failed rc=%d\n", rc);
604                 return rc;
605         }
606
607         while (1) {
608                 smb2_data = kzalloc(ea_buf_size, GFP_KERNEL);
609                 if (smb2_data == NULL) {
610                         SMB2_close(xid, tcon, fid.persistent_fid,
611                                    fid.volatile_fid);
612                         return -ENOMEM;
613                 }
614
615                 rc = SMB2_query_eas(xid, tcon, fid.persistent_fid,
616                                     fid.volatile_fid,
617                                     ea_buf_size, smb2_data);
618
619                 if (rc != -E2BIG)
620                         break;
621
622                 kfree(smb2_data);
623                 ea_buf_size <<= 1;
624
625                 if (ea_buf_size > SMB2_MAX_EA_BUF) {
626                         cifs_dbg(VFS, "EA size is too large\n");
627                         SMB2_close(xid, tcon, fid.persistent_fid,
628                                    fid.volatile_fid);
629                         return -ENOMEM;
630                 }
631         }
632
633         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
634
635         /*
636          * If ea_name is NULL (listxattr) and there are no EAs, return 0 as it's
637          * not an error. Otherwise, the specified ea_name was not found.
638          */
639         if (!rc)
640                 rc = move_smb2_ea_to_cifs(ea_data, buf_size, smb2_data,
641                                           SMB2_MAX_EA_BUF, ea_name);
642         else if (!ea_name && rc == -ENODATA)
643                 rc = 0;
644
645         kfree(smb2_data);
646         return rc;
647 }
648
649
650 static int
651 smb2_set_ea(const unsigned int xid, struct cifs_tcon *tcon,
652             const char *path, const char *ea_name, const void *ea_value,
653             const __u16 ea_value_len, const struct nls_table *nls_codepage,
654             struct cifs_sb_info *cifs_sb)
655 {
656         int rc;
657         __le16 *utf16_path;
658         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
659         struct cifs_open_parms oparms;
660         struct cifs_fid fid;
661         struct smb2_file_full_ea_info *ea;
662         int ea_name_len = strlen(ea_name);
663         int len;
664
665         if (ea_name_len > 255)
666                 return -EINVAL;
667
668         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
669         if (!utf16_path)
670                 return -ENOMEM;
671
672         oparms.tcon = tcon;
673         oparms.desired_access = FILE_WRITE_EA;
674         oparms.disposition = FILE_OPEN;
675         oparms.create_options = 0;
676         oparms.fid = &fid;
677         oparms.reconnect = false;
678
679         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
680         kfree(utf16_path);
681         if (rc) {
682                 cifs_dbg(FYI, "open failed rc=%d\n", rc);
683                 return rc;
684         }
685
686         len = sizeof(ea) + ea_name_len + ea_value_len + 1;
687         ea = kzalloc(len, GFP_KERNEL);
688         if (ea == NULL) {
689                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
690                 return -ENOMEM;
691         }
692
693         ea->ea_name_length = ea_name_len;
694         ea->ea_value_length = cpu_to_le16(ea_value_len);
695         memcpy(ea->ea_data, ea_name, ea_name_len + 1);
696         memcpy(ea->ea_data + ea_name_len + 1, ea_value, ea_value_len);
697
698         rc = SMB2_set_ea(xid, tcon, fid.persistent_fid, fid.volatile_fid, ea,
699                          len);
700         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
701
702         return rc;
703 }
704 #endif
705
706 static bool
707 smb2_can_echo(struct TCP_Server_Info *server)
708 {
709         return server->echoes;
710 }
711
712 static void
713 smb2_clear_stats(struct cifs_tcon *tcon)
714 {
715 #ifdef CONFIG_CIFS_STATS
716         int i;
717         for (i = 0; i < NUMBER_OF_SMB2_COMMANDS; i++) {
718                 atomic_set(&tcon->stats.smb2_stats.smb2_com_sent[i], 0);
719                 atomic_set(&tcon->stats.smb2_stats.smb2_com_failed[i], 0);
720         }
721 #endif
722 }
723
724 static void
725 smb2_dump_share_caps(struct seq_file *m, struct cifs_tcon *tcon)
726 {
727         seq_puts(m, "\n\tShare Capabilities:");
728         if (tcon->capabilities & SMB2_SHARE_CAP_DFS)
729                 seq_puts(m, " DFS,");
730         if (tcon->capabilities & SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY)
731                 seq_puts(m, " CONTINUOUS AVAILABILITY,");
732         if (tcon->capabilities & SMB2_SHARE_CAP_SCALEOUT)
733                 seq_puts(m, " SCALEOUT,");
734         if (tcon->capabilities & SMB2_SHARE_CAP_CLUSTER)
735                 seq_puts(m, " CLUSTER,");
736         if (tcon->capabilities & SMB2_SHARE_CAP_ASYMMETRIC)
737                 seq_puts(m, " ASYMMETRIC,");
738         if (tcon->capabilities == 0)
739                 seq_puts(m, " None");
740         if (tcon->ss_flags & SSINFO_FLAGS_ALIGNED_DEVICE)
741                 seq_puts(m, " Aligned,");
742         if (tcon->ss_flags & SSINFO_FLAGS_PARTITION_ALIGNED_ON_DEVICE)
743                 seq_puts(m, " Partition Aligned,");
744         if (tcon->ss_flags & SSINFO_FLAGS_NO_SEEK_PENALTY)
745                 seq_puts(m, " SSD,");
746         if (tcon->ss_flags & SSINFO_FLAGS_TRIM_ENABLED)
747                 seq_puts(m, " TRIM-support,");
748
749         seq_printf(m, "\tShare Flags: 0x%x", tcon->share_flags);
750         if (tcon->perf_sector_size)
751                 seq_printf(m, "\tOptimal sector size: 0x%x",
752                            tcon->perf_sector_size);
753 }
754
755 static void
756 smb2_print_stats(struct seq_file *m, struct cifs_tcon *tcon)
757 {
758 #ifdef CONFIG_CIFS_STATS
759         atomic_t *sent = tcon->stats.smb2_stats.smb2_com_sent;
760         atomic_t *failed = tcon->stats.smb2_stats.smb2_com_failed;
761         seq_printf(m, "\nNegotiates: %d sent %d failed",
762                    atomic_read(&sent[SMB2_NEGOTIATE_HE]),
763                    atomic_read(&failed[SMB2_NEGOTIATE_HE]));
764         seq_printf(m, "\nSessionSetups: %d sent %d failed",
765                    atomic_read(&sent[SMB2_SESSION_SETUP_HE]),
766                    atomic_read(&failed[SMB2_SESSION_SETUP_HE]));
767         seq_printf(m, "\nLogoffs: %d sent %d failed",
768                    atomic_read(&sent[SMB2_LOGOFF_HE]),
769                    atomic_read(&failed[SMB2_LOGOFF_HE]));
770         seq_printf(m, "\nTreeConnects: %d sent %d failed",
771                    atomic_read(&sent[SMB2_TREE_CONNECT_HE]),
772                    atomic_read(&failed[SMB2_TREE_CONNECT_HE]));
773         seq_printf(m, "\nTreeDisconnects: %d sent %d failed",
774                    atomic_read(&sent[SMB2_TREE_DISCONNECT_HE]),
775                    atomic_read(&failed[SMB2_TREE_DISCONNECT_HE]));
776         seq_printf(m, "\nCreates: %d sent %d failed",
777                    atomic_read(&sent[SMB2_CREATE_HE]),
778                    atomic_read(&failed[SMB2_CREATE_HE]));
779         seq_printf(m, "\nCloses: %d sent %d failed",
780                    atomic_read(&sent[SMB2_CLOSE_HE]),
781                    atomic_read(&failed[SMB2_CLOSE_HE]));
782         seq_printf(m, "\nFlushes: %d sent %d failed",
783                    atomic_read(&sent[SMB2_FLUSH_HE]),
784                    atomic_read(&failed[SMB2_FLUSH_HE]));
785         seq_printf(m, "\nReads: %d sent %d failed",
786                    atomic_read(&sent[SMB2_READ_HE]),
787                    atomic_read(&failed[SMB2_READ_HE]));
788         seq_printf(m, "\nWrites: %d sent %d failed",
789                    atomic_read(&sent[SMB2_WRITE_HE]),
790                    atomic_read(&failed[SMB2_WRITE_HE]));
791         seq_printf(m, "\nLocks: %d sent %d failed",
792                    atomic_read(&sent[SMB2_LOCK_HE]),
793                    atomic_read(&failed[SMB2_LOCK_HE]));
794         seq_printf(m, "\nIOCTLs: %d sent %d failed",
795                    atomic_read(&sent[SMB2_IOCTL_HE]),
796                    atomic_read(&failed[SMB2_IOCTL_HE]));
797         seq_printf(m, "\nCancels: %d sent %d failed",
798                    atomic_read(&sent[SMB2_CANCEL_HE]),
799                    atomic_read(&failed[SMB2_CANCEL_HE]));
800         seq_printf(m, "\nEchos: %d sent %d failed",
801                    atomic_read(&sent[SMB2_ECHO_HE]),
802                    atomic_read(&failed[SMB2_ECHO_HE]));
803         seq_printf(m, "\nQueryDirectories: %d sent %d failed",
804                    atomic_read(&sent[SMB2_QUERY_DIRECTORY_HE]),
805                    atomic_read(&failed[SMB2_QUERY_DIRECTORY_HE]));
806         seq_printf(m, "\nChangeNotifies: %d sent %d failed",
807                    atomic_read(&sent[SMB2_CHANGE_NOTIFY_HE]),
808                    atomic_read(&failed[SMB2_CHANGE_NOTIFY_HE]));
809         seq_printf(m, "\nQueryInfos: %d sent %d failed",
810                    atomic_read(&sent[SMB2_QUERY_INFO_HE]),
811                    atomic_read(&failed[SMB2_QUERY_INFO_HE]));
812         seq_printf(m, "\nSetInfos: %d sent %d failed",
813                    atomic_read(&sent[SMB2_SET_INFO_HE]),
814                    atomic_read(&failed[SMB2_SET_INFO_HE]));
815         seq_printf(m, "\nOplockBreaks: %d sent %d failed",
816                    atomic_read(&sent[SMB2_OPLOCK_BREAK_HE]),
817                    atomic_read(&failed[SMB2_OPLOCK_BREAK_HE]));
818 #endif
819 }
820
821 static void
822 smb2_set_fid(struct cifsFileInfo *cfile, struct cifs_fid *fid, __u32 oplock)
823 {
824         struct cifsInodeInfo *cinode = CIFS_I(d_inode(cfile->dentry));
825         struct TCP_Server_Info *server = tlink_tcon(cfile->tlink)->ses->server;
826
827         cfile->fid.persistent_fid = fid->persistent_fid;
828         cfile->fid.volatile_fid = fid->volatile_fid;
829         server->ops->set_oplock_level(cinode, oplock, fid->epoch,
830                                       &fid->purge_cache);
831         cinode->can_cache_brlcks = CIFS_CACHE_WRITE(cinode);
832         memcpy(cfile->fid.create_guid, fid->create_guid, 16);
833 }
834
835 static void
836 smb2_close_file(const unsigned int xid, struct cifs_tcon *tcon,
837                 struct cifs_fid *fid)
838 {
839         SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
840 }
841
842 static int
843 SMB2_request_res_key(const unsigned int xid, struct cifs_tcon *tcon,
844                      u64 persistent_fid, u64 volatile_fid,
845                      struct copychunk_ioctl *pcchunk)
846 {
847         int rc;
848         unsigned int ret_data_len;
849         struct resume_key_req *res_key;
850
851         rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
852                         FSCTL_SRV_REQUEST_RESUME_KEY, true /* is_fsctl */,
853                         NULL, 0 /* no input */,
854                         (char **)&res_key, &ret_data_len);
855
856         if (rc) {
857                 cifs_dbg(VFS, "refcpy ioctl error %d getting resume key\n", rc);
858                 goto req_res_key_exit;
859         }
860         if (ret_data_len < sizeof(struct resume_key_req)) {
861                 cifs_dbg(VFS, "Invalid refcopy resume key length\n");
862                 rc = -EINVAL;
863                 goto req_res_key_exit;
864         }
865         memcpy(pcchunk->SourceKey, res_key->ResumeKey, COPY_CHUNK_RES_KEY_SIZE);
866
867 req_res_key_exit:
868         kfree(res_key);
869         return rc;
870 }
871
872 static ssize_t
873 smb2_copychunk_range(const unsigned int xid,
874                         struct cifsFileInfo *srcfile,
875                         struct cifsFileInfo *trgtfile, u64 src_off,
876                         u64 len, u64 dest_off)
877 {
878         int rc;
879         unsigned int ret_data_len;
880         struct copychunk_ioctl *pcchunk;
881         struct copychunk_ioctl_rsp *retbuf = NULL;
882         struct cifs_tcon *tcon;
883         int chunks_copied = 0;
884         bool chunk_sizes_updated = false;
885         ssize_t bytes_written, total_bytes_written = 0;
886
887         pcchunk = kmalloc(sizeof(struct copychunk_ioctl), GFP_KERNEL);
888
889         if (pcchunk == NULL)
890                 return -ENOMEM;
891
892         cifs_dbg(FYI, "in smb2_copychunk_range - about to call request res key\n");
893         /* Request a key from the server to identify the source of the copy */
894         rc = SMB2_request_res_key(xid, tlink_tcon(srcfile->tlink),
895                                 srcfile->fid.persistent_fid,
896                                 srcfile->fid.volatile_fid, pcchunk);
897
898         /* Note: request_res_key sets res_key null only if rc !=0 */
899         if (rc)
900                 goto cchunk_out;
901
902         /* For now array only one chunk long, will make more flexible later */
903         pcchunk->ChunkCount = cpu_to_le32(1);
904         pcchunk->Reserved = 0;
905         pcchunk->Reserved2 = 0;
906
907         tcon = tlink_tcon(trgtfile->tlink);
908
909         while (len > 0) {
910                 pcchunk->SourceOffset = cpu_to_le64(src_off);
911                 pcchunk->TargetOffset = cpu_to_le64(dest_off);
912                 pcchunk->Length =
913                         cpu_to_le32(min_t(u32, len, tcon->max_bytes_chunk));
914
915                 /* Request server copy to target from src identified by key */
916                 rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
917                         trgtfile->fid.volatile_fid, FSCTL_SRV_COPYCHUNK_WRITE,
918                         true /* is_fsctl */, (char *)pcchunk,
919                         sizeof(struct copychunk_ioctl), (char **)&retbuf,
920                         &ret_data_len);
921                 if (rc == 0) {
922                         if (ret_data_len !=
923                                         sizeof(struct copychunk_ioctl_rsp)) {
924                                 cifs_dbg(VFS, "invalid cchunk response size\n");
925                                 rc = -EIO;
926                                 goto cchunk_out;
927                         }
928                         if (retbuf->TotalBytesWritten == 0) {
929                                 cifs_dbg(FYI, "no bytes copied\n");
930                                 rc = -EIO;
931                                 goto cchunk_out;
932                         }
933                         /*
934                          * Check if server claimed to write more than we asked
935                          */
936                         if (le32_to_cpu(retbuf->TotalBytesWritten) >
937                             le32_to_cpu(pcchunk->Length)) {
938                                 cifs_dbg(VFS, "invalid copy chunk response\n");
939                                 rc = -EIO;
940                                 goto cchunk_out;
941                         }
942                         if (le32_to_cpu(retbuf->ChunksWritten) != 1) {
943                                 cifs_dbg(VFS, "invalid num chunks written\n");
944                                 rc = -EIO;
945                                 goto cchunk_out;
946                         }
947                         chunks_copied++;
948
949                         bytes_written = le32_to_cpu(retbuf->TotalBytesWritten);
950                         src_off += bytes_written;
951                         dest_off += bytes_written;
952                         len -= bytes_written;
953                         total_bytes_written += bytes_written;
954
955                         cifs_dbg(FYI, "Chunks %d PartialChunk %d Total %zu\n",
956                                 le32_to_cpu(retbuf->ChunksWritten),
957                                 le32_to_cpu(retbuf->ChunkBytesWritten),
958                                 bytes_written);
959                 } else if (rc == -EINVAL) {
960                         if (ret_data_len != sizeof(struct copychunk_ioctl_rsp))
961                                 goto cchunk_out;
962
963                         cifs_dbg(FYI, "MaxChunks %d BytesChunk %d MaxCopy %d\n",
964                                 le32_to_cpu(retbuf->ChunksWritten),
965                                 le32_to_cpu(retbuf->ChunkBytesWritten),
966                                 le32_to_cpu(retbuf->TotalBytesWritten));
967
968                         /*
969                          * Check if this is the first request using these sizes,
970                          * (ie check if copy succeed once with original sizes
971                          * and check if the server gave us different sizes after
972                          * we already updated max sizes on previous request).
973                          * if not then why is the server returning an error now
974                          */
975                         if ((chunks_copied != 0) || chunk_sizes_updated)
976                                 goto cchunk_out;
977
978                         /* Check that server is not asking us to grow size */
979                         if (le32_to_cpu(retbuf->ChunkBytesWritten) <
980                                         tcon->max_bytes_chunk)
981                                 tcon->max_bytes_chunk =
982                                         le32_to_cpu(retbuf->ChunkBytesWritten);
983                         else
984                                 goto cchunk_out; /* server gave us bogus size */
985
986                         /* No need to change MaxChunks since already set to 1 */
987                         chunk_sizes_updated = true;
988                 } else
989                         goto cchunk_out;
990         }
991
992 cchunk_out:
993         kfree(pcchunk);
994         kfree(retbuf);
995         if (rc)
996                 return rc;
997         else
998                 return total_bytes_written;
999 }
1000
1001 static int
1002 smb2_flush_file(const unsigned int xid, struct cifs_tcon *tcon,
1003                 struct cifs_fid *fid)
1004 {
1005         return SMB2_flush(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1006 }
1007
1008 static unsigned int
1009 smb2_read_data_offset(char *buf)
1010 {
1011         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1012         return rsp->DataOffset;
1013 }
1014
1015 static unsigned int
1016 smb2_read_data_length(char *buf, bool in_remaining)
1017 {
1018         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
1019
1020         if (in_remaining)
1021                 return le32_to_cpu(rsp->DataRemaining);
1022
1023         return le32_to_cpu(rsp->DataLength);
1024 }
1025
1026
1027 static int
1028 smb2_sync_read(const unsigned int xid, struct cifs_fid *pfid,
1029                struct cifs_io_parms *parms, unsigned int *bytes_read,
1030                char **buf, int *buf_type)
1031 {
1032         parms->persistent_fid = pfid->persistent_fid;
1033         parms->volatile_fid = pfid->volatile_fid;
1034         return SMB2_read(xid, parms, bytes_read, buf, buf_type);
1035 }
1036
1037 static int
1038 smb2_sync_write(const unsigned int xid, struct cifs_fid *pfid,
1039                 struct cifs_io_parms *parms, unsigned int *written,
1040                 struct kvec *iov, unsigned long nr_segs)
1041 {
1042
1043         parms->persistent_fid = pfid->persistent_fid;
1044         parms->volatile_fid = pfid->volatile_fid;
1045         return SMB2_write(xid, parms, written, iov, nr_segs);
1046 }
1047
1048 /* Set or clear the SPARSE_FILE attribute based on value passed in setsparse */
1049 static bool smb2_set_sparse(const unsigned int xid, struct cifs_tcon *tcon,
1050                 struct cifsFileInfo *cfile, struct inode *inode, __u8 setsparse)
1051 {
1052         struct cifsInodeInfo *cifsi;
1053         int rc;
1054
1055         cifsi = CIFS_I(inode);
1056
1057         /* if file already sparse don't bother setting sparse again */
1058         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && setsparse)
1059                 return true; /* already sparse */
1060
1061         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && !setsparse)
1062                 return true; /* already not sparse */
1063
1064         /*
1065          * Can't check for sparse support on share the usual way via the
1066          * FS attribute info (FILE_SUPPORTS_SPARSE_FILES) on the share
1067          * since Samba server doesn't set the flag on the share, yet
1068          * supports the set sparse FSCTL and returns sparse correctly
1069          * in the file attributes. If we fail setting sparse though we
1070          * mark that server does not support sparse files for this share
1071          * to avoid repeatedly sending the unsupported fsctl to server
1072          * if the file is repeatedly extended.
1073          */
1074         if (tcon->broken_sparse_sup)
1075                 return false;
1076
1077         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1078                         cfile->fid.volatile_fid, FSCTL_SET_SPARSE,
1079                         true /* is_fctl */,
1080                         &setsparse, 1, NULL, NULL);
1081         if (rc) {
1082                 tcon->broken_sparse_sup = true;
1083                 cifs_dbg(FYI, "set sparse rc = %d\n", rc);
1084                 return false;
1085         }
1086
1087         if (setsparse)
1088                 cifsi->cifsAttrs |= FILE_ATTRIBUTE_SPARSE_FILE;
1089         else
1090                 cifsi->cifsAttrs &= (~FILE_ATTRIBUTE_SPARSE_FILE);
1091
1092         return true;
1093 }
1094
1095 static int
1096 smb2_set_file_size(const unsigned int xid, struct cifs_tcon *tcon,
1097                    struct cifsFileInfo *cfile, __u64 size, bool set_alloc)
1098 {
1099         __le64 eof = cpu_to_le64(size);
1100         struct inode *inode;
1101
1102         /*
1103          * If extending file more than one page make sparse. Many Linux fs
1104          * make files sparse by default when extending via ftruncate
1105          */
1106         inode = d_inode(cfile->dentry);
1107
1108         if (!set_alloc && (size > inode->i_size + 8192)) {
1109                 __u8 set_sparse = 1;
1110
1111                 /* whether set sparse succeeds or not, extend the file */
1112                 smb2_set_sparse(xid, tcon, cfile, inode, set_sparse);
1113         }
1114
1115         return SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
1116                             cfile->fid.volatile_fid, cfile->pid, &eof, false);
1117 }
1118
1119 static int
1120 smb2_duplicate_extents(const unsigned int xid,
1121                         struct cifsFileInfo *srcfile,
1122                         struct cifsFileInfo *trgtfile, u64 src_off,
1123                         u64 len, u64 dest_off)
1124 {
1125         int rc;
1126         unsigned int ret_data_len;
1127         struct duplicate_extents_to_file dup_ext_buf;
1128         struct cifs_tcon *tcon = tlink_tcon(trgtfile->tlink);
1129
1130         /* server fileays advertise duplicate extent support with this flag */
1131         if ((le32_to_cpu(tcon->fsAttrInfo.Attributes) &
1132              FILE_SUPPORTS_BLOCK_REFCOUNTING) == 0)
1133                 return -EOPNOTSUPP;
1134
1135         dup_ext_buf.VolatileFileHandle = srcfile->fid.volatile_fid;
1136         dup_ext_buf.PersistentFileHandle = srcfile->fid.persistent_fid;
1137         dup_ext_buf.SourceFileOffset = cpu_to_le64(src_off);
1138         dup_ext_buf.TargetFileOffset = cpu_to_le64(dest_off);
1139         dup_ext_buf.ByteCount = cpu_to_le64(len);
1140         cifs_dbg(FYI, "duplicate extents: src off %lld dst off %lld len %lld",
1141                 src_off, dest_off, len);
1142
1143         rc = smb2_set_file_size(xid, tcon, trgtfile, dest_off + len, false);
1144         if (rc)
1145                 goto duplicate_extents_out;
1146
1147         rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
1148                         trgtfile->fid.volatile_fid,
1149                         FSCTL_DUPLICATE_EXTENTS_TO_FILE,
1150                         true /* is_fsctl */,
1151                         (char *)&dup_ext_buf,
1152                         sizeof(struct duplicate_extents_to_file),
1153                         NULL,
1154                         &ret_data_len);
1155
1156         if (ret_data_len > 0)
1157                 cifs_dbg(FYI, "non-zero response length in duplicate extents");
1158
1159 duplicate_extents_out:
1160         return rc;
1161 }
1162
1163 static int
1164 smb2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
1165                    struct cifsFileInfo *cfile)
1166 {
1167         return SMB2_set_compression(xid, tcon, cfile->fid.persistent_fid,
1168                             cfile->fid.volatile_fid);
1169 }
1170
1171 static int
1172 smb3_set_integrity(const unsigned int xid, struct cifs_tcon *tcon,
1173                    struct cifsFileInfo *cfile)
1174 {
1175         struct fsctl_set_integrity_information_req integr_info;
1176         unsigned int ret_data_len;
1177
1178         integr_info.ChecksumAlgorithm = cpu_to_le16(CHECKSUM_TYPE_UNCHANGED);
1179         integr_info.Flags = 0;
1180         integr_info.Reserved = 0;
1181
1182         return SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1183                         cfile->fid.volatile_fid,
1184                         FSCTL_SET_INTEGRITY_INFORMATION,
1185                         true /* is_fsctl */,
1186                         (char *)&integr_info,
1187                         sizeof(struct fsctl_set_integrity_information_req),
1188                         NULL,
1189                         &ret_data_len);
1190
1191 }
1192
1193 static int
1194 smb3_enum_snapshots(const unsigned int xid, struct cifs_tcon *tcon,
1195                    struct cifsFileInfo *cfile, void __user *ioc_buf)
1196 {
1197         char *retbuf = NULL;
1198         unsigned int ret_data_len = 0;
1199         int rc;
1200         struct smb_snapshot_array snapshot_in;
1201
1202         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1203                         cfile->fid.volatile_fid,
1204                         FSCTL_SRV_ENUMERATE_SNAPSHOTS,
1205                         true /* is_fsctl */,
1206                         NULL, 0 /* no input data */,
1207                         (char **)&retbuf,
1208                         &ret_data_len);
1209         cifs_dbg(FYI, "enum snaphots ioctl returned %d and ret buflen is %d\n",
1210                         rc, ret_data_len);
1211         if (rc)
1212                 return rc;
1213
1214         if (ret_data_len && (ioc_buf != NULL) && (retbuf != NULL)) {
1215                 /* Fixup buffer */
1216                 if (copy_from_user(&snapshot_in, ioc_buf,
1217                     sizeof(struct smb_snapshot_array))) {
1218                         rc = -EFAULT;
1219                         kfree(retbuf);
1220                         return rc;
1221                 }
1222                 if (snapshot_in.snapshot_array_size < sizeof(struct smb_snapshot_array)) {
1223                         rc = -ERANGE;
1224                         kfree(retbuf);
1225                         return rc;
1226                 }
1227
1228                 if (ret_data_len > snapshot_in.snapshot_array_size)
1229                         ret_data_len = snapshot_in.snapshot_array_size;
1230
1231                 if (copy_to_user(ioc_buf, retbuf, ret_data_len))
1232                         rc = -EFAULT;
1233         }
1234
1235         kfree(retbuf);
1236         return rc;
1237 }
1238
1239 static int
1240 smb2_query_dir_first(const unsigned int xid, struct cifs_tcon *tcon,
1241                      const char *path, struct cifs_sb_info *cifs_sb,
1242                      struct cifs_fid *fid, __u16 search_flags,
1243                      struct cifs_search_info *srch_inf)
1244 {
1245         __le16 *utf16_path;
1246         int rc;
1247         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1248         struct cifs_open_parms oparms;
1249
1250         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1251         if (!utf16_path)
1252                 return -ENOMEM;
1253
1254         oparms.tcon = tcon;
1255         oparms.desired_access = FILE_READ_ATTRIBUTES | FILE_READ_DATA;
1256         oparms.disposition = FILE_OPEN;
1257         oparms.create_options = 0;
1258         oparms.fid = fid;
1259         oparms.reconnect = false;
1260
1261         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
1262         kfree(utf16_path);
1263         if (rc) {
1264                 cifs_dbg(FYI, "open dir failed rc=%d\n", rc);
1265                 return rc;
1266         }
1267
1268         srch_inf->entries_in_buffer = 0;
1269         srch_inf->index_of_last_entry = 0;
1270
1271         rc = SMB2_query_directory(xid, tcon, fid->persistent_fid,
1272                                   fid->volatile_fid, 0, srch_inf);
1273         if (rc) {
1274                 cifs_dbg(FYI, "query directory failed rc=%d\n", rc);
1275                 SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1276         }
1277         return rc;
1278 }
1279
1280 static int
1281 smb2_query_dir_next(const unsigned int xid, struct cifs_tcon *tcon,
1282                     struct cifs_fid *fid, __u16 search_flags,
1283                     struct cifs_search_info *srch_inf)
1284 {
1285         return SMB2_query_directory(xid, tcon, fid->persistent_fid,
1286                                     fid->volatile_fid, 0, srch_inf);
1287 }
1288
1289 static int
1290 smb2_close_dir(const unsigned int xid, struct cifs_tcon *tcon,
1291                struct cifs_fid *fid)
1292 {
1293         return SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
1294 }
1295
1296 /*
1297 * If we negotiate SMB2 protocol and get STATUS_PENDING - update
1298 * the number of credits and return true. Otherwise - return false.
1299 */
1300 static bool
1301 smb2_is_status_pending(char *buf, struct TCP_Server_Info *server, int length)
1302 {
1303         struct smb2_sync_hdr *shdr = get_sync_hdr(buf);
1304
1305         if (shdr->Status != STATUS_PENDING)
1306                 return false;
1307
1308         if (!length) {
1309                 spin_lock(&server->req_lock);
1310                 server->credits += le16_to_cpu(shdr->CreditRequest);
1311                 spin_unlock(&server->req_lock);
1312                 wake_up(&server->request_q);
1313         }
1314
1315         return true;
1316 }
1317
1318 static bool
1319 smb2_is_session_expired(char *buf)
1320 {
1321         struct smb2_sync_hdr *shdr = get_sync_hdr(buf);
1322
1323         if (shdr->Status != STATUS_NETWORK_SESSION_EXPIRED)
1324                 return false;
1325
1326         cifs_dbg(FYI, "Session expired\n");
1327         return true;
1328 }
1329
1330 static int
1331 smb2_oplock_response(struct cifs_tcon *tcon, struct cifs_fid *fid,
1332                      struct cifsInodeInfo *cinode)
1333 {
1334         if (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_LEASING)
1335                 return SMB2_lease_break(0, tcon, cinode->lease_key,
1336                                         smb2_get_lease_state(cinode));
1337
1338         return SMB2_oplock_break(0, tcon, fid->persistent_fid,
1339                                  fid->volatile_fid,
1340                                  CIFS_CACHE_READ(cinode) ? 1 : 0);
1341 }
1342
1343 static int
1344 smb2_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
1345              struct kstatfs *buf)
1346 {
1347         int rc;
1348         __le16 srch_path = 0; /* Null - open root of share */
1349         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1350         struct cifs_open_parms oparms;
1351         struct cifs_fid fid;
1352
1353         oparms.tcon = tcon;
1354         oparms.desired_access = FILE_READ_ATTRIBUTES;
1355         oparms.disposition = FILE_OPEN;
1356         oparms.create_options = 0;
1357         oparms.fid = &fid;
1358         oparms.reconnect = false;
1359
1360         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
1361         if (rc)
1362                 return rc;
1363         buf->f_type = SMB2_MAGIC_NUMBER;
1364         rc = SMB2_QFS_info(xid, tcon, fid.persistent_fid, fid.volatile_fid,
1365                            buf);
1366         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
1367         return rc;
1368 }
1369
1370 static bool
1371 smb2_compare_fids(struct cifsFileInfo *ob1, struct cifsFileInfo *ob2)
1372 {
1373         return ob1->fid.persistent_fid == ob2->fid.persistent_fid &&
1374                ob1->fid.volatile_fid == ob2->fid.volatile_fid;
1375 }
1376
1377 static int
1378 smb2_mand_lock(const unsigned int xid, struct cifsFileInfo *cfile, __u64 offset,
1379                __u64 length, __u32 type, int lock, int unlock, bool wait)
1380 {
1381         if (unlock && !lock)
1382                 type = SMB2_LOCKFLAG_UNLOCK;
1383         return SMB2_lock(xid, tlink_tcon(cfile->tlink),
1384                          cfile->fid.persistent_fid, cfile->fid.volatile_fid,
1385                          current->tgid, length, offset, type, wait);
1386 }
1387
1388 static void
1389 smb2_get_lease_key(struct inode *inode, struct cifs_fid *fid)
1390 {
1391         memcpy(fid->lease_key, CIFS_I(inode)->lease_key, SMB2_LEASE_KEY_SIZE);
1392 }
1393
1394 static void
1395 smb2_set_lease_key(struct inode *inode, struct cifs_fid *fid)
1396 {
1397         memcpy(CIFS_I(inode)->lease_key, fid->lease_key, SMB2_LEASE_KEY_SIZE);
1398 }
1399
1400 static void
1401 smb2_new_lease_key(struct cifs_fid *fid)
1402 {
1403         generate_random_uuid(fid->lease_key);
1404 }
1405
1406 static int
1407 smb2_get_dfs_refer(const unsigned int xid, struct cifs_ses *ses,
1408                    const char *search_name,
1409                    struct dfs_info3_param **target_nodes,
1410                    unsigned int *num_of_nodes,
1411                    const struct nls_table *nls_codepage, int remap)
1412 {
1413         int rc;
1414         __le16 *utf16_path = NULL;
1415         int utf16_path_len = 0;
1416         struct cifs_tcon *tcon;
1417         struct fsctl_get_dfs_referral_req *dfs_req = NULL;
1418         struct get_dfs_referral_rsp *dfs_rsp = NULL;
1419         u32 dfs_req_size = 0, dfs_rsp_size = 0;
1420
1421         cifs_dbg(FYI, "smb2_get_dfs_refer path <%s>\n", search_name);
1422
1423         /*
1424          * Try to use the IPC tcon, otherwise just use any
1425          */
1426         tcon = ses->tcon_ipc;
1427         if (tcon == NULL) {
1428                 spin_lock(&cifs_tcp_ses_lock);
1429                 tcon = list_first_entry_or_null(&ses->tcon_list,
1430                                                 struct cifs_tcon,
1431                                                 tcon_list);
1432                 if (tcon)
1433                         tcon->tc_count++;
1434                 spin_unlock(&cifs_tcp_ses_lock);
1435         }
1436
1437         if (tcon == NULL) {
1438                 cifs_dbg(VFS, "session %p has no tcon available for a dfs referral request\n",
1439                          ses);
1440                 rc = -ENOTCONN;
1441                 goto out;
1442         }
1443
1444         utf16_path = cifs_strndup_to_utf16(search_name, PATH_MAX,
1445                                            &utf16_path_len,
1446                                            nls_codepage, remap);
1447         if (!utf16_path) {
1448                 rc = -ENOMEM;
1449                 goto out;
1450         }
1451
1452         dfs_req_size = sizeof(*dfs_req) + utf16_path_len;
1453         dfs_req = kzalloc(dfs_req_size, GFP_KERNEL);
1454         if (!dfs_req) {
1455                 rc = -ENOMEM;
1456                 goto out;
1457         }
1458
1459         /* Highest DFS referral version understood */
1460         dfs_req->MaxReferralLevel = DFS_VERSION;
1461
1462         /* Path to resolve in an UTF-16 null-terminated string */
1463         memcpy(dfs_req->RequestFileName, utf16_path, utf16_path_len);
1464
1465         do {
1466                 rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
1467                                 FSCTL_DFS_GET_REFERRALS,
1468                                 true /* is_fsctl */,
1469                                 (char *)dfs_req, dfs_req_size,
1470                                 (char **)&dfs_rsp, &dfs_rsp_size);
1471         } while (rc == -EAGAIN);
1472
1473         if (rc) {
1474                 if ((rc != -ENOENT) && (rc != -EOPNOTSUPP))
1475                         cifs_dbg(VFS, "ioctl error in smb2_get_dfs_refer rc=%d\n", rc);
1476                 goto out;
1477         }
1478
1479         rc = parse_dfs_referrals(dfs_rsp, dfs_rsp_size,
1480                                  num_of_nodes, target_nodes,
1481                                  nls_codepage, remap, search_name,
1482                                  true /* is_unicode */);
1483         if (rc) {
1484                 cifs_dbg(VFS, "parse error in smb2_get_dfs_refer rc=%d\n", rc);
1485                 goto out;
1486         }
1487
1488  out:
1489         if (tcon && !tcon->ipc) {
1490                 /* ipc tcons are not refcounted */
1491                 spin_lock(&cifs_tcp_ses_lock);
1492                 tcon->tc_count--;
1493                 spin_unlock(&cifs_tcp_ses_lock);
1494         }
1495         kfree(utf16_path);
1496         kfree(dfs_req);
1497         kfree(dfs_rsp);
1498         return rc;
1499 }
1500 #define SMB2_SYMLINK_STRUCT_SIZE \
1501         (sizeof(struct smb2_err_rsp) - 1 + sizeof(struct smb2_symlink_err_rsp))
1502
1503 static int
1504 smb2_query_symlink(const unsigned int xid, struct cifs_tcon *tcon,
1505                    const char *full_path, char **target_path,
1506                    struct cifs_sb_info *cifs_sb)
1507 {
1508         int rc;
1509         __le16 *utf16_path;
1510         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1511         struct cifs_open_parms oparms;
1512         struct cifs_fid fid;
1513         struct kvec err_iov = {NULL, 0};
1514         struct smb2_err_rsp *err_buf;
1515         struct smb2_symlink_err_rsp *symlink;
1516         unsigned int sub_len;
1517         unsigned int sub_offset;
1518         unsigned int print_len;
1519         unsigned int print_offset;
1520         struct cifs_ses *ses = tcon->ses;
1521         struct TCP_Server_Info *server = ses->server;
1522
1523         cifs_dbg(FYI, "%s: path: %s\n", __func__, full_path);
1524
1525         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
1526         if (!utf16_path)
1527                 return -ENOMEM;
1528
1529         oparms.tcon = tcon;
1530         oparms.desired_access = FILE_READ_ATTRIBUTES;
1531         oparms.disposition = FILE_OPEN;
1532         oparms.create_options = 0;
1533         oparms.fid = &fid;
1534         oparms.reconnect = false;
1535
1536         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, &err_iov);
1537
1538         if (!rc || !err_iov.iov_base) {
1539                 kfree(utf16_path);
1540                 return -ENOENT;
1541         }
1542
1543         err_buf = err_iov.iov_base;
1544         if (le32_to_cpu(err_buf->ByteCount) < sizeof(struct smb2_symlink_err_rsp) ||
1545             err_iov.iov_len + server->vals->header_preamble_size < SMB2_SYMLINK_STRUCT_SIZE) {
1546                 kfree(utf16_path);
1547                 return -ENOENT;
1548         }
1549
1550         /* open must fail on symlink - reset rc */
1551         rc = 0;
1552         symlink = (struct smb2_symlink_err_rsp *)err_buf->ErrorData;
1553         sub_len = le16_to_cpu(symlink->SubstituteNameLength);
1554         sub_offset = le16_to_cpu(symlink->SubstituteNameOffset);
1555         print_len = le16_to_cpu(symlink->PrintNameLength);
1556         print_offset = le16_to_cpu(symlink->PrintNameOffset);
1557
1558         if (err_iov.iov_len + server->vals->header_preamble_size <
1559                         SMB2_SYMLINK_STRUCT_SIZE + sub_offset + sub_len) {
1560                 kfree(utf16_path);
1561                 return -ENOENT;
1562         }
1563
1564         if (err_iov.iov_len + server->vals->header_preamble_size <
1565                         SMB2_SYMLINK_STRUCT_SIZE + print_offset + print_len) {
1566                 kfree(utf16_path);
1567                 return -ENOENT;
1568         }
1569
1570         *target_path = cifs_strndup_from_utf16(
1571                                 (char *)symlink->PathBuffer + sub_offset,
1572                                 sub_len, true, cifs_sb->local_nls);
1573         if (!(*target_path)) {
1574                 kfree(utf16_path);
1575                 return -ENOMEM;
1576         }
1577         convert_delimiter(*target_path, '/');
1578         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
1579         kfree(utf16_path);
1580         return rc;
1581 }
1582
1583 #ifdef CONFIG_CIFS_ACL
1584 static struct cifs_ntsd *
1585 get_smb2_acl_by_fid(struct cifs_sb_info *cifs_sb,
1586                 const struct cifs_fid *cifsfid, u32 *pacllen)
1587 {
1588         struct cifs_ntsd *pntsd = NULL;
1589         unsigned int xid;
1590         int rc = -EOPNOTSUPP;
1591         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
1592
1593         if (IS_ERR(tlink))
1594                 return ERR_CAST(tlink);
1595
1596         xid = get_xid();
1597         cifs_dbg(FYI, "trying to get acl\n");
1598
1599         rc = SMB2_query_acl(xid, tlink_tcon(tlink), cifsfid->persistent_fid,
1600                             cifsfid->volatile_fid, (void **)&pntsd, pacllen);
1601         free_xid(xid);
1602
1603         cifs_put_tlink(tlink);
1604
1605         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
1606         if (rc)
1607                 return ERR_PTR(rc);
1608         return pntsd;
1609
1610 }
1611
1612 static struct cifs_ntsd *
1613 get_smb2_acl_by_path(struct cifs_sb_info *cifs_sb,
1614                 const char *path, u32 *pacllen)
1615 {
1616         struct cifs_ntsd *pntsd = NULL;
1617         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1618         unsigned int xid;
1619         int rc;
1620         struct cifs_tcon *tcon;
1621         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
1622         struct cifs_fid fid;
1623         struct cifs_open_parms oparms;
1624         __le16 *utf16_path;
1625
1626         cifs_dbg(FYI, "get smb3 acl for path %s\n", path);
1627         if (IS_ERR(tlink))
1628                 return ERR_CAST(tlink);
1629
1630         tcon = tlink_tcon(tlink);
1631         xid = get_xid();
1632
1633         if (backup_cred(cifs_sb))
1634                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
1635         else
1636                 oparms.create_options = 0;
1637
1638         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1639         if (!utf16_path)
1640                 return ERR_PTR(-ENOMEM);
1641
1642         oparms.tcon = tcon;
1643         oparms.desired_access = READ_CONTROL;
1644         oparms.disposition = FILE_OPEN;
1645         oparms.fid = &fid;
1646         oparms.reconnect = false;
1647
1648         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
1649         kfree(utf16_path);
1650         if (!rc) {
1651                 rc = SMB2_query_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
1652                             fid.volatile_fid, (void **)&pntsd, pacllen);
1653                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
1654         }
1655
1656         cifs_put_tlink(tlink);
1657         free_xid(xid);
1658
1659         cifs_dbg(FYI, "%s: rc = %d ACL len %d\n", __func__, rc, *pacllen);
1660         if (rc)
1661                 return ERR_PTR(rc);
1662         return pntsd;
1663 }
1664
1665 #ifdef CONFIG_CIFS_ACL
1666 static int
1667 set_smb2_acl(struct cifs_ntsd *pnntsd, __u32 acllen,
1668                 struct inode *inode, const char *path, int aclflag)
1669 {
1670         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
1671         unsigned int xid;
1672         int rc, access_flags = 0;
1673         struct cifs_tcon *tcon;
1674         struct cifs_sb_info *cifs_sb = CIFS_SB(inode->i_sb);
1675         struct tcon_link *tlink = cifs_sb_tlink(cifs_sb);
1676         struct cifs_fid fid;
1677         struct cifs_open_parms oparms;
1678         __le16 *utf16_path;
1679
1680         cifs_dbg(FYI, "set smb3 acl for path %s\n", path);
1681         if (IS_ERR(tlink))
1682                 return PTR_ERR(tlink);
1683
1684         tcon = tlink_tcon(tlink);
1685         xid = get_xid();
1686
1687         if (backup_cred(cifs_sb))
1688                 oparms.create_options = CREATE_OPEN_BACKUP_INTENT;
1689         else
1690                 oparms.create_options = 0;
1691
1692         if (aclflag == CIFS_ACL_OWNER || aclflag == CIFS_ACL_GROUP)
1693                 access_flags = WRITE_OWNER;
1694         else
1695                 access_flags = WRITE_DAC;
1696
1697         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
1698         if (!utf16_path)
1699                 return -ENOMEM;
1700
1701         oparms.tcon = tcon;
1702         oparms.desired_access = access_flags;
1703         oparms.disposition = FILE_OPEN;
1704         oparms.path = path;
1705         oparms.fid = &fid;
1706         oparms.reconnect = false;
1707
1708         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
1709         kfree(utf16_path);
1710         if (!rc) {
1711                 rc = SMB2_set_acl(xid, tlink_tcon(tlink), fid.persistent_fid,
1712                             fid.volatile_fid, pnntsd, acllen, aclflag);
1713                 SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
1714         }
1715
1716         cifs_put_tlink(tlink);
1717         free_xid(xid);
1718         return rc;
1719 }
1720 #endif /* CIFS_ACL */
1721
1722 /* Retrieve an ACL from the server */
1723 static struct cifs_ntsd *
1724 get_smb2_acl(struct cifs_sb_info *cifs_sb,
1725                                       struct inode *inode, const char *path,
1726                                       u32 *pacllen)
1727 {
1728         struct cifs_ntsd *pntsd = NULL;
1729         struct cifsFileInfo *open_file = NULL;
1730
1731         if (inode)
1732                 open_file = find_readable_file(CIFS_I(inode), true);
1733         if (!open_file)
1734                 return get_smb2_acl_by_path(cifs_sb, path, pacllen);
1735
1736         pntsd = get_smb2_acl_by_fid(cifs_sb, &open_file->fid, pacllen);
1737         cifsFileInfo_put(open_file);
1738         return pntsd;
1739 }
1740 #endif
1741
1742 static long smb3_zero_range(struct file *file, struct cifs_tcon *tcon,
1743                             loff_t offset, loff_t len, bool keep_size)
1744 {
1745         struct inode *inode;
1746         struct cifsInodeInfo *cifsi;
1747         struct cifsFileInfo *cfile = file->private_data;
1748         struct file_zero_data_information fsctl_buf;
1749         long rc;
1750         unsigned int xid;
1751
1752         xid = get_xid();
1753
1754         inode = d_inode(cfile->dentry);
1755         cifsi = CIFS_I(inode);
1756
1757         /* if file not oplocked can't be sure whether asking to extend size */
1758         if (!CIFS_CACHE_READ(cifsi))
1759                 if (keep_size == false)
1760                         return -EOPNOTSUPP;
1761
1762         /*
1763          * Must check if file sparse since fallocate -z (zero range) assumes
1764          * non-sparse allocation
1765          */
1766         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE))
1767                 return -EOPNOTSUPP;
1768
1769         /*
1770          * need to make sure we are not asked to extend the file since the SMB3
1771          * fsctl does not change the file size. In the future we could change
1772          * this to zero the first part of the range then set the file size
1773          * which for a non sparse file would zero the newly extended range
1774          */
1775         if (keep_size == false)
1776                 if (i_size_read(inode) < offset + len)
1777                         return -EOPNOTSUPP;
1778
1779         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
1780
1781         fsctl_buf.FileOffset = cpu_to_le64(offset);
1782         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
1783
1784         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1785                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
1786                         true /* is_fctl */, (char *)&fsctl_buf,
1787                         sizeof(struct file_zero_data_information), NULL, NULL);
1788         free_xid(xid);
1789         return rc;
1790 }
1791
1792 static long smb3_punch_hole(struct file *file, struct cifs_tcon *tcon,
1793                             loff_t offset, loff_t len)
1794 {
1795         struct inode *inode;
1796         struct cifsInodeInfo *cifsi;
1797         struct cifsFileInfo *cfile = file->private_data;
1798         struct file_zero_data_information fsctl_buf;
1799         long rc;
1800         unsigned int xid;
1801         __u8 set_sparse = 1;
1802
1803         xid = get_xid();
1804
1805         inode = d_inode(cfile->dentry);
1806         cifsi = CIFS_I(inode);
1807
1808         /* Need to make file sparse, if not already, before freeing range. */
1809         /* Consider adding equivalent for compressed since it could also work */
1810         if (!smb2_set_sparse(xid, tcon, cfile, inode, set_sparse))
1811                 return -EOPNOTSUPP;
1812
1813         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
1814
1815         fsctl_buf.FileOffset = cpu_to_le64(offset);
1816         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
1817
1818         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1819                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
1820                         true /* is_fctl */, (char *)&fsctl_buf,
1821                         sizeof(struct file_zero_data_information), NULL, NULL);
1822         free_xid(xid);
1823         return rc;
1824 }
1825
1826 static long smb3_simple_falloc(struct file *file, struct cifs_tcon *tcon,
1827                             loff_t off, loff_t len, bool keep_size)
1828 {
1829         struct inode *inode;
1830         struct cifsInodeInfo *cifsi;
1831         struct cifsFileInfo *cfile = file->private_data;
1832         long rc = -EOPNOTSUPP;
1833         unsigned int xid;
1834
1835         xid = get_xid();
1836
1837         inode = d_inode(cfile->dentry);
1838         cifsi = CIFS_I(inode);
1839
1840         /* if file not oplocked can't be sure whether asking to extend size */
1841         if (!CIFS_CACHE_READ(cifsi))
1842                 if (keep_size == false)
1843                         return -EOPNOTSUPP;
1844
1845         /*
1846          * Files are non-sparse by default so falloc may be a no-op
1847          * Must check if file sparse. If not sparse, and not extending
1848          * then no need to do anything since file already allocated
1849          */
1850         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) == 0) {
1851                 if (keep_size == true)
1852                         return 0;
1853                 /* check if extending file */
1854                 else if (i_size_read(inode) >= off + len)
1855                         /* not extending file and already not sparse */
1856                         return 0;
1857                 /* BB: in future add else clause to extend file */
1858                 else
1859                         return -EOPNOTSUPP;
1860         }
1861
1862         if ((keep_size == true) || (i_size_read(inode) >= off + len)) {
1863                 /*
1864                  * Check if falloc starts within first few pages of file
1865                  * and ends within a few pages of the end of file to
1866                  * ensure that most of file is being forced to be
1867                  * fallocated now. If so then setting whole file sparse
1868                  * ie potentially making a few extra pages at the beginning
1869                  * or end of the file non-sparse via set_sparse is harmless.
1870                  */
1871                 if ((off > 8192) || (off + len + 8192 < i_size_read(inode)))
1872                         return -EOPNOTSUPP;
1873
1874                 rc = smb2_set_sparse(xid, tcon, cfile, inode, false);
1875         }
1876         /* BB: else ... in future add code to extend file and set sparse */
1877
1878
1879         free_xid(xid);
1880         return rc;
1881 }
1882
1883
1884 static long smb3_fallocate(struct file *file, struct cifs_tcon *tcon, int mode,
1885                            loff_t off, loff_t len)
1886 {
1887         /* KEEP_SIZE already checked for by do_fallocate */
1888         if (mode & FALLOC_FL_PUNCH_HOLE)
1889                 return smb3_punch_hole(file, tcon, off, len);
1890         else if (mode & FALLOC_FL_ZERO_RANGE) {
1891                 if (mode & FALLOC_FL_KEEP_SIZE)
1892                         return smb3_zero_range(file, tcon, off, len, true);
1893                 return smb3_zero_range(file, tcon, off, len, false);
1894         } else if (mode == FALLOC_FL_KEEP_SIZE)
1895                 return smb3_simple_falloc(file, tcon, off, len, true);
1896         else if (mode == 0)
1897                 return smb3_simple_falloc(file, tcon, off, len, false);
1898
1899         return -EOPNOTSUPP;
1900 }
1901
1902 static void
1903 smb2_downgrade_oplock(struct TCP_Server_Info *server,
1904                         struct cifsInodeInfo *cinode, bool set_level2)
1905 {
1906         if (set_level2)
1907                 server->ops->set_oplock_level(cinode, SMB2_OPLOCK_LEVEL_II,
1908                                                 0, NULL);
1909         else
1910                 server->ops->set_oplock_level(cinode, 0, 0, NULL);
1911 }
1912
1913 static void
1914 smb2_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1915                       unsigned int epoch, bool *purge_cache)
1916 {
1917         oplock &= 0xFF;
1918         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
1919                 return;
1920         if (oplock == SMB2_OPLOCK_LEVEL_BATCH) {
1921                 cinode->oplock = CIFS_CACHE_RHW_FLG;
1922                 cifs_dbg(FYI, "Batch Oplock granted on inode %p\n",
1923                          &cinode->vfs_inode);
1924         } else if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE) {
1925                 cinode->oplock = CIFS_CACHE_RW_FLG;
1926                 cifs_dbg(FYI, "Exclusive Oplock granted on inode %p\n",
1927                          &cinode->vfs_inode);
1928         } else if (oplock == SMB2_OPLOCK_LEVEL_II) {
1929                 cinode->oplock = CIFS_CACHE_READ_FLG;
1930                 cifs_dbg(FYI, "Level II Oplock granted on inode %p\n",
1931                          &cinode->vfs_inode);
1932         } else
1933                 cinode->oplock = 0;
1934 }
1935
1936 static void
1937 smb21_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1938                        unsigned int epoch, bool *purge_cache)
1939 {
1940         char message[5] = {0};
1941
1942         oplock &= 0xFF;
1943         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
1944                 return;
1945
1946         cinode->oplock = 0;
1947         if (oplock & SMB2_LEASE_READ_CACHING_HE) {
1948                 cinode->oplock |= CIFS_CACHE_READ_FLG;
1949                 strcat(message, "R");
1950         }
1951         if (oplock & SMB2_LEASE_HANDLE_CACHING_HE) {
1952                 cinode->oplock |= CIFS_CACHE_HANDLE_FLG;
1953                 strcat(message, "H");
1954         }
1955         if (oplock & SMB2_LEASE_WRITE_CACHING_HE) {
1956                 cinode->oplock |= CIFS_CACHE_WRITE_FLG;
1957                 strcat(message, "W");
1958         }
1959         if (!cinode->oplock)
1960                 strcat(message, "None");
1961         cifs_dbg(FYI, "%s Lease granted on inode %p\n", message,
1962                  &cinode->vfs_inode);
1963 }
1964
1965 static void
1966 smb3_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1967                       unsigned int epoch, bool *purge_cache)
1968 {
1969         unsigned int old_oplock = cinode->oplock;
1970
1971         smb21_set_oplock_level(cinode, oplock, epoch, purge_cache);
1972
1973         if (purge_cache) {
1974                 *purge_cache = false;
1975                 if (old_oplock == CIFS_CACHE_READ_FLG) {
1976                         if (cinode->oplock == CIFS_CACHE_READ_FLG &&
1977                             (epoch - cinode->epoch > 0))
1978                                 *purge_cache = true;
1979                         else if (cinode->oplock == CIFS_CACHE_RH_FLG &&
1980                                  (epoch - cinode->epoch > 1))
1981                                 *purge_cache = true;
1982                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
1983                                  (epoch - cinode->epoch > 1))
1984                                 *purge_cache = true;
1985                         else if (cinode->oplock == 0 &&
1986                                  (epoch - cinode->epoch > 0))
1987                                 *purge_cache = true;
1988                 } else if (old_oplock == CIFS_CACHE_RH_FLG) {
1989                         if (cinode->oplock == CIFS_CACHE_RH_FLG &&
1990                             (epoch - cinode->epoch > 0))
1991                                 *purge_cache = true;
1992                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
1993                                  (epoch - cinode->epoch > 1))
1994                                 *purge_cache = true;
1995                 }
1996                 cinode->epoch = epoch;
1997         }
1998 }
1999
2000 static bool
2001 smb2_is_read_op(__u32 oplock)
2002 {
2003         return oplock == SMB2_OPLOCK_LEVEL_II;
2004 }
2005
2006 static bool
2007 smb21_is_read_op(__u32 oplock)
2008 {
2009         return (oplock & SMB2_LEASE_READ_CACHING_HE) &&
2010                !(oplock & SMB2_LEASE_WRITE_CACHING_HE);
2011 }
2012
2013 static __le32
2014 map_oplock_to_lease(u8 oplock)
2015 {
2016         if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE)
2017                 return SMB2_LEASE_WRITE_CACHING | SMB2_LEASE_READ_CACHING;
2018         else if (oplock == SMB2_OPLOCK_LEVEL_II)
2019                 return SMB2_LEASE_READ_CACHING;
2020         else if (oplock == SMB2_OPLOCK_LEVEL_BATCH)
2021                 return SMB2_LEASE_HANDLE_CACHING | SMB2_LEASE_READ_CACHING |
2022                        SMB2_LEASE_WRITE_CACHING;
2023         return 0;
2024 }
2025
2026 static char *
2027 smb2_create_lease_buf(u8 *lease_key, u8 oplock)
2028 {
2029         struct create_lease *buf;
2030
2031         buf = kzalloc(sizeof(struct create_lease), GFP_KERNEL);
2032         if (!buf)
2033                 return NULL;
2034
2035         buf->lcontext.LeaseKeyLow = cpu_to_le64(*((u64 *)lease_key));
2036         buf->lcontext.LeaseKeyHigh = cpu_to_le64(*((u64 *)(lease_key + 8)));
2037         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
2038
2039         buf->ccontext.DataOffset = cpu_to_le16(offsetof
2040                                         (struct create_lease, lcontext));
2041         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context));
2042         buf->ccontext.NameOffset = cpu_to_le16(offsetof
2043                                 (struct create_lease, Name));
2044         buf->ccontext.NameLength = cpu_to_le16(4);
2045         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
2046         buf->Name[0] = 'R';
2047         buf->Name[1] = 'q';
2048         buf->Name[2] = 'L';
2049         buf->Name[3] = 's';
2050         return (char *)buf;
2051 }
2052
2053 static char *
2054 smb3_create_lease_buf(u8 *lease_key, u8 oplock)
2055 {
2056         struct create_lease_v2 *buf;
2057
2058         buf = kzalloc(sizeof(struct create_lease_v2), GFP_KERNEL);
2059         if (!buf)
2060                 return NULL;
2061
2062         buf->lcontext.LeaseKeyLow = cpu_to_le64(*((u64 *)lease_key));
2063         buf->lcontext.LeaseKeyHigh = cpu_to_le64(*((u64 *)(lease_key + 8)));
2064         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
2065
2066         buf->ccontext.DataOffset = cpu_to_le16(offsetof
2067                                         (struct create_lease_v2, lcontext));
2068         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2));
2069         buf->ccontext.NameOffset = cpu_to_le16(offsetof
2070                                 (struct create_lease_v2, Name));
2071         buf->ccontext.NameLength = cpu_to_le16(4);
2072         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
2073         buf->Name[0] = 'R';
2074         buf->Name[1] = 'q';
2075         buf->Name[2] = 'L';
2076         buf->Name[3] = 's';
2077         return (char *)buf;
2078 }
2079
2080 static __u8
2081 smb2_parse_lease_buf(void *buf, unsigned int *epoch)
2082 {
2083         struct create_lease *lc = (struct create_lease *)buf;
2084
2085         *epoch = 0; /* not used */
2086         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
2087                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
2088         return le32_to_cpu(lc->lcontext.LeaseState);
2089 }
2090
2091 static __u8
2092 smb3_parse_lease_buf(void *buf, unsigned int *epoch)
2093 {
2094         struct create_lease_v2 *lc = (struct create_lease_v2 *)buf;
2095
2096         *epoch = le16_to_cpu(lc->lcontext.Epoch);
2097         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
2098                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
2099         return le32_to_cpu(lc->lcontext.LeaseState);
2100 }
2101
2102 static unsigned int
2103 smb2_wp_retry_size(struct inode *inode)
2104 {
2105         return min_t(unsigned int, CIFS_SB(inode->i_sb)->wsize,
2106                      SMB2_MAX_BUFFER_SIZE);
2107 }
2108
2109 static bool
2110 smb2_dir_needs_close(struct cifsFileInfo *cfile)
2111 {
2112         return !cfile->invalidHandle;
2113 }
2114
2115 static void
2116 fill_transform_hdr(struct TCP_Server_Info *server,
2117                    struct smb2_transform_hdr *tr_hdr, struct smb_rqst *old_rq)
2118 {
2119         struct smb2_sync_hdr *shdr =
2120                         (struct smb2_sync_hdr *)old_rq->rq_iov[1].iov_base;
2121         unsigned int orig_len = get_rfc1002_length(old_rq->rq_iov[0].iov_base);
2122
2123         memset(tr_hdr, 0, sizeof(struct smb2_transform_hdr));
2124         tr_hdr->ProtocolId = SMB2_TRANSFORM_PROTO_NUM;
2125         tr_hdr->OriginalMessageSize = cpu_to_le32(orig_len);
2126         tr_hdr->Flags = cpu_to_le16(0x01);
2127         get_random_bytes(&tr_hdr->Nonce, SMB3_AES128CMM_NONCE);
2128         memcpy(&tr_hdr->SessionId, &shdr->SessionId, 8);
2129         inc_rfc1001_len(tr_hdr, sizeof(struct smb2_transform_hdr) - server->vals->header_preamble_size);
2130         inc_rfc1001_len(tr_hdr, orig_len);
2131 }
2132
2133 /* We can not use the normal sg_set_buf() as we will sometimes pass a
2134  * stack object as buf.
2135  */
2136 static inline void smb2_sg_set_buf(struct scatterlist *sg, const void *buf,
2137                                    unsigned int buflen)
2138 {
2139         sg_set_page(sg, virt_to_page(buf), buflen, offset_in_page(buf));
2140 }
2141
2142 static struct scatterlist *
2143 init_sg(struct smb_rqst *rqst, u8 *sign)
2144 {
2145         unsigned int sg_len = rqst->rq_nvec + rqst->rq_npages + 1;
2146         unsigned int assoc_data_len = sizeof(struct smb2_transform_hdr) - 24;
2147         struct scatterlist *sg;
2148         unsigned int i;
2149         unsigned int j;
2150
2151         sg = kmalloc_array(sg_len, sizeof(struct scatterlist), GFP_KERNEL);
2152         if (!sg)
2153                 return NULL;
2154
2155         sg_init_table(sg, sg_len);
2156         smb2_sg_set_buf(&sg[0], rqst->rq_iov[0].iov_base + 24, assoc_data_len);
2157         for (i = 1; i < rqst->rq_nvec; i++)
2158                 smb2_sg_set_buf(&sg[i], rqst->rq_iov[i].iov_base,
2159                                                 rqst->rq_iov[i].iov_len);
2160         for (j = 0; i < sg_len - 1; i++, j++) {
2161                 unsigned int len = (j < rqst->rq_npages - 1) ? rqst->rq_pagesz
2162                                                         : rqst->rq_tailsz;
2163                 sg_set_page(&sg[i], rqst->rq_pages[j], len, 0);
2164         }
2165         smb2_sg_set_buf(&sg[sg_len - 1], sign, SMB2_SIGNATURE_SIZE);
2166         return sg;
2167 }
2168
2169 static int
2170 smb2_get_enc_key(struct TCP_Server_Info *server, __u64 ses_id, int enc, u8 *key)
2171 {
2172         struct cifs_ses *ses;
2173         u8 *ses_enc_key;
2174
2175         spin_lock(&cifs_tcp_ses_lock);
2176         list_for_each_entry(ses, &server->smb_ses_list, smb_ses_list) {
2177                 if (ses->Suid != ses_id)
2178                         continue;
2179                 ses_enc_key = enc ? ses->smb3encryptionkey :
2180                                                         ses->smb3decryptionkey;
2181                 memcpy(key, ses_enc_key, SMB3_SIGN_KEY_SIZE);
2182                 spin_unlock(&cifs_tcp_ses_lock);
2183                 return 0;
2184         }
2185         spin_unlock(&cifs_tcp_ses_lock);
2186
2187         return 1;
2188 }
2189 /*
2190  * Encrypt or decrypt @rqst message. @rqst has the following format:
2191  * iov[0] - transform header (associate data),
2192  * iov[1-N] and pages - data to encrypt.
2193  * On success return encrypted data in iov[1-N] and pages, leave iov[0]
2194  * untouched.
2195  */
2196 static int
2197 crypt_message(struct TCP_Server_Info *server, struct smb_rqst *rqst, int enc)
2198 {
2199         struct smb2_transform_hdr *tr_hdr =
2200                         (struct smb2_transform_hdr *)rqst->rq_iov[0].iov_base;
2201         unsigned int assoc_data_len = sizeof(struct smb2_transform_hdr) - 20 - server->vals->header_preamble_size;
2202         int rc = 0;
2203         struct scatterlist *sg;
2204         u8 sign[SMB2_SIGNATURE_SIZE] = {};
2205         u8 key[SMB3_SIGN_KEY_SIZE];
2206         struct aead_request *req;
2207         char *iv;
2208         unsigned int iv_len;
2209         DECLARE_CRYPTO_WAIT(wait);
2210         struct crypto_aead *tfm;
2211         unsigned int crypt_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
2212
2213         rc = smb2_get_enc_key(server, tr_hdr->SessionId, enc, key);
2214         if (rc) {
2215                 cifs_dbg(VFS, "%s: Could not get %scryption key\n", __func__,
2216                          enc ? "en" : "de");
2217                 return 0;
2218         }
2219
2220         rc = smb3_crypto_aead_allocate(server);
2221         if (rc) {
2222                 cifs_dbg(VFS, "%s: crypto alloc failed\n", __func__);
2223                 return rc;
2224         }
2225
2226         tfm = enc ? server->secmech.ccmaesencrypt :
2227                                                 server->secmech.ccmaesdecrypt;
2228         rc = crypto_aead_setkey(tfm, key, SMB3_SIGN_KEY_SIZE);
2229         if (rc) {
2230                 cifs_dbg(VFS, "%s: Failed to set aead key %d\n", __func__, rc);
2231                 return rc;
2232         }
2233
2234         rc = crypto_aead_setauthsize(tfm, SMB2_SIGNATURE_SIZE);
2235         if (rc) {
2236                 cifs_dbg(VFS, "%s: Failed to set authsize %d\n", __func__, rc);
2237                 return rc;
2238         }
2239
2240         req = aead_request_alloc(tfm, GFP_KERNEL);
2241         if (!req) {
2242                 cifs_dbg(VFS, "%s: Failed to alloc aead request", __func__);
2243                 return -ENOMEM;
2244         }
2245
2246         if (!enc) {
2247                 memcpy(sign, &tr_hdr->Signature, SMB2_SIGNATURE_SIZE);
2248                 crypt_len += SMB2_SIGNATURE_SIZE;
2249         }
2250
2251         sg = init_sg(rqst, sign);
2252         if (!sg) {
2253                 cifs_dbg(VFS, "%s: Failed to init sg", __func__);
2254                 rc = -ENOMEM;
2255                 goto free_req;
2256         }
2257
2258         iv_len = crypto_aead_ivsize(tfm);
2259         iv = kzalloc(iv_len, GFP_KERNEL);
2260         if (!iv) {
2261                 cifs_dbg(VFS, "%s: Failed to alloc IV", __func__);
2262                 rc = -ENOMEM;
2263                 goto free_sg;
2264         }
2265         iv[0] = 3;
2266         memcpy(iv + 1, (char *)tr_hdr->Nonce, SMB3_AES128CMM_NONCE);
2267
2268         aead_request_set_crypt(req, sg, sg, crypt_len, iv);
2269         aead_request_set_ad(req, assoc_data_len);
2270
2271         aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
2272                                   crypto_req_done, &wait);
2273
2274         rc = crypto_wait_req(enc ? crypto_aead_encrypt(req)
2275                                 : crypto_aead_decrypt(req), &wait);
2276
2277         if (!rc && enc)
2278                 memcpy(&tr_hdr->Signature, sign, SMB2_SIGNATURE_SIZE);
2279
2280         kfree(iv);
2281 free_sg:
2282         kfree(sg);
2283 free_req:
2284         kfree(req);
2285         return rc;
2286 }
2287
2288 static int
2289 smb3_init_transform_rq(struct TCP_Server_Info *server, struct smb_rqst *new_rq,
2290                        struct smb_rqst *old_rq)
2291 {
2292         struct kvec *iov;
2293         struct page **pages;
2294         struct smb2_transform_hdr *tr_hdr;
2295         unsigned int npages = old_rq->rq_npages;
2296         int i;
2297         int rc = -ENOMEM;
2298
2299         pages = kmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
2300         if (!pages)
2301                 return rc;
2302
2303         new_rq->rq_pages = pages;
2304         new_rq->rq_npages = old_rq->rq_npages;
2305         new_rq->rq_pagesz = old_rq->rq_pagesz;
2306         new_rq->rq_tailsz = old_rq->rq_tailsz;
2307
2308         for (i = 0; i < npages; i++) {
2309                 pages[i] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
2310                 if (!pages[i])
2311                         goto err_free_pages;
2312         }
2313
2314         iov = kmalloc_array(old_rq->rq_nvec, sizeof(struct kvec), GFP_KERNEL);
2315         if (!iov)
2316                 goto err_free_pages;
2317
2318         /* copy all iovs from the old except the 1st one (rfc1002 length) */
2319         memcpy(&iov[1], &old_rq->rq_iov[1],
2320                                 sizeof(struct kvec) * (old_rq->rq_nvec - 1));
2321         new_rq->rq_iov = iov;
2322         new_rq->rq_nvec = old_rq->rq_nvec;
2323
2324         tr_hdr = kmalloc(sizeof(struct smb2_transform_hdr), GFP_KERNEL);
2325         if (!tr_hdr)
2326                 goto err_free_iov;
2327
2328         /* fill the 1st iov with a transform header */
2329         fill_transform_hdr(server, tr_hdr, old_rq);
2330         new_rq->rq_iov[0].iov_base = tr_hdr;
2331         new_rq->rq_iov[0].iov_len = sizeof(struct smb2_transform_hdr);
2332
2333         /* copy pages form the old */
2334         for (i = 0; i < npages; i++) {
2335                 char *dst = kmap(new_rq->rq_pages[i]);
2336                 char *src = kmap(old_rq->rq_pages[i]);
2337                 unsigned int len = (i < npages - 1) ? new_rq->rq_pagesz :
2338                                                         new_rq->rq_tailsz;
2339                 memcpy(dst, src, len);
2340                 kunmap(new_rq->rq_pages[i]);
2341                 kunmap(old_rq->rq_pages[i]);
2342         }
2343
2344         rc = crypt_message(server, new_rq, 1);
2345         cifs_dbg(FYI, "encrypt message returned %d", rc);
2346         if (rc)
2347                 goto err_free_tr_hdr;
2348
2349         return rc;
2350
2351 err_free_tr_hdr:
2352         kfree(tr_hdr);
2353 err_free_iov:
2354         kfree(iov);
2355 err_free_pages:
2356         for (i = i - 1; i >= 0; i--)
2357                 put_page(pages[i]);
2358         kfree(pages);
2359         return rc;
2360 }
2361
2362 static void
2363 smb3_free_transform_rq(struct smb_rqst *rqst)
2364 {
2365         int i = rqst->rq_npages - 1;
2366
2367         for (; i >= 0; i--)
2368                 put_page(rqst->rq_pages[i]);
2369         kfree(rqst->rq_pages);
2370         /* free transform header */
2371         kfree(rqst->rq_iov[0].iov_base);
2372         kfree(rqst->rq_iov);
2373 }
2374
2375 static int
2376 smb3_is_transform_hdr(void *buf)
2377 {
2378         struct smb2_transform_hdr *trhdr = buf;
2379
2380         return trhdr->ProtocolId == SMB2_TRANSFORM_PROTO_NUM;
2381 }
2382
2383 static int
2384 decrypt_raw_data(struct TCP_Server_Info *server, char *buf,
2385                  unsigned int buf_data_size, struct page **pages,
2386                  unsigned int npages, unsigned int page_data_size)
2387 {
2388         struct kvec iov[2];
2389         struct smb_rqst rqst = {NULL};
2390         struct smb2_hdr *hdr;
2391         int rc;
2392
2393         iov[0].iov_base = buf;
2394         iov[0].iov_len = sizeof(struct smb2_transform_hdr);
2395         iov[1].iov_base = buf + sizeof(struct smb2_transform_hdr);
2396         iov[1].iov_len = buf_data_size;
2397
2398         rqst.rq_iov = iov;
2399         rqst.rq_nvec = 2;
2400         rqst.rq_pages = pages;
2401         rqst.rq_npages = npages;
2402         rqst.rq_pagesz = PAGE_SIZE;
2403         rqst.rq_tailsz = (page_data_size % PAGE_SIZE) ? : PAGE_SIZE;
2404
2405         rc = crypt_message(server, &rqst, 0);
2406         cifs_dbg(FYI, "decrypt message returned %d\n", rc);
2407
2408         if (rc)
2409                 return rc;
2410
2411         memmove(buf + server->vals->header_preamble_size, iov[1].iov_base, buf_data_size);
2412         hdr = (struct smb2_hdr *)buf;
2413         hdr->smb2_buf_length = cpu_to_be32(buf_data_size + page_data_size);
2414         server->total_read = buf_data_size + page_data_size + server->vals->header_preamble_size;
2415
2416         return rc;
2417 }
2418
2419 static int
2420 read_data_into_pages(struct TCP_Server_Info *server, struct page **pages,
2421                      unsigned int npages, unsigned int len)
2422 {
2423         int i;
2424         int length;
2425
2426         for (i = 0; i < npages; i++) {
2427                 struct page *page = pages[i];
2428                 size_t n;
2429
2430                 n = len;
2431                 if (len >= PAGE_SIZE) {
2432                         /* enough data to fill the page */
2433                         n = PAGE_SIZE;
2434                         len -= n;
2435                 } else {
2436                         zero_user(page, len, PAGE_SIZE - len);
2437                         len = 0;
2438                 }
2439                 length = cifs_read_page_from_socket(server, page, n);
2440                 if (length < 0)
2441                         return length;
2442                 server->total_read += length;
2443         }
2444
2445         return 0;
2446 }
2447
2448 static int
2449 init_read_bvec(struct page **pages, unsigned int npages, unsigned int data_size,
2450                unsigned int cur_off, struct bio_vec **page_vec)
2451 {
2452         struct bio_vec *bvec;
2453         int i;
2454
2455         bvec = kcalloc(npages, sizeof(struct bio_vec), GFP_KERNEL);
2456         if (!bvec)
2457                 return -ENOMEM;
2458
2459         for (i = 0; i < npages; i++) {
2460                 bvec[i].bv_page = pages[i];
2461                 bvec[i].bv_offset = (i == 0) ? cur_off : 0;
2462                 bvec[i].bv_len = min_t(unsigned int, PAGE_SIZE, data_size);
2463                 data_size -= bvec[i].bv_len;
2464         }
2465
2466         if (data_size != 0) {
2467                 cifs_dbg(VFS, "%s: something went wrong\n", __func__);
2468                 kfree(bvec);
2469                 return -EIO;
2470         }
2471
2472         *page_vec = bvec;
2473         return 0;
2474 }
2475
2476 static int
2477 handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid,
2478                  char *buf, unsigned int buf_len, struct page **pages,
2479                  unsigned int npages, unsigned int page_data_size)
2480 {
2481         unsigned int data_offset;
2482         unsigned int data_len;
2483         unsigned int cur_off;
2484         unsigned int cur_page_idx;
2485         unsigned int pad_len;
2486         struct cifs_readdata *rdata = mid->callback_data;
2487         struct smb2_sync_hdr *shdr = get_sync_hdr(buf);
2488         struct bio_vec *bvec = NULL;
2489         struct iov_iter iter;
2490         struct kvec iov;
2491         int length;
2492         bool use_rdma_mr = false;
2493
2494         if (shdr->Command != SMB2_READ) {
2495                 cifs_dbg(VFS, "only big read responses are supported\n");
2496                 return -ENOTSUPP;
2497         }
2498
2499         if (server->ops->is_session_expired &&
2500             server->ops->is_session_expired(buf)) {
2501                 cifs_reconnect(server);
2502                 wake_up(&server->response_q);
2503                 return -1;
2504         }
2505
2506         if (server->ops->is_status_pending &&
2507                         server->ops->is_status_pending(buf, server, 0))
2508                 return -1;
2509
2510         rdata->result = server->ops->map_error(buf, false);
2511         if (rdata->result != 0) {
2512                 cifs_dbg(FYI, "%s: server returned error %d\n",
2513                          __func__, rdata->result);
2514                 dequeue_mid(mid, rdata->result);
2515                 return 0;
2516         }
2517
2518         data_offset = server->ops->read_data_offset(buf) + server->vals->header_preamble_size;
2519 #ifdef CONFIG_CIFS_SMB_DIRECT
2520         use_rdma_mr = rdata->mr;
2521 #endif
2522         data_len = server->ops->read_data_length(buf, use_rdma_mr);
2523
2524         if (data_offset < server->vals->read_rsp_size) {
2525                 /*
2526                  * win2k8 sometimes sends an offset of 0 when the read
2527                  * is beyond the EOF. Treat it as if the data starts just after
2528                  * the header.
2529                  */
2530                 cifs_dbg(FYI, "%s: data offset (%u) inside read response header\n",
2531                          __func__, data_offset);
2532                 data_offset = server->vals->read_rsp_size;
2533         } else if (data_offset > MAX_CIFS_SMALL_BUFFER_SIZE) {
2534                 /* data_offset is beyond the end of smallbuf */
2535                 cifs_dbg(FYI, "%s: data offset (%u) beyond end of smallbuf\n",
2536                          __func__, data_offset);
2537                 rdata->result = -EIO;
2538                 dequeue_mid(mid, rdata->result);
2539                 return 0;
2540         }
2541
2542         pad_len = data_offset - server->vals->read_rsp_size;
2543
2544         if (buf_len <= data_offset) {
2545                 /* read response payload is in pages */
2546                 cur_page_idx = pad_len / PAGE_SIZE;
2547                 cur_off = pad_len % PAGE_SIZE;
2548
2549                 if (cur_page_idx != 0) {
2550                         /* data offset is beyond the 1st page of response */
2551                         cifs_dbg(FYI, "%s: data offset (%u) beyond 1st page of response\n",
2552                                  __func__, data_offset);
2553                         rdata->result = -EIO;
2554                         dequeue_mid(mid, rdata->result);
2555                         return 0;
2556                 }
2557
2558                 if (data_len > page_data_size - pad_len) {
2559                         /* data_len is corrupt -- discard frame */
2560                         rdata->result = -EIO;
2561                         dequeue_mid(mid, rdata->result);
2562                         return 0;
2563                 }
2564
2565                 rdata->result = init_read_bvec(pages, npages, page_data_size,
2566                                                cur_off, &bvec);
2567                 if (rdata->result != 0) {
2568                         dequeue_mid(mid, rdata->result);
2569                         return 0;
2570                 }
2571
2572                 iov_iter_bvec(&iter, WRITE | ITER_BVEC, bvec, npages, data_len);
2573         } else if (buf_len >= data_offset + data_len) {
2574                 /* read response payload is in buf */
2575                 WARN_ONCE(npages > 0, "read data can be either in buf or in pages");
2576                 iov.iov_base = buf + data_offset;
2577                 iov.iov_len = data_len;
2578                 iov_iter_kvec(&iter, WRITE | ITER_KVEC, &iov, 1, data_len);
2579         } else {
2580                 /* read response payload cannot be in both buf and pages */
2581                 WARN_ONCE(1, "buf can not contain only a part of read data");
2582                 rdata->result = -EIO;
2583                 dequeue_mid(mid, rdata->result);
2584                 return 0;
2585         }
2586
2587         /* set up first iov for signature check */
2588         rdata->iov[0].iov_base = buf;
2589         rdata->iov[0].iov_len = 4;
2590         rdata->iov[1].iov_base = buf + 4;
2591         rdata->iov[1].iov_len = server->vals->read_rsp_size - 4;
2592         cifs_dbg(FYI, "0: iov_base=%p iov_len=%zu\n",
2593                  rdata->iov[0].iov_base, server->vals->read_rsp_size);
2594
2595         length = rdata->copy_into_pages(server, rdata, &iter);
2596
2597         kfree(bvec);
2598
2599         if (length < 0)
2600                 return length;
2601
2602         dequeue_mid(mid, false);
2603         return length;
2604 }
2605
2606 static int
2607 receive_encrypted_read(struct TCP_Server_Info *server, struct mid_q_entry **mid)
2608 {
2609         char *buf = server->smallbuf;
2610         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
2611         unsigned int npages;
2612         struct page **pages;
2613         unsigned int len;
2614         unsigned int buflen = server->pdu_size + server->vals->header_preamble_size;
2615         int rc;
2616         int i = 0;
2617
2618         len = min_t(unsigned int, buflen, server->vals->read_rsp_size -
2619                 server->vals->header_preamble_size +
2620                 sizeof(struct smb2_transform_hdr)) - HEADER_SIZE(server) + 1;
2621
2622         rc = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1, len);
2623         if (rc < 0)
2624                 return rc;
2625         server->total_read += rc;
2626
2627         len = le32_to_cpu(tr_hdr->OriginalMessageSize) +
2628                 server->vals->header_preamble_size -
2629                 server->vals->read_rsp_size;
2630         npages = DIV_ROUND_UP(len, PAGE_SIZE);
2631
2632         pages = kmalloc_array(npages, sizeof(struct page *), GFP_KERNEL);
2633         if (!pages) {
2634                 rc = -ENOMEM;
2635                 goto discard_data;
2636         }
2637
2638         for (; i < npages; i++) {
2639                 pages[i] = alloc_page(GFP_KERNEL|__GFP_HIGHMEM);
2640                 if (!pages[i]) {
2641                         rc = -ENOMEM;
2642                         goto discard_data;
2643                 }
2644         }
2645
2646         /* read read data into pages */
2647         rc = read_data_into_pages(server, pages, npages, len);
2648         if (rc)
2649                 goto free_pages;
2650
2651         rc = cifs_discard_remaining_data(server);
2652         if (rc)
2653                 goto free_pages;
2654
2655         rc = decrypt_raw_data(server, buf, server->vals->read_rsp_size -
2656                               server->vals->header_preamble_size,
2657                               pages, npages, len);
2658         if (rc)
2659                 goto free_pages;
2660
2661         *mid = smb2_find_mid(server, buf);
2662         if (*mid == NULL)
2663                 cifs_dbg(FYI, "mid not found\n");
2664         else {
2665                 cifs_dbg(FYI, "mid found\n");
2666                 (*mid)->decrypted = true;
2667                 rc = handle_read_data(server, *mid, buf,
2668                                       server->vals->read_rsp_size,
2669                                       pages, npages, len);
2670         }
2671
2672 free_pages:
2673         for (i = i - 1; i >= 0; i--)
2674                 put_page(pages[i]);
2675         kfree(pages);
2676         return rc;
2677 discard_data:
2678         cifs_discard_remaining_data(server);
2679         goto free_pages;
2680 }
2681
2682 static int
2683 receive_encrypted_standard(struct TCP_Server_Info *server,
2684                            struct mid_q_entry **mid)
2685 {
2686         int length;
2687         char *buf = server->smallbuf;
2688         unsigned int pdu_length = server->pdu_size;
2689         unsigned int buf_size;
2690         struct mid_q_entry *mid_entry;
2691
2692         /* switch to large buffer if too big for a small one */
2693         if (pdu_length + server->vals->header_preamble_size > MAX_CIFS_SMALL_BUFFER_SIZE) {
2694                 server->large_buf = true;
2695                 memcpy(server->bigbuf, buf, server->total_read);
2696                 buf = server->bigbuf;
2697         }
2698
2699         /* now read the rest */
2700         length = cifs_read_from_socket(server, buf + HEADER_SIZE(server) - 1,
2701                                 pdu_length - HEADER_SIZE(server) + 1 +
2702                                 server->vals->header_preamble_size);
2703         if (length < 0)
2704                 return length;
2705         server->total_read += length;
2706
2707         buf_size = pdu_length + server->vals->header_preamble_size - sizeof(struct smb2_transform_hdr);
2708         length = decrypt_raw_data(server, buf, buf_size, NULL, 0, 0);
2709         if (length)
2710                 return length;
2711
2712         mid_entry = smb2_find_mid(server, buf);
2713         if (mid_entry == NULL)
2714                 cifs_dbg(FYI, "mid not found\n");
2715         else {
2716                 cifs_dbg(FYI, "mid found\n");
2717                 mid_entry->decrypted = true;
2718         }
2719
2720         *mid = mid_entry;
2721
2722         if (mid_entry && mid_entry->handle)
2723                 return mid_entry->handle(server, mid_entry);
2724
2725         return cifs_handle_standard(server, mid_entry);
2726 }
2727
2728 static int
2729 smb3_receive_transform(struct TCP_Server_Info *server, struct mid_q_entry **mid)
2730 {
2731         char *buf = server->smallbuf;
2732         unsigned int pdu_length = server->pdu_size;
2733         struct smb2_transform_hdr *tr_hdr = (struct smb2_transform_hdr *)buf;
2734         unsigned int orig_len = le32_to_cpu(tr_hdr->OriginalMessageSize);
2735
2736         if (pdu_length + server->vals->header_preamble_size < sizeof(struct smb2_transform_hdr) +
2737                                                 sizeof(struct smb2_sync_hdr)) {
2738                 cifs_dbg(VFS, "Transform message is too small (%u)\n",
2739                          pdu_length);
2740                 cifs_reconnect(server);
2741                 wake_up(&server->response_q);
2742                 return -ECONNABORTED;
2743         }
2744
2745         if (pdu_length + server->vals->header_preamble_size < orig_len + sizeof(struct smb2_transform_hdr)) {
2746                 cifs_dbg(VFS, "Transform message is broken\n");
2747                 cifs_reconnect(server);
2748                 wake_up(&server->response_q);
2749                 return -ECONNABORTED;
2750         }
2751
2752         if (pdu_length + server->vals->header_preamble_size > CIFSMaxBufSize + MAX_HEADER_SIZE(server))
2753                 return receive_encrypted_read(server, mid);
2754
2755         return receive_encrypted_standard(server, mid);
2756 }
2757
2758 int
2759 smb3_handle_read_data(struct TCP_Server_Info *server, struct mid_q_entry *mid)
2760 {
2761         char *buf = server->large_buf ? server->bigbuf : server->smallbuf;
2762
2763         return handle_read_data(server, mid, buf, server->pdu_size +
2764                                 server->vals->header_preamble_size,
2765                                 NULL, 0, 0);
2766 }
2767
2768 struct smb_version_operations smb20_operations = {
2769         .compare_fids = smb2_compare_fids,
2770         .setup_request = smb2_setup_request,
2771         .setup_async_request = smb2_setup_async_request,
2772         .check_receive = smb2_check_receive,
2773         .add_credits = smb2_add_credits,
2774         .set_credits = smb2_set_credits,
2775         .get_credits_field = smb2_get_credits_field,
2776         .get_credits = smb2_get_credits,
2777         .wait_mtu_credits = cifs_wait_mtu_credits,
2778         .get_next_mid = smb2_get_next_mid,
2779         .read_data_offset = smb2_read_data_offset,
2780         .read_data_length = smb2_read_data_length,
2781         .map_error = map_smb2_to_linux_error,
2782         .find_mid = smb2_find_mid,
2783         .check_message = smb2_check_message,
2784         .dump_detail = smb2_dump_detail,
2785         .clear_stats = smb2_clear_stats,
2786         .print_stats = smb2_print_stats,
2787         .is_oplock_break = smb2_is_valid_oplock_break,
2788         .handle_cancelled_mid = smb2_handle_cancelled_mid,
2789         .downgrade_oplock = smb2_downgrade_oplock,
2790         .need_neg = smb2_need_neg,
2791         .negotiate = smb2_negotiate,
2792         .negotiate_wsize = smb2_negotiate_wsize,
2793         .negotiate_rsize = smb2_negotiate_rsize,
2794         .sess_setup = SMB2_sess_setup,
2795         .logoff = SMB2_logoff,
2796         .tree_connect = SMB2_tcon,
2797         .tree_disconnect = SMB2_tdis,
2798         .qfs_tcon = smb2_qfs_tcon,
2799         .is_path_accessible = smb2_is_path_accessible,
2800         .can_echo = smb2_can_echo,
2801         .echo = SMB2_echo,
2802         .query_path_info = smb2_query_path_info,
2803         .get_srv_inum = smb2_get_srv_inum,
2804         .query_file_info = smb2_query_file_info,
2805         .set_path_size = smb2_set_path_size,
2806         .set_file_size = smb2_set_file_size,
2807         .set_file_info = smb2_set_file_info,
2808         .set_compression = smb2_set_compression,
2809         .mkdir = smb2_mkdir,
2810         .mkdir_setinfo = smb2_mkdir_setinfo,
2811         .rmdir = smb2_rmdir,
2812         .unlink = smb2_unlink,
2813         .rename = smb2_rename_path,
2814         .create_hardlink = smb2_create_hardlink,
2815         .query_symlink = smb2_query_symlink,
2816         .query_mf_symlink = smb3_query_mf_symlink,
2817         .create_mf_symlink = smb3_create_mf_symlink,
2818         .open = smb2_open_file,
2819         .set_fid = smb2_set_fid,
2820         .close = smb2_close_file,
2821         .flush = smb2_flush_file,
2822         .async_readv = smb2_async_readv,
2823         .async_writev = smb2_async_writev,
2824         .sync_read = smb2_sync_read,
2825         .sync_write = smb2_sync_write,
2826         .query_dir_first = smb2_query_dir_first,
2827         .query_dir_next = smb2_query_dir_next,
2828         .close_dir = smb2_close_dir,
2829         .calc_smb_size = smb2_calc_size,
2830         .is_status_pending = smb2_is_status_pending,
2831         .is_session_expired = smb2_is_session_expired,
2832         .oplock_response = smb2_oplock_response,
2833         .queryfs = smb2_queryfs,
2834         .mand_lock = smb2_mand_lock,
2835         .mand_unlock_range = smb2_unlock_range,
2836         .push_mand_locks = smb2_push_mandatory_locks,
2837         .get_lease_key = smb2_get_lease_key,
2838         .set_lease_key = smb2_set_lease_key,
2839         .new_lease_key = smb2_new_lease_key,
2840         .calc_signature = smb2_calc_signature,
2841         .is_read_op = smb2_is_read_op,
2842         .set_oplock_level = smb2_set_oplock_level,
2843         .create_lease_buf = smb2_create_lease_buf,
2844         .parse_lease_buf = smb2_parse_lease_buf,
2845         .copychunk_range = smb2_copychunk_range,
2846         .wp_retry_size = smb2_wp_retry_size,
2847         .dir_needs_close = smb2_dir_needs_close,
2848         .get_dfs_refer = smb2_get_dfs_refer,
2849         .select_sectype = smb2_select_sectype,
2850 #ifdef CONFIG_CIFS_XATTR
2851         .query_all_EAs = smb2_query_eas,
2852         .set_EA = smb2_set_ea,
2853 #endif /* CIFS_XATTR */
2854 #ifdef CONFIG_CIFS_ACL
2855         .get_acl = get_smb2_acl,
2856         .get_acl_by_fid = get_smb2_acl_by_fid,
2857         .set_acl = set_smb2_acl,
2858 #endif /* CIFS_ACL */
2859 };
2860
2861 struct smb_version_operations smb21_operations = {
2862         .compare_fids = smb2_compare_fids,
2863         .setup_request = smb2_setup_request,
2864         .setup_async_request = smb2_setup_async_request,
2865         .check_receive = smb2_check_receive,
2866         .add_credits = smb2_add_credits,
2867         .set_credits = smb2_set_credits,
2868         .get_credits_field = smb2_get_credits_field,
2869         .get_credits = smb2_get_credits,
2870         .wait_mtu_credits = smb2_wait_mtu_credits,
2871         .get_next_mid = smb2_get_next_mid,
2872         .read_data_offset = smb2_read_data_offset,
2873         .read_data_length = smb2_read_data_length,
2874         .map_error = map_smb2_to_linux_error,
2875         .find_mid = smb2_find_mid,
2876         .check_message = smb2_check_message,
2877         .dump_detail = smb2_dump_detail,
2878         .clear_stats = smb2_clear_stats,
2879         .print_stats = smb2_print_stats,
2880         .is_oplock_break = smb2_is_valid_oplock_break,
2881         .handle_cancelled_mid = smb2_handle_cancelled_mid,
2882         .downgrade_oplock = smb2_downgrade_oplock,
2883         .need_neg = smb2_need_neg,
2884         .negotiate = smb2_negotiate,
2885         .negotiate_wsize = smb2_negotiate_wsize,
2886         .negotiate_rsize = smb2_negotiate_rsize,
2887         .sess_setup = SMB2_sess_setup,
2888         .logoff = SMB2_logoff,
2889         .tree_connect = SMB2_tcon,
2890         .tree_disconnect = SMB2_tdis,
2891         .qfs_tcon = smb2_qfs_tcon,
2892         .is_path_accessible = smb2_is_path_accessible,
2893         .can_echo = smb2_can_echo,
2894         .echo = SMB2_echo,
2895         .query_path_info = smb2_query_path_info,
2896         .get_srv_inum = smb2_get_srv_inum,
2897         .query_file_info = smb2_query_file_info,
2898         .set_path_size = smb2_set_path_size,
2899         .set_file_size = smb2_set_file_size,
2900         .set_file_info = smb2_set_file_info,
2901         .set_compression = smb2_set_compression,
2902         .mkdir = smb2_mkdir,
2903         .mkdir_setinfo = smb2_mkdir_setinfo,
2904         .rmdir = smb2_rmdir,
2905         .unlink = smb2_unlink,
2906         .rename = smb2_rename_path,
2907         .create_hardlink = smb2_create_hardlink,
2908         .query_symlink = smb2_query_symlink,
2909         .query_mf_symlink = smb3_query_mf_symlink,
2910         .create_mf_symlink = smb3_create_mf_symlink,
2911         .open = smb2_open_file,
2912         .set_fid = smb2_set_fid,
2913         .close = smb2_close_file,
2914         .flush = smb2_flush_file,
2915         .async_readv = smb2_async_readv,
2916         .async_writev = smb2_async_writev,
2917         .sync_read = smb2_sync_read,
2918         .sync_write = smb2_sync_write,
2919         .query_dir_first = smb2_query_dir_first,
2920         .query_dir_next = smb2_query_dir_next,
2921         .close_dir = smb2_close_dir,
2922         .calc_smb_size = smb2_calc_size,
2923         .is_status_pending = smb2_is_status_pending,
2924         .is_session_expired = smb2_is_session_expired,
2925         .oplock_response = smb2_oplock_response,
2926         .queryfs = smb2_queryfs,
2927         .mand_lock = smb2_mand_lock,
2928         .mand_unlock_range = smb2_unlock_range,
2929         .push_mand_locks = smb2_push_mandatory_locks,
2930         .get_lease_key = smb2_get_lease_key,
2931         .set_lease_key = smb2_set_lease_key,
2932         .new_lease_key = smb2_new_lease_key,
2933         .calc_signature = smb2_calc_signature,
2934         .is_read_op = smb21_is_read_op,
2935         .set_oplock_level = smb21_set_oplock_level,
2936         .create_lease_buf = smb2_create_lease_buf,
2937         .parse_lease_buf = smb2_parse_lease_buf,
2938         .copychunk_range = smb2_copychunk_range,
2939         .wp_retry_size = smb2_wp_retry_size,
2940         .dir_needs_close = smb2_dir_needs_close,
2941         .enum_snapshots = smb3_enum_snapshots,
2942         .get_dfs_refer = smb2_get_dfs_refer,
2943         .select_sectype = smb2_select_sectype,
2944 #ifdef CONFIG_CIFS_XATTR
2945         .query_all_EAs = smb2_query_eas,
2946         .set_EA = smb2_set_ea,
2947 #endif /* CIFS_XATTR */
2948 #ifdef CONFIG_CIFS_ACL
2949         .get_acl = get_smb2_acl,
2950         .get_acl_by_fid = get_smb2_acl_by_fid,
2951         .set_acl = set_smb2_acl,
2952 #endif /* CIFS_ACL */
2953 };
2954
2955 struct smb_version_operations smb30_operations = {
2956         .compare_fids = smb2_compare_fids,
2957         .setup_request = smb2_setup_request,
2958         .setup_async_request = smb2_setup_async_request,
2959         .check_receive = smb2_check_receive,
2960         .add_credits = smb2_add_credits,
2961         .set_credits = smb2_set_credits,
2962         .get_credits_field = smb2_get_credits_field,
2963         .get_credits = smb2_get_credits,
2964         .wait_mtu_credits = smb2_wait_mtu_credits,
2965         .get_next_mid = smb2_get_next_mid,
2966         .read_data_offset = smb2_read_data_offset,
2967         .read_data_length = smb2_read_data_length,
2968         .map_error = map_smb2_to_linux_error,
2969         .find_mid = smb2_find_mid,
2970         .check_message = smb2_check_message,
2971         .dump_detail = smb2_dump_detail,
2972         .clear_stats = smb2_clear_stats,
2973         .print_stats = smb2_print_stats,
2974         .dump_share_caps = smb2_dump_share_caps,
2975         .is_oplock_break = smb2_is_valid_oplock_break,
2976         .handle_cancelled_mid = smb2_handle_cancelled_mid,
2977         .downgrade_oplock = smb2_downgrade_oplock,
2978         .need_neg = smb2_need_neg,
2979         .negotiate = smb2_negotiate,
2980         .negotiate_wsize = smb2_negotiate_wsize,
2981         .negotiate_rsize = smb2_negotiate_rsize,
2982         .sess_setup = SMB2_sess_setup,
2983         .logoff = SMB2_logoff,
2984         .tree_connect = SMB2_tcon,
2985         .tree_disconnect = SMB2_tdis,
2986         .qfs_tcon = smb3_qfs_tcon,
2987         .is_path_accessible = smb2_is_path_accessible,
2988         .can_echo = smb2_can_echo,
2989         .echo = SMB2_echo,
2990         .query_path_info = smb2_query_path_info,
2991         .get_srv_inum = smb2_get_srv_inum,
2992         .query_file_info = smb2_query_file_info,
2993         .set_path_size = smb2_set_path_size,
2994         .set_file_size = smb2_set_file_size,
2995         .set_file_info = smb2_set_file_info,
2996         .set_compression = smb2_set_compression,
2997         .mkdir = smb2_mkdir,
2998         .mkdir_setinfo = smb2_mkdir_setinfo,
2999         .rmdir = smb2_rmdir,
3000         .unlink = smb2_unlink,
3001         .rename = smb2_rename_path,
3002         .create_hardlink = smb2_create_hardlink,
3003         .query_symlink = smb2_query_symlink,
3004         .query_mf_symlink = smb3_query_mf_symlink,
3005         .create_mf_symlink = smb3_create_mf_symlink,
3006         .open = smb2_open_file,
3007         .set_fid = smb2_set_fid,
3008         .close = smb2_close_file,
3009         .flush = smb2_flush_file,
3010         .async_readv = smb2_async_readv,
3011         .async_writev = smb2_async_writev,
3012         .sync_read = smb2_sync_read,
3013         .sync_write = smb2_sync_write,
3014         .query_dir_first = smb2_query_dir_first,
3015         .query_dir_next = smb2_query_dir_next,
3016         .close_dir = smb2_close_dir,
3017         .calc_smb_size = smb2_calc_size,
3018         .is_status_pending = smb2_is_status_pending,
3019         .is_session_expired = smb2_is_session_expired,
3020         .oplock_response = smb2_oplock_response,
3021         .queryfs = smb2_queryfs,
3022         .mand_lock = smb2_mand_lock,
3023         .mand_unlock_range = smb2_unlock_range,
3024         .push_mand_locks = smb2_push_mandatory_locks,
3025         .get_lease_key = smb2_get_lease_key,
3026         .set_lease_key = smb2_set_lease_key,
3027         .new_lease_key = smb2_new_lease_key,
3028         .generate_signingkey = generate_smb30signingkey,
3029         .calc_signature = smb3_calc_signature,
3030         .set_integrity  = smb3_set_integrity,
3031         .is_read_op = smb21_is_read_op,
3032         .set_oplock_level = smb3_set_oplock_level,
3033         .create_lease_buf = smb3_create_lease_buf,
3034         .parse_lease_buf = smb3_parse_lease_buf,
3035         .copychunk_range = smb2_copychunk_range,
3036         .duplicate_extents = smb2_duplicate_extents,
3037         .validate_negotiate = smb3_validate_negotiate,
3038         .wp_retry_size = smb2_wp_retry_size,
3039         .dir_needs_close = smb2_dir_needs_close,
3040         .fallocate = smb3_fallocate,
3041         .enum_snapshots = smb3_enum_snapshots,
3042         .init_transform_rq = smb3_init_transform_rq,
3043         .free_transform_rq = smb3_free_transform_rq,
3044         .is_transform_hdr = smb3_is_transform_hdr,
3045         .receive_transform = smb3_receive_transform,
3046         .get_dfs_refer = smb2_get_dfs_refer,
3047         .select_sectype = smb2_select_sectype,
3048 #ifdef CONFIG_CIFS_XATTR
3049         .query_all_EAs = smb2_query_eas,
3050         .set_EA = smb2_set_ea,
3051 #endif /* CIFS_XATTR */
3052 #ifdef CONFIG_CIFS_ACL
3053         .get_acl = get_smb2_acl,
3054         .get_acl_by_fid = get_smb2_acl_by_fid,
3055         .set_acl = set_smb2_acl,
3056 #endif /* CIFS_ACL */
3057 };
3058
3059 #ifdef CONFIG_CIFS_SMB311
3060 struct smb_version_operations smb311_operations = {
3061         .compare_fids = smb2_compare_fids,
3062         .setup_request = smb2_setup_request,
3063         .setup_async_request = smb2_setup_async_request,
3064         .check_receive = smb2_check_receive,
3065         .add_credits = smb2_add_credits,
3066         .set_credits = smb2_set_credits,
3067         .get_credits_field = smb2_get_credits_field,
3068         .get_credits = smb2_get_credits,
3069         .wait_mtu_credits = smb2_wait_mtu_credits,
3070         .get_next_mid = smb2_get_next_mid,
3071         .read_data_offset = smb2_read_data_offset,
3072         .read_data_length = smb2_read_data_length,
3073         .map_error = map_smb2_to_linux_error,
3074         .find_mid = smb2_find_mid,
3075         .check_message = smb2_check_message,
3076         .dump_detail = smb2_dump_detail,
3077         .clear_stats = smb2_clear_stats,
3078         .print_stats = smb2_print_stats,
3079         .dump_share_caps = smb2_dump_share_caps,
3080         .is_oplock_break = smb2_is_valid_oplock_break,
3081         .handle_cancelled_mid = smb2_handle_cancelled_mid,
3082         .downgrade_oplock = smb2_downgrade_oplock,
3083         .need_neg = smb2_need_neg,
3084         .negotiate = smb2_negotiate,
3085         .negotiate_wsize = smb2_negotiate_wsize,
3086         .negotiate_rsize = smb2_negotiate_rsize,
3087         .sess_setup = SMB2_sess_setup,
3088         .logoff = SMB2_logoff,
3089         .tree_connect = SMB2_tcon,
3090         .tree_disconnect = SMB2_tdis,
3091         .qfs_tcon = smb3_qfs_tcon,
3092         .is_path_accessible = smb2_is_path_accessible,
3093         .can_echo = smb2_can_echo,
3094         .echo = SMB2_echo,
3095         .query_path_info = smb2_query_path_info,
3096         .get_srv_inum = smb2_get_srv_inum,
3097         .query_file_info = smb2_query_file_info,
3098         .set_path_size = smb2_set_path_size,
3099         .set_file_size = smb2_set_file_size,
3100         .set_file_info = smb2_set_file_info,
3101         .set_compression = smb2_set_compression,
3102         .mkdir = smb2_mkdir,
3103         .mkdir_setinfo = smb2_mkdir_setinfo,
3104         .rmdir = smb2_rmdir,
3105         .unlink = smb2_unlink,
3106         .rename = smb2_rename_path,
3107         .create_hardlink = smb2_create_hardlink,
3108         .query_symlink = smb2_query_symlink,
3109         .query_mf_symlink = smb3_query_mf_symlink,
3110         .create_mf_symlink = smb3_create_mf_symlink,
3111         .open = smb2_open_file,
3112         .set_fid = smb2_set_fid,
3113         .close = smb2_close_file,
3114         .flush = smb2_flush_file,
3115         .async_readv = smb2_async_readv,
3116         .async_writev = smb2_async_writev,
3117         .sync_read = smb2_sync_read,
3118         .sync_write = smb2_sync_write,
3119         .query_dir_first = smb2_query_dir_first,
3120         .query_dir_next = smb2_query_dir_next,
3121         .close_dir = smb2_close_dir,
3122         .calc_smb_size = smb2_calc_size,
3123         .is_status_pending = smb2_is_status_pending,
3124         .is_session_expired = smb2_is_session_expired,
3125         .oplock_response = smb2_oplock_response,
3126         .queryfs = smb2_queryfs,
3127         .mand_lock = smb2_mand_lock,
3128         .mand_unlock_range = smb2_unlock_range,
3129         .push_mand_locks = smb2_push_mandatory_locks,
3130         .get_lease_key = smb2_get_lease_key,
3131         .set_lease_key = smb2_set_lease_key,
3132         .new_lease_key = smb2_new_lease_key,
3133         .generate_signingkey = generate_smb311signingkey,
3134         .calc_signature = smb3_calc_signature,
3135         .set_integrity  = smb3_set_integrity,
3136         .is_read_op = smb21_is_read_op,
3137         .set_oplock_level = smb3_set_oplock_level,
3138         .create_lease_buf = smb3_create_lease_buf,
3139         .parse_lease_buf = smb3_parse_lease_buf,
3140         .copychunk_range = smb2_copychunk_range,
3141         .duplicate_extents = smb2_duplicate_extents,
3142 /*      .validate_negotiate = smb3_validate_negotiate, */ /* not used in 3.11 */
3143         .wp_retry_size = smb2_wp_retry_size,
3144         .dir_needs_close = smb2_dir_needs_close,
3145         .fallocate = smb3_fallocate,
3146         .enum_snapshots = smb3_enum_snapshots,
3147         .init_transform_rq = smb3_init_transform_rq,
3148         .free_transform_rq = smb3_free_transform_rq,
3149         .is_transform_hdr = smb3_is_transform_hdr,
3150         .receive_transform = smb3_receive_transform,
3151         .get_dfs_refer = smb2_get_dfs_refer,
3152         .select_sectype = smb2_select_sectype,
3153 #ifdef CONFIG_CIFS_XATTR
3154         .query_all_EAs = smb2_query_eas,
3155         .set_EA = smb2_set_ea,
3156 #endif /* CIFS_XATTR */
3157 };
3158 #endif /* CIFS_SMB311 */
3159
3160 struct smb_version_values smb20_values = {
3161         .version_string = SMB20_VERSION_STRING,
3162         .protocol_id = SMB20_PROT_ID,
3163         .req_capabilities = 0, /* MBZ */
3164         .large_lock_type = 0,
3165         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3166         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3167         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3168         .header_size = sizeof(struct smb2_hdr),
3169         .header_preamble_size = 4,
3170         .max_header_size = MAX_SMB2_HDR_SIZE,
3171         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3172         .lock_cmd = SMB2_LOCK,
3173         .cap_unix = 0,
3174         .cap_nt_find = SMB2_NT_FIND,
3175         .cap_large_files = SMB2_LARGE_FILES,
3176         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3177         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3178         .create_lease_size = sizeof(struct create_lease),
3179 };
3180
3181 struct smb_version_values smb21_values = {
3182         .version_string = SMB21_VERSION_STRING,
3183         .protocol_id = SMB21_PROT_ID,
3184         .req_capabilities = 0, /* MBZ on negotiate req until SMB3 dialect */
3185         .large_lock_type = 0,
3186         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3187         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3188         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3189         .header_size = sizeof(struct smb2_hdr),
3190         .header_preamble_size = 4,
3191         .max_header_size = MAX_SMB2_HDR_SIZE,
3192         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3193         .lock_cmd = SMB2_LOCK,
3194         .cap_unix = 0,
3195         .cap_nt_find = SMB2_NT_FIND,
3196         .cap_large_files = SMB2_LARGE_FILES,
3197         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3198         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3199         .create_lease_size = sizeof(struct create_lease),
3200 };
3201
3202 struct smb_version_values smb3any_values = {
3203         .version_string = SMB3ANY_VERSION_STRING,
3204         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
3205         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION,
3206         .large_lock_type = 0,
3207         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3208         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3209         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3210         .header_size = sizeof(struct smb2_hdr),
3211         .header_preamble_size = 4,
3212         .max_header_size = MAX_SMB2_HDR_SIZE,
3213         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3214         .lock_cmd = SMB2_LOCK,
3215         .cap_unix = 0,
3216         .cap_nt_find = SMB2_NT_FIND,
3217         .cap_large_files = SMB2_LARGE_FILES,
3218         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3219         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3220         .create_lease_size = sizeof(struct create_lease_v2),
3221 };
3222
3223 struct smb_version_values smbdefault_values = {
3224         .version_string = SMBDEFAULT_VERSION_STRING,
3225         .protocol_id = SMB302_PROT_ID, /* doesn't matter, send protocol array */
3226         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION,
3227         .large_lock_type = 0,
3228         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3229         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3230         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3231         .header_size = sizeof(struct smb2_hdr),
3232         .header_preamble_size = 4,
3233         .max_header_size = MAX_SMB2_HDR_SIZE,
3234         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3235         .lock_cmd = SMB2_LOCK,
3236         .cap_unix = 0,
3237         .cap_nt_find = SMB2_NT_FIND,
3238         .cap_large_files = SMB2_LARGE_FILES,
3239         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3240         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3241         .create_lease_size = sizeof(struct create_lease_v2),
3242 };
3243
3244 struct smb_version_values smb30_values = {
3245         .version_string = SMB30_VERSION_STRING,
3246         .protocol_id = SMB30_PROT_ID,
3247         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION,
3248         .large_lock_type = 0,
3249         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3250         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3251         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3252         .header_size = sizeof(struct smb2_hdr),
3253         .header_preamble_size = 4,
3254         .max_header_size = MAX_SMB2_HDR_SIZE,
3255         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3256         .lock_cmd = SMB2_LOCK,
3257         .cap_unix = 0,
3258         .cap_nt_find = SMB2_NT_FIND,
3259         .cap_large_files = SMB2_LARGE_FILES,
3260         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3261         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3262         .create_lease_size = sizeof(struct create_lease_v2),
3263 };
3264
3265 struct smb_version_values smb302_values = {
3266         .version_string = SMB302_VERSION_STRING,
3267         .protocol_id = SMB302_PROT_ID,
3268         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION,
3269         .large_lock_type = 0,
3270         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3271         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3272         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3273         .header_size = sizeof(struct smb2_hdr),
3274         .header_preamble_size = 4,
3275         .max_header_size = MAX_SMB2_HDR_SIZE,
3276         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3277         .lock_cmd = SMB2_LOCK,
3278         .cap_unix = 0,
3279         .cap_nt_find = SMB2_NT_FIND,
3280         .cap_large_files = SMB2_LARGE_FILES,
3281         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3282         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3283         .create_lease_size = sizeof(struct create_lease_v2),
3284 };
3285
3286 #ifdef CONFIG_CIFS_SMB311
3287 struct smb_version_values smb311_values = {
3288         .version_string = SMB311_VERSION_STRING,
3289         .protocol_id = SMB311_PROT_ID,
3290         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU | SMB2_GLOBAL_CAP_PERSISTENT_HANDLES | SMB2_GLOBAL_CAP_ENCRYPTION,
3291         .large_lock_type = 0,
3292         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
3293         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
3294         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
3295         .header_size = sizeof(struct smb2_hdr),
3296         .header_preamble_size = 4,
3297         .max_header_size = MAX_SMB2_HDR_SIZE,
3298         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
3299         .lock_cmd = SMB2_LOCK,
3300         .cap_unix = 0,
3301         .cap_nt_find = SMB2_NT_FIND,
3302         .cap_large_files = SMB2_LARGE_FILES,
3303         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
3304         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
3305         .create_lease_size = sizeof(struct create_lease_v2),
3306 };
3307 #endif /* SMB311 */