PXE-scripts/create-stock-menus.sh at master · OSgenie/PXE-scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env bash
# Kirtley Wienbroer
# kirtley@osgenie.com
scriptdir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
source $scriptdir/pxe.config

nfs_path=$nfs_server:/pxeboot
nfs_root_path=$nfs_server:/pxeboot/stock
seed_path=http://\$server_ip/preseed
seed_file="ubuntu.seed"

function check_for_sudo ()
{
if [ $UID != 0 ]; then
		echo "You need root privileges"
		exit 2
fi
}

function distro_title ()
{
cat > $menupath << EOM
MENU TITLE --== $distro ==--
LABEL rootmenu
MENU LABEL <---- Stock Menu
    kernel vesamenu.c32
    append menus/stock.conf

EOM
}

function pxe_boot_stock_iso ()
{
kernelpath=$tftp_boot_folder/$boot_folder
mkdir -p $tftp_folder/$kernelpath
cp -uv $subfolder/$boot_folder/$distro_kernel $tftp_folder/$kernelpath/
cp -uv $subfolder/$boot_folder/$distro_ram_disk $tftp_folder/$kernelpath/
cat >> $menupath << EOM
LABEL $revision
MENU LABEL $revision
    kernel $kernelpath/$distro_kernel
    append initrd=$kernelpath/$distro_ram_disk noprompt boot=$boot_folder url=$seed_path/$seed_file netboot=nfs root=/dev/nfs nfsroot=$nfs_root_path/$distro/$revision ro toram -

EOM
}

function generate_stock_menu ()
{
	echo "**********************************************"
	echo "			Stock Menus"
	echo "**********************************************"

	mount -t nfs4 $nfs_path /mnt/
	for folder in /mnt/stock/*; do
	distro=$(basename "$folder")
	menupath="$tftp_folder/menus/stock/$distro.conf"
	echo "creating Stock - $distro menu..."
	distro_title
		# PXE boot menu entry for each iso
		revisions=$( ls -r $folder )
		for revision in $revisions; do
		subfolderarray=$folder/$revision
			for subfolder in $subfolderarray; do
			revision=$(basename "$subfolder")
			tftp_boot_folder=boot/$distro/$revision
			if [ -e $subfolder/casper ]; then
					boot_folder=casper
						if [ -f $subfolder/casper/vmlinuz ]; then
								distro_kernel=vmlinuz
						elif [ -f $subfolder/casper/vmlinuz.efi ]; then
								distro_kernel=vmlinuz.efi
						else
							echo "ERROR - $distro-$revision"
							echo "Kernel Not Found!!"
						fi
						if [ -e $subfolder/casper/initrd.lz ]; then
								distro_ram_disk=initrd.lz
						elif [ -e $subfolder/casper/initrd.gz ]; then
								distro_ram_disk=initrd.gz
						elif [ -e $subfolder/casper/initrd ]; then
								distro_ram_disk=initrd
						fi
			elif [ -e $subfolder/live ]; then
					boot_folder=live
						if [ -e $subfolder/live/initrd.img ]; then
						distro_ram_disk=initrd.img
						fi
						if [ -f $subfolder/live/vmlinuz ]; then
							distro_kernel=vmlinuz
						elif [ -f $subfolder/live/vmlinuz.efi ]; then
							distro_kernel=vmlinuz.efi
						else
							echo "ERROR - $distro-$revision"
							echo "Kernel Not Found!!"
						fi
			elif [ -e $subfolder/install ]; then
					boot_folder=install
						if [ -e $subfolder/install/initrd.gz ]; then
							distro_ram_disk=initrd.gz
						fi
						if [ -f $subfolder/install/vmlinuz ]; then
							distro_kernel=vmlinuz
						fi
			fi
			if [ -z $distro_ram_disk ]; then
				echo "ERROR - $distro-$revision"
				echo "RAM Disk Not Found!!"
				rm $menupath
			elif [ -z $distro_kernel ]; then
				echo "ERROR - $distro-$revision"
				echo "Kernel Not Found!!"
				rm $menupath
			else
				pxe_boot_stock_iso
			fi
			done
		done
	done
	umount -d /mnt/
}

check_for_sudo
generate_stock_menu