// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
 */

#include <linux/delay.h>
#include "qcomtee.h"

/**
 * DOC: Primordial Object
 *
 * After boot, the kernel provides a static object of type
 * %QCOMTEE_OBJECT_TYPE_CB called the primordial object. This object is used
 * for native kernel services or privileged operations.
 *
 * We support:
 *  - %QCOMTEE_OBJECT_OP_MAP_REGION to map a memory object and return mapping
 *    object and mapping information (see qcomtee_mem_object_map()).
 *  - %QCOMTEE_OBJECT_OP_YIELD to yield by the thread running in QTEE.
 *  - %QCOMTEE_OBJECT_OP_SLEEP to wait for a period of time.
 */

#define QCOMTEE_OBJECT_OP_MAP_REGION 0
#define QCOMTEE_OBJECT_OP_YIELD 1
#define QCOMTEE_OBJECT_OP_SLEEP 2

/* Mapping information format as expected by QTEE. */
struct qcomtee_mapping_info {
	u64 paddr;
	u64 len;
	u32 perms;
} __packed;

static int
qcomtee_primordial_obj_dispatch(struct qcomtee_object_invoke_ctx *oic,
				struct qcomtee_object *primordial_object_unused,
				u32 op, struct qcomtee_arg *args)
{
	struct qcomtee_mapping_info *map_info;
	struct qcomtee_object *mem_object;
	struct qcomtee_object *map_object;
	int err = 0;

	switch (op) {
	case QCOMTEE_OBJECT_OP_YIELD:
		cond_resched();
		/* No output object. */
		oic->data = NULL;

		break;
	case QCOMTEE_OBJECT_OP_SLEEP:
		/* Check message format matched QCOMTEE_OBJECT_OP_SLEEP op. */
		if (qcomtee_args_len(args) != 1 ||
		    args[0].type != QCOMTEE_ARG_TYPE_IB ||
		    args[0].b.size < sizeof(u32))
			return -EINVAL;

		msleep(*(u32 *)(args[0].b.addr));
		/* No output object. */
		oic->data = NULL;

		break;
	case QCOMTEE_OBJECT_OP_MAP_REGION:
		if (qcomtee_args_len(args) != 3 ||
		    args[0].type != QCOMTEE_ARG_TYPE_OB ||
		    args[1].type != QCOMTEE_ARG_TYPE_IO ||
		    args[2].type != QCOMTEE_ARG_TYPE_OO ||
		    args[0].b.size < sizeof(struct qcomtee_mapping_info))
			return -EINVAL;

		map_info = args[0].b.addr;
		mem_object = args[1].o;

		qcomtee_mem_object_map(mem_object, &map_object,
				       &map_info->paddr, &map_info->len,
				       &map_info->perms);

		args[2].o = map_object;
		/* One output object; pass it for cleanup to notify. */
		oic->data = map_object;

		qcomtee_object_put(mem_object);

		break;
	default:
		err = -EINVAL;
	}

	return err;
}

/* Called after submitting the callback response. */
static void qcomtee_primordial_obj_notify(struct qcomtee_object_invoke_ctx *oic,
					  struct qcomtee_object *unused,
					  int err)
{
	struct qcomtee_object *object = oic->data;

	/* If err, QTEE did not obtain mapping object. Drop it. */
	if (object && err)
		qcomtee_object_put(object);
}

static struct qcomtee_object_operations qcomtee_primordial_obj_ops = {
	.dispatch = qcomtee_primordial_obj_dispatch,
	.notify = qcomtee_primordial_obj_notify,
};

struct qcomtee_object qcomtee_primordial_object = {
	.name = "primordial",
	.object_type = QCOMTEE_OBJECT_TYPE_CB,
	.ops = &qcomtee_primordial_obj_ops
};