1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31#include "ttm/ttm_module.h"
32#include "ttm/ttm_bo_driver.h"
33#include "ttm/ttm_placement.h"
34#include "drm_mm.h"
35#include <linux/slab.h>
36#include <linux/spinlock.h>
37#include <linux/module.h>
38
39
40
41
42
43
44
45struct ttm_range_manager {
46 struct drm_mm mm;
47 spinlock_t lock;
48};
49
50static int ttm_bo_man_get_node(struct ttm_mem_type_manager *man,
51 struct ttm_buffer_object *bo,
52 struct ttm_placement *placement,
53 struct ttm_mem_reg *mem)
54{
55 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
56 struct drm_mm *mm = &rman->mm;
57 struct drm_mm_node *node = NULL;
58 unsigned long lpfn;
59 int ret;
60
61 lpfn = placement->lpfn;
62 if (!lpfn)
63 lpfn = man->size;
64 do {
65 ret = drm_mm_pre_get(mm);
66 if (unlikely(ret))
67 return ret;
68
69 spin_lock(&rman->lock);
70 node = drm_mm_search_free_in_range(mm,
71 mem->num_pages, mem->page_alignment,
72 placement->fpfn, lpfn, 1);
73 if (unlikely(node == NULL)) {
74 spin_unlock(&rman->lock);
75 return 0;
76 }
77 node = drm_mm_get_block_atomic_range(node, mem->num_pages,
78 mem->page_alignment,
79 placement->fpfn,
80 lpfn);
81 spin_unlock(&rman->lock);
82 } while (node == NULL);
83
84 mem->mm_node = node;
85 mem->start = node->start;
86 return 0;
87}
88
89static void ttm_bo_man_put_node(struct ttm_mem_type_manager *man,
90 struct ttm_mem_reg *mem)
91{
92 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
93
94 if (mem->mm_node) {
95 spin_lock(&rman->lock);
96 drm_mm_put_block(mem->mm_node);
97 spin_unlock(&rman->lock);
98 mem->mm_node = NULL;
99 }
100}
101
102static int ttm_bo_man_init(struct ttm_mem_type_manager *man,
103 unsigned long p_size)
104{
105 struct ttm_range_manager *rman;
106 int ret;
107
108 rman = kzalloc(sizeof(*rman), GFP_KERNEL);
109 if (!rman)
110 return -ENOMEM;
111
112 ret = drm_mm_init(&rman->mm, 0, p_size);
113 if (ret) {
114 kfree(rman);
115 return ret;
116 }
117
118 spin_lock_init(&rman->lock);
119 man->priv = rman;
120 return 0;
121}
122
123static int ttm_bo_man_takedown(struct ttm_mem_type_manager *man)
124{
125 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
126 struct drm_mm *mm = &rman->mm;
127
128 spin_lock(&rman->lock);
129 if (drm_mm_clean(mm)) {
130 drm_mm_takedown(mm);
131 spin_unlock(&rman->lock);
132 kfree(rman);
133 man->priv = NULL;
134 return 0;
135 }
136 spin_unlock(&rman->lock);
137 return -EBUSY;
138}
139
140static void ttm_bo_man_debug(struct ttm_mem_type_manager *man,
141 const char *prefix)
142{
143 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
144
145 spin_lock(&rman->lock);
146 drm_mm_debug_table(&rman->mm, prefix);
147 spin_unlock(&rman->lock);
148}
149
150const struct ttm_mem_type_manager_func ttm_bo_manager_func = {
151 ttm_bo_man_init,
152 ttm_bo_man_takedown,
153 ttm_bo_man_get_node,
154 ttm_bo_man_put_node,
155 ttm_bo_man_debug
156};
157EXPORT_SYMBOL(ttm_bo_manager_func);
158