| Index: src/wasm/decoder.h
|
| diff --git a/src/wasm/decoder.h b/src/wasm/decoder.h
|
| index 4b35685bd89c7027ed39ee4d6657a5441ed84c4d..2dfdd02aed912387d75b4c31cef0656a26ca5263 100644
|
| --- a/src/wasm/decoder.h
|
| +++ b/src/wasm/decoder.h
|
| @@ -77,40 +77,49 @@
|
| return check(base, offset, 8, msg) ? read_u64(base + offset) : 0;
|
| }
|
|
|
| - // Reads a variable-length unsigned integer (little endian).
|
| uint32_t checked_read_u32v(const byte* base, int offset, int* length,
|
| - const char* msg = "expected LEB32") {
|
| - return checked_read_leb<uint32_t>(base, offset, length, msg);
|
| + const char* msg = "expected LEB128") {
|
| + if (!check(base, offset, 1, msg)) {
|
| + *length = 0;
|
| + return 0;
|
| + }
|
| +
|
| + const ptrdiff_t kMaxDiff = 5; // maximum 5 bytes.
|
| + const byte* ptr = base + offset;
|
| + const byte* end = ptr + kMaxDiff;
|
| + if (end > limit_) end = limit_;
|
| + int shift = 0;
|
| + byte b = 0;
|
| + uint32_t result = 0;
|
| + while (ptr < end) {
|
| + b = *ptr++;
|
| + result = result | ((b & 0x7F) << shift);
|
| + if ((b & 0x80) == 0) break;
|
| + shift += 7;
|
| + }
|
| + DCHECK_LE(ptr - (base + offset), kMaxDiff);
|
| + *length = static_cast<int>(ptr - (base + offset));
|
| + if (ptr == end) {
|
| + if (*length == kMaxDiff && (b & 0xF0) != 0) {
|
| + error(base, ptr, "extra bits in LEB128");
|
| + return 0;
|
| + }
|
| + if ((b & 0x80) != 0) {
|
| + error(base, ptr, msg);
|
| + return 0;
|
| + }
|
| + }
|
| + return result;
|
| }
|
|
|
| // Reads a variable-length signed integer (little endian).
|
| int32_t checked_read_i32v(const byte* base, int offset, int* length,
|
| - const char* msg = "expected SLEB32") {
|
| + const char* msg = "expected SLEB128") {
|
| uint32_t result = checked_read_u32v(base, offset, length, msg);
|
| if (*length == 5) return bit_cast<int32_t>(result);
|
| if (*length > 0) {
|
| int shift = 32 - 7 * *length;
|
| - // Perform sign extension.
|
| return bit_cast<int32_t>(result << shift) >> shift;
|
| - }
|
| - return 0;
|
| - }
|
| -
|
| - // Reads a variable-length unsigned integer (little endian).
|
| - uint64_t checked_read_u64v(const byte* base, int offset, int* length,
|
| - const char* msg = "expected LEB64") {
|
| - return checked_read_leb<uint64_t>(base, offset, length, msg);
|
| - }
|
| -
|
| - // Reads a variable-length signed integer (little endian).
|
| - int64_t checked_read_i64v(const byte* base, int offset, int* length,
|
| - const char* msg = "expected SLEB64") {
|
| - uint64_t result = checked_read_u64v(base, offset, length, msg);
|
| - if (*length == 10) return bit_cast<int64_t>(result);
|
| - if (*length > 0) {
|
| - int shift = 64 - 7 * *length;
|
| - // Perform sign extension.
|
| - return bit_cast<int64_t>(result << shift) >> shift;
|
| }
|
| return 0;
|
| }
|
| @@ -324,46 +333,6 @@
|
| const byte* error_pc_;
|
| const byte* error_pt_;
|
| base::SmartArrayPointer<char> error_msg_;
|
| -
|
| - private:
|
| - template <typename IntType>
|
| - IntType checked_read_leb(const byte* base, int offset, int* length,
|
| - const char* msg) {
|
| - if (!check(base, offset, 1, msg)) {
|
| - *length = 0;
|
| - return 0;
|
| - }
|
| -
|
| - const int kMaxLength = (sizeof(IntType) * 8 + 6) / 7;
|
| - const byte* ptr = base + offset;
|
| - const byte* end = ptr + kMaxLength;
|
| - if (end > limit_) end = limit_;
|
| - int shift = 0;
|
| - byte b = 0;
|
| - IntType result = 0;
|
| - while (ptr < end) {
|
| - b = *ptr++;
|
| - result = result | (static_cast<IntType>(b & 0x7F) << shift);
|
| - if ((b & 0x80) == 0) break;
|
| - shift += 7;
|
| - }
|
| - DCHECK_LE(ptr - (base + offset), kMaxLength);
|
| - *length = static_cast<int>(ptr - (base + offset));
|
| - if (ptr == end) {
|
| - // Check there are no bits set beyond the bitwidth of {IntType}.
|
| - const int kExtraBits = (1 + kMaxLength * 7) - (sizeof(IntType) * 8);
|
| - const byte kExtraBitsMask = static_cast<byte>(0xFF << (8 - kExtraBits));
|
| - if (*length == kMaxLength && (b & kExtraBitsMask) != 0) {
|
| - error(base, ptr, "extra bits in varint");
|
| - return 0;
|
| - }
|
| - if ((b & 0x80) != 0) {
|
| - error(base, ptr, msg);
|
| - return 0;
|
| - }
|
| - }
|
| - return result;
|
| - }
|
| };
|
|
|
| #undef TRACE
|
|
|