cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

ummu.c (4851B)


      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}