1 /* SPDX-License-Identifier: GPL-2.0 OR MIT */
2 /**************************************************************************
3 *
4 * Copyright (c) 2007-2010 VMware, Inc., Palo Alto, CA., USA
5 * All Rights Reserved.
6 *
7 * Permission is hereby granted, free of charge, to any person obtaining a
8 * copy of this software and associated documentation files (the
9 * "Software"), to deal in the Software without restriction, including
10 * without limitation the rights to use, copy, modify, merge, publish,
11 * distribute, sub license, and/or sell copies of the Software, and to
12 * permit persons to whom the Software is furnished to do so, subject to
13 * the following conditions:
14 *
15 * The above copyright notice and this permission notice (including the
16 * next paragraph) shall be included in all copies or substantial portions
17 * of the Software.
18 *
19 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
20 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
21 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
22 * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
23 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
24 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
25 * USE OR OTHER DEALINGS IN THE SOFTWARE.
26 *
27 **************************************************************************/
28 /*
29 * Authors: Thomas Hellstrom <thellstrom-at-vmware-dot-com>
30 */
31
32 #include <drm/ttm/ttm_module.h>
33 #include <drm/ttm/ttm_bo_driver.h>
34 #include <drm/ttm/ttm_placement.h>
35 #include <drm/drm_mm.h>
36 #include <linux/slab.h>
37 #include <linux/spinlock.h>
38 #include <linux/module.h>
39
40 /**
41 * Currently we use a spinlock for the lock, but a mutex *may* be
42 * more appropriate to reduce scheduling latency if the range manager
43 * ends up with very fragmented allocation patterns.
44 */
45
46 struct ttm_range_manager {
47 struct drm_mm mm;
48 spinlock_t lock;
49 };
50
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)51 static int ttm_bo_man_get_node(struct ttm_mem_type_manager *man,
52 struct ttm_buffer_object *bo,
53 const struct ttm_place *place,
54 struct ttm_mem_reg *mem)
55 {
56 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
57 struct drm_mm *mm = &rman->mm;
58 struct drm_mm_node *node;
59 enum drm_mm_insert_mode mode;
60 unsigned long lpfn;
61 int ret;
62
63 lpfn = place->lpfn;
64 if (!lpfn)
65 lpfn = man->size;
66
67 node = kzalloc(sizeof(*node), GFP_KERNEL);
68 if (!node)
69 return -ENOMEM;
70
71 mode = DRM_MM_INSERT_BEST;
72 if (place->flags & TTM_PL_FLAG_TOPDOWN)
73 mode = DRM_MM_INSERT_HIGH;
74
75 spin_lock(&rman->lock);
76 ret = drm_mm_insert_node_in_range(mm, node,
77 mem->num_pages,
78 mem->page_alignment, 0,
79 place->fpfn, lpfn, mode);
80 spin_unlock(&rman->lock);
81
82 if (unlikely(ret)) {
83 kfree(node);
84 } else {
85 mem->mm_node = node;
86 mem->start = node->start;
87 }
88
89 return 0;
90 }
91
ttm_bo_man_put_node(struct ttm_mem_type_manager * man,struct ttm_mem_reg * mem)92 static void ttm_bo_man_put_node(struct ttm_mem_type_manager *man,
93 struct ttm_mem_reg *mem)
94 {
95 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
96
97 if (mem->mm_node) {
98 spin_lock(&rman->lock);
99 drm_mm_remove_node(mem->mm_node);
100 spin_unlock(&rman->lock);
101
102 kfree(mem->mm_node);
103 mem->mm_node = NULL;
104 }
105 }
106
ttm_bo_man_init(struct ttm_mem_type_manager * man,unsigned long p_size)107 static int ttm_bo_man_init(struct ttm_mem_type_manager *man,
108 unsigned long p_size)
109 {
110 struct ttm_range_manager *rman;
111
112 rman = kzalloc(sizeof(*rman), GFP_KERNEL);
113 if (!rman)
114 return -ENOMEM;
115
116 drm_mm_init(&rman->mm, 0, p_size);
117 spin_lock_init(&rman->lock);
118 man->priv = rman;
119 return 0;
120 }
121
ttm_bo_man_takedown(struct ttm_mem_type_manager * man)122 static int ttm_bo_man_takedown(struct ttm_mem_type_manager *man)
123 {
124 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
125 struct drm_mm *mm = &rman->mm;
126
127 spin_lock(&rman->lock);
128 if (drm_mm_clean(mm)) {
129 drm_mm_takedown(mm);
130 spin_unlock(&rman->lock);
131 kfree(rman);
132 man->priv = NULL;
133 return 0;
134 }
135 spin_unlock(&rman->lock);
136 return -EBUSY;
137 }
138
ttm_bo_man_debug(struct ttm_mem_type_manager * man,struct drm_printer * printer)139 static void ttm_bo_man_debug(struct ttm_mem_type_manager *man,
140 struct drm_printer *printer)
141 {
142 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv;
143
144 spin_lock(&rman->lock);
145 drm_mm_print(&rman->mm, printer);
146 spin_unlock(&rman->lock);
147 }
148
149 const struct ttm_mem_type_manager_func ttm_bo_manager_func = {
150 .init = ttm_bo_man_init,
151 .takedown = ttm_bo_man_takedown,
152 .get_node = ttm_bo_man_get_node,
153 .put_node = ttm_bo_man_put_node,
154 .debug = ttm_bo_man_debug
155 };
156 EXPORT_SYMBOL(ttm_bo_manager_func);
157
