1/*
 2 * Copyright (C) 2016 Red Hat
 3 * Author: Rob Clark <robdclark@gmail.com>
 4 *
 5 * This program is free software; you can redistribute it and/or modify it
 6 * under the terms of the GNU General Public License version 2 as published by
 7 * the Free Software Foundation.
 8 *
 9 * This program is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
12 * more details.
13 *
14 * You should have received a copy of the GNU General Public License along with
15 * this program.  If not, see <http://www.gnu.org/licenses/>.
16 */
17
18#include "msm_drv.h"
19#include "msm_gem.h"
20#include "msm_mmu.h"
21
22void
23msm_gem_unmap_vma(struct msm_gem_address_space *aspace,
24		struct msm_gem_vma *vma, struct sg_table *sgt)
25{
26	if (!vma->iova)
27		return;
28
29	if (aspace->mmu) {
30		unsigned size = vma->node.size << PAGE_SHIFT;
31		aspace->mmu->funcs->unmap(aspace->mmu, vma->iova, sgt, size);
32	}
33
34	drm_mm_remove_node(&vma->node);
35
36	vma->iova = 0;
37}
38
39int
40msm_gem_map_vma(struct msm_gem_address_space *aspace,
41		struct msm_gem_vma *vma, struct sg_table *sgt, int npages)
42{
43	int ret;
44
45	if (WARN_ON(drm_mm_node_allocated(&vma->node)))
46		return 0;
47
48	ret = drm_mm_insert_node(&aspace->mm, &vma->node, npages,
49			0, DRM_MM_SEARCH_DEFAULT);
50	if (ret)
51		return ret;
52
53	vma->iova = vma->node.start << PAGE_SHIFT;
54
55	if (aspace->mmu) {
56		unsigned size = npages << PAGE_SHIFT;
57		ret = aspace->mmu->funcs->map(aspace->mmu, vma->iova, sgt,
58				size, IOMMU_READ | IOMMU_WRITE);
59	}
60
61	return ret;
62}
63
64void
65msm_gem_address_space_destroy(struct msm_gem_address_space *aspace)
66{
67	drm_mm_takedown(&aspace->mm);
68	if (aspace->mmu)
69		aspace->mmu->funcs->destroy(aspace->mmu);
70	kfree(aspace);
71}
72
73struct msm_gem_address_space *
74msm_gem_address_space_create(struct device *dev, struct iommu_domain *domain,
75		const char *name)
76{
77	struct msm_gem_address_space *aspace;
78
79	aspace = kzalloc(sizeof(*aspace), GFP_KERNEL);
80	if (!aspace)
81		return ERR_PTR(-ENOMEM);
82
83	aspace->name = name;
84	aspace->mmu = msm_iommu_new(dev, domain);
85
86	drm_mm_init(&aspace->mm, (domain->geometry.aperture_start >> PAGE_SHIFT),
87			(domain->geometry.aperture_end >> PAGE_SHIFT) - 1);
88
89	return aspace;
90}