1/*
2 * Copyright (C) 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#if ENABLE(JIT)
29
30#include "MacroAssembler.h"
31#include <wtf/Vector.h>
32#include <wtf/text/CString.h>
33
34
35namespace JSC {
36
37class LinkBuffer;
38
39namespace Yarr {
40
41class YarrCodeBlock;
42
43class YarrJITInfo {
44public:
45 virtual ~YarrJITInfo() { };
46 virtual const char* variant() = 0;
47 virtual unsigned opCount() = 0;
48 virtual void dumpPatternString(PrintStream&) = 0;
49 virtual int dumpFor(PrintStream&, unsigned) = 0;
50};
51
52class YarrDisassembler {
53 WTF_MAKE_FAST_ALLOCATED;
54public:
55 YarrDisassembler(YarrJITInfo*);
56 ~YarrDisassembler();
57
58 void setStartOfCode(MacroAssembler::Label label) { m_startOfCode = label; }
59 void setForGenerate(unsigned opIndex, MacroAssembler::Label label)
60 {
61 m_labelForGenerateYarrOp[opIndex] = label;
62 }
63
64 void setForBacktrack(unsigned opIndex, MacroAssembler::Label label)
65 {
66 m_labelForBacktrackYarrOp[opIndex] = label;
67 }
68
69 void setEndOfGenerate(MacroAssembler::Label label) { m_endOfGenerate = label; }
70 void setEndOfBacktrack(MacroAssembler::Label label) { m_endOfBacktrack = label; }
71 void setEndOfCode(MacroAssembler::Label label) { m_endOfCode = label; }
72
73 void dump(LinkBuffer&);
74 void dump(PrintStream&, LinkBuffer&);
75
76private:
77 enum class VectorOrder {
78 IterateForward,
79 IterateReverse
80 };
81
82 void dumpHeader(PrintStream&, LinkBuffer&);
83 MacroAssembler::Label firstSlowLabel();
84
85 struct DumpedOp {
86 unsigned index;
87 CString disassembly;
88 };
89
90 const char* indentString(unsigned);
91 const char* indentString()
92 {
93 return indentString(m_indentLevel);
94 }
95
96 Vector<DumpedOp> dumpVectorForInstructions(LinkBuffer&, Vector<MacroAssembler::Label>& labels, MacroAssembler::Label endLabel, YarrDisassembler::VectorOrder vectorOrder = VectorOrder::IterateForward);
97
98 void dumpForInstructions(PrintStream&, LinkBuffer&, Vector<MacroAssembler::Label>& labels, MacroAssembler::Label endLabel, YarrDisassembler::VectorOrder vectorOrder = VectorOrder::IterateForward);
99
100 void dumpDisassembly(PrintStream&, const char* prefix, LinkBuffer&, MacroAssembler::Label from, MacroAssembler::Label to);
101
102 YarrJITInfo* m_jitInfo;
103 MacroAssembler::Label m_startOfCode;
104 Vector<MacroAssembler::Label> m_labelForGenerateYarrOp;
105 Vector<MacroAssembler::Label> m_labelForBacktrackYarrOp;
106 MacroAssembler::Label m_endOfGenerate;
107 MacroAssembler::Label m_endOfBacktrack;
108 MacroAssembler::Label m_endOfCode;
109 unsigned m_indentLevel { 0 };
110};
111
112}} // namespace Yarr namespace JSC
113
114#endif // ENABLE(JIT)
115