2022-11-02 23:26:02 +03:00
|
|
|
/*
|
2023-01-12 23:06:51 +03:00
|
|
|
* Copyright (c) 2022-2023, Liav A. <liavalb@hotmail.co.il>
|
2022-11-02 23:26:02 +03:00
|
|
|
*
|
|
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
|
|
*/
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
|
2023-04-11 00:15:48 +03:00
|
|
|
#include <AK/AtomicRefCounted.h>
|
2022-11-02 23:26:02 +03:00
|
|
|
#include <AK/DistinctNumeric.h>
|
|
|
|
#include <AK/Error.h>
|
|
|
|
#include <AK/IntrusiveList.h>
|
2023-01-12 23:06:51 +03:00
|
|
|
#include <AK/IntrusiveListRelaxedConst.h>
|
2022-11-02 23:26:02 +03:00
|
|
|
#include <AK/OwnPtr.h>
|
|
|
|
#include <AK/RefPtr.h>
|
|
|
|
#include <AK/Try.h>
|
|
|
|
#include <AK/Types.h>
|
|
|
|
#include <Kernel/KString.h>
|
|
|
|
#include <Kernel/Locking/SpinlockProtected.h>
|
2023-03-16 11:27:02 +03:00
|
|
|
#include <Kernel/Process.h>
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
namespace Kernel {
|
|
|
|
|
2023-01-12 23:06:51 +03:00
|
|
|
class ProcessList;
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
AK_TYPEDEF_DISTINCT_ORDERED_ID(u64, JailIndex);
|
|
|
|
|
2023-04-11 00:15:48 +03:00
|
|
|
class Jail : public AtomicRefCounted<Jail> {
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
public:
|
2023-01-12 23:47:09 +03:00
|
|
|
RefPtr<ProcessList> process_list();
|
2023-01-12 23:06:51 +03:00
|
|
|
|
2023-04-11 00:24:47 +03:00
|
|
|
static RefPtr<Jail> find_by_index(JailIndex);
|
2023-01-12 23:47:09 +03:00
|
|
|
static ErrorOr<NonnullRefPtr<Jail>> create(NonnullOwnPtr<KString> name, unsigned flags);
|
2023-01-12 23:06:51 +03:00
|
|
|
static ErrorOr<void> for_each_when_process_is_not_jailed(Function<ErrorOr<void>(Jail const&)> callback);
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
StringView name() const { return m_name->view(); }
|
|
|
|
JailIndex index() const { return m_index; }
|
|
|
|
|
|
|
|
void detach(Badge<Process>);
|
2022-11-09 13:39:58 +03:00
|
|
|
SpinlockProtected<size_t, LockRank::None>& attach_count() { return m_attach_count; }
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
private:
|
2023-01-12 23:47:09 +03:00
|
|
|
Jail(NonnullOwnPtr<KString>, JailIndex, RefPtr<ProcessList>);
|
2022-11-02 23:26:02 +03:00
|
|
|
|
|
|
|
NonnullOwnPtr<KString> m_name;
|
|
|
|
JailIndex const m_index;
|
|
|
|
|
2023-04-11 00:24:47 +03:00
|
|
|
IntrusiveListNode<Jail, NonnullRefPtr<Jail>> m_list_node;
|
2023-01-12 23:06:51 +03:00
|
|
|
|
|
|
|
public:
|
|
|
|
using List = IntrusiveListRelaxedConst<&Jail::m_list_node>;
|
|
|
|
|
|
|
|
private:
|
2023-01-12 23:47:09 +03:00
|
|
|
RefPtr<ProcessList> const m_process_list;
|
2023-01-12 23:06:51 +03:00
|
|
|
|
2022-11-09 13:39:58 +03:00
|
|
|
SpinlockProtected<size_t, LockRank::None> m_attach_count { 0 };
|
2022-11-02 23:26:02 +03:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|