sergio/takana

1#!/bin/sh
2# instalador-initramfs.sh — empaqueta el instalador ENTERO en un initramfs que corre en RAM.
3#
4# ── PARA QUÉ ───────────────────────────────────────────────────────────────────────────────────
5# Es lo que hace posible reemplazar el Linux que está corriendo **sin un USB**: se salta a este
6# initramfs por `kexec` (`takana kernel kexec --jump`, ADR 0017), el sistema viejo muere en el salto
7# y su disco queda libre ⇒ recién entonces su raíz se puede formatear. El mismo initramfs sirve
8# arrancado por QEMU o por un medio cualquiera; el salto es sólo una forma de llegar.
9#
10# ── POR QUÉ EL PAYLOAD VIAJA ADENTRO Y NO EN EL DISCO ──────────────────────────────────────────
11# Porque el disco es justamente lo que se va a formatear. Dejar el rootfs en la partición destino y
12# formatearla después es morderse la cola; dejarlo en otra partición ata la instalación a que exista
13# esa otra partición. En RAM no depende de nada: el precio es que el initramfs pesa lo que pese el
14# rootfs, y por eso el script COMPRUEBA que entre en la memoria de la máquina antes de armarlo.
15#
16# Uso:
17#   ROOTFS=<product-rootfs> scripts/instalador-initramfs.sh --root-part /dev/nvme0n1p2 --esp /dev/nvme0n1p1
18#   # o sin destino fijo: lo decide el instalador al arrancar (--use-free-space)
19#
20# Salida: work/takana-instalador.cpio.gz  (+ el `takana kernel kexec` que lo salta, impreso al final)
21set -eu
22ROOT="$(cd "$(dirname "$0")/.." && pwd)"; cd "$ROOT"
23
24ARGS_INSTALADOR=""
25while [ $# -gt 0 ]; do ARGS_INSTALADOR="$ARGS_INSTALADOR $1"; shift; done
26[ -n "$ARGS_INSTALADOR" ] || { echo "uso: $0 <argumentos que se le pasarán a takana-install-partition.sh>" >&2; exit 2; }
27
28OUT="${OUT:-work/takana-instalador.cpio.gz}"
29STAGE="${STAGE:-work/.instalador-initramfs-stage}"
30IRD_TREE="${IRD_TREE:-work/.instalador-initramfs-tree}"
31KERNEL="${KERNEL:-$(ls -dt store/*-linux-metal/boot/bzImage /store/*-linux-metal/boot/bzImage 2>/dev/null | head -1)}"
32[ -n "${KERNEL:-}" ] && [ -r "$KERNEL" ] || { echo "no encuentro un bzImage de linux-metal (set KERNEL=)" >&2; exit 1; }
33
34. "$(dirname "$0")/lib/herramientas-desde-store.sh"
35MKE2FS=$(herramienta e2fsprogs sbin/mke2fs mke2fs)  || exit 1
36SFDISK=$(herramienta util-linux sbin/sfdisk sfdisk) || exit 1
37MMD=$(herramienta    mtools usr/bin/mmd   mmd)      || exit 1
38MCOPY=$(herramienta  mtools usr/bin/mcopy mcopy)    || exit 1
39MDIR=$(herramienta   mtools usr/bin/mdir  mdir)     || exit 1
40CPIO=$(herramienta   libarchive usr/bin/bsdcpio cpio) || exit 1
41GZIP=$(herramienta   gzip bin/gzip gzip)            || exit 1
42
43# ── 1) el stage (rootfs a instalar + initramfs de pivote) lo arma el script validado ───────────
44echo "==> stage del sistema a instalar"
45STAGE_ONLY=1 KEEP_STORE=1 STAGE="$STAGE" ./scripts/install-image-efi.sh
46[ -d "$STAGE/root" ] && [ -r "$STAGE/initramfs.cpio.gz" ] || { echo "el stage no quedó completo" >&2; exit 1; }
47
48# ── 2) ¿entra en la RAM de esta máquina? ───────────────────────────────────────────────────────
49# Un initramfs vive ENTERO en memoria, y el kernel no lo avisa con elegancia: se queda sin RAM al
50# desempaquetar y el arranque muere sin un mensaje que diga por qué. Medirlo antes es la diferencia
51# entre «no entra, usá el USB» y un cuelgue mudo en la máquina del usuario.
52_mib_payload=$(du -sm "$STAGE/root" | cut -f1)
53_mib_ram=$(( $(awk '/^MemTotal:/ {print $2}' /proc/meminfo) / 1024 ))
54echo "    payload ${_mib_payload} MiB · RAM de esta máquina ${_mib_ram} MiB"
55if [ "$(( _mib_payload * 100 / _mib_ram ))" -gt 40 ]; then
56    echo "instalador-initramfs: el payload es más del 40 % de la RAM ($_mib_payload de $_mib_ram MiB)." >&2
57    echo "                      En RAM van el initramfs comprimido Y su contenido desempaquetado." >&2
58    echo "                      Este camino no sirve para este rootfs: instalá desde un medio." >&2
59    exit 1
60fi
61
62# ── 3) el árbol del initramfs ──────────────────────────────────────────────────────────────────
63echo "==> árbol del initramfs-instalador"
64rm -rf "$IRD_TREE"; mkdir -p "$IRD_TREE/bin" "$IRD_TREE/dev" "$IRD_TREE/proc" "$IRD_TREE/sys" \
65    "$IRD_TREE/instalador/lib" "$IRD_TREE/payload"
66BUSYBOX="${BUSYBOX:-$STAGE/root/bin/busybox}"
67[ -r "$BUSYBOX" ] || { echo "no encuentro busybox estático en $BUSYBOX" >&2; exit 1; }
68cp "$BUSYBOX" "$IRD_TREE/bin/busybox"; chmod 0755 "$IRD_TREE/bin/busybox"
69for a in sh mount umount mkdir echo sleep ls cat sed awk grep cut tr basename dirname readlink \
70         find sync reboot dmesg blkid setsid tee du wc head tail sort printf test sha256sum; do
71    ln -sf busybox "$IRD_TREE/bin/$a"
72done
73# Las herramientas de disco NO salen de busybox: son las SELLADAS del corpus, estáticas musl. Dentro
74# del initramfs no hay store del que resolverlas por hash, así que viajan copiadas y el instalador
75# las recibe por `TOOL_<binario>` — el override que existe exactamente para esto.
76for t in "$MKE2FS" "$SFDISK" "$MMD" "$MCOPY" "$MDIR"; do
77    cp "$t" "$IRD_TREE/bin/$(basename "$t")"; chmod 0755 "$IRD_TREE/bin/$(basename "$t")"
78done
79TAKANA_MUSL="${TAKANA_MUSL:-$ROOT/target/x86_64-unknown-linux-musl/release/takana}"
80[ -x "$TAKANA_MUSL" ] || TAKANA_MUSL="$ROOT/target/release/takana"
81cp "$TAKANA_MUSL" "$IRD_TREE/bin/takana"; chmod 0755 "$IRD_TREE/bin/takana"
82cp scripts/takana-install-partition.sh "$IRD_TREE/instalador/"
83cp scripts/lib/herramientas-desde-store.sh "$IRD_TREE/instalador/lib/"
84cp "$KERNEL" "$IRD_TREE/payload/bzImage"
85mkdir -p "$IRD_TREE/payload/stage"
86cp -a "$STAGE/root" "$IRD_TREE/payload/stage/root"
87cp "$STAGE/initramfs.cpio.gz" "$IRD_TREE/payload/stage/initramfs.cpio.gz"
88printf '%s\n' "$ARGS_INSTALADOR" > "$IRD_TREE/instalador/args"
89
90# ── 4) /init — lo que corre cuando el kernel nuevo toma el control ─────────────────────────────
91cat > "$IRD_TREE/init" <<'INIT'
92#!/bin/busybox sh
93# PID 1 del instalador. No pivota a ningún disco: instala y reinicia.
94export PATH=/bin
95mount -t proc     proc /proc 2>/dev/null || true
96mount -t sysfs    sys  /sys  2>/dev/null || true
97mount -t devtmpfs dev  /dev  2>/dev/null || true
98# ⚠ El log va a tty0 **y** al serial, por lo mismo que el initramfs de pivote: el cmdline horneado
99# deja /dev/console en ttyS0 y en un metal sin puerto serie todo esto sería invisible.
100log() { echo "$*"; for d in /dev/tty0 /dev/ttyS0; do echo "$*" > "$d" 2>/dev/null || true; done; }
101rescate() {
102    log "TAKANA-INSTALADOR-FALLO: $*"
103    log "  bloques visibles: $(ls /dev/sd* /dev/nvme* /dev/vd* /dev/mmcblk* 2>/dev/null | tr '\n' ' ')"
104    log "  shell de rescate en tty1 — probá: blkid ; dmesg | tail -30"
105    setsid -c sh -c 'exec /bin/sh </dev/tty1 >/dev/tty1 2>&1' >/dev/null 2>&1 &
106    exec sh
107}
108log "TAKANA-INSTALADOR: arrancado en RAM. El sistema viejo YA NO CORRE: su disco está libre."
109
110# Los discos tardan en enumerar (USB y NVMe sobre todo). Misma espera larga que el pivote, y por el
111# mismo motivo: en QEMU aparecen al instante y la carrera sólo se ve en metal.
112i=0
113while [ "$i" -lt 60 ]; do
114    ls /dev/nvme* /dev/sd* /dev/vd* /dev/mmcblk* >/dev/null 2>&1 && break
115    i=$(( i + 1 )); [ $(( i % 5 )) -eq 0 ] && log "TAKANA-INSTALADOR: esperando discos… ${i}s"
116    sleep 1
117done
118
119cd /instalador
120ARGS=$(cat /instalador/args)
121log "TAKANA-INSTALADOR: takana-install-partition.sh $ARGS"
122# Las herramientas viajan en /bin: el instalador las toma por TOOL_<binario> en vez de buscarlas en
123# un store que acá no existe. STAGE_LISTO dice que el árbol ya vino armado.
124export TOOL_mke2fs=/bin/mke2fs TOOL_sfdisk=/bin/sfdisk TOOL_mmd=/bin/mmd \
125       TOOL_mcopy=/bin/mcopy TOOL_mdir=/bin/mdir
126export STAGE=/payload/stage STAGE_LISTO=1 KERNEL=/payload/bzImage HAMMER=/bin/takana
127# shellcheck disable=SC2086
128if sh /instalador/takana-install-partition.sh $ARGS --yes 2>&1 | tee /dev/tty0 > /dev/ttyS0; then
129    log "TAKANA-INSTALADOR-OK: instalado. Reiniciando en 10 s (sacá el medio si lo hubiera)."
130    sync; sleep 10; reboot -f
131else
132    rescate "el instalador salió con error — NADA se reinicia solo, mirá el log de arriba"
133fi
134INIT
135chmod +x "$IRD_TREE/init"
136
137# ── 5) empaquetar ──────────────────────────────────────────────────────────────────────────────
138mkdir -p "$(dirname "$OUT")"
139( cd "$IRD_TREE" && find . -print0 | "$CPIO" --null -o -H newc --owner=root:root 2>/dev/null | "$GZIP" -1 ) > "$OUT"
140echo "==> $OUT: $(du -h "$OUT" | cut -f1)"
141echo
142echo "Para SALTAR a él sin pasar por el firmware (el userspace de esta máquina muere ACÁ):"
143echo "  sudo takana kernel kexec --probe          # ¿se puede? exit 0 = sí"
144echo "  sudo takana kernel kexec <linux-metal> --initrd $OUT \\"
145echo "       --cmdline 'console=tty0 console=ttyS0,115200' --jump"
146echo
147echo "Para probarlo SIN saltar (y sin riesgo), en QEMU:"
148echo "  qemu-system-x86_64 -m 4096 -kernel $KERNEL -initrd $OUT \\"
149echo "       -append 'console=ttyS0' -drive file=<disco.img>,format=raw -nographic"