mirror of
https://github.com/ElementsProject/lightning.git
synced 2024-12-27 09:04:40 +01:00
65be18d355
Things allocated by libwally all get the tal_name "wally_tal", which cost me a few hours trying to find a leak. In the case where we're making one of the allocations the parent of the others (e.g. a wally_psbt), we can do better: supply a name for the tal_wally_end(). So I add a new tal_wally_end_onto() which does the standard tal_steal() trick, and also changes the (typechecked!) name. Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
160 lines
5.3 KiB
C
160 lines
5.3 KiB
C
#ifndef LIGHTNING_COMMON_UTILS_H
|
|
#define LIGHTNING_COMMON_UTILS_H
|
|
#include "config.h"
|
|
#include <ccan/build_assert/build_assert.h>
|
|
#include <ccan/crypto/ripemd160/ripemd160.h>
|
|
#include <ccan/crypto/sha256/sha256.h>
|
|
#include <ccan/short_types/short_types.h>
|
|
#include <ccan/structeq/structeq.h>
|
|
#include <ccan/tal/tal.h>
|
|
#include <secp256k1.h>
|
|
|
|
extern secp256k1_context *secp256k1_ctx;
|
|
|
|
extern const struct chainparams *chainparams;
|
|
|
|
/* Unsigned min/max macros: BUILD_ASSERT make sure types are unsigned */
|
|
#if HAVE_TYPEOF
|
|
#define MUST_BE_UNSIGNED_INT(x) BUILD_ASSERT_OR_ZERO((typeof(x))(-1)>=0)
|
|
#else
|
|
#define MUST_BE_UNSIGNED_INT(x) 0
|
|
#endif
|
|
|
|
#define min_unsigned(a, b) \
|
|
(MUST_BE_UNSIGNED_INT(a) + MUST_BE_UNSIGNED_INT(b) + min_u64((a), (b)))
|
|
|
|
#define max_unsigned(a, b) \
|
|
(MUST_BE_UNSIGNED_INT(a) + MUST_BE_UNSIGNED_INT(b) + max_u64((a), (b)))
|
|
|
|
static inline u64 min_u64(u64 a, u64 b)
|
|
{
|
|
return a < b ? a : b;
|
|
}
|
|
|
|
static inline u64 max_u64(u64 a, u64 b)
|
|
{
|
|
return a < b ? b : a;
|
|
}
|
|
|
|
/* Marker which indicates an (tal) pointer argument is stolen
|
|
* (i.e. eventually freed) by the function. Unlike TAKEN, which
|
|
* indicates it's only stolen if caller says take() */
|
|
#define STEALS
|
|
|
|
/* Simple accessor function for our own dependencies to use, in order to avoid
|
|
* circular dependencies (should only be used in `bitcoin/y`). */
|
|
bool is_elements(const struct chainparams *chainparams);
|
|
|
|
/* Allocate and fill in a hex-encoded string of this data. */
|
|
char *tal_hexstr(const tal_t *ctx, const void *data, size_t len);
|
|
|
|
/* Allocate and fill a hex-encoding of this tal pointer. */
|
|
char *tal_hex(const tal_t *ctx, const tal_t *data);
|
|
|
|
/* Allocate and fill a buffer with the data of this hex string. */
|
|
u8 *tal_hexdata(const tal_t *ctx, const void *str, size_t len);
|
|
|
|
/* Macro to set memberptr in tal object outer to point to tal object obj,
|
|
* if it isn't NULL.
|
|
* The 0*sizeof() checks that *memberptr = obj is valid */
|
|
#define set_softref(outer, memberptr, obj) \
|
|
set_softref_((outer), sizeof(*(outer)) + 0*sizeof(*(memberptr) = obj), \
|
|
(void **)(memberptr), (obj))
|
|
|
|
/* Macro to clear a (set) softref ptr to NULL */
|
|
#define clear_softref(outer, memberptr) \
|
|
clear_softref_((outer), sizeof(*(outer)), (void **)(memberptr))
|
|
|
|
void set_softref_(const tal_t *outer, size_t outersize, void **ptr, tal_t *obj);
|
|
void clear_softref_(const tal_t *outer, size_t outersize, void **ptr);
|
|
|
|
/* Note: p is never a complex expression, otherwise this multi-evaluates! */
|
|
#define tal_arr_expand(p, s) \
|
|
do { \
|
|
size_t n_ = tal_count(*(p)); \
|
|
tal_resize((p), n_+1); \
|
|
(*(p))[n_] = (s); \
|
|
} while(0)
|
|
|
|
/**
|
|
* Remove an element from an array
|
|
*
|
|
* This will shift the elements past the removed element, changing
|
|
* their position in memory, so only use this for arrays of pointers.
|
|
*/
|
|
#define tal_arr_remove(p, n) tal_arr_remove_((p), sizeof(**p), (n))
|
|
void tal_arr_remove_(void *p, size_t elemsize, size_t n);
|
|
|
|
/* Check for valid UTF-8 */
|
|
bool utf8_check(const void *buf, size_t buflen);
|
|
|
|
/* Check it's UTF-8, return copy (or same if TAKES), or NULL if not valid. */
|
|
char *utf8_str(const tal_t *ctx, const u8 *buf TAKES, size_t buflen);
|
|
|
|
/* Use the POSIX C locale. */
|
|
void setup_locale(void);
|
|
|
|
/* Global temporary convenience context: children freed in io loop core. */
|
|
extern const tal_t *tmpctx;
|
|
|
|
/* Initial creation of tmpctx. */
|
|
void setup_tmpctx(void);
|
|
|
|
/* Free any children of tmpctx. */
|
|
void clean_tmpctx(void);
|
|
|
|
/* Call this before any libwally function which allocates. */
|
|
void tal_wally_start(void);
|
|
|
|
/* Then call this to reparent everything onto this parent */
|
|
void tal_wally_end(const tal_t *parent);
|
|
|
|
/* ... or this if you want to reparent onto something which is
|
|
* allocated by libwally here. Fixes up this from_wally obj to have a
|
|
* proper tal_name, too! */
|
|
#define tal_wally_end_onto(parent, from_wally, type) \
|
|
tal_wally_end_onto_((parent), \
|
|
(from_wally) + 0*sizeof((from_wally) == (type *)0), \
|
|
stringify(type))
|
|
void tal_wally_end_onto_(const tal_t *parent,
|
|
tal_t *from_wally,
|
|
const char *from_wally_name);
|
|
|
|
/* Define sha256_eq. */
|
|
STRUCTEQ_DEF(sha256, 0, u);
|
|
|
|
/* Define ripemd160_eq. */
|
|
STRUCTEQ_DEF(ripemd160, 0, u);
|
|
|
|
/* If gcc complains about 'may be uninitialized' even at -O3, and the code is
|
|
* clear, use this to suppress it. Argument should be gcc version it
|
|
* complained on, so we can re-test as gcc evolves. */
|
|
#define COMPILER_WANTS_INIT(compiler_versions) = 0
|
|
|
|
/* For case where we want one thing if DEVELOPER, another if not, particularly
|
|
* for function parameters.
|
|
*
|
|
* Usefully, you can refer to DEVELOPER-only fields here. */
|
|
#if DEVELOPER
|
|
/* Make sure that nondev is evaluated, and valid */
|
|
#define IFDEV(dev, nondev) ((void)(nondev), (dev))
|
|
#else
|
|
#define IFDEV(dev, nondev) (nondev)
|
|
#endif
|
|
|
|
#if EXPERIMENTAL_FEATURES
|
|
/* Make sure that nondev is evaluated, and valid, but is a constant */
|
|
#define IFEXPERIMENTAL(exp, nonexp) (0 ? (nonexp) : (exp))
|
|
#else
|
|
#define IFEXPERIMENTAL(exp, nonexp) (nonexp)
|
|
#endif
|
|
|
|
/* Context which all wally allocations use (see common/setup.c) */
|
|
extern const tal_t *wally_tal_ctx;
|
|
|
|
/* Like mkstemp but resolves template relative to $TMPDIR (or /tmp if unset).
|
|
* Returns created temporary path name at *created if successful. */
|
|
int tmpdir_mkstemp(const tal_t *ctx, const char *template TAKES, char **created);
|
|
|
|
#endif /* LIGHTNING_COMMON_UTILS_H */
|