1 /**************************************************************************
2 *
3 * Copyright (c) 2007-2010 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 * Authors: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
29 */
30
31 #include <drm/ttm/ttm_module.h>
32 #include <drm/ttm/ttm_bo_driver.h>
33 #include <drm/ttm/ttm_placement.h>
34 #include <drm/drm_mm.h>
35 #include <linux/slab.h>
36 #include <linux/spinlock.h>
37 #include <linux/module.h>
38
39 /**
40 * Currently we use a spinlock for the lock, but a mutex *may* be
41 * more appropriate to reduce scheduling latency if the range manager
42 * ends up with very fragmented allocation patterns.
43 */
44
45 struct ttm_range_manager {
46 struct drm_mm mm;
47 spinlock_t lock;
48 };
49
ttm_bo_man_get_node(struct ttm_mem_type_manager * man,struct ttm_buffer_object * bo,const struct ttm_place * place,struct ttm_mem_reg * mem)50 static int ttm_bo_man_get_node(struct ttm_mem_type_manager *man,
51 struct ttm_buffer_object *bo,
52 const struct ttm_place *place,
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;
58 enum drm_mm_insert_mode mode;
59 unsigned long lpfn;
60 int ret;
61
62 lpfn = place->lpfn;
63 if (!lpfn)
64 lpfn = man->size;
65
66 node = kzalloc(sizeof(*node), GFP_KERNEL);
67 if (!node)
68 return -ENOMEM;
69
70 mode = DRM_MM_INSERT_BEST;
71 if (place->flags & TTM_PL_FLAG_TOPDOWN)
72 mode = DRM_MM_INSERT_HIGH;
73
74 lockmgr(&rman->lock, LK_EXCLUSIVE);
75 ret = drm_mm_insert_node_in_range(mm, node,
76 mem->num_pages,
77 mem->page_alignment, 0,
78 place->fpfn, lpfn, mode);
79 lockmgr(&rman->lock, LK_RELEASE);
80
81 if (unlikely(ret)) {
82 kfree(node);
83 } else {
84 mem->mm_node = node;
85 mem->start = node->start;
86 }
87
88 return 0;
89 }
90
ttm_bo_man_put_node(struct ttm_mem_type_manager * man,struct ttm_mem_reg * mem)91 static void ttm_bo_man_put_node(struct ttm_mem_type_manager *man,
92 struct ttm_mem_reg *mem)
93 {
94 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
95
96 if (mem->mm_node) {
97 lockmgr(&rman->lock, LK_EXCLUSIVE);
98 drm_mm_remove_node(mem->mm_node);
99 lockmgr(&rman->lock, LK_RELEASE);
100
101 kfree(mem->mm_node);
102 mem->mm_node = NULL;
103 }
104 }
105
ttm_bo_man_init(struct ttm_mem_type_manager * man,unsigned long p_size)106 static int ttm_bo_man_init(struct ttm_mem_type_manager *man,
107 unsigned long p_size)
108 {
109 struct ttm_range_manager *rman;
110
111 rman = kzalloc(sizeof(*rman), GFP_KERNEL);
112 if (!rman)
113 return -ENOMEM;
114
115 drm_mm_init(&rman->mm, 0, p_size);
116 lockinit(&rman->lock, "ttmrman", 0, 0);
117 man->priv = rman;
118 return 0;
119 }
120
ttm_bo_man_takedown(struct ttm_mem_type_manager * man)121 static int ttm_bo_man_takedown(struct ttm_mem_type_manager *man)
122 {
123 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
124 struct drm_mm *mm = &rman->mm;
125
126 lockmgr(&rman->lock, LK_EXCLUSIVE);
127 if (drm_mm_clean(mm)) {
128 drm_mm_takedown(mm);
129 lockmgr(&rman->lock, LK_RELEASE);
130 kfree(rman);
131 man->priv = NULL;
132 return 0;
133 }
134 lockmgr(&rman->lock, LK_RELEASE);
135 return -EBUSY;
136 }
137
ttm_bo_man_debug(struct ttm_mem_type_manager * man,struct drm_printer * printer)138 static void ttm_bo_man_debug(struct ttm_mem_type_manager *man,
139 struct drm_printer *printer)
140 {
141 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
142
143 lockmgr(&rman->lock, LK_EXCLUSIVE);
144 drm_mm_print(&rman->mm, printer);
145 lockmgr(&rman->lock, LK_RELEASE);
146 }
147
148 const struct ttm_mem_type_manager_func ttm_bo_manager_func = {
149 .init = ttm_bo_man_init,
150 .takedown = ttm_bo_man_takedown,
151 .get_node = ttm_bo_man_get_node,
152 .put_node = ttm_bo_man_put_node,
153 .debug = ttm_bo_man_debug
154 };
155 EXPORT_SYMBOL(ttm_bo_manager_func);
156