Files
kmscon/src/shl_hook.h
T
David Herrmann 46898b1090 shl: hook: allow callbacks to destroy the hook
If we currently dispatch a callback and they unregister themself and then
destroy the hook, we currently do not free the hook as we would free
memory that is used in the callstack.
However, it is a valid use-case for hook-callbacks to destroy the hook.
Therefore, we set a "dead" flag and destroy the hook after all callbacks
are done.

This fixes some memory-leaks inside the shared-signal callbacks of eloop.

Signed-off-by: David Herrmann <dh.herrmann@googlemail.com>
2012-12-02 13:56:31 +01:00

178 lines
4.1 KiB
C

/*
* shl - Hook Handling
*
* Copyright (c) 2011-2012 David Herrmann <dh.herrmann@googlemail.com>
* Copyright (c) 2011 University of Tuebingen
*
* 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, sublicense, 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 above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* 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 NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS 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.
*/
/*
* Simply hook-implementation
*/
#ifndef SHL_HOOK_H
#define SHL_HOOK_H
#include <errno.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include "shl_dlist.h"
struct shl_hook;
struct shl_hook_entry;
typedef void (*shl_hook_cb) (void *parent, void *arg, void *data);
#define shl_hook_add_cast(hook, cb, data) \
shl_hook_add((hook), (shl_hook_cb)(cb), (data))
#define shl_hook_rm_cast(hook, cb, data) \
shl_hook_rm((hook), (shl_hook_cb)(cb), (data))
struct shl_hook_entry {
struct shl_dlist list;
shl_hook_cb cb;
void *data;
};
struct shl_hook {
unsigned int num;
struct shl_dlist entries;
struct shl_dlist *cur_entry;
bool dead;
};
static inline int shl_hook_new(struct shl_hook **out)
{
struct shl_hook *hook;
if (!out)
return -EINVAL;
hook = malloc(sizeof(*hook));
if (!hook)
return -ENOMEM;
memset(hook, 0, sizeof(*hook));
shl_dlist_init(&hook->entries);
*out = hook;
return 0;
}
static inline void shl_hook_free(struct shl_hook *hook)
{
struct shl_hook_entry *entry;
if (!hook)
return;
if (hook->cur_entry) {
hook->dead = true;
return;
}
while (!shl_dlist_empty(&hook->entries)) {
entry = shl_dlist_entry(hook->entries.prev,
struct shl_hook_entry,
list);
shl_dlist_unlink(&entry->list);
free(entry);
}
free(hook);
}
static inline unsigned int shl_hook_num(struct shl_hook *hook)
{
if (!hook)
return 0;
return hook->num;
}
static inline int shl_hook_add(struct shl_hook *hook, shl_hook_cb cb,
void *data)
{
struct shl_hook_entry *entry;
if (!hook || !cb)
return -EINVAL;
entry = malloc(sizeof(*entry));
if (!entry)
return -ENOMEM;
memset(entry, 0, sizeof(*entry));
entry->cb = cb;
entry->data = data;
shl_dlist_link_tail(&hook->entries, &entry->list);
hook->num++;
return 0;
}
static inline void shl_hook_rm(struct shl_hook *hook, shl_hook_cb cb,
void *data)
{
struct shl_dlist *iter;
struct shl_hook_entry *entry;
if (!hook || !cb)
return;
shl_dlist_for_each_reverse(iter, &hook->entries) {
entry = shl_dlist_entry(iter, struct shl_hook_entry, list);
if (entry->cb == cb && entry->data == data) {
/* if *_call() is running we must not disturb it */
if (hook->cur_entry == iter)
hook->cur_entry = iter->next;
shl_dlist_unlink(&entry->list);
free(entry);
hook->num--;
return;
}
}
}
static inline void shl_hook_call(struct shl_hook *hook, void *parent,
void *arg)
{
struct shl_hook_entry *entry;
if (!hook || hook->cur_entry)
return;
for (hook->cur_entry = hook->entries.next;
hook->cur_entry != &hook->entries; ) {
entry = shl_dlist_entry(hook->cur_entry,
struct shl_hook_entry, list);
entry->cb(parent, arg, entry->data);
if (hook->cur_entry == &entry->list)
hook->cur_entry = hook->cur_entry->next;
}
hook->cur_entry = NULL;
if (hook->dead)
shl_hook_free(hook);
}
#endif /* SHL_HOOK_H */