#! /usr/bin/env bash # # Script for upload the image to yadisk & change the release message on GitHub # Copyright (C) 2018 Copter Express Technologies # # Author: Artem Smirnov # Author: Andrey Dvornikov # # Distributed under MIT License (available at https://opensource.org/licenses/MIT). # The above copyright notice and this permission notice shall be included in all # copies or substantial portions of the Software. # set -e # Exit immidiately on non-zero result echo_stamp() { # TEMPLATE: echo_stamp # TYPE: SUCCESS, ERROR, INFO # More info there https://www.shellhacks.com/ru/bash-colors/ TEXT="$(date '+[%Y-%m-%d %H:%M:%S]') $1" TEXT="\e[1m$TEXT\e[0m" # BOLD case "$2" in SUCCESS) TEXT="\e[32m${TEXT}\e[0m";; # GREEN ERROR) TEXT="\e[31m${TEXT}\e[0m";; # RED *) TEXT="\e[34m${TEXT}\e[0m";; # BLUE esac echo -e ${TEXT} } # This script doesn't work on Ubuntu because OS`s losetup does not consist --partscan (-P). # Idea: use `mount -o loop,offset` # https://stefanoprenna.com/blog/2014/09/22/tutorial-how-to-mount-raw-images-img-images-on-linux/ # REPO_DIR=$(mktemp -d --suffix=.builder_repo) # mount -t ext4 -o loop,offset=$((94208 * 512)) image/clever_qemu_test_2_20180822_163141.img "$REPO_DIR" # mount -t vfat -o loop,offset=$((8192 * 512)) image/clever_qemu_test_2_20180822_163141.img "$REPO_DIR/boot" execute() { # execute [ [...]] echo_stamp "Mount loop-image: $1" local DEV_IMAGE=$(losetup -Pf $1 --show) sleep 0.5 local MOUNT_POINT=$(mktemp -d --suffix=.builder_image) echo_stamp "Mount dirs ${MOUNT_POINT} & ${MOUNT_POINT}/boot" mount "${DEV_IMAGE}p2" ${MOUNT_POINT} mount "${DEV_IMAGE}p1" ${MOUNT_POINT}/boot echo_stamp "Bind system dirs" echo_stamp "Mounting /proc in chroot... " if [ ! -d ${MOUNT_POINT}/proc ]; then mkdir -p ${MOUNT_POINT}/proc \ && mount -t proc -o nosuid,noexec,nodev proc ${MOUNT_POINT}/proc \ && echo_stamp "OK" "SUCCESS" \ || (echo_stamp "Failed" "ERROR"; exit 1) else echo_stamp "/proc already exist" "SUCCESS"; fi echo_stamp "Mounting /sys in chroot... " if [ ! -d ${MOUNT_POINT}/sys ]; then mkdir -p ${MOUNT_POINT}/sys \ && mount -t sysfs -o nosuid,noexec,nodev sysfs ${MOUNT_POINT}/sys \ && echo_stamp "OK" "SUCCESS" \ || (echo_stamp "Failed" "ERROR"; exit 1) else echo_stamp "/sys already exist" "SUCCESS"; fi echo_stamp "Mounting /dev/ and /dev/pts in chroot... " \ && mkdir -p -m 755 ${MOUNT_POINT}/dev/pts \ && mount -t devtmpfs -o mode=0755,nosuid devtmpfs ${MOUNT_POINT}/dev \ && mount -t devpts -o gid=5,mode=620 devpts ${MOUNT_POINT}/dev/pts \ && echo_stamp "OK" "SUCCESS" \ || (echo_stamp "Failed" "ERROR"; exit 1) echo_stamp "Copy DNS records" \ && cp -L /etc/resolv.conf ${MOUNT_POINT}/etc/resolv.conf \ && echo_stamp "OK" "SUCCESS" \ || (echo_stamp "Failed" "ERROR"; exit 1) if [[ $# > 1 ]]; then echo_stamp "Copy script into chroot fs" local SCRIPT_NAME=$(basename $2) local SCRIPT_PATH="$(mktemp -d -p ${MOUNT_POINT}/tmp --suffix=.tmp_builder_script)" local script_name=$(basename $2) local script_path_root="${MOUNT_POINT}/root/${script_name}" # TODO: maybe copy to tmp-dir # TODO: Find more suitable location for temporary script storage # $(mktemp -p ${MOUNT_POINT}/tmp --suffix=.tmp_builder_script) cp "$2" "${script_path_root}" # Run script in chroot with additional arguments chroot ${MOUNT_POINT} /bin/sh -c "/root/${script_name} ${@:3}" # Removing script from chroot fs rm "${script_path_root}" else # https://wiki.archlinux.org/index.php/Change_root_(%D0%A0%D1%83%D1%81%D1%81%D0%BA%D0%B8%D0%B9) # http://www.unix-lab.org/posts/chroot/ # https://habrahabr.ru/post/141012/ # https://losst.ru/vosstanovlenie-grub2 # http://unixteam.ru/content/virtualizaciya-ili-zapuskaem-prilozhenie-v-chroot-okruzhenii-razmyshleniya # http://help.ubuntu.ru/wiki/%D0%B2%D0%BE%D1%81%D1%81%D1%82%D0%B0%D0%BD%D0%BE%D0%B2%D0%BB%D0%B5%D0%BD%D0%B8%D0%B5_grub echo_stamp "Entering to chroot" \ && chroot ${MOUNT_POINT} /bin/bash fi umount_system ${MOUNT_POINT} ${DEV_IMAGE} } umount_system() { # TEMPLATE: umount_system echo_stamp "Unmount chroot rootfs and boot partition: $1" umount_ok=false # Repeat 5 times for i in {1..5}; do umount -fR $1 \ && (echo_stamp "OK" "SUCCESS"; umount_ok=true; break) \ || (echo_stamp "Failed #$i (try 5 times)" "ERROR"; sleep 2) done [[ "$umount_ok" == true ]] \ || (echo_stamp "Umount loop-image was failed" "ERROR"; exit 1) losetup -d $2 } copy_to_chroot() { # copy_to_chroot echo_stamp "Mount loop-image: $1" local DEV_IMAGE=$(losetup -Pf $1 --show) sleep 0.5 local MOUNT_POINT=$(mktemp -d --suffix=.builder_image) echo_stamp "Mount dirs ${MOUNT_POINT} & ${MOUNT_POINT}/boot" mount "${DEV_IMAGE}p2" ${MOUNT_POINT} mount "${DEV_IMAGE}p1" ${MOUNT_POINT}/boot local dir_name=$(dirname "${MOUNT_POINT}$3 /") [[ ! -d ${dir_name} ]] && mkdir -p ${dir_name} \ && echo_stamp "Created ${dir_name}" "SUCCESS" cp -r "$2" "${MOUNT_POINT}$3" umount_system ${MOUNT_POINT} ${DEV_IMAGE} } if [ $(whoami) != "root" ]; then echo "" echo "********************************************************************" echo "******************** This should be run as root ********************" echo "********************************************************************" echo "" exit 1 fi if [[ $# > 0 ]]; then echo "================================================================================" for ((i=1; i<=$#; i++)); do echo "\$$i: ${!i}"; done echo "================================================================================" [[ -f $1 ]] || (echo_stamp "$1 does not exist" "ERROR"; exit 1) if [[ -z $2 ]] && [[ -f $3 ]]; then case "$2" in exec) execute $1 $3 ${@:4};; copy) copy_to_chroot $1 $3 $4;; *) echo "Template: image-chroot.sh [ exec