/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
 * Copyright 2012 Nexenta Systems, Inc. All rights reserved.
 */

#include <stdio.h>
#include <errno.h>
#include <unistd.h>
#include <fcntl.h>
#include <assert.h>
#include <locale.h>
#include <strings.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/multiboot.h>
#include <sys/sysmacros.h>

#include "installboot.h"
#include "./../common/bblk_einfo.h"
#include "./../common/boot_utils.h"
#include "./../common/mboot_extra.h"

#ifndef	TEXT_DOMAIN
#define	TEXT_DOMAIN	"SUNW_OST_OSCMD"
#endif

/*
 * SPARC bootblock installation:
 *
 * The bootblock resides in blocks 1 to 15 (disk label is at block 0).
 * The ZFS boot block is larger than what will fit into these first 7.5K so we
 * break it up and write the remaining portion into the ZFS provided boot block
 * region at offset 512K. If versioning is requested, we add a multiboot
 * header at the end of the bootblock, followed by the extra payload area and
 * place the extended information structure within the latter.
 */

static boolean_t	force_update = B_FALSE;
static boolean_t	do_getinfo = B_FALSE;
static boolean_t	do_version = B_FALSE;
static boolean_t	do_mirror_bblk = B_FALSE;
static boolean_t	strip = B_FALSE;
static boolean_t	verbose_dump = B_FALSE;

static char		*update_str;
static int		tgt_fs_type = TARGET_IS_UFS;
char			mboot_scan[MBOOT_SCAN_SIZE];

/* Function prototypes. */
static int read_bootblock_from_file(char *, ib_data_t *data);
static int read_bootblock_from_disk(int, ib_bootblock_t *);
static void add_bootblock_einfo(ib_bootblock_t *, char *);
static int prepare_bootblock(ib_data_t *, char *);
static int write_zfs_bootblock(ib_data_t *);
static int write_bootblock(ib_data_t *);
static int open_device(ib_device_t *);
static int init_device(ib_device_t *, char *);
static void cleanup_device(ib_device_t *);
static int commit_to_disk(ib_data_t *, char *);
static int handle_install(char *, char **);
static int handle_getinfo(char *, char **);
static int handle_mirror(char *, char **);
static boolean_t is_update_necessary(ib_data_t *, char *);
static int propagate_bootblock(ib_data_t *, ib_data_t *, char *);
static void usage(char *);

static int
read_bootblock_from_file(char *file, ib_data_t *data)
{
	ib_device_t	*device = &data->device;
	ib_bootblock_t	*bblock = &data->bootblock;
	struct stat 	sb;
	uint32_t	buf_size;
	int		fd = -1;
	int		retval = BC_ERROR;

	assert(data != NULL);
	assert(file != NULL);

	fd = open(file, O_RDONLY);
	if (fd == -1) {
		BOOT_DEBUG("Error opening %s\n", file);
		perror("open");
		goto out;
	}

	if (fstat(fd, &sb) == -1) {
		BOOT_DEBUG("Error getting information (stat) about %s", file);
		perror("stat");
		goto outfd;
	}

	bblock->file_size = sb.st_size;
	BOOT_DEBUG("bootblock file size is %x\n", bblock->file_size);

	/* UFS and HSFS bootblocks need to fit in the reserved 7.5K. */
	if (!is_zfs(device->type)) {
		buf_size = P2ROUNDUP(bblock->file_size, SECTOR_SIZE);
		if (buf_size > BBLK_DATA_RSVD_SIZE) {
			BOOT_DEBUG("boot block size is bigger than allowed\n");
			goto outfd;
		}
	} else {
		buf_size = P2ROUNDUP(bblock->file_size + SECTOR_SIZE,
		    SECTOR_SIZE);
		if (buf_size > BBLK_DATA_RSVD_SIZE + MBOOT_SCAN_SIZE) {
			(void) fprintf(stderr, gettext("WARNING, bootblock size"
			    " does not allow to place extended versioning "
			    "information.. skipping\n"));
			do_version = B_FALSE;
		}
	}

	bblock->buf_size = buf_size;
	BOOT_DEBUG("bootblock in-memory buffer size is %x\n",
	    bblock->buf_size);

	bblock->buf = malloc(buf_size);
	if (bblock->buf == NULL) {
		perror(gettext("Memory allocation failure"));
		goto outbuf;
	}
	bblock->file = bblock->buf;

	if (read(fd, bblock->file, bblock->file_size) != bblock->file_size) {
		BOOT_DEBUG("Read from %s failed\n", file);
		perror("read");
		goto outfd;
	}

	/* If not on ZFS, we are done here. */
	if (!is_zfs(device->type)) {
		BOOT_DEBUG("Reading of the bootblock done\n");
		retval = BC_SUCCESS;
		goto outfd;
	}
	/*
	 * We place the multiboot header right after the file, followed by
	 * the extended information structure.
	 */
	bblock->mboot = (multiboot_header_t *)(bblock->file +
	    P2ROUNDUP(bblock->file_size, 8));
	bblock->extra = (char *)bblock->mboot + sizeof (multiboot_header_t);
	BOOT_DEBUG("mboot at %p, extra at %p, buf=%p (size=%d)\n",
	    bblock->mboot, bblock->extra, bblock->buf, bblock->buf_size);

	(void) close(fd);
	return (BC_SUCCESS);

outbuf:
	(void) free(bblock->buf);
	bblock->buf = NULL;
outfd:
	(void) close(fd);
out:
	return (retval);
}

static int
read_bootblock_from_disk(int dev_fd, ib_bootblock_t *bblock)
{
	char			*dest;
	uint32_t		size;
	uint32_t		buf_size;
	uint32_t		mboot_off;
	multiboot_header_t	*mboot;

	assert(bblock != NULL);
	assert(dev_fd != -1);

	/*
	 * The ZFS bootblock is divided in two parts, but the fake multiboot
	 * header can only be in the second part (the one contained in the ZFS
	 * reserved area).
	 */
	if (read_in(dev_fd, mboot_scan, sizeof (mboot_scan),
	    BBLK_ZFS_EXTRA_OFF) != BC_SUCCESS) {
		BOOT_DEBUG("Error reading ZFS reserved area\n");
		perror("read");
		return (BC_ERROR);
	}

	/* No multiboot means no chance of knowing bootblock size */
	if (find_multiboot(mboot_scan, sizeof (mboot_scan), &mboot_off)
	    != BC_SUCCESS) {
		BOOT_DEBUG("Unable to find multiboot header\n");
		return (BC_NOEXTRA);
	}
	mboot = (multiboot_header_t *)(mboot_scan + mboot_off);

	/*
	 * Currently, the amount of space reserved for extra information
	 * is "fixed". We may have to scan for the terminating extra payload
	 * in the future.
	 */
	size = mboot->load_end_addr - mboot->load_addr;
	buf_size = P2ROUNDUP(size + SECTOR_SIZE, SECTOR_SIZE);
	bblock->file_size = size;

	bblock->buf = malloc(buf_size);
	if (bblock->buf == NULL) {
		BOOT_DEBUG("Unable to allocate enough memory to read"
		    " the extra bootblock from the disk\n");
		perror(gettext("Memory allocation failure"));
		return (BC_ERROR);
	}
	bblock->buf_size = buf_size;

	dest = bblock->buf;
	size = BBLK_DATA_RSVD_SIZE;

	if (read_in(dev_fd, dest, size, SECTOR_SIZE) != BC_SUCCESS) {
		BOOT_DEBUG("Error reading first %d bytes of the bootblock\n",
		    size);
		(void) free(bblock->buf);
		bblock->buf = NULL;
		return (BC_ERROR);
	}

	dest += BBLK_DATA_RSVD_SIZE;
	size = bblock->buf_size - BBLK_DATA_RSVD_SIZE;

	if (read_in(dev_fd, dest, size, BBLK_ZFS_EXTRA_OFF) != BC_SUCCESS) {
		BOOT_DEBUG("Error reading ZFS reserved area the second time\n");
		(void) free(bblock->buf);
		bblock->buf = NULL;
		return (BC_ERROR);
	}

	/* Update pointers. */
	bblock->file = bblock->buf;
	bblock->mboot_off = mboot_off;
	bblock->mboot = (multiboot_header_t *)(bblock->buf + bblock->mboot_off
	    + BBLK_DATA_RSVD_SIZE);
	bblock->extra = (char *)bblock->mboot + sizeof (multiboot_header_t);
	bblock->extra_size = bblock->buf_size - bblock->mboot_off
	    - BBLK_DATA_RSVD_SIZE - sizeof (multiboot_header_t);
	return (BC_SUCCESS);
}

static boolean_t
is_update_necessary(ib_data_t *data, char *updt_str)
{
	bblk_einfo_t	*einfo;
	bblk_hs_t	bblock_hs;
	ib_bootblock_t	bblock_disk;
	ib_bootblock_t	*bblock_file = &data->bootblock;
	ib_device_t	*device = &data->device;
	int		dev_fd = device->fd;

	assert(data != NULL);
	assert(device->fd != -1);

	/* Nothing to do if we are not updating a ZFS bootblock. */
	if (!is_zfs(device->type))
		return (B_TRUE);

	bzero(&bblock_disk, sizeof (ib_bootblock_t));

	if (read_bootblock_from_disk(dev_fd, &bblock_disk) != BC_SUCCESS) {
		BOOT_DEBUG("Unable to read bootblock from %s\n", device->path);
		return (B_TRUE);
	}

	einfo = find_einfo(bblock_disk.extra, bblock_disk.extra_size);
	if (einfo == NULL) {
		BOOT_DEBUG("No extended information available\n");
		return (B_TRUE);
	}

	if (!do_version || updt_str == NULL) {
		(void) fprintf(stdout, "WARNING: target device %s has a "
		    "versioned bootblock that is going to be overwritten by a "
		    "non versioned one\n", device->path);
		return (B_TRUE);
	}

	if (force_update) {
		BOOT_DEBUG("Forcing update of %s bootblock\n", device->path);
		return (B_TRUE);
	}

	BOOT_DEBUG("Ready to check installed version vs %s\n", updt_str);

	bblock_hs.src_buf = (unsigned char *)bblock_file->file;
	bblock_hs.src_size = bblock_file->file_size;

	return (einfo_should_update(einfo, &bblock_hs, updt_str));
}

static void
add_bootblock_einfo(ib_bootblock_t *bblock, char *updt_str)
{
	bblk_hs_t	hs;
	uint32_t	avail_space;

	assert(bblock != NULL);

	if (updt_str == NULL) {
		BOOT_DEBUG("WARNING: no update string passed to "
		    "add_bootblock_einfo()\n");
		return;
	}

	/* Fill bootblock hashing source information. */
	hs.src_buf = (unsigned char *)bblock->file;
	hs.src_size = bblock->file_size;
	/* How much space for the extended information structure? */
	avail_space = bblock->buf_size - P2ROUNDUP(bblock->file_size, 8);
	/* Place the extended information structure. */
	add_einfo(bblock->extra, updt_str, &hs, avail_space);
}


static int
prepare_bootblock(ib_data_t *data, char *updt_str)
{
	ib_device_t		*device = &data->device;
	ib_bootblock_t		*bblock = &data->bootblock;
	multiboot_header_t	*mboot;

	assert(data != NULL);

	/* Nothing to do if we are not on ZFS. */
	if (!is_zfs(device->type))
		return (BC_SUCCESS);

	/*
	 * Write the fake multiboot structure followed by the extra information
	 * data. Both mboot and extra pointers have already been filled up to
	 * point to the right location in the buffer. We prepare the fake
	 * multiboot regardless if versioning was requested or not because
	 * we need it for mirroring support.
	 */
	assert(bblock->mboot != NULL);
	assert(bblock->extra != NULL);

	mboot = bblock->mboot;

	mboot->magic = MB_HEADER_MAGIC;
	mboot->flags = MB_HEADER_FLAGS_64;
	mboot->checksum = -(mboot->flags + mboot->magic);
	/*
	 * Flags include the AOUT_KLUDGE and we use the extra members to specify
	 * the size of the bootblock.
	 */
	mboot->header_addr = bblock->mboot_off;
	mboot->load_addr = 0;
	mboot->load_end_addr = bblock->file_size;

	/*
	 * Now that we have the mboot header in place, we can add the extended
	 * versioning information. Since the multiboot header has been placed
	 * after the file image, the hashing will still reflect the one of the
	 * file on the disk.
	 */
	if (do_version)
		add_bootblock_einfo(bblock, updt_str);

	return (BC_SUCCESS);
}

static int
write_zfs_bootblock(ib_data_t *data)
{
	ib_device_t	*device = &data->device;
	ib_bootblock_t	*bblock = &data->bootblock;
	char		*bufptr;
	uint32_t	size;

	assert(data != NULL);
	assert(device->fd != -1);

	/*
	 * In the ZFS case we actually perform two different steps:
	 * - write the first 15 blocks of the bootblock to the reserved disk
	 *   blocks.
	 * - write the remaining blocks in the ZFS reserved area at offset
	 *   512K.
	 */
	bufptr = bblock->buf;
	size = BBLK_DATA_RSVD_SIZE;

	if (write_out(device->fd, bufptr, size, SECTOR_SIZE) != BC_SUCCESS) {
		BOOT_DEBUG("Error writing first 15 blocks of %s\n",
		    device->path);
		perror("write");
		return (BC_ERROR);
	}

	bufptr += BBLK_DATA_RSVD_SIZE;
	size = bblock->buf_size - BBLK_DATA_RSVD_SIZE;

	if (write_out(device->fd, bufptr, size, BBLK_ZFS_EXTRA_OFF)
	    != BC_SUCCESS) {
		BOOT_DEBUG("Error writing the second part of ZFS bootblock "
		    "to %s at offset %d\n", device->path, BBLK_ZFS_EXTRA_OFF);
		return (BC_ERROR);
	}
	return (BC_SUCCESS);
}

static int
write_bootblock(ib_data_t *data)
{
	ib_device_t	*device = &data->device;
	ib_bootblock_t	*bblock = &data->bootblock;
	int		ret;

	assert(data != NULL);

	/*
	 * If we are on UFS or HSFS we simply write out to the reserved
	 * blocks (1 to 15) the boot block.
	 */
	if (!is_zfs(device->type)) {
		if (write_out(device->fd, bblock->buf, bblock->buf_size,
		    SECTOR_SIZE) != BC_SUCCESS) {
			BOOT_DEBUG("Error writing bootblock to %s\n",
			    device->path);
			return (BC_ERROR);
		} else {
			return (BC_SUCCESS);
		}
	} else {
		ret = write_zfs_bootblock(data);
		return (ret);
	}
}

static int
open_device(ib_device_t *device)
{
	struct stat	statbuf;

	device->fd = open(device->path, O_RDWR);
	if (device->fd == -1) {
		BOOT_DEBUG("Unable to open %s\n", device->path);
		perror("open");
		return (BC_ERROR);
	}

	if (fstat(device->fd, &statbuf) != 0) {
		BOOT_DEBUG("Unable to stat %s\n", device->path);
		perror("stat");
		(void) close(device->fd);
		return (BC_ERROR);
	}

	if (S_ISCHR(statbuf.st_mode) == 0) {
		(void) fprintf(stderr, gettext("%s: Not a character device\n"),
		    device->path);
		return (BC_ERROR);
	}

	return (BC_SUCCESS);
}

static int
init_device(ib_device_t *device, char *path)
{
	bzero(device, sizeof (*device));
	device->fd = -1;

	device->path = strdup(path);
	if (path == NULL) {
		perror(gettext("Memory allocation failure"));
		return (BC_ERROR);
	}

	device->type = tgt_fs_type;
	if (open_device(device) != BC_SUCCESS)
		return (BC_ERROR);

	return (BC_SUCCESS);
}

static void
cleanup_device(ib_device_t *device)
{
	free(device->path);
	bzero(device, sizeof (*device));

	if (device->fd != -1)
		(void) close(device->fd);
}

static void
cleanup_bootblock(ib_bootblock_t *bblock)
{
	free(bblock->buf);
	bzero(bblock, sizeof (ib_bootblock_t));
}

/*
 * Propagate the bootblock on the source disk to the destination disk and
 * version it with 'updt_str' in the process. Since we cannot trust any data
 * on the attaching disk, we do not perform any specific check on a potential
 * target extended information structure and we just blindly update.
 */
static int
propagate_bootblock(ib_data_t *src, ib_data_t *dest, char *updt_str)
{
	ib_bootblock_t	*src_bblock = &src->bootblock;
	ib_bootblock_t	*dest_bblock = &dest->bootblock;
	uint32_t	buf_size;

	assert(src != NULL);
	assert(dest != NULL);

	cleanup_bootblock(dest_bblock);

	if (updt_str != NULL) {
		do_version = B_TRUE;
	} else {
		do_version = B_FALSE;
	}

	buf_size = src_bblock->file_size + SECTOR_SIZE;

	dest_bblock->buf_size = P2ROUNDUP(buf_size, SECTOR_SIZE);
	dest_bblock->buf = malloc(dest_bblock->buf_size);
	if (dest_bblock->buf == NULL) {
		perror(gettext("Memory Allocation Failure"));
		return (BC_ERROR);
	}
	dest_bblock->file = dest_bblock->buf;
	dest_bblock->file_size = src_bblock->file_size;
	(void) memcpy(dest_bblock->file, src_bblock->file,
	    dest_bblock->file_size);

	dest_bblock->mboot = (multiboot_header_t *)(dest_bblock->file +
	    P2ROUNDUP(dest_bblock->file_size, 8));
	dest_bblock->extra = (char *)dest_bblock->mboot +
	    sizeof (multiboot_header_t);

	(void) fprintf(stdout, gettext("Propagating %s bootblock to %s\n"),
	    src->device.path, dest->device.path);

	return (commit_to_disk(dest, updt_str));
}

static int
commit_to_disk(ib_data_t *data, char *update_str)
{
	assert(data != NULL);

	if (prepare_bootblock(data, update_str) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Error updating the bootblock "
		    "image\n"));
		return (BC_ERROR);
	}

	if (write_bootblock(data) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Error writing bootblock to "
		    "disk\n"));
		return (BC_ERROR);
	}

	return (BC_SUCCESS);
}


/*
 * Install a new bootblock on the given device. handle_install() expects argv
 * to contain 2 parameters (the target device path and the path to the
 * bootblock.
 *
 * Returns:	BC_SUCCESS - if the installation is successful
 *		BC_ERROR   - if the installation failed
 *		BC_NOUPDT  - if no installation was performed because the
 *		             version currently installed is more recent than the
 *			     supplied one.
 *
 */
static int
handle_install(char *progname, char **argv)
{
	ib_data_t	install_data;
	char		*bootblock = NULL;
	char		*device_path = NULL;
	int		ret = BC_ERROR;

	bootblock = strdup(argv[0]);
	device_path = strdup(argv[1]);

	if (!device_path || !bootblock) {
		(void) fprintf(stderr, gettext("Missing parameter"));
		usage(progname);
		goto out;
	}

	BOOT_DEBUG("device path: %s, bootblock file path: %s\n", device_path,
	    bootblock);
	bzero(&install_data, sizeof (ib_data_t));

	if (init_device(&install_data.device, device_path) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Unable to open device %s\n"),
		    device_path);
		goto out;
	}

	if (read_bootblock_from_file(bootblock, &install_data) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Error reading %s\n"),
		    bootblock);
		goto out_dev;
	}
	/* Versioning is only supported for the ZFS bootblock. */
	if (do_version && !is_zfs(install_data.device.type)) {
		(void) fprintf(stderr, gettext("Versioning is only supported on"
		    " ZFS... skipping.\n"));
		do_version = B_FALSE;
	}

	/*
	 * is_update_necessary() will take care of checking if versioning and/or
	 * forcing the update have been specified. It will also emit a warning
	 * if a non-versioned update is attempted over a versioned bootblock.
	 */
	if (!is_update_necessary(&install_data, update_str)) {
		(void) fprintf(stderr, gettext("bootblock version installed "
		    "on %s is more recent or identical\n"
		    "Use -F to override or install without the -u option\n"),
		    device_path);
		ret = BC_NOUPDT;
		goto out_dev;
	}

	BOOT_DEBUG("Ready to commit to disk\n");
	ret = commit_to_disk(&install_data, update_str);

out_dev:
	cleanup_device(&install_data.device);
out:
	free(bootblock);
	free(device_path);
	return (ret);
}

/*
 * Retrieves from a device the extended information (einfo) associated to the
 * installed bootblock.
 * Expects one parameter, the device path, in the form: /dev/rdsk/c?[t?]d?s0.
 * Returns:
 *        - BC_SUCCESS (and prints out einfo contents depending on 'flags')
 *	  - BC_ERROR (on error)
 *        - BC_NOEINFO (no extended information available)
 */
static int
handle_getinfo(char *progname, char **argv)
{

	ib_data_t	data;
	ib_bootblock_t	*bblock = &data.bootblock;
	ib_device_t	*device = &data.device;
	bblk_einfo_t	*einfo;
	uint8_t		flags = 0;
	uint32_t	size;
	char		*device_path;
	int		retval = BC_ERROR;
	int		ret;

	device_path = strdup(argv[0]);
	if (!device_path) {
		(void) fprintf(stderr, gettext("Missing parameter"));
		usage(progname);
		goto out;
	}

	bzero(&data, sizeof (ib_data_t));
	BOOT_DEBUG("device path: %s\n", device_path);

	if (init_device(device, device_path) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Unable to gather device "
		    "information from %s\n"), device_path);
		goto out_dev;
	}

	if (!is_zfs(device->type)) {
		(void) fprintf(stderr, gettext("Versioning only supported on "
		    "ZFS\n"));
		goto out_dev;
	}

	ret = read_bootblock_from_disk(device->fd, bblock);
	if (ret == BC_ERROR) {
		(void) fprintf(stderr, gettext("Error reading bootblock from "
		    "%s\n"), device_path);
		goto out_dev;
	}

	if (ret == BC_NOEXTRA) {
		BOOT_DEBUG("No multiboot header found on %s, unable "
		    "to locate extra information area (old/non versioned "
		    "bootblock?) \n", device_path);
		(void) fprintf(stderr, gettext("No extended information "
		    "found\n"));
		retval = BC_NOEINFO;
		goto out_dev;
	}

	einfo = find_einfo(bblock->extra, bblock->extra_size);
	if (einfo == NULL) {
		retval = BC_NOEINFO;
		(void) fprintf(stderr, gettext("No extended information "
		    "found\n"));
		goto out_dev;
	}

	/* Print the extended information. */
	if (strip)
		flags |= EINFO_EASY_PARSE;
	if (verbose_dump)
		flags |= EINFO_PRINT_HEADER;

	size = bblock->buf_size - P2ROUNDUP(bblock->file_size, 8) -
	    sizeof (multiboot_header_t);
	print_einfo(flags, einfo, size);
	retval = BC_SUCCESS;

out_dev:
	cleanup_device(&data.device);
out:
	free(device_path);
	return (retval);

}

/*
 * Attempt to mirror (propagate) the current bootblock over the attaching disk.
 *
 * Returns:
 *	- BC_SUCCESS (a successful propagation happened)
 *	- BC_ERROR (an error occurred)
 *	- BC_NOEXTRA (it is not possible to dump the current bootblock since
 *			there is no multiboot information)
 */
static int
handle_mirror(char *progname, char **argv)
{
	ib_data_t	curr_data;
	ib_data_t	attach_data;
	ib_device_t	*curr_device = &curr_data.device;
	ib_device_t	*attach_device = &attach_data.device;
	ib_bootblock_t	*bblock_curr = &curr_data.bootblock;
	ib_bootblock_t	*bblock_attach = &attach_data.bootblock;
	bblk_einfo_t	*einfo_curr = NULL;
	char		*curr_device_path;
	char		*attach_device_path;
	char		*updt_str = NULL;
	int		retval = BC_ERROR;
	int		ret;

	curr_device_path = strdup(argv[0]);
	attach_device_path = strdup(argv[1]);

	if (!curr_device_path || !attach_device_path) {
		(void) fprintf(stderr, gettext("Missing parameter"));
		usage(progname);
		goto out;
	}
	BOOT_DEBUG("Current device path is: %s, attaching device path is: "
	    " %s\n", curr_device_path, attach_device_path);

	bzero(&curr_data, sizeof (ib_data_t));
	bzero(&attach_data, sizeof (ib_data_t));

	if (tgt_fs_type != TARGET_IS_ZFS) {
		(void) fprintf(stderr, gettext("Mirroring is only supported on "
		    "ZFS\n"));
		return (BC_ERROR);
	}

	if (init_device(curr_device, curr_device_path) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Unable to gather device "
		    "information from %s (current device)\n"),
		    curr_device_path);
		goto out_currdev;
	}

	if (init_device(attach_device, attach_device_path) != BC_SUCCESS) {
		(void) fprintf(stderr, gettext("Unable to gather device "
		    "information from %s (attaching device)\n"),
		    attach_device_path);
		goto out_devs;
	}

	ret = read_bootblock_from_disk(curr_device->fd, bblock_curr);
	if (ret == BC_ERROR) {
		BOOT_DEBUG("Error reading bootblock from %s\n",
		    curr_device->path);
		retval = BC_ERROR;
		goto out_devs;
	}

	if (ret == BC_NOEXTRA) {
		BOOT_DEBUG("No multiboot header found on %s, unable to retrieve"
		    " the bootblock\n", curr_device->path);
		retval = BC_NOEXTRA;
		goto out_devs;
	}

	einfo_curr = find_einfo(bblock_curr->extra, bblock_curr->extra_size);
	if (einfo_curr != NULL)
		updt_str = einfo_get_string(einfo_curr);

	retval = propagate_bootblock(&curr_data, &attach_data, updt_str);
	cleanup_bootblock(bblock_curr);
	cleanup_bootblock(bblock_attach);
out_devs:
	cleanup_device(attach_device);
out_currdev:
	cleanup_device(curr_device);
out:
	free(curr_device_path);
	free(attach_device_path);
	return (retval);
}

#define	USAGE_STRING	"Usage: %s [-h|-f|-F fstype|-u verstr] bootblk "       \
			"raw-device\n"					       \
			"\t%s [-e|-V] -i -F zfs raw-device\n"	               \
			"\t%s -M -F zfs raw-device attach-raw-device\n"        \
			"\tfstype is one of: 'ufs', 'hsfs' or 'zfs'\n"

#define	CANON_USAGE_STR	gettext(USAGE_STRING)

static void
usage(char *progname)
{
	(void) fprintf(stdout, CANON_USAGE_STR, progname, progname, progname);
}

int
main(int argc, char **argv)
{
	int	opt;
	int	params = 2;
	int	ret;
	char	*progname;
	char	**handle_args;

	(void) setlocale(LC_ALL, "");
	(void) textdomain(TEXT_DOMAIN);

	while ((opt = getopt(argc, argv, "F:efiVMndhu:")) != EOF) {
		switch (opt) {
		case 'F':
			if (strcmp(optarg, "ufs") == 0) {
				tgt_fs_type = TARGET_IS_UFS;
			} else if (strcmp(optarg, "hsfs") == 0) {
				tgt_fs_type = TARGET_IS_HSFS;
			} else if (strcmp(optarg, "zfs") == 0) {
				tgt_fs_type = TARGET_IS_ZFS;
			} else {
				(void) fprintf(stderr, gettext("Wrong "
				    "filesystem specified\n\n"));
				usage(argv[0]);
				exit(BC_ERROR);
			}
			break;
		case 'e':
			strip = B_TRUE;
			break;
		case 'f':
			force_update = B_TRUE;
			break;
		case 'V':
			verbose_dump = B_TRUE;
			break;
		case 'i':
			do_getinfo = B_TRUE;
			params = 1;
			break;
		case 'u':
			do_version = B_TRUE;

			update_str = malloc(strlen(optarg) + 1);
			if (update_str == NULL) {
				perror(gettext("Memory allocation failure"));
				exit(BC_ERROR);
			}
			(void) strlcpy(update_str, optarg, strlen(optarg) + 1);
			break;
		case 'M':
			do_mirror_bblk = B_TRUE;
			break;
		case 'h':
			usage(argv[0]);
			exit(BC_SUCCESS);
			break;
		case 'd':
			boot_debug = B_TRUE;
			break;
		case 'n':
			nowrite = B_TRUE;
			break;
		default:
			/* fall through to process non-optional args */
			break;
		}
	}

	/* check arguments */
	if (argc != optind + params) {
		usage(argv[0]);
		exit(BC_ERROR);
	}
	progname = argv[0];
	handle_args = argv + optind;

	/* check options. */
	if (do_getinfo && do_mirror_bblk) {
		(void) fprintf(stderr, gettext("Only one of -M and -i can be "
		    "specified at the same time\n"));
		usage(progname);
		exit(BC_ERROR);
	}

	if (nowrite)
		(void) fprintf(stdout, gettext("Dry run requested. Nothing will"
		    " be written to disk.\n"));

	if (do_getinfo) {
		ret = handle_getinfo(progname, handle_args);
	} else if (do_mirror_bblk) {
		ret = handle_mirror(progname, handle_args);
	} else {
		ret = handle_install(progname, handle_args);
	}
	return (ret);
}