Skip to content

Commit abc1e74

Browse files
committed
fix: use admin .ssh folder, delay deleting ssh key
1 parent 4619bd8 commit abc1e74

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

command-ctrl.sh

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -30,27 +30,27 @@ remote_ctrl() {
3030
SSH_REMOTE_PORT=22
3131
fi
3232

33-
PRIVATE_KEY_FILE=/tmp/temp_ssh_key
33+
export PRIVATE_KEY_FILE=/tmp/temp_ssh_key
3434
echo -e "$SSH_PRIVATE_KEY" > "$PRIVATE_KEY_FILE"
3535
chmod 600 "$PRIVATE_KEY_FILE"
3636

3737
if [ -z "$SSH_HOST_KEY" ]; then
3838
echo "\$SSH_HOST_KEY is empty"
3939
else
40-
KNOWN_HOSTS_FILE=~/.ssh/known_hosts
40+
KNOWN_HOSTS_FILE=~admin/.ssh/known_hosts
4141
KNOWN_HOST="$SSH_HOST $SSH_HOST_KEY"
4242
if [ ! -f "$KNOWN_HOSTS_FILE" ] || ! grep -Fxq "$KNOWN_HOST" "$KNOWN_HOSTS_FILE"; then
43-
mkdir -p ~/.ssh
43+
mkdir -p ~admin/.ssh
4444
echo "$KNOWN_HOST" >> "$KNOWN_HOSTS_FILE"
4545
fi
4646
fi
4747

4848
if [ -z "$SSH_AUTH_KEY" ]; then
4949
echo "\$SSH_AUTH_KEY is empty"
5050
else
51-
AUTHORIZED_KEYS_FILE=~/.ssh/authorized_keys
51+
AUTHORIZED_KEYS_FILE=~admin/.ssh/authorized_keys
5252
if [ ! -f "$AUTHORIZED_KEYS_FILE" ] || ! grep -Fxq "$SSH_AUTH_KEY" "$AUTHORIZED_KEYS_FILE"; then
53-
mkdir -p ~/.ssh
53+
mkdir -p ~admin/.ssh
5454
echo "$SSH_AUTH_KEY" >> "$AUTHORIZED_KEYS_FILE"
5555
fi
5656
fi
@@ -60,8 +60,7 @@ remote_ctrl() {
6060
$(which sshd) -p $SSH_LOCAL_PORT -o "PubkeyAuthentication yes"
6161
fi
6262
ssh -o "ExitOnForwardFailure yes" -N -R $SSH_FORWARD_PORT:localhost:$SSH_LOCAL_PORT $SSH_USERNAME@$SSH_HOST -p $SSH_REMOTE_PORT -i $PRIVATE_KEY_FILE &
63-
sleep 2
64-
rm $PRIVATE_KEY_FILE
63+
sh -c 'sleep 10; rm $PRIVATE_KEY_FILE'
6564
}
6665

6766
update() {

0 commit comments

Comments
 (0)