/*
    0xFFFF - Open Free Fiasco Firmware Flasher
    Copyright (C) 2012  Pali Rohár <pali.rohar@gmail.com>

    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 3 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, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <sys/statvfs.h>
#include <sys/types.h>
#include <sys/mman.h>

#include <libgen.h>
#include <unistd.h>
#include <fcntl.h>

#ifdef WITH_CAL
#include <cal.h>
#endif

#include "local.h"
#include "global.h"
#include "device.h"
#include "image.h"

static enum device device = DEVICE_UNKNOWN;
static int failed;

int local_init(void) {

	char buf[1024];
	char * ptr;
	char * ptr2;
	FILE * file;

	if ( failed )
		return -1;

	file = fopen("/proc/cpuinfo", "r");
	if ( ! file ) {
		failed = 1;
		return -1;
	}

	while ( fgets(buf, sizeof(buf), file) ) {

		if ( strncmp(buf, "Hardware", strlen("Hardware")) == 0 ) {

			ptr = buf + strlen("Hardware");

			while ( ptr < buf + sizeof(buf) && *ptr > 0 && *ptr <= 32 )
				++ptr;

			if ( *ptr != ':' )
				continue;

			++ptr;

			while ( ptr < buf + sizeof(buf) && *ptr > 0 && *ptr <= 32 )
				++ptr;

			if ( ! *ptr )
				continue;

			ptr2 = buf + strlen(buf);

			while ( *ptr2 <= 32 ) {
				*ptr2 = 0;
				--ptr2;
			}

			if ( strcmp(ptr, "Nokia 770") == 0 )
				device = DEVICE_SU_18;
			else if ( strcmp(ptr, "Nokia N800") == 0 )
				device = DEVICE_RX_34;
			else if ( strcmp(ptr, "Nokia N810") == 0 )
				device = DEVICE_RX_44;
			else if ( strcmp(ptr, "Nokia N810 WiMAX") == 0 )
				device = DEVICE_RX_48;
			else if ( strcmp(ptr, "Nokia RX-51 board") == 0 )
				device = DEVICE_RX_51;

			if ( device ) {
				printf("Found local device: %s\n", device_to_long_string(device));
				fclose(file);
				return 0;
			}

		}

	}

	failed = 1;
	printf("Not a local device\n");
	fclose(file);
	return -1;

}

enum device local_get_device(void) {

	return device;

}

int local_flash_image(struct image * image) {

	ERROR("Not implemented yet");
	(void)image;
	return -1;

}

static int local_nanddump(const char * file, int mtd, int offset, int length) {

	struct statvfs buf;
	char * command;
	char * path;
	size_t size;
	int ret;

	path = strdup(file);
	if ( ! path )
		return 1;

	ret = statvfs(dirname(path), &buf);

	free(path);

	if ( ret == 0 && buf.f_bsize * buf.f_bfree < (long unsigned int)length ) {
		ERROR("Not enought free space (have: %lu, need: %d)", buf.f_bsize * buf.f_bfree, length);
		return 1;
	}

	size = snprintf(NULL, 0, "nanddump -i -o -b -s %d -l %d -f %s /dev/mtd%dro", offset, length, file, mtd);

	command = malloc(size+1);
	if ( ! command )
		return 1;

	snprintf(command, size+1, "nanddump -i -o -b -s %d -l %d -f %s /dev/mtd%dro", offset, length, file, mtd);

	ret = system(command);

	free(command);

	return ret;

}

struct nanddump_args {
	int valid;
	int mtd;
	int offset;
	int length;
};

static struct nanddump_args nanddump_n900[] = {
	[IMAGE_XLOADER]   = { 1, 0, 0x00000000, 0x00004000 },
	[IMAGE_SECONDARY] = { 1, 0, 0x00004000, 0x0001C000 },
	[IMAGE_KERNEL]    = { 1, 3, 0x00000800, 0x001FF800 },
	[IMAGE_INITFS]    = { 1, 4, 0x00000800, 0x001FF800 },
	[IMAGE_ROOTFS]    = { 1, 5, 0x00000000, 0x0fb40000 },
};

/* FIXME: Is this table correct? */
static struct nanddump_args nanddump[] = {
	[IMAGE_XLOADER]   = { 1, 0, 0x00000200, 0x00003E00 },
	[IMAGE_SECONDARY] = { 1, 0, 0x00004000, 0x0001C000 },
	[IMAGE_KERNEL]    = { 1, 2, 0x00000800, 0x001FF800 },
	[IMAGE_INITFS]    = { 1, 3, 0x00000800, 0x001FF800 },
	[IMAGE_ROOTFS]    = { 1, 4, 0x00000000, 0x0fb80000 },
};

int local_dump_image(enum image_type image, const char * file) {

	int ret = -1;
	int fd = -1;
	unsigned char * addr = NULL;
	off_t nlen, len;

	if ( image == IMAGE_ROOTFS )
		return -1;

	printf("Dumping %s image to file %s...\n", image_type_to_string(image), file);

	if ( device == DEVICE_RX_51 ) {

		if ( image >= sizeof(nanddump_n900)/sizeof(nanddump_n900[0]) || ! nanddump_n900[image].valid ) {
			ERROR("Unsuported image type: %s", image_type_to_string(image));
			goto clean;
		}

		ret = local_nanddump(file, nanddump_n900[image].mtd, nanddump_n900[image].offset, nanddump_n900[image].length);

	} else {

		if ( image >= sizeof(nanddump)/sizeof(nanddump[0]) || ! nanddump[image].valid ) {
			ERROR("Unsuported image type: %s", image_type_to_string(image));
			goto clean;
		}

		ret = local_nanddump(file, nanddump[image].mtd, nanddump[image].offset, nanddump[image].length);

	}

	if ( ret != 0 )
		goto clean;

	fd = open(file, O_RDWR);
	if ( fd < 0 )
		goto clean;

	len = lseek(fd, 0, SEEK_END);
	if ( len == (off_t)-1 || len == 0 )
		goto clean;

	addr = mmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);

	if ( addr == MAP_FAILED )
		addr = NULL;

	if ( ! addr )
		goto clean;

	for ( nlen = len; nlen > 0; --nlen )
		if ( addr[nlen-1] != 0xFF )
			break;

	if ( ( nlen & ( ( 1ULL << 7 ) - 1 ) ) != 0 )
		nlen = ((nlen >> 7) + 1) << 7;

	if ( nlen != len ) {
		printf("Truncating file %s to %d bytes...\n", file, (int)nlen);
		ftruncate(fd, nlen);
	}

clean:
	if ( addr )
		munmap(addr, len);

	if ( fd >= 0 )
		close(fd);

	printf("\n");
	return ret;

}

int local_check_badblocks(const char * device) {

	ERROR("Not implemented yet");
	(void)device;
	return -1;

}

int local_reboot_device(void) {

	if ( system("dsmetool -b") == 0 )
		return 0;
	return system("reboot");

}

static int local_read_cal(const char * str, void * buf, size_t size) {

#ifdef WITH_CAL
	unsigned long int len;
	struct cal * cal;
	void * ptr;
	int ret;

	if ( cal_init(&cal) < 0 )
		return -1;

	ret = cal_read_block(cal, "r&d_mode", &ptr, &len, CAL_FLAG_USER);

	if ( ret < 0 || ! ptr ) {
		cal_finish(cal);
		return -1;
	}

	if ( len > size )
		len = size;

	memcpy(str, ptr, len);
	cal_finish(cal);
	return len;
#else
	(void)str;
	(void)buf;
	(void)size;
	return -1;
#endif

}

static int local_write_cal(const char * str, const void * buf, size_t len) {

	(void)str;
	(void)buf;
	(void)len;
	return -1;

}

int local_get_root_device(void) {

	ERROR("Not implemented yet");
	return -1;

}

int local_set_root_device(int device) {

	ERROR("Not implemented yet");
	(void)device;
	return -1;

}

int local_get_usb_host_mode(void) {

	ERROR("Not implemented yet");
	return -1;

}

int local_set_usb_host_mode(int enable) {

	ERROR("Not implemented yet");
	(void)enable;
	return -1;

}

int local_get_rd_mode(void) {

	char buf[10];

	if ( local_read_cal("r&d_mode", buf, sizeof(buf)) < 0 )
		return -1;

	if ( strncmp(buf, "master", strlen("master")) == 0 )
		return 1;
	else
		return 0;

}

int local_set_rd_mode(int enable) {

	ERROR("Not implemented yet");
	(void)enable;
	return -1;

}

int local_get_rd_flags(char * flags, size_t size) {

	char buf[512];
	char * ptr;

	if ( local_read_cal("r&d_mode", buf, sizeof(buf)) < 0 )
		return -1;

	if ( strncmp(buf, "master", strlen("master")) == 0 )
		ptr = buf + strlen("master");
	else
		ptr = buf;

	if ( *ptr == ',' )
		++ptr;

	if ( ! *ptr ) {
		flags[0] = 0;
		return 0;
	}

	strncpy(flags, ptr, size);
	flags[size-1] = 0;
	return 0;

}

int local_set_rd_flags(const char * flags) {

	char buf[512];

	if ( strlen(flags) > sizeof(buf) + strlen("master,") )
		return -1;

	if ( flags[0] == 0 )
		return local_write_cal("r&d_mode", "master", strlen("master") + 1);

	strcpy(buf, "master,");
	strcpy(buf+strlen("master,"), flags);
	return local_write_cal("r&d_mode", buf, strlen(buf) + 1);

}

int16_t local_get_hwrev(void) {

	char buf[5];
	int16_t hwrev;

	if ( local_read_cal("phone-info", buf, sizeof(buf)) < 0 )
		return 0;

	buf[4] = 0;

	if ( sscanf(buf, "%hd", &hwrev) != 1 )
		return -1;

	return hwrev;

}

int local_set_hwrev(int16_t hwrev) {

	ERROR("Not implemented yet");
	(void)hwrev;
	return -1;

}

int local_get_kernel_ver(char * ver, size_t size) {

	return local_read_cal("kernel-ver", ver, size);

}

int local_set_kernel_ver(const char * ver) {

	return local_write_cal("kernel-ver", ver, strlen(ver));

}

int local_get_nolo_ver(char * ver, size_t size) {

	return local_read_cal("nolo-ver", ver, size);

}

int local_set_nolo_ver(const char * ver) {

	return local_write_cal("nolo-ver", ver, strlen(ver));

}

int local_get_sw_ver(char * ver, size_t size) {

	return local_read_cal("sw-release-ver", ver, size);

}

int local_set_sw_ver(const char * ver) {

	return local_write_cal("sw-release-ver", ver, strlen(ver));

}

int local_get_content_ver(char * ver, size_t size) {

	return local_read_cal("content-ver", ver, size);

}

int local_set_content_ver(const char * ver) {

	return local_write_cal("content-ver", ver, strlen(ver));

}