1 files changed, 119 insertions, 0 deletions
diff --git a/util/cbfstool/fmaptool.c b/util/cbfstool/fmaptool.c
new file mode 100644
index 0000000000..961e9dc918
--- /dev/null
+++ b/util/cbfstool/fmaptool.c
@@ -0,0 +1,119 @@
+/*
+ * 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;
+}