1 //===-- WinCFGuard.h - Windows Control Flow Guard Handling ----*- C++ -*--===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file contains support for writing the metadata for Windows Control Flow 10 // Guard, including address-taken functions, and valid longjmp targets. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_LIB_CODEGEN_ASMPRINTER_WINCFGUARD_H 15 #define LLVM_LIB_CODEGEN_ASMPRINTER_WINCFGUARD_H 16 17 #include "llvm/CodeGen/AsmPrinterHandler.h" 18 #include "llvm/Support/Compiler.h" 19 #include <vector> 20 21 namespace llvm { 22 23 class LLVM_LIBRARY_VISIBILITY WinCFGuard : public AsmPrinterHandler { 24 /// Target of directive emission. 25 AsmPrinter *Asm; 26 std::vector<const MCSymbol *> LongjmpTargets; 27 MCSymbol *lookupImpSymbol(const MCSymbol *Sym); 28 29 public: 30 WinCFGuard(AsmPrinter *A); 31 ~WinCFGuard() override; 32 33 /// Emit the Control Flow Guard function ID table. 34 void endModule() override; 35 36 /// Gather pre-function debug information. 37 /// Every beginFunction(MF) call should be followed by an endFunction(MF) 38 /// call. 39 void beginFunction(const MachineFunction *MF) override {} 40 41 /// Gather post-function debug information. 42 /// Please note that some AsmPrinter implementations may not call 43 /// beginFunction at all. 44 void endFunction(const MachineFunction *MF) override; 45 }; 46 47 } // namespace llvm 48 49 #endif 50