/*	$OpenBSD: fdisk.c,v 1.25 1997/10/16 01:47:10 deraadt Exp $	*/

/*
 * Copyright (c) 1997 Tobias Weingartner
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    This product includes software developed by Tobias Weingartner.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <paths.h>
#include <sys/types.h>
#include <sys/fcntl.h>
#include <sys/disklabel.h>
#include "disk.h"
#include "user.h"

#define _PATH_MBR _PATH_BOOTDIR "mbr"


void
usage()
{
	extern char * __progname;
	fprintf(stderr, "usage: %s "
		"[-ie] [-f mbrboot] [-c cyl -h head -s sect] disk\n"
		"\t-i: initialize disk with virgin MBR\n"
		"\t-e: edit MBRs on disk interactively\n"
		"\t-f: specify non-standard MBR template\n"
		"\t-chs: specify disk geometry\n"
		"`disk' may be of the forms: sd0 or /dev/rsd0c.\n",
		__progname);
	exit(1);
}


int
main(argc, argv)
	int argc;
	char **argv;
{
	int ch, fd;
	int i_flag = 0, m_flag = 0;
	int c_arg = 0, h_arg = 0, s_arg = 0;
	disk_t disk;
	DISK_metrics *usermetrics;
	char *mbrfile = _PATH_MBR;
	mbr_t mbr;
	char mbr_buf[DEV_BSIZE];

	while((ch = getopt(argc, argv, "ief:c:h:s:")) != -1){
		switch(ch){
			case 'i':
				i_flag = 1;
				break;

			case 'e':
				m_flag = 1;
				break;

			case 'f':
				mbrfile = optarg;
				break;

			case 'c':
				c_arg = atoi(optarg);
				if(c_arg < 0 || c_arg > 1023)
					errx(1, "Cylinder argument out of range.");
				break;

			case 'h':
				h_arg = atoi(optarg);
				if(h_arg < 0 || h_arg > 255)
					errx(1, "Head argument out of range.");
				break;

			case 's':
				s_arg = atoi(optarg);
				if(s_arg < 1 || s_arg > 63)
					errx(1, "Sector argument out of range.");
				break;

			default:
				usage();
		}
	}
	argc -= optind;
	argv += optind;

	/* Argument checking */
	if(argc != 1)
		usage();
	else
		disk.name = argv[0];

	/* Put in supplied geometry if there */
	if(c_arg | h_arg | s_arg){
		usermetrics = malloc(sizeof(DISK_metrics));
		if(usermetrics != NULL){
			if(c_arg) usermetrics->cylinders = c_arg;
			else errx(1, "Please specify a full geometry with [-chs].");
			if(h_arg) usermetrics->heads = h_arg;
			else errx(1, "Please specify a full geometry with [-chs].");
			if(s_arg) usermetrics->sectors = s_arg;
			else errx(1, "Please specify a full geometry with [-chs].");
		}
		usermetrics->size = c_arg * h_arg * s_arg;
	}else
		usermetrics = NULL;

	/* Get the geometry */
	if(DISK_getmetrics(&disk, usermetrics))
		errx(1, "Can't get disk geometry, please use [-chs] to specify.");

	/* Parse mbr template, to pass on later */
	if((fd = open(mbrfile, O_RDONLY)) < 0)
		err(1, "open mbr file");
	MBR_read(fd, 0, mbr_buf);
	close(fd);
	MBR_parse(mbr_buf, &mbr);


	/* Print out current MBRs on disk */
	if((i_flag + m_flag) == 0)
		exit(USER_print_disk(&disk));

	/* Punt if no i or m */
	if((i_flag + m_flag) != 1)
		usage();


	/* Now do what we are supposed to */
	if(i_flag)
		USER_init(&disk, &mbr);

	if(m_flag)
		USER_modify(&disk, &mbr, 0);

	return(0);
}