From bf4630076762d9c20c16c80c1c791f352b046dd1 Mon Sep 17 00:00:00 2001 From: Brad Bishop Date: Mon, 30 Jun 2014 22:10:16 -0500 Subject: Port FFS tools over from Building Block repository. --- fpart/src/cmd_write.c | 229 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 229 insertions(+) create mode 100644 fpart/src/cmd_write.c (limited to 'fpart/src/cmd_write.c') diff --git a/fpart/src/cmd_write.c b/fpart/src/cmd_write.c new file mode 100644 index 0000000..83f71d3 --- /dev/null +++ b/fpart/src/cmd_write.c @@ -0,0 +1,229 @@ +/* + * Copyright (c) International Business Machines Corp., 2012 + * + * 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; either version 2 of the License, or + * (at your option) any later version. + * + * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* + * File: write.c + * Author: Shaun Wetzstein + * Descr: --write implementation + * Date: 01/30/2013 + */ + +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include + +#include "main.h" + +int command_write(args_t * args) +{ + assert(args != NULL); + + char name[strlen(args->name) + 2]; + strcpy(name, args->name); + if (strrchr(name, '$') == NULL) + strcat(name, "$"); + + char full_name[page_size]; + struct stat st; + regex_t rx; + + off_t __poffset; + ffs_t *__ffs; + + RAII(FILE*, in, fopen(args->path, "r"), fclose); + + list_t list; + list_init(&list); + + list_iter_t it; + ffs_entry_node_t * node; + + /* ========================= */ + + int write_entry(ffs_entry_t * entry) + { + assert(entry != NULL); + + if (__ffs_entry_name(__ffs, entry, full_name, + sizeof full_name) < 0) + return -1; + + if (regexec(&rx, full_name, 0, NULL, 0) == REG_NOMATCH) + return 0; + + if (entry->flags & FFS_FLAGS_PROTECTED && + args->force != f_FORCE) { + if (args->verbose == f_VERBOSE) + printf("%llx: %s: protected, skipping write\n", + __poffset, full_name); + return 0; + } + + if (entry->actual < st.st_size) { + if (__ffs_entry_truncate(__ffs, full_name, + st.st_size) < 0) { + ERRNO(errno); + return -1; + } + + if (args->verbose == f_VERBOSE) + printf("%llx: %s: truncate size '%llx'\n", + __poffset, full_name, st.st_size); + } + + list_iter_init(&it, &list, LI_FLAG_FWD); + list_for_each(&it, node, node) { + if (node->entry.base == entry->base && + node->entry.size == entry->size) + return 0; + } + + node = (ffs_entry_node_t *)malloc(sizeof(*node)); + assert(node != NULL); + + memcpy(&node->entry, entry, sizeof(node->entry)); + list_add_tail(&list, &node->node); + + size_t block_size = __ffs->hdr->block_size; + size_t entry_size = entry->size * block_size; + size_t data_size = st.st_size; + off_t data_offset = 0; + + if (entry_size < data_size) { + UNEXPECTED("'%s' of size '%x' too big for partition " + "'%s' of size '%x'", args->path, data_size, + args->name, entry_size); + return -1; + } + + if (args->block != NULL) { + if (parse_size(args->block, &block_size) < 0) + return -1; + if (block_size & (__ffs->hdr->block_size-1)) { + UNEXPECTED("'%x' block size must be multiple " + "of target block size '%x'", + block_size, __ffs->hdr->block_size); + return -1; + } + } + + if (__ffs_buffer(__ffs, block_size) < 0) + return -1; + + RAII(void*, block, malloc(block_size), free); + if (block == NULL) { + ERRNO(errno); + return -1; + } + + if (fseeko(in, 0, SEEK_SET) != 0) { + ERRNO(errno); + return -1; + } + + while (0 < data_size) { + ssize_t rc = fread(block, 1, + min(block_size, data_size), in); + if (rc <= 0 && ferror(in)) { + ERRNO(errno); + return -1; + } + + rc = __ffs_entry_write(__ffs, full_name, block, + data_offset, rc); + if (rc < 0) + return -1; + + __ffs_fsync(__ffs); + + data_offset += rc; + data_size -= rc; + } + + if (args->verbose == f_VERBOSE) + printf("%llx: %s: read '%llx' bytes from file '%s'\n", + __poffset, full_name, st.st_size, args->path); + + return 0; + } + + int write(args_t * args, off_t poffset) + { + __poffset = poffset; + + const char * target = args->target; + int debug = args->debug; + + RAII(FILE*, file, fopen_generic(target, "r+", debug), fclose); + if (file == NULL) + return -1; + RAII(ffs_t*, ffs, __ffs_fopen(file, poffset), __ffs_fclose); + if (ffs == NULL) + return -1; + + __ffs = ffs; + + int rc = __ffs_iterate_entries(ffs, write_entry); + if (rc == 1) + rc = -1; + + return rc; + } + + /* ========================= */ + + if (stat(args->path, &st) < 0) { + ERRNO(errno); + return -1; + } + + if (regcomp(&rx, name, REG_ICASE | REG_NOSUB) != 0) { + ERRNO(errno); + return -1; + } + + int rc = command(args, write); + + regfree(&rx); + + while (!list_empty(&list)) + free(container_of(list_remove_head(&list), + ffs_entry_node_t, node)); + + return rc; +} -- cgit v1.2.1