Suzero. X8664. NaCl Sandboxing.

src/IceAssembler.h

 //===- subzero/src/IceAssembler.h - Integrated assembler --------*- C++ -*-===//
 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 //
 // Modified by the Subzero authors.
 //
 //===----------------------------------------------------------------------===//
 //
 //                        The Subzero Code Generator
@@ skipping 219 matching lines @@
 
   void extendCapacity();
 };
 
 class Assembler {
   Assembler() = delete;
   Assembler(const Assembler &) = delete;
   Assembler &operator=(const Assembler &) = delete;
 
 public:
+  using InternalRelocationList =
+      std::vector<std::pair<const IceString, const SizeT>>;
+
   enum AssemblerKind {
     Asm_ARM32,
     Asm_MIPS32,
     Asm_X8632,
     Asm_X8664,
   };
 
   virtual ~Assembler() = default;
 
   /// Allocate a chunk of bytes using the per-Assembler allocator.
@@ skipping 66 matching lines @@
   const IceString &getFunctionName() { return FunctionName; }
   void setFunctionName(const IceString &NewName) { FunctionName = NewName; }
   intptr_t getBufferSize() const { return Buffer.size(); }
   /// Roll back to a (smaller) size.
   void setBufferSize(intptr_t NewSize) { Buffer.setSize(NewSize); }
   void setPreliminary(bool Value) { Preliminary = Value; }
   bool getPreliminary() const { return Preliminary; }
 
   AssemblerKind getKind() const { return Kind; }
 
+  void addRelocationAtCurrentPosition(const IceString &RelocName) {
+    if (!getPreliminary()) {
+      InternalRelocs.emplace_back(RelocName, getBufferSize());
+    }
+  }
+
+  const InternalRelocationList &getInternalRelocations() const {
+    return InternalRelocs;
+  }
+
 protected:
   explicit Assembler(AssemblerKind Kind)
       : Kind(Kind), Allocator(), Buffer(*this) {}
 
 private:
   const AssemblerKind Kind;
+  InternalRelocationList InternalRelocs;
 
   ArenaAllocator<32 * 1024> Allocator;
   /// FunctionName and IsInternal are transferred from the original Cfg object,
   /// since the Cfg object may be deleted by the time the assembler buffer is
   /// emitted.
   IceString FunctionName = "";
   bool IsInternal = false;
   /// Preliminary indicates whether a preliminary pass is being made for
   /// calculating bundle padding (Preliminary=true), versus the final pass where
   /// all changes to label bindings, label links, and relocation fixups are
   /// fully committed (Preliminary=false).
   bool Preliminary = false;
 
   /// Installs a created fixup, after it has been allocated.
   void installFixup(AssemblerFixup *F) { Buffer.installFixup(F); }
 
 protected:
   // Buffer's constructor uses the Allocator, so it needs to appear after it.
   // TODO(jpp): dependencies on construction order are a nice way of shooting
   // yourself in the foot. Fix this.
   AssemblerBuffer Buffer;
 };
 
 } // end of namespace Ice
 
 #endif // SUBZERO_SRC_ICEASSEMBLER_H_