function-body-decoder.h 6.39 KB
Newer Older
1 2 3 4
// Copyright 2015 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

5 6
#ifndef V8_WASM_FUNCTION_BODY_DECODER_H_
#define V8_WASM_FUNCTION_BODY_DECODER_H_
7

8
#include "src/base/compiler-specific.h"
9
#include "src/base/iterator.h"
10
#include "src/common/globals.h"
11
#include "src/wasm/decoder.h"
12 13 14 15 16 17
#include "src/wasm/wasm-opcodes.h"
#include "src/wasm/wasm-result.h"

namespace v8 {
namespace internal {

18 19
class BitVector;  // forward declaration

20 21
namespace wasm {

22
class WasmFeatures;
23
struct WasmModule;  // forward declaration of module interface.
24

25
// A wrapper around the signature and bytes of a function.
26
struct FunctionBody {
27 28 29 30
  const FunctionSig* sig;  // function signature
  uint32_t offset;         // offset in the module bytes, for error reporting
  const byte* start;       // start of the function body
  const byte* end;         // end of the function body
31

32
  FunctionBody(const FunctionSig* sig, uint32_t offset, const byte* start,
33 34 35
               const byte* end)
      : sig(sig), offset(offset), start(start), end(end) {}
};
36

37 38
enum class LoadTransformationKind : uint8_t { kSplat, kExtend };

39 40 41 42 43
V8_EXPORT_PRIVATE DecodeResult VerifyWasmCode(AccountingAllocator* allocator,
                                              const WasmFeatures& enabled,
                                              const WasmModule* module,
                                              WasmFeatures* detected,
                                              const FunctionBody& body);
44

45 46
enum PrintLocals { kPrintLocals, kOmitLocals };
V8_EXPORT_PRIVATE
47
bool PrintRawWasmCode(AccountingAllocator* allocator, const FunctionBody& body,
48
                      const WasmModule* module, PrintLocals print_locals);
49

50 51
V8_EXPORT_PRIVATE
bool PrintRawWasmCode(AccountingAllocator* allocator, const FunctionBody& body,
52
                      const WasmModule* module, PrintLocals print_locals,
53 54
                      std::ostream& out,
                      std::vector<int>* line_numbers = nullptr);
55

56
// A simplified form of AST printing, e.g. from a debugger.
57
void PrintRawWasmCode(const byte* start, const byte* end);
58

59
struct BodyLocalDecls {
60
  // The size of the encoded declarations.
61
  uint32_t encoded_size = 0;  // size of encoded declarations
62

63
  ZoneVector<ValueType> type_list;
64

65
  explicit BodyLocalDecls(Zone* zone) : type_list(zone) {}
66 67
};

68 69
V8_EXPORT_PRIVATE bool DecodeLocalDecls(const WasmFeatures& enabled,
                                        BodyLocalDecls* decls,
70
                                        const byte* start, const byte* end);
71

72 73 74 75
V8_EXPORT_PRIVATE BitVector* AnalyzeLoopAssignmentForTesting(Zone* zone,
                                                             size_t num_locals,
                                                             const byte* start,
                                                             const byte* end);
76

77
// Computes the length of the opcode at the given address.
78
V8_EXPORT_PRIVATE unsigned OpcodeLength(const byte* pc, const byte* end);
79

80 81 82 83 84
// Computes the stack effect of the opcode at the given address.
// Returns <pop count, push count>.
// Be cautious with control opcodes: This function only covers their immediate,
// local stack effect (e.g. BrIf pops 1, Br pops 0). Those opcodes can have
// non-local stack effect though, which are not covered here.
85 86 87 88
// TODO(clemensb): This is only used by the interpreter; move there.
V8_EXPORT_PRIVATE std::pair<uint32_t, uint32_t> StackEffect(
    const WasmModule* module, const FunctionSig* sig, const byte* pc,
    const byte* end);
89

90
// A simple forward iterator for bytecodes.
91
class V8_EXPORT_PRIVATE BytecodeIterator : public NON_EXPORTED_BASE(Decoder) {
92 93
  // Base class for both iterators defined below.
  class iterator_base {
94
   public:
95
    inline iterator_base& operator++() {
96 97 98 99
      DCHECK_LT(ptr_, end_);
      ptr_ += OpcodeLength(ptr_, end_);
      return *this;
    }
100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
    inline bool operator==(const iterator_base& that) {
      return this->ptr_ == that.ptr_;
    }
    inline bool operator!=(const iterator_base& that) {
      return this->ptr_ != that.ptr_;
    }

   protected:
    const byte* ptr_;
    const byte* end_;
    iterator_base(const byte* ptr, const byte* end) : ptr_(ptr), end_(end) {}
  };

 public:
  // If one wants to iterate over the bytecode without looking at {pc_offset()}.
  class opcode_iterator
      : public iterator_base,
117
        public base::iterator<std::input_iterator_tag, WasmOpcode> {
118
   public:
119 120 121 122
    inline WasmOpcode operator*() {
      DCHECK_LT(ptr_, end_);
      return static_cast<WasmOpcode>(*ptr_);
    }
123 124 125 126 127 128 129 130 131 132

   private:
    friend class BytecodeIterator;
    opcode_iterator(const byte* ptr, const byte* end)
        : iterator_base(ptr, end) {}
  };
  // If one wants to iterate over the instruction offsets without looking at
  // opcodes.
  class offset_iterator
      : public iterator_base,
133
        public base::iterator<std::input_iterator_tag, uint32_t> {
134 135 136 137
   public:
    inline uint32_t operator*() {
      DCHECK_LT(ptr_, end_);
      return static_cast<uint32_t>(ptr_ - start_);
138 139 140
    }

   private:
141
    const byte* start_;
142
    friend class BytecodeIterator;
143 144
    offset_iterator(const byte* start, const byte* ptr, const byte* end)
        : iterator_base(ptr, end), start_(start) {}
145 146 147 148 149 150
  };

  // Create a new {BytecodeIterator}. If the {decls} pointer is non-null,
  // assume the bytecode starts with local declarations and decode them.
  // Otherwise, do not decode local decls.
  BytecodeIterator(const byte* start, const byte* end,
151
                   BodyLocalDecls* decls = nullptr);
152

153 154 155 156 157 158 159 160 161 162
  base::iterator_range<opcode_iterator> opcodes() {
    return base::iterator_range<opcode_iterator>(opcode_iterator(pc_, end_),
                                                 opcode_iterator(end_, end_));
  }

  base::iterator_range<offset_iterator> offsets() {
    return base::iterator_range<offset_iterator>(
        offset_iterator(start_, pc_, end_),
        offset_iterator(start_, end_, end_));
  }
163 164

  WasmOpcode current() {
165 166
    return static_cast<WasmOpcode>(
        read_u8<Decoder::kNoValidate>(pc_, "expected bytecode"));
167 168 169 170 171 172 173 174 175 176
  }

  void next() {
    if (pc_ < end_) {
      pc_ += OpcodeLength(pc_, end_);
      if (pc_ >= end_) pc_ = end_;
    }
  }

  bool has_next() { return pc_ < end_; }
177 178

  WasmOpcode prefixed_opcode() {
179
    return read_prefixed_opcode<Decoder::kNoValidate>(pc_);
180
  }
181 182
};

183 184 185 186
}  // namespace wasm
}  // namespace internal
}  // namespace v8

187
#endif  // V8_WASM_FUNCTION_BODY_DECODER_H_