// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <ddk/debug.h>
#include <ddk/device.h>
#include <ddk/platform-defs.h>
#include <ddk/protocol/platform/bus.h>
#include <zircon/syscalls/smc.h>

#include "astro.h"

// The Astro Secure OS memory region is defined within the bootloader image. The ZBI provided to the
// kernel must mark this memory space as reserved. The OP-TEE driver will query OP-TEE for the exact
// sub-range of this memory space to be used by the driver.
#define ASTRO_SECURE_OS_BASE 0x05300000
#define ASTRO_SECURE_OS_LENGTH 0x02000000

static const pbus_mmio_t astro_tee_mmios[] = {
    {
        .base = ASTRO_SECURE_OS_BASE,
        .length = ASTRO_SECURE_OS_LENGTH,
    },
};

static const pbus_bti_t astro_tee_btis[] = {
    {
        .iommu_index = 0,
        .bti_id = BTI_TEE,
    },
};

static const pbus_smc_t astro_tee_smcs[] = {
    {
        .service_call_num_base = ARM_SMC_SERVICE_CALL_NUM_TRUSTED_OS_BASE,
        .count = ARM_SMC_SERVICE_CALL_NUM_TRUSTED_OS_LENGTH,
    },
};

static const pbus_dev_t tee_dev = {
    .name = "tee",
    .vid = PDEV_VID_GENERIC,
    .pid = PDEV_PID_GENERIC,
    .did = PDEV_DID_OPTEE,
    .mmio_list = astro_tee_mmios,
    .mmio_count = countof(astro_tee_mmios),
    .bti_list = astro_tee_btis,
    .bti_count = countof(astro_tee_btis),
    .smc_list = astro_tee_smcs,
    .smc_count = countof(astro_tee_smcs),
};

zx_status_t astro_tee_init(aml_bus_t* bus) {
    zx_status_t status = pbus_device_add(&bus->pbus, &tee_dev);

    if (status != ZX_OK) {
        zxlogf(ERROR, "%s: pbus_device_add tee failed: %d\n", __FUNCTION__, status);
        return status;
    }
    return ZX_OK;
}