46a7b39b848e8e7881928a02ec552800c4572ffc
[WebKit-https.git] / Source / JavaScriptCore / wasm / WasmParser.h
1 /*
2  * Copyright (C) 2016 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(WEBASSEMBLY)
29
30 #include "B3Compilation.h"
31 #include "B3Procedure.h"
32 #include "WasmFormat.h"
33 #include "WasmOps.h"
34 #include "WasmSections.h"
35 #include <wtf/LEBDecoder.h>
36
37 namespace JSC { namespace Wasm {
38
39 class Parser {
40 protected:
41     Parser(const uint8_t*, size_t);
42
43     bool WARN_UNUSED_RETURN consumeCharacter(char);
44     bool WARN_UNUSED_RETURN consumeString(const char*);
45
46     bool WARN_UNUSED_RETURN parseVarUInt1(uint8_t& result);
47     bool WARN_UNUSED_RETURN parseUInt7(uint8_t& result);
48     bool WARN_UNUSED_RETURN parseUInt32(uint32_t& result);
49     bool WARN_UNUSED_RETURN parseVarUInt32(uint32_t& result) { return WTF::LEBDecoder::decodeUInt32(m_source, m_sourceLength, m_offset, result); }
50     bool WARN_UNUSED_RETURN parseVarUInt64(uint64_t& result) { return WTF::LEBDecoder::decodeUInt64(m_source, m_sourceLength, m_offset, result); }
51
52     bool WARN_UNUSED_RETURN parseValueType(Type& result);
53
54     const uint8_t* source() const { return m_source; }
55     size_t length() const { return m_sourceLength; }
56
57     size_t m_offset = 0;
58
59 private:
60     const uint8_t* m_source;
61     size_t m_sourceLength;
62 };
63
64 ALWAYS_INLINE Parser::Parser(const uint8_t* sourceBuffer, size_t sourceLength)
65     : m_source(sourceBuffer)
66     , m_sourceLength(sourceLength)
67 {
68 }
69
70 ALWAYS_INLINE bool Parser::consumeCharacter(char c)
71 {
72     if (m_offset >= length())
73         return false;
74     if (c == source()[m_offset]) {
75         m_offset++;
76         return true;
77     }
78     return false;
79 }
80
81 ALWAYS_INLINE bool Parser::consumeString(const char* str)
82 {
83     unsigned start = m_offset;
84     if (m_offset >= length())
85         return false;
86     for (size_t i = 0; str[i]; i++) {
87         if (!consumeCharacter(str[i])) {
88             m_offset = start;
89             return false;
90         }
91     }
92     return true;
93 }
94
95 ALWAYS_INLINE bool Parser::parseUInt32(uint32_t& result)
96 {
97     if (length() < 4 || m_offset > length() - 4)
98         return false;
99     result = *reinterpret_cast<const uint32_t*>(source() + m_offset);
100     m_offset += 4;
101     return true;
102 }
103
104 ALWAYS_INLINE bool Parser::parseUInt7(uint8_t& result)
105 {
106     if (m_offset >= length())
107         return false;
108     result = source()[m_offset++];
109     return result < 0x80;
110 }
111
112 ALWAYS_INLINE bool Parser::parseVarUInt1(uint8_t& result)
113 {
114     uint32_t temp;
115     if (!parseVarUInt32(temp))
116         return false;
117     result = static_cast<uint8_t>(temp);
118     return temp <= 1;
119 }
120
121 ALWAYS_INLINE bool Parser::parseValueType(Type& result)
122 {
123     uint8_t value;
124     if (!parseUInt7(value))
125         return false;
126     if (value >= static_cast<uint8_t>(Type::LastValueType))
127         return false;
128     result = static_cast<Type>(value);
129     return true;
130 }
131
132 } } // namespace JSC::Wasm
133
134 #endif // ENABLE(WEBASSEMBLY)