mirror of
https://github.com/RGBCube/serenity
synced 2025-07-26 06:37:35 +00:00
Net: Add a basic sys$shutdown() implementation
Calling shutdown prevents further reads and/or writes on a socket. We should do a few more things based on the type of socket, but this initial implementation just puts the basic mechanism in place. Work towards #428.
This commit is contained in:
parent
a3f39fe789
commit
2b0b7cc5a4
8 changed files with 60 additions and 2 deletions
|
@ -149,10 +149,23 @@ KResult Socket::getsockopt(FileDescription&, int level, int option, void* value,
|
|||
|
||||
ssize_t Socket::read(FileDescription& description, u8* buffer, ssize_t size)
|
||||
{
|
||||
if (is_shut_down_for_reading())
|
||||
return 0;
|
||||
return recvfrom(description, buffer, size, 0, nullptr, 0);
|
||||
}
|
||||
|
||||
ssize_t Socket::write(FileDescription& description, const u8* data, ssize_t size)
|
||||
{
|
||||
if (is_shut_down_for_writing())
|
||||
return -EPIPE;
|
||||
return sendto(description, data, size, 0, nullptr, 0);
|
||||
}
|
||||
|
||||
KResult Socket::shutdown(int how)
|
||||
{
|
||||
if (type() == SOCK_STREAM && !is_connected())
|
||||
return KResult(-ENOTCONN);
|
||||
m_shut_down_for_reading |= how & SHUT_RD;
|
||||
m_shut_down_for_writing |= how & SHUT_WR;
|
||||
return KSuccess;
|
||||
}
|
||||
|
|
|
@ -51,6 +51,9 @@ public:
|
|||
int type() const { return m_type; }
|
||||
int protocol() const { return m_protocol; }
|
||||
|
||||
bool is_shut_down_for_writing() const { return m_shut_down_for_writing; }
|
||||
bool is_shut_down_for_reading() const { return m_shut_down_for_reading; }
|
||||
|
||||
enum class SetupState {
|
||||
Unstarted, // we haven't tried to set the socket up yet
|
||||
InProgress, // we're in the process of setting things up - for TCP maybe we've sent a SYN packet
|
||||
|
@ -90,6 +93,8 @@ public:
|
|||
bool can_accept() const { return !m_pending.is_empty(); }
|
||||
RefPtr<Socket> accept();
|
||||
|
||||
KResult shutdown(int how);
|
||||
|
||||
virtual KResult bind(const sockaddr*, socklen_t) = 0;
|
||||
virtual KResult connect(FileDescription&, const sockaddr*, socklen_t, ShouldBlock) = 0;
|
||||
virtual KResult listen(int) = 0;
|
||||
|
@ -153,6 +158,8 @@ private:
|
|||
int m_backlog { 0 };
|
||||
SetupState m_setup_state { SetupState::Unstarted };
|
||||
bool m_connected { false };
|
||||
bool m_shut_down_for_reading { false };
|
||||
bool m_shut_down_for_writing { false };
|
||||
|
||||
timeval m_receive_timeout { 0, 0 };
|
||||
timeval m_send_timeout { 0, 0 };
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue