1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * efi_selftest_loaded_image
4  *
5  * Copyright (c) 2018 Heinrich Schuchardt <xypron.glpk@gmx.de>
6  *
7  * This unit test checks the Loaded Image Protocol.
8  */
9 
10 #include <efi_selftest.h>
11 
12 static efi_guid_t loaded_image_protocol_guid =
13 	EFI_GUID(0x5b1b31a1, 0x9562, 0x11d2,
14 		 0x8e, 0x3f, 0x00, 0xa0, 0xc9, 0x69, 0x72, 0x3b);
15 static struct efi_boot_services *boottime;
16 efi_handle_t image_handle;
17 
18 /*
19  * Setup unit test.
20  *
21  * @handle:	handle of the loaded image
22  * @systable:	system table
23  */
24 static int setup(const efi_handle_t img_handle,
25 		 const struct efi_system_table *systable)
26 {
27 	boottime = systable->boottime;
28 	image_handle = img_handle;
29 
30 	return EFI_ST_SUCCESS;
31 }
32 
33 /*
34  * Execute unit test.
35  *
36  * Verify that the loaded image protocol is installed on the image handle.
37  * Verify that the loaded image protocol points to the system table.
38  */
39 static int execute(void)
40 {
41 	efi_status_t ret;
42 	efi_uintn_t i, protocol_buffer_count = 0;
43 	efi_guid_t **protocol_buffer = NULL;
44 	bool found = false;
45 	struct efi_loaded_image *loaded_image_protocol;
46 
47 	/*
48 	 * Get the GUIDs of all protocols installed on the handle.
49 	 */
50 	ret = boottime->protocols_per_handle(image_handle, &protocol_buffer,
51 					     &protocol_buffer_count);
52 	if (ret != EFI_SUCCESS) {
53 		efi_st_error("ProtocolsPerHandle failed\n");
54 		return EFI_ST_FAILURE;
55 	}
56 	if (!protocol_buffer_count | !protocol_buffer) {
57 		efi_st_error("ProtocolsPerHandle returned no protocol\n");
58 		return EFI_ST_FAILURE;
59 	}
60 	efi_st_printf("%u protocols installed on image handle\n",
61 		      (unsigned int)protocol_buffer_count);
62 	for (i = 0; i < protocol_buffer_count; ++i) {
63 		if (efi_st_memcmp(protocol_buffer[i],
64 				  &loaded_image_protocol_guid,
65 				  sizeof(efi_guid_t)))
66 			found = true;
67 	}
68 	if (!found) {
69 		efi_st_printf("LoadedImageProtocol not found\n");
70 		return EFI_ST_FAILURE;
71 	}
72 	ret = boottime->free_pool(protocol_buffer);
73 	if (ret != EFI_SUCCESS) {
74 		efi_st_error("FreePool failed\n");
75 		return EFI_ST_FAILURE;
76 	}
77 
78 	/*
79 	 * Open the loaded image protocol.
80 	 */
81 	ret = boottime->open_protocol(image_handle, &loaded_image_protocol_guid,
82 				      (void **)&loaded_image_protocol, NULL,
83 				      NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
84 	if (ret != EFI_SUCCESS) {
85 		efi_st_error("OpenProtocol failed\n");
86 		return EFI_ST_FAILURE;
87 	}
88 	if (loaded_image_protocol->revision !=
89 	    EFI_LOADED_IMAGE_PROTOCOL_REVISION) {
90 		efi_st_printf("Incorrect revision\n");
91 		return EFI_ST_FAILURE;
92 	}
93 	if (!loaded_image_protocol->system_table ||
94 	    loaded_image_protocol->system_table->hdr.signature !=
95 	    EFI_SYSTEM_TABLE_SIGNATURE) {
96 		efi_st_printf("System table reference missing\n");
97 		return EFI_ST_FAILURE;
98 	}
99 
100 	return EFI_ST_SUCCESS;
101 }
102 
103 EFI_UNIT_TEST(loadedimage) = {
104 	.name = "loaded image",
105 	.phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
106 	.setup = setup,
107 	.execute = execute,
108 };
109