Matiq/soothe2-re
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

24mm11: wine ptrace tracer works; FIR chain verified BIT-EXACT live (ratio=1.0, q=1 exact); df0 complex-mul confirmed; NEW: track_i != exp(scr) -> gamma born in detector cascade (Stage B target)

Browse Source
This commit is contained in:
Matiq
2026-08-26 14:15:03 +03:00
parent 6bc0120286
commit 60421c32c9
3 changed files with 808 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/wine_chain_trace.py
+306
View File
@@ -0,0 +1,306 @@
#!/usr/bin/env python3
"""wine_chain_trace.py — живой захват промежуточных состояний FIR-цепи
soothe2 через winedbg (wine) + /proc/<pid>/mem.
Брейкпоинты:
EXP 0x1803831c0 комплексная экспонента FIR-цепи (rcx=buf, r8d=count float)
DF0 0x18000b3c0 финальный complex-mul (rcx=FIR, rdx=track, r8d=n пар)
На хите: читаем rcx/rdx/r8 (info reg), буферы — через /proc/<pid>/mem,
копим сэмплы, отпускаем (c). Рендер не убивается.
Запуск: python3 scripts/wine_chain_trace.py <rpp> [n_hits] [outdir]
"""
import os
import pickle
import re
import signal
import struct
import subprocess
import sys
import threading
import time
import numpy as np
BP_EXP = 0x1803831c0
BP_DF0 = 0x18000b3c0
CTX_SLOTS = {'scr': 0x540628, 'trk': 0x540688, 'cur': 0x540678,
'fir_ptr': 0x540668}
def find_host():
import glob
for p in glob.glob('/proc/[0-9]*'):
pid = int(os.path.basename(p))
try:
cmd = open(f'/proc/{pid}/cmdline', 'rb').read().replace(b'\0', b' ').decode('utf8', 'replace')
maps = open(f'/proc/{pid}/maps').read()
except Exception:
continue
if 'soothe2' in maps and 'reaper' not in cmd:
return pid, cmd[:80]
return None, None
def find_ctx(fd, pid):
vt = struct.pack('<Q', 0x1824AC210)
m48 = struct.pack('<I', 0x47380000)
for line in open(f'/proc/{pid}/maps'):
parts = line.split()
if 'rw' not in parts[1]:
continue
lo, hi = (int(x, 16) for x in parts[0].split('-'))
CH = 16 * 1024 * 1024
a = lo
while a < hi:
n = min(CH, hi - a)
try:
d = os.pread(fd, n, a)
except OSError:
break
j = d.find(vt)
while j >= 0:
cand = a + j
sb = os.pread(fd, 4, cand + 0x540870)
if sb and struct.unpack('<f', sb)[0] > 100:
return cand
j = d.find(vt, j + 1)
j = d.find(m48)
while j >= 0:
cand = a + j - 0x24
try:
sb = os.pread(fd, 4, cand + 0x540870)
if sb and struct.unpack('<f', sb)[0] > 100:
return cand
except OSError:
pass
j = d.find(m48, j + 1)
a += n
return None
class WineDbg:
"""Асинхронный ридер stdout winedbg + обмен командами по приглашению."""
PROMPT = 'Wine-dbg>'
def __init__(self, pid):
self.p = subprocess.Popen(
['winedbg', '--pid', str(pid)],
stdin=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.STDOUT, text=True, bufsize=1)
self.buf = ''
self.lock = threading.Lock()
self.ev = threading.Event()
self.alive = True
self.t = threading.Thread(target=self._reader, daemon=True)
self.t.start()
if not self.ev.wait(30):
raise TimeoutError('winedbg не показал приглашение')
def _reader(self):
while self.alive:
ch = self.p.stdout.read(1)
if not ch:
self.alive = False
self.ev.set()
return
with self.lock:
self.buf += ch
if self.PROMPT in self.buf:
self.ev.set()
def cmd(self, c, timeout=90):
with self.lock:
self.buf = ''
self.ev.clear()
self.p.stdin.write(c + '\n')
self.p.stdin.flush()
if not self.ev.wait(timeout):
with self.lock:
tail = self.buf[-300:]
raise TimeoutError('winedbg timeout после %r; tail=%r' % (c, tail))
with self.lock:
out = self.buf.replace(self.PROMPT, '').strip()
self.buf = ''
self.ev.clear()
return out
def close(self):
self.alive = False
try:
self.p.stdin.write('quit\n')
self.p.stdin.flush()
except Exception:
pass
try:
self.p.kill()
except Exception:
pass
def parse_regs(text):
regs = {}
for mm in re.finditer(r'\b([re]?[a-z]{2,3}|r\d+d?)\s*[:=]\s*([0-9a-fA-F]{4,16})\b', text):
name = mm.group(1).lower()
val = int(mm.group(2), 16)
if name not in regs:
regs[name] = val
# нормализация имён к 64-битным
alias = {'eax': 'rax', 'ecx': 'rcx', 'edx': 'rdx', 'ebx': 'rbx',
'esi': 'rsi', 'edi': 'rdi', 'ebp': 'rbp', 'esp': 'rsp'}
out = {}
for k, v in regs.items():
k64 = alias.get(k, k)
if k64.startswith('r') and k64.endswith('d') and k64[1:-1].isdigit():
k64 = k64[:-1]
if len(k64) <= 3 or k64.startswith('r'):
out[k64] = v
return out
def main():
rpp = sys.argv[1] if len(sys.argv) > 1 else '/home/m/soothe-bt/dual_b1q_0.5.rpp'
n_target = int(sys.argv[2]) if len(sys.argv) > 2 else 60
outdir = sys.argv[3] if len(sys.argv) > 3 else '/tmp/opencode/winetrace'
os.makedirs(outdir, exist_ok=True)
wav = None
for ln in open(rpp, errors='replace'):
if 'RENDER_FILE' in ln and '"' in ln:
wav = ln.split('"')[1]
break
if wav and os.path.exists(wav):
os.remove(wav)
subprocess.run("pkill -9 -x reaper; pkill -9 -f '[y]abridge'; "
"rm -rf /run/user/1000/yabridge-soothe2_x64-*; sleep 1",
shell=True)
proc = subprocess.Popen(['/usr/bin/reaper', '-nosplash', '-ignoreerrors',
'-renderproject', rpp],
stdout=open('/dev/null', 'w'), stderr=subprocess.STDOUT)
t0 = time.time()
host = None
while time.time() - t0 < 30 and not host:
host, cmdl = find_host()
if not host:
time.sleep(0.002)
if not host:
print('NO HOST')
return 1
print('host %d (%s)' % (host, cmdl), flush=True)
fd = os.open(f'/proc/{host}/mem', os.O_RDONLY)
ctx = None
while ctx is None and time.time() - t0 < 25:
try:
os.kill(host, signal.SIGSTOP)
except ProcessLookupError:
break
ctx = find_ctx(fd, host)
os.kill(host, signal.SIGCONT)
if not ctx:
time.sleep(0.005)
if not ctx:
print('NO CTX')
return 1
print('ctx %#x' % ctx, flush=True)
dbg = WineDbg(host)
print(dbg.cmd('break *%#x' % BP_EXP)[:160], flush=True)
print(dbg.cmd('break *%#x' % BP_DF0)[:160], flush=True)
def rd(a, n):
return os.pread(fd, n, a)
def rd_f32(a, n):
return np.frombuffer(rd(a, 4*n), dtype='<f4').astype(np.float64)
def rd_q(a):
return struct.unpack('<Q', rd(a, 8))[0]
samples = []
hits = {'EXP': 0, 'DF0': 0}
t_start = time.time()
stall = 0
while sum(hits.values()) < n_target and time.time() - t_start < 300:
try:
out = dbg.cmd('c', timeout=120)
except TimeoutError as e:
print('timeout:', str(e)[-200:], flush=True)
stall += 1
if stall >= 3:
break
continue
addrs = [int(x, 16) for x in re.findall(r'0x[0-9a-fA-F]{9,}', out)]
pc = None
for a in addrs:
if abs(a - BP_EXP) < 64:
pc = a; kind = 'EXP'; break
if abs(a - BP_DF0) < 64:
pc = a; kind = 'DF0'; break
if pc is None:
ir = dbg.cmd('info reg', timeout=30)
rr = parse_regs(ir)
pc = rr.get('rip', 0)
kind = 'EXP' if abs(pc-BP_EXP) < 64 else ('DF0' if abs(pc-BP_DF0) < 64 else None)
if kind is None:
stall += 1
if stall >= 5:
print('неопознанные остановки; tail:', out[-200:], flush=True)
break
continue
ir = dbg.cmd('info reg', timeout=30)
rr = parse_regs(ir)
rcx = rr.get('rcx', 0); rdx = rr.get('rdx', 0); r8 = rr.get('r8', 0)
rec = {'kind': kind, 'rip': pc, 'rcx': rcx, 'rdx': rdx, 'r8': r8,
't': round(time.time()-t_start, 4)}
try:
if kind == 'EXP':
rec['buf'] = rd_f32(rcx, 4098)
rec['count'] = r8
else:
rec['fir'] = rd_f32(rcx, 4098)
if rdx > 0x10000:
rec['track'] = rd_f32(rdx, 2049*2)
# слоты контекста тем же мгновением (процесс остановлен!)
rec['scr'] = rd_f32(ctx+CTX_SLOTS['scr'], 2049)
rec['trk'] = rd_f32(ctx+CTX_SLOTS['trk'], 2049)
rec['cur'] = rd_f32(ctx+CTX_SLOTS['cur'], 2049)
fp = rd_q(ctx+CTX_SLOTS['fir_ptr'])
rec['fir_via_ctx'] = rd_f32(fp, 4098)
except OSError as e:
rec['err'] = str(e)
samples.append(rec)
hits[kind] += 1
if sum(hits.values()) % 10 == 0:
print('hits:', hits, flush=True)
print('сбор завершён:', hits, flush=True)
snap_ptrs = {}
snap_arr = {}
for nm, off in CTX_SLOTS.items():
try:
p = rd_q(ctx+off)
if p > 0x10000:
snap_ptrs[nm] = p
snap_arr[nm] = rd_f32(p, 4100)
except OSError:
pass
dbg.close()
with open(os.path.join(outdir, 'chain_samples.pkl'), 'wb') as f:
pickle.dump({'samples': samples, 'snap_ptrs': snap_ptrs, 'ctx': ctx}, f)
np.savez_compressed(os.path.join(outdir, 'ctx_snap.npz'), **snap_arr)
print('saved', len(samples), 'samples ->', outdir, flush=True)
for _ in range(600):
if proc.poll() is not None:
break
time.sleep(0.1)
print('reaper_rc=%s wav=%s' % (proc.poll(),
os.path.getsize(wav) if wav and os.path.exists(wav) else 'NONE'), flush=True)
return 0
if __name__ == '__main__':
sys.exit(main())
scripts/wine_ptrace_trace.py
+460
View File
@@ -0,0 +1,460 @@
#!/usr/bin/env python3
"""wine_ptrace_trace.py — точный пер-оп захват FIR-цепи soothe2 через ptrace.
Запускает reaper -renderproject как ребёнок (=> ptrace разрешён при любом
yama scope), находит wine-хост yabridge (soothe2 в maps), прицепляется ко
всем тредам, ставит int3 на входах EXP/DF0 ядра, на хитах читает регистры
(PTRACE_GETREGS) и буферы через /proc/tid/mem; между хитами CONT.
Брейкпоинты:
EXP 0x1803831c0 rcx=buf, r8d=count(float)
DF0 0x18000b3c0 rcx=FIR, rdx=track, r8d=n(пар)
Плюс слоты контекста тем же мгновением (scr/trk/cur/FIR@540668).
Запуск: python3 scripts/wine_ptrace_trace.py <rpp> [n_hits] [outdir]
"""
import ctypes
import os
import pickle
import signal
import struct
import subprocess
import sys
import time
import numpy as np
BP_EXP = 0x1803831c0
BP_DF0 = 0x18000b3c0
BP_COPY = 0x1800136e0
BP_DF0RET = 0x18052b898
CTX_SLOTS = {'scr': 0x540628, 'trk': 0x540688, 'cur': 0x540678,
'fir_ptr': 0x540668}
libc = ctypes.CDLL('libc.so.6', use_errno=True)
PTRACE_ATTACH = 16
PTRACE_DETACH = 17
PTRACE_CONT = 7
PTRACE_SINGLESTEP = 9
PTRACE_PEEKDATA = 2
PTRACE_POKEDATA = 5
PTRACE_GETREGS = 12
PTRACE_SETOPTIONS = 0x4200
PTRACE_O_TRACECLONE = 1 << 22
__WALL = 0x40000000
libc.ptrace.restype = ctypes.c_long
libc.ptrace.argtypes = [ctypes.c_long, ctypes.c_long,
ctypes.c_void_p, ctypes.c_void_p]
class UserRegs(ctypes.Structure):
_fields_ = [('r15', ctypes.c_uint64), ('r14', ctypes.c_uint64),
('r13', ctypes.c_uint64), ('r12', ctypes.c_uint64),
('rbp', ctypes.c_uint64), ('rbx', ctypes.c_uint64),
('r11', ctypes.c_uint64), ('r10', ctypes.c_uint64),
('r9', ctypes.c_uint64), ('r8', ctypes.c_uint64),
('rax', ctypes.c_uint64), ('rcx', ctypes.c_uint64),
('rdx', ctypes.c_uint64), ('rsi', ctypes.c_uint64),
('rdi', ctypes.c_uint64), ('orig_rax', ctypes.c_uint64),
('rip', ctypes.c_uint64), ('cs', ctypes.c_uint64),
('eflags', ctypes.c_uint64), ('rsp', ctypes.c_uint64),
('ss', ctypes.c_uint64),
('fs_base', ctypes.c_uint64), ('gs_base', ctypes.c_uint64),
('ds', ctypes.c_uint64), ('es', ctypes.c_uint64),
('fs', ctypes.c_uint64), ('gs', ctypes.c_uint64)]
def pt(req, pid, addr=0, data=0):
if not isinstance(data, int):
data = ctypes.cast(data, ctypes.c_void_p)
else:
data = ctypes.c_void_p(data)
return libc.ptrace(req, pid, ctypes.c_void_p(addr), data)
def getregs(tid):
r = UserRegs()
if pt(PTRACE_GETREGS, tid, 0, ctypes.byref(r)) != 0:
raise OSError('GETREGS tid=%d' % tid)
return r
def setregs(tid, r):
if pt(PTRACE_SETREGS := 13, tid, 0, ctypes.byref(r)) != 0:
raise OSError('SETREGS tid=%d' % tid)
def peek(tid, addr):
v = pt(PTRACE_PEEKDATA, tid, addr, 0)
if v == -1:
e = ctypes.get_errno()
if e != 0:
raise OSError(e)
return v & 0xFFFFFFFFFFFFFFFF
def poke(tid, addr, val):
if pt(PTRACE_POKEDATA, tid, addr, val) == -1 and ctypes.get_errno():
raise OSError('POKEDATA %#x tid=%d: %d' % (addr, tid, ctypes.get_errno()))
def find_host():
import glob
for p in glob.glob('/proc/[0-9]*'):
pid = int(os.path.basename(p))
try:
cmd = open(f'/proc/{pid}/cmdline', 'rb').read().replace(b'\0', b' ').decode('utf8', 'replace')
maps = open(f'/proc/{pid}/maps').read()
except Exception:
continue
if 'soothe2' in maps and 'reaper' not in cmd:
return pid
return None
def find_ctx(fd, pid):
vt = struct.pack('<Q', 0x1824AC210)
m48 = struct.pack('<I', 0x47380000)
for line in open(f'/proc/{pid}/maps'):
parts = line.split()
if 'rw' not in parts[1]:
continue
lo, hi = (int(x, 16) for x in parts[0].split('-'))
CH = 16 * 1024 * 1024
a = lo
while a < hi:
n = min(CH, hi - a)
try:
d = os.pread(fd, n, a)
except OSError:
break
j = d.find(vt)
while j >= 0:
cand = a + j
sb = os.pread(fd, 4, cand + 0x540870)
if sb and struct.unpack('<f', sb)[0] > 100:
return cand
j = d.find(vt, j + 1)
j = d.find(m48)
while j >= 0:
cand = a + j - 0x24
try:
sb = os.pread(fd, 4, cand + 0x540870)
if sb and struct.unpack('<f', sb)[0] > 100:
return cand
except OSError:
pass
j = d.find(m48, j + 1)
a += n
return None
def find_ctx_candidates(fd, pid, fir_ptr):
"""Все адреса X (кратные 8), где [X+0x540668]==fir_ptr => кандидат X."""
val = struct.pack('<Q', fir_ptr)
out = []
for line in open(f'/proc/{pid}/maps'):
parts = line.split()
if 'rw' not in parts[1]:
continue
lo, hi = (int(x, 16) for x in parts[0].split('-'))
CH = 16 * 1024 * 1024
a = lo
while a < hi:
n = min(CH, hi - a)
try:
d = os.pread(fd, n, a)
except OSError:
break
j = d.find(val)
while j >= 0:
if j % 8 == 0:
out.append(a + j - 0x540668)
j = d.find(val, j + 1)
a += n
return out
def main():
rpp = sys.argv[1] if len(sys.argv) > 1 else '/home/m/soothe-bt/dual_b1q_0.5.rpp'
n_target = int(sys.argv[2]) if len(sys.argv) > 2 else 80
outdir = sys.argv[3] if len(sys.argv) > 3 else '/tmp/opencode/winetrace'
os.makedirs(outdir, exist_ok=True)
wav = None
for ln in open(rpp, errors='replace'):
if 'RENDER_FILE' in ln and '"' in ln:
wav = ln.split('"')[1]
break
if wav and os.path.exists(wav):
os.remove(wav)
subprocess.run("pkill -9 -x reaper; pkill -9 -f '[y]abridge'; "
"rm -rf /run/user/1000/yabridge-soothe2_x64-*; sleep 1",
shell=True)
proc = subprocess.Popen(['/usr/bin/reaper', '-nosplash', '-ignoreerrors',
'-renderproject', rpp],
stdout=open('/dev/null', 'w'), stderr=subprocess.STDOUT)
t0 = time.time()
host = None
ctx_fd = None
ctx = None
# Фаза 1: ждём появления хоста и контекста ЧИТАЮЧЕЙ памятью (без ptrace),
# чтобы не мешать загрузке плагина
while time.time() - t0 < 25:
if host is None:
host = find_host()
if host:
try:
ctx_fd = os.open(f'/proc/{host}/mem', os.O_RDONLY)
print('host %d (+%.3fs)' % (host, time.time()-t0), flush=True)
except OSError:
host = None
time.sleep(0.001)
continue
if host is not None:
try:
ctx = find_ctx(ctx_fd, host)
except (ProcessLookupError, OSError):
ctx = None
host = None
time.sleep(0.001)
continue
if ctx:
break
time.sleep(0.002)
if not host or not ctx:
print('NO HOST/CTX (host=%s ctx=%s)' % (host, ctx))
return 1
print('ctx %#x (+%.3fs)' % (ctx, time.time()-t0), flush=True)
fd = ctx_fd
def rd(a, n):
return os.pread(fd, n, a)
def rd_f32(a, n):
return np.frombuffer(rd(a, 4*n), dtype='<f4').astype(np.float64)
def rd_q(a):
return struct.unpack('<Q', rd(a, 8))[0]
# Фаза 2: аттач ко всем текущим тредам хоста
tids = [int(t) for t in os.listdir(f'/proc/{host}/task')]
attached = []
for tid in tids:
try:
if pt(PTRACE_ATTACH, tid) == -1 and ctypes.get_errno():
raise OSError(ctypes.get_errno())
os.waitpid(tid, __WALL)
pt(PTRACE_SETOPTIONS, tid, 0, PTRACE_O_TRACECLONE)
attached.append(tid)
except OSError as e:
print('attach fail tid=%d: %s' % (tid, e), flush=True)
print('attached %d/%d' % (len(attached), len(tids)), flush=True)
# Фаза 3: int3 и запуск
bps = {}
for name, addr in (('COPY', BP_COPY), ('EXP', BP_EXP), ('DF0', BP_DF0),
('DF0RET', BP_DF0RET)):
orig = peek(host, addr)
poke(host, addr, (orig & ~0xFF) | 0xCC)
bps[addr] = (name, orig & 0xFF)
print('int3 installed:', {hex(a): n for a, (n, _) in bps.items()}, flush=True)
for tid in attached:
pt(PTRACE_CONT, tid, 0, 0)
samples = []
hits = {'COPY': 0, 'EXP': 0, 'DF0': 0, 'DF0RET': 0}
track_by_tid = {}
t_start = time.time()
def snapshot_slots(rec):
rec['scr'] = rd_f32(ctx+CTX_SLOTS['scr'], 2049)
rec['trk'] = rd_f32(ctx+CTX_SLOTS['trk'], 2049)
rec['cur'] = rd_f32(ctx+CTX_SLOTS['cur'], 2049)
fp = rd_q(ctx+CTX_SLOTS['fir_ptr'])
rec['fir_via_ctx'] = rd_f32(fp, 4098)
try:
while sum(hits.values()) < n_target and time.time() - t_start < 300:
try:
pid, status = os.waitpid(-1, __WALL | os.WNOHANG)
except ChildProcessError:
print('нет отслеживаемых процессов', flush=True)
break
if (pid, status) == (0, 0):
# никого не остановлено — короткий сон, дедлайн проверится сверху
time.sleep(0.0005)
continue
if not os.WIFSTOPPED(status):
# выход треда/процесса
if pid in attached:
attached.remove(pid)
if pid == host:
print('host exited', flush=True)
break
continue
sig = os.WSTOPSIG(status)
if sig == signal.SIGTRAP:
try:
regs = getregs(pid)
except OSError:
continue
site = regs.rip - 1
info = bps.get(site)
if info is None:
# чужой SIGTRAP (clone/event) — просто продолжить
pt(PTRACE_CONT, pid, 0, 0)
continue
kind, obyte = info
if kind == 'DF0':
track_by_tid[pid] = regs.rdx
if kind == 'DF0RET':
tp = track_by_tid.get(pid)
rec_r = {'kind': 'DF0RET', 'tid': pid,
't': round(time.time()-t_start, 4)}
try:
if tp and tp > 0x10000:
rec_r['track'] = rd_f32(tp, 2049*2)
samples.append(rec_r)
hits['DF0RET'] += 1
except OSError as e:
rec_r['err'] = str(e)
samples.append(rec_r)
poke(pid, site, (peek(pid, site) & ~0xFF) | obyte)
regs.rip = site
setregs(pid, regs)
pt(PTRACE_SINGLESTEP, pid, 0, 0)
os.waitpid(pid, __WALL)
poke(pid, site, (peek(pid, site) & ~0xFF) | 0xCC)
pt(PTRACE_CONT, pid, 0, 0)
continue
if kind == 'COPY':
try:
cnt = min(regs.r9 & 0xFFFFFFFF, 2049)
rec_c = {'kind': 'COPY', 'tid': pid,
't': round(time.time()-t_start, 4),
'src': rd_f32(regs.rcx, cnt),
'dst': regs.r8}
# снять int3/step/restore как у остальных — общий код ниже
poke(pid, site, (peek(pid, site) & ~0xFF) | obyte)
regs.rip = site
setregs(pid, regs)
pt(PTRACE_SINGLESTEP, pid, 0, 0)
os.waitpid(pid, __WALL)
poke(pid, site, (peek(pid, site) & ~0xFF) | 0xCC)
pt(PTRACE_CONT, pid, 0, 0)
samples.append(rec_c)
hits['COPY'] += 1
continue
except OSError as e:
print('copy err', e, flush=True)
continue
# ctx по фактическому указателю FIR из хита + валидация
# инварианта trk==exp(scr) (24mm3), строгая
if kind == 'DF0':
good = None
cands = find_ctx_candidates(fd, host, regs.rcx)
for cand in cands:
if cand <= 0x10000:
continue
try:
v_sc = rd_f32(cand+CTX_SLOTS['scr'], 2049)
v_tr = rd_f32(cand+CTX_SLOTS['trk'], 2049)
except OSError:
continue
if not (np.isfinite(v_sc).all() and np.isfinite(v_tr).all()):
continue
if np.abs(v_sc).max() > 40:
continue
if np.allclose(v_tr, np.exp(v_sc), rtol=1e-3, atol=1e-9):
good = cand
break
if pc_dbg := True:
for cand in cands[:4]:
try:
vs = rd_f32(cand+CTX_SLOTS['scr'], 2049)
vt = rd_f32(cand+CTX_SLOTS['trk'], 2049)
except OSError:
continue
dmax = np.abs(vt-np.exp(np.clip(vs,-80,80))).max()
print(' cand %#x: |scr|=%.4g |trk|=%.4g maxdiff=%.4g'
% (cand, np.abs(vs).max(), np.abs(vt).max(), dmax),
flush=True)
print('cands=%d good=%s' % (len(cands), hex(good) if good else '-'),
flush=True)
if good:
ctx = good
if kind == 'DF0':
track_by_tid[pid] = regs.rdx
rec = {'kind': kind, 'tid': pid,
'rcx': regs.rcx, 'rdx': regs.rdx, 'r8': regs.r8 & 0xFFFFFFFF,
't': round(time.time()-t_start, 4)}
try:
if kind == 'EXP':
rec['buf'] = rd_f32(regs.rcx, 4098)
rec['count'] = rec['r8']
else:
rec['fir'] = rd_f32(regs.rcx, 4098)
if regs.rdx > 0x10000:
rec['track'] = rd_f32(regs.rdx, 2049*2)
if ctx:
snapshot_slots(rec)
if kind == 'DF0':
rec['fir_via_ctx'] = rec.get('fir_via_ctx')
except OSError as e:
rec['err'] = str(e)
samples.append(rec)
hits[kind] += 1
# снять int3 -> шаг назад -> singlestep -> вернуть int3 -> cont
poke(pid, site, (peek(pid, site) & ~0xFF) | obyte)
regs.rip = site
setregs(pid, regs)
pt(PTRACE_SINGLESTEP, pid, 0, 0)
os.waitpid(pid, __WALL)
poke(pid, site, (peek(pid, site) & ~0xFF) | 0xCC)
pt(PTRACE_CONT, pid, 0, 0)
if sum(hits.values()) % 10 == 0:
print('hits:', hits, flush=True)
elif sig in (signal.SIGSTOP, signal.SIGCHLD, signal.SIGWINCH):
pt(PTRACE_CONT, pid, 0, 0)
else:
# посторонний сигнал — доставить
pt(PTRACE_CONT, pid, 0, sig)
finally:
# снять int3 и отсоединиться
for addr, (name, obyte) in bps.items():
try:
poke(host, addr, (peek(host, addr) & ~0xFF) | obyte)
except OSError:
pass
for tid in list(attached):
try:
pt(PTRACE_DETACH, tid, 0, 0)
except OSError:
pass
print('сбор завершён:', hits, flush=True)
snap_ptrs, snap_arr = {}, {}
for nm, off in CTX_SLOTS.items():
p = rd_q(ctx+off)
if p > 0x10000:
snap_ptrs[nm] = p
snap_arr[nm] = rd_f32(p, 4100)
with open(os.path.join(outdir, 'chain_samples.pkl'), 'wb') as f:
pickle.dump({'samples': samples, 'snap_ptrs': snap_ptrs, 'ctx': ctx}, f)
np.savez_compressed(os.path.join(outdir, 'ctx_snap.npz'), **snap_arr)
print('saved %d -> %s' % (len(samples), outdir), flush=True)
for _ in range(600):
if proc.poll() is not None:
break
time.sleep(0.1)
print('reaper_rc=%s wav=%s' % (proc.poll(),
os.path.getsize(wav) if wav and os.path.exists(wav) else 'NONE'), flush=True)
return 0
if __name__ == '__main__':
sys.exit(main())