diff --git a/scripts/vuln01_regression.sh b/scripts/vuln01_regression.sh index c6cd0a1..3e62c58 100755 --- a/scripts/vuln01_regression.sh +++ b/scripts/vuln01_regression.sh @@ -4,6 +4,7 @@ set -euo pipefail BASE_URL="${BASE_URL:-https://wx.sstbc.com}" COURSE_ID="${COURSE_ID:-}" CURL_TIMEOUT="${CURL_TIMEOUT:-20}" +CURL_RESOLVE="${CURL_RESOLVE:-}" TMP_DIR="$(mktemp -d)" trap 'rm -rf "$TMP_DIR"' EXIT @@ -16,6 +17,9 @@ CURL_ARGS=(-sS -L --max-time "$CURL_TIMEOUT") if [[ "${CURL_INSECURE:-0}" == "1" ]]; then CURL_ARGS+=(-k) fi +if [[ -n "$CURL_RESOLVE" ]]; then + CURL_ARGS+=(--resolve "$CURL_RESOLVE") +fi request() { local name="$1"