Merge pull request #7 from AndrewEastwood/main

Improved entrypoint script: accept ip
nanabanano · Mar 13, 2022 · 1eaf356 · 1eaf356
2 parents dfaf3c4 + 74ae733
commit 1eaf356
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/docker_entrypoint.sh b/docker_entrypoint.sh
@@ -1,5 +1,13 @@
 #!/bin/bash
 
 HOSTNAME=$1
-IP=$(dig +short "$HOSTNAME"| tail -n 1)
-python3 -u DRipper.py -s "$IP" -t 135 -p 443
+if [ -z "$HOSTNAME" ] ; then
+  echo error: check your hostname.
+else
+  IP=$(dig +short "$HOSTNAME"| tail -n 1)
+  if [ ! -z "$IP" ] ; then
+    python3 -u DRipper.py -s "$IP" -t 135 -p 443
+  else
+    python3 -u DRipper.py -s "$HOSTNAME" -t 135 -p 443
+  fi
+fi