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