From 26b0786a4ed052573c8b0a13ca908eba1dc5d81f Mon Sep 17 00:00:00 2001
From: 0xallam <ahmed39652003@gmail.com>
Date: Fri, 16 Jan 2026 17:16:45 -0800
Subject: [PATCH] fix: replace pgrep with health check for tool server
 validation

---
 containers/docker-entrypoint.sh | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/containers/docker-entrypoint.sh b/containers/docker-entrypoint.sh
index 1a905dc..ee57114 100644
--- a/containers/docker-entrypoint.sh
+++ b/containers/docker-entrypoint.sh
@@ -168,14 +168,19 @@ sudo -E -u pentester \
   --port="$TOOL_SERVER_PORT" \
   --timeout="$TOOL_SERVER_TIMEOUT" > "$TOOL_SERVER_LOG" 2>&1 &
 
-sleep 3
-if ! pgrep -f "tool_server.py" > /dev/null; then
-  echo "ERROR: Tool server process failed to start"
-  echo "=== Tool server log ==="
-  cat "$TOOL_SERVER_LOG" 2>/dev/null || echo "(no log)"
-  exit 1
-fi
-echo "✅ Tool server started on port $TOOL_SERVER_PORT"
+for i in {1..10}; do
+  if curl -s "http://127.0.0.1:$TOOL_SERVER_PORT/health" | grep -q '"status":"healthy"'; then
+    echo "✅ Tool server healthy on port $TOOL_SERVER_PORT"
+    break
+  fi
+  if [ $i -eq 10 ]; then
+    echo "ERROR: Tool server failed to become healthy"
+    echo "=== Tool server log ==="
+    cat "$TOOL_SERVER_LOG" 2>/dev/null || echo "(no log)"
+    exit 1
+  fi
+  sleep 1
+done
 
 echo "✅ Container ready"