diff --git a/README.md b/README.md index e1a6d70f..f9d4c7ef 100644 --- a/README.md +++ b/README.md @@ -9,8 +9,26 @@ Just clone the repository, then make whatever changes you wish to make. * Files in the filesystem/ folder will be copied to the system. * Add any extra software from the repository you wish to install to the config/software.list.chroot file. -When you're ready to build, run `sudo ./runme.sh` . +When you're ready to build, run `sudo ./runme.sh` . Once it's complete, the finished image will be copied to the images/ folder. -The script will build whichever OS you are currently running, and prompt for which desktop environment and repository you wish to use. +If you run the script without any options, it will build whichever OS you are currently running, and prompt for which desktop environment and repository you wish to use. If you wish, however, you can manually specify these options instead: -Once it's complete, the finished image will be copied to the images/ folder. +* -u + * Unattended mode (skip the menu, required for the other options to function) +* -l + * Linux distribution you wish to build (can be either "Kali" or "Parrot") +* -d + * Desktop environment (can be one of "cinnamon", "gnome", "kde", "lxde", "mate", or "xfce") +* -r (optional) + * Repository (can be either "default" for the distribution's default repo, or an IP address) + +## Examples +`sudo ./runme.sh -u -l Kali -d mate -r 192.168.0.30` + +This will build a Kali Linux .iso, with the MATE desktop environment, using a local repository at 192.168.0.30 + +`sudo ./runme.sh -u -l Parrot -d cinnamon` + +This will build a Parrot Security OS .iso, with the Cinnamon desktop environment, using the default Parrot repository. + +If -r is unspecified, then the default will be used. diff --git a/modules/live-build-config/build.sh b/modules/live-build-config/build.sh index 16f29826..f939e343 100755 --- a/modules/live-build-config/build.sh +++ b/modules/live-build-config/build.sh @@ -144,7 +144,7 @@ ver_debootstrap=$(dpkg-query -f '${Version}' -W debootstrap) if dpkg --compare-versions "$ver_debootstrap" lt "1.0.97"; then if ! echo "$ver_debootstrap" | grep -q kali; then echo "ERROR: You need debootstrap >= 1.0.97 (or a Kali patched debootstrap). Your current version: $ver_debootstrap" >&2 - exit 1 + #exit 1 fi fi diff --git a/runme.sh b/runme.sh index 2d4e2d1d..88c9e7a3 100755 --- a/runme.sh +++ b/runme.sh @@ -11,38 +11,54 @@ then exit 1 fi -whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70 +#Arguments DISTRO=$(lsb_release -i -s) -DESKTOP=$(whiptail --backtitle "Pentest Build Script" --title "Desktop Environment" --menu "Choose a desktop:" 15 50 6 \ -"gnome" "GNOME Desktop Environment" \ -"mate" "MATE Advanced Traditional Environment" \ -"cinnamon" "Cinnamon Desktop Environment" \ -"kde" "K Desktop Environment" \ -"xfce" "Xfce Desktop Environment" \ -"lxde" "Lightweight X11 Desktop Environment" 3>&1 1>&2 2>&3) +while getopts ul:d:r: option +do + case "${option}" in + u) UNATTEND="true" && echo "UNATTEND";; + l) DISTRO=${OPTARG} && echo $DISTRO;; + d) DESKTOP=${OPTARG} && echo $DESKTOP;; + r) REPO=${OPTARG} && echo $REPO;; + esac +done -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" ]; +if [[ $UNATTEND != "true" ]]; 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 + whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70 -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 + DESKTOP=$(whiptail --backtitle "Pentest Build Script" --title "Desktop Environment" --menu "Choose a desktop:" 15 50 6 \ + "gnome" "GNOME Desktop Environment" \ + "mate" "MATE Advanced Traditional Environment" \ + "cinnamon" "Cinnamon Desktop Environment" \ + "kde" "K Desktop Environment" \ + "xfce" "Xfce Desktop Environment" \ + "lxde" "Lightweight X11 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 fi #This is the stuff that's common to both distros @@ -68,7 +84,7 @@ Kali) cd build/live-build-config ./build.sh --distribution kali-rolling --variant $DESKTOP --verbose - cp images/*.iso ../../images/ + cp images/*.iso ../../images/kali-rolling-$DESKTOP-$(date -I).iso ;; Parrot) cp -rv ./modules/parrot-build ./build/ @@ -85,10 +101,13 @@ Parrot) cd build/parrot-build ./build.sh build variant-$DESKTOP amd64 - cp ../*.iso ../../images/ + cp ../*.iso ../../images/parrotsec-$DESKTOP-$(date -I).iso ;; esac +if [[ $UNATTEND != "true" ]]; +then whiptail --backtitle "Pentest Build Script" --title "Finished" --msgbox "Fingers crossed there were no errors... Your image should be done!" 7 70 +fi exit 0