typetapper/typetapper/knowledge.py

13 lines
374 B
Python

from typing import Dict
from collections import defaultdict
import angr
import networkx
from .data import BlockInfo
class TypeTapperManager(angr.knowledge_plugins.plugin.KnowledgeBasePlugin):
def __init__(self, kb: angr.KnowledgeBase):
self.kb = kb
self.graph = networkx.DiGraph()
self.block_info: Dict[int, BlockInfo] = defaultdict(BlockInfo)