/*
 * Xenogears extractor by Nicolas "Pixel" Noble (nicolas@nobis-crew.org)
 * Highly based upon Yazoo's Chrono Cross CD extractor
 * 
 * ******** Original disclaimer by Yazoo ********
 * 
 * Chrono Cross CD extractor Copyright 2000-2001 by Yazoo (hamm@efrei.fr) Revision 0.1b ANSI C
 * 
 * 
 * Features:
 * 
 * Dump the complete content of Chrono Chross CD1/CD2 US and Japanese version It requires a iso
 * named Chrono1.iso in the same directory
 * 
 * Todo list:
 * 
 * Find a way to locate end of the last file Better support for CD2 Dump in subdirectory according
 * to CD1/CD2 repartition Recompilation in Visual C++ 6 for disk speed optimisation Source comment
 * and reorganisation Log feature (Optional since you can redirect output with > ) Progression
 * indicator Better detection of the ISO with error control Major code optimisation Integration in
 * main Chrono Cross Hacking tool
 * 
 * ******** End of original disclaimer by Yazoo ********
 * 
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include "cdutils.h"
#include "generic.h"
#include "Input.h"
#include "Output.h"
#include "Main.h"
#include "cdabstract.h"

CODE_BEGINS
public:
Appli() : tourne(0), nb_seqs(0), f_def(0), f_iso(0), cdutil(0) {}
virtual ~Appli() {
    delete cdutil;
    delete f_def;
    delete f_iso;
    delete f_out;
}
private:

unsigned int tourne;

struct t_index_tab {
    unsigned long address;
    long size;
    long type;
    long index;
};

struct t_sequence {
    unsigned int n;
    unsigned int sum;
    String prefix;
    String name;
    int type;
};

String title, iso_filename, prefix;
unsigned long iso_size;
unsigned int nb_records, nb_seqs;
struct t_sequence sequences[100];
Handle * f_def, * f_iso, * f_out;

cdutils * cdutil;

Byte user_data[2352];

int startup()
{
    
    verbosity = 3;
    
    printm(M_BARE, 
"Xenogears File Extractor by Nicolas \"Pixel\" Noble\n"
"Highly based upon the Chrono Cross File Extractor By Yazoo\n\n");

    if (argc != 3) {
	printm(M_BARE, "Usage: %s <definition_file.sqr> <iso_file_name>\nSee readme.txt for details\n",
	       argv[0]);
	exit(-1);
    }

    printm(M_STATUS, "Processing file %s...\n", argv[1]);

    f_def = new Input(argv[1]);

    if (process_def_file()) {
	printm(M_ERROR, "Unable to process the definition file \"%s\"...\n", argv[1]);
	exit(-1);
    }

    iso_filename = argv[2];

    printm(M_STATUS, "Begin processing iso file.\n");
    f_iso = open_iso(iso_filename);

    if (check_iso()) {
	printm(M_ERROR, "Invalid iso file for " + title + "\n");
	printm(M_ERROR, "===> Make sure you are using a Genuine iso file.\n");
    } else {
	printm(M_INFO, "Genuine " + title + " iso detected.\n");
    }
    
    cdutil = new cdutils(f_iso);
    
    printm(M_STATUS, "Entering files read sequence\n");
    read_files();
    exit(0);
}

/*
 * Ugly but working... for now 
 */
int process_def_file()
{
    String t;
    unsigned int n, sum = 0;

    *f_def >> t;
    printm(M_INFO, "Read title: " + t + "\n");
    title = t;

    *f_def >> t;
    iso_size = t.to_int();
    printm(M_INFO, "Read iso size: %lu bytes\n", iso_size);

    *f_def >> t;
    printm(M_INFO, "Read global directory prefix: " + t + "\n");
    prefix = t;

    *f_def >> t;
    nb_records = t.to_int();
    printm(M_INFO, "Read total of records: %u\n", nb_records);

    while (1) {
	*f_def >> t;
	n = t.to_int();
	if (!n) {
	    if (sum == nb_records) {
		printm(M_INFO, "Definition file seems coherent\n");
		return 0;
	    } else {
		printm(M_ERROR, "Definition file incoherent\n");
		return 1;
	    }
	}
	sum += n;
	sequences[nb_seqs].n = n;
	sequences[nb_seqs].sum = sum;
	*f_def >> t;
	sequences[nb_seqs].prefix = t;
	*f_def >> t;
	n = t.to_int();
	sequences[nb_seqs].type = n;
	*f_def >> t;
	sequences[nb_seqs].name = t;
	printm(M_INFO, "Read definition of sequence %i:\n===> %5i (sum = %5i) chunks of " + t +
	               " (" + sequences[nb_seqs].prefix + ")\n", nb_seqs, n, sum);
	nb_seqs++;
    }
}

long check_iso()
{
    unsigned long length;

    length = f_iso->GetSize();
    if (length < 0) {
	printm(M_INFO, "Can not get file size, assuming reading from CD.\n");
    } else {
	printm(M_INFO, "Filesize of iso file " + iso_filename + " is %ld bytes\n", length);
	if (length != iso_size) {
	    return 1;
	}
    }
    return 0;
}

void read_files()
{
    t_index_tab index_tab[10000];
    unsigned char t[8];
    unsigned long i;
    unsigned long j;
    unsigned int seq = 0;
    unsigned long indexer;
    struct t_index_tab *p = (struct t_index_tab *) t;

    Byte fat[32768];

#define INDEXPOS 24

    cdutil->sector_seek(INDEXPOS);
    for (i = INDEXPOS; i < (INDEXPOS + 16); i++) {
	printm(M_INFO, "Reading fat sector %lu\n", i);
	cdutil->read_sector(&fat[2048 * (i - INDEXPOS)], MODE_2_FORM_1);
    }

    indexer = 0;
    for (j = 0; j < 32768; j = j + 7) {
	t[0] = 0;
	bcopy((char *) &fat[j], (char *) t + 1, 7);
	p->address >>= 8;
	index_tab[indexer] = *p;
	if (p->size > 0 && p->address != 0) {
	    index_tab[indexer].index = j / 7;
	    printm(M_INFO, "Found a quite valid index: number %4lu, address %6lu, size %3li\n",
		   indexer, index_tab[indexer].address, index_tab[indexer].size);
	    indexer++;
	    if (indexer == nb_records)
		break;
	} else {
	    printm(M_WARNING, "Ignored invalid index chunk number %4lu (size = %lu) (%02x %02x %02x %02x %02x %02x %02x)\n",
		 j / 7, -index_tab[indexer].size, t[0], t[1], t[2], t[4], t[5], t[6], t[7]);
	}
    }
    printm(M_STATUS, "Index file generation complete.\n\n");

    for (i = 0; i < nb_records; i++) {
	if (sequences[seq].sum == i)
	    seq++;
	index_tab[i].type = sequences[seq].type;
	if (sequences[seq].type == 0) {
	    printm(M_INFO, "%6lu (%10lu): ignored\n", index_tab[i].address, index_tab[i].size);
	} else {
	    printm(M_INFO, "%6lu (%10lu): ", index_tab[i].address, index_tab[i].size);
	    file_dump(index_tab[i].address, index_tab[i].size, i, seq);
	    if (verbosity >= M_INFO) {
		fprintf(stderr, "\n");
	    }
	}
    }
    fprintf(stderr, "\n");
}

void file_dump(unsigned long debut, unsigned long taille, long num, int seq)
{
    long i;
    long nbsects;
    String nom;
    String extension = ".out";
    String nom_t;
    char type = sequences[seq].type;
    char ptitbidule[] = "-\\|/";
    
    nom_t.set("%04ld", num);

    nom = "./" + prefix + "/";

    MKDIR(nom.to_charp());

    nom += sequences[seq].prefix + "/";
    MKDIR(nom.to_charp());

    nom += nom_t + extension;;

    f_out = new Output(nom);
    nbsects = taille / sec_sizes[type];
    if (taille % sec_sizes[type])
	nbsects++;
    cdutil->sector_seek(debut);
    for (i = 0; i < nbsects; i++) {
	if (verbosity < M_INFO) 
	    fprintf(stderr, "  (%c)\010\010\010\010\010", ptitbidule[((tourne++) >> 8) % 4]);
	cdutil->read_sector(user_data, type);
	if (i != (nbsects - 1)) {
	    f_out->write(user_data, sec_sizes[type]);
	} else {
	    f_out->write(user_data, taille % sec_sizes[type] ? taille % sec_sizes[type] : sec_sizes[type]);
	}
    }
    delete f_out;
    f_out = 0;
    printm(M_BARE, "  (*) Dumped file number %4ld - type \"" + sequences[seq].name + "\"           \r", num);
}

} * Application = new Appli();