--- a/grub-core/kern/ieee1275/openfw.c +++ b/grub-core/kern/ieee1275/openfw.c @@ -302,6 +302,34 @@ 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 @@ return grub_errno; } + grub_track_map (addr, size); return GRUB_ERR_NONE; } --- a/include/grub/ieee1275/ieee1275.h +++ b/include/grub/ieee1275/ieee1275.h @@ -33,6 +33,12 @@ 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 @@ -228,6 +234,7 @@ 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, --- a/grub-core/kern/ieee1275/init.c +++ b/grub-core/kern/ieee1275/init.c @@ -111,6 +111,7 @@ void grub_exit (void) { + grub_releasemap(); grub_ieee1275_exit (); }