core-lightning/common/bigsize.c
Rusty Russell 93d04d08d0 wire: update to latest version of the spec.
The main change here is that the previously-optional open/accept
fields and reestablish fields are now compulsory (everyone was
including them anyway).  In fact, the open/accept is a TLV
because it was actually the same format.

For more details, see lightning-rfc/f068dd0d8dfa5ae75feedd99f269e23be4777381

Changelog-Removed: protocol: support for optioned form of reestablish messages now compulsory.
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
2020-06-23 18:49:15 +02:00

125 lines
2.2 KiB
C

#include <assert.h>
#include <common/bigsize.h>
#include <wire/wire.h>
#ifndef SUPERVERBOSE
#define SUPERVERBOSE(...)
#endif
size_t bigsize_len(bigsize_t v)
{
if (v < 0xfd) {
return 1;
} else if (v <= 0xffff) {
return 3;
} else if (v <= 0xffffffff) {
return 5;
} else {
return 9;
}
}
size_t bigsize_put(u8 buf[BIGSIZE_MAX_LEN], bigsize_t v)
{
u8 *p = buf;
if (v < 0xfd) {
*(p++) = v;
} else if (v <= 0xffff) {
(*p++) = 0xfd;
(*p++) = v >> 8;
(*p++) = v;
} else if (v <= 0xffffffff) {
(*p++) = 0xfe;
(*p++) = v >> 24;
(*p++) = v >> 16;
(*p++) = v >> 8;
(*p++) = v;
} else {
(*p++) = 0xff;
(*p++) = v >> 56;
(*p++) = v >> 48;
(*p++) = v >> 40;
(*p++) = v >> 32;
(*p++) = v >> 24;
(*p++) = v >> 16;
(*p++) = v >> 8;
(*p++) = v;
}
return p - buf;
}
size_t bigsize_get(const u8 *p, size_t max, bigsize_t *val)
{
if (max < 1) {
SUPERVERBOSE("EOF");
return 0;
}
switch (*p) {
case 0xfd:
if (max < 3) {
SUPERVERBOSE("unexpected EOF");
return 0;
}
*val = ((u64)p[1] << 8) + p[2];
if (*val < 0xfd) {
SUPERVERBOSE("decoded bigsize is not canonical");
return 0;
}
return 3;
case 0xfe:
if (max < 5) {
SUPERVERBOSE("unexpected EOF");
return 0;
}
*val = ((u64)p[1] << 24) + ((u64)p[2] << 16)
+ ((u64)p[3] << 8) + p[4];
if ((*val >> 16) == 0) {
SUPERVERBOSE("decoded bigsize is not canonical");
return 0;
}
return 5;
case 0xff:
if (max < 9) {
SUPERVERBOSE("unexpected EOF");
return 0;
}
*val = ((u64)p[1] << 56) + ((u64)p[2] << 48)
+ ((u64)p[3] << 40) + ((u64)p[4] << 32)
+ ((u64)p[5] << 24) + ((u64)p[6] << 16)
+ ((u64)p[7] << 8) + p[8];
if ((*val >> 32) == 0) {
SUPERVERBOSE("decoded bigsize is not canonical");
return 0;
}
return 9;
default:
*val = *p;
return 1;
}
}
bigsize_t fromwire_bigsize(const u8 **cursor, size_t *max)
{
bigsize_t v;
size_t len = bigsize_get(*cursor, *max, &v);
if (len == 0) {
fromwire_fail(cursor, max);
return 0;
}
assert(len <= *max);
fromwire(cursor, max, NULL, len);
return v;
}
void towire_bigsize(u8 **pptr, const bigsize_t val)
{
u8 buf[BIGSIZE_MAX_LEN];
size_t len;
len = bigsize_put(buf, val);
towire(pptr, buf, len);
}