80 lines
2.5 KiB
Bash
80 lines
2.5 KiB
Bash
|
#!/bin/bash
|
||
|
|
if ! command -v jq >/dev/null 2>&1 ; then
|
||
|
|
wget -O /usr/bin/jq http://download.jsaix.cn/d/linux/jq-1.7-linux64 && chmod +x /usr/bin/jq
|
||
|
|
if ! command -v jq >/dev/null 2>&1 ; then
|
||
|
|
echo "Unsupported Linux distribution"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
UUID="$1"
|
||
|
|
if [ -z "$UUID" ]; then
|
||
|
|
echo "Error: Missing UUID parameter."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
INSTALL_DIR="/usr/local/aix-agent"
|
||
|
|
MAX_RETRIES=3
|
||
|
|
API_ENDPOINT="https://gateway.jsaix.cn/api/servers/agent_update/$UUID"
|
||
|
|
REMOTE_RESPONSE=$(curl -s "$API_ENDPOINT")
|
||
|
|
CODE=$(echo "$REMOTE_RESPONSE" | jq -r '.code')
|
||
|
|
if [ "$CODE" != "1" ]; then
|
||
|
|
echo "Error: Remote API returned a non-successful code. Exiting."
|
||
|
|
echo "$(echo "$REMOTE_RESPONSE" | jq -r '.message')"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
REMOTE_VERSION=$(curl -s "$API_ENDPOINT" | jq -r '.data.version')
|
||
|
|
DOWNLOAD_URL=$(echo "$REMOTE_RESPONSE" | jq -r '.data.url')
|
||
|
|
EXPECTED_MD5=$(echo "$REMOTE_RESPONSE" | jq -r '.data.md5')
|
||
|
|
if [ -z "$DOWNLOAD_URL" ] || [ -z "$EXPECTED_MD5" ] || [ -z "$REMOTE_VERSION" ]; then
|
||
|
|
echo "Error: Failed to fetch valid download information from the remote API."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
cd "$INSTALL_DIR"
|
||
|
|
download_and_check_md5() {
|
||
|
|
echo "Downloading archive..."
|
||
|
|
curl -O "$DOWNLOAD_URL"
|
||
|
|
if [ $? -ne 0 ]; then
|
||
|
|
echo "Error downloading archive."
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
ARCHIVE_FILE=$(basename "$DOWNLOAD_URL")
|
||
|
|
CALCULATED_MD5=$(md5sum "$ARCHIVE_FILE" | awk '{print $1}' | tr '[:lower:]' '[:upper:]')
|
||
|
|
if [ "$CALCULATED_MD5" != "$EXPECTED_MD5" ]; then
|
||
|
|
echo "Error: MD5 checksum does not match."
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
LOCAL_VERSION=$(/usr/local/aix-agent/aix-agent --version)
|
||
|
|
IFS='.' read -r -a array1 <<< "$REMOTE_VERSION"
|
||
|
|
IFS='.' read -r -a array2 <<< "$LOCAL_VERSION"
|
||
|
|
IS_UPDATE=0
|
||
|
|
for i in {0..2}; do
|
||
|
|
if ((array1[i] > array2[i])); then
|
||
|
|
IS_UPDATE=1
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
if [ "$IS_UPDATE" -eq 1 ]; then
|
||
|
|
retries=0
|
||
|
|
while [ $retries -lt $MAX_RETRIES ]; do
|
||
|
|
((retries++))
|
||
|
|
echo "Attempt $retries of $MAX_RETRIES"
|
||
|
|
download_and_check_md5 && break
|
||
|
|
if [ $retries -lt $MAX_RETRIES ]; then
|
||
|
|
echo "Retrying in 5 seconds..."
|
||
|
|
sleep 5
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
if [ $retries -eq $MAX_RETRIES ]; then
|
||
|
|
echo "Max retries reached. Exiting."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
tar -xvzf "$ARCHIVE_FILE" -C "$INSTALL_DIR"
|
||
|
|
rm "$ARCHIVE_FILE"
|
||
|
|
chmod +x /usr/local/aix-agent/aix-agent
|
||
|
|
sudo systemctl restart aix-agent
|
||
|
|
echo "Update successful!"
|
||
|
|
else
|
||
|
|
echo "No update available. Current version: $LOCAL_VERSION"
|
||
|
|
fi
|