summaryrefslogtreecommitdiff
path: root/util/cbfstool/resize.c
diff options
context:
space:
mode:
authorPeter Stuge <peter@stuge.se>2009-04-14 00:08:34 +0000
committerPeter Stuge <peter@stuge.se>2009-04-14 00:08:34 +0000
commit1d862ded11d4a2ca42d3d0aff4ea947dd28bcb68 (patch)
treed1ffd93a52e3b218827ac85fb19fb5bd51140044 /util/cbfstool/resize.c
parent450b23fb2ea2f08bd2e1343e0ce34ea72f19c4a9 (diff)
downloadcoreboot-1d862ded11d4a2ca42d3d0aff4ea947dd28bcb68.tar.xz
v2/util: romfs -> cbfs rename
It's all sed here. romfs->cbfs, ROMFS->CBFS, romtool->cbfstool Signed-off-by: Peter Stuge <peter@stuge.se> Acked-by: Ronald G. Minnich <rminnich@gmail.com> git-svn-id: svn://svn.coreboot.org/coreboot/trunk@4110 2b7e53f0-3cfb-0310-b3e9-8179ed1497e1
Diffstat (limited to 'util/cbfstool/resize.c')
-rw-r--r--util/cbfstool/resize.c168
1 files changed, 168 insertions, 0 deletions
diff --git a/util/cbfstool/resize.c b/util/cbfstool/resize.c
new file mode 100644
index 0000000000..157515619c
--- /dev/null
+++ b/util/cbfstool/resize.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 <ctype.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <sys/mman.h>
+
+#include "cbfstool.h"
+
+void resize_usage(void)
+{
+ printf("resize [SIZE] [ERASEBLOCK]\tResize the ROM\n");
+}
+
+int resize_handler(struct rom *rom, int argc, char **argv)
+{
+ unsigned int size, align, offset;
+ char null = '\0';
+ int bootblocksize = ntohl(rom->header->bootblocksize);
+
+ if (argc < 1) {
+ resize_usage();
+ return -1;
+ }
+
+ if (rom->fd <= 0) {
+ ERROR("ROM file %s does not exist\n", rom->name);
+ return -1;
+ }
+
+ align = ntohl(rom->header->align);
+ size = get_size(argv[0]);
+
+ if (argc >= 2)
+ align = strtoul(argv[1], NULL, 0);
+
+ if (size == rom->size && align == ntohl(rom->header->align)) {
+ ERROR("New parameters are the same as the old\n");
+ return 0;
+ }
+
+ if (size < bootblocksize) {
+ ERROR("The new size is smaller then the bootblock size\n");
+ return -1;
+ }
+
+ /* if the current ROM is too big for the new size, then complain */
+
+ if (rom_used_space(rom) > size - bootblocksize) {
+ ERROR("The new size is too small for the current ROM\n");
+ return -1;
+ }
+
+ /* Grow the rom if we need to */
+
+ if (size > rom->size) {
+ munmap(rom->ptr, rom->size);
+
+ lseek(rom->fd, size - 1, SEEK_SET);
+ write(rom->fd, &null, 1);
+
+ rom->ptr = mmap(0, size, PROT_READ | PROT_WRITE, MAP_SHARED,
+ rom->fd, 0);
+
+ if (rom->ptr == MAP_FAILED) {
+ ERROR("Unable to grow the ROM\n");
+ return -1;
+ }
+ }
+
+ /* We only have to rewrite the entries if the alignment changed */
+
+ if (align != ntohl(rom->header->align)) {
+ struct cbfs_file *c;
+
+ /* The first entry doesn't have to move */
+
+ c = rom_find(rom, rom->header->offset);
+ offset = rom->header->offset;
+
+ while (c) {
+ struct cbfs_file *n = rom_find_next(rom, c);
+ unsigned int next;
+
+ if (n == NULL)
+ break;
+
+ /* Calculate a new location for the entry */
+ next =
+ ROM_OFFSET(rom,
+ c) + ALIGN(ntohl(c->offset) +
+ ntohl(c->len), align);
+
+ /* Copy the next entry there */
+ memmove(ROM_PTR(rom, next), n,
+ ntohl(n->offset) + ntohl(n->len));
+
+ c = (struct cbfs_file *)ROM_PTR(rom, next);
+
+ /* If the previous header wasn't overwritten by the change,
+ corrupt the header so we don't accidently find it */
+
+ if (ROM_OFFSET(rom, n) >
+ next + ntohl(c->len) + ntohl(c->offset))
+ memset(n->magic, 0, sizeof(n->magic));
+ }
+ }
+
+ /* Copy the bootblock */
+
+ memmove(rom->ptr + size - bootblocksize,
+ rom->ptr + rom->size - bootblocksize, bootblocksize);
+
+ /* Recacluate the location of the header */
+
+ offset = ROM_READL(rom, size - 12);
+
+ rom->header = (struct cbfs_header *)
+ ROM_PTR(rom, size - (0xFFFFFFFF - offset) - 1);
+
+ /* Put the new values in the header */
+ rom->header->romsize = htonl(size);
+ rom->header->align = htonl(align);
+
+ /* Truncate the file if we have to */
+
+ if (size < rom->size) {
+ munmap(rom->ptr, rom->size);
+
+ rom->ptr = NULL;
+
+ if (ftruncate(rom->fd, size)) {
+ ERROR("Unable to truncate the ROM\n");
+ return -1;
+ }
+
+ rom->ptr = mmap(0, size, PROT_READ | PROT_WRITE, MAP_SHARED,
+ rom->fd, 0);
+
+ if (rom->ptr == MAP_FAILED) {
+ ERROR("Unable to truncate the ROM\n");
+ return -1;
+ }
+ }
+
+ rom->size = size;
+ rom->fssize = size - bootblocksize;
+
+ return 0;
+}