Re: [RFT] PXE chainloader

["Vladimir 'φ-coder/phcoder' Serbinenko" <phcoder@gmail.com>]

[-- Attachment #1.1: Type: text/plain, Size: 2976 bytes --]

On 09/06/2010 01:08 AM, Omni Flux wrote:
> The output of two tests follow. It works, to an extent. Is PXE being
> left in a state where the next bootloader can call it's initialization
> routines again?
>
Attached patch may help. I noticed that last file may still remain opened.
>
> Test 1: Boot grub PXE, chainload grub PXE, chainload grub PXE again.
> Result: Chainloaded grub PXEs load in rescue mode.
>
> Pre-boot eXecution Environment (PXE) 2.43
> (C) Copyright 1999 Intel Corporation
> (C) Copyright 2000 Lanworks Technologies Co. a subsidiary of 3Com
> Corporation
> All rights reserved
>
>
> CLIENT MAC ADDR: XX XX XX XX XX XX  GUID:
> 00020003-0004-0005-0006-000700080009
> CLIENT IP: 192.168.2.101  MASK: 255.255.255.0  DHCP IP: 192.168.1.1
> GATEWAY IP: 192.168.2.1
> Welcome to GRUB!
>
>                              GNU GRUB  version 1.98
>
>    Minimal BASH-like line editing is supported. For the first word, TAB
>    lists possible command completions. Anywhere else TAB lists possible
>    device or file completions.
>
>
> grub> pxechainloader (pxe)/boot/grub/grub2pxe
> grub> boot
> Welcome to GRUB!
>
> error: file not found.
> Entering rescue mode...
> grub rescue> insmod pxechain
> grub rescue> pxechainloader (pxe)/boot/grub/grub2pxe
> grub rescue> boot
> Welcome to GRUB!
>
> error: file not found.
> Entering rescue mode...
> grub rescue>
>
>
>
>
>
>
>
>
>
> Test 2: Boot grub PXE, chainload pxelinux (Debian Lenny net installer)
> Result: Chainloaded pxelinux fails to initialize UDP stack.
>
> Pre-boot eXecution Environment (PXE) 2.43
> (C) Copyright 1999 Intel Corporation
> (C) Copyright 2000 Lanworks Technologies Co. a subsidiary of 3Com
> Corporation
> All rights reserved
>
>
> CLIENT MAC ADDR: XX XX XX XX XX XX  GUID:
> 00020003-0004-0005-0006-000700080009
> CLIENT IP: 192.168.2.101  MASK: 255.255.255.0  DHCP IP: 192.168.1.1
> GATEWAY IP: 192.168.2.1
> Welcome to GRUB!
>
>                              GNU GRUB  version 1.98
>
>    Minimal BASH-like line editing is supported. For the first word, TAB
>    lists possible command completions. Anywhere else TAB lists possible
>    device or file completions.
>
>
> grub> pxechainloader (pxe)/boot/debian/lenny/i386/pxelinux.0
> grub> boot
>
> PXELINUX 3.71 Debian-2008-09-06  Copyright (C) 1994-2008 H. Peter Anvin
> Found PXENV+ structure
> PXE API version is 0201
> UNDI data segment at:   00096690
> UNDI data segment size: 4801
> UNDI code segment at:   0009AEA0
> UNDI code segment size: 3160
> PXE entry point found (we hope) at 9AEA:00FA
> Getting cached packet 01
> Getting cached packet 02
> Getting cached packet 03
> My IP address seemes to be 45A84865 192.168.2.101
> ip=192.168.2.101:192.168.1.1:192.168.2.1:255.255.255.0
> Failed to initialize UDP stack
>
> Boot failed: press a key to retry, or wait for reset...
>


-- 
Regards
Vladimir 'φ-coder/phcoder' Serbinenko


[-- Warning: decoded text below may be mangled, UTF-8 assumed --]
[-- Attachment #1.2: pxechain.diff --]
[-- Type: text/x-diff; name="pxechain.diff", Size: 4431 bytes --]

=== modified file 'grub-core/Makefile.core.def'
--- grub-core/Makefile.core.def	2010-08-30 13:13:38 +0000
+++ grub-core/Makefile.core.def	2010-09-03 23:57:56 +0000
@@ -1064,6 +1064,12 @@
 };
 
 module = {
+  name = pxechain;
+  i386_pc = loader/i386/pc/pxechainloader.c;
+  enable = i386_pc;
+};
+
+module = {
   name = multiboot2;
   cppflags = "-DGRUB_USE_MULTIBOOT2";
 

=== modified file 'grub-core/fs/i386/pc/pxe.c'
--- grub-core/fs/i386/pc/pxe.c	2010-09-02 13:47:03 +0000
+++ grub-core/fs/i386/pc/pxe.c	2010-09-05 23:28:35 +0000
@@ -522,12 +522,17 @@
 void
 grub_pxe_unload (void)
 {
+  struct grub_pxenv_tftp_close c;
+
   if (grub_pxe_pxenv)
     {
       grub_fs_unregister (&grub_pxefs_fs);
       grub_disk_dev_unregister (&grub_pxe_dev);
-
-      grub_pxe_pxenv = 0;
+      if (curr_file != 0)
+	{
+	  grub_pxe_call (GRUB_PXENV_TFTP_CLOSE, &c, pxe_rm_entry);
+	  curr_file = 0;
+	}
     }
 }
 

=== added file 'grub-core/loader/i386/pc/pxechainloader.c'
--- grub-core/loader/i386/pc/pxechainloader.c	1970-01-01 00:00:00 +0000
+++ grub-core/loader/i386/pc/pxechainloader.c	2010-09-05 23:29:09 +0000
@@ -0,0 +1,132 @@
+/* chainloader.c - boot another boot loader */
+/*
+ *  GRUB  --  GRand Unified Bootloader
+ *  Copyright (C) 2002,2004,2007,2009,2010  Free Software Foundation, Inc.
+ *
+ *  GRUB is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  GRUB is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <grub/loader.h>
+#include <grub/file.h>
+#include <grub/err.h>
+#include <grub/device.h>
+#include <grub/disk.h>
+#include <grub/misc.h>
+#include <grub/types.h>
+#include <grub/partition.h>
+#include <grub/dl.h>
+#include <grub/command.h>
+#include <grub/machine/biosnum.h>
+#include <grub/i18n.h>
+#include <grub/video.h>
+#include <grub/mm.h>
+#include <grub/cpu/relocator.h>
+
+static grub_dl_t my_mod;
+static struct grub_relocator *rel;
+static grub_uint32_t edx = 0xffffffff;
+
+static grub_err_t
+grub_pxechain_boot (void)
+{
+  struct grub_relocator16_state state = { 
+    .cs = 0,
+    .ip = 0x7c00,
+    .ds = 0,
+    .es = 0,
+    .fs = 0,
+    .gs = 0,
+    .ss = 0,
+    .sp = 0x7c00,
+    .edx = edx
+  };
+  grub_video_set_mode ("text", 0, 0);
+
+  grub_pxe_unload ();
+
+  return grub_relocator16_boot (rel, state);
+}
+
+static grub_err_t
+grub_pxechain_unload (void)
+{
+  grub_relocator_unload (rel);
+  rel = NULL;
+  grub_dl_unref (my_mod);
+  return GRUB_ERR_NONE;
+}
+
+static grub_err_t
+grub_cmd_pxechain (grub_command_t cmd __attribute__ ((unused)),
+		int argc, char *argv[])
+{
+  grub_file_t file = 0;
+  grub_err_t err;
+  void *image;
+  grub_size_t imagesize;
+
+  if (argc == 0)
+    return grub_error (GRUB_ERR_BAD_ARGUMENT, "no file specified");
+
+  grub_dl_ref (my_mod);
+
+  rel = grub_relocator_new ();
+  if (!rel)
+    goto fail;
+
+  file = grub_file_open (argv[0]);
+  if (! file)
+    goto fail;
+
+
+  edx = grub_get_root_biosnumber ();
+
+  imagesize = grub_file_size (file);
+  {
+    grub_relocator_chunk_t ch;
+    err = grub_relocator_alloc_chunk_addr (rel, &ch, 0x7c00, imagesize);
+    if (err)
+      goto fail;
+    image = get_virtual_current_address (ch);
+  }
+
+  if (grub_file_read (file, image, imagesize) != (grub_ssize_t) imagesize)
+    goto fail;
+ 
+  grub_loader_set (grub_pxechain_boot, grub_pxechain_unload, 1);
+  return GRUB_ERR_NONE;
+
+ fail:
+
+  if (file)
+    grub_file_close (file);
+
+  grub_pxechain_unload ();
+
+  return grub_errno;
+}
+
+static grub_command_t cmd;
+
+GRUB_MOD_INIT(pxechainloader)
+{
+  cmd = grub_register_command ("pxechainloader", grub_cmd_pxechain,
+			       0, N_("Chainload a PXE image."));
+  my_mod = mod;
+}
+
+GRUB_MOD_FINI(pxechainloader)
+{
+  grub_unregister_command (cmd);
+}


[-- Attachment #2: OpenPGP digital signature --]
[-- Type: application/pgp-signature, Size: 294 bytes --]