summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--Userland/Libraries/LibCore/Stream.cpp11
-rw-r--r--Userland/Libraries/LibCore/Stream.h3
2 files changed, 10 insertions, 4 deletions
diff --git a/Userland/Libraries/LibCore/Stream.cpp b/Userland/Libraries/LibCore/Stream.cpp
index 58d99e6349..b24b56514f 100644
--- a/Userland/Libraries/LibCore/Stream.cpp
+++ b/Userland/Libraries/LibCore/Stream.cpp
@@ -341,15 +341,15 @@ ErrorOr<void> Socket::connect_inet(int fd, SocketAddress const& address)
return {};
}
-ErrorOr<size_t> PosixSocketHelper::read(Bytes buffer)
+ErrorOr<size_t> PosixSocketHelper::read(Bytes buffer, int flags)
{
if (!is_open()) {
return Error::from_errno(ENOTCONN);
}
- ssize_t rc = ::recv(m_fd, buffer.data(), buffer.size(), 0);
+ ssize_t rc = ::recv(m_fd, buffer.data(), buffer.size(), flags);
if (rc < 0) {
- return Error::from_errno(errno);
+ return Error::from_syscall("recv", -errno);
}
m_last_read_was_eof = rc == 0;
@@ -556,4 +556,9 @@ ErrorOr<void> LocalSocket::send_fd(int fd)
#endif
}
+ErrorOr<size_t> LocalSocket::read_without_waiting(Bytes buffer)
+{
+ return m_helper.read(buffer, MSG_DONTWAIT);
+}
+
}
diff --git a/Userland/Libraries/LibCore/Stream.h b/Userland/Libraries/LibCore/Stream.h
index 7999eef53c..d3820ab379 100644
--- a/Userland/Libraries/LibCore/Stream.h
+++ b/Userland/Libraries/LibCore/Stream.h
@@ -229,7 +229,7 @@ public:
int fd() const { return m_fd; }
void set_fd(int fd) { m_fd = fd; }
- ErrorOr<size_t> read(Bytes);
+ ErrorOr<size_t> read(Bytes, int flags = 0);
ErrorOr<size_t> write(ReadonlyBytes);
bool is_eof() const { return !is_open() || m_last_read_was_eof; }
@@ -412,6 +412,7 @@ public:
ErrorOr<int> receive_fd(int flags);
ErrorOr<void> send_fd(int fd);
+ ErrorOr<size_t> read_without_waiting(Bytes buffer);
virtual ~LocalSocket() { close(); }