#include "test.h"

static void
run_test(jit_state_t *j, uint8_t *arena_base, size_t arena_size)
{
	jit_begin(j, arena_base, arena_size);

	size_t align = jit_enter_jit_abi(j, 3, 0, 0);

	jit_operand_t args[] =
	{ jit_operand_gpr (JIT_OPERAND_ABI_POINTER, JIT_R0),
	  jit_operand_gpr (JIT_OPERAND_ABI_POINTER, JIT_R1),
	  jit_operand_gpr (JIT_OPERAND_ABI_WORD, JIT_R2),
	  jit_operand_gpr (JIT_OPERAND_ABI_WORD, JIT_V0) };
	jit_load_args(j, 4, args);

	jit_qmulr(j, JIT_V1, JIT_V2, JIT_R2, JIT_V0);
	jit_str(j, JIT_R0, JIT_V1);
	jit_str(j, JIT_R1, JIT_V2);

	jit_leave_jit_abi(j, 3, 0, align);

	jit_ret(j);

	size_t size = 0;
	void* ret = jit_end(j, &size);

	void (*f)(jit_word_t*, jit_word_t*, jit_word_t, jit_word_t) = ret;

#define QMUL(a, b, c, d)                                                  \
	do {                                                              \
		jit_word_t C = 0, D = 0; f(&C, &D, a, b); ASSERT(C == c); \
		ASSERT(D == d);                                           \
	} while (0)

	QMUL(-2, -1, 2, 0);
	QMUL(0, -1, 0, 0);
	QMUL(-1, 0, 0, 0);
	QMUL(1, -1, -1, -1);
#if EJIT_WORDSIZE == 32
	QMUL(0x7ffff, 0x7ffff, 0xfff00001, 0x3f);
	QMUL(0x80000000, -2, 0, 1);
	QMUL(0x80000000, 2, 0, -1);
	QMUL(0x80000001, 3, 0x80000003, -2);
	QMUL(0x80000001, -3, 0x7ffffffd, 1);
#else
	QMUL(0x7ffffffff, 0x7ffffffff, 0xfffffff000000001, 0x3f);
	QMUL(0x8000000000000000, -2, 0, 1);
	QMUL(0x8000000000000000, 2, 0, -1);
	QMUL(0x8000000000000001, 3, 0x8000000000000003, -2);
	QMUL(0x8000000000000001, -3, 0x7ffffffffffffffd, 1);
#endif
}

int
main (int argc, char *argv[])
{
	return main_helper(argc, argv, run_test);
}