mirrors/posix-uefi
1
0
Fork 0
mirror of https://gitlab.com/bztsrc/posix-uefi.git synced 2025-01-01 03:25:52 +01:00
Code Issues Projects Releases Wiki Activity

Make sure returned address is NULL on error

Browse source
This commit is contained in:
bzt 2021-02-09 17:55:31 +01:00
parent 61758230ba
commit e90f4a44bc
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
uefi/stdlib.c
View file

@ -76,7 +76,7 @@ void *malloc (size_t __size)
{ {
void *ret = NULL; void *ret = NULL;
efi_status_t status = BS->AllocatePool(LIP ? LIP->ImageDataType : EfiLoaderData, __size, &ret); efi_status_t status = BS->AllocatePool(LIP ? LIP->ImageDataType : EfiLoaderData, __size, &ret);
if(!EFI_ERROR(status) || !ret) errno = ENOMEM; if(EFI_ERROR(status) || !ret) { errno = ENOMEM; ret = NULL; }
return ret; return ret;
} }
@ -93,7 +93,7 @@ void *realloc (void *__ptr, size_t __size)
void *ret = __ptr; void *ret = __ptr;
/* not sure if this works */ /* not sure if this works */
efi_status_t status = BS->AllocatePool(LIP ? LIP->ImageDataType : EfiLoaderData, __size, &ret); efi_status_t status = BS->AllocatePool(LIP ? LIP->ImageDataType : EfiLoaderData, __size, &ret);
if(!EFI_ERROR(status) || !ret) errno = ENOMEM; if(EFI_ERROR(status) || !ret) { errno = ENOMEM; ret = NULL; }
return ret; return ret;
#else #else
void *ret = malloc(__size); void *ret = malloc(__size);
@ -107,7 +107,7 @@ void *realloc (void *__ptr, size_t __size)
void free (void *__ptr) void free (void *__ptr)
{ {
efi_status_t status = BS->FreePool(__ptr); efi_status_t status = BS->FreePool(__ptr);
if(!EFI_ERROR(status)) errno = ENOMEM; if(EFI_ERROR(status)) errno = ENOMEM;
} }
void abort () void abort ()