162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * Copyright 2014 Red Hat Inc.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Permission is hereby granted, free of charge, to any person obtaining a
562306a36Sopenharmony_ci * copy of this software and associated documentation files (the "Software"),
662306a36Sopenharmony_ci * to deal in the Software without restriction, including without limitation
762306a36Sopenharmony_ci * the rights to use, copy, modify, merge, publish, distribute, sublicense,
862306a36Sopenharmony_ci * and/or sell copies of the Software, and to permit persons to whom the
962306a36Sopenharmony_ci * Software is furnished to do so, subject to the following conditions:
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * The above copyright notice and this permission notice shall be included in
1262306a36Sopenharmony_ci * all copies or substantial portions of the Software.
1362306a36Sopenharmony_ci *
1462306a36Sopenharmony_ci * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1562306a36Sopenharmony_ci * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1662306a36Sopenharmony_ci * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1762306a36Sopenharmony_ci * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
1862306a36Sopenharmony_ci * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
1962306a36Sopenharmony_ci * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
2062306a36Sopenharmony_ci * OTHER DEALINGS IN THE SOFTWARE.
2162306a36Sopenharmony_ci *
2262306a36Sopenharmony_ci * Authors: Ben Skeggs <bskeggs@redhat.com>
2362306a36Sopenharmony_ci */
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci#include <nvif/object.h>
2662306a36Sopenharmony_ci#include <nvif/client.h>
2762306a36Sopenharmony_ci#include <nvif/driver.h>
2862306a36Sopenharmony_ci#include <nvif/ioctl.h>
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ciint
3162306a36Sopenharmony_cinvif_object_ioctl(struct nvif_object *object, void *data, u32 size, void **hack)
3262306a36Sopenharmony_ci{
3362306a36Sopenharmony_ci	struct nvif_client *client = object->client;
3462306a36Sopenharmony_ci	union {
3562306a36Sopenharmony_ci		struct nvif_ioctl_v0 v0;
3662306a36Sopenharmony_ci	} *args = data;
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci	if (size >= sizeof(*args) && args->v0.version == 0) {
3962306a36Sopenharmony_ci		if (object != &client->object)
4062306a36Sopenharmony_ci			args->v0.object = nvif_handle(object);
4162306a36Sopenharmony_ci		else
4262306a36Sopenharmony_ci			args->v0.object = 0;
4362306a36Sopenharmony_ci		args->v0.owner = NVIF_IOCTL_V0_OWNER_ANY;
4462306a36Sopenharmony_ci	} else
4562306a36Sopenharmony_ci		return -ENOSYS;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	return client->driver->ioctl(client->object.priv, data, size, hack);
4862306a36Sopenharmony_ci}
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_civoid
5162306a36Sopenharmony_cinvif_object_sclass_put(struct nvif_sclass **psclass)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	kfree(*psclass);
5462306a36Sopenharmony_ci	*psclass = NULL;
5562306a36Sopenharmony_ci}
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ciint
5862306a36Sopenharmony_cinvif_object_sclass_get(struct nvif_object *object, struct nvif_sclass **psclass)
5962306a36Sopenharmony_ci{
6062306a36Sopenharmony_ci	struct {
6162306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
6262306a36Sopenharmony_ci		struct nvif_ioctl_sclass_v0 sclass;
6362306a36Sopenharmony_ci	} *args = NULL;
6462306a36Sopenharmony_ci	int ret, cnt = 0, i;
6562306a36Sopenharmony_ci	u32 size;
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	while (1) {
6862306a36Sopenharmony_ci		size = sizeof(*args) + cnt * sizeof(args->sclass.oclass[0]);
6962306a36Sopenharmony_ci		if (!(args = kmalloc(size, GFP_KERNEL)))
7062306a36Sopenharmony_ci			return -ENOMEM;
7162306a36Sopenharmony_ci		args->ioctl.version = 0;
7262306a36Sopenharmony_ci		args->ioctl.type = NVIF_IOCTL_V0_SCLASS;
7362306a36Sopenharmony_ci		args->sclass.version = 0;
7462306a36Sopenharmony_ci		args->sclass.count = cnt;
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci		ret = nvif_object_ioctl(object, args, size, NULL);
7762306a36Sopenharmony_ci		if (ret == 0 && args->sclass.count <= cnt)
7862306a36Sopenharmony_ci			break;
7962306a36Sopenharmony_ci		cnt = args->sclass.count;
8062306a36Sopenharmony_ci		kfree(args);
8162306a36Sopenharmony_ci		if (ret != 0)
8262306a36Sopenharmony_ci			return ret;
8362306a36Sopenharmony_ci	}
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	*psclass = kcalloc(args->sclass.count, sizeof(**psclass), GFP_KERNEL);
8662306a36Sopenharmony_ci	if (*psclass) {
8762306a36Sopenharmony_ci		for (i = 0; i < args->sclass.count; i++) {
8862306a36Sopenharmony_ci			(*psclass)[i].oclass = args->sclass.oclass[i].oclass;
8962306a36Sopenharmony_ci			(*psclass)[i].minver = args->sclass.oclass[i].minver;
9062306a36Sopenharmony_ci			(*psclass)[i].maxver = args->sclass.oclass[i].maxver;
9162306a36Sopenharmony_ci		}
9262306a36Sopenharmony_ci		ret = args->sclass.count;
9362306a36Sopenharmony_ci	} else {
9462306a36Sopenharmony_ci		ret = -ENOMEM;
9562306a36Sopenharmony_ci	}
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	kfree(args);
9862306a36Sopenharmony_ci	return ret;
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ciu32
10262306a36Sopenharmony_cinvif_object_rd(struct nvif_object *object, int size, u64 addr)
10362306a36Sopenharmony_ci{
10462306a36Sopenharmony_ci	struct {
10562306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
10662306a36Sopenharmony_ci		struct nvif_ioctl_rd_v0 rd;
10762306a36Sopenharmony_ci	} args = {
10862306a36Sopenharmony_ci		.ioctl.type = NVIF_IOCTL_V0_RD,
10962306a36Sopenharmony_ci		.rd.size = size,
11062306a36Sopenharmony_ci		.rd.addr = addr,
11162306a36Sopenharmony_ci	};
11262306a36Sopenharmony_ci	int ret = nvif_object_ioctl(object, &args, sizeof(args), NULL);
11362306a36Sopenharmony_ci	if (ret) {
11462306a36Sopenharmony_ci		/*XXX: warn? */
11562306a36Sopenharmony_ci		return 0;
11662306a36Sopenharmony_ci	}
11762306a36Sopenharmony_ci	return args.rd.data;
11862306a36Sopenharmony_ci}
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_civoid
12162306a36Sopenharmony_cinvif_object_wr(struct nvif_object *object, int size, u64 addr, u32 data)
12262306a36Sopenharmony_ci{
12362306a36Sopenharmony_ci	struct {
12462306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
12562306a36Sopenharmony_ci		struct nvif_ioctl_wr_v0 wr;
12662306a36Sopenharmony_ci	} args = {
12762306a36Sopenharmony_ci		.ioctl.type = NVIF_IOCTL_V0_WR,
12862306a36Sopenharmony_ci		.wr.size = size,
12962306a36Sopenharmony_ci		.wr.addr = addr,
13062306a36Sopenharmony_ci		.wr.data = data,
13162306a36Sopenharmony_ci	};
13262306a36Sopenharmony_ci	int ret = nvif_object_ioctl(object, &args, sizeof(args), NULL);
13362306a36Sopenharmony_ci	if (ret) {
13462306a36Sopenharmony_ci		/*XXX: warn? */
13562306a36Sopenharmony_ci	}
13662306a36Sopenharmony_ci}
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ciint
13962306a36Sopenharmony_cinvif_object_mthd(struct nvif_object *object, u32 mthd, void *data, u32 size)
14062306a36Sopenharmony_ci{
14162306a36Sopenharmony_ci	struct {
14262306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
14362306a36Sopenharmony_ci		struct nvif_ioctl_mthd_v0 mthd;
14462306a36Sopenharmony_ci	} *args;
14562306a36Sopenharmony_ci	u8 stack[128];
14662306a36Sopenharmony_ci	int ret;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	if (sizeof(*args) + size > sizeof(stack)) {
14962306a36Sopenharmony_ci		if (!(args = kmalloc(sizeof(*args) + size, GFP_KERNEL)))
15062306a36Sopenharmony_ci			return -ENOMEM;
15162306a36Sopenharmony_ci	} else {
15262306a36Sopenharmony_ci		args = (void *)stack;
15362306a36Sopenharmony_ci	}
15462306a36Sopenharmony_ci	args->ioctl.version = 0;
15562306a36Sopenharmony_ci	args->ioctl.type = NVIF_IOCTL_V0_MTHD;
15662306a36Sopenharmony_ci	args->mthd.version = 0;
15762306a36Sopenharmony_ci	args->mthd.method = mthd;
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci	memcpy(args->mthd.data, data, size);
16062306a36Sopenharmony_ci	ret = nvif_object_ioctl(object, args, sizeof(*args) + size, NULL);
16162306a36Sopenharmony_ci	memcpy(data, args->mthd.data, size);
16262306a36Sopenharmony_ci	if (args != (void *)stack)
16362306a36Sopenharmony_ci		kfree(args);
16462306a36Sopenharmony_ci	return ret;
16562306a36Sopenharmony_ci}
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_civoid
16862306a36Sopenharmony_cinvif_object_unmap_handle(struct nvif_object *object)
16962306a36Sopenharmony_ci{
17062306a36Sopenharmony_ci	struct {
17162306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
17262306a36Sopenharmony_ci		struct nvif_ioctl_unmap unmap;
17362306a36Sopenharmony_ci	} args = {
17462306a36Sopenharmony_ci		.ioctl.type = NVIF_IOCTL_V0_UNMAP,
17562306a36Sopenharmony_ci	};
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_ci	nvif_object_ioctl(object, &args, sizeof(args), NULL);
17862306a36Sopenharmony_ci}
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ciint
18162306a36Sopenharmony_cinvif_object_map_handle(struct nvif_object *object, void *argv, u32 argc,
18262306a36Sopenharmony_ci		       u64 *handle, u64 *length)
18362306a36Sopenharmony_ci{
18462306a36Sopenharmony_ci	struct {
18562306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
18662306a36Sopenharmony_ci		struct nvif_ioctl_map_v0 map;
18762306a36Sopenharmony_ci	} *args;
18862306a36Sopenharmony_ci	u32 argn = sizeof(*args) + argc;
18962306a36Sopenharmony_ci	int ret, maptype;
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	if (!(args = kzalloc(argn, GFP_KERNEL)))
19262306a36Sopenharmony_ci		return -ENOMEM;
19362306a36Sopenharmony_ci	args->ioctl.type = NVIF_IOCTL_V0_MAP;
19462306a36Sopenharmony_ci	memcpy(args->map.data, argv, argc);
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	ret = nvif_object_ioctl(object, args, argn, NULL);
19762306a36Sopenharmony_ci	*handle = args->map.handle;
19862306a36Sopenharmony_ci	*length = args->map.length;
19962306a36Sopenharmony_ci	maptype = args->map.type;
20062306a36Sopenharmony_ci	kfree(args);
20162306a36Sopenharmony_ci	return ret ? ret : (maptype == NVIF_IOCTL_MAP_V0_IO);
20262306a36Sopenharmony_ci}
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_civoid
20562306a36Sopenharmony_cinvif_object_unmap(struct nvif_object *object)
20662306a36Sopenharmony_ci{
20762306a36Sopenharmony_ci	struct nvif_client *client = object->client;
20862306a36Sopenharmony_ci	if (object->map.ptr) {
20962306a36Sopenharmony_ci		if (object->map.size) {
21062306a36Sopenharmony_ci			client->driver->unmap(client, object->map.ptr,
21162306a36Sopenharmony_ci						      object->map.size);
21262306a36Sopenharmony_ci			object->map.size = 0;
21362306a36Sopenharmony_ci		}
21462306a36Sopenharmony_ci		object->map.ptr = NULL;
21562306a36Sopenharmony_ci		nvif_object_unmap_handle(object);
21662306a36Sopenharmony_ci	}
21762306a36Sopenharmony_ci}
21862306a36Sopenharmony_ci
21962306a36Sopenharmony_ciint
22062306a36Sopenharmony_cinvif_object_map(struct nvif_object *object, void *argv, u32 argc)
22162306a36Sopenharmony_ci{
22262306a36Sopenharmony_ci	struct nvif_client *client = object->client;
22362306a36Sopenharmony_ci	u64 handle, length;
22462306a36Sopenharmony_ci	int ret = nvif_object_map_handle(object, argv, argc, &handle, &length);
22562306a36Sopenharmony_ci	if (ret >= 0) {
22662306a36Sopenharmony_ci		if (ret) {
22762306a36Sopenharmony_ci			object->map.ptr = client->driver->map(client,
22862306a36Sopenharmony_ci							      handle,
22962306a36Sopenharmony_ci							      length);
23062306a36Sopenharmony_ci			if (ret = -ENOMEM, object->map.ptr) {
23162306a36Sopenharmony_ci				object->map.size = length;
23262306a36Sopenharmony_ci				return 0;
23362306a36Sopenharmony_ci			}
23462306a36Sopenharmony_ci		} else {
23562306a36Sopenharmony_ci			object->map.ptr = (void *)(unsigned long)handle;
23662306a36Sopenharmony_ci			return 0;
23762306a36Sopenharmony_ci		}
23862306a36Sopenharmony_ci		nvif_object_unmap_handle(object);
23962306a36Sopenharmony_ci	}
24062306a36Sopenharmony_ci	return ret;
24162306a36Sopenharmony_ci}
24262306a36Sopenharmony_ci
24362306a36Sopenharmony_civoid
24462306a36Sopenharmony_cinvif_object_dtor(struct nvif_object *object)
24562306a36Sopenharmony_ci{
24662306a36Sopenharmony_ci	struct {
24762306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
24862306a36Sopenharmony_ci		struct nvif_ioctl_del del;
24962306a36Sopenharmony_ci	} args = {
25062306a36Sopenharmony_ci		.ioctl.type = NVIF_IOCTL_V0_DEL,
25162306a36Sopenharmony_ci	};
25262306a36Sopenharmony_ci
25362306a36Sopenharmony_ci	if (!nvif_object_constructed(object))
25462306a36Sopenharmony_ci		return;
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_ci	nvif_object_unmap(object);
25762306a36Sopenharmony_ci	nvif_object_ioctl(object, &args, sizeof(args), NULL);
25862306a36Sopenharmony_ci	object->client = NULL;
25962306a36Sopenharmony_ci}
26062306a36Sopenharmony_ci
26162306a36Sopenharmony_ciint
26262306a36Sopenharmony_cinvif_object_ctor(struct nvif_object *parent, const char *name, u32 handle,
26362306a36Sopenharmony_ci		 s32 oclass, void *data, u32 size, struct nvif_object *object)
26462306a36Sopenharmony_ci{
26562306a36Sopenharmony_ci	struct {
26662306a36Sopenharmony_ci		struct nvif_ioctl_v0 ioctl;
26762306a36Sopenharmony_ci		struct nvif_ioctl_new_v0 new;
26862306a36Sopenharmony_ci	} *args;
26962306a36Sopenharmony_ci	int ret = 0;
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ci	object->client = NULL;
27262306a36Sopenharmony_ci	object->name = name ? name : "nvifObject";
27362306a36Sopenharmony_ci	object->handle = handle;
27462306a36Sopenharmony_ci	object->oclass = oclass;
27562306a36Sopenharmony_ci	object->map.ptr = NULL;
27662306a36Sopenharmony_ci	object->map.size = 0;
27762306a36Sopenharmony_ci
27862306a36Sopenharmony_ci	if (parent) {
27962306a36Sopenharmony_ci		if (!(args = kmalloc(sizeof(*args) + size, GFP_KERNEL))) {
28062306a36Sopenharmony_ci			nvif_object_dtor(object);
28162306a36Sopenharmony_ci			return -ENOMEM;
28262306a36Sopenharmony_ci		}
28362306a36Sopenharmony_ci
28462306a36Sopenharmony_ci		object->parent = parent->parent;
28562306a36Sopenharmony_ci
28662306a36Sopenharmony_ci		args->ioctl.version = 0;
28762306a36Sopenharmony_ci		args->ioctl.type = NVIF_IOCTL_V0_NEW;
28862306a36Sopenharmony_ci		args->new.version = 0;
28962306a36Sopenharmony_ci		args->new.route = parent->client->route;
29062306a36Sopenharmony_ci		args->new.token = nvif_handle(object);
29162306a36Sopenharmony_ci		args->new.object = nvif_handle(object);
29262306a36Sopenharmony_ci		args->new.handle = handle;
29362306a36Sopenharmony_ci		args->new.oclass = oclass;
29462306a36Sopenharmony_ci
29562306a36Sopenharmony_ci		memcpy(args->new.data, data, size);
29662306a36Sopenharmony_ci		ret = nvif_object_ioctl(parent, args, sizeof(*args) + size,
29762306a36Sopenharmony_ci					&object->priv);
29862306a36Sopenharmony_ci		memcpy(data, args->new.data, size);
29962306a36Sopenharmony_ci		kfree(args);
30062306a36Sopenharmony_ci		if (ret == 0)
30162306a36Sopenharmony_ci			object->client = parent->client;
30262306a36Sopenharmony_ci	}
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	if (ret)
30562306a36Sopenharmony_ci		nvif_object_dtor(object);
30662306a36Sopenharmony_ci	return ret;
30762306a36Sopenharmony_ci}
308