aboutsummaryrefslogtreecommitdiff
path: root/src/util
diff options
context:
space:
mode:
authorVasil Dimov <vd@FreeBSD.org>2021-04-13 14:29:14 +0200
committerVasil Dimov <vd@FreeBSD.org>2022-07-20 16:26:23 +0200
commitb4bac556791b5bb8aa118d4c1fed42c3fe45550c (patch)
tree2e254ac7dbd624be6afc4bb90c1143911208a119 /src/util
parent5db7d2ca0aa51ff25f97bf21ce0cbc9e6b741cbd (diff)
net: convert standalone IsSelectableSocket() to Sock::IsSelectable()
This makes the callers mockable.
Diffstat (limited to 'src/util')
-rw-r--r--src/util/sock.cpp9
-rw-r--r--src/util/sock.h8
2 files changed, 11 insertions, 6 deletions
diff --git a/src/util/sock.cpp b/src/util/sock.cpp
index fed50444a9..c312a88276 100644
--- a/src/util/sock.cpp
+++ b/src/util/sock.cpp
@@ -117,11 +117,12 @@ int Sock::GetSockName(sockaddr* name, socklen_t* name_len) const
return getsockname(m_socket, name, name_len);
}
-bool IsSelectableSocket(const SOCKET& s) {
+bool Sock::IsSelectable() const
+{
#if defined(USE_POLL) || defined(WIN32)
return true;
#else
- return (s < FD_SETSIZE);
+ return m_socket < FD_SETSIZE;
#endif
}
@@ -193,10 +194,10 @@ bool Sock::WaitMany(std::chrono::milliseconds timeout, EventsPerSock& events_per
SOCKET socket_max{0};
for (const auto& [sock, events] : events_per_sock) {
- const auto& s = sock->m_socket;
- if (!IsSelectableSocket(s)) {
+ if (!sock->IsSelectable()) {
return false;
}
+ const auto& s = sock->m_socket;
if (events.requested & RECV) {
FD_SET(s, &recv);
}
diff --git a/src/util/sock.h b/src/util/sock.h
index 9ec53ec91d..ab9c6737c6 100644
--- a/src/util/sock.h
+++ b/src/util/sock.h
@@ -133,6 +133,12 @@ public:
*/
[[nodiscard]] virtual int GetSockName(sockaddr* name, socklen_t* name_len) const;
+ /**
+ * Check if the underlying socket can be used for `select(2)` (or the `Wait()` method).
+ * @return true if selectable
+ */
+ [[nodiscard]] virtual bool IsSelectable() const;
+
using Event = uint8_t;
/**
@@ -267,8 +273,6 @@ private:
void Close();
};
-bool IsSelectableSocket(const SOCKET& s);
-
/** Return readable error string for a network error code */
std::string NetworkErrorString(int err);