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 struct lock lock; 48 }; 49 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 = NULL; 58 enum drm_mm_search_flags sflags = DRM_MM_SEARCH_BEST; 59 enum drm_mm_allocator_flags aflags = DRM_MM_CREATE_DEFAULT; 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 if (place->flags & TTM_PL_FLAG_TOPDOWN) { 72 sflags = DRM_MM_SEARCH_BELOW; 73 aflags = DRM_MM_CREATE_TOP; 74 } 75 76 lockmgr(&rman->lock, LK_EXCLUSIVE); 77 ret = drm_mm_insert_node_in_range_generic(mm, node, mem->num_pages, 78 mem->page_alignment, 0, 79 place->fpfn, lpfn, 80 sflags, aflags); 81 lockmgr(&rman->lock, LK_RELEASE); 82 83 if (unlikely(ret)) { 84 kfree(node); 85 } else { 86 mem->mm_node = node; 87 mem->start = node->start; 88 } 89 90 return 0; 91 } 92 93 static void ttm_bo_man_put_node(struct ttm_mem_type_manager *man, 94 struct ttm_mem_reg *mem) 95 { 96 struct ttm_range_manager *rman = (struct ttm_range_manager *) man->priv; 97 98 if (mem->mm_node) { 99 lockmgr(&rman->lock, LK_EXCLUSIVE); 100 drm_mm_remove_node(mem->mm_node); 101 lockmgr(&rman->lock, LK_RELEASE); 102 103 kfree(mem->mm_node); 104 mem->mm_node = NULL; 105 } 106 } 107 108 static int ttm_bo_man_init(struct ttm_mem_type_manager *man, 109 unsigned long p_size) 110 { 111 struct ttm_range_manager *rman; 112 113 rman = kzalloc(sizeof(*rman), GFP_KERNEL); 114 if (!rman) 115 return -ENOMEM; 116 117 drm_mm_init(&rman->mm, 0, p_size); 118 lockinit(&rman->lock, "ttmrman", 0, LK_CANRECURSE); 119 man->priv = rman; 120 return 0; 121 } 122 123 static 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 lockmgr(&rman->lock, LK_EXCLUSIVE); 129 if (drm_mm_clean(mm)) { 130 drm_mm_takedown(mm); 131 lockmgr(&rman->lock, LK_RELEASE); 132 kfree(rman); 133 man->priv = NULL; 134 return 0; 135 } 136 lockmgr(&rman->lock, LK_RELEASE); 137 return -EBUSY; 138 } 139 140 static 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 lockmgr(&rman->lock, LK_EXCLUSIVE); 146 drm_mm_debug_table(&rman->mm, prefix); 147 lockmgr(&rman->lock, LK_RELEASE); 148 } 149 150 const 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 }; 157 EXPORT_SYMBOL(ttm_bo_manager_func); 158