mirror/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2026-09-28 17:11:11 -09:00
Code Issues Packages Projects Releases Wiki Activity

GP-0: from review

Browse Source
This commit is contained in:
d-millar
2025-01-10 20:50:09 +00:00
parent 4fd092bb73
commit fe71611803
6 changed files with 119 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

195
Ghidra/Debug/Debugger-agent-drgn/src/main/py/src/ghidradrgn/commands.py
View File

@@ -973,106 +973,111 @@ def ghidra_trace_put_environment():
put_environment()
def put_regions():
nproc = util.selected_process()
if nproc is None:
return
try:
regions = prog.loaded_modules()
except Exception as e:
regions = []
#if len(regions) == 0:
# regions = util.full_mem()
mapper = STATE.trace.memory_mapper
keys = []
# r : MEMORY_BASIC_INFORMATION64
for r in regions:
start = r[0].address_range[0]
end = r[0].address_range[1]
size = end - start + 1
rpath = REGION_PATTERN.format(procnum=nproc, start=start)
keys.append(REGION_KEY_PATTERN.format(start=start))
regobj = STATE.trace.create_object(rpath)
(start_base, start_addr) = map_address(start)
regobj.set_value('Range', start_addr.extend(size))
regobj.set_value('Name', r[0].name)
regobj.set_value('Object File', r[0].loaded_file_path)
regobj.set_value('_readable', True)
regobj.set_value('_writable', True)
regobj.set_value('_executable', True)
regobj.set_value('_display', '{:x} {}'.format(start, r[0].name))
regobj.insert()
STATE.trace.proxy_object_path(
MEMORY_PATTERN.format(procnum=nproc)).retain_values(keys)
def ghidra_trace_put_regions():
"""
Read the memory map, if applicable, and write to the trace's Regions
"""
STATE.require_tx()
with STATE.client.batch() as b:
put_regions()
def put_modules():
nproc = util.selected_process()
if nproc is None:
return
# Detect whether this is supported before defining the command
if hasattr(drgn, 'RelocatableModule'):
def put_regions():
nproc = util.selected_process()
if nproc is None:
return
try:
modules = prog.modules()
except Exception as e:
return
mapper = STATE.trace.memory_mapper
mod_keys = []
for m in modules:
name = m.name
# m[1] : _DEBUG_MODULE_PARAMETERS
base = m.address_range[0]
hbase = hex(base)
size = m.address_range[1] - base
mpath = MODULE_PATTERN.format(procnum=nproc, modpath=hbase)
modobj = STATE.trace.create_object(mpath)
mod_keys.append(MODULE_KEY_PATTERN.format(modpath=hbase))
base_base, base_addr = mapper.map(nproc, base)
if base_base != base_addr.space:
STATE.trace.create_overlay_space(base_base, base_addr.space)
modobj.set_value('Range', base_addr.extend(size))
modobj.set_value('Name', name)
modobj.set_value('_display', '{:x} {}'.format(base, name))
modobj.insert()
attrobj = STATE.trace.create_object(mpath+".Attributes")
attrobj.set_value('BuildId', m.build_id)
attrobj.set_value('DebugBias', m.debug_file_bias)
attrobj.set_value('DebugPath', m.debug_file_path)
attrobj.set_value('DebugStatus', str(m.debug_file_status))
attrobj.set_value('LoadBias', m.loaded_file_bias)
attrobj.set_value('LoadPath', m.loaded_file_path)
attrobj.set_value('LoadStatus', str(m.loaded_file_status))
attrobj.insert()
if type(m) == drgn.RelocatableModule:
secobj = STATE.trace.create_object(mpath+".Sections")
secobj.insert()
STATE.trace.proxy_object_path(MODULES_PATTERN.format(
procnum=nproc)).retain_values(mod_keys)
try:
regions = prog.loaded_modules()
except Exception as e:
regions = []
#if len(regions) == 0:
# regions = util.full_mem()
mapper = STATE.trace.memory_mapper
keys = []
# r : MEMORY_BASIC_INFORMATION64
for r in regions:
start = r[0].address_range[0]
end = r[0].address_range[1]
size = end - start + 1
rpath = REGION_PATTERN.format(procnum=nproc, start=start)
keys.append(REGION_KEY_PATTERN.format(start=start))
regobj = STATE.trace.create_object(rpath)
(start_base, start_addr) = map_address(start)
regobj.set_value('Range', start_addr.extend(size))
regobj.set_value('Name', r[0].name)
regobj.set_value('Object File', r[0].loaded_file_path)
regobj.set_value('_readable', True)
regobj.set_value('_writable', True)
regobj.set_value('_executable', True)
regobj.set_value('_display', '{:x} {}'.format(start, r[0].name))
regobj.insert()
STATE.trace.proxy_object_path(
MEMORY_PATTERN.format(procnum=nproc)).retain_values(keys)
def ghidra_trace_put_modules():
"""
Gather object files, if applicable, and write to the trace's Modules
"""
STATE.require_tx()
with STATE.client.batch() as b:
put_modules()
def ghidra_trace_put_regions():
"""
Read the memory map, if applicable, and write to the trace's Regions
"""
STATE.require_tx()
with STATE.client.batch() as b:
put_regions()
# Detect whether this is supported before defining the command
if hasattr(drgn, 'RelocatableModule'):
def put_modules():
nproc = util.selected_process()
if nproc is None:
return
try:
modules = prog.modules()
except Exception as e:
return
mapper = STATE.trace.memory_mapper
mod_keys = []
for m in modules:
name = m.name
# m[1] : _DEBUG_MODULE_PARAMETERS
base = m.address_range[0]
hbase = hex(base)
size = m.address_range[1] - base
mpath = MODULE_PATTERN.format(procnum=nproc, modpath=hbase)
modobj = STATE.trace.create_object(mpath)
mod_keys.append(MODULE_KEY_PATTERN.format(modpath=hbase))
base_base, base_addr = mapper.map(nproc, base)
if base_base != base_addr.space:
STATE.trace.create_overlay_space(base_base, base_addr.space)
modobj.set_value('Range', base_addr.extend(size))
modobj.set_value('Name', name)
modobj.set_value('_display', '{:x} {}'.format(base, name))
modobj.insert()
attrobj = STATE.trace.create_object(mpath+".Attributes")
attrobj.set_value('BuildId', m.build_id)
attrobj.set_value('DebugBias', m.debug_file_bias)
attrobj.set_value('DebugPath', m.debug_file_path)
attrobj.set_value('DebugStatus', str(m.debug_file_status))
attrobj.set_value('LoadBias', m.loaded_file_bias)
attrobj.set_value('LoadPath', m.loaded_file_path)
attrobj.set_value('LoadStatus', str(m.loaded_file_status))
attrobj.insert()
if type(m) == drgn.RelocatableModule:
secobj = STATE.trace.create_object(mpath+".Sections")
secobj.insert()
STATE.trace.proxy_object_path(MODULES_PATTERN.format(
procnum=nproc)).retain_values(mod_keys)
def ghidra_trace_put_modules():
"""
Gather object files, if applicable, and write to the trace's Modules
"""
STATE.require_tx()
with STATE.client.batch() as b:
put_modules()
# Detect whether this is supported before defining the command
if hasattr(drgn, 'RelocatableModule'):
def put_sections(m : drgn.RelocatableModule):
nproc = util.selected_process()

26
Ghidra/Debug/Debugger-agent-drgn/src/main/py/src/ghidradrgn/methods.py
View File

@@ -263,20 +263,22 @@ def refresh_locals(node: sch.Schema('LocalsContainer')):
commands.ghidra_trace_put_locals()
@REGISTRY.method(action='refresh', display='Refresh Memory')
def refresh_mappings(node: sch.Schema('Memory')):
"""Refresh the list of memory regions for the process."""
with commands.open_tracked_tx('Refresh Memory Regions'):
commands.ghidra_trace_put_regions()
if hasattr(drgn, 'RelocatableModule'):
@REGISTRY.method(action='refresh', display='Refresh Memory')
def refresh_mappings(node: sch.Schema('Memory')):
"""Refresh the list of memory regions for the process."""
with commands.open_tracked_tx('Refresh Memory Regions'):
commands.ghidra_trace_put_regions()
@REGISTRY.method(action='refresh', display='Refresh Modules')
def refresh_modules(node: sch.Schema('ModuleContainer')):
"""
Refresh the modules list for the process.
"""
with commands.open_tracked_tx('Refresh Modules'):
commands.ghidra_trace_put_modules()
if hasattr(drgn, 'RelocatableModule'):
@REGISTRY.method(action='refresh', display='Refresh Modules')
def refresh_modules(node: sch.Schema('ModuleContainer')):
"""
Refresh the modules list for the process.
"""
with commands.open_tracked_tx('Refresh Modules'):
commands.ghidra_trace_put_modules()
@REGISTRY.method(action='activate')