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