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
 46struct ttm_range_manager {
 47	struct drm_mm mm;
 48	spinlock_t lock;
 49};
 50
 51static 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
 92static 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
107static 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
122static 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
139static 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
149const 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};
156EXPORT_SYMBOL(ttm_bo_manager_func);