#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[]=
".init\n"
"_start:\n"
"  LDI r64@0x00000, 40\n"
"  END\n"
".text\n"
"_life:\n"
"  IINC r64@0x00000, r64@0x00000\n"
"  END\n"
"  IINC r64@0x00000, r64@0x00000\n"
"  LDI r32@0x00020, 7\n"
"  EXIT r32@0x00020\n"
".fini\n"
"_fini:\n"
"  LDI r64@0x00008, 99\n"
"  END\n";
static __mpu_uint64_t
get64( unsigned a )
{
  return( __cpu_data_get_u64( __cpu_reg_ptr( 64, a ) ) );
}

int
main( void )
{
  cpu_asm_image im;
  char err[256];
  int rc;
  __cpu_init();
  CHECK( cpu_assemble_text( src, &im, err, sizeof( err ) ) == 0, err );
  CHECK( im.has_init && im.has_life && im.has_fini, "three lifecycle entries present" );
  CHECK( im.init_entry < im.life_entry && im.life_entry < im.fini_entry, "entry order in PROGRAM" );
  CHECK( __cpu_program_set( im.code, im.code_size ) == 0, "program set" );
  CHECK( __cpu_data_set( im.data, im.data_size ) == 0, "data set" );
  CHECK( __cpu_stack_alloc( 256 ) == 0, "stack alloc" );
  __cpu_pc = im.init_entry;
  rc = __cpu_run();
  CHECK( rc == CPU_EXEC_END, ".init END returned" );
  CHECK( __cpu_get_stop_reason() == CPU_STOP_END, "INIT stop reason END" );
  CHECK( get64( 0 ) == 40, "init result" );
  __cpu_pc = im.life_entry;
  rc = __cpu_run();
  CHECK( rc == CPU_EXEC_END, "first LIFE END returned" );
  CHECK( get64( 0 ) == 41, "first life step" );
  rc = __cpu_run();
  CHECK( rc == CPU_EXEC_EXIT, "resume reaches EXIT" );
  CHECK( __cpu_get_stop_reason() == CPU_STOP_EXIT, "EXIT stop reason" );
  CHECK( __cpu_get_exit_status() == 7, "EXIT status" );
  CHECK( get64( 0 ) == 42, "resume PC is after END" );
  __cpu_pc = im.fini_entry;
  rc = __cpu_run();
  CHECK( rc == CPU_EXEC_END, ".fini END returned" );
  CHECK( get64( 8 ) == 99, "fini result" );
  cpu_asm_image_free( &im );
  __cpu_free_context();
  puts( "libcpu lifecycle ABI regression: PASS" );
  return( 0 );
}