52 lines
1.9 KiB
Plaintext
52 lines
1.9 KiB
Plaintext
Refactor retguard to make adding additional arches easier.
|
|
|
|
Index: lib/Target/X86/X86ReturnProtectorLowering.h
|
|
--- lib/Target/X86/X86ReturnProtectorLowering.h.orig
|
|
+++ lib/Target/X86/X86ReturnProtectorLowering.h
|
|
@@ -0,0 +1,45 @@
|
|
+//===-- X86ReturnProtectorLowering.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 X86 implementation of ReturnProtectorLowering class.
|
|
+//
|
|
+//===----------------------------------------------------------------------===//
|
|
+
|
|
+#ifndef LLVM_LIB_TARGET_X86_X86RETURNPROTECTORLOWERING_H
|
|
+#define LLVM_LIB_TARGET_X86_X86RETURNPROTECTORLOWERING_H
|
|
+
|
|
+#include "llvm/CodeGen/ReturnProtectorLowering.h"
|
|
+
|
|
+namespace llvm {
|
|
+
|
|
+class X86ReturnProtectorLowering : 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;
|
|
+};
|
|
+
|
|
+} // namespace llvm
|
|
+
|
|
+#endif
|