freebsd-dev/sys/dev/drm2/drm_gem_names.c
Jean-Sébastien Pédron 735f427563 drm: In drm_gem_name_create(), verify argument before acquiring lock
Submitted by:	J.R. Oldroyd <jr@opal.com>
2013-08-25 10:04:10 +00:00

212 lines
5.2 KiB
C

/*-
* Copyright (c) 2011 The FreeBSD Foundation
* All rights reserved.
*
* This software was developed by Konstantin Belousov under sponsorship from
* the FreeBSD Foundation.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <sys/cdefs.h>
__FBSDID("$FreeBSD$");
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/limits.h>
#include <sys/malloc.h>
#include <dev/drm2/drm_gem_names.h>
MALLOC_DEFINE(M_GEM_NAMES, "gem_name", "Hash headers for the gem names");
static void drm_gem_names_delete_name(struct drm_gem_names *names,
struct drm_gem_name *np);
void
drm_gem_names_init(struct drm_gem_names *names)
{
names->unr = new_unrhdr(1, INT_MAX, NULL); /* XXXKIB */
names->names_hash = hashinit(1000 /* XXXKIB */, M_GEM_NAMES,
&names->hash_mask);
mtx_init(&names->lock, "drmnames", NULL, MTX_DEF);
}
void
drm_gem_names_fini(struct drm_gem_names *names)
{
struct drm_gem_name *np;
int i;
mtx_lock(&names->lock);
for (i = 0; i <= names->hash_mask; i++) {
while ((np = LIST_FIRST(&names->names_hash[i])) != NULL) {
drm_gem_names_delete_name(names, np);
mtx_lock(&names->lock);
}
}
mtx_unlock(&names->lock);
mtx_destroy(&names->lock);
hashdestroy(names->names_hash, M_GEM_NAMES, names->hash_mask);
delete_unrhdr(names->unr);
}
static struct drm_gem_names_head *
gem_name_hash_index(struct drm_gem_names *names, int name)
{
return (&names->names_hash[name & names->hash_mask]);
}
void *
drm_gem_name_ref(struct drm_gem_names *names, uint32_t name,
void (*ref)(void *))
{
struct drm_gem_name *n;
mtx_lock(&names->lock);
LIST_FOREACH(n, gem_name_hash_index(names, name), link) {
if (n->name == name) {
if (ref != NULL)
ref(n->ptr);
mtx_unlock(&names->lock);
return (n->ptr);
}
}
mtx_unlock(&names->lock);
return (NULL);
}
struct drm_gem_ptr_match_arg {
uint32_t res;
void *ptr;
};
static int
drm_gem_ptr_match(uint32_t name, void *ptr, void *arg)
{
struct drm_gem_ptr_match_arg *a;
a = arg;
if (ptr == a->ptr) {
a->res = name;
return (1);
} else
return (0);
}
uint32_t
drm_gem_find_name(struct drm_gem_names *names, void *ptr)
{
struct drm_gem_ptr_match_arg arg;
arg.res = 0;
arg.ptr = ptr;
drm_gem_names_foreach(names, drm_gem_ptr_match, &arg);
return (arg.res);
}
int
drm_gem_name_create(struct drm_gem_names *names, void *p, uint32_t *name)
{
struct drm_gem_name *np;
if (*name != 0) {
return (EALREADY);
}
np = malloc(sizeof(struct drm_gem_name), M_GEM_NAMES, M_WAITOK);
mtx_lock(&names->lock);
np->name = alloc_unr(names->unr);
if (np->name == -1) {
mtx_unlock(&names->lock);
free(np, M_GEM_NAMES);
return (ENOMEM);
}
*name = np->name;
np->ptr = p;
LIST_INSERT_HEAD(gem_name_hash_index(names, np->name), np, link);
mtx_unlock(&names->lock);
return (0);
}
static void
drm_gem_names_delete_name(struct drm_gem_names *names, struct drm_gem_name *np)
{
mtx_assert(&names->lock, MA_OWNED);
LIST_REMOVE(np, link);
mtx_unlock(&names->lock);
free_unr(names->unr, np->name);
free(np, M_GEM_NAMES);
}
void *
drm_gem_names_remove(struct drm_gem_names *names, uint32_t name)
{
struct drm_gem_name *n;
void *res;
mtx_lock(&names->lock);
LIST_FOREACH(n, gem_name_hash_index(names, name), link) {
if (n->name == name) {
res = n->ptr;
drm_gem_names_delete_name(names, n);
return (res);
}
}
mtx_unlock(&names->lock);
return (NULL);
}
void
drm_gem_names_foreach(struct drm_gem_names *names,
int (*f)(uint32_t, void *, void *), void *arg)
{
struct drm_gem_name *np;
struct drm_gem_name marker;
int i, fres;
bzero(&marker, sizeof(marker));
marker.name = -1;
mtx_lock(&names->lock);
for (i = 0; i <= names->hash_mask; i++) {
for (np = LIST_FIRST(&names->names_hash[i]); np != NULL; ) {
if (np->name == -1) {
np = LIST_NEXT(np, link);
continue;
}
LIST_INSERT_AFTER(np, &marker, link);
mtx_unlock(&names->lock);
fres = f(np->name, np->ptr, arg);
mtx_lock(&names->lock);
np = LIST_NEXT(&marker, link);
LIST_REMOVE(&marker, link);
if (fres)
break;
}
}
mtx_unlock(&names->lock);
}