event/net/SocketWrapper: pass std::span to WriteV()

CM4all · Nov 17, 2023 · d5d7e05 · d5d7e05
1 parent 5472ea3
commit d5d7e05
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/src/event/net/BufferedSocket.cxx b/src/event/net/BufferedSocket.cxx
@@ -8,6 +8,7 @@
 #include "net/TimeoutError.hxx"
 
 #include <errno.h>
+#include <sys/uio.h> // for struct iovec
 
 BufferedSocket::BufferedSocket(EventLoop &_event_loop) noexcept
 	:base(_event_loop, *this),
@@ -784,9 +785,9 @@ BufferedSocket::Write(std::span<const std::byte> src) noexcept
 }
 
 ssize_t
-BufferedSocket::WriteV(const struct iovec *v, std::size_t n) noexcept
+BufferedSocket::WriteV(std::span<const struct iovec> v) noexcept
 {
-	ssize_t nbytes = base.WriteV(v, n);
+	ssize_t nbytes = base.WriteV(v);
 
 	if (nbytes < 0) [[unlikely]] {
 		if (const int e = errno; e == EAGAIN) [[likely]] {

diff --git a/src/event/net/BufferedSocket.hxx b/src/event/net/BufferedSocket.hxx
@@ -604,7 +604,7 @@ public:
 	 */
 	ssize_t Write(std::span<const std::byte> src) noexcept;
 
-	ssize_t WriteV(const struct iovec *v, std::size_t n) noexcept;
+	ssize_t WriteV(std::span<const struct iovec> v) noexcept;
 
 	/**
 	 * Transfer data from the given file descriptor to the socket.

diff --git a/src/event/net/SocketWrapper.cxx b/src/event/net/SocketWrapper.cxx
@@ -136,11 +136,11 @@ SocketWrapper::Write(std::span<const std::byte> src) noexcept
 }
 
 ssize_t
-SocketWrapper::WriteV(const struct iovec *v, std::size_t n) noexcept
+SocketWrapper::WriteV(std::span<const struct iovec> v) noexcept
 {
 	assert(IsValid());
 
-	auto m = MakeMsgHdr({v, n});
+	auto m = MakeMsgHdr(v);
 
 	return sendmsg(GetSocket().Get(), &m, MSG_DONTWAIT|MSG_NOSIGNAL);
 }

diff --git a/src/event/net/SocketWrapper.hxx b/src/event/net/SocketWrapper.hxx
@@ -166,7 +166,7 @@ public:
 
 	ssize_t Write(std::span<const std::byte> src) noexcept;
 
-	ssize_t WriteV(const struct iovec *v, std::size_t n) noexcept;
+	ssize_t WriteV(std::span<const struct iovec> v) noexcept;
 
 	ssize_t WriteFrom(FileDescriptor other_fd, FdType other_fd_type,
 			  off_t *other_offset,