1643
static int next_csum(struct btrfs_root *root,
1644
struct extent_buffer **leaf,
1645
struct btrfs_path *path,
1650
struct btrfs_key *key)
1653
struct btrfs_root *csum_root = root->fs_info->csum_root;
1654
struct btrfs_csum_item *csum_item;
1655
u32 blocksize = root->sectorsize;
1656
u16 csum_size = btrfs_super_csum_size(root->fs_info->super_copy);
1657
int csums_in_item = btrfs_item_size_nr(*leaf, *slot) / csum_size;
1659
if (*csum_offset >= csums_in_item) {
1662
if (*slot >= btrfs_header_nritems(*leaf)) {
1663
ret = btrfs_next_leaf(csum_root, path);
1668
*leaf = path->nodes[0];
1669
*slot = path->slots[0];
1671
btrfs_item_key_to_cpu(*leaf, key, *slot);
1674
if (key->offset + (*csum_offset) * blocksize >= end)
1676
csum_item = btrfs_item_ptr(*leaf, *slot, struct btrfs_csum_item);
1677
csum_item = (struct btrfs_csum_item *)((unsigned char *)csum_item
1678
+ (*csum_offset) * csum_size);
1679
read_extent_buffer(*leaf, tree_csum,
1680
(unsigned long)csum_item, csum_size);
1684
static u64 calc_data_offset(struct btrfs_key *key,
1685
struct chunk_record *chunk,
1691
int logical_stripe_nr;
1693
int nr_data_stripes;
1695
data_offset = key->offset + csum_offset * blocksize - chunk->offset;
1696
nr_data_stripes = chunk->num_stripes;
1698
if (chunk->type_flags & BTRFS_BLOCK_GROUP_RAID5)
1699
nr_data_stripes -= 1;
1700
else if (chunk->type_flags & BTRFS_BLOCK_GROUP_RAID6)
1701
nr_data_stripes -= 2;
1703
logical_stripe_nr = data_offset / chunk->stripe_len;
1704
dev_stripe_nr = logical_stripe_nr / nr_data_stripes;
1706
data_offset -= logical_stripe_nr * chunk->stripe_len;
1707
data_offset += dev_stripe_nr * chunk->stripe_len;
1709
return dev_offset + data_offset;
1712
static int check_one_csum(int fd, u64 start, u32 len, u32 tree_csum)
1716
u32 csum_result = ~(u32)0;
1721
ret = pread64(fd, data, len, start);
1722
if (ret < 0 || ret != len) {
1727
csum_result = btrfs_csum_data(NULL, data, csum_result, len);
1728
btrfs_csum_final(csum_result, (char *)&csum_result);
1729
if (csum_result != tree_csum)
1736
static u64 item_end_offset(struct btrfs_root *root, struct btrfs_key *key,
1737
struct extent_buffer *leaf, int slot) {
1738
u32 blocksize = root->sectorsize;
1739
u16 csum_size = btrfs_super_csum_size(root->fs_info->super_copy);
1741
u64 offset = btrfs_item_size_nr(leaf, slot);
1742
offset /= csum_size;
1743
offset *= blocksize;
1744
offset += key->offset;
1749
static int insert_stripe(struct list_head *devexts,
1750
struct recover_control *rc,
1751
struct chunk_record *chunk,
1753
struct device_extent_record *devext;
1754
struct btrfs_device *dev;
1756
devext = list_entry(devexts->next, struct device_extent_record,
1758
dev = btrfs_find_device_by_devid(rc->fs_devices, devext->objectid,
1762
BUG_ON(btrfs_find_device_by_devid(rc->fs_devices, devext->objectid,
1765
chunk->stripes[index].devid = devext->objectid;
1766
chunk->stripes[index].offset = devext->offset;
1767
memcpy(chunk->stripes[index].dev_uuid, dev->uuid, BTRFS_UUID_SIZE);
1769
list_move(&devext->chunk_list, &chunk->dextents);
1774
#define EQUAL_STRIPE (1 << 0)
1776
static int rebuild_raid_data_chunk_stripes(struct recover_control *rc,
1777
struct btrfs_root *root,
1778
struct chunk_record *chunk,
1784
struct btrfs_path path;
1785
struct btrfs_key prev_key;
1786
struct btrfs_key key;
1787
struct btrfs_root *csum_root;
1788
struct extent_buffer *leaf;
1789
struct device_extent_record *devext;
1790
struct device_extent_record *next;
1791
struct btrfs_device *dev;
1792
u64 start = chunk->offset;
1793
u64 end = start + chunk->stripe_len;
1794
u64 chunk_end = chunk->offset + chunk->length;
1795
u64 csum_offset = 0;
1797
u32 blocksize = root->sectorsize;
1800
int num_unordered = 0;
1801
LIST_HEAD(unordered);
1802
LIST_HEAD(candidates);
1804
csum_root = root->fs_info->csum_root;
1805
btrfs_init_path(&path);
1806
list_splice_init(&chunk->dextents, &candidates);
1808
if (list_is_last(candidates.next, &candidates))
1811
key.objectid = BTRFS_EXTENT_CSUM_OBJECTID;
1812
key.type = BTRFS_EXTENT_CSUM_KEY;
1815
ret = btrfs_search_slot(NULL, csum_root, &key, &path, 0, 0);
1817
fprintf(stderr, "Search csum failed(%d)\n", ret);
1820
leaf = path.nodes[0];
1821
slot = path.slots[0];
1823
if (slot >= btrfs_header_nritems(leaf)) {
1824
ret = btrfs_next_leaf(csum_root, &path);
1827
"Walk tree failed(%d)\n", ret);
1829
} else if (ret > 0) {
1830
slot = btrfs_header_nritems(leaf) - 1;
1831
btrfs_item_key_to_cpu(leaf, &key, slot);
1832
if (item_end_offset(root, &key, leaf, slot)
1834
csum_offset = start - key.offset;
1835
csum_offset /= blocksize;
1840
leaf = path.nodes[0];
1841
slot = path.slots[0];
1843
btrfs_item_key_to_cpu(leaf, &key, slot);
1844
ret = btrfs_previous_item(csum_root, &path, 0,
1845
BTRFS_EXTENT_CSUM_KEY);
1849
if (key.offset >= end)
1854
leaf = path.nodes[0];
1855
slot = path.slots[0];
1857
btrfs_item_key_to_cpu(leaf, &prev_key, slot);
1858
if (item_end_offset(root, &prev_key, leaf, slot) > start) {
1859
csum_offset = start - prev_key.offset;
1860
csum_offset /= blocksize;
1861
btrfs_item_key_to_cpu(leaf, &key, slot);
1863
if (key.offset >= end)
1867
if (key.offset + csum_offset * blocksize > chunk_end)
1871
ret = next_csum(root, &leaf, &path, &slot, &csum_offset, &tree_csum,
1874
fprintf(stderr, "Fetch csum failed\n");
1876
} else if (ret == 1) {
1877
list_for_each_entry(devext, &unordered, chunk_list)
1879
if (!(*flags & EQUAL_STRIPE))
1880
*flags |= EQUAL_STRIPE;
1882
} else if (ret == 2)
1885
list_for_each_entry_safe(devext, next, &candidates, chunk_list) {
1886
data_offset = calc_data_offset(&key, chunk, devext->offset,
1887
csum_offset, blocksize);
1888
dev = btrfs_find_device_by_devid(rc->fs_devices,
1889
devext->objectid, 0);
1894
BUG_ON(btrfs_find_device_by_devid(rc->fs_devices,
1895
devext->objectid, 1));
1897
ret = check_one_csum(dev->fd, data_offset, blocksize,
1902
list_move(&devext->chunk_list, &unordered);
1905
if (list_empty(&candidates)) {
1906
list_for_each_entry(devext, &unordered, chunk_list)
1908
if (chunk->type_flags & BTRFS_BLOCK_GROUP_RAID6
1909
&& num_unordered == 2) {
1910
list_splice_init(&unordered, &chunk->dextents);
1911
btrfs_release_path(&path);
1919
if (list_is_last(candidates.next, &candidates)) {
1920
index = btrfs_calc_stripe_index(chunk,
1921
key.offset + csum_offset * blocksize);
1922
if (chunk->stripes[index].devid)
1924
ret = insert_stripe(&candidates, rc, chunk, index);
1932
start = btrfs_next_stripe_logical_offset(chunk, start);
1933
end = min(start + chunk->stripe_len, chunk_end);
1934
list_splice_init(&unordered, &candidates);
1935
btrfs_release_path(&path);
1937
if (end < chunk_end)
1941
list_splice_init(&candidates, &unordered);
1942
list_for_each_entry(devext, &unordered, chunk_list)
1944
if (num_unordered == 1) {
1945
for (i = 0; i < chunk->num_stripes; i++) {
1946
if (!chunk->stripes[i].devid) {
1951
ret = insert_stripe(&unordered, rc, chunk, index);
1955
if ((num_unordered == 2 && chunk->type_flags
1956
& BTRFS_BLOCK_GROUP_RAID5)
1957
|| (num_unordered == 3 && chunk->type_flags
1958
& BTRFS_BLOCK_GROUP_RAID6)) {
1959
for (i = 0; i < chunk->num_stripes; i++) {
1960
if (!chunk->stripes[i].devid) {
1961
ret = insert_stripe(&unordered, rc,
1970
ret = !!ret || (list_empty(&unordered) ? 0 : 1);
1971
list_splice_init(&candidates, &chunk->dextents);
1972
list_splice_init(&unordered, &chunk->dextents);
1973
btrfs_release_path(&path);
1978
static int btrfs_rebuild_ordered_data_chunk_stripes(struct recover_control *rc,
1979
struct btrfs_root *root)
1981
struct chunk_record *chunk;
1982
struct chunk_record *next;
1987
list_for_each_entry_safe(chunk, next, &rc->unrepaired_chunks, list) {
1988
if ((chunk->type_flags & BTRFS_BLOCK_GROUP_DATA)
1989
&& (chunk->type_flags & BTRFS_ORDERED_RAID)) {
1991
err = rebuild_raid_data_chunk_stripes(rc, root, chunk,
1994
list_move(&chunk->list, &rc->bad_chunks);
1995
if (flags & EQUAL_STRIPE)
1997
"Failure: too many equal stripes in chunk[%llu %llu]\n",
1998
chunk->offset, chunk->length);
2002
list_move(&chunk->list, &rc->good_chunks);
1568
2008
static int btrfs_recover_chunks(struct recover_control *rc)
1570
2010
struct chunk_record *chunk;