Mission Statement: Given a source code repository in C++/C, indicate areas that have a higher probability to contain a vulnerability
This repo contains:
- Modified implementations of the
LineVul
model - A Fastapi server for model inference
- A VSCode extension that communicates with the server (not yet)
Server installation might be a bit tricky.
git clone https://github.com/Royz2123/IMDBug.git
cd IMDBug
The project runs on Python3
, and is dependent on:
pip install gdown
pip install transformers
pip install captum
pip install torch
pip install numpy
pip install tqdm
pip install pickle
pip install sklearn
pip install pandas
pip install tokenizers
pip install language_data
pip install tree_sitter
You can also refer to scripts/make_env.sh
for a more detailed description of the installation press.
In order to install the models, go to scripts
and then run download_models
. This will attempt to
download all the supported models.
Just go to scripts
and run the run.bat
script (which in turn runs the imdbug_app.py
module, you can just run
this directly yourself)
cd \path\to\IMDBug
cd scripts
run.bat
And you're done!