#ifndef LIGHTNING_COMMON_UTILS_H
#define LIGHTNING_COMMON_UTILS_H
#include "config.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>

struct membuf;
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);

/**
 * mlock_tal_memory - lock a tal-allocated memory with sodium_mlock.
 * @ptr - the tal-allocated memory to lock
 *
 * This is a generic function to replace the pattern of sodium_mlock + tal_add_destructor.
 *
 * Aborts on failure (memory locking is mandatory for security).
 */
void mlock_tal_memory(const tal_t *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)

/* Checks if two tal_arr are equal: sizeof checks types are the same */
#define tal_arr_eq(a, b) tal_arr_eq_((a), (b), sizeof((a) == (b)))

/* Avoids double-evaluation in macro */
bool tal_arr_eq_(const void *a, const void *b, size_t unused);

/**
 * 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 simple arrays.
 */
#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);

/**
 * Insert an element in an array
 */
#define tal_arr_insert(p, n, v) \
	do {								\
		size_t n_ = tal_count(*(p));				\
		tal_resize((p), n_+1);					\
		memmove(*(p) + n + 1, *(p) + n, (n_ - n) * sizeof(**(p))); \
		(*(p))[n] = (v);					\
	} while(0)

/* 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);

/* Strdup, or pass through NULL */
char *tal_strdup_or_null(const tal_t *ctx, const char *str);

/* 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.
 *
 * This method should *not* be used for creation of psbt objects,
 * tal_wally_end_onto should be used instead!
 *
 * Typical usage is to end on the PSBT being modified.
 */
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!
 *
 * This results in a tal hierarchy like so:
 * parent -> from_wally -> [everything wally allocated]
 *
 * This is typically used when allocating a new psbt and ensuring all the psbt's
 * member objects are children of the PSBT itself.
 *
 * For example:
 * struct wally_psbt *clone;
 * tal_wally_start();
 * if (wally_psbt_clone_alloc(psbt, 0, &clone) != WALLY_OK)
 * 	abort();
 * tal_wally_end_onto(ctx, clone, struct wally_psbt);
 */
#define tal_wally_end_onto(parent, from_wally, type)                           \
	tal_wally_end_onto_(                                                   \
	    (parent), (from_wally),                                            \
	    &stringify(type)[0 * sizeof((from_wally) == (type *)0)])
void tal_wally_end_onto_(const tal_t *parent,
			 tal_t *from_wally,
			 const char *from_wally_name);

/* ... or this if libwally only used temporary allocations. */
void tal_wally_discard(void);

/* 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

/* 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);

/* For use with membuf_init */
void *membuf_tal_resize(struct membuf *mb, void *rawelems, size_t newsize);

/**
 * tal_strlowering - return the same string by in lower case.
 * @ctx: the context to tal from (often NULL)
 * @string: the string that is going to be lowered (can be take())
 *
 * FIXME: move this in ccan
 */
char *str_lowering(const void *ctx, const char *string TAKES);

/**
 * tal_struppering - return the same string by in upper case.
 * @ctx: the context to tal from (often NULL)
 * @string: the string that is going to be UPPERED (can be take())
 *
 * FIXME: move this in ccan
 */
char *str_uppering(const void *ctx, const char *string TAKES);

/**
 * Assign two different structs which are the same size.
 * We use this for assigning secrets <-> sha256 for example.
 */
#define CROSS_TYPE_ASSIGNMENT(dst, src)					\
	memcpy((dst), (src),						\
	       sizeof(*(dst)) + BUILD_ASSERT_OR_ZERO(sizeof(*(dst)) == sizeof(*(src))))

/**
 * Compare two different structs which are the same size.
 * We use this for comparing bitcoin_txid <-> sha256 for example.
 */
#define CROSS_TYPE_EQ(a, b)						\
	(memcmp((a), (b),						\
		sizeof(*(a)) + BUILD_ASSERT_OR_ZERO(sizeof(*(a)) == sizeof(*(b)))) == 0)

#endif /* LIGHTNING_COMMON_UTILS_H */
