tinyramfs/generate

454 lines
12 KiB
Plaintext
Raw Normal View History

2020-01-06 00:20:12 +05:30
#!/bin/sh
2020-01-05 23:31:39 +05:30
#
# tiny initramfs generation tool
2020-02-06 22:05:49 +05:30
# print message
msg() {
case "$1" in
2020-02-11 23:32:23 +05:30
info)
printf "info >> %s\n" "$2" >&2
;;
warn)
printf "warning >> %s\n" "$2" >&2
printf "do you want to continue? press enter or ctrl+c to exit\n"
read -r _
;;
panic)
printf "panic >> %s\n" "$2" >&2
exit 1
;;
2020-02-06 22:05:49 +05:30
esac
2020-01-30 19:23:17 +05:30
}
2020-01-05 23:31:39 +05:30
2020-02-11 04:18:51 +05:30
# create tmpdir
create_tmpdir() {
2020-02-13 07:48:53 +05:30
msg info "creating working directory"
2020-02-11 04:18:51 +05:30
if [ -n "$XDG_CACHE_HOME" ]; then
tmpdir="${XDG_CACHE_HOME}/initramfs.$$"
elif [ -n "$TMPDIR" ]; then
tmpdir="${TMPDIR}/initramfs.$$"
else
tmpdir="/tmp/initramfs.$$"
fi
2020-02-13 06:06:13 +05:30
mkdir "$tmpdir" || msg panic "failed to create working directory"
2020-02-11 04:18:51 +05:30
}
2020-02-02 17:49:20 +05:30
# remove tmpdir
2020-01-30 19:23:17 +05:30
remove_tmpdir() {
2020-02-13 08:03:18 +05:30
msg info "removing working directory"
2020-01-30 19:23:17 +05:30
rm -rf "$tmpdir"
}
2020-02-02 17:49:20 +05:30
# change current directory to script directory if user haven't do it
check_currentdir() {
2020-02-12 07:35:53 +05:30
script_dir=$(dirname "$0")
2020-02-02 17:49:20 +05:30
[ "$PWD" = "$script_dir" ] || {
2020-02-13 08:03:18 +05:30
msg info "changing current directory to script directory"
2020-02-11 23:32:23 +05:30
cd "$script_dir" || msg panic "failed to change directory"
2020-02-02 17:49:20 +05:30
}
}
# check needed files
2020-01-30 19:23:17 +05:30
check_requirements() {
2020-02-07 18:16:34 +05:30
msg info "checking requirements"
2020-01-30 19:23:17 +05:30
2020-02-09 06:09:08 +05:30
# check busybox installed
2020-02-12 07:14:41 +05:30
if command -v busybox > /dev/null 2>&1; then
2020-02-11 23:32:23 +05:30
# check busybox supports CONFIG_FEATURE_INSTALLER
busybox --help | grep -q "\-\-install \[\-s\]" || msg panic "recompile busybox with CONFIG_FEATURE_INSTALLER"
# check requirements for init
for busybox_dep in mdev uevent switch_root; do
busybox $busybox_dep --help > /dev/null 2>&1 || msg panic "recompile busybox with $busybox_dep"
done
2020-02-12 07:14:41 +05:30
else
msg panic "busybox doesn't installed"
fi
2020-02-09 06:09:08 +05:30
# check kmod modprobe installed
2020-02-12 07:14:41 +05:30
if command -v modprobe > /dev/null 2>&1; then
2020-02-11 23:32:23 +05:30
# busybox modprobe doesn't supported(yet)
modprobe --version 2>&1 | grep -q "kmod" || msg panic "kmod modprobe version doesn't installed"
2020-02-12 07:14:41 +05:30
else
msg panic "modprobe doesn't installed"
fi
2020-02-09 06:09:08 +05:30
# check util-linux tools
[ "$use_util_linux" = 1 ] && {
2020-02-11 23:32:23 +05:30
# check mount installed
if command -v mount > /dev/null 2>&1; then
mount --version 2>&1 | grep -q "util-linux" || {
msg warning "util-linux mount version doesn't installed. PARTUUID and filesystem type autodetection support will be missing"
use_util_linux=0
}
else
msg panic "mount doesn't installed"
fi
# check blkid installed
if command -v blkid > /dev/null 2>&1; then
blkid --version 2>&1 | grep -q "util-linux" || {
msg warning "util-linux blkid version doesn't installed. PARTUUID support will be missing"
use_util_linux=0
}
else
msg panic "blkid doesn't installed"
fi
}
2020-02-09 06:09:08 +05:30
}
# install requirements
2020-02-09 06:09:08 +05:30
install_requirements() {
msg info "installing requirements"
# install user specified binaries
[ -n "$binaries" ] && install_binaries $binaries
# install util-linux binaries
[ "$use_util_linux" = 1 ] && install_binaries mount blkid
# install mandatory binaries
install_binaries busybox modprobe
2020-01-30 19:23:17 +05:30
}
2020-02-02 17:49:20 +05:30
# create FHS directory structure
2020-01-30 19:23:17 +05:30
create_structure() {
2020-02-07 18:16:34 +05:30
msg info "creating directory structure"
2020-02-09 06:09:08 +05:30
for dir in dev tmp var run etc usr/lib usr/bin mnt/root proc root sys; do
2020-02-11 23:32:23 +05:30
mkdir -p "${tmpdir}/${dir}"
2020-01-30 19:23:17 +05:30
done
}
2020-02-02 17:49:20 +05:30
# some dynamically linked libraries and binaries compiled with hardcoded
# dependencies path. to make it worked we need create symlinks for them.
# also POSIX ln doesn't have --relative flag like in GNU ln. as workaround
# we change directory to tmpdir and make needed symlinks.
2020-01-30 19:23:17 +05:30
create_symlinks() {
2020-02-07 18:16:34 +05:30
msg info "creating symlinks"
2020-01-30 19:23:17 +05:30
( cd "$tmpdir" && {
2020-02-11 23:32:23 +05:30
ln -s usr/lib lib
ln -s usr/lib lib64
ln -s usr/bin bin
ln -s usr/bin sbin
ln -s ../run var/run
cd "${tmpdir}/usr"
ln -s bin sbin
ln -s lib lib64
2020-01-30 19:23:17 +05:30
} )
}
#parse_fstab() {
# TODO parse fstab
2020-01-05 23:31:39 +05:30
#while [ "$use_fstab" -eq 1 ] && read fs dir type opts; do thing; done < /etc/fstab
2020-01-30 19:23:17 +05:30
#}
#parse_crypttab() {
# TODO parse crypttab
#}
2020-01-05 23:31:39 +05:30
2020-02-02 17:49:20 +05:30
# install mdev
2020-02-02 18:40:43 +05:30
install_mdev() {
2020-02-07 18:16:34 +05:30
msg info "installing mdev"
2020-02-02 18:30:23 +05:30
install -m644 mdev.conf -t "${tmpdir}/etc"
install -Dm755 storage-device -t "${tmpdir}/lib/mdev"
2020-01-30 19:23:17 +05:30
}
2020-01-19 02:31:21 +05:30
2020-02-02 17:49:20 +05:30
# install mdevd
2020-02-02 18:40:43 +05:30
install_mdevd() {
2020-02-07 18:16:34 +05:30
msg info "installing mdevd"
2020-01-30 19:23:17 +05:30
install_binaries mdevd mdevd-coldplug
2020-02-02 18:30:23 +05:30
install -m644 mdev.conf -t "${tmpdir}/etc"
install -Dm755 storage-device -t "${tmpdir}/lib/mdev"
2020-01-30 19:23:17 +05:30
}
2020-01-25 16:57:02 +05:30
2020-02-02 17:49:20 +05:30
# install udev
2020-02-02 18:40:43 +05:30
install_udev() {
2020-02-07 18:16:34 +05:30
msg info "installing udev"
2020-01-30 19:23:17 +05:30
install_binaries udevd udevadm dmsetup
# FIXME rewrite this piece of crap
2020-02-11 23:32:23 +05:30
find /usr/lib/udev -type f | grep -v "rc_keymaps\|hwdb.d" | cpio -pd "$tmpdir" > /dev/null 2>&1
2020-01-30 19:23:17 +05:30
}
2020-01-25 16:57:02 +05:30
2020-01-28 20:43:42 +05:30
# handle lvm
2020-02-02 18:40:43 +05:30
install_lvm() {
2020-02-07 18:16:34 +05:30
msg info "installing LVM"
2020-01-30 19:23:17 +05:30
install_binaries lvm
2020-02-06 21:51:00 +05:30
# if hostonly mode enabled install only needed drivers
2020-02-12 00:17:36 +05:30
[ "$hostonly" = 1 ] && install_drivers dm-thin-pool dm-multipath dm-snapshot dm-cache dm-log dm-mirror
2020-02-06 21:51:00 +05:30
2020-02-09 01:35:03 +05:30
if [ "$lvm_conf" = 1 ]; then
2020-02-11 23:32:23 +05:30
install -Dm644 /etc/lvm/*.conf -t "${tmpdir}/etc/lvm" || msg panic "failed to install LVM config"
2020-02-09 01:35:03 +05:30
else
2020-02-11 23:32:23 +05:30
mkdir "${tmpdir}/etc/lvm"
cat << EOF > "${tmpdir}/etc/lvm/lvm.conf"
devices {
2020-02-09 01:35:03 +05:30
issue_discards = ${lvm_discard:-0}
}
global {
2020-02-09 01:35:03 +05:30
use_lvmetad = 0
}
EOF
fi
2020-01-30 19:23:17 +05:30
}
2020-01-19 02:31:21 +05:30
2020-01-25 16:57:02 +05:30
# handle luks
2020-02-02 18:40:43 +05:30
install_luks() {
2020-02-07 18:16:34 +05:30
msg info "installing LUKS"
2020-01-30 19:23:17 +05:30
install_binaries cryptsetup
2020-01-25 16:57:02 +05:30
2020-02-06 21:51:00 +05:30
# if hostonly mode enabled install only needed drivers
2020-02-12 00:17:36 +05:30
[ "$hostonly" = 1 ] && install_drivers aes dm-crypt sha256 sha512 wp512 ecb lrw xts twofish serpent
2020-02-06 21:51:00 +05:30
2020-01-25 16:57:02 +05:30
# avoid locking directory missing warning message
2020-01-30 19:23:17 +05:30
mkdir "${tmpdir}/run/cryptsetup"
2020-01-25 16:57:02 +05:30
# TODO get rid of this workaround
# workaround for luks2
2020-02-07 18:16:34 +05:30
install -s -m755 /usr/lib/libgcc_s.so.1 -t "${tmpdir}/usr/lib" || msg panic "failed to install LUKS libraries"
2020-01-25 16:57:02 +05:30
2020-02-06 03:59:59 +05:30
# block discard support
2020-01-28 20:43:42 +05:30
[ "$luks_discard" = 1 ] && luks_args="--allow-discards $luks_args"
2020-01-25 16:57:02 +05:30
2020-02-06 03:59:59 +05:30
# copy luks header
[ -f "$luks_header" ] && {
2020-02-11 23:32:23 +05:30
install -m400 "$luks_header" "${tmpdir}/root/luks_header" || msg panic "failed to copy LUKS header"
luks_args="--header=/root/luks_header $luks_args"
2020-02-06 03:59:59 +05:30
}
# copy luks keyfile
[ -f "$luks_keyfile" ] && {
2020-02-11 23:32:23 +05:30
install -m400 "$luks_keyfile" "${tmpdir}/root/luks_keyfile" || msg panic "failed to copy LUKS keyfile"
luks_args="--key-file=/root/luks_keyfile $luks_args"
2020-02-06 03:59:59 +05:30
}
2020-01-30 19:23:17 +05:30
}
2020-01-19 02:31:21 +05:30
2020-02-12 00:17:36 +05:30
# install drivers and deps
2020-01-30 19:23:17 +05:30
install_drivers() {
2020-02-12 00:17:36 +05:30
printf "%s\n" "$@" | while read -r driver; do
2020-02-12 06:48:02 +05:30
# strip path and extension
driver="${driver##*/}"
driver="${driver%%.*}"
2020-02-11 23:32:23 +05:30
for driver_dep in $(modprobe -D "$driver" 2> /dev/null | grep -v builtin | cut -d " " -f 2); do
install -Dm644 "$driver_dep" "${tmpdir}${driver_dep}"
2020-02-06 21:51:00 +05:30
done
done
2020-02-12 01:02:25 +05:30
# TODO implement monolithic kernel support
[ -e "${tmpdir}/lib/modules" ] || msg panic "failed to install drivers"
2020-02-12 00:17:36 +05:30
}
# install hostonly drivers
install_hostonly_drivers() {
msg info "installing hostonly drivers"
[ -n "$root_type" ] || msg panic "hostonly mode required root_type option to be configured"
# perform autodetection of drivers via /sys
2020-02-12 06:48:02 +05:30
install_drivers $(find /sys -name modalias -exec sort -u "{}" "+")
2020-02-05 19:28:58 +05:30
2020-02-06 21:51:00 +05:30
# TODO autodetect root fs driver
# TODO separate root type option
# install root fs driver
2020-02-12 00:17:36 +05:30
install_drivers "$root_type"
2020-02-06 21:51:00 +05:30
# install user specified drivers
2020-02-12 00:17:36 +05:30
[ -n "$drivers" ] && install_drivers "$drivers"
2020-02-06 21:51:00 +05:30
}
# find and install all drivers
install_all_drivers() {
2020-02-07 18:16:34 +05:30
msg info "installing all drivers"
2020-02-12 06:48:02 +05:30
modker="${moddir}/${kernel}/kernel"
install_drivers \
$(find \
"${modker}/arch" \
"${modker}/crypto" \
"${modker}/fs" \
"${modker}/lib" \
"${modker}/drivers/block" \
"${modker}/drivers/ata" \
"${modker}/drivers/md" \
"${modker}/drivers/scsi" \
"${modker}/drivers/usb/storage" \
"${modker}/drivers/usb/host" \
"${modker}/drivers/virtio" \
-type f 2> /dev/null)
2020-02-06 21:51:00 +05:30
}
2020-01-30 19:23:17 +05:30
2020-02-06 21:51:00 +05:30
# generate "modules" files
generate_depmod() {
2020-02-13 07:48:53 +05:30
msg info "running depmod"
modker="${moddir}/${kernel}"
2020-01-30 19:23:17 +05:30
# install list of drivers
cp "${modker}/modules.softdep" "${modker}/modules.builtin" "${modker}/modules.order" "${tmpdir}/${modker}"
# generate dependencies list of drivers
depmod -b "$tmpdir" "$kernel"
}
# TODO make strip optional
# handle binaries
install_binaries() {
2020-02-09 06:09:08 +05:30
printf "%s\n" "$@" | while read -r binary; do
2020-02-11 23:32:23 +05:30
msg info "installing binary $binary"
# check binary existence
command -v "$binary" > /dev/null 2>&1 || msg panic "$binary doesn't exists"
2020-01-30 19:23:17 +05:30
2020-02-11 23:32:23 +05:30
# install and strip binary
install -s -m755 "$(command -v $binary)" -t "${tmpdir}/usr/bin"
2020-01-30 19:23:17 +05:30
# check statically linking
2020-02-11 23:32:23 +05:30
ldd "$(command -v $binary)" > /dev/null 2>&1 || continue
2020-01-30 19:23:17 +05:30
2020-02-11 23:32:23 +05:30
# install libraries
install_libraries $binary
2020-01-30 19:23:17 +05:30
done
}
2020-01-05 23:31:39 +05:30
2020-01-13 07:52:45 +05:30
# TODO make strip optional
2020-01-30 19:23:17 +05:30
# handle libraries
install_libraries() {
2020-02-09 06:09:08 +05:30
for library in $(ldd "$(command -v $1)" | sed -nre 's,.* (/.*lib.*/.*.so.*) .*,\1,p' -e 's,.*(/lib.*/ld.*.so.*) .*,\1,p'); do
2020-02-11 23:32:23 +05:30
# check symlink
if [ -h "$library" ]; then
# check lib already existence
if [ ! -e "${tmpdir}/usr/lib/${library##*/}" ] && [ ! -e "${tmpdir}/$(readlink -f $library)" ]; then
# regular
install -s -m755 "$(readlink -f $library)" -t "${tmpdir}/usr/lib"
# FIXME handle all symlinks
# symlink may link to symlink
[ -h "/usr/lib/$(readlink $library)" ] && cp -a "/usr/lib/$(readlink $library)" "${tmpdir}/usr/lib"
# symlink
cp -a "$library" "${tmpdir}/usr/lib"
fi
else
if [ ! -e "${tmpdir}/usr/lib/${library##*/}" ]; then
install -s -m755 "$library" -t "${tmpdir}/usr/lib"
fi
fi
2020-01-07 03:09:15 +05:30
done
2020-01-30 19:23:17 +05:30
}
2020-01-05 23:31:39 +05:30
2020-02-07 18:16:34 +05:30
# install important files used by init
2020-01-30 19:23:17 +05:30
install_files() {
2020-02-07 18:16:34 +05:30
msg info "installing files"
2020-02-06 21:51:00 +05:30
# FIXME eof broken
# initialize config
2020-02-11 23:32:23 +05:30
cat << EOF > "${tmpdir}/config"
2020-01-30 20:17:23 +05:30
debug="$debug"
2020-01-05 23:31:39 +05:30
root="$root"
2020-01-25 16:57:02 +05:30
root_type="$root_type"
root_args="$root_args"
2020-01-30 19:23:17 +05:30
devmgr="$devmgr"
2020-01-25 16:57:02 +05:30
#drivers="$drivers"
use_lvm="$use_lvm"
use_luks="$use_luks"
luks_root="$luks_root"
luks_args="$luks_args"
2020-01-05 23:31:39 +05:30
EOF
2020-02-06 21:51:00 +05:30
# needed for devmgr
2020-02-11 23:32:23 +05:30
cat << EOF > "${tmpdir}/etc/group"
2020-01-30 19:23:17 +05:30
root:x:0:
tty:x:5:
dialout:x:11:
uucp:x:14:
kmem:x:3:
input:x:25:
video:x:13:
audio:x:12:
lp:x:10:
disk:x:9:
cdrom:x:16:
tape:x:6:
kvm:x:24:
floppy:x:8:
EOF
2020-02-06 21:51:00 +05:30
# needed for devmgr
2020-02-11 23:32:23 +05:30
cat << EOF > "${tmpdir}/etc/passwd"
2020-01-30 19:23:17 +05:30
root:x:0:0::/root:/bin/sh
nobody:x:99:99::/:/bin/false
EOF
# install init script
install -m755 ./init -t "$tmpdir"
}
# TODO add more compession tools
2020-01-30 21:04:33 +05:30
# create and compress cpio archive
2020-01-30 19:23:17 +05:30
create_initramfs() {
2020-02-07 18:16:34 +05:30
msg info "creating initramfs image"
2020-01-30 19:23:17 +05:30
{
2020-02-11 23:32:23 +05:30
(
cd "$tmpdir"
find . \
| cpio -oH newc \
| gzip -9
) \
| tee "${script_dir}/initramfs-${kernel}.img.gz"
} > /dev/null 2>&1 || msg panic "failed to generate initramfs image"
2020-01-30 19:23:17 +05:30
}
2020-02-02 17:49:20 +05:30
# check root
2020-02-06 22:05:49 +05:30
[ "$(id -u)" = 0 ] || msg panic "must be run as root"
2020-01-30 19:23:17 +05:30
2020-02-13 06:06:13 +05:30
create_tmpdir
2020-02-05 19:05:17 +05:30
# remove tmpdir on exit or unexpected error
trap remove_tmpdir EXIT INT
2020-02-02 17:49:20 +05:30
check_currentdir
2020-01-30 19:23:17 +05:30
2020-02-02 17:49:20 +05:30
# source config
2020-02-13 07:48:53 +05:30
. ./config || msg panic "failed to source config"
2020-01-30 20:58:03 +05:30
# handle debug mode
[ "$debug" = 1 ] && {
2020-01-30 21:04:33 +05:30
# debug shell commands
2020-01-30 20:58:03 +05:30
set -x
2020-01-30 21:04:33 +05:30
# don't remove anything
2020-01-30 20:58:03 +05:30
trap : EXIT INT
}
2020-02-02 17:49:20 +05:30
# variables
2020-02-12 02:00:20 +05:30
kernel="${kernel:-$(uname -r)}"
2020-02-11 04:18:51 +05:30
moddir="/lib/modules"
2020-02-02 17:49:20 +05:30
check_requirements
2020-01-30 19:23:17 +05:30
create_structure
create_symlinks
#parse_fstab
#parse_crypttab
2020-02-09 06:09:08 +05:30
install_requirements
2020-02-06 21:51:00 +05:30
if [ "$hostonly" = 1 ]; then
2020-02-12 00:17:36 +05:30
install_hostonly_drivers
2020-02-06 21:51:00 +05:30
else
install_all_drivers
fi
generate_depmod
2020-01-30 19:23:17 +05:30
2020-02-02 17:49:20 +05:30
# handle device manager
2020-01-30 19:23:17 +05:30
case "$devmgr" in
2020-02-02 18:40:43 +05:30
mdev) install_mdev ;;
mdevd) install_mdevd ;;
udev) install_udev ;;
2020-02-06 22:05:49 +05:30
*) msg panic "devmgr option broken" ;;
2020-01-30 19:23:17 +05:30
esac
2020-02-02 18:40:43 +05:30
[ "$use_luks" = 1 ] && install_luks
[ "$use_lvm" = 1 ] && install_lvm
2020-01-05 23:31:39 +05:30
2020-02-06 19:45:29 +05:30
install_files
2020-01-30 19:23:17 +05:30
create_initramfs
2020-01-05 23:31:39 +05:30
2020-02-06 22:05:49 +05:30
msg info "done! check out initramfs-${kernel}.img.gz"