1/*
2 * Copyright (C) 2013 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#ifndef FTLThunks_h
27#define FTLThunks_h
28
29#if ENABLE(FTL_JIT)
30
31#include "FTLLocation.h"
32#include "FTLSlowPathCallKey.h"
33#include "MacroAssemblerCodeRef.h"
34#include <wtf/HashMap.h>
35
36namespace JSC {
37
38class VM;
39
40namespace FTL {
41
42MacroAssemblerCodeRef osrExitGenerationThunkGenerator(VM*);
43MacroAssemblerCodeRef slowPathCallThunkGenerator(VM&, const SlowPathCallKey&);
44
45template<typename KeyTypeArgument>
46struct ThunkMap {
47    typedef KeyTypeArgument KeyType;
48    typedef HashMap<KeyType, MacroAssemblerCodeRef> ToThunkMap;
49    typedef HashMap<MacroAssemblerCodePtr, KeyType> FromThunkMap;
50
51    ToThunkMap m_toThunk;
52    FromThunkMap m_fromThunk;
53};
54
55template<typename MapType, typename GeneratorType>
56MacroAssemblerCodeRef 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 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 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 {
78public:
79    MacroAssemblerCodeRef getSlowPathCallThunk(VM& vm, const SlowPathCallKey& key)
80    {
81        return generateIfNecessary(
82            vm, m_slowPathCallThunks, key, slowPathCallThunkGenerator);
83    }
84
85    SlowPathCallKey keyForSlowPathCallThunk(MacroAssemblerCodePtr ptr)
86    {
87        return keyForThunk(m_slowPathCallThunks, ptr);
88    }
89
90private:
91    ThunkMap<SlowPathCallKey> m_slowPathCallThunks;
92};
93
94} } // namespace JSC::FTL
95
96#endif // ENABLE(FTL_JIT)
97
98#endif // FTLTHunks_h
99