/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 * Copyright (C) 2016 Furrtek
 *
 * This file is part of PortaPack.
 *
 * 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, 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; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ui_loadmodule.hpp"

#include "ch.h"

#include "ff.h"
#include "event_m0.hpp"
#include "hackrf_gpio.hpp"
#include "portapack.hpp"
#include "portapack_shared_memory.hpp"
#include "hackrf_hal.hpp"
#include "string_format.hpp"

#include <cstring>
#include <stdio.h>

using namespace hackrf::one;

namespace ui {

void LoadModuleView::focus() {
	button_ok.focus();
}

void LoadModuleView::paint(Painter& painter) {
	(void)painter;
}

void LoadModuleView::on_hide() {
	EventDispatcher::message_map().unregister_handler(Message::ID::ReadyForSwitch);
	EventDispatcher::message_map().unregister_handler(Message::ID::ModuleID);
}

void LoadModuleView::on_show() {
	// Ask for MD5 signature and compare
	ModuleIDMessage message;

	//message_map.unregister_handler(Message::ID::ModuleID);
	
	EventDispatcher::message_map().register_handler(Message::ID::ModuleID,
		[this](Message* const p) {
			uint8_t c;
			const auto message = static_cast<const ModuleIDMessage*>(p);
			if (message->query == false) {	// Shouldn't be needed
				for (c=0;c<16;c++) {
					if (message->md5_signature[c] != _hash[c]) break;
				}
				if (c == 16) {
					text_info.set("Module already loaded :)");
					_mod_loaded = true;
				} else {
					loadmodule();
				}
			}
		}
	);
	
	message.query = true;
	shared_memory.baseband_queue.push(message);
}

int LoadModuleView::load_image() {
	const char magic[6] = {'P', 'P', 'M', ' ', 0x02, 0x00};
	UINT bw;
	uint8_t i;
	uint32_t cnt;
	char md5sum[16];
	FILINFO modinfo;
	FIL modfile;
	DIR rootdir;
	FRESULT res;
	
	// Scan SD card root directory for files with the right MD5 fingerprint at the right location
	if (f_opendir(&rootdir, "/") == FR_OK) {
		for (;;) {
			res = f_readdir(&rootdir, &modinfo);
			if (res != FR_OK || modinfo.fname[0] == 0) break;	// Reached last file, abort
			// Only care about files with .bin extension
			if ((!(modinfo.fattrib & AM_DIR)) && (modinfo.fname[9] == 'B') && (modinfo.fname[10] == 'I') && (modinfo.fname[11] == 'N')) {
				res = f_open(&modfile, modinfo.fname, FA_OPEN_EXISTING | FA_READ);
				if (res != FR_OK) return 0;
				// Magic bytes and version check
				f_read(&modfile, &md5sum, 6, &bw);
				for (i = 0; i < 6; i++)
					if (md5sum[i] != magic[i]) break;
				if (i == 6) {
					f_lseek(&modfile, 26);
					f_read(&modfile, &md5sum, 16, &bw);
					for (i = 0; i < 16; i++)
						if (md5sum[i] != _hash[i]) break;
					// f_read can't read more than 512 bytes at a time ?
					if (i == 16) {
						f_lseek(&modfile, 512);
						for (cnt = 0; cnt < 64; cnt++) {
							if (f_read(&modfile, reinterpret_cast<void*>(portapack::memory::map::m4_code.base() + (cnt * 512)), 512, &bw)) return 0;
						}
						f_close(&modfile);
						f_closedir(&rootdir);
						LPC_RGU->RESET_CTRL[0] = (1 << 13);
						return 1;
					}
				}
				f_close(&modfile);
			}
		}
		f_closedir(&rootdir);
	}
	
	return 0;
}

void LoadModuleView::loadmodule() {
	//message_map.unregister_handler(Message::ID::ReadyForSwitch);
	
	EventDispatcher::message_map().register_handler(Message::ID::ReadyForSwitch,
		[this](Message* const p) {
			(void)p;
			if (load_image()) {
				text_info.set(to_string_hex(*((unsigned int*)0x10080000),8));
				//text_infob.set(to_string_hex(*((unsigned int*)0x10080004),8));
				text_infob.set("Module loaded :)");
				_mod_loaded = true;
			} else {
				text_info.set("Module not found :(");
				_mod_loaded = false;
			}
		}
	);
	
	m4_switch(_hash);
}

LoadModuleView::LoadModuleView(
	NavigationView& nav,
	const char * hash,
	View* new_view
)
{
	add_children({ {
		&text_info,
		&text_infob,
		&button_ok
	} });
	
	_hash = hash;
	
	button_ok.on_select = [this,&nav,new_view](Button&){
		nav.pop();
		if (_mod_loaded == true) nav.push(new_view);
	};
}

} /* namespace ui */