From 53e50c2bf84bf098e600de61a46df5a41bbb7898 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=B5=A9=E6=96=8C?= <newsclan@gmail.com>
Date: Wed, 22 Jan 2025 09:18:02 +0800
Subject: [PATCH] Update restore.sh

---
 trunk/install/restore.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/trunk/install/restore.sh b/trunk/install/restore.sh
index 322d6b2a100..855afb7a269 100755
--- a/trunk/install/restore.sh
+++ b/trunk/install/restore.sh
@@ -14,6 +14,10 @@ SERVER=`cat $config|grep 'OJ_HOST_NAME' |awk -F= '{print $2}'`
 USER=`cat $config|grep 'OJ_USER_NAME' |awk -F= '{print $2}'`
 PASSWORD=`cat $config|grep 'OJ_PASSWORD' |awk -F= '{print $2}'`
 DATABASE=`cat $config|grep 'OJ_DB_NAME' |awk -F= '{print $2}'`
+web_user=`grep www /etc/passwd|awk -F: '{print $1}'`
+
+chmod 770 /home/judge/src/web/upload
+chown $web_user -R /home/judge/src/web/upload
 mkdir hustoj-restore
 cd hustoj-restore
 MAIN="../$1"
@@ -21,10 +25,10 @@ MAIN="../$1"
 tar xjf $MAIN
 mv /home/judge/data /home/judge/data.del.$DATE
 mv home/judge/data /home/judge/
-chown www-data -R /home/judge/data
+chown  $web_user  -R /home/judge/data
 mv /home/judge/src/web/upload /home/judge/src/web/upload.del.$DATE
 mv home/judge/src/web/upload /home/judge/src/web/
-chown www-data -R /home/judge/src/web/
+chown  $web_user -R /home/judge/src/web/
 bzip2 -d var/backups/db_${BAKDATE}.sql.bz2
 sed -i 's/COLLATE=utf8mb4_0900_ai_ci//g' var/backups/db_${BAKDATE}.sql
 sed -i 's/COLLATE utf8mb4_0900_ai_ci//g' var/backups/db_${BAKDATE}.sql