115 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			115 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===- FaultMaps.cpp ------------------------------------------------------===//
 | |
| //
 | |
| // 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
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #include "llvm/CodeGen/FaultMaps.h"
 | |
| #include "llvm/ADT/Twine.h"
 | |
| #include "llvm/CodeGen/AsmPrinter.h"
 | |
| #include "llvm/MC/MCContext.h"
 | |
| #include "llvm/MC/MCExpr.h"
 | |
| #include "llvm/MC/MCObjectFileInfo.h"
 | |
| #include "llvm/MC/MCStreamer.h"
 | |
| #include "llvm/Support/Debug.h"
 | |
| #include "llvm/Support/ErrorHandling.h"
 | |
| 
 | |
| using namespace llvm;
 | |
| 
 | |
| #define DEBUG_TYPE "faultmaps"
 | |
| 
 | |
| static const int FaultMapVersion = 1;
 | |
| const char *FaultMaps::WFMP = "Fault Maps: ";
 | |
| 
 | |
| FaultMaps::FaultMaps(AsmPrinter &AP) : AP(AP) {}
 | |
| 
 | |
| void FaultMaps::recordFaultingOp(FaultKind FaultTy,
 | |
|                                  const MCSymbol *FaultingLabel,
 | |
|                                  const MCSymbol *HandlerLabel) {
 | |
|   MCContext &OutContext = AP.OutStreamer->getContext();
 | |
| 
 | |
|   const MCExpr *FaultingOffset = MCBinaryExpr::createSub(
 | |
|       MCSymbolRefExpr::create(FaultingLabel, OutContext),
 | |
|       MCSymbolRefExpr::create(AP.CurrentFnSymForSize, OutContext), OutContext);
 | |
| 
 | |
|   const MCExpr *HandlerOffset = MCBinaryExpr::createSub(
 | |
|       MCSymbolRefExpr::create(HandlerLabel, OutContext),
 | |
|       MCSymbolRefExpr::create(AP.CurrentFnSymForSize, OutContext), OutContext);
 | |
| 
 | |
|   FunctionInfos[AP.CurrentFnSym].emplace_back(FaultTy, FaultingOffset,
 | |
|                                               HandlerOffset);
 | |
| }
 | |
| 
 | |
| void FaultMaps::serializeToFaultMapSection() {
 | |
|   if (FunctionInfos.empty())
 | |
|     return;
 | |
| 
 | |
|   MCContext &OutContext = AP.OutStreamer->getContext();
 | |
|   MCStreamer &OS = *AP.OutStreamer;
 | |
| 
 | |
|   // Create the section.
 | |
|   MCSection *FaultMapSection =
 | |
|       OutContext.getObjectFileInfo()->getFaultMapSection();
 | |
|   OS.switchSection(FaultMapSection);
 | |
| 
 | |
|   // Emit a dummy symbol to force section inclusion.
 | |
|   OS.emitLabel(OutContext.getOrCreateSymbol(Twine("__LLVM_FaultMaps")));
 | |
| 
 | |
|   LLVM_DEBUG(dbgs() << "********** Fault Map Output **********\n");
 | |
| 
 | |
|   // Header
 | |
|   OS.emitIntValue(FaultMapVersion, 1); // Version.
 | |
|   OS.emitIntValue(0, 1);               // Reserved.
 | |
|   OS.emitInt16(0);                     // Reserved.
 | |
| 
 | |
|   LLVM_DEBUG(dbgs() << WFMP << "#functions = " << FunctionInfos.size() << "\n");
 | |
|   OS.emitInt32(FunctionInfos.size());
 | |
| 
 | |
|   LLVM_DEBUG(dbgs() << WFMP << "functions:\n");
 | |
| 
 | |
|   for (const auto &FFI : FunctionInfos)
 | |
|     emitFunctionInfo(FFI.first, FFI.second);
 | |
| }
 | |
| 
 | |
| void FaultMaps::emitFunctionInfo(const MCSymbol *FnLabel,
 | |
|                                  const FunctionFaultInfos &FFI) {
 | |
|   MCStreamer &OS = *AP.OutStreamer;
 | |
| 
 | |
|   LLVM_DEBUG(dbgs() << WFMP << "  function addr: " << *FnLabel << "\n");
 | |
|   OS.emitSymbolValue(FnLabel, 8);
 | |
| 
 | |
|   LLVM_DEBUG(dbgs() << WFMP << "  #faulting PCs: " << FFI.size() << "\n");
 | |
|   OS.emitInt32(FFI.size());
 | |
| 
 | |
|   OS.emitInt32(0); // Reserved
 | |
| 
 | |
|   for (auto &Fault : FFI) {
 | |
|     LLVM_DEBUG(dbgs() << WFMP << "    fault type: "
 | |
|                       << faultTypeToString(Fault.Kind) << "\n");
 | |
|     OS.emitInt32(Fault.Kind);
 | |
| 
 | |
|     LLVM_DEBUG(dbgs() << WFMP << "    faulting PC offset: "
 | |
|                       << *Fault.FaultingOffsetExpr << "\n");
 | |
|     OS.emitValue(Fault.FaultingOffsetExpr, 4);
 | |
| 
 | |
|     LLVM_DEBUG(dbgs() << WFMP << "    fault handler PC offset: "
 | |
|                       << *Fault.HandlerOffsetExpr << "\n");
 | |
|     OS.emitValue(Fault.HandlerOffsetExpr, 4);
 | |
|   }
 | |
| }
 | |
| 
 | |
| const char *FaultMaps::faultTypeToString(FaultMaps::FaultKind FT) {
 | |
|   switch (FT) {
 | |
|   default:
 | |
|     llvm_unreachable("unhandled fault type!");
 | |
|   case FaultMaps::FaultingLoad:
 | |
|     return "FaultingLoad";
 | |
|   case FaultMaps::FaultingLoadStore:
 | |
|     return "FaultingLoadStore";
 | |
|   case FaultMaps::FaultingStore:
 | |
|     return "FaultingStore";
 | |
|   }
 | |
| }
 |