Update make_dict_v2.sh

(-) removed default vars ;
(+) added LD_PRELOAD_PATH check
This commit is contained in:
Alexander Shvedov
2024-06-01 01:18:03 +03:00
committed by GitHub
parent c202d80dea
commit 5e708b23c6

View File

@ -1,32 +1,28 @@
#default values
timeout_sec=5
LD_PRELOAD_PATH="/home/${USER}/AFLplusplus/utils/libtokencap/libtokencap.so"
#help
usage() {
echo "Usage: $0 -o <target_output> -b <target_bin> [-t <timeout_sec>] [-p <LD_PRELOAD_PATH>]"
echo "Usage: $0 -o <target_output> -b <target_bin> -p <LD_PRELOAD_PATH> [-t <timeout_sec>]"
echo "Options:"
echo " -o Path to target output directory"
echo " -b Path to target program binary"
echo " -t Timeout in seconds (default: 5)"
echo " -p Path to LD_PRELOAD library (default: ${LD_PRELOAD_PATH})"
echo " -p Path to LD_PRELOAD library"
echo " -t Timeout in seconds"
exit 1
}
#parse cli options
while getopts ":o:b:t:p:" opt; do
while getopts ":o:b:p:t:" opt; do
case $opt in
o) target_output="$OPTARG" ;;
b) target_bin="$OPTARG" ;;
t) timeout_sec="$OPTARG" ;;
p) LD_PRELOAD_PATH="$OPTARG" ;;
t) timeout_sec="$OPTARG" ;;
\?) echo "Invalid option: -$OPTARG" >&2; usage ;;
:) echo "Option -$OPTARG requires an args" >&2; usage ;;
:) echo "Option -$OPTARG requires an argument." >&2; usage ;;
esac
done
#check options
if [ -z "$target_output" ] || [ -z "$target_bin" ]; then
if [ -z "$target_output" ] || [ -z "$target_bin" ] || [ -z "$LD_PRELOAD_PATH" ]; then
echo "Error: Missing mandatory opts" >&2
usage
fi