fuzz-tests: Add a test for the onion message handler

tests/fuzz/Makefile

@@ -9,6 +9,16 @@ tests/fuzz/fuzz-hmac-sha256: LDLIBS += -lcrypto
 tests/fuzz/fuzz-wire-*.o: tests/fuzz/wire.h
 tests/fuzz/fuzz-bolt12-*.o: tests/fuzz/bolt12.h
 
+tests/fuzz/fuzz-handle_onion_message:	common/sphinx.o				\
+					common/blindedpath.o			\
+					common/hmac.o				\
+					common/blinding.o			\
+					common/onionreply.o			\
+					common/dev_disconnect.o			\
+					common/onion_message_parse.o		\
+	                                connectd/onion_message.o		\
+					connectd/connectd_wiregen.o
+
 FUZZ_TARGETS_SRC := $(wildcard tests/fuzz/fuzz-*.c)
 FUZZ_TARGETS_OBJS := $(FUZZ_TARGETS_SRC:.c=.o)
 FUZZ_TARGETS_BIN := $(FUZZ_TARGETS_SRC:.c=)

tests/fuzz/corpora/fuzz-handle_onion_message/a294a6a9839b43adb866e3583a5dbfd2cb75c9f0

@@ -0,0 +1 @@
+
ÕÕõÕ

tests/fuzz/corpora/fuzz-handle_onion_message/bf8b4530d8d246dd74ac53a13471bba17941dff7

@@ -0,0 +1 @@
+

tests/fuzz/corpora/fuzz-handle_onion_message/d46ab8e2c7abdbd2b04fd2f8787eb2196d98ebe4

@@ -0,0 +1 @@
+
ÿÿÿÿÿÿ

tests/fuzz/corpora/fuzz-handle_onion_message/fa5d81c0adfec72bf5fbe705b756cb40e40de91e

@@ -0,0 +1 @@
+
ƒÕÕõÕá

tests/fuzz/fuzz-handle_onion_message.c

@@ -0,0 +1,105 @@
+#include "config.h"
+#include <fcntl.h>
+#include <setjmp.h>
+#include <secp256k1_ecdh.h>
+#include <common/daemon_conn.h>
+#include <common/ecdh.h>
+#include <common/setup.h>
+#include <common/status.h>
+#include <common/wire_error.h>
+#include <connectd/connectd_wiregen.h>
+#include <connectd/connectd.h>
+#include <connectd/multiplex.h>
+#include <connectd/onion_message.h>
+#include <wire/peer_wiregen.h>
+#include <tests/fuzz/libfuzz.h>
+
+static int lightningd_fd;
+static struct privkey priv;
+static struct siphash_seed siphashseed;
+jmp_buf fuzz_env;
+
+/* MOCKS START */
+void inject_peer_msg(struct peer *peer UNNEEDED, const u8 *msg TAKES UNNEEDED)
+{ longjmp(fuzz_env, 1); }
+
+u8 *towire_warningfmt(const tal_t *ctx UNNEEDED,
+		      const struct channel_id *channel UNNEEDED,
+		      const char *fmt UNNEEDED, ...)
+{ longjmp(fuzz_env, 1); }
+
+const struct siphash_seed *siphash_seed(void)
+{ return &siphashseed; }
+/* MOCKS END */
+
+void ecdh(const struct pubkey *point, struct secret *ss)
+{
+	assert(secp256k1_ecdh(secp256k1_ctx, ss->data, &point->pubkey,
+				priv.secret.data, NULL, NULL) == 1);
+}
+
+static struct daemon *new_daemon(const tal_t *ctx)
+{
+	struct daemon *daemon = talz(ctx, struct daemon);
+
+	daemon->our_features = tal(ctx, struct feature_set);
+	daemon->our_features->bits[NODE_ANNOUNCE_FEATURE] = tal_arr(ctx, u8, 0);
+	set_feature_bit(&daemon->our_features->bits[NODE_ANNOUNCE_FEATURE], OPT_ONION_MESSAGES);
+
+	daemon->scid_htable = tal(ctx, struct scid_htable);
+	scid_htable_init(daemon->scid_htable);
+
+	daemon->peers = tal(ctx, struct peer_htable);
+	peer_htable_init(daemon->peers);
+
+	memset(&daemon->mykey, 'a', sizeof(daemon->mykey));
+	node_id_from_pubkey(&daemon->id, &daemon->mykey);
+
+	daemon->master = daemon_conn_new(ctx, lightningd_fd, NULL, NULL, daemon);
+
+	return daemon;
+}
+
+void init(int *argc, char ***argv)
+{
+	common_setup("fuzzer");
+	lightningd_fd = open("/dev/null", O_WRONLY);
+	status_setup_sync(lightningd_fd);
+	chainparams = chainparams_for_network("bitcoin");
+
+	memset(&priv, 'b', sizeof(priv));
+	memset(&siphashseed, 1, sizeof(siphashseed));
+}
+
+void run(const uint8_t *data, size_t size)
+{
+	if (setjmp(fuzz_env) != 0)
+		goto cleanup;
+
+	struct daemon *daemon;
+	struct peer *peer;
+	struct pubkey dummy_key;
+
+	memset(&dummy_key, 'c', sizeof(dummy_key));
+
+	daemon = new_daemon(tmpctx);
+	if (!daemon)
+		goto cleanup;
+
+	peer = talz(tmpctx, struct peer);
+
+	peer->daemon = daemon;
+	node_id_from_pubkey(&peer->id, &dummy_key);
+	peer->onionmsg_incoming_tokens = ONION_MSG_MSEC;
+
+	/* Use fuzzer data as payload of the onion message. */
+	const u8 *onion_msg = towire_onion_message(tmpctx, &dummy_key,
+					tal_dup_arr(tmpctx, u8, data, size, 0));
+
+	handle_onion_message(daemon, peer, onion_msg);
+
+cleanup:
+	if (daemon)
+		tal_free(daemon->master);
+	clean_tmpctx();
+}