cbfstool: Rework to use getopt style parameters
- Adding more and more optional and non-optional parameters bloated cbfstool and made the code hard to read with a lot of parsing in the actual cbfs handling functions. This change switches over to use getopt style options for everything but command and cbfs file name. - This allows us to simplify the coreboot Makefiles a bit - Also, add guards to include files - Fix some 80+ character lines - Add more detailed error reporting - Free memory we're allocating Signed-off-by: Stefan Reinauer <reinauer@google.com> Change-Id: Ia9137942deb8d26bbb30068e6de72466afe9b0a7 Reviewed-on: http://review.coreboot.org/1800 Tested-by: build bot (Jenkins) Reviewed-by: Ronald G. Minnich <rminnich@gmail.com>
This commit is contained in:
committed by
Stefan Reinauer
parent
e5a0a5d6df
commit
632175802e
@@ -22,401 +22,491 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <ctype.h>
|
||||
#include <unistd.h>
|
||||
#include <getopt.h>
|
||||
#include "common.h"
|
||||
#include "cbfs.h"
|
||||
|
||||
struct command {
|
||||
const char *name;
|
||||
int (*function) (int argc, char **argv);
|
||||
const char *optstring;
|
||||
int (*function) (void);
|
||||
};
|
||||
|
||||
static int cbfs_add(int argc, char **argv)
|
||||
int verbose = 0;
|
||||
static char *cbfs_name = NULL;
|
||||
static char *rom_name = NULL;
|
||||
static char *rom_filename = NULL;
|
||||
static char *rom_bootblock = NULL;
|
||||
static uint32_t rom_type = 0;
|
||||
static uint32_t rom_baseaddress = 0;
|
||||
static uint32_t rom_loadaddress = 0;
|
||||
static uint32_t rom_entrypoint = 0;
|
||||
static uint32_t rom_size = 0;
|
||||
static uint32_t rom_alignment = 0;
|
||||
static uint32_t rom_offset = 0;
|
||||
static comp_algo rom_algo = CBFS_COMPRESS_NONE;
|
||||
|
||||
static int cbfs_add(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
char *cmd = argv[2];
|
||||
void *rom = loadrom(romname);
|
||||
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc < 5) {
|
||||
printf("not enough arguments to '%s'.\n", cmd);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char *filename = argv[3];
|
||||
char *cbfsname = argv[4];
|
||||
|
||||
uint32_t filesize = 0;
|
||||
void *filedata = loadfile(filename, &filesize, 0, SEEK_SET);
|
||||
void *rom, *filedata, *cbfsfile;
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (rom_type == 0) {
|
||||
fprintf(stderr, "E: You need to specify a valid -t/--type.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
filedata = loadfile(rom_filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
printf("Could not load file '%s'.\n", filename);
|
||||
fprintf(stderr, "E: Could not load file '%s'.\n",
|
||||
rom_filename);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
uint32_t base = 0;
|
||||
void *cbfsfile = NULL;
|
||||
cbfsfile = create_cbfs_file(rom_name, filedata, &filesize,
|
||||
rom_type, &rom_baseaddress);
|
||||
free(filedata);
|
||||
|
||||
if (argc < 6) {
|
||||
printf("not enough arguments to 'add'.\n");
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, rom_baseaddress)) {
|
||||
fprintf(stderr, "E: Adding file '%s' failed.\n", rom_filename);
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
uint32_t type;
|
||||
if (intfiletype(argv[5]) != ((uint64_t) - 1))
|
||||
type = intfiletype(argv[5]);
|
||||
else
|
||||
type = strtoul(argv[5], NULL, 0);
|
||||
if (argc > 6) {
|
||||
base = strtoul(argv[6], NULL, 0);
|
||||
}
|
||||
cbfsfile = create_cbfs_file(cbfsname, filedata, &filesize, type, &base);
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, base)) {
|
||||
printf("Adding file '%s' failed.\n", filename);
|
||||
if (writerom(cbfs_name, rom, romsize)) {
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
if (writerom(romname, rom, romsize))
|
||||
return 1;
|
||||
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_add_payload(int argc, char **argv)
|
||||
static int cbfs_add_payload(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
char *cmd = argv[2];
|
||||
void *rom = loadrom(romname);
|
||||
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc < 5) {
|
||||
printf("not enough arguments to '%s'.\n", cmd);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char *filename = argv[3];
|
||||
char *cbfsname = argv[4];
|
||||
|
||||
uint32_t filesize = 0;
|
||||
void *filedata = loadfile(filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
printf("Could not load file '%s'.\n", filename);
|
||||
return 1;
|
||||
}
|
||||
|
||||
uint32_t base = 0;
|
||||
void *cbfsfile = NULL;
|
||||
|
||||
comp_algo algo = CBFS_COMPRESS_NONE;
|
||||
if (argc > 5) {
|
||||
if (argv[5][0] == 'l')
|
||||
algo = CBFS_COMPRESS_LZMA;
|
||||
}
|
||||
if (argc > 6) {
|
||||
base = strtoul(argv[6], NULL, 0);
|
||||
}
|
||||
void *rom, *filedata, *cbfsfile;
|
||||
unsigned char *payload;
|
||||
filesize = parse_elf_to_payload(filedata, &payload, algo);
|
||||
cbfsfile =
|
||||
create_cbfs_file(cbfsname, payload, &filesize,
|
||||
CBFS_COMPONENT_PAYLOAD, &base);
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, base)) {
|
||||
printf("Adding payload '%s' failed.\n", filename);
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
if (writerom(romname, rom, romsize))
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_add_stage(int argc, char **argv)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
char *cmd = argv[2];
|
||||
void *rom = loadrom(romname);
|
||||
}
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc < 5) {
|
||||
printf("not enough arguments to '%s'.\n", cmd);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char *filename = argv[3];
|
||||
char *cbfsname = argv[4];
|
||||
|
||||
uint32_t filesize = 0;
|
||||
void *filedata = loadfile(filename, &filesize, 0, SEEK_SET);
|
||||
filedata = loadfile(rom_filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
printf("Could not load file '%s'.\n", filename);
|
||||
fprintf(stderr, "E: Could not load file '%s'.\n",
|
||||
rom_filename);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
uint32_t base = 0;
|
||||
void *cbfsfile = NULL;
|
||||
filesize = parse_elf_to_payload(filedata, &payload, rom_algo);
|
||||
|
||||
comp_algo algo = CBFS_COMPRESS_NONE;
|
||||
if (argc > 5) {
|
||||
if (argv[5][0] == 'l')
|
||||
algo = CBFS_COMPRESS_LZMA;
|
||||
}
|
||||
if (argc > 6) {
|
||||
base = strtoul(argv[6], NULL, 0);
|
||||
}
|
||||
unsigned char *stage;
|
||||
filesize = parse_elf_to_stage(filedata, &stage, algo, &base);
|
||||
cbfsfile =
|
||||
create_cbfs_file(cbfsname, stage, &filesize,
|
||||
CBFS_COMPONENT_STAGE, &base);
|
||||
cbfsfile = create_cbfs_file(rom_name, payload, &filesize,
|
||||
CBFS_COMPONENT_PAYLOAD, &rom_baseaddress);
|
||||
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, base)) {
|
||||
printf("Adding stage '%s' failed.\n", filename);
|
||||
free(filedata);
|
||||
free(payload);
|
||||
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, rom_baseaddress)) {
|
||||
fprintf(stderr, "E: Adding payload '%s' failed.\n",
|
||||
rom_filename);
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
if (writerom(romname, rom, romsize))
|
||||
|
||||
if (writerom(cbfs_name, rom, romsize)) {
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_add_flat_binary(int argc, char **argv)
|
||||
static int cbfs_add_stage(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
char *cmd = argv[2];
|
||||
void *rom = loadrom(romname);
|
||||
uint32_t filesize = 0;
|
||||
void *rom, *filedata, *cbfsfile;
|
||||
unsigned char *stage;
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc < 7) {
|
||||
printf("not enough arguments to '%s'.\n", cmd);
|
||||
filedata = loadfile(rom_filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
fprintf(stderr, "E: Could not load file '%s'.\n",
|
||||
rom_filename);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char *filename = argv[3];
|
||||
char *cbfsname = argv[4];
|
||||
unsigned long load_address = strtoul(argv[5], NULL, 0);
|
||||
unsigned long entry_point = strtoul(argv[6], NULL, 0);
|
||||
filesize = parse_elf_to_stage(filedata, &stage, rom_algo, &rom_baseaddress);
|
||||
|
||||
uint32_t base = 0;
|
||||
void *cbfsfile = NULL;
|
||||
cbfsfile = create_cbfs_file(rom_name, stage, &filesize,
|
||||
CBFS_COMPONENT_STAGE, &rom_baseaddress);
|
||||
|
||||
comp_algo algo = CBFS_COMPRESS_NONE;
|
||||
if (argc > 7) {
|
||||
if (argv[7][0] == 'l')
|
||||
algo = CBFS_COMPRESS_LZMA;
|
||||
free(filedata);
|
||||
free(stage);
|
||||
|
||||
if (add_file_to_cbfs(cbfsfile, filesize, rom_baseaddress)) {
|
||||
fprintf(stderr, "E: Adding stage '%s' failed.\n",
|
||||
rom_filename);
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
if (argc > 8) {
|
||||
base = strtoul(argv[8], NULL, 0);
|
||||
|
||||
if (writerom(cbfs_name, rom, romsize)) {
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
comp_func_ptr compress = compression_function(algo);
|
||||
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_add_flat_binary(void)
|
||||
{
|
||||
uint32_t filesize = 0;
|
||||
uint32_t final_size;
|
||||
void *rom, *filedata, *cbfsfile;
|
||||
unsigned char *payload;
|
||||
comp_func_ptr compress;
|
||||
struct cbfs_payload_segment *segs;
|
||||
int doffset, len = 0;
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (rom_loadaddress == 0) {
|
||||
fprintf(stderr, "E: You need to specify a valid "
|
||||
"-l/--load-address.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (rom_entrypoint == 0) {
|
||||
fprintf(stderr, "You need to specify a valid "
|
||||
"-e/--entry-point.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
compress = compression_function(rom_algo);
|
||||
if (!compress)
|
||||
return 1;
|
||||
|
||||
uint32_t filesize = 0;
|
||||
void *filedata = loadfile(filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
printf("Could not load file '%s'.\n", filename);
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
fprintf(stderr, "Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
unsigned char *payload;
|
||||
filedata = loadfile(rom_filename, &filesize, 0, SEEK_SET);
|
||||
if (filedata == NULL) {
|
||||
fprintf(stderr, "E: Could not load file '%s'.\n",
|
||||
rom_filename);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* FIXME compressed file size might be bigger than original file */
|
||||
payload = calloc((2 * sizeof(struct cbfs_payload_segment)) + filesize, 1);
|
||||
if (payload == NULL) {
|
||||
printf("Could not allocate memory.\n");
|
||||
fprintf(stderr, "E: Could not allocate memory.\n");
|
||||
free(filedata);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
struct cbfs_payload_segment *segs;
|
||||
segs = (struct cbfs_payload_segment *)payload;
|
||||
int doffset = (2 * sizeof(struct cbfs_payload_segment));
|
||||
doffset = (2 * sizeof(struct cbfs_payload_segment));
|
||||
|
||||
/* Prepare code segment */
|
||||
segs[0].type = PAYLOAD_SEGMENT_CODE;
|
||||
segs[0].load_addr = (uint64_t)htonll(load_address);
|
||||
segs[0].load_addr = (uint64_t)htonll(rom_loadaddress);
|
||||
segs[0].mem_len = (uint32_t)htonl(filesize);
|
||||
segs[0].offset = htonl(doffset);
|
||||
|
||||
int len = 0;
|
||||
segs[0].offset = (uint32_t)htonl(doffset);
|
||||
|
||||
compress(filedata, filesize, (char *)(payload + doffset), &len);
|
||||
segs[0].compression = htonl(algo);
|
||||
segs[0].compression = htonl(rom_algo);
|
||||
segs[0].len = htonl(len);
|
||||
|
||||
if ((unsigned int)len > filesize) {
|
||||
if ((unsigned int)len >= filesize) {
|
||||
segs[0].compression = 0;
|
||||
segs[0].len = htonl(filesize);
|
||||
memcpy((char *)(payload + doffset), filedata, filesize);
|
||||
}
|
||||
uint32_t final_size = doffset + ntohl(segs[0].len);
|
||||
|
||||
/* prepare entry point segment */
|
||||
segs[1].type = PAYLOAD_SEGMENT_ENTRY;
|
||||
segs[1].load_addr = (uint64_t)htonll(entry_point);
|
||||
segs[1].load_addr = (uint64_t)htonll(rom_entrypoint);
|
||||
|
||||
final_size = doffset + ntohl(segs[0].len);
|
||||
cbfsfile =
|
||||
create_cbfs_file(cbfsname, payload, &final_size,
|
||||
CBFS_COMPONENT_PAYLOAD, &base);
|
||||
if (add_file_to_cbfs(cbfsfile, final_size, base)) {
|
||||
printf("Adding payload '%s' failed.\n", filename);
|
||||
create_cbfs_file(rom_name, payload, &final_size,
|
||||
CBFS_COMPONENT_PAYLOAD, &rom_baseaddress);
|
||||
|
||||
free(filedata);
|
||||
free(payload);
|
||||
|
||||
if (add_file_to_cbfs(cbfsfile, final_size, rom_baseaddress)) {
|
||||
fprintf(stderr, "E: Adding payload '%s' failed.\n",
|
||||
rom_filename);
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
if (writerom(romname, rom, romsize))
|
||||
if (writerom(cbfs_name, rom, romsize)) {
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
free(cbfsfile);
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_remove(int argc, char **argv)
|
||||
static int cbfs_remove(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
char *cmd = argv[2];
|
||||
void *rom = loadrom(romname);
|
||||
void *rom;
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc < 4) {
|
||||
printf("not enough arguments to '%s'.\n", cmd);
|
||||
if (remove_file_from_cbfs(rom_name)) {
|
||||
fprintf(stderr, "E: Removing file '%s' failed.\n",
|
||||
rom_name);
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char *cbfsname = argv[3];
|
||||
|
||||
if (remove_file_from_cbfs(cbfsname)) {
|
||||
printf("Removing file '%s' failed.\n", cbfsname);
|
||||
if (writerom(cbfs_name, rom, romsize)) {
|
||||
free(rom);
|
||||
return 1;
|
||||
}
|
||||
if (writerom(romname, rom, romsize))
|
||||
return 1;
|
||||
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_create(int argc, char **argv)
|
||||
static int cbfs_create(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
if (argc < 5) {
|
||||
printf("not enough arguments to 'create'.\n");
|
||||
if (rom_size == 0) {
|
||||
printf("You need to specify a valid -s/--size.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
char* suffix;
|
||||
uint32_t size = strtoul(argv[3], &suffix, 0);
|
||||
if (tolower(suffix[0])=='k') {
|
||||
size *= 1024;
|
||||
if (!rom_bootblock) {
|
||||
printf("You need to specify -b/--bootblock.\n");
|
||||
return 1;
|
||||
}
|
||||
if (tolower(suffix[0])=='m') {
|
||||
size *= 1024 * 1024;
|
||||
}
|
||||
char *bootblock = argv[4];
|
||||
uint32_t align = 0;
|
||||
|
||||
if (argc > 5)
|
||||
align = strtoul(argv[5], NULL, 0);
|
||||
|
||||
uint32_t offs = 0;
|
||||
if (argc > 6)
|
||||
offs = strtoul(argv[6], NULL, 0);
|
||||
|
||||
return create_cbfs_image(romname, size, bootblock, align, offs);
|
||||
return create_cbfs_image(cbfs_name, rom_size, rom_bootblock,
|
||||
rom_alignment, rom_offset);
|
||||
}
|
||||
|
||||
static int cbfs_locate(int argc, char **argv)
|
||||
static int cbfs_locate(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
if (argc < 6) {
|
||||
printf("not enough arguments to 'locate'.\n");
|
||||
uint32_t filesize, location;
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
const char *file = argv[3];
|
||||
uint32_t filesize = getfilesize(file);
|
||||
const char *filename = argv[4];
|
||||
int align = strtoul(argv[5], NULL, 0);
|
||||
uint32_t location = cbfs_find_location(romname, filesize, filename, align);
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
filesize = getfilesize(rom_filename);
|
||||
|
||||
location = cbfs_find_location(cbfs_name, filesize,
|
||||
rom_name, rom_alignment);
|
||||
|
||||
printf("%x\n", location);
|
||||
return location == 0 ? 1 : 0;
|
||||
}
|
||||
|
||||
static int cbfs_print(int argc, char **argv)
|
||||
static int cbfs_print(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
void *rom = loadrom(romname);
|
||||
void *rom;
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
print_cbfs_directory(romname);
|
||||
print_cbfs_directory(cbfs_name);
|
||||
|
||||
free(rom);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cbfs_extract(int argc, char **argv)
|
||||
static int cbfs_extract(void)
|
||||
{
|
||||
char *romname = argv[1];
|
||||
void *rom = loadrom(romname);
|
||||
void *rom;
|
||||
int ret;
|
||||
|
||||
if (!rom_filename) {
|
||||
fprintf(stderr, "E: You need to specify -f/--filename.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (!rom_name) {
|
||||
fprintf(stderr, "E: You need to specify -n/--name.\n");
|
||||
return 1;
|
||||
}
|
||||
|
||||
rom = loadrom(cbfs_name);
|
||||
if (rom == NULL) {
|
||||
printf("Could not load ROM image '%s'.\n", romname);
|
||||
fprintf(stderr, "E: Could not load ROM image '%s'.\n",
|
||||
cbfs_name);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc != 5)
|
||||
{
|
||||
printf("Error: you must specify a CBFS name and a file to dump it in.\n");
|
||||
return 1;
|
||||
}
|
||||
ret = extract_file_from_cbfs(cbfs_name, rom_name, rom_filename);
|
||||
|
||||
return extract_file_from_cbfs(romname, argv[3], argv[4]);
|
||||
free(rom);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static const struct command commands[] = {
|
||||
{"add", cbfs_add},
|
||||
{"add-payload", cbfs_add_payload},
|
||||
{"add-stage", cbfs_add_stage},
|
||||
{"add-flat-binary", cbfs_add_flat_binary},
|
||||
{"remove", cbfs_remove},
|
||||
{"create", cbfs_create},
|
||||
{"locate", cbfs_locate},
|
||||
{"print", cbfs_print},
|
||||
{"extract", cbfs_extract},
|
||||
{"add", "f:n:t:b:h?", cbfs_add},
|
||||
{"add-payload", "f:n:t:c:b:h?", cbfs_add_payload},
|
||||
{"add-stage", "f:n:t:c:b:h?", cbfs_add_stage},
|
||||
{"add-flat-binary", "f:n:l:e:c:b:h?", cbfs_add_flat_binary},
|
||||
{"remove", "n:h?", cbfs_remove},
|
||||
{"create", "s:B:a:o:h?", cbfs_create},
|
||||
{"locate", "f:n:a:h?", cbfs_locate},
|
||||
{"print", "h?", cbfs_print},
|
||||
{"extract", "n:f:h?", cbfs_extract},
|
||||
};
|
||||
|
||||
static void usage(void)
|
||||
static struct option long_options[] = {
|
||||
{"name", required_argument, 0, 'n' },
|
||||
{"type", required_argument, 0, 't' },
|
||||
{"compression", required_argument, 0, 'c' },
|
||||
{"base-address", required_argument, 0, 'b' },
|
||||
{"load-address", required_argument, 0, 'l' },
|
||||
{"entry-point", required_argument, 0, 'e' },
|
||||
{"size", required_argument, 0, 's' },
|
||||
{"bootblock", required_argument, 0, 'B' },
|
||||
{"alignment", required_argument, 0, 'a' },
|
||||
{"offset", required_argument, 0, 'o' },
|
||||
{"file", required_argument, 0, 'f' },
|
||||
{"verbose", no_argument, 0, 'v' },
|
||||
{"help", no_argument, 0, 'h' },
|
||||
{NULL, 0, 0, 0 }
|
||||
};
|
||||
|
||||
static void usage(char *name)
|
||||
{
|
||||
printf
|
||||
("cbfstool: Management utility for CBFS formatted ROM images\n\n"
|
||||
"USAGE:\n" " cbfstool [-h]\n"
|
||||
" cbfstool FILE COMMAND [PARAMETERS]...\n\n" "OPTIONs:\n"
|
||||
"USAGE:\n" " %s [-h]\n"
|
||||
" %s FILE COMMAND [PARAMETERS]...\n\n" "OPTIONs:\n"
|
||||
" -h Display this help message\n\n"
|
||||
"COMMANDs:\n"
|
||||
" add FILE NAME TYPE [base address] Add a component\n"
|
||||
" add-payload FILE NAME [COMP] [base] Add a payload to the ROM\n"
|
||||
" add-stage FILE NAME [COMP] [base] Add a stage to the ROM\n"
|
||||
" add-flat-binary FILE NAME LOAD ENTRY \\\n"
|
||||
" [COMP] [base] Add a 32bit flat mode binary\n"
|
||||
" remove FILE NAME Remove a component\n"
|
||||
" create SIZE BOOTBLOCK [ALIGN] [offset] Create a ROM file\n"
|
||||
" locate FILE NAME ALIGN Find a place for a file of that size\n"
|
||||
" print Show the contents of the ROM\n"
|
||||
" extract NAME FILE Extracts a raw payload from ROM\n"
|
||||
" add -f FILE -n NAME -t TYPE [-b base-address] "
|
||||
"Add a component\n"
|
||||
" add-payload -f FILE -n NAME [-c compression] [-b base] "
|
||||
"Add a payload to the ROM\n"
|
||||
" add-stage -f FILE -n NAME [-c compression] [-b base] "
|
||||
"Add a stage to the ROM\n"
|
||||
" add-flat-binary -f FILE -n NAME -l load-address \\\n"
|
||||
" -e entry-point [-c compression] [-b base] "
|
||||
"Add a 32bit flat mode binary\n"
|
||||
" remove -n NAME "
|
||||
"Remove a component\n"
|
||||
" create -s size -B bootblock [-a align] [-o offset] "
|
||||
"Create a ROM file\n"
|
||||
" locate -f FILE -n NAME -a align "
|
||||
"Find a place for a file of that size\n"
|
||||
" print "
|
||||
"Show the contents of the ROM\n"
|
||||
" extract -n NAME -f FILE "
|
||||
"Extracts a raw payload from ROM\n"
|
||||
"\n"
|
||||
"TYPEs:\n"
|
||||
);
|
||||
"TYPEs:\n", name, name
|
||||
);
|
||||
print_supported_filetypes();
|
||||
}
|
||||
|
||||
/* Small, OS/libc independent runtime check
|
||||
* for endianess
|
||||
*/
|
||||
/* Small, OS/libc independent runtime check for endianess */
|
||||
int host_bigendian = 0;
|
||||
|
||||
static void which_endian(void)
|
||||
@@ -431,23 +521,108 @@ static void which_endian(void)
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
size_t i;
|
||||
int c;
|
||||
|
||||
if (argc < 3) {
|
||||
usage();
|
||||
usage(argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
which_endian();
|
||||
|
||||
cbfs_name = argv[1];
|
||||
char *cmd = argv[2];
|
||||
optind += 2;
|
||||
|
||||
for (i = 0; i < ARRAY_SIZE(commands); i++) {
|
||||
if (strcmp(cmd, commands[i].name) != 0)
|
||||
continue;
|
||||
return commands[i].function(argc, argv);
|
||||
|
||||
while (1) {
|
||||
char *suffix = NULL;
|
||||
int option_index = 0;
|
||||
|
||||
c = getopt_long(argc, argv, commands[i].optstring,
|
||||
long_options, &option_index);
|
||||
if (c == -1)
|
||||
break;
|
||||
|
||||
/* filter out illegal long options */
|
||||
if (index(commands[i].optstring, c) == NULL) {
|
||||
/* TODO maybe print actual long option instead */
|
||||
printf("%s: invalid option -- '%c'\n",
|
||||
argv[0], c);
|
||||
c = '?';
|
||||
}
|
||||
|
||||
switch(c) {
|
||||
case 'n':
|
||||
rom_name = optarg;
|
||||
break;
|
||||
case 't':
|
||||
if (intfiletype(optarg) != ((uint64_t) - 1))
|
||||
rom_type = intfiletype(optarg);
|
||||
else
|
||||
rom_type = strtoul(optarg, NULL, 0);
|
||||
if (rom_type == 0)
|
||||
printf("W: Unknown type '%s' ignored\n",
|
||||
optarg);
|
||||
break;
|
||||
case 'c':
|
||||
if (!strncasecmp(optarg, "lzma", 5))
|
||||
rom_algo = CBFS_COMPRESS_LZMA;
|
||||
else if (!strncasecmp(optarg, "none", 5))
|
||||
rom_algo = CBFS_COMPRESS_NONE;
|
||||
else
|
||||
printf("W: Unknown compression '%s'"
|
||||
" ignored.\n", optarg);
|
||||
break;
|
||||
case 'b':
|
||||
rom_baseaddress = strtoul(optarg, NULL, 0);
|
||||
break;
|
||||
case 'l':
|
||||
rom_loadaddress = strtoul(optarg, NULL, 0);
|
||||
|
||||
break;
|
||||
case 'e':
|
||||
rom_entrypoint = strtoul(optarg, NULL, 0);
|
||||
break;
|
||||
case 's':
|
||||
rom_size = strtoul(optarg, &suffix, 0);
|
||||
if (tolower(suffix[0])=='k') {
|
||||
rom_size *= 1024;
|
||||
}
|
||||
if (tolower(suffix[0])=='m') {
|
||||
rom_size *= 1024 * 1024;
|
||||
}
|
||||
case 'B':
|
||||
rom_bootblock = optarg;
|
||||
break;
|
||||
case 'a':
|
||||
rom_alignment = strtoul(optarg, NULL, 0);
|
||||
break;
|
||||
case 'o':
|
||||
rom_offset = strtoul(optarg, NULL, 0);
|
||||
break;
|
||||
case 'f':
|
||||
rom_filename = optarg;
|
||||
break;
|
||||
case 'v':
|
||||
verbose++;
|
||||
break;
|
||||
case 'h':
|
||||
case '?':
|
||||
usage(argv[0]);
|
||||
return 1;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
return commands[i].function();
|
||||
}
|
||||
|
||||
printf("Unknown command '%s'.\n", cmd);
|
||||
usage();
|
||||
usage(argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
Reference in New Issue
Block a user