Kernel: Use KResultOr better in ProcessGroup construction

This allows us to use TRY() more.
This commit is contained in:
Andreas Kling 2021-09-05 21:48:04 +02:00
parent 540d62d3b2
commit 98dc08fe56
3 changed files with 11 additions and 21 deletions

View file

@ -24,27 +24,24 @@ ProcessGroup::~ProcessGroup()
});
}
RefPtr<ProcessGroup> ProcessGroup::try_create(ProcessGroupID pgid)
KResultOr<NonnullRefPtr<ProcessGroup>> ProcessGroup::try_create(ProcessGroupID pgid)
{
auto process_group = adopt_ref_if_nonnull(new (nothrow) ProcessGroup(pgid));
if (!process_group)
return {};
auto process_group = TRY(adopt_nonnull_ref_or_enomem(new (nothrow) ProcessGroup(pgid)));
process_groups().with([&](auto& groups) {
groups.prepend(*process_group);
});
return process_group;
}
RefPtr<ProcessGroup> ProcessGroup::try_find_or_create(ProcessGroupID pgid)
KResultOr<NonnullRefPtr<ProcessGroup>> ProcessGroup::try_find_or_create(ProcessGroupID pgid)
{
return process_groups().with([&](auto& groups) -> RefPtr<ProcessGroup> {
return process_groups().with([&](auto& groups) -> KResultOr<NonnullRefPtr<ProcessGroup>> {
for (auto& group : groups) {
if (group.pgid() == pgid)
return &group;
return group;
}
auto process_group = adopt_ref_if_nonnull(new (nothrow) ProcessGroup(pgid));
if (process_group)
groups.prepend(*process_group);
auto process_group = TRY(adopt_nonnull_ref_or_enomem(new (nothrow) ProcessGroup(pgid)));
groups.prepend(*process_group);
return process_group;
});
}

View file

@ -24,8 +24,8 @@ class ProcessGroup
public:
~ProcessGroup();
static RefPtr<ProcessGroup> try_create(ProcessGroupID);
static RefPtr<ProcessGroup> try_find_or_create(ProcessGroupID);
static KResultOr<NonnullRefPtr<ProcessGroup>> try_create(ProcessGroupID);
static KResultOr<NonnullRefPtr<ProcessGroup>> try_find_or_create(ProcessGroupID);
static RefPtr<ProcessGroup> from_pgid(ProcessGroupID);
const ProcessGroupID& pgid() const { return m_pgid; }

View file

@ -38,11 +38,7 @@ KResultOr<FlatPtr> Process::sys$setsid()
return EPERM;
// Create a new Session and a new ProcessGroup.
auto new_pg = ProcessGroup::try_create(ProcessGroupID(pid().value()));
if (!new_pg)
return ENOMEM;
m_pg = move(new_pg);
m_pg = TRY(ProcessGroup::try_create(ProcessGroupID(pid().value())));
m_tty = nullptr;
ProtectedDataMutationScope scope { *this };
m_protected_values.sid = pid().value();
@ -122,10 +118,7 @@ KResultOr<FlatPtr> Process::sys$setpgid(pid_t specified_pid, pid_t specified_pgi
return EPERM;
}
// FIXME: There are more EPERM conditions to check for here..
auto process_group = ProcessGroup::try_find_or_create(new_pgid);
if (!process_group)
return ENOMEM;
process->m_pg = move(process_group);
process->m_pg = TRY(ProcessGroup::try_find_or_create(new_pgid));
return 0;
}