net: add new method Sock::GetSockName() that wraps getsockname()

This will help to increase `Sock` usage and make more code mockable.

2 files changed, 12 insertions, 0 deletions

diff --git a/src/util/sock.cpp b/src/util/sock.cpp
index 7d5069423a..b4c0aa4205 100644
--- a/src/util/sock.cpp
+++ b/src/util/sock.cpp
@@ -111,6 +111,11 @@ int Sock::SetSockOpt(int level, int opt_name, const void* opt_val, socklen_t opt
     return setsockopt(m_socket, level, opt_name, static_cast<const char*>(opt_val), opt_len);
 }
 
+int Sock::GetSockName(sockaddr* name, socklen_t* name_len) const
+{
+    return getsockname(m_socket, name, name_len);
+}
+
 bool Sock::Wait(std::chrono::milliseconds timeout, Event requested, Event* occurred) const
 {
     // We need a `shared_ptr` owning `this` for `WaitMany()`, but don't want
diff --git a/src/util/sock.h b/src/util/sock.h
index 3245820995..96d0b3b56b 100644
--- a/src/util/sock.h
+++ b/src/util/sock.h
@@ -126,6 +126,13 @@ public:
                                          const void* opt_val,
                                          socklen_t opt_len) const;
 
+    /**
+     * getsockname(2) wrapper. Equivalent to
+     * `getsockname(this->Get(), name, name_len)`. Code that uses this
+     * wrapper can be unit tested if this method is overridden by a mock Sock implementation.
+     */
+    [[nodiscard]] virtual int GetSockName(sockaddr* name, socklen_t* name_len) const;
+
     using Event = uint8_t;
 
     /**