1920d2b5eSBen Skeggs /* 2920d2b5eSBen Skeggs * Copyright 2017 Red Hat Inc. 3920d2b5eSBen Skeggs * 4920d2b5eSBen Skeggs * Permission is hereby granted, free of charge, to any person obtaining a 5920d2b5eSBen Skeggs * copy of this software and associated documentation files (the "Software"), 6920d2b5eSBen Skeggs * to deal in the Software without restriction, including without limitation 7920d2b5eSBen Skeggs * the rights to use, copy, modify, merge, publish, distribute, sublicense, 8920d2b5eSBen Skeggs * and/or sell copies of the Software, and to permit persons to whom the 9920d2b5eSBen Skeggs * Software is furnished to do so, subject to the following conditions: 10920d2b5eSBen Skeggs * 11920d2b5eSBen Skeggs * The above copyright notice and this permission notice shall be included in 12920d2b5eSBen Skeggs * all copies or substantial portions of the Software. 13920d2b5eSBen Skeggs * 14920d2b5eSBen Skeggs * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15920d2b5eSBen Skeggs * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16920d2b5eSBen Skeggs * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 17920d2b5eSBen Skeggs * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR 18920d2b5eSBen Skeggs * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 19920d2b5eSBen Skeggs * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 20920d2b5eSBen Skeggs * OTHER DEALINGS IN THE SOFTWARE. 21920d2b5eSBen Skeggs */ 22920d2b5eSBen Skeggs #include <nvif/vmm.h> 23920d2b5eSBen Skeggs #include <nvif/mem.h> 24920d2b5eSBen Skeggs 25920d2b5eSBen Skeggs #include <nvif/if000c.h> 26920d2b5eSBen Skeggs 27920d2b5eSBen Skeggs int 28920d2b5eSBen Skeggs nvif_vmm_unmap(struct nvif_vmm *vmm, u64 addr) 29920d2b5eSBen Skeggs { 30920d2b5eSBen Skeggs return nvif_object_mthd(&vmm->object, NVIF_VMM_V0_UNMAP, 31920d2b5eSBen Skeggs &(struct nvif_vmm_unmap_v0) { .addr = addr }, 32920d2b5eSBen Skeggs sizeof(struct nvif_vmm_unmap_v0)); 33920d2b5eSBen Skeggs } 34920d2b5eSBen Skeggs 35920d2b5eSBen Skeggs int 36920d2b5eSBen Skeggs nvif_vmm_map(struct nvif_vmm *vmm, u64 addr, u64 size, void *argv, u32 argc, 37920d2b5eSBen Skeggs struct nvif_mem *mem, u64 offset) 38920d2b5eSBen Skeggs { 39920d2b5eSBen Skeggs struct nvif_vmm_map_v0 *args; 409dfbd731SArnd Bergmann u8 stack[48]; 41920d2b5eSBen Skeggs int ret; 42920d2b5eSBen Skeggs 43920d2b5eSBen Skeggs if (sizeof(*args) + argc > sizeof(stack)) { 44920d2b5eSBen Skeggs if (!(args = kmalloc(sizeof(*args) + argc, GFP_KERNEL))) 45920d2b5eSBen Skeggs return -ENOMEM; 46920d2b5eSBen Skeggs } else { 47920d2b5eSBen Skeggs args = (void *)stack; 48920d2b5eSBen Skeggs } 49920d2b5eSBen Skeggs 50920d2b5eSBen Skeggs args->version = 0; 51920d2b5eSBen Skeggs args->addr = addr; 52920d2b5eSBen Skeggs args->size = size; 53920d2b5eSBen Skeggs args->memory = nvif_handle(&mem->object); 54920d2b5eSBen Skeggs args->offset = offset; 55920d2b5eSBen Skeggs memcpy(args->data, argv, argc); 56920d2b5eSBen Skeggs 57920d2b5eSBen Skeggs ret = nvif_object_mthd(&vmm->object, NVIF_VMM_V0_MAP, 58920d2b5eSBen Skeggs args, sizeof(*args) + argc); 59920d2b5eSBen Skeggs if (args != (void *)stack) 60920d2b5eSBen Skeggs kfree(args); 61920d2b5eSBen Skeggs return ret; 62920d2b5eSBen Skeggs } 63920d2b5eSBen Skeggs 64920d2b5eSBen Skeggs void 65920d2b5eSBen Skeggs nvif_vmm_put(struct nvif_vmm *vmm, struct nvif_vma *vma) 66920d2b5eSBen Skeggs { 67920d2b5eSBen Skeggs if (vma->size) { 68920d2b5eSBen Skeggs WARN_ON(nvif_object_mthd(&vmm->object, NVIF_VMM_V0_PUT, 69920d2b5eSBen Skeggs &(struct nvif_vmm_put_v0) { 70920d2b5eSBen Skeggs .addr = vma->addr, 71920d2b5eSBen Skeggs }, sizeof(struct nvif_vmm_put_v0))); 72920d2b5eSBen Skeggs vma->size = 0; 73920d2b5eSBen Skeggs } 74920d2b5eSBen Skeggs } 75920d2b5eSBen Skeggs 76920d2b5eSBen Skeggs int 77920d2b5eSBen Skeggs nvif_vmm_get(struct nvif_vmm *vmm, enum nvif_vmm_get type, bool sparse, 78920d2b5eSBen Skeggs u8 page, u8 align, u64 size, struct nvif_vma *vma) 79920d2b5eSBen Skeggs { 80920d2b5eSBen Skeggs struct nvif_vmm_get_v0 args; 81920d2b5eSBen Skeggs int ret; 82920d2b5eSBen Skeggs 83920d2b5eSBen Skeggs args.version = vma->size = 0; 84920d2b5eSBen Skeggs args.sparse = sparse; 85920d2b5eSBen Skeggs args.page = page; 86920d2b5eSBen Skeggs args.align = align; 87920d2b5eSBen Skeggs args.size = size; 88920d2b5eSBen Skeggs 89920d2b5eSBen Skeggs switch (type) { 90920d2b5eSBen Skeggs case ADDR: args.type = NVIF_VMM_GET_V0_ADDR; break; 91920d2b5eSBen Skeggs case PTES: args.type = NVIF_VMM_GET_V0_PTES; break; 92920d2b5eSBen Skeggs case LAZY: args.type = NVIF_VMM_GET_V0_LAZY; break; 93920d2b5eSBen Skeggs default: 94920d2b5eSBen Skeggs WARN_ON(1); 95920d2b5eSBen Skeggs return -EINVAL; 96920d2b5eSBen Skeggs } 97920d2b5eSBen Skeggs 98920d2b5eSBen Skeggs ret = nvif_object_mthd(&vmm->object, NVIF_VMM_V0_GET, 99920d2b5eSBen Skeggs &args, sizeof(args)); 100920d2b5eSBen Skeggs if (ret == 0) { 101920d2b5eSBen Skeggs vma->addr = args.addr; 102920d2b5eSBen Skeggs vma->size = args.size; 103920d2b5eSBen Skeggs } 104920d2b5eSBen Skeggs return ret; 105920d2b5eSBen Skeggs } 106920d2b5eSBen Skeggs 107920d2b5eSBen Skeggs void 108920d2b5eSBen Skeggs nvif_vmm_fini(struct nvif_vmm *vmm) 109920d2b5eSBen Skeggs { 110920d2b5eSBen Skeggs kfree(vmm->page); 111920d2b5eSBen Skeggs nvif_object_fini(&vmm->object); 112920d2b5eSBen Skeggs } 113920d2b5eSBen Skeggs 114920d2b5eSBen Skeggs int 115920d2b5eSBen Skeggs nvif_vmm_init(struct nvif_mmu *mmu, s32 oclass, u64 addr, u64 size, 116920d2b5eSBen Skeggs void *argv, u32 argc, struct nvif_vmm *vmm) 117920d2b5eSBen Skeggs { 118920d2b5eSBen Skeggs struct nvif_vmm_v0 *args; 119920d2b5eSBen Skeggs u32 argn = sizeof(*args) + argc; 120920d2b5eSBen Skeggs int ret = -ENOSYS, i; 121920d2b5eSBen Skeggs 122920d2b5eSBen Skeggs vmm->object.client = NULL; 123920d2b5eSBen Skeggs vmm->page = NULL; 124920d2b5eSBen Skeggs 125920d2b5eSBen Skeggs if (!(args = kmalloc(argn, GFP_KERNEL))) 126920d2b5eSBen Skeggs return -ENOMEM; 127920d2b5eSBen Skeggs args->version = 0; 128920d2b5eSBen Skeggs args->addr = addr; 129920d2b5eSBen Skeggs args->size = size; 130920d2b5eSBen Skeggs memcpy(args->data, argv, argc); 131920d2b5eSBen Skeggs 132920d2b5eSBen Skeggs ret = nvif_object_init(&mmu->object, 0, oclass, args, argn, 133920d2b5eSBen Skeggs &vmm->object); 134920d2b5eSBen Skeggs if (ret) 135920d2b5eSBen Skeggs goto done; 136920d2b5eSBen Skeggs 137920d2b5eSBen Skeggs vmm->start = args->addr; 138920d2b5eSBen Skeggs vmm->limit = args->size; 139920d2b5eSBen Skeggs 140920d2b5eSBen Skeggs vmm->page_nr = args->page_nr; 141*6da2ec56SKees Cook vmm->page = kmalloc_array(vmm->page_nr, sizeof(*vmm->page), 142*6da2ec56SKees Cook GFP_KERNEL); 143920d2b5eSBen Skeggs if (!vmm->page) { 144920d2b5eSBen Skeggs ret = -ENOMEM; 145920d2b5eSBen Skeggs goto done; 146920d2b5eSBen Skeggs } 147920d2b5eSBen Skeggs 148920d2b5eSBen Skeggs for (i = 0; i < vmm->page_nr; i++) { 149920d2b5eSBen Skeggs struct nvif_vmm_page_v0 args = { .index = i }; 150920d2b5eSBen Skeggs 151920d2b5eSBen Skeggs ret = nvif_object_mthd(&vmm->object, NVIF_VMM_V0_PAGE, 152920d2b5eSBen Skeggs &args, sizeof(args)); 153920d2b5eSBen Skeggs if (ret) 154920d2b5eSBen Skeggs break; 155920d2b5eSBen Skeggs 156920d2b5eSBen Skeggs vmm->page[i].shift = args.shift; 157920d2b5eSBen Skeggs vmm->page[i].sparse = args.sparse; 158920d2b5eSBen Skeggs vmm->page[i].vram = args.vram; 159920d2b5eSBen Skeggs vmm->page[i].host = args.host; 160920d2b5eSBen Skeggs vmm->page[i].comp = args.comp; 161920d2b5eSBen Skeggs } 162920d2b5eSBen Skeggs 163920d2b5eSBen Skeggs done: 164920d2b5eSBen Skeggs if (ret) 165920d2b5eSBen Skeggs nvif_vmm_fini(vmm); 166920d2b5eSBen Skeggs kfree(args); 167920d2b5eSBen Skeggs return ret; 168920d2b5eSBen Skeggs } 169