62 lines
2.4 KiB
Plaintext
62 lines
2.4 KiB
Plaintext
- Add retguard for arm64.
|
|
- Do not store the retguard cookie in frame in leaf functions if possible.
|
|
Makes things slightly faster and also improves security in these functions,
|
|
since the retguard cookie can't leak via the stack.
|
|
|
|
Index: lib/Target/AArch64/AArch64ReturnProtectorLowering.h
|
|
--- lib/Target/AArch64/AArch64ReturnProtectorLowering.h.orig
|
|
+++ lib/Target/AArch64/AArch64ReturnProtectorLowering.h
|
|
@@ -0,0 +1,52 @@
|
|
+//===-- AArch64ReturnProtectorLowering.h - --------------------- -*- C++ -*-==//
|
|
+//
|
|
+// The LLVM Compiler Infrastructure
|
|
+//
|
|
+// This file is distributed under the University of Illinois Open Source
|
|
+// License. See LICENSE.TXT for details.
|
|
+//
|
|
+//===----------------------------------------------------------------------===//
|
|
+//
|
|
+// This file contains the AArch64 implementation of ReturnProtectorLowering
|
|
+// class.
|
|
+//
|
|
+//===----------------------------------------------------------------------===//
|
|
+
|
|
+#ifndef LLVM_LIB_TARGET_AARCH64_AARCH64RETURNPROTECTORLOWERING_H
|
|
+#define LLVM_LIB_TARGET_AARCH64_AARCH64RETURNPROTECTORLOWERING_H
|
|
+
|
|
+#include "llvm/CodeGen/ReturnProtectorLowering.h"
|
|
+
|
|
+namespace llvm {
|
|
+
|
|
+class AArch64ReturnProtectorLowering : public ReturnProtectorLowering {
|
|
+public:
|
|
+ /// insertReturnProtectorPrologue/Epilogue - insert return protector
|
|
+ /// instrumentation in prologue or epilogue.
|
|
+ virtual void
|
|
+ insertReturnProtectorPrologue(MachineFunction &MF, MachineBasicBlock &MBB,
|
|
+ GlobalVariable *cookie) const override;
|
|
+ virtual void
|
|
+ insertReturnProtectorEpilogue(MachineFunction &MF, MachineInstr &MI,
|
|
+ GlobalVariable *cookie) const override;
|
|
+
|
|
+ /// opcodeIsReturn - Reuturn true is the given opcode is a return
|
|
+ /// instruction needing return protection, false otherwise.
|
|
+ virtual bool opcodeIsReturn(unsigned opcode) const override;
|
|
+
|
|
+ /// fillTempRegisters - Fill the list of available temp registers we can
|
|
+ /// use as a return protector register.
|
|
+ virtual void
|
|
+ fillTempRegisters(MachineFunction &MF,
|
|
+ std::vector<unsigned> &TempRegs) const override;
|
|
+
|
|
+ /// saveReturnProtectorRegister - Allows the target to save the
|
|
+ /// CalculationRegister in the CalleeSavedInfo vector if needed.
|
|
+ virtual void
|
|
+ saveReturnProtectorRegister(MachineFunction &MF,
|
|
+ std::vector<CalleeSavedInfo> &CSI) const override;
|
|
+};
|
|
+
|
|
+} // namespace llvm
|
|
+
|
|
+#endif
|