feat: restart ssh tunnel

Author: hiroTochigi

src/aws/dependencies/getProcessNumber.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+
+getProcessNumber() {
+    local targetString="$1"
+    local processInfo="$2"
+    local processNumber=$(
+            echo "$processInfo" \
+            | \
+            awk -v target="$targetString" '$0 ~ /\/usr\/lib\/autossh\/autossh/ && $0 ~ target {print $2}'
+        )
+
+    if [ -n "$processNumber" ]; then
+        echo "$processNumber"
+    else
+        echo ""
+    fi
+}
+
+getProcessInfo() {
+    local processInfo=$(ps aux | grep ssh)
+    echo "$processInfo"
+}
+

src/aws/dependencies/reverseShell.sh

@@ -36,3 +36,17 @@ function openSSHTunnel(){
     createSshConfig $instanceName $instanceIp "root" "22" "~/.ssh/id_rsa" $sshtunnelPortArray
     autossh  -f  -T -N -q -4 -M $monitorPort $instanceName
 }
+
+function closeSSHTunnel(){
+    local instanceName=$1
+    local processNumber=$(getProcessNumber "$instanceName" "$(getProcessInfo)")
+    if [ -n "$processNumber" ]; then
+        kill -9 $processNumber
+    fi
+}
+
+function restartSSHTunnel(){
+    local instanceName=$1
+    local instanceIp=$2
+    updateSshConfigInterface $instanceName HostName $instanceIp
+}

src/aws/load.sh

@@ -11,6 +11,7 @@ source $manageConfigPath/src/aws/dependencies/manageConfig.sh
 source $manageConfigPath/src/aws/dependencies/sshtunnelFunction.sh
 source $manageConfigPath/src/aws/dependencies/reverseShell.sh
 source $manageConfigPath/src/aws/dependencies/updateOrAppend.sh
+source $manageConfigPath/src/aws/dependencies/getProcessNumber.sh
 
 source $manageConfigPath/src/utils/dependencies/config.sh
 source $manageConfigPath/src/utils/dependencies/array.sh

src/aws/restart.sh

@@ -30,4 +30,7 @@ function restart(){
     echo "the new ip address is $publicIp"
     updateIPAddress $balloonName $publicIp
 
+    closeSSHTunnel
+    restartSSHTunnel
+    echo "open ssh tunnel"
 }

allTest.sh renamed to tests/allTest.sh

@@ -1,5 +1,3 @@
-source array.sh
-source replace.sh
 
 test_extractBlocks() {
     local input_string="$1"

tests/load.sh

@@ -0,0 +1,7 @@
+
+manageConfigPath=$(pwd)
+
+source $manageConfigPath/src/utils/load.sh
+source $manageConfigPath/src/aws/load.sh
+
+source $manageConfigPath/tests/testGetProcessNumber.sh

tests/testGetProcessNumber.sh

@@ -0,0 +1,27 @@
+testGetProcessNumber() {
+    local luftballon="luftballon"
+    
+    # Test case 1: Find process number
+    local processInfo1="root     1923245  0.0  0.0   2216    84 ?        Ss   03:18   0:00 /usr/lib/autossh/autossh    -T -N -q -4 -M 2200 luftballon"
+    local result1=$(getProcessNumber "$luftballon" "$processInfo1")
+    echo "Test case 1 - Process number: $result1"
+
+    # Test case 2: No /usr/lib/autossh/autossh
+    local processInfo2="root     1922602  0.0  0.1  19736 10404 ?        Ss   03:16   0:00 sshd: root@pts/0"
+    local result2=$(getProcessNumber "$luftballon" "$processInfo2")
+    echo "Test case 2 - Process number: $result2"
+
+    # Test case 3: No luftballon
+    local processInfo3="root     1923245  0.0  0.0   2216    84 ?        Ss   03:18   0:00 /usr/lib/autossh/autossh    -T -N -q -4 -M 2200 not_found"
+    local result3=$(getProcessNumber "$luftballon" "$processInfo3")
+    echo "Test case 3 - Process number: $result3"
+
+    # Test case 4: Multiple /usr/lib/autossh/autossh
+    local processInfo4="root     1923245  0.0  0.0   2216    84 ?        Ss   03:18   0:00 /usr/lib/autossh/autossh    -T -N -q -4 -M 2200 luftballon
+root     1923246  0.3  0.1  14476  8072 ?        S    03:18   0:00 /usr/lib/autossh/autossh    -T -N -q -4 -M 2200 luftballon"
+    local result4=$(getProcessNumber "$luftballon" "$processInfo4")
+    echo "Test case 4 - Process number: $result4"
+
+    local result5=$(getProcessNumber "$luftballon" "$(getProcessInfo)")
+    echo "Test case 5 - Process number: $result5"
+}