From dacd1f91a6951dbc74869f54ae3b85151e56a643 Mon Sep 17 00:00:00 2001
From: Download-Fritz <download-fritz@outlook.com>
Date: Sun, 7 Apr 2019 13:00:01 +0200
Subject: [PATCH] OcAppleDiskImageLib: Add Chunklist-aided VerifyData API

---
 Include/Library/OcAppleDiskImageLib.h           |  8 ++++++++
 .../OcAppleDiskImageLib/OcAppleDiskImageLib.c   | 17 +++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/Include/Library/OcAppleDiskImageLib.h b/Include/Library/OcAppleDiskImageLib.h
index 90ad4dd7..31618cab 100644
--- a/Include/Library/OcAppleDiskImageLib.h
+++ b/Include/Library/OcAppleDiskImageLib.h
@@ -15,6 +15,8 @@
 
 #include <IndustryStandard/AppleDiskImage.h>
 
+#include <Library/OcAppleChunklistLib.h>
+
 //
 // Disk image context.
 //
@@ -45,6 +47,12 @@ OcAppleDiskImageFreeContext(
   IN OC_APPLE_DISK_IMAGE_CONTEXT *Context
   );
 
+BOOLEAN
+OcAppleDiskImageVerifyData (
+  IN OUT OC_APPLE_DISK_IMAGE_CONTEXT  *Context,
+  IN OUT OC_APPLE_CHUNKLIST_CONTEXT   *ChunklistContext
+  );
+
 BOOLEAN
 OcAppleDiskImageRead (
   IN  OC_APPLE_DISK_IMAGE_CONTEXT  *Context,
diff --git a/Library/OcAppleDiskImageLib/OcAppleDiskImageLib.c b/Library/OcAppleDiskImageLib/OcAppleDiskImageLib.c
index 73351e96..c243be4f 100644
--- a/Library/OcAppleDiskImageLib/OcAppleDiskImageLib.c
+++ b/Library/OcAppleDiskImageLib/OcAppleDiskImageLib.c
@@ -16,6 +16,7 @@
 #include <Library/BaseMemoryLib.h>
 #include <Library/DebugLib.h>
 #include <Library/MemoryAllocationLib.h>
+#include <Library/OcAppleChunklistLib.h>
 #include <Library/OcAppleDiskImageLib.h>
 #include <Library/OcCompressionLib.h>
 #include <Library/OcGuardLib.h>
@@ -170,6 +171,22 @@ OcAppleDiskImageInitializeContext (
   return TRUE;
 }
 
+BOOLEAN
+OcAppleDiskImageVerifyData (
+  IN OUT OC_APPLE_DISK_IMAGE_CONTEXT  *Context,
+  IN OUT OC_APPLE_CHUNKLIST_CONTEXT   *ChunklistContext
+  )
+{
+  ASSERT (Context != NULL);
+  ASSERT (ChunklistContext != NULL);
+
+  return OcAppleChunklistVerifyData (
+           ChunklistContext,
+           Context->Buffer,
+           Context->Length
+           );
+}
+
 VOID
 OcAppleDiskImageFreeContext (
   IN OC_APPLE_DISK_IMAGE_CONTEXT  *Context