fix: imports and add tests

mandiant · Apr 7, 2022 · 7507917 · 7507917
1 parent 1614ed3
commit 7507917
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -4,7 +4,8 @@ on:
   push:
     branches: [ master, dotnet-main ]
   pull_request:
-    branches: [ master, dotnet-main ]
+    # TODO remove dotnet-main-2
+    branches: [ master, dotnet-main, dotnet-main-2 ]
 
 # save workspaces to speed up testing
 env:

diff --git a/capa/features/extractors/dnfile_.py b/capa/features/extractors/dnfile_.py
@@ -68,6 +68,7 @@ def get_base_address(self) -> int:
 
     def get_entry_point(self) -> int:
         # self.pe.net.Flags.CLT_NATIVE_ENTRYPOINT
+        #  True: native EP: Token
         #  False: managed EP: RVA
         return self.pe.net.struct.EntryPointTokenOrRva