71 lines · cpp
1//===- llvm/CodeGen/MachineModuleInfoImpls.cpp ----------------------------===//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-exception6//7//===----------------------------------------------------------------------===//8//9// This file implements object-file format specific implementations of10// MachineModuleInfoImpl.11//12//===----------------------------------------------------------------------===//13 14#include "llvm/CodeGen/MachineModuleInfoImpls.h"15#include "llvm/ADT/DenseMap.h"16#include "llvm/ADT/STLExtras.h"17#include "llvm/IR/Constants.h"18#include "llvm/IR/Module.h"19#include "llvm/MC/MCSymbol.h"20 21using namespace llvm;22 23//===----------------------------------------------------------------------===//24// MachineModuleInfoMachO25//===----------------------------------------------------------------------===//26 27// Out of line virtual method.28void MachineModuleInfoMachO::anchor() {}29void MachineModuleInfoELF::anchor() {}30void MachineModuleInfoCOFF::anchor() {}31void MachineModuleInfoWasm::anchor() {}32 33using PairTy = std::pair<MCSymbol *, MachineModuleInfoImpl::StubValueTy>;34static int SortSymbolPair(const PairTy *LHS, const PairTy *RHS) {35 return LHS->first->getName().compare(RHS->first->getName());36}37 38MachineModuleInfoImpl::SymbolListTy MachineModuleInfoImpl::getSortedStubs(39 DenseMap<MCSymbol *, MachineModuleInfoImpl::StubValueTy> &Map) {40 MachineModuleInfoImpl::SymbolListTy List(Map.begin(), Map.end());41 42 array_pod_sort(List.begin(), List.end(), SortSymbolPair);43 44 Map.clear();45 return List;46}47 48using ExprStubPairTy = std::pair<MCSymbol *, const MCExpr *>;49static int SortAuthStubPair(const ExprStubPairTy *LHS,50 const ExprStubPairTy *RHS) {51 return LHS->first->getName().compare(RHS->first->getName());52}53 54MachineModuleInfoImpl::ExprStubListTy MachineModuleInfoImpl::getSortedExprStubs(55 DenseMap<MCSymbol *, const MCExpr *> &ExprStubs) {56 MachineModuleInfoImpl::ExprStubListTy List(ExprStubs.begin(),57 ExprStubs.end());58 59 array_pod_sort(List.begin(), List.end(), SortAuthStubPair);60 61 ExprStubs.clear();62 return List;63}64 65MachineModuleInfoELF::MachineModuleInfoELF(const MachineModuleInfo &MMI) {66 const Module *M = MMI.getModule();67 const auto *Flag = mdconst::extract_or_null<ConstantInt>(68 M->getModuleFlag("ptrauth-sign-personality"));69 HasSignedPersonality = Flag && Flag->getZExtValue() == 1;70}71