{ config , pkgs , lib , ... }: let inherit (lib) mkOption types concatStringsSep; cfg = config.boot.tftp; in { imports = [ ./ramdisk.nix ]; options.boot.tftp.freeSpaceBytes = mkOption { type = types.int; default = 0; }; options.system.outputs = { tftpboot = mkOption { type = types.package; description = '' Directory containing files needed for TFTP booting ''; }; boot-scr = mkOption { type = types.package; description = '' U-Boot commands to load and boot a kernel and rootfs over TFTP. Copy-paste into the device boot monitor ''; }; }; config = { boot.ramdisk.enable = true; system.outputs = rec { tftpboot = let o = config.system.outputs; in pkgs.runCommand "tftpboot" {} '' mkdir $out cd $out ln -s ${o.rootfs} rootfs ln -s ${o.kernel} vmlinux ln -s ${o.manifest} manifest ln -s ${o.kernel.headers} build ln -s ${o.uimage} uimage ln -s ${o.boot-scr}/dtb dtb ln -s ${o.boot-scr}/script boot.scr ''; boot-scr = let inherit (pkgs.lib.trivial) toHexString; o = config.system.outputs; cmdline = concatStringsSep " " config.boot.commandLine; in pkgs.buildPackages.runCommand "boot-scr" { nativeBuildInputs = [ pkgs.pkgsBuildBuild.dtc ]; } '' uimageSize=$(($(stat -L -c %s ${o.uimage}) + 0x1000 &(~0xfff))) rootfsStart=0x$(printf %x $((${cfg.loadAddress} + 0x100000 + $uimageSize &(~0xfffff) ))) rootfsBytes=$(($(stat -L -c %s ${o.rootfs}) + 0x100000 &(~0xfffff))) rootfsMb=$(($rootfsBytes >> 20)) rootfsBytes=$(($rootfsBytes + ${toString cfg.freeSpaceBytes} )) cmd="mtdparts=phram0:''${rootfsMb}M(rootfs) phram.phram=phram0,''${rootfsStart},''${rootfsBytes},${config.hardware.flash.eraseBlockSize} root=/dev/mtdblock0"; dtbStart=$(printf %x $((${cfg.loadAddress} + $rootfsBytes + 0x100000 + $uimageSize ))) mkdir $out dtc -I dtb -O dts -o tmp.dts ${o.dtb} echo "/ { reserved-memory { phram: phram@$rootfsStart {compatible = \"phram\"; reg = <0x0 $rootfsStart 0x0 $(printf "0x%x" $rootfsBytes )>; }; }; } ;" >> tmp.dts cat tmp.dts dtc -I dts -O dtb -o $out/dtb tmp.dts dtbBytes=$(($(stat -L -c %s $out/dtb) + 0x1000 &(~0xfff))) cat > $out/script << EOF setenv serverip ${cfg.serverip} setenv ipaddr ${cfg.ipaddr} setenv bootargs 'liminix ${cmdline} $cmd' tftp 0x$(printf %x ${cfg.loadAddress}) result/uimage ; tftp 0x$(printf %x $rootfsStart) result/rootfs ; tftp 0x$dtbStart result/dtb bootm 0x$(printf %x ${cfg.loadAddress}) - 0x$dtbStart # EOF ''; }; }; }