ys/GenericAgent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(mem_scanner): fix context_size logic and update sops

Browse Source
This commit is contained in:
Liang Jiaqing
2026-02-09 18:32:15 +08:00
parent 656a3fe45e
commit 3a9a5ee2aa
6 changed files with 174 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
memory/mem_scanner.py
View File

@@ -91,9 +91,12 @@ def scan_memory(pid, pattern, context_size=256, mode='auto', llm_mode=False):
matched_data = inst.instances[0].matched_data
base = mbi.BaseAddress if mbi.BaseAddress else 0
if llm_mode:
results.append(format_llm_context(data, offset, base))
results.append(format_llm_context(data, offset, base, length=context_size))
else:
results.append(f"Addr: {hex(base+offset)}\nHex: {data[max(0,offset-16):offset+32].hex()}")
# Expand context based on context_size to capture full KEY+SALT
start = max(0, offset - context_size)
end = min(len(data), offset + len(matched_data) + context_size)
results.append(f"Addr: {hex(base+offset)}\nHex: {data[start:end].hex()}")
# Update address using the region size
next_addr = (mbi.BaseAddress if mbi.BaseAddress else 0) + mbi.RegionSize