#!/bin/bash
# Original kali-pxe script (C) by IceF0x
# Under licence GPLv3 http://www.gnu.org/licenses/gpl.txt

echo " _  __     _ _   ______  _______ 
| |/ /__ _| (_) |  _ \ \/ / ____|
| ' // _\` | | | | |_) \  /|  _|  
| . \ (_| | | | |  __//  \| |___ 
|_|\_\__,_|_|_| |_|  /_/\_\_____|"

if [ "$#" != 1 ]
then
  echo -e "Usage: $0 <arch>
       Exemple: $0 32
       Use -h or --help to display this help"
        exit 1
fi                      

function dnsmasqinstall(){
apt-get install dnsmasq
mv /etc/dnsmasq.conf /etc/dnsmasq.conf.bak


echo "interface=eth0" > /etc/dnsmasq.conf
echo "dhcp-range=192.168.101.100,192.168.101.200,12h" >> /etc/dnsmasq.conf
echo "dhcp-boot=pxelinux.0" >> /etc/dnsmasq.conf
echo "enable-tftp" >> /etc/dnsmasq.conf
echo "tftp-root=/tftpboot/" >> /etc/dnsmasq.conf
echo "dhcp-option=3,192.168.101.1" >> /etc/dnsmasq.conf
echo "dhcp-option=6,8.8.8.8,8.8.4.4" >> /etc/dnsmasq.conf

service dnsmasq restart

mkdir -p /tftpboot
cd /tftpboot
}

case "$1" in
        32)
        dnsmasqinstall
        # for 32 bit systems:
        wget http://repo.kali.org/kali/dists/kali/main/installer-i386/current/images/netboot/netboot.tar.gz
        tar zxpf netboot.tar.gz
        rm netboot.tar.gz
        ;;
        64)
        dnsmasqinstall
        # for 64 bit systems:
        wget http://repo.kali.org/kali/dists/kali/main/installer-amd64/current/images/netboot/netboot.tar.gz
        tar zxpf netboot.tar.gz
        rm netboot.tar.gz
        ;;
        *|-h|--help)
        echo -e "Usage: $0 <arch>
       Exemple: $0 32
       Use -h or --help to display this help"
        exit 1
        ;;
esac

exit 0