1/*
2 * Copyright 2017 Red Hat Inc.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
21 */
22#include "ummu.h"
23#include "umem.h"
24#include "uvmm.h"
25
26#include <core/client.h>
27
28#include <nvif/if0008.h>
29#include <nvif/unpack.h>
30
31static int
32nvkm_ummu_sclass(struct nvkm_object *object, int index,
33		 struct nvkm_oclass *oclass)
34{
35	struct nvkm_mmu *mmu = nvkm_ummu(object)->mmu;
36
37	if (mmu->func->mem.user.oclass) {
38		if (index-- == 0) {
39			oclass->base = mmu->func->mem.user;
40			oclass->ctor = nvkm_umem_new;
41			return 0;
42		}
43	}
44
45	if (mmu->func->vmm.user.oclass) {
46		if (index-- == 0) {
47			oclass->base = mmu->func->vmm.user;
48			oclass->ctor = nvkm_uvmm_new;
49			return 0;
50		}
51	}
52
53	return -EINVAL;
54}
55
56static int
57nvkm_ummu_heap(struct nvkm_ummu *ummu, void *argv, u32 argc)
58{
59	struct nvkm_mmu *mmu = ummu->mmu;
60	union {
61		struct nvif_mmu_heap_v0 v0;
62	} *args = argv;
63	int ret = -ENOSYS;
64	u8 index;
65
66	if (!(ret = nvif_unpack(ret, &argv, &argc, args->v0, 0, 0, false))) {
67		if ((index = args->v0.index) >= mmu->heap_nr)
68			return -EINVAL;
69		args->v0.size = mmu->heap[index].size;
70	} else
71		return ret;
72
73	return 0;
74}
75
76static int
77nvkm_ummu_type(struct nvkm_ummu *ummu, void *argv, u32 argc)
78{
79	struct nvkm_mmu *mmu = ummu->mmu;
80	union {
81		struct nvif_mmu_type_v0 v0;
82	} *args = argv;
83	int ret = -ENOSYS;
84	u8 type, index;
85
86	if (!(ret = nvif_unpack(ret, &argv, &argc, args->v0, 0, 0, false))) {
87		if ((index = args->v0.index) >= mmu->type_nr)
88			return -EINVAL;
89		type = mmu->type[index].type;
90		args->v0.heap = mmu->type[index].heap;
91		args->v0.vram = !!(type & NVKM_MEM_VRAM);
92		args->v0.host = !!(type & NVKM_MEM_HOST);
93		args->v0.comp = !!(type & NVKM_MEM_COMP);
94		args->v0.disp = !!(type & NVKM_MEM_DISP);
95		args->v0.kind = !!(type & NVKM_MEM_KIND);
96		args->v0.mappable = !!(type & NVKM_MEM_MAPPABLE);
97		args->v0.coherent = !!(type & NVKM_MEM_COHERENT);
98		args->v0.uncached = !!(type & NVKM_MEM_UNCACHED);
99	} else
100		return ret;
101
102	return 0;
103}
104
105static int
106nvkm_ummu_kind(struct nvkm_ummu *ummu, void *argv, u32 argc)
107{
108	struct nvkm_mmu *mmu = ummu->mmu;
109	union {
110		struct nvif_mmu_kind_v0 v0;
111	} *args = argv;
112	const u8 *kind = NULL;
113	int ret = -ENOSYS, count = 0;
114	u8 kind_inv = 0;
115
116	if (mmu->func->kind)
117		kind = mmu->func->kind(mmu, &count, &kind_inv);
118
119	if (!(ret = nvif_unpack(ret, &argv, &argc, args->v0, 0, 0, true))) {
120		if (argc != args->v0.count * sizeof(*args->v0.data))
121			return -EINVAL;
122		if (args->v0.count > count)
123			return -EINVAL;
124		args->v0.kind_inv = kind_inv;
125		memcpy(args->v0.data, kind, args->v0.count);
126	} else
127		return ret;
128
129	return 0;
130}
131
132static int
133nvkm_ummu_mthd(struct nvkm_object *object, u32 mthd, void *argv, u32 argc)
134{
135	struct nvkm_ummu *ummu = nvkm_ummu(object);
136	switch (mthd) {
137	case NVIF_MMU_V0_HEAP: return nvkm_ummu_heap(ummu, argv, argc);
138	case NVIF_MMU_V0_TYPE: return nvkm_ummu_type(ummu, argv, argc);
139	case NVIF_MMU_V0_KIND: return nvkm_ummu_kind(ummu, argv, argc);
140	default:
141		break;
142	}
143	return -EINVAL;
144}
145
146static const struct nvkm_object_func
147nvkm_ummu = {
148	.mthd = nvkm_ummu_mthd,
149	.sclass = nvkm_ummu_sclass,
150};
151
152int
153nvkm_ummu_new(struct nvkm_device *device, const struct nvkm_oclass *oclass,
154	      void *argv, u32 argc, struct nvkm_object **pobject)
155{
156	union {
157		struct nvif_mmu_v0 v0;
158	} *args = argv;
159	struct nvkm_mmu *mmu = device->mmu;
160	struct nvkm_ummu *ummu;
161	int ret = -ENOSYS, kinds = 0;
162	u8 unused = 0;
163
164	if (mmu->func->kind)
165		mmu->func->kind(mmu, &kinds, &unused);
166
167	if (!(ret = nvif_unpack(ret, &argv, &argc, args->v0, 0, 0, false))) {
168		args->v0.dmabits = mmu->dma_bits;
169		args->v0.heap_nr = mmu->heap_nr;
170		args->v0.type_nr = mmu->type_nr;
171		args->v0.kind_nr = kinds;
172	} else
173		return ret;
174
175	if (!(ummu = kzalloc(sizeof(*ummu), GFP_KERNEL)))
176		return -ENOMEM;
177	nvkm_object_ctor(&nvkm_ummu, oclass, &ummu->object);
178	ummu->mmu = mmu;
179	*pobject = &ummu->object;
180	return 0;
181}
182