2j<dZddlmZddlZddlZddlZddlZej5ddlm Z m Z m Z m Z m Z mZdddn #1swxYwYdZdZdZdZd Zd Zd Zed kr edSdS) aAtheris fuzz harness for onnx.compose. Two input paths are exercised per iteration, selected by a fuzzer-controlled toggle byte read from the *tail* of the input: * Raw -> a 4-byte big-endian length prefix splits the remaining bytes into m1 | m2, each parsed via onnx.load_model_from_string. Catches protobuf parser bugs and bugs reachable only through crafted serialized models the structured builder will not produce. * Structured -> helper.make_model from FuzzedDataProvider builds two small linear graphs with predictable input/output names, so merge_models reaches its connect_io/add_prefix logic on most iterations rather than only when the parser happens to accept a random byte string. Further bits of the toggle byte select prefix1/prefix2 (add_prefix collision-resolution path) and a randomized (vs. derived) io_map, so a single harness covers merge_models, merge_graphs, check_overlapping_names, the recursive connect_io subgraph rewrite, and add_prefix. ) annotationsN) TensorProtocheckercomposedefshelperload_model_from_string)ReluSigmoidTanhAbsNegIdentitycDtj|tjdgS)NN)rmake_tensor_value_inforFLOATnames GC:\Projects\Historical\venv\Lib\site-packages\onnx/fuzz/fuzz_compose.py _value_infor-s  ({/@3% H HHc\|dd}d|}g}t|D]m}t|dttdz }d|d|}|t j||g|g|}nd|} |t jd|g| gt j|d |td|gt| g} t j | t j d |g S) a6Build a small linear graph with predictable input/output names. Naming the sole input/output `In`/`Out` lets the derived io_map (m1's outputs -> m2's inputs) connect on most iterations, so merge_models' rewrite logic is reached without relying on the raw protobuf parser to produce compatible names. `opset` is shared between both models: merge_models rejects mismatched opset_import up front, so picking it independently per model would make the structured path rarely reach the merge/connect logic it exists to exercise. rInv_Outrg) opset_imports) ConsumeIntInRangerange_UNARYlenappendr make_node make_graphr make_model make_opsetid) fdptagopsetn_opslastnodesiopnxtout_namegraphs r _build_modelr71s4  ! !!Q ' 'E ::D E 5\\ C))!S[[1_== >#mmmm V%b4&3%88999S{{H LL!*tfxjAABBB   y3yy;zCzz223k(6K6K5L  E  U63Fr53Q3Q2R S S SSrcd|jjD}d|jjD}tt ||dS)Ncg|] }|j Sr.0os r z#_derived_io_map..M , , ,qAF , , ,rcg|] }|j Sr:rr<r2s rr>z#_derived_io_map..N * * *a16 * * *rFstrict)r6outputinputlistzip)m1m2outsinss r_derived_io_maprMLsM , ,BHO , , ,D * *28> * * *C D#e,,, - --rc d|jjD}d|jjD}|r|sgStj|}|||||dtt|t|}tt|d||d|dS)Ncg|] }|j Sr:rr;s rr>z"_random_io_map..Sr?rcg|] }|j Sr:rrAs rr>z"_random_io_map..TrBrrTrC) r6rErFrandomRandomshufflerandintminr&rGrH)datarIrJrKrLrngns r_random_io_maprYRs , ,BHO , , ,D * *28> * * *C s -  CKKKK As3t99c#hh//00A D!Hc"1"gd333 4 44rcbt|dkrdS|d}|dd}t|dz}t|dz}t|dz} |r^tj|}|dt j}t|d|}t|d|} nt|dkrdStj d|dd\} |dd} | t| krdSt| d| }t| | d} |rt||| nt|| } |rdnd} |rd nd}tj|| | | | }tj|d dS#t"$rYdSwxYw) Nrrz>Ig1_g2_)prefix1prefix2T) full_check)r&boolatherisFuzzedDataProviderr#ronnx_opset_versionr7structunpackr rYrMr merge_modelsr check_model Exception)rVtogglesbody use_prefixuse_structureduse_random_io_mapr,r.rIrJn1restio_maprarbmergeds r TestOneInputrv^s 4yy1}}2hG 9Dgn%%J'D.))NWt^,,  3,T22C))!T-D-F-FGGEc1e,,Bc1e,,BB4yy1}}M$RaR11ER8DCII~~'SbS 22B'RSS 22B! )N4R ( ( ( R((  &/%%4%/%%4%b"fgwWWWFt444444   s A3F rs*#""""" W!! ?IIITTT6... 5 5 5+++\  zDFFFFFsAA  A