blob: e6a356b5897f31d59046b05707e2a5b55b762b4b [file] [log] [blame]
Tom Rini70df9d62018-05-07 17:02:21 -04001// SPDX-License-Identifier: GPL-2.0+
Alexander Grafe83be452016-03-04 01:10:02 +01002/*
3 * EFI application disk support
4 *
5 * Copyright (c) 2016 Alexander Graf
Alexander Grafe83be452016-03-04 01:10:02 +01006 */
7
Heinrich Schuchardt37240572020-07-17 20:33:05 +02008#define LOG_CATEGORY LOGC_EFI
9
Alexander Grafe83be452016-03-04 01:10:02 +010010#include <common.h>
Simon Glasse3dfa912016-05-01 13:52:32 -060011#include <blk.h>
Simon Glass302d4f82016-05-14 14:03:05 -060012#include <dm.h>
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +090013#include <dm/device-internal.h>
14#include <dm/tag.h>
15#include <event.h>
Alexander Grafe83be452016-03-04 01:10:02 +010016#include <efi_loader.h>
AKASHI Takahiro120a01a2019-10-07 14:59:38 +090017#include <fs.h>
Heinrich Schuchardt37240572020-07-17 20:33:05 +020018#include <log.h>
Alexander Grafe83be452016-03-04 01:10:02 +010019#include <part.h>
20#include <malloc.h>
21
Heinrich Schuchardta9931732020-03-19 15:16:31 +010022struct efi_system_partition efi_system_partition;
23
Heinrich Schuchardt788ad412019-04-20 07:39:11 +020024const efi_guid_t efi_block_io_guid = EFI_BLOCK_IO_PROTOCOL_GUID;
Heinrich Schuchardt26397622021-02-02 17:53:14 +010025const efi_guid_t efi_system_partition_guid = PARTITION_SYSTEM_GUID;
Alexander Grafe83be452016-03-04 01:10:02 +010026
Heinrich Schuchardt72928722018-09-26 05:27:56 +020027/**
28 * struct efi_disk_obj - EFI disk object
29 *
30 * @header: EFI object header
31 * @ops: EFI disk I/O protocol interface
Heinrich Schuchardt72928722018-09-26 05:27:56 +020032 * @dev_index: device index of block device
33 * @media: block I/O media information
34 * @dp: device path to the block device
35 * @part: partition
36 * @volume: simple file system protocol of the partition
AKASHI Takahiro808ece12022-04-19 10:05:17 +090037 * @dev: associated DM device
Heinrich Schuchardt72928722018-09-26 05:27:56 +020038 */
Alexander Grafe83be452016-03-04 01:10:02 +010039struct efi_disk_obj {
Heinrich Schuchardt72928722018-09-26 05:27:56 +020040 struct efi_object header;
Alexander Grafe83be452016-03-04 01:10:02 +010041 struct efi_block_io ops;
Alexander Grafe83be452016-03-04 01:10:02 +010042 int dev_index;
Alexander Grafe83be452016-03-04 01:10:02 +010043 struct efi_block_io_media media;
Rob Clarka83272f2017-09-13 18:05:31 -040044 struct efi_device_path *dp;
Rob Clarka83272f2017-09-13 18:05:31 -040045 unsigned int part;
Rob Clark53142032017-09-13 18:05:34 -040046 struct efi_simple_file_system_protocol *volume;
Alexander Grafe83be452016-03-04 01:10:02 +010047};
48
Heinrich Schuchardt670198c2019-09-05 20:13:46 +020049/**
50 * efi_disk_reset() - reset block device
51 *
52 * This function implements the Reset service of the EFI_BLOCK_IO_PROTOCOL.
53 *
54 * As U-Boot's block devices do not have a reset function simply return
55 * EFI_SUCCESS.
56 *
57 * See the Unified Extensible Firmware Interface (UEFI) specification for
58 * details.
59 *
60 * @this: pointer to the BLOCK_IO_PROTOCOL
61 * @extended_verification: extended verification
62 * Return: status code
63 */
Alexander Grafe83be452016-03-04 01:10:02 +010064static efi_status_t EFIAPI efi_disk_reset(struct efi_block_io *this,
65 char extended_verification)
66{
67 EFI_ENTRY("%p, %x", this, extended_verification);
Heinrich Schuchardt670198c2019-09-05 20:13:46 +020068 return EFI_EXIT(EFI_SUCCESS);
Alexander Grafe83be452016-03-04 01:10:02 +010069}
70
AKASHI Takahiroba647af2022-05-12 11:29:01 +090071/**
72 * efi_disk_is_removable() - check if the device is removable media
73 * @handle: efi object handle;
74 *
75 * Examine the device and determine if the device is a local block device
76 * and removable media.
77 *
78 * Return: true if removable, false otherwise
79 */
80bool efi_disk_is_removable(efi_handle_t handle)
81{
82 struct efi_handler *handler;
83 struct efi_block_io *io;
84 efi_status_t ret;
85
86 ret = efi_search_protocol(handle, &efi_block_io_guid, &handler);
87 if (ret != EFI_SUCCESS)
88 return false;
89
90 io = handler->protocol_interface;
91
92 if (!io || !io->media)
93 return false;
94
95 return (bool)io->media->removable_media;
96}
97
Alexander Grafe83be452016-03-04 01:10:02 +010098enum efi_disk_direction {
99 EFI_DISK_READ,
100 EFI_DISK_WRITE,
101};
102
Heinrich Schuchardt9ce06172017-08-26 22:33:13 +0200103static efi_status_t efi_disk_rw_blocks(struct efi_block_io *this,
Alexander Grafe83be452016-03-04 01:10:02 +0100104 u32 media_id, u64 lba, unsigned long buffer_size,
105 void *buffer, enum efi_disk_direction direction)
106{
107 struct efi_disk_obj *diskobj;
Alexander Grafe83be452016-03-04 01:10:02 +0100108 int blksz;
109 int blocks;
110 unsigned long n;
111
Alexander Grafe83be452016-03-04 01:10:02 +0100112 diskobj = container_of(this, struct efi_disk_obj, ops);
AKASHI Takahiro808ece12022-04-19 10:05:17 +0900113 blksz = diskobj->media.block_size;
Alexander Grafe83be452016-03-04 01:10:02 +0100114 blocks = buffer_size / blksz;
115
Heinrich Schuchardtee6ef8b2019-09-05 19:43:17 +0200116 EFI_PRINT("blocks=%x lba=%llx blksz=%x dir=%d\n",
117 blocks, lba, blksz, direction);
Alexander Grafe83be452016-03-04 01:10:02 +0100118
119 /* We only support full block access */
120 if (buffer_size & (blksz - 1))
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200121 return EFI_BAD_BUFFER_SIZE;
Alexander Grafe83be452016-03-04 01:10:02 +0100122
AKASHI Takahiro8c591132022-04-28 13:49:16 +0900123 if (CONFIG_IS_ENABLED(PARTITIONS) &&
Masahisa Kojimac9611082022-07-22 11:39:10 +0900124 device_get_uclass_id(diskobj->header.dev) == UCLASS_PARTITION) {
AKASHI Takahiro8c591132022-04-28 13:49:16 +0900125 if (direction == EFI_DISK_READ)
Masahisa Kojimac9611082022-07-22 11:39:10 +0900126 n = dev_read(diskobj->header.dev, lba, blocks, buffer);
AKASHI Takahiro8c591132022-04-28 13:49:16 +0900127 else
Masahisa Kojimac9611082022-07-22 11:39:10 +0900128 n = dev_write(diskobj->header.dev, lba, blocks, buffer);
AKASHI Takahiro8c591132022-04-28 13:49:16 +0900129 } else {
130 /* dev is a block device (UCLASS_BLK) */
131 struct blk_desc *desc;
AKASHI Takahiro808ece12022-04-19 10:05:17 +0900132
Masahisa Kojimac9611082022-07-22 11:39:10 +0900133 desc = dev_get_uclass_plat(diskobj->header.dev);
AKASHI Takahiro8c591132022-04-28 13:49:16 +0900134 if (direction == EFI_DISK_READ)
135 n = blk_dread(desc, lba, blocks, buffer);
136 else
137 n = blk_dwrite(desc, lba, blocks, buffer);
138 }
Alexander Grafe83be452016-03-04 01:10:02 +0100139
140 /* We don't do interrupts, so check for timers cooperatively */
141 efi_timer_check();
142
Heinrich Schuchardtee6ef8b2019-09-05 19:43:17 +0200143 EFI_PRINT("n=%lx blocks=%x\n", n, blocks);
Alexander Graf62a67482016-06-02 11:38:27 +0200144
Alexander Grafe83be452016-03-04 01:10:02 +0100145 if (n != blocks)
Rob Clark99c43872017-07-25 20:28:29 -0400146 return EFI_DEVICE_ERROR;
Alexander Grafe83be452016-03-04 01:10:02 +0100147
Rob Clark99c43872017-07-25 20:28:29 -0400148 return EFI_SUCCESS;
Alexander Grafe83be452016-03-04 01:10:02 +0100149}
150
Heinrich Schuchardt76f57892020-04-10 17:10:34 +0200151/**
152 * efi_disk_read_blocks() - reads blocks from device
153 *
154 * This function implements the ReadBlocks service of the EFI_BLOCK_IO_PROTOCOL.
155 *
156 * See the Unified Extensible Firmware Interface (UEFI) specification for
157 * details.
158 *
159 * @this: pointer to the BLOCK_IO_PROTOCOL
160 * @media_id: id of the medium to be read from
161 * @lba: starting logical block for reading
162 * @buffer_size: size of the read buffer
163 * @buffer: pointer to the destination buffer
164 * Return: status code
165 */
Simon Glassc2194bf2016-09-25 15:27:32 -0600166static efi_status_t EFIAPI efi_disk_read_blocks(struct efi_block_io *this,
Heinrich Schuchardt0844f792018-01-19 20:24:48 +0100167 u32 media_id, u64 lba, efi_uintn_t buffer_size,
Alexander Grafe83be452016-03-04 01:10:02 +0100168 void *buffer)
169{
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200170 void *real_buffer = buffer;
171 efi_status_t r;
172
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200173 if (!this)
174 return EFI_INVALID_PARAMETER;
175 /* TODO: check for media changes */
176 if (media_id != this->media->media_id)
177 return EFI_MEDIA_CHANGED;
178 if (!this->media->media_present)
179 return EFI_NO_MEDIA;
Heinrich Schuchardt0bb73892021-01-23 19:33:11 +0100180 /* media->io_align is a power of 2 or 0 */
181 if (this->media->io_align &&
182 (uintptr_t)buffer & (this->media->io_align - 1))
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200183 return EFI_INVALID_PARAMETER;
184 if (lba * this->media->block_size + buffer_size >
Jesper Schmitz Mouridsenf1424932021-02-09 17:17:17 +0100185 (this->media->last_block + 1) * this->media->block_size)
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200186 return EFI_INVALID_PARAMETER;
187
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200188#ifdef CONFIG_EFI_LOADER_BOUNCE_BUFFER
189 if (buffer_size > EFI_LOADER_BOUNCE_BUFFER_SIZE) {
190 r = efi_disk_read_blocks(this, media_id, lba,
191 EFI_LOADER_BOUNCE_BUFFER_SIZE, buffer);
192 if (r != EFI_SUCCESS)
193 return r;
194 return efi_disk_read_blocks(this, media_id, lba +
195 EFI_LOADER_BOUNCE_BUFFER_SIZE / this->media->block_size,
196 buffer_size - EFI_LOADER_BOUNCE_BUFFER_SIZE,
197 buffer + EFI_LOADER_BOUNCE_BUFFER_SIZE);
198 }
199
200 real_buffer = efi_bounce_buffer;
201#endif
202
Masahiro Yamadac7570a32018-08-06 20:47:40 +0900203 EFI_ENTRY("%p, %x, %llx, %zx, %p", this, media_id, lba,
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200204 buffer_size, buffer);
205
206 r = efi_disk_rw_blocks(this, media_id, lba, buffer_size, real_buffer,
207 EFI_DISK_READ);
208
209 /* Copy from bounce buffer to real buffer if necessary */
210 if ((r == EFI_SUCCESS) && (real_buffer != buffer))
211 memcpy(buffer, real_buffer, buffer_size);
212
213 return EFI_EXIT(r);
Alexander Grafe83be452016-03-04 01:10:02 +0100214}
215
Heinrich Schuchardt76f57892020-04-10 17:10:34 +0200216/**
217 * efi_disk_write_blocks() - writes blocks to device
218 *
219 * This function implements the WriteBlocks service of the
220 * EFI_BLOCK_IO_PROTOCOL.
221 *
222 * See the Unified Extensible Firmware Interface (UEFI) specification for
223 * details.
224 *
225 * @this: pointer to the BLOCK_IO_PROTOCOL
226 * @media_id: id of the medium to be written to
227 * @lba: starting logical block for writing
228 * @buffer_size: size of the write buffer
229 * @buffer: pointer to the source buffer
230 * Return: status code
231 */
Simon Glassc2194bf2016-09-25 15:27:32 -0600232static efi_status_t EFIAPI efi_disk_write_blocks(struct efi_block_io *this,
Heinrich Schuchardt0844f792018-01-19 20:24:48 +0100233 u32 media_id, u64 lba, efi_uintn_t buffer_size,
Alexander Grafe83be452016-03-04 01:10:02 +0100234 void *buffer)
235{
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200236 void *real_buffer = buffer;
237 efi_status_t r;
238
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200239 if (!this)
240 return EFI_INVALID_PARAMETER;
241 if (this->media->read_only)
242 return EFI_WRITE_PROTECTED;
243 /* TODO: check for media changes */
244 if (media_id != this->media->media_id)
245 return EFI_MEDIA_CHANGED;
246 if (!this->media->media_present)
247 return EFI_NO_MEDIA;
Heinrich Schuchardt0bb73892021-01-23 19:33:11 +0100248 /* media->io_align is a power of 2 or 0 */
249 if (this->media->io_align &&
250 (uintptr_t)buffer & (this->media->io_align - 1))
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200251 return EFI_INVALID_PARAMETER;
252 if (lba * this->media->block_size + buffer_size >
Jesper Schmitz Mouridsenf1424932021-02-09 17:17:17 +0100253 (this->media->last_block + 1) * this->media->block_size)
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200254 return EFI_INVALID_PARAMETER;
255
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200256#ifdef CONFIG_EFI_LOADER_BOUNCE_BUFFER
257 if (buffer_size > EFI_LOADER_BOUNCE_BUFFER_SIZE) {
258 r = efi_disk_write_blocks(this, media_id, lba,
259 EFI_LOADER_BOUNCE_BUFFER_SIZE, buffer);
260 if (r != EFI_SUCCESS)
261 return r;
262 return efi_disk_write_blocks(this, media_id, lba +
263 EFI_LOADER_BOUNCE_BUFFER_SIZE / this->media->block_size,
264 buffer_size - EFI_LOADER_BOUNCE_BUFFER_SIZE,
265 buffer + EFI_LOADER_BOUNCE_BUFFER_SIZE);
266 }
267
268 real_buffer = efi_bounce_buffer;
269#endif
270
Masahiro Yamadac7570a32018-08-06 20:47:40 +0900271 EFI_ENTRY("%p, %x, %llx, %zx, %p", this, media_id, lba,
Alexander Graf7c00a3c2016-05-11 18:25:48 +0200272 buffer_size, buffer);
273
274 /* Populate bounce buffer if necessary */
275 if (real_buffer != buffer)
276 memcpy(real_buffer, buffer, buffer_size);
277
278 r = efi_disk_rw_blocks(this, media_id, lba, buffer_size, real_buffer,
279 EFI_DISK_WRITE);
280
281 return EFI_EXIT(r);
Alexander Grafe83be452016-03-04 01:10:02 +0100282}
283
Heinrich Schuchardt76f57892020-04-10 17:10:34 +0200284/**
285 * efi_disk_flush_blocks() - flushes modified data to the device
286 *
287 * This function implements the FlushBlocks service of the
288 * EFI_BLOCK_IO_PROTOCOL.
289 *
290 * As we always write synchronously nothing is done here.
291 *
292 * See the Unified Extensible Firmware Interface (UEFI) specification for
293 * details.
294 *
295 * @this: pointer to the BLOCK_IO_PROTOCOL
296 * Return: status code
297 */
Alexander Grafe83be452016-03-04 01:10:02 +0100298static efi_status_t EFIAPI efi_disk_flush_blocks(struct efi_block_io *this)
299{
Alexander Grafe83be452016-03-04 01:10:02 +0100300 EFI_ENTRY("%p", this);
301 return EFI_EXIT(EFI_SUCCESS);
302}
303
304static const struct efi_block_io block_io_disk_template = {
305 .reset = &efi_disk_reset,
306 .read_blocks = &efi_disk_read_blocks,
307 .write_blocks = &efi_disk_write_blocks,
308 .flush_blocks = &efi_disk_flush_blocks,
309};
310
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100311/**
312 * efi_fs_from_path() - retrieve simple file system protocol
313 *
314 * Gets the simple file system protocol for a file device path.
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100315 *
316 * The full path provided is split into device part and into a file
317 * part. The device part is used to find the handle on which the
318 * simple file system protocol is installed.
319 *
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100320 * @full_path: device path including device and file
321 * Return: simple file system protocol
Rob Clark53142032017-09-13 18:05:34 -0400322 */
323struct efi_simple_file_system_protocol *
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100324efi_fs_from_path(struct efi_device_path *full_path)
Rob Clark53142032017-09-13 18:05:34 -0400325{
326 struct efi_object *efiobj;
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100327 struct efi_handler *handler;
328 struct efi_device_path *device_path;
329 struct efi_device_path *file_path;
330 efi_status_t ret;
Rob Clark53142032017-09-13 18:05:34 -0400331
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100332 /* Split the path into a device part and a file part */
333 ret = efi_dp_split_file_path(full_path, &device_path, &file_path);
334 if (ret != EFI_SUCCESS)
335 return NULL;
336 efi_free_pool(file_path);
337
338 /* Get the EFI object for the partition */
Heinrich Schuchardt0a04a412022-03-19 06:35:43 +0100339 efiobj = efi_dp_find_obj(device_path, NULL, NULL);
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100340 efi_free_pool(device_path);
Rob Clark53142032017-09-13 18:05:34 -0400341 if (!efiobj)
342 return NULL;
343
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100344 /* Find the simple file system protocol */
345 ret = efi_search_protocol(efiobj, &efi_simple_file_system_protocol_guid,
346 &handler);
347 if (ret != EFI_SUCCESS)
348 return NULL;
Rob Clark53142032017-09-13 18:05:34 -0400349
Heinrich Schuchardt39578762018-01-19 20:24:39 +0100350 /* Return the simple file system protocol for the partition */
351 return handler->protocol_interface;
Rob Clark53142032017-09-13 18:05:34 -0400352}
353
AKASHI Takahiro120a01a2019-10-07 14:59:38 +0900354/**
355 * efi_fs_exists() - check if a partition bears a file system
356 *
357 * @desc: block device descriptor
358 * @part: partition number
359 * Return: 1 if a file system exists on the partition
360 * 0 otherwise
361 */
362static int efi_fs_exists(struct blk_desc *desc, int part)
363{
364 if (fs_set_blk_dev_with_part(desc, part))
365 return 0;
366
367 if (fs_get_type() == FS_TYPE_ANY)
368 return 0;
369
370 fs_close();
371
372 return 1;
373}
374
Heinrich Schuchardt76f57892020-04-10 17:10:34 +0200375/**
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100376 * efi_disk_add_dev() - create a handle for a partition or disk
Heinrich Schuchardt34f51832017-10-26 19:25:46 +0200377 *
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100378 * @parent: parent handle
379 * @dp_parent: parent device path
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100380 * @desc: internal block device
381 * @dev_index: device index for block device
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100382 * @part_info: partition info
Heinrich Schuchardt76f57892020-04-10 17:10:34 +0200383 * @part: partition
384 * @disk: pointer to receive the created handle
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100385 * Return: disk object
Heinrich Schuchardt34f51832017-10-26 19:25:46 +0200386 */
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100387static efi_status_t efi_disk_add_dev(
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100388 efi_handle_t parent,
389 struct efi_device_path *dp_parent,
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100390 struct blk_desc *desc,
391 int dev_index,
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100392 struct disk_partition *part_info,
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100393 unsigned int part,
394 struct efi_disk_obj **disk)
Alexander Graf06fe57b2016-04-11 16:16:17 +0200395{
396 struct efi_disk_obj *diskobj;
Heinrich Schuchardt2321b552020-05-21 09:22:06 +0200397 struct efi_object *handle;
Heinrich Schuchardt26397622021-02-02 17:53:14 +0100398 const efi_guid_t *guid = NULL;
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100399 efi_status_t ret;
Alexander Graf06fe57b2016-04-11 16:16:17 +0200400
Alexander Graf601db1f2016-08-05 14:51:47 +0200401 /* Don't add empty devices */
402 if (!desc->lba)
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100403 return EFI_NOT_READY;
Alexander Graf601db1f2016-08-05 14:51:47 +0200404
Rob Clarka83272f2017-09-13 18:05:31 -0400405 diskobj = calloc(1, sizeof(*diskobj));
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100406 if (!diskobj)
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100407 return EFI_OUT_OF_RESOURCES;
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100408
409 /* Hook up to the device list */
Heinrich Schuchardt72928722018-09-26 05:27:56 +0200410 efi_add_handle(&diskobj->header);
Alexander Graf06fe57b2016-04-11 16:16:17 +0200411
412 /* Fill in object data */
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100413 if (part_info) {
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100414 struct efi_device_path *node = efi_dp_part_node(desc, part);
Heinrich Schuchardtcb9ff4e2020-01-10 12:36:01 +0100415 struct efi_handler *handler;
416 void *protocol_interface;
417
Heinrich Schuchardte29fbb32022-10-06 13:36:02 +0200418 if (!node) {
419 ret = EFI_OUT_OF_RESOURCES;
420 goto error;
421 }
422
Heinrich Schuchardtcb9ff4e2020-01-10 12:36:01 +0100423 /* Parent must expose EFI_BLOCK_IO_PROTOCOL */
424 ret = efi_search_protocol(parent, &efi_block_io_guid, &handler);
425 if (ret != EFI_SUCCESS)
426 goto error;
427
428 /*
429 * Link the partition (child controller) to the block device
430 * (controller).
431 */
432 ret = efi_protocol_open(handler, &protocol_interface, NULL,
433 &diskobj->header,
434 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER);
435 if (ret != EFI_SUCCESS)
436 goto error;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100437
438 diskobj->dp = efi_dp_append_node(dp_parent, node);
439 efi_free_pool(node);
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100440 diskobj->media.last_block = part_info->size - 1;
Heinrich Schuchardt26397622021-02-02 17:53:14 +0100441 if (part_info->bootable & PART_EFI_SYSTEM_PARTITION)
442 guid = &efi_system_partition_guid;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100443 } else {
444 diskobj->dp = efi_dp_from_part(desc, part);
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100445 diskobj->media.last_block = desc->lba - 1;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100446 }
Rob Clarka83272f2017-09-13 18:05:31 -0400447 diskobj->part = part;
Heinrich Schuchardt2321b552020-05-21 09:22:06 +0200448
449 /*
450 * Install the device path and the block IO protocol.
451 *
452 * InstallMultipleProtocolInterfaces() checks if the device path is
453 * already installed on an other handle and returns EFI_ALREADY_STARTED
454 * in this case.
455 */
456 handle = &diskobj->header;
Ilias Apalodimas8ac0ebe2022-10-06 16:08:46 +0300457 ret = efi_install_multiple_protocol_interfaces(&handle,
458 &efi_guid_device_path,
459 diskobj->dp,
460 &efi_block_io_guid,
461 &diskobj->ops, guid,
462 NULL, NULL);
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100463 if (ret != EFI_SUCCESS)
Heinrich Schuchardte47b68b2021-11-20 13:56:02 +0100464 goto error;
Heinrich Schuchardt2321b552020-05-21 09:22:06 +0200465
466 /*
467 * On partitions or whole disks without partitions install the
468 * simple file system protocol if a file system is available.
469 */
AKASHI Takahiro356ef902019-10-07 14:59:39 +0900470 if ((part || desc->part_type == PART_TYPE_UNKNOWN) &&
471 efi_fs_exists(desc, part)) {
Rob Clark53142032017-09-13 18:05:34 -0400472 diskobj->volume = efi_simple_file_system(desc, part,
473 diskobj->dp);
Heinrich Schuchardt72928722018-09-26 05:27:56 +0200474 ret = efi_add_protocol(&diskobj->header,
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100475 &efi_simple_file_system_protocol_guid,
Heinrich Schuchardtce721a82018-01-19 20:24:38 +0100476 diskobj->volume);
Heinrich Schuchardt9fb0f2f2017-11-26 14:05:12 +0100477 if (ret != EFI_SUCCESS)
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100478 return ret;
Rob Clark53142032017-09-13 18:05:34 -0400479 }
Alexander Graf06fe57b2016-04-11 16:16:17 +0200480 diskobj->ops = block_io_disk_template;
Alexander Graf06fe57b2016-04-11 16:16:17 +0200481 diskobj->dev_index = dev_index;
482
483 /* Fill in EFI IO Media info (for read/write callbacks) */
484 diskobj->media.removable_media = desc->removable;
485 diskobj->media.media_present = 1;
Heinrich Schuchardt35922522019-09-05 19:31:23 +0200486 /*
487 * MediaID is just an arbitrary counter.
488 * We have to change it if the medium is removed or changed.
489 */
490 diskobj->media.media_id = 1;
Alexander Graf06fe57b2016-04-11 16:16:17 +0200491 diskobj->media.block_size = desc->blksz;
492 diskobj->media.io_align = desc->blksz;
Heinrich Schuchardtaf8c5b62020-03-19 15:45:52 +0100493 if (part)
Emmanuel Vadot5bf29642017-12-11 19:22:33 +0100494 diskobj->media.logical_partition = 1;
Alexander Graf06fe57b2016-04-11 16:16:17 +0200495 diskobj->ops.media = &diskobj->media;
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100496 if (disk)
497 *disk = diskobj;
Heinrich Schuchardta9931732020-03-19 15:16:31 +0100498
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100499 EFI_PRINT("BlockIO: part %u, present %d, logical %d, removable %d"
Paul Barbieriab814f12022-06-30 07:02:04 -0400500 ", last_block %llu\n",
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100501 diskobj->part,
502 diskobj->media.media_present,
503 diskobj->media.logical_partition,
504 diskobj->media.removable_media,
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100505 diskobj->media.last_block);
506
Heinrich Schuchardta9931732020-03-19 15:16:31 +0100507 /* Store first EFI system partition */
Simon Glassfada3f92022-09-17 09:00:09 -0600508 if (part && !efi_system_partition.uclass_id) {
Heinrich Schuchardt26397622021-02-02 17:53:14 +0100509 if (part_info->bootable & PART_EFI_SYSTEM_PARTITION) {
Simon Glassfada3f92022-09-17 09:00:09 -0600510 efi_system_partition.uclass_id = desc->uclass_id;
Heinrich Schuchardta9931732020-03-19 15:16:31 +0100511 efi_system_partition.devnum = desc->devnum;
512 efi_system_partition.part = part;
Heinrich Schuchardt54f180d2021-05-25 18:00:13 +0200513 EFI_PRINT("EFI system partition: %s %x:%x\n",
Simon Glassfada3f92022-09-17 09:00:09 -0600514 blk_get_uclass_name(desc->uclass_id),
Heinrich Schuchardta9931732020-03-19 15:16:31 +0100515 desc->devnum, part);
516 }
517 }
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100518 return EFI_SUCCESS;
Heinrich Schuchardtcb9ff4e2020-01-10 12:36:01 +0100519error:
520 efi_delete_handle(&diskobj->header);
521 return ret;
Alexander Graf06fe57b2016-04-11 16:16:17 +0200522}
523
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900524/*
525 * Create a handle for a whole raw disk
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100526 *
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900527 * @dev uclass device (UCLASS_BLK)
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100528 *
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900529 * Create an efi_disk object which is associated with @dev.
530 * The type of @dev must be UCLASS_BLK.
531 *
532 * @return 0 on success, -1 otherwise
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100533 */
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900534static int efi_disk_create_raw(struct udevice *dev)
Alexander Grafeb8d82c2016-04-11 16:16:18 +0200535{
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900536 struct efi_disk_obj *disk;
537 struct blk_desc *desc;
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900538 int diskid;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100539 efi_status_t ret;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100540
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900541 desc = dev_get_uclass_plat(dev);
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900542 diskid = desc->devnum;
Alexander Grafeb8d82c2016-04-11 16:16:18 +0200543
AKASHI Takahiro7ec925d2022-08-19 09:48:30 +0900544 ret = efi_disk_add_dev(NULL, NULL, desc,
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900545 diskid, NULL, 0, &disk);
546 if (ret != EFI_SUCCESS) {
547 if (ret == EFI_NOT_READY)
548 log_notice("Disk %s not ready\n", dev->name);
549 else
550 log_err("Adding disk for %s failed\n", dev->name);
Heinrich Schuchardtb56e7192021-01-23 06:52:21 +0100551
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900552 return -1;
Alexander Grafeb8d82c2016-04-11 16:16:18 +0200553 }
Masahisa Kojimac9611082022-07-22 11:39:10 +0900554 if (efi_link_dev(&disk->header, dev)) {
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900555 efi_free_pool(disk->dp);
556 efi_delete_handle(&disk->header);
Rob Clarka83272f2017-09-13 18:05:31 -0400557
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900558 return -1;
559 }
560
561 return 0;
Alexander Grafeb8d82c2016-04-11 16:16:18 +0200562}
563
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900564/*
565 * Create a handle for a disk partition
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100566 *
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900567 * @dev uclass device (UCLASS_PARTITION)
Alexander Grafe83be452016-03-04 01:10:02 +0100568 *
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900569 * Create an efi_disk object which is associated with @dev.
570 * The type of @dev must be UCLASS_PARTITION.
Heinrich Schuchardt62036062020-03-19 13:49:34 +0100571 *
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900572 * @return 0 on success, -1 otherwise
Alexander Grafe83be452016-03-04 01:10:02 +0100573 */
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900574static int efi_disk_create_part(struct udevice *dev)
Alexander Grafe83be452016-03-04 01:10:02 +0100575{
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900576 efi_handle_t parent;
577 struct blk_desc *desc;
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900578 struct disk_part *part_data;
579 struct disk_partition *info;
580 unsigned int part;
581 int diskid;
582 struct efi_handler *handler;
583 struct efi_device_path *dp_parent;
Heinrich Schuchardt6f3de6b2018-01-19 20:24:47 +0100584 struct efi_disk_obj *disk;
Heinrich Schuchardtcde7a362018-02-09 20:55:47 +0100585 efi_status_t ret;
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900586
587 if (dev_tag_get_ptr(dev_get_parent(dev), DM_TAG_EFI, (void **)&parent))
588 return -1;
589
590 desc = dev_get_uclass_plat(dev_get_parent(dev));
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900591 diskid = desc->devnum;
592
593 part_data = dev_get_uclass_plat(dev);
594 part = part_data->partnum;
595 info = &part_data->gpt_part_info;
596
597 ret = efi_search_protocol(parent, &efi_guid_device_path, &handler);
598 if (ret != EFI_SUCCESS)
599 return -1;
600 dp_parent = (struct efi_device_path *)handler->protocol_interface;
601
AKASHI Takahiro7ec925d2022-08-19 09:48:30 +0900602 ret = efi_disk_add_dev(parent, dp_parent, desc, diskid,
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900603 info, part, &disk);
604 if (ret != EFI_SUCCESS) {
605 log_err("Adding partition for %s failed\n", dev->name);
606 return -1;
607 }
Masahisa Kojimac9611082022-07-22 11:39:10 +0900608 if (efi_link_dev(&disk->header, dev)) {
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900609 efi_free_pool(disk->dp);
610 efi_delete_handle(&disk->header);
611
612 return -1;
613 }
614
615 return 0;
616}
617
618/*
619 * Create efi_disk objects for a block device
620 *
621 * @dev uclass device (UCLASS_BLK)
622 *
623 * Create efi_disk objects for partitions as well as a raw disk
624 * which is associated with @dev.
625 * The type of @dev must be UCLASS_BLK.
626 * This function is expected to be called at EV_PM_POST_PROBE.
627 *
628 * @return 0 on success, -1 otherwise
629 */
Heinrich Schuchardtda620cf2022-10-06 07:29:41 +0200630int efi_disk_probe(void *ctx, struct event *event)
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900631{
Simon Glass302d4f82016-05-14 14:03:05 -0600632 struct udevice *dev;
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900633 enum uclass_id id;
AKASHI Takahirocdd245a2022-04-19 10:05:13 +0900634 struct blk_desc *desc;
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900635 struct udevice *child;
636 int ret;
Simon Glass302d4f82016-05-14 14:03:05 -0600637
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900638 dev = event->data.dm.dev;
639 id = device_get_uclass_id(dev);
Simon Glass302d4f82016-05-14 14:03:05 -0600640
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900641 /* TODO: We won't support partitions in a partition */
642 if (id != UCLASS_BLK)
643 return 0;
644
AKASHI Takahirocdd245a2022-04-19 10:05:13 +0900645 /*
646 * avoid creating duplicated objects now that efi_driver
647 * has already created an efi_disk at this moment.
648 */
649 desc = dev_get_uclass_plat(dev);
Simon Glassfada3f92022-09-17 09:00:09 -0600650 if (desc->uclass_id != UCLASS_EFI_LOADER) {
AKASHI Takahirocdd245a2022-04-19 10:05:13 +0900651 ret = efi_disk_create_raw(dev);
652 if (ret)
653 return -1;
654 }
Simon Glass302d4f82016-05-14 14:03:05 -0600655
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900656 device_foreach_child(child, dev) {
657 ret = efi_disk_create_part(child);
658 if (ret)
659 return -1;
Simon Glass302d4f82016-05-14 14:03:05 -0600660 }
Alexander Grafe83be452016-03-04 01:10:02 +0100661
AKASHI Takahiro2381f2e2022-04-19 10:05:12 +0900662 return 0;
663}
664
AKASHI Takahiro7cebb752022-04-19 10:05:14 +0900665/*
666 * Delete an efi_disk object for a whole raw disk
667 *
668 * @dev uclass device (UCLASS_BLK)
669 *
670 * Delete an efi_disk object which is associated with @dev.
671 * The type of @dev must be UCLASS_BLK.
672 *
673 * @return 0 on success, -1 otherwise
674 */
675static int efi_disk_delete_raw(struct udevice *dev)
676{
677 efi_handle_t handle;
AKASHI Takahiroaff330a2022-04-19 10:05:15 +0900678 struct blk_desc *desc;
AKASHI Takahiro7cebb752022-04-19 10:05:14 +0900679 struct efi_disk_obj *diskobj;
680
681 if (dev_tag_get_ptr(dev, DM_TAG_EFI, (void **)&handle))
682 return -1;
683
AKASHI Takahiroaff330a2022-04-19 10:05:15 +0900684 desc = dev_get_uclass_plat(dev);
Simon Glassfada3f92022-09-17 09:00:09 -0600685 if (desc->uclass_id != UCLASS_EFI_LOADER) {
AKASHI Takahiroaff330a2022-04-19 10:05:15 +0900686 diskobj = container_of(handle, struct efi_disk_obj, header);
687 efi_free_pool(diskobj->dp);
688 }
AKASHI Takahiro7cebb752022-04-19 10:05:14 +0900689
690 efi_delete_handle(handle);
691 dev_tag_del(dev, DM_TAG_EFI);
692
693 return 0;
694}
695
696/*
697 * Delete an efi_disk object for a disk partition
698 *
699 * @dev uclass device (UCLASS_PARTITION)
700 *
701 * Delete an efi_disk object which is associated with @dev.
702 * The type of @dev must be UCLASS_PARTITION.
703 *
704 * @return 0 on success, -1 otherwise
705 */
706static int efi_disk_delete_part(struct udevice *dev)
707{
708 efi_handle_t handle;
709 struct efi_disk_obj *diskobj;
710
711 if (dev_tag_get_ptr(dev, DM_TAG_EFI, (void **)&handle))
712 return -1;
713
714 diskobj = container_of(handle, struct efi_disk_obj, header);
715
716 efi_free_pool(diskobj->dp);
717 efi_delete_handle(handle);
718 dev_tag_del(dev, DM_TAG_EFI);
719
720 return 0;
721}
722
723/*
724 * Delete an efi_disk object for a block device
725 *
726 * @dev uclass device (UCLASS_BLK or UCLASS_PARTITION)
727 *
728 * Delete an efi_disk object which is associated with @dev.
729 * The type of @dev must be either UCLASS_BLK or UCLASS_PARTITION.
730 * This function is expected to be called at EV_PM_PRE_REMOVE.
731 *
732 * @return 0 on success, -1 otherwise
733 */
Heinrich Schuchardtda620cf2022-10-06 07:29:41 +0200734int efi_disk_remove(void *ctx, struct event *event)
AKASHI Takahiro7cebb752022-04-19 10:05:14 +0900735{
736 enum uclass_id id;
737 struct udevice *dev;
738
739 dev = event->data.dm.dev;
740 id = device_get_uclass_id(dev);
741
742 if (id == UCLASS_BLK)
743 return efi_disk_delete_raw(dev);
744 else if (id == UCLASS_PARTITION)
745 return efi_disk_delete_part(dev);
746 else
747 return 0;
748}
749
Masahisa Kojimac5ff0a02022-09-12 17:33:50 +0900750/**
751 * efi_disk_get_device_name() - get U-Boot device name associated with EFI handle
752 *
753 * @handle: pointer to the EFI handle
754 * @buf: pointer to the buffer to store the string
755 * @size: size of buffer
756 * Return: status code
757 */
758efi_status_t efi_disk_get_device_name(const efi_handle_t handle, char *buf, int size)
759{
760 int count;
761 int diskid;
762 enum uclass_id id;
763 unsigned int part;
764 struct udevice *dev;
765 struct blk_desc *desc;
766 const char *if_typename;
767 bool is_partition = false;
768 struct disk_part *part_data;
769
770 if (!handle || !buf || !size)
771 return EFI_INVALID_PARAMETER;
772
773 dev = handle->dev;
774 id = device_get_uclass_id(dev);
775 if (id == UCLASS_BLK) {
776 desc = dev_get_uclass_plat(dev);
777 } else if (id == UCLASS_PARTITION) {
778 desc = dev_get_uclass_plat(dev_get_parent(dev));
779 is_partition = true;
780 } else {
781 return EFI_INVALID_PARAMETER;
782 }
Simon Glassfada3f92022-09-17 09:00:09 -0600783 if_typename = blk_get_uclass_name(desc->uclass_id);
Masahisa Kojimac5ff0a02022-09-12 17:33:50 +0900784 diskid = desc->devnum;
785
786 if (is_partition) {
787 part_data = dev_get_uclass_plat(dev);
788 part = part_data->partnum;
789 count = snprintf(buf, size, "%s %d:%d", if_typename, diskid, part);
790 } else {
791 count = snprintf(buf, size, "%s %d", if_typename, diskid);
792 }
793
794 if (count < 0 || (count + 1) > size)
795 return EFI_INVALID_PARAMETER;
796
797 return EFI_SUCCESS;
798}
Tom Rini586c22b2022-09-19 13:19:39 -0400799
800/**
Heinrich Schuchardta02228f2022-08-31 16:37:35 +0200801 * efi_disks_register() - ensure all block devices are available in UEFI
802 *
803 * The function probes all block devices. As we store UEFI variables on the
804 * EFI system partition this function has to be called before enabling
805 * variable services.
806 */
807efi_status_t efi_disks_register(void)
808{
809 struct udevice *dev;
810
811 uclass_foreach_dev_probe(UCLASS_BLK, dev) {
812 }
813
814 return EFI_SUCCESS;
815}