/*
 * Copyright (C) 2025 - Tillitis AB
 * SPDX-License-Identifier: GPL-2.0-only
 */

#include <stdint.h>
#include <tkey/assert.h>
#include <tkey/debug.h>
#include <tkey/led.h>
#include <tkey/lib.h>

#include "mgmt_app.h"
#include "partition_table.h"
#include "preload_app.h"
#include "storage.h"

#include "../tk1/resetinfo.h"
#include "../tk1/syscall_num.h"

// clang-format off
static volatile uint32_t *system_reset = (volatile uint32_t *)TK1_MMIO_TK1_SYSTEM_RESET;
static volatile uint32_t *udi          = (volatile uint32_t *)TK1_MMIO_TK1_UDI_FIRST;
static volatile uint8_t  *resetinfo    = (volatile uint8_t *) TK1_MMIO_RESETINFO_BASE;
// clang-format on

extern struct partition_table part_table;

int32_t syscall_handler(uint32_t number, uint32_t arg1, uint32_t arg2,
			uint32_t arg3)
{
	switch (number) {
	case TK1_SYSCALL_RESET:
		// TODO: Take length from user
		memcpy((uint8_t *)resetinfo, (uint8_t *)arg1, sizeof(struct reset));
		*system_reset = 1;

		return 0;
	case TK1_SYSCALL_ALLOC_AREA:
		if (storage_allocate_area(&part_table) < 0) {
			debug_puts("couldn't allocate storage area\n");
			return -1;
		}

		return 0;
	case TK1_SYSCALL_DEALLOC_AREA:
		if (storage_deallocate_area(&part_table) < 0) {
			debug_puts("couldn't deallocate storage area\n");
			return -1;
		}

		return 0;
	case TK1_SYSCALL_WRITE_DATA:
		if (storage_write_data(&part_table, arg1, (uint8_t *)arg2,
				       arg3) < 0) {
			debug_puts("couldn't write storage area\n");
			return -1;
		}

		return 0;
	case TK1_SYSCALL_READ_DATA:
		if (storage_read_data(&part_table, arg1, (uint8_t *)arg2,
				       arg3) < 0) {
			debug_puts("couldn't read storage area\n");
			return -1;
		}

		return 0;
	case TK1_SYSCALL_SET_LED:
		led_set(arg1);
		return 0;
	case TK1_SYSCALL_GET_VIDPID:
		// UDI is 2 words: VID/PID & serial. Return just the
		// first word. Serial is kept secret to the device
		// app.
		return udi[0];

	case TK1_SYSCALL_PRELOAD_DELETE:
		return preload_delete(&part_table, 1);

	case TK1_SYSCALL_PRELOAD_STORE:
		// arg1 offset
		// arg2 data
		// arg3 size
		// always using slot 1
		return preload_store(&part_table, arg1, (uint8_t *)arg2, arg3, 1);

	case TK1_SYSCALL_PRELOAD_STORE_FIN:
		// arg1 app_size
		// arg2 app_digest
		// arg3 app_signature
		// always using slot 1
		return preload_store_finalize(&part_table, arg1, (uint8_t *)arg2, (uint8_t *)arg3, 1);

	case TK1_SYSCALL_PRELOAD_GET_DIGSIG:
		return preload_get_digsig(&part_table, (uint8_t *)arg1, (uint8_t *)arg2, 1);

	case TK1_SYSCALL_REG_MGMT:
		return mgmt_app_register(&part_table);

	default:
		assert(1 == 2);
	}

	assert(1 == 2);
	return -1; // This should never run
}