mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-03-08 14:34:53 -05:00
Merge bitcoin/bitcoin#24378: refactor: make bind() and listen() mockable/testable
b2733ab6a8
net: add new method Sock::Listen() that wraps listen() (Vasil Dimov)3ad7de225e
net: add new method Sock::Bind() that wraps bind() (Vasil Dimov) Pull request description: _This is a piece of #21878, chopped off to ease review._ Add new methods `Sock::Bind()` and `Sock::Listen()` that wrap `bind()` and `listen()`. This will help to increase `Sock` usage and make more code mockable. ACKs for top commit: pk-b2: ACKb2733ab6a8
laanwj: Code review ACKb2733ab6a8
Tree-SHA512: c6e737606703e2106fe60cc000cfbbae3a7f43deadb25f70531e2cac0457e0b0581440279d14c76c492eb85c12af4adde52c30baf74542c41597e419817488e8
This commit is contained in:
commit
55c9e2d790
6 changed files with 71 additions and 3 deletions
|
@ -2323,8 +2323,7 @@ bool CConnman::BindListenPort(const CService& addrBind, bilingual_str& strError,
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
if (::bind(sock->Get(), (struct sockaddr*)&sockaddr, len) == SOCKET_ERROR)
|
if (sock->Bind(reinterpret_cast<struct sockaddr*>(&sockaddr), len) == SOCKET_ERROR) {
|
||||||
{
|
|
||||||
int nErr = WSAGetLastError();
|
int nErr = WSAGetLastError();
|
||||||
if (nErr == WSAEADDRINUSE)
|
if (nErr == WSAEADDRINUSE)
|
||||||
strError = strprintf(_("Unable to bind to %s on this computer. %s is probably already running."), addrBind.ToString(), PACKAGE_NAME);
|
strError = strprintf(_("Unable to bind to %s on this computer. %s is probably already running."), addrBind.ToString(), PACKAGE_NAME);
|
||||||
|
@ -2336,7 +2335,7 @@ bool CConnman::BindListenPort(const CService& addrBind, bilingual_str& strError,
|
||||||
LogPrintf("Bound to %s\n", addrBind.ToString());
|
LogPrintf("Bound to %s\n", addrBind.ToString());
|
||||||
|
|
||||||
// Listen for incoming connections
|
// Listen for incoming connections
|
||||||
if (listen(sock->Get(), SOMAXCONN) == SOCKET_ERROR)
|
if (sock->Listen(SOMAXCONN) == SOCKET_ERROR)
|
||||||
{
|
{
|
||||||
strError = strprintf(_("Listening for incoming connections failed (listen returned error %s)"), NetworkErrorString(WSAGetLastError()));
|
strError = strprintf(_("Listening for incoming connections failed (listen returned error %s)"), NetworkErrorString(WSAGetLastError()));
|
||||||
LogPrintLevel(BCLog::NET, BCLog::Level::Error, "%s\n", strError.original);
|
LogPrintLevel(BCLog::NET, BCLog::Level::Error, "%s\n", strError.original);
|
||||||
|
|
|
@ -155,6 +155,45 @@ int FuzzedSock::Connect(const sockaddr*, socklen_t) const
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int FuzzedSock::Bind(const sockaddr*, socklen_t) const
|
||||||
|
{
|
||||||
|
// Have a permanent error at bind_errnos[0] because when the fuzzed data is exhausted
|
||||||
|
// SetFuzzedErrNo() will always set the global errno to bind_errnos[0]. We want to
|
||||||
|
// avoid this method returning -1 and setting errno to a temporary error (like EAGAIN)
|
||||||
|
// repeatedly because proper code should retry on temporary errors, leading to an
|
||||||
|
// infinite loop.
|
||||||
|
constexpr std::array bind_errnos{
|
||||||
|
EACCES,
|
||||||
|
EADDRINUSE,
|
||||||
|
EADDRNOTAVAIL,
|
||||||
|
EAGAIN,
|
||||||
|
};
|
||||||
|
if (m_fuzzed_data_provider.ConsumeBool()) {
|
||||||
|
SetFuzzedErrNo(m_fuzzed_data_provider, bind_errnos);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int FuzzedSock::Listen(int) const
|
||||||
|
{
|
||||||
|
// Have a permanent error at listen_errnos[0] because when the fuzzed data is exhausted
|
||||||
|
// SetFuzzedErrNo() will always set the global errno to listen_errnos[0]. We want to
|
||||||
|
// avoid this method returning -1 and setting errno to a temporary error (like EAGAIN)
|
||||||
|
// repeatedly because proper code should retry on temporary errors, leading to an
|
||||||
|
// infinite loop.
|
||||||
|
constexpr std::array listen_errnos{
|
||||||
|
EADDRINUSE,
|
||||||
|
EINVAL,
|
||||||
|
EOPNOTSUPP,
|
||||||
|
};
|
||||||
|
if (m_fuzzed_data_provider.ConsumeBool()) {
|
||||||
|
SetFuzzedErrNo(m_fuzzed_data_provider, listen_errnos);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
std::unique_ptr<Sock> FuzzedSock::Accept(sockaddr* addr, socklen_t* addr_len) const
|
std::unique_ptr<Sock> FuzzedSock::Accept(sockaddr* addr, socklen_t* addr_len) const
|
||||||
{
|
{
|
||||||
constexpr std::array accept_errnos{
|
constexpr std::array accept_errnos{
|
||||||
|
|
|
@ -61,6 +61,10 @@ public:
|
||||||
|
|
||||||
int Connect(const sockaddr*, socklen_t) const override;
|
int Connect(const sockaddr*, socklen_t) const override;
|
||||||
|
|
||||||
|
int Bind(const sockaddr*, socklen_t) const override;
|
||||||
|
|
||||||
|
int Listen(int backlog) const override;
|
||||||
|
|
||||||
std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override;
|
std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override;
|
||||||
|
|
||||||
int GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* opt_len) const override;
|
int GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* opt_len) const override;
|
||||||
|
|
|
@ -122,6 +122,10 @@ public:
|
||||||
|
|
||||||
int Connect(const sockaddr*, socklen_t) const override { return 0; }
|
int Connect(const sockaddr*, socklen_t) const override { return 0; }
|
||||||
|
|
||||||
|
int Bind(const sockaddr*, socklen_t) const override { return 0; }
|
||||||
|
|
||||||
|
int Listen(int) const override { return 0; }
|
||||||
|
|
||||||
std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override
|
std::unique_ptr<Sock> Accept(sockaddr* addr, socklen_t* addr_len) const override
|
||||||
{
|
{
|
||||||
if (addr != nullptr) {
|
if (addr != nullptr) {
|
||||||
|
|
|
@ -66,6 +66,16 @@ int Sock::Connect(const sockaddr* addr, socklen_t addr_len) const
|
||||||
return connect(m_socket, addr, addr_len);
|
return connect(m_socket, addr, addr_len);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int Sock::Bind(const sockaddr* addr, socklen_t addr_len) const
|
||||||
|
{
|
||||||
|
return bind(m_socket, addr, addr_len);
|
||||||
|
}
|
||||||
|
|
||||||
|
int Sock::Listen(int backlog) const
|
||||||
|
{
|
||||||
|
return listen(m_socket, backlog);
|
||||||
|
}
|
||||||
|
|
||||||
std::unique_ptr<Sock> Sock::Accept(sockaddr* addr, socklen_t* addr_len) const
|
std::unique_ptr<Sock> Sock::Accept(sockaddr* addr, socklen_t* addr_len) const
|
||||||
{
|
{
|
||||||
#ifdef WIN32
|
#ifdef WIN32
|
||||||
|
|
|
@ -86,6 +86,18 @@ public:
|
||||||
*/
|
*/
|
||||||
[[nodiscard]] virtual int Connect(const sockaddr* addr, socklen_t addr_len) const;
|
[[nodiscard]] virtual int Connect(const sockaddr* addr, socklen_t addr_len) const;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* bind(2) wrapper. Equivalent to `bind(this->Get(), addr, addr_len)`. Code that uses this
|
||||||
|
* wrapper can be unit tested if this method is overridden by a mock Sock implementation.
|
||||||
|
*/
|
||||||
|
[[nodiscard]] virtual int Bind(const sockaddr* addr, socklen_t addr_len) const;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* listen(2) wrapper. Equivalent to `listen(this->Get(), backlog)`. Code that uses this
|
||||||
|
* wrapper can be unit tested if this method is overridden by a mock Sock implementation.
|
||||||
|
*/
|
||||||
|
[[nodiscard]] virtual int Listen(int backlog) const;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* accept(2) wrapper. Equivalent to `std::make_unique<Sock>(accept(this->Get(), addr, addr_len))`.
|
* accept(2) wrapper. Equivalent to `std::make_unique<Sock>(accept(this->Get(), addr, addr_len))`.
|
||||||
* Code that uses this wrapper can be unit tested if this method is overridden by a mock Sock
|
* Code that uses this wrapper can be unit tested if this method is overridden by a mock Sock
|
||||||
|
|
Loading…
Add table
Reference in a new issue