nostrdb: ccan: copy ccan files into their own subdirectory.
This lets them be updated/bugfixed together. I just copied them for now, didn't change anything else. Signed-off-by: Rusty Russell <rusty@rustcorp.com.au> Signed-off-by: William Casarin <jb55@jb55.com>
This commit is contained in:
committed by
Daniel D’Aquino
parent
6c53bc75f2
commit
494386d211
315
nostrdb/ccan/ccan/tal/str/str.c
Normal file
315
nostrdb/ccan/ccan/tal/str/str.c
Normal file
@@ -0,0 +1,315 @@
|
||||
/* Licensed under BSD-MIT - see LICENSE file for details */
|
||||
#include <unistd.h>
|
||||
#include <stdint.h>
|
||||
#include <string.h>
|
||||
#include <limits.h>
|
||||
#include <stdlib.h>
|
||||
#include "talstr.h"
|
||||
#include <sys/types.h>
|
||||
#include <regex.h>
|
||||
#include <stdarg.h>
|
||||
#include <unistd.h>
|
||||
#include <stdio.h>
|
||||
#include "str.h"
|
||||
|
||||
char *tal_strdup_(const tal_t *ctx, const char *p, const char *label)
|
||||
{
|
||||
/* We have to let through NULL for take(). */
|
||||
return tal_dup_arr_label(ctx, char, p, p ? strlen(p) + 1: 1, 0, label);
|
||||
}
|
||||
|
||||
char *tal_strndup_(const tal_t *ctx, const char *p, size_t n, const char *label)
|
||||
{
|
||||
size_t len;
|
||||
char *ret;
|
||||
|
||||
/* We have to let through NULL for take(). */
|
||||
if (likely(p))
|
||||
len = strnlen(p, n);
|
||||
else
|
||||
len = n;
|
||||
|
||||
ret = tal_dup_arr_label(ctx, char, p, len, 1, label);
|
||||
if (ret)
|
||||
ret[len] = '\0';
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *tal_fmt_(const tal_t *ctx, const char *label, const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
char *ret;
|
||||
|
||||
va_start(ap, fmt);
|
||||
ret = tal_vfmt_(ctx, fmt, ap, label);
|
||||
va_end(ap);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
static bool do_vfmt(char **buf, size_t off, const char *fmt, va_list ap)
|
||||
{
|
||||
/* A decent guess to start. */
|
||||
size_t max = strlen(fmt) * 2 + 1;
|
||||
bool ok;
|
||||
|
||||
for (;;) {
|
||||
va_list ap2;
|
||||
int ret;
|
||||
|
||||
if (!tal_resize(buf, off + max)) {
|
||||
ok = false;
|
||||
break;
|
||||
}
|
||||
|
||||
va_copy(ap2, ap);
|
||||
ret = vsnprintf(*buf + off, max, fmt, ap2);
|
||||
va_end(ap2);
|
||||
|
||||
if (ret < max) {
|
||||
ok = true;
|
||||
/* Make sure tal_count() is correct! */
|
||||
tal_resize(buf, off + ret + 1);
|
||||
break;
|
||||
}
|
||||
max *= 2;
|
||||
}
|
||||
|
||||
if (taken(fmt))
|
||||
tal_free(fmt);
|
||||
return ok;
|
||||
}
|
||||
|
||||
char *tal_vfmt_(const tal_t *ctx, const char *fmt, va_list ap, const char *label)
|
||||
{
|
||||
char *buf;
|
||||
|
||||
if (!fmt && taken(fmt))
|
||||
return NULL;
|
||||
|
||||
/* A decent guess to start. */
|
||||
buf = tal_arr_label(ctx, char, strlen(fmt) * 2, label);
|
||||
if (!do_vfmt(&buf, 0, fmt, ap))
|
||||
buf = tal_free(buf);
|
||||
return buf;
|
||||
}
|
||||
|
||||
bool tal_append_vfmt(char **baseptr, const char *fmt, va_list ap)
|
||||
{
|
||||
if (!fmt && taken(fmt))
|
||||
return false;
|
||||
|
||||
return do_vfmt(baseptr, strlen(*baseptr), fmt, ap);
|
||||
}
|
||||
|
||||
bool tal_append_fmt(char **baseptr, const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
bool ret;
|
||||
|
||||
va_start(ap, fmt);
|
||||
ret = tal_append_vfmt(baseptr, fmt, ap);
|
||||
va_end(ap);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
char *tal_strcat_(const tal_t *ctx, const char *s1, const char *s2,
|
||||
const char *label)
|
||||
{
|
||||
size_t len1, len2;
|
||||
char *ret;
|
||||
|
||||
if (unlikely(!s2) && taken(s2)) {
|
||||
if (taken(s1))
|
||||
tal_free(s1);
|
||||
return NULL;
|
||||
}
|
||||
/* We have to let through NULL for take(). */
|
||||
len1 = s1 ? strlen(s1) : 0;
|
||||
len2 = strlen(s2);
|
||||
|
||||
ret = tal_dup_arr_label(ctx, char, s1, len1, len2 + 1, label);
|
||||
if (likely(ret))
|
||||
memcpy(ret + len1, s2, len2 + 1);
|
||||
|
||||
if (taken(s2))
|
||||
tal_free(s2);
|
||||
return ret;
|
||||
}
|
||||
|
||||
char **tal_strsplit_(const tal_t *ctx,
|
||||
const char *string, const char *delims, enum strsplit flags,
|
||||
const char *label)
|
||||
{
|
||||
char **parts, *str;
|
||||
size_t max = 64, num = 0;
|
||||
|
||||
parts = tal_arr(ctx, char *, max + 1);
|
||||
if (unlikely(!parts)) {
|
||||
if (taken(string))
|
||||
tal_free(string);
|
||||
if (taken(delims))
|
||||
tal_free(delims);
|
||||
return NULL;
|
||||
}
|
||||
str = tal_strdup(parts, string);
|
||||
if (unlikely(!str))
|
||||
goto fail;
|
||||
if (unlikely(!delims) && is_taken(delims))
|
||||
goto fail;
|
||||
|
||||
if (flags == STR_NO_EMPTY)
|
||||
str += strspn(str, delims);
|
||||
|
||||
while (*str != '\0') {
|
||||
size_t len = strcspn(str, delims), dlen;
|
||||
|
||||
parts[num] = str;
|
||||
dlen = strspn(str + len, delims);
|
||||
parts[num][len] = '\0';
|
||||
if (flags == STR_EMPTY_OK && dlen)
|
||||
dlen = 1;
|
||||
str += len + dlen;
|
||||
if (++num == max && !tal_resize(&parts, max*=2 + 1))
|
||||
goto fail;
|
||||
}
|
||||
parts[num] = NULL;
|
||||
|
||||
/* Ensure that tal_count() is correct. */
|
||||
if (unlikely(!tal_resize(&parts, num+1)))
|
||||
goto fail;
|
||||
|
||||
if (taken(delims))
|
||||
tal_free(delims);
|
||||
return parts;
|
||||
|
||||
fail:
|
||||
tal_free(parts);
|
||||
if (taken(delims))
|
||||
tal_free(delims);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
char *tal_strjoin_(const tal_t *ctx,
|
||||
char *strings[], const char *delim, enum strjoin flags,
|
||||
const char *label)
|
||||
{
|
||||
unsigned int i;
|
||||
char *ret = NULL;
|
||||
size_t totlen = 0, dlen;
|
||||
|
||||
if (unlikely(!strings) && is_taken(strings))
|
||||
goto fail;
|
||||
|
||||
if (unlikely(!delim) && is_taken(delim))
|
||||
goto fail;
|
||||
|
||||
dlen = strlen(delim);
|
||||
ret = tal_arr_label(ctx, char, dlen*2+1, label);
|
||||
if (!ret)
|
||||
goto fail;
|
||||
|
||||
ret[0] = '\0';
|
||||
for (i = 0; strings[i]; i++) {
|
||||
size_t len = strlen(strings[i]);
|
||||
|
||||
if (flags == STR_NO_TRAIL && !strings[i+1])
|
||||
dlen = 0;
|
||||
if (!tal_resize(&ret, totlen + len + dlen + 1))
|
||||
goto fail;
|
||||
memcpy(ret + totlen, strings[i], len);
|
||||
totlen += len;
|
||||
memcpy(ret + totlen, delim, dlen);
|
||||
totlen += dlen;
|
||||
}
|
||||
ret[totlen] = '\0';
|
||||
/* Make sure tal_count() is correct! */
|
||||
tal_resize(&ret, totlen+1);
|
||||
out:
|
||||
if (taken(strings))
|
||||
tal_free(strings);
|
||||
if (taken(delim))
|
||||
tal_free(delim);
|
||||
return ret;
|
||||
fail:
|
||||
ret = tal_free(ret);
|
||||
goto out;
|
||||
}
|
||||
|
||||
static size_t count_open_braces(const char *string)
|
||||
{
|
||||
#if 1
|
||||
size_t num = 0, esc = 0;
|
||||
|
||||
while (*string) {
|
||||
if (*string == '\\')
|
||||
esc++;
|
||||
else {
|
||||
/* An odd number of \ means it's escaped. */
|
||||
if (*string == '(' && (esc & 1) == 0)
|
||||
num++;
|
||||
esc = 0;
|
||||
}
|
||||
string++;
|
||||
}
|
||||
return num;
|
||||
#else
|
||||
return strcount(string, "(");
|
||||
#endif
|
||||
}
|
||||
|
||||
bool tal_strreg_(const tal_t *ctx, const char *string, const char *label,
|
||||
const char *regex, ...)
|
||||
{
|
||||
size_t nmatch = 1 + count_open_braces(regex);
|
||||
regmatch_t matches[nmatch];
|
||||
regex_t r;
|
||||
bool ret = false;
|
||||
unsigned int i;
|
||||
va_list ap;
|
||||
|
||||
if (unlikely(!regex) && is_taken(regex))
|
||||
goto fail_no_re;
|
||||
|
||||
if (regcomp(&r, regex, REG_EXTENDED) != 0)
|
||||
goto fail_no_re;
|
||||
|
||||
if (unlikely(!string) && is_taken(string))
|
||||
goto fail;
|
||||
|
||||
if (regexec(&r, string, nmatch, matches, 0) != 0)
|
||||
goto fail;
|
||||
|
||||
ret = true;
|
||||
va_start(ap, regex);
|
||||
for (i = 1; i < nmatch; i++) {
|
||||
char **arg = va_arg(ap, char **);
|
||||
if (arg) {
|
||||
/* eg. ([a-z])? can give "no match". */
|
||||
if (matches[i].rm_so == -1)
|
||||
*arg = NULL;
|
||||
else {
|
||||
*arg = tal_strndup_(ctx,
|
||||
string + matches[i].rm_so,
|
||||
matches[i].rm_eo
|
||||
- matches[i].rm_so,
|
||||
label);
|
||||
/* FIXME: If we fail, we set some and leak! */
|
||||
if (!*arg) {
|
||||
ret = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
va_end(ap);
|
||||
fail:
|
||||
regfree(&r);
|
||||
fail_no_re:
|
||||
if (taken(regex))
|
||||
tal_free(regex);
|
||||
if (taken(string))
|
||||
tal_free(string);
|
||||
return ret;
|
||||
}
|
||||
225
nostrdb/ccan/ccan/tal/str/str.h
Normal file
225
nostrdb/ccan/ccan/tal/str/str.h
Normal file
@@ -0,0 +1,225 @@
|
||||
/* Licensed under BSD-MIT - see LICENSE file for details */
|
||||
#ifndef CCAN_STR_TAL_H
|
||||
#define CCAN_STR_TAL_H
|
||||
#ifdef TAL_USE_TALLOC
|
||||
#include <ccan/tal/talloc/talloc.h>
|
||||
#else
|
||||
#include "tal.h"
|
||||
#endif
|
||||
#include <string.h>
|
||||
#include <stdbool.h>
|
||||
|
||||
/**
|
||||
* tal_strdup - duplicate a string
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @p: the string to copy (can be take()).
|
||||
*
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*/
|
||||
#define tal_strdup(ctx, p) tal_strdup_(ctx, p, TAL_LABEL(char, "[]"))
|
||||
char *tal_strdup_(const tal_t *ctx, const char *p TAKES, const char *label);
|
||||
|
||||
/**
|
||||
* tal_strndup - duplicate a limited amount of a string.
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @p: the string to copy (can be take()).
|
||||
* @n: the maximum length to copy.
|
||||
*
|
||||
* Always gives a nul-terminated string, with strlen() <= @n.
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*/
|
||||
#define tal_strndup(ctx, p, n) tal_strndup_(ctx, p, n, TAL_LABEL(char, "[]"))
|
||||
char *tal_strndup_(const tal_t *ctx, const char *p TAKES, size_t n,
|
||||
const char *label);
|
||||
|
||||
/**
|
||||
* tal_fmt - allocate a formatted string
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @fmt: the printf-style format (can be take()).
|
||||
*
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*/
|
||||
#define tal_fmt(ctx, ...) \
|
||||
tal_fmt_(ctx, TAL_LABEL(char, "[]"), __VA_ARGS__)
|
||||
char *tal_fmt_(const tal_t *ctx, const char *label, const char *fmt TAKES,
|
||||
...) PRINTF_FMT(3,4);
|
||||
|
||||
/**
|
||||
* tal_vfmt - allocate a formatted string (va_list version)
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @fmt: the printf-style format (can be take()).
|
||||
* @va: the va_list containing the format args.
|
||||
*
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*/
|
||||
#define tal_vfmt(ctx, fmt, va) \
|
||||
tal_vfmt_(ctx, fmt, va, TAL_LABEL(char, "[]"))
|
||||
char *tal_vfmt_(const tal_t *ctx, const char *fmt TAKES, va_list ap,
|
||||
const char *label)
|
||||
PRINTF_FMT(2,0);
|
||||
|
||||
/**
|
||||
* tal_append_fmt - append a formatted string to a talloc string.
|
||||
* @baseptr: a pointer to the tal string to be appended to.
|
||||
* @fmt: the printf-style format (can be take()).
|
||||
*
|
||||
* Returns false on allocation failure.
|
||||
* Otherwise tal_count(*@baseptr) == strlen(*@baseptr) + 1.
|
||||
*/
|
||||
bool tal_append_fmt(char **baseptr, const char *fmt TAKES, ...) PRINTF_FMT(2,3);
|
||||
|
||||
/**
|
||||
* tal_append_vfmt - append a formatted string to a talloc string (va_list)
|
||||
* @baseptr: a pointer to the tal string to be appended to.
|
||||
* @fmt: the printf-style format (can be take()).
|
||||
* @va: the va_list containing the format args.
|
||||
*
|
||||
* Returns false on allocation failure.
|
||||
* Otherwise tal_count(*@baseptr) == strlen(*@baseptr) + 1.
|
||||
*/
|
||||
bool tal_append_vfmt(char **baseptr, const char *fmt TAKES, va_list ap);
|
||||
|
||||
/**
|
||||
* tal_strcat - join two strings together
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @s1: the first string (can be take()).
|
||||
* @s2: the second string (can be take()).
|
||||
*
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*/
|
||||
#define tal_strcat(ctx, s1, s2) tal_strcat_(ctx, s1, s2, TAL_LABEL(char, "[]"))
|
||||
char *tal_strcat_(const tal_t *ctx, const char *s1 TAKES, const char *s2 TAKES,
|
||||
const char *label);
|
||||
|
||||
enum strsplit {
|
||||
STR_EMPTY_OK,
|
||||
STR_NO_EMPTY
|
||||
};
|
||||
|
||||
/**
|
||||
* tal_strsplit - Split string into an array of substrings
|
||||
* @ctx: the context to tal from (often NULL).
|
||||
* @string: the string to split (can be take()).
|
||||
* @delims: delimiters where lines should be split (can be take()).
|
||||
* @flags: whether to include empty substrings.
|
||||
*
|
||||
* This function splits a single string into multiple strings.
|
||||
*
|
||||
* If @string is take(), the returned array will point into the
|
||||
* mangled @string.
|
||||
*
|
||||
* Multiple delimiters result in empty substrings. By definition, no
|
||||
* delimiters will appear in the substrings.
|
||||
*
|
||||
* The final char * in the array will be NULL, and tal_count() will
|
||||
* return the number of elements plus 1 (for that NULL).
|
||||
*
|
||||
* Example:
|
||||
* #include <ccan/tal/str/str.h>
|
||||
* ...
|
||||
* static unsigned int count_long_lines(const char *string)
|
||||
* {
|
||||
* char **lines;
|
||||
* unsigned int i, long_lines = 0;
|
||||
*
|
||||
* // Can only fail on out-of-memory.
|
||||
* lines = tal_strsplit(NULL, string, "\n", STR_NO_EMPTY);
|
||||
* for (i = 0; lines[i] != NULL; i++)
|
||||
* if (strlen(lines[i]) > 80)
|
||||
* long_lines++;
|
||||
* tal_free(lines);
|
||||
* return long_lines;
|
||||
* }
|
||||
*/
|
||||
#define tal_strsplit(ctx, string, delims, flag) \
|
||||
tal_strsplit_(ctx, string, delims, flag, TAL_LABEL(char *, "[]"))
|
||||
char **tal_strsplit_(const tal_t *ctx,
|
||||
const char *string TAKES,
|
||||
const char *delims TAKES,
|
||||
enum strsplit flag,
|
||||
const char *label);
|
||||
|
||||
enum strjoin {
|
||||
STR_TRAIL,
|
||||
STR_NO_TRAIL
|
||||
};
|
||||
|
||||
/**
|
||||
* tal_strjoin - Join an array of substrings into one long string
|
||||
* @ctx: the context to tal from (often NULL).
|
||||
* @strings: the NULL-terminated array of strings to join (can be take())
|
||||
* @delim: the delimiter to insert between the strings (can be take())
|
||||
* @flags: whether to add a delimieter to the end
|
||||
*
|
||||
* This function joins an array of strings into a single string. The
|
||||
* return value is allocated using tal. Each string in @strings is
|
||||
* followed by a copy of @delim.
|
||||
*
|
||||
* The returned string will have tal_count() == strlen() + 1.
|
||||
*
|
||||
* Example:
|
||||
* // Append the string "--EOL" to each line.
|
||||
* static char *append_to_all_lines(const char *string)
|
||||
* {
|
||||
* char **lines, *ret;
|
||||
*
|
||||
* lines = tal_strsplit(NULL, string, "\n", STR_EMPTY_OK);
|
||||
* ret = tal_strjoin(NULL, lines, "-- EOL\n", STR_TRAIL);
|
||||
* tal_free(lines);
|
||||
* return ret;
|
||||
* }
|
||||
*/
|
||||
#define tal_strjoin(ctx, strings, delim, flags) \
|
||||
tal_strjoin_(ctx, strings, delim, flags, TAL_LABEL(char, "[]"))
|
||||
char *tal_strjoin_(const void *ctx,
|
||||
char *strings[] TAKES,
|
||||
const char *delim TAKES,
|
||||
enum strjoin flags,
|
||||
const char *label);
|
||||
|
||||
/**
|
||||
* tal_strreg - match/extract from a string via (extended) regular expressions.
|
||||
* @ctx: the context to tal from (often NULL)
|
||||
* @string: the string to try to match (can be take())
|
||||
* @regex: the regular expression to match (can be take())
|
||||
* ...: pointers to strings to allocate for subexpressions.
|
||||
*
|
||||
* Returns true if we matched, in which case any parenthesized
|
||||
* expressions in @regex are allocated and placed in the char **
|
||||
* arguments following @regex. NULL arguments mean the match is not
|
||||
* saved. The order of the strings is the order
|
||||
* of opening braces in the expression: in the case of repeated
|
||||
* expressions (eg "([a-z])*") the last one is saved, in the case of
|
||||
* non-existent matches (eg "([a-z]*)?") the pointer is set to NULL.
|
||||
*
|
||||
* Allocation failures or malformed regular expressions return false.
|
||||
* The allocated strings will have tal_count() == strlen() + 1.
|
||||
*
|
||||
* See Also:
|
||||
* regcomp(3), regex(3).
|
||||
*
|
||||
* Example:
|
||||
* // Given "My name is Rusty" outputs "Hello Rusty!\n"
|
||||
* // Given "my first name is Rusty Russell" outputs "Hello Rusty Russell!\n"
|
||||
* // Given "My name isnt Rusty Russell" outputs "Hello there!\n"
|
||||
* int main(int argc, char *argv[])
|
||||
* {
|
||||
* char *person, *input;
|
||||
*
|
||||
* (void)argc;
|
||||
* // Join args and trim trailing space.
|
||||
* input = tal_strjoin(NULL, argv+1, " ", STR_NO_TRAIL);
|
||||
* if (tal_strreg(NULL, input,
|
||||
* "[Mm]y (first )?name is ([A-Za-z ]+)",
|
||||
* NULL, &person))
|
||||
* printf("Hello %s!\n", person);
|
||||
* else
|
||||
* printf("Hello there!\n");
|
||||
* return 0;
|
||||
* }
|
||||
*/
|
||||
#define tal_strreg(ctx, string, ...) \
|
||||
tal_strreg_(ctx, string, TAL_LABEL(char, "[]"), __VA_ARGS__)
|
||||
bool tal_strreg_(const void *ctx, const char *string TAKES,
|
||||
const char *label, const char *regex, ...);
|
||||
#endif /* CCAN_STR_TAL_H */
|
||||
972
nostrdb/ccan/ccan/tal/tal.c
Normal file
972
nostrdb/ccan/ccan/tal/tal.c
Normal file
@@ -0,0 +1,972 @@
|
||||
/* Licensed under BSD-MIT - see LICENSE file for details */
|
||||
#include "tal.h"
|
||||
#include "../compiler.h"
|
||||
#include "list.h"
|
||||
#include "alignof.h"
|
||||
|
||||
#include <assert.h>
|
||||
#include <stdio.h>
|
||||
#include <stddef.h>
|
||||
#include <string.h>
|
||||
#include <limits.h>
|
||||
#include <stdint.h>
|
||||
#include <errno.h>
|
||||
|
||||
//#define TAL_DEBUG 1
|
||||
|
||||
#define NOTIFY_IS_DESTRUCTOR 512
|
||||
#define NOTIFY_EXTRA_ARG 1024
|
||||
|
||||
/* This makes our parent_child ptr stand out for to_tal_hdr checks */
|
||||
#define TAL_PTR_OBFUSTICATOR ((intptr_t)0x1984200820142016ULL)
|
||||
|
||||
/* 32-bit type field, first byte 0 in either endianness. */
|
||||
enum prop_type {
|
||||
CHILDREN = 0x00c1d500,
|
||||
NAME = 0x00111100,
|
||||
NOTIFIER = 0x00071f00,
|
||||
};
|
||||
|
||||
struct tal_hdr {
|
||||
struct list_node list;
|
||||
struct prop_hdr *prop;
|
||||
/* XOR with TAL_PTR_OBFUSTICATOR */
|
||||
intptr_t parent_child;
|
||||
size_t bytelen;
|
||||
};
|
||||
|
||||
struct prop_hdr {
|
||||
enum prop_type type;
|
||||
struct prop_hdr *next;
|
||||
};
|
||||
|
||||
struct children {
|
||||
struct prop_hdr hdr; /* CHILDREN */
|
||||
struct tal_hdr *parent;
|
||||
struct list_head children; /* Head of siblings. */
|
||||
};
|
||||
|
||||
struct name {
|
||||
struct prop_hdr hdr; /* NAME */
|
||||
char name[];
|
||||
};
|
||||
|
||||
struct notifier {
|
||||
struct prop_hdr hdr; /* NOTIFIER */
|
||||
enum tal_notify_type types;
|
||||
union notifier_cb {
|
||||
void (*notifyfn)(tal_t *, enum tal_notify_type, void *);
|
||||
void (*destroy)(tal_t *); /* If NOTIFY_IS_DESTRUCTOR set */
|
||||
void (*destroy2)(tal_t *, void *); /* If NOTIFY_EXTRA_ARG */
|
||||
} u;
|
||||
};
|
||||
|
||||
/* Extra arg */
|
||||
struct notifier_extra_arg {
|
||||
struct notifier n;
|
||||
void *arg;
|
||||
};
|
||||
|
||||
#define EXTRA_ARG(n) (((struct notifier_extra_arg *)(n))->arg)
|
||||
|
||||
static struct {
|
||||
struct tal_hdr hdr;
|
||||
struct children c;
|
||||
} null_parent = { { { &null_parent.hdr.list, &null_parent.hdr.list },
|
||||
&null_parent.c.hdr, TAL_PTR_OBFUSTICATOR, 0 },
|
||||
{ { CHILDREN, NULL },
|
||||
&null_parent.hdr,
|
||||
{ { &null_parent.c.children.n,
|
||||
&null_parent.c.children.n } }
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
static void *(*allocfn)(size_t size) = malloc;
|
||||
static void *(*resizefn)(void *, size_t size) = realloc;
|
||||
static void (*freefn)(void *) = free;
|
||||
static void (*errorfn)(const char *msg) = (void *)abort;
|
||||
/* Count on non-destrutor notifiers; often stays zero. */
|
||||
static size_t notifiers = 0;
|
||||
|
||||
static inline void COLD call_error(const char *msg)
|
||||
{
|
||||
errorfn(msg);
|
||||
}
|
||||
|
||||
static bool get_destroying_bit(intptr_t parent_child)
|
||||
{
|
||||
return parent_child & 1;
|
||||
}
|
||||
|
||||
static void set_destroying_bit(intptr_t *parent_child)
|
||||
{
|
||||
*parent_child |= 1;
|
||||
}
|
||||
|
||||
static struct children *ignore_destroying_bit(intptr_t parent_child)
|
||||
{
|
||||
return (void *)((parent_child ^ TAL_PTR_OBFUSTICATOR) & ~(intptr_t)1);
|
||||
}
|
||||
|
||||
/* This means valgrind can see leaks. */
|
||||
void tal_cleanup(void)
|
||||
{
|
||||
struct tal_hdr *i;
|
||||
|
||||
while ((i = list_top(&null_parent.c.children, struct tal_hdr, list))) {
|
||||
list_del(&i->list);
|
||||
memset(i, 0, sizeof(*i));
|
||||
}
|
||||
|
||||
/* Cleanup any taken pointers. */
|
||||
take_cleanup();
|
||||
}
|
||||
|
||||
/* We carefully start all real properties with a zero byte. */
|
||||
static bool is_literal(const struct prop_hdr *prop)
|
||||
{
|
||||
return ((char *)prop)[0] != 0;
|
||||
}
|
||||
|
||||
#ifndef NDEBUG
|
||||
static const void *bounds_start, *bounds_end;
|
||||
|
||||
static void update_bounds(const void *new, size_t size)
|
||||
{
|
||||
if (unlikely(!bounds_start)) {
|
||||
bounds_start = new;
|
||||
bounds_end = (char *)new + size;
|
||||
} else if (new < bounds_start)
|
||||
bounds_start = new;
|
||||
else if ((char *)new + size > (char *)bounds_end)
|
||||
bounds_end = (char *)new + size;
|
||||
}
|
||||
|
||||
static bool in_bounds(const void *p)
|
||||
{
|
||||
return !p
|
||||
|| (p >= (void *)&null_parent && p <= (void *)(&null_parent + 1))
|
||||
|| (p >= bounds_start && p <= bounds_end);
|
||||
}
|
||||
#else
|
||||
static void update_bounds(const void *new, size_t size)
|
||||
{
|
||||
}
|
||||
|
||||
static bool in_bounds(const void *p)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
#endif
|
||||
|
||||
static void check_bounds(const void *p)
|
||||
{
|
||||
if (!in_bounds(p))
|
||||
call_error("Not a valid header");
|
||||
}
|
||||
|
||||
static struct tal_hdr *to_tal_hdr(const void *ctx)
|
||||
{
|
||||
struct tal_hdr *t;
|
||||
|
||||
t = (struct tal_hdr *)((char *)ctx - sizeof(struct tal_hdr));
|
||||
check_bounds(t);
|
||||
check_bounds(ignore_destroying_bit(t->parent_child));
|
||||
check_bounds(t->list.next);
|
||||
check_bounds(t->list.prev);
|
||||
if (t->prop && !is_literal(t->prop))
|
||||
check_bounds(t->prop);
|
||||
return t;
|
||||
}
|
||||
|
||||
static struct tal_hdr *to_tal_hdr_or_null(const void *ctx)
|
||||
{
|
||||
if (!ctx)
|
||||
return &null_parent.hdr;
|
||||
return to_tal_hdr(ctx);
|
||||
}
|
||||
|
||||
static void *from_tal_hdr(const struct tal_hdr *hdr)
|
||||
{
|
||||
return (void *)(hdr + 1);
|
||||
}
|
||||
|
||||
static void *from_tal_hdr_or_null(const struct tal_hdr *hdr)
|
||||
{
|
||||
if (hdr == &null_parent.hdr)
|
||||
return NULL;
|
||||
return from_tal_hdr(hdr);
|
||||
}
|
||||
|
||||
#ifdef TAL_DEBUG
|
||||
static struct tal_hdr *debug_tal(struct tal_hdr *tal)
|
||||
{
|
||||
tal_check(from_tal_hdr_or_null(tal), "TAL_DEBUG ");
|
||||
return tal;
|
||||
}
|
||||
#else
|
||||
static struct tal_hdr *debug_tal(struct tal_hdr *tal)
|
||||
{
|
||||
return tal;
|
||||
}
|
||||
#endif
|
||||
|
||||
static void notify(const struct tal_hdr *ctx,
|
||||
enum tal_notify_type type, const void *info,
|
||||
int saved_errno)
|
||||
{
|
||||
const struct prop_hdr *p;
|
||||
|
||||
for (p = ctx->prop; p; p = p->next) {
|
||||
struct notifier *n;
|
||||
|
||||
if (is_literal(p))
|
||||
break;
|
||||
if (p->type != NOTIFIER)
|
||||
continue;
|
||||
n = (struct notifier *)p;
|
||||
if (n->types & type) {
|
||||
errno = saved_errno;
|
||||
if (n->types & NOTIFY_IS_DESTRUCTOR) {
|
||||
/* Blatt this notifier in case it tries to
|
||||
* tal_del_destructor() from inside */
|
||||
union notifier_cb cb = n->u;
|
||||
/* It's a union, so this NULLs destroy2 too! */
|
||||
n->u.destroy = NULL;
|
||||
if (n->types & NOTIFY_EXTRA_ARG)
|
||||
cb.destroy2(from_tal_hdr(ctx),
|
||||
EXTRA_ARG(n));
|
||||
else
|
||||
cb.destroy(from_tal_hdr(ctx));
|
||||
} else
|
||||
n->u.notifyfn(from_tal_hdr_or_null(ctx), type,
|
||||
(void *)info);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void *allocate(size_t size)
|
||||
{
|
||||
void *ret = allocfn(size);
|
||||
if (!ret)
|
||||
call_error("allocation failed");
|
||||
else
|
||||
update_bounds(ret, size);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static struct prop_hdr **find_property_ptr(const struct tal_hdr *t,
|
||||
enum prop_type type)
|
||||
{
|
||||
struct prop_hdr **p;
|
||||
|
||||
for (p = (struct prop_hdr **)&t->prop; *p; p = &(*p)->next) {
|
||||
if (is_literal(*p)) {
|
||||
if (type == NAME)
|
||||
return p;
|
||||
break;
|
||||
}
|
||||
if ((*p)->type == type)
|
||||
return p;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void *find_property(const struct tal_hdr *parent, enum prop_type type)
|
||||
{
|
||||
struct prop_hdr **p = find_property_ptr(parent, type);
|
||||
|
||||
if (p)
|
||||
return *p;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static void init_property(struct prop_hdr *hdr,
|
||||
struct tal_hdr *parent,
|
||||
enum prop_type type)
|
||||
{
|
||||
hdr->type = type;
|
||||
hdr->next = parent->prop;
|
||||
parent->prop = hdr;
|
||||
}
|
||||
|
||||
static struct notifier *add_notifier_property(struct tal_hdr *t,
|
||||
enum tal_notify_type types,
|
||||
void (*fn)(void *,
|
||||
enum tal_notify_type,
|
||||
void *),
|
||||
void *extra_arg)
|
||||
{
|
||||
struct notifier *prop;
|
||||
|
||||
if (types & NOTIFY_EXTRA_ARG)
|
||||
prop = allocate(sizeof(struct notifier_extra_arg));
|
||||
else
|
||||
prop = allocate(sizeof(struct notifier));
|
||||
|
||||
if (prop) {
|
||||
init_property(&prop->hdr, t, NOTIFIER);
|
||||
prop->types = types;
|
||||
prop->u.notifyfn = fn;
|
||||
if (types & NOTIFY_EXTRA_ARG)
|
||||
EXTRA_ARG(prop) = extra_arg;
|
||||
}
|
||||
return prop;
|
||||
}
|
||||
|
||||
static enum tal_notify_type del_notifier_property(struct tal_hdr *t,
|
||||
void (*fn)(tal_t *,
|
||||
enum tal_notify_type,
|
||||
void *),
|
||||
bool match_extra_arg,
|
||||
void *extra_arg)
|
||||
{
|
||||
struct prop_hdr **p;
|
||||
|
||||
for (p = (struct prop_hdr **)&t->prop; *p; p = &(*p)->next) {
|
||||
struct notifier *n;
|
||||
enum tal_notify_type types;
|
||||
|
||||
if (is_literal(*p))
|
||||
break;
|
||||
if ((*p)->type != NOTIFIER)
|
||||
continue;
|
||||
n = (struct notifier *)*p;
|
||||
if (n->u.notifyfn != fn)
|
||||
continue;
|
||||
|
||||
types = n->types;
|
||||
if ((types & NOTIFY_EXTRA_ARG)
|
||||
&& match_extra_arg
|
||||
&& extra_arg != EXTRA_ARG(n))
|
||||
continue;
|
||||
|
||||
*p = (*p)->next;
|
||||
freefn(n);
|
||||
return types & ~(NOTIFY_IS_DESTRUCTOR|NOTIFY_EXTRA_ARG);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static struct name *add_name_property(struct tal_hdr *t, const char *name)
|
||||
{
|
||||
struct name *prop;
|
||||
|
||||
prop = allocate(sizeof(*prop) + strlen(name) + 1);
|
||||
if (prop) {
|
||||
init_property(&prop->hdr, t, NAME);
|
||||
strcpy(prop->name, name);
|
||||
}
|
||||
return prop;
|
||||
}
|
||||
|
||||
static struct children *add_child_property(struct tal_hdr *parent,
|
||||
struct tal_hdr *child UNNEEDED)
|
||||
{
|
||||
struct children *prop = allocate(sizeof(*prop));
|
||||
if (prop) {
|
||||
init_property(&prop->hdr, parent, CHILDREN);
|
||||
prop->parent = parent;
|
||||
list_head_init(&prop->children);
|
||||
}
|
||||
return prop;
|
||||
}
|
||||
|
||||
static bool add_child(struct tal_hdr *parent, struct tal_hdr *child)
|
||||
{
|
||||
struct children *children = find_property(parent, CHILDREN);
|
||||
|
||||
if (!children) {
|
||||
children = add_child_property(parent, child);
|
||||
if (!children)
|
||||
return false;
|
||||
}
|
||||
list_add(&children->children, &child->list);
|
||||
child->parent_child = (intptr_t)children ^ TAL_PTR_OBFUSTICATOR;
|
||||
return true;
|
||||
}
|
||||
|
||||
static void del_tree(struct tal_hdr *t, const tal_t *orig, int saved_errno)
|
||||
{
|
||||
struct prop_hdr **prop, *p, *next;
|
||||
|
||||
assert(!taken(from_tal_hdr(t)));
|
||||
|
||||
/* Already being destroyed? Don't loop. */
|
||||
if (unlikely(get_destroying_bit(t->parent_child)))
|
||||
return;
|
||||
|
||||
set_destroying_bit(&t->parent_child);
|
||||
|
||||
/* Call free notifiers. */
|
||||
notify(t, TAL_NOTIFY_FREE, (tal_t *)orig, saved_errno);
|
||||
|
||||
/* Now free children and groups. */
|
||||
prop = find_property_ptr(t, CHILDREN);
|
||||
if (prop) {
|
||||
struct tal_hdr *i;
|
||||
struct children *c = (struct children *)*prop;
|
||||
|
||||
while ((i = list_top(&c->children, struct tal_hdr, list))) {
|
||||
list_del(&i->list);
|
||||
del_tree(i, orig, saved_errno);
|
||||
}
|
||||
}
|
||||
|
||||
/* Finally free our properties. */
|
||||
for (p = t->prop; p && !is_literal(p); p = next) {
|
||||
next = p->next;
|
||||
freefn(p);
|
||||
}
|
||||
freefn(t);
|
||||
}
|
||||
|
||||
void *tal_alloc_(const tal_t *ctx, size_t size, bool clear, const char *label)
|
||||
{
|
||||
struct tal_hdr *child, *parent = debug_tal(to_tal_hdr_or_null(ctx));
|
||||
|
||||
child = allocate(sizeof(struct tal_hdr) + size);
|
||||
if (!child)
|
||||
return NULL;
|
||||
if (clear)
|
||||
memset(from_tal_hdr(child), 0, size);
|
||||
child->prop = (void *)label;
|
||||
child->bytelen = size;
|
||||
|
||||
if (!add_child(parent, child)) {
|
||||
freefn(child);
|
||||
return NULL;
|
||||
}
|
||||
debug_tal(parent);
|
||||
if (notifiers)
|
||||
notify(parent, TAL_NOTIFY_ADD_CHILD, from_tal_hdr(child), 0);
|
||||
return from_tal_hdr(debug_tal(child));
|
||||
}
|
||||
|
||||
static bool adjust_size(size_t *size, size_t count)
|
||||
{
|
||||
const size_t extra = sizeof(struct tal_hdr);
|
||||
|
||||
/* Multiplication wrap */
|
||||
if (count && unlikely(*size * count / *size != count))
|
||||
goto overflow;
|
||||
|
||||
*size *= count;
|
||||
|
||||
/* Make sure we don't wrap adding header. */
|
||||
if (*size + extra < extra)
|
||||
goto overflow;
|
||||
return true;
|
||||
overflow:
|
||||
call_error("allocation size overflow");
|
||||
return false;
|
||||
}
|
||||
|
||||
void *tal_alloc_arr_(const tal_t *ctx, size_t size, size_t count, bool clear,
|
||||
const char *label)
|
||||
{
|
||||
if (!adjust_size(&size, count))
|
||||
return NULL;
|
||||
|
||||
return tal_alloc_(ctx, size, clear, label);
|
||||
}
|
||||
|
||||
void *tal_free(const tal_t *ctx)
|
||||
{
|
||||
if (ctx) {
|
||||
struct tal_hdr *t;
|
||||
int saved_errno = errno;
|
||||
t = debug_tal(to_tal_hdr(ctx));
|
||||
if (unlikely(get_destroying_bit(t->parent_child)))
|
||||
return NULL;
|
||||
if (notifiers)
|
||||
notify(ignore_destroying_bit(t->parent_child)->parent,
|
||||
TAL_NOTIFY_DEL_CHILD, ctx, saved_errno);
|
||||
list_del(&t->list);
|
||||
del_tree(t, ctx, saved_errno);
|
||||
errno = saved_errno;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void *tal_steal_(const tal_t *new_parent, const tal_t *ctx)
|
||||
{
|
||||
if (ctx) {
|
||||
struct tal_hdr *newpar, *t, *old_parent;
|
||||
|
||||
newpar = debug_tal(to_tal_hdr_or_null(new_parent));
|
||||
t = debug_tal(to_tal_hdr(ctx));
|
||||
|
||||
/* Unlink it from old parent. */
|
||||
list_del(&t->list);
|
||||
old_parent = ignore_destroying_bit(t->parent_child)->parent;
|
||||
|
||||
if (unlikely(!add_child(newpar, t))) {
|
||||
/* We can always add to old parent, because it has a
|
||||
* children property already. */
|
||||
if (!add_child(old_parent, t))
|
||||
abort();
|
||||
return NULL;
|
||||
}
|
||||
debug_tal(newpar);
|
||||
if (notifiers)
|
||||
notify(t, TAL_NOTIFY_STEAL, new_parent, 0);
|
||||
}
|
||||
return (void *)ctx;
|
||||
}
|
||||
|
||||
bool tal_add_destructor_(const tal_t *ctx, void (*destroy)(void *me))
|
||||
{
|
||||
tal_t *t = debug_tal(to_tal_hdr(ctx));
|
||||
return add_notifier_property(t, TAL_NOTIFY_FREE|NOTIFY_IS_DESTRUCTOR,
|
||||
(void *)destroy, NULL);
|
||||
}
|
||||
|
||||
bool tal_add_destructor2_(const tal_t *ctx, void (*destroy)(void *me, void *arg),
|
||||
void *arg)
|
||||
{
|
||||
tal_t *t = debug_tal(to_tal_hdr(ctx));
|
||||
return add_notifier_property(t, TAL_NOTIFY_FREE|NOTIFY_IS_DESTRUCTOR
|
||||
|NOTIFY_EXTRA_ARG,
|
||||
(void *)destroy, arg);
|
||||
}
|
||||
|
||||
/* We could support notifiers with an extra arg, but we didn't add to API */
|
||||
bool tal_add_notifier_(const tal_t *ctx, enum tal_notify_type types,
|
||||
void (*callback)(tal_t *, enum tal_notify_type, void *))
|
||||
{
|
||||
struct tal_hdr *t = debug_tal(to_tal_hdr_or_null(ctx));
|
||||
struct notifier *n;
|
||||
|
||||
assert(types);
|
||||
assert((types & ~(TAL_NOTIFY_FREE | TAL_NOTIFY_STEAL | TAL_NOTIFY_MOVE
|
||||
| TAL_NOTIFY_RESIZE | TAL_NOTIFY_RENAME
|
||||
| TAL_NOTIFY_ADD_CHILD | TAL_NOTIFY_DEL_CHILD
|
||||
| TAL_NOTIFY_ADD_NOTIFIER
|
||||
| TAL_NOTIFY_DEL_NOTIFIER)) == 0);
|
||||
|
||||
/* Don't call notifier about itself: set types after! */
|
||||
n = add_notifier_property(t, 0, callback, NULL);
|
||||
if (unlikely(!n))
|
||||
return false;
|
||||
|
||||
if (notifiers)
|
||||
notify(t, TAL_NOTIFY_ADD_NOTIFIER, callback, 0);
|
||||
|
||||
n->types = types;
|
||||
if (types != TAL_NOTIFY_FREE)
|
||||
notifiers++;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool tal_del_notifier_(const tal_t *ctx,
|
||||
void (*callback)(tal_t *, enum tal_notify_type, void *),
|
||||
bool match_extra_arg, void *extra_arg)
|
||||
{
|
||||
struct tal_hdr *t = debug_tal(to_tal_hdr_or_null(ctx));
|
||||
enum tal_notify_type types;
|
||||
|
||||
types = del_notifier_property(t, callback, match_extra_arg, extra_arg);
|
||||
if (types) {
|
||||
notify(t, TAL_NOTIFY_DEL_NOTIFIER, callback, 0);
|
||||
if (types != TAL_NOTIFY_FREE)
|
||||
notifiers--;
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
bool tal_del_destructor_(const tal_t *ctx, void (*destroy)(void *me))
|
||||
{
|
||||
return tal_del_notifier_(ctx, (void *)destroy, false, NULL);
|
||||
}
|
||||
|
||||
bool tal_del_destructor2_(const tal_t *ctx, void (*destroy)(void *me, void *arg),
|
||||
void *arg)
|
||||
{
|
||||
return tal_del_notifier_(ctx, (void *)destroy, true, arg);
|
||||
}
|
||||
|
||||
bool tal_set_name_(tal_t *ctx, const char *name, bool literal)
|
||||
{
|
||||
struct tal_hdr *t = debug_tal(to_tal_hdr(ctx));
|
||||
struct prop_hdr **prop = find_property_ptr(t, NAME);
|
||||
|
||||
/* Get rid of any old name */
|
||||
if (prop) {
|
||||
struct name *name = (struct name *)*prop;
|
||||
if (is_literal(&name->hdr))
|
||||
*prop = NULL;
|
||||
else {
|
||||
*prop = name->hdr.next;
|
||||
freefn(name);
|
||||
}
|
||||
}
|
||||
|
||||
if (literal && name[0]) {
|
||||
struct prop_hdr **p;
|
||||
|
||||
/* Append literal. */
|
||||
for (p = &t->prop; *p && !is_literal(*p); p = &(*p)->next);
|
||||
*p = (struct prop_hdr *)name;
|
||||
} else if (!add_name_property(t, name))
|
||||
return false;
|
||||
|
||||
debug_tal(t);
|
||||
if (notifiers)
|
||||
notify(t, TAL_NOTIFY_RENAME, name, 0);
|
||||
return true;
|
||||
}
|
||||
|
||||
const char *tal_name(const tal_t *t)
|
||||
{
|
||||
struct name *n;
|
||||
|
||||
n = find_property(debug_tal(to_tal_hdr(t)), NAME);
|
||||
if (!n)
|
||||
return NULL;
|
||||
|
||||
if (is_literal(&n->hdr))
|
||||
return (const char *)n;
|
||||
return n->name;
|
||||
}
|
||||
|
||||
size_t tal_bytelen(const tal_t *ptr)
|
||||
{
|
||||
/* NULL -> null_parent which has bytelen 0 */
|
||||
struct tal_hdr *t = debug_tal(to_tal_hdr_or_null(ptr));
|
||||
|
||||
return t->bytelen;
|
||||
}
|
||||
|
||||
/* Start one past first child: make stopping natural in circ. list. */
|
||||
static struct tal_hdr *first_child(struct tal_hdr *parent)
|
||||
{
|
||||
struct children *child;
|
||||
|
||||
child = find_property(parent, CHILDREN);
|
||||
if (!child)
|
||||
return NULL;
|
||||
|
||||
return list_top(&child->children, struct tal_hdr, list);
|
||||
}
|
||||
|
||||
tal_t *tal_first(const tal_t *root)
|
||||
{
|
||||
struct tal_hdr *c, *t = debug_tal(to_tal_hdr_or_null(root));
|
||||
|
||||
c = first_child(t);
|
||||
if (!c)
|
||||
return NULL;
|
||||
return from_tal_hdr(c);
|
||||
}
|
||||
|
||||
tal_t *tal_next(const tal_t *prev)
|
||||
{
|
||||
struct tal_hdr *next, *prevhdr = debug_tal(to_tal_hdr(prev));
|
||||
struct list_head *head;
|
||||
|
||||
head = &ignore_destroying_bit(prevhdr->parent_child)->children;
|
||||
next = list_next(head, prevhdr, list);
|
||||
if (!next)
|
||||
return NULL;
|
||||
return from_tal_hdr(next);
|
||||
}
|
||||
|
||||
tal_t *tal_parent(const tal_t *ctx)
|
||||
{
|
||||
struct tal_hdr *t;
|
||||
|
||||
if (!ctx)
|
||||
return NULL;
|
||||
|
||||
t = debug_tal(to_tal_hdr(ctx));
|
||||
if (ignore_destroying_bit(t->parent_child)->parent == &null_parent.hdr)
|
||||
return NULL;
|
||||
return from_tal_hdr(ignore_destroying_bit(t->parent_child)->parent);
|
||||
}
|
||||
|
||||
bool tal_resize_(tal_t **ctxp, size_t size, size_t count, bool clear)
|
||||
{
|
||||
struct tal_hdr *old_t, *t;
|
||||
struct children *child;
|
||||
|
||||
old_t = debug_tal(to_tal_hdr(*ctxp));
|
||||
|
||||
if (!adjust_size(&size, count))
|
||||
return false;
|
||||
|
||||
t = resizefn(old_t, sizeof(struct tal_hdr) + size);
|
||||
if (!t) {
|
||||
call_error("Reallocation failure");
|
||||
return false;
|
||||
}
|
||||
|
||||
/* Clear between old end and new end. */
|
||||
if (clear && size > t->bytelen) {
|
||||
char *old_end = (char *)(t + 1) + t->bytelen;
|
||||
memset(old_end, 0, size - t->bytelen);
|
||||
}
|
||||
|
||||
/* Update length. */
|
||||
t->bytelen = size;
|
||||
update_bounds(t, sizeof(struct tal_hdr) + size);
|
||||
|
||||
/* If it didn't move, we're done! */
|
||||
if (t != old_t) {
|
||||
/* Fix up linked list pointers. */
|
||||
t->list.next->prev = t->list.prev->next = &t->list;
|
||||
|
||||
/* Copy take() property. */
|
||||
if (taken(from_tal_hdr(old_t)))
|
||||
take(from_tal_hdr(t));
|
||||
|
||||
/* Fix up child property's parent pointer. */
|
||||
child = find_property(t, CHILDREN);
|
||||
if (child) {
|
||||
assert(child->parent == old_t);
|
||||
child->parent = t;
|
||||
}
|
||||
*ctxp = from_tal_hdr(debug_tal(t));
|
||||
if (notifiers)
|
||||
notify(t, TAL_NOTIFY_MOVE, from_tal_hdr(old_t), 0);
|
||||
}
|
||||
if (notifiers)
|
||||
notify(t, TAL_NOTIFY_RESIZE, (void *)size, 0);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool tal_expand_(tal_t **ctxp, const void *src, size_t size, size_t count)
|
||||
{
|
||||
size_t old_len;
|
||||
bool ret = false;
|
||||
|
||||
old_len = debug_tal(to_tal_hdr(*ctxp))->bytelen;
|
||||
|
||||
/* Check for additive overflow */
|
||||
if (old_len + count * size < old_len) {
|
||||
call_error("dup size overflow");
|
||||
goto out;
|
||||
}
|
||||
|
||||
/* Don't point src inside thing we're expanding! */
|
||||
assert(src < *ctxp
|
||||
|| (char *)src >= (char *)(*ctxp) + old_len);
|
||||
|
||||
if (!tal_resize_(ctxp, size, old_len/size + count, false))
|
||||
goto out;
|
||||
|
||||
memcpy((char *)*ctxp + old_len, src, count * size);
|
||||
ret = true;
|
||||
|
||||
out:
|
||||
if (taken(src))
|
||||
tal_free(src);
|
||||
return ret;
|
||||
}
|
||||
|
||||
void *tal_dup_(const tal_t *ctx, const void *p, size_t size,
|
||||
size_t n, size_t extra, bool nullok, const char *label)
|
||||
{
|
||||
void *ret;
|
||||
size_t nbytes = size;
|
||||
|
||||
if (nullok && p == NULL) {
|
||||
/* take(NULL) works. */
|
||||
(void)taken(p);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (!adjust_size(&nbytes, n)) {
|
||||
if (taken(p))
|
||||
tal_free(p);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* Beware addition overflow! */
|
||||
if (n + extra < n) {
|
||||
call_error("dup size overflow");
|
||||
if (taken(p))
|
||||
tal_free(p);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (taken(p)) {
|
||||
if (unlikely(!p))
|
||||
return NULL;
|
||||
if (unlikely(!tal_resize_((void **)&p, size, n + extra, false)))
|
||||
return tal_free(p);
|
||||
if (unlikely(!tal_steal(ctx, p)))
|
||||
return tal_free(p);
|
||||
return (void *)p;
|
||||
}
|
||||
|
||||
ret = tal_alloc_arr_(ctx, size, n + extra, false, label);
|
||||
if (ret)
|
||||
memcpy(ret, p, nbytes);
|
||||
return ret;
|
||||
}
|
||||
|
||||
void *tal_dup_talarr_(const tal_t *ctx, const tal_t *src TAKES, const char *label)
|
||||
{
|
||||
return tal_dup_(ctx, src, 1, tal_bytelen(src), 0, true, label);
|
||||
}
|
||||
|
||||
void tal_set_backend(void *(*alloc_fn)(size_t size),
|
||||
void *(*resize_fn)(void *, size_t size),
|
||||
void (*free_fn)(void *),
|
||||
void (*error_fn)(const char *msg))
|
||||
{
|
||||
if (alloc_fn)
|
||||
allocfn = alloc_fn;
|
||||
if (resize_fn)
|
||||
resizefn = resize_fn;
|
||||
if (free_fn)
|
||||
freefn = free_fn;
|
||||
if (error_fn)
|
||||
errorfn = error_fn;
|
||||
}
|
||||
|
||||
#ifdef CCAN_TAL_DEBUG
|
||||
static void dump_node(unsigned int indent, const struct tal_hdr *t)
|
||||
{
|
||||
unsigned int i;
|
||||
const struct prop_hdr *p;
|
||||
|
||||
for (i = 0; i < indent; i++)
|
||||
fprintf(stderr, " ");
|
||||
fprintf(stderr, "%p len=%zu", t, t->bytelen);
|
||||
for (p = t->prop; p; p = p->next) {
|
||||
struct children *c;
|
||||
struct name *n;
|
||||
struct notifier *no;
|
||||
if (is_literal(p)) {
|
||||
fprintf(stderr, " \"%s\"", (const char *)p);
|
||||
break;
|
||||
}
|
||||
switch (p->type) {
|
||||
case CHILDREN:
|
||||
c = (struct children *)p;
|
||||
fprintf(stderr, " CHILDREN(%p):parent=%p,children={%p,%p}",
|
||||
p, c->parent,
|
||||
c->children.n.prev, c->children.n.next);
|
||||
break;
|
||||
case NAME:
|
||||
n = (struct name *)p;
|
||||
fprintf(stderr, " NAME(%p):%s", p, n->name);
|
||||
break;
|
||||
case NOTIFIER:
|
||||
no = (struct notifier *)p;
|
||||
fprintf(stderr, " NOTIFIER(%p):fn=%p", p, no->u.notifyfn);
|
||||
break;
|
||||
default:
|
||||
fprintf(stderr, " **UNKNOWN(%p):%i**", p, p->type);
|
||||
}
|
||||
}
|
||||
fprintf(stderr, "\n");
|
||||
}
|
||||
|
||||
static void tal_dump_(unsigned int level, const struct tal_hdr *t)
|
||||
{
|
||||
struct children *children;
|
||||
|
||||
dump_node(level, t);
|
||||
|
||||
children = find_property(t, CHILDREN);
|
||||
if (children) {
|
||||
struct tal_hdr *i;
|
||||
|
||||
list_for_each(&children->children, i, list)
|
||||
tal_dump_(level + 1, i);
|
||||
}
|
||||
}
|
||||
|
||||
void tal_dump(void)
|
||||
{
|
||||
tal_dump_(0, &null_parent.hdr);
|
||||
}
|
||||
#endif /* CCAN_TAL_DEBUG */
|
||||
|
||||
#ifndef NDEBUG
|
||||
static bool check_err(struct tal_hdr *t, const char *errorstr,
|
||||
const char *errmsg)
|
||||
{
|
||||
if (errorstr) {
|
||||
/* Try not to malloc: it may be corrupted. */
|
||||
char msg[strlen(errorstr) + 20 + strlen(errmsg) + 1];
|
||||
sprintf(msg, "%s:%p %s", errorstr, from_tal_hdr(t), errmsg);
|
||||
call_error(msg);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool check_node(struct children *parent_child,
|
||||
struct tal_hdr *t, const char *errorstr)
|
||||
{
|
||||
struct prop_hdr *p;
|
||||
struct name *name = NULL;
|
||||
struct children *children = NULL;
|
||||
|
||||
if (!in_bounds(t))
|
||||
return check_err(t, errorstr, "invalid pointer");
|
||||
|
||||
if (ignore_destroying_bit(t->parent_child) != parent_child)
|
||||
return check_err(t, errorstr, "incorrect parent");
|
||||
|
||||
for (p = t->prop; p; p = p->next) {
|
||||
if (is_literal(p)) {
|
||||
if (name)
|
||||
return check_err(t, errorstr,
|
||||
"has extra literal");
|
||||
break;
|
||||
}
|
||||
if (!in_bounds(p))
|
||||
return check_err(t, errorstr,
|
||||
"has bad property pointer");
|
||||
|
||||
switch (p->type) {
|
||||
case CHILDREN:
|
||||
if (children)
|
||||
return check_err(t, errorstr,
|
||||
"has two child nodes");
|
||||
children = (struct children *)p;
|
||||
break;
|
||||
case NOTIFIER:
|
||||
break;
|
||||
case NAME:
|
||||
if (name)
|
||||
return check_err(t, errorstr,
|
||||
"has two names");
|
||||
name = (struct name *)p;
|
||||
break;
|
||||
default:
|
||||
return check_err(t, errorstr, "has unknown property");
|
||||
}
|
||||
}
|
||||
if (children) {
|
||||
struct tal_hdr *i;
|
||||
|
||||
if (!list_check(&children->children, errorstr))
|
||||
return false;
|
||||
list_for_each(&children->children, i, list) {
|
||||
if (!check_node(children, i, errorstr))
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool tal_check(const tal_t *ctx, const char *errorstr)
|
||||
{
|
||||
struct tal_hdr *t = to_tal_hdr_or_null(ctx);
|
||||
|
||||
return check_node(ignore_destroying_bit(t->parent_child), t, errorstr);
|
||||
}
|
||||
#else /* NDEBUG */
|
||||
bool tal_check(const tal_t *ctx, const char *errorstr)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
#endif
|
||||
553
nostrdb/ccan/ccan/tal/tal.h
Normal file
553
nostrdb/ccan/ccan/tal/tal.h
Normal file
@@ -0,0 +1,553 @@
|
||||
/* Licensed under BSD-MIT - see LICENSE file for details */
|
||||
#ifndef CCAN_TAL_H
|
||||
#define CCAN_TAL_H
|
||||
#include "../config.h"
|
||||
#include "../compiler.h"
|
||||
#include "likely.h"
|
||||
#include "typesafe_cb.h"
|
||||
#include "str.h"
|
||||
#include "take.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdarg.h>
|
||||
|
||||
/**
|
||||
* tal_t - convenient alias for void to mark tal pointers.
|
||||
*
|
||||
* Since any pointer can be a tal-allocated pointer, it's often
|
||||
* useful to use this typedef to mark them explicitly.
|
||||
*/
|
||||
typedef void tal_t;
|
||||
|
||||
/**
|
||||
* tal - basic allocator function
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @type: the type to allocate.
|
||||
*
|
||||
* Allocates a specific type, with a given parent context. The name
|
||||
* of the object is a string of the type, but if CCAN_TAL_DEBUG is
|
||||
* defined it also contains the file and line which allocated it.
|
||||
*
|
||||
* tal_count() of the return will be 1.
|
||||
*
|
||||
* Example:
|
||||
* int *p = tal(NULL, int);
|
||||
* *p = 1;
|
||||
*/
|
||||
#define tal(ctx, type) \
|
||||
tal_label(ctx, type, TAL_LABEL(type, ""))
|
||||
|
||||
/**
|
||||
* talz - zeroing allocator function
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @type: the type to allocate.
|
||||
*
|
||||
* Equivalent to tal() followed by memset() to zero.
|
||||
*
|
||||
* Example:
|
||||
* p = talz(NULL, int);
|
||||
* assert(*p == 0);
|
||||
*/
|
||||
#define talz(ctx, type) \
|
||||
talz_label(ctx, type, TAL_LABEL(type, ""))
|
||||
|
||||
/**
|
||||
* tal_free - free a tal-allocated pointer.
|
||||
* @p: NULL, or tal allocated object to free.
|
||||
*
|
||||
* This calls the destructors for p (if any), then does the same for all its
|
||||
* children (recursively) before finally freeing the memory. It returns
|
||||
* NULL, for convenience.
|
||||
*
|
||||
* Note: errno is preserved by this call, and also saved and restored
|
||||
* for any destructors or notifiers.
|
||||
*
|
||||
* Example:
|
||||
* p = tal_free(p);
|
||||
*/
|
||||
void *tal_free(const tal_t *p);
|
||||
|
||||
/**
|
||||
* tal_arr - allocate an array of objects.
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @type: the type to allocate.
|
||||
* @count: the number to allocate.
|
||||
*
|
||||
* tal_count() of the returned pointer will be @count.
|
||||
*
|
||||
* Example:
|
||||
* p = tal_arr(NULL, int, 2);
|
||||
* p[0] = 0;
|
||||
* p[1] = 1;
|
||||
*/
|
||||
#define tal_arr(ctx, type, count) \
|
||||
tal_arr_label(ctx, type, count, TAL_LABEL(type, "[]"))
|
||||
|
||||
/**
|
||||
* tal_arrz - allocate an array of zeroed objects.
|
||||
* @ctx: NULL, or tal allocated object to be parent.
|
||||
* @type: the type to allocate.
|
||||
* @count: the number to allocate.
|
||||
*
|
||||
* Equivalent to tal_arr() followed by memset() to zero.
|
||||
*
|
||||
* Example:
|
||||
* p = tal_arrz(NULL, int, 2);
|
||||
* assert(p[0] == 0 && p[1] == 0);
|
||||
*/
|
||||
#define tal_arrz(ctx, type, count) \
|
||||
tal_arrz_label(ctx, type, count, TAL_LABEL(type, "[]"))
|
||||
|
||||
/**
|
||||
* tal_resize - enlarge or reduce a tal object.
|
||||
* @p: A pointer to the tal allocated array to resize.
|
||||
* @count: the number to allocate.
|
||||
*
|
||||
* This returns true on success (and may move *@p), or false on failure.
|
||||
* On success, tal_count() of *@p will be @count.
|
||||
*
|
||||
* Note: if *p is take(), it will still be take() upon return, even if it
|
||||
* has been moved.
|
||||
*
|
||||
* Example:
|
||||
* tal_resize(&p, 100);
|
||||
*/
|
||||
#define tal_resize(p, count) \
|
||||
tal_resize_((void **)(p), sizeof**(p), (count), false)
|
||||
|
||||
/**
|
||||
* tal_resizez - enlarge or reduce a tal object; zero out extra.
|
||||
* @p: A pointer to the tal allocated array to resize.
|
||||
* @count: the number to allocate.
|
||||
*
|
||||
* This returns true on success (and may move *@p), or false on failure.
|
||||
*
|
||||
* Example:
|
||||
* tal_resizez(&p, 200);
|
||||
*/
|
||||
#define tal_resizez(p, count) \
|
||||
tal_resize_((void **)(p), sizeof**(p), (count), true)
|
||||
|
||||
/**
|
||||
* tal_steal - change the parent of a tal-allocated pointer.
|
||||
* @ctx: The new parent.
|
||||
* @ptr: The tal allocated object to move, or NULL.
|
||||
*
|
||||
* This may need to perform an allocation, in which case it may fail; thus
|
||||
* it can return NULL, otherwise returns @ptr. If @ptr is NULL, this function does
|
||||
* nothing.
|
||||
*/
|
||||
#if HAVE_STATEMENT_EXPR
|
||||
/* Weird macro avoids gcc's 'warning: value computed is not used'. */
|
||||
#define tal_steal(ctx, ptr) \
|
||||
({ (tal_typeof(ptr) tal_steal_((ctx),(ptr))); })
|
||||
#else
|
||||
#define tal_steal(ctx, ptr) \
|
||||
(tal_typeof(ptr) tal_steal_((ctx),(ptr)))
|
||||
#endif
|
||||
|
||||
/**
|
||||
* tal_add_destructor - add a callback function when this context is destroyed.
|
||||
* @ptr: The tal allocated object.
|
||||
* @function: the function to call before it's freed.
|
||||
*
|
||||
* This is a more convenient form of tal_add_notifier(@ptr,
|
||||
* TAL_NOTIFY_FREE, ...), in that the function prototype takes only @ptr.
|
||||
*
|
||||
* Note that this can only fail if your allocfn fails and your errorfn returns.
|
||||
*/
|
||||
#define tal_add_destructor(ptr, function) \
|
||||
tal_add_destructor_((ptr), typesafe_cb(void, void *, (function), (ptr)))
|
||||
|
||||
/**
|
||||
* tal_del_destructor - remove a destructor callback function.
|
||||
* @ptr: The tal allocated object.
|
||||
* @function: the function to call before it's freed.
|
||||
*
|
||||
* If @function has not been successfully added as a destructor, this returns
|
||||
* false. Note that if we're inside the destructor call itself, this will
|
||||
* return false.
|
||||
*/
|
||||
#define tal_del_destructor(ptr, function) \
|
||||
tal_del_destructor_((ptr), typesafe_cb(void, void *, (function), (ptr)))
|
||||
|
||||
/**
|
||||
* tal_add_destructor2 - add a 2-arg callback function when context is destroyed.
|
||||
* @ptr: The tal allocated object.
|
||||
* @function: the function to call before it's freed.
|
||||
* @arg: the extra argument to the function.
|
||||
*
|
||||
* Sometimes an extra argument is required for a destructor; this
|
||||
* saves the extra argument internally to avoid the caller having to
|
||||
* do an extra allocation.
|
||||
*
|
||||
* Note that this can only fail if your allocfn fails and your errorfn returns.
|
||||
*/
|
||||
#define tal_add_destructor2(ptr, function, arg) \
|
||||
tal_add_destructor2_((ptr), \
|
||||
typesafe_cb_cast(void (*)(tal_t *, void *), \
|
||||
void (*)(__typeof__(ptr), \
|
||||
__typeof__(arg)), \
|
||||
(function)), \
|
||||
(arg))
|
||||
|
||||
/**
|
||||
* tal_del_destructor - remove a destructor callback function.
|
||||
* @ptr: The tal allocated object.
|
||||
* @function: the function to call before it's freed.
|
||||
*
|
||||
* If @function has not been successfully added as a destructor, this returns
|
||||
* false. Note that if we're inside the destructor call itself, this will
|
||||
* return false.
|
||||
*/
|
||||
#define tal_del_destructor(ptr, function) \
|
||||
tal_del_destructor_((ptr), typesafe_cb(void, void *, (function), (ptr)))
|
||||
|
||||
/**
|
||||
* tal_del_destructor2 - remove 2-arg callback function.
|
||||
* @ptr: The tal allocated object.
|
||||
* @function: the function to call before it's freed.
|
||||
* @arg: the extra argument to the function.
|
||||
*
|
||||
* If @function has not been successfully added as a destructor with
|
||||
* @arg, this returns false.
|
||||
*/
|
||||
#define tal_del_destructor2(ptr, function, arg) \
|
||||
tal_del_destructor2_((ptr), \
|
||||
typesafe_cb_cast(void (*)(tal_t *, void *), \
|
||||
void (*)(__typeof__(ptr), \
|
||||
__typeof__(arg)), \
|
||||
(function)), \
|
||||
(arg))
|
||||
enum tal_notify_type {
|
||||
TAL_NOTIFY_FREE = 1,
|
||||
TAL_NOTIFY_STEAL = 2,
|
||||
TAL_NOTIFY_MOVE = 4,
|
||||
TAL_NOTIFY_RESIZE = 8,
|
||||
TAL_NOTIFY_RENAME = 16,
|
||||
TAL_NOTIFY_ADD_CHILD = 32,
|
||||
TAL_NOTIFY_DEL_CHILD = 64,
|
||||
TAL_NOTIFY_ADD_NOTIFIER = 128,
|
||||
TAL_NOTIFY_DEL_NOTIFIER = 256
|
||||
};
|
||||
|
||||
/**
|
||||
* tal_add_notifier - add a callback function when this context changes.
|
||||
* @ptr: The tal allocated object, or NULL.
|
||||
* @types: Bitwise OR of the types the callback is interested in.
|
||||
* @callback: the function to call.
|
||||
*
|
||||
* Note that this can only fail if your allocfn fails and your errorfn
|
||||
* returns. Also note that notifiers are not reliable in the case
|
||||
* where an allocation fails, as they may be called before any
|
||||
* allocation is actually done.
|
||||
*
|
||||
* TAL_NOTIFY_FREE is called when @ptr is freed, either directly or
|
||||
* because an ancestor is freed: @info is the argument to tal_free().
|
||||
* It is exactly equivalent to a destructor, with more information.
|
||||
* errno is set to the value it was at the call of tal_free().
|
||||
*
|
||||
* TAL_NOTIFY_STEAL is called when @ptr's parent changes: @info is the
|
||||
* new parent.
|
||||
*
|
||||
* TAL_NOTIFY_MOVE is called when @ptr is realloced (via tal_resize)
|
||||
* and moved. In this case, @ptr arg here is the new memory, and
|
||||
* @info is the old pointer.
|
||||
*
|
||||
* TAL_NOTIFY_RESIZE is called when @ptr is realloced via tal_resize:
|
||||
* @info is the new size, in bytes. If the pointer has moved,
|
||||
* TAL_NOTIFY_MOVE callbacks are called first.
|
||||
*
|
||||
* TAL_NOTIFY_ADD_CHILD/TAL_NOTIFY_DEL_CHILD are called when @ptr is
|
||||
* the context for a tal() allocating call, or a direct child is
|
||||
* tal_free()d: @info is the child. Note that TAL_NOTIFY_DEL_CHILD is
|
||||
* not called when this context is tal_free()d: TAL_NOTIFY_FREE is
|
||||
* considered sufficient for that case.
|
||||
*
|
||||
* TAL_NOTIFY_ADD_NOTIFIER/TAL_NOTIFIER_DEL_NOTIFIER are called when a
|
||||
* notifier is added or removed (not for this notifier): @info is the
|
||||
* callback. This is also called for tal_add_destructor and
|
||||
* tal_del_destructor.
|
||||
*/
|
||||
#define tal_add_notifier(ptr, types, callback) \
|
||||
tal_add_notifier_((ptr), (types), \
|
||||
typesafe_cb_postargs(void, tal_t *, (callback), \
|
||||
(ptr), \
|
||||
enum tal_notify_type, void *))
|
||||
|
||||
/**
|
||||
* tal_del_notifier - remove a notifier callback function.
|
||||
* @ptr: The tal allocated object.
|
||||
* @callback: the function to call.
|
||||
*/
|
||||
#define tal_del_notifier(ptr, callback) \
|
||||
tal_del_notifier_((ptr), \
|
||||
typesafe_cb_postargs(void, void *, (callback), \
|
||||
(ptr), \
|
||||
enum tal_notify_type, void *), \
|
||||
false, NULL)
|
||||
|
||||
/**
|
||||
* tal_set_name - attach a name to a tal pointer.
|
||||
* @ptr: The tal allocated object.
|
||||
* @name: The name to use.
|
||||
*
|
||||
* The name is copied, unless we're certain it's a string literal.
|
||||
*/
|
||||
#define tal_set_name(ptr, name) \
|
||||
tal_set_name_((ptr), (name), TAL_IS_LITERAL(name))
|
||||
|
||||
/**
|
||||
* tal_name - get the name for a tal pointer.
|
||||
* @ptr: The tal allocated object.
|
||||
*
|
||||
* Returns NULL if no name has been set.
|
||||
*/
|
||||
const char *tal_name(const tal_t *ptr);
|
||||
|
||||
/**
|
||||
* tal_count - get the count of objects in a tal object.
|
||||
* @ptr: The tal allocated object (or NULL)
|
||||
*
|
||||
* Returns 0 if @ptr is NULL. Note that if the allocation was done as a
|
||||
* different type to @ptr, the result may not match the @count argument
|
||||
* (or implied 1) of that allocation!
|
||||
*/
|
||||
#define tal_count(p) (tal_bytelen(p) / sizeof(*p))
|
||||
|
||||
/**
|
||||
* tal_bytelen - get the count of bytes in a tal object.
|
||||
* @ptr: The tal allocated object (or NULL)
|
||||
*
|
||||
* Returns 0 if @ptr is NULL.
|
||||
*/
|
||||
size_t tal_bytelen(const tal_t *ptr);
|
||||
|
||||
/**
|
||||
* tal_first - get the first immediate tal object child.
|
||||
* @root: The tal allocated object to start with, or NULL.
|
||||
*
|
||||
* Returns NULL if there are no children.
|
||||
*/
|
||||
tal_t *tal_first(const tal_t *root);
|
||||
|
||||
/**
|
||||
* tal_next - get the next immediate tal object child.
|
||||
* @prev: The return value from tal_first or tal_next.
|
||||
*
|
||||
* Returns NULL if there are no more immediate children. This should be safe to
|
||||
* call on an altering tree unless @prev is no longer valid.
|
||||
*/
|
||||
tal_t *tal_next(const tal_t *prev);
|
||||
|
||||
/**
|
||||
* tal_parent - get the parent of a tal object.
|
||||
* @ctx: The tal allocated object.
|
||||
*
|
||||
* Returns the parent, which may be NULL. Returns NULL if @ctx is NULL.
|
||||
*/
|
||||
tal_t *tal_parent(const tal_t *ctx);
|
||||
|
||||
/**
|
||||
* tal_dup - duplicate an object.
|
||||
* @ctx: The tal allocated object to be parent of the result (may be NULL).
|
||||
* @type: the type (should match type of @p!)
|
||||
* @p: the object to copy (or reparented if take()). Must not be NULL.
|
||||
*/
|
||||
#define tal_dup(ctx, type, p) \
|
||||
tal_dup_label(ctx, type, p, TAL_LABEL(type, ""), false)
|
||||
|
||||
/**
|
||||
* tal_dup_or_null - duplicate an object, or just pass NULL.
|
||||
* @ctx: The tal allocated object to be parent of the result (may be NULL).
|
||||
* @type: the type (should match type of @p!)
|
||||
* @p: the object to copy (or reparented if take())
|
||||
*
|
||||
* if @p is NULL, just return NULL, otherwise to tal_dup().
|
||||
*/
|
||||
#define tal_dup_or_null(ctx, type, p) \
|
||||
tal_dup_label(ctx, type, p, TAL_LABEL(type, ""), true)
|
||||
|
||||
/**
|
||||
* tal_dup_arr - duplicate an array.
|
||||
* @ctx: The tal allocated object to be parent of the result (may be NULL).
|
||||
* @type: the type (should match type of @p!)
|
||||
* @p: the array to copy (or resized & reparented if take())
|
||||
* @n: the number of sizeof(type) entries to copy.
|
||||
* @extra: the number of extra sizeof(type) entries to allocate.
|
||||
*/
|
||||
#define tal_dup_arr(ctx, type, p, n, extra) \
|
||||
tal_dup_arr_label(ctx, type, p, n, extra, TAL_LABEL(type, "[]"))
|
||||
|
||||
|
||||
/**
|
||||
* tal_dup_arr - duplicate a tal array.
|
||||
* @ctx: The tal allocated object to be parent of the result (may be NULL).
|
||||
* @type: the type (should match type of @p!)
|
||||
* @p: the tal array to copy (or resized & reparented if take())
|
||||
*
|
||||
* The common case of duplicating an entire tal array.
|
||||
*/
|
||||
#define tal_dup_talarr(ctx, type, p) \
|
||||
((type *)tal_dup_talarr_((ctx), tal_typechk_(p, type *), \
|
||||
TAL_LABEL(type, "[]")))
|
||||
/* Lower-level interfaces, where you want to supply your own label string. */
|
||||
#define tal_label(ctx, type, label) \
|
||||
((type *)tal_alloc_((ctx), sizeof(type), false, label))
|
||||
#define talz_label(ctx, type, label) \
|
||||
((type *)tal_alloc_((ctx), sizeof(type), true, label))
|
||||
#define tal_arr_label(ctx, type, count, label) \
|
||||
((type *)tal_alloc_arr_((ctx), sizeof(type), (count), false, label))
|
||||
#define tal_arrz_label(ctx, type, count, label) \
|
||||
((type *)tal_alloc_arr_((ctx), sizeof(type), (count), true, label))
|
||||
#define tal_dup_label(ctx, type, p, label, nullok) \
|
||||
((type *)tal_dup_((ctx), tal_typechk_(p, type *), \
|
||||
sizeof(type), 1, 0, nullok, \
|
||||
label))
|
||||
#define tal_dup_arr_label(ctx, type, p, n, extra, label) \
|
||||
((type *)tal_dup_((ctx), tal_typechk_(p, type *), \
|
||||
sizeof(type), (n), (extra), false, \
|
||||
label))
|
||||
|
||||
/**
|
||||
* tal_set_backend - set the allocation or error functions to use
|
||||
* @alloc_fn: allocator or NULL (default is malloc)
|
||||
* @resize_fn: re-allocator or NULL (default is realloc)
|
||||
* @free_fn: free function or NULL (default is free)
|
||||
* @error_fn: called on errors or NULL (default is abort)
|
||||
*
|
||||
* The defaults are set up so tal functions never return NULL, but you
|
||||
* can override error_fn to change that. error_fn can return, and is
|
||||
* called if alloc_fn or resize_fn fail.
|
||||
*
|
||||
* If any parameter is NULL, that function is unchanged.
|
||||
*/
|
||||
void tal_set_backend(void *(*alloc_fn)(size_t size),
|
||||
void *(*resize_fn)(void *, size_t size),
|
||||
void (*free_fn)(void *),
|
||||
void (*error_fn)(const char *msg));
|
||||
|
||||
/**
|
||||
* tal_expand - expand a tal array with contents.
|
||||
* @a1p: a pointer to the tal array to expand.
|
||||
* @a2: the second array (can be take()).
|
||||
* @num2: the number of elements in the second array.
|
||||
*
|
||||
* Note that *@a1 and @a2 should be the same type. tal_count(@a1) will
|
||||
* be increased by @num2.
|
||||
*
|
||||
* Example:
|
||||
* int *arr1 = tal_arrz(NULL, int, 2);
|
||||
* int arr2[2] = { 1, 3 };
|
||||
*
|
||||
* tal_expand(&arr1, arr2, 2);
|
||||
* assert(tal_count(arr1) == 4);
|
||||
* assert(arr1[2] == 1);
|
||||
* assert(arr1[3] == 3);
|
||||
*/
|
||||
#define tal_expand(a1p, a2, num2) \
|
||||
tal_expand_((void **)(a1p), (a2), sizeof**(a1p), \
|
||||
(num2) + 0*sizeof(*(a1p) == (a2)))
|
||||
|
||||
/**
|
||||
* tal_cleanup - remove pointers from NULL node
|
||||
*
|
||||
* Internally, tal keeps a list of nodes allocated from @ctx NULL; this
|
||||
* prevents valgrind from noticing memory leaks. This re-initializes
|
||||
* that list to empty.
|
||||
*
|
||||
* It also calls take_cleanup() for you.
|
||||
*/
|
||||
void tal_cleanup(void);
|
||||
|
||||
|
||||
/**
|
||||
* tal_check - sanity check a tal context and its children.
|
||||
* @ctx: a tal context, or NULL.
|
||||
* @errorstr: a string to prepend calls to error_fn, or NULL.
|
||||
*
|
||||
* This sanity-checks a tal tree (unless NDEBUG is defined, in which case
|
||||
* it simply returns true). If errorstr is not null, error_fn is called
|
||||
* when a problem is found, otherwise it is not.
|
||||
*
|
||||
* See also:
|
||||
* tal_set_backend()
|
||||
*/
|
||||
bool tal_check(const tal_t *ctx, const char *errorstr);
|
||||
|
||||
#ifdef CCAN_TAL_DEBUG
|
||||
/**
|
||||
* tal_dump - dump entire tal tree to stderr.
|
||||
*
|
||||
* This is a helper for debugging tal itself, which dumps all the tal internal
|
||||
* state.
|
||||
*/
|
||||
void tal_dump(void);
|
||||
#endif
|
||||
|
||||
/* Internal support functions */
|
||||
#ifndef TAL_LABEL
|
||||
#ifdef CCAN_TAL_NO_LABELS
|
||||
#define TAL_LABEL(type, arr) NULL
|
||||
#else
|
||||
#ifdef CCAN_TAL_DEBUG
|
||||
#define TAL_LABEL(type, arr) \
|
||||
__FILE__ ":" stringify(__LINE__) ":" stringify(type) arr
|
||||
#else
|
||||
#define TAL_LABEL(type, arr) stringify(type) arr
|
||||
#endif /* CCAN_TAL_DEBUG */
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#if HAVE_BUILTIN_CONSTANT_P
|
||||
#define TAL_IS_LITERAL(str) __builtin_constant_p(str)
|
||||
#else
|
||||
#define TAL_IS_LITERAL(str) (sizeof(&*(str)) != sizeof(char *))
|
||||
#endif
|
||||
|
||||
bool tal_set_name_(tal_t *ctx, const char *name, bool literal);
|
||||
|
||||
#if HAVE_TYPEOF
|
||||
#define tal_typeof(ptr) (__typeof__(ptr))
|
||||
#if HAVE_STATEMENT_EXPR
|
||||
/* Careful: ptr can be const foo *, ptype is foo *. Also, ptr could
|
||||
* be an array, eg "hello". */
|
||||
#define tal_typechk_(ptr, ptype) ({ __typeof__((ptr)+0) _p = (ptype)(ptr); _p; })
|
||||
#else
|
||||
#define tal_typechk_(ptr, ptype) (ptr)
|
||||
#endif
|
||||
#else /* !HAVE_TYPEOF */
|
||||
#define tal_typeof(ptr)
|
||||
#define tal_typechk_(ptr, ptype) (ptr)
|
||||
#endif
|
||||
|
||||
void *tal_alloc_(const tal_t *ctx, size_t bytes, bool clear, const char *label);
|
||||
void *tal_alloc_arr_(const tal_t *ctx, size_t bytes, size_t count, bool clear,
|
||||
const char *label);
|
||||
|
||||
void *tal_dup_(const tal_t *ctx, const void *p TAKES, size_t size,
|
||||
size_t n, size_t extra, bool nullok, const char *label);
|
||||
void *tal_dup_talarr_(const tal_t *ctx, const tal_t *src TAKES,
|
||||
const char *label);
|
||||
|
||||
tal_t *tal_steal_(const tal_t *new_parent, const tal_t *t);
|
||||
|
||||
bool tal_resize_(tal_t **ctxp, size_t size, size_t count, bool clear);
|
||||
bool tal_expand_(tal_t **ctxp, const void *src TAKES, size_t size, size_t count);
|
||||
|
||||
bool tal_add_destructor_(const tal_t *ctx, void (*destroy)(void *me));
|
||||
bool tal_add_destructor2_(const tal_t *ctx, void (*destroy)(void *me, void *arg),
|
||||
void *arg);
|
||||
bool tal_del_destructor_(const tal_t *ctx, void (*destroy)(void *me));
|
||||
bool tal_del_destructor2_(const tal_t *ctx, void (*destroy)(void *me, void *arg),
|
||||
void *arg);
|
||||
|
||||
bool tal_add_notifier_(const tal_t *ctx, enum tal_notify_type types,
|
||||
void (*notify)(tal_t *ctx, enum tal_notify_type,
|
||||
void *info));
|
||||
bool tal_del_notifier_(const tal_t *ctx,
|
||||
void (*notify)(tal_t *ctx, enum tal_notify_type,
|
||||
void *info),
|
||||
bool match_extra_arg, void *arg);
|
||||
#endif /* CCAN_TAL_H */
|
||||
Reference in New Issue
Block a user