tools: add mkeficapsule command for UEFI capsule update

This is a utility mainly for test purpose.
  mkeficapsule -f: create a test capsule file for FIT image firmware

Having said that, you will be able to customize the code to fit
your specific requirements for your platform.

Signed-off-by: AKASHI Takahiro <takahiro.akashi@linaro.org>

diff --git a/tools/Makefile b/tools/Makefile
index 51123fd..66d9376 100644 (file)
--- a/tools/Makefile
+++ b/tools/Makefile
@@ -218,6 +218,8 @@ hostprogs-$(CONFIG_MIPS) += mips-relocs
 hostprogs-$(CONFIG_ASN1_COMPILER)      += asn1_compiler
 HOSTCFLAGS_asn1_compiler.o = -idirafter $(srctree)/include
 
+hostprogs-$(CONFIG_EFI_HAVE_CAPSULE_SUPPORT) += mkeficapsule
+
 # We build some files with extra pedantic flags to try to minimize things
 # that won't build on some weird host compiler -- though there are lots of
 # exceptions for files that aren't complaint.

diff --git a/tools/mkeficapsule.c b/tools/mkeficapsule.c
new file mode 100644 (file)
index 0000000..426038e
--- /dev/null
+++ b/tools/mkeficapsule.c
@@ -0,0 +1,236 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * Copyright 2018 Linaro Limited
+ *             Author: AKASHI Takahiro
+ */
+
+#include <getopt.h>
+#include <malloc.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <linux/types.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+typedef __u8 u8;
+typedef __u16 u16;
+typedef __u32 u32;
+typedef __u64 u64;
+typedef __s16 s16;
+typedef __s32 s32;
+
+#define aligned_u64 __aligned_u64
+
+#ifndef __packed
+#define __packed __attribute__((packed))
+#endif
+
+#include <efi.h>
+#include <efi_api.h>
+
+static const char *tool_name = "mkeficapsule";
+
+efi_guid_t efi_guid_fm_capsule = EFI_FIRMWARE_MANAGEMENT_CAPSULE_ID_GUID;
+efi_guid_t efi_guid_image_type_uboot_fit =
+               EFI_FIRMWARE_IMAGE_TYPE_UBOOT_FIT_GUID;
+efi_guid_t efi_guid_image_type_uboot_raw =
+               EFI_FIRMWARE_IMAGE_TYPE_UBOOT_RAW_GUID;
+
+static struct option options[] = {
+       {"fit", required_argument, NULL, 'f'},
+       {"raw", required_argument, NULL, 'r'},
+       {"index", required_argument, NULL, 'i'},
+       {"instance", required_argument, NULL, 'I'},
+       {"help", no_argument, NULL, 'h'},
+       {NULL, 0, NULL, 0},
+};
+
+static void print_usage(void)
+{
+       printf("Usage: %s [options] <output file>\n"
+              "Options:\n"
+              "\t--fit <fit image>      new FIT image file\n"
+              "\t--raw <raw image>      new raw image file\n"
+              "\t--index <index>        update image index\n"
+              "\t--instance <instance>  update hardware instance\n"
+              "\t--help                 print a help message\n",
+              tool_name);
+}
+
+static int create_fwbin(char *path, char *bin, efi_guid_t *guid,
+                       unsigned long index, unsigned long instance)
+{
+       struct efi_capsule_header header;
+       struct efi_firmware_management_capsule_header capsule;
+       struct efi_firmware_management_capsule_image_header image;
+       FILE *f, *g;
+       struct stat bin_stat;
+       u8 *data;
+       size_t size;
+       u64 offset;
+
+#ifdef DEBUG
+       printf("For output: %s\n", path);
+       printf("\tbin: %s\n\ttype: %pUl\n" bin, guid);
+       printf("\tindex: %ld\n\tinstance: %ld\n", index, instance);
+#endif
+
+       g = fopen(bin, "r");
+       if (!g) {
+               printf("cannot open %s\n", bin);
+               return -1;
+       }
+       if (stat(bin, &bin_stat) < 0) {
+               printf("cannot determine the size of %s\n", bin);
+               goto err_1;
+       }
+       data = malloc(bin_stat.st_size);
+       if (!data) {
+               printf("cannot allocate memory: %lx\n", bin_stat.st_size);
+               goto err_1;
+       }
+       f = fopen(path, "w");
+       if (!f) {
+               printf("cannot open %s\n", path);
+               goto err_2;
+       }
+       header.capsule_guid = efi_guid_fm_capsule;
+       header.header_size = sizeof(header);
+       header.flags = CAPSULE_FLAGS_PERSIST_ACROSS_RESET; /* TODO */
+       header.capsule_image_size = sizeof(header)
+                                       + sizeof(capsule) + sizeof(u64)
+                                       + sizeof(image)
+                                       + bin_stat.st_size;
+
+       size = fwrite(&header, 1, sizeof(header), f);
+       if (size < sizeof(header)) {
+               printf("write failed (%lx)\n", size);
+               goto err_3;
+       }
+
+       capsule.version = 0x00000001;
+       capsule.embedded_driver_count = 0;
+       capsule.payload_item_count = 1;
+       size = fwrite(&capsule, 1, sizeof(capsule), f);
+       if (size < (sizeof(capsule))) {
+               printf("write failed (%lx)\n", size);
+               goto err_3;
+       }
+       offset = sizeof(capsule) + sizeof(u64);
+       size = fwrite(&offset, 1, sizeof(offset), f);
+       if (size < sizeof(offset)) {
+               printf("write failed (%lx)\n", size);
+               goto err_3;
+       }
+
+       image.version = 0x00000003;
+       memcpy(&image.update_image_type_id, guid, sizeof(*guid));
+       image.update_image_index = index;
+       image.update_image_size = bin_stat.st_size;
+       image.update_vendor_code_size = 0; /* none */
+       image.update_hardware_instance = instance;
+       image.image_capsule_support = 0;
+
+       size = fwrite(&image, 1, sizeof(image), f);
+       if (size < sizeof(image)) {
+               printf("write failed (%lx)\n", size);
+               goto err_3;
+       }
+       size = fread(data, 1, bin_stat.st_size, g);
+       if (size < bin_stat.st_size) {
+               printf("read failed (%lx)\n", size);
+               goto err_3;
+       }
+       size = fwrite(data, 1, bin_stat.st_size, f);
+       if (size < bin_stat.st_size) {
+               printf("write failed (%lx)\n", size);
+               goto err_3;
+       }
+
+       fclose(f);
+       fclose(g);
+       free(data);
+
+       return 0;
+
+err_3:
+       fclose(f);
+err_2:
+       free(data);
+err_1:
+       fclose(g);
+
+       return -1;
+}
+
+/*
+ * Usage:
+ *   $ mkeficapsule -f <firmware binary> <output file>
+ */
+int main(int argc, char **argv)
+{
+       char *file;
+       efi_guid_t *guid;
+       unsigned long index, instance;
+       int c, idx;
+
+       file = NULL;
+       guid = NULL;
+       index = 0;
+       instance = 0;
+       for (;;) {
+               c = getopt_long(argc, argv, "f:r:i:I:v:h", options, &idx);
+               if (c == -1)
+                       break;
+
+               switch (c) {
+               case 'f':
+                       if (file) {
+                               printf("Image already specified\n");
+                               return -1;
+                       }
+                       file = optarg;
+                       guid = &efi_guid_image_type_uboot_fit;
+                       break;
+               case 'r':
+                       if (file) {
+                               printf("Image already specified\n");
+                               return -1;
+                       }
+                       file = optarg;
+                       guid = &efi_guid_image_type_uboot_raw;
+                       break;
+               case 'i':
+                       index = strtoul(optarg, NULL, 0);
+                       break;
+               case 'I':
+                       instance = strtoul(optarg, NULL, 0);
+                       break;
+               case 'h':
+                       print_usage();
+                       return 0;
+               }
+       }
+
+       /* need a output file */
+       if (argc != optind + 1) {
+               print_usage();
+               return -1;
+       }
+
+       /* need a fit image file or raw image file */
+       if (!file) {
+               print_usage();
+               return -1;
+       }
+
+       if (create_fwbin(argv[optind], file, guid, index, instance)
+                       < 0) {
+               printf("Creating firmware capsule failed\n");
+               return -1;
+       }
+
+       return 0;
+}