smb2misc.c 20 KB

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