/*
 * 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 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <sys/types.h>
#include <stdlib.h>
#include <libintl.h>
#include <unistd.h>
#include "trackio.h"
#include "main.h"
#include "util.h"
#include "bstream.h"
#include "misc_scsi.h"
#include "msgs.h"
#include "device.h"
#include "mmc.h"
#include "transport.h"

void
write_image(void)
{
	bstreamhandle h;
	off_t size;
	int no_size, ret;

	get_media_type(target->d_fd);

	/* DVD+RW does not have blanking and can be overwritten */
	if (device_type != DVD_PLUS_W) {
	(void) check_device(target, CHECK_DEVICE_NOT_READY |
	    CHECK_DEVICE_NOT_WRITABLE | CHECK_MEDIA_IS_NOT_WRITABLE |
	    EXIT_IF_CHECK_FAILED);
	} else {
		(void) check_device(target, CHECK_DEVICE_NOT_READY |
		    EXIT_IF_CHECK_FAILED);
	}

	/*
	 * Simulation writing can't happen on DVD+RW's
	 * or DVD+R's. According to the MMC spec this
	 * operation is not supported. So we should
	 * bail out if the user tries to do a simulation
	 * write.
	 */
	if (simulation && (device_type == DVD_PLUS_W ||
	    device_type == DVD_PLUS)) {
		err_msg(gettext("Media does not support simulated writing.\n"));
		exit(1);
	}

	write_init(TRACK_MODE_DATA);

	if (image_file) {
		h = open_iso_read_stream(image_file);
	} else {
		h = open_stdin_read_stream();
	}

	if (h == NULL) {
		err_msg(gettext("Cannot open %s: %s\n"),
		    image_file ? image_file : "stdin", get_err_str());
		exit(1);
	}
	no_size = 0;
	ret = h->bstr_size(h, &size);
	if (ret == 0) {
		if ((str_errno == STR_ERR_NO_REG_FILE)) {
			no_size = 1;
		} else {
			err_msg(gettext("Cannot stat input file: %s\n"),
			    get_err_str());
			exit(1);
		}
	}
	if ((no_size == 0) && (size == 0)) {
		err_msg(gettext("Input size(0) not valid\n"));
		exit(1);
	}
	if (no_size == 0) {
		off_t cap;
		struct track_info *ti;
		uint_t bsize;

		ti = (struct track_info *)my_zalloc(sizeof (*ti));
		if (write_mode == TAO_MODE)
			if (!build_track_info(target, -1, ti)) {
				err_msg(
				    gettext("Unable to find out writable "
				    "address\n"));
				exit(1);
			}
		if (device_type == CD_RW) {
			if ((cap = get_last_possible_lba(target)) <= 0) {
				if ((cap = read_format_capacity(target->d_fd,
				    &bsize)) <= 0) {
					err_msg(gettext("Unable to determine "
					    "media capacity.  Defaulting to "
					    "650 MB (74 minute) disc.\n"));
					cap = MAX_CD_BLKS;
				}
			}
		} else {
			/*
			 * For DVD drives use read_format_capacity to
			 * find media size, it can be 3.6, 3.9, 4.2,
			 * 4.7, 9.2
			 */
			cap = read_format_capacity(target->d_fd,
			    &bsize);
			/*
			 * Sanity check; Default to 4.7 GB if cap unreasonable
			 */
			if (cap < MAX_CD_BLKS)
				cap = MAX_DVD_BLKS;
		}

		if (device_type == CD_RW)
			cap = (cap + 1 - ti->ti_start_address) * 2048;
		else
			cap *= 2048 + 1;

		if (size > cap) {
			err_msg(gettext("Size required (%lld bytes) is greater "
			    "than available space (%lld bytes).\n"), size, cap);
			exit(1);
		}

		if (device_type == DVD_MINUS) {
			(void) printf(gettext("Preparing to write DVD\n"));

			/* streamed file, we dont know the size to reserve */
			if (no_size == 1) {
				size = cap - 1;
			}

			/* DAO requires that we reserve the size to write */
			if (debug)
				(void) printf(
				    "DAO_MODE:reserving track size of = 0x%x\n",
				    (uint32_t)(size/2048));

			if (!set_reservation(target->d_fd, size/2048)) {
				(void) printf(gettext(
				    "Setting reservation failed\n"));
				exit(1);
			}
		} else if (device_type == DVD_PLUS_W) {
			/*
			 * DVD+RW requires that we format the media before
			 * writing.
			 */
			(void) print_n_flush(gettext("Formatting media..."));
			if (!format_media(target->d_fd)) {
				(void) printf(gettext(
				    "Could not format media\n"));
				exit(1);
			} else {
				int counter;
				uchar_t *di;

				/* poll until format is done */
				di = (uchar_t *)my_zalloc(DISC_INFO_BLOCK_SIZE);
				(void) sleep(10);
				for (counter = 0; counter < 200; counter++) {

				ret = read_disc_info(target->d_fd, di);

				if ((SENSE_KEY(rqbuf) == 2) &&
				    (ASC(rqbuf) == 4)) {
					(void) print_n_flush(".");
					(void) sleep(5);
				} else {
					break;
					}
				}
			}

			(void) printf(gettext("done\n"));
		}


		free(ti);
	}

	write_next_track(TRACK_MODE_DATA, h);

	h->bstr_close(h);
	write_fini();
	fini_device(target);
	exit(0);
}