diff options
| -rw-r--r-- | plugins/efi_resolver/include/ModuleType.h | 6 |
1 files changed, 3 insertions, 3 deletions
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; |