diff options
Diffstat (limited to 'fpart/src/cmd_read.c')
| -rw-r--r-- | fpart/src/cmd_read.c | 166 |
1 files changed, 166 insertions, 0 deletions
diff --git a/fpart/src/cmd_read.c b/fpart/src/cmd_read.c new file mode 100644 index 0000000..e058377 --- /dev/null +++ b/fpart/src/cmd_read.c @@ -0,0 +1,166 @@ +/* + * 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: read.c + * Author: Shaun Wetzstein <shaun@us.ibm.com> + * Descr: --read implementation + * Date: 01/30/2013 + */ + +#include <sys/types.h> +#include <sys/stat.h> + +#include <fcntl.h> +#include <string.h> +#include <stdint.h> +#include <stdlib.h> +#include <stdio.h> +#include <stdbool.h> +#include <unistd.h> +#include <getopt.h> +#include <errno.h> +#include <ctype.h> +#include <regex.h> + +#include <clib/attribute.h> +#include <clib/assert.h> +#include <clib/list.h> +#include <clib/list_iter.h> +#include <clib/misc.h> +#include <clib/min.h> +#include <clib/err.h> +#include <clib/raii.h> + +#include "main.h" + +int command_read(args_t * args) +{ + assert(args != NULL); + + if (access(args->path, F_OK) == 0 && args->force != f_FORCE) { + UNEXPECTED("output file '%s' already exists, use --force " + "to overwrite\n", args->path); + return -1; + } + + list_t list; + list_init(&list); + + list_iter_t it; + ffs_entry_node_t * node; + + /* ========================= */ + + int read(args_t * args, off_t 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; + + size_t block_size = ffs->hdr->block_size; + + 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; + } + + ffs_entry_t entry; + if (__ffs_entry_find(ffs, args->name, &entry) == false) { + UNEXPECTED("'%s' partition not found => %s", + args->target, args->name); + return -1; + } + + 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); + + RAII(FILE*, out, fopen(args->path, "w+"), fclose); + if (out == NULL) { + ERRNO(errno); + return -1; + } + + size_t data_size = entry.actual; + off_t data_offset = 0; + + while (0 < data_size) { + ssize_t rc; + rc = __ffs_entry_read(ffs, args->name, + block, data_offset, + min(block_size, data_size)); + + rc = fwrite(block, 1, rc, out); + if (rc <= 0 && ferror(out)) { + ERRNO(errno); + return -1; + } + + data_size -= rc; + data_offset += rc; + } + + if (args->verbose == f_VERBOSE) + printf("%llx: %s: wrote '%x' bytes to file '%s'\n", + poffset, args->name, entry.actual, args->path); + + return 0; + } + + /* ========================= */ + + int rc = command(args, read); + + while (!list_empty(&list)) + free(container_of(list_remove_head(&list), + ffs_entry_node_t, node)); + + return rc; +} + |
