1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (C) 2013 Red Hat
4  * Author: Rob Clark <robdclark@gmail.com>
5  */
6 
7 #include <linux/dma-buf.h>
8 
9 #include <drm/drm_prime.h>
10 
11 #include "msm_drv.h"
12 #include "msm_gem.h"
13 
14 struct sg_table *msm_gem_prime_get_sg_table(struct drm_gem_object *obj)
15 {
16 	struct msm_gem_object *msm_obj = to_msm_bo(obj);
17 	int npages = obj->size >> PAGE_SHIFT;
18 
19 	if (WARN_ON(!msm_obj->pages))  /* should have already pinned! */
20 		return NULL;
21 
22 	return drm_prime_pages_to_sg(obj->dev, msm_obj->pages, npages);
23 }
24 
25 int msm_gem_prime_vmap(struct drm_gem_object *obj, struct dma_buf_map *map)
26 {
27 	void *vaddr;
28 
29 	vaddr = msm_gem_get_vaddr(obj);
30 	if (IS_ERR(vaddr))
31 		return PTR_ERR(vaddr);
32 	dma_buf_map_set_vaddr(map, vaddr);
33 
34 	return 0;
35 }
36 
37 void msm_gem_prime_vunmap(struct drm_gem_object *obj, struct dma_buf_map *map)
38 {
39 	msm_gem_put_vaddr(obj);
40 }
41 
42 int msm_gem_prime_mmap(struct drm_gem_object *obj, struct vm_area_struct *vma)
43 {
44 	int ret;
45 
46 	ret = drm_gem_mmap_obj(obj, obj->size, vma);
47 	if (ret < 0)
48 		return ret;
49 
50 	return msm_gem_mmap_obj(vma->vm_private_data, vma);
51 }
52 
53 struct drm_gem_object *msm_gem_prime_import_sg_table(struct drm_device *dev,
54 		struct dma_buf_attachment *attach, struct sg_table *sg)
55 {
56 	return msm_gem_import(dev, attach->dmabuf, sg);
57 }
58 
59 int msm_gem_prime_pin(struct drm_gem_object *obj)
60 {
61 	if (!obj->import_attach)
62 		msm_gem_get_pages(obj);
63 	return 0;
64 }
65 
66 void msm_gem_prime_unpin(struct drm_gem_object *obj)
67 {
68 	if (!obj->import_attach)
69 		msm_gem_put_pages(obj);
70 }
71