/*
 * $Id$
 *
 * Copyright (C) 2002, 2003 ETC s.r.o.
 *
 * 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., 59 Temple Place - Suite 330, Boston, MA
 * 02111-1307, USA.
 *
 * Written by Marcel Telka <marcel@telka.sk>, 2002, 2003.
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "gettext.h"
#define	_(s)		gettext(s)
#define	N_(s)		gettext_noop(s)
#define	P_(s,p,n)	ngettext(s,p,n)

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <readline/readline.h>
#include <readline/history.h>

#include "part.h"
#include "tap.h"
#include "parport.h"

#include "bus.h"

#include "jtag.h"

#ifndef HAVE_GETLINE
ssize_t getline( char **lineptr, size_t *n, FILE *stream );
#endif

chain_t *chain = NULL;
bus_t *bus = NULL;

static char *
get_token( char *buf )
{
	return strtok( buf, " \f\n\r\t\v" );
}

static int jtag_parse_file( const char *filename );

#define	JTAGDIR		".jtag"
#define	HISTORYFILE	"history"
#define	RCFILE		"rc"

static void
jtag_create_jtagdir( void )
{
	char *home = getenv( "HOME" );
	char *jdir;

	if (!home)
		return;

	jdir = malloc( strlen(home) + strlen(JTAGDIR) + 2 );	/* "/" and trailing \0 */
	if (!jdir)
		return;

	strcpy( jdir, home );
	strcat( jdir, "/" );
	strcat( jdir, JTAGDIR );

	/* Create the directory if it doesn't exists. */
	mkdir( jdir, S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH );

	free( jdir );
}
						 
static void
jtag_load_history( void )
{
	char *home = getenv( "HOME" );
	char *file;

	using_history();

	if (!home)
		return;

	file = malloc( strlen(home) + strlen(JTAGDIR) + strlen(HISTORYFILE) + 3 );	/* 2 x "/" and trailing \0 */
	if (!file)
		return;

	strcpy( file, home );
	strcat( file, "/" );
	strcat( file, JTAGDIR );
	strcat( file, "/" );
	strcat( file, HISTORYFILE );

	read_history( file );

	free( file );
}

static void
jtag_save_history( void )
{
	char *home = getenv( "HOME" );
	char *file;

	if (!home)
		return;

	file = malloc( strlen(home) + strlen(JTAGDIR) + strlen(HISTORYFILE) + 3);	/* 2 x "/" and trailing \0 */
	if (!file)
		return;

	strcpy( file, home );
	strcat( file, "/" );
	strcat( file, JTAGDIR );
	strcat( file, "/");
	strcat( file, HISTORYFILE );

	write_history( file );

	free( file );
}

static int
jtag_parse_line( char *line )
{
		char *t;

		if (!line || !(strlen( line ) > 0))
			return 1;

		t = get_token( line );
		if (!t)
			return 1;

		if (strcmp( t, "quit" ) == 0) {
			if (get_token( NULL )) {
				printf( _("quit: syntax error\n\nType \"help\" for help.\n\n") );
				return 1;
			}
			return 0;
		}

		if (strcmp( t, "help" ) == 0) {
			t = get_token( NULL );
			if (get_token( NULL ))
				printf( _("help: Syntax error!\n") );
			else
				help( t );
			return 1;
		}

		if (strcmp( t, "frequency" ) == 0) {
			uint32_t freq;

			t = get_token( NULL );
			if (!t) {
				printf( _("Missing argument(s)\n") );
				return 1;
			}
			if ((sscanf( t, "0x%x", &freq ) != 1) && (sscanf( t, "%u", &freq ) != 1)) {
				printf( _("syntax error\n") );
				return 1;
			}

			if (get_token( NULL )) {
				printf( _("frequency: syntax error\n") );
				return 1;
			}

			printf( _("Setting TCK frequency to %u Hz\n"), freq );
			frequency = freq;

			return 1;
		}

		if (strcmp( t, "cable" ) == 0) {
			int i;
			const char *params[10];
			int numpar;

			t = get_token( NULL );
			if (!t) {
				printf( _("Missing argument(s)\n") );
				return 1;
			}
			for (i = 0; parport_drivers[i]; i++)
				if (strcmp( t, parport_drivers[i]->type ) == 0)
					break;
			if (!parport_drivers[i]) {
				printf( _("Unknown connection type: %s\n"), t );
				return 1;
			}

			for (numpar = 0; numpar < 10; numpar++) {
				params[numpar] = get_token( NULL );
				if (!params[numpar])
					break;
			}
			if (get_token( NULL )) {
				printf( _("syntax error!\n") );
				return 1;
			}

			chain_disconnect( chain );
			chain->cable = parport_drivers[i]->connect( params, numpar );
			if (!chain->cable) {
				printf( _("Error: Cable connection failed!\n") );
				return 1;
			}

			if (cable_init( chain->cable )) {
				printf( _("Error: Cable initialization failed!\n") );
				chain_disconnect( chain );
				return 1;
			}
			chain_set_trst( chain, 0 );
			chain_set_trst( chain, 1 );
			tap_reset( chain );

			return 1;
		}

		if (strcmp( t, "discovery" ) == 0) {
			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("discovery: missing filename\n") );
				return 1;
			}
			if (get_token( NULL )) {
				printf( _("syntax error!\n") );
				return 1;
			}
			discovery( chain, t );
			return 1;
		}

		if (strcmp( t, "detect" ) == 0) {
			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (get_token( NULL )) {
				printf( _("detect: syntax error\n") );
				return 1;
			}

			if (bus) {
				bus->free( bus );
				bus = NULL;
			}
			parts_free( chain->parts );
			chain->parts = detect_parts( chain, JTAG_DATA_DIR );
			if (!chain->parts->len) {
				parts_free( chain->parts );
				chain->parts = NULL;
				return 1;
			}
			parts_set_instruction( chain->parts, "SAMPLE/PRELOAD" );
			chain_shift_instructions( chain );
			chain_shift_data_registers( chain );
			parts_set_instruction( chain->parts, "BYPASS" );
			chain_shift_instructions( chain );
			if (strcmp( chain->parts->parts[0]->part, "SA1110" ) == 0)
				bus = new_sa1110_bus( chain, 0 );
			if (strcmp( chain->parts->parts[0]->part, "PXA250" ) == 0)
				bus = new_pxa250_bus( chain, 0 );
			if (strcmp( chain->parts->parts[0]->part, "IXP425" ) == 0)
				bus = new_ixp425_bus( chain, 0 );
			return 1;
		}

		if (strcmp( t, "flashmem" ) == 0) {
			FILE *f;
			int msbin = 0;
			uint32_t addr = 0;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("flashmem: Missing argument(s)\n") );
				return 1;
			}
			if (strcmp( t, "msbin" ) != 0) {
				if ((sscanf( t, "0x%x", &addr ) != 1) && (sscanf( t, "%d", &addr ) != 1)) {
					printf( _("error\n") );
					return 1;
				}
				printf( "0x%08X\n", addr );
			} else
				msbin = 1;
			/* filename */
			t = get_token( NULL );
			if (!t) {
				printf( _("flashmem: missing filename\n") );
				return 1;
			}
			if (get_token( NULL )) {
				printf( _("syntax error!\n") );
				return 1;
			}
			f = fopen( t, "r" );
			if (!f) {
				printf( _("Unable to open file `%s'!\n"), t );
				return 1;
			}
			if (msbin) 
				flashmsbin( bus, f );
			else
				flashmem( bus, f, addr );
			fclose( f );
			return 1;
		}

		if (strcmp( t, "readmem" ) == 0) {
			FILE *f;
			uint32_t addr = 0;
			uint32_t len = 0;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("flashmem: Missing argument(s)\n") );
				return 1;
			}
			if ((sscanf( t, "0x%x", &addr ) != 1) && (sscanf( t, "%d", &addr ) != 1)) {
				printf( _("syntax error\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("flashmem: Missing argument(s)\n") );
				return 1;
			}
			if ((sscanf( t, "0x%x", &len ) != 1) && (sscanf( t, "%d", &len ) != 1)) {
				printf( _("syntax error\n") );
				return 1;
			}

			/* filename */
			t = get_token( NULL );
			if (!t) {
				printf( _("flashmem: missing filename\n") );
				return 1;
			}
			if (get_token( NULL )) {
				printf( _("syntax error!\n") );
				return 1;
			}

			f = fopen( t, "w" );
			if (!f) {
				printf( _("Unable to create file `%s'!\n"), t );
				return 1;
			}
			readmem( bus, f, addr, len );

			fclose( f );
			return 1;
		}

		if (strcmp( t, "detectflash" ) == 0) {
			if (get_token( NULL )) {
				printf( _("detectflash: syntax error\n") );
				return 1;
			}

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			detectflash( bus );
			return 1;
		}

		if (strcmp( t, "print" ) == 0) {
			if (get_token( NULL )) {
				printf( _("print: syntax error\n") );
				return 1;
			}

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			parts_print( chain->parts, 1 );
			return 1;
		}

		if (strcmp( t, "instruction" ) == 0) {
			int n;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("instruction: syntax error\n") );
				return 1;
			}

			n = strtol( t, &t, 10 );
			if (t && *t) {
				printf( _("instruction: syntax error\n") );
				return 1;
			}

			if ((n < 0) || (n >= chain->parts->len)) {
				printf( _("instruction: invalid part number\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("instruction: missing instruction name\n") );
				return 1;
			}

			if (get_token( NULL )) {
				printf( _("instruction: syntax error\n") );
				return 1;
			}

			part_set_instruction( chain->parts->parts[n], t );
			if (chain->parts->parts[n]->active_instruction == NULL)
				printf( _("instruction: unknown instruction %s\n"), t );

			return 1;
		}

		if (strcmp( t, "shift" ) == 0) {
			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			t = get_token( NULL );

			if (t && (strcmp( t, "ir" ) == 0)) {
				chain_shift_instructions( chain );
				return 1;
			}

			if (t && (strcmp( t, "dr" ) == 0)) {
				chain_shift_data_registers( chain );
				return 1;
			}

			printf( _("shift: syntax error\n") );
			return 1;
		}

		if (strcmp( t, "dr" ) == 0) {
			int n;
			int dir;
			tap_register *r;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("dr: syntax error\n") );
				return 1;
			}

			n = strtol( t, &t, 10 );
			if (t && *t) {
				printf( _("dr: syntax error\n") );
				return 1;
			}
			
			if ((n < 0) || (n >= chain->parts->len)) {
				printf( _("dr: invalid part number\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t)
				dir = 1;
			else {
				if (strcmp( t, "in" ) == 0)
					dir = 0;
				else if (strcmp( t, "out" ) == 0)
					dir = 1;
				else {
					printf( _("dr: syntax error\n") );
					return 1;
				}

				if (get_token( NULL )) {
					printf( _("dr: syntax error\n") );
					return 1;
				}
			}

			if (dir)
				r = chain->parts->parts[n]->active_instruction->data_register->out;
			else
				r = chain->parts->parts[n]->active_instruction->data_register->in;
			printf( "%s\n", register_get_string( r ) );

			return 1;
		}

		if (strcmp( t, "set" ) == 0) {
			int n;
			int data;
			int dir;
			char *s;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t || strcmp( t, "signal" ) != 0) {
				printf( _("set: syntax error\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("set: syntax error\n") );
				return 1;
			}
			n = strtol( t, &t, 10 );
			if (t && *t) {
				printf( _("set: syntax error\n") );
				return 1;
			}

			if ((n < 0) || (n >= chain->parts->len)) {
				printf( _("set: invalid part number\n") );
				return 1;
			}

			s = get_token( NULL );		/* signal name */
			if (!s) {
				printf( _("set: syntax error\n") );
				return 1;
			}

			t = get_token( NULL );		/* direction */
			if (!t || (strcmp( t, "in" ) != 0 && strcmp( t, "out" ) != 0)) {
				printf( _("set: syntax error\n") );
				return 1;
			}

			dir = (strcmp( t, "in" ) == 0) ? 0 : 1;
			if (dir) {
				t = get_token( NULL );
				if (!t) {
					printf( _("set: syntax error\n") );
					return 1;
				}
				data = strtol( t, &t, 10 );
				if (t && *t) {
					printf( _("set: syntax error\n") );
					return 1;
				}

				if ((data < 0) || (data > 1)) {
					printf( _("set: invalid data value\n") );
					return 1;
				}
			} else
				data = 0;

			if (get_token( NULL )) {
				printf( _("set: syntax error\n") );
				return 1;
			}

			part_set_signal( chain->parts->parts[n], s, dir, data );

			return 1;
		}

		if (strcmp( t, "get" ) == 0) {
			int n;
			int data;

			if (!chain->cable) {
				printf( _("Error: Cable not configured. Use 'cable' command first!\n") );
				return 1;
			}

			if (!chain->parts) {
				printf( _("Run \"detect\" first.\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t || strcmp( t, "signal" ) != 0) {
				printf( _("get: syntax error\n") );
				return 1;
			}

			t = get_token( NULL );
			if (!t) {
				printf( _("get: syntax error\n") );
				return 1;
			}
			n = strtol( t, &t, 10 );
			if (t && *t) {
				printf( _("get: syntax error\n") );
				return 1;
			}

			if ((n < 0) || (n >= chain->parts->len)) {
				printf( _("get: invalid part number\n") );
				return 1;
			}

			t = get_token( NULL );		/* signal name */
			if (!t || get_token( NULL )) {
				printf( _("get: syntax error\n") );
				return 1;
			}

			data = part_get_signal( chain->parts->parts[n], t );
			if (data != -1)
				printf( _("%s = %d\n"), t, data );

			return 1;
		}

		if (strcmp( t, "script" ) == 0) {
			int go;

			t = get_token( NULL );          /* filename */
			if (!t) {
				printf( _("script: missing filename\n") );
				return 1;
			}
			if (get_token( NULL )) {
				printf( _("script: syntax error\n") );
				return 1;
			}

			go = jtag_parse_file( t );
			if (go < 0)
				printf( _("Unable to open file `%s'!\n"), t );
			return go;
		}

		printf( _("%s: unknown command\n"), t );

	return 1;
}

static void
jtag_readline_loop( const char *prompt )
{
	char *line = NULL;

	/* Iterate */
	while (jtag_parse_line( line )) {
		free( line );

		/* Read a line from the terminal */
		line = readline( prompt );

		/* Check if we actually got something */
		if (line && (strlen( line ) > 0))
			add_history( line );
	}
	free( line );
}

static int
jtag_parse_file( const char *filename )
{
	FILE *f;
	int go = 1;
	char *line = NULL;
	int n = 0;

	if (strcmp( filename, "-" ) != 0)
		f = fopen( filename, "r" );
	else		
		f = stdin;
	if (!f)
		return -1;

	while (go && (getline( &line, &n, f ) != -1))
		if ((strlen(line) > 0) && (line[0] != '#'))
			go = jtag_parse_line(line);

	free(line);
	if (strcmp( filename, "-" ) != 0)
		fclose(f);

	return go;
}

static int
jtag_parse_rc( void )
{
	char *home = getenv( "HOME" );
	char *file;
	int go;

	if (!home)
		return 1;

	file = malloc( strlen(home) + strlen(JTAGDIR) + strlen(RCFILE) + 3 );	/* 2 x "/" and trailing \0 */
	if (!file)
		return 1;

	strcpy( file, home );
	strcat( file, "/" );
	strcat( file, JTAGDIR );
	strcat( file, "/" );
	strcat( file, RCFILE );

	go = jtag_parse_file( file );

	free( file );

	return go;
}

int
main( int argc, const char **argv )
{
	int go = 1;
	int i;

#ifdef ENABLE_NLS
	/* l10n support */
	setlocale( LC_MESSAGES, "" );
	bindtextdomain( PACKAGE, LOCALEDIR );
	textdomain( PACKAGE );
#endif /* ENABLE_NLS */

	printf(
			_("%s\n"
			"Copyright (C) 2002, 2003 ETC s.r.o.\n"
			"%s is free software, covered by the GNU General Public License, and you are\n"
			"welcome to change it and/or distribute copies of it under certain conditions.\n"
			"There is absolutely no warranty for %s.\n\n"), PACKAGE_STRING, PACKAGE, PACKAGE
	);

	chain = chain_alloc();
	if (!chain) {
		printf( _("Out of memory\n") );
		return -1;
	}

	printf( _("Warning: %s may damage your hardware! Type \"quit\" for exit!\n\n"), PACKAGE );
	printf( _("Type \"help\" for help.\n\n") );

	for (i = 1; i < argc; i++) {
		go = jtag_parse_file( argv[i] );
		if (go < 0)
			printf( _("Unable to open file `%s'!\n"), argv[i] );
		if (!go)
			break;
	}

	if (go) {
		/* Create ~/.jtag */
		jtag_create_jtagdir();

		/* Parse and execute the RC file */
		go = jtag_parse_rc();

		if (go) {
			/* Load history */
			jtag_load_history();

			/* main loop */
			jtag_readline_loop( "jtag> " );

			/* Save history */
			jtag_save_history();
		}
	}

	if (bus) {
		bus->free( bus );
		bus = NULL;
	}
	chain_free( chain );

	return 0;
}