drivers/char/drm/drm_sman.c - maze/linux - Git at Google

 /**************************************************************************
  *
  * 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"

 typedef struct drm_owner_item {
 	drm_hash_item_t owner_hash;
 	struct list_head sman_list;
 	struct list_head mem_blocks;
 } drm_owner_item_t;

 void drm_sman_takedown(drm_sman_t * sman)
 {
 	drm_ht_remove(&sman->user_hash_tab);
 	drm_ht_remove(&sman->owner_hash_tab);
 	if (sman->mm)
 		drm_free(sman->mm, sman->num_managers * sizeof(*sman->mm),
 			 DRM_MEM_MM);
 }

 EXPORT_SYMBOL(drm_sman_takedown);

 int
 drm_sman_init(drm_sman_t * sman, unsigned int num_managers,
 	      unsigned int user_order, unsigned int owner_order)
 {
 	int ret = 0;

 	sman->mm = (drm_sman_mm_t *) drm_calloc(num_managers, sizeof(*sman->mm),
 						DRM_MEM_MM);
 	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:
 	drm_free(sman->mm, num_managers * sizeof(*sman->mm), DRM_MEM_MM);
 out:
 	return ret;
 }

 EXPORT_SYMBOL(drm_sman_init);

 static void *drm_sman_mm_allocate(void *private, unsigned long size,
 				  unsigned alignment)
 {
 	drm_mm_t *mm = (drm_mm_t *) private;
 	drm_mm_node_t *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)
 {
 	drm_mm_t *mm = (drm_mm_t *) private;
 	drm_mm_node_t *node = (drm_mm_node_t *) ref;

 	drm_mm_put_block(mm, node);
 }

 static void drm_sman_mm_destroy(void *private)
 {
 	drm_mm_t *mm = (drm_mm_t *) private;
 	drm_mm_takedown(mm);
 	drm_free(mm, sizeof(*mm), DRM_MEM_MM);
 }

 static unsigned long drm_sman_mm_offset(void *private, void *ref)
 {
 	drm_mm_node_t *node = (drm_mm_node_t *) ref;
 	return node->start;
 }

 int
 drm_sman_set_range(drm_sman_t * sman, unsigned int manager,
 		   unsigned long start, unsigned long size)
 {
 	drm_sman_mm_t *sman_mm;
 	drm_mm_t *mm;
 	int ret;

 	BUG_ON(manager >= sman->num_managers);

 	sman_mm = &sman->mm[manager];
 	mm = drm_calloc(1, sizeof(*mm), DRM_MEM_MM);
 	if (!mm) {
 		return -ENOMEM;
 	}
 	sman_mm->private = mm;
 	ret = drm_mm_init(mm, start, size);

 	if (ret) {
 		drm_free(mm, sizeof(*mm), DRM_MEM_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(drm_sman_t * sman, unsigned int manager,
 		     drm_sman_mm_t * allocator)
 {
 	BUG_ON(manager >= sman->num_managers);
 	sman->mm[manager] = *allocator;

 	return 0;
 }

 static drm_owner_item_t *drm_sman_get_owner_item(drm_sman_t * sman,
 						 unsigned long owner)
 {
 	int ret;
 	drm_hash_item_t *owner_hash_item;
 	drm_owner_item_t *owner_item;

 	ret = drm_ht_find_item(&sman->owner_hash_tab, owner, &owner_hash_item);
 	if (!ret) {
 		return drm_hash_entry(owner_hash_item, drm_owner_item_t,
 				      owner_hash);
 	}

 	owner_item = drm_calloc(1, sizeof(*owner_item), DRM_MEM_MM);
 	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:
 	drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
 out:
 	return NULL;
 }

 drm_memblock_item_t *drm_sman_alloc(drm_sman_t *sman, unsigned int manager,
 				    unsigned long size, unsigned alignment,
 				    unsigned long owner)
 {
 	void *tmp;
 	drm_sman_mm_t *sman_mm;
 	drm_owner_item_t *owner_item;
 	drm_memblock_item_t *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 = drm_calloc(1, sizeof(*memblock), DRM_MEM_MM);

 	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:
 	drm_free(memblock, sizeof(*memblock), DRM_MEM_MM);
 out:
 	sman_mm->free(sman_mm->private, tmp);

 	return NULL;
 }

 EXPORT_SYMBOL(drm_sman_alloc);

 static void drm_sman_free(drm_memblock_item_t *item)
 {
 	drm_sman_t *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);
 	drm_free(item, sizeof(*item), DRM_MEM_MM);
 }

 int drm_sman_free_key(drm_sman_t *sman, unsigned int key)
 {
 	drm_hash_item_t *hash_item;
 	drm_memblock_item_t *memblock_item;

 	if (drm_ht_find_item(&sman->user_hash_tab, key, &hash_item))
 		return -EINVAL;

 	memblock_item = drm_hash_entry(hash_item, drm_memblock_item_t, user_hash);
 	drm_sman_free(memblock_item);
 	return 0;
 }

 EXPORT_SYMBOL(drm_sman_free_key);

 static void drm_sman_remove_owner(drm_sman_t *sman,
 				  drm_owner_item_t *owner_item)
 {
 	list_del(&owner_item->sman_list);
 	drm_ht_remove_item(&sman->owner_hash_tab, &owner_item->owner_hash);
 	drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
 }

 int drm_sman_owner_clean(drm_sman_t *sman, unsigned long owner)
 {

 	drm_hash_item_t *hash_item;
 	drm_owner_item_t *owner_item;

 	if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
 		return -1;
 	}

 	owner_item = drm_hash_entry(hash_item, drm_owner_item_t, 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(drm_sman_t *sman,
 				      drm_owner_item_t *owner_item)
 {
 	drm_memblock_item_t *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(drm_sman_t *sman, unsigned long owner)
 {

 	drm_hash_item_t *hash_item;
 	drm_owner_item_t *owner_item;

 	if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {

 		return;
 	}

 	owner_item = drm_hash_entry(hash_item, drm_owner_item_t, owner_hash);
 	drm_sman_do_owner_cleanup(sman, owner_item);
 }

 EXPORT_SYMBOL(drm_sman_owner_cleanup);

 void drm_sman_cleanup(drm_sman_t *sman)
 {
 	drm_owner_item_t *entry, *next;
 	unsigned int i;
 	drm_sman_mm_t *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);
	/**************************************************************************
	*
	* 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"

	typedef struct drm_owner_item {
	drm_hash_item_t owner_hash;
	struct list_head sman_list;
	struct list_head mem_blocks;
	} drm_owner_item_t;

	void drm_sman_takedown(drm_sman_t * sman)
	{
	drm_ht_remove(&sman->user_hash_tab);
	drm_ht_remove(&sman->owner_hash_tab);
	if (sman->mm)
	drm_free(sman->mm, sman->num_managers * sizeof(*sman->mm),
	DRM_MEM_MM);
	}

	EXPORT_SYMBOL(drm_sman_takedown);

	int
	drm_sman_init(drm_sman_t * sman, unsigned int num_managers,
	unsigned int user_order, unsigned int owner_order)
	{
	int ret = 0;

	sman->mm = (drm_sman_mm_t ) drm_calloc(num_managers, sizeof(sman->mm),
	DRM_MEM_MM);
	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:
	drm_free(sman->mm, num_managers * sizeof(*sman->mm), DRM_MEM_MM);
	out:
	return ret;
	}

	EXPORT_SYMBOL(drm_sman_init);

	static void drm_sman_mm_allocate(void private, unsigned long size,
	unsigned alignment)
	{
	drm_mm_t mm = (drm_mm_t ) private;
	drm_mm_node_t *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)
	{
	drm_mm_t mm = (drm_mm_t ) private;
	drm_mm_node_t node = (drm_mm_node_t ) ref;

	drm_mm_put_block(mm, node);
	}

	static void drm_sman_mm_destroy(void *private)
	{
	drm_mm_t mm = (drm_mm_t ) private;
	drm_mm_takedown(mm);
	drm_free(mm, sizeof(*mm), DRM_MEM_MM);
	}

	static unsigned long drm_sman_mm_offset(void private, void ref)
	{
	drm_mm_node_t node = (drm_mm_node_t ) ref;
	return node->start;
	}

	int
	drm_sman_set_range(drm_sman_t * sman, unsigned int manager,
	unsigned long start, unsigned long size)
	{
	drm_sman_mm_t *sman_mm;
	drm_mm_t *mm;
	int ret;

	BUG_ON(manager >= sman->num_managers);

	sman_mm = &sman->mm[manager];
	mm = drm_calloc(1, sizeof(*mm), DRM_MEM_MM);
	if (!mm) {
	return -ENOMEM;
	}
	sman_mm->private = mm;
	ret = drm_mm_init(mm, start, size);

	if (ret) {
	drm_free(mm, sizeof(*mm), DRM_MEM_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(drm_sman_t * sman, unsigned int manager,
	drm_sman_mm_t * allocator)
	{
	BUG_ON(manager >= sman->num_managers);
	sman->mm[manager] = *allocator;

	return 0;
	}

	static drm_owner_item_t drm_sman_get_owner_item(drm_sman_t sman,
	unsigned long owner)
	{
	int ret;
	drm_hash_item_t *owner_hash_item;
	drm_owner_item_t *owner_item;

	ret = drm_ht_find_item(&sman->owner_hash_tab, owner, &owner_hash_item);
	if (!ret) {
	return drm_hash_entry(owner_hash_item, drm_owner_item_t,
	owner_hash);
	}

	owner_item = drm_calloc(1, sizeof(*owner_item), DRM_MEM_MM);
	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:
	drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
	out:
	return NULL;
	}

	drm_memblock_item_t drm_sman_alloc(drm_sman_t sman, unsigned int manager,
	unsigned long size, unsigned alignment,
	unsigned long owner)
	{
	void *tmp;
	drm_sman_mm_t *sman_mm;
	drm_owner_item_t *owner_item;
	drm_memblock_item_t *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 = drm_calloc(1, sizeof(*memblock), DRM_MEM_MM);

	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:
	drm_free(memblock, sizeof(*memblock), DRM_MEM_MM);
	out:
	sman_mm->free(sman_mm->private, tmp);

	return NULL;
	}

	EXPORT_SYMBOL(drm_sman_alloc);

	static void drm_sman_free(drm_memblock_item_t *item)
	{
	drm_sman_t *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);
	drm_free(item, sizeof(*item), DRM_MEM_MM);
	}

	int drm_sman_free_key(drm_sman_t *sman, unsigned int key)
	{
	drm_hash_item_t *hash_item;
	drm_memblock_item_t *memblock_item;

	if (drm_ht_find_item(&sman->user_hash_tab, key, &hash_item))
	return -EINVAL;

	memblock_item = drm_hash_entry(hash_item, drm_memblock_item_t, user_hash);
	drm_sman_free(memblock_item);
	return 0;
	}

	EXPORT_SYMBOL(drm_sman_free_key);

	static void drm_sman_remove_owner(drm_sman_t *sman,
	drm_owner_item_t *owner_item)
	{
	list_del(&owner_item->sman_list);
	drm_ht_remove_item(&sman->owner_hash_tab, &owner_item->owner_hash);
	drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
	}

	int drm_sman_owner_clean(drm_sman_t *sman, unsigned long owner)
	{

	drm_hash_item_t *hash_item;
	drm_owner_item_t *owner_item;

	if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
	return -1;
	}

	owner_item = drm_hash_entry(hash_item, drm_owner_item_t, 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(drm_sman_t *sman,
	drm_owner_item_t *owner_item)
	{
	drm_memblock_item_t 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(drm_sman_t *sman, unsigned long owner)
	{

	drm_hash_item_t *hash_item;
	drm_owner_item_t *owner_item;

	if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {

	return;
	}

	owner_item = drm_hash_entry(hash_item, drm_owner_item_t, owner_hash);
	drm_sman_do_owner_cleanup(sman, owner_item);
	}

	EXPORT_SYMBOL(drm_sman_owner_cleanup);

	void drm_sman_cleanup(drm_sman_t *sman)
	{
	drm_owner_item_t entry, next;
	unsigned int i;
	drm_sman_mm_t *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);