drivers/char/drm/drm_sman.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2006 Tungsten Graphics, Inc., Bismarck., ND., 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 SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
  17  * THE COPYRIGHT HOLDERS, AUTHORS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM,
  18  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  19  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  20  * USE OR OTHER DEALINGS IN THE SOFTWARE.
  21  *
  22  * The above copyright notice and this permission notice (including the
  23  * next paragraph) shall be included in all copies or substantial portions
  24  * of the Software.
  25  *
  26  *
  27  **************************************************************************/
  28 /*
  29  * Simple memory manager interface that keeps track on allocate regions on a
  30  * per "owner" basis. All regions associated with an "owner" can be released
  31  * with a simple call. Typically if the "owner" exists. The owner is any
  32  * "unsigned long" identifier. Can typically be a pointer to a file private
  33  * struct or a context identifier.
  34  *
  35  * Authors:
  36  * Thomas Hellström <thomas-at-tungstengraphics-dot-com>
  37  */
  38
  39 #include "drm_sman.h"
  40
  41 typedef struct drm_owner_item {
  42         drm_hash_item_t owner_hash;
  43         struct list_head sman_list;
  44         struct list_head mem_blocks;
  45 } drm_owner_item_t;
  46
  47 void drm_sman_takedown(drm_sman_t * sman)
  48 {
  49         drm_ht_remove(&sman->user_hash_tab);
  50         drm_ht_remove(&sman->owner_hash_tab);
  51         if (sman->mm)
  52                 drm_free(sman->mm, sman->num_managers * sizeof(*sman->mm),
  53                          DRM_MEM_MM);
  54 }
  55
  56 EXPORT_SYMBOL(drm_sman_takedown);
  57
  58 int
  59 drm_sman_init(drm_sman_t * sman, unsigned int num_managers,
  60               unsigned int user_order, unsigned int owner_order)
  61 {
  62         int ret = 0;
  63
  64         sman->mm = (drm_sman_mm_t *) drm_calloc(num_managers, sizeof(*sman->mm),
  65                                                 DRM_MEM_MM);
  66         if (!sman->mm) {
  67                 ret = -ENOMEM;
  68                 goto out;
  69         }
  70         sman->num_managers = num_managers;
  71         INIT_LIST_HEAD(&sman->owner_items);
  72         ret = drm_ht_create(&sman->owner_hash_tab, owner_order);
  73         if (ret)
  74                 goto out1;
  75         ret = drm_ht_create(&sman->user_hash_tab, user_order);
  76         if (!ret)
  77                 goto out;
  78
  79         drm_ht_remove(&sman->owner_hash_tab);
  80 out1:
  81         drm_free(sman->mm, num_managers * sizeof(*sman->mm), DRM_MEM_MM);
  82 out:
  83         return ret;
  84 }
  85
  86 EXPORT_SYMBOL(drm_sman_init);
  87
  88 static void *drm_sman_mm_allocate(void *private, unsigned long size,
  89                                   unsigned alignment)
  90 {
  91         drm_mm_t *mm = (drm_mm_t *) private;
  92         drm_mm_node_t *tmp;
  93
  94         tmp = drm_mm_search_free(mm, size, alignment, 1);
  95         if (!tmp) {
  96                 return NULL;
  97         }
  98         tmp = drm_mm_get_block(tmp, size, alignment);
  99         return tmp;
 100 }
 101
 102 static void drm_sman_mm_free(void *private, void *ref)
 103 {
 104         drm_mm_t *mm = (drm_mm_t *) private;
 105         drm_mm_node_t *node = (drm_mm_node_t *) ref;
 106
 107         drm_mm_put_block(mm, node);
 108 }
 109
 110 static void drm_sman_mm_destroy(void *private)
 111 {
 112         drm_mm_t *mm = (drm_mm_t *) private;
 113         drm_mm_takedown(mm);
 114         drm_free(mm, sizeof(*mm), DRM_MEM_MM);
 115 }
 116
 117 static unsigned long drm_sman_mm_offset(void *private, void *ref)
 118 {
 119         drm_mm_node_t *node = (drm_mm_node_t *) ref;
 120         return node->start;
 121 }
 122
 123 int
 124 drm_sman_set_range(drm_sman_t * sman, unsigned int manager,
 125                    unsigned long start, unsigned long size)
 126 {
 127         drm_sman_mm_t *sman_mm;
 128         drm_mm_t *mm;
 129         int ret;
 130
 131         BUG_ON(manager >= sman->num_managers);
 132
 133         sman_mm = &sman->mm[manager];
 134         mm = drm_calloc(1, sizeof(*mm), DRM_MEM_MM);
 135         if (!mm) {
 136                 return -ENOMEM;
 137         }
 138         sman_mm->private = mm;
 139         ret = drm_mm_init(mm, start, size);
 140
 141         if (ret) {
 142                 drm_free(mm, sizeof(*mm), DRM_MEM_MM);
 143                 return ret;
 144         }
 145
 146         sman_mm->allocate = drm_sman_mm_allocate;
 147         sman_mm->free = drm_sman_mm_free;
 148         sman_mm->destroy = drm_sman_mm_destroy;
 149         sman_mm->offset = drm_sman_mm_offset;
 150
 151         return 0;
 152 }
 153
 154 EXPORT_SYMBOL(drm_sman_set_range);
 155
 156 int
 157 drm_sman_set_manager(drm_sman_t * sman, unsigned int manager,
 158                      drm_sman_mm_t * allocator)
 159 {
 160         BUG_ON(manager >= sman->num_managers);
 161         sman->mm[manager] = *allocator;
 162
 163         return 0;
 164 }
 165 EXPORT_SYMBOL(drm_sman_set_manager);
 166
 167 static drm_owner_item_t *drm_sman_get_owner_item(drm_sman_t * sman,
 168                                                  unsigned long owner)
 169 {
 170         int ret;
 171         drm_hash_item_t *owner_hash_item;
 172         drm_owner_item_t *owner_item;
 173
 174         ret = drm_ht_find_item(&sman->owner_hash_tab, owner, &owner_hash_item);
 175         if (!ret) {
 176                 return drm_hash_entry(owner_hash_item, drm_owner_item_t,
 177                                       owner_hash);
 178         }
 179
 180         owner_item = drm_calloc(1, sizeof(*owner_item), DRM_MEM_MM);
 181         if (!owner_item)
 182                 goto out;
 183
 184         INIT_LIST_HEAD(&owner_item->mem_blocks);
 185         owner_item->owner_hash.key = owner;
 186         if (drm_ht_insert_item(&sman->owner_hash_tab, &owner_item->owner_hash))
 187                 goto out1;
 188
 189         list_add_tail(&owner_item->sman_list, &sman->owner_items);
 190         return owner_item;
 191
 192 out1:
 193         drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
 194 out:
 195         return NULL;
 196 }
 197
 198 drm_memblock_item_t *drm_sman_alloc(drm_sman_t *sman, unsigned int manager,
 199                                     unsigned long size, unsigned alignment,
 200                                     unsigned long owner)
 201 {
 202         void *tmp;
 203         drm_sman_mm_t *sman_mm;
 204         drm_owner_item_t *owner_item;
 205         drm_memblock_item_t *memblock;
 206
 207         BUG_ON(manager >= sman->num_managers);
 208
 209         sman_mm = &sman->mm[manager];
 210         tmp = sman_mm->allocate(sman_mm->private, size, alignment);
 211
 212         if (!tmp) {
 213                 return NULL;
 214         }
 215
 216         memblock = drm_calloc(1, sizeof(*memblock), DRM_MEM_MM);
 217
 218         if (!memblock)
 219                 goto out;
 220
 221         memblock->mm_info = tmp;
 222         memblock->mm = sman_mm;
 223         memblock->sman = sman;
 224
 225         if (drm_ht_just_insert_please
 226             (&sman->user_hash_tab, &memblock->user_hash,
 227              (unsigned long)memblock, 32, 0, 0))
 228                 goto out1;
 229
 230         owner_item = drm_sman_get_owner_item(sman, owner);
 231         if (!owner_item)
 232                 goto out2;
 233
 234         list_add_tail(&memblock->owner_list, &owner_item->mem_blocks);
 235
 236         return memblock;
 237
 238 out2:
 239         drm_ht_remove_item(&sman->user_hash_tab, &memblock->user_hash);
 240 out1:
 241         drm_free(memblock, sizeof(*memblock), DRM_MEM_MM);
 242 out:
 243         sman_mm->free(sman_mm->private, tmp);
 244
 245         return NULL;
 246 }
 247
 248 EXPORT_SYMBOL(drm_sman_alloc);
 249
 250 static void drm_sman_free(drm_memblock_item_t *item)
 251 {
 252         drm_sman_t *sman = item->sman;
 253
 254         list_del(&item->owner_list);
 255         drm_ht_remove_item(&sman->user_hash_tab, &item->user_hash);
 256         item->mm->free(item->mm->private, item->mm_info);
 257         drm_free(item, sizeof(*item), DRM_MEM_MM);
 258 }
 259
 260 int drm_sman_free_key(drm_sman_t *sman, unsigned int key)
 261 {
 262         drm_hash_item_t *hash_item;
 263         drm_memblock_item_t *memblock_item;
 264
 265         if (drm_ht_find_item(&sman->user_hash_tab, key, &hash_item))
 266                 return -EINVAL;
 267
 268         memblock_item = drm_hash_entry(hash_item, drm_memblock_item_t, user_hash);
 269         drm_sman_free(memblock_item);
 270         return 0;
 271 }
 272
 273 EXPORT_SYMBOL(drm_sman_free_key);
 274
 275 static void drm_sman_remove_owner(drm_sman_t *sman,
 276                                   drm_owner_item_t *owner_item)
 277 {
 278         list_del(&owner_item->sman_list);
 279         drm_ht_remove_item(&sman->owner_hash_tab, &owner_item->owner_hash);
 280         drm_free(owner_item, sizeof(*owner_item), DRM_MEM_MM);
 281 }
 282
 283 int drm_sman_owner_clean(drm_sman_t *sman, unsigned long owner)
 284 {
 285
 286         drm_hash_item_t *hash_item;
 287         drm_owner_item_t *owner_item;
 288
 289         if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
 290                 return -1;
 291         }
 292
 293         owner_item = drm_hash_entry(hash_item, drm_owner_item_t, owner_hash);
 294         if (owner_item->mem_blocks.next == &owner_item->mem_blocks) {
 295                 drm_sman_remove_owner(sman, owner_item);
 296                 return -1;
 297         }
 298
 299         return 0;
 300 }
 301
 302 EXPORT_SYMBOL(drm_sman_owner_clean);
 303
 304 static void drm_sman_do_owner_cleanup(drm_sman_t *sman,
 305                                       drm_owner_item_t *owner_item)
 306 {
 307         drm_memblock_item_t *entry, *next;
 308
 309         list_for_each_entry_safe(entry, next, &owner_item->mem_blocks,
 310                                  owner_list) {
 311                 drm_sman_free(entry);
 312         }
 313         drm_sman_remove_owner(sman, owner_item);
 314 }
 315
 316 void drm_sman_owner_cleanup(drm_sman_t *sman, unsigned long owner)
 317 {
 318
 319         drm_hash_item_t *hash_item;
 320         drm_owner_item_t *owner_item;
 321
 322         if (drm_ht_find_item(&sman->owner_hash_tab, owner, &hash_item)) {
 323
 324                 return;
 325         }
 326
 327         owner_item = drm_hash_entry(hash_item, drm_owner_item_t, owner_hash);
 328         drm_sman_do_owner_cleanup(sman, owner_item);
 329 }
 330
 331 EXPORT_SYMBOL(drm_sman_owner_cleanup);
 332
 333 void drm_sman_cleanup(drm_sman_t *sman)
 334 {
 335         drm_owner_item_t *entry, *next;
 336         unsigned int i;
 337         drm_sman_mm_t *sman_mm;
 338
 339         list_for_each_entry_safe(entry, next, &sman->owner_items, sman_list) {
 340                 drm_sman_do_owner_cleanup(sman, entry);
 341         }
 342         if (sman->mm) {
 343                 for (i = 0; i < sman->num_managers; ++i) {
 344                         sman_mm = &sman->mm[i];
 345                         if (sman_mm->private) {
 346                                 sman_mm->destroy(sman_mm->private);
 347                                 sman_mm->private = NULL;
 348                         }
 349                 }
 350         }
 351 }
 352
 353 EXPORT_SYMBOL(drm_sman_cleanup);