/****************************************************************************
 * Automated test cases
 *
 * The contents of this file are subject to the Mozilla Public
 * License Version 1.1 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of
 * the License at http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS
 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * rights and limitations under the License.
 *
 * The Original Code is Ndless code.
 *
 * The Initial Developer of the Original Code is Olivier ARMAND
 * <olivier.calc@gmail.com>.
 * Portions created by the Initial Developer are Copyright (C) 2010
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s): 
 ****************************************************************************/

#include <os.h>
//------  define and test new fuuncions -------
static const unsigned atol_addrs[] = {0x101e0ec8, 0x101e29f8}; // NCAS 1.3 and CAS 1.3 addresses
#define atol SYSCALL_CUSTOM(atol_addrs, long, char *s)
static const unsigned ftell_addrs[] = {0x101e09d4, 0x101e2504}; // NCAS 1.3 and CAS 1.3 addresses
#define ftell SYSCALL_CUSTOM(ftell_addrs, long, FILE * )
static const unsigned write_unaligned_longword_addrs[] = {0x1013b878, 0x1013d3b0}; // NCAS 1.3 and CAS 1.3 addresses
#define write_unaligned_longword SYSCALL_CUSTOM(write_unaligned_longword_addrs, void, void *, unsigned long )
static const unsigned write_unaligned_word_addrs[] = {0x1013b860, 0x1013d398}; // NCAS 1.3 and CAS 1.3 addresses
#define write_unaligned_word SYSCALL_CUSTOM(write_unaligned_word_addrs, void, void *, int )
static const unsigned strtoul_addrs[] = {0x102a73a4, 0x102a8ed4}; // NCAS 1.3 and CAS 1.3 addresses
#define strtoul SYSCALL_CUSTOM(strtoul_addrs, unsigned long, const char *, char **, int,int)
// from Ndless1.0 for OS1.3(untested): set_current_path
//static const unsigned set_current_path_addrs[] = {0x101e31e4, 0x0x101e4d14}; // NCAS 1.3 and CAS 1.3 addresses
//#define set_current_path SYSCALL_CUSTOM(set_current_path_addrs, void, const char *) 

// TODO: use snprintf instead of sprintf

unsigned errcount = 0;

static void error(const char *tstname, const char *errmsg) {
	printf("[%s] FAILED: %s\n", tstname, errmsg);
	errcount++;
}

static void assertUIntEquals(const char *tstname, unsigned expected, unsigned actual) {
	char sbuf[100];
	if (expected != actual) {
		sprintf(sbuf, "%s(%u, %u)", __func__, expected, actual);
		error(tstname, sbuf);
	}
}

static void assertZero(const char *tstname, unsigned actual) {
	assertUIntEquals(tstname, 0, actual);
}

static void assertStrEquals(const char *tstname, const char *expected, const char *actual) {
	char sbuf[100];
	if (strcmp(expected, actual)) {
		sprintf(sbuf, "%s(\"%s\", \"%s\")", __func__, expected, actual);
		error(tstname, sbuf);
	}
}

static void assertNotNull(const char *tstname, void *actual) {
	char sbuf[100];
	if (!actual) {
		sprintf(sbuf, "%s(\"%p\")", __func__, actual);
		error(tstname, sbuf);
	}
}

int global_int;
int* nl_relocdata_data[] = {&global_int};

static const unsigned custom_sprintf_addrs[] = {0x101de274}; // only non-CAS 1.3
#define custom_sprintf SYSCALL_CUSTOM(custom_sprintf_addrs, int __attribute__((__format__(__printf__,2,3))), char *s, const char *format, ...)

int main(int argc, char *argv[]) {
	char buf[100];
	char buf2[100];
	int ret;
	char *ptr;
	FILE *file;
	struct stat sstat;
	DSTAT dstat;
        unsigned intmask = TCT_Local_Control_Interrupts(0);   // crashes if not present on NCAS1.3 
	assertUIntEquals("argc", 1, argc); 
	assertStrEquals("argv", "ndless_tests.tns", strrchr(argv[0], '/') + 1); 
	
	ret = sprintf(buf, "%i%i%i", 1, 2, 3);
	assertStrEquals("_syscallsvar >4 params", "123", buf); // tests sprintf. uses _syscallvar_savedlr.
	assertUIntEquals("_syscallsvar return", 3, ret);
	
	if (nl_osvalue((int*)custom_sprintf_addrs, 1)) { // we are on non-CAS 1.3: execute tests which only work on this version.
		unsigned nl_osvalue_data[] = {1, 2, 3};  
		assertUIntEquals("nl_osvalue", 1, nl_osvalue((int*)nl_osvalue_data, 3)); // Also tests syscalls extensions
		custom_sprintf(buf, "%s", "custom");
		assertStrEquals("_syscall_custom", "custom", buf); 
	}
	
	global_int = 1; // tests relocation of global variables 
	nl_relocdata((unsigned*)nl_relocdata_data, 1);
	assertUIntEquals("nl_relocdata", 1, (unsigned)*nl_relocdata_data[0]);
	
	/* syscalls */
	buf[0] = 1; buf[1] = 2; buf[2] = 3; buf[3] = 4; buf[4] = 5;
	assertUIntEquals("read_unaligned_longword", 0x05040302, read_unaligned_longword(buf + 1));
	assertUIntEquals("read_unaligned_word", 0x0302, read_unaligned_word(buf + 1));
	ascii2utf16(buf, "abc", sizeof(buf));
	assertUIntEquals("ascii2utf16", 'a', buf[0]);
	
	assertUIntEquals("isalpha", TRUE, isalpha('a'));
	assertUIntEquals("isascii", TRUE, isascii('+'));
	assertUIntEquals("isdigit", TRUE, isdigit('0'));
	assertUIntEquals("islower", TRUE, islower('a'));
	assertUIntEquals("isprint", TRUE, isprint('a'));
	assertUIntEquals("isspace", TRUE, isspace(' '));
	assertUIntEquals("isupper", TRUE, isupper('A'));
	//assertUIntEquals("isxdigit", TRUE, isxdigit('f'));  
	assertUIntEquals("tolower", 'a', tolower('A'));
	assertUIntEquals("atoi", 1, atoi("1"));
	// assertDblEquals("atof", 1.1, atof("1.1")); // TODO fails
        printf("atof() test: Pi = %9.7f\n",atof("3.1415926"));
	
	ptr = malloc(100); 
	assertNotNull("malloc", ptr);
	free(ptr);  
	ptr = calloc(5, 4); 
	assertNotNull("calloc", ptr);
	assertZero("calloc[0]", *(char*)ptr);
	ptr = realloc(ptr, 100); 
	assertNotNull("realloc", ptr);
	free(ptr);
	
	memset(buf, 1, sizeof(buf)); 
	assertUIntEquals("memset", 1, buf[10]);
	buf[0] = 0; buf[10] = 1;
	memcpy(buf, buf + 10, 1); 
	assertUIntEquals("memcpy", 1, buf[0]);
	buf[0] = 0; buf[10] = 1;
	memmove(buf, buf + 10, 1); 
	assertUIntEquals("memmove", 1, buf[0]);
	buf[0] = 1; buf[10] = 1;
	assertZero("memcmp", memcmp(buf, buf + 10, 1));
	buf[0] = 0; ; buf[10] = 1;
	memrev(buf, 11);  
	assertUIntEquals("memrev", 1, buf[0]);
	assertUIntEquals("strlen", 3, strlen("abc"));
	strcpy(buf, "abc");  
	assertZero("strcpy,strcmp", strcmp(buf, "abc"));
	buf[2] = 0;
	strncpy(buf, "abc", 2);  
	assertZero("strncpy", strcmp(buf, "ab"));
	strcpy(buf, "abc");  
	assertZero("strncmp", strncmp(buf, "a", 1));
	strcpy(buf, "a");
	strncat(buf, "bc", 1);  
	assertZero("strncat", strcmp(buf, "ab"));
	ptr = strchr("abc", 'b');
	assertUIntEquals("strchr", 'b', *ptr);
	ptr = strrchr("abc", 'a');
	assertUIntEquals("strrchr", 'a', *ptr);
	ptr = strpbrk("abc", "dc"); 
	assertUIntEquals("strpbrk", 'c', *ptr);
	
	sprintf(buf, "%s", "abc");
	assertZero("sprintf", strcmp(buf, "abc"));
	
	strncpy(buf, argv[0], sizeof(buf));
	*(strrchr(buf, '/') + 1) = '\0'; // keep the folder
	strncat(buf, "__testfile.tns", sizeof(buf)); // buf = temp file path
	file = fopen(buf, "w+");
	assertNotNull("fopen", file);
	assertUIntEquals("fwrite", 4, fwrite("abc", 1, 4, file));
	assertZero("fseek", fseek(file, 0, SEEK_SET));
	assertUIntEquals("fread-1", 4, fread(buf2, 1, 4, file));
	assertZero("fread-2", strcmp(buf2, "abc"));
	assertUIntEquals("fprintf", 3, fprintf(file, "%s", "abc"));
	fseek(file, 0, SEEK_SET);
	assertUIntEquals("fputc", 'a', fputc('a', file));
	fseek(file, -1, SEEK_CUR);
	assertUIntEquals("fgetc", 'a', fgetc(file));
	fseek(file, 0, SEEK_SET); 
	fprintf(file, "%s", "abc\n"); 
	fseek(file, 0, SEEK_SET);
	fgets(buf2, sizeof(buf2), file);
	assertZero("fgets", strcmp(buf2, "abc\n"));
	assertZero("fclose", fclose(file));
	
	assertZero("stat", stat(buf, &sstat)); 
	assertZero("unlink", unlink(buf));     
	assertZero("mkdir", mkdir("/tmp/__testdir", 0)); 

	assertZero("rename", rename("/tmp/__testdir", "/tmp/__testdir2")); 
	assertZero("chdir", chdir("/tmp"));
	assertZero("rmdir", rmdir("__testdir2")); 
	
	assertZero("NU_Set_Current_Dir", NU_Set_Current_Dir("A:\\tmp")); 
	assertZero("NU_Current_Dir-1", NU_Current_Dir("A:", buf)); 
	assertZero("NU_Current_Dir-2", strcmp(buf, "\\tmp\\"));
	assertZero("NU_Get_First", NU_Get_First(&dstat, "A:\\*.*"));
	assertZero("NU_Get_Next-1", NU_Get_Next(&dstat));
	assertZero("NU_Get_Next-2", strcmp(dstat.filepath, "tmp")); 
	NU_Done(&dstat);
        printf("Misceallaneous & Custom calls-----\n");
        printf("atol test: -12345678==%ld\n",atol("-12345678"));
	file = fopen("/documents/ndless/hello.tns", "rb");
	fseek(file, 0, SEEK_END);
        printf("ftell test: hello.tns file size =%ld\n",ftell(file));
        fclose(file);
	buf[0] = 1; buf[1] = 2; buf[2] = 3; buf[3] = 4; buf[4] = 5;
        write_unaligned_longword(buf + 1,0x07020304);
	assertUIntEquals("write_unaligned_longword", 0x07020304, read_unaligned_longword(buf + 1));
        write_unaligned_word(buf + 1,(int)0x00000708);
	assertUIntEquals("write_unaligned_word", 0x07020708, read_unaligned_longword(buf + 1));
        printf("strtoul test: 0xfedcba98==%x\n",strtoul("fedcba98",NULL,16,0)); // always pass  '0' fourth parameter
	
	/* libndls */
	assertUIntEquals("abs,min,max", 4, max(min(abs(-3), 2), 4));
	sleep(100);
	
	if (!errcount)
		puts("Successful!");
	else
		printf("%u test(s) failed.\n", errcount);
        TCT_Local_Control_Interrupts(intmask);
	exit(0); // tests exit()
}