1/*
2 * Copyright (C) 2012 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 JITStubRoutineSet_h
27#define JITStubRoutineSet_h
28
29#include "JITStubRoutine.h"
30#include <wtf/FastMalloc.h>
31#include <wtf/HashMap.h>
32#include <wtf/Vector.h>
33
34namespace JSC {
35
36class GCAwareJITStubRoutine;
37class SlotVisitor;
38
39#if ENABLE(JIT)
40
41class JITStubRoutineSet {
42    WTF_MAKE_NONCOPYABLE(JITStubRoutineSet);
43    WTF_MAKE_FAST_ALLOCATED;
44
45public:
46    JITStubRoutineSet();
47    ~JITStubRoutineSet();
48
49    void add(GCAwareJITStubRoutine*);
50
51    void clearMarks();
52
53    void mark(void* candidateAddress)
54    {
55        uintptr_t address = reinterpret_cast<uintptr_t>(candidateAddress);
56        if (!JITStubRoutine::passesFilter(address))
57            return;
58
59        markSlow(address);
60    }
61
62    void deleteUnmarkedJettisonedStubRoutines();
63
64    void traceMarkedStubRoutines(SlotVisitor&);
65
66    unsigned size() const { return m_listOfRoutines.size(); }
67    GCAwareJITStubRoutine* at(unsigned i) const { return m_listOfRoutines[i]; }
68
69private:
70    void markSlow(uintptr_t address);
71
72    HashMap<uintptr_t, GCAwareJITStubRoutine*> m_addressToRoutineMap;
73    Vector<GCAwareJITStubRoutine*> m_listOfRoutines;
74};
75
76#else // !ENABLE(JIT)
77
78class JITStubRoutineSet {
79    WTF_MAKE_NONCOPYABLE(JITStubRoutineSet);
80    WTF_MAKE_FAST_ALLOCATED;
81
82public:
83    JITStubRoutineSet() { }
84    ~JITStubRoutineSet() { }
85
86    void add(GCAwareJITStubRoutine*) { }
87    void clearMarks() { }
88    void mark(void*) { }
89    void deleteUnmarkedJettisonedStubRoutines() { }
90    void traceMarkedStubRoutines(SlotVisitor&) { }
91};
92
93#endif // !ENABLE(JIT)
94
95} // namespace JSC
96
97#endif // JITStubRoutineSet_h
98
99