[mirror_ubuntu-bionic-kernel.git] / drivers / gpu / drm / nouveau / nvkm / subdev / mmu / base.c

/*
 * Copyright 2010 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.
 *
 * Authors: Ben Skeggs
 */
#include "priv.h"
#include "vmm.h"

#include <subdev/fb.h>

#include <nvif/if500d.h>
#include <nvif/if900d.h>

struct nvkm_mmu_ptp {
	struct nvkm_mmu_pt *pt;
	struct list_head head;
	u8  shift;
	u16 mask;
	u16 free;
};

static void
nvkm_mmu_ptp_put(struct nvkm_mmu *mmu, bool force, struct nvkm_mmu_pt *pt)
{
	const int slot = pt->base >> pt->ptp->shift;
	struct nvkm_mmu_ptp *ptp = pt->ptp;

	/* If there were no free slots in the parent allocation before,
	 * there will be now, so return PTP to the cache.
	 */
	if (!ptp->free)
		list_add(&ptp->head, &mmu->ptp.list);
	ptp->free |= BIT(slot);

	/* If there's no more sub-allocations, destroy PTP. */
	if (ptp->free == ptp->mask) {
		nvkm_mmu_ptc_put(mmu, force, &ptp->pt);
		list_del(&ptp->head);
		kfree(ptp);
	}

	kfree(pt);
}

struct nvkm_mmu_pt *
nvkm_mmu_ptp_get(struct nvkm_mmu *mmu, u32 size, bool zero)
{
	struct nvkm_mmu_pt *pt;
	struct nvkm_mmu_ptp *ptp;
	int slot;

	if (!(pt = kzalloc(sizeof(*pt), GFP_KERNEL)))
		return NULL;

	ptp = list_first_entry_or_null(&mmu->ptp.list, typeof(*ptp), head);
	if (!ptp) {
		/* Need to allocate a new parent to sub-allocate from. */
		if (!(ptp = kmalloc(sizeof(*ptp), GFP_KERNEL))) {
			kfree(pt);
			return NULL;
		}

		ptp->pt = nvkm_mmu_ptc_get(mmu, 0x1000, 0x1000, false);
		if (!ptp->pt) {
			kfree(ptp);
			kfree(pt);
			return NULL;
		}

		ptp->shift = order_base_2(size);
		slot = nvkm_memory_size(ptp->pt->memory) >> ptp->shift;
		ptp->mask = (1 << slot) - 1;
		ptp->free = ptp->mask;
		list_add(&ptp->head, &mmu->ptp.list);
	}
	pt->ptp = ptp;
	pt->sub = true;

	/* Sub-allocate from parent object, removing PTP from cache
	 * if there's no more free slots left.
	 */
	slot = __ffs(ptp->free);
	ptp->free &= ~BIT(slot);
	if (!ptp->free)
		list_del(&ptp->head);

	pt->memory = pt->ptp->pt->memory;
	pt->base = slot << ptp->shift;
	pt->addr = pt->ptp->pt->addr + pt->base;
	return pt;
}

struct nvkm_mmu_ptc {
	struct list_head head;
	struct list_head item;
	u32 size;
	u32 refs;
};

static inline struct nvkm_mmu_ptc *
nvkm_mmu_ptc_find(struct nvkm_mmu *mmu, u32 size)
{
	struct nvkm_mmu_ptc *ptc;

	list_for_each_entry(ptc, &mmu->ptc.list, head) {
		if (ptc->size == size)
			return ptc;
	}

	ptc = kmalloc(sizeof(*ptc), GFP_KERNEL);
	if (ptc) {
		INIT_LIST_HEAD(&ptc->item);
		ptc->size = size;
		ptc->refs = 0;
		list_add(&ptc->head, &mmu->ptc.list);
	}

	return ptc;
}

void
nvkm_mmu_ptc_put(struct nvkm_mmu *mmu, bool force, struct nvkm_mmu_pt **ppt)
{
	struct nvkm_mmu_pt *pt = *ppt;
	if (pt) {
		/* Handle sub-allocated page tables. */
		if (pt->sub) {
			mutex_lock(&mmu->ptp.mutex);
			nvkm_mmu_ptp_put(mmu, force, pt);
			mutex_unlock(&mmu->ptp.mutex);
			return;
		}

		/* Either cache or free the object. */
		mutex_lock(&mmu->ptc.mutex);
		if (pt->ptc->refs < 8 /* Heuristic. */ && !force) {
			list_add_tail(&pt->head, &pt->ptc->item);
			pt->ptc->refs++;
		} else {
			nvkm_memory_unref(&pt->memory);
			kfree(pt);
		}
		mutex_unlock(&mmu->ptc.mutex);
	}
}

struct nvkm_mmu_pt *
nvkm_mmu_ptc_get(struct nvkm_mmu *mmu, u32 size, u32 align, bool zero)
{
	struct nvkm_mmu_ptc *ptc;
	struct nvkm_mmu_pt *pt;
	int ret;

	/* Sub-allocated page table (ie. GP100 LPT). */
	if (align < 0x1000) {
		mutex_lock(&mmu->ptp.mutex);
		pt = nvkm_mmu_ptp_get(mmu, align, zero);
		mutex_unlock(&mmu->ptp.mutex);
		return pt;
	}

	/* Lookup cache for this page table size. */
	mutex_lock(&mmu->ptc.mutex);
	ptc = nvkm_mmu_ptc_find(mmu, size);
	if (!ptc) {
		mutex_unlock(&mmu->ptc.mutex);
		return NULL;
	}

	/* If there's a free PT in the cache, reuse it. */
	pt = list_first_entry_or_null(&ptc->item, typeof(*pt), head);
	if (pt) {
		if (zero)
			nvkm_fo64(pt->memory, 0, 0, size >> 3);
		list_del(&pt->head);
		ptc->refs--;
		mutex_unlock(&mmu->ptc.mutex);
		return pt;
	}
	mutex_unlock(&mmu->ptc.mutex);

	/* No such luck, we need to allocate. */
	if (!(pt = kmalloc(sizeof(*pt), GFP_KERNEL)))
		return NULL;
	pt->ptc = ptc;
	pt->sub = false;

	ret = nvkm_memory_new(mmu->subdev.device, NVKM_MEM_TARGET_INST,
			      size, align, zero, &pt->memory);
	if (ret) {
		kfree(pt);
		return NULL;
	}

	pt->base = 0;
	pt->addr = nvkm_memory_addr(pt->memory);
	return pt;
}

static void
nvkm_vm_map_(const struct nvkm_vmm_page *page, struct nvkm_vma *vma, u64 delta,
	     struct nvkm_mem *mem, nvkm_vmm_pte_func fn,
	     struct nvkm_vmm_map *map)
{
	union {
		struct nv50_vmm_map_v0 nv50;
		struct gf100_vmm_map_v0 gf100;
	} args;
	struct nvkm_vmm *vmm = vma->vm;
	void *argv = NULL;
	u32 argc = 0;
	int ret;

	map->memory = mem->memory;
	map->page = page;

	if (vmm->func->valid) {
		switch (vmm->mmu->subdev.device->card_type) {
		case NV_50:
			args.nv50.version = 0;
			args.nv50.ro = !(vma->access & NV_MEM_ACCESS_WO);
			args.nv50.priv = !!(vma->access & NV_MEM_ACCESS_SYS);
			args.nv50.kind = (mem->memtype & 0x07f);
			args.nv50.comp = (mem->memtype & 0x180) >> 7;
			argv = &args.nv50;
			argc = sizeof(args.nv50);
			break;
		case NV_C0:
		case NV_E0:
		case GM100:
		case GP100: {
			args.gf100.version = 0;
			args.gf100.vol = (nvkm_memory_target(map->memory) != NVKM_MEM_TARGET_VRAM);
			args.gf100.ro = !(vma->access & NV_MEM_ACCESS_WO);
			args.gf100.priv = !!(vma->access & NV_MEM_ACCESS_SYS);
			args.gf100.kind = (mem->memtype & 0x0ff);
			argv = &args.gf100;
			argc = sizeof(args.gf100);
		}
			break;
		default:
			break;
		}

		ret = vmm->func->valid(vmm, argv, argc, map);
		if (WARN_ON(ret))
			return;
	}

	mutex_lock(&vmm->mutex);
	nvkm_vmm_ptes_map(vmm, page, ((u64)vma->node->offset << 12) + delta,
				      (u64)vma->node->length << 12, map, fn);
	mutex_unlock(&vmm->mutex);

	nvkm_memory_tags_put(vma->memory, vmm->mmu->subdev.device, &vma->tags);
	nvkm_memory_unref(&vma->memory);
	vma->memory = nvkm_memory_ref(map->memory);
	vma->tags = map->tags;
}

void
nvkm_mmu_ptc_dump(struct nvkm_mmu *mmu)
{
	struct nvkm_mmu_ptc *ptc;
	list_for_each_entry(ptc, &mmu->ptc.list, head) {
		struct nvkm_mmu_pt *pt, *tt;
		list_for_each_entry_safe(pt, tt, &ptc->item, head) {
			nvkm_memory_unref(&pt->memory);
			list_del(&pt->head);
			kfree(pt);
		}
	}
}

static void
nvkm_mmu_ptc_fini(struct nvkm_mmu *mmu)
{
	struct nvkm_mmu_ptc *ptc, *ptct;

	list_for_each_entry_safe(ptc, ptct, &mmu->ptc.list, head) {
		WARN_ON(!list_empty(&ptc->item));
		list_del(&ptc->head);
		kfree(ptc);
	}
}

static void
nvkm_mmu_ptc_init(struct nvkm_mmu *mmu)
{
	mutex_init(&mmu->ptc.mutex);
	INIT_LIST_HEAD(&mmu->ptc.list);
	mutex_init(&mmu->ptp.mutex);
	INIT_LIST_HEAD(&mmu->ptp.list);
}

void
nvkm_vm_map_at(struct nvkm_vma *vma, u64 delta, struct nvkm_mem *node)
{
	const struct nvkm_vmm_page *page = vma->vm->func->page;
	if (page->desc->func->unmap) {
		struct nvkm_vmm_map map = { .mem = node->mem };
		while (page->shift != vma->node->type)
			page++;
		nvkm_vm_map_(page, vma, delta, node, page->desc->func->mem, &map);
		return;
	}
}

static void
nvkm_vm_map_sg_table(struct nvkm_vma *vma, u64 delta, u64 length,
		     struct nvkm_mem *mem)
{
	const struct nvkm_vmm_page *page = vma->vm->func->page;
	if (page->desc->func->unmap) {
		struct nvkm_vmm_map map = { .sgl = mem->sg->sgl };
		while (page->shift != vma->node->type)
			page++;
		nvkm_vm_map_(page, vma, delta, mem, page->desc->func->sgl, &map);
		return;
	}
}

static void
nvkm_vm_map_sg(struct nvkm_vma *vma, u64 delta, u64 length,
	       struct nvkm_mem *mem)
{
	const struct nvkm_vmm_page *page = vma->vm->func->page;
	if (page->desc->func->unmap) {
		struct nvkm_vmm_map map = { .dma = mem->pages };
		while (page->shift != vma->node->type)
			page++;
		nvkm_vm_map_(page, vma, delta, mem, page->desc->func->dma, &map);
		return;
	}
}

void
nvkm_vm_map(struct nvkm_vma *vma, struct nvkm_mem *node)
{
	if (node->sg)
		nvkm_vm_map_sg_table(vma, 0, node->size << 12, node);
	else
	if (node->pages)
		nvkm_vm_map_sg(vma, 0, node->size << 12, node);
	else
		nvkm_vm_map_at(vma, 0, node);
}

void
nvkm_vm_unmap_at(struct nvkm_vma *vma, u64 delta, u64 length)
{
	struct nvkm_vm *vm = vma->vm;
	if (vm->func->page->desc->func->unmap) {
		const struct nvkm_vmm_page *page = vm->func->page;
		while (page->shift != vma->node->type)
			page++;
		mutex_lock(&vm->mutex);
		nvkm_vmm_ptes_unmap(vm, page, (vma->node->offset << 12) + delta,
					       vma->node->length << 12, false);
		mutex_unlock(&vm->mutex);
		return;
	}
}

void
nvkm_vm_unmap(struct nvkm_vma *vma)
{
	nvkm_vm_unmap_at(vma, 0, (u64)vma->node->length << 12);

	nvkm_memory_tags_put(vma->memory, vma->vm->mmu->subdev.device, &vma->tags);
	nvkm_memory_unref(&vma->memory);
}

int
nvkm_vm_get(struct nvkm_vm *vm, u64 size, u32 page_shift, u32 access,
	    struct nvkm_vma *vma)
{
	u32 align = (1 << page_shift) >> 12;
	u32 msize = size >> 12;
	int ret;

	mutex_lock(&vm->mutex);
	ret = nvkm_mm_head(&vm->mm, 0, page_shift, msize, msize, align,
			   &vma->node);
	if (unlikely(ret != 0)) {
		mutex_unlock(&vm->mutex);
		return ret;
	}

	if (vm->func->page->desc->func->unmap) {
		const struct nvkm_vmm_page *page = vm->func->page;
		while (page->shift != page_shift)
			page++;

		ret = nvkm_vmm_ptes_get(vm, page, vma->node->offset << 12,
						  vma->node->length << 12);
		if (ret) {
			nvkm_mm_free(&vm->mm, &vma->node);
			mutex_unlock(&vm->mutex);
			return ret;
		}
	}
	mutex_unlock(&vm->mutex);

	vma->memory = NULL;
	vma->tags = NULL;
	vma->vm = NULL;
	nvkm_vm_ref(vm, &vma->vm, NULL);
	vma->offset = (u64)vma->node->offset << 12;
	vma->access = access;
	return 0;
}

void
nvkm_vm_put(struct nvkm_vma *vma)
{
	struct nvkm_mmu *mmu;
	struct nvkm_vm *vm;

	if (unlikely(vma->node == NULL))
		return;
	vm = vma->vm;
	mmu = vm->mmu;

	nvkm_memory_tags_put(vma->memory, mmu->subdev.device, &vma->tags);
	nvkm_memory_unref(&vma->memory);

	mutex_lock(&vm->mutex);
	if (vm->func->page->desc->func->unmap) {
		const struct nvkm_vmm_page *page = vm->func->page;
		while (page->shift != vma->node->type)
			page++;

		nvkm_vmm_ptes_put(vm, page, vma->node->offset << 12,
					    vma->node->length << 12);
	}

	nvkm_mm_free(&vm->mm, &vma->node);
	mutex_unlock(&vm->mutex);

	nvkm_vm_ref(NULL, &vma->vm, NULL);
}

int
nvkm_vm_boot(struct nvkm_vm *vm, u64 size)
{
	return nvkm_vmm_boot(vm);
}

static int
nvkm_vm_legacy(struct nvkm_mmu *mmu, u64 offset, u64 length, u64 mm_offset,
	       u32 block, struct nvkm_vm *vm)
{
	u64 mm_length = (offset + length) - mm_offset;
	int ret;

	kref_init(&vm->refcount);

	if (block > length)
		block = length;

	ret = nvkm_mm_init(&vm->mm, 0, mm_offset >> 12, mm_length >> 12,
			   block >> 12);
	if (ret)
		return ret;

	return 0;
}

int
nvkm_vm_new(struct nvkm_device *device, u64 offset, u64 length, u64 mm_offset,
	    struct lock_class_key *key, struct nvkm_vm **pvm)
{
	struct nvkm_mmu *mmu = device->mmu;

	*pvm = NULL;
	if (mmu->func->vmm.ctor) {
		int ret = mmu->func->vmm.ctor(mmu, mm_offset,
					      offset + length - mm_offset,
					      NULL, 0, key, "legacy", pvm);
		if (ret) {
			nvkm_vm_ref(NULL, pvm, NULL);
			return ret;
		}

		ret = nvkm_vm_legacy(mmu, offset, length, mm_offset,
				     (*pvm)->func->page_block ?
				     (*pvm)->func->page_block : 4096, *pvm);
		if (ret)
			nvkm_vm_ref(NULL, pvm, NULL);

		return ret;
	}

	return -EINVAL;
}

static void
nvkm_vm_del(struct kref *kref)
{
	struct nvkm_vm *vm = container_of(kref, typeof(*vm), refcount);

	nvkm_mm_fini(&vm->mm);
	if (vm->func)
		nvkm_vmm_dtor(vm);
	kfree(vm);
}

int
nvkm_vm_ref(struct nvkm_vm *ref, struct nvkm_vm **ptr, struct nvkm_memory *inst)
{
	if (ref) {
		if (ref->func->join && inst) {
			int ret = ref->func->join(ref, inst);
			if (ret)
				return ret;
		}

		kref_get(&ref->refcount);
	}

	if (*ptr) {
		if ((*ptr)->func->part && inst)
			(*ptr)->func->part(*ptr, inst);
		kref_put(&(*ptr)->refcount, nvkm_vm_del);
	}

	*ptr = ref;
	return 0;
}

static int
nvkm_mmu_oneinit(struct nvkm_subdev *subdev)
{
	struct nvkm_mmu *mmu = nvkm_mmu(subdev);

	if (mmu->func->vmm.global) {
		int ret = nvkm_vm_new(subdev->device, 0, mmu->limit, 0,
				      NULL, &mmu->vmm);
		if (ret)
			return ret;
	}

	return 0;
}

static int
nvkm_mmu_init(struct nvkm_subdev *subdev)
{
	struct nvkm_mmu *mmu = nvkm_mmu(subdev);
	if (mmu->func->init)
		mmu->func->init(mmu);
	return 0;
}

static void *
nvkm_mmu_dtor(struct nvkm_subdev *subdev)
{
	struct nvkm_mmu *mmu = nvkm_mmu(subdev);

	nvkm_vm_ref(NULL, &mmu->vmm, NULL);

	nvkm_mmu_ptc_fini(mmu);
	return mmu;
}

static const struct nvkm_subdev_func
nvkm_mmu = {
	.dtor = nvkm_mmu_dtor,
	.oneinit = nvkm_mmu_oneinit,
	.init = nvkm_mmu_init,
};

void
nvkm_mmu_ctor(const struct nvkm_mmu_func *func, struct nvkm_device *device,
	      int index, struct nvkm_mmu *mmu)
{
	nvkm_subdev_ctor(&nvkm_mmu, device, index, &mmu->subdev);
	mmu->func = func;
	mmu->limit = func->limit;
	mmu->dma_bits = func->dma_bits;
	mmu->lpg_shift = func->lpg_shift;
	nvkm_mmu_ptc_init(mmu);
}

int
nvkm_mmu_new_(const struct nvkm_mmu_func *func, struct nvkm_device *device,
	      int index, struct nvkm_mmu **pmmu)
{
	if (!(*pmmu = kzalloc(sizeof(**pmmu), GFP_KERNEL)))
		return -ENOMEM;
	nvkm_mmu_ctor(func, device, index, *pmmu);
	return 0;
}
Commit	Line	Data
a11c3198 BS	1	/*
	2	* Copyright 2010 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	* Authors: Ben Skeggs
	23	*/
c9582455	24	#include "priv.h"
806a7335	25	#include "vmm.h"
a11c3198	26
c9582455	27	#include <subdev/fb.h>
a11c3198	28
fd542a3e	29	#include <nvif/if500d.h>
b77791da	30	#include <nvif/if900d.h>
fd542a3e	31
f1280394 BS	32	struct nvkm_mmu_ptp {
	33	struct nvkm_mmu_pt *pt;
	34	struct list_head head;
	35	u8 shift;
	36	u16 mask;
	37	u16 free;
	38	};
	39
	40	static void
	41	nvkm_mmu_ptp_put(struct nvkm_mmu mmu, bool force, struct nvkm_mmu_pt pt)
	42	{
	43	const int slot = pt->base >> pt->ptp->shift;
	44	struct nvkm_mmu_ptp *ptp = pt->ptp;
	45
	46	/* If there were no free slots in the parent allocation before,
	47	* there will be now, so return PTP to the cache.
	48	*/
	49	if (!ptp->free)
	50	list_add(&ptp->head, &mmu->ptp.list);
	51	ptp->free \|= BIT(slot);
	52
	53	/* If there's no more sub-allocations, destroy PTP. */
	54	if (ptp->free == ptp->mask) {
	55	nvkm_mmu_ptc_put(mmu, force, &ptp->pt);
	56	list_del(&ptp->head);
	57	kfree(ptp);
	58	}
	59
	60	kfree(pt);
	61	}
	62
	63	struct nvkm_mmu_pt *
	64	nvkm_mmu_ptp_get(struct nvkm_mmu *mmu, u32 size, bool zero)
	65	{
	66	struct nvkm_mmu_pt *pt;
	67	struct nvkm_mmu_ptp *ptp;
	68	int slot;
	69
	70	if (!(pt = kzalloc(sizeof(*pt), GFP_KERNEL)))
	71	return NULL;
	72
	73	ptp = list_first_entry_or_null(&mmu->ptp.list, typeof(*ptp), head);
	74	if (!ptp) {
	75	/* Need to allocate a new parent to sub-allocate from. */
	76	if (!(ptp = kmalloc(sizeof(*ptp), GFP_KERNEL))) {
	77	kfree(pt);
	78	return NULL;
	79	}
	80
	81	ptp->pt = nvkm_mmu_ptc_get(mmu, 0x1000, 0x1000, false);
	82	if (!ptp->pt) {
	83	kfree(ptp);
	84	kfree(pt);
	85	return NULL;
	86	}
	87
	88	ptp->shift = order_base_2(size);
	89	slot = nvkm_memory_size(ptp->pt->memory) >> ptp->shift;
	90	ptp->mask = (1 << slot) - 1;
	91	ptp->free = ptp->mask;
	92	list_add(&ptp->head, &mmu->ptp.list);
	93	}
	94	pt->ptp = ptp;
	95	pt->sub = true;
96
97	/* Sub-allocate from parent object, removing PTP from cache
98	* if there's no more free slots left.
99	*/
100	slot = __ffs(ptp->free);
101	ptp->free &= ~BIT(slot);
102	if (!ptp->free)
103	list_del(&ptp->head);
104
105	pt->memory = pt->ptp->pt->memory;
106	pt->base = slot << ptp->shift;
107	pt->addr = pt->ptp->pt->addr + pt->base;
108	return pt;
109	}
110
9a45ddaa BS	111	struct nvkm_mmu_ptc {
	112	struct list_head head;
	113	struct list_head item;
	114	u32 size;
	115	u32 refs;
	116	};
	117
	118	static inline struct nvkm_mmu_ptc *
	119	nvkm_mmu_ptc_find(struct nvkm_mmu *mmu, u32 size)
	120	{
	121	struct nvkm_mmu_ptc *ptc;
	122
	123	list_for_each_entry(ptc, &mmu->ptc.list, head) {
	124	if (ptc->size == size)
	125	return ptc;
	126	}
	127
	128	ptc = kmalloc(sizeof(*ptc), GFP_KERNEL);
	129	if (ptc) {
	130	INIT_LIST_HEAD(&ptc->item);
	131	ptc->size = size;
	132	ptc->refs = 0;
	133	list_add(&ptc->head, &mmu->ptc.list);
	134	}
	135
	136	return ptc;
	137	}
	138
	139	void
	140	nvkm_mmu_ptc_put(struct nvkm_mmu mmu, bool force, struct nvkm_mmu_pt *ppt)
	141	{
	142	struct nvkm_mmu_pt pt = ppt;
	143	if (pt) {
f1280394 BS	144	/* Handle sub-allocated page tables. */
	145	if (pt->sub) {
	146	mutex_lock(&mmu->ptp.mutex);
	147	nvkm_mmu_ptp_put(mmu, force, pt);
	148	mutex_unlock(&mmu->ptp.mutex);
	149	return;
	150	}
	151
9a45ddaa BS	152	/* Either cache or free the object. */
	153	mutex_lock(&mmu->ptc.mutex);
	154	if (pt->ptc->refs < 8 /* Heuristic. */ && !force) {
	155	list_add_tail(&pt->head, &pt->ptc->item);
	156	pt->ptc->refs++;
	157	} else {
	158	nvkm_memory_unref(&pt->memory);
	159	kfree(pt);
	160	}
	161	mutex_unlock(&mmu->ptc.mutex);
	162	}
	163	}
	164
	165	struct nvkm_mmu_pt *
	166	nvkm_mmu_ptc_get(struct nvkm_mmu *mmu, u32 size, u32 align, bool zero)
	167	{
	168	struct nvkm_mmu_ptc *ptc;
	169	struct nvkm_mmu_pt *pt;
	170	int ret;
	171
f1280394 BS	172	/* Sub-allocated page table (ie. GP100 LPT). */
	173	if (align < 0x1000) {
	174	mutex_lock(&mmu->ptp.mutex);
	175	pt = nvkm_mmu_ptp_get(mmu, align, zero);
	176	mutex_unlock(&mmu->ptp.mutex);
	177	return pt;
	178	}
	179
9a45ddaa BS	180	/* Lookup cache for this page table size. */
	181	mutex_lock(&mmu->ptc.mutex);
	182	ptc = nvkm_mmu_ptc_find(mmu, size);
	183	if (!ptc) {
	184	mutex_unlock(&mmu->ptc.mutex);
	185	return NULL;
	186	}
	187
	188	/* If there's a free PT in the cache, reuse it. */
	189	pt = list_first_entry_or_null(&ptc->item, typeof(*pt), head);
	190	if (pt) {
	191	if (zero)
	192	nvkm_fo64(pt->memory, 0, 0, size >> 3);
	193	list_del(&pt->head);
	194	ptc->refs--;
	195	mutex_unlock(&mmu->ptc.mutex);
	196	return pt;
	197	}
	198	mutex_unlock(&mmu->ptc.mutex);
	199
	200	/* No such luck, we need to allocate. */
	201	if (!(pt = kmalloc(sizeof(*pt), GFP_KERNEL)))
	202	return NULL;
	203	pt->ptc = ptc;
f1280394	204	pt->sub = false;
9a45ddaa BS	205
	206	ret = nvkm_memory_new(mmu->subdev.device, NVKM_MEM_TARGET_INST,
	207	size, align, zero, &pt->memory);
	208	if (ret) {
	209	kfree(pt);
	210	return NULL;
	211	}
	212
	213	pt->base = 0;
	214	pt->addr = nvkm_memory_addr(pt->memory);
	215	return pt;
	216	}
	217
eb813999 BS	218	static void
	219	nvkm_vm_map_(const struct nvkm_vmm_page page, struct nvkm_vma vma, u64 delta,
	220	struct nvkm_mem *mem, nvkm_vmm_pte_func fn,
	221	struct nvkm_vmm_map *map)
	222	{
fd542a3e BS	223	union {
fd542a3e BS	224	struct nv50_vmm_map_v0 nv50;
b77791da	225	struct gf100_vmm_map_v0 gf100;
fd542a3e	226	} args;
eb813999 BS	227	struct nvkm_vmm *vmm = vma->vm;
	228	void *argv = NULL;
	229	u32 argc = 0;
	230	int ret;
	231
	232	map->memory = mem->memory;
	233	map->page = page;
	234
	235	if (vmm->func->valid) {
fd542a3e BS	236	switch (vmm->mmu->subdev.device->card_type) {
	237	case NV_50:
	238	args.nv50.version = 0;
	239	args.nv50.ro = !(vma->access & NV_MEM_ACCESS_WO);
	240	args.nv50.priv = !!(vma->access & NV_MEM_ACCESS_SYS);
	241	args.nv50.kind = (mem->memtype & 0x07f);
	242	args.nv50.comp = (mem->memtype & 0x180) >> 7;
	243	argv = &args.nv50;
	244	argc = sizeof(args.nv50);
	245	break;
b77791da BS	246	case NV_C0:
	247	case NV_E0:
	248	case GM100:
	249	case GP100: {
	250	args.gf100.version = 0;
	251	args.gf100.vol = (nvkm_memory_target(map->memory) != NVKM_MEM_TARGET_VRAM);
	252	args.gf100.ro = !(vma->access & NV_MEM_ACCESS_WO);
	253	args.gf100.priv = !!(vma->access & NV_MEM_ACCESS_SYS);
	254	args.gf100.kind = (mem->memtype & 0x0ff);
	255	argv = &args.gf100;
	256	argc = sizeof(args.gf100);
	257	}
	258	break;
fd542a3e BS	259	default:
	260	break;
	261	}
	262
eb813999 BS	263	ret = vmm->func->valid(vmm, argv, argc, map);
	264	if (WARN_ON(ret))
	265	return;
	266	}
	267
	268	mutex_lock(&vmm->mutex);
	269	nvkm_vmm_ptes_map(vmm, page, ((u64)vma->node->offset << 12) + delta,
	270	(u64)vma->node->length << 12, map, fn);
	271	mutex_unlock(&vmm->mutex);
	272
	273	nvkm_memory_tags_put(vma->memory, vmm->mmu->subdev.device, &vma->tags);
	274	nvkm_memory_unref(&vma->memory);
	275	vma->memory = nvkm_memory_ref(map->memory);
	276	vma->tags = map->tags;
	277	}
	278
9a45ddaa BS	279	void
	280	nvkm_mmu_ptc_dump(struct nvkm_mmu *mmu)
	281	{
	282	struct nvkm_mmu_ptc *ptc;
	283	list_for_each_entry(ptc, &mmu->ptc.list, head) {
	284	struct nvkm_mmu_pt pt, tt;
	285	list_for_each_entry_safe(pt, tt, &ptc->item, head) {
	286	nvkm_memory_unref(&pt->memory);
	287	list_del(&pt->head);
	288	kfree(pt);
	289	}
	290	}
	291	}
	292
	293	static void
	294	nvkm_mmu_ptc_fini(struct nvkm_mmu *mmu)
	295	{
	296	struct nvkm_mmu_ptc ptc, ptct;
	297
	298	list_for_each_entry_safe(ptc, ptct, &mmu->ptc.list, head) {
	299	WARN_ON(!list_empty(&ptc->item));
	300	list_del(&ptc->head);
	301	kfree(ptc);
	302	}
	303	}
	304
	305	static void
	306	nvkm_mmu_ptc_init(struct nvkm_mmu *mmu)
	307	{
	308	mutex_init(&mmu->ptc.mutex);
	309	INIT_LIST_HEAD(&mmu->ptc.list);
f1280394 BS	310	mutex_init(&mmu->ptp.mutex);
f1280394 BS	311	INIT_LIST_HEAD(&mmu->ptp.list);
9a45ddaa BS	312	}
9a45ddaa BS	313
a11c3198	314	void
42594600	315	nvkm_vm_map_at(struct nvkm_vma vma, u64 delta, struct nvkm_mem node)
a11c3198	316	{
eb813999	317	const struct nvkm_vmm_page *page = vma->vm->func->page;
eb813999 BS	318	if (page->desc->func->unmap) {
	319	struct nvkm_vmm_map map = { .mem = node->mem };
	320	while (page->shift != vma->node->type)
	321	page++;
	322	nvkm_vm_map_(page, vma, delta, node, page->desc->func->mem, &map);
	323	return;
	324	}
a11c3198 BS	325	}
a11c3198 BS	326
2e2cfbe6	327	static void
42594600 BS	328	nvkm_vm_map_sg_table(struct nvkm_vma *vma, u64 delta, u64 length,
42594600 BS	329	struct nvkm_mem *mem)
22b33e8e	330	{
eb813999	331	const struct nvkm_vmm_page *page = vma->vm->func->page;
eb813999 BS	332	if (page->desc->func->unmap) {
	333	struct nvkm_vmm_map map = { .sgl = mem->sg->sgl };
	334	while (page->shift != vma->node->type)
	335	page++;
	336	nvkm_vm_map_(page, vma, delta, mem, page->desc->func->sgl, &map);
	337	return;
	338	}
22b33e8e DA	339	}
22b33e8e DA	340
2e2cfbe6	341	static void
42594600 BS	342	nvkm_vm_map_sg(struct nvkm_vma *vma, u64 delta, u64 length,
42594600 BS	343	struct nvkm_mem *mem)
a11c3198	344	{
eb813999	345	const struct nvkm_vmm_page *page = vma->vm->func->page;
eb813999 BS	346	if (page->desc->func->unmap) {
	347	struct nvkm_vmm_map map = { .dma = mem->pages };
	348	while (page->shift != vma->node->type)
	349	page++;
	350	nvkm_vm_map_(page, vma, delta, mem, page->desc->func->dma, &map);
	351	return;
	352	}
a11c3198 BS	353	}
a11c3198 BS	354
2e2cfbe6	355	void
42594600	356	nvkm_vm_map(struct nvkm_vma vma, struct nvkm_mem node)
2e2cfbe6 BS	357	{
2e2cfbe6 BS	358	if (node->sg)
42594600	359	nvkm_vm_map_sg_table(vma, 0, node->size << 12, node);
2e2cfbe6 BS	360	else
2e2cfbe6 BS	361	if (node->pages)
42594600	362	nvkm_vm_map_sg(vma, 0, node->size << 12, node);
2e2cfbe6	363	else
42594600	364	nvkm_vm_map_at(vma, 0, node);
2e2cfbe6 BS	365	}
2e2cfbe6 BS	366
a11c3198	367	void
42594600	368	nvkm_vm_unmap_at(struct nvkm_vma *vma, u64 delta, u64 length)
a11c3198	369	{
42594600	370	struct nvkm_vm *vm = vma->vm;
eb813999 BS	371	if (vm->func->page->desc->func->unmap) {
	372	const struct nvkm_vmm_page *page = vm->func->page;
	373	while (page->shift != vma->node->type)
	374	page++;
	375	mutex_lock(&vm->mutex);
	376	nvkm_vmm_ptes_unmap(vm, page, (vma->node->offset << 12) + delta,
	377	vma->node->length << 12, false);
	378	mutex_unlock(&vm->mutex);
	379	return;
	380	}
a11c3198 BS	381	}
	382
	383	void
42594600	384	nvkm_vm_unmap(struct nvkm_vma *vma)
a11c3198	385	{
42594600	386	nvkm_vm_unmap_at(vma, 0, (u64)vma->node->length << 12);
eb813999 BS	387
	388	nvkm_memory_tags_put(vma->memory, vma->vm->mmu->subdev.device, &vma->tags);
	389	nvkm_memory_unref(&vma->memory);
a11c3198 BS	390	}
a11c3198 BS	391
a11c3198	392	int
42594600 BS	393	nvkm_vm_get(struct nvkm_vm *vm, u64 size, u32 page_shift, u32 access,
42594600 BS	394	struct nvkm_vma *vma)
a11c3198 BS	395	{
	396	u32 align = (1 << page_shift) >> 12;
	397	u32 msize = size >> 12;
a11c3198 BS	398	int ret;
a11c3198 BS	399
1de68568	400	mutex_lock(&vm->mutex);
42594600 BS	401	ret = nvkm_mm_head(&vm->mm, 0, page_shift, msize, msize, align,
42594600 BS	402	&vma->node);
a11c3198	403	if (unlikely(ret != 0)) {
1de68568	404	mutex_unlock(&vm->mutex);
a11c3198 BS	405	return ret;
	406	}
	407
eb813999 BS	408	if (vm->func->page->desc->func->unmap) {
	409	const struct nvkm_vmm_page *page = vm->func->page;
	410	while (page->shift != page_shift)
	411	page++;
	412
	413	ret = nvkm_vmm_ptes_get(vm, page, vma->node->offset << 12,
	414	vma->node->length << 12);
	415	if (ret) {
	416	nvkm_mm_free(&vm->mm, &vma->node);
	417	mutex_unlock(&vm->mutex);
	418	return ret;
	419	}
a11c3198	420	}
1de68568	421	mutex_unlock(&vm->mutex);
a11c3198	422
eb813999 BS	423	vma->memory = NULL;
eb813999 BS	424	vma->tags = NULL;
780194b1	425	vma->vm = NULL;
42594600	426	nvkm_vm_ref(vm, &vma->vm, NULL);
a11c3198 BS	427	vma->offset = (u64)vma->node->offset << 12;
	428	vma->access = access;
	429	return 0;
	430	}
	431
	432	void
42594600	433	nvkm_vm_put(struct nvkm_vma *vma)
a11c3198	434	{
227c95d9 BS	435	struct nvkm_mmu *mmu;
227c95d9 BS	436	struct nvkm_vm *vm;
a11c3198 BS	437
	438	if (unlikely(vma->node == NULL))
	439	return;
227c95d9 BS	440	vm = vma->vm;
	441	mmu = vm->mmu;
	442
eb813999 BS	443	nvkm_memory_tags_put(vma->memory, mmu->subdev.device, &vma->tags);
	444	nvkm_memory_unref(&vma->memory);
	445
1de68568	446	mutex_lock(&vm->mutex);
eb813999 BS	447	if (vm->func->page->desc->func->unmap) {
	448	const struct nvkm_vmm_page *page = vm->func->page;
	449	while (page->shift != vma->node->type)
	450	page++;
	451
	452	nvkm_vmm_ptes_put(vm, page, vma->node->offset << 12,
	453	vma->node->length << 12);
eb813999 BS	454	}
eb813999 BS	455
42594600	456	nvkm_mm_free(&vm->mm, &vma->node);
1de68568	457	mutex_unlock(&vm->mutex);
780194b1	458
42594600	459	nvkm_vm_ref(NULL, &vma->vm, NULL);
a11c3198 BS	460	}
a11c3198 BS	461
d8e83994 BS	462	int
	463	nvkm_vm_boot(struct nvkm_vm *vm, u64 size)
	464	{
26880e76	465	return nvkm_vmm_boot(vm);
d8e83994 BS	466	}
d8e83994 BS	467
806a7335 BS	468	static int
	469	nvkm_vm_legacy(struct nvkm_mmu *mmu, u64 offset, u64 length, u64 mm_offset,
	470	u32 block, struct nvkm_vm *vm)
a11c3198	471	{
a11c3198	472	u64 mm_length = (offset + length) - mm_offset;
a11c3198 BS	473	int ret;
a11c3198 BS	474
e0bacd2f	475	kref_init(&vm->refcount);
a11c3198	476
806a7335 BS	477	if (block > length)
	478	block = length;
	479
4d058fab	480	ret = nvkm_mm_init(&vm->mm, 0, mm_offset >> 12, mm_length >> 12,
42594600	481	block >> 12);
26880e76	482	if (ret)
806a7335	483	return ret;
806a7335 BS	484
	485	return 0;
	486	}
	487
3863c9bc	488	int
42594600	489	nvkm_vm_new(struct nvkm_device *device, u64 offset, u64 length, u64 mm_offset,
1de68568	490	struct lock_class_key key, struct nvkm_vm *pvm)
3863c9bc	491	{
c9582455	492	struct nvkm_mmu *mmu = device->mmu;
806a7335 BS	493
	494	*pvm = NULL;
	495	if (mmu->func->vmm.ctor) {
	496	int ret = mmu->func->vmm.ctor(mmu, mm_offset,
	497	offset + length - mm_offset,
	498	NULL, 0, key, "legacy", pvm);
	499	if (ret) {
	500	nvkm_vm_ref(NULL, pvm, NULL);
	501	return ret;
	502	}
	503
	504	ret = nvkm_vm_legacy(mmu, offset, length, mm_offset,
	505	(*pvm)->func->page_block ?
	506	(pvm)->func->page_block : 4096, pvm);
	507	if (ret)
	508	nvkm_vm_ref(NULL, pvm, NULL);
	509
	510	return ret;
	511	}
	512
af3b8d53	513	return -EINVAL;
3863c9bc BS	514	}
3863c9bc BS	515
a11c3198	516	static void
42594600	517	nvkm_vm_del(struct kref *kref)
a11c3198	518	{
42594600	519	struct nvkm_vm vm = container_of(kref, typeof(vm), refcount);
a11c3198	520
42594600	521	nvkm_mm_fini(&vm->mm);
806a7335 BS	522	if (vm->func)
806a7335 BS	523	nvkm_vmm_dtor(vm);
a11c3198 BS	524	kfree(vm);
	525	}
	526
	527	int
d30af7ce	528	nvkm_vm_ref(struct nvkm_vm ref, struct nvkm_vm ptr, struct nvkm_memory inst)
a11c3198	529	{
e0bacd2f	530	if (ref) {
d30af7ce	531	if (ref->func->join && inst) {
26880e76	532	int ret = ref->func->join(ref, inst);
d30af7ce BS	533	if (ret)
d30af7ce BS	534	return ret;
d30af7ce	535	}
a11c3198	536
e0bacd2f	537	kref_get(&ref->refcount);
a11c3198 BS	538	}
a11c3198 BS	539
e0bacd2f	540	if (*ptr) {
d30af7ce BS	541	if ((*ptr)->func->part && inst)
d30af7ce BS	542	(ptr)->func->part(ptr, inst);
42594600	543	kref_put(&(*ptr)->refcount, nvkm_vm_del);
a11c3198 BS	544	}
a11c3198 BS	545
e0bacd2f	546	*ptr = ref;
a11c3198 BS	547	return 0;
a11c3198 BS	548	}
c9582455 BS	549
	550	static int
	551	nvkm_mmu_oneinit(struct nvkm_subdev *subdev)
	552	{
	553	struct nvkm_mmu *mmu = nvkm_mmu(subdev);
806a7335 BS	554
	555	if (mmu->func->vmm.global) {
	556	int ret = nvkm_vm_new(subdev->device, 0, mmu->limit, 0,
	557	NULL, &mmu->vmm);
	558	if (ret)
	559	return ret;
	560	}
	561
c9582455 BS	562	return 0;
	563	}
	564
	565	static int
	566	nvkm_mmu_init(struct nvkm_subdev *subdev)
	567	{
	568	struct nvkm_mmu *mmu = nvkm_mmu(subdev);
	569	if (mmu->func->init)
	570	mmu->func->init(mmu);
	571	return 0;
	572	}
	573
	574	static void *
	575	nvkm_mmu_dtor(struct nvkm_subdev *subdev)
	576	{
	577	struct nvkm_mmu *mmu = nvkm_mmu(subdev);
9a45ddaa	578
806a7335	579	nvkm_vm_ref(NULL, &mmu->vmm, NULL);
9a45ddaa BS	580
9a45ddaa BS	581	nvkm_mmu_ptc_fini(mmu);
03b0ba7b	582	return mmu;
c9582455 BS	583	}
	584
	585	static const struct nvkm_subdev_func
	586	nvkm_mmu = {
	587	.dtor = nvkm_mmu_dtor,
	588	.oneinit = nvkm_mmu_oneinit,
	589	.init = nvkm_mmu_init,
	590	};
	591
	592	void
	593	nvkm_mmu_ctor(const struct nvkm_mmu_func func, struct nvkm_device device,
	594	int index, struct nvkm_mmu *mmu)
	595	{
56d06fa2	596	nvkm_subdev_ctor(&nvkm_mmu, device, index, &mmu->subdev);
c9582455 BS	597	mmu->func = func;
	598	mmu->limit = func->limit;
	599	mmu->dma_bits = func->dma_bits;
	600	mmu->lpg_shift = func->lpg_shift;
9a45ddaa	601	nvkm_mmu_ptc_init(mmu);
c9582455 BS	602	}
	603
	604	int
	605	nvkm_mmu_new_(const struct nvkm_mmu_func func, struct nvkm_device device,
	606	int index, struct nvkm_mmu **pmmu)
	607	{
	608	if (!(pmmu = kzalloc(sizeof(*pmmu), GFP_KERNEL)))
	609	return -ENOMEM;
	610	nvkm_mmu_ctor(func, device, index, *pmmu);
	611	return 0;
	612	}