diff --git a/Include/Library/OcMachoLib.h b/Include/Library/OcMachoLib.h index 16e59817..31b59d4a 100644 --- a/Include/Library/OcMachoLib.h +++ b/Include/Library/OcMachoLib.h @@ -56,6 +56,17 @@ MachoGetMachHeader64 ( IN OUT OC_MACHO_CONTEXT *Context ); +/** + Returns the Mach-O's file size. + + @param[in,out] Context Context of the Mach-O. + +**/ +UINTN +MachoGetFileSize ( + IN OUT OC_MACHO_CONTEXT *Context + ); + /** Returns the last virtual address of a Mach-O. diff --git a/Library/OcMachoLib/Header.c b/Library/OcMachoLib/Header.c index a83a0d05..0418f51b 100644 --- a/Library/OcMachoLib/Header.c +++ b/Library/OcMachoLib/Header.c @@ -51,6 +51,23 @@ MachoGetMachHeader64 ( return (MACH_HEADER_64 *)Context->MachHeader; } +/** + Returns the Mach-O's file size. + + @param[in,out] Context Context of the Mach-O. + +**/ +UINTN +MachoGetFileSize ( + IN OUT OC_MACHO_CONTEXT *Context + ) +{ + ASSERT (Context != NULL); + ASSERT (Context->FileSize != 0); + + return Context->FileSize; +} + /** Initializes a Mach-O Context.