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_compare.c | 231 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 231 insertions(+) create mode 100644 fpart/src/cmd_compare.c (limited to 'fpart/src/cmd_compare.c') diff --git a/fpart/src/cmd_compare.c b/fpart/src/cmd_compare.c new file mode 100644 index 0000000..95b64d0 --- /dev/null +++ b/fpart/src/cmd_compare.c @@ -0,0 +1,231 @@ +/* + * 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: compare.c + * Author: Shaun Wetzstein + * Descr: --compare 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_compare(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]; + regex_t rx; + + off_t __poffset; + ffs_t * __in, * __out; + + list_t list; + list_init(&list); + + list_iter_t it; + ffs_entry_node_t * node; + + /* ========================= */ + + int compare_entry(ffs_entry_t * src) + { + if (__ffs_entry_name(__in, src, full_name, + sizeof full_name) < 0) + return -1; + if (regexec(&rx, full_name, 0, NULL, 0) == REG_NOMATCH) + return 0; + + if (src->flags & FFS_FLAGS_PROTECTED && + args->force != f_FORCE) { + if (args->verbose == f_VERBOSE) + printf("%llx: %s: protected, skipping " + "compare\n", __poffset, full_name); + return 0; + } + + list_iter_init(&it, &list, LI_FLAG_FWD); + list_for_each(&it, node, node) { + if (node->entry.base == src->base && + node->entry.size == src->size) + return 0; + } + + node = (ffs_entry_node_t *)malloc(sizeof(*node)); + assert(node != NULL); + + memcpy(&node->entry, src, sizeof(node->entry)); + list_add_tail(&list, &node->node); + + ffs_entry_t dst; + if (__ffs_entry_find(__out, full_name, &dst) == false) { + UNEXPECTED("'%s' entry not found '%s'", + args->target, args->name); + return -1; + } + + if (verify_operation(full_name, __in, src, __out, &dst) < 0) + return -1; + + size_t block_size = __in->hdr->block_size; + + if (args->block != NULL) { + if (parse_size(args->block, &block_size) < 0) + return -1; + if (block_size & (__in->hdr->block_size-1)) { + UNEXPECTED("'%x' block size must be multiple " + "of target block size '%x'", + block_size, __in->hdr->block_size); + return -1; + } + } + + if (__ffs_buffer(__in, block_size) < 0) + return -1; + if (__ffs_buffer(__out, block_size) < 0) + return -1; + + RAII(void*, src_block, malloc(block_size), free); + if (src_block == NULL) { + ERRNO(errno); + return -1; + } + RAII(void*, dst_block, malloc(block_size), free); + if (dst_block == NULL) { + ERRNO(errno); + return -1; + } + + size_t data_size = src->actual; + off_t data_offset = 0; + + while (0 < data_size) { + ssize_t rc; + + rc = __ffs_entry_read(__in, full_name, src_block, + data_offset, + min(block_size, data_size)); + if (rc < 0) + return -1; + + rc = __ffs_entry_read(__out, full_name, dst_block, + data_offset, rc); + if (rc < 0) + return -1; + + if (memcmp(src_block, dst_block, block_size) != 0) { + if (args->verbose == f_VERBOSE) + printf("%llx: %s: miscompare at offset" + "'%llx'\n", __poffset, full_name, + data_offset); + UNEXPECTED("data miscompare at offset range " + "[%llx..%llx]", data_offset, + data_offset + block_size - 1); + return -1; + } + + + data_offset += rc; + data_size -= rc; + } + + if (args->verbose == f_VERBOSE) + printf("%llx: %s: compare to '%s'\n", __poffset, + full_name, args->path); + + return 0; + } + + int compare(args_t * args, off_t poffset) + { + __poffset = poffset; + + const char * path = args->path; + const char * target = args->target; + int debug = args->debug; + + RAII(FILE*, inf, fopen_generic(path, "r", debug), fclose); + if (inf == NULL) + return -1; + RAII(FILE*, outf, fopen_generic(target, "r", debug), fclose); + if (outf == NULL) + return -1; + + RAII(ffs_t*, in, __ffs_fopen(inf, poffset), __ffs_fclose); + if (in == NULL) + return -1; + RAII(ffs_t*, out, __ffs_fopen(outf, poffset), __ffs_fclose); + if (out == NULL) + return -1; + + __in = in; + __out = out; + + int rc = __ffs_iterate_entries(in, compare_entry); + if (rc == 1) + rc = -1; + + return rc; + } + + /* ========================= */ + + if (regcomp(&rx, name, REG_ICASE | REG_NOSUB) != 0) { + ERRNO(errno); + return -1; + } + + int rc = command(args, compare); + + regfree(&rx); + + while (!list_empty(&list)) + free(container_of(list_remove_head(&list), + ffs_entry_node_t, node)); + + return rc; +} -- cgit v1.2.1