Files
2014-07-17 15:54:20 -05:00

89 lines
3.4 KiB
Python

# Volatility
# Copyright (C) 2008-2013 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 as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# 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.plugins.mac.pstasks as pstasks
import volatility.plugins.volshell as volshell
import volatility.plugins.mac.lsmod as lsmod
import volatility.obj as obj
class mac_volshell(volshell.volshell):
"""Shell in the memory image"""
@staticmethod
def is_valid_profile(profile):
return profile.metadata.get('os', 'Unknown').lower() == 'mac'
def modules(self):
mods = lsmod.mac_lsmod(self._config).calculate()
for mod in mods:
print "{3:16x} {0:48} {1:16x} {2:6d}".format(mod.name, mod.address, mod.m('size'), mod.obj_offset)
def getpidlist(self):
return list(pstasks.mac_tasks(self._config).allprocs())
def ps(self, procs = None):
print "{0:16} {1:6} {2:8}".format("Name", "PID", "Offset")
for proc in procs or self.getpidlist():
print "{0:16} {1:<6} {2:#08x}".format(proc.p_comm, proc.p_pid, proc.obj_offset)
def context_display(self):
dtb = self._proc.task.dereference_as("task").map.pmap.pm_cr3
print "Current context: process {0}, pid={1} DTB={2:#x}".format(self._proc.p_comm,
self._proc.p_pid, dtb)
def set_context(self, offset = None, pid = None, name = None, physical = None):
if pid is not None:
offsets = []
for p in self.getpidlist():
if p.p_pid.v() == pid:
offsets.append(p)
if not offsets:
print "Unable to find process matching pid {0}".format(pid)
return
elif len(offsets) > 1:
print "Multiple processes match {0}, please specify by offset".format(pid)
print "Matching processes:"
self.ps(offsets)
return
else:
offset = offsets[0].v()
elif name is not None:
offsets = []
for p in self.getpidlist():
if p.p_comm.find(name) >= 0:
offsets.append(p)
if not offsets:
print "Unable to find process matching name {0}".format(name)
return
elif len(offsets) > 1:
print "Multiple processes match name {0}, please specify by PID or offset".format(name)
print "Matching processes:"
self.ps(offsets)
return
else:
offset = offsets[0].v()
elif offset is None:
print "Must provide one of: offset, name, or pid as a argument."
return
self._proc = obj.Object("proc", offset = offset, vm = self._addrspace)
self.context_display()