Yapf: Minor reformats for recent plugins

This commit is contained in:
Mike Auty
2020-08-03 16:33:10 +01:00
parent 2e6dc851d3
commit ff32383f82
4 changed files with 36 additions and 42 deletions
+19 -18
View File
@@ -15,29 +15,27 @@ from volatility.framework.renderers import format_hints
vollog = logging.getLogger(__name__)
class tty_check(plugins.PluginInterface):
"""Checks tty devices for hooks"""
@classmethod
def get_requirements(cls) -> List[interfaces.configuration.RequirementInterface]:
return [
requirements.TranslationLayerRequirement(name='primary',
description='Memory layer for the kernel',
architectures=["Intel32", "Intel64"]),
requirements.SymbolTableRequirement(
name="vmlinux", description="Linux kernel symbols"),
requirements.PluginRequirement(
name='lsmod', plugin=lsmod.Lsmod, version=(1, 0, 0))
requirements.TranslationLayerRequirement(name = 'primary',
description = 'Memory layer for the kernel',
architectures = ["Intel32", "Intel64"]),
requirements.SymbolTableRequirement(name = "vmlinux", description = "Linux kernel symbols"),
requirements.PluginRequirement(name = 'lsmod', plugin = lsmod.Lsmod, version = (1, 0, 0))
]
def _generator(self):
vmlinux = contexts.Module(
self.context, self.config['vmlinux'], self.config['primary'], 0)
vmlinux = contexts.Module(self.context, self.config['vmlinux'], self.config['primary'], 0)
modules = lsmod.Lsmod.list_modules(self.context, self.config['primary'], self.config['vmlinux'])
handlers = linux.LinuxUtilities.generate_kernel_handler_info(self.context, self.config['primary'], self.config['vmlinux'], modules)
handlers = linux.LinuxUtilities.generate_kernel_handler_info(self.context, self.config['primary'],
self.config['vmlinux'], modules)
try:
tty_drivers = vmlinux.object_from_symbol("tty_drivers")
@@ -46,7 +44,7 @@ class tty_check(plugins.PluginInterface):
if not tty_drivers:
raise TypeError(
"This plugin requires the tty_drivers structure."
"This plugin requires the tty_drivers structure."
"This structure is not present in the supplied symbol table."
"This means you are either analyzing an unsupported kernel version or that your symbol table is corrupt."
)
@@ -54,10 +52,13 @@ class tty_check(plugins.PluginInterface):
for tty in tty_drivers.to_list(vmlinux.name + constants.BANG + "tty_driver", "tty_drivers"):
try:
ttys = utility.array_of_pointers(tty.ttys.dereference(), count=tty.num, subtype=vmlinux.name + constants.BANG + "tty_struct", context=self.context)
ttys = utility.array_of_pointers(tty.ttys.dereference(),
count = tty.num,
subtype = vmlinux.name + constants.BANG + "tty_struct",
context = self.context)
except exceptions.PagedInvalidAddressException:
continue
for tty_dev in ttys:
if tty_dev == 0:
@@ -70,7 +71,7 @@ class tty_check(plugins.PluginInterface):
module_name, symbol_name = linux.LinuxUtilities.lookup_module_address(self.context, handlers, recv_buf)
yield (0, (name, format_hints.Hex(recv_buf), module_name, symbol_name))
def run(self):
return renderers.TreeGrid([("Name", str), ("Address", format_hints.Hex), ("Module", str),
("Symbol", str)], self._generator())
return renderers.TreeGrid([("Name", str), ("Address", format_hints.Hex), ("Module", str), ("Symbol", str)],
self._generator())