| // Copyright 2019 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <log.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <xefi.h> |
| |
| #include <efi/boot-services.h> |
| #include <efi/protocol/loaded-image.h> |
| |
| // Caller frees memory. |
| efi_status xefi_get_load_options(size_t *load_options_size, void **load_options) { |
| efi_loaded_image_protocol *loaded; |
| efi_status status; |
| |
| DLOG("open loaded image"); |
| status = xefi_open_protocol(gImg, &LoadedImageProtocol, (void **)&loaded); |
| if (status != EFI_SUCCESS) { |
| ELOG_S(status, "%s: Cannot open LoadedImageProtocol", __func__); |
| goto exit0; |
| } |
| |
| LOG("image load options len = %u", loaded->LoadOptionsSize); |
| status = |
| gBS->AllocatePool(EfiLoaderData, loaded->LoadOptionsSize + sizeof(char16_t), load_options); |
| |
| if (status != EFI_SUCCESS) { |
| ELOG_S(status, "%s: failed to allocate memory", __func__); |
| goto exit1; |
| } |
| |
| if (loaded->LoadOptionsSize) { |
| DLOG("copy load options"); |
| gBS->CopyMem(*load_options, loaded->LoadOptions, loaded->LoadOptionsSize); |
| } |
| |
| // TODO(https://fxbug.dev/100791): either null-terminate this string or remove |
| // the extra char16_t. |
| |
| *load_options_size = loaded->LoadOptionsSize + sizeof(char16_t); |
| |
| exit1: |
| DLOG("close protocol"); |
| xefi_close_protocol(gImg, &LoadedImageProtocol); |
| exit0: |
| return status; |
| } |