129 lines
3.6 KiB
Bash
Executable File
129 lines
3.6 KiB
Bash
Executable File
#!/bin/sh
|
|
# Ensure memtest binaries are present even if live-build's built-in memtest
|
|
# stage does not copy them. Boot menu entries come from the canonical templates
|
|
# enforced by build.sh after live-build finishes.
|
|
set -e
|
|
|
|
: "${BEE_REQUIRE_MEMTEST:=0}"
|
|
|
|
# Debian Bookworm's pinned memtest86+ package installs these exact paths.
|
|
MEMTEST_FILES="memtest86+x64.bin memtest86+x64.efi"
|
|
BINARY_BOOT_DIR="binary/boot"
|
|
|
|
log() {
|
|
echo "memtest hook: $*"
|
|
}
|
|
|
|
fail_or_warn() {
|
|
msg="$1"
|
|
if [ "${BEE_REQUIRE_MEMTEST}" = "1" ]; then
|
|
log "ERROR: ${msg}"
|
|
exit 1
|
|
fi
|
|
log "WARNING: ${msg}"
|
|
return 0
|
|
}
|
|
|
|
copy_memtest_file() {
|
|
src="$1"
|
|
dst_name="${2:-$(basename "$src")}"
|
|
dst="${BINARY_BOOT_DIR}/${dst_name}"
|
|
|
|
[ -f "$src" ] || return 1
|
|
mkdir -p "${BINARY_BOOT_DIR}"
|
|
cp "$src" "$dst"
|
|
log "copied ${dst_name} from ${src}"
|
|
}
|
|
|
|
extract_memtest_from_deb() {
|
|
deb="$1"
|
|
tmpdir="$(mktemp -d)"
|
|
|
|
log "extracting memtest payload from ${deb}"
|
|
dpkg-deb -x "$deb" "$tmpdir"
|
|
|
|
if [ -f "${tmpdir}/boot/memtest86+x64.efi" ]; then
|
|
copy_memtest_file "${tmpdir}/boot/memtest86+x64.efi"
|
|
fi
|
|
|
|
if [ -f "${tmpdir}/boot/memtest86+x64.bin" ]; then
|
|
copy_memtest_file "${tmpdir}/boot/memtest86+x64.bin"
|
|
fi
|
|
|
|
rm -rf "$tmpdir"
|
|
}
|
|
|
|
download_and_extract_memtest() {
|
|
tmpdl="$(mktemp -d)"
|
|
if [ -n "${MEMTEST_VERSION:-}" ]; then
|
|
pkg_spec="memtest86+=${MEMTEST_VERSION}"
|
|
else
|
|
pkg_spec="memtest86+"
|
|
fi
|
|
log "downloading ${pkg_spec} from apt"
|
|
if ! ( cd "$tmpdl" && apt-get download "$pkg_spec" 2>/dev/null ); then
|
|
log "apt download failed, retrying after apt-get update"
|
|
apt-get update -qq >/dev/null 2>&1 || true
|
|
( cd "$tmpdl" && apt-get download "$pkg_spec" 2>/dev/null ) || true
|
|
fi
|
|
deb="$(find "$tmpdl" -maxdepth 1 -type f -name 'memtest86+*.deb' 2>/dev/null | head -1)"
|
|
if [ -n "$deb" ]; then
|
|
extract_memtest_from_deb "$deb"
|
|
else
|
|
log "apt download of memtest86+ failed"
|
|
fi
|
|
rm -rf "$tmpdl"
|
|
}
|
|
|
|
ensure_memtest_binaries() {
|
|
missing=0
|
|
for f in ${MEMTEST_FILES}; do
|
|
[ -f "${BINARY_BOOT_DIR}/${f}" ] || missing=1
|
|
done
|
|
[ "$missing" -eq 1 ] || return 0
|
|
|
|
# 1. Try files already placed by lb binary_memtest or chroot
|
|
for root in chroot/boot /boot; do
|
|
for f in ${MEMTEST_FILES}; do
|
|
[ -f "${BINARY_BOOT_DIR}/${f}" ] || copy_memtest_file "${root}/${f}" || true
|
|
done
|
|
done
|
|
|
|
missing=0
|
|
for f in ${MEMTEST_FILES}; do
|
|
[ -f "${BINARY_BOOT_DIR}/${f}" ] || missing=1
|
|
done
|
|
[ "$missing" -eq 1 ] || return 0
|
|
|
|
# 2. Try apt package cache (may be empty if lb binary_memtest already purged)
|
|
for root in cache chroot/var/cache/apt/archives /var/cache/apt/archives; do
|
|
[ -d "$root" ] || continue
|
|
deb="$(find "$root" -type f \( -name 'memtest86+_*.deb' -o -name 'memtest86+*.deb' \) 2>/dev/null | head -1)"
|
|
[ -n "$deb" ] || continue
|
|
extract_memtest_from_deb "$deb"
|
|
break
|
|
done
|
|
|
|
missing=0
|
|
for f in ${MEMTEST_FILES}; do
|
|
[ -f "${BINARY_BOOT_DIR}/${f}" ] || missing=1
|
|
done
|
|
[ "$missing" -eq 1 ] || return 0
|
|
|
|
# 3. Fallback: download fresh from apt (lb binary_memtest purges the cache)
|
|
download_and_extract_memtest
|
|
|
|
missing=0
|
|
for f in ${MEMTEST_FILES}; do
|
|
if [ ! -f "${BINARY_BOOT_DIR}/${f}" ]; then
|
|
fail_or_warn "missing ${BINARY_BOOT_DIR}/${f}"
|
|
missing=1
|
|
fi
|
|
done
|
|
[ "$missing" -eq 0 ] || return 0
|
|
}
|
|
|
|
log "ensuring memtest binaries in binary image"
|
|
ensure_memtest_binaries
|
|
log "memtest assets ready"
|