add CILFile.classes and CILFile.interfaces

This commit is contained in:
Shiz 2020-06-02 00:58:25 +02:00
parent 21d55f5a2d
commit 39f9cdaf2e
1 changed files with 15 additions and 0 deletions

View File

@ -137,6 +137,21 @@ class CLIFile:
def get_by_token(self, t):
return self.get_table_entry(t.table, t.row)
@property
def entrypoint(self):
return self.get_by_token(self.header.entrypoint)
@property
def classes(self):
return (
t for t in self.get_table(CLITableType.TypeDef)
if t.flags.semantics == CLITypeClassSemantics.Class
)
@property
def interfaces(self):
return (
t for t in self.get_table(CLITableType.TypeDef)
if t.flags.semantics == CLITypeClassSemantics.Interface
)