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