// SPDX-License-Identifier: GPL-2.0-only
/*
 * IRQ routing offset tests.
 *
 * Copyright IBM Corp. 2026
 *
 * Authors:
 *  Janosch Frank <frankja@linux.ibm.com>
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>

#include "test_util.h"
#include "kvm_util.h"
#include "kselftest.h"
#include "ucall_common.h"

extern char guest_code[];
asm("guest_code:\n"
    "diag %r0,%r0,0\n"
    "j .\n");

static void test(void)
{
	struct kvm_irq_routing *routing;
	struct kvm_vcpu *vcpu;
	struct kvm_vm *vm;
	gpa_t mem;
	int ret;

	struct kvm_irq_routing_entry ue = {
		.type = KVM_IRQ_ROUTING_S390_ADAPTER,
		.gsi = 1,
	};

	vm = vm_create_with_one_vcpu(&vcpu, guest_code);
	mem = vm_phy_pages_alloc(vm, 2, 4096 * 42, 0);

	routing = kvm_gsi_routing_create();
	routing->nr = 1;
	routing->entries[0] = ue;
	routing->entries[0].u.adapter.summary_addr = (uintptr_t)mem;
	routing->entries[0].u.adapter.ind_addr = (uintptr_t)mem;

	routing->entries[0].u.adapter.summary_offset = 4096 * 8;
	ret = __vm_ioctl(vm, KVM_SET_GSI_ROUTING, routing);
	ksft_test_result(ret == -1 && errno == EINVAL, "summary offset outside of page\n");

	routing->entries[0].u.adapter.summary_offset -= 4;
	ret = __vm_ioctl(vm, KVM_SET_GSI_ROUTING, routing);
	ksft_test_result(ret == 0, "summary offset inside of page\n");

	routing->entries[0].u.adapter.ind_offset = 4096 * 8;
	ret = __vm_ioctl(vm, KVM_SET_GSI_ROUTING, routing);
	ksft_test_result(ret == -1 && errno == EINVAL, "ind offset outside of page\n");

	routing->entries[0].u.adapter.ind_offset -= 4;
	ret = __vm_ioctl(vm, KVM_SET_GSI_ROUTING, routing);
	ksft_test_result(ret == 0, "ind offset inside of page\n");

	kvm_vm_free(vm);
}

int main(int argc, char *argv[])
{
	TEST_REQUIRE(kvm_has_cap(KVM_CAP_IRQ_ROUTING));

	ksft_print_header();
	ksft_set_plan(4);
	test();

	ksft_finished();	/* Print results and exit() accordingly */
}