/* IBM_PROLOG_BEGIN_TAG                                                   */
/* This is an automatically generated prolog.                             */
/*                                                                        */
/* $Source: fpart/src/cmd_copy.c $                                        */
/*                                                                        */
/* OpenPOWER FFS Project                                                  */
/*                                                                        */
/* Contributors Listed Below - COPYRIGHT 2014,2015                        */
/* [+] International Business Machines Corp.                              */
/*                                                                        */
/*                                                                        */
/* Licensed under the Apache License, Version 2.0 (the "License");        */
/* you may not use this file except in compliance with the License.       */
/* You may obtain a copy of the License at                                */
/*                                                                        */
/*     http://www.apache.org/licenses/LICENSE-2.0                         */
/*                                                                        */
/* Unless required by applicable law or agreed to in writing, software    */
/* distributed under the License is distributed on an "AS IS" BASIS,      */
/* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or        */
/* implied. See the License for the specific language governing           */
/* permissions and limitations under the License.                         */
/*                                                                        */
/* IBM_PROLOG_END_TAG                                                     */

/*
 *    File: copy.c
 *  Author: Shaun Wetzstein <shaun@us.ibm.com>
 *   Descr: --copy 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_copy(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;

	ffs_t *__in, *__out;
  	off_t __poffset;

	list_t list;
	list_init(&list);

	list_iter_t it;
	ffs_entry_node_t * node;

	/* ========================= */

	int copy_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 copy\n",
			               __poffset, full_name);
			return 0;
		}

		__ffs_entry_truncate(__out, full_name, src->actual);
		if (args->verbose == f_VERBOSE)
      			printf("%llx: %s: truncate size '%x'\n",
				__poffset, full_name, src->actual);

		uint32_t src_val, dst_val;
		for (uint32_t i=0; i<FFS_USER_WORDS; i++) {
			__ffs_entry_user_get(__in, full_name, i, &src_val);
			__ffs_entry_user_get(__out, full_name, i, &dst_val);
			if (src_val != dst_val)
				if (__ffs_entry_user_put(__out, full_name,
						     i, src_val) < 0)
					return -1;
		}

		if (args->verbose == f_VERBOSE)
			printf("%llx: %s: user[] copy from '%s'\n", __poffset,
				full_name, args->path);

		list_iter_init(&it, &list, LI_FLAG_FWD);
		list_for_each(&it, node, node) {
			ffs_entry_t * e = &node->entry;
			if (e->base == src->base && e->size == src->size) {
				if (args->verbose == f_VERBOSE)
					printf("%llx: %s: skipping copy from "
					       "'%s'\n", __poffset, full_name,
					       args->path);
				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*, block, malloc(block_size), free);
		if (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, block,
					      data_offset,
					      min(block_size, data_size));
			if (rc < 0)
				return -1;

			rc = __ffs_entry_write(__out, full_name, block,
					       data_offset, rc);
			if (rc < 0)
				return -1;

			data_offset += rc;
			data_size -= rc;
		}

		__ffs_fsync(__out);

		if (args->verbose == f_VERBOSE)
			printf("%llx: %s: copy from '%s'\n", __poffset,
				full_name, args->path);

		return 0;
	}

	int copy(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(ffs_t*, in, __ffs_fopen(inf, poffset), __ffs_fclose);
		if (in == NULL)
			return -1;

		if (args->force == f_FORCE) {
			size_t block_size = in->hdr->block_size;
			char block[block_size];

			size_t rc;
			rc = __ffs_entry_read(in, FFS_PARTITION_NAME, block,
					      0, block_size);
			if (rc != block_size) {
				UNEXPECTED("'%s' unable to read '%s' from "
					   "offset '%llx'", args->path,
					   FFS_PARTITION_NAME, poffset);
				return -1;
			}

			RAII(FILE*, outf,
			     fopen_generic(target, "r+", debug), fclose);
			if (outf == NULL)
				return -1;

			if (fseeko(outf, poffset, SEEK_SET) != 0) {
				ERRNO(errno);
				return -1;
			}

			rc = fwrite(block, 1, block_size, outf);
			if (rc <= 0) {
				if (ferror(outf)) {
					ERRNO(errno);
					return -1;
				}
			}

			fflush(outf);
			if (fileno(outf) != -1)
				fsync(fileno(outf));

			if (args->verbose == f_VERBOSE)
				printf("%llx: %s: force partition table\n",
				       poffset, FFS_PARTITION_NAME);
		}

		RAII(FILE*, outf, fopen_generic(target, "r+", debug), fclose);
		if (outf == NULL)
			return -1;
		RAII(ffs_t*, out, __ffs_fopen(outf, poffset), __ffs_fclose);
		if (in == NULL)
			return -1;

		__in = in;
		__out = out;

		int rc = __ffs_iterate_entries(in, copy_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, copy);

	regfree(&rx);

	while (!list_empty(&list))
		free(container_of(list_remove_head(&list),
				  ffs_entry_node_t, node));

	return rc;
}