mirror of
https://github.com/volatilityfoundation/volatility
synced 2026-06-08 18:04:46 +00:00
gleeda
f5aabe2368
crashed when supplying a base address due to missing data that was not yielded. also made the "ADDR" arg clearer
104 lines
4.3 KiB
Python
104 lines
4.3 KiB
Python
# Volatility
|
|
# Copyright (c) 2008-2015 Volatility Foundation
|
|
#
|
|
# This file is part of Volatility.
|
|
#
|
|
# Volatility is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License Version 2 as
|
|
# published by the Free Software Foundation. You may not use, modify or
|
|
# distribute this program under any other version of the GNU General
|
|
# Public License.
|
|
#
|
|
# Volatility is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Volatility. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
import volatility.utils as utils
|
|
import volatility.obj as obj
|
|
import volatility.debug as debug
|
|
import volatility.plugins.common as common
|
|
import volatility.plugins.malware.devicetree as dtree
|
|
import volatility.win32.modules as modules
|
|
import volatility.win32.tasks as tasks
|
|
from volatility.renderers import TreeGrid
|
|
from volatility.renderers.basic import Address
|
|
|
|
class drivermodule(common.AbstractWindowsCommand):
|
|
"""Associate driver objects to kernel modules"""
|
|
|
|
def __init__(self, config, *args, **kwargs):
|
|
common.AbstractWindowsCommand.__init__(self, config, *args, **kwargs)
|
|
config.add_option('ADDR', short_option = 'a', default = None,
|
|
help = 'Show info on module at or containing this (base) address',
|
|
action = 'store', type = 'int')
|
|
|
|
def calculate(self):
|
|
addr_space = utils.load_as(self._config)
|
|
|
|
modlist = list(modules.lsmod(addr_space))
|
|
mods = dict((addr_space.address_mask(mod.DllBase), mod) for mod in modlist)
|
|
mod_addrs = sorted(mods.keys())
|
|
|
|
drivers = dtree.DriverIrp(self._config).calculate()
|
|
driver_name = "UNKNOWN"
|
|
service_key = "UNKNOWN"
|
|
driver_name3 = "UNKNOWN"
|
|
module_name = "UNKNOWN"
|
|
|
|
if self._config.ADDR:
|
|
find_address = self._config.ADDR
|
|
|
|
module_name = tasks.find_module(mods, mod_addrs, mods.values()[0].obj_vm.address_mask(find_address))
|
|
if module_name:
|
|
module_name = module_name.BaseDllName or module_name.FullDllName
|
|
|
|
for driver in drivers:
|
|
if driver.DriverStart <= find_address < driver.DriverStart + driver.DriverSize:
|
|
header = driver.get_object_header()
|
|
driver_name = header.NameInfo.Name
|
|
driver_name = str(driver.get_object_header().NameInfo.Name or '')
|
|
service_key = str(driver.DriverExtension.ServiceKeyName or '')
|
|
driver_name3 = str(driver.DriverName or '')
|
|
break
|
|
|
|
yield (module_name, driver_name, service_key, driver_name3)
|
|
|
|
else:
|
|
for driver in drivers:
|
|
driver_name = str(driver.get_object_header().NameInfo.Name or '')
|
|
service_key = str(driver.DriverExtension.ServiceKeyName or '')
|
|
driver_name3 = str(driver.DriverName or '')
|
|
|
|
owning_module = tasks.find_module(mods, mod_addrs, mods.values()[0].obj_vm.address_mask(driver.DriverStart))
|
|
module_name = "UNKNOWN"
|
|
if owning_module:
|
|
module_name = owning_module.BaseDllName or owning_module.FullDllName
|
|
|
|
yield (module_name, driver_name, service_key, driver_name3)
|
|
|
|
def generator(self, data):
|
|
|
|
for module_name, driver_name, service_key, driver_name3 in data:
|
|
yield( 0, [str(module_name), str(driver_name), str(service_key), str(driver_name3)])
|
|
|
|
def unified_output(self, data):
|
|
return TreeGrid([("Module", str),
|
|
("Driver", str),
|
|
("Alt. Name", str),
|
|
("Service Key", str)],
|
|
self.generator(data))
|
|
|
|
def render_text(self, outfd, data):
|
|
self.table_header(outfd, [("Module", "36"), ("Driver", "24"), ("Alt. Name", "24"), ("Service Key", "")])
|
|
|
|
for module_name, driver_name, service_key, driver_name3 in data:
|
|
self.table_row(outfd, module_name, driver_name, service_key, driver_name3)
|
|
|
|
|
|
|
|
|