]>
Commit | Line | Data |
---|---|---|
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 |
22 | struct efi_system_partition efi_system_partition; |
23 | ||
dec88e41 | 24 | const efi_guid_t efi_block_io_guid = EFI_BLOCK_IO_PROTOCOL_GUID; |
b9b0ea30 | 25 | const 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 | 39 | struct 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 |
64 | static 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 | */ | |
80 | bool 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 |
98 | enum efi_disk_direction { |
99 | EFI_DISK_READ, | |
100 | EFI_DISK_WRITE, | |
101 | }; | |
102 | ||
a80a232e | 103 | static 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 | 166 | static 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 | 232 | static 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 |
298 | static 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 | ||
304 | static 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 | */ |
323 | struct efi_simple_file_system_protocol * | |
110d80a1 | 324 | efi_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 | */ | |
362 | static 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 | 387 | static 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 |
512 | error: |
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 | 527 | static 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 | 567 | static 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 | */ | |
623 | static 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 | */ | |
668 | static 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 | */ | |
699 | static 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 | */ | |
727 | static 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 |
743 | efi_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 | */ | |
772 | efi_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 | } |