From c4efe0ab81183d4e75a36557a00acb2086935947 Mon Sep 17 00:00:00 2001 From: Michael Krasnitski Date: Tue, 14 Jan 2025 13:07:52 -0500 Subject: Fix running in headless mode --- plugins/efi_resolver/include/ModuleType.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'plugins') diff --git a/plugins/efi_resolver/include/ModuleType.h b/plugins/efi_resolver/include/ModuleType.h index 187dcc36..2bad23b1 100644 --- a/plugins/efi_resolver/include/ModuleType.h +++ b/plugins/efi_resolver/include/ModuleType.h @@ -13,10 +13,10 @@ enum EFIModuleType static inline EFIModuleType identifyModuleType(BinaryView* bv) { - std::string viewType = bv->GetCurrentView(); - if (viewType == "Linear:PE") + FileMetadata* file = bv->GetFile(); + if (file->GetViewOfType("PE")) return DXE; - else if (viewType == "Linear:TE") + else if (file->GetViewOfType("TE")) return PEI; else return UNKNOWN; -- cgit v1.3.1