StartBaselineTests.sh

#!/bin/bash

RED='\033[0;31m'
GREEN='\033[0;32m'
GREY='\033[1;30m'
NC='\033[0m'

ENB_LOGFILE="${PWD}/fuzzLogs/enbFuzzLogs.txt"
UE_LOGFILE="${PWD}/fuzzLogs/epcFuzzLogs.txt"
EPC_LOGFILE="${PWD}/fuzzLogs/ueFuzzLogs.txt"
ENB_PCAP_DIR="${PWD}/fuzzLogs/pcap/"

for (( i=$1; i<=$2; i++ ))
do
    now=$(date +"%T")
    # contruct header for logs and stdout
    curr_test="DEFAULT"
    test_header="-------------FUZZ TEST $i ($curr_test)------------------"

    # print to logs and stdout
    echo -e "\n${GREEN}$test_header${NC}\n"
    echo $test_header >> fuzzLogs/epcFuzzLogs.txt
    echo $test_header >> fuzzLogs/enbFuzzLogs.txt
    echo $test_header >> fuzzLogs/epcFuzzLogs.txt

    # start epc, append output to log file, store pid, echo pid
    /home/dsetareh/srsRAN/build/srsepc/src/srsepc >> fuzzLogs/epcFuzzLogs.txt &
    epc_pid=$!

    # start enb, append output to log file, store pid, echo pid
    /home/dsetareh/srsRAN/build/srsenb/src/srsenb --rf.device_name=zmq --rf.device_args="fail_on_disconnect=true,tx_port=tcp://*:2000,rx_port=tcp://localhost:2001,id=enb,base_srate=23.04e6" >> fuzzLogs/enbFuzzLogs.txt  &
    enb_pid=$!
    sleep 1 # * 1 second wait

    # start ue, append output to log file, store pid, echo pid
    /home/dsetareh/srsRAN/build/srsue/src/srsue >> fuzzLogs/ueFuzzLogs.txt &
    ue_pid=$!
    echo -e "${GREEN}STARTING${NC}: UE($ue_pid) - EPC ($epc_pid) - ENB ($enb_pid)\n"


    echo -e "${RED}!!${NC} FUZZING ${RED}!!${NC}\n" # fuzzing must be occuring in the bg
    sleep 3 # ! 3 second wait, this is how long the full env lasts before killing begins

    # use stored pid's to kill ue, then enb, then epc
    echo -e "${RED}STOPPING${NC}: UE($ue_pid) - EPC ($epc_pid) - ENB ($enb_pid)\n"
    kill -KILL $ue_pid
    kill -KILL $enb_pid
    kill -KILL $epc_pid

    echo -e "${GREY}" # kill output is dimmed, as not useful
    sleep 3 # ! 3 second wait
    echo -e "${NC}"

    # save pcap file for enb
    
    echo -e "\n${GREEN}Saving ENB PCAP as fuzzLogs/pcap/DEFAULT/$now.pcap${NC}\n"
    cp enb.pcap fuzzLogs/pcap/DEFAULT/$now.pcap
    

    # log competion of iteration, continue
    echo -e "Test complete, Log written, starting next test in 1 seconds...\n"
    sleep 3 # ! 1 second wait
done

echo Fuzzing tests complete!