1 files changed, 45 insertions, 0 deletions
diff --git a/build_tools/archiso/archiso/hooks/archiso_pxe_nbd b/build_tools/archiso/archiso/hooks/archiso_pxe_nbd
new file mode 100644
index 0000000..6147d06
--- /dev/null
+++ b/build_tools/archiso/archiso/hooks/archiso_pxe_nbd
@@ -0,0 +1,45 @@
+# vim: set ft=sh:
+
+run_hook() {
+    if [[ -n "${ip}" && -n "${archiso_nbd_srv}" ]]; then
+
+        archiso_nbd_srv=$(eval echo ${archiso_nbd_srv})
+        [[ -z "${archiso_nbd_name}" ]] && archiso_nbd_name="archiso"
+
+        mount_handler="archiso_pxe_nbd_mount_handler"
+    fi
+}
+
+archiso_pxe_nbd_mount_handler () {
+    newroot="${1}"
+
+    # Module autoloading like with loop devices does not work, doing manually...
+    modprobe nbd 2> /dev/null
+    msg ":: Waiting for boot device..."
+    while ! poll_device /dev/nbd0 30; do
+        echo "ERROR: boot device didn't show up after 30 seconds..."
+        echo "   Falling back to interactive prompt"
+        echo "   You can try to fix the problem manually, log out when you are finished"
+        launch_interactive_shell
+    done
+
+    msg ":: Setup NBD from ${archiso_nbd_srv} at /dev/nbd0"
+    if [[ "${copytoram}" = "y" ]]; then
+        nbd-client ${archiso_nbd_srv} -N ${archiso_nbd_name} /dev/nbd0
+    else
+        nbd-client ${archiso_nbd_srv} -N ${archiso_nbd_name} /dev/nbd0 -persist
+    fi
+
+    archisodevice=/dev/nbd0
+
+    archiso_mount_handler ${newroot}
+
+    if [[ "${copytoram}" = "y" ]]; then
+        msg ":: Disconnect NBD from ${archiso_nbd_srv} at /dev/nbd0"
+        nbd-client -d /dev/nbd0
+    else
+        mkdir -p /run/archiso
+        pidof nbd-client > /run/archiso/nbd_client.pid
+        cp /archiso_pxe_nbd ${newroot}/etc/rc.d/functions.d/
+    fi
+}