UHj+8dZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZmZddl mZmZmZeej)j*dZed z Zdd Zdd Zdd Zdd ZddZdddZddZddZefddZ ddZ!e"dk(re!yy)zITrace save-selector leaf streams using script handler advance candidates.) annotationsN)Path)find_cns_strings read_sections va_to_offset)DEFAULT_HANDLER_VAHANDLER_TABLE_VA)analyze_stream_effectdword_atsection_name_for_vaoutcFt||}||t|k\ry||S)N)rlenexesectionsvaoffsets B/home/exedev/hwanse/tools/summarize_save_selector_stream_traces.pybyte_atrs+ (B 'F ~3s8+ v;cvt||}||dzt|kDrytjd||dS)Nz"VXY  Gw$6&sHjAVZZQS5TjTX\X`X`ahXitxx(jkH&&8C.)1;1G"Z,-T!&*<<!!',C!D!LVZZ78zz-06B  ks ,C>C>c*|dk(r y|dk(ry|dk(ryy)Ni@zif table[(context+0xa8)[stream+2]] == 1 then fall through +8; table is 0x59e370 when stream+1 is 0, otherwise 0x59e360; else jump to dword [stream+4]im@zVjump to dword [stream+4] when encounter-state flags are set; otherwise fall through +8i&@z@condition helper can jump to dword [stream+4] or fall through +8)r3s r branch_noter;;s/Z , ZgZQ rcg}t}|}t|D]a}||vr|j|d|ddd|S|j|t |||} t |||} | | |j|d|ddd|St ||| } ||d|d| d| dd| } | |vr || | d<nt|| d| d <d| d| d <| d rJt |||d z} | d| dnd| d <d|dzd| d<t| jd}|r|| d<|j| | d}t|dk(r| d s ||dz }7t|dk(r | d rd| d<|S|sd| d<|Sd| d<|S|S)Nr#r%loop)stepvaHex stopReason unreadable)r>rr?valuevalueHexcnsTpointer pointerHexr&rbranchTargetHexfallthroughVaHexr* branchNoter r rzbranch-or-fallthroughr@zno-fixed-advancezmultiple-advances) setrangeappendaddrrr8rr;r0r)rrstrings stream_va max_stepsrowsseenrr>r'rBhandlerrowtargetnoter7s r trace_streamrXIs7 D 5D Bi + : KK"RH VT U R KQ  h+sHb) >U] KK"RH \Z [ F KE Xv6"S]U3K(     G  CJ (E * 6!C N"$U3K 0C  * +C262F:@:Lr&%6RVC! "(*26#,&7C" #w{{;78D$(L! C?+ x=A g.E&F (1+ B  x=A '*A"B 7C   K  2C   K0L KY+X Krc>t|}t||}g}|D]~}dD]w\}}|j|} | st| d} |j |jd|jd|jd|| t |||| |dy|S)N))leafleafPointerHex)nestednestedPointerHexsourcerVr[)r_rVr[ streamKind streamVaHextrace)rrr0intrMrX) r leaf_rowsrQrrOrRrZkindkey stream_hexrPs r build_rowsrh|sS!HsH-G D U ID##JJ+I KK((8,((8,"&((+;"<")%c8WiS     Krchddddddt|ddg}|D]}|jd|dd |d d |d d dddg|dD]A}djd|jdgDxsd}|dk7rd|ndg}|jdr|j d|jdr|j d|j d|dd|dd|jddd|jd dd|jd!dd"d#j|d|jd$xsdd|jd%xsdd"|jd&xsdd'|jd(xsdd)D|j dd*j|S)+Nz# Save Selector Stream TraceszwLinear bytecode traces for save-selector leaf streams. These traces follow only handlers with one fixed stream advance.zrRows with branch targets, missing advances, or multiple possible advances are stopped rather than guessed through.zTraced streams: .z## ra (r`, leaf r[)z]| step | va | value | opcode | handler | effect | branch target | fallthrough | stop | note |z>| ---: | --- | --- | --- | --- | --- | --- | --- | --- | --- |rb/c3&K|] }d| yw+Nr:.0rBs r zmarkdown..W!E7 Wr -advance r. writes streamr&branchz| r>z | `r?z` | `rCr(r+z` | ; rGrIr@z | rJz | )rextendjoinr0rM)rRlinesrUr6r7 effect_bitss rmarkdownrs'  B| 3t9+Q'  E #m$%RL(9':'#FVBWAXXY Z  k L   L DxxW/SU9VWW^[^H4<OXhZ0MKxx'""?3xx/0""8, LLT&\N$tG}oU488JPS;T:UVHH[#./uTXXnc5R4SSW99[)*$txx8I/J/Qc.RSHH/07C8TXXl=S=ZWZ<[[^88L)0S1 5    R+, 99U rcrg}|D]l}g}|dD]}djd|jdgDxsd}|dk7rd|ndg}|jdr|jd|jd r|jd |jd |d d tj|ddtj|jdddtj|jdddtj|jdxsddtjdj|d tj|jdxsddtj|jdxsddtj|jdxsddtj|jdxsdd|jdtj|ddtj|ddtj|d d!d"j|d# od$jd%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7t |d8d$j|d9d:d"gS);Nrbroc3&K|] }d| ywrqr:rss rruzhtml_page..rvrwr rxryr.rzr&r{zr>zr?zrCr(r+zr|rGrIr@z rJz z

rarlr`rmr[z)

rjz
stepvavalueopcodehandlereffectbranch targetfallthroughstopnote
r}zzzz zF z, Save Selector Stream Tracesz zzz&

Save Selector Stream Traces

z

Linear bytecode traces for save-selector leaf streams. These traces follow only handlers with one fixed stream advance.

z{

Rows with branch targets, missing advances, or multiple possible advances are stopped rather than guessed through.

z

Traced streams: z.

zz)rr0rMhtmlescaper)rRrrUbodyr6r7r5s r html_pagersH L DxxW/SU9VWW^[^H/73 +CHFxx' o.xx/0 h' KKF|n%![[g789![[*c)BCDE![[+s)CDEF![[.)A)HSIJK{{499V#4567![[2C)D)KLMN![[2D)E)LMNO{{488L#9#@SABC{{488L#9#@SABC   * 4;;s=1232dkk#lBS6T5UU\]a]h]hilm}i~]]AAggdm_$4 6 / 8 99"P6hTWC&0 K F D {%0 ( / rc|jdd|dz jtj|dddzd y) NT)parentsexist_ok save_selector_stream_traces.jsonF) ensure_asciiindentr}utf-8encoding)mkdir write_textjsondumps)rRout_dirs r write_outputsrsC MM$M. 11== 4eA6=>rcntjt}|jdtt dz |jdtt dz |jdtt |jdtd |j}t|jjtj|jjd |j }t#||j$t'd t)|d |j$dz y)N) descriptionz--exez Hwanse2.exe)typedefaultz--leaf-streamszsave_selector_leaf_streams.jsonz --out-dirz --max-steps rrzwrote z save selector stream traces -> r)argparseArgumentParser__doc__ add_argumentrROOTOUTrc parse_argsrhr read_bytesrloads leaf_streams read_textrQrrprintr)parserargsrRs rmainrs  $ $ 9F dD=4HI (tSCd=de  $<  C<    D   4$$...@A  D $ % F3t9+=dllMo>o=p qrr__main__)rrr list[dict]rrcreturn int | None)rrrrr'rcrdict)r3rrz str | None) rrrrrOzdict[int, str]rPrcrQrcrr)r)rrrdrrQrcrr)rRrrstr)rRrrrrNone)rr)#r __future__rrrrrpathlibrprobe_exe_scene_tablesrrrsummarize_script_handler_tablerr r r r __file__resolverrrrrr8r;rXrhrrrr__name__r:rrrsO" PPO__ H~''* Ul42 0f* F6r58 s  zFr