firmware-utils: oseama: support extracting entity
authorRafał Miłecki <zajec5@gmail.com>
Thu, 5 May 2016 12:32:00 +0000 (14:32 +0200)
committerRafał Miłecki <zajec5@gmail.com>
Wed, 15 Jun 2016 08:54:16 +0000 (10:54 +0200)
This updates oseama in firmware-utils to the version from packages
keeping them synced.

Signed-off-by: Rafał Miłecki <zajec5@gmail.com>
tools/firmware-utils/src/oseama.c

index 37a2171..4434b11 100644 (file)
@@ -57,6 +57,7 @@ struct seama_entity_header {
 
 char *seama_path;
 int entity_idx = -1;
+char *out_path;
 
 static inline size_t oseama_min(size_t x, size_t y) {
        return x < y ? x : y;
@@ -391,6 +392,132 @@ out:
        return err;
 }
 
+/**************************************************
+ * Extract
+ **************************************************/
+
+static void oseama_extract_parse_options(int argc, char **argv) {
+       int c;
+
+       while ((c = getopt(argc, argv, "e:o:")) != -1) {
+               switch (c) {
+               case 'e':
+                       entity_idx = atoi(optarg);
+                       break;
+               case 'o':
+                       out_path = optarg;
+                       break;
+               }
+       }
+}
+
+static int oseama_extract_entity(FILE *seama, FILE *out) {
+       struct seama_entity_header hdr;
+       size_t bytes, metasize, imagesize, length;
+       uint8_t buf[1024];
+       int i = 0;
+       int err = 0;
+
+       while ((bytes = fread(&hdr, 1, sizeof(hdr), seama)) == sizeof(hdr)) {
+               if (be32_to_cpu(hdr.magic) != SEAMA_MAGIC) {
+                       fprintf(stderr, "Invalid Seama magic: 0x%08x\n", be32_to_cpu(hdr.magic));
+                       err =  -EINVAL;
+                       break;
+               }
+               metasize = be16_to_cpu(hdr.metasize);
+               imagesize = be32_to_cpu(hdr.imagesize);
+
+               if (i != entity_idx) {
+                       fseek(seama, metasize + imagesize, SEEK_CUR);
+                       i++;
+                       continue;
+               }
+
+               fseek(seama, -sizeof(hdr), SEEK_CUR);
+
+               length = sizeof(hdr) + metasize + imagesize;
+               while ((bytes = fread(buf, 1, oseama_min(sizeof(buf), length), seama)) > 0) {
+                       if (fwrite(buf, 1, bytes, out) != bytes) {
+                               fprintf(stderr, "Couldn't write %zu B to %s\n", bytes, out_path);
+                               err = -EIO;
+                               break;
+                       }
+                       length -= bytes;
+               }
+
+               if (length) {
+                       fprintf(stderr, "Couldn't extract whole entity %d from %s (%zu B left)\n", entity_idx, seama_path, length);
+                       err = -EIO;
+                       break;
+               }
+
+               break;
+       }
+
+       return err;
+}
+
+static int oseama_extract(int argc, char **argv) {
+       FILE *seama;
+       FILE *out;
+       struct seama_seal_header hdr;
+       size_t bytes;
+       uint16_t metasize;
+       int err = 0;
+
+       if (argc < 3) {
+               fprintf(stderr, "No Seama file passed\n");
+               err = -EINVAL;
+               goto out;
+       }
+       seama_path = argv[2];
+
+       optind = 3;
+       oseama_extract_parse_options(argc, argv);
+       if (entity_idx < 0) {
+               fprintf(stderr, "No entity specified\n");
+               err = -EINVAL;
+               goto out;
+       } else if (!out_path) {
+               fprintf(stderr, "No output file specified\n");
+               err = -EINVAL;
+               goto out;
+       }
+
+       seama = fopen(seama_path, "r");
+       if (!seama) {
+               fprintf(stderr, "Couldn't open %s\n", seama_path);
+               err = -EACCES;
+               goto out;
+       }
+
+       out = fopen(out_path, "w");
+       if (!out) {
+               fprintf(stderr, "Couldn't open %s\n", out_path);
+               err = -EACCES;
+               goto err_close_seama;
+       }
+
+       bytes = fread(&hdr, 1, sizeof(hdr), seama);
+       if (bytes != sizeof(hdr)) {
+               fprintf(stderr, "Couldn't read %s header\n", seama_path);
+               err =  -EIO;
+               goto err_close_out;
+       }
+       metasize = be16_to_cpu(hdr.metasize);
+
+       fseek(seama, metasize, SEEK_CUR);
+
+       oseama_extract_entity(seama, out);
+
+err_close_out:
+       fclose(out);
+err_close_seama:
+       fclose(seama);
+out:
+       return err;
+}
+
 /**************************************************
  * Start
  **************************************************/
@@ -407,6 +534,11 @@ static void usage() {
        printf("\t-m meta\t\t\t\tmeta into to put in header\n");
        printf("\t-f file\t\t\t\tappend content from file\n");
        printf("\t-b offset\t\t\tappend zeros till reaching absolute offset\n");
+       printf("\n");
+       printf("Extract from Seama seal (container):\n");
+       printf("\toseama extract <file> [options]\n");
+       printf("\t-e\t\t\t\tindex of entity to extract\n");
+       printf("\t-o file\t\t\t\toutput file\n");
 }
 
 int main(int argc, char **argv) {
@@ -415,6 +547,8 @@ int main(int argc, char **argv) {
                        return oseama_info(argc, argv);
                else if (!strcmp(argv[1], "entity"))
                        return oseama_entity(argc, argv);
+               else if (!strcmp(argv[1], "extract"))
+                       return oseama_extract(argc, argv);
        }
 
        usage();