diff options
Diffstat (limited to 'fcp/src/cmd_erase.c')
| -rw-r--r-- | fcp/src/cmd_erase.c | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/fcp/src/cmd_erase.c b/fcp/src/cmd_erase.c new file mode 100644 index 0000000..972f0be --- /dev/null +++ b/fcp/src/cmd_erase.c @@ -0,0 +1,203 @@ +/* + * 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: cmd_erase.c + * Author: Shaun Wetzstein <shaun@us.ibm.com> + * Descr: erase 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 "misc.h" +#include "main.h" + +static int __erase(args_t * args, off_t offset, entry_list_t * done_list) +{ + assert(args != NULL); + + char * type = args->dst_type; + char * target = args->dst_target; + char * name = args->dst_name; + + size_t fill; + if (args->opt_nr == 1) { + fill = 0xFF; + } else if (args->opt_nr == 2) { + if (parse_number(args->opt[1], &fill) < 0) + return -1; + } + + RAII(FILE*, file, __fopen(type, target, "r+", debug), fclose); + if (file == NULL) + return -1; + if (check_file(target, file, offset) < 0) + return -1; + RAII(ffs_t*, ffs, __ffs_fopen(file, offset), __ffs_fclose); + if (ffs == NULL) + return -1; + + ffs->path = basename(target); + done_list->ffs = ffs; + + if (ffs->count <= 0) + return 0; + + size_t block_size; + if (__ffs_info(ffs, FFS_INFO_BLOCK_SIZE, &block_size) < 0) + return -1; + + if (args->buffer != NULL) { + size_t buffer; + if (parse_size(args->buffer, &buffer) < 0) + return -1; + if (__ffs_buffer(ffs, buffer) < 0) + return -1; + } + + ffs_entry_t parent; + if (__ffs_entry_find(ffs, name, &parent) == false) { + UNEXPECTED("partition entry '%s' not found\n", name); + return -1; + } + + RAII(entry_list_t*, entry_list, entry_list_create(ffs), + entry_list_delete); + if (entry_list == NULL) + return -1; + if (entry_list_add(entry_list, &parent) < 0) + return -1; + + while (list_empty(&entry_list->list) == 0) { + RAII(entry_node_t*, entry_node, + container_of(list_head(&entry_list->list), + entry_node_t, node), free); + + entry_list_remove(entry_list, entry_node); + ffs_entry_t * entry = &entry_node->entry; + + if (entry->type == FFS_TYPE_LOGICAL) { + if (entry_list_add_child(entry_list, entry) < 0) + return -1; + continue; + } + + char full_name[page_size]; + if (__ffs_entry_name(ffs, entry, full_name, + sizeof full_name) < 0) + return -1; + + if (args->protected != f_PROTECTED) { + if (entry->flags & FFS_FLAGS_PROTECTED) { + if (args->verbose == f_VERBOSE) + fprintf(stderr, "%8llx: %s: protected " + "(skip)\n", offset, full_name); + continue; + } + } + + if (__ffs_entry_truncate(ffs, full_name, 0ULL) < 0) { + ERRNO(errno); + return -1; + } + if (args->verbose == f_VERBOSE) + fprintf(stderr, "%8llx: %s: trunc size '%x' (done)\n", + offset, full_name, 0); + + if (entry_list_exists(done_list, entry) == 1) { + if (args->verbose == f_VERBOSE) + fprintf(stderr, "%8llx: %s: erase partition " + "(skip)\n", offset, full_name); + continue; + } + + if (entry_list_add(done_list, entry) < 0) + return -1; + + if (fcp_erase_entry(ffs, full_name, (char)fill) < 0) + return -1; + + if (args->verbose == f_VERBOSE) + fprintf(stderr, "%8llx: %s: erase partition (done)\n", + offset, full_name); + } + + return 0; +} + +int command_erase(args_t * args) +{ + assert(args != NULL); + + int rc = 0; + + RAII(entry_list_t*, done_list, entry_list_create(NULL), + entry_list_delete); + if (done_list == NULL) + return -1; + + char * end = (char *)args->offset; + while (rc == 0 && end != NULL && *end != '\0') { + errno = 0; + off_t offset = strtoull(end, &end, 0); + if (end == NULL || errno != 0) { + UNEXPECTED("invalid --offset specified '%s'", + args->offset); + return -1; + } + + if (*end != ',' && *end != ':' && *end != '\0') { + UNEXPECTED("invalid --offset separator " + "character '%c'", *end); + return -1; + } + + rc = __erase(args, offset, done_list); + if (rc < 0) + break; + + if (*end == '\0') + break; + end++; + } + + return rc; +} |
