Source code for angr.knowledge_plugins.plugin

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from angr.knowledge_base.knowledge_base import KnowledgeBase

default_plugins = {}


class KnowledgeBasePlugin:
    def __init__(self, kb: "KnowledgeBase"):
        # This call is needed, because some Plugins like TypesStore inherit from KnowledgeBasePlugin and something else
        # So we still need to call the constructor of the other class.
        super().__init__()
        self._kb = kb

    def copy(self):
        raise NotImplementedError

    @staticmethod
    def register_default(name, cls):
        if name in default_plugins:
            raise Exception(f"{default_plugins[name]} is already set as the default for {name}")
        default_plugins[name] = cls