Skip to content

README.md

Latest commit

 

History

History
158 lines (121 loc) · 4.26 KB

README.md

File metadata and controls

158 lines (121 loc) · 4.26 KB

Artifacts for the Paper "Exploring LLMs for Initial Seed Augmentation in Smart Contract Fuzzing with Vulnerability-Guided Prioritization"

This repository contains the experimental artifacts supporting our research.

System Requirements

  • Ubuntu 18.04+ (or similar Linux distribution)
  • 8GB+ RAM, 20GB+ disk space
  • Python 3.8+

Setup Instructions

To reproduce the results, follow the steps below:

1. Clone the Repository and Submodules

git clone https://github.com/pamunb/SmartChat-Artifact.git
cd SmartChat-Artifact
git submodule update --init --recursive

2. Install Dependencies

  • Install Poetry for Python package management.
  • Install .NET SDK 5.0 to build SmartChat.
  • Install Docker if not already available.

3. Build and Configure

cd SmartChat
make

Then, install Python dependencies using Poetry:

cd <repo-root-dir>/SmartChat/script
poetry install

cd <repo-root-dir>/scripts
poetry install

RQ1: LLMs for Initial Fuzzing Seed Generation

To generate the graphs for RQ1:

cd rq1
find . -name "*.tar.xz" -exec tar -xJf {} \;
./get_metrics.sh
./get_coverage.sh
./graphs.sh

RQ2: Ablation and Temperature Comparison

To generate the results and plots:

cd rq2
find . -name "*.tar.xz" -exec tar -xJf {} \;

./create_data.sh ./output_B1_llm_only_buggain llm_only_buggain
./create_data.sh ./output_B1_llm_only llm_only

Generate bar plots:

python3.8 ../scripts/plot_b1_bar.py ./llm_only_buggain/merged_all_llm_only_buggain* \
  --models "GPT4.1-Mini,Llama3.3-70B" \
  --colors "#8facd2,#dd7e7e" \
  --output llm_only_buggain.pdf

python3.8 ../scripts/plot_b1_bar.py ./llm_only/merged_all_llm_only* \
  --models "GPT4.1-Mini,Llama3.3-70B" \
  --colors "#8facd2,#dd7e7e" \
  --output llm_only.pdf

Statistical Evaluation

For comparison with the Smartian Data Flow version:

python3.8 ../scripts/statistical_eval.py ./llm_only_buggain/llm_only_buggain.csv base_dfa.csv

For comparison with the Smartian Random Seeds version:

python3.8 ../scripts/statistical_eval.py ./llm_only_buggain/llm_only_buggain.csv base_rand.csv

RQ3: Comparison with Smartian and ConFuzzius

To extract and prepare data:

cd rq3
find . -name "*.tar.xz" -exec tar -xJf {} \;
./create_data.sh

Plotting for GPT4.1-Mini:

python3.8 ../scripts/plot_b2_graph.py \
  ./smartian.cov ./confuzzius.cov \
  ./smartchat_cot_gpt4.1mini_0.4.cov \
  ./smartchat_code_gpt4.1mini_0.4.cov \
  ./smartchat_nocode_gpt4.1mini_0.4.cov \
  -o comparison_plot_combined_gpt.pdf \
  --baseline-name "Smartian" \
  --method-names "ConFuzzius" "SᴍᴀʀᴛCʜᴀᴛ CoT Code (GPT4.1-Mini)" "SᴍᴀʀᴛCʜᴀᴛ Code (GPT4.1-Mini)" "SᴍᴀʀᴛCʜᴀᴛ ABI (GPT4.1-Mini)" \
  --left-ylim 60 95 \
  --legend-loc "lower center" \
  --subplot-layout \
  --left-data-files ./smartian.bug ./confuzzius.bug ./smartchat_cot_gpt4.1mini_0.4.bug ./smartchat_code_gpt4.1mini_0.4.bug ./smartchat_nocode_gpt4.1mini_0.4.bug \
  --left-ylabel "Total # of Bugs found" \
  --ylabel "Instruction Coverage (%)"

Plotting for Llama3.3-70B:

  python3.8 ../scripts/plot_b2_graph.py \
  ./smartian.cov ./confuzzius.cov \
  ./smartchat_cot_Llama3.3-70B_0.4.cov \
  ./smartchat_code_Llama3.3-70B_0.4.cov \
  ./smartchat_nocode_Llama3.3-70B_0.4.cov \
  -o comparison_plot_combined_llama.pdf \
  --baseline-name "Smartian" \
  --method-names "ConFuzzius" "SᴍᴀʀᴛCʜᴀᴛ CoT Code (Llama3.3-70B)" "SᴍᴀʀᴛCʜᴀᴛ Code (Llama3.3-70B)" "SᴍᴀʀᴛCʜᴀᴛ ABI (Llama3.3-70B)" \
  --left-ylim 60 95 \
  --legend-loc "lower center" \
  --subplot-layout \
  --left-data-files ./smartian.bug ./confuzzius.bug ./smartchat_cot_Llama3.3-70B_0.4.bug ./smartchat_code_Llama3.3-70B_0.4.bug ./smartchat_nocode_Llama3.3-70B_0.4.bug \
  --left-ylabel "Total # of Bugs found" \
  --ylabel "Instruction Coverage (%)"

Statistical Evaluation

For comparison with the Smartian:

python3.8 ../scripts/statistical_eval.py ./smartchat.csv ./smartian.csv

For comparison with the ConFuzzius:

python3.8 ../scripts/statistical_eval.py ./smartchat.csv ./confuzzius.csv