mirror of https://github.com/hak5/openwrt.git
oseama: support extracting entity
Signed-off-by: Rafał Miłecki <zajec5@gmail.com>lede-17.01
Rafał Miłecki
parent
cb0de9a68e
commit
399d214d05
|
|
@ -57,6 +57,7 @@ struct seama_entity_header {
|
||||||
|
|
||||||
char *seama_path;
|
char *seama_path;
|
||||||
int entity_idx = -1;
|
int entity_idx = -1;
|
||||||
|
char *out_path;
|
||||||
|
|
||||||
static inline size_t oseama_min(size_t x, size_t y) {
|
static inline size_t oseama_min(size_t x, size_t y) {
|
||||||
return x < y ? x : y;
|
return x < y ? x : y;
|
||||||
|
|
@ -391,6 +392,132 @@ out:
|
||||||
return err;
|
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
|
* Start
|
||||||
**************************************************/
|
**************************************************/
|
||||||
|
|
@ -407,6 +534,11 @@ static void usage() {
|
||||||
printf("\t-m meta\t\t\t\tmeta into to put in header\n");
|
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-f file\t\t\t\tappend content from file\n");
|
||||||
printf("\t-b offset\t\t\tappend zeros till reaching absolute offset\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) {
|
int main(int argc, char **argv) {
|
||||||
|
|
@ -415,6 +547,8 @@ int main(int argc, char **argv) {
|
||||||
return oseama_info(argc, argv);
|
return oseama_info(argc, argv);
|
||||||
else if (!strcmp(argv[1], "entity"))
|
else if (!strcmp(argv[1], "entity"))
|
||||||
return oseama_entity(argc, argv);
|
return oseama_entity(argc, argv);
|
||||||
|
else if (!strcmp(argv[1], "extract"))
|
||||||
|
return oseama_extract(argc, argv);
|
||||||
}
|
}
|
||||||
|
|
||||||
usage();
|
usage();
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue