mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-24 18:32:28 -05:00
046d6855f5
This makes the Scheduler a lot leaner by not having to evaluate block conditions every time it is invoked. Instead evaluate them as the states change, and unblock threads at that point. This also implements some more waitid/waitpid/wait features and behavior. For example, WUNTRACED and WNOWAIT are now supported. And wait will now not return EINTR when SIGCHLD is delivered at the same time.
272 lines
9.1 KiB
C++
272 lines
9.1 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
|
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <AK/StringBuilder.h>
|
|
#include <AK/StringView.h>
|
|
#include <Kernel/FileSystem/FileDescription.h>
|
|
#include <Kernel/Net/IPv4Socket.h>
|
|
#include <Kernel/Net/LocalSocket.h>
|
|
#include <Kernel/Net/Socket.h>
|
|
#include <Kernel/Process.h>
|
|
#include <Kernel/UnixTypes.h>
|
|
#include <LibC/errno_numbers.h>
|
|
|
|
//#define SOCKET_DEBUG
|
|
|
|
namespace Kernel {
|
|
|
|
KResultOr<NonnullRefPtr<Socket>> Socket::create(int domain, int type, int protocol)
|
|
{
|
|
switch (domain) {
|
|
case AF_LOCAL:
|
|
return LocalSocket::create(type & SOCK_TYPE_MASK);
|
|
case AF_INET:
|
|
return IPv4Socket::create(type & SOCK_TYPE_MASK, protocol);
|
|
default:
|
|
return KResult(-EAFNOSUPPORT);
|
|
}
|
|
}
|
|
|
|
Socket::Socket(int domain, int type, int protocol)
|
|
: m_domain(domain)
|
|
, m_type(type)
|
|
, m_protocol(protocol)
|
|
{
|
|
auto& process = *Process::current();
|
|
m_origin = { process.pid().value(), process.uid(), process.gid() };
|
|
}
|
|
|
|
Socket::~Socket()
|
|
{
|
|
}
|
|
|
|
void Socket::set_setup_state(SetupState new_setup_state)
|
|
{
|
|
#ifdef SOCKET_DEBUG
|
|
dbg() << "Socket{" << this << "} setup state moving from " << to_string(m_setup_state) << " to " << to_string(new_setup_state);
|
|
#endif
|
|
|
|
m_setup_state = new_setup_state;
|
|
evaluate_block_conditions();
|
|
}
|
|
|
|
RefPtr<Socket> Socket::accept()
|
|
{
|
|
LOCKER(m_lock);
|
|
if (m_pending.is_empty())
|
|
return nullptr;
|
|
#ifdef SOCKET_DEBUG
|
|
dbg() << "Socket{" << this << "} de-queueing connection";
|
|
#endif
|
|
auto client = m_pending.take_first();
|
|
ASSERT(!client->is_connected());
|
|
auto& process = *Process::current();
|
|
client->m_acceptor = { process.pid().value(), process.uid(), process.gid() };
|
|
client->m_connected = true;
|
|
client->m_role = Role::Accepted;
|
|
if (!m_pending.is_empty())
|
|
evaluate_block_conditions();
|
|
return client;
|
|
}
|
|
|
|
KResult Socket::queue_connection_from(NonnullRefPtr<Socket> peer)
|
|
{
|
|
#ifdef SOCKET_DEBUG
|
|
dbg() << "Socket{" << this << "} queueing connection";
|
|
#endif
|
|
LOCKER(m_lock);
|
|
if (m_pending.size() >= m_backlog)
|
|
return KResult(-ECONNREFUSED);
|
|
m_pending.append(peer);
|
|
evaluate_block_conditions();
|
|
return KSuccess;
|
|
}
|
|
|
|
KResult Socket::setsockopt(int level, int option, Userspace<const void*> user_value, socklen_t user_value_size)
|
|
{
|
|
ASSERT(level == SOL_SOCKET);
|
|
switch (option) {
|
|
case SO_SNDTIMEO:
|
|
if (user_value_size != sizeof(timeval))
|
|
return KResult(-EINVAL);
|
|
if (!copy_from_user(&m_send_timeout, static_ptr_cast<const timeval*>(user_value)))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
case SO_RCVTIMEO:
|
|
if (user_value_size != sizeof(timeval))
|
|
return KResult(-EINVAL);
|
|
if (!copy_from_user(&m_receive_timeout, static_ptr_cast<const timeval*>(user_value)))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
case SO_BINDTODEVICE: {
|
|
if (user_value_size != IFNAMSIZ)
|
|
return KResult(-EINVAL);
|
|
auto user_string = static_ptr_cast<const char*>(user_value);
|
|
auto ifname = copy_string_from_user(user_string, user_value_size);
|
|
if (ifname.is_null())
|
|
return KResult(-EFAULT);
|
|
auto device = NetworkAdapter::lookup_by_name(ifname);
|
|
if (!device)
|
|
return KResult(-ENODEV);
|
|
m_bound_interface = device;
|
|
return KSuccess;
|
|
}
|
|
case SO_KEEPALIVE:
|
|
// FIXME: Obviously, this is not a real keepalive.
|
|
return KSuccess;
|
|
case SO_TIMESTAMP:
|
|
if (user_value_size != sizeof(int))
|
|
return KResult(-EINVAL);
|
|
if (!copy_from_user(&m_timestamp, static_ptr_cast<const int*>(user_value)))
|
|
return KResult(-EFAULT);
|
|
if (m_timestamp && (domain() != AF_INET || type() == SOCK_STREAM)) {
|
|
// FIXME: Support SO_TIMESTAMP for more protocols?
|
|
m_timestamp = 0;
|
|
return KResult(-ENOTSUP);
|
|
}
|
|
return KSuccess;
|
|
default:
|
|
dbg() << "setsockopt(" << option << ") at SOL_SOCKET not implemented.";
|
|
return KResult(-ENOPROTOOPT);
|
|
}
|
|
}
|
|
|
|
KResult Socket::getsockopt(FileDescription&, int level, int option, Userspace<void*> value, Userspace<socklen_t*> value_size)
|
|
{
|
|
socklen_t size;
|
|
if (!copy_from_user(&size, value_size.unsafe_userspace_ptr()))
|
|
return KResult(-EFAULT);
|
|
|
|
// FIXME: Add TCP_NODELAY, IPPROTO_TCP and IPPROTO_IP (used in OpenSSH)
|
|
if (level != SOL_SOCKET) {
|
|
// Not sure if this is the correct error code, but it's only temporary until other levels are implemented.
|
|
return KResult(-ENOPROTOOPT);
|
|
}
|
|
|
|
switch (option) {
|
|
case SO_SNDTIMEO:
|
|
if (size < sizeof(timeval))
|
|
return KResult(-EINVAL);
|
|
if (!copy_to_user(static_ptr_cast<timeval*>(value), &m_send_timeout))
|
|
return KResult(-EFAULT);
|
|
size = sizeof(timeval);
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
case SO_RCVTIMEO:
|
|
if (size < sizeof(timeval))
|
|
return KResult(-EINVAL);
|
|
if (!copy_to_user(static_ptr_cast<timeval*>(value), &m_receive_timeout))
|
|
return KResult(-EFAULT);
|
|
size = sizeof(timeval);
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
case SO_ERROR: {
|
|
if (size < sizeof(int))
|
|
return KResult(-EINVAL);
|
|
dbg() << "getsockopt(SO_ERROR): FIXME!";
|
|
int errno = 0;
|
|
if (!copy_to_user(static_ptr_cast<int*>(value), &errno))
|
|
return KResult(-EFAULT);
|
|
size = sizeof(int);
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
}
|
|
case SO_BINDTODEVICE:
|
|
if (size < IFNAMSIZ)
|
|
return KResult(-EINVAL);
|
|
if (m_bound_interface) {
|
|
const auto& name = m_bound_interface->name();
|
|
auto length = name.length() + 1;
|
|
if (!copy_to_user(static_ptr_cast<char*>(value), name.characters(), length))
|
|
return KResult(-EFAULT);
|
|
size = length;
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
} else {
|
|
size = 0;
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
|
|
return KResult(-EFAULT);
|
|
}
|
|
case SO_TIMESTAMP:
|
|
if (size < sizeof(int))
|
|
return KResult(-EINVAL);
|
|
if (!copy_to_user(static_ptr_cast<int*>(value), &m_timestamp))
|
|
return KResult(-EFAULT);
|
|
size = sizeof(int);
|
|
if (!copy_to_user(value_size, &size))
|
|
return KResult(-EFAULT);
|
|
return KSuccess;
|
|
default:
|
|
dbg() << "getsockopt(" << option << ") at SOL_SOCKET not implemented.";
|
|
return KResult(-ENOPROTOOPT);
|
|
}
|
|
}
|
|
|
|
KResultOr<size_t> Socket::read(FileDescription& description, size_t, UserOrKernelBuffer& buffer, size_t size)
|
|
{
|
|
if (is_shut_down_for_reading())
|
|
return 0;
|
|
timeval tv;
|
|
return recvfrom(description, buffer, size, 0, {}, 0, tv);
|
|
}
|
|
|
|
KResultOr<size_t> Socket::write(FileDescription& description, size_t, const UserOrKernelBuffer& data, size_t size)
|
|
{
|
|
if (is_shut_down_for_writing())
|
|
return -EPIPE;
|
|
return sendto(description, data, size, 0, {}, 0);
|
|
}
|
|
|
|
KResult Socket::shutdown(int how)
|
|
{
|
|
LOCKER(lock());
|
|
if (type() == SOCK_STREAM && !is_connected())
|
|
return KResult(-ENOTCONN);
|
|
if (m_role == Role::Listener)
|
|
return KResult(-ENOTCONN);
|
|
if (!m_shut_down_for_writing && (how & SHUT_WR))
|
|
shut_down_for_writing();
|
|
if (!m_shut_down_for_reading && (how & SHUT_RD))
|
|
shut_down_for_reading();
|
|
m_shut_down_for_reading |= (how & SHUT_RD) != 0;
|
|
m_shut_down_for_writing |= (how & SHUT_WR) != 0;
|
|
return KSuccess;
|
|
}
|
|
|
|
KResult Socket::stat(::stat& st) const
|
|
{
|
|
memset(&st, 0, sizeof(st));
|
|
st.st_mode = S_IFSOCK;
|
|
return KSuccess;
|
|
}
|
|
|
|
}
|