Skip to content

Commit

Permalink
Merge pull request #885 from eve-mem/vmayarascan
Browse files Browse the repository at this point in the history
add linux.vmayarascan
  • Loading branch information
ikelos authored Nov 23, 2023
2 parents d56297c + faa8b2e commit 7012edb
Show file tree
Hide file tree
Showing 2 changed files with 128 additions and 3 deletions.
113 changes: 113 additions & 0 deletions volatility3/framework/plugins/linux/vmayarascan.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,113 @@
# This file is Copyright 2023 Volatility Foundation and licensed under the Volatility Software License 1.0
# which is available at https://www.volatilityfoundation.org/license/vsl-v1.0
#

from typing import Iterable, List, Tuple

from volatility3.framework import interfaces, renderers
from volatility3.framework.configuration import requirements
from volatility3.framework.renderers import format_hints
from volatility3.plugins import yarascan
from volatility3.plugins.linux import pslist


class VmaYaraScan(interfaces.plugins.PluginInterface):
"""Scans all virtual memory areas for tasks using yara."""

_required_framework_version = (2, 4, 0)
_version = (1, 0, 0)

@classmethod
def get_requirements(cls) -> List[interfaces.configuration.RequirementInterface]:
# create a list of requirements for vmayarascan
vmayarascan_requirements = [
requirements.ListRequirement(
name="pid",
element_type=int,
description="Process IDs to include (all other processes are excluded)",
optional=True,
),
requirements.PluginRequirement(
name="pslist", plugin=pslist.PsList, version=(2, 0, 0)
),
requirements.PluginRequirement(
name="yarascan", plugin=yarascan.YaraScan, version=(1, 2, 0)
),
requirements.VersionRequirement(
name="yarascanner", component=yarascan.YaraScanner, version=(2, 0, 0)
),
requirements.ModuleRequirement(
name="kernel",
description="Linux kernel",
architectures=["Intel32", "Intel64"],
),
]

# get base yarascan requirements for command line options
yarascan_requirements = yarascan.YaraScan.get_yarascan_option_requirements()

# return the combined requirements
return yarascan_requirements + vmayarascan_requirements

def _generator(self):
# use yarascan to parse the yara options provided and create the rules
rules = yarascan.YaraScan.process_yara_options(dict(self.config))

# filter based on the pid option if provided
filter_func = pslist.PsList.create_pid_filter(self.config.get("pid", None))
for task in pslist.PsList.list_tasks(
context=self.context,
vmlinux_module_name=self.config["kernel"],
filter_func=filter_func,
):
# attempt to create a process layer for each task and skip those
# that cannot (e.g. kernel threads)
proc_layer_name = task.add_process_layer()
if not proc_layer_name:
continue

# get the proc_layer object from the context
proc_layer = self.context.layers[proc_layer_name]

# scan the process layer with the yarascanner
for offset, rule_name, name, value in proc_layer.scan(
context=self.context,
scanner=yarascan.YaraScanner(rules=rules),
sections=self.get_vma_maps(task),
):
yield 0, (
format_hints.Hex(offset),
task.tgid,
rule_name,
name,
value,
)

@staticmethod
def get_vma_maps(
task: interfaces.objects.ObjectInterface,
) -> Iterable[Tuple[int, int]]:
"""Creates a map of start/end addresses for each virtual memory area in a task.
Args:
task: The task object of which to read the vmas from
Returns:
An iterable of tuples containing start and end addresses for each descriptor
"""
if task.mm:
for vma in task.mm.get_vma_iter():
vm_size = vma.vm_end - vma.vm_start
yield (vma.vm_start, vm_size)

def run(self):
return renderers.TreeGrid(
[
("Offset", format_hints.Hex),
("PID", int),
("Rule", str),
("Component", str),
("Value", bytes),
],
self._generator(),
)
18 changes: 15 additions & 3 deletions volatility3/framework/plugins/yarascan.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,19 +61,31 @@ class YaraScan(plugins.PluginInterface):
"""Scans kernel memory using yara rules (string or file)."""

_required_framework_version = (2, 0, 0)
_version = (1, 1, 0)
_version = (1, 2, 0)

# TODO: When the major version is bumped, take the opportunity to rename the yara_rules config to yara_string
# or something that makes more sense

@classmethod
def get_requirements(cls) -> List[interfaces.configuration.RequirementInterface]:
return [
"""Returns the requirements needed to run yarascan directly, combining the TranslationLayerRequirement
and the requirements from get_yarascan_option_requirements."""
return cls.get_yarascan_option_requirements() + [
requirements.TranslationLayerRequirement(
name="primary",
description="Memory layer for the kernel",
architectures=["Intel32", "Intel64"],
),
)
]

@classmethod
def get_yarascan_option_requirements(
cls,
) -> List[interfaces.configuration.RequirementInterface]:
"""Returns the requirements needed for the command lines options used by yarascan. This can
then also be used by other plugins that are using yarascan. This does not include a
TranslationLayerRequirement or a ModuleRequirement."""
return [
requirements.BooleanRequirement(
name="insensitive",
description="Makes the search case insensitive",
Expand Down

0 comments on commit 7012edb

Please sign in to comment.