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
14struct 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
25int 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
37void msm_gem_prime_vunmap(struct drm_gem_object *obj, struct dma_buf_map *map)
38{
39	msm_gem_put_vaddr(obj);
40}
41
42int 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
53struct 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
59int 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
66void msm_gem_prime_unpin(struct drm_gem_object *obj)
67{
68	if (!obj->import_attach)
69		msm_gem_put_pages(obj);
70}