Add necessary functions to allow upgrade using installation assistant

unattended_installer/install_functions/checks.sh

@@ -154,8 +154,8 @@ function checks_arguments() {
 
     # -------------- Global -----------------------------------------
 
-    if [ -z "${AIO}" ] && [ -z "${indexer}" ] && [ -z "${dashboard}" ] && [ -z "${wazuh}" ] && [ -z "${start_indexer_cluster}" ] && [ -z "${configurations}" ] && [ -z "${uninstall}" ] && [ -z "${download}" ]; then
-        common_logger -e "At least one of these arguments is necessary -a|--all-in-one, -g|--generate-config-files, -wi|--wazuh-indexer, -wd|--wazuh-dashboard, -s|--start-cluster, -ws|--wazuh-server, -u|--uninstall, -dw|--download-wazuh."
+    if [ -z "${AIO}" ] && [ -z "${indexer}" ] && [ -z "${dashboard}" ] && [ -z "${wazuh}" ] && [ -z "${start_indexer_cluster}" ] && [ -z "${configurations}" ] && [ -z "${uninstall}" ] && [ -z "${download}" ] && [ -z "${upgrade}" ]; then
+        common_logger -e "At least one of these arguments is necessary -a|--all-in-one, -g|--generate-config-files, , -up|--upgrade, -wi|--wazuh-indexer, -wd|--wazuh-dashboard, -s|--start-cluster, -ws|--wazuh-server, -u|--uninstall, -dw|--download-wazuh."
         exit 1
     fi
 

unattended_installer/install_functions/dashboard.sh

@@ -191,7 +191,9 @@ function dashboard_install() {
     common_checkInstalled
     if [  "$install_result" != 0  ] || [ -z "${dashboard_installed}" ]; then
         common_logger -e "Wazuh dashboard installation failed."
-        installCommon_rollBack
+        if [ -z ${upgrade} ]; then
+            installCommon_rollBack
+        fi
         exit 1
     else
         common_logger "Wazuh dashboard installation finished."

unattended_installer/install_functions/indexer.sh

@@ -148,7 +148,9 @@ function indexer_install() {
     common_checkInstalled
     if [  "$install_result" != 0  ] || [ -z "${indexer_installed}" ]; then
         common_logger -e "Wazuh indexer installation failed."
-        installCommon_rollBack
+        if [ -z ${upgrade} ]; then
+            installCommon_rollBack
+        fi
         exit 1
     else
         common_logger "Wazuh indexer installation finished."

unattended_installer/install_functions/installMain.sh

@@ -49,6 +49,9 @@ function getHelp() {
     echo -e "        -u,  --uninstall"
     echo -e "                Uninstalls all Wazuh components. This will erase all the existing configuration and data."
     echo -e ""
+    echo -e "        -up,  --upgrade"
+    echo -e "                Upgrades installed Wazuh components."
+    echo -e ""
     echo -e "        -v,  --verbose"
     echo -e "                Shows the complete installation output."
     echo -e ""
@@ -129,6 +132,10 @@ function main() {
                 uninstall=1
                 shift 1
                 ;;
+            "-up"|"--upgrade")
+                upgrade=1
+                shift 1
+                ;;
             "-v"|"--verbose")
                 debugEnabled=1
                 debug="2>&1 | tee -a ${logfile}"
@@ -209,11 +216,20 @@ function main() {
     common_checkSystem
     common_checkInstalled
     checks_arguments
+# -------------- Uninstall case  ------------------------------------
+
     if [ -n "${uninstall}" ]; then
         installCommon_rollBack
         exit 0
     fi
 
+# -------------- Upgrade case  ------------------------------------
+
+    if [ -n "${upgrade}" ]; then
+        upgrade_upgradeInstalled
+        exit 0
+    fi
+
 # -------------- Preliminary checks  --------------------------------
 
     if [ -z "${configurations}" ] && [ -z "${AIO}" ] && [ -z "${download}" ]; then

unattended_installer/install_functions/manager.sh

@@ -54,7 +54,9 @@ function manager_install() {
     common_checkInstalled
     if [  "$install_result" != 0  ] || [ -z "${wazuh_installed}" ]; then
         common_logger -e "Wazuh installation failed."
-        installCommon_rollBack
+        if [ -z ${upgrade} ]; then
+            installCommon_rollBack 
+        fi
         exit 1
     else
         common_logger "Wazuh manager installation finished."

unattended_installer/install_functions/upgrade.sh

@@ -0,0 +1,105 @@
+# Wazuh installer - common.sh functions.
+# Copyright (C) 2015, Wazuh Inc.
+#
+# This program is a free software; you can redistribute it
+# and/or modify it under the terms of the GNU General Public
+# License (version 2) as published by the FSF - Free Software
+# Foundation.
+
+
+function upgrade_getUpgradable {
+
+  if [ -n "${wazuh_installed}" ]; then
+    if [ "${sys_type}" == "yum" ]; then
+        manager_upgradable=$(yum check-update wazuh-manager | grep wazuh-manager | sed 's/  */ /g'| cut -d' ' -f2 | sed "s/-.*//g")
+    elif [ "${sys_type}" == "apt-get" ]; then
+        manager_upgradable=$(apt list wazuh-manager -a 2>/dev/null | grep "upgradable from" | cut -d' ' -f2 | sed -e "s/-.*//")
+    fi
+  fi
+
+  if [ -n "${filebeat_installed}" ]; then
+    if [ "${sys_type}" == "yum" ]; then
+        filebeat_upgradable=$(yum check-update filebeat | grep filebeat | sed 's/  */ /g'| cut -d' ' -f2 | sed "s/-.*//g")
+    elif [ "${sys_type}" == "apt-get" ]; then
+        filebeat_upgradable=$(apt list filebeat -a 2>/dev/null | grep "upgradable from" | cut -d' ' -f2 | sed -e "s/-.*//")
+    fi
+  fi
+
+  if [ -n "${indexer_installed}" ]; then
+    if [ "${sys_type}" == "yum" ]; then
+        indexer_upgradable=$(yum check-update wazuh-indexer | grep wazuh-indexer | sed 's/  */ /g'| cut -d' ' -f2 | sed "s/-.*//g")
+    elif [ "${sys_type}" == "apt-get" ]; then
+        indexer_upgradable=$(apt list wazuh-indexer -a 2>/dev/null | grep "upgradable from" | cut -d' ' -f2 | sed -e "s/-.*//")
+    fi
+  fi
+
+  if [ -n "${dashboard_installed}" ]; then
+    if [ "${sys_type}" == "yum" ]; then
+        dashboard_upgradable=$(yum check-update wazuh-dashboard | grep wazuh-dashboard | sed 's/  */ /g'| cut -d' ' -f2 | sed "s/-.*//g")
+    elif [ "${sys_type}" == "apt-get" ]; then
+        dashboard_upgradable=$(apt list wazuh-dashboard -a 2>/dev/null | grep "upgradable from" | cut -d' ' -f2 | sed -e "s/-.*//")
+    fi
+  fi
+
+}
+
+
+function upgrade_upgradeInstalled(){
+
+  common_logger "--- Upgrading existing Wazuh installation ---"
+
+  upgrade_getUpgradable
+
+  if [ -n "${wazuh_installed}" ]; then
+    if [ -n "${manager_upgradable}" ]; then
+      if [ "${manager_upgradable}" == "${wazuh_version}" ]; then
+        common_logger "Upgrading Wazuh Manager to ${manager_upgradable}"
+        eval "manager_install ${debug}"
+      else
+        common_logger -w "Wazuh manager can be upgraded but version does not match with the installation assistant version"
+      fi
+    else
+      common_logger -w "Wazuh manager is already installed and is up to date."
+    fi
+  fi
+
+  if [ -n "${filebeat_installed}" ]; then
+    if [ -n "${filebeat_upgradable}" ]; then
+      if [ "${filebeat_upgradable}" == "${filebeat_version}" ]; then
+        common_logger "Upgrading Filebeat to ${filebeat_upgradable}"
+        eval "filebeat_install ${debug}"
+      else
+        common_logger -w "Filebeat can be upgraded but version does not match with the installation assistant version"
+      fi
+    else
+      common_logger -w "Filebeat is already installed and is up to date."
+    fi
+  fi
+
+  if [ -n "${indexer_installed}" ]; then
+    if [ -n "${indexer_upgradable}" ]; then
+      if [ "${indexer_upgradable}" == "${wazuh_version}" ]; then
+        common_logger "Upgrading Wazuh Indexer to ${indexer_upgradable}"
+        eval "indexer_install ${debug}"
+      else
+        common_logger -w "Wazuh Indexer can be upgraded but version does not match with the installation assistant version"
+      fi
+    else
+      common_logger -w "Wazuh Indexer is already installed and is up to date."
+    fi
+  fi
+
+  if [ -n "${dashboard_installed}" ]; then
+    if [ -n "${dashboard_upgradable}" ]; then
+      if [ "${dashboard_upgradable}" == "${wazuh_version}" ]; then
+        common_logger "Upgrading Wazuh Dashboard to ${dashboard_upgradable}"
+        eval "dashboard_install ${debug}"
+      else
+        common_logger -w "Wazuh Dashboard can be upgraded but version does not match with the installation assistant version"
+      fi
+    else
+      common_logger -w "Wazuh Dashboard is already installed and is up to date."
+    fi
+  fi
+
+}