treforbenbow commited on
Commit
cb26259
·
verified ·
1 Parent(s): 80c824d

Upload reproduce.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. reproduce.py +13 -0
reproduce.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """VULN-012: Heap OOB Read in convertInt32Data<T>"""
2
+ import sys, tensorrt as trt
3
+
4
+ models = ['int8_oob_read.onnx','bool_oob_read.onnx','float16_oob_read.onnx','bfloat16_oob_read.onnx']
5
+ for m in models:
6
+ print(f'Testing: {m}')
7
+ logger = trt.Logger(trt.Logger.ERROR)
8
+ builder = trt.Builder(logger)
9
+ network = builder.create_network(1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH))
10
+ parser = trt.OnnxParser(network, logger)
11
+ ok = parser.parse_from_file(m)
12
+ print(f' Result: {"parsed" if ok else "parse fail"}')
13
+ print('Done. If no crash, vuln may be patched.')