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/Range.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 = removeCodePtrTag<uintptr_t>(candidateAddress);
56 if (!m_range.contains(address))
57 return;
58 markSlow(address);
59 }
60
61 void prepareForConservativeScan();
62
63 void deleteUnmarkedJettisonedStubRoutines();
64
65 void traceMarkedStubRoutines(SlotVisitor&);
66
67private:
68 void markSlow(uintptr_t address);
69
70 struct Routine {
71 uintptr_t startAddress;
72 GCAwareJITStubRoutine* routine;
73 };
74 Vector<Routine> m_routines;
75 Range<uintptr_t> m_range { 0, 0 };
76};
77
78#else // !ENABLE(JIT)
79
80class JITStubRoutineSet {
81 WTF_MAKE_NONCOPYABLE(JITStubRoutineSet);
82 WTF_MAKE_FAST_ALLOCATED;
83
84public:
85 JITStubRoutineSet() { }
86 ~JITStubRoutineSet() { }
87
88 void add(GCAwareJITStubRoutine*) { }
89 void clearMarks() { }
90 void mark(void*) { }
91 void prepareForConservativeScan() { }
92 void deleteUnmarkedJettisonedStubRoutines() { }
93 void traceMarkedStubRoutines(SlotVisitor&) { }
94};
95
96#endif // !ENABLE(JIT)
97
98} // namespace JSC
99