1/*
2 * Copyright (C) 2013-2019 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(FTL_JIT)
29
30#include "FTLLocation.h"
31#include "FTLSlowPathCallKey.h"
32#include "MacroAssemblerCodeRef.h"
33#include <wtf/HashMap.h>
34
35namespace JSC {
36
37class VM;
38
39namespace FTL {
40
41MacroAssemblerCodeRef<JITThunkPtrTag> osrExitGenerationThunkGenerator(VM&);
42MacroAssemblerCodeRef<JITThunkPtrTag> lazySlowPathGenerationThunkGenerator(VM&);
43MacroAssemblerCodeRef<JITThunkPtrTag> slowPathCallThunkGenerator(VM&, const SlowPathCallKey&);
44
45template<typename KeyTypeArgument>
46struct ThunkMap {
47 typedef KeyTypeArgument KeyType;
48 typedef HashMap<KeyType, MacroAssemblerCodeRef<JITThunkPtrTag>> ToThunkMap;
49 typedef HashMap<MacroAssemblerCodePtr<JITThunkPtrTag>, KeyType> FromThunkMap;
50
51 ToThunkMap m_toThunk;
52 FromThunkMap m_fromThunk;
53};
54
55template<typename MapType, typename GeneratorType>
56MacroAssemblerCodeRef<JITThunkPtrTag> generateIfNecessary(
57 VM& vm, MapType& map, const typename MapType::KeyType& key, GeneratorType generator)
58{
59 typename MapType::ToThunkMap::iterator iter = map.m_toThunk.find(key);
60 if (iter != map.m_toThunk.end())
61 return iter->value;
62
63 MacroAssemblerCodeRef<JITThunkPtrTag> result = generator(vm, key);
64 map.m_toThunk.add(key, result);
65 map.m_fromThunk.add(result.code(), key);
66 return result;
67}
68
69template<typename MapType>
70typename MapType::KeyType keyForThunk(MapType& map, MacroAssemblerCodePtr<JITThunkPtrTag> ptr)
71{
72 typename MapType::FromThunkMap::iterator iter = map.m_fromThunk.find(ptr);
73 RELEASE_ASSERT(iter != map.m_fromThunk.end());
74 return iter->value;
75}
76
77class Thunks {
78 WTF_MAKE_FAST_ALLOCATED;
79 WTF_MAKE_NONCOPYABLE(Thunks);
80public:
81 Thunks() = default;
82 MacroAssemblerCodeRef<JITThunkPtrTag> getSlowPathCallThunk(VM& vm, const SlowPathCallKey& key)
83 {
84 return generateIfNecessary(vm, m_slowPathCallThunks, key, slowPathCallThunkGenerator);
85 }
86
87 SlowPathCallKey keyForSlowPathCallThunk(MacroAssemblerCodePtr<JITThunkPtrTag> ptr)
88 {
89 return keyForThunk(m_slowPathCallThunks, ptr);
90 }
91
92private:
93 ThunkMap<SlowPathCallKey> m_slowPathCallThunks;
94};
95
96} } // namespace JSC::FTL
97
98#endif // ENABLE(FTL_JIT)
99