b6a1c97c41
Currently, the test only verifies the 5-to-8 bit conversion. Replace it with a roundtrip check that verifies 8-to-5 bit conversion as well.
80 lines
2.4 KiB
C
80 lines
2.4 KiB
C
#include "config.h"
|
|
#include <assert.h>
|
|
#include <common/utils.h>
|
|
#include <common/bech32.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <tests/fuzz/libfuzz.h>
|
|
|
|
void init(int *argc, char ***argv)
|
|
{
|
|
}
|
|
|
|
void run(const uint8_t *data, size_t size)
|
|
{
|
|
const char hrp_inv[5] = "lnbc\0", hrp_addr[3] = "bc\0";
|
|
char *bech32_str, *hrp_out, *addr;
|
|
uint8_t *data_out;
|
|
size_t data_out_len, bech32_str_cap;
|
|
int decode_wit_version;
|
|
bech32_encoding benc, benc_decoded;
|
|
|
|
if (size < 1)
|
|
return;
|
|
|
|
/* Buffer size is defined in each function's doc comment. */
|
|
benc = data[0] ? BECH32_ENCODING_BECH32 : BECH32_ENCODING_BECH32M;
|
|
bech32_str_cap = (size - 1) + strlen(hrp_inv) + 8;
|
|
bech32_str = tal_arr(tmpctx, char, bech32_str_cap);
|
|
if (bech32_encode(bech32_str, hrp_inv, data + 1, size - 1,
|
|
bech32_str_cap, benc) == 1) {
|
|
hrp_out = tal_arr(tmpctx, char, strlen(bech32_str) - 6);
|
|
data_out = tal_arr(tmpctx, uint8_t, strlen(bech32_str) - 8);
|
|
|
|
benc_decoded = bech32_decode(hrp_out, data_out, &data_out_len,
|
|
bech32_str, bech32_str_cap);
|
|
assert(benc_decoded == benc);
|
|
assert(strcmp(hrp_inv, hrp_out) == 0);
|
|
assert(data_out_len == size - 1);
|
|
assert(memcmp(data_out, data + 1, data_out_len) == 0);
|
|
}
|
|
|
|
/* Convert data to 5-bit values (0-31) */
|
|
u8 *five_bit_data = tal_dup_arr(tmpctx, u8, data, size, 0);
|
|
for (size_t i = 0; i < size; i++)
|
|
five_bit_data[i] &= 0x1F;
|
|
|
|
u8 *eight_bit_data = tal_arr(tmpctx, u8, size);
|
|
size_t eight_bit_len = 0;
|
|
/* Convert 5-to-8 without padding */
|
|
if (bech32_convert_bits(eight_bit_data, &eight_bit_len, 8,
|
|
five_bit_data, size, 5, 0)) {
|
|
u8 *five_bit_deconv = tal_arr(tmpctx, u8, size);
|
|
size_t five_bit_deconv_len = 0;
|
|
/* Convert 8-to-5 with padding */
|
|
if (bech32_convert_bits(five_bit_deconv, &five_bit_deconv_len, 5,
|
|
eight_bit_data, eight_bit_len, 8, 1)) {
|
|
assert(five_bit_deconv_len == size);
|
|
assert(memcmp(five_bit_data, five_bit_deconv, five_bit_deconv_len) == 0);
|
|
}
|
|
}
|
|
|
|
data_out = tal_arr(tmpctx, uint8_t, size);
|
|
data_out_len = 0;
|
|
|
|
addr = tal_arr(tmpctx, char, 73 + strlen(hrp_addr));
|
|
for (int wit_version = 0; wit_version <= 16; ++wit_version) {
|
|
if (segwit_addr_encode(addr, hrp_addr, wit_version, data,
|
|
size) == 0)
|
|
continue;
|
|
|
|
assert(segwit_addr_decode(&decode_wit_version, data_out,
|
|
&data_out_len, hrp_addr, addr) == 1);
|
|
assert(decode_wit_version == wit_version);
|
|
assert(data_out_len == size);
|
|
assert(memcmp(data_out, data, data_out_len) == 0);
|
|
}
|
|
|
|
clean_tmpctx();
|
|
}
|