mirror of
https://github.com/dobin/avred
synced 2026-06-08 13:54:13 +00:00
Dobin Rutishauser
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from model.plugin_model import Plugin
|
|
from model.model_base import Scanner, OutflankPatch, ScanInfo
|
|
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
|
|
from reducer import Reducer
|
|
from model.file_model import BaseFile
|
|
|
|
|
|
class PluginPe(Plugin):
|
|
|
|
def makeFile(self, filepath: str):
|
|
file = FilePe()
|
|
file.loadFromFile(filepath)
|
|
return file
|
|
|
|
|
|
def analyzeFile(self, file: BaseFile, scanner: Scanner, reducer: Reducer, analyzerOptions={}) -> Tuple[Match, ScanInfo]:
|
|
# We use the simple PE analyzer
|
|
return analyzeFilePe(file, scanner, reducer, 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)
|