OSDN Git Service

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