pentest-distro-builder/runme.sh
Kaj Forney 5b5761fe13 Ask for repository.
Former-commit-id: 4863d4b5144ef737b429dc79cc85dc8e9de7da7a
2018-09-26 15:44:51 -06:00

94 lines
3.7 KiB
Bash
Executable file

#!/bin/bash
function pause(){
read -p "$*"
}
#Root check!
if [ "$EUID" -ne 0 ]
then
whiptail --backtitle "Pentest Build Script" --title "NOT ROOT" --msgbox "YOU ARE NOT ROOT. PLEASE RUN THIS SCRIPT AS ROOT." 10 55
exit 1
fi
whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70
DISTRO=$(lsb_release -i -s)
DESKTOP=$(whiptail --backtitle "Pentest Build Script" --title "Desktop Environment" --menu "Choose a desktop environment:" 10 40 4 \
"gnome" "GNOME" \
"mate" "MATE Desktop Environment" \
"cinnamon" "Cinnamon Desktop Environment" \
"kde" "K Desktop Environment" 3>&1 1>&2 2>&3)
REPO=$(whiptail --backtitle "Pentest Build Script" --title "Repository" --menu "Which repository are you going to use?" 10 60 2 \
"default" "The distribution's default repository" \
"local" "A repository hosted on your local network" 3>&1 1>&2 2>&3)
if [ $REPO == "local" ];
then
REPO=$(whiptail --backtitle "Pentest Build Script" --title "Local Repository" --inputbox "What is the IP address of your local repository?" \
10 60 3>&1 1>&2 2>&3)
fi
if (whiptail --backtitle "Pentest Build Script" --title "Confirmation" --yesno "We're going to build:
Distribution: $DISTRO
Desktop Environment: $DESKTOP
Repository: $REPO
Shall we proceed?" 10 50)
then
whiptail --backtitle "Pentest Build Script" --title "Build Time!" --msgbox "This will take a while. Press OK to proceed." 8 30
else
whiptail --backtitle "Pentest Build Script" --title "OK" --msgbox "This script will now terminate." 8 35
exit 1
fi
#This is the stuff that's common to both distros
dpkg-name ./debs/*.deb
rm -rf build
mkdir build
apt install git build-essential cdebootstrap live-build curl wget
case $DISTRO in
Kali)
cp -rv ./modules/live-build-config ./build/
mkdir ./build/live-build-config/kali-config/variant-$DESKTOP/packages.chroot
cp ./debs/*.deb build/live-build-config/kali-config/variant-$DESKTOP/packages.chroot/
cp -rv filesystem/* ./build/live-build-config/kali-config/common/includes.chroot/
cp config/02-unattended-boot.binary ./build/live-build-config/kali-config/common/includes.installer/
cp config/preseed.cfg ./build/live-build-config/kali-config/common/includes.installer/
cp config/software.list.chroot ./build/live-build-config/kali-config/variant-$DESKTOP/
if [ $REPO != "default" ];
then
sed -i 's/archive.kali.org/$REPO/g' ./build/live-build-config/auto/config
fi
cd build/live-build-config
pause "Are you all ready to build? Hit enter when you're ready (this will take a while...)"
./build.sh --distribution kali-rolling --variant $DESKTOP --verbose
cp images/*.iso ../../images/
;;
Parrot)
cp -rv ./modules/parrot-build ./build/
mkdir ./build/parrot-build/templates/parrot-variant-$DESKTOP/packages.chroot
cp ./debs/*.deb build/parrot-build/templates/parrot-variant-$DESKTOP/packages.chroot/
cp -rv filesystem/* ./build/parrot-build/templates/common/includes.chroot/
cp config/software.list.chroot ./build/parrot-build/templates/parrot-variant-$DESKTOP/
if [ $REPO != "default" ];
then
sed -i "s|archive.parrotsec.org|$REPO|g" ./build/parrot-build/auto/config
sed -i "s|deb.parrotsec.org|$REPO|g" ./build/parrot-build/templates/common/archives/parrot.list
fi
cd build/parrot-build
pause "Are you all ready to build? Hit enter when you're ready (this will take a while...)"
./build.sh build variant-$DESKTOP amd64
cp *.iso ../../images/
;;
esac
whiptail --backtitle "Pentest Build Script" --title "Finished" --msgbox "Fingers crossed there were no errors... Your image should be done!" 7 70
exit 0