]> Git Repo - u-boot.git/blame - lib/efi_loader/efi_disk.c
blk: Switch over to using uclass IDs
[u-boot.git] / lib / efi_loader / efi_disk.c
CommitLineData
f739fcd8 1// SPDX-License-Identifier: GPL-2.0+
2a22d05d
AG
2/*
3 * EFI application disk support
4 *
5 * Copyright (c) 2016 Alexander Graf
2a22d05d
AG
6 */
7
af457cfc
HS
8#define LOG_CATEGORY LOGC_EFI
9
2a22d05d 10#include <common.h>
6dd9faf8 11#include <blk.h>
487d756f 12#include <dm.h>
a9bf024b
AT
13#include <dm/device-internal.h>
14#include <dm/tag.h>
15#include <event.h>
2a22d05d 16#include <efi_loader.h>
86740067 17#include <fs.h>
af457cfc 18#include <log.h>
2a22d05d
AG
19#include <part.h>
20#include <malloc.h>
21
11078bb2
HS
22struct efi_system_partition efi_system_partition;
23
dec88e41 24const efi_guid_t efi_block_io_guid = EFI_BLOCK_IO_PROTOCOL_GUID;
b9b0ea30 25const efi_guid_t efi_system_partition_guid = PARTITION_SYSTEM_GUID;
2a22d05d 26
d39646a3
HS
27/**
28 * struct efi_disk_obj - EFI disk object
29 *
30 * @header: EFI object header
31 * @ops: EFI disk I/O protocol interface
d39646a3
HS
32 * @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
d97e98c8 37 * @dev: associated DM device
d39646a3 38 */
2a22d05d 39struct efi_disk_obj {
d39646a3 40 struct efi_object header;
2a22d05d 41 struct efi_block_io ops;
2a22d05d 42 int dev_index;
2a22d05d 43 struct efi_block_io_media media;
884bcf6f 44 struct efi_device_path *dp;
884bcf6f 45 unsigned int part;
2a92080d 46 struct efi_simple_file_system_protocol *volume;
2a22d05d
AG
47};
48
cda9b352
HS
49/**
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 */
2a22d05d
AG
64static 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);
cda9b352 68 return EFI_EXIT(EFI_SUCCESS);
2a22d05d
AG
69}
70
05f391e2
AT
71/**
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
2a22d05d
AG
98enum efi_disk_direction {
99 EFI_DISK_READ,
100 EFI_DISK_WRITE,
101};
102
a80a232e 103static efi_status_t efi_disk_rw_blocks(struct efi_block_io *this,
2a22d05d
AG
104 u32 media_id, u64 lba, unsigned long buffer_size,
105 void *buffer, enum efi_disk_direction direction)
106{
107 struct efi_disk_obj *diskobj;
2a22d05d
AG
108 int blksz;
109 int blocks;
110 unsigned long n;
111
2a22d05d 112 diskobj = container_of(this, struct efi_disk_obj, ops);
d97e98c8 113 blksz = diskobj->media.block_size;
2a22d05d
AG
114 blocks = buffer_size / blksz;
115
9d3f3398
HS
116 EFI_PRINT("blocks=%x lba=%llx blksz=%x dir=%d\n",
117 blocks, lba, blksz, direction);
2a22d05d
AG
118
119 /* We only support full block access */
120 if (buffer_size & (blksz - 1))
f59f0825 121 return EFI_BAD_BUFFER_SIZE;
2a22d05d 122
6c640422 123 if (CONFIG_IS_ENABLED(PARTITIONS) &&
ee576662 124 device_get_uclass_id(diskobj->header.dev) == UCLASS_PARTITION) {
6c640422 125 if (direction == EFI_DISK_READ)
ee576662 126 n = dev_read(diskobj->header.dev, lba, blocks, buffer);
6c640422 127 else
ee576662 128 n = dev_write(diskobj->header.dev, lba, blocks, buffer);
6c640422
AT
129 } else {
130 /* dev is a block device (UCLASS_BLK) */
131 struct blk_desc *desc;
132
ee576662 133 desc = dev_get_uclass_plat(diskobj->header.dev);
6c640422
AT
134 if (direction == EFI_DISK_READ)
135 n = blk_dread(desc, lba, blocks, buffer);
136 else
137 n = blk_dwrite(desc, lba, blocks, buffer);
138 }
2a22d05d
AG
139
140 /* We don't do interrupts, so check for timers cooperatively */
141 efi_timer_check();
142
9d3f3398 143 EFI_PRINT("n=%lx blocks=%x\n", n, blocks);
edcef3ba 144
2a22d05d 145 if (n != blocks)
3304990b 146 return EFI_DEVICE_ERROR;
2a22d05d 147
3304990b 148 return EFI_SUCCESS;
2a22d05d
AG
149}
150
55976b7f
HS
151/**
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 */
e275458c 166static efi_status_t EFIAPI efi_disk_read_blocks(struct efi_block_io *this,
4f94865b 167 u32 media_id, u64 lba, efi_uintn_t buffer_size,
2a22d05d
AG
168 void *buffer)
169{
51735ae0
AG
170 void *real_buffer = buffer;
171 efi_status_t r;
172
f59f0825
HS
173 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;
688e8825
HS
180 /* 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))
f59f0825
HS
183 return EFI_INVALID_PARAMETER;
184 if (lba * this->media->block_size + buffer_size >
e67beffe 185 (this->media->last_block + 1) * this->media->block_size)
f59f0825
HS
186 return EFI_INVALID_PARAMETER;
187
51735ae0
AG
188#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
dee37fc9 203 EFI_ENTRY("%p, %x, %llx, %zx, %p", this, media_id, lba,
51735ae0
AG
204 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);
2a22d05d
AG
214}
215
55976b7f
HS
216/**
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 */
e275458c 232static efi_status_t EFIAPI efi_disk_write_blocks(struct efi_block_io *this,
4f94865b 233 u32 media_id, u64 lba, efi_uintn_t buffer_size,
2a22d05d
AG
234 void *buffer)
235{
51735ae0
AG
236 void *real_buffer = buffer;
237 efi_status_t r;
238
f59f0825
HS
239 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;
688e8825
HS
248 /* 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))
f59f0825
HS
251 return EFI_INVALID_PARAMETER;
252 if (lba * this->media->block_size + buffer_size >
e67beffe 253 (this->media->last_block + 1) * this->media->block_size)
f59f0825
HS
254 return EFI_INVALID_PARAMETER;
255
51735ae0
AG
256#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
dee37fc9 271 EFI_ENTRY("%p, %x, %llx, %zx, %p", this, media_id, lba,
51735ae0
AG
272 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);
2a22d05d
AG
282}
283
55976b7f
HS
284/**
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 */
2a22d05d
AG
298static efi_status_t EFIAPI efi_disk_flush_blocks(struct efi_block_io *this)
299{
2a22d05d
AG
300 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
47a95963
HS
311/**
312 * efi_fs_from_path() - retrieve simple file system protocol
313 *
314 * Gets the simple file system protocol for a file device path.
110d80a1
HS
315 *
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 *
47a95963
HS
320 * @full_path: device path including device and file
321 * Return: simple file system protocol
2a92080d
RC
322 */
323struct efi_simple_file_system_protocol *
110d80a1 324efi_fs_from_path(struct efi_device_path *full_path)
2a92080d
RC
325{
326 struct efi_object *efiobj;
110d80a1
HS
327 struct efi_handler *handler;
328 struct efi_device_path *device_path;
329 struct efi_device_path *file_path;
330 efi_status_t ret;
2a92080d 331
110d80a1
HS
332 /* 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 */
e46ef1db 339 efiobj = efi_dp_find_obj(device_path, NULL, NULL);
110d80a1 340 efi_free_pool(device_path);
2a92080d
RC
341 if (!efiobj)
342 return NULL;
343
110d80a1
HS
344 /* 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;
2a92080d 349
110d80a1
HS
350 /* Return the simple file system protocol for the partition */
351 return handler->protocol_interface;
2a92080d
RC
352}
353
86740067
AT
354/**
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
55976b7f 375/**
47a95963 376 * efi_disk_add_dev() - create a handle for a partition or disk
93945f2c 377 *
47a95963
HS
378 * @parent: parent handle
379 * @dp_parent: parent device path
47a95963
HS
380 * @desc: internal block device
381 * @dev_index: device index for block device
8d0949b3 382 * @part_info: partition info
55976b7f
HS
383 * @part: partition
384 * @disk: pointer to receive the created handle
47a95963 385 * Return: disk object
93945f2c 386 */
df9cf561 387static efi_status_t efi_disk_add_dev(
64e4db0f
HS
388 efi_handle_t parent,
389 struct efi_device_path *dp_parent,
64e4db0f
HS
390 struct blk_desc *desc,
391 int dev_index,
8d0949b3 392 struct disk_partition *part_info,
df9cf561
HS
393 unsigned int part,
394 struct efi_disk_obj **disk)
4a12a97c
AG
395{
396 struct efi_disk_obj *diskobj;
0a87e05d 397 struct efi_object *handle;
b9b0ea30 398 const efi_guid_t *guid = NULL;
4b9f7aaf 399 efi_status_t ret;
4a12a97c 400
0812d1a0
AG
401 /* Don't add empty devices */
402 if (!desc->lba)
df9cf561 403 return EFI_NOT_READY;
0812d1a0 404
884bcf6f 405 diskobj = calloc(1, sizeof(*diskobj));
4b9f7aaf 406 if (!diskobj)
df9cf561 407 return EFI_OUT_OF_RESOURCES;
4b9f7aaf
HS
408
409 /* Hook up to the device list */
d39646a3 410 efi_add_handle(&diskobj->header);
4a12a97c
AG
411
412 /* Fill in object data */
8d0949b3 413 if (part_info) {
64e4db0f 414 struct efi_device_path *node = efi_dp_part_node(desc, part);
26448513
HS
415 struct efi_handler *handler;
416 void *protocol_interface;
417
418 /* Parent must expose EFI_BLOCK_IO_PROTOCOL */
419 ret = efi_search_protocol(parent, &efi_block_io_guid, &handler);
420 if (ret != EFI_SUCCESS)
421 goto error;
422
423 /*
424 * Link the partition (child controller) to the block device
425 * (controller).
426 */
427 ret = efi_protocol_open(handler, &protocol_interface, NULL,
428 &diskobj->header,
429 EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER);
430 if (ret != EFI_SUCCESS)
431 goto error;
64e4db0f
HS
432
433 diskobj->dp = efi_dp_append_node(dp_parent, node);
434 efi_free_pool(node);
8d0949b3 435 diskobj->media.last_block = part_info->size - 1;
b9b0ea30
HS
436 if (part_info->bootable & PART_EFI_SYSTEM_PARTITION)
437 guid = &efi_system_partition_guid;
64e4db0f
HS
438 } else {
439 diskobj->dp = efi_dp_from_part(desc, part);
8d0949b3 440 diskobj->media.last_block = desc->lba - 1;
64e4db0f 441 }
884bcf6f 442 diskobj->part = part;
0a87e05d
HS
443
444 /*
445 * Install the device path and the block IO protocol.
446 *
447 * InstallMultipleProtocolInterfaces() checks if the device path is
448 * already installed on an other handle and returns EFI_ALREADY_STARTED
449 * in this case.
450 */
451 handle = &diskobj->header;
452 ret = EFI_CALL(efi_install_multiple_protocol_interfaces(
453 &handle, &efi_guid_device_path, diskobj->dp,
b9b0ea30
HS
454 &efi_block_io_guid, &diskobj->ops,
455 guid, NULL, NULL));
4b9f7aaf 456 if (ret != EFI_SUCCESS)
cd9a26bf 457 goto error;
0a87e05d
HS
458
459 /*
460 * On partitions or whole disks without partitions install the
461 * simple file system protocol if a file system is available.
462 */
89cb6a5d
AT
463 if ((part || desc->part_type == PART_TYPE_UNKNOWN) &&
464 efi_fs_exists(desc, part)) {
2a92080d
RC
465 diskobj->volume = efi_simple_file_system(desc, part,
466 diskobj->dp);
d39646a3 467 ret = efi_add_protocol(&diskobj->header,
4b9f7aaf 468 &efi_simple_file_system_protocol_guid,
22de1de9 469 diskobj->volume);
4b9f7aaf 470 if (ret != EFI_SUCCESS)
df9cf561 471 return ret;
2a92080d 472 }
4a12a97c 473 diskobj->ops = block_io_disk_template;
4a12a97c
AG
474 diskobj->dev_index = dev_index;
475
476 /* Fill in EFI IO Media info (for read/write callbacks) */
477 diskobj->media.removable_media = desc->removable;
478 diskobj->media.media_present = 1;
f59f0825
HS
479 /*
480 * MediaID is just an arbitrary counter.
481 * We have to change it if the medium is removed or changed.
482 */
483 diskobj->media.media_id = 1;
4a12a97c
AG
484 diskobj->media.block_size = desc->blksz;
485 diskobj->media.io_align = desc->blksz;
9f888969 486 if (part)
5f770836 487 diskobj->media.logical_partition = 1;
4a12a97c 488 diskobj->ops.media = &diskobj->media;
df9cf561
HS
489 if (disk)
490 *disk = diskobj;
11078bb2 491
8d0949b3 492 EFI_PRINT("BlockIO: part %u, present %d, logical %d, removable %d"
7a85f324 493 ", last_block %llu\n",
8d0949b3
HS
494 diskobj->part,
495 diskobj->media.media_present,
496 diskobj->media.logical_partition,
497 diskobj->media.removable_media,
8d0949b3
HS
498 diskobj->media.last_block);
499
11078bb2
HS
500 /* Store first EFI system partition */
501 if (part && !efi_system_partition.if_type) {
b9b0ea30 502 if (part_info->bootable & PART_EFI_SYSTEM_PARTITION) {
11078bb2
HS
503 efi_system_partition.if_type = desc->if_type;
504 efi_system_partition.devnum = desc->devnum;
505 efi_system_partition.part = part;
3dca77b1 506 EFI_PRINT("EFI system partition: %s %x:%x\n",
11078bb2
HS
507 blk_get_if_type_name(desc->if_type),
508 desc->devnum, part);
509 }
510 }
df9cf561 511 return EFI_SUCCESS;
26448513
HS
512error:
513 efi_delete_handle(&diskobj->header);
514 return ret;
4a12a97c
AG
515}
516
a9bf024b
AT
517/*
518 * Create a handle for a whole raw disk
519 *
520 * @dev uclass device (UCLASS_BLK)
64e4db0f 521 *
a9bf024b
AT
522 * Create an efi_disk object which is associated with @dev.
523 * The type of @dev must be UCLASS_BLK.
47a95963 524 *
a9bf024b 525 * @return 0 on success, -1 otherwise
64e4db0f 526 */
a9bf024b 527static int efi_disk_create_raw(struct udevice *dev)
8c3df0bf 528{
a9bf024b
AT
529 struct efi_disk_obj *disk;
530 struct blk_desc *desc;
a9bf024b 531 int diskid;
64e4db0f 532 efi_status_t ret;
64e4db0f 533
a9bf024b 534 desc = dev_get_uclass_plat(dev);
a9bf024b
AT
535 diskid = desc->devnum;
536
ab31c8a1 537 ret = efi_disk_add_dev(NULL, NULL, desc,
a9bf024b
AT
538 diskid, NULL, 0, &disk);
539 if (ret != EFI_SUCCESS) {
540 if (ret == EFI_NOT_READY)
541 log_notice("Disk %s not ready\n", dev->name);
542 else
543 log_err("Adding disk for %s failed\n", dev->name);
544
545 return -1;
546 }
ee576662 547 if (efi_link_dev(&disk->header, dev)) {
a9bf024b
AT
548 efi_free_pool(disk->dp);
549 efi_delete_handle(&disk->header);
550
551 return -1;
8c3df0bf 552 }
884bcf6f 553
a9bf024b 554 return 0;
8c3df0bf
AG
555}
556
a9bf024b
AT
557/*
558 * Create a handle for a disk partition
2a22d05d 559 *
a9bf024b 560 * @dev uclass device (UCLASS_PARTITION)
47a95963 561 *
a9bf024b
AT
562 * Create an efi_disk object which is associated with @dev.
563 * The type of @dev must be UCLASS_PARTITION.
487d756f 564 *
a9bf024b 565 * @return 0 on success, -1 otherwise
2a22d05d 566 */
a9bf024b 567static int efi_disk_create_part(struct udevice *dev)
2a22d05d 568{
a9bf024b
AT
569 efi_handle_t parent;
570 struct blk_desc *desc;
a9bf024b
AT
571 struct disk_part *part_data;
572 struct disk_partition *info;
573 unsigned int part;
574 int diskid;
575 struct efi_handler *handler;
576 struct efi_device_path *dp_parent;
64e4db0f 577 struct efi_disk_obj *disk;
df9cf561 578 efi_status_t ret;
a9bf024b
AT
579
580 if (dev_tag_get_ptr(dev_get_parent(dev), DM_TAG_EFI, (void **)&parent))
581 return -1;
582
583 desc = dev_get_uclass_plat(dev_get_parent(dev));
a9bf024b
AT
584 diskid = desc->devnum;
585
586 part_data = dev_get_uclass_plat(dev);
587 part = part_data->partnum;
588 info = &part_data->gpt_part_info;
589
590 ret = efi_search_protocol(parent, &efi_guid_device_path, &handler);
591 if (ret != EFI_SUCCESS)
592 return -1;
593 dp_parent = (struct efi_device_path *)handler->protocol_interface;
594
ab31c8a1 595 ret = efi_disk_add_dev(parent, dp_parent, desc, diskid,
a9bf024b
AT
596 info, part, &disk);
597 if (ret != EFI_SUCCESS) {
598 log_err("Adding partition for %s failed\n", dev->name);
599 return -1;
600 }
ee576662 601 if (efi_link_dev(&disk->header, dev)) {
a9bf024b
AT
602 efi_free_pool(disk->dp);
603 efi_delete_handle(&disk->header);
604
605 return -1;
606 }
607
608 return 0;
609}
610
611/*
612 * Create efi_disk objects for a block device
613 *
614 * @dev uclass device (UCLASS_BLK)
615 *
616 * Create efi_disk objects for partitions as well as a raw disk
617 * which is associated with @dev.
618 * The type of @dev must be UCLASS_BLK.
619 * This function is expected to be called at EV_PM_POST_PROBE.
620 *
621 * @return 0 on success, -1 otherwise
622 */
623static int efi_disk_probe(void *ctx, struct event *event)
624{
487d756f 625 struct udevice *dev;
a9bf024b 626 enum uclass_id id;
3c809dfe 627 struct blk_desc *desc;
a9bf024b
AT
628 struct udevice *child;
629 int ret;
487d756f 630
a9bf024b
AT
631 dev = event->data.dm.dev;
632 id = device_get_uclass_id(dev);
487d756f 633
a9bf024b
AT
634 /* TODO: We won't support partitions in a partition */
635 if (id != UCLASS_BLK)
636 return 0;
637
3c809dfe
AT
638 /*
639 * avoid creating duplicated objects now that efi_driver
640 * has already created an efi_disk at this moment.
641 */
642 desc = dev_get_uclass_plat(dev);
e33a5c6b 643 if (desc->if_type != UCLASS_EFI_LOADER) {
3c809dfe
AT
644 ret = efi_disk_create_raw(dev);
645 if (ret)
646 return -1;
647 }
487d756f 648
a9bf024b
AT
649 device_foreach_child(child, dev) {
650 ret = efi_disk_create_part(child);
651 if (ret)
652 return -1;
487d756f 653 }
2a22d05d 654
a9bf024b
AT
655 return 0;
656}
657
b406eb04
AT
658/*
659 * Delete an efi_disk object for a whole raw disk
660 *
661 * @dev uclass device (UCLASS_BLK)
662 *
663 * Delete an efi_disk object which is associated with @dev.
664 * The type of @dev must be UCLASS_BLK.
665 *
666 * @return 0 on success, -1 otherwise
667 */
668static int efi_disk_delete_raw(struct udevice *dev)
669{
670 efi_handle_t handle;
a3cb34e9 671 struct blk_desc *desc;
b406eb04
AT
672 struct efi_disk_obj *diskobj;
673
674 if (dev_tag_get_ptr(dev, DM_TAG_EFI, (void **)&handle))
675 return -1;
676
a3cb34e9 677 desc = dev_get_uclass_plat(dev);
e33a5c6b 678 if (desc->if_type != UCLASS_EFI_LOADER) {
a3cb34e9
AT
679 diskobj = container_of(handle, struct efi_disk_obj, header);
680 efi_free_pool(diskobj->dp);
681 }
b406eb04
AT
682
683 efi_delete_handle(handle);
684 dev_tag_del(dev, DM_TAG_EFI);
685
686 return 0;
687}
688
689/*
690 * Delete an efi_disk object for a disk partition
691 *
692 * @dev uclass device (UCLASS_PARTITION)
693 *
694 * Delete an efi_disk object which is associated with @dev.
695 * The type of @dev must be UCLASS_PARTITION.
696 *
697 * @return 0 on success, -1 otherwise
698 */
699static int efi_disk_delete_part(struct udevice *dev)
700{
701 efi_handle_t handle;
702 struct efi_disk_obj *diskobj;
703
704 if (dev_tag_get_ptr(dev, DM_TAG_EFI, (void **)&handle))
705 return -1;
706
707 diskobj = container_of(handle, struct efi_disk_obj, header);
708
709 efi_free_pool(diskobj->dp);
710 efi_delete_handle(handle);
711 dev_tag_del(dev, DM_TAG_EFI);
712
713 return 0;
714}
715
716/*
717 * Delete an efi_disk object for a block device
718 *
719 * @dev uclass device (UCLASS_BLK or UCLASS_PARTITION)
720 *
721 * Delete an efi_disk object which is associated with @dev.
722 * The type of @dev must be either UCLASS_BLK or UCLASS_PARTITION.
723 * This function is expected to be called at EV_PM_PRE_REMOVE.
724 *
725 * @return 0 on success, -1 otherwise
726 */
727static int efi_disk_remove(void *ctx, struct event *event)
728{
729 enum uclass_id id;
730 struct udevice *dev;
731
732 dev = event->data.dm.dev;
733 id = device_get_uclass_id(dev);
734
735 if (id == UCLASS_BLK)
736 return efi_disk_delete_raw(dev);
737 else if (id == UCLASS_PARTITION)
738 return efi_disk_delete_part(dev);
739 else
740 return 0;
741}
742
a9bf024b
AT
743efi_status_t efi_disk_init(void)
744{
745 int ret;
746
747 ret = event_register("efi_disk add", EVT_DM_POST_PROBE,
748 efi_disk_probe, NULL);
749 if (ret) {
750 log_err("Event registration for efi_disk add failed\n");
751 return EFI_OUT_OF_RESOURCES;
752 }
2a22d05d 753
b406eb04
AT
754 ret = event_register("efi_disk del", EVT_DM_PRE_REMOVE,
755 efi_disk_remove, NULL);
756 if (ret) {
757 log_err("Event registration for efi_disk del failed\n");
758 return EFI_OUT_OF_RESOURCES;
759 }
760
df9cf561 761 return EFI_SUCCESS;
2a22d05d 762}
87d79142
MK
763
764/**
765 * efi_disk_get_device_name() - get U-Boot device name associated with EFI handle
766 *
767 * @handle: pointer to the EFI handle
768 * @buf: pointer to the buffer to store the string
769 * @size: size of buffer
770 * Return: status code
771 */
772efi_status_t efi_disk_get_device_name(const efi_handle_t handle, char *buf, int size)
773{
774 int count;
775 int diskid;
776 enum uclass_id id;
777 unsigned int part;
778 struct udevice *dev;
779 struct blk_desc *desc;
780 const char *if_typename;
781 bool is_partition = false;
782 struct disk_part *part_data;
783
784 if (!handle || !buf || !size)
785 return EFI_INVALID_PARAMETER;
786
787 dev = handle->dev;
788 id = device_get_uclass_id(dev);
789 if (id == UCLASS_BLK) {
790 desc = dev_get_uclass_plat(dev);
791 } else if (id == UCLASS_PARTITION) {
792 desc = dev_get_uclass_plat(dev_get_parent(dev));
793 is_partition = true;
794 } else {
795 return EFI_INVALID_PARAMETER;
796 }
797 if_typename = blk_get_if_type_name(desc->if_type);
798 diskid = desc->devnum;
799
800 if (is_partition) {
801 part_data = dev_get_uclass_plat(dev);
802 part = part_data->partnum;
803 count = snprintf(buf, size, "%s %d:%d", if_typename, diskid, part);
804 } else {
805 count = snprintf(buf, size, "%s %d", if_typename, diskid);
806 }
807
808 if (count < 0 || (count + 1) > size)
809 return EFI_INVALID_PARAMETER;
810
811 return EFI_SUCCESS;
812}
This page took 0.299033 seconds and 4 git commands to generate.