efi_memory: use list_count_nodes() to count list entries

Use the API function list_count_nodes() to count the number of EFI
memory map entries.

Signed-off-by: Sughosh Ganu <sughosh.ganu@linaro.org>
Reviewed-by: Simon Glass <sjg@chromium.org>
Reviewed-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>

diff --git a/lib/efi_loader/efi_memory.c b/lib/efi_loader/efi_memory.c
index 12cf23fa3fa8d76d7cd0cb84b5406bfa58a02c54..9a62864a0026edbe741cd39a798d17493577a341 100644 (file)
--- a/lib/efi_loader/efi_memory.c
+++ b/lib/efi_loader/efi_memory.c
@@ -742,8 +742,8 @@ efi_status_t efi_get_memory_map(efi_uintn_t *memory_map_size,
                                efi_uintn_t *descriptor_size,
                                uint32_t *descriptor_version)
 {
+       size_t map_entries;
        efi_uintn_t map_size = 0;
-       int map_entries = 0;
        struct list_head *lhandle;
        efi_uintn_t provided_map_size;
 
@@ -752,8 +752,7 @@ efi_status_t efi_get_memory_map(efi_uintn_t *memory_map_size,
 
        provided_map_size = *memory_map_size;
 
-       list_for_each(lhandle, &efi_mem)
-               map_entries++;
+       map_entries = list_count_nodes(&efi_mem);
 
        map_size = map_entries * sizeof(struct efi_mem_desc);