#include <stdio.h>
#include <string.h>
#include <libmcpu.h>
#include "../utils/mcpu-asm.h"

#define CHECK(c,m) do { if(!(c)) { fprintf(stderr,"FAIL: %s\n",m); return 1; } } while(0)

static const char src[] =
".equ REG0, REGBASE+0\n"
".equ REGBASE, 0x03000\n"
".equ STEP, 4*512\n"
".data\n"
"ucs:\n"
"  .ucs2 \"Андрей\"\n"
"utf:\n"
"  .utf8 \"Андрей\"\n"
"raw:\n"
"  .byte 1, 2+3, STEP/512, 0xff\n"
"ints:\n"
"  .integer 16, 0x0410, 0x043d, 0\n"
"ints8:\n"
"  .int 8, 65, 66, 0\n"
"  .align 4*4\n"
"vals:\n"
"  .integer 64, 11, 42\n"
".text\n"
"start:\n"
"  LDI r64@REG0, 40+2\n"
"  LOAD r64@(REG0+8), vals+8\n"
"  HALT\n";

int
main( void )
{
  cpu_asm_image im;
  char err[256];
  int rc, i;
  static const __mpu_uint16_t ucs_expect[] =
  {
    0x0410, 0x043d, 0x0434, 0x0440, 0x0435, 0x0439, 0
  };
  static const unsigned char utf_expect[] =
  {
    0xd0, 0x90, 0xd0, 0xbd, 0xd0, 0xb4, 0xd1, 0x80, 0xd0, 0xb5, 0xd0, 0xb9, 0
  };
  static const unsigned char raw_expect[] =
  {
    1, 5, 4, 0xff
  };
  static const char bad_ucs[] = ".data\n.ucs2 \"\\U0001F600\"\n.text\nHALT\n";
  static const char bad_byte[] = ".data\n.byte 256\n.text\nHALT\n";
  static const char bad_cycle[] = ".equ A, B+1\n.equ B, A+1\n.text\nHALT\n";

  __cpu_init();
  CHECK( cpu_assemble_text( src, &im, err, sizeof( err ) ) == 0, err );
  CHECK( im.data_size == 80, "DATA size including NULL guard, strings, raw bytes, integer lists and alignment" );
  for( i = 0; i < 7; i++ ) CHECK( __cpu_data_get_u16( im.data + 16 + i * 2 ) == ucs_expect[i], ".ucs2 code unit" );
  CHECK( !memcmp( im.data + 30, utf_expect, sizeof( utf_expect ) ), ".utf8 exact UTF-8 bytes plus NUL" );
  CHECK( !memcmp( im.data + 43, raw_expect, sizeof( raw_expect ) ), ".byte expressions" );
  CHECK( __cpu_data_get_u16( im.data + 47 ) == 0x0410, ".integer first list element" );
  CHECK( __cpu_data_get_u16( im.data + 49 ) == 0x043d, ".integer second list element" );
  CHECK( __cpu_data_get_u16( im.data + 51 ) == 0, ".integer terminator" );
  CHECK( im.data[53] == 65 && im.data[54] == 66 && im.data[55] == 0, ".int alias and list" );
  for( i = 56; i < 64; i++ ) CHECK( im.data[i] == 0, ".align expression padding" );
  CHECK( __cpu_data_get_u64( im.data + 64 ) == 11, "first 64-bit integer list element" );
  CHECK( __cpu_data_get_u64( im.data + 72 ) == 42, "second 64-bit integer list element" );

  CHECK( __cpu_program_set( im.code, im.code_size ) == 0, "program set" );
  CHECK( __cpu_data_set( im.data, im.data_size ) == 0, "data set" );
  rc = __cpu_run();
  CHECK( rc == 0, "assembled expression program run" );
  CHECK( __cpu_data_get_u64( __cpu_reg_ptr( 64, 0x03000 ) ) == 42, "symbolic register address via forward .equ" );
  CHECK( __cpu_data_get_u64( __cpu_reg_ptr( 64, 0x03008 ) ) == 42, "DATA label+offset expression" );
  cpu_asm_image_free( &im );

  CHECK( cpu_assemble_text( bad_ucs, &im, err, sizeof( err ) ) != 0, "strict UCS-2 rejects non-BMP character" );
  CHECK( strstr( err, "strict UCS-2" ) != 0, "strict UCS-2 diagnostic" );
  CHECK( cpu_assemble_text( bad_byte, &im, err, sizeof( err ) ) != 0, ".byte rejects >8-bit value" );
  CHECK( cpu_assemble_text( bad_cycle, &im, err, sizeof( err ) ) != 0, "cyclic .equ rejected" );

  __cpu_free_context();
  puts( "libcpu assembler directives/expressions regression: PASS" );
  return( 0 );
}