LilithAdam5 commited on
Commit
b472971
·
verified ·
1 Parent(s): 47d3e39

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +123 -3
README.md CHANGED
@@ -1,3 +1,123 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ # Model Card for tensorrt-deserialization-poc
6
+
7
+ This repository contains a **Proof-of-Concept (PoC)** demonstrating unsafe deserialization in NVIDIA TensorRT engines (`IRuntime::deserializeCudaEngine`). The PoC triggers a type hash mismatch that may lead to remote code execution or GPU crash. This card documents the PoC, environment, and usage instructions for security research and bug bounty submissions.
8
+
9
+ ## Model Details
10
+
11
+ ### Model Description
12
+
13
+ - **Developed by:** ZEUS / ATHENA
14
+ - **Shared by:** ZEUS
15
+ - **Model type:** Security PoC / Exploit Demonstration
16
+ - **Language(s) (NLP):** Python
17
+ - **License:** Apache 2.0
18
+ - **Finetuned from model [optional]:** N/A
19
+
20
+ ### Model Sources
21
+
22
+ - **Repository:** [tensorrt-deserialization-poc](https://huggingface.co/LilithAdam5/tensorrt-deserialization-poc)
23
+ - **Paper [optional]:** N/A
24
+ - **Demo [optional]:** N/A
25
+
26
+ ## Uses
27
+
28
+ ### Direct Use
29
+
30
+ This PoC is intended for **security researchers and bug bounty programs** to safely reproduce the unsafe deserialization behavior in TensorRT.
31
+
32
+ ### Downstream Use
33
+
34
+ - Could be integrated into internal security testing pipelines to validate TensorRT engine safety.
35
+ - Not intended for production use; execution may crash GPUs or systems if misused.
36
+
37
+ ### Out-of-Scope Use
38
+
39
+ - This PoC is **not a machine learning model** and should not be used for training, inference, or production ML pipelines.
40
+ - Should not be executed on unisolated production environments.
41
+
42
+ ## Bias, Risks, and Limitations
43
+
44
+ - **Risks:** Triggering the PoC may crash GPUs or expose unsafe execution paths.
45
+ - **Limitations:** Only tested with TensorRT 10.13.3.9 on CUDA 13.x and Python 3.13.
46
+ - Users should run in isolated virtual environments.
47
+
48
+ ### Recommendations
49
+
50
+ - Always run in a **sandboxed GPU environment**.
51
+ - Use the provided safe wrapper for triage and bug bounty submissions.
52
+
53
+ ## How to Get Started with the PoC
54
+
55
+ 1. Create and activate a Python virtual environment:
56
+ ```bash
57
+ python3 -m venv lilith_venv
58
+ source lilith_venv/bin/activate
59
+
60
+ pip install tensorrt
61
+ python poc_trt_rce.py
62
+ import tensorrt as trt
63
+
64
+ with open("safe_trt_crash.trt", "rb") as f:
65
+ engine_data = f.read()
66
+
67
+ runtime = trt.Runtime(trt.Logger(trt.Logger.WARNING))
68
+
69
+ try:
70
+ engine = runtime.deserialize_cuda_engine(engine_data)
71
+ if engine:
72
+ print("[!] Deserialization succeeded (unexpected)")
73
+ except Exception as e:
74
+ print("[TRT] Error during deserialization:", e)
75
+ Environment Details
76
+
77
+ OS: Ubuntu 22.04
78
+
79
+ Python: 3.13
80
+
81
+ CUDA: 13.x
82
+
83
+ TensorRT: 10.13.3.9
84
+
85
+ Hardware: NVIDIA GPU (for runtime deserialization)
86
+
87
+ Technical Specifications
88
+
89
+ Objective: Demonstrate unsafe deserialization in TensorRT engines for security research.
90
+
91
+ PoC Language: Python
92
+
93
+ Serialized Engine File: safe_trt_crash.trt
94
+
95
+ Citation
96
+
97
+ Use this repository reference when citing in security reports or bug bounty submissions:
98
+
99
+ BibTeX:
100
+
101
+ @misc{LilithAdam5_2025_tensorrt,
102
+ title={tensorrt-deserialization-poc},
103
+ author={ZEUS},
104
+ year={2025},
105
+ howpublished={Hugging Face Hub},
106
+ url={https://huggingface.co/LilithAdam5/tensorrt-deserialization-poc}
107
+ }
108
+
109
+
110
+ APA:
111
+ ZEUS. (2025). tensorrt-deserialization-poc. Hugging Face Hub. https://huggingface.co/LilithAdam5/tensorrt-deserialization-poc
112
+
113
+ Model Card Authors
114
+
115
+ ZEUS
116
+
117
+ ATHENA
118
+
119
+ Model Card Contact
120
+
121
+ Email: [optional]
122
+
123
+ GitHub / Hugging Face: LilithAdam5