353 lines
8.6 KiB
C
353 lines
8.6 KiB
C
|
/**************************************************************************
|
||
|
|
*
|
||
|
|
* Copyright 2006 Tungsten Graphics, Inc., Bismarck., ND., USA.
|
||
|
|
* All Rights Reserved.
|
||
|
|
*
|
||
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
||
|
|
* copy of this software and associated documentation files (the
|
||
|
|
* "Software"), to deal in the Software without restriction, including
|
||
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
||
|
|
* distribute, sub license, and/or sell copies of the Software, and to
|
||
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
||
|
|
* the following conditions:
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
|
||
|
|
* THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
|
||
|
|
* DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
|
||
|
|
* OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
||
|
|
* USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
|
*
|
||
|
|
* The above copyright notice and this permission notice (including the
|
||
|
|
* next paragraph) shall be included in all copies or substantial portions
|
||
|
|
* of the Software.
|
||
|
|
*
|
||
|
|
*
|
||
|
|
**************************************************************************/
|
||
|
|
/*
|
||
|
|
* Simple memory manager interface that keeps track on allocate regions on a
|
||
|
|
* per "owner" basis. All regions associated with an "owner" can be released
|
||
|
|
* with a simple call. Typically if the "owner" exists. The owner is any
|
||
|
|
* "unsigned long" identifier. Can typically be a pointer to a file private
|
||
|
|
* struct or a context identifier.
|
||
|
|
*
|
||
|
|
* Authors:
|
||
|
|
* Thomas Hellström <thomas-at-tungstengraphics-dot-com>
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "drm_sman.h"
|
||
|
|
|
||
|
|
struct drm_owner_item {
|
||
|
|
struct drm_hash_item owner_hash;
|
||
|
|
struct list_head sman_list;
|
||
|
|
struct list_head mem_blocks;
|
||
|
|
};
|
||
|
|
|
||
|
|
void drm_sman_takedown(struct drm_sman * sman)
|
||
|
|
{
|
||
|
|
drm_ht_remove(&sman->user_hash_tab);
|
||
|
|
drm_ht_remove(&sman->owner_hash_tab);
|
||
|
|
kfree(sman->mm);
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_takedown);
|
||
|
|
|
||
|
|
int
|
||
|
|
drm_sman_init(struct drm_sman * sman, unsigned int num_managers,
|
||
|
|
unsigned int user_order, unsigned int owner_order)
|
||
|
|
{
|
||
|
|
int ret = 0;
|
||
|
|
|
||
|
|
sman->mm = (struct drm_sman_mm *) kcalloc(num_managers,
|
||
|
|
sizeof(*sman->mm),
|
||
|
|
GFP_KERNEL);
|
||
|
|
if (!sman->mm) {
|
||
|
|
ret = -ENOMEM;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
sman->num_managers = num_managers;
|
||
|
|
INIT_LIST_HEAD(&sman->owner_items);
|
||
|
|
ret = drm_ht_create(&sman->owner_hash_tab, owner_order);
|
||
|
|
if (ret)
|
||
|
|
goto out1;
|
||
|
|
ret = drm_ht_create(&sman->user_hash_tab, user_order);
|
||
|
|
if (!ret)
|
||
|
|
goto out;
|
||
|
|
|
||
|
|
drm_ht_remove(&sman->owner_hash_tab);
|
||
|
|
out1:
|
||
|
|
kfree(sman->mm);
|
||
|
|
out:
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_init);
|
||
|
|
|
||
|
|
static void *drm_sman_mm_allocate(void *private, unsigned long size,
|
||
|
|
unsigned alignment)
|
||
|
|
{
|
||
|
|
struct drm_mm *mm = (struct drm_mm *) private;
|
||
|
|
struct drm_mm_node *tmp;
|
||
|
|
|
||
|
|
tmp = drm_mm_search_free(mm, size, alignment, 1);
|
||
|
|
if (!tmp) {
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
tmp = drm_mm_get_block(tmp, size, alignment);
|
||
|
|
return tmp;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void drm_sman_mm_free(void *private, void *ref)
|
||
|
|
{
|
||
|
|
struct drm_mm_node *node = (struct drm_mm_node *) ref;
|
||
|
|
|
||
|
|
drm_mm_put_block(node);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void drm_sman_mm_destroy(void *private)
|
||
|
|
{
|
||
|
|
struct drm_mm *mm = (struct drm_mm *) private;
|
||
|
|
drm_mm_takedown(mm);
|
||
|
|
kfree(mm);
|
||
|
|
}
|
||
|
|
|
||
|
|
static unsigned long drm_sman_mm_offset(void *private, void *ref)
|
||
|
|
{
|
||
|
|
struct drm_mm_node *node = (struct drm_mm_node *) ref;
|
||
|
|
return node->start;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
drm_sman_set_range(struct drm_sman * sman, unsigned int manager,
|
||
|
|
unsigned long start, unsigned long size)
|
||
|
|
{
|
||
|
|
struct drm_sman_mm *sman_mm;
|
||
|
|
struct drm_mm *mm;
|
||
|
|
int ret;
|
||
|
|
|
||
|
|
BUG_ON(manager >= sman->num_managers);
|
||
|
|
|
||
|
|
sman_mm = &sman->mm[manager];
|
||
|
|
mm = kzalloc(sizeof(*mm), GFP_KERNEL);
|
||
|
|
if (!mm) {
|
||
|
|
return -ENOMEM;
|
||
|
|
}
|
||
|
|
sman_mm->private = mm;
|
||
|
|
ret = drm_mm_init(mm, start, size);
|
||
|
|
|
||
|
|
if (ret) {
|
||
|
|
kfree(mm);
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
sman_mm->allocate = drm_sman_mm_allocate;
|
||
|
|
sman_mm->free = drm_sman_mm_free;
|
||
|
|
sman_mm->destroy = drm_sman_mm_destroy;
|
||
|
|
sman_mm->offset = drm_sman_mm_offset;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_set_range);
|
||
|
|
|
||
|
|
int
|
||
|
|
drm_sman_set_manager(struct drm_sman * sman, unsigned int manager,
|
||
|
|
struct drm_sman_mm * allocator)
|
||
|
|
{
|
||
|
|
BUG_ON(manager >= sman->num_managers);
|
||
|
|
sman->mm[manager] = *allocator;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
EXPORT_SYMBOL(drm_sman_set_manager);
|
||
|
|
|
||
|
|
static struct drm_owner_item *drm_sman_get_owner_item(struct drm_sman * sman,
|
||
|
|
unsigned long owner)
|
||
|
|
{
|
||
|
|
int ret;
|
||
|
|
struct drm_hash_item *owner_hash_item;
|
||
|
|
struct drm_owner_item *owner_item;
|
||
|
|
|
||
|
|
ret = drm_ht_find_item(&sman->owner_hash_tab, owner, &owner_hash_item);
|
||
|
|
if (!ret) {
|
||
|
|
return drm_hash_entry(owner_hash_item, struct drm_owner_item,
|
||
|
|
owner_hash);
|
||
|
|
}
|
||
|
|
|
||
|
|
owner_item = kzalloc(sizeof(*owner_item), GFP_KERNEL);
|
||
|
|
if (!owner_item)
|
||
|
|
goto out;
|
||
|
|
|
||
|
|
INIT_LIST_HEAD(&owner_item->mem_blocks);
|
||
|
|
owner_item->owner_hash.key = owner;
|
||
|
|
if (drm_ht_insert_item(&sman->owner_hash_tab, &owner_item->owner_hash))
|
||
|
|
goto out1;
|
||
|
|
|
||
|
|
list_add_tail(&owner_item->sman_list, &sman->owner_items);
|
||
|
|
return owner_item;
|
||
|
|
|
||
|
|
out1:
|
||
|
|
kfree(owner_item);
|
||
|
|
out:
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct drm_memblock_item *drm_sman_alloc(struct drm_sman *sman, unsigned int manager,
|
||
|
|
unsigned long size, unsigned alignment,
|
||
|
|
unsigned long owner)
|
||
|
|
{
|
||
|
|
void *tmp;
|
||
|
|
struct drm_sman_mm *sman_mm;
|
||
|
|
struct drm_owner_item *owner_item;
|
||
|
|
struct drm_memblock_item *memblock;
|
||
|
|
|
||
|
|
BUG_ON(manager >= sman->num_managers);
|
||
|
|
|
||
|
|
sman_mm = &sman->mm[manager];
|
||
|
|
tmp = sman_mm->allocate(sman_mm->private, size, alignment);
|
||
|
|
|
||
|
|
if (!tmp) {
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
memblock = kzalloc(sizeof(*memblock), GFP_KERNEL);
|
||
|
|
|
||
|
|
if (!memblock)
|
||
|
|
goto out;
|
||
|
|
|
||
|
|
memblock->mm_info = tmp;
|
||
|
|
memblock->mm = sman_mm;
|
||
|
|
memblock->sman = sman;
|
||
|
|
|
||
|
|
if (drm_ht_just_insert_please
|
||
|
|
(&sman->user_hash_tab, &memblock->user_hash,
|
||
|
|
(unsigned long)memblock, 32, 0, 0))
|
||
|
|
goto out1;
|
||
|
|
|
||
|
|
owner_item = drm_sman_get_owner_item(sman, owner);
|
||
|
|
if (!owner_item)
|
||
|
|
goto out2;
|
||
|
|
|
||
|
|
list_add_tail(&memblock->owner_list, &owner_item->mem_blocks);
|
||
|
|
|
||
|
|
return memblock;
|
||
|
|
|
||
|
|
out2:
|
||
|
|
drm_ht_remove_item(&sman->user_hash_tab, &memblock->user_hash);
|
||
|
|
out1:
|
||
|
|
kfree(memblock);
|
||
|
|
out:
|
||
|
|
sman_mm->free(sman_mm->private, tmp);
|
||
|
|
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_alloc);
|
||
|
|
|
||
|
|
static void drm_sman_free(struct drm_memblock_item *item)
|
||
|
|
{
|
||
|
|
struct drm_sman *sman = item->sman;
|
||
|
|
|
||
|
|
list_del(&item->owner_list);
|
||
|
|
drm_ht_remove_item(&sman->user_hash_tab, &item->user_hash);
|
||
|
|
item->mm->free(item->mm->private, item->mm_info);
|
||
|
|
kfree(item);
|
||
|
|
}
|
||
|
|
|
||
|
|
int drm_sman_free_key(struct drm_sman *sman, unsigned int key)
|
||
|
|
{
|
||
|
|
struct drm_hash_item *hash_item;
|
||
|
|
struct drm_memblock_item *memblock_item;
|
||
|
|
|
||
|
|
if (drm_ht_find_item(&sman->user_hash_tab, key, &hash_item))
|
||
|
|
return -EINVAL;
|
||
|
|
|
||
|
|
memblock_item = drm_hash_entry(hash_item, struct drm_memblock_item,
|
||
|
|
user_hash);
|
||
|
|
drm_sman_free(memblock_item);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_free_key);
|
||
|
|
|
||
|
|
static void drm_sman_remove_owner(struct drm_sman *sman,
|
||
|
|
struct drm_owner_item *owner_item)
|
||
|
|
{
|
||
|
|
list_del(&owner_item->sman_list);
|
||
|
|
drm_ht_remove_item(&sman->owner_hash_tab, &owner_item->owner_hash);
|
||
|
|
kfree(owner_item);
|
||
|
|
}
|
||
|
|
|
||
|
|
int drm_sman_owner_clean(struct drm_sman *sman, unsigned long owner)
|
||
|
|
{
|
||
|
|
|
||
|
|
struct drm_hash_item *hash_item;
|
||
|
|
struct drm_owner_item *owner_item;
|
||
|
|
|
||
|
|
if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
owner_item = drm_hash_entry(hash_item, struct drm_owner_item, owner_hash);
|
||
|
|
if (owner_item->mem_blocks.next == &owner_item->mem_blocks) {
|
||
|
|
drm_sman_remove_owner(sman, owner_item);
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_owner_clean);
|
||
|
|
|
||
|
|
static void drm_sman_do_owner_cleanup(struct drm_sman *sman,
|
||
|
|
struct drm_owner_item *owner_item)
|
||
|
|
{
|
||
|
|
struct drm_memblock_item *entry, *next;
|
||
|
|
|
||
|
|
list_for_each_entry_safe(entry, next, &owner_item->mem_blocks,
|
||
|
|
owner_list) {
|
||
|
|
drm_sman_free(entry);
|
||
|
|
}
|
||
|
|
drm_sman_remove_owner(sman, owner_item);
|
||
|
|
}
|
||
|
|
|
||
|
|
void drm_sman_owner_cleanup(struct drm_sman *sman, unsigned long owner)
|
||
|
|
{
|
||
|
|
|
||
|
|
struct drm_hash_item *hash_item;
|
||
|
|
struct drm_owner_item *owner_item;
|
||
|
|
|
||
|
|
if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
owner_item = drm_hash_entry(hash_item, struct drm_owner_item, owner_hash);
|
||
|
|
drm_sman_do_owner_cleanup(sman, owner_item);
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_owner_cleanup);
|
||
|
|
|
||
|
|
void drm_sman_cleanup(struct drm_sman *sman)
|
||
|
|
{
|
||
|
|
struct drm_owner_item *entry, *next;
|
||
|
|
unsigned int i;
|
||
|
|
struct drm_sman_mm *sman_mm;
|
||
|
|
|
||
|
|
list_for_each_entry_safe(entry, next, &sman->owner_items, sman_list) {
|
||
|
|
drm_sman_do_owner_cleanup(sman, entry);
|
||
|
|
}
|
||
|
|
if (sman->mm) {
|
||
|
|
for (i = 0; i < sman->num_managers; ++i) {
|
||
|
|
sman_mm = &sman->mm[i];
|
||
|
|
if (sman_mm->private) {
|
||
|
|
sman_mm->destroy(sman_mm->private);
|
||
|
|
sman_mm->private = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
EXPORT_SYMBOL(drm_sman_cleanup);
|