#!/usr/bin/env bash
#
# Скрипт установки linux-агента для Vulns.io VM
#
# "Script version 1.15"
#

# Лог файл
LOG_FILE=/tmp/deploy-linuxsc.txt

# Временный файл с инсталлятором
FILE_TMP=agent-installer_$$.sh

# IP адрес сервера с установленным решением Vulns.io VM или сенсора
SERVER_IP=xxx.xxx.xxx.xxx

# Ключ для доступа к API установленного решения Vulns.io VM
# (см. раздел документации "Создание API-токена")
API_KEY="abcdef1234657890"

# Указать Sensor ID, если агенты будут работать через сенсор
SENSOR_ID=""

# accountid - id организации в базе OnPremise
# По умолчанию accountid="0" - организация по умолчанию
accountid="0"

# userid - id пользователя в базе OnPremise
# По умолчанию userid="0" - пользователь по умолчанию
userid="0"

# Имена сервисов агента для проверки их работоспособности
agent_service_name="vulns.io-agent-linux"
updater_service_name="vulns.io-agent-updater-linux"

# Параметр 'DELETE_ASSET_IF_INSTALL_FAIL' определяет поведение скрипта в случае если службы агента не запустились.
#  Значение 1 - указывает, что в случае не удачного запуска службы (по любой причине) удаляем установленный агент и очищаем из базы данных OnPremise asset и agent
#  Значение 0 - не предпринимаем никаких действий по удалению агента
DELETE_ASSET_IF_INSTALL_FAIL=1

# Разрешено ли выводить информацию о действиях скрипта на экран или в лог-файл
LOG_TO_SCREEN=1
LOG_TO_FILE=1

# коды выхода из скрипта
# =0  - Агент установлен и запущен в системе корректно
# =4  - Не найдена команда curl для работы с API OnPremise
# =8  - Пользователь не входит в группу 'root', необходимого для создания служб агентов в системе и их запуска
# =16 - Вызов справки по аргументам запуска скрипта
# =32 - Ошибка доступа к API OnPremise
# =64 - Ошибка скачивания инсталятора агента с сервера OnPremise


out_data_log()
{
# Вывод информации о работе скрипта на экран или в лог-файл в зависимости от переменных LOG_TO_SCREEN и LOG_TO_FILE
if  [ $LOG_TO_SCREEN -eq 1 ]; then
  echo $1
fi
if  [ $LOG_TO_FILE -eq 1 ]; then
  echo $1 >> $LOG_FILE
fi
}

init()
{
# Проверка существования curl. Необходимо для доступа к API OnPremise через протокол https.

if ! command -v curl >/dev/null; then
  out_data_log "Attention: You may need to install curl for the deploy script!"
  exit 4
fi

# Проверка суперправ у пользователя запустившего скрипт
if groups | grep -q "\broot\b"; then
  out_data_log "The corrent user is in group 'root'. Continuing the work of the script"
else
  out_data_log "Attention - the user is not in the 'root' group!"
  exit 8
fi

if [[ -n "$1" ]]; then
    echo " "
    echo "Требуется указать корректный аргумент вызова скрипта"
    echo "Команды запуска скрипта:"
    echo $(basename $0)"  [ -h | --help ]"
    echo $'\t'"где '--help или -h '"$'\t'" - вызов этой справки по командам"
    exit 16
fi

# Получаем текущий IP-адрес машины
IP_ADR=$(hostname --all-ip-addresses | awk '{print $1}')
out_data_log "Current IP="$IP_ADR
}

check_response_api()
{
# Проверка ответа от API и результата выполнения curl

#  если после вызова curl получаем код ошибки больше 0, то сообщаем об ошибки доступности API
if [ $1 -ne 0 ]; then
 if [ $1 -eq 22 ]; then
  out_data_log "Request failure. API OnPremise not available, response code >= 400"
 else
  out_data_log "CURL: Error code = "$1
 fi
 RETURN_VAL="error"
else
 RETURN_VAL="ok"
fi
}

get_installer()
{
# Отправка запроса на скачивание установщика и получение его url
local osType="linux"

GET_URL=$(curl -X GET -k "https://$SERVER_IP/integration/v1/agents/download/installer/url?osType=$osType&accountId=$accountid&userId=$userid&sensorId=$SENSOR_ID" -H "Content-Type: application/json" -H "x-api-key: $API_KEY" --compressed -s)

check_response_api $?

if [ "$RETURN_VAL" == "error" ]; then
  out_data_log "Error send request to API OnPremise, server IP "$SERVER_IP
# Если после первого обращения к API сервер был не доступен, то прекращаяем выполнения скрипта, считая, что проблемы с сетевым доступом или сервером OnPremise
  exit 32
fi

url_id=$(echo $GET_URL | sed -e "s/^.*url\"\:\"//g" | sed -e "s/\".*$//g")
agentid=$(echo $GET_URL | sed -e "s/^.*agentId\"\:\"//g" | sed -e "s/\".*$//g")
assetid=$(echo $GET_URL | sed -e "s/^.*assetId\"\:\"//g" | sed -e "s/\".*$//g")
}

delete_asset()
{
# Удаление asset'а

# assetId - получить из ответа полученного при создании агента

ASSET=$(curl -X DELETE -k "https://$SERVER_IP/integration/v1/assets/$assetid" -H "Content-Type: application/json" -H "x-api-key: $API_KEY" --compressed -s)

check_response_api $?

if [ "$RETURN_VAL" == "error" ]; then
  out_data_log "Error deleting asset '$assetid' data in database OnPremise, server IP $SERVER_IP"
else
  out_data_log "Asset deleted correctly."
fi

}

deploy_agent()
{
# Скачивание и запуск установки Linux-установщика

out_data_log "---- Download and deploy agent installer ----"
# Скачиваем файл, игнорируя самоподписанный SSL сертификат сервера (-k)

# Вариант #1 - скачиваем файл и сразу его запускаем
#curl -k -s $url_id | bash

# Вариант #2 - скачиваем файл, выдаем разрешение на запуск и запускаем исталлятор
# В этом случае можно обработать ответ от сервера и вывести необходимую информацию на экран/лог
curl -k -s $url_id -o ./$FILE_TMP
check_response_api $?
if [ "$RETURN_VAL" == "error" ]; then
  out_data_log "Error downloading agent installer script from server IP $SERVER_IP"
# Не смогли скачать агента, прекращаем работу скрипта
  exit 64
else
  out_data_log "Agent installer correctly downloaded."
fi
chmod +x ./$FILE_TMP
./$FILE_TMP

echo "------------ Deploy agent finish ------------"
}

check_agent()
{
# Проверка, что хотя бы одна из служб агента запущена
if systemctl -q is-active $agent_service_name == 'active'; then
  out_data_log "Service $agent_service_name still running."
  AGENT_INSTALLED=1
else
  if systemctl -q is-active $updater_service_name == 'active'; then
   out_data_log "Service $updater_service_name still running."
   AGENT_INSTALLED=1
   else

     # Не нашли запущенную службу агента (vulns.io-agent-linux) или апдейтера (vulns.io-agent-updater-linux)
     AGENT_INSTALLED=0

     out_data_log "Linux-agent installation failed with an error, agent not installed"

     # В зависимости от параметра 'DELETE_ASSET_IF_INSTALL_FAIL' удаляем или нет установленный агент и (не)очищаем из базы данных OnPremise asset и agent
     if [ $DELETE_ASSET_IF_INSTALL_FAIL -eq 0 ]; then
      out_data_log "Set internal variable 'DELETE_ASSET_IF_INSTALL_FAIL=0'"
     else
     # Удаляем агента с помощью uninstaller'а и удаляем asset и agent в бд OnPremise
       if [ -f /opt/vulns.io/agent-linux/agent_uninstall.sh ]; then
         out_data_log "Run - Uninstall agent script"
         /opt/vulns.io/agent-linux/agent_uninstall.sh
       fi

# - fix on удаляем забытые символьные ссылки (для поддержки старых версий деинсталляторов агентов)
  local AGENT_SERVICE="/usr/lib/systemd/system/vulns.io-agent-linux.service"
  local AGENT_UPD_SERVICE="/usr/lib/systemd/system/vulns.io-agent-updater-linux.service"
       if [ -L $AGENT_SERVICE ]; then
        rm -f $AGENT_SERVICE
        out_data_log "FIX: Удалили файл $AGENT_SERVICE"
       fi
       if [ -L $AGENT_UPD_SERVICE ]; then
        rm -f $AGENT_UPD_SERVICE
        out_data_log "FIX: Удалили файл $AGENT_UPD_SERVICE"
       fi
# - fix off

       out_data_log "Deleting data on OnPremise"

       delete_asset

       # Удаляем скачанный файл инсталятора linux-агента
       rm -f ./$FILE_TMP
     fi
  fi
fi
if [ $AGENT_INSTALLED -eq 1 ]; then
  out_data_log "Linux agent installation completed correctly"
fi
}


# --------- MAIN ---------

out_data_log "-- Start deploy linux-agent at $(date +"%Y-%m-%d %T") ---"

init $1

get_installer

out_data_log "agentid='$agentid'"
out_data_log "assetid='$assetid'"

deploy_agent

WAIT_TIME=5
out_data_log "Wait $WAIT_TIME second before check linux-agent"
sleep $WAIT_TIME

check_agent

out_data_log "-- End deploy linux-agent. Total script running time $SECONDS seconds ---"