Kernel/Syscalls/pledge.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

/*
 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <AK/StringView.h>
#include <Kernel/Process.h>

namespace Kernel {

KResultOr<FlatPtr> Process::sys$pledge(Userspace<const Syscall::SC_pledge_params*> user_params)
{
    VERIFY_PROCESS_BIG_LOCK_ACQUIRED(this)
    Syscall::SC_pledge_params params;
    if (!copy_from_user(&params, user_params))
        return EFAULT;

    if (params.promises.length > 1024 || params.execpromises.length > 1024)
        return E2BIG;

    OwnPtr<KString> promises;
    if (params.promises.characters) {
        auto promises_or_error = try_copy_kstring_from_user(params.promises);
        if (promises_or_error.is_error())
            return promises_or_error.error();
        promises = promises_or_error.release_value();
    }

    OwnPtr<KString> execpromises;
    if (params.execpromises.characters) {
        auto execpromises_or_error = try_copy_kstring_from_user(params.execpromises);
        if (execpromises_or_error.is_error())
            return execpromises_or_error.error();
        execpromises = execpromises_or_error.release_value();
    }

    auto parse_pledge = [&](auto pledge_spec, u32& mask) {
        auto parts = pledge_spec.split_view(' ');
        for (auto& part : parts) {
#define __ENUMERATE_PLEDGE_PROMISE(x)   \
    if (part == #x) {                   \
        mask |= (1u << (u32)Pledge::x); \
        continue;                       \
    }
            ENUMERATE_PLEDGE_PROMISES
#undef __ENUMERATE_PLEDGE_PROMISE
            return false;
        }
        return true;
    };

    ProtectedDataMutationScope scope { *this };

    if (promises) {
        u32 new_promises = 0;
        if (!parse_pledge(promises->view(), new_promises))
            return EINVAL;
        if (m_has_promises && (new_promises & ~m_promises))
            return EPERM;
        m_has_promises = true;
        m_promises = new_promises;
    }

    if (execpromises) {
        u32 new_execpromises = 0;
        if (!parse_pledge(execpromises->view(), new_execpromises))
            return EINVAL;
        if (m_has_execpromises && (new_execpromises & ~m_execpromises))
            return EPERM;
        m_has_execpromises = true;
        m_execpromises = new_execpromises;
    }

    return 0;
}

}