KumiSystems/openwrtv4
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

oseama: support extracting entity

Browse source 
Signed-off-by: Rafał Miłecki <zajec5@gmail.com>
This commit is contained in:
Rafał Miłecki 2016-06-10 14:38:04 +02:00
parent cb0de9a68e
commit 399d214d05
1 changed files with 134 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

134
package/utils/oseama/src/oseama.c
View 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();