Commit | Line | Data |
---|---|---|
2048e328 MY |
1 | /* |
2 | * Copyright (C) Fuzhou Rockchip Electronics Co.Ltd | |
3 | * Author:Mark Yao <mark.yao@rock-chips.com> | |
4 | * | |
5 | * This software is licensed under the terms of the GNU General Public | |
6 | * License version 2, as published by the Free Software Foundation, and | |
7 | * may be copied, distributed, and modified under those terms. | |
8 | * | |
9 | * This program is distributed in the hope that it will be useful, | |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | * GNU General Public License for more details. | |
13 | */ | |
14 | ||
15 | #include <drm/drm.h> | |
16 | #include <drm/drmP.h> | |
17 | #include <drm/drm_gem.h> | |
18 | #include <drm/drm_vma_manager.h> | |
19 | ||
2048e328 MY |
20 | #include "rockchip_drm_drv.h" |
21 | #include "rockchip_drm_gem.h" | |
22 | ||
f76c83b5 DK |
23 | static int rockchip_gem_alloc_buf(struct rockchip_gem_object *rk_obj, |
24 | bool alloc_kmap) | |
2048e328 MY |
25 | { |
26 | struct drm_gem_object *obj = &rk_obj->base; | |
27 | struct drm_device *drm = obj->dev; | |
28 | ||
00085f1e | 29 | rk_obj->dma_attrs = DMA_ATTR_WRITE_COMBINE; |
2048e328 | 30 | |
f76c83b5 | 31 | if (!alloc_kmap) |
00085f1e | 32 | rk_obj->dma_attrs |= DMA_ATTR_NO_KERNEL_MAPPING; |
f76c83b5 | 33 | |
2048e328 MY |
34 | rk_obj->kvaddr = dma_alloc_attrs(drm->dev, obj->size, |
35 | &rk_obj->dma_addr, GFP_KERNEL, | |
00085f1e | 36 | rk_obj->dma_attrs); |
4b9a90c0 | 37 | if (!rk_obj->kvaddr) { |
913bb40a | 38 | DRM_ERROR("failed to allocate %zu byte dma buffer", obj->size); |
4b9a90c0 | 39 | return -ENOMEM; |
2048e328 MY |
40 | } |
41 | ||
42 | return 0; | |
43 | } | |
44 | ||
45 | static void rockchip_gem_free_buf(struct rockchip_gem_object *rk_obj) | |
46 | { | |
47 | struct drm_gem_object *obj = &rk_obj->base; | |
48 | struct drm_device *drm = obj->dev; | |
49 | ||
50 | dma_free_attrs(drm->dev, obj->size, rk_obj->kvaddr, rk_obj->dma_addr, | |
00085f1e | 51 | rk_obj->dma_attrs); |
2048e328 MY |
52 | } |
53 | ||
41315b79 DK |
54 | static int rockchip_drm_gem_object_mmap(struct drm_gem_object *obj, |
55 | struct vm_area_struct *vma) | |
56 | ||
2048e328 | 57 | { |
41315b79 | 58 | int ret; |
2048e328 MY |
59 | struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj); |
60 | struct drm_device *drm = obj->dev; | |
2048e328 | 61 | |
41315b79 DK |
62 | /* |
63 | * dma_alloc_attrs() allocated a struct page table for rk_obj, so clear | |
64 | * VM_PFNMAP flag that was set by drm_gem_mmap_obj()/drm_gem_mmap(). | |
65 | */ | |
66 | vma->vm_flags &= ~VM_PFNMAP; | |
a8594f20 | 67 | vma->vm_pgoff = 0; |
2048e328 | 68 | |
41315b79 | 69 | ret = dma_mmap_attrs(drm->dev, vma, rk_obj->kvaddr, rk_obj->dma_addr, |
00085f1e | 70 | obj->size, rk_obj->dma_attrs); |
41315b79 DK |
71 | if (ret) |
72 | drm_gem_vm_close(vma); | |
73 | ||
74 | return ret; | |
75 | } | |
76 | ||
77 | int rockchip_gem_mmap_buf(struct drm_gem_object *obj, | |
78 | struct vm_area_struct *vma) | |
79 | { | |
41315b79 DK |
80 | int ret; |
81 | ||
41315b79 | 82 | ret = drm_gem_mmap_obj(obj, obj->size, vma); |
41315b79 DK |
83 | if (ret) |
84 | return ret; | |
85 | ||
86 | return rockchip_drm_gem_object_mmap(obj, vma); | |
2048e328 MY |
87 | } |
88 | ||
89 | /* drm driver mmap file operations */ | |
90 | int rockchip_gem_mmap(struct file *filp, struct vm_area_struct *vma) | |
91 | { | |
2048e328 | 92 | struct drm_gem_object *obj; |
2048e328 MY |
93 | int ret; |
94 | ||
41315b79 DK |
95 | ret = drm_gem_mmap(filp, vma); |
96 | if (ret) | |
97 | return ret; | |
2048e328 | 98 | |
41315b79 | 99 | obj = vma->vm_private_data; |
2048e328 | 100 | |
41315b79 | 101 | return rockchip_drm_gem_object_mmap(obj, vma); |
2048e328 MY |
102 | } |
103 | ||
104 | struct rockchip_gem_object * | |
f76c83b5 DK |
105 | rockchip_gem_create_object(struct drm_device *drm, unsigned int size, |
106 | bool alloc_kmap) | |
2048e328 MY |
107 | { |
108 | struct rockchip_gem_object *rk_obj; | |
109 | struct drm_gem_object *obj; | |
110 | int ret; | |
111 | ||
112 | size = round_up(size, PAGE_SIZE); | |
113 | ||
114 | rk_obj = kzalloc(sizeof(*rk_obj), GFP_KERNEL); | |
115 | if (!rk_obj) | |
116 | return ERR_PTR(-ENOMEM); | |
117 | ||
118 | obj = &rk_obj->base; | |
119 | ||
120 | drm_gem_private_object_init(drm, obj, size); | |
121 | ||
f76c83b5 | 122 | ret = rockchip_gem_alloc_buf(rk_obj, alloc_kmap); |
2048e328 MY |
123 | if (ret) |
124 | goto err_free_rk_obj; | |
125 | ||
126 | return rk_obj; | |
127 | ||
128 | err_free_rk_obj: | |
129 | kfree(rk_obj); | |
130 | return ERR_PTR(ret); | |
131 | } | |
132 | ||
133 | /* | |
134 | * rockchip_gem_free_object - (struct drm_driver)->gem_free_object callback | |
135 | * function | |
136 | */ | |
137 | void rockchip_gem_free_object(struct drm_gem_object *obj) | |
138 | { | |
139 | struct rockchip_gem_object *rk_obj; | |
140 | ||
141 | drm_gem_free_mmap_offset(obj); | |
142 | ||
143 | rk_obj = to_rockchip_obj(obj); | |
144 | ||
145 | rockchip_gem_free_buf(rk_obj); | |
146 | ||
147 | kfree(rk_obj); | |
148 | } | |
149 | ||
150 | /* | |
151 | * rockchip_gem_create_with_handle - allocate an object with the given | |
152 | * size and create a gem handle on it | |
153 | * | |
154 | * returns a struct rockchip_gem_object* on success or ERR_PTR values | |
155 | * on failure. | |
156 | */ | |
157 | static struct rockchip_gem_object * | |
158 | rockchip_gem_create_with_handle(struct drm_file *file_priv, | |
159 | struct drm_device *drm, unsigned int size, | |
160 | unsigned int *handle) | |
161 | { | |
162 | struct rockchip_gem_object *rk_obj; | |
163 | struct drm_gem_object *obj; | |
164 | int ret; | |
165 | ||
f76c83b5 | 166 | rk_obj = rockchip_gem_create_object(drm, size, false); |
2048e328 MY |
167 | if (IS_ERR(rk_obj)) |
168 | return ERR_CAST(rk_obj); | |
169 | ||
170 | obj = &rk_obj->base; | |
171 | ||
172 | /* | |
173 | * allocate a id of idr table where the obj is registered | |
174 | * and handle has the id what user can see. | |
175 | */ | |
176 | ret = drm_gem_handle_create(file_priv, obj, handle); | |
177 | if (ret) | |
178 | goto err_handle_create; | |
179 | ||
180 | /* drop reference from allocate - handle holds it now. */ | |
181 | drm_gem_object_unreference_unlocked(obj); | |
182 | ||
183 | return rk_obj; | |
184 | ||
185 | err_handle_create: | |
186 | rockchip_gem_free_object(obj); | |
187 | ||
188 | return ERR_PTR(ret); | |
189 | } | |
190 | ||
191 | int rockchip_gem_dumb_map_offset(struct drm_file *file_priv, | |
192 | struct drm_device *dev, uint32_t handle, | |
193 | uint64_t *offset) | |
194 | { | |
195 | struct drm_gem_object *obj; | |
196 | int ret; | |
197 | ||
a8ad0bd8 | 198 | obj = drm_gem_object_lookup(file_priv, handle); |
2048e328 MY |
199 | if (!obj) { |
200 | DRM_ERROR("failed to lookup gem object.\n"); | |
648a4ce7 | 201 | return -EINVAL; |
2048e328 MY |
202 | } |
203 | ||
204 | ret = drm_gem_create_mmap_offset(obj); | |
205 | if (ret) | |
206 | goto out; | |
207 | ||
208 | *offset = drm_vma_node_offset_addr(&obj->vma_node); | |
209 | DRM_DEBUG_KMS("offset = 0x%llx\n", *offset); | |
210 | ||
211 | out: | |
648a4ce7 DV |
212 | drm_gem_object_unreference_unlocked(obj); |
213 | ||
214 | return 0; | |
2048e328 MY |
215 | } |
216 | ||
217 | /* | |
218 | * rockchip_gem_dumb_create - (struct drm_driver)->dumb_create callback | |
219 | * function | |
220 | * | |
221 | * This aligns the pitch and size arguments to the minimum required. wrap | |
222 | * this into your own function if you need bigger alignment. | |
223 | */ | |
224 | int rockchip_gem_dumb_create(struct drm_file *file_priv, | |
225 | struct drm_device *dev, | |
226 | struct drm_mode_create_dumb *args) | |
227 | { | |
228 | struct rockchip_gem_object *rk_obj; | |
229 | int min_pitch = DIV_ROUND_UP(args->width * args->bpp, 8); | |
230 | ||
231 | /* | |
232 | * align to 64 bytes since Mali requires it. | |
233 | */ | |
e3c4abdb MY |
234 | args->pitch = ALIGN(min_pitch, 64); |
235 | args->size = args->pitch * args->height; | |
2048e328 MY |
236 | |
237 | rk_obj = rockchip_gem_create_with_handle(file_priv, dev, args->size, | |
238 | &args->handle); | |
239 | ||
240 | return PTR_ERR_OR_ZERO(rk_obj); | |
241 | } | |
242 | ||
243 | /* | |
244 | * Allocate a sg_table for this GEM object. | |
245 | * Note: Both the table's contents, and the sg_table itself must be freed by | |
246 | * the caller. | |
247 | * Returns a pointer to the newly allocated sg_table, or an ERR_PTR() error. | |
248 | */ | |
249 | struct sg_table *rockchip_gem_prime_get_sg_table(struct drm_gem_object *obj) | |
250 | { | |
251 | struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj); | |
252 | struct drm_device *drm = obj->dev; | |
253 | struct sg_table *sgt; | |
254 | int ret; | |
255 | ||
256 | sgt = kzalloc(sizeof(*sgt), GFP_KERNEL); | |
257 | if (!sgt) | |
258 | return ERR_PTR(-ENOMEM); | |
259 | ||
260 | ret = dma_get_sgtable_attrs(drm->dev, sgt, rk_obj->kvaddr, | |
261 | rk_obj->dma_addr, obj->size, | |
00085f1e | 262 | rk_obj->dma_attrs); |
2048e328 MY |
263 | if (ret) { |
264 | DRM_ERROR("failed to allocate sgt, %d\n", ret); | |
265 | kfree(sgt); | |
266 | return ERR_PTR(ret); | |
267 | } | |
268 | ||
269 | return sgt; | |
270 | } | |
271 | ||
272 | void *rockchip_gem_prime_vmap(struct drm_gem_object *obj) | |
273 | { | |
274 | struct rockchip_gem_object *rk_obj = to_rockchip_obj(obj); | |
275 | ||
00085f1e | 276 | if (rk_obj->dma_attrs & DMA_ATTR_NO_KERNEL_MAPPING) |
f76c83b5 DK |
277 | return NULL; |
278 | ||
2048e328 MY |
279 | return rk_obj->kvaddr; |
280 | } | |
281 | ||
282 | void rockchip_gem_prime_vunmap(struct drm_gem_object *obj, void *vaddr) | |
283 | { | |
284 | /* Nothing to do */ | |
285 | } |