From 91cbf4dbd864b65ba6b107957f087d1d305914b2 Mon Sep 17 00:00:00 2001 From: Vasil Dimov Date: Thu, 15 Apr 2021 14:01:52 +0200 Subject: [PATCH] fuzz: add CConnman::CreateNodeFromAcceptedSocket() to the tests --- src/test/fuzz/connman.cpp | 9 +++++++++ src/test/util/net.h | 8 ++++++++ 2 files changed, 17 insertions(+) diff --git a/src/test/fuzz/connman.cpp b/src/test/fuzz/connman.cpp index e65cf5ce58b..ea7ad3c4574 100644 --- a/src/test/fuzz/connman.cpp +++ b/src/test/fuzz/connman.cpp @@ -178,6 +178,15 @@ FUZZ_TARGET(connman, .init = initialize_connman) /*strDest=*/fuzzed_data_provider.ConsumeBool() ? nullptr : random_string.c_str(), /*conn_type=*/conn_type, /*use_v2transport=*/fuzzed_data_provider.ConsumeBool()); + }, + [&] { + connman.SetNetworkActive(fuzzed_data_provider.ConsumeBool()); + const auto peer = ConsumeAddress(fuzzed_data_provider); + connman.CreateNodeFromAcceptedSocketPublic( + /*sock=*/CreateSock(AF_INET, SOCK_STREAM, IPPROTO_TCP), + /*permissions=*/ConsumeWeakEnum(fuzzed_data_provider, ALL_NET_PERMISSION_FLAGS), + /*addr_bind=*/ConsumeAddress(fuzzed_data_provider), + /*addr_peer=*/peer); }); } (void)connman.GetAddedNodeInfo(fuzzed_data_provider.ConsumeBool()); diff --git a/src/test/util/net.h b/src/test/util/net.h index 6bf2bf73007..19264ca5570 100644 --- a/src/test/util/net.h +++ b/src/test/util/net.h @@ -68,6 +68,14 @@ struct ConnmanTestMsg : public CConnman { m_nodes.clear(); } + void CreateNodeFromAcceptedSocketPublic(std::unique_ptr sock, + NetPermissionFlags permissions, + const CAddress& addr_bind, + const CAddress& addr_peer) + { + CreateNodeFromAcceptedSocket(std::move(sock), permissions, addr_bind, addr_peer); + } + void Handshake(CNode& node, bool successfully_connected, ServiceFlags remote_services,