smb2misc.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683
  1. /*
  2. * fs/cifs/smb2misc.c
  3. *
  4. * Copyright (C) International Business Machines Corp., 2002,2011
  5. * Etersoft, 2012
  6. * Author(s): Steve French (sfrench@us.ibm.com)
  7. * Pavel Shilovsky (pshilovsky@samba.org) 2012
  8. *
  9. * This library is free software; you can redistribute it and/or modify
  10. * it under the terms of the GNU Lesser General Public License as published
  11. * by the Free Software Foundation; either version 2.1 of the License, or
  12. * (at your option) any later version.
  13. *
  14. * This library is distributed in the hope that it will be useful,
  15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
  17. * the GNU Lesser General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU Lesser General Public License
  20. * along with this library; if not, write to the Free Software
  21. * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22. */
  23. #include <linux/ctype.h>
  24. #include "smb2pdu.h"
  25. #include "cifsglob.h"
  26. #include "cifsproto.h"
  27. #include "smb2proto.h"
  28. #include "cifs_debug.h"
  29. #include "cifs_unicode.h"
  30. #include "smb2status.h"
  31. static int
  32. check_smb2_hdr(struct smb2_hdr *hdr, __u64 mid)
  33. {
  34. __u64 wire_mid = le64_to_cpu(hdr->MessageId);
  35. /*
  36. * Make sure that this really is an SMB, that it is a response,
  37. * and that the message ids match.
  38. */
  39. if ((*(__le32 *)hdr->ProtocolId == SMB2_PROTO_NUMBER) &&
  40. (mid == wire_mid)) {
  41. if (hdr->Flags & SMB2_FLAGS_SERVER_TO_REDIR)
  42. return 0;
  43. else {
  44. /* only one valid case where server sends us request */
  45. if (hdr->Command == SMB2_OPLOCK_BREAK)
  46. return 0;
  47. else
  48. cifs_dbg(VFS, "Received Request not response\n");
  49. }
  50. } else { /* bad signature or mid */
  51. if (*(__le32 *)hdr->ProtocolId != SMB2_PROTO_NUMBER)
  52. cifs_dbg(VFS, "Bad protocol string signature header %x\n",
  53. *(unsigned int *) hdr->ProtocolId);
  54. if (mid != wire_mid)
  55. cifs_dbg(VFS, "Mids do not match: %llu and %llu\n",
  56. mid, wire_mid);
  57. }
  58. cifs_dbg(VFS, "Bad SMB detected. The Mid=%llu\n", wire_mid);
  59. return 1;
  60. }
  61. /*
  62. * The following table defines the expected "StructureSize" of SMB2 responses
  63. * in order by SMB2 command. This is similar to "wct" in SMB/CIFS responses.
  64. *
  65. * Note that commands are defined in smb2pdu.h in le16 but the array below is
  66. * indexed by command in host byte order
  67. */
  68. static const __le16 smb2_rsp_struct_sizes[NUMBER_OF_SMB2_COMMANDS] = {
  69. /* SMB2_NEGOTIATE */ cpu_to_le16(65),
  70. /* SMB2_SESSION_SETUP */ cpu_to_le16(9),
  71. /* SMB2_LOGOFF */ cpu_to_le16(4),
  72. /* SMB2_TREE_CONNECT */ cpu_to_le16(16),
  73. /* SMB2_TREE_DISCONNECT */ cpu_to_le16(4),
  74. /* SMB2_CREATE */ cpu_to_le16(89),
  75. /* SMB2_CLOSE */ cpu_to_le16(60),
  76. /* SMB2_FLUSH */ cpu_to_le16(4),
  77. /* SMB2_READ */ cpu_to_le16(17),
  78. /* SMB2_WRITE */ cpu_to_le16(17),
  79. /* SMB2_LOCK */ cpu_to_le16(4),
  80. /* SMB2_IOCTL */ cpu_to_le16(49),
  81. /* BB CHECK this ... not listed in documentation */
  82. /* SMB2_CANCEL */ cpu_to_le16(0),
  83. /* SMB2_ECHO */ cpu_to_le16(4),
  84. /* SMB2_QUERY_DIRECTORY */ cpu_to_le16(9),
  85. /* SMB2_CHANGE_NOTIFY */ cpu_to_le16(9),
  86. /* SMB2_QUERY_INFO */ cpu_to_le16(9),
  87. /* SMB2_SET_INFO */ cpu_to_le16(2),
  88. /* BB FIXME can also be 44 for lease break */
  89. /* SMB2_OPLOCK_BREAK */ cpu_to_le16(24)
  90. };
  91. int
  92. smb2_check_message(char *buf, unsigned int length)
  93. {
  94. struct smb2_hdr *hdr = (struct smb2_hdr *)buf;
  95. struct smb2_pdu *pdu = (struct smb2_pdu *)hdr;
  96. __u64 mid = le64_to_cpu(hdr->MessageId);
  97. __u32 len = get_rfc1002_length(buf);
  98. __u32 clc_len; /* calculated length */
  99. int command;
  100. /* BB disable following printk later */
  101. cifs_dbg(FYI, "%s length: 0x%x, smb_buf_length: 0x%x\n",
  102. __func__, length, len);
  103. /*
  104. * Add function to do table lookup of StructureSize by command
  105. * ie Validate the wct via smb2_struct_sizes table above
  106. */
  107. if (length < sizeof(struct smb2_pdu)) {
  108. if ((length >= sizeof(struct smb2_hdr)) && (hdr->Status != 0)) {
  109. pdu->StructureSize2 = 0;
  110. /*
  111. * As with SMB/CIFS, on some error cases servers may
  112. * not return wct properly
  113. */
  114. return 0;
  115. } else {
  116. cifs_dbg(VFS, "Length less than SMB header size\n");
  117. }
  118. return 1;
  119. }
  120. if (len > CIFSMaxBufSize + MAX_SMB2_HDR_SIZE - 4) {
  121. cifs_dbg(VFS, "SMB length greater than maximum, mid=%llu\n",
  122. mid);
  123. return 1;
  124. }
  125. if (check_smb2_hdr(hdr, mid))
  126. return 1;
  127. if (hdr->StructureSize != SMB2_HEADER_STRUCTURE_SIZE) {
  128. cifs_dbg(VFS, "Illegal structure size %u\n",
  129. le16_to_cpu(hdr->StructureSize));
  130. return 1;
  131. }
  132. command = le16_to_cpu(hdr->Command);
  133. if (command >= NUMBER_OF_SMB2_COMMANDS) {
  134. cifs_dbg(VFS, "Illegal SMB2 command %d\n", command);
  135. return 1;
  136. }
  137. if (smb2_rsp_struct_sizes[command] != pdu->StructureSize2) {
  138. if (command != SMB2_OPLOCK_BREAK_HE && (hdr->Status == 0 ||
  139. pdu->StructureSize2 != SMB2_ERROR_STRUCTURE_SIZE2)) {
  140. /* error packets have 9 byte structure size */
  141. cifs_dbg(VFS, "Illegal response size %u for command %d\n",
  142. le16_to_cpu(pdu->StructureSize2), command);
  143. return 1;
  144. } else if (command == SMB2_OPLOCK_BREAK_HE && (hdr->Status == 0)
  145. && (le16_to_cpu(pdu->StructureSize2) != 44)
  146. && (le16_to_cpu(pdu->StructureSize2) != 36)) {
  147. /* special case for SMB2.1 lease break message */
  148. cifs_dbg(VFS, "Illegal response size %d for oplock break\n",
  149. le16_to_cpu(pdu->StructureSize2));
  150. return 1;
  151. }
  152. }
  153. if (4 + len != length) {
  154. cifs_dbg(VFS, "Total length %u RFC1002 length %u mismatch mid %llu\n",
  155. length, 4 + len, mid);
  156. return 1;
  157. }
  158. clc_len = smb2_calc_size(hdr);
  159. if (4 + len != clc_len) {
  160. cifs_dbg(FYI, "Calculated size %u length %u mismatch mid %llu\n",
  161. clc_len, 4 + len, mid);
  162. /* create failed on symlink */
  163. if (command == SMB2_CREATE_HE &&
  164. hdr->Status == STATUS_STOPPED_ON_SYMLINK)
  165. return 0;
  166. /* Windows 7 server returns 24 bytes more */
  167. if (clc_len + 20 == len && command == SMB2_OPLOCK_BREAK_HE)
  168. return 0;
  169. /* server can return one byte more due to implied bcc[0] */
  170. if (clc_len == 4 + len + 1)
  171. return 0;
  172. /*
  173. * Some windows servers (win2016) will pad also the final
  174. * PDU in a compound to 8 bytes.
  175. */
  176. if (((clc_len + 7) & ~7) == len)
  177. return 0;
  178. /*
  179. * MacOS server pads after SMB2.1 write response with 3 bytes
  180. * of junk. Other servers match RFC1001 len to actual
  181. * SMB2/SMB3 frame length (header + smb2 response specific data)
  182. * Log the server error (once), but allow it and continue
  183. * since the frame is parseable.
  184. */
  185. if (clc_len < 4 /* RFC1001 header size */ + len) {
  186. printk_once(KERN_WARNING
  187. "SMB2 server sent bad RFC1001 len %d not %d\n",
  188. len, clc_len - 4);
  189. return 0;
  190. }
  191. return 1;
  192. }
  193. return 0;
  194. }
  195. /*
  196. * The size of the variable area depends on the offset and length fields
  197. * located in different fields for various SMB2 responses. SMB2 responses
  198. * with no variable length info, show an offset of zero for the offset field.
  199. */
  200. static const bool has_smb2_data_area[NUMBER_OF_SMB2_COMMANDS] = {
  201. /* SMB2_NEGOTIATE */ true,
  202. /* SMB2_SESSION_SETUP */ true,
  203. /* SMB2_LOGOFF */ false,
  204. /* SMB2_TREE_CONNECT */ false,
  205. /* SMB2_TREE_DISCONNECT */ false,
  206. /* SMB2_CREATE */ true,
  207. /* SMB2_CLOSE */ false,
  208. /* SMB2_FLUSH */ false,
  209. /* SMB2_READ */ true,
  210. /* SMB2_WRITE */ false,
  211. /* SMB2_LOCK */ false,
  212. /* SMB2_IOCTL */ true,
  213. /* SMB2_CANCEL */ false, /* BB CHECK this not listed in documentation */
  214. /* SMB2_ECHO */ false,
  215. /* SMB2_QUERY_DIRECTORY */ true,
  216. /* SMB2_CHANGE_NOTIFY */ true,
  217. /* SMB2_QUERY_INFO */ true,
  218. /* SMB2_SET_INFO */ false,
  219. /* SMB2_OPLOCK_BREAK */ false
  220. };
  221. /*
  222. * Returns the pointer to the beginning of the data area. Length of the data
  223. * area and the offset to it (from the beginning of the smb are also returned.
  224. */
  225. char *
  226. smb2_get_data_area_len(int *off, int *len, struct smb2_hdr *hdr)
  227. {
  228. *off = 0;
  229. *len = 0;
  230. /* error responses do not have data area */
  231. if (hdr->Status && hdr->Status != STATUS_MORE_PROCESSING_REQUIRED &&
  232. (((struct smb2_err_rsp *)hdr)->StructureSize) ==
  233. SMB2_ERROR_STRUCTURE_SIZE2)
  234. return NULL;
  235. /*
  236. * Following commands have data areas so we have to get the location
  237. * of the data buffer offset and data buffer length for the particular
  238. * command.
  239. */
  240. switch (hdr->Command) {
  241. case SMB2_NEGOTIATE:
  242. *off = le16_to_cpu(
  243. ((struct smb2_negotiate_rsp *)hdr)->SecurityBufferOffset);
  244. *len = le16_to_cpu(
  245. ((struct smb2_negotiate_rsp *)hdr)->SecurityBufferLength);
  246. break;
  247. case SMB2_SESSION_SETUP:
  248. *off = le16_to_cpu(
  249. ((struct smb2_sess_setup_rsp *)hdr)->SecurityBufferOffset);
  250. *len = le16_to_cpu(
  251. ((struct smb2_sess_setup_rsp *)hdr)->SecurityBufferLength);
  252. break;
  253. case SMB2_CREATE:
  254. *off = le32_to_cpu(
  255. ((struct smb2_create_rsp *)hdr)->CreateContextsOffset);
  256. *len = le32_to_cpu(
  257. ((struct smb2_create_rsp *)hdr)->CreateContextsLength);
  258. break;
  259. case SMB2_QUERY_INFO:
  260. *off = le16_to_cpu(
  261. ((struct smb2_query_info_rsp *)hdr)->OutputBufferOffset);
  262. *len = le32_to_cpu(
  263. ((struct smb2_query_info_rsp *)hdr)->OutputBufferLength);
  264. break;
  265. case SMB2_READ:
  266. *off = ((struct smb2_read_rsp *)hdr)->DataOffset;
  267. *len = le32_to_cpu(((struct smb2_read_rsp *)hdr)->DataLength);
  268. break;
  269. case SMB2_QUERY_DIRECTORY:
  270. *off = le16_to_cpu(
  271. ((struct smb2_query_directory_rsp *)hdr)->OutputBufferOffset);
  272. *len = le32_to_cpu(
  273. ((struct smb2_query_directory_rsp *)hdr)->OutputBufferLength);
  274. break;
  275. case SMB2_IOCTL:
  276. *off = le32_to_cpu(
  277. ((struct smb2_ioctl_rsp *)hdr)->OutputOffset);
  278. *len = le32_to_cpu(((struct smb2_ioctl_rsp *)hdr)->OutputCount);
  279. break;
  280. case SMB2_CHANGE_NOTIFY:
  281. default:
  282. /* BB FIXME for unimplemented cases above */
  283. cifs_dbg(VFS, "no length check for command\n");
  284. break;
  285. }
  286. /*
  287. * Invalid length or offset probably means data area is invalid, but
  288. * we have little choice but to ignore the data area in this case.
  289. */
  290. if (*off > 4096) {
  291. cifs_dbg(VFS, "offset %d too large, data area ignored\n", *off);
  292. *len = 0;
  293. *off = 0;
  294. } else if (*off < 0) {
  295. cifs_dbg(VFS, "negative offset %d to data invalid ignore data area\n",
  296. *off);
  297. *off = 0;
  298. *len = 0;
  299. } else if (*len < 0) {
  300. cifs_dbg(VFS, "negative data length %d invalid, data area ignored\n",
  301. *len);
  302. *len = 0;
  303. } else if (*len > 128 * 1024) {
  304. cifs_dbg(VFS, "data area larger than 128K: %d\n", *len);
  305. *len = 0;
  306. }
  307. /* return pointer to beginning of data area, ie offset from SMB start */
  308. if ((*off != 0) && (*len != 0))
  309. return (char *)(&hdr->ProtocolId[0]) + *off;
  310. else
  311. return NULL;
  312. }
  313. /*
  314. * Calculate the size of the SMB message based on the fixed header
  315. * portion, the number of word parameters and the data portion of the message.
  316. */
  317. unsigned int
  318. smb2_calc_size(void *buf)
  319. {
  320. struct smb2_hdr *hdr = (struct smb2_hdr *)buf;
  321. struct smb2_pdu *pdu = (struct smb2_pdu *)hdr;
  322. int offset; /* the offset from the beginning of SMB to data area */
  323. int data_length; /* the length of the variable length data area */
  324. /* Structure Size has already been checked to make sure it is 64 */
  325. int len = 4 + le16_to_cpu(pdu->hdr.StructureSize);
  326. /*
  327. * StructureSize2, ie length of fixed parameter area has already
  328. * been checked to make sure it is the correct length.
  329. */
  330. len += le16_to_cpu(pdu->StructureSize2);
  331. if (has_smb2_data_area[le16_to_cpu(hdr->Command)] == false)
  332. goto calc_size_exit;
  333. smb2_get_data_area_len(&offset, &data_length, hdr);
  334. cifs_dbg(FYI, "SMB2 data length %d offset %d\n", data_length, offset);
  335. if (data_length > 0) {
  336. /*
  337. * Check to make sure that data area begins after fixed area,
  338. * Note that last byte of the fixed area is part of data area
  339. * for some commands, typically those with odd StructureSize,
  340. * so we must add one to the calculation (and 4 to account for
  341. * the size of the RFC1001 hdr.
  342. */
  343. if (offset + 4 + 1 < len) {
  344. cifs_dbg(VFS, "data area offset %d overlaps SMB2 header %d\n",
  345. offset + 4 + 1, len);
  346. data_length = 0;
  347. } else {
  348. len = 4 + offset + data_length;
  349. }
  350. }
  351. calc_size_exit:
  352. cifs_dbg(FYI, "SMB2 len %d\n", len);
  353. return len;
  354. }
  355. /* Note: caller must free return buffer */
  356. __le16 *
  357. cifs_convert_path_to_utf16(const char *from, struct cifs_sb_info *cifs_sb)
  358. {
  359. int len;
  360. const char *start_of_path;
  361. __le16 *to;
  362. int map_type;
  363. if (cifs_sb->mnt_cifs_flags & CIFS_MOUNT_MAP_SFM_CHR)
  364. map_type = SFM_MAP_UNI_RSVD;
  365. else if (cifs_sb->mnt_cifs_flags & CIFS_MOUNT_MAP_SPECIAL_CHR)
  366. map_type = SFU_MAP_UNI_RSVD;
  367. else
  368. map_type = NO_MAP_UNI_RSVD;
  369. /* Windows doesn't allow paths beginning with \ */
  370. if (from[0] == '\\')
  371. start_of_path = from + 1;
  372. else
  373. start_of_path = from;
  374. to = cifs_strndup_to_utf16(start_of_path, PATH_MAX, &len,
  375. cifs_sb->local_nls, map_type);
  376. return to;
  377. }
  378. __le32
  379. smb2_get_lease_state(struct cifsInodeInfo *cinode)
  380. {
  381. __le32 lease = 0;
  382. if (CIFS_CACHE_WRITE(cinode))
  383. lease |= SMB2_LEASE_WRITE_CACHING;
  384. if (CIFS_CACHE_HANDLE(cinode))
  385. lease |= SMB2_LEASE_HANDLE_CACHING;
  386. if (CIFS_CACHE_READ(cinode))
  387. lease |= SMB2_LEASE_READ_CACHING;
  388. return lease;
  389. }
  390. struct smb2_lease_break_work {
  391. struct work_struct lease_break;
  392. struct tcon_link *tlink;
  393. __u8 lease_key[16];
  394. __le32 lease_state;
  395. };
  396. static void
  397. cifs_ses_oplock_break(struct work_struct *work)
  398. {
  399. struct smb2_lease_break_work *lw = container_of(work,
  400. struct smb2_lease_break_work, lease_break);
  401. int rc;
  402. rc = SMB2_lease_break(0, tlink_tcon(lw->tlink), lw->lease_key,
  403. lw->lease_state);
  404. cifs_dbg(FYI, "Lease release rc %d\n", rc);
  405. cifs_put_tlink(lw->tlink);
  406. kfree(lw);
  407. }
  408. static bool
  409. smb2_tcon_has_lease(struct cifs_tcon *tcon, struct smb2_lease_break *rsp,
  410. struct smb2_lease_break_work *lw)
  411. {
  412. bool found;
  413. __u8 lease_state;
  414. struct list_head *tmp;
  415. struct cifsFileInfo *cfile;
  416. struct TCP_Server_Info *server = tcon->ses->server;
  417. struct cifs_pending_open *open;
  418. struct cifsInodeInfo *cinode;
  419. int ack_req = le32_to_cpu(rsp->Flags &
  420. SMB2_NOTIFY_BREAK_LEASE_FLAG_ACK_REQUIRED);
  421. lease_state = le32_to_cpu(rsp->NewLeaseState);
  422. list_for_each(tmp, &tcon->openFileList) {
  423. cfile = list_entry(tmp, struct cifsFileInfo, tlist);
  424. cinode = CIFS_I(d_inode(cfile->dentry));
  425. if (memcmp(cinode->lease_key, rsp->LeaseKey,
  426. SMB2_LEASE_KEY_SIZE))
  427. continue;
  428. cifs_dbg(FYI, "found in the open list\n");
  429. cifs_dbg(FYI, "lease key match, lease break 0x%x\n",
  430. le32_to_cpu(rsp->NewLeaseState));
  431. server->ops->set_oplock_level(cinode, lease_state, 0, NULL);
  432. if (ack_req)
  433. cfile->oplock_break_cancelled = false;
  434. else
  435. cfile->oplock_break_cancelled = true;
  436. queue_work(cifsiod_wq, &cfile->oplock_break);
  437. kfree(lw);
  438. return true;
  439. }
  440. found = false;
  441. list_for_each_entry(open, &tcon->pending_opens, olist) {
  442. if (memcmp(open->lease_key, rsp->LeaseKey,
  443. SMB2_LEASE_KEY_SIZE))
  444. continue;
  445. if (!found && ack_req) {
  446. found = true;
  447. memcpy(lw->lease_key, open->lease_key,
  448. SMB2_LEASE_KEY_SIZE);
  449. lw->tlink = cifs_get_tlink(open->tlink);
  450. queue_work(cifsiod_wq, &lw->lease_break);
  451. }
  452. cifs_dbg(FYI, "found in the pending open list\n");
  453. cifs_dbg(FYI, "lease key match, lease break 0x%x\n",
  454. le32_to_cpu(rsp->NewLeaseState));
  455. open->oplock = lease_state;
  456. }
  457. return found;
  458. }
  459. static bool
  460. smb2_is_valid_lease_break(char *buffer)
  461. {
  462. struct smb2_lease_break *rsp = (struct smb2_lease_break *)buffer;
  463. struct list_head *tmp, *tmp1, *tmp2;
  464. struct TCP_Server_Info *server;
  465. struct cifs_ses *ses;
  466. struct cifs_tcon *tcon;
  467. struct smb2_lease_break_work *lw;
  468. lw = kmalloc(sizeof(struct smb2_lease_break_work), GFP_KERNEL);
  469. if (!lw)
  470. return false;
  471. INIT_WORK(&lw->lease_break, cifs_ses_oplock_break);
  472. lw->lease_state = rsp->NewLeaseState;
  473. cifs_dbg(FYI, "Checking for lease break\n");
  474. /* look up tcon based on tid & uid */
  475. spin_lock(&cifs_tcp_ses_lock);
  476. list_for_each(tmp, &cifs_tcp_ses_list) {
  477. server = list_entry(tmp, struct TCP_Server_Info, tcp_ses_list);
  478. list_for_each(tmp1, &server->smb_ses_list) {
  479. ses = list_entry(tmp1, struct cifs_ses, smb_ses_list);
  480. list_for_each(tmp2, &ses->tcon_list) {
  481. tcon = list_entry(tmp2, struct cifs_tcon,
  482. tcon_list);
  483. spin_lock(&tcon->open_file_lock);
  484. cifs_stats_inc(
  485. &tcon->stats.cifs_stats.num_oplock_brks);
  486. if (smb2_tcon_has_lease(tcon, rsp, lw)) {
  487. spin_unlock(&tcon->open_file_lock);
  488. spin_unlock(&cifs_tcp_ses_lock);
  489. return true;
  490. }
  491. spin_unlock(&tcon->open_file_lock);
  492. }
  493. }
  494. }
  495. spin_unlock(&cifs_tcp_ses_lock);
  496. kfree(lw);
  497. cifs_dbg(FYI, "Can not process lease break - no lease matched\n");
  498. return false;
  499. }
  500. bool
  501. smb2_is_valid_oplock_break(char *buffer, struct TCP_Server_Info *server)
  502. {
  503. struct smb2_oplock_break *rsp = (struct smb2_oplock_break *)buffer;
  504. struct list_head *tmp, *tmp1, *tmp2;
  505. struct cifs_ses *ses;
  506. struct cifs_tcon *tcon;
  507. struct cifsInodeInfo *cinode;
  508. struct cifsFileInfo *cfile;
  509. cifs_dbg(FYI, "Checking for oplock break\n");
  510. if (rsp->hdr.Command != SMB2_OPLOCK_BREAK)
  511. return false;
  512. if (rsp->StructureSize !=
  513. smb2_rsp_struct_sizes[SMB2_OPLOCK_BREAK_HE]) {
  514. if (le16_to_cpu(rsp->StructureSize) == 44)
  515. return smb2_is_valid_lease_break(buffer);
  516. else
  517. return false;
  518. }
  519. cifs_dbg(FYI, "oplock level 0x%x\n", rsp->OplockLevel);
  520. /* look up tcon based on tid & uid */
  521. spin_lock(&cifs_tcp_ses_lock);
  522. list_for_each(tmp, &server->smb_ses_list) {
  523. ses = list_entry(tmp, struct cifs_ses, smb_ses_list);
  524. list_for_each(tmp1, &ses->tcon_list) {
  525. tcon = list_entry(tmp1, struct cifs_tcon, tcon_list);
  526. cifs_stats_inc(&tcon->stats.cifs_stats.num_oplock_brks);
  527. spin_lock(&tcon->open_file_lock);
  528. list_for_each(tmp2, &tcon->openFileList) {
  529. cfile = list_entry(tmp2, struct cifsFileInfo,
  530. tlist);
  531. if (rsp->PersistentFid !=
  532. cfile->fid.persistent_fid ||
  533. rsp->VolatileFid !=
  534. cfile->fid.volatile_fid)
  535. continue;
  536. cifs_dbg(FYI, "file id match, oplock break\n");
  537. cinode = CIFS_I(d_inode(cfile->dentry));
  538. spin_lock(&cfile->file_info_lock);
  539. if (!CIFS_CACHE_WRITE(cinode) &&
  540. rsp->OplockLevel == SMB2_OPLOCK_LEVEL_NONE)
  541. cfile->oplock_break_cancelled = true;
  542. else
  543. cfile->oplock_break_cancelled = false;
  544. set_bit(CIFS_INODE_PENDING_OPLOCK_BREAK,
  545. &cinode->flags);
  546. /*
  547. * Set flag if the server downgrades the oplock
  548. * to L2 else clear.
  549. */
  550. if (rsp->OplockLevel)
  551. set_bit(
  552. CIFS_INODE_DOWNGRADE_OPLOCK_TO_L2,
  553. &cinode->flags);
  554. else
  555. clear_bit(
  556. CIFS_INODE_DOWNGRADE_OPLOCK_TO_L2,
  557. &cinode->flags);
  558. spin_unlock(&cfile->file_info_lock);
  559. queue_work(cifsiod_wq, &cfile->oplock_break);
  560. spin_unlock(&tcon->open_file_lock);
  561. spin_unlock(&cifs_tcp_ses_lock);
  562. return true;
  563. }
  564. spin_unlock(&tcon->open_file_lock);
  565. spin_unlock(&cifs_tcp_ses_lock);
  566. cifs_dbg(FYI, "No matching file for oplock break\n");
  567. return true;
  568. }
  569. }
  570. spin_unlock(&cifs_tcp_ses_lock);
  571. cifs_dbg(FYI, "Can not process oplock break for non-existent connection\n");
  572. return false;
  573. }
  574. void
  575. smb2_cancelled_close_fid(struct work_struct *work)
  576. {
  577. struct close_cancelled_open *cancelled = container_of(work,
  578. struct close_cancelled_open, work);
  579. cifs_dbg(VFS, "Close unmatched open\n");
  580. SMB2_close(0, cancelled->tcon, cancelled->fid.persistent_fid,
  581. cancelled->fid.volatile_fid);
  582. cifs_put_tcon(cancelled->tcon);
  583. kfree(cancelled);
  584. }
  585. int
  586. smb2_handle_cancelled_mid(char *buffer, struct TCP_Server_Info *server)
  587. {
  588. struct smb2_hdr *hdr = (struct smb2_hdr *)buffer;
  589. struct smb2_create_rsp *rsp = (struct smb2_create_rsp *)buffer;
  590. struct cifs_tcon *tcon;
  591. struct close_cancelled_open *cancelled;
  592. if (hdr->Command != SMB2_CREATE || hdr->Status != STATUS_SUCCESS)
  593. return 0;
  594. cancelled = kzalloc(sizeof(*cancelled), GFP_KERNEL);
  595. if (!cancelled)
  596. return -ENOMEM;
  597. tcon = smb2_find_smb_tcon(server, hdr->SessionId, hdr->TreeId);
  598. if (!tcon) {
  599. kfree(cancelled);
  600. return -ENOENT;
  601. }
  602. cancelled->fid.persistent_fid = rsp->PersistentFileId;
  603. cancelled->fid.volatile_fid = rsp->VolatileFileId;
  604. cancelled->tcon = tcon;
  605. INIT_WORK(&cancelled->work, smb2_cancelled_close_fid);
  606. queue_work(cifsiod_wq, &cancelled->work);
  607. return 0;
  608. }