#include <ejit/ejit.h>
#include <assert.h>
#include "do_jit.h"

static uint32_t data[] = { 0x12121212, 0x00000000, 0x34343434 };

int main(int argc, char *argv[])
{
	(void)argv;
	bool do_jit = argc > 1;
	struct ejit_operand operands[3] = {
		EJIT_OPERAND_GPR(0, EJIT_TYPE(void *)),
		EJIT_OPERAND_GPR(1, EJIT_TYPE(int)),
		EJIT_OPERAND_GPR(2, EJIT_TYPE(int32_t))
	};
	struct ejit_func *f = ejit_create_func(EJIT_VOID, 3, operands);

	ejit_stxr_32(f, EJIT_GPR(2), EJIT_GPR(0), EJIT_GPR(1));
	ejit_ret(f);

	ejit_select_compile_func(f, 3, 0, EJIT_USE64(void *), do_jit, true);

	assert(data[0] == 0x12121212);
	assert(data[1] == 0x00);
	assert(data[2] == 0x34343434);
	struct ejit_arg args[3] = {
		EJIT_ARG(data, void *),
		EJIT_ARG(sizeof(int32_t), int),
		EJIT_ARG(-1, int32_t)
	};
	ejit_run_func_i(f, 3, args);
	assert(data[0] == 0x12121212);
	assert(data[1] == 0xffffffff);
	assert(data[2] == 0x34343434);

	ejit_destroy_func(f);
}