Skip to content

Commit f81b79e

Browse files
authored
Merge pull request #55 from YOMXXX/fix/install-root-su-recursion
fix(install): prevent infinite su recursion when running as root
2 parents 3d965df + e2223bd commit f81b79e

1 file changed

Lines changed: 22 additions & 5 deletions

File tree

scripts/install_hermes_memory_tencentdb.sh

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,17 @@
3333

3434
set -e
3535

36-
# 动态获取当前执行用户及 HOME 目录
37-
USERNAME=$(whoami)
36+
# 动态获取目标安装用户及其 HOME 目录。
37+
# 优先级:
38+
# 1. 显式 ``INSTALL_AS_USER`` 环境变量(管理员脚本场景:root 跑安装但
39+
# 想为另一个用户配置)
40+
# 2. ``SUDO_USER``(被 ``sudo`` 调用时,切回原用户而不是 root)
41+
# 3. ``whoami`` —— 当前 EUID 对应的用户
42+
#
43+
# 注意:当 root 直接 ssh 登录跑(非 sudo)时,前两个都不会被设置,
44+
# ``whoami`` 返回 ``root``。下面的 ``id -u`` == 0 分支会识别这种"目标
45+
# 就是 root"的情况、跳过 ``su - root`` 递归。
46+
USERNAME="${INSTALL_AS_USER:-${SUDO_USER:-$(whoami)}}"
3847
USER_HOME=$(eval echo ~$USERNAME)
3948

4049
# npm 包名
@@ -62,10 +71,14 @@ LEGACY_INSTALL_DIR="$USER_HOME/tdai-memory-openclaw-plugin"
6271
LEGACY_DATA_DIR="$USER_HOME/memory-tdai"
6372

6473
# ==================== root → 自动切换到目标用户 ====================
65-
# 与 install_hermes_ubuntu.sh 保持一致:如果以 root 执行,自动 su 切到
66-
# 目标用户运行实际安装逻辑。也可以直接以目标用户身份执行,跳过此段。
74+
# 与 install_hermes_ubuntu.sh 保持一致:如果以 root 执行且目标用户不是
75+
# root,自动 su 切到目标用户运行实际安装逻辑。
76+
#
77+
# 如果当前是 root 且目标用户也是 root(``USERNAME=root``,例如直接 ssh
78+
# 登录 root 跑安装),跳过 ``su - root`` —— 否则会无限递归(``su - root``
79+
# 进入的仍是 root,又走到这个分支,再次 su,永远停不下来)。见 issue #20。
6780

68-
if [ "$(id -u)" -eq 0 ]; then
81+
if [ "$(id -u)" -eq 0 ] && [ "$USERNAME" != "root" ]; then
6982
echo "[memory-tencentdb] Running as root, switching to $USERNAME for installation..."
7083

7184
# 验证前置条件
@@ -88,6 +101,10 @@ if [ "$(id -u)" -eq 0 ]; then
88101
rm -f "$TEMP_SCRIPT"
89102
echo "[memory-tencentdb] Installation completed successfully"
90103
exit 0
104+
elif [ "$(id -u)" -eq 0 ]; then
105+
# 当前是 root 且目标用户也是 root:直接以 root 跑后续安装逻辑,
106+
# 不再走 ``su -`` 切换(避免 #20 的递归)。
107+
echo "[memory-tencentdb] Running as root; target user is also root — installing in place."
91108
fi
92109

93110
# ==================== 用户阶段(核心安装逻辑) ====================

0 commit comments

Comments
 (0)