mirror of
https://github.com/aljazceru/lightning.git
synced 2025-12-18 22:54:25 +01:00
utils: add set_softref() / clear_softref().
We often want a pointer which will turn to NULL if the pointed-to thing is freed. This is possible with tal objects, so create it. Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
committed by
neil saitug
parent
296868daf4
commit
d5eca470dc
@@ -15,6 +15,78 @@ bool is_elements(const struct chainparams *chainparams)
|
||||
return chainparams->is_elements;
|
||||
}
|
||||
|
||||
#if DEVELOPER
|
||||
/* If you've got a softref, we assume no reallocs. */
|
||||
static void dont_move_softref(tal_t *ctx, enum tal_notify_type ntype, void *info)
|
||||
{
|
||||
abort();
|
||||
}
|
||||
#endif
|
||||
|
||||
static void softref_nullify(tal_t *obj, void **ptr)
|
||||
{
|
||||
*ptr = NULL;
|
||||
#if DEVELOPER
|
||||
tal_del_notifier(obj, dont_move_softref);
|
||||
#endif
|
||||
}
|
||||
|
||||
static void softref_cleanup(const tal_t *outer, void **ptr)
|
||||
{
|
||||
if (*ptr) {
|
||||
tal_del_destructor2(*ptr, softref_nullify, ptr);
|
||||
}
|
||||
#if DEVELOPER
|
||||
tal_del_notifier(outer, dont_move_softref);
|
||||
#endif
|
||||
}
|
||||
|
||||
void set_softref_(const tal_t *outer, size_t outersize, void **ptr, tal_t *obj)
|
||||
{
|
||||
/* pointer is inside outer, right? */
|
||||
assert((char *)ptr >= (char *)outer);
|
||||
assert((char *)ptr < (char *)outer + outersize);
|
||||
|
||||
/* This is harmless if there was no prior, otherwise constrains the
|
||||
* leak: we don't have enough information in softref_nullify to
|
||||
* clear softref_cleanup */
|
||||
tal_del_destructor2(outer, softref_cleanup, ptr);
|
||||
|
||||
if (obj) {
|
||||
tal_add_destructor2(outer, softref_cleanup, ptr);
|
||||
tal_add_destructor2(obj, softref_nullify, ptr);
|
||||
#if DEVELOPER
|
||||
tal_add_notifier(obj, TAL_NOTIFY_MOVE, dont_move_softref);
|
||||
#endif
|
||||
}
|
||||
|
||||
#if DEVELOPER
|
||||
tal_add_notifier(outer, TAL_NOTIFY_MOVE, dont_move_softref);
|
||||
#endif
|
||||
|
||||
*ptr = obj;
|
||||
}
|
||||
|
||||
void clear_softref_(const tal_t *outer, size_t outersize, void **ptr)
|
||||
{
|
||||
assert((char *)ptr >= (char *)outer);
|
||||
assert((char *)ptr < (char *)outer + outersize);
|
||||
|
||||
if (*ptr) {
|
||||
tal_del_destructor2(outer, softref_cleanup, ptr);
|
||||
tal_del_destructor2(*ptr, softref_nullify, ptr);
|
||||
#if DEVELOPER
|
||||
tal_del_notifier(*ptr, dont_move_softref);
|
||||
#endif
|
||||
}
|
||||
|
||||
#if DEVELOPER
|
||||
tal_del_notifier(outer, dont_move_softref);
|
||||
#endif
|
||||
|
||||
*ptr = NULL;
|
||||
}
|
||||
|
||||
char *tal_hexstr(const tal_t *ctx, const void *data, size_t len)
|
||||
{
|
||||
char *str = tal_arr(ctx, char, hex_str_size(len));
|
||||
|
||||
Reference in New Issue
Block a user