1/*
2 * Copyright (C) 2016-2018 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#if ENABLE(WEBASSEMBLY)
29
30#include "CompilationResult.h"
31#include "WasmB3IRGenerator.h"
32#include "WasmEntryPlan.h"
33#include "WasmModuleInformation.h"
34#include "WasmTierUpCount.h"
35#include <wtf/Bag.h>
36#include <wtf/Function.h>
37#include <wtf/SharedTask.h>
38#include <wtf/ThreadSafeRefCounted.h>
39#include <wtf/Vector.h>
40
41namespace JSC {
42
43class CallLinkInfo;
44
45namespace Wasm {
46
47class CodeBlock;
48
49class BBQPlan final : public EntryPlan {
50public:
51 using Base = EntryPlan;
52
53 using Base::Base;
54
55 BBQPlan(Context*, Ref<ModuleInformation>, uint32_t functionIndex, CodeBlock*, CompletionTask&&);
56
57 void work(CompilationEffort) override;
58 void initializeCallees(const CalleeInitializer&) override;
59
60protected:
61 bool prepareImpl() override;
62 void compileFunction(uint32_t functionIndex) override;
63 void didCompleteCompilation(const AbstractLocker&) override;
64
65private:
66 std::unique_ptr<InternalFunction> compileFunction(uint32_t functionIndex, CompilationContext&, Vector<UnlinkedWasmToWasmCall>&, TierUpCount*);
67
68 Vector<std::unique_ptr<InternalFunction>> m_wasmInternalFunctions;
69 HashMap<uint32_t, std::unique_ptr<InternalFunction>, typename DefaultHash<uint32_t>::Hash, WTF::UnsignedWithZeroKeyHashTraits<uint32_t>> m_embedderToWasmInternalFunctions;
70 Vector<CompilationContext> m_compilationContexts;
71 Vector<std::unique_ptr<TierUpCount>> m_tierUpCounts;
72
73 RefPtr<CodeBlock> m_codeBlock { nullptr };
74 uint32_t m_functionIndex;
75};
76
77
78} } // namespace JSC::Wasm
79
80#endif // ENABLE(WEBASSEMBLY)
81