from abc import ABCMeta, abstractmethod import collections.abc from volatility.framework import validity __author__ = 'mike' NAMESPACE_DIVIDER = "." def namespace_join(pathlist): return NAMESPACE_DIVIDER.join(pathlist) class ConfigurationItem(validity.ValidityRoutines): """Class to distinguish configuration elements from everything else""" def __init__(self, name, optional): validity.ValidityRoutines.__init__(self) self._type_check(name, str) if NAMESPACE_DIVIDER in name: raise ValueError("Name cannot contain the namespace divider (" + NAMESPACE_DIVIDER + ")") self._name = name self._optional = optional @property def name(self): """The name of the Option.""" return self._name @abstractmethod def validate(self, context): """Validates the currently set value""" pass @property def optional(self): """Whether the option is required for or not""" return self._optional class ConfigurationGroup(ConfigurationItem, collections.abc.Mapping): """Class to hold and provide a namespace for plugins and core options""" def __init__(self, name): ConfigurationItem.__init__(self, name, optional = False) self._namespace = {} def add_item(self, item, namespace = None): if not isinstance(item, ConfigurationItem): raise TypeError("Only ConfigurationItem objects can be added to a ConfigurationGroup") if namespace: ns_split = namespace.split(NAMESPACE_DIVIDER) if ns_split[0] not in self: self._namespace[ns_split[0]] = ConfigurationGroup(ns_split[0]) return self._namespace[ns_split[0]].add_item(item, namespace_join(ns_split[1:])) self._namespace[item.name] = item def __iter__(self): return iter(self._namespace) def __getitem__(self, item): self._type_check(item, str) item_split = item.split(NAMESPACE_DIVIDER) if len(item_split) > 1: return self._namespace[item_split[0]][namespace_join(item_split[1:])] # Let namespace produce the index error if necessary return self._namespace[item_split[0]] def get_value(self, item): """Returns the value of the requirement, not the requirement itself""" return self.get(item).value def __contains__(self, item): item_split = item.split(NAMESPACE_DIVIDER) if len(item_split) > 1: if item_split[0] in self._namespace: return namespace_join(item_split[1:]) in self._namespace[item_split[0]] else: return False return item in self._namespace def __len__(self): return len(self._namespace) def validate(self, context): """Validates the current value, which for groups cannot be set, so always returns True""" return all([self[subitem].validate(context) for subitem in self._namespace if not self[subitem].optional]) class GenericRequirement(ConfigurationItem, metaclass = ABCMeta): """Class to handle a single specific configuration option""" def __init__(self, name, description = None, default = None, optional = None): """Creates a new option""" ConfigurationItem.__init__(self, name, optional) self._default = default self._description = description self._value = None @property def value(self): """Returns the value or the default if the value is not set""" if self._value is None: return self._default return self._value @value.setter def value(self, data): """Sets the value to that of the input data""" self._value = data @property def default(self): """Returns the default value if one is set""" return self._default @property def description(self): """A short description of what the Option is designed to affect or achieve.""" return self._description @abstractmethod def validate_input(self, value, context): """Validates the value against a context Throws exceptions if the valid is invalid""" pass def validate(self, context): """Validates the currently set value""" return self.validate_input(self.value, context)