/*
 * Copyright 2018 The Hafnium Authors.
 *
 * Use of this source code is governed by a BSD-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/BSD-3-Clause.
 */

#include "hf/arch/irq.h"
#include "hf/arch/vm/interrupts.h"

#include "hf/ffa.h"
#include "hf/std.h"

#include "vmapi/hf/call.h"

#include "test/hftest.h"
#include "test/vmapi/arch/exception_handler.h"
#include "test/vmapi/ffa.h"

TEST_SERVICE(echo_msg_send2_v1_1)
{
	void *send_buf = SERVICE_SEND_BUFFER();
	void *recv_buf = SERVICE_RECV_BUFFER();

	/* Setup handling of NPI, to handle RX buffer full notification. */
	exception_setup(check_npi, NULL);
	arch_irq_enable();

	for (;;) {
		char payload[255] = {0};
		ffa_id_t echo_sender;

		echo_sender = receive_indirect_message(
				      &payload, sizeof(payload), recv_buf)
				      .sender;

		HFTEST_LOG("Message received: %s", payload);

		/* Echo message back. */
		send_indirect_message_v1_1(hf_vm_get_id(), echo_sender,
					   send_buf, &payload, sizeof(payload),
					   0);

		/* Give back control to PVM. */
		ffa_yield();
	}
}

TEST_SERVICE(echo_msg_send2_v1_2)
{
	void *send_buf = SERVICE_SEND_BUFFER();
	void *recv_buf = SERVICE_RECV_BUFFER();
	struct ffa_partition_rxtx_header header;

	/* Setup handling of NPI, to handle RX buffer full notification. */
	exception_setup(check_npi, NULL);
	arch_irq_enable();

	for (;;) {
		char payload[255] = {0};

		header = receive_indirect_message(payload, sizeof(payload),
						  recv_buf);

		HFTEST_LOG("Message received: %s", payload);

		/* Echo message back. */
		send_indirect_message_with_uuid(
			hf_vm_get_id(), header.sender, send_buf, payload,
			sizeof(payload), header.uuid, 0);

		/* Give back control to PVM. */
		ffa_yield();
	}
}

TEST_SERVICE(echo_msg_send2_service)
{
	void *send_buf = SERVICE_SEND_BUFFER();
	void *recv_buf = SERVICE_RECV_BUFFER();
	struct ffa_uuid target_uuid;
	struct ffa_partition_info target_info;
	uint8_t echo_payload[255] = {0};
	ffa_id_t echo_sender;
	const uint8_t payload[255] = "hello world\n";
	struct ffa_value ret;
	const ffa_id_t own_id = hf_vm_get_id();

	/* Setup handling of NPI, to handle RX buffer notification. */
	exception_setup(check_npi, NULL);
	arch_irq_enable();

	/* Retrieve uuid of target endpoint. */
	receive_indirect_message((void *)&target_uuid, sizeof(target_uuid),
				 recv_buf);

	/* From uuid to respective partition info. */
	ASSERT_EQ(get_ffa_partition_info(target_uuid, &target_info,
					 sizeof(target_info), recv_buf),
		  1);

	/* Send message to target. */
	ret = send_indirect_message(own_id, target_info.vm_id, send_buf,
				    &payload, sizeof(payload), 0);
	ASSERT_EQ(ret.func, FFA_SUCCESS_32);

	ffa_yield();

	echo_sender = receive_indirect_message(&echo_payload,
					       sizeof(echo_payload), recv_buf)
			      .sender;

	HFTEST_LOG("Message received: %s", echo_payload);

	EXPECT_EQ(echo_sender, target_info.vm_id);

	/* Give back control to caller VM. */
	ffa_yield();
}