update_armitage.sh

#!/bin/bash
KVER=`uname -a`

function print_good ()
{
    echo -e "\x1B[01;32m[*]\x1B[0m $1"
}
########################################

function print_error ()
{
    echo -e "\x1B[01;31m[*]\x1B[0m $1"
}
########################################

function print_status ()
{
    echo -e "\x1B[01;34m[*]\x1B[0m $1"
}
########################################

function install_armitage_osx
{
    if [ -e /usr/bin/curl ]; then
        print_status "Downloading latest version of Armitage"
    curl -# -o /tmp/armitage.tgz http://www.fastandeasyhacking.com/download/armitage-latest.tgz && print_good "Finished"
    if [ $? -eq 1 ] ; then
            print_error "Failed to download the latest version of Armitage make sure you"
        print_error "are connected to the internet and can reach http://www.fastandeasyhacking.com"
    else
        print_status "Decompressing package to /usr/local/share/armitage"
        tar -xvzf /tmp/armitage.tgz -C /usr/local/share
    fi

    # Check if links exists and if they do not create them
    if [ ! -e /usr/local/bin/armitage ]; then
        print_status "Linking Armitage in /usr/local/bin/armitage"
        sh -c "echo java -jar /usr/local/share/armitage/armitage.jar \$\* > /usr/local/share/armitage/armitage"
        ln -s /usr/local/share/armitage/armitage /usr/local/bin/armitage
    else
        print_good "Armitage is already linked to /usr/local/bin/armitage"
        echo java -jar /usr/local/share/armitage/armitage.jar \$\* > /usr/local/bin/armitage
    fi

    if [ ! -e /usr/local/bin/teamserver ]; then
        print_status "Copying Teamserver in /usr/local/bin/teamserver"
        ln -s /usr/local/armitage/teamserver /usr/local/bin/teamserver
        perl -pi -e 's/armitage.jar/\/usr\/local\/share\/armitage\/armitage.jar/g' /usr/local/share/armitage/teamserver
    else
        print_good "Teamserver is already linked to /usr/local/bin/teamserver"
        perl -pi -e 's/armitage.jar/\/usr\/local\/share\/armitage\/armitage.jar/g' /usr/local/share/armitage/teamserver
    fi
    print_good "Finished"
    fi
}
########################################

function install_armitage_linux
{
    if [ -e /usr/bin/curl ]; then
    print_status "Downloading latest version of Armitage"
    curl -# -o /tmp/armitage.tgz http://www.fastandeasyhacking.com/download/armitage-latest.tgz && print_good "Finished"
    if [ $? -eq 1 ] ; then
        print_error "Failed to download the latest version of Armitage make sure you"
        print_error "are connected to the internet and can reach http://www.fastandeasyhacking.com"
        else
        print_status "Decompressing package to /usr/local/share/armitage"
        sudo tar -xvzf /tmp/armitage.tgz -C /usr/local/share >>outfile 2>&1
    fi

    # Check if links exists and if they do not create them
    if [ ! -e /usr/local/bin/armitage ]; then
        print_status "Creating link for Armitage in /usr/local/bin/armitage"
        sudo sh -c "echo java -jar /usr/local/share/armitage/armitage.jar \$\* > /usr/local/share/armitage/armitage"
        sudo ln -s /usr/local/share/armitage/armitage /usr/local/bin/armitage
    else
        print_good "Armitage is already linked to /usr/local/bin/armitage"
        sudo sh -c "echo java -jar /usr/local/share/armitage/armitage.jar \$\* > /usr/local/share/armitage/armitage"
    fi

    if [ ! -e /usr/local/bin/teamserver ]; then
        print_status "Creating link for Teamserver in /usr/local/bin/teamserver"
        sudo ln -s /usr/local/share/armitage/teamserver /usr/local/bin/teamserver
        sudo perl -pi -e 's/armitage.jar/\/usr\/local\/share\/armitage\/armitage.jar/g' /usr/local/share/armitage/teamserver
    else
        print_good "Teamserver is already linked to /usr/local/bin/teamserver"
        sudo perl -pi -e 's/armitage.jar/\/usr\/local\/share\/armitage\/armitage.jar/g' /usr/local/share/armitage/teamserver
    fi
    print_good "Finished"
    fi
}

if [[ "$KVER" =~ Darwin ]]; then
    install_armitage_osx

elif [[ "$KVER" =~ buntu ]]; then
    install_armitage_linux

else
    print_error "The script does not support this platform at this moment."
    exit 1
fi