file.c 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416
  1. /*
  2. * Copyright (C) 2007 Oracle. All rights reserved.
  3. *
  4. * This program is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU General Public
  6. * License v2 as published by the Free Software Foundation.
  7. *
  8. * This program is distributed in the hope that it will be useful,
  9. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11. * General Public License for more details.
  12. *
  13. * You should have received a copy of the GNU General Public
  14. * License along with this program; if not, write to the
  15. * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  16. * Boston, MA 021110-1307, USA.
  17. */
  18. #include <linux/fs.h>
  19. #include <linux/pagemap.h>
  20. #include <linux/highmem.h>
  21. #include <linux/time.h>
  22. #include <linux/init.h>
  23. #include <linux/string.h>
  24. #include <linux/backing-dev.h>
  25. #include <linux/mpage.h>
  26. #include <linux/falloc.h>
  27. #include <linux/swap.h>
  28. #include <linux/writeback.h>
  29. #include <linux/statfs.h>
  30. #include <linux/compat.h>
  31. #include <linux/slab.h>
  32. #include "ctree.h"
  33. #include "disk-io.h"
  34. #include "transaction.h"
  35. #include "btrfs_inode.h"
  36. #include "ioctl.h"
  37. #include "print-tree.h"
  38. #include "tree-log.h"
  39. #include "locking.h"
  40. #include "compat.h"
  41. #include "volumes.h"
  42. static struct kmem_cache *btrfs_inode_defrag_cachep;
  43. /*
  44. * when auto defrag is enabled we
  45. * queue up these defrag structs to remember which
  46. * inodes need defragging passes
  47. */
  48. struct inode_defrag {
  49. struct rb_node rb_node;
  50. /* objectid */
  51. u64 ino;
  52. /*
  53. * transid where the defrag was added, we search for
  54. * extents newer than this
  55. */
  56. u64 transid;
  57. /* root objectid */
  58. u64 root;
  59. /* last offset we were able to defrag */
  60. u64 last_offset;
  61. /* if we've wrapped around back to zero once already */
  62. int cycled;
  63. };
  64. static int __compare_inode_defrag(struct inode_defrag *defrag1,
  65. struct inode_defrag *defrag2)
  66. {
  67. if (defrag1->root > defrag2->root)
  68. return 1;
  69. else if (defrag1->root < defrag2->root)
  70. return -1;
  71. else if (defrag1->ino > defrag2->ino)
  72. return 1;
  73. else if (defrag1->ino < defrag2->ino)
  74. return -1;
  75. else
  76. return 0;
  77. }
  78. /* pop a record for an inode into the defrag tree. The lock
  79. * must be held already
  80. *
  81. * If you're inserting a record for an older transid than an
  82. * existing record, the transid already in the tree is lowered
  83. *
  84. * If an existing record is found the defrag item you
  85. * pass in is freed
  86. */
  87. static int __btrfs_add_inode_defrag(struct inode *inode,
  88. struct inode_defrag *defrag)
  89. {
  90. struct btrfs_root *root = BTRFS_I(inode)->root;
  91. struct inode_defrag *entry;
  92. struct rb_node **p;
  93. struct rb_node *parent = NULL;
  94. int ret;
  95. p = &root->fs_info->defrag_inodes.rb_node;
  96. while (*p) {
  97. parent = *p;
  98. entry = rb_entry(parent, struct inode_defrag, rb_node);
  99. ret = __compare_inode_defrag(defrag, entry);
  100. if (ret < 0)
  101. p = &parent->rb_left;
  102. else if (ret > 0)
  103. p = &parent->rb_right;
  104. else {
  105. /* if we're reinserting an entry for
  106. * an old defrag run, make sure to
  107. * lower the transid of our existing record
  108. */
  109. if (defrag->transid < entry->transid)
  110. entry->transid = defrag->transid;
  111. if (defrag->last_offset > entry->last_offset)
  112. entry->last_offset = defrag->last_offset;
  113. return -EEXIST;
  114. }
  115. }
  116. set_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags);
  117. rb_link_node(&defrag->rb_node, parent, p);
  118. rb_insert_color(&defrag->rb_node, &root->fs_info->defrag_inodes);
  119. return 0;
  120. }
  121. static inline int __need_auto_defrag(struct btrfs_root *root)
  122. {
  123. if (!btrfs_test_opt(root, AUTO_DEFRAG))
  124. return 0;
  125. if (btrfs_fs_closing(root->fs_info))
  126. return 0;
  127. return 1;
  128. }
  129. /*
  130. * insert a defrag record for this inode if auto defrag is
  131. * enabled
  132. */
  133. int btrfs_add_inode_defrag(struct btrfs_trans_handle *trans,
  134. struct inode *inode)
  135. {
  136. struct btrfs_root *root = BTRFS_I(inode)->root;
  137. struct inode_defrag *defrag;
  138. u64 transid;
  139. int ret;
  140. if (!__need_auto_defrag(root))
  141. return 0;
  142. if (test_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags))
  143. return 0;
  144. if (trans)
  145. transid = trans->transid;
  146. else
  147. transid = BTRFS_I(inode)->root->last_trans;
  148. defrag = kmem_cache_zalloc(btrfs_inode_defrag_cachep, GFP_NOFS);
  149. if (!defrag)
  150. return -ENOMEM;
  151. defrag->ino = btrfs_ino(inode);
  152. defrag->transid = transid;
  153. defrag->root = root->root_key.objectid;
  154. spin_lock(&root->fs_info->defrag_inodes_lock);
  155. if (!test_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags)) {
  156. /*
  157. * If we set IN_DEFRAG flag and evict the inode from memory,
  158. * and then re-read this inode, this new inode doesn't have
  159. * IN_DEFRAG flag. At the case, we may find the existed defrag.
  160. */
  161. ret = __btrfs_add_inode_defrag(inode, defrag);
  162. if (ret)
  163. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  164. } else {
  165. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  166. }
  167. spin_unlock(&root->fs_info->defrag_inodes_lock);
  168. return 0;
  169. }
  170. /*
  171. * Requeue the defrag object. If there is a defrag object that points to
  172. * the same inode in the tree, we will merge them together (by
  173. * __btrfs_add_inode_defrag()) and free the one that we want to requeue.
  174. */
  175. void btrfs_requeue_inode_defrag(struct inode *inode,
  176. struct inode_defrag *defrag)
  177. {
  178. struct btrfs_root *root = BTRFS_I(inode)->root;
  179. int ret;
  180. if (!__need_auto_defrag(root))
  181. goto out;
  182. /*
  183. * Here we don't check the IN_DEFRAG flag, because we need merge
  184. * them together.
  185. */
  186. spin_lock(&root->fs_info->defrag_inodes_lock);
  187. ret = __btrfs_add_inode_defrag(inode, defrag);
  188. spin_unlock(&root->fs_info->defrag_inodes_lock);
  189. if (ret)
  190. goto out;
  191. return;
  192. out:
  193. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  194. }
  195. /*
  196. * pick the defragable inode that we want, if it doesn't exist, we will get
  197. * the next one.
  198. */
  199. static struct inode_defrag *
  200. btrfs_pick_defrag_inode(struct btrfs_fs_info *fs_info, u64 root, u64 ino)
  201. {
  202. struct inode_defrag *entry = NULL;
  203. struct inode_defrag tmp;
  204. struct rb_node *p;
  205. struct rb_node *parent = NULL;
  206. int ret;
  207. tmp.ino = ino;
  208. tmp.root = root;
  209. spin_lock(&fs_info->defrag_inodes_lock);
  210. p = fs_info->defrag_inodes.rb_node;
  211. while (p) {
  212. parent = p;
  213. entry = rb_entry(parent, struct inode_defrag, rb_node);
  214. ret = __compare_inode_defrag(&tmp, entry);
  215. if (ret < 0)
  216. p = parent->rb_left;
  217. else if (ret > 0)
  218. p = parent->rb_right;
  219. else
  220. goto out;
  221. }
  222. if (parent && __compare_inode_defrag(&tmp, entry) > 0) {
  223. parent = rb_next(parent);
  224. if (parent)
  225. entry = rb_entry(parent, struct inode_defrag, rb_node);
  226. else
  227. entry = NULL;
  228. }
  229. out:
  230. if (entry)
  231. rb_erase(parent, &fs_info->defrag_inodes);
  232. spin_unlock(&fs_info->defrag_inodes_lock);
  233. return entry;
  234. }
  235. void btrfs_cleanup_defrag_inodes(struct btrfs_fs_info *fs_info)
  236. {
  237. struct inode_defrag *defrag;
  238. struct rb_node *node;
  239. spin_lock(&fs_info->defrag_inodes_lock);
  240. node = rb_first(&fs_info->defrag_inodes);
  241. while (node) {
  242. rb_erase(node, &fs_info->defrag_inodes);
  243. defrag = rb_entry(node, struct inode_defrag, rb_node);
  244. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  245. if (need_resched()) {
  246. spin_unlock(&fs_info->defrag_inodes_lock);
  247. cond_resched();
  248. spin_lock(&fs_info->defrag_inodes_lock);
  249. }
  250. node = rb_first(&fs_info->defrag_inodes);
  251. }
  252. spin_unlock(&fs_info->defrag_inodes_lock);
  253. }
  254. #define BTRFS_DEFRAG_BATCH 1024
  255. static int __btrfs_run_defrag_inode(struct btrfs_fs_info *fs_info,
  256. struct inode_defrag *defrag)
  257. {
  258. struct btrfs_root *inode_root;
  259. struct inode *inode;
  260. struct btrfs_key key;
  261. struct btrfs_ioctl_defrag_range_args range;
  262. int num_defrag;
  263. /* get the inode */
  264. key.objectid = defrag->root;
  265. btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY);
  266. key.offset = (u64)-1;
  267. inode_root = btrfs_read_fs_root_no_name(fs_info, &key);
  268. if (IS_ERR(inode_root)) {
  269. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  270. return PTR_ERR(inode_root);
  271. }
  272. key.objectid = defrag->ino;
  273. btrfs_set_key_type(&key, BTRFS_INODE_ITEM_KEY);
  274. key.offset = 0;
  275. inode = btrfs_iget(fs_info->sb, &key, inode_root, NULL);
  276. if (IS_ERR(inode)) {
  277. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  278. return PTR_ERR(inode);
  279. }
  280. /* do a chunk of defrag */
  281. clear_bit(BTRFS_INODE_IN_DEFRAG, &BTRFS_I(inode)->runtime_flags);
  282. memset(&range, 0, sizeof(range));
  283. range.len = (u64)-1;
  284. range.start = defrag->last_offset;
  285. sb_start_write(fs_info->sb);
  286. num_defrag = btrfs_defrag_file(inode, NULL, &range, defrag->transid,
  287. BTRFS_DEFRAG_BATCH);
  288. sb_end_write(fs_info->sb);
  289. /*
  290. * if we filled the whole defrag batch, there
  291. * must be more work to do. Queue this defrag
  292. * again
  293. */
  294. if (num_defrag == BTRFS_DEFRAG_BATCH) {
  295. defrag->last_offset = range.start;
  296. btrfs_requeue_inode_defrag(inode, defrag);
  297. } else if (defrag->last_offset && !defrag->cycled) {
  298. /*
  299. * we didn't fill our defrag batch, but
  300. * we didn't start at zero. Make sure we loop
  301. * around to the start of the file.
  302. */
  303. defrag->last_offset = 0;
  304. defrag->cycled = 1;
  305. btrfs_requeue_inode_defrag(inode, defrag);
  306. } else {
  307. kmem_cache_free(btrfs_inode_defrag_cachep, defrag);
  308. }
  309. iput(inode);
  310. return 0;
  311. }
  312. /*
  313. * run through the list of inodes in the FS that need
  314. * defragging
  315. */
  316. int btrfs_run_defrag_inodes(struct btrfs_fs_info *fs_info)
  317. {
  318. struct inode_defrag *defrag;
  319. u64 first_ino = 0;
  320. u64 root_objectid = 0;
  321. atomic_inc(&fs_info->defrag_running);
  322. while(1) {
  323. if (!__need_auto_defrag(fs_info->tree_root))
  324. break;
  325. /* find an inode to defrag */
  326. defrag = btrfs_pick_defrag_inode(fs_info, root_objectid,
  327. first_ino);
  328. if (!defrag) {
  329. if (root_objectid || first_ino) {
  330. root_objectid = 0;
  331. first_ino = 0;
  332. continue;
  333. } else {
  334. break;
  335. }
  336. }
  337. first_ino = defrag->ino + 1;
  338. root_objectid = defrag->root;
  339. __btrfs_run_defrag_inode(fs_info, defrag);
  340. }
  341. atomic_dec(&fs_info->defrag_running);
  342. /*
  343. * during unmount, we use the transaction_wait queue to
  344. * wait for the defragger to stop
  345. */
  346. wake_up(&fs_info->transaction_wait);
  347. return 0;
  348. }
  349. /* simple helper to fault in pages and copy. This should go away
  350. * and be replaced with calls into generic code.
  351. */
  352. static noinline int btrfs_copy_from_user(loff_t pos, int num_pages,
  353. size_t write_bytes,
  354. struct page **prepared_pages,
  355. struct iov_iter *i)
  356. {
  357. size_t copied = 0;
  358. size_t total_copied = 0;
  359. int pg = 0;
  360. int offset = pos & (PAGE_CACHE_SIZE - 1);
  361. while (write_bytes > 0) {
  362. size_t count = min_t(size_t,
  363. PAGE_CACHE_SIZE - offset, write_bytes);
  364. struct page *page = prepared_pages[pg];
  365. /*
  366. * Copy data from userspace to the current page
  367. *
  368. * Disable pagefault to avoid recursive lock since
  369. * the pages are already locked
  370. */
  371. pagefault_disable();
  372. copied = iov_iter_copy_from_user_atomic(page, i, offset, count);
  373. pagefault_enable();
  374. /* Flush processor's dcache for this page */
  375. flush_dcache_page(page);
  376. /*
  377. * if we get a partial write, we can end up with
  378. * partially up to date pages. These add
  379. * a lot of complexity, so make sure they don't
  380. * happen by forcing this copy to be retried.
  381. *
  382. * The rest of the btrfs_file_write code will fall
  383. * back to page at a time copies after we return 0.
  384. */
  385. if (!PageUptodate(page) && copied < count)
  386. copied = 0;
  387. iov_iter_advance(i, copied);
  388. write_bytes -= copied;
  389. total_copied += copied;
  390. /* Return to btrfs_file_aio_write to fault page */
  391. if (unlikely(copied == 0))
  392. break;
  393. if (unlikely(copied < PAGE_CACHE_SIZE - offset)) {
  394. offset += copied;
  395. } else {
  396. pg++;
  397. offset = 0;
  398. }
  399. }
  400. return total_copied;
  401. }
  402. /*
  403. * unlocks pages after btrfs_file_write is done with them
  404. */
  405. void btrfs_drop_pages(struct page **pages, size_t num_pages)
  406. {
  407. size_t i;
  408. for (i = 0; i < num_pages; i++) {
  409. /* page checked is some magic around finding pages that
  410. * have been modified without going through btrfs_set_page_dirty
  411. * clear it here
  412. */
  413. ClearPageChecked(pages[i]);
  414. unlock_page(pages[i]);
  415. mark_page_accessed(pages[i]);
  416. page_cache_release(pages[i]);
  417. }
  418. }
  419. /*
  420. * after copy_from_user, pages need to be dirtied and we need to make
  421. * sure holes are created between the current EOF and the start of
  422. * any next extents (if required).
  423. *
  424. * this also makes the decision about creating an inline extent vs
  425. * doing real data extents, marking pages dirty and delalloc as required.
  426. */
  427. int btrfs_dirty_pages(struct btrfs_root *root, struct inode *inode,
  428. struct page **pages, size_t num_pages,
  429. loff_t pos, size_t write_bytes,
  430. struct extent_state **cached)
  431. {
  432. int err = 0;
  433. int i;
  434. u64 num_bytes;
  435. u64 start_pos;
  436. u64 end_of_last_block;
  437. u64 end_pos = pos + write_bytes;
  438. loff_t isize = i_size_read(inode);
  439. start_pos = pos & ~((u64)root->sectorsize - 1);
  440. num_bytes = (write_bytes + pos - start_pos +
  441. root->sectorsize - 1) & ~((u64)root->sectorsize - 1);
  442. end_of_last_block = start_pos + num_bytes - 1;
  443. err = btrfs_set_extent_delalloc(inode, start_pos, end_of_last_block,
  444. cached);
  445. if (err)
  446. return err;
  447. for (i = 0; i < num_pages; i++) {
  448. struct page *p = pages[i];
  449. SetPageUptodate(p);
  450. ClearPageChecked(p);
  451. set_page_dirty(p);
  452. }
  453. /*
  454. * we've only changed i_size in ram, and we haven't updated
  455. * the disk i_size. There is no need to log the inode
  456. * at this time.
  457. */
  458. if (end_pos > isize)
  459. i_size_write(inode, end_pos);
  460. return 0;
  461. }
  462. /*
  463. * this drops all the extents in the cache that intersect the range
  464. * [start, end]. Existing extents are split as required.
  465. */
  466. void btrfs_drop_extent_cache(struct inode *inode, u64 start, u64 end,
  467. int skip_pinned)
  468. {
  469. struct extent_map *em;
  470. struct extent_map *split = NULL;
  471. struct extent_map *split2 = NULL;
  472. struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
  473. u64 len = end - start + 1;
  474. u64 gen;
  475. int ret;
  476. int testend = 1;
  477. unsigned long flags;
  478. int compressed = 0;
  479. WARN_ON(end < start);
  480. if (end == (u64)-1) {
  481. len = (u64)-1;
  482. testend = 0;
  483. }
  484. while (1) {
  485. int no_splits = 0;
  486. if (!split)
  487. split = alloc_extent_map();
  488. if (!split2)
  489. split2 = alloc_extent_map();
  490. if (!split || !split2)
  491. no_splits = 1;
  492. write_lock(&em_tree->lock);
  493. em = lookup_extent_mapping(em_tree, start, len);
  494. if (!em) {
  495. write_unlock(&em_tree->lock);
  496. break;
  497. }
  498. flags = em->flags;
  499. gen = em->generation;
  500. if (skip_pinned && test_bit(EXTENT_FLAG_PINNED, &em->flags)) {
  501. if (testend && em->start + em->len >= start + len) {
  502. free_extent_map(em);
  503. write_unlock(&em_tree->lock);
  504. break;
  505. }
  506. start = em->start + em->len;
  507. if (testend)
  508. len = start + len - (em->start + em->len);
  509. free_extent_map(em);
  510. write_unlock(&em_tree->lock);
  511. continue;
  512. }
  513. compressed = test_bit(EXTENT_FLAG_COMPRESSED, &em->flags);
  514. clear_bit(EXTENT_FLAG_PINNED, &em->flags);
  515. remove_extent_mapping(em_tree, em);
  516. if (no_splits)
  517. goto next;
  518. if (em->block_start < EXTENT_MAP_LAST_BYTE &&
  519. em->start < start) {
  520. split->start = em->start;
  521. split->len = start - em->start;
  522. split->orig_start = em->orig_start;
  523. split->block_start = em->block_start;
  524. if (compressed)
  525. split->block_len = em->block_len;
  526. else
  527. split->block_len = split->len;
  528. split->orig_block_len = max(split->block_len,
  529. em->orig_block_len);
  530. split->generation = gen;
  531. split->bdev = em->bdev;
  532. split->flags = flags;
  533. split->compress_type = em->compress_type;
  534. ret = add_extent_mapping(em_tree, split);
  535. BUG_ON(ret); /* Logic error */
  536. list_move(&split->list, &em_tree->modified_extents);
  537. free_extent_map(split);
  538. split = split2;
  539. split2 = NULL;
  540. }
  541. if (em->block_start < EXTENT_MAP_LAST_BYTE &&
  542. testend && em->start + em->len > start + len) {
  543. u64 diff = start + len - em->start;
  544. split->start = start + len;
  545. split->len = em->start + em->len - (start + len);
  546. split->bdev = em->bdev;
  547. split->flags = flags;
  548. split->compress_type = em->compress_type;
  549. split->generation = gen;
  550. split->orig_block_len = max(em->block_len,
  551. em->orig_block_len);
  552. if (compressed) {
  553. split->block_len = em->block_len;
  554. split->block_start = em->block_start;
  555. split->orig_start = em->orig_start;
  556. } else {
  557. split->block_len = split->len;
  558. split->block_start = em->block_start + diff;
  559. split->orig_start = em->orig_start;
  560. }
  561. ret = add_extent_mapping(em_tree, split);
  562. BUG_ON(ret); /* Logic error */
  563. list_move(&split->list, &em_tree->modified_extents);
  564. free_extent_map(split);
  565. split = NULL;
  566. }
  567. next:
  568. write_unlock(&em_tree->lock);
  569. /* once for us */
  570. free_extent_map(em);
  571. /* once for the tree*/
  572. free_extent_map(em);
  573. }
  574. if (split)
  575. free_extent_map(split);
  576. if (split2)
  577. free_extent_map(split2);
  578. }
  579. /*
  580. * this is very complex, but the basic idea is to drop all extents
  581. * in the range start - end. hint_block is filled in with a block number
  582. * that would be a good hint to the block allocator for this file.
  583. *
  584. * If an extent intersects the range but is not entirely inside the range
  585. * it is either truncated or split. Anything entirely inside the range
  586. * is deleted from the tree.
  587. */
  588. int __btrfs_drop_extents(struct btrfs_trans_handle *trans,
  589. struct btrfs_root *root, struct inode *inode,
  590. struct btrfs_path *path, u64 start, u64 end,
  591. u64 *drop_end, int drop_cache)
  592. {
  593. struct extent_buffer *leaf;
  594. struct btrfs_file_extent_item *fi;
  595. struct btrfs_key key;
  596. struct btrfs_key new_key;
  597. u64 ino = btrfs_ino(inode);
  598. u64 search_start = start;
  599. u64 disk_bytenr = 0;
  600. u64 num_bytes = 0;
  601. u64 extent_offset = 0;
  602. u64 extent_end = 0;
  603. int del_nr = 0;
  604. int del_slot = 0;
  605. int extent_type;
  606. int recow;
  607. int ret;
  608. int modify_tree = -1;
  609. int update_refs = (root->ref_cows || root == root->fs_info->tree_root);
  610. int found = 0;
  611. if (drop_cache)
  612. btrfs_drop_extent_cache(inode, start, end - 1, 0);
  613. if (start >= BTRFS_I(inode)->disk_i_size)
  614. modify_tree = 0;
  615. while (1) {
  616. recow = 0;
  617. ret = btrfs_lookup_file_extent(trans, root, path, ino,
  618. search_start, modify_tree);
  619. if (ret < 0)
  620. break;
  621. if (ret > 0 && path->slots[0] > 0 && search_start == start) {
  622. leaf = path->nodes[0];
  623. btrfs_item_key_to_cpu(leaf, &key, path->slots[0] - 1);
  624. if (key.objectid == ino &&
  625. key.type == BTRFS_EXTENT_DATA_KEY)
  626. path->slots[0]--;
  627. }
  628. ret = 0;
  629. next_slot:
  630. leaf = path->nodes[0];
  631. if (path->slots[0] >= btrfs_header_nritems(leaf)) {
  632. BUG_ON(del_nr > 0);
  633. ret = btrfs_next_leaf(root, path);
  634. if (ret < 0)
  635. break;
  636. if (ret > 0) {
  637. ret = 0;
  638. break;
  639. }
  640. leaf = path->nodes[0];
  641. recow = 1;
  642. }
  643. btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
  644. if (key.objectid > ino ||
  645. key.type > BTRFS_EXTENT_DATA_KEY || key.offset >= end)
  646. break;
  647. fi = btrfs_item_ptr(leaf, path->slots[0],
  648. struct btrfs_file_extent_item);
  649. extent_type = btrfs_file_extent_type(leaf, fi);
  650. if (extent_type == BTRFS_FILE_EXTENT_REG ||
  651. extent_type == BTRFS_FILE_EXTENT_PREALLOC) {
  652. disk_bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
  653. num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
  654. extent_offset = btrfs_file_extent_offset(leaf, fi);
  655. extent_end = key.offset +
  656. btrfs_file_extent_num_bytes(leaf, fi);
  657. } else if (extent_type == BTRFS_FILE_EXTENT_INLINE) {
  658. extent_end = key.offset +
  659. btrfs_file_extent_inline_len(leaf, fi);
  660. } else {
  661. WARN_ON(1);
  662. extent_end = search_start;
  663. }
  664. if (extent_end <= search_start) {
  665. path->slots[0]++;
  666. goto next_slot;
  667. }
  668. found = 1;
  669. search_start = max(key.offset, start);
  670. if (recow || !modify_tree) {
  671. modify_tree = -1;
  672. btrfs_release_path(path);
  673. continue;
  674. }
  675. /*
  676. * | - range to drop - |
  677. * | -------- extent -------- |
  678. */
  679. if (start > key.offset && end < extent_end) {
  680. BUG_ON(del_nr > 0);
  681. BUG_ON(extent_type == BTRFS_FILE_EXTENT_INLINE);
  682. memcpy(&new_key, &key, sizeof(new_key));
  683. new_key.offset = start;
  684. ret = btrfs_duplicate_item(trans, root, path,
  685. &new_key);
  686. if (ret == -EAGAIN) {
  687. btrfs_release_path(path);
  688. continue;
  689. }
  690. if (ret < 0)
  691. break;
  692. leaf = path->nodes[0];
  693. fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
  694. struct btrfs_file_extent_item);
  695. btrfs_set_file_extent_num_bytes(leaf, fi,
  696. start - key.offset);
  697. fi = btrfs_item_ptr(leaf, path->slots[0],
  698. struct btrfs_file_extent_item);
  699. extent_offset += start - key.offset;
  700. btrfs_set_file_extent_offset(leaf, fi, extent_offset);
  701. btrfs_set_file_extent_num_bytes(leaf, fi,
  702. extent_end - start);
  703. btrfs_mark_buffer_dirty(leaf);
  704. if (update_refs && disk_bytenr > 0) {
  705. ret = btrfs_inc_extent_ref(trans, root,
  706. disk_bytenr, num_bytes, 0,
  707. root->root_key.objectid,
  708. new_key.objectid,
  709. start - extent_offset, 0);
  710. BUG_ON(ret); /* -ENOMEM */
  711. }
  712. key.offset = start;
  713. }
  714. /*
  715. * | ---- range to drop ----- |
  716. * | -------- extent -------- |
  717. */
  718. if (start <= key.offset && end < extent_end) {
  719. BUG_ON(extent_type == BTRFS_FILE_EXTENT_INLINE);
  720. memcpy(&new_key, &key, sizeof(new_key));
  721. new_key.offset = end;
  722. btrfs_set_item_key_safe(trans, root, path, &new_key);
  723. extent_offset += end - key.offset;
  724. btrfs_set_file_extent_offset(leaf, fi, extent_offset);
  725. btrfs_set_file_extent_num_bytes(leaf, fi,
  726. extent_end - end);
  727. btrfs_mark_buffer_dirty(leaf);
  728. if (update_refs && disk_bytenr > 0)
  729. inode_sub_bytes(inode, end - key.offset);
  730. break;
  731. }
  732. search_start = extent_end;
  733. /*
  734. * | ---- range to drop ----- |
  735. * | -------- extent -------- |
  736. */
  737. if (start > key.offset && end >= extent_end) {
  738. BUG_ON(del_nr > 0);
  739. BUG_ON(extent_type == BTRFS_FILE_EXTENT_INLINE);
  740. btrfs_set_file_extent_num_bytes(leaf, fi,
  741. start - key.offset);
  742. btrfs_mark_buffer_dirty(leaf);
  743. if (update_refs && disk_bytenr > 0)
  744. inode_sub_bytes(inode, extent_end - start);
  745. if (end == extent_end)
  746. break;
  747. path->slots[0]++;
  748. goto next_slot;
  749. }
  750. /*
  751. * | ---- range to drop ----- |
  752. * | ------ extent ------ |
  753. */
  754. if (start <= key.offset && end >= extent_end) {
  755. if (del_nr == 0) {
  756. del_slot = path->slots[0];
  757. del_nr = 1;
  758. } else {
  759. BUG_ON(del_slot + del_nr != path->slots[0]);
  760. del_nr++;
  761. }
  762. if (update_refs &&
  763. extent_type == BTRFS_FILE_EXTENT_INLINE) {
  764. inode_sub_bytes(inode,
  765. extent_end - key.offset);
  766. extent_end = ALIGN(extent_end,
  767. root->sectorsize);
  768. } else if (update_refs && disk_bytenr > 0) {
  769. ret = btrfs_free_extent(trans, root,
  770. disk_bytenr, num_bytes, 0,
  771. root->root_key.objectid,
  772. key.objectid, key.offset -
  773. extent_offset, 0);
  774. BUG_ON(ret); /* -ENOMEM */
  775. inode_sub_bytes(inode,
  776. extent_end - key.offset);
  777. }
  778. if (end == extent_end)
  779. break;
  780. if (path->slots[0] + 1 < btrfs_header_nritems(leaf)) {
  781. path->slots[0]++;
  782. goto next_slot;
  783. }
  784. ret = btrfs_del_items(trans, root, path, del_slot,
  785. del_nr);
  786. if (ret) {
  787. btrfs_abort_transaction(trans, root, ret);
  788. break;
  789. }
  790. del_nr = 0;
  791. del_slot = 0;
  792. btrfs_release_path(path);
  793. continue;
  794. }
  795. BUG_ON(1);
  796. }
  797. if (!ret && del_nr > 0) {
  798. ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
  799. if (ret)
  800. btrfs_abort_transaction(trans, root, ret);
  801. }
  802. if (drop_end)
  803. *drop_end = found ? min(end, extent_end) : end;
  804. btrfs_release_path(path);
  805. return ret;
  806. }
  807. int btrfs_drop_extents(struct btrfs_trans_handle *trans,
  808. struct btrfs_root *root, struct inode *inode, u64 start,
  809. u64 end, int drop_cache)
  810. {
  811. struct btrfs_path *path;
  812. int ret;
  813. path = btrfs_alloc_path();
  814. if (!path)
  815. return -ENOMEM;
  816. ret = __btrfs_drop_extents(trans, root, inode, path, start, end, NULL,
  817. drop_cache);
  818. btrfs_free_path(path);
  819. return ret;
  820. }
  821. static int extent_mergeable(struct extent_buffer *leaf, int slot,
  822. u64 objectid, u64 bytenr, u64 orig_offset,
  823. u64 *start, u64 *end)
  824. {
  825. struct btrfs_file_extent_item *fi;
  826. struct btrfs_key key;
  827. u64 extent_end;
  828. if (slot < 0 || slot >= btrfs_header_nritems(leaf))
  829. return 0;
  830. btrfs_item_key_to_cpu(leaf, &key, slot);
  831. if (key.objectid != objectid || key.type != BTRFS_EXTENT_DATA_KEY)
  832. return 0;
  833. fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
  834. if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG ||
  835. btrfs_file_extent_disk_bytenr(leaf, fi) != bytenr ||
  836. btrfs_file_extent_offset(leaf, fi) != key.offset - orig_offset ||
  837. btrfs_file_extent_compression(leaf, fi) ||
  838. btrfs_file_extent_encryption(leaf, fi) ||
  839. btrfs_file_extent_other_encoding(leaf, fi))
  840. return 0;
  841. extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
  842. if ((*start && *start != key.offset) || (*end && *end != extent_end))
  843. return 0;
  844. *start = key.offset;
  845. *end = extent_end;
  846. return 1;
  847. }
  848. /*
  849. * Mark extent in the range start - end as written.
  850. *
  851. * This changes extent type from 'pre-allocated' to 'regular'. If only
  852. * part of extent is marked as written, the extent will be split into
  853. * two or three.
  854. */
  855. int btrfs_mark_extent_written(struct btrfs_trans_handle *trans,
  856. struct inode *inode, u64 start, u64 end)
  857. {
  858. struct btrfs_root *root = BTRFS_I(inode)->root;
  859. struct extent_buffer *leaf;
  860. struct btrfs_path *path;
  861. struct btrfs_file_extent_item *fi;
  862. struct btrfs_key key;
  863. struct btrfs_key new_key;
  864. u64 bytenr;
  865. u64 num_bytes;
  866. u64 extent_end;
  867. u64 orig_offset;
  868. u64 other_start;
  869. u64 other_end;
  870. u64 split;
  871. int del_nr = 0;
  872. int del_slot = 0;
  873. int recow;
  874. int ret;
  875. u64 ino = btrfs_ino(inode);
  876. path = btrfs_alloc_path();
  877. if (!path)
  878. return -ENOMEM;
  879. again:
  880. recow = 0;
  881. split = start;
  882. key.objectid = ino;
  883. key.type = BTRFS_EXTENT_DATA_KEY;
  884. key.offset = split;
  885. ret = btrfs_search_slot(trans, root, &key, path, -1, 1);
  886. if (ret < 0)
  887. goto out;
  888. if (ret > 0 && path->slots[0] > 0)
  889. path->slots[0]--;
  890. leaf = path->nodes[0];
  891. btrfs_item_key_to_cpu(leaf, &key, path->slots[0]);
  892. BUG_ON(key.objectid != ino || key.type != BTRFS_EXTENT_DATA_KEY);
  893. fi = btrfs_item_ptr(leaf, path->slots[0],
  894. struct btrfs_file_extent_item);
  895. BUG_ON(btrfs_file_extent_type(leaf, fi) !=
  896. BTRFS_FILE_EXTENT_PREALLOC);
  897. extent_end = key.offset + btrfs_file_extent_num_bytes(leaf, fi);
  898. BUG_ON(key.offset > start || extent_end < end);
  899. bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
  900. num_bytes = btrfs_file_extent_disk_num_bytes(leaf, fi);
  901. orig_offset = key.offset - btrfs_file_extent_offset(leaf, fi);
  902. memcpy(&new_key, &key, sizeof(new_key));
  903. if (start == key.offset && end < extent_end) {
  904. other_start = 0;
  905. other_end = start;
  906. if (extent_mergeable(leaf, path->slots[0] - 1,
  907. ino, bytenr, orig_offset,
  908. &other_start, &other_end)) {
  909. new_key.offset = end;
  910. btrfs_set_item_key_safe(trans, root, path, &new_key);
  911. fi = btrfs_item_ptr(leaf, path->slots[0],
  912. struct btrfs_file_extent_item);
  913. btrfs_set_file_extent_generation(leaf, fi,
  914. trans->transid);
  915. btrfs_set_file_extent_num_bytes(leaf, fi,
  916. extent_end - end);
  917. btrfs_set_file_extent_offset(leaf, fi,
  918. end - orig_offset);
  919. fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
  920. struct btrfs_file_extent_item);
  921. btrfs_set_file_extent_generation(leaf, fi,
  922. trans->transid);
  923. btrfs_set_file_extent_num_bytes(leaf, fi,
  924. end - other_start);
  925. btrfs_mark_buffer_dirty(leaf);
  926. goto out;
  927. }
  928. }
  929. if (start > key.offset && end == extent_end) {
  930. other_start = end;
  931. other_end = 0;
  932. if (extent_mergeable(leaf, path->slots[0] + 1,
  933. ino, bytenr, orig_offset,
  934. &other_start, &other_end)) {
  935. fi = btrfs_item_ptr(leaf, path->slots[0],
  936. struct btrfs_file_extent_item);
  937. btrfs_set_file_extent_num_bytes(leaf, fi,
  938. start - key.offset);
  939. btrfs_set_file_extent_generation(leaf, fi,
  940. trans->transid);
  941. path->slots[0]++;
  942. new_key.offset = start;
  943. btrfs_set_item_key_safe(trans, root, path, &new_key);
  944. fi = btrfs_item_ptr(leaf, path->slots[0],
  945. struct btrfs_file_extent_item);
  946. btrfs_set_file_extent_generation(leaf, fi,
  947. trans->transid);
  948. btrfs_set_file_extent_num_bytes(leaf, fi,
  949. other_end - start);
  950. btrfs_set_file_extent_offset(leaf, fi,
  951. start - orig_offset);
  952. btrfs_mark_buffer_dirty(leaf);
  953. goto out;
  954. }
  955. }
  956. while (start > key.offset || end < extent_end) {
  957. if (key.offset == start)
  958. split = end;
  959. new_key.offset = split;
  960. ret = btrfs_duplicate_item(trans, root, path, &new_key);
  961. if (ret == -EAGAIN) {
  962. btrfs_release_path(path);
  963. goto again;
  964. }
  965. if (ret < 0) {
  966. btrfs_abort_transaction(trans, root, ret);
  967. goto out;
  968. }
  969. leaf = path->nodes[0];
  970. fi = btrfs_item_ptr(leaf, path->slots[0] - 1,
  971. struct btrfs_file_extent_item);
  972. btrfs_set_file_extent_generation(leaf, fi, trans->transid);
  973. btrfs_set_file_extent_num_bytes(leaf, fi,
  974. split - key.offset);
  975. fi = btrfs_item_ptr(leaf, path->slots[0],
  976. struct btrfs_file_extent_item);
  977. btrfs_set_file_extent_generation(leaf, fi, trans->transid);
  978. btrfs_set_file_extent_offset(leaf, fi, split - orig_offset);
  979. btrfs_set_file_extent_num_bytes(leaf, fi,
  980. extent_end - split);
  981. btrfs_mark_buffer_dirty(leaf);
  982. ret = btrfs_inc_extent_ref(trans, root, bytenr, num_bytes, 0,
  983. root->root_key.objectid,
  984. ino, orig_offset, 0);
  985. BUG_ON(ret); /* -ENOMEM */
  986. if (split == start) {
  987. key.offset = start;
  988. } else {
  989. BUG_ON(start != key.offset);
  990. path->slots[0]--;
  991. extent_end = end;
  992. }
  993. recow = 1;
  994. }
  995. other_start = end;
  996. other_end = 0;
  997. if (extent_mergeable(leaf, path->slots[0] + 1,
  998. ino, bytenr, orig_offset,
  999. &other_start, &other_end)) {
  1000. if (recow) {
  1001. btrfs_release_path(path);
  1002. goto again;
  1003. }
  1004. extent_end = other_end;
  1005. del_slot = path->slots[0] + 1;
  1006. del_nr++;
  1007. ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
  1008. 0, root->root_key.objectid,
  1009. ino, orig_offset, 0);
  1010. BUG_ON(ret); /* -ENOMEM */
  1011. }
  1012. other_start = 0;
  1013. other_end = start;
  1014. if (extent_mergeable(leaf, path->slots[0] - 1,
  1015. ino, bytenr, orig_offset,
  1016. &other_start, &other_end)) {
  1017. if (recow) {
  1018. btrfs_release_path(path);
  1019. goto again;
  1020. }
  1021. key.offset = other_start;
  1022. del_slot = path->slots[0];
  1023. del_nr++;
  1024. ret = btrfs_free_extent(trans, root, bytenr, num_bytes,
  1025. 0, root->root_key.objectid,
  1026. ino, orig_offset, 0);
  1027. BUG_ON(ret); /* -ENOMEM */
  1028. }
  1029. if (del_nr == 0) {
  1030. fi = btrfs_item_ptr(leaf, path->slots[0],
  1031. struct btrfs_file_extent_item);
  1032. btrfs_set_file_extent_type(leaf, fi,
  1033. BTRFS_FILE_EXTENT_REG);
  1034. btrfs_set_file_extent_generation(leaf, fi, trans->transid);
  1035. btrfs_mark_buffer_dirty(leaf);
  1036. } else {
  1037. fi = btrfs_item_ptr(leaf, del_slot - 1,
  1038. struct btrfs_file_extent_item);
  1039. btrfs_set_file_extent_type(leaf, fi,
  1040. BTRFS_FILE_EXTENT_REG);
  1041. btrfs_set_file_extent_generation(leaf, fi, trans->transid);
  1042. btrfs_set_file_extent_num_bytes(leaf, fi,
  1043. extent_end - key.offset);
  1044. btrfs_mark_buffer_dirty(leaf);
  1045. ret = btrfs_del_items(trans, root, path, del_slot, del_nr);
  1046. if (ret < 0) {
  1047. btrfs_abort_transaction(trans, root, ret);
  1048. goto out;
  1049. }
  1050. }
  1051. out:
  1052. btrfs_free_path(path);
  1053. return 0;
  1054. }
  1055. /*
  1056. * on error we return an unlocked page and the error value
  1057. * on success we return a locked page and 0
  1058. */
  1059. static int prepare_uptodate_page(struct page *page, u64 pos,
  1060. bool force_uptodate)
  1061. {
  1062. int ret = 0;
  1063. if (((pos & (PAGE_CACHE_SIZE - 1)) || force_uptodate) &&
  1064. !PageUptodate(page)) {
  1065. ret = btrfs_readpage(NULL, page);
  1066. if (ret)
  1067. return ret;
  1068. lock_page(page);
  1069. if (!PageUptodate(page)) {
  1070. unlock_page(page);
  1071. return -EIO;
  1072. }
  1073. }
  1074. return 0;
  1075. }
  1076. /*
  1077. * this gets pages into the page cache and locks them down, it also properly
  1078. * waits for data=ordered extents to finish before allowing the pages to be
  1079. * modified.
  1080. */
  1081. static noinline int prepare_pages(struct btrfs_root *root, struct file *file,
  1082. struct page **pages, size_t num_pages,
  1083. loff_t pos, unsigned long first_index,
  1084. size_t write_bytes, bool force_uptodate)
  1085. {
  1086. struct extent_state *cached_state = NULL;
  1087. int i;
  1088. unsigned long index = pos >> PAGE_CACHE_SHIFT;
  1089. struct inode *inode = fdentry(file)->d_inode;
  1090. gfp_t mask = btrfs_alloc_write_mask(inode->i_mapping);
  1091. int err = 0;
  1092. int faili = 0;
  1093. u64 start_pos;
  1094. u64 last_pos;
  1095. start_pos = pos & ~((u64)root->sectorsize - 1);
  1096. last_pos = ((u64)index + num_pages) << PAGE_CACHE_SHIFT;
  1097. again:
  1098. for (i = 0; i < num_pages; i++) {
  1099. pages[i] = find_or_create_page(inode->i_mapping, index + i,
  1100. mask | __GFP_WRITE);
  1101. if (!pages[i]) {
  1102. faili = i - 1;
  1103. err = -ENOMEM;
  1104. goto fail;
  1105. }
  1106. if (i == 0)
  1107. err = prepare_uptodate_page(pages[i], pos,
  1108. force_uptodate);
  1109. if (i == num_pages - 1)
  1110. err = prepare_uptodate_page(pages[i],
  1111. pos + write_bytes, false);
  1112. if (err) {
  1113. page_cache_release(pages[i]);
  1114. faili = i - 1;
  1115. goto fail;
  1116. }
  1117. wait_on_page_writeback(pages[i]);
  1118. }
  1119. err = 0;
  1120. if (start_pos < inode->i_size) {
  1121. struct btrfs_ordered_extent *ordered;
  1122. lock_extent_bits(&BTRFS_I(inode)->io_tree,
  1123. start_pos, last_pos - 1, 0, &cached_state);
  1124. ordered = btrfs_lookup_first_ordered_extent(inode,
  1125. last_pos - 1);
  1126. if (ordered &&
  1127. ordered->file_offset + ordered->len > start_pos &&
  1128. ordered->file_offset < last_pos) {
  1129. btrfs_put_ordered_extent(ordered);
  1130. unlock_extent_cached(&BTRFS_I(inode)->io_tree,
  1131. start_pos, last_pos - 1,
  1132. &cached_state, GFP_NOFS);
  1133. for (i = 0; i < num_pages; i++) {
  1134. unlock_page(pages[i]);
  1135. page_cache_release(pages[i]);
  1136. }
  1137. btrfs_wait_ordered_range(inode, start_pos,
  1138. last_pos - start_pos);
  1139. goto again;
  1140. }
  1141. if (ordered)
  1142. btrfs_put_ordered_extent(ordered);
  1143. clear_extent_bit(&BTRFS_I(inode)->io_tree, start_pos,
  1144. last_pos - 1, EXTENT_DIRTY | EXTENT_DELALLOC |
  1145. EXTENT_DO_ACCOUNTING | EXTENT_DEFRAG,
  1146. 0, 0, &cached_state, GFP_NOFS);
  1147. unlock_extent_cached(&BTRFS_I(inode)->io_tree,
  1148. start_pos, last_pos - 1, &cached_state,
  1149. GFP_NOFS);
  1150. }
  1151. for (i = 0; i < num_pages; i++) {
  1152. if (clear_page_dirty_for_io(pages[i]))
  1153. account_page_redirty(pages[i]);
  1154. set_page_extent_mapped(pages[i]);
  1155. WARN_ON(!PageLocked(pages[i]));
  1156. }
  1157. return 0;
  1158. fail:
  1159. while (faili >= 0) {
  1160. unlock_page(pages[faili]);
  1161. page_cache_release(pages[faili]);
  1162. faili--;
  1163. }
  1164. return err;
  1165. }
  1166. static noinline ssize_t __btrfs_buffered_write(struct file *file,
  1167. struct iov_iter *i,
  1168. loff_t pos)
  1169. {
  1170. struct inode *inode = fdentry(file)->d_inode;
  1171. struct btrfs_root *root = BTRFS_I(inode)->root;
  1172. struct page **pages = NULL;
  1173. unsigned long first_index;
  1174. size_t num_written = 0;
  1175. int nrptrs;
  1176. int ret = 0;
  1177. bool force_page_uptodate = false;
  1178. nrptrs = min((iov_iter_count(i) + PAGE_CACHE_SIZE - 1) /
  1179. PAGE_CACHE_SIZE, PAGE_CACHE_SIZE /
  1180. (sizeof(struct page *)));
  1181. nrptrs = min(nrptrs, current->nr_dirtied_pause - current->nr_dirtied);
  1182. nrptrs = max(nrptrs, 8);
  1183. pages = kmalloc(nrptrs * sizeof(struct page *), GFP_KERNEL);
  1184. if (!pages)
  1185. return -ENOMEM;
  1186. first_index = pos >> PAGE_CACHE_SHIFT;
  1187. while (iov_iter_count(i) > 0) {
  1188. size_t offset = pos & (PAGE_CACHE_SIZE - 1);
  1189. size_t write_bytes = min(iov_iter_count(i),
  1190. nrptrs * (size_t)PAGE_CACHE_SIZE -
  1191. offset);
  1192. size_t num_pages = (write_bytes + offset +
  1193. PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
  1194. size_t dirty_pages;
  1195. size_t copied;
  1196. WARN_ON(num_pages > nrptrs);
  1197. /*
  1198. * Fault pages before locking them in prepare_pages
  1199. * to avoid recursive lock
  1200. */
  1201. if (unlikely(iov_iter_fault_in_readable(i, write_bytes))) {
  1202. ret = -EFAULT;
  1203. break;
  1204. }
  1205. ret = btrfs_delalloc_reserve_space(inode,
  1206. num_pages << PAGE_CACHE_SHIFT);
  1207. if (ret)
  1208. break;
  1209. /*
  1210. * This is going to setup the pages array with the number of
  1211. * pages we want, so we don't really need to worry about the
  1212. * contents of pages from loop to loop
  1213. */
  1214. ret = prepare_pages(root, file, pages, num_pages,
  1215. pos, first_index, write_bytes,
  1216. force_page_uptodate);
  1217. if (ret) {
  1218. btrfs_delalloc_release_space(inode,
  1219. num_pages << PAGE_CACHE_SHIFT);
  1220. break;
  1221. }
  1222. copied = btrfs_copy_from_user(pos, num_pages,
  1223. write_bytes, pages, i);
  1224. /*
  1225. * if we have trouble faulting in the pages, fall
  1226. * back to one page at a time
  1227. */
  1228. if (copied < write_bytes)
  1229. nrptrs = 1;
  1230. if (copied == 0) {
  1231. force_page_uptodate = true;
  1232. dirty_pages = 0;
  1233. } else {
  1234. force_page_uptodate = false;
  1235. dirty_pages = (copied + offset +
  1236. PAGE_CACHE_SIZE - 1) >>
  1237. PAGE_CACHE_SHIFT;
  1238. }
  1239. /*
  1240. * If we had a short copy we need to release the excess delaloc
  1241. * bytes we reserved. We need to increment outstanding_extents
  1242. * because btrfs_delalloc_release_space will decrement it, but
  1243. * we still have an outstanding extent for the chunk we actually
  1244. * managed to copy.
  1245. */
  1246. if (num_pages > dirty_pages) {
  1247. if (copied > 0) {
  1248. spin_lock(&BTRFS_I(inode)->lock);
  1249. BTRFS_I(inode)->outstanding_extents++;
  1250. spin_unlock(&BTRFS_I(inode)->lock);
  1251. }
  1252. btrfs_delalloc_release_space(inode,
  1253. (num_pages - dirty_pages) <<
  1254. PAGE_CACHE_SHIFT);
  1255. }
  1256. if (copied > 0) {
  1257. ret = btrfs_dirty_pages(root, inode, pages,
  1258. dirty_pages, pos, copied,
  1259. NULL);
  1260. if (ret) {
  1261. btrfs_delalloc_release_space(inode,
  1262. dirty_pages << PAGE_CACHE_SHIFT);
  1263. btrfs_drop_pages(pages, num_pages);
  1264. break;
  1265. }
  1266. }
  1267. btrfs_drop_pages(pages, num_pages);
  1268. cond_resched();
  1269. balance_dirty_pages_ratelimited(inode->i_mapping);
  1270. if (dirty_pages < (root->leafsize >> PAGE_CACHE_SHIFT) + 1)
  1271. btrfs_btree_balance_dirty(root);
  1272. pos += copied;
  1273. num_written += copied;
  1274. }
  1275. kfree(pages);
  1276. return num_written ? num_written : ret;
  1277. }
  1278. static ssize_t __btrfs_direct_write(struct kiocb *iocb,
  1279. const struct iovec *iov,
  1280. unsigned long nr_segs, loff_t pos,
  1281. loff_t *ppos, size_t count, size_t ocount)
  1282. {
  1283. struct file *file = iocb->ki_filp;
  1284. struct iov_iter i;
  1285. ssize_t written;
  1286. ssize_t written_buffered;
  1287. loff_t endbyte;
  1288. int err;
  1289. written = generic_file_direct_write(iocb, iov, &nr_segs, pos, ppos,
  1290. count, ocount);
  1291. if (written < 0 || written == count)
  1292. return written;
  1293. pos += written;
  1294. count -= written;
  1295. iov_iter_init(&i, iov, nr_segs, count, written);
  1296. written_buffered = __btrfs_buffered_write(file, &i, pos);
  1297. if (written_buffered < 0) {
  1298. err = written_buffered;
  1299. goto out;
  1300. }
  1301. endbyte = pos + written_buffered - 1;
  1302. err = filemap_write_and_wait_range(file->f_mapping, pos, endbyte);
  1303. if (err)
  1304. goto out;
  1305. written += written_buffered;
  1306. *ppos = pos + written_buffered;
  1307. invalidate_mapping_pages(file->f_mapping, pos >> PAGE_CACHE_SHIFT,
  1308. endbyte >> PAGE_CACHE_SHIFT);
  1309. out:
  1310. return written ? written : err;
  1311. }
  1312. static void update_time_for_write(struct inode *inode)
  1313. {
  1314. struct timespec now;
  1315. if (IS_NOCMTIME(inode))
  1316. return;
  1317. now = current_fs_time(inode->i_sb);
  1318. if (!timespec_equal(&inode->i_mtime, &now))
  1319. inode->i_mtime = now;
  1320. if (!timespec_equal(&inode->i_ctime, &now))
  1321. inode->i_ctime = now;
  1322. if (IS_I_VERSION(inode))
  1323. inode_inc_iversion(inode);
  1324. }
  1325. static ssize_t btrfs_file_aio_write(struct kiocb *iocb,
  1326. const struct iovec *iov,
  1327. unsigned long nr_segs, loff_t pos)
  1328. {
  1329. struct file *file = iocb->ki_filp;
  1330. struct inode *inode = fdentry(file)->d_inode;
  1331. struct btrfs_root *root = BTRFS_I(inode)->root;
  1332. loff_t *ppos = &iocb->ki_pos;
  1333. u64 start_pos;
  1334. ssize_t num_written = 0;
  1335. ssize_t err = 0;
  1336. size_t count, ocount;
  1337. bool sync = (file->f_flags & O_DSYNC) || IS_SYNC(file->f_mapping->host);
  1338. sb_start_write(inode->i_sb);
  1339. mutex_lock(&inode->i_mutex);
  1340. err = generic_segment_checks(iov, &nr_segs, &ocount, VERIFY_READ);
  1341. if (err) {
  1342. mutex_unlock(&inode->i_mutex);
  1343. goto out;
  1344. }
  1345. count = ocount;
  1346. current->backing_dev_info = inode->i_mapping->backing_dev_info;
  1347. err = generic_write_checks(file, &pos, &count, S_ISBLK(inode->i_mode));
  1348. if (err) {
  1349. mutex_unlock(&inode->i_mutex);
  1350. goto out;
  1351. }
  1352. if (count == 0) {
  1353. mutex_unlock(&inode->i_mutex);
  1354. goto out;
  1355. }
  1356. err = file_remove_suid(file);
  1357. if (err) {
  1358. mutex_unlock(&inode->i_mutex);
  1359. goto out;
  1360. }
  1361. /*
  1362. * If BTRFS flips readonly due to some impossible error
  1363. * (fs_info->fs_state now has BTRFS_SUPER_FLAG_ERROR),
  1364. * although we have opened a file as writable, we have
  1365. * to stop this write operation to ensure FS consistency.
  1366. */
  1367. if (root->fs_info->fs_state & BTRFS_SUPER_FLAG_ERROR) {
  1368. mutex_unlock(&inode->i_mutex);
  1369. err = -EROFS;
  1370. goto out;
  1371. }
  1372. /*
  1373. * We reserve space for updating the inode when we reserve space for the
  1374. * extent we are going to write, so we will enospc out there. We don't
  1375. * need to start yet another transaction to update the inode as we will
  1376. * update the inode when we finish writing whatever data we write.
  1377. */
  1378. update_time_for_write(inode);
  1379. start_pos = round_down(pos, root->sectorsize);
  1380. if (start_pos > i_size_read(inode)) {
  1381. err = btrfs_cont_expand(inode, i_size_read(inode), start_pos);
  1382. if (err) {
  1383. mutex_unlock(&inode->i_mutex);
  1384. goto out;
  1385. }
  1386. }
  1387. if (sync)
  1388. atomic_inc(&BTRFS_I(inode)->sync_writers);
  1389. if (unlikely(file->f_flags & O_DIRECT)) {
  1390. num_written = __btrfs_direct_write(iocb, iov, nr_segs,
  1391. pos, ppos, count, ocount);
  1392. } else {
  1393. struct iov_iter i;
  1394. iov_iter_init(&i, iov, nr_segs, count, num_written);
  1395. num_written = __btrfs_buffered_write(file, &i, pos);
  1396. if (num_written > 0)
  1397. *ppos = pos + num_written;
  1398. }
  1399. mutex_unlock(&inode->i_mutex);
  1400. /*
  1401. * we want to make sure fsync finds this change
  1402. * but we haven't joined a transaction running right now.
  1403. *
  1404. * Later on, someone is sure to update the inode and get the
  1405. * real transid recorded.
  1406. *
  1407. * We set last_trans now to the fs_info generation + 1,
  1408. * this will either be one more than the running transaction
  1409. * or the generation used for the next transaction if there isn't
  1410. * one running right now.
  1411. *
  1412. * We also have to set last_sub_trans to the current log transid,
  1413. * otherwise subsequent syncs to a file that's been synced in this
  1414. * transaction will appear to have already occured.
  1415. */
  1416. BTRFS_I(inode)->last_trans = root->fs_info->generation + 1;
  1417. BTRFS_I(inode)->last_sub_trans = root->log_transid;
  1418. if (num_written > 0 || num_written == -EIOCBQUEUED) {
  1419. err = generic_write_sync(file, pos, num_written);
  1420. if (err < 0 && num_written > 0)
  1421. num_written = err;
  1422. }
  1423. out:
  1424. if (sync)
  1425. atomic_dec(&BTRFS_I(inode)->sync_writers);
  1426. sb_end_write(inode->i_sb);
  1427. current->backing_dev_info = NULL;
  1428. return num_written ? num_written : err;
  1429. }
  1430. int btrfs_release_file(struct inode *inode, struct file *filp)
  1431. {
  1432. /*
  1433. * ordered_data_close is set by settattr when we are about to truncate
  1434. * a file from a non-zero size to a zero size. This tries to
  1435. * flush down new bytes that may have been written if the
  1436. * application were using truncate to replace a file in place.
  1437. */
  1438. if (test_and_clear_bit(BTRFS_INODE_ORDERED_DATA_CLOSE,
  1439. &BTRFS_I(inode)->runtime_flags)) {
  1440. btrfs_add_ordered_operation(NULL, BTRFS_I(inode)->root, inode);
  1441. if (inode->i_size > BTRFS_ORDERED_OPERATIONS_FLUSH_LIMIT)
  1442. filemap_flush(inode->i_mapping);
  1443. }
  1444. if (filp->private_data)
  1445. btrfs_ioctl_trans_end(filp);
  1446. return 0;
  1447. }
  1448. /*
  1449. * fsync call for both files and directories. This logs the inode into
  1450. * the tree log instead of forcing full commits whenever possible.
  1451. *
  1452. * It needs to call filemap_fdatawait so that all ordered extent updates are
  1453. * in the metadata btree are up to date for copying to the log.
  1454. *
  1455. * It drops the inode mutex before doing the tree log commit. This is an
  1456. * important optimization for directories because holding the mutex prevents
  1457. * new operations on the dir while we write to disk.
  1458. */
  1459. int btrfs_sync_file(struct file *file, loff_t start, loff_t end, int datasync)
  1460. {
  1461. struct dentry *dentry = file->f_path.dentry;
  1462. struct inode *inode = dentry->d_inode;
  1463. struct btrfs_root *root = BTRFS_I(inode)->root;
  1464. int ret = 0;
  1465. struct btrfs_trans_handle *trans;
  1466. trace_btrfs_sync_file(file, datasync);
  1467. /*
  1468. * We write the dirty pages in the range and wait until they complete
  1469. * out of the ->i_mutex. If so, we can flush the dirty pages by
  1470. * multi-task, and make the performance up.
  1471. */
  1472. atomic_inc(&BTRFS_I(inode)->sync_writers);
  1473. ret = filemap_write_and_wait_range(inode->i_mapping, start, end);
  1474. atomic_dec(&BTRFS_I(inode)->sync_writers);
  1475. if (ret)
  1476. return ret;
  1477. mutex_lock(&inode->i_mutex);
  1478. /*
  1479. * We flush the dirty pages again to avoid some dirty pages in the
  1480. * range being left.
  1481. */
  1482. atomic_inc(&root->log_batch);
  1483. btrfs_wait_ordered_range(inode, start, end - start + 1);
  1484. atomic_inc(&root->log_batch);
  1485. /*
  1486. * check the transaction that last modified this inode
  1487. * and see if its already been committed
  1488. */
  1489. if (!BTRFS_I(inode)->last_trans) {
  1490. mutex_unlock(&inode->i_mutex);
  1491. goto out;
  1492. }
  1493. /*
  1494. * if the last transaction that changed this file was before
  1495. * the current transaction, we can bail out now without any
  1496. * syncing
  1497. */
  1498. smp_mb();
  1499. if (btrfs_inode_in_log(inode, root->fs_info->generation) ||
  1500. BTRFS_I(inode)->last_trans <=
  1501. root->fs_info->last_trans_committed) {
  1502. BTRFS_I(inode)->last_trans = 0;
  1503. /*
  1504. * We'v had everything committed since the last time we were
  1505. * modified so clear this flag in case it was set for whatever
  1506. * reason, it's no longer relevant.
  1507. */
  1508. clear_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
  1509. &BTRFS_I(inode)->runtime_flags);
  1510. mutex_unlock(&inode->i_mutex);
  1511. goto out;
  1512. }
  1513. /*
  1514. * ok we haven't committed the transaction yet, lets do a commit
  1515. */
  1516. if (file->private_data)
  1517. btrfs_ioctl_trans_end(file);
  1518. trans = btrfs_start_transaction(root, 0);
  1519. if (IS_ERR(trans)) {
  1520. ret = PTR_ERR(trans);
  1521. mutex_unlock(&inode->i_mutex);
  1522. goto out;
  1523. }
  1524. ret = btrfs_log_dentry_safe(trans, root, dentry);
  1525. if (ret < 0) {
  1526. mutex_unlock(&inode->i_mutex);
  1527. goto out;
  1528. }
  1529. /* we've logged all the items and now have a consistent
  1530. * version of the file in the log. It is possible that
  1531. * someone will come in and modify the file, but that's
  1532. * fine because the log is consistent on disk, and we
  1533. * have references to all of the file's extents
  1534. *
  1535. * It is possible that someone will come in and log the
  1536. * file again, but that will end up using the synchronization
  1537. * inside btrfs_sync_log to keep things safe.
  1538. */
  1539. mutex_unlock(&inode->i_mutex);
  1540. if (ret != BTRFS_NO_LOG_SYNC) {
  1541. if (ret > 0) {
  1542. ret = btrfs_commit_transaction(trans, root);
  1543. } else {
  1544. ret = btrfs_sync_log(trans, root);
  1545. if (ret == 0)
  1546. ret = btrfs_end_transaction(trans, root);
  1547. else
  1548. ret = btrfs_commit_transaction(trans, root);
  1549. }
  1550. } else {
  1551. ret = btrfs_end_transaction(trans, root);
  1552. }
  1553. out:
  1554. return ret > 0 ? -EIO : ret;
  1555. }
  1556. static const struct vm_operations_struct btrfs_file_vm_ops = {
  1557. .fault = filemap_fault,
  1558. .page_mkwrite = btrfs_page_mkwrite,
  1559. .remap_pages = generic_file_remap_pages,
  1560. };
  1561. static int btrfs_file_mmap(struct file *filp, struct vm_area_struct *vma)
  1562. {
  1563. struct address_space *mapping = filp->f_mapping;
  1564. if (!mapping->a_ops->readpage)
  1565. return -ENOEXEC;
  1566. file_accessed(filp);
  1567. vma->vm_ops = &btrfs_file_vm_ops;
  1568. return 0;
  1569. }
  1570. static int hole_mergeable(struct inode *inode, struct extent_buffer *leaf,
  1571. int slot, u64 start, u64 end)
  1572. {
  1573. struct btrfs_file_extent_item *fi;
  1574. struct btrfs_key key;
  1575. if (slot < 0 || slot >= btrfs_header_nritems(leaf))
  1576. return 0;
  1577. btrfs_item_key_to_cpu(leaf, &key, slot);
  1578. if (key.objectid != btrfs_ino(inode) ||
  1579. key.type != BTRFS_EXTENT_DATA_KEY)
  1580. return 0;
  1581. fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item);
  1582. if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG)
  1583. return 0;
  1584. if (btrfs_file_extent_disk_bytenr(leaf, fi))
  1585. return 0;
  1586. if (key.offset == end)
  1587. return 1;
  1588. if (key.offset + btrfs_file_extent_num_bytes(leaf, fi) == start)
  1589. return 1;
  1590. return 0;
  1591. }
  1592. static int fill_holes(struct btrfs_trans_handle *trans, struct inode *inode,
  1593. struct btrfs_path *path, u64 offset, u64 end)
  1594. {
  1595. struct btrfs_root *root = BTRFS_I(inode)->root;
  1596. struct extent_buffer *leaf;
  1597. struct btrfs_file_extent_item *fi;
  1598. struct extent_map *hole_em;
  1599. struct extent_map_tree *em_tree = &BTRFS_I(inode)->extent_tree;
  1600. struct btrfs_key key;
  1601. int ret;
  1602. key.objectid = btrfs_ino(inode);
  1603. key.type = BTRFS_EXTENT_DATA_KEY;
  1604. key.offset = offset;
  1605. ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
  1606. if (ret < 0)
  1607. return ret;
  1608. BUG_ON(!ret);
  1609. leaf = path->nodes[0];
  1610. if (hole_mergeable(inode, leaf, path->slots[0]-1, offset, end)) {
  1611. u64 num_bytes;
  1612. path->slots[0]--;
  1613. fi = btrfs_item_ptr(leaf, path->slots[0],
  1614. struct btrfs_file_extent_item);
  1615. num_bytes = btrfs_file_extent_num_bytes(leaf, fi) +
  1616. end - offset;
  1617. btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
  1618. btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
  1619. btrfs_set_file_extent_offset(leaf, fi, 0);
  1620. btrfs_mark_buffer_dirty(leaf);
  1621. goto out;
  1622. }
  1623. if (hole_mergeable(inode, leaf, path->slots[0]+1, offset, end)) {
  1624. u64 num_bytes;
  1625. path->slots[0]++;
  1626. key.offset = offset;
  1627. btrfs_set_item_key_safe(trans, root, path, &key);
  1628. fi = btrfs_item_ptr(leaf, path->slots[0],
  1629. struct btrfs_file_extent_item);
  1630. num_bytes = btrfs_file_extent_num_bytes(leaf, fi) + end -
  1631. offset;
  1632. btrfs_set_file_extent_num_bytes(leaf, fi, num_bytes);
  1633. btrfs_set_file_extent_ram_bytes(leaf, fi, num_bytes);
  1634. btrfs_set_file_extent_offset(leaf, fi, 0);
  1635. btrfs_mark_buffer_dirty(leaf);
  1636. goto out;
  1637. }
  1638. btrfs_release_path(path);
  1639. ret = btrfs_insert_file_extent(trans, root, btrfs_ino(inode), offset,
  1640. 0, 0, end - offset, 0, end - offset,
  1641. 0, 0, 0);
  1642. if (ret)
  1643. return ret;
  1644. out:
  1645. btrfs_release_path(path);
  1646. hole_em = alloc_extent_map();
  1647. if (!hole_em) {
  1648. btrfs_drop_extent_cache(inode, offset, end - 1, 0);
  1649. set_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
  1650. &BTRFS_I(inode)->runtime_flags);
  1651. } else {
  1652. hole_em->start = offset;
  1653. hole_em->len = end - offset;
  1654. hole_em->orig_start = offset;
  1655. hole_em->block_start = EXTENT_MAP_HOLE;
  1656. hole_em->block_len = 0;
  1657. hole_em->orig_block_len = 0;
  1658. hole_em->bdev = root->fs_info->fs_devices->latest_bdev;
  1659. hole_em->compress_type = BTRFS_COMPRESS_NONE;
  1660. hole_em->generation = trans->transid;
  1661. do {
  1662. btrfs_drop_extent_cache(inode, offset, end - 1, 0);
  1663. write_lock(&em_tree->lock);
  1664. ret = add_extent_mapping(em_tree, hole_em);
  1665. if (!ret)
  1666. list_move(&hole_em->list,
  1667. &em_tree->modified_extents);
  1668. write_unlock(&em_tree->lock);
  1669. } while (ret == -EEXIST);
  1670. free_extent_map(hole_em);
  1671. if (ret)
  1672. set_bit(BTRFS_INODE_NEEDS_FULL_SYNC,
  1673. &BTRFS_I(inode)->runtime_flags);
  1674. }
  1675. return 0;
  1676. }
  1677. static int btrfs_punch_hole(struct inode *inode, loff_t offset, loff_t len)
  1678. {
  1679. struct btrfs_root *root = BTRFS_I(inode)->root;
  1680. struct extent_state *cached_state = NULL;
  1681. struct btrfs_path *path;
  1682. struct btrfs_block_rsv *rsv;
  1683. struct btrfs_trans_handle *trans;
  1684. u64 lockstart = round_up(offset, BTRFS_I(inode)->root->sectorsize);
  1685. u64 lockend = round_down(offset + len,
  1686. BTRFS_I(inode)->root->sectorsize) - 1;
  1687. u64 cur_offset = lockstart;
  1688. u64 min_size = btrfs_calc_trunc_metadata_size(root, 1);
  1689. u64 drop_end;
  1690. int ret = 0;
  1691. int err = 0;
  1692. bool same_page = ((offset >> PAGE_CACHE_SHIFT) ==
  1693. ((offset + len - 1) >> PAGE_CACHE_SHIFT));
  1694. btrfs_wait_ordered_range(inode, offset, len);
  1695. mutex_lock(&inode->i_mutex);
  1696. /*
  1697. * We needn't truncate any page which is beyond the end of the file
  1698. * because we are sure there is no data there.
  1699. */
  1700. /*
  1701. * Only do this if we are in the same page and we aren't doing the
  1702. * entire page.
  1703. */
  1704. if (same_page && len < PAGE_CACHE_SIZE) {
  1705. if (offset < round_up(inode->i_size, PAGE_CACHE_SIZE))
  1706. ret = btrfs_truncate_page(inode, offset, len, 0);
  1707. mutex_unlock(&inode->i_mutex);
  1708. return ret;
  1709. }
  1710. /* zero back part of the first page */
  1711. if (offset < round_up(inode->i_size, PAGE_CACHE_SIZE)) {
  1712. ret = btrfs_truncate_page(inode, offset, 0, 0);
  1713. if (ret) {
  1714. mutex_unlock(&inode->i_mutex);
  1715. return ret;
  1716. }
  1717. }
  1718. /* zero the front end of the last page */
  1719. if (offset + len < round_up(inode->i_size, PAGE_CACHE_SIZE)) {
  1720. ret = btrfs_truncate_page(inode, offset + len, 0, 1);
  1721. if (ret) {
  1722. mutex_unlock(&inode->i_mutex);
  1723. return ret;
  1724. }
  1725. }
  1726. if (lockend < lockstart) {
  1727. mutex_unlock(&inode->i_mutex);
  1728. return 0;
  1729. }
  1730. while (1) {
  1731. struct btrfs_ordered_extent *ordered;
  1732. truncate_pagecache_range(inode, lockstart, lockend);
  1733. lock_extent_bits(&BTRFS_I(inode)->io_tree, lockstart, lockend,
  1734. 0, &cached_state);
  1735. ordered = btrfs_lookup_first_ordered_extent(inode, lockend);
  1736. /*
  1737. * We need to make sure we have no ordered extents in this range
  1738. * and nobody raced in and read a page in this range, if we did
  1739. * we need to try again.
  1740. */
  1741. if ((!ordered ||
  1742. (ordered->file_offset + ordered->len < lockstart ||
  1743. ordered->file_offset > lockend)) &&
  1744. !test_range_bit(&BTRFS_I(inode)->io_tree, lockstart,
  1745. lockend, EXTENT_UPTODATE, 0,
  1746. cached_state)) {
  1747. if (ordered)
  1748. btrfs_put_ordered_extent(ordered);
  1749. break;
  1750. }
  1751. if (ordered)
  1752. btrfs_put_ordered_extent(ordered);
  1753. unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart,
  1754. lockend, &cached_state, GFP_NOFS);
  1755. btrfs_wait_ordered_range(inode, lockstart,
  1756. lockend - lockstart + 1);
  1757. }
  1758. path = btrfs_alloc_path();
  1759. if (!path) {
  1760. ret = -ENOMEM;
  1761. goto out;
  1762. }
  1763. rsv = btrfs_alloc_block_rsv(root, BTRFS_BLOCK_RSV_TEMP);
  1764. if (!rsv) {
  1765. ret = -ENOMEM;
  1766. goto out_free;
  1767. }
  1768. rsv->size = btrfs_calc_trunc_metadata_size(root, 1);
  1769. rsv->failfast = 1;
  1770. /*
  1771. * 1 - update the inode
  1772. * 1 - removing the extents in the range
  1773. * 1 - adding the hole extent
  1774. */
  1775. trans = btrfs_start_transaction(root, 3);
  1776. if (IS_ERR(trans)) {
  1777. err = PTR_ERR(trans);
  1778. goto out_free;
  1779. }
  1780. ret = btrfs_block_rsv_migrate(&root->fs_info->trans_block_rsv, rsv,
  1781. min_size);
  1782. BUG_ON(ret);
  1783. trans->block_rsv = rsv;
  1784. while (cur_offset < lockend) {
  1785. ret = __btrfs_drop_extents(trans, root, inode, path,
  1786. cur_offset, lockend + 1,
  1787. &drop_end, 1);
  1788. if (ret != -ENOSPC)
  1789. break;
  1790. trans->block_rsv = &root->fs_info->trans_block_rsv;
  1791. ret = fill_holes(trans, inode, path, cur_offset, drop_end);
  1792. if (ret) {
  1793. err = ret;
  1794. break;
  1795. }
  1796. cur_offset = drop_end;
  1797. ret = btrfs_update_inode(trans, root, inode);
  1798. if (ret) {
  1799. err = ret;
  1800. break;
  1801. }
  1802. btrfs_end_transaction(trans, root);
  1803. btrfs_btree_balance_dirty(root);
  1804. trans = btrfs_start_transaction(root, 3);
  1805. if (IS_ERR(trans)) {
  1806. ret = PTR_ERR(trans);
  1807. trans = NULL;
  1808. break;
  1809. }
  1810. ret = btrfs_block_rsv_migrate(&root->fs_info->trans_block_rsv,
  1811. rsv, min_size);
  1812. BUG_ON(ret); /* shouldn't happen */
  1813. trans->block_rsv = rsv;
  1814. }
  1815. if (ret) {
  1816. err = ret;
  1817. goto out_trans;
  1818. }
  1819. trans->block_rsv = &root->fs_info->trans_block_rsv;
  1820. ret = fill_holes(trans, inode, path, cur_offset, drop_end);
  1821. if (ret) {
  1822. err = ret;
  1823. goto out_trans;
  1824. }
  1825. out_trans:
  1826. if (!trans)
  1827. goto out_free;
  1828. inode_inc_iversion(inode);
  1829. inode->i_mtime = inode->i_ctime = CURRENT_TIME;
  1830. trans->block_rsv = &root->fs_info->trans_block_rsv;
  1831. ret = btrfs_update_inode(trans, root, inode);
  1832. btrfs_end_transaction(trans, root);
  1833. btrfs_btree_balance_dirty(root);
  1834. out_free:
  1835. btrfs_free_path(path);
  1836. btrfs_free_block_rsv(root, rsv);
  1837. out:
  1838. unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart, lockend,
  1839. &cached_state, GFP_NOFS);
  1840. mutex_unlock(&inode->i_mutex);
  1841. if (ret && !err)
  1842. err = ret;
  1843. return err;
  1844. }
  1845. static long btrfs_fallocate(struct file *file, int mode,
  1846. loff_t offset, loff_t len)
  1847. {
  1848. struct inode *inode = file->f_path.dentry->d_inode;
  1849. struct extent_state *cached_state = NULL;
  1850. u64 cur_offset;
  1851. u64 last_byte;
  1852. u64 alloc_start;
  1853. u64 alloc_end;
  1854. u64 alloc_hint = 0;
  1855. u64 locked_end;
  1856. struct extent_map *em;
  1857. int blocksize = BTRFS_I(inode)->root->sectorsize;
  1858. int ret;
  1859. alloc_start = round_down(offset, blocksize);
  1860. alloc_end = round_up(offset + len, blocksize);
  1861. /* Make sure we aren't being give some crap mode */
  1862. if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
  1863. return -EOPNOTSUPP;
  1864. if (mode & FALLOC_FL_PUNCH_HOLE)
  1865. return btrfs_punch_hole(inode, offset, len);
  1866. /*
  1867. * Make sure we have enough space before we do the
  1868. * allocation.
  1869. */
  1870. ret = btrfs_check_data_free_space(inode, alloc_end - alloc_start);
  1871. if (ret)
  1872. return ret;
  1873. /*
  1874. * wait for ordered IO before we have any locks. We'll loop again
  1875. * below with the locks held.
  1876. */
  1877. btrfs_wait_ordered_range(inode, alloc_start, alloc_end - alloc_start);
  1878. mutex_lock(&inode->i_mutex);
  1879. ret = inode_newsize_ok(inode, alloc_end);
  1880. if (ret)
  1881. goto out;
  1882. if (alloc_start > inode->i_size) {
  1883. ret = btrfs_cont_expand(inode, i_size_read(inode),
  1884. alloc_start);
  1885. if (ret)
  1886. goto out;
  1887. }
  1888. locked_end = alloc_end - 1;
  1889. while (1) {
  1890. struct btrfs_ordered_extent *ordered;
  1891. /* the extent lock is ordered inside the running
  1892. * transaction
  1893. */
  1894. lock_extent_bits(&BTRFS_I(inode)->io_tree, alloc_start,
  1895. locked_end, 0, &cached_state);
  1896. ordered = btrfs_lookup_first_ordered_extent(inode,
  1897. alloc_end - 1);
  1898. if (ordered &&
  1899. ordered->file_offset + ordered->len > alloc_start &&
  1900. ordered->file_offset < alloc_end) {
  1901. btrfs_put_ordered_extent(ordered);
  1902. unlock_extent_cached(&BTRFS_I(inode)->io_tree,
  1903. alloc_start, locked_end,
  1904. &cached_state, GFP_NOFS);
  1905. /*
  1906. * we can't wait on the range with the transaction
  1907. * running or with the extent lock held
  1908. */
  1909. btrfs_wait_ordered_range(inode, alloc_start,
  1910. alloc_end - alloc_start);
  1911. } else {
  1912. if (ordered)
  1913. btrfs_put_ordered_extent(ordered);
  1914. break;
  1915. }
  1916. }
  1917. cur_offset = alloc_start;
  1918. while (1) {
  1919. u64 actual_end;
  1920. em = btrfs_get_extent(inode, NULL, 0, cur_offset,
  1921. alloc_end - cur_offset, 0);
  1922. if (IS_ERR_OR_NULL(em)) {
  1923. if (!em)
  1924. ret = -ENOMEM;
  1925. else
  1926. ret = PTR_ERR(em);
  1927. break;
  1928. }
  1929. last_byte = min(extent_map_end(em), alloc_end);
  1930. actual_end = min_t(u64, extent_map_end(em), offset + len);
  1931. last_byte = ALIGN(last_byte, blocksize);
  1932. if (em->block_start == EXTENT_MAP_HOLE ||
  1933. (cur_offset >= inode->i_size &&
  1934. !test_bit(EXTENT_FLAG_PREALLOC, &em->flags))) {
  1935. ret = btrfs_prealloc_file_range(inode, mode, cur_offset,
  1936. last_byte - cur_offset,
  1937. 1 << inode->i_blkbits,
  1938. offset + len,
  1939. &alloc_hint);
  1940. if (ret < 0) {
  1941. free_extent_map(em);
  1942. break;
  1943. }
  1944. } else if (actual_end > inode->i_size &&
  1945. !(mode & FALLOC_FL_KEEP_SIZE)) {
  1946. /*
  1947. * We didn't need to allocate any more space, but we
  1948. * still extended the size of the file so we need to
  1949. * update i_size.
  1950. */
  1951. inode->i_ctime = CURRENT_TIME;
  1952. i_size_write(inode, actual_end);
  1953. btrfs_ordered_update_i_size(inode, actual_end, NULL);
  1954. }
  1955. free_extent_map(em);
  1956. cur_offset = last_byte;
  1957. if (cur_offset >= alloc_end) {
  1958. ret = 0;
  1959. break;
  1960. }
  1961. }
  1962. unlock_extent_cached(&BTRFS_I(inode)->io_tree, alloc_start, locked_end,
  1963. &cached_state, GFP_NOFS);
  1964. out:
  1965. mutex_unlock(&inode->i_mutex);
  1966. /* Let go of our reservation. */
  1967. btrfs_free_reserved_data_space(inode, alloc_end - alloc_start);
  1968. return ret;
  1969. }
  1970. static int find_desired_extent(struct inode *inode, loff_t *offset, int whence)
  1971. {
  1972. struct btrfs_root *root = BTRFS_I(inode)->root;
  1973. struct extent_map *em;
  1974. struct extent_state *cached_state = NULL;
  1975. u64 lockstart = *offset;
  1976. u64 lockend = i_size_read(inode);
  1977. u64 start = *offset;
  1978. u64 orig_start = *offset;
  1979. u64 len = i_size_read(inode);
  1980. u64 last_end = 0;
  1981. int ret = 0;
  1982. lockend = max_t(u64, root->sectorsize, lockend);
  1983. if (lockend <= lockstart)
  1984. lockend = lockstart + root->sectorsize;
  1985. len = lockend - lockstart + 1;
  1986. len = max_t(u64, len, root->sectorsize);
  1987. if (inode->i_size == 0)
  1988. return -ENXIO;
  1989. lock_extent_bits(&BTRFS_I(inode)->io_tree, lockstart, lockend, 0,
  1990. &cached_state);
  1991. /*
  1992. * Delalloc is such a pain. If we have a hole and we have pending
  1993. * delalloc for a portion of the hole we will get back a hole that
  1994. * exists for the entire range since it hasn't been actually written
  1995. * yet. So to take care of this case we need to look for an extent just
  1996. * before the position we want in case there is outstanding delalloc
  1997. * going on here.
  1998. */
  1999. if (whence == SEEK_HOLE && start != 0) {
  2000. if (start <= root->sectorsize)
  2001. em = btrfs_get_extent_fiemap(inode, NULL, 0, 0,
  2002. root->sectorsize, 0);
  2003. else
  2004. em = btrfs_get_extent_fiemap(inode, NULL, 0,
  2005. start - root->sectorsize,
  2006. root->sectorsize, 0);
  2007. if (IS_ERR(em)) {
  2008. ret = PTR_ERR(em);
  2009. goto out;
  2010. }
  2011. last_end = em->start + em->len;
  2012. if (em->block_start == EXTENT_MAP_DELALLOC)
  2013. last_end = min_t(u64, last_end, inode->i_size);
  2014. free_extent_map(em);
  2015. }
  2016. while (1) {
  2017. em = btrfs_get_extent_fiemap(inode, NULL, 0, start, len, 0);
  2018. if (IS_ERR(em)) {
  2019. ret = PTR_ERR(em);
  2020. break;
  2021. }
  2022. if (em->block_start == EXTENT_MAP_HOLE) {
  2023. if (test_bit(EXTENT_FLAG_VACANCY, &em->flags)) {
  2024. if (last_end <= orig_start) {
  2025. free_extent_map(em);
  2026. ret = -ENXIO;
  2027. break;
  2028. }
  2029. }
  2030. if (whence == SEEK_HOLE) {
  2031. *offset = start;
  2032. free_extent_map(em);
  2033. break;
  2034. }
  2035. } else {
  2036. if (whence == SEEK_DATA) {
  2037. if (em->block_start == EXTENT_MAP_DELALLOC) {
  2038. if (start >= inode->i_size) {
  2039. free_extent_map(em);
  2040. ret = -ENXIO;
  2041. break;
  2042. }
  2043. }
  2044. *offset = start;
  2045. free_extent_map(em);
  2046. break;
  2047. }
  2048. }
  2049. start = em->start + em->len;
  2050. last_end = em->start + em->len;
  2051. if (em->block_start == EXTENT_MAP_DELALLOC)
  2052. last_end = min_t(u64, last_end, inode->i_size);
  2053. if (test_bit(EXTENT_FLAG_VACANCY, &em->flags)) {
  2054. free_extent_map(em);
  2055. ret = -ENXIO;
  2056. break;
  2057. }
  2058. free_extent_map(em);
  2059. cond_resched();
  2060. }
  2061. if (!ret)
  2062. *offset = min(*offset, inode->i_size);
  2063. out:
  2064. unlock_extent_cached(&BTRFS_I(inode)->io_tree, lockstart, lockend,
  2065. &cached_state, GFP_NOFS);
  2066. return ret;
  2067. }
  2068. static loff_t btrfs_file_llseek(struct file *file, loff_t offset, int whence)
  2069. {
  2070. struct inode *inode = file->f_mapping->host;
  2071. int ret;
  2072. mutex_lock(&inode->i_mutex);
  2073. switch (whence) {
  2074. case SEEK_END:
  2075. case SEEK_CUR:
  2076. offset = generic_file_llseek(file, offset, whence);
  2077. goto out;
  2078. case SEEK_DATA:
  2079. case SEEK_HOLE:
  2080. if (offset >= i_size_read(inode)) {
  2081. mutex_unlock(&inode->i_mutex);
  2082. return -ENXIO;
  2083. }
  2084. ret = find_desired_extent(inode, &offset, whence);
  2085. if (ret) {
  2086. mutex_unlock(&inode->i_mutex);
  2087. return ret;
  2088. }
  2089. }
  2090. if (offset < 0 && !(file->f_mode & FMODE_UNSIGNED_OFFSET)) {
  2091. offset = -EINVAL;
  2092. goto out;
  2093. }
  2094. if (offset > inode->i_sb->s_maxbytes) {
  2095. offset = -EINVAL;
  2096. goto out;
  2097. }
  2098. /* Special lock needed here? */
  2099. if (offset != file->f_pos) {
  2100. file->f_pos = offset;
  2101. file->f_version = 0;
  2102. }
  2103. out:
  2104. mutex_unlock(&inode->i_mutex);
  2105. return offset;
  2106. }
  2107. const struct file_operations btrfs_file_operations = {
  2108. .llseek = btrfs_file_llseek,
  2109. .read = do_sync_read,
  2110. .write = do_sync_write,
  2111. .aio_read = generic_file_aio_read,
  2112. .splice_read = generic_file_splice_read,
  2113. .aio_write = btrfs_file_aio_write,
  2114. .mmap = btrfs_file_mmap,
  2115. .open = generic_file_open,
  2116. .release = btrfs_release_file,
  2117. .fsync = btrfs_sync_file,
  2118. .fallocate = btrfs_fallocate,
  2119. .unlocked_ioctl = btrfs_ioctl,
  2120. #ifdef CONFIG_COMPAT
  2121. .compat_ioctl = btrfs_ioctl,
  2122. #endif
  2123. };
  2124. void btrfs_auto_defrag_exit(void)
  2125. {
  2126. if (btrfs_inode_defrag_cachep)
  2127. kmem_cache_destroy(btrfs_inode_defrag_cachep);
  2128. }
  2129. int btrfs_auto_defrag_init(void)
  2130. {
  2131. btrfs_inode_defrag_cachep = kmem_cache_create("btrfs_inode_defrag",
  2132. sizeof(struct inode_defrag), 0,
  2133. SLAB_RECLAIM_ACCOUNT | SLAB_MEM_SPREAD,
  2134. NULL);
  2135. if (!btrfs_inode_defrag_cachep)
  2136. return -ENOMEM;
  2137. return 0;
  2138. }