Merge pull request #4 from kforney/master

Add CLI options for use with unattended builds.

Former-commit-id: 0e31d805b2
Former-commit-id: 5217c50cd00463eeeec33c6585b38e7e9201eee8
This commit is contained in:
Don Bowers 2018-10-04 15:41:16 -06:00 committed by GitHub
commit af87f177c4
3 changed files with 68 additions and 31 deletions

View file

@ -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.

View file

@ -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

View file

@ -11,10 +11,25 @@ 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)
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
if [[ $UNATTEND != "true" ]];
then
whiptail --backtitle "Pentest Build Script" --title "Welcome" --msgbox "This script will help you build a custom pentesting distribution." 7 70
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" \
@ -44,6 +59,7 @@ 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
dpkg-name ./debs/*.deb
@ -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