80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
|
"""scan_lutsub.py v2 — range-filtered scan for detector-LUT param subs.
|
|
Filter: -60<A<-1, 1<B<100, 0.2<G<8, mode in {0,1}, [+0x90] valid pointer.
|
|
Prints unique (A,B,G,mode) tuples with addresses.
|
|
"""
|
|
import struct, subprocess, sys, time
|
|
import glob, os
|
|
import numpy as np
|
|
|
|
def find_host():
|
|
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
|
|
|
|
subprocess.run("pkill -9 -x reaper; pkill -9 -f '[y]abridge'; "
|
|
"rm -rf /run/user/1000/yabridge-soothe2_x64-*; sleep 1", shell=True)
|
|
rpp = sys.argv[1] if len(sys.argv) > 1 else '/home/m/soothe-bt/dual_b1q_0.5.rpp'
|
|
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 = find_host(); time.sleep(0.002)
|
|
print('host', host, flush=True)
|
|
if not host:
|
|
sys.exit(1)
|
|
fd = os.open(f'/proc/{host}/mem', os.O_RDONLY)
|
|
|
|
def rd(a,n):
|
|
try: return os.pread(fd,n,a)
|
|
except OSError: return None
|
|
|
|
seen={}
|
|
alive=True
|
|
t_end=time.time()+12
|
|
while time.time()<t_end and alive:
|
|
try:
|
|
for line in open(f'/proc/{host}/maps'):
|
|
parts=line.split()
|
|
if 'rw' not in parts[1]: continue
|
|
lo,hi=(int(x,16) for x in parts[0].split('-'))
|
|
CH=8*1024*1024
|
|
a=lo
|
|
while a<hi:
|
|
d=rd(a,min(CH+64,hi-a))
|
|
if not d:
|
|
alive=False; break
|
|
n=len(d)//4
|
|
arr=np.frombuffer(d[:n*4],dtype='<f4')
|
|
# vectorized filter on A at even offsets
|
|
m=(arr>-60)&(arr<-1)
|
|
for i in np.nonzero(m)[0]:
|
|
Bv=float(arr[i+1]) if i+1<n else 0
|
|
if not (1<Bv<100): continue
|
|
addr=a+i*4
|
|
s=rd(addr,0x98)
|
|
if not s or len(s)<0x98: continue
|
|
G=struct.unpack('<f',s[12:16])[0]
|
|
if not (0.2<G<8): continue
|
|
mode=s[16]
|
|
if mode>1: continue
|
|
p=struct.unpack('<Q',s[0x90:0x98])[0]
|
|
if not (0x10000<p<0x7ffff0000000): continue
|
|
key=(round(float(arr[i]),2),round(Bv,2),round(G,3),mode)
|
|
if key not in seen:
|
|
seen[key]=(addr,p)
|
|
print('NEW sub@%x vt=%x A=%.3f B=%.3f G=%.4f mode=%d' %
|
|
(addr,p,arr[i],Bv,G,mode), flush=True)
|
|
a+=CH
|
|
except (FileNotFoundError, ProcessLookupError):
|
|
alive=False
|
|
break
|
|
print('done; unique tuples:', len(seen))
|