70 lines
1.6 KiB
Bash
70 lines
1.6 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
# Copyright (C) 2022 Intel Corporation
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
|
||
|
|
# Store pids of parallel fuzzer tests
|
||
|
|
pids=()
|
||
|
|
|
||
|
|
function cleanup() {
|
||
|
|
rm -rf "$@"
|
||
|
|
# Make sure that there is no process left hanging
|
||
|
|
kill -9 "${pids[@]}" || :
|
||
|
|
}
|
||
|
|
|
||
|
|
function get_testn() {
|
||
|
|
local fuzz_num=$1
|
||
|
|
local mem_size=$2
|
||
|
|
local nproc
|
||
|
|
nproc=$(nproc)
|
||
|
|
# Choose lower value, best case scenario is one test per core
|
||
|
|
local testn=$((fuzz_num < nproc ? fuzz_num : nproc))
|
||
|
|
|
||
|
|
export HUGEMEM=$((mem_size * testn))
|
||
|
|
setup
|
||
|
|
TESTN=$testn
|
||
|
|
}
|
||
|
|
|
||
|
|
function start_llvm_fuzz_all() {
|
||
|
|
local testn=$1 # Number of test to run in parallel
|
||
|
|
local fuzz_num=$2 # Number of fuzzer tests
|
||
|
|
local time=$3 # Time available for all fuzzers
|
||
|
|
local core
|
||
|
|
|
||
|
|
# Calculate time for a single test and multiply it by number
|
||
|
|
# of test execute in parallel and round it up to 1 sek per test
|
||
|
|
timen=$(printf %d $(((time / fuzz_num) * testn)))
|
||
|
|
timen=$((timen == 0 ? 1 : timen))
|
||
|
|
|
||
|
|
for ((i = 0; i < fuzz_num; i++)); do
|
||
|
|
core=$(printf "0x%x" $((0x1 << (i % testn))))
|
||
|
|
start_llvm_fuzz $i $timen $core &> $output_dir/llvm/llvm_"$FUZZER"_$i.txt &
|
||
|
|
pids+=($!)
|
||
|
|
|
||
|
|
# Wait for processes to finish
|
||
|
|
if (((i + 1) % testn == 0 || fuzz_num - i - 1 == 0)); then
|
||
|
|
(
|
||
|
|
sleep $((timen * 10 + 100))
|
||
|
|
echo "Timeout $time"
|
||
|
|
exit 1
|
||
|
|
) &
|
||
|
|
timeout_pid=$!
|
||
|
|
for pid in "${pids[@]}"; do
|
||
|
|
wait $pid
|
||
|
|
done
|
||
|
|
kill $timeout_pid || :
|
||
|
|
pids=()
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
function start_llvm_fuzz_short() {
|
||
|
|
local fuzz_num=$1
|
||
|
|
local time=$2
|
||
|
|
|
||
|
|
for ((i = 0; i < fuzz_num; i++)); do
|
||
|
|
start_llvm_fuzz $i $time 0x1
|
||
|
|
done
|
||
|
|
}
|