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 #ifdef CONFIG_NVMAP_FORCE_ZEROED_USER_PAGES
59 #define NVMAP_ZEROED_PAGES __GFP_ZERO
61 #define NVMAP_ZEROED_PAGES 0
64 #define GFP_NVMAP (__GFP_NVMAP | __GFP_NOWARN | NVMAP_ZEROED_PAGES)
66 extern bool zero_memory;
69 #define NVMAP_LAZY_VFREE
75 void _nvmap_handle_free(struct nvmap_handle *h);
76 /* holds max number of handles allocted per process at any time */
77 extern u32 nvmap_max_handle_count;
79 /* If set force zeroed memory to userspace. */
80 extern bool zero_memory;
83 #define PG_PROT_KERNEL PAGE_KERNEL
84 #define FLUSH_DCACHE_AREA __flush_dcache_area
85 #define outer_flush_range(s, e)
86 #define outer_inv_range(s, e)
87 #define outer_clean_range(s, e)
88 #define outer_flush_all()
89 #define outer_clean_all()
90 extern void __clean_dcache_page(struct page *);
91 extern void __flush_dcache_page(struct page *);
93 #define PG_PROT_KERNEL pgprot_kernel
94 #define FLUSH_DCACHE_AREA __cpuc_flush_dcache_area
95 extern void __flush_dcache_page(struct address_space *, struct page *);
98 struct nvmap_vma_list {
99 struct list_head list;
100 struct vm_area_struct *vma;
103 /* handles allocated using shared system memory (either IOVMM- or high-order
104 * page allocations */
105 struct nvmap_pgalloc {
107 bool contig; /* contiguous system memory */
108 atomic_t ndirty; /* count number of dirty pages */
111 struct nvmap_handle {
112 struct rb_node node; /* entry on global handle tree */
113 atomic_t ref; /* reference count (i.e., # of duplications) */
114 atomic_t pin; /* pin count */
115 u32 flags; /* caching flags */
116 size_t size; /* padded (as-allocated) size */
117 size_t orig_size; /* original (as-requested) size */
119 u8 kind; /* memory kind (0=pitch, !0 -> blocklinear) */
120 struct nvmap_client *owner;
123 * dma_buf necessities. An attachment is made on dma_buf allocation to
124 * facilitate the nvmap_pin* APIs.
126 struct dma_buf *dmabuf;
127 struct dma_buf_attachment *attachment;
130 struct nvmap_pgalloc pgalloc;
131 struct nvmap_heap_block *carveout;
133 bool heap_pgalloc; /* handle is page allocated (sysmem / iovmm) */
134 bool alloc; /* handle has memory allocated */
135 u32 heap_type; /* handle heap is allocated from */
136 u32 userflags; /* flags passed from userspace */
137 void *vaddr; /* mapping used inside kernel */
138 struct list_head vmas; /* list of all user vma's */
139 atomic_t share_count; /* number of processes sharing the handle */
141 void *nvhost_priv; /* nvhost private data */
142 void (*nvhost_priv_delete)(void *priv);
145 /* handle_ref objects are client-local references to an nvmap_handle;
146 * they are distinct objects so that handles can be unpinned and
147 * unreferenced the correct number of times when a client abnormally
149 struct nvmap_handle_ref {
150 struct nvmap_handle *handle;
152 atomic_t dupes; /* number of times to free on file close */
153 atomic_t pin; /* number of times to unpin on free */
156 #ifdef CONFIG_NVMAP_PAGE_POOLS
159 * This is the default ratio defining pool size. It can be thought of as pool
160 * size in either MB per GB or KB per MB. That means the max this number can
161 * be is 1024 (all physical memory - not a very good idea) or 0 (no page pool
164 #define NVMAP_PP_POOL_SIZE (128)
167 * The wakeup threshold is how many empty page slots there need to be in order
168 * for the background allocater to be woken up.
170 #define NVMAP_PP_DEF_FILL_THRESH (4096)
173 * For when memory does not require zeroing this is the minimum number of pages
174 * remaining in the page pools before the background allocer is woken up. This
175 * essentially disables the page pools (unless its extremely small).
177 #define NVMAP_PP_ZERO_MEM_FILL_MIN (2048)
179 struct nvmap_page_pool {
181 u32 alloc; /* Alloc index. */
182 u32 fill; /* Fill index. */
183 u32 count; /* Number of pages in the table. */
184 u32 length; /* Length of the pages array. */
185 struct page **page_array;
187 #ifdef CONFIG_NVMAP_PAGE_POOL_DEBUG
195 #define pp_empty(pp) \
196 ((pp)->fill == (pp)->alloc && !(pp)->page_array[(pp)->alloc])
197 #define pp_full(pp) \
198 ((pp)->fill == (pp)->alloc && (pp)->page_array[(pp)->alloc])
200 #define nvmap_pp_alloc_inc(pp) nvmap_pp_inc_index((pp), &(pp)->alloc)
201 #define nvmap_pp_fill_inc(pp) nvmap_pp_inc_index((pp), &(pp)->fill)
203 /* Handle wrap around. */
204 static inline void nvmap_pp_inc_index(struct nvmap_page_pool *pp, u32 *ind)
208 /* Wrap condition. */
209 if (*ind >= pp->length)
213 static inline void nvmap_page_pool_lock(struct nvmap_page_pool *pool)
215 mutex_lock(&pool->lock);
218 static inline void nvmap_page_pool_unlock(struct nvmap_page_pool *pool)
220 mutex_unlock(&pool->lock);
223 int nvmap_page_pool_init(struct nvmap_device *dev);
224 int nvmap_page_pool_fini(struct nvmap_device *dev);
225 struct page *nvmap_page_pool_alloc(struct nvmap_page_pool *pool);
226 bool nvmap_page_pool_fill(struct nvmap_page_pool *pool, struct page *page);
227 int __nvmap_page_pool_alloc_lots_locked(struct nvmap_page_pool *pool,
228 struct page **pages, u32 nr);
229 int __nvmap_page_pool_fill_lots_locked(struct nvmap_page_pool *pool,
230 struct page **pages, u32 nr);
231 int nvmap_page_pool_clear(void);
232 int nvmap_page_pool_debugfs_init(struct dentry *nvmap_root);
235 struct nvmap_carveout_commit {
237 struct list_head list;
240 struct nvmap_client {
242 struct rb_root handle_refs;
243 struct mutex ref_lock;
246 struct task_struct *task;
247 struct list_head list;
250 struct nvmap_carveout_commit carveout_commit[0];
254 struct nvmap_vma_priv {
255 struct nvmap_handle *handle;
257 atomic_t count; /* number of processes cloning the VMA */
260 struct nvmap_device {
261 struct rb_root handles;
262 spinlock_t handle_lock;
263 struct miscdevice dev_user;
264 struct nvmap_carveout_node *heaps;
266 #ifdef CONFIG_NVMAP_PAGE_POOLS
267 struct nvmap_page_pool pool;
269 struct list_head clients;
270 spinlock_t clients_lock;
291 atomic64_t stats[NS_NUM];
295 extern struct nvmap_stats nvmap_stats;
296 extern struct nvmap_device *nvmap_dev;
298 void nvmap_stats_inc(enum nvmap_stats_t, size_t size);
299 void nvmap_stats_dec(enum nvmap_stats_t, size_t size);
300 u64 nvmap_stats_read(enum nvmap_stats_t);
302 static inline void nvmap_ref_lock(struct nvmap_client *priv)
304 mutex_lock(&priv->ref_lock);
307 static inline void nvmap_ref_unlock(struct nvmap_client *priv)
309 mutex_unlock(&priv->ref_lock);
313 * NOTE: this does not ensure the continued existence of the underlying
314 * dma_buf. If you want ensure the existence of the dma_buf you must get an
315 * nvmap_handle_ref as that is what tracks the dma_buf refs.
317 static inline struct nvmap_handle *nvmap_handle_get(struct nvmap_handle *h)
319 if (WARN_ON(!virt_addr_valid(h))) {
320 pr_err("%s: invalid handle\n", current->group_leader->comm);
324 if (unlikely(atomic_inc_return(&h->ref) <= 1)) {
325 pr_err("%s: %s attempt to get a freed handle\n",
326 __func__, current->group_leader->comm);
334 static inline pgprot_t nvmap_pgprot(struct nvmap_handle *h, pgprot_t prot)
336 if (h->flags == NVMAP_HANDLE_UNCACHEABLE) {
338 if (h->owner && !h->owner->warned) {
339 char task_comm[TASK_COMM_LEN];
340 h->owner->warned = 1;
341 get_task_comm(task_comm, h->owner->task);
342 pr_err("PID %d: %s: WARNING: "
343 "NVMAP_HANDLE_WRITE_COMBINE "
344 "should be used in place of "
345 "NVMAP_HANDLE_UNCACHEABLE on ARM64\n",
346 h->owner->task->pid, task_comm);
349 return pgprot_noncached(prot);
351 else if (h->flags == NVMAP_HANDLE_WRITE_COMBINE)
352 return pgprot_dmacoherent(prot);
356 struct nvmap_heap_block *nvmap_carveout_alloc(struct nvmap_client *dev,
357 struct nvmap_handle *handle,
360 unsigned long nvmap_carveout_usage(struct nvmap_client *c,
361 struct nvmap_heap_block *b);
363 struct nvmap_carveout_node;
364 void nvmap_carveout_commit_add(struct nvmap_client *client,
365 struct nvmap_carveout_node *node, size_t len);
367 void nvmap_carveout_commit_subtract(struct nvmap_client *client,
368 struct nvmap_carveout_node *node,
371 void nvmap_handle_put(struct nvmap_handle *h);
373 struct nvmap_handle_ref *__nvmap_validate_locked(struct nvmap_client *priv,
374 struct nvmap_handle *h);
376 struct nvmap_handle *nvmap_validate_get(struct nvmap_handle *h);
378 struct nvmap_handle_ref *nvmap_create_handle(struct nvmap_client *client,
381 struct nvmap_handle_ref *nvmap_duplicate_handle(struct nvmap_client *client,
382 struct nvmap_handle *h, bool skip_val);
384 struct nvmap_handle_ref *nvmap_create_handle_from_fd(
385 struct nvmap_client *client, int fd);
387 int nvmap_alloc_handle(struct nvmap_client *client,
388 struct nvmap_handle *h, unsigned int heap_mask,
389 size_t align, u8 kind,
392 void nvmap_free_handle(struct nvmap_client *c, struct nvmap_handle *h);
394 void nvmap_free_handle_user_id(struct nvmap_client *c, unsigned long user_id);
396 int nvmap_pin_ids(struct nvmap_client *client,
397 unsigned int nr, struct nvmap_handle * const *ids);
399 void nvmap_unpin_ids(struct nvmap_client *priv,
400 unsigned int nr, struct nvmap_handle * const *ids);
402 int nvmap_handle_remove(struct nvmap_device *dev, struct nvmap_handle *h);
404 void nvmap_handle_add(struct nvmap_device *dev, struct nvmap_handle *h);
406 int is_nvmap_vma(struct vm_area_struct *vma);
408 int nvmap_get_dmabuf_fd(struct nvmap_client *client, struct nvmap_handle *h);
409 struct nvmap_handle *nvmap_get_id_from_dmabuf_fd(struct nvmap_client *client,
412 int nvmap_get_handle_param(struct nvmap_client *client,
413 struct nvmap_handle_ref *ref, u32 param, u64 *result);
415 struct nvmap_client *nvmap_client_get(struct nvmap_client *client);
417 void nvmap_client_put(struct nvmap_client *c);
419 struct nvmap_handle *unmarshal_user_id(u32 id);
421 /* MM definitions. */
422 extern size_t cache_maint_inner_threshold;
423 extern size_t cache_maint_outer_threshold;
425 extern void v7_flush_kern_cache_all(void);
426 extern void v7_clean_kern_cache_all(void *);
427 extern void __flush_dcache_all(void *arg);
428 extern void __clean_dcache_all(void *arg);
430 void inner_flush_cache_all(void);
431 void inner_clean_cache_all(void);
432 void nvmap_clean_cache(struct page **pages, int numpages);
433 void nvmap_flush_cache(struct page **pages, int numpages);
435 int nvmap_do_cache_maint_list(struct nvmap_handle **handles, u32 *offsets,
436 u32 *sizes, int op, int nr);
438 /* Internal API to support dmabuf */
439 struct dma_buf *__nvmap_dmabuf_export(struct nvmap_client *client,
440 struct nvmap_handle *handle);
441 struct dma_buf *__nvmap_make_dmabuf(struct nvmap_client *client,
442 struct nvmap_handle *handle);
443 struct sg_table *__nvmap_sg_table(struct nvmap_client *client,
444 struct nvmap_handle *h);
445 void __nvmap_free_sg_table(struct nvmap_client *client,
446 struct nvmap_handle *h, struct sg_table *sgt);
447 void *__nvmap_kmap(struct nvmap_handle *h, unsigned int pagenum);
448 void __nvmap_kunmap(struct nvmap_handle *h, unsigned int pagenum, void *addr);
449 void *__nvmap_mmap(struct nvmap_handle *h);
450 void __nvmap_munmap(struct nvmap_handle *h, void *addr);
451 int __nvmap_map(struct nvmap_handle *h, struct vm_area_struct *vma);
452 int __nvmap_get_handle_param(struct nvmap_client *client,
453 struct nvmap_handle *h, u32 param, u64 *result);
454 int __nvmap_do_cache_maint(struct nvmap_client *client, struct nvmap_handle *h,
455 unsigned long start, unsigned long end,
456 unsigned int op, bool clean_only_dirty);
457 struct nvmap_client *__nvmap_create_client(struct nvmap_device *dev,
459 struct dma_buf *__nvmap_dmabuf_export_from_ref(struct nvmap_handle_ref *ref);
460 struct nvmap_handle *__nvmap_ref_to_id(struct nvmap_handle_ref *ref);
461 int __nvmap_pin(struct nvmap_handle_ref *ref, phys_addr_t *phys);
462 void __nvmap_unpin(struct nvmap_handle_ref *ref);
463 int __nvmap_dmabuf_fd(struct nvmap_client *client,
464 struct dma_buf *dmabuf, int flags);
466 void nvmap_dmabuf_debugfs_init(struct dentry *nvmap_root);
467 int nvmap_dmabuf_stash_init(void);
469 void *nvmap_altalloc(size_t len);
470 void nvmap_altfree(void *ptr, size_t len);
472 static inline struct page *nvmap_to_page(struct page *page)
474 return (struct page *)((unsigned long)page & ~3UL);
477 static inline bool nvmap_page_dirty(struct page *page)
479 return (unsigned long)page & 1UL;
482 static inline void nvmap_page_mkdirty(struct page **page)
484 *page = (struct page *)((unsigned long)*page | 1UL);
487 static inline void nvmap_page_mkclean(struct page **page)
489 *page = (struct page *)((unsigned long)*page & ~1UL);
492 static inline bool nvmap_page_reserved(struct page *page)
494 return !!((unsigned long)page & 2UL);
497 static inline void nvmap_page_mkreserved(struct page **page)
499 *page = (struct page *)((unsigned long)*page | 2UL);
502 static inline void nvmap_page_mkunreserved(struct page **page)
504 *page = (struct page *)((unsigned long)*page & ~2UL);
508 * FIXME: assume user space requests for reserve operations
511 static inline void nvmap_handle_mk(struct nvmap_handle *h,
512 u32 offset, u32 size,
513 void (*fn)(struct page **))
516 int start_page = PAGE_ALIGN(offset) >> PAGE_SHIFT;
517 int end_page = (offset + size) >> PAGE_SHIFT;
519 if (h->heap_pgalloc) {
520 for (i = start_page; i < end_page; i++)
521 fn(&h->pgalloc.pages[i]);
525 static inline void nvmap_handle_mkclean(struct nvmap_handle *h,
526 u32 offset, u32 size)
528 nvmap_handle_mk(h, offset, size, nvmap_page_mkclean);
531 static inline void nvmap_handle_mkunreserved(struct nvmap_handle *h,
532 u32 offset, u32 size)
534 nvmap_handle_mk(h, offset, size, nvmap_page_mkunreserved);
537 static inline void nvmap_handle_mkreserved(struct nvmap_handle *h,
538 u32 offset, u32 size)
540 nvmap_handle_mk(h, offset, size, nvmap_page_mkreserved);
543 static inline struct page **nvmap_pages(struct page **pg_pages, u32 nr_pages)
548 pages = nvmap_altalloc(sizeof(*pages) * nr_pages);
552 for (i = 0; i < nr_pages; i++)
553 pages[i] = nvmap_to_page(pg_pages[i]);
558 void nvmap_zap_handle(struct nvmap_handle *handle, u32 offset, u32 size);
560 void nvmap_zap_handles(struct nvmap_handle **handles, u32 *offsets,
563 void nvmap_vma_open(struct vm_area_struct *vma);
565 int nvmap_reserve_pages(struct nvmap_handle **handles, u32 *offsets,
566 u32 *sizes, u32 nr, u32 op);
568 #endif /* __VIDEO_TEGRA_NVMAP_NVMAP_H */