ath10k: split ce pipe init/alloc further
[cascardo/linux.git] / fs / cifs / smb2ops.c
1 /*
2  *  SMB2 version specific operations
3  *
4  *  Copyright (c) 2012, Jeff Layton <jlayton@redhat.com>
5  *
6  *  This library is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License v2 as published
8  *  by the Free Software Foundation.
9  *
10  *  This library is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
13  *  the GNU Lesser General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Lesser General Public License
16  *  along with this library; if not, write to the Free Software
17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18  */
19
20 #include <linux/pagemap.h>
21 #include <linux/vfs.h>
22 #include <linux/falloc.h>
23 #include "cifsglob.h"
24 #include "smb2pdu.h"
25 #include "smb2proto.h"
26 #include "cifsproto.h"
27 #include "cifs_debug.h"
28 #include "cifs_unicode.h"
29 #include "smb2status.h"
30 #include "smb2glob.h"
31
32 static int
33 change_conf(struct TCP_Server_Info *server)
34 {
35         server->credits += server->echo_credits + server->oplock_credits;
36         server->oplock_credits = server->echo_credits = 0;
37         switch (server->credits) {
38         case 0:
39                 return -1;
40         case 1:
41                 server->echoes = false;
42                 server->oplocks = false;
43                 cifs_dbg(VFS, "disabling echoes and oplocks\n");
44                 break;
45         case 2:
46                 server->echoes = true;
47                 server->oplocks = false;
48                 server->echo_credits = 1;
49                 cifs_dbg(FYI, "disabling oplocks\n");
50                 break;
51         default:
52                 server->echoes = true;
53                 server->oplocks = true;
54                 server->echo_credits = 1;
55                 server->oplock_credits = 1;
56         }
57         server->credits -= server->echo_credits + server->oplock_credits;
58         return 0;
59 }
60
61 static void
62 smb2_add_credits(struct TCP_Server_Info *server, const unsigned int add,
63                  const int optype)
64 {
65         int *val, rc = 0;
66         spin_lock(&server->req_lock);
67         val = server->ops->get_credits_field(server, optype);
68         *val += add;
69         server->in_flight--;
70         if (server->in_flight == 0 && (optype & CIFS_OP_MASK) != CIFS_NEG_OP)
71                 rc = change_conf(server);
72         /*
73          * Sometimes server returns 0 credits on oplock break ack - we need to
74          * rebalance credits in this case.
75          */
76         else if (server->in_flight > 0 && server->oplock_credits == 0 &&
77                  server->oplocks) {
78                 if (server->credits > 1) {
79                         server->credits--;
80                         server->oplock_credits++;
81                 }
82         }
83         spin_unlock(&server->req_lock);
84         wake_up(&server->request_q);
85         if (rc)
86                 cifs_reconnect(server);
87 }
88
89 static void
90 smb2_set_credits(struct TCP_Server_Info *server, const int val)
91 {
92         spin_lock(&server->req_lock);
93         server->credits = val;
94         spin_unlock(&server->req_lock);
95 }
96
97 static int *
98 smb2_get_credits_field(struct TCP_Server_Info *server, const int optype)
99 {
100         switch (optype) {
101         case CIFS_ECHO_OP:
102                 return &server->echo_credits;
103         case CIFS_OBREAK_OP:
104                 return &server->oplock_credits;
105         default:
106                 return &server->credits;
107         }
108 }
109
110 static unsigned int
111 smb2_get_credits(struct mid_q_entry *mid)
112 {
113         return le16_to_cpu(((struct smb2_hdr *)mid->resp_buf)->CreditRequest);
114 }
115
116 static int
117 smb2_wait_mtu_credits(struct TCP_Server_Info *server, unsigned int size,
118                       unsigned int *num, unsigned int *credits)
119 {
120         int rc = 0;
121         unsigned int scredits;
122
123         spin_lock(&server->req_lock);
124         while (1) {
125                 if (server->credits <= 0) {
126                         spin_unlock(&server->req_lock);
127                         cifs_num_waiters_inc(server);
128                         rc = wait_event_killable(server->request_q,
129                                         has_credits(server, &server->credits));
130                         cifs_num_waiters_dec(server);
131                         if (rc)
132                                 return rc;
133                         spin_lock(&server->req_lock);
134                 } else {
135                         if (server->tcpStatus == CifsExiting) {
136                                 spin_unlock(&server->req_lock);
137                                 return -ENOENT;
138                         }
139
140                         scredits = server->credits;
141                         /* can deadlock with reopen */
142                         if (scredits == 1) {
143                                 *num = SMB2_MAX_BUFFER_SIZE;
144                                 *credits = 0;
145                                 break;
146                         }
147
148                         /* leave one credit for a possible reopen */
149                         scredits--;
150                         *num = min_t(unsigned int, size,
151                                      scredits * SMB2_MAX_BUFFER_SIZE);
152
153                         *credits = DIV_ROUND_UP(*num, SMB2_MAX_BUFFER_SIZE);
154                         server->credits -= *credits;
155                         server->in_flight++;
156                         break;
157                 }
158         }
159         spin_unlock(&server->req_lock);
160         return rc;
161 }
162
163 static __u64
164 smb2_get_next_mid(struct TCP_Server_Info *server)
165 {
166         __u64 mid;
167         /* for SMB2 we need the current value */
168         spin_lock(&GlobalMid_Lock);
169         mid = server->CurrentMid++;
170         spin_unlock(&GlobalMid_Lock);
171         return mid;
172 }
173
174 static struct mid_q_entry *
175 smb2_find_mid(struct TCP_Server_Info *server, char *buf)
176 {
177         struct mid_q_entry *mid;
178         struct smb2_hdr *hdr = (struct smb2_hdr *)buf;
179
180         spin_lock(&GlobalMid_Lock);
181         list_for_each_entry(mid, &server->pending_mid_q, qhead) {
182                 if ((mid->mid == hdr->MessageId) &&
183                     (mid->mid_state == MID_REQUEST_SUBMITTED) &&
184                     (mid->command == hdr->Command)) {
185                         spin_unlock(&GlobalMid_Lock);
186                         return mid;
187                 }
188         }
189         spin_unlock(&GlobalMid_Lock);
190         return NULL;
191 }
192
193 static void
194 smb2_dump_detail(void *buf)
195 {
196 #ifdef CONFIG_CIFS_DEBUG2
197         struct smb2_hdr *smb = (struct smb2_hdr *)buf;
198
199         cifs_dbg(VFS, "Cmd: %d Err: 0x%x Flags: 0x%x Mid: %llu Pid: %d\n",
200                  smb->Command, smb->Status, smb->Flags, smb->MessageId,
201                  smb->ProcessId);
202         cifs_dbg(VFS, "smb buf %p len %u\n", smb, smb2_calc_size(smb));
203 #endif
204 }
205
206 static bool
207 smb2_need_neg(struct TCP_Server_Info *server)
208 {
209         return server->max_read == 0;
210 }
211
212 static int
213 smb2_negotiate(const unsigned int xid, struct cifs_ses *ses)
214 {
215         int rc;
216         ses->server->CurrentMid = 0;
217         rc = SMB2_negotiate(xid, ses);
218         /* BB we probably don't need to retry with modern servers */
219         if (rc == -EAGAIN)
220                 rc = -EHOSTDOWN;
221         return rc;
222 }
223
224 static unsigned int
225 smb2_negotiate_wsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
226 {
227         struct TCP_Server_Info *server = tcon->ses->server;
228         unsigned int wsize;
229
230         /* start with specified wsize, or default */
231         wsize = volume_info->wsize ? volume_info->wsize : CIFS_DEFAULT_IOSIZE;
232         wsize = min_t(unsigned int, wsize, server->max_write);
233
234         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
235                 wsize = min_t(unsigned int, wsize, SMB2_MAX_BUFFER_SIZE);
236
237         return wsize;
238 }
239
240 static unsigned int
241 smb2_negotiate_rsize(struct cifs_tcon *tcon, struct smb_vol *volume_info)
242 {
243         struct TCP_Server_Info *server = tcon->ses->server;
244         unsigned int rsize;
245
246         /* start with specified rsize, or default */
247         rsize = volume_info->rsize ? volume_info->rsize : CIFS_DEFAULT_IOSIZE;
248         rsize = min_t(unsigned int, rsize, server->max_read);
249
250         if (!(server->capabilities & SMB2_GLOBAL_CAP_LARGE_MTU))
251                 rsize = min_t(unsigned int, rsize, SMB2_MAX_BUFFER_SIZE);
252
253         return rsize;
254 }
255
256 #ifdef CONFIG_CIFS_STATS2
257 static int
258 SMB3_request_interfaces(const unsigned int xid, struct cifs_tcon *tcon)
259 {
260         int rc;
261         unsigned int ret_data_len = 0;
262         struct network_interface_info_ioctl_rsp *out_buf;
263
264         rc = SMB2_ioctl(xid, tcon, NO_FILE_ID, NO_FILE_ID,
265                         FSCTL_QUERY_NETWORK_INTERFACE_INFO, true /* is_fsctl */,
266                         NULL /* no data input */, 0 /* no data input */,
267                         (char **)&out_buf, &ret_data_len);
268
269         if ((rc == 0)  && (ret_data_len > 0)) {
270                 /* Dump info on first interface */
271                 cifs_dbg(FYI, "Adapter Capability 0x%x\t",
272                         le32_to_cpu(out_buf->Capability));
273                 cifs_dbg(FYI, "Link Speed %lld\n",
274                         le64_to_cpu(out_buf->LinkSpeed));
275         } else
276                 cifs_dbg(VFS, "error %d on ioctl to get interface list\n", rc);
277
278         return rc;
279 }
280 #endif /* STATS2 */
281
282 static void
283 smb3_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
284 {
285         int rc;
286         __le16 srch_path = 0; /* Null - open root of share */
287         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
288         struct cifs_open_parms oparms;
289         struct cifs_fid fid;
290
291         oparms.tcon = tcon;
292         oparms.desired_access = FILE_READ_ATTRIBUTES;
293         oparms.disposition = FILE_OPEN;
294         oparms.create_options = 0;
295         oparms.fid = &fid;
296         oparms.reconnect = false;
297
298         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
299         if (rc)
300                 return;
301
302 #ifdef CONFIG_CIFS_STATS2
303         SMB3_request_interfaces(xid, tcon);
304 #endif /* STATS2 */
305
306         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
307                         FS_ATTRIBUTE_INFORMATION);
308         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
309                         FS_DEVICE_INFORMATION);
310         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
311                         FS_SECTOR_SIZE_INFORMATION); /* SMB3 specific */
312         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
313         return;
314 }
315
316 static void
317 smb2_qfs_tcon(const unsigned int xid, struct cifs_tcon *tcon)
318 {
319         int rc;
320         __le16 srch_path = 0; /* Null - open root of share */
321         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
322         struct cifs_open_parms oparms;
323         struct cifs_fid fid;
324
325         oparms.tcon = tcon;
326         oparms.desired_access = FILE_READ_ATTRIBUTES;
327         oparms.disposition = FILE_OPEN;
328         oparms.create_options = 0;
329         oparms.fid = &fid;
330         oparms.reconnect = false;
331
332         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
333         if (rc)
334                 return;
335
336         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
337                         FS_ATTRIBUTE_INFORMATION);
338         SMB2_QFS_attr(xid, tcon, fid.persistent_fid, fid.volatile_fid,
339                         FS_DEVICE_INFORMATION);
340         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
341         return;
342 }
343
344 static int
345 smb2_is_path_accessible(const unsigned int xid, struct cifs_tcon *tcon,
346                         struct cifs_sb_info *cifs_sb, const char *full_path)
347 {
348         int rc;
349         __le16 *utf16_path;
350         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
351         struct cifs_open_parms oparms;
352         struct cifs_fid fid;
353
354         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
355         if (!utf16_path)
356                 return -ENOMEM;
357
358         oparms.tcon = tcon;
359         oparms.desired_access = FILE_READ_ATTRIBUTES;
360         oparms.disposition = FILE_OPEN;
361         oparms.create_options = 0;
362         oparms.fid = &fid;
363         oparms.reconnect = false;
364
365         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
366         if (rc) {
367                 kfree(utf16_path);
368                 return rc;
369         }
370
371         rc = SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
372         kfree(utf16_path);
373         return rc;
374 }
375
376 static int
377 smb2_get_srv_inum(const unsigned int xid, struct cifs_tcon *tcon,
378                   struct cifs_sb_info *cifs_sb, const char *full_path,
379                   u64 *uniqueid, FILE_ALL_INFO *data)
380 {
381         *uniqueid = le64_to_cpu(data->IndexNumber);
382         return 0;
383 }
384
385 static int
386 smb2_query_file_info(const unsigned int xid, struct cifs_tcon *tcon,
387                      struct cifs_fid *fid, FILE_ALL_INFO *data)
388 {
389         int rc;
390         struct smb2_file_all_info *smb2_data;
391
392         smb2_data = kzalloc(sizeof(struct smb2_file_all_info) + MAX_NAME * 2,
393                             GFP_KERNEL);
394         if (smb2_data == NULL)
395                 return -ENOMEM;
396
397         rc = SMB2_query_info(xid, tcon, fid->persistent_fid, fid->volatile_fid,
398                              smb2_data);
399         if (!rc)
400                 move_smb2_info_to_cifs(data, smb2_data);
401         kfree(smb2_data);
402         return rc;
403 }
404
405 static bool
406 smb2_can_echo(struct TCP_Server_Info *server)
407 {
408         return server->echoes;
409 }
410
411 static void
412 smb2_clear_stats(struct cifs_tcon *tcon)
413 {
414 #ifdef CONFIG_CIFS_STATS
415         int i;
416         for (i = 0; i < NUMBER_OF_SMB2_COMMANDS; i++) {
417                 atomic_set(&tcon->stats.smb2_stats.smb2_com_sent[i], 0);
418                 atomic_set(&tcon->stats.smb2_stats.smb2_com_failed[i], 0);
419         }
420 #endif
421 }
422
423 static void
424 smb2_dump_share_caps(struct seq_file *m, struct cifs_tcon *tcon)
425 {
426         seq_puts(m, "\n\tShare Capabilities:");
427         if (tcon->capabilities & SMB2_SHARE_CAP_DFS)
428                 seq_puts(m, " DFS,");
429         if (tcon->capabilities & SMB2_SHARE_CAP_CONTINUOUS_AVAILABILITY)
430                 seq_puts(m, " CONTINUOUS AVAILABILITY,");
431         if (tcon->capabilities & SMB2_SHARE_CAP_SCALEOUT)
432                 seq_puts(m, " SCALEOUT,");
433         if (tcon->capabilities & SMB2_SHARE_CAP_CLUSTER)
434                 seq_puts(m, " CLUSTER,");
435         if (tcon->capabilities & SMB2_SHARE_CAP_ASYMMETRIC)
436                 seq_puts(m, " ASYMMETRIC,");
437         if (tcon->capabilities == 0)
438                 seq_puts(m, " None");
439         if (tcon->ss_flags & SSINFO_FLAGS_ALIGNED_DEVICE)
440                 seq_puts(m, " Aligned,");
441         if (tcon->ss_flags & SSINFO_FLAGS_PARTITION_ALIGNED_ON_DEVICE)
442                 seq_puts(m, " Partition Aligned,");
443         if (tcon->ss_flags & SSINFO_FLAGS_NO_SEEK_PENALTY)
444                 seq_puts(m, " SSD,");
445         if (tcon->ss_flags & SSINFO_FLAGS_TRIM_ENABLED)
446                 seq_puts(m, " TRIM-support,");
447
448         seq_printf(m, "\tShare Flags: 0x%x", tcon->share_flags);
449         if (tcon->perf_sector_size)
450                 seq_printf(m, "\tOptimal sector size: 0x%x",
451                            tcon->perf_sector_size);
452 }
453
454 static void
455 smb2_print_stats(struct seq_file *m, struct cifs_tcon *tcon)
456 {
457 #ifdef CONFIG_CIFS_STATS
458         atomic_t *sent = tcon->stats.smb2_stats.smb2_com_sent;
459         atomic_t *failed = tcon->stats.smb2_stats.smb2_com_failed;
460         seq_printf(m, "\nNegotiates: %d sent %d failed",
461                    atomic_read(&sent[SMB2_NEGOTIATE_HE]),
462                    atomic_read(&failed[SMB2_NEGOTIATE_HE]));
463         seq_printf(m, "\nSessionSetups: %d sent %d failed",
464                    atomic_read(&sent[SMB2_SESSION_SETUP_HE]),
465                    atomic_read(&failed[SMB2_SESSION_SETUP_HE]));
466         seq_printf(m, "\nLogoffs: %d sent %d failed",
467                    atomic_read(&sent[SMB2_LOGOFF_HE]),
468                    atomic_read(&failed[SMB2_LOGOFF_HE]));
469         seq_printf(m, "\nTreeConnects: %d sent %d failed",
470                    atomic_read(&sent[SMB2_TREE_CONNECT_HE]),
471                    atomic_read(&failed[SMB2_TREE_CONNECT_HE]));
472         seq_printf(m, "\nTreeDisconnects: %d sent %d failed",
473                    atomic_read(&sent[SMB2_TREE_DISCONNECT_HE]),
474                    atomic_read(&failed[SMB2_TREE_DISCONNECT_HE]));
475         seq_printf(m, "\nCreates: %d sent %d failed",
476                    atomic_read(&sent[SMB2_CREATE_HE]),
477                    atomic_read(&failed[SMB2_CREATE_HE]));
478         seq_printf(m, "\nCloses: %d sent %d failed",
479                    atomic_read(&sent[SMB2_CLOSE_HE]),
480                    atomic_read(&failed[SMB2_CLOSE_HE]));
481         seq_printf(m, "\nFlushes: %d sent %d failed",
482                    atomic_read(&sent[SMB2_FLUSH_HE]),
483                    atomic_read(&failed[SMB2_FLUSH_HE]));
484         seq_printf(m, "\nReads: %d sent %d failed",
485                    atomic_read(&sent[SMB2_READ_HE]),
486                    atomic_read(&failed[SMB2_READ_HE]));
487         seq_printf(m, "\nWrites: %d sent %d failed",
488                    atomic_read(&sent[SMB2_WRITE_HE]),
489                    atomic_read(&failed[SMB2_WRITE_HE]));
490         seq_printf(m, "\nLocks: %d sent %d failed",
491                    atomic_read(&sent[SMB2_LOCK_HE]),
492                    atomic_read(&failed[SMB2_LOCK_HE]));
493         seq_printf(m, "\nIOCTLs: %d sent %d failed",
494                    atomic_read(&sent[SMB2_IOCTL_HE]),
495                    atomic_read(&failed[SMB2_IOCTL_HE]));
496         seq_printf(m, "\nCancels: %d sent %d failed",
497                    atomic_read(&sent[SMB2_CANCEL_HE]),
498                    atomic_read(&failed[SMB2_CANCEL_HE]));
499         seq_printf(m, "\nEchos: %d sent %d failed",
500                    atomic_read(&sent[SMB2_ECHO_HE]),
501                    atomic_read(&failed[SMB2_ECHO_HE]));
502         seq_printf(m, "\nQueryDirectories: %d sent %d failed",
503                    atomic_read(&sent[SMB2_QUERY_DIRECTORY_HE]),
504                    atomic_read(&failed[SMB2_QUERY_DIRECTORY_HE]));
505         seq_printf(m, "\nChangeNotifies: %d sent %d failed",
506                    atomic_read(&sent[SMB2_CHANGE_NOTIFY_HE]),
507                    atomic_read(&failed[SMB2_CHANGE_NOTIFY_HE]));
508         seq_printf(m, "\nQueryInfos: %d sent %d failed",
509                    atomic_read(&sent[SMB2_QUERY_INFO_HE]),
510                    atomic_read(&failed[SMB2_QUERY_INFO_HE]));
511         seq_printf(m, "\nSetInfos: %d sent %d failed",
512                    atomic_read(&sent[SMB2_SET_INFO_HE]),
513                    atomic_read(&failed[SMB2_SET_INFO_HE]));
514         seq_printf(m, "\nOplockBreaks: %d sent %d failed",
515                    atomic_read(&sent[SMB2_OPLOCK_BREAK_HE]),
516                    atomic_read(&failed[SMB2_OPLOCK_BREAK_HE]));
517 #endif
518 }
519
520 static void
521 smb2_set_fid(struct cifsFileInfo *cfile, struct cifs_fid *fid, __u32 oplock)
522 {
523         struct cifsInodeInfo *cinode = CIFS_I(cfile->dentry->d_inode);
524         struct TCP_Server_Info *server = tlink_tcon(cfile->tlink)->ses->server;
525
526         cfile->fid.persistent_fid = fid->persistent_fid;
527         cfile->fid.volatile_fid = fid->volatile_fid;
528         server->ops->set_oplock_level(cinode, oplock, fid->epoch,
529                                       &fid->purge_cache);
530         cinode->can_cache_brlcks = CIFS_CACHE_WRITE(cinode);
531 }
532
533 static void
534 smb2_close_file(const unsigned int xid, struct cifs_tcon *tcon,
535                 struct cifs_fid *fid)
536 {
537         SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
538 }
539
540 static int
541 SMB2_request_res_key(const unsigned int xid, struct cifs_tcon *tcon,
542                      u64 persistent_fid, u64 volatile_fid,
543                      struct copychunk_ioctl *pcchunk)
544 {
545         int rc;
546         unsigned int ret_data_len;
547         struct resume_key_req *res_key;
548
549         rc = SMB2_ioctl(xid, tcon, persistent_fid, volatile_fid,
550                         FSCTL_SRV_REQUEST_RESUME_KEY, true /* is_fsctl */,
551                         NULL, 0 /* no input */,
552                         (char **)&res_key, &ret_data_len);
553
554         if (rc) {
555                 cifs_dbg(VFS, "refcpy ioctl error %d getting resume key\n", rc);
556                 goto req_res_key_exit;
557         }
558         if (ret_data_len < sizeof(struct resume_key_req)) {
559                 cifs_dbg(VFS, "Invalid refcopy resume key length\n");
560                 rc = -EINVAL;
561                 goto req_res_key_exit;
562         }
563         memcpy(pcchunk->SourceKey, res_key->ResumeKey, COPY_CHUNK_RES_KEY_SIZE);
564
565 req_res_key_exit:
566         kfree(res_key);
567         return rc;
568 }
569
570 static int
571 smb2_clone_range(const unsigned int xid,
572                         struct cifsFileInfo *srcfile,
573                         struct cifsFileInfo *trgtfile, u64 src_off,
574                         u64 len, u64 dest_off)
575 {
576         int rc;
577         unsigned int ret_data_len;
578         struct copychunk_ioctl *pcchunk;
579         struct copychunk_ioctl_rsp *retbuf = NULL;
580         struct cifs_tcon *tcon;
581         int chunks_copied = 0;
582         bool chunk_sizes_updated = false;
583
584         pcchunk = kmalloc(sizeof(struct copychunk_ioctl), GFP_KERNEL);
585
586         if (pcchunk == NULL)
587                 return -ENOMEM;
588
589         cifs_dbg(FYI, "in smb2_clone_range - about to call request res key\n");
590         /* Request a key from the server to identify the source of the copy */
591         rc = SMB2_request_res_key(xid, tlink_tcon(srcfile->tlink),
592                                 srcfile->fid.persistent_fid,
593                                 srcfile->fid.volatile_fid, pcchunk);
594
595         /* Note: request_res_key sets res_key null only if rc !=0 */
596         if (rc)
597                 goto cchunk_out;
598
599         /* For now array only one chunk long, will make more flexible later */
600         pcchunk->ChunkCount = __constant_cpu_to_le32(1);
601         pcchunk->Reserved = 0;
602         pcchunk->Reserved2 = 0;
603
604         tcon = tlink_tcon(trgtfile->tlink);
605
606         while (len > 0) {
607                 pcchunk->SourceOffset = cpu_to_le64(src_off);
608                 pcchunk->TargetOffset = cpu_to_le64(dest_off);
609                 pcchunk->Length =
610                         cpu_to_le32(min_t(u32, len, tcon->max_bytes_chunk));
611
612                 /* Request server copy to target from src identified by key */
613                 rc = SMB2_ioctl(xid, tcon, trgtfile->fid.persistent_fid,
614                         trgtfile->fid.volatile_fid, FSCTL_SRV_COPYCHUNK_WRITE,
615                         true /* is_fsctl */, (char *)pcchunk,
616                         sizeof(struct copychunk_ioctl), (char **)&retbuf,
617                         &ret_data_len);
618                 if (rc == 0) {
619                         if (ret_data_len !=
620                                         sizeof(struct copychunk_ioctl_rsp)) {
621                                 cifs_dbg(VFS, "invalid cchunk response size\n");
622                                 rc = -EIO;
623                                 goto cchunk_out;
624                         }
625                         if (retbuf->TotalBytesWritten == 0) {
626                                 cifs_dbg(FYI, "no bytes copied\n");
627                                 rc = -EIO;
628                                 goto cchunk_out;
629                         }
630                         /*
631                          * Check if server claimed to write more than we asked
632                          */
633                         if (le32_to_cpu(retbuf->TotalBytesWritten) >
634                             le32_to_cpu(pcchunk->Length)) {
635                                 cifs_dbg(VFS, "invalid copy chunk response\n");
636                                 rc = -EIO;
637                                 goto cchunk_out;
638                         }
639                         if (le32_to_cpu(retbuf->ChunksWritten) != 1) {
640                                 cifs_dbg(VFS, "invalid num chunks written\n");
641                                 rc = -EIO;
642                                 goto cchunk_out;
643                         }
644                         chunks_copied++;
645
646                         src_off += le32_to_cpu(retbuf->TotalBytesWritten);
647                         dest_off += le32_to_cpu(retbuf->TotalBytesWritten);
648                         len -= le32_to_cpu(retbuf->TotalBytesWritten);
649
650                         cifs_dbg(FYI, "Chunks %d PartialChunk %d Total %d\n",
651                                 le32_to_cpu(retbuf->ChunksWritten),
652                                 le32_to_cpu(retbuf->ChunkBytesWritten),
653                                 le32_to_cpu(retbuf->TotalBytesWritten));
654                 } else if (rc == -EINVAL) {
655                         if (ret_data_len != sizeof(struct copychunk_ioctl_rsp))
656                                 goto cchunk_out;
657
658                         cifs_dbg(FYI, "MaxChunks %d BytesChunk %d MaxCopy %d\n",
659                                 le32_to_cpu(retbuf->ChunksWritten),
660                                 le32_to_cpu(retbuf->ChunkBytesWritten),
661                                 le32_to_cpu(retbuf->TotalBytesWritten));
662
663                         /*
664                          * Check if this is the first request using these sizes,
665                          * (ie check if copy succeed once with original sizes
666                          * and check if the server gave us different sizes after
667                          * we already updated max sizes on previous request).
668                          * if not then why is the server returning an error now
669                          */
670                         if ((chunks_copied != 0) || chunk_sizes_updated)
671                                 goto cchunk_out;
672
673                         /* Check that server is not asking us to grow size */
674                         if (le32_to_cpu(retbuf->ChunkBytesWritten) <
675                                         tcon->max_bytes_chunk)
676                                 tcon->max_bytes_chunk =
677                                         le32_to_cpu(retbuf->ChunkBytesWritten);
678                         else
679                                 goto cchunk_out; /* server gave us bogus size */
680
681                         /* No need to change MaxChunks since already set to 1 */
682                         chunk_sizes_updated = true;
683                 }
684         }
685
686 cchunk_out:
687         kfree(pcchunk);
688         return rc;
689 }
690
691 static int
692 smb2_flush_file(const unsigned int xid, struct cifs_tcon *tcon,
693                 struct cifs_fid *fid)
694 {
695         return SMB2_flush(xid, tcon, fid->persistent_fid, fid->volatile_fid);
696 }
697
698 static unsigned int
699 smb2_read_data_offset(char *buf)
700 {
701         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
702         return rsp->DataOffset;
703 }
704
705 static unsigned int
706 smb2_read_data_length(char *buf)
707 {
708         struct smb2_read_rsp *rsp = (struct smb2_read_rsp *)buf;
709         return le32_to_cpu(rsp->DataLength);
710 }
711
712
713 static int
714 smb2_sync_read(const unsigned int xid, struct cifsFileInfo *cfile,
715                struct cifs_io_parms *parms, unsigned int *bytes_read,
716                char **buf, int *buf_type)
717 {
718         parms->persistent_fid = cfile->fid.persistent_fid;
719         parms->volatile_fid = cfile->fid.volatile_fid;
720         return SMB2_read(xid, parms, bytes_read, buf, buf_type);
721 }
722
723 static int
724 smb2_sync_write(const unsigned int xid, struct cifsFileInfo *cfile,
725                 struct cifs_io_parms *parms, unsigned int *written,
726                 struct kvec *iov, unsigned long nr_segs)
727 {
728
729         parms->persistent_fid = cfile->fid.persistent_fid;
730         parms->volatile_fid = cfile->fid.volatile_fid;
731         return SMB2_write(xid, parms, written, iov, nr_segs);
732 }
733
734 /* Set or clear the SPARSE_FILE attribute based on value passed in setsparse */
735 static bool smb2_set_sparse(const unsigned int xid, struct cifs_tcon *tcon,
736                 struct cifsFileInfo *cfile, struct inode *inode, __u8 setsparse)
737 {
738         struct cifsInodeInfo *cifsi;
739         int rc;
740
741         cifsi = CIFS_I(inode);
742
743         /* if file already sparse don't bother setting sparse again */
744         if ((cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && setsparse)
745                 return true; /* already sparse */
746
747         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE) && !setsparse)
748                 return true; /* already not sparse */
749
750         /*
751          * Can't check for sparse support on share the usual way via the
752          * FS attribute info (FILE_SUPPORTS_SPARSE_FILES) on the share
753          * since Samba server doesn't set the flag on the share, yet
754          * supports the set sparse FSCTL and returns sparse correctly
755          * in the file attributes. If we fail setting sparse though we
756          * mark that server does not support sparse files for this share
757          * to avoid repeatedly sending the unsupported fsctl to server
758          * if the file is repeatedly extended.
759          */
760         if (tcon->broken_sparse_sup)
761                 return false;
762
763         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
764                         cfile->fid.volatile_fid, FSCTL_SET_SPARSE,
765                         true /* is_fctl */, &setsparse, 1, NULL, NULL);
766         if (rc) {
767                 tcon->broken_sparse_sup = true;
768                 cifs_dbg(FYI, "set sparse rc = %d\n", rc);
769                 return false;
770         }
771
772         if (setsparse)
773                 cifsi->cifsAttrs |= FILE_ATTRIBUTE_SPARSE_FILE;
774         else
775                 cifsi->cifsAttrs &= (~FILE_ATTRIBUTE_SPARSE_FILE);
776
777         return true;
778 }
779
780 static int
781 smb2_set_file_size(const unsigned int xid, struct cifs_tcon *tcon,
782                    struct cifsFileInfo *cfile, __u64 size, bool set_alloc)
783 {
784         __le64 eof = cpu_to_le64(size);
785         struct inode *inode;
786
787         /*
788          * If extending file more than one page make sparse. Many Linux fs
789          * make files sparse by default when extending via ftruncate
790          */
791         inode = cfile->dentry->d_inode;
792
793         if (!set_alloc && (size > inode->i_size + 8192)) {
794                 __u8 set_sparse = 1;
795
796                 /* whether set sparse succeeds or not, extend the file */
797                 smb2_set_sparse(xid, tcon, cfile, inode, set_sparse);
798         }
799
800         return SMB2_set_eof(xid, tcon, cfile->fid.persistent_fid,
801                             cfile->fid.volatile_fid, cfile->pid, &eof, false);
802 }
803
804 static int
805 smb2_set_compression(const unsigned int xid, struct cifs_tcon *tcon,
806                    struct cifsFileInfo *cfile)
807 {
808         return SMB2_set_compression(xid, tcon, cfile->fid.persistent_fid,
809                             cfile->fid.volatile_fid);
810 }
811
812 static int
813 smb2_query_dir_first(const unsigned int xid, struct cifs_tcon *tcon,
814                      const char *path, struct cifs_sb_info *cifs_sb,
815                      struct cifs_fid *fid, __u16 search_flags,
816                      struct cifs_search_info *srch_inf)
817 {
818         __le16 *utf16_path;
819         int rc;
820         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
821         struct cifs_open_parms oparms;
822
823         utf16_path = cifs_convert_path_to_utf16(path, cifs_sb);
824         if (!utf16_path)
825                 return -ENOMEM;
826
827         oparms.tcon = tcon;
828         oparms.desired_access = FILE_READ_ATTRIBUTES | FILE_READ_DATA;
829         oparms.disposition = FILE_OPEN;
830         oparms.create_options = 0;
831         oparms.fid = fid;
832         oparms.reconnect = false;
833
834         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, NULL);
835         kfree(utf16_path);
836         if (rc) {
837                 cifs_dbg(VFS, "open dir failed\n");
838                 return rc;
839         }
840
841         srch_inf->entries_in_buffer = 0;
842         srch_inf->index_of_last_entry = 0;
843
844         rc = SMB2_query_directory(xid, tcon, fid->persistent_fid,
845                                   fid->volatile_fid, 0, srch_inf);
846         if (rc) {
847                 cifs_dbg(VFS, "query directory failed\n");
848                 SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
849         }
850         return rc;
851 }
852
853 static int
854 smb2_query_dir_next(const unsigned int xid, struct cifs_tcon *tcon,
855                     struct cifs_fid *fid, __u16 search_flags,
856                     struct cifs_search_info *srch_inf)
857 {
858         return SMB2_query_directory(xid, tcon, fid->persistent_fid,
859                                     fid->volatile_fid, 0, srch_inf);
860 }
861
862 static int
863 smb2_close_dir(const unsigned int xid, struct cifs_tcon *tcon,
864                struct cifs_fid *fid)
865 {
866         return SMB2_close(xid, tcon, fid->persistent_fid, fid->volatile_fid);
867 }
868
869 /*
870 * If we negotiate SMB2 protocol and get STATUS_PENDING - update
871 * the number of credits and return true. Otherwise - return false.
872 */
873 static bool
874 smb2_is_status_pending(char *buf, struct TCP_Server_Info *server, int length)
875 {
876         struct smb2_hdr *hdr = (struct smb2_hdr *)buf;
877
878         if (hdr->Status != STATUS_PENDING)
879                 return false;
880
881         if (!length) {
882                 spin_lock(&server->req_lock);
883                 server->credits += le16_to_cpu(hdr->CreditRequest);
884                 spin_unlock(&server->req_lock);
885                 wake_up(&server->request_q);
886         }
887
888         return true;
889 }
890
891 static int
892 smb2_oplock_response(struct cifs_tcon *tcon, struct cifs_fid *fid,
893                      struct cifsInodeInfo *cinode)
894 {
895         if (tcon->ses->server->capabilities & SMB2_GLOBAL_CAP_LEASING)
896                 return SMB2_lease_break(0, tcon, cinode->lease_key,
897                                         smb2_get_lease_state(cinode));
898
899         return SMB2_oplock_break(0, tcon, fid->persistent_fid,
900                                  fid->volatile_fid,
901                                  CIFS_CACHE_READ(cinode) ? 1 : 0);
902 }
903
904 static int
905 smb2_queryfs(const unsigned int xid, struct cifs_tcon *tcon,
906              struct kstatfs *buf)
907 {
908         int rc;
909         __le16 srch_path = 0; /* Null - open root of share */
910         u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
911         struct cifs_open_parms oparms;
912         struct cifs_fid fid;
913
914         oparms.tcon = tcon;
915         oparms.desired_access = FILE_READ_ATTRIBUTES;
916         oparms.disposition = FILE_OPEN;
917         oparms.create_options = 0;
918         oparms.fid = &fid;
919         oparms.reconnect = false;
920
921         rc = SMB2_open(xid, &oparms, &srch_path, &oplock, NULL, NULL);
922         if (rc)
923                 return rc;
924         buf->f_type = SMB2_MAGIC_NUMBER;
925         rc = SMB2_QFS_info(xid, tcon, fid.persistent_fid, fid.volatile_fid,
926                            buf);
927         SMB2_close(xid, tcon, fid.persistent_fid, fid.volatile_fid);
928         return rc;
929 }
930
931 static bool
932 smb2_compare_fids(struct cifsFileInfo *ob1, struct cifsFileInfo *ob2)
933 {
934         return ob1->fid.persistent_fid == ob2->fid.persistent_fid &&
935                ob1->fid.volatile_fid == ob2->fid.volatile_fid;
936 }
937
938 static int
939 smb2_mand_lock(const unsigned int xid, struct cifsFileInfo *cfile, __u64 offset,
940                __u64 length, __u32 type, int lock, int unlock, bool wait)
941 {
942         if (unlock && !lock)
943                 type = SMB2_LOCKFLAG_UNLOCK;
944         return SMB2_lock(xid, tlink_tcon(cfile->tlink),
945                          cfile->fid.persistent_fid, cfile->fid.volatile_fid,
946                          current->tgid, length, offset, type, wait);
947 }
948
949 static void
950 smb2_get_lease_key(struct inode *inode, struct cifs_fid *fid)
951 {
952         memcpy(fid->lease_key, CIFS_I(inode)->lease_key, SMB2_LEASE_KEY_SIZE);
953 }
954
955 static void
956 smb2_set_lease_key(struct inode *inode, struct cifs_fid *fid)
957 {
958         memcpy(CIFS_I(inode)->lease_key, fid->lease_key, SMB2_LEASE_KEY_SIZE);
959 }
960
961 static void
962 smb2_new_lease_key(struct cifs_fid *fid)
963 {
964         get_random_bytes(fid->lease_key, SMB2_LEASE_KEY_SIZE);
965 }
966
967 static int
968 smb2_query_symlink(const unsigned int xid, struct cifs_tcon *tcon,
969                    const char *full_path, char **target_path,
970                    struct cifs_sb_info *cifs_sb)
971 {
972         int rc;
973         __le16 *utf16_path;
974         __u8 oplock = SMB2_OPLOCK_LEVEL_NONE;
975         struct cifs_open_parms oparms;
976         struct cifs_fid fid;
977         struct smb2_err_rsp *err_buf = NULL;
978         struct smb2_symlink_err_rsp *symlink;
979         unsigned int sub_len, sub_offset;
980
981         cifs_dbg(FYI, "%s: path: %s\n", __func__, full_path);
982
983         utf16_path = cifs_convert_path_to_utf16(full_path, cifs_sb);
984         if (!utf16_path)
985                 return -ENOMEM;
986
987         oparms.tcon = tcon;
988         oparms.desired_access = FILE_READ_ATTRIBUTES;
989         oparms.disposition = FILE_OPEN;
990         oparms.create_options = 0;
991         oparms.fid = &fid;
992         oparms.reconnect = false;
993
994         rc = SMB2_open(xid, &oparms, utf16_path, &oplock, NULL, &err_buf);
995
996         if (!rc || !err_buf) {
997                 kfree(utf16_path);
998                 return -ENOENT;
999         }
1000         /* open must fail on symlink - reset rc */
1001         rc = 0;
1002         symlink = (struct smb2_symlink_err_rsp *)err_buf->ErrorData;
1003         sub_len = le16_to_cpu(symlink->SubstituteNameLength);
1004         sub_offset = le16_to_cpu(symlink->SubstituteNameOffset);
1005         *target_path = cifs_strndup_from_utf16(
1006                                 (char *)symlink->PathBuffer + sub_offset,
1007                                 sub_len, true, cifs_sb->local_nls);
1008         if (!(*target_path)) {
1009                 kfree(utf16_path);
1010                 return -ENOMEM;
1011         }
1012         convert_delimiter(*target_path, '/');
1013         cifs_dbg(FYI, "%s: target path: %s\n", __func__, *target_path);
1014         kfree(utf16_path);
1015         return rc;
1016 }
1017
1018 static long smb3_zero_range(struct file *file, struct cifs_tcon *tcon,
1019                             loff_t offset, loff_t len, bool keep_size)
1020 {
1021         struct inode *inode;
1022         struct cifsInodeInfo *cifsi;
1023         struct cifsFileInfo *cfile = file->private_data;
1024         struct file_zero_data_information fsctl_buf;
1025         long rc;
1026         unsigned int xid;
1027
1028         xid = get_xid();
1029
1030         inode = cfile->dentry->d_inode;
1031         cifsi = CIFS_I(inode);
1032
1033         /* if file not oplocked can't be sure whether asking to extend size */
1034         if (!CIFS_CACHE_READ(cifsi))
1035                 if (keep_size == false)
1036                         return -EOPNOTSUPP;
1037
1038         /* 
1039          * Must check if file sparse since fallocate -z (zero range) assumes
1040          * non-sparse allocation
1041          */
1042         if (!(cifsi->cifsAttrs & FILE_ATTRIBUTE_SPARSE_FILE))
1043                 return -EOPNOTSUPP;
1044
1045         /*
1046          * need to make sure we are not asked to extend the file since the SMB3
1047          * fsctl does not change the file size. In the future we could change
1048          * this to zero the first part of the range then set the file size
1049          * which for a non sparse file would zero the newly extended range
1050          */
1051         if (keep_size == false)
1052                 if (i_size_read(inode) < offset + len)
1053                         return -EOPNOTSUPP;
1054
1055         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
1056
1057         fsctl_buf.FileOffset = cpu_to_le64(offset);
1058         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
1059
1060         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1061                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
1062                         true /* is_fctl */, (char *)&fsctl_buf,
1063                         sizeof(struct file_zero_data_information), NULL, NULL);
1064         free_xid(xid);
1065         return rc;
1066 }
1067
1068 static long smb3_punch_hole(struct file *file, struct cifs_tcon *tcon,
1069                             loff_t offset, loff_t len)
1070 {
1071         struct inode *inode;
1072         struct cifsInodeInfo *cifsi;
1073         struct cifsFileInfo *cfile = file->private_data;
1074         struct file_zero_data_information fsctl_buf;
1075         long rc;
1076         unsigned int xid;
1077         __u8 set_sparse = 1;
1078
1079         xid = get_xid();
1080
1081         inode = cfile->dentry->d_inode;
1082         cifsi = CIFS_I(inode);
1083
1084         /* Need to make file sparse, if not already, before freeing range. */
1085         /* Consider adding equivalent for compressed since it could also work */
1086         if (!smb2_set_sparse(xid, tcon, cfile, inode, set_sparse))
1087                 return -EOPNOTSUPP;
1088
1089         cifs_dbg(FYI, "offset %lld len %lld", offset, len);
1090
1091         fsctl_buf.FileOffset = cpu_to_le64(offset);
1092         fsctl_buf.BeyondFinalZero = cpu_to_le64(offset + len);
1093
1094         rc = SMB2_ioctl(xid, tcon, cfile->fid.persistent_fid,
1095                         cfile->fid.volatile_fid, FSCTL_SET_ZERO_DATA,
1096                         true /* is_fctl */, (char *)&fsctl_buf,
1097                         sizeof(struct file_zero_data_information), NULL, NULL);
1098         free_xid(xid);
1099         return rc;
1100 }
1101
1102 static long smb3_fallocate(struct file *file, struct cifs_tcon *tcon, int mode,
1103                            loff_t off, loff_t len)
1104 {
1105         /* KEEP_SIZE already checked for by do_fallocate */
1106         if (mode & FALLOC_FL_PUNCH_HOLE)
1107                 return smb3_punch_hole(file, tcon, off, len);
1108         else if (mode & FALLOC_FL_ZERO_RANGE) {
1109                 if (mode & FALLOC_FL_KEEP_SIZE)
1110                         return smb3_zero_range(file, tcon, off, len, true);
1111                 return smb3_zero_range(file, tcon, off, len, false);
1112         }
1113
1114         return -EOPNOTSUPP;
1115 }
1116
1117 static void
1118 smb2_downgrade_oplock(struct TCP_Server_Info *server,
1119                         struct cifsInodeInfo *cinode, bool set_level2)
1120 {
1121         if (set_level2)
1122                 server->ops->set_oplock_level(cinode, SMB2_OPLOCK_LEVEL_II,
1123                                                 0, NULL);
1124         else
1125                 server->ops->set_oplock_level(cinode, 0, 0, NULL);
1126 }
1127
1128 static void
1129 smb2_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1130                       unsigned int epoch, bool *purge_cache)
1131 {
1132         oplock &= 0xFF;
1133         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
1134                 return;
1135         if (oplock == SMB2_OPLOCK_LEVEL_BATCH) {
1136                 cinode->oplock = CIFS_CACHE_RHW_FLG;
1137                 cifs_dbg(FYI, "Batch Oplock granted on inode %p\n",
1138                          &cinode->vfs_inode);
1139         } else if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE) {
1140                 cinode->oplock = CIFS_CACHE_RW_FLG;
1141                 cifs_dbg(FYI, "Exclusive Oplock granted on inode %p\n",
1142                          &cinode->vfs_inode);
1143         } else if (oplock == SMB2_OPLOCK_LEVEL_II) {
1144                 cinode->oplock = CIFS_CACHE_READ_FLG;
1145                 cifs_dbg(FYI, "Level II Oplock granted on inode %p\n",
1146                          &cinode->vfs_inode);
1147         } else
1148                 cinode->oplock = 0;
1149 }
1150
1151 static void
1152 smb21_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1153                        unsigned int epoch, bool *purge_cache)
1154 {
1155         char message[5] = {0};
1156
1157         oplock &= 0xFF;
1158         if (oplock == SMB2_OPLOCK_LEVEL_NOCHANGE)
1159                 return;
1160
1161         cinode->oplock = 0;
1162         if (oplock & SMB2_LEASE_READ_CACHING_HE) {
1163                 cinode->oplock |= CIFS_CACHE_READ_FLG;
1164                 strcat(message, "R");
1165         }
1166         if (oplock & SMB2_LEASE_HANDLE_CACHING_HE) {
1167                 cinode->oplock |= CIFS_CACHE_HANDLE_FLG;
1168                 strcat(message, "H");
1169         }
1170         if (oplock & SMB2_LEASE_WRITE_CACHING_HE) {
1171                 cinode->oplock |= CIFS_CACHE_WRITE_FLG;
1172                 strcat(message, "W");
1173         }
1174         if (!cinode->oplock)
1175                 strcat(message, "None");
1176         cifs_dbg(FYI, "%s Lease granted on inode %p\n", message,
1177                  &cinode->vfs_inode);
1178 }
1179
1180 static void
1181 smb3_set_oplock_level(struct cifsInodeInfo *cinode, __u32 oplock,
1182                       unsigned int epoch, bool *purge_cache)
1183 {
1184         unsigned int old_oplock = cinode->oplock;
1185
1186         smb21_set_oplock_level(cinode, oplock, epoch, purge_cache);
1187
1188         if (purge_cache) {
1189                 *purge_cache = false;
1190                 if (old_oplock == CIFS_CACHE_READ_FLG) {
1191                         if (cinode->oplock == CIFS_CACHE_READ_FLG &&
1192                             (epoch - cinode->epoch > 0))
1193                                 *purge_cache = true;
1194                         else if (cinode->oplock == CIFS_CACHE_RH_FLG &&
1195                                  (epoch - cinode->epoch > 1))
1196                                 *purge_cache = true;
1197                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
1198                                  (epoch - cinode->epoch > 1))
1199                                 *purge_cache = true;
1200                         else if (cinode->oplock == 0 &&
1201                                  (epoch - cinode->epoch > 0))
1202                                 *purge_cache = true;
1203                 } else if (old_oplock == CIFS_CACHE_RH_FLG) {
1204                         if (cinode->oplock == CIFS_CACHE_RH_FLG &&
1205                             (epoch - cinode->epoch > 0))
1206                                 *purge_cache = true;
1207                         else if (cinode->oplock == CIFS_CACHE_RHW_FLG &&
1208                                  (epoch - cinode->epoch > 1))
1209                                 *purge_cache = true;
1210                 }
1211                 cinode->epoch = epoch;
1212         }
1213 }
1214
1215 static bool
1216 smb2_is_read_op(__u32 oplock)
1217 {
1218         return oplock == SMB2_OPLOCK_LEVEL_II;
1219 }
1220
1221 static bool
1222 smb21_is_read_op(__u32 oplock)
1223 {
1224         return (oplock & SMB2_LEASE_READ_CACHING_HE) &&
1225                !(oplock & SMB2_LEASE_WRITE_CACHING_HE);
1226 }
1227
1228 static __le32
1229 map_oplock_to_lease(u8 oplock)
1230 {
1231         if (oplock == SMB2_OPLOCK_LEVEL_EXCLUSIVE)
1232                 return SMB2_LEASE_WRITE_CACHING | SMB2_LEASE_READ_CACHING;
1233         else if (oplock == SMB2_OPLOCK_LEVEL_II)
1234                 return SMB2_LEASE_READ_CACHING;
1235         else if (oplock == SMB2_OPLOCK_LEVEL_BATCH)
1236                 return SMB2_LEASE_HANDLE_CACHING | SMB2_LEASE_READ_CACHING |
1237                        SMB2_LEASE_WRITE_CACHING;
1238         return 0;
1239 }
1240
1241 static char *
1242 smb2_create_lease_buf(u8 *lease_key, u8 oplock)
1243 {
1244         struct create_lease *buf;
1245
1246         buf = kzalloc(sizeof(struct create_lease), GFP_KERNEL);
1247         if (!buf)
1248                 return NULL;
1249
1250         buf->lcontext.LeaseKeyLow = cpu_to_le64(*((u64 *)lease_key));
1251         buf->lcontext.LeaseKeyHigh = cpu_to_le64(*((u64 *)(lease_key + 8)));
1252         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
1253
1254         buf->ccontext.DataOffset = cpu_to_le16(offsetof
1255                                         (struct create_lease, lcontext));
1256         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context));
1257         buf->ccontext.NameOffset = cpu_to_le16(offsetof
1258                                 (struct create_lease, Name));
1259         buf->ccontext.NameLength = cpu_to_le16(4);
1260         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
1261         buf->Name[0] = 'R';
1262         buf->Name[1] = 'q';
1263         buf->Name[2] = 'L';
1264         buf->Name[3] = 's';
1265         return (char *)buf;
1266 }
1267
1268 static char *
1269 smb3_create_lease_buf(u8 *lease_key, u8 oplock)
1270 {
1271         struct create_lease_v2 *buf;
1272
1273         buf = kzalloc(sizeof(struct create_lease_v2), GFP_KERNEL);
1274         if (!buf)
1275                 return NULL;
1276
1277         buf->lcontext.LeaseKeyLow = cpu_to_le64(*((u64 *)lease_key));
1278         buf->lcontext.LeaseKeyHigh = cpu_to_le64(*((u64 *)(lease_key + 8)));
1279         buf->lcontext.LeaseState = map_oplock_to_lease(oplock);
1280
1281         buf->ccontext.DataOffset = cpu_to_le16(offsetof
1282                                         (struct create_lease_v2, lcontext));
1283         buf->ccontext.DataLength = cpu_to_le32(sizeof(struct lease_context_v2));
1284         buf->ccontext.NameOffset = cpu_to_le16(offsetof
1285                                 (struct create_lease_v2, Name));
1286         buf->ccontext.NameLength = cpu_to_le16(4);
1287         /* SMB2_CREATE_REQUEST_LEASE is "RqLs" */
1288         buf->Name[0] = 'R';
1289         buf->Name[1] = 'q';
1290         buf->Name[2] = 'L';
1291         buf->Name[3] = 's';
1292         return (char *)buf;
1293 }
1294
1295 static __u8
1296 smb2_parse_lease_buf(void *buf, unsigned int *epoch)
1297 {
1298         struct create_lease *lc = (struct create_lease *)buf;
1299
1300         *epoch = 0; /* not used */
1301         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
1302                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
1303         return le32_to_cpu(lc->lcontext.LeaseState);
1304 }
1305
1306 static __u8
1307 smb3_parse_lease_buf(void *buf, unsigned int *epoch)
1308 {
1309         struct create_lease_v2 *lc = (struct create_lease_v2 *)buf;
1310
1311         *epoch = le16_to_cpu(lc->lcontext.Epoch);
1312         if (lc->lcontext.LeaseFlags & SMB2_LEASE_FLAG_BREAK_IN_PROGRESS)
1313                 return SMB2_OPLOCK_LEVEL_NOCHANGE;
1314         return le32_to_cpu(lc->lcontext.LeaseState);
1315 }
1316
1317 static unsigned int
1318 smb2_wp_retry_size(struct inode *inode)
1319 {
1320         return min_t(unsigned int, CIFS_SB(inode->i_sb)->wsize,
1321                      SMB2_MAX_BUFFER_SIZE);
1322 }
1323
1324 static bool
1325 smb2_dir_needs_close(struct cifsFileInfo *cfile)
1326 {
1327         return !cfile->invalidHandle;
1328 }
1329
1330 struct smb_version_operations smb20_operations = {
1331         .compare_fids = smb2_compare_fids,
1332         .setup_request = smb2_setup_request,
1333         .setup_async_request = smb2_setup_async_request,
1334         .check_receive = smb2_check_receive,
1335         .add_credits = smb2_add_credits,
1336         .set_credits = smb2_set_credits,
1337         .get_credits_field = smb2_get_credits_field,
1338         .get_credits = smb2_get_credits,
1339         .wait_mtu_credits = cifs_wait_mtu_credits,
1340         .get_next_mid = smb2_get_next_mid,
1341         .read_data_offset = smb2_read_data_offset,
1342         .read_data_length = smb2_read_data_length,
1343         .map_error = map_smb2_to_linux_error,
1344         .find_mid = smb2_find_mid,
1345         .check_message = smb2_check_message,
1346         .dump_detail = smb2_dump_detail,
1347         .clear_stats = smb2_clear_stats,
1348         .print_stats = smb2_print_stats,
1349         .is_oplock_break = smb2_is_valid_oplock_break,
1350         .downgrade_oplock = smb2_downgrade_oplock,
1351         .need_neg = smb2_need_neg,
1352         .negotiate = smb2_negotiate,
1353         .negotiate_wsize = smb2_negotiate_wsize,
1354         .negotiate_rsize = smb2_negotiate_rsize,
1355         .sess_setup = SMB2_sess_setup,
1356         .logoff = SMB2_logoff,
1357         .tree_connect = SMB2_tcon,
1358         .tree_disconnect = SMB2_tdis,
1359         .qfs_tcon = smb2_qfs_tcon,
1360         .is_path_accessible = smb2_is_path_accessible,
1361         .can_echo = smb2_can_echo,
1362         .echo = SMB2_echo,
1363         .query_path_info = smb2_query_path_info,
1364         .get_srv_inum = smb2_get_srv_inum,
1365         .query_file_info = smb2_query_file_info,
1366         .set_path_size = smb2_set_path_size,
1367         .set_file_size = smb2_set_file_size,
1368         .set_file_info = smb2_set_file_info,
1369         .set_compression = smb2_set_compression,
1370         .mkdir = smb2_mkdir,
1371         .mkdir_setinfo = smb2_mkdir_setinfo,
1372         .rmdir = smb2_rmdir,
1373         .unlink = smb2_unlink,
1374         .rename = smb2_rename_path,
1375         .create_hardlink = smb2_create_hardlink,
1376         .query_symlink = smb2_query_symlink,
1377         .open = smb2_open_file,
1378         .set_fid = smb2_set_fid,
1379         .close = smb2_close_file,
1380         .flush = smb2_flush_file,
1381         .async_readv = smb2_async_readv,
1382         .async_writev = smb2_async_writev,
1383         .sync_read = smb2_sync_read,
1384         .sync_write = smb2_sync_write,
1385         .query_dir_first = smb2_query_dir_first,
1386         .query_dir_next = smb2_query_dir_next,
1387         .close_dir = smb2_close_dir,
1388         .calc_smb_size = smb2_calc_size,
1389         .is_status_pending = smb2_is_status_pending,
1390         .oplock_response = smb2_oplock_response,
1391         .queryfs = smb2_queryfs,
1392         .mand_lock = smb2_mand_lock,
1393         .mand_unlock_range = smb2_unlock_range,
1394         .push_mand_locks = smb2_push_mandatory_locks,
1395         .get_lease_key = smb2_get_lease_key,
1396         .set_lease_key = smb2_set_lease_key,
1397         .new_lease_key = smb2_new_lease_key,
1398         .calc_signature = smb2_calc_signature,
1399         .is_read_op = smb2_is_read_op,
1400         .set_oplock_level = smb2_set_oplock_level,
1401         .create_lease_buf = smb2_create_lease_buf,
1402         .parse_lease_buf = smb2_parse_lease_buf,
1403         .clone_range = smb2_clone_range,
1404         .wp_retry_size = smb2_wp_retry_size,
1405         .dir_needs_close = smb2_dir_needs_close,
1406 };
1407
1408 struct smb_version_operations smb21_operations = {
1409         .compare_fids = smb2_compare_fids,
1410         .setup_request = smb2_setup_request,
1411         .setup_async_request = smb2_setup_async_request,
1412         .check_receive = smb2_check_receive,
1413         .add_credits = smb2_add_credits,
1414         .set_credits = smb2_set_credits,
1415         .get_credits_field = smb2_get_credits_field,
1416         .get_credits = smb2_get_credits,
1417         .wait_mtu_credits = smb2_wait_mtu_credits,
1418         .get_next_mid = smb2_get_next_mid,
1419         .read_data_offset = smb2_read_data_offset,
1420         .read_data_length = smb2_read_data_length,
1421         .map_error = map_smb2_to_linux_error,
1422         .find_mid = smb2_find_mid,
1423         .check_message = smb2_check_message,
1424         .dump_detail = smb2_dump_detail,
1425         .clear_stats = smb2_clear_stats,
1426         .print_stats = smb2_print_stats,
1427         .is_oplock_break = smb2_is_valid_oplock_break,
1428         .downgrade_oplock = smb2_downgrade_oplock,
1429         .need_neg = smb2_need_neg,
1430         .negotiate = smb2_negotiate,
1431         .negotiate_wsize = smb2_negotiate_wsize,
1432         .negotiate_rsize = smb2_negotiate_rsize,
1433         .sess_setup = SMB2_sess_setup,
1434         .logoff = SMB2_logoff,
1435         .tree_connect = SMB2_tcon,
1436         .tree_disconnect = SMB2_tdis,
1437         .qfs_tcon = smb2_qfs_tcon,
1438         .is_path_accessible = smb2_is_path_accessible,
1439         .can_echo = smb2_can_echo,
1440         .echo = SMB2_echo,
1441         .query_path_info = smb2_query_path_info,
1442         .get_srv_inum = smb2_get_srv_inum,
1443         .query_file_info = smb2_query_file_info,
1444         .set_path_size = smb2_set_path_size,
1445         .set_file_size = smb2_set_file_size,
1446         .set_file_info = smb2_set_file_info,
1447         .set_compression = smb2_set_compression,
1448         .mkdir = smb2_mkdir,
1449         .mkdir_setinfo = smb2_mkdir_setinfo,
1450         .rmdir = smb2_rmdir,
1451         .unlink = smb2_unlink,
1452         .rename = smb2_rename_path,
1453         .create_hardlink = smb2_create_hardlink,
1454         .query_symlink = smb2_query_symlink,
1455         .open = smb2_open_file,
1456         .set_fid = smb2_set_fid,
1457         .close = smb2_close_file,
1458         .flush = smb2_flush_file,
1459         .async_readv = smb2_async_readv,
1460         .async_writev = smb2_async_writev,
1461         .sync_read = smb2_sync_read,
1462         .sync_write = smb2_sync_write,
1463         .query_dir_first = smb2_query_dir_first,
1464         .query_dir_next = smb2_query_dir_next,
1465         .close_dir = smb2_close_dir,
1466         .calc_smb_size = smb2_calc_size,
1467         .is_status_pending = smb2_is_status_pending,
1468         .oplock_response = smb2_oplock_response,
1469         .queryfs = smb2_queryfs,
1470         .mand_lock = smb2_mand_lock,
1471         .mand_unlock_range = smb2_unlock_range,
1472         .push_mand_locks = smb2_push_mandatory_locks,
1473         .get_lease_key = smb2_get_lease_key,
1474         .set_lease_key = smb2_set_lease_key,
1475         .new_lease_key = smb2_new_lease_key,
1476         .calc_signature = smb2_calc_signature,
1477         .is_read_op = smb21_is_read_op,
1478         .set_oplock_level = smb21_set_oplock_level,
1479         .create_lease_buf = smb2_create_lease_buf,
1480         .parse_lease_buf = smb2_parse_lease_buf,
1481         .clone_range = smb2_clone_range,
1482         .wp_retry_size = smb2_wp_retry_size,
1483         .dir_needs_close = smb2_dir_needs_close,
1484 };
1485
1486 struct smb_version_operations smb30_operations = {
1487         .compare_fids = smb2_compare_fids,
1488         .setup_request = smb2_setup_request,
1489         .setup_async_request = smb2_setup_async_request,
1490         .check_receive = smb2_check_receive,
1491         .add_credits = smb2_add_credits,
1492         .set_credits = smb2_set_credits,
1493         .get_credits_field = smb2_get_credits_field,
1494         .get_credits = smb2_get_credits,
1495         .wait_mtu_credits = smb2_wait_mtu_credits,
1496         .get_next_mid = smb2_get_next_mid,
1497         .read_data_offset = smb2_read_data_offset,
1498         .read_data_length = smb2_read_data_length,
1499         .map_error = map_smb2_to_linux_error,
1500         .find_mid = smb2_find_mid,
1501         .check_message = smb2_check_message,
1502         .dump_detail = smb2_dump_detail,
1503         .clear_stats = smb2_clear_stats,
1504         .print_stats = smb2_print_stats,
1505         .dump_share_caps = smb2_dump_share_caps,
1506         .is_oplock_break = smb2_is_valid_oplock_break,
1507         .downgrade_oplock = smb2_downgrade_oplock,
1508         .need_neg = smb2_need_neg,
1509         .negotiate = smb2_negotiate,
1510         .negotiate_wsize = smb2_negotiate_wsize,
1511         .negotiate_rsize = smb2_negotiate_rsize,
1512         .sess_setup = SMB2_sess_setup,
1513         .logoff = SMB2_logoff,
1514         .tree_connect = SMB2_tcon,
1515         .tree_disconnect = SMB2_tdis,
1516         .qfs_tcon = smb3_qfs_tcon,
1517         .is_path_accessible = smb2_is_path_accessible,
1518         .can_echo = smb2_can_echo,
1519         .echo = SMB2_echo,
1520         .query_path_info = smb2_query_path_info,
1521         .get_srv_inum = smb2_get_srv_inum,
1522         .query_file_info = smb2_query_file_info,
1523         .set_path_size = smb2_set_path_size,
1524         .set_file_size = smb2_set_file_size,
1525         .set_file_info = smb2_set_file_info,
1526         .set_compression = smb2_set_compression,
1527         .mkdir = smb2_mkdir,
1528         .mkdir_setinfo = smb2_mkdir_setinfo,
1529         .rmdir = smb2_rmdir,
1530         .unlink = smb2_unlink,
1531         .rename = smb2_rename_path,
1532         .create_hardlink = smb2_create_hardlink,
1533         .query_symlink = smb2_query_symlink,
1534         .open = smb2_open_file,
1535         .set_fid = smb2_set_fid,
1536         .close = smb2_close_file,
1537         .flush = smb2_flush_file,
1538         .async_readv = smb2_async_readv,
1539         .async_writev = smb2_async_writev,
1540         .sync_read = smb2_sync_read,
1541         .sync_write = smb2_sync_write,
1542         .query_dir_first = smb2_query_dir_first,
1543         .query_dir_next = smb2_query_dir_next,
1544         .close_dir = smb2_close_dir,
1545         .calc_smb_size = smb2_calc_size,
1546         .is_status_pending = smb2_is_status_pending,
1547         .oplock_response = smb2_oplock_response,
1548         .queryfs = smb2_queryfs,
1549         .mand_lock = smb2_mand_lock,
1550         .mand_unlock_range = smb2_unlock_range,
1551         .push_mand_locks = smb2_push_mandatory_locks,
1552         .get_lease_key = smb2_get_lease_key,
1553         .set_lease_key = smb2_set_lease_key,
1554         .new_lease_key = smb2_new_lease_key,
1555         .generate_signingkey = generate_smb3signingkey,
1556         .calc_signature = smb3_calc_signature,
1557         .is_read_op = smb21_is_read_op,
1558         .set_oplock_level = smb3_set_oplock_level,
1559         .create_lease_buf = smb3_create_lease_buf,
1560         .parse_lease_buf = smb3_parse_lease_buf,
1561         .clone_range = smb2_clone_range,
1562         .validate_negotiate = smb3_validate_negotiate,
1563         .wp_retry_size = smb2_wp_retry_size,
1564         .dir_needs_close = smb2_dir_needs_close,
1565         .fallocate = smb3_fallocate,
1566 };
1567
1568 struct smb_version_values smb20_values = {
1569         .version_string = SMB20_VERSION_STRING,
1570         .protocol_id = SMB20_PROT_ID,
1571         .req_capabilities = 0, /* MBZ */
1572         .large_lock_type = 0,
1573         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
1574         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
1575         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
1576         .header_size = sizeof(struct smb2_hdr),
1577         .max_header_size = MAX_SMB2_HDR_SIZE,
1578         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
1579         .lock_cmd = SMB2_LOCK,
1580         .cap_unix = 0,
1581         .cap_nt_find = SMB2_NT_FIND,
1582         .cap_large_files = SMB2_LARGE_FILES,
1583         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
1584         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
1585         .create_lease_size = sizeof(struct create_lease),
1586 };
1587
1588 struct smb_version_values smb21_values = {
1589         .version_string = SMB21_VERSION_STRING,
1590         .protocol_id = SMB21_PROT_ID,
1591         .req_capabilities = 0, /* MBZ on negotiate req until SMB3 dialect */
1592         .large_lock_type = 0,
1593         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
1594         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
1595         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
1596         .header_size = sizeof(struct smb2_hdr),
1597         .max_header_size = MAX_SMB2_HDR_SIZE,
1598         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
1599         .lock_cmd = SMB2_LOCK,
1600         .cap_unix = 0,
1601         .cap_nt_find = SMB2_NT_FIND,
1602         .cap_large_files = SMB2_LARGE_FILES,
1603         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
1604         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
1605         .create_lease_size = sizeof(struct create_lease),
1606 };
1607
1608 struct smb_version_values smb30_values = {
1609         .version_string = SMB30_VERSION_STRING,
1610         .protocol_id = SMB30_PROT_ID,
1611         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU,
1612         .large_lock_type = 0,
1613         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
1614         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
1615         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
1616         .header_size = sizeof(struct smb2_hdr),
1617         .max_header_size = MAX_SMB2_HDR_SIZE,
1618         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
1619         .lock_cmd = SMB2_LOCK,
1620         .cap_unix = 0,
1621         .cap_nt_find = SMB2_NT_FIND,
1622         .cap_large_files = SMB2_LARGE_FILES,
1623         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
1624         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
1625         .create_lease_size = sizeof(struct create_lease_v2),
1626 };
1627
1628 struct smb_version_values smb302_values = {
1629         .version_string = SMB302_VERSION_STRING,
1630         .protocol_id = SMB302_PROT_ID,
1631         .req_capabilities = SMB2_GLOBAL_CAP_DFS | SMB2_GLOBAL_CAP_LEASING | SMB2_GLOBAL_CAP_LARGE_MTU,
1632         .large_lock_type = 0,
1633         .exclusive_lock_type = SMB2_LOCKFLAG_EXCLUSIVE_LOCK,
1634         .shared_lock_type = SMB2_LOCKFLAG_SHARED_LOCK,
1635         .unlock_lock_type = SMB2_LOCKFLAG_UNLOCK,
1636         .header_size = sizeof(struct smb2_hdr),
1637         .max_header_size = MAX_SMB2_HDR_SIZE,
1638         .read_rsp_size = sizeof(struct smb2_read_rsp) - 1,
1639         .lock_cmd = SMB2_LOCK,
1640         .cap_unix = 0,
1641         .cap_nt_find = SMB2_NT_FIND,
1642         .cap_large_files = SMB2_LARGE_FILES,
1643         .signing_enabled = SMB2_NEGOTIATE_SIGNING_ENABLED | SMB2_NEGOTIATE_SIGNING_REQUIRED,
1644         .signing_required = SMB2_NEGOTIATE_SIGNING_REQUIRED,
1645         .create_lease_size = sizeof(struct create_lease_v2),
1646 };