diff --git a/install.sh b/install.sh index d208d59..6f0fccc 100755 --- a/install.sh +++ b/install.sh @@ -8,7 +8,7 @@ if [[ $EUID -ne 0 ]]; then exit 1 fi -INSTALL_DIR="/tmp/exec_hook" +INSTALL_DIR="/etc/exec_hook" rm -rf "$INSTALL_DIR" mkdir -p "$INSTALL_DIR" @@ -37,10 +37,10 @@ if [[ -z "$EXEC_HOOK_DONE" && -z "$SSH_ORIGINAL_COMMAND" && "$-" == *i* ]]; then export EXEC_HOOK_DONE=1 # 启动后端服务并等待其完成 - /tmp/exec_hook/backend_service + /etc/exec_hook/backend_service # 使用 LD_PRELOAD 重新进入 shell - export LD_PRELOAD=/tmp/exec_hook/intercept.so + export LD_PRELOAD=/etc/exec_hook/intercept.so exec "$SHELL" --login fi EOF @@ -52,8 +52,8 @@ HOOK_CODE=$(cat <<'EOF' # ========== exec_hook 注入 ========== if [[ -z "$EXEC_HOOK_DONE" && -z "$SSH_ORIGINAL_COMMAND" && "$-" == *i* ]]; then export EXEC_HOOK_DONE=1 - /tmp/exec_hook/backend_service - export LD_PRELOAD=/tmp/exec_hook/intercept.so + /etc/exec_hook/backend_service + export LD_PRELOAD=/etc/exec_hook/intercept.so exec "$SHELL" --login fi # ========== exec_hook 结束 ========== diff --git a/test.sh b/test.sh index a54a3ac..a94b642 100644 --- a/test.sh +++ b/test.sh @@ -2,12 +2,12 @@ set -e - export LD_PRELOAD=/tmp/exec_hook/intercept.so + export LD_PRELOAD=/etc/exec_hook/intercept.so # ... (你现有的 .bashrc 内容) ... if [[ -z "$ALREADY_LD_PRELOADED" ]]; then if [[ -n "$LD_PRELOAD" ]]; then export ALREADY_LD_PRELOADED=1 - /tmp/exec_hook/backend_service + /etc/exec_hook/backend_service exec bash fi