/*
	FIPS - the First nondestructive Interactive Partition Splitting program

	Module main.cpp

	RCS - Header:
	$Header: c:/daten/fips/source/main/RCS/main.cpp 1.1 1994/05/25 22:20:03 schaefer Exp schaefer $

	Copyright (C) 1993 Arno Schaefer

	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., 675 Mass Ave, Cambridge, MA 02139, USA.


	Report problems and direct all questions to:

	schaefer@rbg.informatik.th-darmstadt.de
*/

#include <stdlib.h>
#include "logdr_st.h"
#include "global.h"
#include "input.h"
#include "fat.h"
#include "fipsspec.h"
#include "host_os.h"


void main (int argc,char *argv[])
{
	evaluate_argument_vector (argc,argv);

	atexit (exit_function);

	if (global.debug_mode) global.open_debugfile(argc,argv);

	notice();

	int drive_number;
	if (global.drive_number_cmdline) drive_number = global.drive_number_cmdline;
	else drive_number = ask_for_drive_number ();

	host_os os;
	char infostring[256];

	if (os.ok () != OK)
	{
		printx ("\nWARNING: FIPS has detected that it is running under %s\n"
			"FIPS should not be used under a multitasking OS. If possible, boot from a DOS\n"
			"disk and then run FIPS. Read FIPS.DOC for more information.\n\n",
			os.information (infostring));
		ask_if_proceed ();
	}

	fips_harddrive harddrive (drive_number);

	if (harddrive.errorcode)
		error ("Error reading Drive Geometry: Errorcode %u",harddrive.errorcode);

	harddrive.reset();

	if (harddrive.errorcode)
		error ("Drive Initialization Failure: Errorcode %u",harddrive.errorcode);

	if (harddrive.rootsector->read ())
		error ("Error reading Root Sector");

	if (global.debug_mode)
	{
		fprintf (global.debugfile,"\nRoot Sector Drive %02Xh:\n\n",drive_number);
		hexwrite (harddrive.rootsector->data,512,global.debugfile);
	}

	fips_partition *partition;

	while (true)
	{
		fips_harddrive hd = harddrive;

		hd.partition_table().get (hd.rootsector);

		printx ("\nPartition Table:\n\n");
		hd.print_partition_table ();

		hd.check();

		int partition_number;
		if (global.partition_number_cmdline) partition_number = global.partition_number_cmdline - 1;
		else partition_number = ask_for_partition_number (hd.partition_table().partition_info);

		partition = new fips_partition (&hd,partition_number);

		int system = partition->partition_info->system;
		if (system == 5)
			error ("Can't split extended Partitions");
		if (system == 0)
			error ("Invalid Partition selected: %u",partition_number + 1);
		if ((system != 1) && (system != 4) && (system != 6))
			error ("Unknown Filesystem: %02Xh",system);

		if (partition->bootsector->read ())
			error ("Error reading Boot Sector");

		if (global.debug_mode)
		{
			fprintf (global.debugfile,"\nBoot Sector Drive %02Xh, Partition %u:\n\n",hd.number,partition->number + 1);
			hexwrite (partition->bootsector->data,512,global.debugfile);
		}

		partition->bpb().get (partition->bootsector);

		printx ("\nBootsector:\n\n");
		partition->print_bpb ();

		partition->info().get (partition->bpb());
		if (global.debug_mode) partition->write_info_debugfile ();

		partition->check();

		fat16 fat1 (partition,1);
		fat16 fat2 (partition,2);

		fat1.check_against (&fat2);

		dword new_part_min_sector = partition->info().start_data + (dword) 4085 * partition->bpb().sectors_per_cluster;
		dword new_part_min_cylinder = (new_part_min_sector + partition->partition_info->start_sector_abs - 1) / (hd.geometry.heads * hd.geometry.sectors) + 1;

		if (new_part_min_cylinder > partition->partition_info->end_cylinder)
			error ("Partition too small - can't split");

		dword min_free_cluster = fat1.min_cluster ();
		dword min_free_sector = partition->info().start_data + (min_free_cluster - 2) * (dword) partition->bpb().sectors_per_cluster;
		dword min_free_cylinder = (min_free_sector + partition->partition_info->start_sector_abs - 1) / (hd.geometry.heads * hd.geometry.sectors) + 1;

		if (min_free_cylinder > partition->partition_info->end_cylinder)
			error ("Last Cylinder is not free");

		if (new_part_min_cylinder < min_free_cylinder) new_part_min_cylinder = min_free_cylinder;

		if (ask_if_save()) save_root_and_boot(&hd,partition);

		dword new_start_cylinder;
		if (global.new_start_cylinder_cmdline)
		{
			new_start_cylinder = global.new_start_cylinder_cmdline;
			if ((new_start_cylinder < new_part_min_cylinder) || (new_start_cylinder > partition->partition_info->end_cylinder))
				error ("Invalid new start cylinder: %lu",new_start_cylinder);
		}
		else new_start_cylinder = ask_for_new_start_cylinder (partition->partition_info->start_cylinder, new_part_min_cylinder,partition->partition_info->end_cylinder, hd.geometry.heads * hd.geometry.sectors);

		fat2.check_empty (new_start_cylinder * hd.geometry.heads * hd.geometry.sectors - partition->partition_info->start_sector_abs);

		hd.calculate_new_root (new_start_cylinder,partition);

		hd.partition_table().put (hd.rootsector);

		hd.partition_table().get (hd.rootsector);

		printx ("\nNew Partition Table:\n\n");
		hd.print_partition_table ();

		hd.check();

		if (ask_if_continue ())
		{
			harddrive = hd;
			break;
		}
	}

	partition->calculate_new_boot ();

	partition->bpb().put (partition->bootsector);

	partition->bpb().get (partition->bootsector);

	printx ("\nNew Bootsector:\n\n");
	partition->print_bpb ();

	partition->info().get (partition->bpb());
	if (global.debug_mode) partition->write_info_debugfile ();

	partition->check();

	if (!global.test_mode)
	{
		ask_for_write_permission();

		if (harddrive.rootsector->write())
			error ("Error writing Root Sector");

		if (partition->bootsector->write ())
			error ("Error writing Boot Sector");

		printx ("Repartitioning complete\n");
	}
}