From 60162d82a2bd33f8df32b6743ae996b62290dbbb Mon Sep 17 00:00:00 2001 From: Michael Chang <mchang@suse.com> Date: Mon, 7 Feb 2022 20:44:40 +0800 Subject: [PATCH 119/272] Add grub_disk_write_tail helper function This helps in writing data to partition where the end of buffer is aligned to end of partition. Signed-off-by: Michael Chang <mchang@suse.com> --- grub-core/lib/disk.c | 18 ++++++++++++++++++ include/grub/disk.h | 3 +++ 2 files changed, 21 insertions(+) diff --git a/grub-core/lib/disk.c b/grub-core/lib/disk.c index 123993dd4..b6076faf1 100644 --- a/grub-core/lib/disk.c +++ b/grub-core/lib/disk.c @@ -51,6 +51,24 @@ grub_disk_cache_invalidate (unsigned long dev_id, unsigned long disk_id, } } +grub_err_t +grub_disk_write_tail (grub_disk_t disk, grub_size_t size, const void *buf) +{ + grub_partition_t part; + grub_disk_addr_t sector; + grub_off_t offset; + + if (!disk->partition) + return GRUB_ERR_NONE; + + part = disk->partition; + sector = part->len; + sector -= (size + GRUB_DISK_SECTOR_SIZE - 1) >> GRUB_DISK_SECTOR_BITS; + offset = size & (GRUB_DISK_SECTOR_SIZE - 1); + + return grub_disk_write (disk, sector, offset, size, buf); +} + grub_err_t grub_disk_write (grub_disk_t disk, grub_disk_addr_t sector, grub_off_t offset, grub_size_t size, const void *buf) diff --git a/include/grub/disk.h b/include/grub/disk.h index fbf23df7f..0858ee40e 100644 --- a/include/grub/disk.h +++ b/include/grub/disk.h @@ -255,6 +255,9 @@ grub_err_t EXPORT_FUNC(grub_disk_read) (grub_disk_t disk, grub_off_t offset, grub_size_t size, void *buf); +grub_err_t grub_disk_write_tail (grub_disk_t disk, + grub_size_t size, + const void *buf); grub_err_t grub_disk_write (grub_disk_t disk, grub_disk_addr_t sector, grub_off_t offset, -- 2.41.0