1 /*
2  * Copyright 2011 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: Dave Airlie
23  */
24 
25 #include <linux/dma-buf.h>
26 
27 #include <drm/drmP.h>
28 
29 #include "nouveau_drm.h"
30 #include "nouveau_gem.h"
31 
32 static struct sg_table *nouveau_gem_map_dma_buf(struct dma_buf_attachment *attachment,
33 					  enum dma_data_direction dir)
34 {
35 	struct nouveau_bo *nvbo = attachment->dmabuf->priv;
36 	struct drm_device *dev = nvbo->gem->dev;
37 	int npages = nvbo->bo.num_pages;
38 	struct sg_table *sg;
39 	int nents;
40 
41 	mutex_lock(&dev->struct_mutex);
42 	sg = drm_prime_pages_to_sg(nvbo->bo.ttm->pages, npages);
43 	nents = dma_map_sg(attachment->dev, sg->sgl, sg->nents, dir);
44 	mutex_unlock(&dev->struct_mutex);
45 	return sg;
46 }
47 
48 static void nouveau_gem_unmap_dma_buf(struct dma_buf_attachment *attachment,
49 				      struct sg_table *sg, enum dma_data_direction dir)
50 {
51 	dma_unmap_sg(attachment->dev, sg->sgl, sg->nents, dir);
52 	sg_free_table(sg);
53 	kfree(sg);
54 }
55 
56 static void nouveau_gem_dmabuf_release(struct dma_buf *dma_buf)
57 {
58 	struct nouveau_bo *nvbo = dma_buf->priv;
59 
60 	if (nvbo->gem->export_dma_buf == dma_buf) {
61 		nvbo->gem->export_dma_buf = NULL;
62 		drm_gem_object_unreference_unlocked(nvbo->gem);
63 	}
64 }
65 
66 static void *nouveau_gem_kmap_atomic(struct dma_buf *dma_buf, unsigned long page_num)
67 {
68 	return NULL;
69 }
70 
71 static void nouveau_gem_kunmap_atomic(struct dma_buf *dma_buf, unsigned long page_num, void *addr)
72 {
73 
74 }
75 static void *nouveau_gem_kmap(struct dma_buf *dma_buf, unsigned long page_num)
76 {
77 	return NULL;
78 }
79 
80 static void nouveau_gem_kunmap(struct dma_buf *dma_buf, unsigned long page_num, void *addr)
81 {
82 
83 }
84 
85 static int nouveau_gem_prime_mmap(struct dma_buf *dma_buf, struct vm_area_struct *vma)
86 {
87 	return -EINVAL;
88 }
89 
90 static void *nouveau_gem_prime_vmap(struct dma_buf *dma_buf)
91 {
92 	struct nouveau_bo *nvbo = dma_buf->priv;
93 	struct drm_device *dev = nvbo->gem->dev;
94 	int ret;
95 
96 	mutex_lock(&dev->struct_mutex);
97 	if (nvbo->vmapping_count) {
98 		nvbo->vmapping_count++;
99 		goto out_unlock;
100 	}
101 
102 	ret = ttm_bo_kmap(&nvbo->bo, 0, nvbo->bo.num_pages,
103 			  &nvbo->dma_buf_vmap);
104 	if (ret) {
105 		mutex_unlock(&dev->struct_mutex);
106 		return ERR_PTR(ret);
107 	}
108 	nvbo->vmapping_count = 1;
109 out_unlock:
110 	mutex_unlock(&dev->struct_mutex);
111 	return nvbo->dma_buf_vmap.virtual;
112 }
113 
114 static void nouveau_gem_prime_vunmap(struct dma_buf *dma_buf, void *vaddr)
115 {
116 	struct nouveau_bo *nvbo = dma_buf->priv;
117 	struct drm_device *dev = nvbo->gem->dev;
118 
119 	mutex_lock(&dev->struct_mutex);
120 	nvbo->vmapping_count--;
121 	if (nvbo->vmapping_count == 0) {
122 		ttm_bo_kunmap(&nvbo->dma_buf_vmap);
123 	}
124 	mutex_unlock(&dev->struct_mutex);
125 }
126 
127 static const struct dma_buf_ops nouveau_dmabuf_ops =  {
128 	.map_dma_buf = nouveau_gem_map_dma_buf,
129 	.unmap_dma_buf = nouveau_gem_unmap_dma_buf,
130 	.release = nouveau_gem_dmabuf_release,
131 	.kmap = nouveau_gem_kmap,
132 	.kmap_atomic = nouveau_gem_kmap_atomic,
133 	.kunmap = nouveau_gem_kunmap,
134 	.kunmap_atomic = nouveau_gem_kunmap_atomic,
135 	.mmap = nouveau_gem_prime_mmap,
136 	.vmap = nouveau_gem_prime_vmap,
137 	.vunmap = nouveau_gem_prime_vunmap,
138 };
139 
140 static int
141 nouveau_prime_new(struct drm_device *dev,
142 		  size_t size,
143 		  struct sg_table *sg,
144 		  struct nouveau_bo **pnvbo)
145 {
146 	struct nouveau_bo *nvbo;
147 	u32 flags = 0;
148 	int ret;
149 
150 	flags = TTM_PL_FLAG_TT;
151 
152 	ret = nouveau_bo_new(dev, size, 0, flags, 0, 0,
153 			     sg, pnvbo);
154 	if (ret)
155 		return ret;
156 	nvbo = *pnvbo;
157 
158 	nvbo->valid_domains = NOUVEAU_GEM_DOMAIN_GART;
159 	nvbo->gem = drm_gem_object_alloc(dev, nvbo->bo.mem.size);
160 	if (!nvbo->gem) {
161 		nouveau_bo_ref(NULL, pnvbo);
162 		return -ENOMEM;
163 	}
164 
165 	nvbo->gem->driver_private = nvbo;
166 	return 0;
167 }
168 
169 struct dma_buf *nouveau_gem_prime_export(struct drm_device *dev,
170 				struct drm_gem_object *obj, int flags)
171 {
172 	struct nouveau_bo *nvbo = nouveau_gem_object(obj);
173 	int ret = 0;
174 
175 	/* pin buffer into GTT */
176 	ret = nouveau_bo_pin(nvbo, TTM_PL_FLAG_TT);
177 	if (ret)
178 		return ERR_PTR(-EINVAL);
179 
180 	return dma_buf_export(nvbo, &nouveau_dmabuf_ops, obj->size, flags);
181 }
182 
183 struct drm_gem_object *nouveau_gem_prime_import(struct drm_device *dev,
184 				struct dma_buf *dma_buf)
185 {
186 	struct dma_buf_attachment *attach;
187 	struct sg_table *sg;
188 	struct nouveau_bo *nvbo;
189 	int ret;
190 
191 	if (dma_buf->ops == &nouveau_dmabuf_ops) {
192 		nvbo = dma_buf->priv;
193 		if (nvbo->gem) {
194 			if (nvbo->gem->dev == dev) {
195 				drm_gem_object_reference(nvbo->gem);
196 				dma_buf_put(dma_buf);
197 				return nvbo->gem;
198 			}
199 		}
200 	}
201 	/* need to attach */
202 	attach = dma_buf_attach(dma_buf, dev->dev);
203 	if (IS_ERR(attach))
204 		return ERR_PTR(PTR_ERR(attach));
205 
206 	sg = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
207 	if (IS_ERR(sg)) {
208 		ret = PTR_ERR(sg);
209 		goto fail_detach;
210 	}
211 
212 	ret = nouveau_prime_new(dev, dma_buf->size, sg, &nvbo);
213 	if (ret)
214 		goto fail_unmap;
215 
216 	nvbo->gem->import_attach = attach;
217 
218 	return nvbo->gem;
219 
220 fail_unmap:
221 	dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
222 fail_detach:
223 	dma_buf_detach(dma_buf, attach);
224 	return ERR_PTR(ret);
225 }
226 
227