torch-pi-gen/scripts/common

101 lines
2.4 KiB
Plaintext
Raw Normal View History

2016-04-11 06:21:07 +00:00
log (){
2018-03-02 20:08:24 +00:00
date +"[%T] $*" | tee -a "${LOG_FILE}"
2016-04-11 06:21:07 +00:00
}
2016-05-04 14:51:41 +00:00
export -f log
2016-04-11 06:21:07 +00:00
bootstrap(){
local BOOTSTRAP_CMD=debootstrap
local BOOTSTRAP_ARGS=()
2016-04-11 06:21:07 +00:00
export http_proxy=${APT_PROXY}
if [ "$(dpkg --print-architecture)" != "armhf" ] && [ "$(dpkg --print-architecture)" != "aarch64" ]; then
BOOTSTRAP_CMD=qemu-debootstrap
2016-04-11 06:21:07 +00:00
fi
BOOTSTRAP_ARGS+=(--arch armhf)
BOOTSTRAP_ARGS+=(--components "main,contrib,non-free")
BOOTSTRAP_ARGS+=(--keyring "${STAGE_DIR}/files/raspberrypi.gpg")
BOOTSTRAP_ARGS+=("$@")
printf -v BOOTSTRAP_STR '%q ' "${BOOTSTRAP_ARGS[@]}"
setarch linux32 capsh --drop=cap_setfcap -- -c "'${BOOTSTRAP_CMD}' $BOOTSTRAP_STR" || true
if [ -d "$2/debootstrap" ]; then
rmdir "$2/debootstrap"
fi
2016-04-11 06:21:07 +00:00
}
2016-05-04 14:51:41 +00:00
export -f bootstrap
2016-04-11 06:21:07 +00:00
copy_previous(){
if [ ! -d "${PREV_ROOTFS_DIR}" ]; then
2016-04-11 06:21:07 +00:00
echo "Previous stage rootfs not found"
false
fi
mkdir -p "${ROOTFS_DIR}"
2017-03-30 16:51:23 +00:00
rsync -aHAXx --exclude var/cache/apt/archives "${PREV_ROOTFS_DIR}/" "${ROOTFS_DIR}/"
2016-04-11 06:21:07 +00:00
}
2016-05-04 14:51:41 +00:00
export -f copy_previous
2016-04-11 06:21:07 +00:00
unmount(){
if [ -z "$1" ]; then
DIR=$PWD
else
DIR=$1
fi
while mount | grep -q "$DIR"; do
local LOCS
LOCS=$(mount | grep "$DIR" | cut -f 3 -d ' ' | sort -r)
2016-04-11 06:21:07 +00:00
for loc in $LOCS; do
umount "$loc"
2016-04-11 06:21:07 +00:00
done
done
}
2016-05-04 14:51:41 +00:00
export -f unmount
unmount_image(){
sync
sleep 1
local LOOP_DEVICES
LOOP_DEVICES=$(losetup --list | grep "$(basename "${1}")" | cut -f1 -d' ')
for LOOP_DEV in ${LOOP_DEVICES}; do
if [ -n "${LOOP_DEV}" ]; then
local MOUNTED_DIR
MOUNTED_DIR=$(mount | grep "$(basename "${LOOP_DEV}")" | head -n 1 | cut -f 3 -d ' ')
if [ -n "${MOUNTED_DIR}" ] && [ "${MOUNTED_DIR}" != "/" ]; then
unmount "$(dirname "${MOUNTED_DIR}")"
fi
sleep 1
losetup -d "${LOOP_DEV}"
2016-05-27 10:54:56 +00:00
fi
done
2016-05-04 14:51:41 +00:00
}
export -f unmount_image
2016-04-11 06:21:07 +00:00
on_chroot() {
if ! mount | grep -q "$(realpath "${ROOTFS_DIR}"/proc)"; then
mount -t proc proc "${ROOTFS_DIR}/proc"
2016-04-11 06:21:07 +00:00
fi
if ! mount | grep -q "$(realpath "${ROOTFS_DIR}"/dev)"; then
mount --bind /dev "${ROOTFS_DIR}/dev"
2016-04-11 06:21:07 +00:00
fi
if ! mount | grep -q "$(realpath "${ROOTFS_DIR}"/dev/pts)"; then
mount --bind /dev/pts "${ROOTFS_DIR}/dev/pts"
fi
2016-04-11 06:21:07 +00:00
if ! mount | grep -q "$(realpath "${ROOTFS_DIR}"/sys)"; then
mount --bind /sys "${ROOTFS_DIR}/sys"
2016-04-11 06:21:07 +00:00
fi
setarch linux32 capsh --drop=cap_setfcap "--chroot=${ROOTFS_DIR}/" -- -e "$@"
2016-04-11 06:21:07 +00:00
}
2016-05-04 14:51:41 +00:00
export -f on_chroot
2016-04-11 06:21:07 +00:00
update_issue() {
echo -e "Raspberry Pi reference ${IMG_DATE}\nGenerated using ${PI_GEN}, ${PI_GEN_REPO}, ${GIT_HASH}, ${1}" > "${ROOTFS_DIR}/etc/rpi-issue"
2016-04-11 06:21:07 +00:00
}
2016-05-04 14:51:41 +00:00
export -f update_issue