Merge pull request #184 from themactep/master

Something to review and discuss.
pull/187/head
Igor Zalatov 2022-02-23 11:04:09 +03:00 committed by GitHub
commit 9e3add8313
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 190 additions and 230 deletions

View File

@ -1,268 +1,228 @@
#!/bin/sh #!/bin/sh
# #
# OpenIPC.org | v.20220221 # OpenIPC.org | v.20220222
# #
set -e set -e
soc=$(fw_printenv -n soc) || (echo -e "\n\e[1;31mSOC not defined in U-Boot ENV...\e[0m\n"; exit 1) echo_c() {
vendor=$(ipcinfo --vendor) [ -z "$HASERLVER" ] && t="\e[1;$1m$2\e[0m" || t="$2"
kernel=$(awk -F ':' '/"kernel"/ {print $1}' /proc/mtd) echo -e "$t"
rootfs=$(awk -F ':' '/"rootfs"/ {print $1}' /proc/mtd)
overlay=$(awk -F ':' '/"rootfs_data"/ {print $1}' /proc/mtd)
kernelrelease=0x$(xxd -l 4 -s 8 -p /dev/"${kernel}" | xargs)
systemrelease=$(cat /etc/os-release | grep "GITHUB_VERSION" | head -1 | cut -d= -f2)
github="https://github.com/OpenIPC/firmware/releases/download/latest"
option="-s --connect-timeout 30 --max-time 60"
stop_services() {
echo -e "\n\e[1;31mForced shutdown of services...\e[0m\n"
killall majestic || true
killall crond klogd ntpd rngd syslogd || true
} }
unload_modules() { die() {
echo -e "\n\e[1;31mUnload vendor specific modules...\e[0m\n" echo_c 31 "$1 Aborting."
load_$vendor -r >/dev/null 2>&1 || true reboot_system
}
check_memory() {
echo -e "\n\e[1;31mCheck RAM size...\e[0m\n"
free
}
download_firmware() {
if [ $remotely -eq 1 ]; then
echo -e "\n\e[1;31mDownloading the latest firmware...\e[0m\n"
echo ${1}
curl ${option} -L "${1}" -o - | gzip -d | tar xvf - -C /tmp || (echo -e "\n\e[1;31mError receiving the update file, execution was interrupted...\e[0m\n"; exit 1)
fi
}
check_sum() {
if [ $md5 -eq 1 ]; then
cd $(dirname "${1}")
md5sum -c -s "${1}.md5sum"
if [ $? -eq 0 ]; then
echo -e "\n\e[1;31m${2}: Checksum verified...\e[0m\n"
else
echo -e "\n\e[1;31m${2}: Checksum error, execution was interrupted...\e[0m\n"
exit 1
fi
else
echo -e "\n\e[1;31m${2}: Checksum skipped...\e[0m\n"
fi
} }
check_soc() { check_soc() {
if [ $host -eq 1 ]; then [ "$skip_soc" -eq 1 ] && echo "Skip SoC validation" && return 0
echo -e "\n\e[1;32m${2}: installed SOC: ${soc}\e[0m" [ "$1" = "$soc" ] && echo "SoC OK" && return 0
echo -e "\e[1;32m${2}: new SOC: ${1}\e[0m\n" die "Wrong SoC!"
if [ "${1}" = "${soc}" ]; then
echo -e "\n\e[1;31m${2}: Hostname verified...\e[0m\n"
else
echo -e "\n\e[1;31m${2} is not for this system, execution was interrupted...\e[0m\n"
exit 1
fi
else
echo -e "\n\e[1;31m${2}: Hostname check skipped...\e[0m\n"
fi
} }
check_release() { compare_versions() {
if [ ${version} -eq 1 ]; then [ "$skip_ver" -eq 1 ] && echo "Skip version checking" && return 1
echo -e "\n\e[1;32m${3}: installed release: ${1}\e[0m" [ "$1" = "$2" ] && echo_c 32 "Same version, nothing to update" && return 0
echo -e "\e[1;32m${3}: new release: ${2}\e[0m\n" echo_c 32 "New version, going to update" && return 1
if [ "${1}" = "${2}" ]; then
echo -e "\n\e[1;31m${3} has the same release version, skipping...\e[0m\n"
return 0
else
return 1
fi
else
echo -e "\n\e[1;31m${3}: Release version check skipped...\e[0m\n"
return 1
fi
} }
mount_rootfs() { do_update_kernel() {
mkdir -p /tmp/rootfs x=$1
loop=$(losetup -f) [ -z "$x" ] && x="/tmp/uImage.$soc"
[ $? -eq 0 ] && losetup "${loop}" "${1}" && mount "${loop}" /tmp/rootfs echo_c 33 "\nKernel"
if [ $? -eq 0 ]; then echo "Update kernel from $x"
new_soc=$(cat /tmp/rootfs/etc/hostname | head -1 | cut -d- -f2) [ ! -f "$x" ] && die "File $x not found"
new_systemrelease=$(cat /tmp/rootfs/etc/os-release | grep "GITHUB_VERSION" | head -1 | cut -d= -f2) check_soc "$(od -j 32 -N 32 -S 1 -A n "$x" | cut -d- -f3)"
umount /tmp/rootfs compare_versions "$kernel_version" "$(get_kernel_version "$x")" && return 0
rm -rf /tmp/rootfs flashcp -v "$x" "$kernel_device"
losetup -d "${loop}" echo_c 32 "Kernel updated to $(get_kernel_version "$kernel_device")"
else unset x
if [ $host -eq 1 ]; then
echo -e "\n\e[1;31mRootfs: Unable to get hostname, execution was interrupted...\e[0m\n"
exit 1
fi
fi
} }
writing_kernel() { do_update_rootfs() {
if [ $update_kernel -eq 1 ]; then x=$1
check_sum "${1}" "Kernel" [ -z "$x" ] && x="/tmp/rootfs.squashfs.$soc"
echo_c 33 "\nRootFS"
echo "Update rootfs from $x"
[ ! -f "$x" ] && die "File ${x} not found"
y=/tmp/rootfs
if mkdir -p "$y" && loop=$(losetup -f) && losetup "$loop" "$x" && mount "$loop" "$y"; then
check_soc "$(head -1 ${y}/etc/hostname | cut -d- -f2)"
compare_versions "$system_version" "$(get_system_version "$y")" && return 0
umount "$y" && rm -rf "$y" && losetup -d "$loop"
else
die "Unable to mount $y!"
fi
unset y
new_kernelrelease=0x$(xxd -l 4 -s 8 -p "${1}" | xargs) flashcp -v "$x" "$(get_device "rootfs")"
check_release "${kernelrelease}" "${new_kernelrelease}" "Kernel" && return 0 echo_c 32 "RootFS updated to $(get_system_version "")"
unset x
new_soc=$(od -N 32 -j 32 "${1}" -S 1 -A none | cut -d- -f3)
check_soc "${new_soc}" "Kernel"
echo -e "\n\e[1;31mWriting a new kernel...\e[0m\n"
flashcp -v "${1}" /dev/${kernel}
fi
} }
writing_rootfs() {
if [ $update_rootfs -eq 1 ]; then
check_sum "${1}" "Rootfs"
mount_rootfs "${1}" do_wipe_overlay() {
check_release "${systemrelease}" "${new_systemrelease}" "Rootfs" && return 0 echo_c 33 "\nOverlayFS"
check_soc "${new_soc}" "Rootfs" echo "Erase overlay partition"
flash_eraseall -j "$(get_device "rootfs_data")"
echo -e "\n\e[1;31mWriting a new rootfs...\e[0m\n"
flashcp -v "${1}" /dev/${rootfs}
fi
} }
clearing_overlayfs() { download_firmware() {
if [ $clear_overlayfs -eq 1 ]; then echo_c 33 "\nFirmware"
echo -e "\n\e[1;31mСlearing the overlayfs partition...\e[0m\n" [ -z "$url" ] && url="https://github.com/OpenIPC/firmware/releases/download/latest/openipc.${soc}-br.tgz"
flash_eraseall -j /dev/${overlay} echo "Download from $url"
fi [ -z "$HASERLVER" ] && progress="-#" || progress="-s"
[ "$(curl -o /dev/null -s -w '%{http_code}\n' "$url")" = "000" ] && die "Check your network!"
curl --connect-timeout 30 -m 60 -L "$url" ${progress} -o - | gzip -d | tar xf - -C /tmp || die "Cannot retrieve $url"
if [ "$skip_md5" -eq 0 ]; then
(cd /tmp && md5sum -s -c *.md5sum) || die "Wrong checksum!"
fi
} }
quick_reboot() { free_resources() {
if [ $no_reboot -eq 0 ]; then echo_c 97 "\nStop services, unload modules"
echo -e "\n\e[1;31mRebooting without any questions...\e[0m\n" killall majestic crond klogd ntpd rngd syslogd >/dev/null 2>&1 || true
sleep 1 "load_$vendor" -r >/dev/null 2>&1 || true
reboot -f }
fi
get_device() {
echo -n "/dev/$(grep "\"$1\"" /proc/mtd | cut -d: -f1)"
}
get_kernel_version() {
echo -n "0x$(xxd -l 4 -s 8 -p "$1" | xargs)"
}
get_system_info() {
vendor=$(ipcinfo --vendor)
soc=$(fw_printenv -n soc) || die "SoC is not defined in U-Boot environment"
kernel_device=$(get_device "kernel")
kernel_version=$(get_kernel_version "$kernel_device")
system_version=$(get_system_version "")
}
get_system_version() {
grep "GITHUB_VERSION" "$1/etc/os-release" | head -1 | cut -d= -f2 | sed 's/"//g'
}
print_sysinfo() {
get_system_info
echo_c 33 "OpenIPC System Updater"
echo_c 96 "\nVendor\t$vendor\nSoC\t$soc\nKernel\t$kernel_version\nRootFS\t$system_version"
} }
print_usage() { print_usage() {
echo "Usage: ./sysupgrade [-option]" echo "
echo "options:" Usage: $0 [options]
echo "Remotely:"
echo " -k update kernel" -k update kernel from online repository
echo " -r update rootfs" -r update rootfs from online repository
echo " --url=URL URL for update file (.tgz format)" --url=[URL] custom URL to update from (.tgz format)
echo "Locally:" --kernel=[FILE] update kernel from file (uImage format)
echo " --kernel=[FILE] update kernel from file (uImage format)" --rootfs=[FILE] update rootfs from file (squashfs format)
echo " --rootfs=[FILE] update rootfs from file (squashfs format)" --force_md5 do not validate MD5 hash
echo --force_soc do not validate processor
echo " -f skip check md5 hash, release version" --force_ver do not validate release version
echo " --force and hostname" --force_all do not validate anything
echo " -n clear overlayfs" -n, --wipe_overlay wipe overlay partition
echo " -x no reboot after success execution" -x, --no_reboot do not reboot after updating
echo " -h --help help information" -h, --help display this help and exit
"
} }
reboot_system() {
[ "$skip_reboot" -eq 1 ] && echo_c 31 "\nReboot needed!" && exit 1
echo_c 97 "\nUnconditional reboot"
reboot -d 1 -f
}
clear_overlay=0
remote_update=0
skip_md5=0
skip_soc=0
skip_ver=0
skip_reboot=0
update_kernel=0 update_kernel=0
update_rootfs=0 update_rootfs=0
clear_overlayfs=0
no_reboot=0
usage=0
from_url=0
remotely=0
locally=0
md5=1
host=1
version=1
for arg in $@ for i in "$@"; do
do case $i in
case $arg in --force_all)
"-k") skip_md5=1
update_kernel=1 skip_soc=1
remotely=1 skip_ver=1
;; shift
"-r") ;;
update_rootfs=1 --force_md5)
remotely=1 skip_md5=1
;; shift
"-n") ;;
clear_overlayfs=1 --force_soc)
;; skip_soc=1
"-x") shift
no_reboot=1 ;;
;; --force_ver)
"-f") skip_ver=1
md5=0 shift
version=0 ;;
;; -h|--help)
"-h") print_sysinfo
usage=1 print_usage
;; exit 0
*) ;;
case $(echo $arg | cut -d= -f1) in -k )
"--kernel") update_kernel=1
kernel_file=$(echo =$arg | cut -d= -f3-) remote_update=1
update_kernel=1 shift
locally=1 ;;
;; --kernel=*)
"--rootfs") update_kernel=1
rootfs_file=$(echo =$arg | cut -d= -f3-) kernel_file="${i#*=}"
update_rootfs=1 shift
locally=1 ;;
;; -n|--wipe_overlay)
"--url") clear_overlay=1
url=$(echo =$arg | cut -d= -f3-) shift
from_url=1 ;;
remotely=1 -r)
;; update_rootfs=1
"--force") remote_update=1
md5=0 shift
host=0 ;;
version=0 --rootfs=*)
;; update_rootfs=1
"--help") rootfs_file="${i#*=}"
usage=1 shift
;; ;;
*) --url=*)
usage=1; url="${i#*=}"
;; shift
esac ;;
-x|--no_reboot)
skip_reboot=1
shift
;;
*)
print_sysinfo
echo_c 97 "\nUnknown option: $1"
print_usage
exit 1
;;
esac esac
done done
if ( [ $remotely -eq 1 ] && [ $locally -eq 1 ] ) || [ $usage -eq 1 ] || \ print_sysinfo
( [ $remotely -eq 0 ] && [ $locally -eq 0 ] && [ $clear_overlayfs -eq 0 ] ); then
print_usage
echo one
exit 1
fi
if [ $remotely -eq 0 ] && [ $locally -eq 0 ] && [ $clear_overlayfs -eq 1 ]; then [ "$update_kernel" -eq 0 ] && \
clearing_overlayfs && quick_reboot [ "$update_rootfs" -eq 0 ] && \
exit 1 [ "$clear_overlay" -eq 0 ] && \
fi echo_c 97 "\nTry '$(basename "$0") --help' for options." && \
exit 0
[ -z "${kernel_file}" ] && kernel_file="/tmp/uImage.${soc}" free_resources
[ -z "${rootfs_file}" ] && rootfs_file="/tmp/rootfs.squashfs.${soc}" [ "$remote_update" -eq 1 ] && download_firmware
[ "$update_kernel" -eq 1 ] && do_update_kernel "$kernel_file"
if [ $remotely -eq 1 ]; then [ "$update_rootfs" -eq 1 ] && do_update_rootfs "$rootfs_file"
if [ $from_url -eq 1 ]; then [ "$clear_overlay" -eq 1 ] && do_wipe_overlay
if [ -z "${url}" ]; then reboot_system
echo -e "\n\e[1;31mThere is no URL to update file...\e[0m\n"
exit 1
fi
else
if [ "$(curl -o /dev/null -s -w '%{http_code}\n' http://github.com)" != "301" ]; then
echo -e "\n\e[1;31mThere is no access to the github.com, please check the Internet connection...\e[0m\n"
exit 1
fi
url="${github}/openipc.${soc}-br.tgz"
fi
fi
stop_services
unload_modules
download_firmware "${url}" && check_memory && writing_kernel "${kernel_file}" && writing_rootfs "${rootfs_file}" && sleep 1 && clearing_overlayfs && quick_reboot