Populate preloaded_undi device when applicable.

diff --git a/src/arch/i386/prefix/pxeprefix.S b/src/arch/i386/prefix/pxeprefix.S
index 8b485e48f8983d590107e49c5484d7c675224ae6..62b9348429822329e29c14d74ca3244c4d25545a 100644 (file)
--- a/src/arch/i386/prefix/pxeprefix.S
+++ b/src/arch/i386/prefix/pxeprefix.S
@@ -691,6 +691,15 @@ run_etherboot:
        /* Install Etherboot */
        call    install
 
+#ifdef PXELOADER_KEEP_UNDI
+       /* Copy our undi_device structure to the preloaded_undi variable */
+       movw    %bx, %es
+       movw    $preloaded_undi, %di
+       movw    $undi_device, %si
+       movw    $undi_device_size, %cx
+       rep movsb
+#endif
+
        /* Jump to .text16 segment with %ds pointing to .data16*/
        movw    %bx, %ds
        pushw   %ax