2 * drivers/video/tegra/nvmap/nvmap.h
4 * GPU memory management driver for Tegra
6 * Copyright (c) 2009-2014, NVIDIA CORPORATION. All rights reserved.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful, but WITHOUT
14 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
15 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
23 #ifndef __VIDEO_TEGRA_NVMAP_NVMAP_H
24 #define __VIDEO_TEGRA_NVMAP_NVMAP_H
26 #include <linux/list.h>
28 #include <linux/mutex.h>
29 #include <linux/rbtree.h>
30 #include <linux/sched.h>
31 #include <linux/wait.h>
32 #include <linux/atomic.h>
33 #include <linux/dma-buf.h>
34 #include <linux/syscalls.h>
36 #include <linux/miscdevice.h>
37 #include <linux/nvmap.h>
38 #include <linux/vmalloc.h>
39 #include <linux/slab.h>
41 #include <linux/workqueue.h>
42 #include <linux/dma-mapping.h>
43 #include <linux/dma-direction.h>
44 #include <linux/platform_device.h>
46 #include <asm/cacheflush.h>
48 #include <asm/outercache.h>
50 #include "nvmap_heap.h"
52 #ifdef CONFIG_NVMAP_HIGHMEM_ONLY
53 #define __GFP_NVMAP __GFP_HIGHMEM
55 #define __GFP_NVMAP (GFP_KERNEL | __GFP_HIGHMEM)
58 #define GFP_NVMAP (__GFP_NVMAP | __GFP_NOWARN)
61 #define NVMAP_LAZY_VFREE
67 void _nvmap_handle_free(struct nvmap_handle *h);
68 /* holds max number of handles allocted per process at any time */
69 extern u32 nvmap_max_handle_count;
71 /* If set force zeroed memory to userspace. */
72 extern bool zero_memory;
75 #define PG_PROT_KERNEL PAGE_KERNEL
76 #define FLUSH_DCACHE_AREA __flush_dcache_area
77 #define outer_flush_range(s, e)
78 #define outer_inv_range(s, e)
79 #define outer_clean_range(s, e)
80 #define outer_flush_all()
81 #define outer_clean_all()
82 extern void __clean_dcache_page(struct page *);
83 extern void __flush_dcache_page(struct page *);
85 #define PG_PROT_KERNEL pgprot_kernel
86 #define FLUSH_DCACHE_AREA __cpuc_flush_dcache_area
87 extern void __flush_dcache_page(struct address_space *, struct page *);
90 struct nvmap_vma_list {
91 struct list_head list;
92 struct vm_area_struct *vma;
96 /* handles allocated using shared system memory (either IOVMM- or high-order
98 struct nvmap_pgalloc {
100 bool contig; /* contiguous system memory */
101 atomic_t ndirty; /* count number of dirty pages */
104 struct nvmap_handle {
105 struct rb_node node; /* entry on global handle tree */
106 atomic_t ref; /* reference count (i.e., # of duplications) */
107 atomic_t pin; /* pin count */
108 u32 flags; /* caching flags */
109 size_t size; /* padded (as-allocated) size */
110 size_t orig_size; /* original (as-requested) size */
112 u8 kind; /* memory kind (0=pitch, !0 -> blocklinear) */
113 struct nvmap_client *owner;
116 * dma_buf necessities. An attachment is made on dma_buf allocation to
117 * facilitate the nvmap_pin* APIs.
119 struct dma_buf *dmabuf;
120 struct dma_buf_attachment *attachment;
123 struct nvmap_pgalloc pgalloc;
124 struct nvmap_heap_block *carveout;
126 bool heap_pgalloc; /* handle is page allocated (sysmem / iovmm) */
127 bool alloc; /* handle has memory allocated */
128 u32 heap_type; /* handle heap is allocated from */
129 u32 userflags; /* flags passed from userspace */
130 void *vaddr; /* mapping used inside kernel */
131 struct list_head vmas; /* list of all user vma's */
132 atomic_t share_count; /* number of processes sharing the handle */
134 void *nvhost_priv; /* nvhost private data */
135 void (*nvhost_priv_delete)(void *priv);
138 /* handle_ref objects are client-local references to an nvmap_handle;
139 * they are distinct objects so that handles can be unpinned and
140 * unreferenced the correct number of times when a client abnormally
142 struct nvmap_handle_ref {
143 struct nvmap_handle *handle;
145 atomic_t dupes; /* number of times to free on file close */
146 atomic_t pin; /* number of times to unpin on free */
149 #ifdef CONFIG_NVMAP_PAGE_POOLS
152 * This is the default ratio defining pool size. It can be thought of as pool
153 * size in either MB per GB or KB per MB. That means the max this number can
154 * be is 1024 (all physical memory - not a very good idea) or 0 (no page pool
157 #define NVMAP_PP_POOL_SIZE (128)
160 * The wakeup threshold is how many empty page slots there need to be in order
161 * for the background allocater to be woken up.
163 #define NVMAP_PP_DEF_FILL_THRESH (4096)
166 * For when memory does not require zeroing this is the minimum number of pages
167 * remaining in the page pools before the background allocer is woken up. This
168 * essentially disables the page pools (unless its extremely small).
170 #define NVMAP_PP_ZERO_MEM_FILL_MIN (2048)
172 struct nvmap_page_pool {
174 u32 count; /* Number of pages in the page list. */
175 u32 max; /* Max length of the page list. */
176 struct list_head page_list;
178 #ifdef CONFIG_NVMAP_PAGE_POOL_DEBUG
186 int nvmap_page_pool_init(struct nvmap_device *dev);
187 int nvmap_page_pool_fini(struct nvmap_device *dev);
188 struct page *nvmap_page_pool_alloc(struct nvmap_page_pool *pool);
189 bool nvmap_page_pool_fill(struct nvmap_page_pool *pool, struct page *page);
190 int nvmap_page_pool_alloc_lots(struct nvmap_page_pool *pool,
191 struct page **pages, u32 nr);
192 int nvmap_page_pool_fill_lots(struct nvmap_page_pool *pool,
193 struct page **pages, u32 nr);
194 int nvmap_page_pool_clear(void);
195 int nvmap_page_pool_debugfs_init(struct dentry *nvmap_root);
198 struct nvmap_client {
200 struct rb_root handle_refs;
201 struct mutex ref_lock;
204 struct task_struct *task;
205 struct list_head list;
211 struct nvmap_vma_priv {
212 struct nvmap_handle *handle;
214 atomic_t count; /* number of processes cloning the VMA */
217 struct nvmap_device {
218 struct rb_root handles;
219 spinlock_t handle_lock;
220 struct miscdevice dev_user;
221 struct nvmap_carveout_node *heaps;
223 #ifdef CONFIG_NVMAP_PAGE_POOLS
224 struct nvmap_page_pool pool;
226 struct list_head clients;
227 spinlock_t clients_lock;
248 atomic64_t stats[NS_NUM];
252 extern struct nvmap_stats nvmap_stats;
253 extern struct nvmap_device *nvmap_dev;
255 void nvmap_stats_inc(enum nvmap_stats_t, size_t size);
256 void nvmap_stats_dec(enum nvmap_stats_t, size_t size);
257 u64 nvmap_stats_read(enum nvmap_stats_t);
259 static inline void nvmap_ref_lock(struct nvmap_client *priv)
261 mutex_lock(&priv->ref_lock);
264 static inline void nvmap_ref_unlock(struct nvmap_client *priv)
266 mutex_unlock(&priv->ref_lock);
270 * NOTE: this does not ensure the continued existence of the underlying
271 * dma_buf. If you want ensure the existence of the dma_buf you must get an
272 * nvmap_handle_ref as that is what tracks the dma_buf refs.
274 static inline struct nvmap_handle *nvmap_handle_get(struct nvmap_handle *h)
276 if (WARN_ON(!virt_addr_valid(h))) {
277 pr_err("%s: invalid handle\n", current->group_leader->comm);
281 if (unlikely(atomic_inc_return(&h->ref) <= 1)) {
282 pr_err("%s: %s attempt to get a freed handle\n",
283 __func__, current->group_leader->comm);
291 static inline pgprot_t nvmap_pgprot(struct nvmap_handle *h, pgprot_t prot)
293 if (h->flags == NVMAP_HANDLE_UNCACHEABLE) {
295 if (h->owner && !h->owner->warned) {
296 char task_comm[TASK_COMM_LEN];
297 h->owner->warned = 1;
298 get_task_comm(task_comm, h->owner->task);
299 pr_err("PID %d: %s: WARNING: "
300 "NVMAP_HANDLE_WRITE_COMBINE "
301 "should be used in place of "
302 "NVMAP_HANDLE_UNCACHEABLE on ARM64\n",
303 h->owner->task->pid, task_comm);
306 return pgprot_noncached(prot);
308 else if (h->flags == NVMAP_HANDLE_WRITE_COMBINE)
309 return pgprot_dmacoherent(prot);
313 struct nvmap_heap_block *nvmap_carveout_alloc(struct nvmap_client *dev,
314 struct nvmap_handle *handle,
317 unsigned long nvmap_carveout_usage(struct nvmap_client *c,
318 struct nvmap_heap_block *b);
320 struct nvmap_carveout_node;
322 void nvmap_handle_put(struct nvmap_handle *h);
324 struct nvmap_handle_ref *__nvmap_validate_locked(struct nvmap_client *priv,
325 struct nvmap_handle *h);
327 struct nvmap_handle *nvmap_validate_get(struct nvmap_handle *h);
329 struct nvmap_handle_ref *nvmap_create_handle(struct nvmap_client *client,
332 struct nvmap_handle_ref *nvmap_duplicate_handle(struct nvmap_client *client,
333 struct nvmap_handle *h, bool skip_val);
335 struct nvmap_handle_ref *nvmap_create_handle_from_fd(
336 struct nvmap_client *client, int fd);
338 int nvmap_alloc_handle(struct nvmap_client *client,
339 struct nvmap_handle *h, unsigned int heap_mask,
340 size_t align, u8 kind,
343 void nvmap_free_handle(struct nvmap_client *c, struct nvmap_handle *h);
345 void nvmap_free_handle_user_id(struct nvmap_client *c, unsigned long user_id);
347 int nvmap_pin_ids(struct nvmap_client *client,
348 unsigned int nr, struct nvmap_handle * const *ids);
350 void nvmap_unpin_ids(struct nvmap_client *priv,
351 unsigned int nr, struct nvmap_handle * const *ids);
353 int nvmap_handle_remove(struct nvmap_device *dev, struct nvmap_handle *h);
355 void nvmap_handle_add(struct nvmap_device *dev, struct nvmap_handle *h);
357 int is_nvmap_vma(struct vm_area_struct *vma);
359 int nvmap_get_dmabuf_fd(struct nvmap_client *client, struct nvmap_handle *h);
360 struct nvmap_handle *nvmap_get_id_from_dmabuf_fd(struct nvmap_client *client,
363 int nvmap_get_handle_param(struct nvmap_client *client,
364 struct nvmap_handle_ref *ref, u32 param, u64 *result);
366 struct nvmap_client *nvmap_client_get(struct nvmap_client *client);
368 void nvmap_client_put(struct nvmap_client *c);
370 struct nvmap_handle *unmarshal_user_handle(__u32 handle);
372 /* MM definitions. */
373 extern size_t cache_maint_inner_threshold;
374 extern size_t cache_maint_outer_threshold;
376 extern void v7_flush_kern_cache_all(void);
377 extern void v7_clean_kern_cache_all(void *);
378 extern void __flush_dcache_all(void *arg);
379 extern void __clean_dcache_all(void *arg);
381 void inner_flush_cache_all(void);
382 void inner_clean_cache_all(void);
383 void nvmap_clean_cache(struct page **pages, int numpages);
384 void nvmap_flush_cache(struct page **pages, int numpages);
386 int nvmap_do_cache_maint_list(struct nvmap_handle **handles, u32 *offsets,
387 u32 *sizes, int op, int nr);
389 /* Internal API to support dmabuf */
390 struct dma_buf *__nvmap_dmabuf_export(struct nvmap_client *client,
391 struct nvmap_handle *handle);
392 struct dma_buf *__nvmap_make_dmabuf(struct nvmap_client *client,
393 struct nvmap_handle *handle);
394 struct sg_table *__nvmap_sg_table(struct nvmap_client *client,
395 struct nvmap_handle *h);
396 void __nvmap_free_sg_table(struct nvmap_client *client,
397 struct nvmap_handle *h, struct sg_table *sgt);
398 void *__nvmap_kmap(struct nvmap_handle *h, unsigned int pagenum);
399 void __nvmap_kunmap(struct nvmap_handle *h, unsigned int pagenum, void *addr);
400 void *__nvmap_mmap(struct nvmap_handle *h);
401 void __nvmap_munmap(struct nvmap_handle *h, void *addr);
402 int __nvmap_map(struct nvmap_handle *h, struct vm_area_struct *vma);
403 int __nvmap_get_handle_param(struct nvmap_client *client,
404 struct nvmap_handle *h, u32 param, u64 *result);
405 int __nvmap_do_cache_maint(struct nvmap_client *client, struct nvmap_handle *h,
406 unsigned long start, unsigned long end,
407 unsigned int op, bool clean_only_dirty);
408 struct nvmap_client *__nvmap_create_client(struct nvmap_device *dev,
410 struct dma_buf *__nvmap_dmabuf_export_from_ref(struct nvmap_handle_ref *ref);
411 struct nvmap_handle *__nvmap_ref_to_id(struct nvmap_handle_ref *ref);
412 int __nvmap_pin(struct nvmap_handle_ref *ref, phys_addr_t *phys);
413 void __nvmap_unpin(struct nvmap_handle_ref *ref);
414 int __nvmap_dmabuf_fd(struct nvmap_client *client,
415 struct dma_buf *dmabuf, int flags);
417 void nvmap_dmabuf_debugfs_init(struct dentry *nvmap_root);
418 int nvmap_dmabuf_stash_init(void);
420 void *nvmap_altalloc(size_t len);
421 void nvmap_altfree(void *ptr, size_t len);
423 static inline struct page *nvmap_to_page(struct page *page)
425 return (struct page *)((unsigned long)page & ~3UL);
428 static inline bool nvmap_page_dirty(struct page *page)
430 return (unsigned long)page & 1UL;
433 static inline void nvmap_page_mkdirty(struct page **page)
435 *page = (struct page *)((unsigned long)*page | 1UL);
438 static inline void nvmap_page_mkclean(struct page **page)
440 *page = (struct page *)((unsigned long)*page & ~1UL);
443 static inline bool nvmap_page_reserved(struct page *page)
445 return !!((unsigned long)page & 2UL);
448 static inline void nvmap_page_mkreserved(struct page **page)
450 *page = (struct page *)((unsigned long)*page | 2UL);
453 static inline void nvmap_page_mkunreserved(struct page **page)
455 *page = (struct page *)((unsigned long)*page & ~2UL);
459 * FIXME: assume user space requests for reserve operations
462 static inline void nvmap_handle_mk(struct nvmap_handle *h,
463 u32 offset, u32 size,
464 void (*fn)(struct page **))
467 int start_page = PAGE_ALIGN(offset) >> PAGE_SHIFT;
468 int end_page = (offset + size) >> PAGE_SHIFT;
470 if (h->heap_pgalloc) {
471 for (i = start_page; i < end_page; i++)
472 fn(&h->pgalloc.pages[i]);
476 static inline void nvmap_handle_mkclean(struct nvmap_handle *h,
477 u32 offset, u32 size)
479 nvmap_handle_mk(h, offset, size, nvmap_page_mkclean);
482 static inline void nvmap_handle_mkunreserved(struct nvmap_handle *h,
483 u32 offset, u32 size)
485 nvmap_handle_mk(h, offset, size, nvmap_page_mkunreserved);
488 static inline void nvmap_handle_mkreserved(struct nvmap_handle *h,
489 u32 offset, u32 size)
491 nvmap_handle_mk(h, offset, size, nvmap_page_mkreserved);
494 static inline struct page **nvmap_pages(struct page **pg_pages, u32 nr_pages)
499 pages = nvmap_altalloc(sizeof(*pages) * nr_pages);
503 for (i = 0; i < nr_pages; i++)
504 pages[i] = nvmap_to_page(pg_pages[i]);
509 void nvmap_zap_handle(struct nvmap_handle *handle, u32 offset, u32 size);
511 void nvmap_zap_handles(struct nvmap_handle **handles, u32 *offsets,
514 void nvmap_vma_open(struct vm_area_struct *vma);
516 int nvmap_reserve_pages(struct nvmap_handle **handles, u32 *offsets,
517 u32 *sizes, u32 nr, u32 op);
519 #endif /* __VIDEO_TEGRA_NVMAP_NVMAP_H */