mirror of
https://github.com/ElementsProject/lightning.git
synced 2025-01-03 20:44:54 +01:00
cb9debc229
Trying to rework the TLV streams to have a more homogenous interface to work with. This is by no means a complete implementation, just the groundwork that is going to be used by the wire code generator to generate the specific accessors, but it's enough so we can manipulate TLV streams in the onion and later just switch to the generated ones.
108 lines
2.4 KiB
C
108 lines
2.4 KiB
C
#include <common/bigsize.h>
|
|
#include <wire/tlvstream.h>
|
|
#include <wire/wire.h>
|
|
|
|
void towire_tlvstream_raw(u8 **pptr, const struct tlv_field *fields)
|
|
{
|
|
if (!fields)
|
|
return;
|
|
|
|
for (size_t i = 0; i < tal_count(fields); i++) {
|
|
const struct tlv_field *field = &fields[i];
|
|
/* BOLT #1:
|
|
*
|
|
* The sending node:
|
|
...
|
|
* - MUST minimally encode `type` and `length`.
|
|
*/
|
|
towire_bigsize(pptr, field->numtype);
|
|
towire_bigsize(pptr, field->length);
|
|
towire(pptr, field->value, field->length);
|
|
}
|
|
}
|
|
|
|
void tlvstream_set_raw(struct tlv_field **stream, u64 type, u8 *value TAKES)
|
|
{
|
|
struct tlv_field f;
|
|
f.length = tal_bytelen(value);
|
|
f.numtype = type;
|
|
f.value = tal_dup_arr(*stream, u8, value, f.length, 0);
|
|
tal_arr_expand(stream, f);
|
|
}
|
|
|
|
void tlvstream_set_short_channel_id(struct tlv_field **stream, u64 type,
|
|
struct short_channel_id *value)
|
|
{
|
|
u8 *ser = tal_arr(NULL, u8, 0);
|
|
towire_short_channel_id(&ser, value);
|
|
tlvstream_set_raw(stream, type, take(ser));
|
|
}
|
|
|
|
void tlvstream_set_tu64(struct tlv_field **stream, u64 type, u64 value)
|
|
{
|
|
u8 *ser = tal_arr(NULL, u8, 0);
|
|
towire_tu64(&ser, value);
|
|
tlvstream_set_raw(stream, type, take(ser));
|
|
}
|
|
|
|
void tlvstream_set_tu32(struct tlv_field **stream, u64 type, u32 value)
|
|
{
|
|
u8 *ser = tal_arr(NULL, u8, 0);
|
|
towire_tu64(&ser, value);
|
|
tlvstream_set_raw(stream, type, take(ser));
|
|
}
|
|
|
|
static struct tlv_field *tlvstream_get_raw(struct tlv_field *stream, u64 type)
|
|
{
|
|
for (size_t i=0; i<tal_count(stream); i++)
|
|
if (stream[i].numtype == type)
|
|
return &stream[i];
|
|
return NULL;
|
|
}
|
|
|
|
|
|
bool tlvstream_get_short_channel_id(struct tlv_field *stream, u64 type,
|
|
struct short_channel_id *value)
|
|
{
|
|
struct tlv_field *raw = tlvstream_get_raw(stream, type);
|
|
const u8 *v;
|
|
size_t max;
|
|
if (raw == NULL || raw->length != 8)
|
|
return false;
|
|
|
|
max = raw->length;
|
|
v = raw->value;
|
|
fromwire_short_channel_id(&v, &max, value);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool tlvstream_get_tu64(struct tlv_field *stream, u64 type, u64 *value)
|
|
{
|
|
struct tlv_field *raw = tlvstream_get_raw(stream, type);
|
|
const u8 *v;
|
|
size_t max;
|
|
if (raw == NULL || raw->length != 8)
|
|
return false;
|
|
|
|
max = raw->length;
|
|
v = raw->value;
|
|
*value = fromwire_tu64(&v, &max);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool tlvstream_get_tu32(struct tlv_field *stream, u64 type, u32 *value)
|
|
{
|
|
struct tlv_field *raw = tlvstream_get_raw(stream, type);
|
|
const u8 *v;
|
|
size_t max;
|
|
if (raw == NULL || raw->length != 8)
|
|
return false;
|
|
|
|
max = raw->length;
|
|
v = raw->value;
|
|
*value = fromwire_tu64(&v, &max);
|
|
return true;
|
|
}
|