1 /**************************************************************************
2 *
3 * Copyright (c) 2006-2009 VMware, Inc., Palo Alto, CA., USA
4 * All Rights Reserved.
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
16 * of the Software.
17 *
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
21 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
22 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
23 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
24 * USE OR OTHER DEALINGS IN THE SOFTWARE.
25 *
26 **************************************************************************/
27
28 #ifndef TTM_MEMORY_H
29 #define TTM_MEMORY_H
30
31 #include <linux/workqueue.h>
32 #include <linux/spinlock.h>
33 #include <linux/bug.h>
34 #include <linux/wait.h>
35 #include <linux/errno.h>
36 #include <linux/kobject.h>
37 #include <linux/mm.h>
38
39 /**
40 * struct ttm_mem_shrink - callback to shrink TTM memory usage.
41 *
42 * @do_shrink: The callback function.
43 *
44 * Arguments to the do_shrink functions are intended to be passed using
45 * inheritance. That is, the argument class derives from struct ttm_mem_shrink,
46 * and can be accessed using container_of().
47 */
48
49 struct ttm_mem_shrink {
50 int (*do_shrink) (struct ttm_mem_shrink *);
51 };
52
53 /**
54 * struct ttm_mem_global - Global memory accounting structure.
55 *
56 * @shrink: A single callback to shrink TTM memory usage. Extend this
57 * to a linked list to be able to handle multiple callbacks when needed.
58 * @swap_queue: A workqueue to handle shrinking in low memory situations. We
59 * need a separate workqueue since it will spend a lot of time waiting
60 * for the GPU, and this will otherwise block other workqueue tasks(?)
61 * At this point we use only a single-threaded workqueue.
62 * @work: The workqueue callback for the shrink queue.
63 * @lock: Lock to protect the @shrink - and the memory accounting members,
64 * that is, essentially the whole structure with some exceptions.
65 * @zones: Array of pointers to accounting zones.
66 * @num_zones: Number of populated entries in the @zones array.
67 * @zone_kernel: Pointer to the kernel zone.
68 * @zone_highmem: Pointer to the highmem zone if there is one.
69 * @zone_dma32: Pointer to the dma32 zone if there is one.
70 *
71 * Note that this structure is not per device. It should be global for all
72 * graphics devices.
73 */
74
75 #define TTM_MEM_MAX_ZONES 2
76 struct ttm_mem_zone;
77 struct ttm_mem_global {
78 struct kobject kobj;
79 struct ttm_mem_shrink *shrink;
80 struct workqueue_struct *swap_queue;
81 struct work_struct work;
82 spinlock_t lock;
83 struct ttm_mem_zone *zones[TTM_MEM_MAX_ZONES];
84 unsigned int num_zones;
85 struct ttm_mem_zone *zone_kernel;
86 #ifdef CONFIG_HIGHMEM
87 struct ttm_mem_zone *zone_highmem;
88 #else
89 struct ttm_mem_zone *zone_dma32;
90 #endif
91 };
92
93 /**
94 * ttm_mem_init_shrink - initialize a struct ttm_mem_shrink object
95 *
96 * @shrink: The object to initialize.
97 * @func: The callback function.
98 */
99
ttm_mem_init_shrink(struct ttm_mem_shrink * shrink,int (* func)(struct ttm_mem_shrink *))100 static inline void ttm_mem_init_shrink(struct ttm_mem_shrink *shrink,
101 int (*func) (struct ttm_mem_shrink *))
102 {
103 shrink->do_shrink = func;
104 }
105
106 /**
107 * ttm_mem_register_shrink - register a struct ttm_mem_shrink object.
108 *
109 * @glob: The struct ttm_mem_global object to register with.
110 * @shrink: An initialized struct ttm_mem_shrink object to register.
111 *
112 * Returns:
113 * -EBUSY: There's already a callback registered. (May change).
114 */
115
ttm_mem_register_shrink(struct ttm_mem_global * glob,struct ttm_mem_shrink * shrink)116 static inline int ttm_mem_register_shrink(struct ttm_mem_global *glob,
117 struct ttm_mem_shrink *shrink)
118 {
119 spin_lock(&glob->lock);
120 if (glob->shrink != NULL) {
121 spin_unlock(&glob->lock);
122 return -EBUSY;
123 }
124 glob->shrink = shrink;
125 spin_unlock(&glob->lock);
126 return 0;
127 }
128
129 /**
130 * ttm_mem_unregister_shrink - unregister a struct ttm_mem_shrink object.
131 *
132 * @glob: The struct ttm_mem_global object to unregister from.
133 * @shrink: A previously registert struct ttm_mem_shrink object.
134 *
135 */
136
ttm_mem_unregister_shrink(struct ttm_mem_global * glob,struct ttm_mem_shrink * shrink)137 static inline void ttm_mem_unregister_shrink(struct ttm_mem_global *glob,
138 struct ttm_mem_shrink *shrink)
139 {
140 spin_lock(&glob->lock);
141 BUG_ON(glob->shrink != shrink);
142 glob->shrink = NULL;
143 spin_unlock(&glob->lock);
144 }
145
146 extern int ttm_mem_global_init(struct ttm_mem_global *glob);
147 extern void ttm_mem_global_release(struct ttm_mem_global *glob);
148 extern int ttm_mem_global_alloc(struct ttm_mem_global *glob, uint64_t memory,
149 bool no_wait, bool interruptible);
150 extern void ttm_mem_global_free(struct ttm_mem_global *glob,
151 uint64_t amount);
152 extern int ttm_mem_global_alloc_page(struct ttm_mem_global *glob,
153 struct page *page,
154 bool no_wait, bool interruptible);
155 extern void ttm_mem_global_free_page(struct ttm_mem_global *glob,
156 struct page *page);
157 extern size_t ttm_round_pot(size_t size);
158 #endif
159