Skip to content

Commit 0acdd65

Browse files
author
Chandra Pratap
committed
fuzz-tests: Add a test for codex32 operations
Changelog-None: Add a test for `codex32_encode()` and `codex32_secret_decode()` defined in `common/codex32.{c, h}`.
1 parent c50bd38 commit 0acdd65

File tree

2 files changed

+199
-0
lines changed

2 files changed

+199
-0
lines changed

tests/fuzz/Makefile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ FUZZ_COMMON_OBJS := \
2929
common/close_tx.o \
3030
common/configdir.o \
3131
common/configvar.o \
32+
common/codex32.o \
3233
common/channel_id.o \
3334
common/channel_type.o \
3435
common/cryptomsg.o \

tests/fuzz/fuzz-codex32.c

Lines changed: 198 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,198 @@
1+
#include "config.h"
2+
#include <assert.h>
3+
#include <ccan/ccan/array_size/array_size.h>
4+
#include <ccan/ccan/tal/str/str.h>
5+
#include <common/setup.h>
6+
#include <common/bech32.h>
7+
#include <common/utils.h>
8+
#include <common/codex32.h>
9+
#include <tests/fuzz/libfuzz.h>
10+
11+
/* Default mutator defined by libFuzzer */
12+
size_t LLVMFuzzerMutate(uint8_t *data, size_t size, size_t max_size);
13+
size_t LLVMFuzzerCustomMutator(uint8_t *fuzz_data, size_t size, size_t max_size,
14+
unsigned int seed);
15+
size_t LLVMFuzzerCustomCrossOver(const u8 *in1, size_t in1_size, const u8 *in2,
16+
size_t in2_size, u8 *out, size_t max_out_size,
17+
unsigned seed);
18+
19+
/* Duplicate codex32 structure */
20+
static struct codex32 *codex32_dup(const tal_t *ctx, const struct codex32 *src)
21+
{
22+
struct codex32 *dup = tal(ctx, struct codex32);
23+
dup->hrp = tal_strdup(dup, src->hrp);
24+
dup->threshold = src->threshold;
25+
memcpy(dup->id, src->id, sizeof(dup->id));
26+
dup->share_idx = src->share_idx;
27+
dup->payload = tal_dup_arr(dup, u8, src->payload,
28+
tal_bytelen(src->payload), 0);
29+
dup->type = src->type;
30+
return dup;
31+
}
32+
33+
void init(int *argc, char ***argv)
34+
{
35+
common_setup("fuzzer");
36+
}
37+
38+
/* Custom mutator with structure-aware and byte-level mutations */
39+
size_t LLVMFuzzerCustomMutator(uint8_t *fuzz_data, size_t size,
40+
size_t max_size, unsigned int seed)
41+
{
42+
srand(seed);
43+
char *str = to_string(tmpctx, fuzz_data, size);
44+
char *fail;
45+
struct codex32 *parts = codex32_decode(tmpctx, NULL, str, &fail);
46+
47+
/* If valid, try structure-aware mutation */
48+
if (parts) {
49+
/* Mutate a random component */
50+
switch(rand() % 4) {
51+
case 0: /* Mutate HRP arbitrarily */
52+
{
53+
size_t hrp_len = strlen(parts->hrp);
54+
size_t max_hrp_len = 10; /* Reasonable max */
55+
char *new_hrp = tal_arr(parts, char, max_hrp_len + 1);
56+
size_t new_hrp_len = LLVMFuzzerMutate((u8 *)new_hrp,
57+
hrp_len, max_hrp_len);
58+
new_hrp[new_hrp_len] = '\0';
59+
parts->hrp = new_hrp;
60+
}
61+
break;
62+
63+
case 1: /* Mutate threshold to any value */
64+
parts->threshold = rand();
65+
break;
66+
67+
case 2: /* Mutate ID arbitrarily */
68+
{
69+
size_t id_len = sizeof(parts->id) - 1;
70+
LLVMFuzzerMutate((u8 *)parts->id, id_len, id_len);
71+
parts->id[id_len] = '\0';
72+
}
73+
break;
74+
75+
case 3: /* Mutate payload */
76+
{
77+
size_t old_size = tal_bytelen(parts->payload);
78+
tal_resize(&parts->payload, max_size);
79+
size_t new_size = LLVMFuzzerMutate((u8 *)parts->payload, old_size, max_size);
80+
tal_resize(&parts->payload, new_size);
81+
}
82+
break;
83+
}
84+
85+
if (strlen(parts->hrp) == 2) {
86+
char *reencoded;
87+
const char *err = codex32_secret_encode(tmpctx, parts->hrp, parts->id,
88+
parts->threshold, parts->payload,
89+
tal_bytelen(parts->payload), &reencoded);
90+
if (!err) {
91+
size_t len = tal_bytelen(reencoded) - 1;
92+
if (len <= max_size) {
93+
memcpy(fuzz_data, reencoded, len);
94+
return len;
95+
}
96+
}
97+
}
98+
}
99+
100+
/* Fallback: byte-level mutation */
101+
return LLVMFuzzerMutate(fuzz_data, size, max_size);
102+
}
103+
104+
/* Custom crossover with structure-aware recombination */
105+
size_t LLVMFuzzerCustomCrossOver(const u8 *in1, size_t in1_size, const u8 *in2, size_t in2_size,
106+
u8 *out, size_t max_out_size, unsigned seed)
107+
{
108+
srand(seed);
109+
char *str1 = to_string(tmpctx, in1, in1_size);
110+
char *str2 = to_string(tmpctx, in2, in2_size);
111+
char *fail;
112+
113+
/* Decode both inputs */
114+
struct codex32 *p1 = codex32_decode(tmpctx, NULL, str1, &fail);
115+
struct codex32 *p2 = codex32_decode(tmpctx, NULL, str2, &fail);
116+
117+
/* If both valid, try structure-aware crossover */
118+
if (p1 && p2) {
119+
/* Create child by combining parts */
120+
struct codex32 *child = codex32_dup(tmpctx, p1);
121+
122+
/* Choose crossover method */
123+
switch(rand() % 4) {
124+
case 0: /* Crossover HRP */
125+
{
126+
size_t hrp1_len = strlen(p1->hrp);
127+
size_t hrp2_len = strlen(p2->hrp);
128+
char *new_hrp = tal_arr(child, char, max_out_size);
129+
size_t new_hrp_len = cross_over((const u8 *)p1->hrp, hrp1_len,
130+
(const u8 *)p2->hrp, hrp2_len,
131+
(u8 *)new_hrp, max_out_size, rand());
132+
new_hrp[new_hrp_len] = '\0';
133+
child->hrp = new_hrp;
134+
}
135+
break;
136+
137+
case 1: /* Crossover threshold */
138+
child->threshold = p2->threshold;
139+
break;
140+
141+
case 2: /* Crossover ID */
142+
{
143+
size_t id_len = sizeof(p1->id) - 1;
144+
cross_over((const u8 *)p1->id, id_len, (const u8 *)p2->id, id_len,
145+
(u8 *)child->id, id_len, rand());
146+
child->id[id_len] = '\0';
147+
}
148+
break;
149+
150+
case 3: /* Crossover payload */
151+
{
152+
size_t p1_len = tal_bytelen(p1->payload);
153+
size_t p2_len = tal_bytelen(p2->payload);
154+
tal_resize(&child->payload, max_out_size);
155+
size_t new_payload_len = cross_over(p1->payload, p1_len,
156+
p2->payload, p2_len,
157+
(u8 *)child->payload, max_out_size, rand());
158+
tal_resize(&child->payload, new_payload_len);
159+
}
160+
break;
161+
}
162+
163+
if (strlen(child->hrp) == 2) {
164+
char *reencoded;
165+
const char *err = codex32_secret_encode(tmpctx, child->hrp, child->id,
166+
child->threshold, child->payload,
167+
tal_bytelen(child->payload), &reencoded);
168+
if (!err) {
169+
size_t len = strlen(reencoded);
170+
if (len <= max_out_size) {
171+
memcpy(out, reencoded, len);
172+
return len;
173+
}
174+
}
175+
}
176+
}
177+
178+
/* Fallback: byte-level crossover */
179+
return cross_over(in1, in1_size, in2, in2_size, out, max_out_size, seed);
180+
}
181+
182+
void run(const uint8_t *data, size_t size)
183+
{
184+
struct codex32 *c32;
185+
char *str, *fail, *bip93;
186+
187+
str = to_string(tmpctx, data, size);
188+
189+
c32 = codex32_decode(tmpctx, NULL, str, &fail);
190+
if (c32) {
191+
const char *ret = codex32_secret_encode(tmpctx, c32->hrp, c32->id, c32->threshold,
192+
c32->payload, tal_bytelen(c32->payload), &bip93);
193+
assert(!ret && bip93);
194+
} else
195+
assert(fail);
196+
197+
clean_tmpctx();
198+
}

0 commit comments

Comments
 (0)