coreboot-kgpe-d16/util/cbfstool/fmaptool.c

120 lines
3.4 KiB
C

/*
* fmaptool, CLI utility for converting plaintext fmd files into fmap blobs
*
* Copyright (C) 2015 Google, Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA, 02110-1301 USA
*/
#include "common.h"
#include "fmap_from_fmd.h"
#include <stdio.h>
#include <string.h>
#define STDIN_FILENAME_SENTINEL "-"
enum fmaptool_return {
FMAPTOOL_EXIT_SUCCESS = 0,
FMAPTOOL_EXIT_BAD_ARGS,
FMAPTOOL_EXIT_BAD_INPUT_PATH,
FMAPTOOL_EXIT_BAD_OUTPUT_PATH,
FMAPTOOL_EXIT_FAILED_DESCRIPTOR,
FMAPTOOL_EXIT_FAILED_FMAP_CONVERSION,
FMAPTOOL_EXIT_UNKNOWN_FMAP_SIZE,
FMAPTOOL_EXIT_FAILED_WRITING_FILE,
};
bool fmd_process_annotation_impl(unused const struct flashmap_descriptor *node,
unused const char *annotation)
{
// We always accept annotations, but never act on them.
return true;
}
int main(int argc, char **argv)
{
if (argc != 3) {
fputs("Convert a human-readable flashmap descriptor (fmd) file into the binary FMAP format for use in firmware images\n",
stderr);
fprintf(stderr,
"USAGE: %s <fmd input file> <binary output file>\n",
argv[0]);
fprintf(stderr,
"To read from standard input, provide '%s' as the input filename.\n",
STDIN_FILENAME_SENTINEL);
return FMAPTOOL_EXIT_BAD_ARGS;
}
const char *fmd_filename = argv[1];
const char *fmap_filename = argv[2];
FILE *fmd_file = stdin;
if (strcmp(fmd_filename, STDIN_FILENAME_SENTINEL) != 0) {
fmd_file = fopen(fmd_filename, "r");
if (!fmd_file) {
fprintf(stderr, "FATAL: Unable to open file '%s'\n",
fmd_filename);
return FMAPTOOL_EXIT_BAD_INPUT_PATH;
}
}
struct flashmap_descriptor *descriptor = fmd_create(fmd_file);
fclose(fmd_file);
if (!descriptor) {
fputs("FATAL: Failed while processing provided descriptor\n",
stderr);
return FMAPTOOL_EXIT_FAILED_DESCRIPTOR;
}
struct fmap *flashmap = fmap_from_fmd(descriptor);
if (!flashmap) {
fputs("FATAL: Failed while constructing FMAP section\n",
stderr);
fmd_cleanup(descriptor);
return FMAPTOOL_EXIT_FAILED_FMAP_CONVERSION;
}
int size = fmap_size(flashmap);
if (size < 0) {
fputs("FATAL: Failed to determine FMAP section size\n",
stderr);
fmap_destroy(flashmap);
fmd_cleanup(descriptor);
return FMAPTOOL_EXIT_UNKNOWN_FMAP_SIZE;
}
FILE *fmap_file = fopen(fmap_filename, "wb");
if (!fmap_file) {
fprintf(stderr, "FATAL: Unable to open file '%s' for writing\n",
fmap_filename);
fmap_destroy(flashmap);
fmd_cleanup(descriptor);
return FMAPTOOL_EXIT_BAD_OUTPUT_PATH;
}
if (!fwrite(flashmap, size, 1, fmap_file)) {
fputs("FATAL: Failed to write final FMAP to file\n", stderr);
fclose(fmap_file);
fmap_destroy(flashmap);
fmd_cleanup(descriptor);
return FMAPTOOL_EXIT_FAILED_WRITING_FILE;
}
fclose(fmap_file);
printf("SUCCESS: Wrote %d bytes to file '%s'\n", size, fmap_filename);
fmap_destroy(flashmap);
fmd_cleanup(descriptor);
return FMAPTOOL_EXIT_SUCCESS;
}