/*
 * Copyright 2019 The Hafnium Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

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

#include "vmapi/hf/call.h"

#include "hftest.h"

alignas(PAGE_SIZE) static uint8_t pages[2 * PAGE_SIZE];

TEST_SERVICE(data_abort)
{
	/* Not using NULL so static analysis doesn't complain. */
	int *p = (int *)1;
	*p = 12;
}

TEST_SERVICE(straddling_data_abort)
{
	/* Give some memory to the primary VM so that it's unmapped. */
	ASSERT_EQ(hf_share_memory(HF_PRIMARY_VM_ID,
				  (hf_ipaddr_t)(&pages[PAGE_SIZE]), PAGE_SIZE,
				  HF_MEMORY_GIVE),
		  0);
	*(volatile uint64_t *)(&pages[PAGE_SIZE - 6]);
}

TEST_SERVICE(instruction_abort)
{
	/* Not using NULL so static analysis doesn't complain. */
	int (*f)(void) = (int (*)(void))4;
	f();
}

TEST_SERVICE(straddling_instruction_abort)
{
	/*
	 * Get a function pointer which, when branched to, will attempt to
	 * execute a 4-byte instruction straddling two pages.
	 */
	int (*f)(void) = (int (*)(void))(&pages[PAGE_SIZE - 2]);

	/* Give second page to the primary VM so that it's unmapped. */
	ASSERT_EQ(hf_share_memory(HF_PRIMARY_VM_ID,
				  (hf_ipaddr_t)(&pages[PAGE_SIZE]), PAGE_SIZE,
				  HF_MEMORY_GIVE),
		  0);

	/* Branch to instruction whose 2 bytes are now in an unmapped page. */
	f();
}