ukupgrade

#!/bin/bash

# @author Mustafa Hasturk
# @mail mustafa.hasturk@yandex.com
# @author Caio Oliveira
# @mail caiooliveirafarias0@gmail.com
# @author Christoph Kepler
# @mail development@kepler.international
# @author Usb Key
# @mail usbkey9@gmail.com

help="Usage: ukupgrade [OPTION]...
Dowload and install last available kernel for ubuntu systems.
Settings can be saved if no options provided or by running setup script

  -r,			Get Release Candidate kernel (default: Stable kernel)
  -l,			Get Low Latency kernel (default: Generic kernel)
  -a,			Run ukpurge script after upgrade
  -h,			Show this help
  -simple,		Use default settings (stable, generic, no purge)

Examples:
  ukupgrade -simple	Download, install the last kernel using defaults settings
  ukupgrade -a		Same behavior of previous but uninstall old kernels
"

#Follow symlink and cd to right directory
SOURCE="${BASH_SOURCE[0]}"
while [ -h "$SOURCE" ]; do
	DIR="$( cd -P "$( dirname "$SOURCE" )" && pwd )"
	SOURCE="$(readlink "$SOURCE")"
	[[ $SOURCE != /* ]] && SOURCE="$DIR/$SOURCE"
done
cd "$( cd -P "$( dirname "$SOURCE" )" && pwd )"

if ! which lynx > /dev/null; then sudo apt-get install lynx -y; fi

arch=$(dpkg --print-architecture)

function download() {
	wget -q --show-progress -P /tmp $(lynx -dump -listonly -dont-wrap-pre $kernelURL | grep "$1" | grep "$2" | grep "$arch" | cut -d ' ' -f 4 | tail -1)
	if [ $? == "1" ]; then
		echo "Download failed!"
		exit 1
	fi
}

function cleanup() {
	sudo rm -rf /tmp/linux*.deb
}

if [ "$1" = "-simple" ]; then
	source ./sample.uku.cfg ;
else
	if [ $# -eq 0 ]; then
		if [ ! -f ./uku.cfg ]; then ./setup ; fi
		source ./uku.cfg ;
	else
		source ./sample.uku.cfg ;

		while getopts 'rlph' flag; do
			case "${flag}" in
				r) kernelrc=1 ;;
				l) kernelmode='lowlatency' ;;
				p) arm=1 ;;
				h) 	echo "${help}";
					exit;
					;;
				*) 	echo "Unexpected option ${flag}";
					echo "${help}";
					exit;
					;;
			esac
		done
	fi ;
fi

case "$kernelrc" in
	1) kernelURL=$(lynx -dump -nonumbers http://kernel.ubuntu.com/~kernel-ppa/mainline/ | tail -1) ;;
	*) kernelURL="http://kernel.ubuntu.com/~kernel-ppa/mainline/"$(lynx -dump -nonumbers http://kernel.ubuntu.com/~kernel-ppa/mainline/ | grep -v rc | rev | cut -d '/' -f 2 | rev | grep -E 'v.+' | sort -V | rev | cut -d ' ' -f 1 | rev | tail -1)"/" ;;
esac

# Same version test
reqVersion=$(echo $kernelURL | awk -F "/v" '{print $2}' | cut -d '/' -f 1)
locVersion=$(dpkg -l | grep linux-image | grep $kernelmode | awk '{print $3}' | sed -e 's/.0-.*rc/.rc/' | cut -d '-' -f 1 | sed -e 's/.rc/-rc/')

if echo "$locVersion" | grep -Fxq "$reqVersion"; then
	echo 'Kernel up to date. Finishing'
	exit
fi

# Cleaning up tmp dir
cat << EOF

Cleaning old downloads in /tmp
EOF
cleanup

# Download Kernel Modules
cat << EOF

Downloading the latest $kernelmode kernel module...
EOF
download $kernelmode module

# Download Kernel Image
cat << EOF

Downloading the latest $kernelmode kernel image...
EOF
download $kernelmode image

# Download Kernel Header
cat << EOF

Downloading the latest $kernelmode kernel header...
EOF
download $kernelmode header


# Download Shared Kernel Header
cat << EOF

Downloading the shared kernel header...
EOF
wget -q --show-progress -P /tmp $(lynx -dump -listonly -dont-wrap-pre $kernelURL | grep all | grep headers | cut -d ' ' -f 4 | tail -1)
if [ $? == '1' ]; then
	echo 'Download failed!'
	exit 1
fi

# Install Kernel
cat << EOF

Installing Kernel and Headers...
EOF

while fuser /var/lib/dpkg/lock >/dev/null 2>&1 ; do
    sleep 0.5
done

sudo dpkg -i /tmp/linux-modules*.deb
sudo dpkg -i /tmp/linux-header*.deb
sudo dpkg -i /tmp/linux-image*.deb
cleanup
touch /tmp/ukupgrade.nr

cat << EOF

Finished.
EOF

if [ $arm -eq 1 ]; then ./ukpurge ; fi

if [ $cron = 0 ]; then
	read -p 'Do you want to reboot system now? (y/n)(10s to No): ' -n 1 -t 10 -s rs
	case "$rs" in
		y | Y)
			echo "Yes, reboot this now!"
			sleep 2
			sudo reboot ;;
		*) echo 'No, stay this wake uped!' ;;
	esac ;
fi