diff --git a/benchmarks/auto_tune/auto_tune.sh b/benchmarks/auto_tune/auto_tune.sh index a245e2022e605..f6435afeaa6be 100644 --- a/benchmarks/auto_tune/auto_tune.sh +++ b/benchmarks/auto_tune/auto_tune.sh @@ -3,7 +3,13 @@ # This script aims to tune the best server parameter combinations to maximize throughput for given requirement. # See details in README (benchmarks/auto_tune/README.md). -TAG=$(date +"%Y_%m_%d_%H_%M") +TAG=${TAG:-$(date +"%Y_%m_%d_%H_%M")} + +# Sanitize TAG to prevent path traversal +if [[ "$TAG" == *..* || "$TAG" == /* ]]; then + echo "Error: TAG cannot contain '..' or be an absolute path." >&2 + exit 1 +fi SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd ) VLLM_LOGGING_LEVEL=${VLLM_LOGGING_LEVEL:-INFO} BASE=${BASE:-"$SCRIPT_DIR/../../.."}