/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2020 Andrew Turner
 *
 * This work was supported by Innovate UK project 105694, "Digital Security
 * by Design (DSbD) Technology Platform Prototype".
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/bus.h>
#include <sys/kernel.h>
#include <sys/module.h>
#include <sys/sysctl.h>

#include <dev/fdt/simplebus.h>

#include <dev/ofw/ofw_bus.h>
#include <dev/ofw/ofw_bus_subr.h>

#include <arm/broadcom/bcm2835/bcm2835_firmware.h>
#include <arm/broadcom/bcm2835/bcm2835_mbox.h>
#include <arm/broadcom/bcm2835/bcm2835_mbox_prop.h>
#include <arm/broadcom/bcm2835/bcm2835_vcbus.h>

struct bcm2835_firmware_softc {
	struct simplebus_softc	sc;
	phandle_t	sc_mbox;
};

static struct ofw_compat_data compat_data[] = {
	{"raspberrypi,bcm2835-firmware",	1},
	{NULL,					0}
};

static int sysctl_bcm2835_firmware_get_revision(SYSCTL_HANDLER_ARGS);

static int
bcm2835_firmware_probe(device_t dev)
{
	if (!ofw_bus_status_okay(dev))
		return (ENXIO);

	if (ofw_bus_search_compatible(dev, compat_data)->ocd_data == 0)
		return (ENXIO);

	device_set_desc(dev, "BCM2835 Firmware");
	return (BUS_PROBE_DEFAULT);
}

static int
bcm2835_firmware_attach(device_t dev)
{
	struct bcm2835_firmware_softc *sc;
	struct sysctl_ctx_list *ctx;
	struct sysctl_oid *tree_node;
	struct sysctl_oid_list *tree;
	phandle_t node, mbox;
	int rv;

	sc = device_get_softc(dev);

	node = ofw_bus_get_node(dev);
	rv = OF_getencprop(node, "mboxes", &mbox, sizeof(mbox));
	if (rv <= 0) {
		device_printf(dev, "can't read mboxes property\n");
		return (ENXIO);
	}
	sc->sc_mbox = mbox;

	OF_device_register_xref(OF_xref_from_node(node), dev);

	ctx = device_get_sysctl_ctx(dev);
	tree_node = device_get_sysctl_tree(dev);
	tree = SYSCTL_CHILDREN(tree_node);
	SYSCTL_ADD_PROC(ctx, tree, OID_AUTO, "revision",
	    CTLTYPE_UINT | CTLFLAG_RD, sc, sizeof(*sc),
	    sysctl_bcm2835_firmware_get_revision, "IU",
	    "Firmware revision");

	/* The firmwaare doesn't have a ranges property */
	sc->sc.flags |= SB_FLAG_NO_RANGES;
	return (simplebus_attach(dev));
}

int
bcm2835_firmware_property(device_t dev, uint32_t prop, void *data, size_t len)
{
	struct {
		struct bcm2835_mbox_hdr hdr;
		struct bcm2835_mbox_tag_hdr tag_hdr;
		uint32_t data[];
	} *msg_hdr;
	size_t msg_len;
	int err;

	/*
	 * The message is processed in 32-bit chunks so must be a multiple
	 * of 32-bits.
	 */
	if ((len & (sizeof(uint32_t) - 1)) != 0)
		return (EINVAL);

	msg_len = sizeof(*msg_hdr) + len + sizeof(uint32_t);
	msg_hdr = malloc(sizeof(*msg_hdr) + msg_len + sizeof(uint32_t),
	    M_DEVBUF, M_WAITOK | M_ZERO);

	msg_hdr->hdr.buf_size = msg_len;
	msg_hdr->hdr.code = BCM2835_MBOX_CODE_REQ;
	msg_hdr->tag_hdr.tag = prop;
	msg_hdr->tag_hdr.val_buf_size = len;
	memcpy(msg_hdr->data, data, len);
	msg_hdr->data[len / sizeof(uint32_t)] = 0;

	err = bcm2835_mbox_property(msg_hdr, msg_len);
	if (err == 0) {
		memcpy(data, msg_hdr->data, len);
	}

	free(msg_hdr, M_DEVBUF);

	return (err);
}

static int
sysctl_bcm2835_firmware_get_revision(SYSCTL_HANDLER_ARGS)
{
	struct bcm2835_firmware_softc *sc = arg1;
	uint32_t rev;
	int err;

	if (bcm2835_firmware_property(sc->sc.dev,
	    BCM2835_MBOX_TAG_FIRMWARE_REVISION, &rev, sizeof(rev)) != 0)
		return (ENXIO);

	err = sysctl_handle_int(oidp, &rev, sizeof(rev), req);
	if (err || !req->newptr) /* error || read request */
		return (err);

	/* write request */
	return (EINVAL);
}

static device_method_t bcm2835_firmware_methods[] = {
	/* Device interface */
	DEVMETHOD(device_probe,		bcm2835_firmware_probe),
	DEVMETHOD(device_attach,	bcm2835_firmware_attach),

	DEVMETHOD_END
};

DEFINE_CLASS_1(bcm2835_firmware, bcm2835_firmware_driver,
    bcm2835_firmware_methods, sizeof(struct bcm2835_firmware_softc),
    simplebus_driver);

EARLY_DRIVER_MODULE(bcm2835_firmware, simplebus, bcm2835_firmware_driver, 0, 0,
    BUS_PASS_BUS + BUS_PASS_ORDER_LATE);
MODULE_DEPEND(bcm2835_firmware, mbox, 1, 1, 1);