/*
 * Copyright 2021 Red Hat Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */
#define nvkm_ucgrp(p) container_of((p), struct nvkm_ucgrp, object)
#include "priv.h"
#include "cgrp.h"
#include "runl.h"

#include <subdev/mmu.h>

#include <nvif/if0021.h>

struct nvkm_ucgrp {
	struct nvkm_object object;
	struct nvkm_cgrp *cgrp;
};

static int
nvkm_ucgrp_chan_new(const struct nvkm_oclass *oclass, void *argv, u32 argc,
		    struct nvkm_object **pobject)
{
	struct nvkm_cgrp *cgrp = nvkm_ucgrp(oclass->parent)->cgrp;

	return nvkm_uchan_new(cgrp->runl->fifo, cgrp, oclass, argv, argc, pobject);
}

static int
nvkm_ucgrp_sclass(struct nvkm_object *object, int index, struct nvkm_oclass *oclass)
{
	struct nvkm_cgrp *cgrp = nvkm_ucgrp(object)->cgrp;
	struct nvkm_fifo *fifo = cgrp->runl->fifo;
	const struct nvkm_fifo_func_chan *chan = &fifo->func->chan;
	int c = 0;

	/* *_CHANNEL_GPFIFO_* */
	if (chan->user.oclass) {
		if (c++ == index) {
			oclass->base = chan->user;
			oclass->ctor = nvkm_ucgrp_chan_new;
			return 0;
		}
	}

	return -EINVAL;
}

static void *
nvkm_ucgrp_dtor(struct nvkm_object *object)
{
	struct nvkm_ucgrp *ucgrp = nvkm_ucgrp(object);

	nvkm_cgrp_unref(&ucgrp->cgrp);
	return ucgrp;
}

static const struct nvkm_object_func
nvkm_ucgrp = {
	.dtor = nvkm_ucgrp_dtor,
	.sclass = nvkm_ucgrp_sclass,
};

int
nvkm_ucgrp_new(struct nvkm_fifo *fifo, const struct nvkm_oclass *oclass, void *argv, u32 argc,
	       struct nvkm_object **pobject)
{
	union nvif_cgrp_args *args = argv;
	struct nvkm_runl *runl;
	struct nvkm_vmm *vmm;
	struct nvkm_ucgrp *ucgrp;
	int ret;

	if (argc < sizeof(args->v0) || args->v0.version != 0)
		return -ENOSYS;
	argc -= sizeof(args->v0);

	if (args->v0.namelen != argc)
		return -EINVAL;

	/* Lookup objects referenced in args. */
	runl = nvkm_runl_get(fifo, args->v0.runlist, 0);
	if (!runl)
		return -EINVAL;

	vmm = nvkm_uvmm_search(oclass->client, args->v0.vmm);
	if (IS_ERR(vmm))
		return PTR_ERR(vmm);

	/* Allocate channel group. */
	if (!(ucgrp = kzalloc(sizeof(*ucgrp), GFP_KERNEL))) {
		ret = -ENOMEM;
		goto done;
	}

	nvkm_object_ctor(&nvkm_ucgrp, oclass, &ucgrp->object);
	*pobject = &ucgrp->object;

	ret = nvkm_cgrp_new(runl, args->v0.name, vmm, true, &ucgrp->cgrp);
	if (ret)
		goto done;

	/* Return channel group info to caller. */
	args->v0.cgid = ucgrp->cgrp->id;

done:
	nvkm_vmm_unref(&vmm);
	return ret;
}