mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-01-08 20:32:56 +03:00
79fa9765ca
We now use AK::Error and AK::ErrorOr<T> in both kernel and userspace! This was a slightly tedious refactoring that took a long time, so it's not unlikely that some bugs crept in. Nevertheless, it does pass basic functionality testing, and it's just real nice to finally see the same pattern in all contexts. :^)
141 lines
3.9 KiB
C++
141 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <Kernel/Arch/x86/InterruptDisabler.h>
|
|
#include <Kernel/Debug.h>
|
|
#include <Kernel/Process.h>
|
|
#include <Kernel/TTY/MasterPTY.h>
|
|
#include <Kernel/TTY/PTYMultiplexer.h>
|
|
#include <Kernel/TTY/SlavePTY.h>
|
|
#include <LibC/errno_numbers.h>
|
|
#include <LibC/signal_numbers.h>
|
|
#include <LibC/sys/ioctl_numbers.h>
|
|
|
|
namespace Kernel {
|
|
|
|
ErrorOr<NonnullRefPtr<MasterPTY>> MasterPTY::try_create(unsigned int index)
|
|
{
|
|
// FIXME: Don't make a temporary String here
|
|
auto pts_name = TRY(KString::try_create(String::formatted("/dev/pts/{}", index)));
|
|
auto tty_name = TRY(pts_name->try_clone());
|
|
|
|
auto buffer = TRY(DoubleBuffer::try_create());
|
|
auto master_pty = TRY(adopt_nonnull_ref_or_enomem(new (nothrow) MasterPTY(index, move(buffer), move(pts_name))));
|
|
auto slave_pty = TRY(adopt_nonnull_ref_or_enomem(new (nothrow) SlavePTY(*master_pty, index, move(tty_name))));
|
|
master_pty->m_slave = slave_pty;
|
|
master_pty->after_inserting();
|
|
slave_pty->after_inserting();
|
|
return master_pty;
|
|
}
|
|
|
|
MasterPTY::MasterPTY(unsigned index, NonnullOwnPtr<DoubleBuffer> buffer, NonnullOwnPtr<KString> pts_name)
|
|
: CharacterDevice(200, index)
|
|
, m_index(index)
|
|
, m_buffer(move(buffer))
|
|
, m_pts_name(move(pts_name))
|
|
{
|
|
auto& process = Process::current();
|
|
set_uid(process.uid());
|
|
set_gid(process.gid());
|
|
|
|
m_buffer->set_unblock_callback([this]() {
|
|
if (m_slave)
|
|
evaluate_block_conditions();
|
|
});
|
|
}
|
|
|
|
MasterPTY::~MasterPTY()
|
|
{
|
|
dbgln_if(MASTERPTY_DEBUG, "~MasterPTY({})", m_index);
|
|
PTYMultiplexer::the().notify_master_destroyed({}, m_index);
|
|
}
|
|
|
|
KString const& MasterPTY::pts_name() const
|
|
{
|
|
return *m_pts_name;
|
|
}
|
|
|
|
ErrorOr<size_t> MasterPTY::read(OpenFileDescription&, u64, UserOrKernelBuffer& buffer, size_t size)
|
|
{
|
|
if (!m_slave && m_buffer->is_empty())
|
|
return 0;
|
|
return m_buffer->read(buffer, size);
|
|
}
|
|
|
|
ErrorOr<size_t> MasterPTY::write(OpenFileDescription&, u64, const UserOrKernelBuffer& buffer, size_t size)
|
|
{
|
|
if (!m_slave)
|
|
return EIO;
|
|
m_slave->on_master_write(buffer, size);
|
|
return size;
|
|
}
|
|
|
|
bool MasterPTY::can_read(const OpenFileDescription&, size_t) const
|
|
{
|
|
if (!m_slave)
|
|
return true;
|
|
return !m_buffer->is_empty();
|
|
}
|
|
|
|
bool MasterPTY::can_write(const OpenFileDescription&, size_t) const
|
|
{
|
|
return true;
|
|
}
|
|
|
|
void MasterPTY::notify_slave_closed(Badge<SlavePTY>)
|
|
{
|
|
dbgln_if(MASTERPTY_DEBUG, "MasterPTY({}): slave closed, my retains: {}, slave retains: {}", m_index, ref_count(), m_slave->ref_count());
|
|
// +1 ref for my MasterPTY::m_slave
|
|
// +1 ref for OpenFileDescription::m_device
|
|
if (m_slave->ref_count() == 2)
|
|
m_slave = nullptr;
|
|
}
|
|
|
|
ErrorOr<size_t> MasterPTY::on_slave_write(const UserOrKernelBuffer& data, size_t size)
|
|
{
|
|
if (m_closed)
|
|
return EIO;
|
|
return m_buffer->write(data, size);
|
|
}
|
|
|
|
bool MasterPTY::can_write_from_slave() const
|
|
{
|
|
if (m_closed)
|
|
return true;
|
|
return m_buffer->space_for_writing();
|
|
}
|
|
|
|
ErrorOr<void> MasterPTY::close()
|
|
{
|
|
InterruptDisabler disabler;
|
|
// After the closing OpenFileDescription dies, slave is the only thing keeping me alive.
|
|
// From this point, let's consider ourselves closed.
|
|
m_closed = true;
|
|
|
|
if (m_slave)
|
|
m_slave->hang_up();
|
|
|
|
return {};
|
|
}
|
|
|
|
ErrorOr<void> MasterPTY::ioctl(OpenFileDescription& description, unsigned request, Userspace<void*> arg)
|
|
{
|
|
REQUIRE_PROMISE(tty);
|
|
if (!m_slave)
|
|
return EIO;
|
|
if (request == TIOCSWINSZ || request == TIOCGPGRP)
|
|
return m_slave->ioctl(description, request, arg);
|
|
return EINVAL;
|
|
}
|
|
|
|
ErrorOr<NonnullOwnPtr<KString>> MasterPTY::pseudo_path(const OpenFileDescription&) const
|
|
{
|
|
// FIXME: Replace this and others of this pattern by KString::formatted()
|
|
return KString::try_create(String::formatted("ptm:{}", m_pts_name));
|
|
}
|
|
|
|
}
|