mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-05-24 02:42:12 +02:00
net: add new method Sock::SetSockOpt() that wraps setsockopt()
This will help to increase `Sock` usage and make more code mockable.
This commit is contained in:
parent
e14f0fa6a3
commit
184e56d668
@ -193,6 +193,19 @@ int FuzzedSock::GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* op
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int FuzzedSock::SetSockOpt(int, int, const void*, socklen_t) const
|
||||||
|
{
|
||||||
|
constexpr std::array setsockopt_errnos{
|
||||||
|
ENOMEM,
|
||||||
|
ENOBUFS,
|
||||||
|
};
|
||||||
|
if (m_fuzzed_data_provider.ConsumeBool()) {
|
||||||
|
SetFuzzedErrNo(m_fuzzed_data_provider, setsockopt_errnos);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
bool FuzzedSock::Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred) const
|
bool FuzzedSock::Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred) const
|
||||||
{
|
{
|
||||||
constexpr std::array wait_errnos{
|
constexpr std::array wait_errnos{
|
||||||
|
@ -68,6 +68,8 @@ public:
|
|||||||
|
|
||||||
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;
|
||||||
|
|
||||||
|
int SetSockOpt(int level, int opt_name, const void* opt_val, socklen_t opt_len) const override;
|
||||||
|
|
||||||
bool Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred = nullptr) const override;
|
bool Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred = nullptr) const override;
|
||||||
|
|
||||||
bool IsConnected(std::string& errmsg) const override;
|
bool IsConnected(std::string& errmsg) const override;
|
||||||
|
@ -150,6 +150,8 @@ public:
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int SetSockOpt(int, int, const void*, socklen_t) const override { return 0; }
|
||||||
|
|
||||||
bool Wait(std::chrono::milliseconds timeout,
|
bool Wait(std::chrono::milliseconds timeout,
|
||||||
Event requested,
|
Event requested,
|
||||||
Event* occurred = nullptr) const override
|
Event* occurred = nullptr) const override
|
||||||
|
@ -105,6 +105,11 @@ int Sock::GetSockOpt(int level, int opt_name, void* opt_val, socklen_t* opt_len)
|
|||||||
return getsockopt(m_socket, level, opt_name, static_cast<char*>(opt_val), opt_len);
|
return getsockopt(m_socket, level, opt_name, static_cast<char*>(opt_val), opt_len);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int Sock::SetSockOpt(int level, int opt_name, const void* opt_val, socklen_t opt_len) const
|
||||||
|
{
|
||||||
|
return setsockopt(m_socket, level, opt_name, static_cast<const char*>(opt_val), opt_len);
|
||||||
|
}
|
||||||
|
|
||||||
bool Sock::Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred) const
|
bool Sock::Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred) const
|
||||||
{
|
{
|
||||||
#ifdef USE_POLL
|
#ifdef USE_POLL
|
||||||
|
@ -115,6 +115,16 @@ public:
|
|||||||
void* opt_val,
|
void* opt_val,
|
||||||
socklen_t* opt_len) const;
|
socklen_t* opt_len) const;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* setsockopt(2) wrapper. Equivalent to
|
||||||
|
* `setsockopt(this->Get(), level, opt_name, opt_val, opt_len)`. Code that uses this
|
||||||
|
* wrapper can be unit tested if this method is overridden by a mock Sock implementation.
|
||||||
|
*/
|
||||||
|
[[nodiscard]] virtual int SetSockOpt(int level,
|
||||||
|
int opt_name,
|
||||||
|
const void* opt_val,
|
||||||
|
socklen_t opt_len) const;
|
||||||
|
|
||||||
using Event = uint8_t;
|
using Event = uint8_t;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
Loading…
x
Reference in New Issue
Block a user