Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 | // SPDX-License-Identifier: GPL-2.0+ /* * Aurora Innovation, Inc. Copyright 2022. * */ #include <blk.h> #include <config.h> #include <command.h> #include <dm.h> #include <env.h> #include <part.h> #if defined(CONFIG_EFI_CLIENT) || defined(CONFIG_EFI_APP) #include <efi.h> #include <efi_api.h> static bool partition_is_on_device(const struct efi_device_path *device, const struct efi_device_path *part, __u32 *part_no) { size_t d_len, p_len; const struct efi_device_path *p, *d; for (d = device; d->type != DEVICE_PATH_TYPE_END; d = (void *)d + d->length) { } d_len = (void *)d - (void *)device; for (p = part; p->type != DEVICE_PATH_TYPE_END && !(p->type == DEVICE_PATH_TYPE_MEDIA_DEVICE && p->sub_type == DEVICE_PATH_SUB_TYPE_HARD_DRIVE_PATH); p = (void *)p + p->length) { } if (p->type != DEVICE_PATH_TYPE_MEDIA_DEVICE || p->sub_type != DEVICE_PATH_SUB_TYPE_HARD_DRIVE_PATH) { // Not a partition. return false; } p_len = (void *)p - (void *)part; if (p_len == d_len && !memcmp(device, part, p_len)) { if (part_no) *part_no = ((__u32 *)p)[1]; return true; } return false; } #endif static int part_find(int argc, char *const argv[]) { #if defined(CONFIG_EFI_CLIENT) || defined(CONFIG_EFI_APP) efi_guid_t efi_devpath_guid = EFI_DEVICE_PATH_PROTOCOL_GUID; struct efi_device_path *loaded_image_path = NULL; struct efi_boot_services *boot = efi_get_boot(); struct efi_priv *priv = efi_get_priv(); bool part_self = false; #endif struct driver *d = ll_entry_start(struct driver, driver); const int n_ents = ll_entry_count(struct driver, driver); struct disk_partition info; struct blk_desc *desc; struct driver *entry; struct udevice *udev; struct uclass *uc; int ret; if (argc != 2) return CMD_RET_USAGE; #if defined(CONFIG_EFI_CLIENT) || defined(CONFIG_EFI_APP) part_self = !strncmp(argv[1], "self", 6); if (part_self) { ret = boot->handle_protocol(priv->loaded_image->device_handle, &efi_devpath_guid, (void **)&loaded_image_path); if (ret) log_warning("failed to get device path for loaded image (ret=%d)", ret); } #endif ret = uclass_get(UCLASS_BLK, &uc); if (ret) { puts("Could not get BLK uclass.\n"); return CMD_RET_FAILURE; } for (entry = d; entry < d + n_ents; entry++) { if (entry->id != UCLASS_BLK) continue; uclass_foreach_dev(udev, uc) { int i; if (udev->driver != entry) continue; desc = dev_get_uclass_plat(udev); #if defined(CONFIG_EFI_CLIENT) || defined(CONFIG_EFI_APP) if (part_self) { if (desc->if_type == IF_TYPE_EFI_MEDIA) { struct efi_media_plat *plat = dev_get_plat(udev->parent); __u32 loader_part_no; if (partition_is_on_device(plat->device_path, loaded_image_path, &loader_part_no)) { char env[256]; ret = snprintf(env, sizeof(env), "%s %d:%d", blk_get_if_type_name(desc->if_type), desc->devnum, loader_part_no); if (ret < 0 || ret == sizeof(env)) return CMD_RET_FAILURE; if (env_set("target_part", env)) return CMD_RET_FAILURE; return CMD_RET_SUCCESS; } } } else { #endif for (i = 1; i <= MAX_SEARCH_PARTITIONS; i++) { ret = part_get_info(desc, i, &info); if (ret) break; if (strcasecmp(argv[1], info.type_guid) == 0) { char env[256]; ret = snprintf(env, sizeof(env), "%s %d:%d", blk_get_if_type_name(desc->if_type), desc->devnum, i); if (ret < 0 || ret == sizeof(env)) return CMD_RET_FAILURE; env_set("target_part", env); debug("Setting target_part to %s\n", env); return CMD_RET_SUCCESS; } } #if defined(CONFIG_EFI_CLIENT) || defined(CONFIG_EFI_APP) } #endif } } return CMD_RET_FAILURE; } static int do_part_find(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) { return part_find(argc, argv); } U_BOOT_CMD( part_find, 2, 0, do_part_find, "Find a partition", "<guid>\n" "- Examine the list of known partitions for one that has a type\n" " GUID that matches 'guid', expressed in the standard text format.\n" " If successful, the target_part environment variable will be set\n" " to the corresponding 'interface dev:part'.\n" ); |