Update Dockerfile

Dockerfile

@@ -21,6 +21,10 @@ RUN apt-get update && apt-get install -y \
     dbus-x11 \
     && rm -rf /var/lib/apt/lists/*
 
+# 创建非root用户
+RUN useradd -m -d /home/chrome chrome \
+    && chown -R chrome:chrome /home/chrome
+
 # 设置工作目录
 WORKDIR /app
 
@@ -36,22 +40,26 @@ ENV DISPLAY=:99
 ENV CHROME_BIN=/usr/bin/chromium
 ENV CHROMEDRIVER_PATH=/usr/bin/chromedriver
 ENV PYTHONUNBUFFERED=1
+ENV HOME=/home/chrome
 
-# 创建必要的目录和文件，设置权限
-RUN mkdir -p /tmp/.X11-unix && \
-    chmod 1777 /tmp/.X11-unix && \
-    mkdir -p /tmp/chrome-data && \
+# 创建必要的目录和文件
+RUN mkdir -p /home/chrome/.X11-unix && \
+    mkdir -p /home/chrome/chrome-data && \
     touch /app/cookies.txt && \
-    chmod 777 /app/cookies.txt
+    chmod 777 /app/cookies.txt && \
+    chown -R chrome:chrome /app /home/chrome
 
 # 创建启动脚本
 RUN echo '#!/bin/bash\n\
-mkdir -p /tmp/.X11-unix\n\
-chmod 1777 /tmp/.X11-unix\n\
+export DISPLAY=:99\n\
 Xvfb :99 -screen 0 1024x768x16 -ac &\n\
 sleep 2\n\
 python nb2.py' > /app/start.sh \
-    && chmod +x /app/start.sh
+    && chmod +x /app/start.sh \
+    && chown chrome:chrome /app/start.sh
+
+# 切换到非root用户
+USER chrome
 
 # 启动脚本
 CMD ["/bin/bash", "/app/start.sh"]