1 files changed, 168 insertions, 0 deletions
diff --git a/util/cbfstool/fs.c b/util/cbfstool/fs.c
new file mode 100644
index 0000000000..c5e9892a8e
--- /dev/null
+++ b/util/cbfstool/fs.c
@@ -0,0 +1,168 @@
+/*
+ * cbfstool
+ *
+ * Copyright (C) 2008 Jordan Crouse <jordan@cosmicpenguin.net>
+ *
+ * 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 <string.h>
+#include "cbfstool.h"
+
+struct cbfs_file *rom_find(struct rom *rom, unsigned int offset)
+{
+	while (offset < rom->fssize) {
+		struct cbfs_file *c =
+		    (struct cbfs_file *)ROM_PTR(rom, offset);
+
+		if (!strcmp(c->magic, COMPONENT_MAGIC))
+			return c;
+
+		offset += ntohl(rom->header->align);
+	}
+
+	return NULL;
+}
+
+struct cbfs_file *rom_find_first(struct rom *rom)
+{
+	return rom_find(rom, ntohl(rom->header->offset));
+}
+
+struct cbfs_file *rom_find_next(struct rom *rom, struct cbfs_file *prev)
+{
+	unsigned int offset = ROM_OFFSET(rom, prev);
+
+	return rom_find(rom, offset +
+			ALIGN(ntohl(prev->offset) + ntohl(prev->len),
+			      ntohl(rom->header->align)));
+}
+
+struct cbfs_file *rom_find_empty(struct rom *rom)
+{
+	unsigned int offset = ntohl(rom->header->offset);
+	unsigned int ret = ntohl(rom->header->offset);
+
+	while (offset < rom->fssize) {
+
+		struct cbfs_file *c =
+		    (struct cbfs_file *)ROM_PTR(rom, offset);
+
+		if (!strcmp(c->magic, COMPONENT_MAGIC)) {
+			offset += ALIGN(ntohl(c->offset) + ntohl(c->len),
+					ntohl(rom->header->align));
+
+			ret = offset;
+		} else
+			offset += ntohl(rom->header->align);
+	}
+
+	return (ret < rom->fssize) ?
+	    (struct cbfs_file *)ROM_PTR(rom, ret) : NULL;
+}
+
+struct cbfs_file *rom_find_by_name(struct rom *rom, const char *name)
+{
+	struct cbfs_file *c = rom_find_first(rom);
+
+	while (c) {
+		if (!strcmp((char *)CBFS_NAME(c), name))
+			return c;
+
+		c = rom_find_next(rom, c);
+	}
+
+	return NULL;
+}
+
+unsigned int rom_used_space(struct rom *rom)
+{
+	struct cbfs_file *c = rom_find_first(rom);
+	unsigned int ret = 0;
+
+	while (c) {
+		ret = ROM_OFFSET(rom, c) + ntohl(c->offset) + ntohl(c->len);
+		c = rom_find_next(rom, c);
+	}
+
+	return ret;
+}
+
+int rom_remove(struct rom *rom, const char *name)
+{
+	struct cbfs_file *c = rom_find_by_name(rom, name);
+	struct cbfs_file *n;
+	int clear;
+
+	if (c == NULL) {
+		ERROR("Component %s does not exist\n", name);
+		return -1;
+	}
+
+	/* Get the next component - and copy it into the current
+	   space */
+
+	n = rom_find_next(rom, c);
+
+	memcpy(c, n, rom->fssize - ROM_OFFSET(rom, n));
+
+	clear = ROM_OFFSET(rom, n) - ROM_OFFSET(rom, c);
+
+	/* Zero the new space */
+	memset(ROM_PTR(rom, rom->fssize - clear), 0, clear);
+	return 0;
+}
+
+int rom_add(struct rom *rom, const char *name, void *buffer, int size, int type)
+{
+	struct cbfs_file *c = rom_find_empty(rom);
+	unsigned int offset;
+	unsigned int csize;
+
+	if (rom_find_by_name(rom, name)) {
+		ERROR("Component %s already exists in this rom\n", name);
+		return -1;
+	}
+
+	if (c == NULL) {
+		ERROR("There is no more room in this ROM\n");
+		return -1;
+	}
+
+	csize = sizeof(struct cbfs_file) + ALIGN(strlen(name), 16) + size;
+
+	offset = ROM_OFFSET(rom, c);
+
+	if (offset + csize >= rom->fssize) {
+		ERROR("There is not enough room in this ROM for this\n");
+		ERROR("component. I need %d bytes, only have %d bytes avail\n",
+		      csize, rom->fssize - offset);
+
+		return -1;
+	}
+
+	strcpy(c->magic, COMPONENT_MAGIC);
+
+	csize = sizeof(struct cbfs_file) + ALIGN(strlen(name) + 1, 16);
+
+	c->len = htonl(size);
+	c->offset = htonl(csize);
+	c->type = htonl(type);
+
+	memset(CBFS_NAME(c), 0, ALIGN(strlen(name) + 1, 16));
+	strcpy((char *)CBFS_NAME(c), name);
+
+	memcpy(((unsigned char *)c) + csize, buffer, size);
+	return 0;
+}