efi: use efi_get_memory_map() to get final map for x86
Replace the open-coded memory map getting with the efi_get_memory_map() that is now general enough to use. Signed-off-by: Roy Franz <roy.franz@linaro.org> Signed-off-by: Matt Fleming <matt.fleming@intel.com>
This commit is contained in:
parent
1c089c65f5
commit
ae8e9060a3
|
@ -527,26 +527,13 @@ static efi_status_t exit_boot(struct boot_params *boot_params,
|
|||
u8 nr_entries;
|
||||
int i;
|
||||
|
||||
size = sizeof(*mem_map) * 32;
|
||||
get_map:
|
||||
status = efi_get_memory_map(sys_table, &mem_map, &size, &desc_size,
|
||||
&desc_version, &key);
|
||||
|
||||
again:
|
||||
size += sizeof(*mem_map) * 2;
|
||||
_size = size;
|
||||
status = efi_low_alloc(sys_table, size, 1, (unsigned long *)&mem_map);
|
||||
if (status != EFI_SUCCESS)
|
||||
return status;
|
||||
|
||||
get_map:
|
||||
status = efi_call_phys5(sys_table->boottime->get_memory_map, &size,
|
||||
mem_map, &key, &desc_size, &desc_version);
|
||||
if (status == EFI_BUFFER_TOO_SMALL) {
|
||||
efi_free(sys_table, _size, (unsigned long)mem_map);
|
||||
goto again;
|
||||
}
|
||||
|
||||
if (status != EFI_SUCCESS)
|
||||
goto free_mem_map;
|
||||
|
||||
memcpy(&efi->efi_loader_signature, EFI_LOADER_SIGNATURE, sizeof(__u32));
|
||||
efi->efi_systab = (unsigned long)sys_table;
|
||||
efi->efi_memdesc_size = desc_size;
|
||||
|
@ -574,6 +561,7 @@ static efi_status_t exit_boot(struct boot_params *boot_params,
|
|||
goto free_mem_map;
|
||||
|
||||
called_exit = true;
|
||||
efi_call_phys1(sys_table->boottime->free_pool, mem_map);
|
||||
goto get_map;
|
||||
}
|
||||
|
||||
|
@ -642,7 +630,7 @@ static efi_status_t exit_boot(struct boot_params *boot_params,
|
|||
return EFI_SUCCESS;
|
||||
|
||||
free_mem_map:
|
||||
efi_free(sys_table, _size, (unsigned long)mem_map);
|
||||
efi_call_phys1(sys_table->boottime->free_pool, mem_map);
|
||||
return status;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue