On Sat, Sep 11, 2021 at 09:28:31AM +0200, Heinrich Schuchardt wrote:
> Simplify efi_sigstore_parse_sigdb() by using existing functions.
> 
> Signed-off-by: Heinrich Schuchardt <xypron.g...@gmx.de>
> ---
>  lib/efi_loader/efi_signature.c | 35 ++++++----------------------------
>  1 file changed, 6 insertions(+), 29 deletions(-)
> 
> diff --git a/lib/efi_loader/efi_signature.c b/lib/efi_loader/efi_signature.c
> index bdd09881fc..b741905a99 100644
> --- a/lib/efi_loader/efi_signature.c
> +++ b/lib/efi_loader/efi_signature.c
> @@ -7,6 +7,7 @@
>  #include <common.h>
>  #include <charset.h>
>  #include <efi_loader.h>
> +#include <efi_variable.h>
>  #include <image.h>
>  #include <hexdump.h>
>  #include <malloc.h>
> @@ -740,44 +741,20 @@ err:
>   */
>  struct efi_signature_store *efi_sigstore_parse_sigdb(u16 *name)
>  {
> -     struct efi_signature_store *sigstore = NULL;
>       const efi_guid_t *vendor;
>       void *db;
>       efi_uintn_t db_size;
> -     efi_status_t ret;
> 
> -     if (!u16_strcmp(name, L"PK") || !u16_strcmp(name, L"KEK")) {
> -             vendor = &efi_global_variable_guid;
> -     } else if (!u16_strcmp(name, L"db") || !u16_strcmp(name, L"dbx")) {
> -             vendor = &efi_guid_image_security_database;
> -     } else {
> +     vendor = efi_auth_var_get_guid(name);
> +     if (!vendor) {
>               EFI_PRINT("unknown signature database, %ls\n", name);
>               return NULL;
>       }

efi_auth_var_get_guid() will return &efi_global_variable_guid if the
GUID for the variable name isn't found.

> 
> -     /* retrieve variable data */
> -     db_size = 0;
> -     ret = EFI_CALL(efi_get_variable(name, vendor, NULL, &db_size, NULL));
> -     if (ret == EFI_NOT_FOUND) {
> -             EFI_PRINT("variable, %ls, not found\n", name);
> -             sigstore = calloc(sizeof(*sigstore), 1);
> -             return sigstore;
> -     } else if (ret != EFI_BUFFER_TOO_SMALL) {
> -             EFI_PRINT("Getting variable, %ls, failed\n", name);
> -             return NULL;
> -     }
> -
> -     db = malloc(db_size);
> +     db = efi_get_var(name, vendor, &db_size);
>       if (!db) {
> -             EFI_PRINT("Out of memory\n");
> -             return NULL;
> -     }
> -
> -     ret = EFI_CALL(efi_get_variable(name, vendor, NULL, &db_size, db));
> -     if (ret != EFI_SUCCESS) {
> -             EFI_PRINT("Getting variable, %ls, failed\n", name);
> -             free(db);
> -             return NULL;
> +             EFI_PRINT("variable, %ls, not found\n", name);
> +             return calloc(sizeof(struct efi_signature_store), 1);
>       }
> 
>       return efi_build_signature_store(db, db_size);
> --
> 2.30.2
> 

Reply via email to