1#!/bin/sh -e 2# 3# Copyright (C) 2008-2011 Intel 4# 5# install.sh [device_name] [rootfs_name] [video_mode] [vga_mode] 6# 7 8PATH=/sbin:/bin:/usr/sbin:/usr/bin 9 10# figure out how big of a boot partition we need 11boot_size=$(du -ms /run/media/$1/ | awk '{print $1}') 12# remove rootfs.img ($2) from the size if it exists, as its not installed to /boot 13if [ -e /run/media/$1/$2 ]; then 14 boot_size=$(( boot_size - $( du -ms /run/media/$1/$2 | awk '{print $1}') )) 15fi 16# remove initrd from size since its not currently installed 17if [ -e /run/media/$1/initrd ]; then 18 boot_size=$(( boot_size - $( du -ms /run/media/$1/initrd | awk '{print $1}') )) 19fi 20# add 10M to provide some extra space for users and account 21# for rounding in the above subtractions 22boot_size=$(( boot_size + 10 )) 23 24# 5% for the swap 25swap_ratio=5 26 27# Get a list of hard drives 28hdnamelist="" 29live_dev_name=`cat /proc/mounts | grep ${1%/} | awk '{print $1}'` 30live_dev_name=${live_dev_name#\/dev/} 31# Only strip the digit identifier if the device is not an mmc 32case $live_dev_name in 33 mmcblk*) 34 ;; 35 nvme*) 36 ;; 37 *) 38 live_dev_name=${live_dev_name%%[0-9]*} 39 ;; 40esac 41 42echo "Searching for hard drives ..." 43 44# Some eMMC devices have special sub devices such as mmcblk0boot0 etc 45# we're currently only interested in the root device so pick them wisely 46devices=`ls /sys/block/ | grep -v mmcblk` || true 47mmc_devices=`ls /sys/block/ | grep "mmcblk[0-9]\{1,\}$"` || true 48devices="$devices $mmc_devices" 49 50for device in $devices; do 51 case $device in 52 loop*) 53 # skip loop device 54 ;; 55 sr*) 56 # skip CDROM device 57 ;; 58 ram*) 59 # skip ram device 60 ;; 61 *) 62 # skip the device LiveOS is on 63 # Add valid hard drive name to the list 64 case $device in 65 $live_dev_name*) 66 # skip the device we are running from 67 ;; 68 *) 69 hdnamelist="$hdnamelist $device" 70 ;; 71 esac 72 ;; 73 esac 74done 75 76TARGET_DEVICE_NAME="" 77for hdname in $hdnamelist; do 78 # Display found hard drives and their basic info 79 echo "-------------------------------" 80 echo /dev/$hdname 81 if [ -r /sys/block/$hdname/device/vendor ]; then 82 echo -n "VENDOR=" 83 cat /sys/block/$hdname/device/vendor 84 fi 85 if [ -r /sys/block/$hdname/device/model ]; then 86 echo -n "MODEL=" 87 cat /sys/block/$hdname/device/model 88 fi 89 if [ -r /sys/block/$hdname/device/uevent ]; then 90 echo -n "UEVENT=" 91 cat /sys/block/$hdname/device/uevent 92 fi 93 echo 94done 95 96# Get user choice 97while true; do 98 echo "Please select an install target or press n to exit ($hdnamelist ): " 99 read answer 100 if [ "$answer" = "n" ]; then 101 echo "Installation manually aborted." 102 exit 1 103 fi 104 for hdname in $hdnamelist; do 105 if [ "$answer" = "$hdname" ]; then 106 TARGET_DEVICE_NAME=$answer 107 break 108 fi 109 done 110 if [ -n "$TARGET_DEVICE_NAME" ]; then 111 break 112 fi 113done 114 115if [ -n "$TARGET_DEVICE_NAME" ]; then 116 echo "Installing image on /dev/$TARGET_DEVICE_NAME ..." 117else 118 echo "No hard drive selected. Installation aborted." 119 exit 1 120fi 121 122device=/dev/$TARGET_DEVICE_NAME 123 124# 125# The udev automounter can cause pain here, kill it 126# 127rm -f /etc/udev/rules.d/automount.rules 128rm -f /etc/udev/scripts/mount* 129 130# 131# Unmount anything the automounter had mounted 132# 133umount ${device}* 2> /dev/null || /bin/true 134 135if [ ! -b /dev/loop0 ] ; then 136 mknod /dev/loop0 b 7 0 137fi 138 139mkdir -p /tmp 140if [ ! -L /etc/mtab ] && [ -e /proc/mounts ]; then 141 ln -sf /proc/mounts /etc/mtab 142fi 143 144disk_size=$(parted ${device} unit mb print | grep '^Disk .*: .*MB' | cut -d" " -f 3 | sed -e "s/MB//") 145 146grub_version=$(grub-install -V|sed 's/.* \([0-9]\).*/\1/') 147 148if [ $grub_version -eq 0 ] ; then 149 bios_boot_size=0 150else 151 # For GRUB 2 we need separate parition to store stage2 grub image 152 # 2Mb value is chosen to align partition for best performance. 153 bios_boot_size=2 154fi 155 156swap_size=$((disk_size*swap_ratio/100)) 157rootfs_size=$((disk_size-bios_boot_size-boot_size-swap_size)) 158 159boot_start=$((bios_boot_size)) 160rootfs_start=$((bios_boot_size+boot_size)) 161rootfs_end=$((rootfs_start+rootfs_size)) 162swap_start=$((rootfs_end)) 163 164# MMC devices are special in a couple of ways 165# 1) they use a partition prefix character 'p' 166# 2) they are detected asynchronously (need rootwait) 167rootwait="" 168part_prefix="" 169if [ ! "${device#/dev/mmcblk}" = "${device}" ] || \ 170 [ ! "${device#/dev/nvme}" = "${device}" ]; then 171 part_prefix="p" 172 rootwait="rootwait" 173fi 174 175# USB devices also require rootwait 176if [ -n `readlink /dev/disk/by-id/usb* | grep $TARGET_DEVICE_NAME` ]; then 177 rootwait="rootwait" 178fi 179 180if [ $grub_version -eq 0 ] ; then 181 bios_boot='' 182 bootfs=${device}${part_prefix}1 183 rootfs=${device}${part_prefix}2 184 swap=${device}${part_prefix}3 185else 186 bios_boot=${device}${part_prefix}1 187 bootfs=${device}${part_prefix}2 188 rootfs=${device}${part_prefix}3 189 swap=${device}${part_prefix}4 190fi 191 192echo "*****************" 193[ $grub_version -ne 0 ] && echo "BIOS boot partition size: $bios_boot_size MB ($bios_boot)" 194echo "Boot partition size: $boot_size MB ($bootfs)" 195echo "Rootfs partition size: $rootfs_size MB ($rootfs)" 196echo "Swap partition size: $swap_size MB ($swap)" 197echo "*****************" 198echo "Deleting partition table on ${device} ..." 199dd if=/dev/zero of=${device} bs=512 count=35 200 201echo "Creating new partition table on ${device} ..." 202if [ $grub_version -eq 0 ] ; then 203 parted ${device} mktable msdos 204 echo "Creating boot partition on $bootfs" 205 parted ${device} mkpart primary ext3 0% $boot_size 206else 207 parted ${device} mktable gpt 208 echo "Creating BIOS boot partition on $bios_boot" 209 parted ${device} mkpart bios_boot 0% $bios_boot_size 210 parted ${device} set 1 bios_grub on 211 echo "Creating boot partition on $bootfs" 212 parted ${device} mkpart boot ext3 $boot_start $boot_size 213fi 214 215echo "Creating rootfs partition on $rootfs" 216[ $grub_version -eq 0 ] && pname='primary' || pname='root' 217parted ${device} mkpart $pname ext4 $rootfs_start $rootfs_end 218 219echo "Creating swap partition on $swap" 220[ $grub_version -eq 0 ] && pname='primary' || pname='swap' 221parted ${device} mkpart $pname linux-swap $swap_start 100% 222 223parted ${device} print 224 225echo "Waiting for device nodes..." 226C=0 227while [ $C -ne 3 ] && [ ! -e $bootfs -o ! -e $rootfs -o ! -e $swap ]; do 228 C=$(( C + 1 )) 229 sleep 1 230done 231 232echo "Formatting $bootfs to ext3..." 233mkfs.ext3 $bootfs 234 235echo "Formatting $rootfs to ext4..." 236mkfs.ext4 $rootfs 237 238echo "Formatting swap partition...($swap)" 239mkswap $swap 240 241mkdir /tgt_root 242mkdir /src_root 243mkdir -p /boot 244 245# Handling of the target root partition 246mount $rootfs /tgt_root 247mount -o rw,loop,noatime,nodiratime /run/media/$1/$2 /src_root 248echo "Copying rootfs files..." 249cp -a /src_root/* /tgt_root 250if [ -d /tgt_root/etc/ ] ; then 251 if [ $grub_version -ne 0 ] ; then 252 boot_uuid=$(blkid -o value -s UUID ${bootfs}) 253 swap_part_uuid=$(blkid -o value -s PARTUUID ${swap}) 254 bootdev="UUID=$boot_uuid" 255 swapdev=/dev/disk/by-partuuid/$swap_part_uuid 256 else 257 bootdev=${bootfs} 258 swapdev=${swap} 259 fi 260 echo "$swapdev swap swap defaults 0 0" >> /tgt_root/etc/fstab 261 echo "$bootdev /boot ext3 defaults 1 2" >> /tgt_root/etc/fstab 262 # We dont want udev to mount our root device while we're booting... 263 if [ -d /tgt_root/etc/udev/ ] ; then 264 echo "${device}" >> /tgt_root/etc/udev/mount.blacklist 265 fi 266fi 267umount /tgt_root 268umount /src_root 269 270echo "Looking for kernels to use as boot target.." 271# Find kernel to boot to 272# Give user options if multiple are found 273kernels="$(find /run/media/$1/ -type f \ 274 -name bzImage* -o -name zImage* \ 275 -o -name vmlinux* -o -name vmlinuz* \ 276 -o -name fitImage* \ 277 | sed s:.*/::)" 278if [ -n "$(echo $kernels)" ]; then 279 # only one kernel entry if no space 280 if [ -z "$(echo $kernels | grep " ")" ]; then 281 kernel=$kernels 282 echo "$kernel will be used as the boot target" 283 else 284 echo "Which kernel do we want to boot by default? The following kernels were found:" 285 echo $kernels 286 read answer 287 kernel=$answer 288 fi 289else 290 echo "No kernels found, exiting..." 291 exit 1 292fi 293 294# Handling of the target boot partition 295mount $bootfs /boot 296echo "Preparing boot partition..." 297 298if [ -f /etc/grub.d/00_header -a $grub_version -ne 0 ] ; then 299 echo "Preparing custom grub2 menu..." 300 root_part_uuid=$(blkid -o value -s PARTUUID ${rootfs}) 301 boot_uuid=$(blkid -o value -s UUID ${bootfs}) 302 GRUBCFG="/boot/grub/grub.cfg" 303 mkdir -p $(dirname $GRUBCFG) 304 cat >$GRUBCFG <<_EOF 305timeout=5 306default=0 307menuentry "Linux" { 308 search --no-floppy --fs-uuid $boot_uuid --set root 309 linux /$kernel root=PARTUUID=$root_part_uuid $rootwait rw $5 $3 $4 quiet 310} 311_EOF 312 chmod 0444 $GRUBCFG 313fi 314grub-install ${device} 315 316if [ $grub_version -eq 0 ] ; then 317 echo "(hd0) ${device}" > /boot/grub/device.map 318 echo "Preparing custom grub menu..." 319 echo "default 0" > /boot/grub/menu.lst 320 echo "timeout 30" >> /boot/grub/menu.lst 321 echo "title Live Boot/Install-Image" >> /boot/grub/menu.lst 322 echo "root (hd0,0)" >> /boot/grub/menu.lst 323 echo "kernel /$kernel root=$rootfs rw $3 $4 quiet" >> /boot/grub/menu.lst 324fi 325 326# Copy kernel artifacts. To add more artifacts just add to types 327# For now just support kernel types already being used by something in OE-core 328for types in bzImage zImage vmlinux vmlinuz fitImage; do 329 for kernel in `find /run/media/$1/ -name $types*`; do 330 cp $kernel /boot 331 done 332done 333 334umount /boot 335 336sync 337 338echo "Remove your installation media, and press ENTER" 339 340read enter 341 342echo "Rebooting..." 343reboot -f 344