mirror of
https://github.com/dobin/avred
synced 2026-06-08 13:54:13 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from plugins.model import Plugin, BaseFile
|
|
from model.model_base import Scanner, OutflankPatch
|
|
from model.model_data import Match
|
|
from model.model_verification import MatchConclusion
|
|
from plugins.pe.file_pe import FilePe
|
|
from typing import List, Tuple, Set
|
|
|
|
from plugins.pe.analyzer_pe import analyzeFilePe
|
|
from plugins.pe.augment_pe import augmentFilePe
|
|
from plugins.pe.outflank_pe import outflankPe
|
|
from plugins.pe.file_pe import FilePe
|
|
|
|
|
|
class PluginPe(Plugin):
|
|
|
|
def makeFile(self, filepath: str):
|
|
file = FilePe()
|
|
file.loadFromFile(filepath)
|
|
return file
|
|
|
|
|
|
def analyzeFile(self, file: BaseFile, scanner: Scanner, analyzerOptions={}):
|
|
# We use the simple PE analyzer
|
|
return analyzeFilePe(file, scanner, analyzerOptions)
|
|
|
|
|
|
def augmentFile(self, file: BaseFile, matches: List[Match]) -> str:
|
|
return augmentFilePe(file, matches)
|
|
|
|
|
|
def outflankFile(
|
|
self, file: BaseFile, matches: List[Match], matchConclusion: MatchConclusion, scanner: Scanner = None
|
|
) -> List[OutflankPatch]:
|
|
return outflankPe(file, matches, matchConclusion, scanner)
|