summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibWeb/WebAssembly/WebAssemblyObject.h
blob: 3758ab9d39f6156a8ee7975cd25b32e627661238 (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
104
105
106
107
108
109
110
111
112
113
114
/*
 * Copyright (c) 2021, Ali Mohammad Pur <mpfard@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibJS/Runtime/Object.h>
#include <LibWasm/AbstractMachine/AbstractMachine.h>
#include <LibWeb/Forward.h>
#include <LibWeb/WebAssembly/WebAssemblyObjectPrototype.h>

namespace Web::Bindings {

class WebAssemblyMemoryObject;

class WebAssemblyObject final : public JS::Object {
    JS_OBJECT(WebAssemblyObject, JS::Object);

public:
    explicit WebAssemblyObject(JS::GlobalObject&);
    virtual void initialize(JS::GlobalObject&) override;
    virtual ~WebAssemblyObject() override = default;

    virtual void visit_edges(Cell::Visitor&) override;

    struct CompiledWebAssemblyModule {
        explicit CompiledWebAssemblyModule(Wasm::Module&& module)
            : module(move(module))
        {
        }

        Wasm::Module module;
    };

    // FIXME: These should just be members of the module (instance) object,
    //        but the module needs to stick around while its instance is alive
    //        so ideally this would be a refcounted object, shared between
    //        WebAssemblyModuleObject's and WebAssemblyInstantiatedModuleObject's.
    struct ModuleCache {
        HashMap<Wasm::FunctionAddress, JS::Function*> function_instances;
        HashMap<Wasm::MemoryAddress, WebAssemblyMemoryObject*> memory_instances;
    };
    struct GlobalModuleCache {
        HashMap<Wasm::FunctionAddress, JS::NativeFunction*> function_instances;
    };

    static NonnullOwnPtrVector<CompiledWebAssemblyModule> s_compiled_modules;
    static NonnullOwnPtrVector<Wasm::ModuleInstance> s_instantiated_modules;
    static Vector<ModuleCache> s_module_caches;
    static GlobalModuleCache s_global_cache;

    static Wasm::AbstractMachine s_abstract_machine;

private:
    JS_DECLARE_NATIVE_FUNCTION(validate);
    JS_DECLARE_NATIVE_FUNCTION(compile);
    JS_DECLARE_NATIVE_FUNCTION(instantiate);
};

class WebAssemblyModuleObject final : public JS::Object {
    JS_OBJECT(WebAssemblyModuleObject, JS::Object);

public:
    explicit WebAssemblyModuleObject(JS::GlobalObject&, size_t index);
    virtual ~WebAssemblyModuleObject() override = default;

    size_t index() const { return m_index; }
    const Wasm::Module& module() const { return WebAssemblyObject::s_compiled_modules.at(m_index).module; }

private:
    size_t m_index { 0 };
};

class WebAssemblyInstanceObject final : public JS::Object {
    JS_OBJECT(WebAssemblyInstanceObject, JS::Object);

public:
    explicit WebAssemblyInstanceObject(JS::GlobalObject&, size_t index);
    virtual void initialize(JS::GlobalObject&) override;
    virtual ~WebAssemblyInstanceObject() override = default;

    size_t index() const { return m_index; }
    Wasm::ModuleInstance& instance() const { return WebAssemblyObject::s_instantiated_modules.at(m_index); }
    auto& cache() { return WebAssemblyObject::s_module_caches.at(m_index); }

    void visit_edges(Cell::Visitor&) override;

    friend class WebAssemblyInstancePrototype;

private:
    size_t m_index { 0 };
    JS::Object* m_exports_object { nullptr };
};

class WebAssemblyMemoryObject final : public JS::Object {
    JS_OBJECT(WebAssemblyModuleObject, JS::Object);

public:
    explicit WebAssemblyMemoryObject(JS::GlobalObject&, Wasm::MemoryAddress);
    virtual void initialize(JS::GlobalObject&) override;
    virtual ~WebAssemblyMemoryObject() override = default;

    auto address() const { return m_address; }

private:
    JS_DECLARE_NATIVE_FUNCTION(grow);
    JS_DECLARE_NATIVE_GETTER(buffer);

    Wasm::MemoryAddress m_address;
};

}