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