fsmap.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright (C) 2017 Oracle. All Rights Reserved.
  4. *
  5. * Author: Darrick J. Wong <darrick.wong@oracle.com>
  6. */
  7. #include "ext4.h"
  8. #include <linux/fsmap.h>
  9. #include "fsmap.h"
  10. #include "mballoc.h"
  11. #include <linux/sort.h>
  12. #include <linux/list_sort.h>
  13. #include <trace/events/ext4.h>
  14. /* Convert an ext4_fsmap to an fsmap. */
  15. void ext4_fsmap_from_internal(struct super_block *sb, struct fsmap *dest,
  16. struct ext4_fsmap *src)
  17. {
  18. dest->fmr_device = src->fmr_device;
  19. dest->fmr_flags = src->fmr_flags;
  20. dest->fmr_physical = src->fmr_physical << sb->s_blocksize_bits;
  21. dest->fmr_owner = src->fmr_owner;
  22. dest->fmr_offset = 0;
  23. dest->fmr_length = src->fmr_length << sb->s_blocksize_bits;
  24. dest->fmr_reserved[0] = 0;
  25. dest->fmr_reserved[1] = 0;
  26. dest->fmr_reserved[2] = 0;
  27. }
  28. /* Convert an fsmap to an ext4_fsmap. */
  29. void ext4_fsmap_to_internal(struct super_block *sb, struct ext4_fsmap *dest,
  30. struct fsmap *src)
  31. {
  32. dest->fmr_device = src->fmr_device;
  33. dest->fmr_flags = src->fmr_flags;
  34. dest->fmr_physical = src->fmr_physical >> sb->s_blocksize_bits;
  35. dest->fmr_owner = src->fmr_owner;
  36. dest->fmr_length = src->fmr_length >> sb->s_blocksize_bits;
  37. }
  38. /* getfsmap query state */
  39. struct ext4_getfsmap_info {
  40. struct ext4_fsmap_head *gfi_head;
  41. ext4_fsmap_format_t gfi_formatter; /* formatting fn */
  42. void *gfi_format_arg;/* format buffer */
  43. ext4_fsblk_t gfi_next_fsblk; /* next fsblock we expect */
  44. u32 gfi_dev; /* device id */
  45. ext4_group_t gfi_agno; /* bg number, if applicable */
  46. struct ext4_fsmap gfi_low; /* low rmap key */
  47. struct ext4_fsmap gfi_high; /* high rmap key */
  48. struct ext4_fsmap gfi_lastfree; /* free ext at end of last bg */
  49. struct list_head gfi_meta_list; /* fixed metadata list */
  50. bool gfi_last; /* last extent? */
  51. };
  52. /* Associate a device with a getfsmap handler. */
  53. struct ext4_getfsmap_dev {
  54. int (*gfd_fn)(struct super_block *sb,
  55. struct ext4_fsmap *keys,
  56. struct ext4_getfsmap_info *info);
  57. u32 gfd_dev;
  58. };
  59. /* Compare two getfsmap device handlers. */
  60. static int ext4_getfsmap_dev_compare(const void *p1, const void *p2)
  61. {
  62. const struct ext4_getfsmap_dev *d1 = p1;
  63. const struct ext4_getfsmap_dev *d2 = p2;
  64. return d1->gfd_dev - d2->gfd_dev;
  65. }
  66. /* Compare a record against our starting point */
  67. static bool ext4_getfsmap_rec_before_low_key(struct ext4_getfsmap_info *info,
  68. struct ext4_fsmap *rec)
  69. {
  70. return rec->fmr_physical + rec->fmr_length <=
  71. info->gfi_low.fmr_physical;
  72. }
  73. /*
  74. * Format a reverse mapping for getfsmap, having translated rm_startblock
  75. * into the appropriate daddr units.
  76. */
  77. static int ext4_getfsmap_helper(struct super_block *sb,
  78. struct ext4_getfsmap_info *info,
  79. struct ext4_fsmap *rec)
  80. {
  81. struct ext4_fsmap fmr;
  82. struct ext4_sb_info *sbi = EXT4_SB(sb);
  83. ext4_fsblk_t rec_fsblk = rec->fmr_physical;
  84. ext4_group_t agno;
  85. ext4_grpblk_t cno;
  86. int error;
  87. if (fatal_signal_pending(current))
  88. return -EINTR;
  89. /*
  90. * Filter out records that start before our startpoint, if the
  91. * caller requested that.
  92. */
  93. if (ext4_getfsmap_rec_before_low_key(info, rec)) {
  94. rec_fsblk += rec->fmr_length;
  95. if (info->gfi_next_fsblk < rec_fsblk)
  96. info->gfi_next_fsblk = rec_fsblk;
  97. return EXT4_QUERY_RANGE_CONTINUE;
  98. }
  99. /* Are we just counting mappings? */
  100. if (info->gfi_head->fmh_count == 0) {
  101. if (info->gfi_head->fmh_entries == UINT_MAX)
  102. return EXT4_QUERY_RANGE_ABORT;
  103. if (rec_fsblk > info->gfi_next_fsblk)
  104. info->gfi_head->fmh_entries++;
  105. if (info->gfi_last)
  106. return EXT4_QUERY_RANGE_CONTINUE;
  107. info->gfi_head->fmh_entries++;
  108. rec_fsblk += rec->fmr_length;
  109. if (info->gfi_next_fsblk < rec_fsblk)
  110. info->gfi_next_fsblk = rec_fsblk;
  111. return EXT4_QUERY_RANGE_CONTINUE;
  112. }
  113. /*
  114. * If the record starts past the last physical block we saw,
  115. * then we've found a gap. Report the gap as being owned by
  116. * whatever the caller specified is the missing owner.
  117. */
  118. if (rec_fsblk > info->gfi_next_fsblk) {
  119. if (info->gfi_head->fmh_entries >= info->gfi_head->fmh_count)
  120. return EXT4_QUERY_RANGE_ABORT;
  121. ext4_get_group_no_and_offset(sb, info->gfi_next_fsblk,
  122. &agno, &cno);
  123. trace_ext4_fsmap_mapping(sb, info->gfi_dev, agno,
  124. EXT4_C2B(sbi, cno),
  125. rec_fsblk - info->gfi_next_fsblk,
  126. EXT4_FMR_OWN_UNKNOWN);
  127. fmr.fmr_device = info->gfi_dev;
  128. fmr.fmr_physical = info->gfi_next_fsblk;
  129. fmr.fmr_owner = EXT4_FMR_OWN_UNKNOWN;
  130. fmr.fmr_length = rec_fsblk - info->gfi_next_fsblk;
  131. fmr.fmr_flags = FMR_OF_SPECIAL_OWNER;
  132. error = info->gfi_formatter(&fmr, info->gfi_format_arg);
  133. if (error)
  134. return error;
  135. info->gfi_head->fmh_entries++;
  136. }
  137. if (info->gfi_last)
  138. goto out;
  139. /* Fill out the extent we found */
  140. if (info->gfi_head->fmh_entries >= info->gfi_head->fmh_count)
  141. return EXT4_QUERY_RANGE_ABORT;
  142. ext4_get_group_no_and_offset(sb, rec_fsblk, &agno, &cno);
  143. trace_ext4_fsmap_mapping(sb, info->gfi_dev, agno, EXT4_C2B(sbi, cno),
  144. rec->fmr_length, rec->fmr_owner);
  145. fmr.fmr_device = info->gfi_dev;
  146. fmr.fmr_physical = rec_fsblk;
  147. fmr.fmr_owner = rec->fmr_owner;
  148. fmr.fmr_flags = FMR_OF_SPECIAL_OWNER;
  149. fmr.fmr_length = rec->fmr_length;
  150. error = info->gfi_formatter(&fmr, info->gfi_format_arg);
  151. if (error)
  152. return error;
  153. info->gfi_head->fmh_entries++;
  154. out:
  155. rec_fsblk += rec->fmr_length;
  156. if (info->gfi_next_fsblk < rec_fsblk)
  157. info->gfi_next_fsblk = rec_fsblk;
  158. return EXT4_QUERY_RANGE_CONTINUE;
  159. }
  160. static inline ext4_fsblk_t ext4_fsmap_next_pblk(struct ext4_fsmap *fmr)
  161. {
  162. return fmr->fmr_physical + fmr->fmr_length;
  163. }
  164. static int ext4_getfsmap_meta_helper(struct super_block *sb,
  165. ext4_group_t agno, ext4_grpblk_t start,
  166. ext4_grpblk_t len, void *priv)
  167. {
  168. struct ext4_getfsmap_info *info = priv;
  169. struct ext4_fsmap *p;
  170. struct ext4_fsmap *tmp;
  171. struct ext4_sb_info *sbi = EXT4_SB(sb);
  172. ext4_fsblk_t fsb, fs_start, fs_end;
  173. int error;
  174. fs_start = fsb = (EXT4_C2B(sbi, start) +
  175. ext4_group_first_block_no(sb, agno));
  176. fs_end = fs_start + EXT4_C2B(sbi, len);
  177. /*
  178. * Return relevant extents from the meta_list. We emit all extents that
  179. * partially/fully overlap with the query range
  180. */
  181. list_for_each_entry_safe(p, tmp, &info->gfi_meta_list, fmr_list) {
  182. if (p->fmr_physical + p->fmr_length <= info->gfi_next_fsblk) {
  183. list_del(&p->fmr_list);
  184. kfree(p);
  185. continue;
  186. }
  187. if (p->fmr_physical <= fs_end &&
  188. p->fmr_physical + p->fmr_length > fs_start) {
  189. /* Emit the retained free extent record if present */
  190. if (info->gfi_lastfree.fmr_owner) {
  191. error = ext4_getfsmap_helper(sb, info,
  192. &info->gfi_lastfree);
  193. if (error)
  194. return error;
  195. info->gfi_lastfree.fmr_owner = 0;
  196. }
  197. error = ext4_getfsmap_helper(sb, info, p);
  198. if (error)
  199. return error;
  200. fsb = p->fmr_physical + p->fmr_length;
  201. if (info->gfi_next_fsblk < fsb)
  202. info->gfi_next_fsblk = fsb;
  203. list_del(&p->fmr_list);
  204. kfree(p);
  205. continue;
  206. }
  207. }
  208. if (info->gfi_next_fsblk < fsb)
  209. info->gfi_next_fsblk = fsb;
  210. return 0;
  211. }
  212. /* Transform a blockgroup's free record into a fsmap */
  213. static int ext4_getfsmap_datadev_helper(struct super_block *sb,
  214. ext4_group_t agno, ext4_grpblk_t start,
  215. ext4_grpblk_t len, void *priv)
  216. {
  217. struct ext4_fsmap irec;
  218. struct ext4_getfsmap_info *info = priv;
  219. struct ext4_fsmap *p;
  220. struct ext4_fsmap *tmp;
  221. struct ext4_sb_info *sbi = EXT4_SB(sb);
  222. ext4_fsblk_t fsb;
  223. ext4_fsblk_t fslen;
  224. int error;
  225. fsb = (EXT4_C2B(sbi, start) + ext4_group_first_block_no(sb, agno));
  226. fslen = EXT4_C2B(sbi, len);
  227. /* If the retained free extent record is set... */
  228. if (info->gfi_lastfree.fmr_owner) {
  229. /* ...and abuts this one, lengthen it and return. */
  230. if (ext4_fsmap_next_pblk(&info->gfi_lastfree) == fsb) {
  231. info->gfi_lastfree.fmr_length += fslen;
  232. return 0;
  233. }
  234. /*
  235. * There's a gap between the two free extents; emit the
  236. * retained extent prior to merging the meta_list.
  237. */
  238. error = ext4_getfsmap_helper(sb, info, &info->gfi_lastfree);
  239. if (error)
  240. return error;
  241. info->gfi_lastfree.fmr_owner = 0;
  242. }
  243. /* Merge in any relevant extents from the meta_list */
  244. list_for_each_entry_safe(p, tmp, &info->gfi_meta_list, fmr_list) {
  245. if (p->fmr_physical + p->fmr_length <= info->gfi_next_fsblk) {
  246. list_del(&p->fmr_list);
  247. kfree(p);
  248. } else if (p->fmr_physical < fsb) {
  249. error = ext4_getfsmap_helper(sb, info, p);
  250. if (error)
  251. return error;
  252. list_del(&p->fmr_list);
  253. kfree(p);
  254. }
  255. }
  256. irec.fmr_device = 0;
  257. irec.fmr_physical = fsb;
  258. irec.fmr_length = fslen;
  259. irec.fmr_owner = EXT4_FMR_OWN_FREE;
  260. irec.fmr_flags = 0;
  261. /* If this is a free extent at the end of a bg, buffer it. */
  262. if (ext4_fsmap_next_pblk(&irec) ==
  263. ext4_group_first_block_no(sb, agno + 1)) {
  264. info->gfi_lastfree = irec;
  265. return 0;
  266. }
  267. /* Otherwise, emit it */
  268. return ext4_getfsmap_helper(sb, info, &irec);
  269. }
  270. /* Execute a getfsmap query against the log device. */
  271. static int ext4_getfsmap_logdev(struct super_block *sb, struct ext4_fsmap *keys,
  272. struct ext4_getfsmap_info *info)
  273. {
  274. journal_t *journal = EXT4_SB(sb)->s_journal;
  275. struct ext4_fsmap irec;
  276. /* Set up search keys */
  277. info->gfi_low = keys[0];
  278. info->gfi_low.fmr_length = 0;
  279. memset(&info->gfi_high, 0xFF, sizeof(info->gfi_high));
  280. trace_ext4_fsmap_low_key(sb, info->gfi_dev, 0,
  281. info->gfi_low.fmr_physical,
  282. info->gfi_low.fmr_length,
  283. info->gfi_low.fmr_owner);
  284. trace_ext4_fsmap_high_key(sb, info->gfi_dev, 0,
  285. info->gfi_high.fmr_physical,
  286. info->gfi_high.fmr_length,
  287. info->gfi_high.fmr_owner);
  288. if (keys[0].fmr_physical > 0)
  289. return 0;
  290. /* Fabricate an rmap entry for the external log device. */
  291. irec.fmr_physical = journal->j_blk_offset;
  292. irec.fmr_length = journal->j_total_len;
  293. irec.fmr_owner = EXT4_FMR_OWN_LOG;
  294. irec.fmr_flags = 0;
  295. return ext4_getfsmap_helper(sb, info, &irec);
  296. }
  297. /* Helper to fill out an ext4_fsmap. */
  298. static inline int ext4_getfsmap_fill(struct list_head *meta_list,
  299. ext4_fsblk_t fsb, ext4_fsblk_t len,
  300. uint64_t owner)
  301. {
  302. struct ext4_fsmap *fsm;
  303. fsm = kmalloc(sizeof(*fsm), GFP_NOFS);
  304. if (!fsm)
  305. return -ENOMEM;
  306. fsm->fmr_device = 0;
  307. fsm->fmr_flags = 0;
  308. fsm->fmr_physical = fsb;
  309. fsm->fmr_owner = owner;
  310. fsm->fmr_length = len;
  311. list_add_tail(&fsm->fmr_list, meta_list);
  312. return 0;
  313. }
  314. /*
  315. * This function returns the number of file system metadata blocks at
  316. * the beginning of a block group, including the reserved gdt blocks.
  317. */
  318. static unsigned int ext4_getfsmap_find_sb(struct super_block *sb,
  319. ext4_group_t agno,
  320. struct list_head *meta_list)
  321. {
  322. struct ext4_sb_info *sbi = EXT4_SB(sb);
  323. ext4_fsblk_t fsb = ext4_group_first_block_no(sb, agno);
  324. ext4_fsblk_t len;
  325. unsigned long first_meta_bg = le32_to_cpu(sbi->s_es->s_first_meta_bg);
  326. unsigned long metagroup = agno / EXT4_DESC_PER_BLOCK(sb);
  327. int error;
  328. /* Record the superblock. */
  329. if (ext4_bg_has_super(sb, agno)) {
  330. error = ext4_getfsmap_fill(meta_list, fsb, 1, EXT4_FMR_OWN_FS);
  331. if (error)
  332. return error;
  333. fsb++;
  334. }
  335. /* Record the group descriptors. */
  336. len = ext4_bg_num_gdb(sb, agno);
  337. if (!len)
  338. return 0;
  339. error = ext4_getfsmap_fill(meta_list, fsb, len,
  340. EXT4_FMR_OWN_GDT);
  341. if (error)
  342. return error;
  343. fsb += len;
  344. /* Reserved GDT blocks */
  345. if (!ext4_has_feature_meta_bg(sb) || metagroup < first_meta_bg) {
  346. len = le16_to_cpu(sbi->s_es->s_reserved_gdt_blocks);
  347. /*
  348. * mkfs.ext4 can set s_reserved_gdt_blocks as 0 in some cases,
  349. * check for that.
  350. */
  351. if (!len)
  352. return 0;
  353. error = ext4_getfsmap_fill(meta_list, fsb, len,
  354. EXT4_FMR_OWN_RESV_GDT);
  355. if (error)
  356. return error;
  357. }
  358. return 0;
  359. }
  360. /* Compare two fsmap items. */
  361. static int ext4_getfsmap_compare(void *priv,
  362. const struct list_head *a,
  363. const struct list_head *b)
  364. {
  365. struct ext4_fsmap *fa;
  366. struct ext4_fsmap *fb;
  367. fa = container_of(a, struct ext4_fsmap, fmr_list);
  368. fb = container_of(b, struct ext4_fsmap, fmr_list);
  369. if (fa->fmr_physical < fb->fmr_physical)
  370. return -1;
  371. else if (fa->fmr_physical > fb->fmr_physical)
  372. return 1;
  373. return 0;
  374. }
  375. /* Merge adjacent extents of fixed metadata. */
  376. static void ext4_getfsmap_merge_fixed_metadata(struct list_head *meta_list)
  377. {
  378. struct ext4_fsmap *p;
  379. struct ext4_fsmap *prev = NULL;
  380. struct ext4_fsmap *tmp;
  381. list_for_each_entry_safe(p, tmp, meta_list, fmr_list) {
  382. if (!prev) {
  383. prev = p;
  384. continue;
  385. }
  386. if (prev->fmr_owner == p->fmr_owner &&
  387. prev->fmr_physical + prev->fmr_length == p->fmr_physical) {
  388. prev->fmr_length += p->fmr_length;
  389. list_del(&p->fmr_list);
  390. kfree(p);
  391. } else
  392. prev = p;
  393. }
  394. }
  395. /* Free a list of fixed metadata. */
  396. static void ext4_getfsmap_free_fixed_metadata(struct list_head *meta_list)
  397. {
  398. struct ext4_fsmap *p;
  399. struct ext4_fsmap *tmp;
  400. list_for_each_entry_safe(p, tmp, meta_list, fmr_list) {
  401. list_del(&p->fmr_list);
  402. kfree(p);
  403. }
  404. }
  405. /* Find all the fixed metadata in the filesystem. */
  406. static int ext4_getfsmap_find_fixed_metadata(struct super_block *sb,
  407. struct list_head *meta_list)
  408. {
  409. struct ext4_group_desc *gdp;
  410. ext4_group_t agno;
  411. int error;
  412. INIT_LIST_HEAD(meta_list);
  413. /* Collect everything. */
  414. for (agno = 0; agno < EXT4_SB(sb)->s_groups_count; agno++) {
  415. gdp = ext4_get_group_desc(sb, agno, NULL);
  416. if (!gdp) {
  417. error = -EFSCORRUPTED;
  418. goto err;
  419. }
  420. /* Superblock & GDT */
  421. error = ext4_getfsmap_find_sb(sb, agno, meta_list);
  422. if (error)
  423. goto err;
  424. /* Block bitmap */
  425. error = ext4_getfsmap_fill(meta_list,
  426. ext4_block_bitmap(sb, gdp), 1,
  427. EXT4_FMR_OWN_BLKBM);
  428. if (error)
  429. goto err;
  430. /* Inode bitmap */
  431. error = ext4_getfsmap_fill(meta_list,
  432. ext4_inode_bitmap(sb, gdp), 1,
  433. EXT4_FMR_OWN_INOBM);
  434. if (error)
  435. goto err;
  436. /* Inodes */
  437. error = ext4_getfsmap_fill(meta_list,
  438. ext4_inode_table(sb, gdp),
  439. EXT4_SB(sb)->s_itb_per_group,
  440. EXT4_FMR_OWN_INODES);
  441. if (error)
  442. goto err;
  443. }
  444. /* Sort the list */
  445. list_sort(NULL, meta_list, ext4_getfsmap_compare);
  446. /* Merge adjacent extents */
  447. ext4_getfsmap_merge_fixed_metadata(meta_list);
  448. return 0;
  449. err:
  450. ext4_getfsmap_free_fixed_metadata(meta_list);
  451. return error;
  452. }
  453. /* Execute a getfsmap query against the buddy bitmaps */
  454. static int ext4_getfsmap_datadev(struct super_block *sb,
  455. struct ext4_fsmap *keys,
  456. struct ext4_getfsmap_info *info)
  457. {
  458. struct ext4_sb_info *sbi = EXT4_SB(sb);
  459. ext4_fsblk_t start_fsb;
  460. ext4_fsblk_t end_fsb;
  461. ext4_fsblk_t bofs;
  462. ext4_fsblk_t eofs;
  463. ext4_group_t start_ag;
  464. ext4_group_t end_ag;
  465. ext4_grpblk_t first_cluster;
  466. ext4_grpblk_t last_cluster;
  467. struct ext4_fsmap irec;
  468. int error = 0;
  469. bofs = le32_to_cpu(sbi->s_es->s_first_data_block);
  470. eofs = ext4_blocks_count(sbi->s_es);
  471. if (keys[0].fmr_physical >= eofs)
  472. return 0;
  473. else if (keys[0].fmr_physical < bofs)
  474. keys[0].fmr_physical = bofs;
  475. if (keys[1].fmr_physical >= eofs)
  476. keys[1].fmr_physical = eofs - 1;
  477. if (keys[1].fmr_physical < keys[0].fmr_physical)
  478. return 0;
  479. start_fsb = keys[0].fmr_physical;
  480. end_fsb = keys[1].fmr_physical;
  481. /* Determine first and last group to examine based on start and end */
  482. ext4_get_group_no_and_offset(sb, start_fsb, &start_ag, &first_cluster);
  483. ext4_get_group_no_and_offset(sb, end_fsb, &end_ag, &last_cluster);
  484. /*
  485. * Convert the fsmap low/high keys to bg based keys. Initialize
  486. * low to the fsmap low key and max out the high key to the end
  487. * of the bg.
  488. */
  489. info->gfi_low = keys[0];
  490. info->gfi_low.fmr_physical = EXT4_C2B(sbi, first_cluster);
  491. info->gfi_low.fmr_length = 0;
  492. memset(&info->gfi_high, 0xFF, sizeof(info->gfi_high));
  493. /* Assemble a list of all the fixed-location metadata. */
  494. error = ext4_getfsmap_find_fixed_metadata(sb, &info->gfi_meta_list);
  495. if (error)
  496. goto err;
  497. /* Query each bg */
  498. for (info->gfi_agno = start_ag;
  499. info->gfi_agno <= end_ag;
  500. info->gfi_agno++) {
  501. /*
  502. * Set the bg high key from the fsmap high key if this
  503. * is the last bg that we're querying.
  504. */
  505. if (info->gfi_agno == end_ag) {
  506. info->gfi_high = keys[1];
  507. info->gfi_high.fmr_physical = EXT4_C2B(sbi,
  508. last_cluster);
  509. info->gfi_high.fmr_length = 0;
  510. }
  511. trace_ext4_fsmap_low_key(sb, info->gfi_dev, info->gfi_agno,
  512. info->gfi_low.fmr_physical,
  513. info->gfi_low.fmr_length,
  514. info->gfi_low.fmr_owner);
  515. trace_ext4_fsmap_high_key(sb, info->gfi_dev, info->gfi_agno,
  516. info->gfi_high.fmr_physical,
  517. info->gfi_high.fmr_length,
  518. info->gfi_high.fmr_owner);
  519. error = ext4_mballoc_query_range(sb, info->gfi_agno,
  520. EXT4_B2C(sbi, info->gfi_low.fmr_physical),
  521. EXT4_B2C(sbi, info->gfi_high.fmr_physical),
  522. ext4_getfsmap_meta_helper,
  523. ext4_getfsmap_datadev_helper, info);
  524. if (error)
  525. goto err;
  526. /*
  527. * Set the bg low key to the start of the bg prior to
  528. * moving on to the next bg.
  529. */
  530. if (info->gfi_agno == start_ag)
  531. memset(&info->gfi_low, 0, sizeof(info->gfi_low));
  532. }
  533. /* Do we have a retained free extent? */
  534. if (info->gfi_lastfree.fmr_owner) {
  535. error = ext4_getfsmap_helper(sb, info, &info->gfi_lastfree);
  536. if (error)
  537. goto err;
  538. }
  539. /*
  540. * The dummy record below will cause ext4_getfsmap_helper() to report
  541. * any allocated blocks at the end of the range.
  542. */
  543. irec.fmr_device = 0;
  544. irec.fmr_physical = end_fsb + 1;
  545. irec.fmr_length = 0;
  546. irec.fmr_owner = EXT4_FMR_OWN_FREE;
  547. irec.fmr_flags = 0;
  548. info->gfi_last = true;
  549. error = ext4_getfsmap_helper(sb, info, &irec);
  550. if (error)
  551. goto err;
  552. err:
  553. ext4_getfsmap_free_fixed_metadata(&info->gfi_meta_list);
  554. return error;
  555. }
  556. /* Do we recognize the device? */
  557. static bool ext4_getfsmap_is_valid_device(struct super_block *sb,
  558. struct ext4_fsmap *fm)
  559. {
  560. if (fm->fmr_device == 0 || fm->fmr_device == UINT_MAX ||
  561. fm->fmr_device == new_encode_dev(sb->s_bdev->bd_dev))
  562. return true;
  563. if (EXT4_SB(sb)->s_journal_bdev_file &&
  564. fm->fmr_device ==
  565. new_encode_dev(file_bdev(EXT4_SB(sb)->s_journal_bdev_file)->bd_dev))
  566. return true;
  567. return false;
  568. }
  569. /* Ensure that the low key is less than the high key. */
  570. static bool ext4_getfsmap_check_keys(struct ext4_fsmap *low_key,
  571. struct ext4_fsmap *high_key)
  572. {
  573. if (low_key->fmr_device > high_key->fmr_device)
  574. return false;
  575. if (low_key->fmr_device < high_key->fmr_device)
  576. return true;
  577. if (low_key->fmr_physical > high_key->fmr_physical)
  578. return false;
  579. if (low_key->fmr_physical < high_key->fmr_physical)
  580. return true;
  581. if (low_key->fmr_owner > high_key->fmr_owner)
  582. return false;
  583. if (low_key->fmr_owner < high_key->fmr_owner)
  584. return true;
  585. return false;
  586. }
  587. #define EXT4_GETFSMAP_DEVS 2
  588. /*
  589. * Get filesystem's extents as described in head, and format for
  590. * output. Calls formatter to fill the user's buffer until all
  591. * extents are mapped, until the passed-in head->fmh_count slots have
  592. * been filled, or until the formatter short-circuits the loop, if it
  593. * is tracking filled-in extents on its own.
  594. *
  595. * Key to Confusion
  596. * ----------------
  597. * There are multiple levels of keys and counters at work here:
  598. * _fsmap_head.fmh_keys -- low and high fsmap keys passed in;
  599. * these reflect fs-wide block addrs.
  600. * dkeys -- fmh_keys used to query each device;
  601. * these are fmh_keys but w/ the low key
  602. * bumped up by fmr_length.
  603. * _getfsmap_info.gfi_next_fsblk-- next fs block we expect to see; this
  604. * is how we detect gaps in the fsmap
  605. * records and report them.
  606. * _getfsmap_info.gfi_low/high -- per-bg low/high keys computed from
  607. * dkeys; used to query the free space.
  608. */
  609. int ext4_getfsmap(struct super_block *sb, struct ext4_fsmap_head *head,
  610. ext4_fsmap_format_t formatter, void *arg)
  611. {
  612. struct ext4_fsmap dkeys[2]; /* per-dev keys */
  613. struct ext4_getfsmap_dev handlers[EXT4_GETFSMAP_DEVS];
  614. struct ext4_getfsmap_info info = { NULL };
  615. int i;
  616. int error = 0;
  617. if (head->fmh_iflags & ~FMH_IF_VALID)
  618. return -EINVAL;
  619. if (!ext4_getfsmap_is_valid_device(sb, &head->fmh_keys[0]) ||
  620. !ext4_getfsmap_is_valid_device(sb, &head->fmh_keys[1]))
  621. return -EINVAL;
  622. head->fmh_entries = 0;
  623. /* Set up our device handlers. */
  624. memset(handlers, 0, sizeof(handlers));
  625. handlers[0].gfd_dev = new_encode_dev(sb->s_bdev->bd_dev);
  626. handlers[0].gfd_fn = ext4_getfsmap_datadev;
  627. if (EXT4_SB(sb)->s_journal_bdev_file) {
  628. handlers[1].gfd_dev = new_encode_dev(
  629. file_bdev(EXT4_SB(sb)->s_journal_bdev_file)->bd_dev);
  630. handlers[1].gfd_fn = ext4_getfsmap_logdev;
  631. }
  632. sort(handlers, EXT4_GETFSMAP_DEVS, sizeof(struct ext4_getfsmap_dev),
  633. ext4_getfsmap_dev_compare, NULL);
  634. /*
  635. * To continue where we left off, we allow userspace to use the
  636. * last mapping from a previous call as the low key of the next.
  637. * This is identified by a non-zero length in the low key. We
  638. * have to increment the low key in this scenario to ensure we
  639. * don't return the same mapping again, and instead return the
  640. * very next mapping.
  641. *
  642. * Bump the physical offset as there can be no other mapping for
  643. * the same physical block range.
  644. */
  645. dkeys[0] = head->fmh_keys[0];
  646. dkeys[0].fmr_physical += dkeys[0].fmr_length;
  647. dkeys[0].fmr_owner = 0;
  648. dkeys[0].fmr_length = 0;
  649. memset(&dkeys[1], 0xFF, sizeof(struct ext4_fsmap));
  650. if (!ext4_getfsmap_check_keys(dkeys, &head->fmh_keys[1]))
  651. return -EINVAL;
  652. info.gfi_next_fsblk = head->fmh_keys[0].fmr_physical +
  653. head->fmh_keys[0].fmr_length;
  654. info.gfi_formatter = formatter;
  655. info.gfi_format_arg = arg;
  656. info.gfi_head = head;
  657. /* For each device we support... */
  658. for (i = 0; i < EXT4_GETFSMAP_DEVS; i++) {
  659. /* Is this device within the range the user asked for? */
  660. if (!handlers[i].gfd_fn)
  661. continue;
  662. if (head->fmh_keys[0].fmr_device > handlers[i].gfd_dev)
  663. continue;
  664. if (head->fmh_keys[1].fmr_device < handlers[i].gfd_dev)
  665. break;
  666. /*
  667. * If this device number matches the high key, we have
  668. * to pass the high key to the handler to limit the
  669. * query results. If the device number exceeds the
  670. * low key, zero out the low key so that we get
  671. * everything from the beginning.
  672. */
  673. if (handlers[i].gfd_dev == head->fmh_keys[1].fmr_device)
  674. dkeys[1] = head->fmh_keys[1];
  675. if (handlers[i].gfd_dev > head->fmh_keys[0].fmr_device)
  676. memset(&dkeys[0], 0, sizeof(struct ext4_fsmap));
  677. info.gfi_dev = handlers[i].gfd_dev;
  678. info.gfi_last = false;
  679. info.gfi_agno = -1;
  680. error = handlers[i].gfd_fn(sb, dkeys, &info);
  681. if (error)
  682. break;
  683. info.gfi_next_fsblk = 0;
  684. }
  685. head->fmh_oflags = FMH_OF_DEV_T;
  686. return error;
  687. }