diff options
Diffstat (limited to 'tools/firmware-utils/src/seama.c')
-rw-r--r-- | tools/firmware-utils/src/seama.c | 499 |
1 files changed, 0 insertions, 499 deletions
diff --git a/tools/firmware-utils/src/seama.c b/tools/firmware-utils/src/seama.c deleted file mode 100644 index 89eeac90c4..0000000000 --- a/tools/firmware-utils/src/seama.c +++ /dev/null @@ -1,499 +0,0 @@ -// SPDX-License-Identifier: BSD-3-Clause -/* vi: set sw=4 ts=4: */ -/* - * Copyright (C) 2008, Alpha Networks, Inc. - * Created by David Hsieh <david_hsieh@alphanetworks.com> - * All right reserved. - * - * (SEA)ttle i(MA)ge is the image which used in project seattle. - */ - -#include <stdio.h> -#include <stdint.h> -#include <stdlib.h> -#include <stdarg.h> -#include <sys/types.h> -#include <sys/stat.h> -#include <unistd.h> -#include <string.h> -#include <arpa/inet.h> - -#include "md5.h" -#include "seama.h" - -#define PROGNAME "seama" -#define VERSION "0.20" -#define MAX_SEAMA_META_SIZE 1024 -#define MAX_META 128 -#define MAX_IMAGE 128 - -extern int optind; -extern char * optarg; - -static int o_verbose = 0; /* verbose mode. */ -static char * o_dump = NULL; /* Seama file to dump. */ -static char * o_seal = NULL; /* Seal the input images when file name exist. */ -static char * o_extract = NULL; /* Extract the seama file. */ -static char * o_images[MAX_IMAGE];/* The image files to pack or seal */ -static int o_isize = 0; /* number of images */ -static char * o_meta[MAX_META]; /* meta data array */ -static int o_msize = 0; /* size of meta array */ - -static void verbose(const char * format, ...) -{ - va_list marker; - if (o_verbose) - { - va_start(marker, format); - vfprintf(stdout, format, marker); - va_end(marker); - } -} - -static void cleanup_exit(int exit_code) -{ - verbose("%s: exit with code %d\n", PROGNAME, exit_code); - exit(exit_code); -} - -static void show_usage(int exit_code) -{ - printf( PROGNAME " version " VERSION "\n" - "usage: " PROGNAME " [OPTIONS]\n" - " -h show this help message.\n" - " -v verbose mode.\n" - " -m {META data} META data.\n" - " -d {file} dump the info of the seama file.\n" - " -i {input file} image file name.\n" - " -s {file} Seal the images to the seama file.\n" - " -x {seama file} Extract the seama file.\n" - "\n" - " SEAMA can pack the input file (with -i) into a seama file.\n" - " ex: seama -i target.file\n" - " SEAMA can also seal multiple seama files into a single seama file.\n" - " ex: seama -s final.file -i taget1.seama -i target2.seama\n" - " To extract the raw image from SEAMA, you need to specify the meta.\n" - " The first image match the specified meta will be extract to\n" - " the output file which was specified with '-x'.\n" - " ex: seama -x output -i seama.image -m file=sealpac\n" - ); - cleanup_exit(exit_code); -} - -static int parse_args(int argc, char * argv[]) -{ - int opt; - - while ((opt = getopt(argc, argv, "hvd:s:i:m:x:")) > 0) - { - switch (opt) - { - default: show_usage(-1); break; - case 'h': show_usage(0); break; - case 'v': o_verbose++; break; - case 'd': o_dump = optarg; break; - case 's': o_seal = optarg; break; - case 'x': o_extract = optarg; break; - case 'i': - if (o_isize < MAX_IMAGE) o_images[o_isize++] = optarg; - else printf("Exceed the maximum acceptable image files.!\n"); - break; - case 'm': - if (o_msize < MAX_META) o_meta[o_msize++] = optarg; - else printf("Exceed the maximum acceptable META data.!\n"); - break; - } - } - return 0; -} - -/*******************************************************************/ - -static size_t calculate_digest(FILE * fh, size_t size, uint8_t * digest) -{ - MD5_CTX ctx; - size_t bytes_left, bytes_read, i; - uint8_t buf[MAX_SEAMA_META_SIZE]; - - bytes_left = size ? size : sizeof(buf); - bytes_read = 0; - - MD5_Init(&ctx); - while (!feof(fh) && !ferror(fh) && bytes_left > 0) - { - i = bytes_left < sizeof(buf) ? bytes_left : sizeof(buf); - i = fread(buf, sizeof(char), i, fh); - if (i > 0) - { - MD5_Update(&ctx, buf, i); - bytes_read += i; - } - if (size) bytes_left -= i; - } - MD5_Final(digest, &ctx); - return bytes_read; -} - -#define READ_BUFF_SIZE 8*1024 -static size_t copy_file(FILE * to, FILE * from) -{ - size_t i, fsize = 0; - uint8_t buf[READ_BUFF_SIZE]; - - while (!feof(from) && !ferror(from)) - { - i = fread(buf, sizeof(uint8_t), READ_BUFF_SIZE, from); - if (i > 0) - { - fsize += i; - fwrite(buf, sizeof(uint8_t), i, to); - } - } - return fsize; -} - -static int verify_seama(const char * fname, int msg) -{ - FILE * fh = NULL; - struct stat st; - seamahdr_t shdr; - uint8_t checksum[16]; - uint8_t digest[16]; - uint8_t buf[MAX_SEAMA_META_SIZE]; - size_t msize, isize, i; - int ret = -1; - -#define ERRBREAK(fmt, args...) { if (msg) printf(fmt, ##args); break; } - - do - { - if (stat(fname, &st) < 0) ERRBREAK("Unable to get the info of '%s'\n",fname); - if ((fh = fopen(fname, "r+"))==NULL) ERRBREAK("Unable to open '%s' for reading!\n",fname); - - /* Dump SEAMA header */ - if (msg) printf("FILE - %s (%d bytes)\n", fname, (int)st.st_size); - - /* SEAMA */ - while (!feof(fh) && !ferror(fh)) - { - /* read header */ - if (fread(&shdr, sizeof(shdr), 1, fh) != 1) break; - - /* Check the magic number */ - if (shdr.magic != htonl(SEAMA_MAGIC)) ERRBREAK("Invalid SEAMA magic. Probably no more SEAMA!\n"); - - /* Get the size */ - isize = ntohl(shdr.size); - msize = ntohs(shdr.metasize); - - /* The checksum exist only if size is greater than zero. */ - if (isize > 0) - { - if (fread(checksum, sizeof(checksum), 1, fh) != 1) - ERRBREAK("Error reading checksum !\n"); - } - - /* Check the META size. */ - if (msize > sizeof(buf)) ERRBREAK("META data in SEAMA header is too large!\n"); - - /* Read META data. */ - if (fread(buf, sizeof(char), msize, fh) != msize) - ERRBREAK("Unable to read SEAMA META data!\n"); - - /* dump header */ - if (msg) - { - printf("SEAMA ==========================================\n"); - printf(" magic : %08x\n", ntohl(shdr.magic)); - printf(" meta size : %zu bytes\n", msize); - for (i=0; i<msize; i+=(strlen((const char *)&buf[i])+1)) - printf(" meta data : %s\n", &buf[i]); - printf(" image size : %zu bytes\n", isize); - } - - /* verify checksum */ - if (isize > 0) - { - if (msg) - { - printf(" checksum : "); - for (i=0; i<16; i++) printf("%02X", checksum[i]); - printf("\n"); - } - - /* Calculate the checksum */ - calculate_digest(fh, isize, digest); - if (msg) - { - printf(" digest : "); - for (i=0; i<16; i++) printf("%02X", digest[i]); - printf("\n"); - } - - if (memcmp(checksum, digest, 16)!=0) ERRBREAK("!!ERROR!! checksum error !!\n"); - ret = 0; - } - } - if (msg) printf("================================================\n"); - } while (0); - if (fh) fclose(fh); - return ret; -} - -static size_t write_seama_header(FILE * fh, char * meta[], size_t msize, size_t size) -{ - seamahdr_t shdr; - size_t i; - uint16_t metasize = 0; - - /* Calculate the META size */ - for (i=0; i<msize; i++) metasize += (strlen(meta[i]) + 1); - //+++ let meta data end on 4 alignment by siyou. 2010/3/1 03:58pm - metasize = ((metasize+3)/4)*4; - verbose("SEAMA META : %d bytes\n", metasize); - - /* Fill up the header, all the data endian should be network byte order. */ - shdr.magic = htonl(SEAMA_MAGIC); - shdr.reserved = 0; - shdr.metasize = htons(metasize); - shdr.size = htonl(size); - - /* Write the header */ - return fwrite(&shdr, sizeof(seamahdr_t), 1, fh); -} - -static size_t write_checksum(FILE * fh, uint8_t * checksum) -{ - return fwrite(checksum, sizeof(uint8_t), 16, fh); -} - -static size_t write_meta_data(FILE * fh, char * meta[], size_t size) -{ - size_t i,j; - size_t ret = 0; - - for (i=0; i<size; i++) - { - verbose("SEAMA META data : %s\n", meta[i]); - j = fwrite(meta[i], sizeof(char), strlen(meta[i])+1, fh); - if (j != strlen(meta[i])+1) return 0; - ret += j; - } - //+++ let meta data end on 4 alignment by siyou. 2010/3/1 03:58pm - j = ((ret+3)/4)*4; - for ( ; ret < j; ret++) - fwrite("", sizeof(char), 1, fh); - - return ret; -} - -/*******************************************************************/ - -static void dump_seama(const char * fname) -{ - verify_seama(fname, 1); -} - -static void seal_files(const char * file) -{ - FILE * fh; - FILE * ifh; - size_t i; - - /* Each image should be seama. */ - for (i = 0; i < o_isize; i++) - { - if (verify_seama(o_images[i], 0) < 0) - { - printf("'%s' is not a seama file !\n",o_images[i]); - return; - } - } - - /* Open file for write */ - fh = fopen(file, "w+"); - if (fh) - { - /* Write the header. */ - write_seama_header(fh, o_meta, o_msize, 0); - write_meta_data(fh, o_meta, o_msize); - - /* Write image files */ - for (i=0; i<o_isize; i++) - { - ifh = fopen(o_images[i], "r+"); - if (ifh) - { - copy_file(fh, ifh); - fclose(ifh); - } - } - - fclose(fh); - } -} - -static void pack_files(void) -{ - FILE * fh; - FILE * ifh; - size_t i, fsize; - char filename[512]; - uint8_t digest[16]; - - for (i=0; i<o_isize; i++) - { - /* Open the input file. */ - ifh = fopen(o_images[i], "r+"); - if (ifh) - { - fsize = calculate_digest(ifh, 0, digest); - verbose("file size (%s) : %d\n", o_images[i], fsize); - rewind(ifh); - - /* Open the output file. */ - sprintf(filename, "%s.seama", o_images[i]); - fh = fopen(filename, "w+"); - if (fh) - { - write_seama_header(fh, o_meta, o_msize, fsize); - write_checksum(fh, digest); - write_meta_data(fh, o_meta, o_msize); - copy_file(fh, ifh); - fclose(fh); - } - fclose(ifh); - } - else - { - printf("Unable to open image file '%s'\n",o_images[i]); - } - } -} - -/**************************************************************************/ - -static int match_meta(const char * meta, size_t size) -{ - size_t i, j; - int match; - - for (i = 0; i < o_msize; i++) - { - for (match = 0, j = 0; j < size; j += (strlen(&meta[j])+1)) - if (strcmp(&meta[j], o_meta[i])==0) { match++; break; } - if (!match) return 0; - } - return 1; -} - - -static void extract_file(const char * output) -{ - FILE * ifh = NULL; - FILE * ofh = NULL; - size_t msize, isize, i, m; - seamahdr_t shdr; - uint8_t buf[MAX_SEAMA_META_SIZE]; - int done = 0; - - /* We need meta for searching the target image. */ - if (o_msize == 0) - { - printf("SEAMA: need meta for searching image.\n"); - return; - } - - /* Walk through each input file */ - for (i = 0; i < o_isize; i++) - { - /* verify the input file */ - if (verify_seama(o_images[i], 0) < 0) - { - printf("SEAMA: '%s' is not a seama file !\n", o_images[i]); - continue; - } - /* open the input file */ - ifh = fopen(o_images[i], "r"); - if (!ifh) continue; - /* read file */ - while (!feof(ifh) && !ferror(ifh)) - { - /* read header */ - fread(&shdr, sizeof(shdr), 1, ifh); - if (shdr.magic != htonl(SEAMA_MAGIC)) break; - /* Get the size */ - isize = ntohl(shdr.size); - msize = ntohs(shdr.metasize); - if (isize == 0) - { - while (msize > 0) - { - m = fread(buf, sizeof(char), (msize < MAX_SEAMA_META_SIZE) ? msize : MAX_SEAMA_META_SIZE, ifh); - if (m <= 0) break; - msize -= m; - } - continue; - } - /* read checksum */ - fread(buf, sizeof(char), 16, ifh); - if (msize > 0) - { - /* read META */ - fread(buf, sizeof(char), msize, ifh); - if (match_meta((const char *)buf, msize)) - { - printf("SEAMA: found image @ '%s', image size: %zu\n", o_images[i], isize); - /* open output file */ - ofh = fopen(output, "w"); - if (!ofh) printf("SEAMA: unable to open '%s' for writting.\n",output); - else - { - while (isize > 0) - { - m = fread(buf, sizeof(char), (isize < MAX_SEAMA_META_SIZE) ? isize : MAX_SEAMA_META_SIZE, ifh); - if (m <= 0) break; - fwrite(buf, sizeof(char), m, ofh); - isize -= m; - } - fclose(ofh); - } - done++; - break; - } - } - while (isize > 0) - { - m = fread(buf, sizeof(char), (isize < MAX_SEAMA_META_SIZE) ? isize : MAX_SEAMA_META_SIZE, ifh); - if (m <= 0) break; - isize -= m; - } - } - /* close the file. */ - fclose(ifh); - if (done) break; - } - return; -} - -/*******************************************************************/ -#ifdef RGBIN_BOX -int seama_main(int argc, char * argv[], char * env[]) -#else -int main(int argc, char * argv[], char * env[]) -#endif -{ - verbose("SEAMA version " VERSION "\n"); - - /* parse the arguments */ - if (parse_args(argc, argv) < 0) show_usage(9); - - /* Do the works */ - if (o_dump) dump_seama(o_dump); - else if (o_seal) seal_files(o_seal); - else if (o_extract) extract_file(o_extract); - else pack_files(); - - cleanup_exit(0); - return 0; -} |