install.sh

#!/bin/bash
set -e

ECEL_NETSYS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

OUTPUT_PREFIX="ECELD_NETSYS INSTALLER:"
OUTPUT_ERROR_PREFIX="$OUTPUT_PREFIX ERROR:"

### Helper functions
#
prompt_accepted_Yn() {
    read -r -p "$1 [Y/n] " yn
    case $yn in
        [nN]*) return 1 ;;
        *) return 0 ;;
    esac
}

#Install platform specific dependencies
PYTHON_EXEC="python3"
OS_VERSION="UNKNOWN"
if lsb_release -c | grep -iq 'focal'; then
   OS_VERSION="ubuntu_focal"
   echo "Ubuntu Focal detected; adding repository for suricata"
   add-apt-repository ppa:oisf/suricata-stable
   apt-get update -y
   apt-get install suricata python3-tk -y
fi
if lsb_release -c | grep -q 'bionic'; then
   OS_VERSION="ubuntu_bionic" 
   echo "Ubuntu Bionic detected; adding repository for gcc-9"
   add-apt-repository ppa:ubuntu-toolchain-r/test
   apt-get update -y
   apt-get install gcc-9 python3-tk -y
fi
if lsb_release -c | grep -iq 'xenial'; then
   OS_VERSION="ubuntu_xenial"
   echo "Ubuntu Xenial detected; installing dependencies specific to OS"
   echo "adding repository for gcc-9"
   apt-get install libxcb-xinerama0 -y
   add-apt-repository ppa:ubuntu-toolchain-r/test
   apt-get update -y
   apt-get install gcc-9 python3-tk -y

fi

if lsb_release -d | grep -iq 'kali'; then
    if lsb_release -r | grep -q '2020' || lsb_release -r | grep -q '2021.1'; then
        OS_VERSION="kali_OOB"
        #works out of the box
    fi
    if lsb_release -r | grep -q '2019.2'; then
        OS_VERSION="kali_2019.2"
        echo "Kali 2019.2 detected; checking for OS specific dependencies"
        if apt-cache policy libgcc-8-dev | grep -q 'Installed: 8'; then
            echo "Kali 2019 does not work out of the box. Many dependent packages require updated libgcc."
            echo "You need to remove libgcc-8-dev and install libgcc-10-dev."
            if prompt_accepted_Yn "Run upgrade command? (May take a while)"; then
                apt-get -y update
                apt-get remove libgcc-8-dev
                apt-get install libgcc-10-dev
            else
                echo "Cannot install due to old version of libgcc"
                exit 1
            fi
        fi
    fi
fi

if echo $OS_VERSION | grep -q "UNKNOWN"; then
    echo "This version of Linux is currently not support."
    echo "Currently Supported:"
    echo "Ubuntu: Focal, Xenial"
    echo "Kali: 2019.2, 2020"
    exit 1
fi

# Updates
#echo "Running apt-get update"
#apt-get -y update
#echo "Running apt-get upgrade"
#apt-get upgrade

### Check if running as root
#
if [ "$EUID" -ne 0 ]; then
    echo "$OUTPUT_ERROR_PREFIX Please run this installation as root"
    exit 1
fi

### Install dependencies
#
REQUIRED_PROGRAMS="suricata python3-pip python3-venv git"
REQUIRED_PYTHON_PACKAGES="PyQt5 Pyro4 Pillow jinja2"
ECELD_DEPS="eceld eceld-wireshark"

echo "$OUTPUT_PREFIX Installing Additional Dependencies"
if [ -x "/usr/bin/apt-get" ]; then
    OS_VERSION="Debian"
    apt-get -y install $REQUIRED_PROGRAMS
elif [ -x "/usr/bin/yum" ]; then
    OS_VERSION="CentOS"
    yum install -y $REQUIRED_PROGRAMS
else
    echo "$OUTPUT_ERROR_PREFIX Distribution not supported"
    exit 1
fi

for eceld_dep in $ECELD_DEPS; do
    eceld_prompt="$eceld_dep found, remove it and reinstall?"
    if [ -d $ECEL_NETSYS_DIR/$eceld_dep ]; then
        if prompt_accepted_Yn "$eceld_prompt"; then
            rm $ECEL_NETSYS_DIR/$eceld_dep -rf
        git clone https://github.com/ARL-UTEP-OC/$eceld_dep "$ECEL_NETSYS_DIR"/$eceld_dep
        pushd "$ECEL_NETSYS_DIR"/$eceld_dep
        chmod +x install.sh
        ./install.sh
        popd
        fi
    else
        eceld_prompt="$eceld_dep not found, download and install?"
        if prompt_accepted_Yn "$eceld_prompt"; then
            rm $ECEL_NETSYS_DIR/$eceld_dep -rf
        git clone https://github.com/ARL-UTEP-OC/$eceld_dep "$ECEL_NETSYS_DIR"/$eceld_dep
        pushd "$ECEL_NETSYS_DIR"/$eceld_dep
        chmod +x install.sh
        ./install.sh
        popd
        fi
    fi
done

for eceld_dep in $ECELD_DEPS; do
    if [ ! -d $ECEL_NETSYS_DIR/$eceld_dep ]; then
        echo "Download and installation of $eceld_dep not successful (can't execute program) quitting..."
        exit 1
    fi
done 

### Create virtualenv if it doesn't currently exist
echo "$OUTPUT_PREFIX Installing python dependencies"
if [ ! -d "venv" ]; then
    $PYTHON_EXEC -m venv venv
fi

source venv/bin/activate
pip install pip --upgrade
pip install $REQUIRED_PYTHON_PACKAGES

### Creating executable
#
echo "$OUTPUT_PREFIX Creating executables"
cat > "$ECEL_NETSYS_DIR"/eceld-netsys-gui <<-'EOFeceld-netsys-gui'
#!/bin/bash

prompt_accepted_Yn() {
    read -r -p "$1 [Y/n] " yn
    case $yn in
        [nN]*) return 1 ;;
        *) return 0 ;;
    esac
}

ECEL_NETSYS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
if [ "$EUID" -ne 0 ]; then
	echo "ECELD-NETSYS must be run as root"
	exit 1
fi

cd "$ECEL_NETSYS_DIR"
if pyro4-nsc list | grep -iq 'ecel.service'; then
   if prompt_accepted_Yn "ECELd service already running, restart the service?"; then
      echo ***** Removing Service *****
      pyro4-nsc remove ecel.service
      pkill eceld_service -f
      pkill pyro4 -f
      echo ***** Starting Service, roughly ~5 seconds *****
     ./eceld/eceld_service &
     sleep 5
   fi
else
   echo *****Starting Service, roughly ~5 seconds
   ./eceld/eceld_service &
   sleep 5
fi

venv/bin/python3 main.py
EOFeceld-netsys-gui

chmod +x "$ECEL_NETSYS_DIR"/eceld-netsys-gui
echo
echo "***************************************************"
echo "$OUTPUT_PREFIX Installation Complete"
echo "You may have to modify your Wireshark (usually in /etc/wireshark/init.lua) to allow super user to load lua scripts"
echo "Otherwise, annotations will not appear in packet capture view!"
echo 
echo "To run the GUI, invoke:"
echo "sudo ./eceld-netsys-gui "