btrfs: Add RAID 6 recovery for a btrfs filesystem

Add the RAID 6 recovery, in order to use a RAID 6 filesystem even if some
disks (up to two) are missing. This code use the md RAID 6 code already
present in grub.

Signed-off-by: Goffredo Baroncelli <kreijack@inwind.it>
Reviewed-by: Daniel Kiper <daniel.kiper@oracle.com>
This commit is contained in:
Goffredo Baroncelli 2018-10-22 19:29:39 +02:00 committed by Vincent Batts
parent 72a2af67ff
commit 83985ac0bf

View file

@ -30,6 +30,7 @@
#include <grub/i18n.h> #include <grub/i18n.h>
#include <grub/btrfs.h> #include <grub/btrfs.h>
#include <grub/crypto.h> #include <grub/crypto.h>
#include <grub/diskfilter.h>
GRUB_MOD_LICENSE ("GPLv3+"); GRUB_MOD_LICENSE ("GPLv3+");
@ -701,11 +702,36 @@ rebuild_raid5 (char *dest, struct raid56_buffer *buffers,
} }
} }
static grub_err_t
raid6_recover_read_buffer (void *data, int disk_nr,
grub_uint64_t addr __attribute__ ((unused)),
void *dest, grub_size_t size)
{
struct raid56_buffer *buffers = data;
if (!buffers[disk_nr].data_is_valid)
return grub_errno = GRUB_ERR_READ_ERROR;
grub_memcpy(dest, buffers[disk_nr].buf, size);
return grub_errno = GRUB_ERR_NONE;
}
static void
rebuild_raid6 (struct raid56_buffer *buffers, grub_uint64_t nstripes,
grub_uint64_t csize, grub_uint64_t parities_pos, void *dest,
grub_uint64_t stripen)
{
grub_raid6_recover_gen (buffers, nstripes, stripen, parities_pos,
dest, 0, csize, 0, raid6_recover_read_buffer);
}
static grub_err_t static grub_err_t
raid56_read_retry (struct grub_btrfs_data *data, raid56_read_retry (struct grub_btrfs_data *data,
struct grub_btrfs_chunk_item *chunk, struct grub_btrfs_chunk_item *chunk,
grub_uint64_t stripe_offset, grub_uint64_t stripe_offset, grub_uint64_t stripen,
grub_uint64_t csize, void *buf) grub_uint64_t csize, void *buf, grub_uint64_t parities_pos)
{ {
struct raid56_buffer *buffers; struct raid56_buffer *buffers;
grub_uint64_t nstripes = grub_le_to_cpu16 (chunk->nstripes); grub_uint64_t nstripes = grub_le_to_cpu16 (chunk->nstripes);
@ -777,6 +803,14 @@ raid56_read_retry (struct grub_btrfs_data *data,
ret = GRUB_ERR_READ_ERROR; ret = GRUB_ERR_READ_ERROR;
goto cleanup; goto cleanup;
} }
else if (failed_devices > 2 && (chunk_type & GRUB_BTRFS_CHUNK_TYPE_RAID6))
{
grub_dprintf ("btrfs", "not enough disks for RAID 6: total %" PRIuGRUB_UINT64_T
", missing %" PRIuGRUB_UINT64_T "\n",
nstripes, failed_devices);
ret = GRUB_ERR_READ_ERROR;
goto cleanup;
}
else else
grub_dprintf ("btrfs", "enough disks for RAID 5: total %" grub_dprintf ("btrfs", "enough disks for RAID 5: total %"
PRIuGRUB_UINT64_T ", missing %" PRIuGRUB_UINT64_T "\n", PRIuGRUB_UINT64_T ", missing %" PRIuGRUB_UINT64_T "\n",
@ -786,7 +820,7 @@ raid56_read_retry (struct grub_btrfs_data *data,
if (chunk_type & GRUB_BTRFS_CHUNK_TYPE_RAID5) if (chunk_type & GRUB_BTRFS_CHUNK_TYPE_RAID5)
rebuild_raid5 (buf, buffers, nstripes, csize); rebuild_raid5 (buf, buffers, nstripes, csize);
else else
grub_dprintf ("btrfs", "called rebuild_raid6(), NOT IMPLEMENTED\n"); rebuild_raid6 (buffers, nstripes, csize, parities_pos, buf, stripen);
ret = GRUB_ERR_NONE; ret = GRUB_ERR_NONE;
cleanup: cleanup:
@ -876,9 +910,11 @@ grub_btrfs_read_logical (struct grub_btrfs_data *data, grub_disk_addr_t addr,
unsigned redundancy = 1; unsigned redundancy = 1;
unsigned i, j; unsigned i, j;
int is_raid56; int is_raid56;
grub_uint64_t parities_pos = 0;
is_raid56 = !!(grub_le_to_cpu64 (chunk->type) & is_raid56 = !!(grub_le_to_cpu64 (chunk->type) &
GRUB_BTRFS_CHUNK_TYPE_RAID5); (GRUB_BTRFS_CHUNK_TYPE_RAID5 |
GRUB_BTRFS_CHUNK_TYPE_RAID6));
if (grub_le_to_cpu64 (chunk->size) <= off) if (grub_le_to_cpu64 (chunk->size) <= off)
{ {
@ -1027,6 +1063,15 @@ grub_btrfs_read_logical (struct grub_btrfs_data *data, grub_disk_addr_t addr,
*/ */
grub_divmod64 (high + stripen, nstripes, &stripen); grub_divmod64 (high + stripen, nstripes, &stripen);
/*
* parities_pos is equal to ((high - nparities) % nstripes)
* (see the diagram above). However, (high - nparities) can
* be negative, e.g. when high == 0, leading to an incorrect
* results. (high + nstripes - nparities) is always positive and
* modulo nstripes is equal to ((high - nparities) % nstripes).
*/
grub_divmod64 (high + nstripes - nparities, nstripes, &parities_pos);
stripe_offset = chunk_stripe_length * high + low; stripe_offset = chunk_stripe_length * high + low;
csize = chunk_stripe_length - low; csize = chunk_stripe_length - low;
@ -1067,7 +1112,7 @@ grub_btrfs_read_logical (struct grub_btrfs_data *data, grub_disk_addr_t addr,
grub_errno = GRUB_ERR_NONE; grub_errno = GRUB_ERR_NONE;
if (err) if (err)
err = raid56_read_retry (data, chunk, stripe_offset, err = raid56_read_retry (data, chunk, stripe_offset,
csize, buf); stripen, csize, buf, parities_pos);
} }
else else
for (i = 0; i < redundancy; i++) for (i = 0; i < redundancy; i++)