grub2/grub2-ppc64le-memory-map.patch

Index: grub-2.04~rc1/grub-core/kern/ieee1275/openfw.c
===================================================================
--- grub-2.04~rc1.orig/grub-core/kern/ieee1275/openfw.c
+++ grub-2.04~rc1/grub-core/kern/ieee1275/openfw.c
@@ -302,6 +302,34 @@ grub_ieee1275_map (grub_addr_t phys, gru
   return args.catch_result;
 }
 
+/* Preallocate IEEE1275_MAX_MAP_RESOURCE map tracks to track the
+ * map regions allocated to us by the firmware. Cannot
+ * dynamically allocate them, since the heap is not set
+ * yet.
+ */
+struct grub_map_track  grub_map_track[IEEE1275_MAX_MAP_RESOURCE];
+int grub_map_track_index=0;
+
+void
+grub_releasemap ()
+{
+  int i=0;
+  for (i=grub_map_track_index-1; i >= 0; i--)
+   grub_ieee1275_release(grub_map_track[i].addr, grub_map_track[i].size);
+  grub_map_track_index = 0;
+  return;
+}
+
+static void
+grub_track_map (grub_addr_t addr, grub_size_t size)
+{
+  if (grub_map_track_index >= IEEE1275_MAX_MAP_RESOURCE)
+   return;
+  grub_map_track[grub_map_track_index].addr = addr;
+  grub_map_track[grub_map_track_index++].size = size;
+  return;
+}
+
 grub_err_t
 grub_claimmap (grub_addr_t addr, grub_size_t size)
 {
@@ -317,6 +345,7 @@ grub_claimmap (grub_addr_t addr, grub_si
       return grub_errno;
     }
 
+  grub_track_map (addr, size);
   return GRUB_ERR_NONE;
 }
 
Index: grub-2.04~rc1/include/grub/ieee1275/ieee1275.h
===================================================================
--- grub-2.04~rc1.orig/include/grub/ieee1275/ieee1275.h
+++ grub-2.04~rc1/include/grub/ieee1275/ieee1275.h
@@ -30,6 +30,12 @@ struct grub_ieee1275_mem_region
   unsigned int size;
 };
 
+#define IEEE1275_MAX_MAP_RESOURCE 10
+struct grub_map_track {
+  grub_addr_t addr;
+  grub_size_t size;
+};
+
 #define IEEE1275_MAX_PROP_LEN	8192
 #define IEEE1275_MAX_PATH_LEN	256
 
@@ -235,6 +241,7 @@ char *EXPORT_FUNC(grub_ieee1275_encode_u
 int EXPORT_FUNC(grub_ieee1275_get_block_size) (grub_ieee1275_ihandle_t ihandle);
 
 grub_err_t EXPORT_FUNC(grub_claimmap) (grub_addr_t addr, grub_size_t size);
+void EXPORT_FUNC(grub_releasemap) (void);
 
 int
 EXPORT_FUNC(grub_ieee1275_map) (grub_addr_t phys, grub_addr_t virt,
Index: grub-2.04~rc1/grub-core/kern/ieee1275/init.c
===================================================================
--- grub-2.04~rc1.orig/grub-core/kern/ieee1275/init.c
+++ grub-2.04~rc1/grub-core/kern/ieee1275/init.c
@@ -73,6 +73,7 @@ grub_addr_t grub_ieee1275_original_stack
 void
 grub_exit (void)
 {
+  grub_releasemap();
   grub_ieee1275_exit ();
 }