#!/usr/bin/env bash

# -------------------------------------------------------
# 0. Create/Configure a Non-Root User (Optional)
# -------------------------------------------------------
# Adjust group names, etc. according to your environment

sudo useradd -m -d /users/user42 -s /bin/bash user42
sudo passwd user42
sudo usermod -aG sudo user42
sudo usermod -aG kclsystemfuzz-PG user42

# Adjust ownership/permissions as desired:
sudo chown -R user42:user42 /users/user42
sudo chmod 777 /users/user42

# -------------------------------------------------------
# 1. Update & Upgrade the System
# -------------------------------------------------------
sudo apt-get update && sudo apt-get upgrade -y

# -------------------------------------------------------
# 2. Install Docker (Host Only)
# -------------------------------------------------------
sudo apt-get install -y \
  ca-certificates \
  curl \
  gnupg \
  lsb-release \
  wget \
  git

sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg \
  | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg

echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] \
  https://download.docker.com/linux/ubuntu \
  $(lsb_release -cs) stable" \
  | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

sudo apt-get update
sudo apt-get install -y \
  docker-ce \
  docker-ce-cli \
  containerd.io \
  docker-buildx-plugin \
  docker-compose-plugin

# Optionally allow user42 to run docker without sudo:
sudo usermod -aG docker user42
# -------------------------------------------------------
# 3. Configure Core Dumps
# -------------------------------------------------------
echo "core" | sudo tee /proc/sys/kernel/core_pattern

# -------------------------------------------------------
# 4. Download Dockerfile & Helper Scripts & Build Fuzzing Image
# -------------------------------------------------------
cd /users/user42
su - user42
wget https://raw.githubusercontent.com/ayseirmak/FuzzdFlags-ASE/refs/heads/main/exp3-fuzz-fuzzdflags-options/exp3-dock.dockerfile
wget https://raw.githubusercontent.com/ayseirmak/FuzzdFlags-ASE/refs/heads/main/extract_fuzz_stat_dir.sh
wget https://raw.githubusercontent.com/ayseirmak/FuzzdFlags-ASE/refs/heads/main/decrypt_queue.sh
chmod +x *.sh

# Build Docker image from the local Dockerfile
docker build -f exp3-dock.dockerfile -t afl-clang-opts-img .

# -------------------------------------------------------
# 5. Prepare Output Directories
# -------------------------------------------------------
mkdir -p fuzz01 fuzz02 fuzz03 fuzz04 fuzz05
chown -R user42:user42 fuzz01 fuzz02 fuzz03 fuzz04 fuzz05
chmod -R 777 fuzz01 fuzz02 fuzz03 fuzz04 fuzz05


# -------------------------------------------------------
# 61. Launch 5 Fuzzing Containers for 1 seed
# -------------------------------------------------------
# NOTE: Adjust --cpuset-cpus according to the actual cores on your m510 node.
# Give each container 3 cores:
# for 16 cpu fuzz01 -> 0-2, fuzz02 -> 3-5, fuzz03 -> 6-8, fuzz04 -> 9-11, fuzz05 -> 12-14

# docker run -d --name fuzz01 --cpuset-cpus="0-2" \
#   -v /users/user42/fuzz01:/users/user42/output-fuzz \
#  afl-clang-opts-img \
#   /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-1 \
#     /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

# docker run -d --name fuzz02 --cpuset-cpus="3-5" \
#   -v /users/user42/fuzz02:/users/user42/output-fuzz \
#  afl-clang-opts-img \
#   /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-1 \
#     /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

# docker run -d --name fuzz03 --cpuset-cpus="6-8" \
#   -v /users/user42/fuzz03:/users/user42/output-fuzz \
#  afl-clang-opts-img \
#   /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-1 \
#     /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

# docker run -d --name fuzz04 --cpuset-cpus="9-11" \
#   -v /users/user42/fuzz04:/users/user42/output-fuzz \
#  afl-clang-opts-img \
#   /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-1 \
#     /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

# docker run -d --name fuzz05 --cpuset-cpus="12-14" \
#   -v /users/user42/fuzz05:/users/user42/output-fuzz \
#  afl-clang-opts-img \
#   /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-1 \
#     /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

# echo "All 5 fuzzing containers started."

# -------------------------------------------------------
# 62. Launch 5 Fuzzing Containers for 30 seed
# -------------------------------------------------------
# NOTE: Adjust --cpuset-cpus according to the actual cores on your m510 node.
# Give each container 3 cores:
# for 16 cpu fuzz01 -> 0-2, fuzz02 -> 3-5, fuzz03 -> 6-8, fuzz04 -> 9-11, fuzz05 -> 12-14
docker run -d --name fuzz01 --cpuset-cpus="0-2" \
  -v /users/user42/fuzz01:/users/user42/output-fuzz \
 afl-clang-opts-img \
  /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-30 \
    /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

docker run -d --name fuzz02 --cpuset-cpus="3-5" \
  -v /users/user42/fuzz02:/users/user42/output-fuzz \
 afl-clang-opts-img \
  /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-30 \
    /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

docker run -d --name fuzz03 --cpuset-cpus="6-8" \
  -v /users/user42/fuzz03:/users/user42/output-fuzz \
 afl-clang-opts-img \
  /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-30 \
    /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

docker run -d --name fuzz04 --cpuset-cpus="9-11" \
  -v /users/user42/fuzz04:/users/user42/output-fuzz \
 afl-clang-opts-img \
  /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-30 \
    /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options

docker run -d --name fuzz05 --cpuset-cpus="12-14" \
  -v /users/user42/fuzz05:/users/user42/output-fuzz \
 afl-clang-opts-img \
  /users/user42/24_fuzz.sh run_AFL_conf_clangopt.sh /users/user42/input-seeds-30 \
    /users/user42/output-fuzz -O0 /users/user42/build-clang-options/bin/clang-options


# -------------------------------------------------------
# 71. After Fuzzing to Extract Fuzz Statistics 1 seed
# -------------------------------------------------------
# sudo chown -R user42:user42 /users/user42
# ./extract_fuzz_stat_dir.sh /users/user42
# tar -czvf exp31-1seed-fuzz-results.tar.gz -C /users/user42/ fuzz01 fuzz02 fuzz03 fuzz04 fuzz05 fuzzer_stats_summary.csv

# wget https://github.com/ayseirmak/FuzzdFlags-ASE/releases/download/v1.0.0-alpha.1/build-clangOpt.tar.gz
# tar -zxvf build-clangOpt.tar.gz
# mkdir -p exp31-1seed-fuzz-queue

# export INSTRUMENTED_CLANG_PATH=/users/user42/build/bin/clang
# export CFILES_DIR=/users/user42/llvmSS-minimised-corpus/
# export FILE_COUNT=1811

# Decrypt the queues from each fuzzing container
# ./decrypt_queue.sh fuzz01/default/queue/ exp31-1seed-fuzz-queue/fuzz01-queue
# ./decrypt_queue.sh fuzz02/default/queue/ exp31-1seed-fuzz-queue/fuzz02-queue
# ./decrypt_queue.sh fuzz03/default/queue/ exp31-1seed-fuzz-queue/fuzz03-queue
# ./decrypt_queue.sh fuzz04/default/queue/ exp31-1seed-fuzz-queue/fuzz04-queue
# ./decrypt_queue.sh fuzz05/default/queue/ exp31-1seed-fuzz-queue/fuzz05-queue
# tar -czvf exp31-1seed-fuzz-queue.tar.gz -C /users/user42 exp31-1seed-fuzz-queue 

# Decrypt the hangs from each fuzzing container
# mkdir -p exp31-1seed-fuzz-hangs
# ./decrypt_queue.sh fuzz01/default/hangs/ exp31-1seed-fuzz-hangs/fuzz01-hangs
# ./decrypt_queue.sh fuzz02/default/hangs/ exp31-1seed-fuzz-hangs/fuzz02-hangs
# ./decrypt_queue.sh fuzz03/default/hangs/ exp31-1seed-fuzz-hangs/fuzz03-hangs
# ./decrypt_queue.sh fuzz04/default/hangs/ exp31-1seed-fuzz-hangs/fuzz04-hangs
# ./decrypt_queue.sh fuzz05/default/hangs/ exp31-1seed-fuzz-hangs/fuzz05-hangs
# tar -czvf exp31-1seed-fuzz-hangs.tar.gz -C /users/user42 exp31-1seed-fuzz-hangs 


# -------------------------------------------------------
# 72. After Fuzzing to Extract Fuzz Statistics 30 seeds
# -------------------------------------------------------
sudo chown -R user42:user42 /users/user42
./extract_fuzz_stat_dir.sh /users/user42
tar -czvf exp32-30seed-fuzz-results.tar.gz -C /users/user42/ fuzz01 fuzz02 fuzz03 fuzz04 fuzz05 fuzzer_stats_summary.csv

wget https://github.com/ayseirmak/FuzzdFlags-ASE/releases/download/v1.0.0-alpha.1/build-clangOpt.tar.gz
tar -zxvf build-clangOpt.tar.gz
mkdir -p exp32-30seed-fuzz-queue

export INSTRUMENTED_CLANG_PATH=/users/user42/build/bin/clang
export CFILES_DIR=/users/user42/llvmSS-minimised-corpus/
export FILE_COUNT=1811

# Decrypt the queues from each fuzzing container
./decrypt_queue.sh fuzz01/default/queue/ exp32-30seed-fuzz-queue/fuzz01-queue
./decrypt_queue.sh fuzz02/default/queue/ exp32-30seed-fuzz-queue/fuzz02-queue
./decrypt_queue.sh fuzz03/default/queue/ exp32-30seed-fuzz-queue/fuzz03-queue
./decrypt_queue.sh fuzz04/default/queue/ exp32-30seed-fuzz-queue/fuzz04-queue
./decrypt_queue.sh fuzz05/default/queue/ exp32-30seed-fuzz-queue/fuzz05-queue
tar -czvf exp32-30seed-fuzz-queue.tar.gz -C /users/user42 exp32-30seed-fuzz-queue

# Decrypt the hangs from each fuzzing container
 mkdir -p exp32-30seed-fuzz-hangs
./decrypt_queue.sh fuzz01/default/hangs/ exp32-30seed-fuzz-hangs/fuzz01-hangs
./decrypt_queue.sh fuzz02/default/hangs/ exp32-30seed-fuzz-hangs/fuzz02-hangs
./decrypt_queue.sh fuzz03/default/hangs/ exp32-30seed-fuzz-hangs/fuzz03-hangs
./decrypt_queue.sh fuzz04/default/hangs/ exp32-30seed-fuzz-hangs/fuzz04-hangs
./decrypt_queue.sh fuzz05/default/hangs/ exp32-30seed-fuzz-hangs/fuzz05-hangs
tar -czvf exp32-30seed-fuzz-hangs.tar.gz -C /users/user42 exp32-30seed-fuzz-hangs