summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibWasm/AbstractMachine/Configuration.cpp
blob: 5bc86280084c0306ec6c131712bd7fcba9274926 (plain)
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
 * Copyright (c) 2021, Ali Mohammad Pur <mpfard@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <LibWasm/AbstractMachine/Configuration.h>
#include <LibWasm/AbstractMachine/Interpreter.h>

namespace Wasm {

Optional<Label> Configuration::nth_label(size_t i)
{
    for (size_t index = m_stack.size(); index > 0; --index) {
        auto& entry = m_stack.entries()[index - 1];
        if (auto ptr = entry.get_pointer<Label>()) {
            if (i == 0)
                return *ptr;
            --i;
        }
    }
    return {};
}

Result Configuration::call(FunctionAddress address, Vector<Value> arguments)
{
    auto* function = m_store.get(address);
    if (!function)
        return Trap {};
    if (auto* wasm_function = function->get_pointer<WasmFunction>()) {
        Vector<Value> locals;
        locals.ensure_capacity(arguments.size() + wasm_function->code().locals().size());
        for (auto& value : arguments)
            locals.append(Value { value });
        for (auto& type : wasm_function->code().locals())
            locals.empend(type, 0ull);

        auto frame = make<Frame>(
            wasm_function->module(),
            move(locals),
            wasm_function->code().body(),
            wasm_function->type().results().size());

        set_frame(move(frame));
        return execute();
    }

    // It better be a host function, else something is really wrong.
    auto& host_function = function->get<HostFunction>();
    return host_function.function()(*this, arguments);
}

Result Configuration::execute()
{
    Interpreter interpreter;
    interpreter.pre_interpret_hook = pre_interpret_hook;
    interpreter.post_interpret_hook = post_interpret_hook;

    interpreter.interpret(*this);
    if (interpreter.did_trap())
        return Trap {};

    Vector<Value> results;
    results.ensure_capacity(m_current_frame->arity());
    for (size_t i = 0; i < m_current_frame->arity(); ++i)
        results.append(move(stack().pop().get<Value>()));
    auto label = stack().pop();
    // ASSERT: label == current frame
    if (!label.has<Label>())
        return Trap {};
    return Result { move(results) };
}

void Configuration::dump_stack()
{
    for (const auto& entry : stack().entries()) {
        entry.visit(
            [](const Value& v) {
                v.value().visit([]<typename T>(const T& v) {
                    if constexpr (IsIntegral<T> || IsFloatingPoint<T>)
                        dbgln("    {}", v);
                    else
                        dbgln("    *{}", v.value());
                });
            },
            [](const NonnullOwnPtr<Frame>& f) {
                dbgln("    frame({})", f->arity());
                for (auto& local : f->locals()) {
                    local.value().visit([]<typename T>(const T& v) {
                        if constexpr (IsIntegral<T> || IsFloatingPoint<T>)
                            dbgln("        {}", v);
                        else
                            dbgln("        *{}", v.value());
                    });
                }
            },
            [](const Label& l) {
                dbgln("    label({}) -> {}", l.arity(), l.continuation());
            });
    }
}

}