AArch64MCLFIRewriter.h

Go to the documentation of this file.

//===- AArch64MCLFIRewriter.h -----------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file declares the AArch64MCLFIRewriter class, the AArch64 specific
// subclass of MCLFIRewriter.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64MCLFIREWRITER_H
#define LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64MCLFIREWRITER_H
 
#include "AArch64AddressingModes.h"
#include "llvm/MC/MCInstrInfo.h"
#include "llvm/MC/MCLFIRewriter.h"
#include "llvm/MC/MCRegister.h"
#include "llvm/MC/MCRegisterInfo.h"
 
#include <optional>
 
namespace llvm {
class MCContext;
class MCExpr;
class MCInst;
class MCOperand;
class MCStreamer;
class MCSubtargetInfo;
class MCSymbol;
 
/// Rewrites AArch64 instructions for LFI sandboxing.
///
/// This class implements the LFI (Lightweight Fault Isolation) rewriting
/// for AArch64 instructions. It transforms instructions to ensure memory
/// accesses and control flow are confined within the sandbox region.
///
/// Reserved registers:
/// - X27: Sandbox base address (always holds the base)
/// - X28: Safe address register (always within sandbox)
/// - X26: Scratch register for intermediate calculations
/// - X25: context register (points to thread-local runtime data)
/// - SP:  Stack pointer (always within sandbox)
/// - X30: Link register (always within sandbox)
class AArch64MCLFIRewriter : public MCLFIRewriter {
public:
  AArch64MCLFIRewriter(MCContext &Ctx, std::unique_ptr<MCRegisterInfo> &&RI,
                       std::unique_ptr<MCInstrInfo> &&II)
      : MCLFIRewriter(Ctx, std::move(RI), std::move(II)) {}
 
  bool rewriteInst(const MCInst &Inst, MCStreamer &Out,
                   const MCSubtargetInfo &STI) override;
 
  void onLabel(const MCSymbol *Symbol) override;
 
private:
  /// Recursion guard to prevent infinite loops when emitting instructions.
  bool Guard = false;
 
  /// Deferred `.tlsdesccall` symbol. The directive attaches a
  /// R_AARCH64_TLSDESC_CALL relocation to the following BLR. Since LFI inserts
  /// a guard before that BLR, the marker is deferred and re-emitted between
  /// the guard and the branch so the relocation stays on the BLR.
  const MCExpr *PendingTLSDescCall = nullptr;
 
  /// Rewriter state for implementing the guard-elimination optimization, which
  /// allows redundant add masks to be skipped. When it holds a value, x28 is
  /// known to already hold the guarded value of that register.
  std::optional<MCRegister> ActiveGuardReg;
 
  // Instruction classification. Returns the reserved register that may be
  // modified, or an invalid register if no reserved register is touched.
  MCRegister mayModifyReserved(const MCInst &Inst) const;
  bool mayModifySP(const MCInst &Inst) const;
 
  // Instruction emission.
  void emitInst(const MCInst &Inst, MCStreamer &Out,
                const MCSubtargetInfo &STI);
  void emitAddMask(MCRegister Dest, MCRegister Src, MCStreamer &Out,
                   const MCSubtargetInfo &STI);
  void emitBranch(unsigned Opcode, MCRegister Target, MCStreamer &Out,
                  const MCSubtargetInfo &STI);
  void emitPendingTLSDescCall(MCStreamer &Out, const MCSubtargetInfo &STI);
  void emitMov(MCRegister Dest, MCRegister Src, MCStreamer &Out,
               const MCSubtargetInfo &STI);
  void emitAddImm(MCRegister Dest, MCRegister Src, int64_t Imm, MCStreamer &Out,
                  const MCSubtargetInfo &STI);
  void emitAddReg(MCRegister Dest, MCRegister Src1, MCRegister Src2,
                  unsigned Shift, MCStreamer &Out, const MCSubtargetInfo &STI);
  void emitAddRegExtend(MCRegister Dest, MCRegister Src1, MCRegister Src2,
                        AArch64_AM::ShiftExtendType ExtType, unsigned Shift,
                        MCStreamer &Out, const MCSubtargetInfo &STI);
  void emitMemRoW(unsigned Opcode, const MCOperand &DataOp, MCRegister BaseReg,
                  MCStreamer &Out, const MCSubtargetInfo &STI);
 
  // Rewriting logic.
  void doRewriteInst(const MCInst &Inst, MCStreamer &Out,
                     const MCSubtargetInfo &STI);
 
  // Control flow.
  void rewriteIndirectBranch(const MCInst &Inst, MCStreamer &Out,
                             const MCSubtargetInfo &STI);
  void rewriteReturn(const MCInst &Inst, MCStreamer &Out,
                     const MCSubtargetInfo &STI);
 
  // Memory access.
  void rewriteLoadStore(const MCInst &Inst, MCStreamer &Out,
                        const MCSubtargetInfo &STI);
  void rewriteLoadStoreBase(const MCInst &Inst, MCStreamer &Out,
                            const MCSubtargetInfo &STI);
  bool rewriteLoadStoreRoW(const MCInst &Inst, MCStreamer &Out,
                           const MCSubtargetInfo &STI);
 
  // SP register modification.
  void rewriteSPModification(const MCInst &Inst, MCStreamer &Out,
                             const MCSubtargetInfo &STI);
 
  // Link register modification.
  void rewriteLRModification(const MCInst &Inst, MCStreamer &Out,
                             const MCSubtargetInfo &STI);
 
  // System instructions.
  void rewriteSyscall(const MCInst &Inst, MCStreamer &Out,
                      const MCSubtargetInfo &STI);
  void rewriteTPRead(const MCInst &Inst, MCStreamer &Out,
                     const MCSubtargetInfo &STI);
  void rewriteTPWrite(const MCInst &Inst, MCStreamer &Out,
                      const MCSubtargetInfo &STI);
  void rewriteVASysOp(const MCInst &Inst, MCStreamer &Out,
                      const MCSubtargetInfo &STI);
};
 
/// Returns true if \p Opcode is a pre- or post-indexed memory access that the
/// LFI rewriter expands with a base-register update (i.e. an extra
/// instruction beyond the guard + access pair).
bool isLFIPrePostMemAccess(unsigned Opcode);
 
} // namespace llvm
 
#endif // LLVM_LIB_TARGET_AARCH64_MCTARGETDESC_AARCH64MCLFIREWRITER_H