Proxy Settings added to OWASP Benchmark Crawler to enable different hosts (OWASP-Benchmark#1)

arunmuthu255 · web-flow · commit e7ac1abc0ee6 · 2021-09-28T14:18:04.000+05:30
diff --git a/runCrawler.bat b/runCrawler.bat
@@ -1,3 +1,17 @@
-# source "scripts/verifyBenchmarkPluginAvailable.sh" - Don't have .bat version of this (yet)
-CALL mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml
-
+@echo off
+REM source "scripts/verifyBenchmarkPluginAvailable.sh" - Don't have .bat version of this (yet)
+set /A args_count=0    
+for %%A in (%*) do set /A args_count+=1
+if %args_count% NEQ 0 (
+    if %args_count% EQU 2 (
+        CALL mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml -DproxyHost="%1" -DproxyPort="%2"
+    ) else (
+        echo Error!!
+        echo -------
+        echo To run the Crawler for localhost, execute runCrawler.bat with no arguments.
+        echo To run the Crawler for remote host, execute runCrawler.bat with only 2 arguments, proxy-host and proxy-port.
+        echo Example: runCrawler.bat 192.168.0.1 53452
+    )
+) else (
+    CALL mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml
+)
diff --git a/runCrawler.sh b/runCrawler.sh
@@ -1,3 +1,13 @@
+#!/bin/sh
 source "scripts/verifyBenchmarkPluginAvailable.sh"
-mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml
-
+if [ $# -eq 2 ]; then
+    mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml -DproxyHost="$1" -DproxyPort="$2"
+elif [ $# -eq 0 ]; then
+    mvn org.owasp:benchmarkutils-maven-plugin:run-crawler -DcrawlerFile=data/benchmark-crawler-http.xml
+else
+    echo "Error!!"
+    echo "-------"
+    echo "To run the Crawler for localhost, execute runCrawler.sh with no arguments."
+    echo "To run the Crawler for remote host, execute runCrawler.sh with only 2 arguments, proxy-host and proxy-port."
+    echo "Example: ./runCrawler.sh 192.168.0.1 53452"
+fi
