#include "libfat.h"
#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "libhost.h"

os_u32 loaded_fat_segment[512];
os_u8 mbr_buffer[512];

int main(int argc, char** argv) {
	if (argc < 3) {
		printf("No files provided\n");
		return 1;
	}

	FILE* f = fopen(argv[1], "rb+");

	LFFileHandle h = libhost_create_handle(f);

	LFControlBlock fat;
	if(libfat_init(h, loaded_fat_segment, 512 * sizeof(os_u32), &fat) != OS_TRUE) {
		fprintf(stderr, "libfat_init failed.\n");
		return 1;
	}

	FILE* f2 = fopen(argv[2], "rb");

	LFFileHandle mbrHandle = libhost_create_handle(f2);

	if(libfat_test_mbr(&mbrHandle) != OS_TRUE) {
		printf("Failed to test MBR...\n");
		return 1;
	}

	printf("FAT Ok.\n");

	FatHeader_32 header = fat.header;
	fseek(f2, 0, SEEK_SET);

	os_u32 read = fread(mbr_buffer, 1, 512, f2);
	if (read < 512) {
		fprintf(stderr, "MBR size is incorrect: %d\n", read);
		return 1;
	}

	if (*(os_u16*)(mbr_buffer + 510) != 0xAA55) {
		fprintf(stderr, "MBR boot signature is missing: %#x\n", *(os_u16*)(mbr_buffer + 510));
		return 1;
	}

	printf("MBR Ok.\n");

	memcpy(mbr_buffer + 3, ((char*)&header) + 3, sizeof(FatHeader_32) - 3);

	fseek(f, 0, SEEK_SET);
	fwrite(mbr_buffer, 1, 512, f);

	fclose(f);
	fclose(f2);
	printf("MBR Written on FAT.\n");

	return 0;
}