budget.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756
  1. /*
  2. * This file is part of UBIFS.
  3. *
  4. * Copyright (C) 2006-2008 Nokia Corporation.
  5. *
  6. * This program is free software; you can redistribute it and/or modify it
  7. * under the terms of the GNU General Public License version 2 as published by
  8. * the Free Software Foundation.
  9. *
  10. * This program is distributed in the hope that it will be useful, but WITHOUT
  11. * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12. * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  13. * more details.
  14. *
  15. * You should have received a copy of the GNU General Public License along with
  16. * this program; if not, write to the Free Software Foundation, Inc., 51
  17. * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  18. *
  19. * Authors: Adrian Hunter
  20. * Artem Bityutskiy (Битюцкий Артём)
  21. */
  22. /*
  23. * This file implements the budgeting sub-system which is responsible for UBIFS
  24. * space management.
  25. *
  26. * Factors such as compression, wasted space at the ends of LEBs, space in other
  27. * journal heads, the effect of updates on the index, and so on, make it
  28. * impossible to accurately predict the amount of space needed. Consequently
  29. * approximations are used.
  30. */
  31. #include "ubifs.h"
  32. #include <linux/writeback.h>
  33. #include <linux/math64.h>
  34. /*
  35. * When pessimistic budget calculations say that there is no enough space,
  36. * UBIFS starts writing back dirty inodes and pages, doing garbage collection,
  37. * or committing. The below constant defines maximum number of times UBIFS
  38. * repeats the operations.
  39. */
  40. #define MAX_MKSPC_RETRIES 3
  41. /*
  42. * The below constant defines amount of dirty pages which should be written
  43. * back at when trying to shrink the liability.
  44. */
  45. #define NR_TO_WRITE 16
  46. /**
  47. * shrink_liability - write-back some dirty pages/inodes.
  48. * @c: UBIFS file-system description object
  49. * @nr_to_write: how many dirty pages to write-back
  50. *
  51. * This function shrinks UBIFS liability by means of writing back some amount
  52. * of dirty inodes and their pages. Returns the amount of pages which were
  53. * written back. The returned value does not include dirty inodes which were
  54. * synchronized.
  55. *
  56. * Note, this function synchronizes even VFS inodes which are locked
  57. * (@i_mutex) by the caller of the budgeting function, because write-back does
  58. * not touch @i_mutex.
  59. */
  60. static int shrink_liability(struct ubifs_info *c, int nr_to_write)
  61. {
  62. int nr_written;
  63. struct writeback_control wbc = {
  64. .sync_mode = WB_SYNC_NONE,
  65. .range_end = LLONG_MAX,
  66. .nr_to_write = nr_to_write,
  67. };
  68. generic_sync_sb_inodes(c->vfs_sb, &wbc);
  69. nr_written = nr_to_write - wbc.nr_to_write;
  70. if (!nr_written) {
  71. /*
  72. * Re-try again but wait on pages/inodes which are being
  73. * written-back concurrently (e.g., by pdflush).
  74. */
  75. memset(&wbc, 0, sizeof(struct writeback_control));
  76. wbc.sync_mode = WB_SYNC_ALL;
  77. wbc.range_end = LLONG_MAX;
  78. wbc.nr_to_write = nr_to_write;
  79. generic_sync_sb_inodes(c->vfs_sb, &wbc);
  80. nr_written = nr_to_write - wbc.nr_to_write;
  81. }
  82. dbg_budg("%d pages were written back", nr_written);
  83. return nr_written;
  84. }
  85. /**
  86. * run_gc - run garbage collector.
  87. * @c: UBIFS file-system description object
  88. *
  89. * This function runs garbage collector to make some more free space. Returns
  90. * zero if a free LEB has been produced, %-EAGAIN if commit is required, and a
  91. * negative error code in case of failure.
  92. */
  93. static int run_gc(struct ubifs_info *c)
  94. {
  95. int err, lnum;
  96. /* Make some free space by garbage-collecting dirty space */
  97. down_read(&c->commit_sem);
  98. lnum = ubifs_garbage_collect(c, 1);
  99. up_read(&c->commit_sem);
  100. if (lnum < 0)
  101. return lnum;
  102. /* GC freed one LEB, return it to lprops */
  103. dbg_budg("GC freed LEB %d", lnum);
  104. err = ubifs_return_leb(c, lnum);
  105. if (err)
  106. return err;
  107. return 0;
  108. }
  109. /**
  110. * get_liability - calculate current liability.
  111. * @c: UBIFS file-system description object
  112. *
  113. * This function calculates and returns current UBIFS liability, i.e. the
  114. * amount of bytes UBIFS has "promised" to write to the media.
  115. */
  116. static long long get_liability(struct ubifs_info *c)
  117. {
  118. long long liab;
  119. spin_lock(&c->space_lock);
  120. liab = c->budg_idx_growth + c->budg_data_growth + c->budg_dd_growth;
  121. spin_unlock(&c->space_lock);
  122. return liab;
  123. }
  124. /**
  125. * make_free_space - make more free space on the file-system.
  126. * @c: UBIFS file-system description object
  127. *
  128. * This function is called when an operation cannot be budgeted because there
  129. * is supposedly no free space. But in most cases there is some free space:
  130. * o budgeting is pessimistic, so it always budgets more than it is actually
  131. * needed, so shrinking the liability is one way to make free space - the
  132. * cached data will take less space then it was budgeted for;
  133. * o GC may turn some dark space into free space (budgeting treats dark space
  134. * as not available);
  135. * o commit may free some LEB, i.e., turn freeable LEBs into free LEBs.
  136. *
  137. * So this function tries to do the above. Returns %-EAGAIN if some free space
  138. * was presumably made and the caller has to re-try budgeting the operation.
  139. * Returns %-ENOSPC if it couldn't do more free space, and other negative error
  140. * codes on failures.
  141. */
  142. static int make_free_space(struct ubifs_info *c)
  143. {
  144. int err, retries = 0;
  145. long long liab1, liab2;
  146. do {
  147. liab1 = get_liability(c);
  148. /*
  149. * We probably have some dirty pages or inodes (liability), try
  150. * to write them back.
  151. */
  152. dbg_budg("liability %lld, run write-back", liab1);
  153. shrink_liability(c, NR_TO_WRITE);
  154. liab2 = get_liability(c);
  155. if (liab2 < liab1)
  156. return -EAGAIN;
  157. dbg_budg("new liability %lld (not shrinked)", liab2);
  158. /* Liability did not shrink again, try GC */
  159. dbg_budg("Run GC");
  160. err = run_gc(c);
  161. if (!err)
  162. return -EAGAIN;
  163. if (err != -EAGAIN && err != -ENOSPC)
  164. /* Some real error happened */
  165. return err;
  166. dbg_budg("Run commit (retries %d)", retries);
  167. err = ubifs_run_commit(c);
  168. if (err)
  169. return err;
  170. } while (retries++ < MAX_MKSPC_RETRIES);
  171. return -ENOSPC;
  172. }
  173. /**
  174. * ubifs_calc_min_idx_lebs - calculate amount of LEBs for the index.
  175. * @c: UBIFS file-system description object
  176. *
  177. * This function calculates and returns the number of LEBs which should be kept
  178. * for index usage.
  179. */
  180. int ubifs_calc_min_idx_lebs(struct ubifs_info *c)
  181. {
  182. int idx_lebs;
  183. long long idx_size;
  184. idx_size = c->old_idx_sz + c->budg_idx_growth + c->budg_uncommitted_idx;
  185. /* And make sure we have thrice the index size of space reserved */
  186. idx_size += idx_size << 1;
  187. /*
  188. * We do not maintain 'old_idx_size' as 'old_idx_lebs'/'old_idx_bytes'
  189. * pair, nor similarly the two variables for the new index size, so we
  190. * have to do this costly 64-bit division on fast-path.
  191. */
  192. idx_lebs = div_u64(idx_size + c->idx_leb_size - 1, c->idx_leb_size);
  193. /*
  194. * The index head is not available for the in-the-gaps method, so add an
  195. * extra LEB to compensate.
  196. */
  197. idx_lebs += 1;
  198. if (idx_lebs < MIN_INDEX_LEBS)
  199. idx_lebs = MIN_INDEX_LEBS;
  200. return idx_lebs;
  201. }
  202. /**
  203. * ubifs_calc_available - calculate available FS space.
  204. * @c: UBIFS file-system description object
  205. * @min_idx_lebs: minimum number of LEBs reserved for the index
  206. *
  207. * This function calculates and returns amount of FS space available for use.
  208. */
  209. long long ubifs_calc_available(const struct ubifs_info *c, int min_idx_lebs)
  210. {
  211. int subtract_lebs;
  212. long long available;
  213. available = c->main_bytes - c->lst.total_used;
  214. /*
  215. * Now 'available' contains theoretically available flash space
  216. * assuming there is no index, so we have to subtract the space which
  217. * is reserved for the index.
  218. */
  219. subtract_lebs = min_idx_lebs;
  220. /* Take into account that GC reserves one LEB for its own needs */
  221. subtract_lebs += 1;
  222. /*
  223. * The GC journal head LEB is not really accessible. And since
  224. * different write types go to different heads, we may count only on
  225. * one head's space.
  226. */
  227. subtract_lebs += c->jhead_cnt - 1;
  228. /* We also reserve one LEB for deletions, which bypass budgeting */
  229. subtract_lebs += 1;
  230. available -= (long long)subtract_lebs * c->leb_size;
  231. /* Subtract the dead space which is not available for use */
  232. available -= c->lst.total_dead;
  233. /*
  234. * Subtract dark space, which might or might not be usable - it depends
  235. * on the data which we have on the media and which will be written. If
  236. * this is a lot of uncompressed or not-compressible data, the dark
  237. * space cannot be used.
  238. */
  239. available -= c->lst.total_dark;
  240. /*
  241. * However, there is more dark space. The index may be bigger than
  242. * @min_idx_lebs. Those extra LEBs are assumed to be available, but
  243. * their dark space is not included in total_dark, so it is subtracted
  244. * here.
  245. */
  246. if (c->lst.idx_lebs > min_idx_lebs) {
  247. subtract_lebs = c->lst.idx_lebs - min_idx_lebs;
  248. available -= subtract_lebs * c->dark_wm;
  249. }
  250. /* The calculations are rough and may end up with a negative number */
  251. return available > 0 ? available : 0;
  252. }
  253. /**
  254. * can_use_rp - check whether the user is allowed to use reserved pool.
  255. * @c: UBIFS file-system description object
  256. *
  257. * UBIFS has so-called "reserved pool" which is flash space reserved
  258. * for the superuser and for uses whose UID/GID is recorded in UBIFS superblock.
  259. * This function checks whether current user is allowed to use reserved pool.
  260. * Returns %1 current user is allowed to use reserved pool and %0 otherwise.
  261. */
  262. static int can_use_rp(struct ubifs_info *c)
  263. {
  264. if (current_fsuid() == c->rp_uid || capable(CAP_SYS_RESOURCE) ||
  265. (c->rp_gid != 0 && in_group_p(c->rp_gid)))
  266. return 1;
  267. return 0;
  268. }
  269. /**
  270. * do_budget_space - reserve flash space for index and data growth.
  271. * @c: UBIFS file-system description object
  272. *
  273. * This function makes sure UBIFS has enough free LEBs for index growth and
  274. * data.
  275. *
  276. * When budgeting index space, UBIFS reserves thrice as many LEBs as the index
  277. * would take if it was consolidated and written to the flash. This guarantees
  278. * that the "in-the-gaps" commit method always succeeds and UBIFS will always
  279. * be able to commit dirty index. So this function basically adds amount of
  280. * budgeted index space to the size of the current index, multiplies this by 3,
  281. * and makes sure this does not exceed the amount of free LEBs.
  282. *
  283. * Notes about @c->min_idx_lebs and @c->lst.idx_lebs variables:
  284. * o @c->lst.idx_lebs is the number of LEBs the index currently uses. It might
  285. * be large, because UBIFS does not do any index consolidation as long as
  286. * there is free space. IOW, the index may take a lot of LEBs, but the LEBs
  287. * will contain a lot of dirt.
  288. * o @c->min_idx_lebs is the number of LEBS the index presumably takes. IOW,
  289. * the index may be consolidated to take up to @c->min_idx_lebs LEBs.
  290. *
  291. * This function returns zero in case of success, and %-ENOSPC in case of
  292. * failure.
  293. */
  294. static int do_budget_space(struct ubifs_info *c)
  295. {
  296. long long outstanding, available;
  297. int lebs, rsvd_idx_lebs, min_idx_lebs;
  298. /* First budget index space */
  299. min_idx_lebs = ubifs_calc_min_idx_lebs(c);
  300. /* Now 'min_idx_lebs' contains number of LEBs to reserve */
  301. if (min_idx_lebs > c->lst.idx_lebs)
  302. rsvd_idx_lebs = min_idx_lebs - c->lst.idx_lebs;
  303. else
  304. rsvd_idx_lebs = 0;
  305. /*
  306. * The number of LEBs that are available to be used by the index is:
  307. *
  308. * @c->lst.empty_lebs + @c->freeable_cnt + @c->idx_gc_cnt -
  309. * @c->lst.taken_empty_lebs
  310. *
  311. * @c->lst.empty_lebs are available because they are empty.
  312. * @c->freeable_cnt are available because they contain only free and
  313. * dirty space, @c->idx_gc_cnt are available because they are index
  314. * LEBs that have been garbage collected and are awaiting the commit
  315. * before they can be used. And the in-the-gaps method will grab these
  316. * if it needs them. @c->lst.taken_empty_lebs are empty LEBs that have
  317. * already been allocated for some purpose.
  318. *
  319. * Note, @c->idx_gc_cnt is included to both @c->lst.empty_lebs (because
  320. * these LEBs are empty) and to @c->lst.taken_empty_lebs (because they
  321. * are taken until after the commit).
  322. *
  323. * Note, @c->lst.taken_empty_lebs may temporarily be higher by one
  324. * because of the way we serialize LEB allocations and budgeting. See a
  325. * comment in 'ubifs_find_free_space()'.
  326. */
  327. lebs = c->lst.empty_lebs + c->freeable_cnt + c->idx_gc_cnt -
  328. c->lst.taken_empty_lebs;
  329. if (unlikely(rsvd_idx_lebs > lebs)) {
  330. dbg_budg("out of indexing space: min_idx_lebs %d (old %d), "
  331. "rsvd_idx_lebs %d", min_idx_lebs, c->min_idx_lebs,
  332. rsvd_idx_lebs);
  333. return -ENOSPC;
  334. }
  335. available = ubifs_calc_available(c, min_idx_lebs);
  336. outstanding = c->budg_data_growth + c->budg_dd_growth;
  337. if (unlikely(available < outstanding)) {
  338. dbg_budg("out of data space: available %lld, outstanding %lld",
  339. available, outstanding);
  340. return -ENOSPC;
  341. }
  342. if (available - outstanding <= c->rp_size && !can_use_rp(c))
  343. return -ENOSPC;
  344. c->min_idx_lebs = min_idx_lebs;
  345. return 0;
  346. }
  347. /**
  348. * calc_idx_growth - calculate approximate index growth from budgeting request.
  349. * @c: UBIFS file-system description object
  350. * @req: budgeting request
  351. *
  352. * For now we assume each new node adds one znode. But this is rather poor
  353. * approximation, though.
  354. */
  355. static int calc_idx_growth(const struct ubifs_info *c,
  356. const struct ubifs_budget_req *req)
  357. {
  358. int znodes;
  359. znodes = req->new_ino + (req->new_page << UBIFS_BLOCKS_PER_PAGE_SHIFT) +
  360. req->new_dent;
  361. return znodes * c->max_idx_node_sz;
  362. }
  363. /**
  364. * calc_data_growth - calculate approximate amount of new data from budgeting
  365. * request.
  366. * @c: UBIFS file-system description object
  367. * @req: budgeting request
  368. */
  369. static int calc_data_growth(const struct ubifs_info *c,
  370. const struct ubifs_budget_req *req)
  371. {
  372. int data_growth;
  373. data_growth = req->new_ino ? c->inode_budget : 0;
  374. if (req->new_page)
  375. data_growth += c->page_budget;
  376. if (req->new_dent)
  377. data_growth += c->dent_budget;
  378. data_growth += req->new_ino_d;
  379. return data_growth;
  380. }
  381. /**
  382. * calc_dd_growth - calculate approximate amount of data which makes other data
  383. * dirty from budgeting request.
  384. * @c: UBIFS file-system description object
  385. * @req: budgeting request
  386. */
  387. static int calc_dd_growth(const struct ubifs_info *c,
  388. const struct ubifs_budget_req *req)
  389. {
  390. int dd_growth;
  391. dd_growth = req->dirtied_page ? c->page_budget : 0;
  392. if (req->dirtied_ino)
  393. dd_growth += c->inode_budget << (req->dirtied_ino - 1);
  394. if (req->mod_dent)
  395. dd_growth += c->dent_budget;
  396. dd_growth += req->dirtied_ino_d;
  397. return dd_growth;
  398. }
  399. /**
  400. * ubifs_budget_space - ensure there is enough space to complete an operation.
  401. * @c: UBIFS file-system description object
  402. * @req: budget request
  403. *
  404. * This function allocates budget for an operation. It uses pessimistic
  405. * approximation of how much flash space the operation needs. The goal of this
  406. * function is to make sure UBIFS always has flash space to flush all dirty
  407. * pages, dirty inodes, and dirty znodes (liability). This function may force
  408. * commit, garbage-collection or write-back. Returns zero in case of success,
  409. * %-ENOSPC if there is no free space and other negative error codes in case of
  410. * failures.
  411. */
  412. int ubifs_budget_space(struct ubifs_info *c, struct ubifs_budget_req *req)
  413. {
  414. int uninitialized_var(cmt_retries), uninitialized_var(wb_retries);
  415. int err, idx_growth, data_growth, dd_growth, retried = 0;
  416. ubifs_assert(req->new_page <= 1);
  417. ubifs_assert(req->dirtied_page <= 1);
  418. ubifs_assert(req->new_dent <= 1);
  419. ubifs_assert(req->mod_dent <= 1);
  420. ubifs_assert(req->new_ino <= 1);
  421. ubifs_assert(req->new_ino_d <= UBIFS_MAX_INO_DATA);
  422. ubifs_assert(req->dirtied_ino <= 4);
  423. ubifs_assert(req->dirtied_ino_d <= UBIFS_MAX_INO_DATA * 4);
  424. ubifs_assert(!(req->new_ino_d & 7));
  425. ubifs_assert(!(req->dirtied_ino_d & 7));
  426. data_growth = calc_data_growth(c, req);
  427. dd_growth = calc_dd_growth(c, req);
  428. if (!data_growth && !dd_growth)
  429. return 0;
  430. idx_growth = calc_idx_growth(c, req);
  431. again:
  432. spin_lock(&c->space_lock);
  433. ubifs_assert(c->budg_idx_growth >= 0);
  434. ubifs_assert(c->budg_data_growth >= 0);
  435. ubifs_assert(c->budg_dd_growth >= 0);
  436. if (unlikely(c->nospace) && (c->nospace_rp || !can_use_rp(c))) {
  437. dbg_budg("no space");
  438. spin_unlock(&c->space_lock);
  439. return -ENOSPC;
  440. }
  441. c->budg_idx_growth += idx_growth;
  442. c->budg_data_growth += data_growth;
  443. c->budg_dd_growth += dd_growth;
  444. err = do_budget_space(c);
  445. if (likely(!err)) {
  446. req->idx_growth = idx_growth;
  447. req->data_growth = data_growth;
  448. req->dd_growth = dd_growth;
  449. spin_unlock(&c->space_lock);
  450. return 0;
  451. }
  452. /* Restore the old values */
  453. c->budg_idx_growth -= idx_growth;
  454. c->budg_data_growth -= data_growth;
  455. c->budg_dd_growth -= dd_growth;
  456. spin_unlock(&c->space_lock);
  457. if (req->fast) {
  458. dbg_budg("no space for fast budgeting");
  459. return err;
  460. }
  461. err = make_free_space(c);
  462. cond_resched();
  463. if (err == -EAGAIN) {
  464. dbg_budg("try again");
  465. goto again;
  466. } else if (err == -ENOSPC) {
  467. if (!retried) {
  468. retried = 1;
  469. dbg_budg("-ENOSPC, but anyway try once again");
  470. goto again;
  471. }
  472. dbg_budg("FS is full, -ENOSPC");
  473. c->nospace = 1;
  474. if (can_use_rp(c) || c->rp_size == 0)
  475. c->nospace_rp = 1;
  476. smp_wmb();
  477. } else
  478. ubifs_err("cannot budget space, error %d", err);
  479. return err;
  480. }
  481. /**
  482. * ubifs_release_budget - release budgeted free space.
  483. * @c: UBIFS file-system description object
  484. * @req: budget request
  485. *
  486. * This function releases the space budgeted by 'ubifs_budget_space()'. Note,
  487. * since the index changes (which were budgeted for in @req->idx_growth) will
  488. * only be written to the media on commit, this function moves the index budget
  489. * from @c->budg_idx_growth to @c->budg_uncommitted_idx. The latter will be
  490. * zeroed by the commit operation.
  491. */
  492. void ubifs_release_budget(struct ubifs_info *c, struct ubifs_budget_req *req)
  493. {
  494. ubifs_assert(req->new_page <= 1);
  495. ubifs_assert(req->dirtied_page <= 1);
  496. ubifs_assert(req->new_dent <= 1);
  497. ubifs_assert(req->mod_dent <= 1);
  498. ubifs_assert(req->new_ino <= 1);
  499. ubifs_assert(req->new_ino_d <= UBIFS_MAX_INO_DATA);
  500. ubifs_assert(req->dirtied_ino <= 4);
  501. ubifs_assert(req->dirtied_ino_d <= UBIFS_MAX_INO_DATA * 4);
  502. ubifs_assert(!(req->new_ino_d & 7));
  503. ubifs_assert(!(req->dirtied_ino_d & 7));
  504. if (!req->recalculate) {
  505. ubifs_assert(req->idx_growth >= 0);
  506. ubifs_assert(req->data_growth >= 0);
  507. ubifs_assert(req->dd_growth >= 0);
  508. }
  509. if (req->recalculate) {
  510. req->data_growth = calc_data_growth(c, req);
  511. req->dd_growth = calc_dd_growth(c, req);
  512. req->idx_growth = calc_idx_growth(c, req);
  513. }
  514. if (!req->data_growth && !req->dd_growth)
  515. return;
  516. c->nospace = c->nospace_rp = 0;
  517. smp_wmb();
  518. spin_lock(&c->space_lock);
  519. c->budg_idx_growth -= req->idx_growth;
  520. c->budg_uncommitted_idx += req->idx_growth;
  521. c->budg_data_growth -= req->data_growth;
  522. c->budg_dd_growth -= req->dd_growth;
  523. c->min_idx_lebs = ubifs_calc_min_idx_lebs(c);
  524. ubifs_assert(c->budg_idx_growth >= 0);
  525. ubifs_assert(c->budg_data_growth >= 0);
  526. ubifs_assert(c->budg_dd_growth >= 0);
  527. ubifs_assert(c->min_idx_lebs < c->main_lebs);
  528. ubifs_assert(!(c->budg_idx_growth & 7));
  529. ubifs_assert(!(c->budg_data_growth & 7));
  530. ubifs_assert(!(c->budg_dd_growth & 7));
  531. spin_unlock(&c->space_lock);
  532. }
  533. /**
  534. * ubifs_convert_page_budget - convert budget of a new page.
  535. * @c: UBIFS file-system description object
  536. *
  537. * This function converts budget which was allocated for a new page of data to
  538. * the budget of changing an existing page of data. The latter is smaller than
  539. * the former, so this function only does simple re-calculation and does not
  540. * involve any write-back.
  541. */
  542. void ubifs_convert_page_budget(struct ubifs_info *c)
  543. {
  544. spin_lock(&c->space_lock);
  545. /* Release the index growth reservation */
  546. c->budg_idx_growth -= c->max_idx_node_sz << UBIFS_BLOCKS_PER_PAGE_SHIFT;
  547. /* Release the data growth reservation */
  548. c->budg_data_growth -= c->page_budget;
  549. /* Increase the dirty data growth reservation instead */
  550. c->budg_dd_growth += c->page_budget;
  551. /* And re-calculate the indexing space reservation */
  552. c->min_idx_lebs = ubifs_calc_min_idx_lebs(c);
  553. spin_unlock(&c->space_lock);
  554. }
  555. /**
  556. * ubifs_release_dirty_inode_budget - release dirty inode budget.
  557. * @c: UBIFS file-system description object
  558. * @ui: UBIFS inode to release the budget for
  559. *
  560. * This function releases budget corresponding to a dirty inode. It is usually
  561. * called when after the inode has been written to the media and marked as
  562. * clean.
  563. */
  564. void ubifs_release_dirty_inode_budget(struct ubifs_info *c,
  565. struct ubifs_inode *ui)
  566. {
  567. struct ubifs_budget_req req;
  568. memset(&req, 0, sizeof(struct ubifs_budget_req));
  569. req.dd_growth = c->inode_budget + ALIGN(ui->data_len, 8);
  570. ubifs_release_budget(c, &req);
  571. }
  572. /**
  573. * ubifs_reported_space - calculate reported free space.
  574. * @c: the UBIFS file-system description object
  575. * @free: amount of free space
  576. *
  577. * This function calculates amount of free space which will be reported to
  578. * user-space. User-space application tend to expect that if the file-system
  579. * (e.g., via the 'statfs()' call) reports that it has N bytes available, they
  580. * are able to write a file of size N. UBIFS attaches node headers to each data
  581. * node and it has to write indexing nodes as well. This introduces additional
  582. * overhead, and UBIFS has to report slightly less free space to meet the above
  583. * expectations.
  584. *
  585. * This function assumes free space is made up of uncompressed data nodes and
  586. * full index nodes (one per data node, tripled because we always allow enough
  587. * space to write the index thrice).
  588. *
  589. * Note, the calculation is pessimistic, which means that most of the time
  590. * UBIFS reports less space than it actually has.
  591. */
  592. long long ubifs_reported_space(const struct ubifs_info *c, long long free)
  593. {
  594. int divisor, factor, f;
  595. /*
  596. * Reported space size is @free * X, where X is UBIFS block size
  597. * divided by UBIFS block size + all overhead one data block
  598. * introduces. The overhead is the node header + indexing overhead.
  599. *
  600. * Indexing overhead calculations are based on the following formula:
  601. * I = N/(f - 1) + 1, where I - number of indexing nodes, N - number
  602. * of data nodes, f - fanout. Because effective UBIFS fanout is twice
  603. * as less than maximum fanout, we assume that each data node
  604. * introduces 3 * @c->max_idx_node_sz / (@c->fanout/2 - 1) bytes.
  605. * Note, the multiplier 3 is because UBIFS reserves thrice as more space
  606. * for the index.
  607. */
  608. f = c->fanout > 3 ? c->fanout >> 1 : 2;
  609. factor = UBIFS_BLOCK_SIZE;
  610. divisor = UBIFS_MAX_DATA_NODE_SZ;
  611. divisor += (c->max_idx_node_sz * 3) / (f - 1);
  612. free *= factor;
  613. return div_u64(free, divisor);
  614. }
  615. /**
  616. * ubifs_get_free_space_nolock - return amount of free space.
  617. * @c: UBIFS file-system description object
  618. *
  619. * This function calculates amount of free space to report to user-space.
  620. *
  621. * Because UBIFS may introduce substantial overhead (the index, node headers,
  622. * alignment, wastage at the end of LEBs, etc), it cannot report real amount of
  623. * free flash space it has (well, because not all dirty space is reclaimable,
  624. * UBIFS does not actually know the real amount). If UBIFS did so, it would
  625. * bread user expectations about what free space is. Users seem to accustomed
  626. * to assume that if the file-system reports N bytes of free space, they would
  627. * be able to fit a file of N bytes to the FS. This almost works for
  628. * traditional file-systems, because they have way less overhead than UBIFS.
  629. * So, to keep users happy, UBIFS tries to take the overhead into account.
  630. */
  631. long long ubifs_get_free_space_nolock(struct ubifs_info *c)
  632. {
  633. int rsvd_idx_lebs, lebs;
  634. long long available, outstanding, free;
  635. ubifs_assert(c->min_idx_lebs == ubifs_calc_min_idx_lebs(c));
  636. outstanding = c->budg_data_growth + c->budg_dd_growth;
  637. available = ubifs_calc_available(c, c->min_idx_lebs);
  638. /*
  639. * When reporting free space to user-space, UBIFS guarantees that it is
  640. * possible to write a file of free space size. This means that for
  641. * empty LEBs we may use more precise calculations than
  642. * 'ubifs_calc_available()' is using. Namely, we know that in empty
  643. * LEBs we would waste only @c->leb_overhead bytes, not @c->dark_wm.
  644. * Thus, amend the available space.
  645. *
  646. * Note, the calculations below are similar to what we have in
  647. * 'do_budget_space()', so refer there for comments.
  648. */
  649. if (c->min_idx_lebs > c->lst.idx_lebs)
  650. rsvd_idx_lebs = c->min_idx_lebs - c->lst.idx_lebs;
  651. else
  652. rsvd_idx_lebs = 0;
  653. lebs = c->lst.empty_lebs + c->freeable_cnt + c->idx_gc_cnt -
  654. c->lst.taken_empty_lebs;
  655. lebs -= rsvd_idx_lebs;
  656. available += lebs * (c->dark_wm - c->leb_overhead);
  657. if (available > outstanding)
  658. free = ubifs_reported_space(c, available - outstanding);
  659. else
  660. free = 0;
  661. return free;
  662. }
  663. /**
  664. * ubifs_get_free_space - return amount of free space.
  665. * @c: UBIFS file-system description object
  666. *
  667. * This function calculates and retuns amount of free space to report to
  668. * user-space.
  669. */
  670. long long ubifs_get_free_space(struct ubifs_info *c)
  671. {
  672. long long free;
  673. spin_lock(&c->space_lock);
  674. free = ubifs_get_free_space_nolock(c);
  675. spin_unlock(&c->space_lock);
  676. return free;
  677. }