Skip to content

Commit c4efe0a

Browse files
mkrasnitskizznop
authored andcommitted
Fix running in headless mode
1 parent fac7ba8 commit c4efe0a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

plugins/efi_resolver/include/ModuleType.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ enum EFIModuleType
1313

1414
static inline EFIModuleType identifyModuleType(BinaryView* bv)
1515
{
16-
std::string viewType = bv->GetCurrentView();
17-
if (viewType == "Linear:PE")
16+
FileMetadata* file = bv->GetFile();
17+
if (file->GetViewOfType("PE"))
1818
return DXE;
19-
else if (viewType == "Linear:TE")
19+
else if (file->GetViewOfType("TE"))
2020
return PEI;
2121
else
2222
return UNKNOWN;

0 commit comments

Comments
 (0)