Create jvm_ops.sh

hsntgm · web-flow · commit 9de07b30662d · 2024-11-30T20:22:18.000+03:00
diff --git a/jvm_ops.sh b/jvm_ops.sh
@@ -0,0 +1,333 @@
+#!/bin/bash
+#
+# Copyright (C) 2021 Hasan CALISIR <hasan.calisir@psauxit.com>
+# Distributed under the GNU General Public License, version 2.0.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# What is doing this script exactly?
+# -Maven build, deploy, jvm control helper script for pspenit app.
+
+# app info
+prog_name="JVM Tools | PSAUXIT-PSPENIT"
+script_name="$0"
+
+# prod path | app name
+app_name="psauxit.jar"
+prod_path="$HOME/psauxitwebtools"
+preprod_path="$HOME/webnettools"
+
+# my jar is exist
+find_prod_app () {
+  prod_app=`find "${prod_path:?}/" -name \*"${app_name}" -print0 | tr -d '\0' 2>/dev/null`
+}
+
+# Set color
+setup_terminal () {
+  green="$(tput setaf 2)"; red="$(tput setaf 1)"; reset="$(tput sgr 0)"
+  cyan="$(tput setaf 6)"; magenta="$(tput setaf 5)"
+  TPUT_BOLD="$(tput bold)"; TPUT_BGRED="$(tput setab 1)"
+  TPUT_WHITE="$(tput setaf 7)";
+}
+setup_terminal
+
+# wait for process exit code
+my_wait () {
+  local my_pid=$!
+  local result
+  # Force kill bg process if script exits
+  trap "kill -9 $my_pid 2>/dev/null" EXIT
+  # Wait stylish while process is alive
+  spin='-\|/'
+  mi=0
+  while kill -0 $my_pid 2>/dev/null # (ps | grep $my_pid) is alternative
+  do
+    mi=$(( (mi+1) %4 ))
+    printf "\r${m_tab}${green}[ ${spin:$mi:1} ]${magenta} ${1}${reset}"
+    sleep .1
+  done
+  # Get bg process exit code
+  wait $my_pid
+  [[ $? -eq 0 ]] && result=ok
+  # Need for tput cuu1
+  echo ""
+  # Reset trap to normal exit
+  trap - EXIT
+  # Return status of function according to bg process status
+  [[ "${result}" ]] && return 0
+  return 1
+}
+
+fatal () {
+  printf >&2 "\n${m_tab}%s ABORTED %s %s \n\n" "${TPUT_BGRED}${TPUT_WHITE}${TPUT_BOLD}" "${reset}" "${*}"
+  exit 1
+}
+
+pretty_suc () {
+  echo "${m_tab}${TPUT_BOLD}${green}[ ✓ ] ${cyan}${1}${reset}"
+}
+
+pretty_fail () {
+  echo "${m_tab}${TPUT_BOLD}${red}[ x ] ${red}${1}${reset}"
+}
+
+replace_suc () {
+  tput cuu 1
+  echo "${m_tab}${TPUT_BOLD}${green}[ ✓ ] ${cyan}${1}${reset}"
+}
+
+replace_fail () {
+  tput cuu 1
+  echo "${m_tab}${TPUT_BOLD}${red}[ x ] ${red}${1}${reset}"
+}
+
+# deployment process
+deploy () {
+  local preprod_app=`find "${preprod_path:?}/" -name \*-runner.jar -print0 | tr -d '\0' 2>/dev/null`
+
+  # check this is a clean deploy
+  if [[ -d "${prod_path:?}" ]]; then
+     find_prod_app
+     [[ $prod_app ]] && clean=0 || clean=1
+  else
+    clean=1
+  fi
+
+  # Start deployment
+  if [[ -d "${preprod_path:?}" ]]; then
+    if [[ $preprod_app ]]; then
+      if [[ "${clean}" -eq 0 ]]; then
+        if [[ "$(md5sum "${preprod_app}" | awk '{print $1}')" != "$(md5sum "${prod_app}" | awk '{print $1}')" ]]; then
+          { rm -rf "${prod_path:?}"; mkdir -p "${prod_path}"/lib; cp "${preprod_app}" "${prod_path}"/"${prod_app##*/}"; cp -r "${preprod_path}"/target/lib/* "${prod_path}"/lib/; } >/dev/null 2>&1
+        else
+          pretty_suc "Application up to date. Deployment SKIPPED!"
+          return 1
+        fi
+      else
+        { mkdir -p "${prod_path}"/lib; cp "${preprod_app}" "${prod_path}"/"${app_name}"; cp -r "${preprod_path}"/target/lib/* "${prod_path}"/lib/; find_prod_app; } >/dev/null 2>&1
+      fi
+    else
+      pretty_fail "Cannot start deployment process, deployment jar cannot found!"
+      return 1
+    fi
+  else
+    pretty_fail "Cannot start deployment process, preprod path not found!"
+    return 1
+  fi
+  return 0
+}
+
+# clean frontend
+clean () {
+  arrb=("node" "node_modules" "package-lock.json")
+  arrf=("vendor" "build")
+
+  for d in "${arrb[@]}"; do
+    if [[ -e "${preprod_path}/src/main/frontend/${d}" ]]; then
+      print_b=1
+      rm -rf "${preprod_path:?}/src/main/frontend/${d}" &>/dev/null &
+    fi
+  done
+
+  if [[ "${print_b}" ]]; then
+    my_wait "Cleaning node,node_modules,lock.json.." && replace_suc "Cleaning node,node_modules,lock.json COMPLETED!" || { replace_fail "Cleaning node,node_modules,lock.json FAILED!"; fatal "QUIT"; }
+  fi
+
+  for d in "${arrf[@]}"; do
+    if [[ -d "${preprod_path}/src/main/frontend/public/${d}" ]]; then
+      print_f=1
+      rm -r "${preprod_path}/src/main/frontend/public/${d}" &>/dev/null &
+    fi
+  done
+
+  if [[ "${print_f}" ]]; then
+    my_wait "Cleaning previous frontend build.." && replace_suc "Cleaning previous frontend build COMPLETED!" || { replace_fail "Cleaning previous frontend build FAILED!"; fatal "QUIT"; }
+  fi
+
+  if [[ $1 == all ]] ; then
+    cd "${preprod_path}"
+    mvn clean &>/dev/null &
+    my_wait "Cleaning previous backend build.." && replace_suc "Cleaning previous backend build COMPLETED!" || { replace_fail "Cleaning previous backend build FAILED!"; fatal "QUIT"; }
+  fi
+}
+
+# Build jar via maven
+mvn_build () {
+  if [[ -d "${preprod_path:?}" ]]; then
+    cd "${preprod_path}"
+    mvn clean &>/dev/null &
+    my_wait "Cleaning previous backend build.." && replace_suc "Cleaning previous backend build COMPLETED!" || { replace_fail "Cleaning previous backend build FAILED!"; fatal "QUIT"; }
+    clean
+    mvn package &>/dev/null &
+    my_wait "Building application.." && replace_suc "Building application COMPLETED! READY TO DEPLOYMENT!" || { replace_fail "Building application FAILED!"; fatal "QUIT"; }
+  else
+    fatal "Preprod path not exist!"
+  fi
+}
+
+# Application global vars
+global_vars () {
+  export PATH=/home/black/.local/bin:/home/black/.cargo/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin:$PATH
+  export AVAILABLE_TOOLS=testssl,ping,dig,dnsrecon,wpscan,ddec,amass,whois,host,rustscan,cidr,mtr,wapiti,asn,xsstrike,nmap
+  export RATE_LIMIT=60000
+  export CA_DIR=/etc/ssl/certs/
+  export PORT=8080
+  export INTRO_TEXT='<div class="accept"><p class="accept-p"><span style="font-size:14px">By using this free service, I accept the <a href="https://tools.psauxit.com/page/terms-and-conditions" onclick="window.open(this.href, ">Terms And Conditions</a> and <a href="https://tools.psauxit.com/page/privacy-policy" onclick="window.open(this.href, ">Privacy Policy</a> of this website (<a href="https://www.psauxit.com/" onclick="window.open( this.href, "><span style="color:#e74c3c"><em>psauxit.com</em></span></a> ).&nbsp;I declare that <strong>I will only scan my own network and domain names I own</strong> for vulnerability testing purposes.</span></p></div>'
+}
+
+# Test application is working after (re)start in background for 5 sec
+# To:do add 'try loop' if not suc after 5 sec
+nohup_java () {
+  startwait=5
+  pid=$!
+  spin='-\|/'
+  mi=0
+  for i in $(seq $startwait); do
+    if ! kill -0 $pid 2>/dev/null; then
+      wait $!
+      exitcode=$?
+      replace_fail "Application cannot (re)started, EXITCODE:$exitcode" >&2
+      break
+    fi
+    mi=$(( (mi+1) %4 ))
+    printf "\r${m_tab}${green}[ ${spin:$mi:1} ]${magenta} ${1}${reset}"
+    sleep 1
+  done
+  if kill -0 $pid; then
+    echo ""
+    replace_suc "Application (re)started successfully!"
+  fi
+}
+
+# run in background
+my_run () {
+  nohup java "-Dquarkus.http.host=10.0.0.3" "-Dquarkus.http.port=${PORT}" "-Djava.util.logging.manager=org.jboss.logmanager.LogManager" -jar "${prod_app}" >/dev/null 2>&1 &
+}
+
+# run in foreground
+my_run_f () {
+  java "-Dquarkus.http.host=10.0.0.3" "-Dquarkus.http.port=${PORT}" "-Djava.util.logging.manager=org.jboss.logmanager.LogManager" -jar "${prod_app}"
+}
+
+# stop jvm
+stop_jvm () {
+  if kill -9 $(ps aux | grep -v grep | grep "${app_name}" | awk '{print $2}') >/dev/null 2>&1; then
+    pretty_suc "Application stopped!"
+  else
+    pretty_suc "Application already stopped!"
+  fi
+}
+
+# start jvm
+start_jvm () {
+  if ! ps aux | grep -v grep | grep "${app_name}" >/dev/null 2>&1; then
+    find_prod_app
+    [[ $prod_app ]] && { global_vars; my_run; nohup_java "Re-starting..."; } || pretty_fail "App not found. Cannot start application!"
+  else
+    pretty_suc "Application already running!"
+  fi
+}
+
+# start {python wapiti virtual env,application} on reboot
+boot () {
+  start_jvm
+}
+
+# restart jvm
+restart_jvm () {
+  stop_jvm
+  start_jvm
+}
+
+# carry process to foreground
+run_foreground () {
+  stop_jvm
+  find_prod_app
+  [[ $prod_app ]] && { global_vars; my_run_f; } || pretty_fail "App not found. Cannot start application!"
+}
+
+help () {
+  echo -e "\n${m_tab}${cyan}# Script Help"
+  echo -e "${m_tab}# ---------------------------------------------------------------------------------"
+  echo -e "${m_tab}#${m_tab}--start              start jvm"
+  echo -e "${m_tab}#${m_tab}--stop               stop jvm"
+  echo -e "${m_tab}#${m_tab}--restart            restart jvm"
+  echo -e "${m_tab}#${m_tab}--foreground         start jvm on foreground"
+  echo -e "${m_tab}#${m_tab}--deploy             start deployment"
+  echo -e "${m_tab}#${m_tab}--build              build app"
+  echo -e "${m_tab}#${m_tab}--build-deploy       build app and start deploy"
+  echo -e "${m_tab}#${m_tab}--clean              clean previous frontend build"
+  echo -e "${m_tab}#${m_tab}--clean-all          clean previous frontend & backend build"
+  echo -e "${m_tab}#${m_tab}--boot               start jvm on boot"
+  echo -e "${m_tab}#${m_tab}--help               display help"
+  echo -e "${m_tab}# ---------------------------------------------------------------------------------${reset}\n"
+}
+
+deployment () {
+  if deploy; then
+    # Restart application
+    [[ $clean -eq 0 ]] && pretty_suc "OK! Deployment COMPLETED!" || pretty_suc "DONE! Clean deployment COMPLETED!"
+    pretty_suc "Will try to (re)start application..."
+    restart_jvm
+  elif ! ps aux | grep -v grep | grep "${prod_app##*/}" >/dev/null 2>&1; then
+    pretty_fail "Application is not working currently.."
+    pretty_suc "Trying to start application..."
+    start_jvm
+  else
+    pretty_suc "Application is running."
+  fi
+}
+
+# Build app and deploy it immediately
+build_deploy () {
+  mvn_build
+  deployment
+}
+
+inv_opt () {
+  printf "%s\\n" "${red}${prog_name}: Invalid option '$1'${reset}"
+  printf "%s\\n" "${cyan}Try '${script_name} --help' for more information.${reset}"
+  exit 1
+}
+
+# Script management
+main () {
+  if [[ "$#" -eq 0 || "$#" -gt 1 ]]; then
+    printf "%s\\n" "${red}${prog_name}: Argument required or too many argument${reset}"
+    printf "%s\\n" "${cyan}Try '${script_name} --help' for more information.${reset}"
+    exit 1
+  fi
+
+  # set script arguments
+  while [[ "$#" -gt 0 ]]; do
+    case "$1" in
+      -s  | --start        ) start_jvm       ;;
+      -t  | --stop         ) stop_jvm        ;;
+      -r  | --restart      ) restart_jvm     ;;
+      -f  | --foreground   ) run_foreground  ;;
+      -d  | --deploy       ) deployment      ;;
+      -c  | --build        ) mvn_build       ;;
+      -cd | --build-deploy ) build_deploy    ;;
+      -e  | --clean        ) clean           ;;
+      -ca | --clean-all    ) clean all       ;;
+      -b  | --boot         ) boot            ;;
+      -h  | --help         ) help            ;;
+      --  | -* | *         ) inv_opt         ;;
+    esac
+    break
+  done
+}
+
+# Call main
+main "${@}"
