From f307c7cd889b0d8a6a13afb2f1113b0161d2e951 Mon Sep 17 00:00:00 2001
From: Summer-Gu <37869445+gubinjie@users.noreply.github.com>
Date: Mon, 10 Feb 2025 10:21:31 +0800
Subject: [PATCH] feat: Docker adds SSRF-related timeout settings (#13395)

---
 docker/.env.example        | 4 ++++
 docker/docker-compose.yaml | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/docker/.env.example b/docker/.env.example
index b21bdc7085..3bc79059dc 100644
--- a/docker/.env.example
+++ b/docker/.env.example
@@ -887,6 +887,10 @@ SSRF_HTTP_PORT=3128
 SSRF_COREDUMP_DIR=/var/spool/squid
 SSRF_REVERSE_PROXY_PORT=8194
 SSRF_SANDBOX_HOST=sandbox
+SSRF_DEFAULT_TIME_OUT=5
+SSRF_DEFAULT_CONNECT_TIME_OUT=5
+SSRF_DEFAULT_READ_TIME_OUT=5
+SSRF_DEFAULT_WRITE_TIME_OUT=5
 
 # ------------------------------
 # docker env var for specifying vector db type at startup
diff --git a/docker/docker-compose.yaml b/docker/docker-compose.yaml
index 67207fd466..33379e4b8c 100644
--- a/docker/docker-compose.yaml
+++ b/docker/docker-compose.yaml
@@ -377,6 +377,10 @@ x-shared-env: &shared-api-worker-env
   SSRF_COREDUMP_DIR: ${SSRF_COREDUMP_DIR:-/var/spool/squid}
   SSRF_REVERSE_PROXY_PORT: ${SSRF_REVERSE_PROXY_PORT:-8194}
   SSRF_SANDBOX_HOST: ${SSRF_SANDBOX_HOST:-sandbox}
+  SSRF_DEFAULT_TIME_OUT: ${SSRF_DEFAULT_TIME_OUT:-5}
+  SSRF_DEFAULT_CONNECT_TIME_OUT: ${SSRF_DEFAULT_CONNECT_TIME_OUT:-5}
+  SSRF_DEFAULT_READ_TIME_OUT: ${SSRF_DEFAULT_READ_TIME_OUT:-5}
+  SSRF_DEFAULT_WRITE_TIME_OUT: ${SSRF_DEFAULT_WRITE_TIME_OUT:-5}
   EXPOSE_NGINX_PORT: ${EXPOSE_NGINX_PORT:-80}
   EXPOSE_NGINX_SSL_PORT: ${EXPOSE_NGINX_SSL_PORT:-443}
   POSITION_TOOL_PINS: ${POSITION_TOOL_PINS:-}