Create MCTargetOptions.

For now it contains a single flag, SanitizeAddress, which enables
AddressSanitizer instrumentation of inline assembly.

Patch by Yuri Gorshenin.

llvm-svn: 206971
This commit is contained in:
Evgeniy Stepanov 2014-04-23 11:16:03 +00:00
parent eeb820f572
commit 0a951b775e
26 changed files with 210 additions and 67 deletions

View File

@ -16,6 +16,7 @@
#ifndef LLVM_CODEGEN_COMMANDFLAGS_H #ifndef LLVM_CODEGEN_COMMANDFLAGS_H
#define LLVM_CODEGEN_COMMANDFLAGS_H #define LLVM_CODEGEN_COMMANDFLAGS_H
#include "llvm/MC/MCTargetOptionsCommandFlags.h"
#include "llvm/Support/CodeGen.h" #include "llvm/Support/CodeGen.h"
#include "llvm/Support/CommandLine.h" #include "llvm/Support/CommandLine.h"
#include "llvm/Target/TargetMachine.h" #include "llvm/Target/TargetMachine.h"
@ -225,6 +226,9 @@ static inline TargetOptions InitTargetOptionsFromCodeGenFlags() {
Options.TrapFuncName = TrapFuncName; Options.TrapFuncName = TrapFuncName;
Options.PositionIndependentExecutable = EnablePIE; Options.PositionIndependentExecutable = EnablePIE;
Options.UseInitArray = UseInitArray; Options.UseInitArray = UseInitArray;
Options.MCOptions = InitMCTargetOptionsFromFlags();
return Options; return Options;
} }

View File

@ -12,14 +12,15 @@
#include "llvm/MC/MCExpr.h" #include "llvm/MC/MCExpr.h"
#include "llvm/MC/MCParser/MCAsmParserExtension.h" #include "llvm/MC/MCParser/MCAsmParserExtension.h"
#include "llvm/MC/MCTargetOptions.h"
namespace llvm { namespace llvm {
class MCStreamer;
class StringRef;
class SMLoc;
class AsmToken; class AsmToken;
class MCParsedAsmOperand;
class MCInst; class MCInst;
class MCParsedAsmOperand;
class MCStreamer;
class SMLoc;
class StringRef;
template <typename T> class SmallVectorImpl; template <typename T> class SmallVectorImpl;
enum AsmRewriteKind { enum AsmRewriteKind {
@ -97,6 +98,9 @@ protected: // Can only create subclasses.
/// ms-style inline assembly. /// ms-style inline assembly.
MCAsmParserSemaCallback *SemaCallback; MCAsmParserSemaCallback *SemaCallback;
/// Set of options which affects instrumentation of inline assembly.
MCTargetOptions MCOptions;
public: public:
virtual ~MCTargetAsmParser(); virtual ~MCTargetAsmParser();

View File

@ -0,0 +1,40 @@
//===- MCTargetOptions.h - MC Target Options -------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_MC_MCTARGETOPTIONS_H
#define LLVM_MC_MCTARGETOPTIONS_H
namespace llvm {
class MCTargetOptions {
public:
enum AsmInstrumentation {
AsmInstrumentationNone,
AsmInstrumentationAddress
};
/// Enables AddressSanitizer instrumentation at machine level.
bool SanitizeAddress : 1;
MCTargetOptions();
};
inline bool operator==(const MCTargetOptions &LHS, const MCTargetOptions &RHS) {
#define ARE_EQUAL(X) LHS.X == RHS.X
return ARE_EQUAL(SanitizeAddress);
#undef ARE_EQUAL
}
inline bool operator!=(const MCTargetOptions &LHS, const MCTargetOptions &RHS) {
return !(LHS == RHS);
}
} // end namespace llvm
#endif

View File

@ -0,0 +1,42 @@
//===-- MCTargetOptionsCommandFlags.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 machine code-specific flags that are shared between
// different command line tools.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_MC_MCTARGETOPTIONSCOMMANDFLAGS_H
#define LLVM_MC_MCTARGETOPTIONSCOMMANDFLAGS_H
#include "llvm/Support/CommandLine.h"
#include "llvm/MC/MCTargetOptions.h"
using namespace llvm;
cl::opt<MCTargetOptions::AsmInstrumentation> AsmInstrumentation(
"asm-instrumentation",
cl::desc("Instrumentation of inline assembly and "
"assembly source files"),
cl::init(MCTargetOptions::AsmInstrumentationNone),
cl::values(clEnumValN(MCTargetOptions::AsmInstrumentationNone,
"none",
"no instrumentation at all"),
clEnumValN(MCTargetOptions::AsmInstrumentationAddress,
"address",
"instrument instructions with memory arguments"),
clEnumValEnd));
static inline MCTargetOptions InitMCTargetOptionsFromFlags() {
MCTargetOptions Options;
Options.SanitizeAddress =
(AsmInstrumentation == MCTargetOptions::AsmInstrumentationAddress);
return Options;
}
#endif

View File

@ -45,6 +45,7 @@ namespace llvm {
class MCSymbolizer; class MCSymbolizer;
class MCRelocationInfo; class MCRelocationInfo;
class MCTargetAsmParser; class MCTargetAsmParser;
class MCTargetOptions;
class TargetMachine; class TargetMachine;
class TargetOptions; class TargetOptions;
class raw_ostream; class raw_ostream;
@ -104,9 +105,11 @@ namespace llvm {
const MCRegisterInfo &MRI, const MCRegisterInfo &MRI,
StringRef TT, StringRef TT,
StringRef CPU); StringRef CPU);
typedef MCTargetAsmParser *(*MCAsmParserCtorTy)(MCSubtargetInfo &STI, typedef MCTargetAsmParser *(*MCAsmParserCtorTy)(
MCAsmParser &P, MCSubtargetInfo &STI,
const MCInstrInfo &MII); MCAsmParser &P,
const MCInstrInfo &MII,
const MCTargetOptions &Options);
typedef MCDisassembler *(*MCDisassemblerCtorTy)(const Target &T, typedef MCDisassembler *(*MCDisassemblerCtorTy)(const Target &T,
const MCSubtargetInfo &STI, const MCSubtargetInfo &STI,
MCContext &Ctx); MCContext &Ctx);
@ -362,12 +365,14 @@ namespace llvm {
/// ///
/// \param Parser The target independent parser implementation to use for /// \param Parser The target independent parser implementation to use for
/// parsing and lexing. /// parsing and lexing.
MCTargetAsmParser *createMCAsmParser(MCSubtargetInfo &STI, MCTargetAsmParser *createMCAsmParser(
MCAsmParser &Parser, MCSubtargetInfo &STI,
const MCInstrInfo &MII) const { MCAsmParser &Parser,
const MCInstrInfo &MII,
const MCTargetOptions &Options) const {
if (!MCAsmParserCtorFn) if (!MCAsmParserCtorFn)
return nullptr; return nullptr;
return MCAsmParserCtorFn(STI, Parser, MII); return MCAsmParserCtorFn(STI, Parser, MII, Options);
} }
/// createAsmPrinter - Create a target specific assembly printer pass. This /// createAsmPrinter - Create a target specific assembly printer pass. This
@ -1099,8 +1104,9 @@ namespace llvm {
private: private:
static MCTargetAsmParser *Allocator(MCSubtargetInfo &STI, MCAsmParser &P, static MCTargetAsmParser *Allocator(MCSubtargetInfo &STI, MCAsmParser &P,
const MCInstrInfo &MII) { const MCInstrInfo &MII,
return new MCAsmParserImpl(STI, P, MII); const MCTargetOptions &Options) {
return new MCAsmParserImpl(STI, P, MII, Options);
} }
}; };

View File

@ -15,6 +15,7 @@
#ifndef LLVM_TARGET_TARGETOPTIONS_H #ifndef LLVM_TARGET_TARGETOPTIONS_H
#define LLVM_TARGET_TARGETOPTIONS_H #define LLVM_TARGET_TARGETOPTIONS_H
#include "llvm/MC/MCTargetOptions.h"
#include <string> #include <string>
namespace llvm { namespace llvm {
@ -197,6 +198,9 @@ namespace llvm {
/// via the llvm.fma.* intrinsic) will always be honored, regardless of /// via the llvm.fma.* intrinsic) will always be honored, regardless of
/// the value of this option. /// the value of this option.
FPOpFusion::FPOpFusionMode AllowFPOpFusion; FPOpFusion::FPOpFusionMode AllowFPOpFusion;
/// Machine level options.
MCTargetOptions MCOptions;
}; };
// Comparison operators: // Comparison operators:
@ -223,7 +227,8 @@ inline bool operator==(const TargetOptions &LHS,
ARE_EQUAL(TrapUnreachable) && ARE_EQUAL(TrapUnreachable) &&
ARE_EQUAL(TrapFuncName) && ARE_EQUAL(TrapFuncName) &&
ARE_EQUAL(FloatABIType) && ARE_EQUAL(FloatABIType) &&
ARE_EQUAL(AllowFPOpFusion); ARE_EQUAL(AllowFPOpFusion) &&
ARE_EQUAL(MCOptions);
#undef ARE_EQUAL #undef ARE_EQUAL
} }

View File

@ -15,6 +15,7 @@
#include "llvm/ADT/SmallString.h" #include "llvm/ADT/SmallString.h"
#include "llvm/ADT/Twine.h" #include "llvm/ADT/Twine.h"
#include "llvm/CodeGen/MachineBasicBlock.h" #include "llvm/CodeGen/MachineBasicBlock.h"
#include "llvm/CodeGen/MachineFunction.h"
#include "llvm/CodeGen/MachineModuleInfo.h" #include "llvm/CodeGen/MachineModuleInfo.h"
#include "llvm/IR/Constants.h" #include "llvm/IR/Constants.h"
#include "llvm/IR/DataLayout.h" #include "llvm/IR/DataLayout.h"
@ -135,8 +136,11 @@ void AsmPrinter::EmitInlineAsm(StringRef Str, const MDNode *LocMDNode,
// emitInlineAsmEnd(). // emitInlineAsmEnd().
MCSubtargetInfo STIOrig = *STI; MCSubtargetInfo STIOrig = *STI;
MCTargetOptions MCOptions;
if (MF)
MCOptions = MF->getTarget().Options.MCOptions;
std::unique_ptr<MCTargetAsmParser> TAP( std::unique_ptr<MCTargetAsmParser> TAP(
TM.getTarget().createMCAsmParser(*STI, *Parser, *MII)); TM.getTarget().createMCAsmParser(*STI, *Parser, *MII, MCOptions));
if (!TAP) if (!TAP)
report_fatal_error("Inline asm not supported by this streamer because" report_fatal_error("Inline asm not supported by this streamer because"
" we don't have an asm parser for this target\n"); " we don't have an asm parser for this target\n");

View File

@ -739,7 +739,8 @@ bool LTOModule::addAsmGlobalSymbols(std::string &errMsg) {
_target->getTargetTriple(), _target->getTargetCPU(), _target->getTargetTriple(), _target->getTargetCPU(),
_target->getTargetFeatureString())); _target->getTargetFeatureString()));
std::unique_ptr<MCTargetAsmParser> TAP( std::unique_ptr<MCTargetAsmParser> TAP(
T.createMCAsmParser(*STI, *Parser.get(), *MCII)); T.createMCAsmParser(*STI, *Parser.get(), *MCII,
_target->Options.MCOptions));
if (!TAP) { if (!TAP) {
errMsg = "target " + std::string(T.getName()) + errMsg = "target " + std::string(T.getName()) +
" does not define AsmParser."; " does not define AsmParser.";

View File

@ -45,6 +45,7 @@ add_llvm_library(LLVMMC
MCSubtargetInfo.cpp MCSubtargetInfo.cpp
MCSymbol.cpp MCSymbol.cpp
MCSymbolizer.cpp MCSymbolizer.cpp
MCTargetOptions.cpp
MCValue.cpp MCValue.cpp
MCWin64EH.cpp MCWin64EH.cpp
MachObjectWriter.cpp MachObjectWriter.cpp

View File

@ -0,0 +1,16 @@
//===- lib/MC/MCTargetOptions.cpp - MC Target Options --------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "llvm/MC/MCTargetOptions.h"
namespace llvm {
MCTargetOptions::MCTargetOptions() : SanitizeAddress(false) {}
} // end namespace llvm

View File

@ -55,7 +55,8 @@ public:
}; };
AArch64AsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser, AArch64AsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(_STI), Parser(_Parser) { : MCTargetAsmParser(), STI(_STI), Parser(_Parser) {
MCAsmParserExtension::Initialize(_Parser); MCAsmParserExtension::Initialize(_Parser);

View File

@ -344,7 +344,8 @@ public:
}; };
ARMAsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser, ARMAsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(_STI), Parser(_Parser), MII(MII), UC(_Parser) { : MCTargetAsmParser(), STI(_STI), Parser(_Parser), MII(MII), UC(_Parser) {
MCAsmParserExtension::Initialize(_Parser); MCAsmParserExtension::Initialize(_Parser);

View File

@ -104,7 +104,8 @@ public:
#include "ARM64GenAsmMatcher.inc" #include "ARM64GenAsmMatcher.inc"
}; };
ARM64AsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser, ARM64AsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(_STI), Parser(_Parser) { : MCTargetAsmParser(), STI(_STI), Parser(_Parser) {
MCAsmParserExtension::Initialize(_Parser); MCAsmParserExtension::Initialize(_Parser);

View File

@ -229,7 +229,8 @@ class MipsAsmParser : public MCTargetAsmParser {
public: public:
MipsAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser, MipsAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(sti), Parser(parser) { : MCTargetAsmParser(), STI(sti), Parser(parser) {
// Initialize the set of available features. // Initialize the set of available features.
setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits())); setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits()));

View File

@ -264,7 +264,8 @@ class PPCAsmParser : public MCTargetAsmParser {
public: public:
PPCAsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser, PPCAsmParser(MCSubtargetInfo &_STI, MCAsmParser &_Parser,
const MCInstrInfo &_MII) const MCInstrInfo &_MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(_STI), Parser(_Parser), MII(_MII) { : MCTargetAsmParser(), STI(_STI), Parser(_Parser), MII(_MII) {
// Check for 64-bit vs. 32-bit pointer mode. // Check for 64-bit vs. 32-bit pointer mode.
Triple TheTriple(STI.getTargetTriple()); Triple TheTriple(STI.getTargetTriple());

View File

@ -83,7 +83,8 @@ class SparcAsmParser : public MCTargetAsmParser {
bool is64Bit() const { return STI.getTargetTriple().startswith("sparcv9"); } bool is64Bit() const { return STI.getTargetTriple().startswith("sparcv9"); }
public: public:
SparcAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser, SparcAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(sti), Parser(parser) { : MCTargetAsmParser(), STI(sti), Parser(parser) {
// Initialize the set of available features. // Initialize the set of available features.
setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits())); setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits()));

View File

@ -331,7 +331,8 @@ private:
public: public:
SystemZAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser, SystemZAsmParser(MCSubtargetInfo &sti, MCAsmParser &parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(sti), Parser(parser) { : MCTargetAsmParser(), STI(sti), Parser(parser) {
MCAsmParserExtension::Initialize(Parser); MCAsmParserExtension::Initialize(Parser);

View File

@ -21,6 +21,7 @@
#include "llvm/MC/MCAsmInfo.h" #include "llvm/MC/MCAsmInfo.h"
#include "llvm/MC/MCCodeGenInfo.h" #include "llvm/MC/MCCodeGenInfo.h"
#include "llvm/MC/MCContext.h" #include "llvm/MC/MCContext.h"
#include "llvm/MC/MCTargetOptions.h"
#include "llvm/MC/SectionKind.h" #include "llvm/MC/SectionKind.h"
#include "llvm/Support/CommandLine.h" #include "llvm/Support/CommandLine.h"
#include "llvm/Target/TargetLowering.h" #include "llvm/Target/TargetLowering.h"
@ -89,6 +90,8 @@ void TargetMachine::resetTargetOptions(const MachineFunction *MF) const {
RESET_OPTION(NoNaNsFPMath, "no-nans-fp-math"); RESET_OPTION(NoNaNsFPMath, "no-nans-fp-math");
RESET_OPTION(UseSoftFloat, "use-soft-float"); RESET_OPTION(UseSoftFloat, "use-soft-float");
RESET_OPTION(DisableTailCalls, "disable-tail-calls"); RESET_OPTION(DisableTailCalls, "disable-tail-calls");
TO.MCOptions.SanitizeAddress = F->hasFnAttribute(Attribute::SanitizeAddress);
} }
/// getRelocationModel - Returns the code generation relocation model. The /// getRelocationModel - Returns the code generation relocation model. The

View File

@ -11,22 +11,18 @@
#include "X86AsmInstrumentation.h" #include "X86AsmInstrumentation.h"
#include "X86Operand.h" #include "X86Operand.h"
#include "llvm/ADT/StringExtras.h" #include "llvm/ADT/StringExtras.h"
#include "llvm/IR/Function.h"
#include "llvm/MC/MCContext.h" #include "llvm/MC/MCContext.h"
#include "llvm/MC/MCInst.h" #include "llvm/MC/MCInst.h"
#include "llvm/MC/MCInstBuilder.h" #include "llvm/MC/MCInstBuilder.h"
#include "llvm/MC/MCParser/MCParsedAsmOperand.h"
#include "llvm/MC/MCStreamer.h" #include "llvm/MC/MCStreamer.h"
#include "llvm/MC/MCSubtargetInfo.h" #include "llvm/MC/MCSubtargetInfo.h"
#include "llvm/Support/CommandLine.h" #include "llvm/MC/MCTargetOptions.h"
#include "llvm/Support/Compiler.h"
#include "llvm/MC/MCParser/MCParsedAsmOperand.h"
namespace llvm { namespace llvm {
namespace { namespace {
static cl::opt<bool> ClAsanInstrumentInlineAssembly(
"asan-instrument-inline-assembly", cl::desc("instrument inline assembly"),
cl::Hidden, cl::init(false));
bool IsStackReg(unsigned Reg) { bool IsStackReg(unsigned Reg) {
return Reg == X86::RSP || Reg == X86::ESP || Reg == X86::SP; return Reg == X86::RSP || Reg == X86::ESP || Reg == X86::SP;
} }
@ -38,7 +34,7 @@ std::string FuncName(unsigned AccessSize, bool IsWrite) {
class X86AddressSanitizer : public X86AsmInstrumentation { class X86AddressSanitizer : public X86AsmInstrumentation {
public: public:
X86AddressSanitizer(MCSubtargetInfo &sti) : STI(sti) {} X86AddressSanitizer(const MCSubtargetInfo &STI) : STI(STI) {}
virtual ~X86AddressSanitizer() {} virtual ~X86AddressSanitizer() {}
// X86AsmInstrumentation implementation: // X86AsmInstrumentation implementation:
@ -63,7 +59,7 @@ public:
} }
protected: protected:
MCSubtargetInfo &STI; const MCSubtargetInfo &STI;
}; };
void X86AddressSanitizer::InstrumentMemOperand( void X86AddressSanitizer::InstrumentMemOperand(
@ -144,7 +140,8 @@ void X86AddressSanitizer::InstrumentMOV(
class X86AddressSanitizer32 : public X86AddressSanitizer { class X86AddressSanitizer32 : public X86AddressSanitizer {
public: public:
X86AddressSanitizer32(MCSubtargetInfo &sti) : X86AddressSanitizer(sti) {} X86AddressSanitizer32(const MCSubtargetInfo &STI)
: X86AddressSanitizer(STI) {}
virtual ~X86AddressSanitizer32() {} virtual ~X86AddressSanitizer32() {}
virtual void InstrumentMemOperandImpl(X86Operand *Op, unsigned AccessSize, virtual void InstrumentMemOperandImpl(X86Operand *Op, unsigned AccessSize,
@ -179,7 +176,8 @@ void X86AddressSanitizer32::InstrumentMemOperandImpl(
class X86AddressSanitizer64 : public X86AddressSanitizer { class X86AddressSanitizer64 : public X86AddressSanitizer {
public: public:
X86AddressSanitizer64(MCSubtargetInfo &sti) : X86AddressSanitizer(sti) {} X86AddressSanitizer64(const MCSubtargetInfo &STI)
: X86AddressSanitizer(STI) {}
virtual ~X86AddressSanitizer64() {} virtual ~X86AddressSanitizer64() {}
virtual void InstrumentMemOperandImpl(X86Operand *Op, unsigned AccessSize, virtual void InstrumentMemOperandImpl(X86Operand *Op, unsigned AccessSize,
@ -223,8 +221,10 @@ void X86AsmInstrumentation::InstrumentInstruction(
const MCInst &Inst, SmallVectorImpl<MCParsedAsmOperand *> &Operands, const MCInst &Inst, SmallVectorImpl<MCParsedAsmOperand *> &Operands,
MCContext &Ctx, MCStreamer &Out) {} MCContext &Ctx, MCStreamer &Out) {}
X86AsmInstrumentation *CreateX86AsmInstrumentation(MCSubtargetInfo &STI) { X86AsmInstrumentation *
if (ClAsanInstrumentInlineAssembly) { CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions, const MCContext &Ctx,
const MCSubtargetInfo &STI) {
if (MCOptions.SanitizeAddress) {
if ((STI.getFeatureBits() & X86::Mode32Bit) != 0) if ((STI.getFeatureBits() & X86::Mode32Bit) != 0)
return new X86AddressSanitizer32(STI); return new X86AddressSanitizer32(STI);
if ((STI.getFeatureBits() & X86::Mode64Bit) != 0) if ((STI.getFeatureBits() & X86::Mode64Bit) != 0)

View File

@ -19,10 +19,13 @@ class MCInst;
class MCParsedAsmOperand; class MCParsedAsmOperand;
class MCStreamer; class MCStreamer;
class MCSubtargetInfo; class MCSubtargetInfo;
class MCTargetOptions;
class X86AsmInstrumentation; class X86AsmInstrumentation;
X86AsmInstrumentation *CreateX86AsmInstrumentation(MCSubtargetInfo &STI); X86AsmInstrumentation *
CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
const MCContext &Ctx, const MCSubtargetInfo &STI);
class X86AsmInstrumentation { class X86AsmInstrumentation {
public: public:
@ -36,11 +39,12 @@ public:
protected: protected:
friend X86AsmInstrumentation * friend X86AsmInstrumentation *
CreateX86AsmInstrumentation(MCSubtargetInfo &STI); CreateX86AsmInstrumentation(const MCTargetOptions &MCOptions,
const MCContext &Ctx, const MCSubtargetInfo &STI);
X86AsmInstrumentation(); X86AsmInstrumentation();
}; };
} // End llvm namespace } // End llvm namespace
#endif // X86_ASM_INSTRUMENTATION_H #endif // X86_ASM_INSTRUMENTATION_H

View File

@ -710,13 +710,16 @@ private:
public: public:
X86AsmParser(MCSubtargetInfo &sti, MCAsmParser &parser, X86AsmParser(MCSubtargetInfo &sti, MCAsmParser &parser,
const MCInstrInfo &MII) const MCInstrInfo &MII,
const MCTargetOptions &Options)
: MCTargetAsmParser(), STI(sti), Parser(parser), InstInfo(0) { : MCTargetAsmParser(), STI(sti), Parser(parser), InstInfo(0) {
// Initialize the set of available features. // Initialize the set of available features.
setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits())); setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits()));
Instrumentation.reset(CreateX86AsmInstrumentation(STI)); Instrumentation.reset(
CreateX86AsmInstrumentation(Options, Parser.getContext(), STI));
} }
bool ParseRegister(unsigned &RegNo, SMLoc &StartLoc, SMLoc &EndLoc) override; bool ParseRegister(unsigned &RegNo, SMLoc &StartLoc, SMLoc &EndLoc) override;
bool bool

View File

@ -0,0 +1,20 @@
; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mcpu=corei7 -mattr=+sse2 -asm-instrumentation=address | FileCheck %s
target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"
; CHECK-LABEL: mov_no_attr
; CHECK-NOT: callq __sanitizer_sanitize_load8@PLT
; CHECK-NOT: callq __sanitizer_sanitize_store8@PLT
define void @mov_no_attr(i64* %dst, i64* %src) {
tail call void asm sideeffect "movq ($1), %rax \0A\09movq %rax, ($0) \0A\09", "r,r,~{memory},~{rax},~{dirflag},~{fpsr},~{flags}"(i64* %dst, i64* %src)
ret void
}
; CHECK-LABEL: mov_sanitize
; CHECK: callq __sanitizer_sanitize_load8@PLT
; CHECK: callq __sanitizer_sanitize_store8@PLT
define void @mov_sanitize(i64* %dst, i64* %src) sanitize_address {
tail call void asm sideeffect "movq ($1), %rax \0A\09movq %rax, ($0) \0A\09", "r,r,~{memory},~{rax},~{dirflag},~{fpsr},~{flags}"(i64* %dst, i64* %src)
ret void
}

View File

@ -1,4 +1,4 @@
; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mcpu=corei7 -mattr=+sse2 -asan-instrument-inline-assembly | FileCheck %s ; RUN: llc < %s -mtriple=x86_64-unknown-linux-gnu -mcpu=corei7 -mattr=+sse2 -asm-instrumentation=address | FileCheck %s
target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128" target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu" target triple = "x86_64-unknown-linux-gnu"
@ -113,7 +113,7 @@ entry:
ret void ret void
} }
attributes #0 = { nounwind uwtable "less-precise-fpmad"="false" "no-frame-pointer-elim"="false" "no-frame-pointer-elim-non-leaf"="false" "no-infs-fp-math"="false" "no-nans-fp-math"="false" "unsafe-fp-math"="false" "use-soft-float"="false" } attributes #0 = { nounwind uwtable sanitize_address "less-precise-fpmad"="false" "no-frame-pointer-elim"="false" "no-frame-pointer-elim-non-leaf"="false" "no-infs-fp-math"="false" "no-nans-fp-math"="false" "unsafe-fp-math"="false" "use-soft-float"="false" }
attributes #1 = { nounwind } attributes #1 = { nounwind }
!0 = metadata !{i32 98, i32 122, i32 160} !0 = metadata !{i32 98, i32 122, i32 160}

View File

@ -1,4 +1,4 @@
# RUN: llvm-mc %s -triple=x86_64-unknown-linux-gnu -mcpu=corei7 -mattr=+sse2 -asan-instrument-inline-assembly | FileCheck %s # RUN: llvm-mc %s -triple=x86_64-unknown-linux-gnu -mcpu=corei7 -mattr=+sse2 -asm-instrumentation=address | FileCheck %s
.text .text
.globl mov1b .globl mov1b

View File

@ -20,25 +20,5 @@ mov1b: # @mov1b
.size mov1b, .Ltmp0-mov1b .size mov1b, .Ltmp0-mov1b
.cfi_endproc .cfi_endproc
.globl mov16b
.align 16, 0x90
.type mov16b,@function
# CHECK-LABEL: mov16b
# CHECK-NOT: callq __sanitizer_sanitize_load16@PLT
# CHECK-NOT: callq __sanitizer_sanitize_store16@PLT
mov16b: # @mov16b
.cfi_startproc
# BB#0:
#APP
movaps (%rsi), %xmm0
movaps %xmm0, (%rdi)
#NO_APP
retq
.Ltmp1:
.size mov16b, .Ltmp1-mov16b
.cfi_endproc
.ident "clang version 3.5 " .ident "clang version 3.5 "
.section ".note.GNU-stack","",@progbits .section ".note.GNU-stack","",@progbits

View File

@ -25,6 +25,7 @@
#include "llvm/MC/MCStreamer.h" #include "llvm/MC/MCStreamer.h"
#include "llvm/MC/MCSubtargetInfo.h" #include "llvm/MC/MCSubtargetInfo.h"
#include "llvm/MC/MCTargetAsmParser.h" #include "llvm/MC/MCTargetAsmParser.h"
#include "llvm/MC/MCTargetOptionsCommandFlags.h"
#include "llvm/Support/CommandLine.h" #include "llvm/Support/CommandLine.h"
#include "llvm/Support/Compression.h" #include "llvm/Support/Compression.h"
#include "llvm/Support/FileUtilities.h" #include "llvm/Support/FileUtilities.h"
@ -320,9 +321,11 @@ static int AsLexInput(SourceMgr &SrcMgr, MCAsmInfo &MAI, tool_output_file *Out)
static int AssembleInput(const char *ProgName, const Target *TheTarget, static int AssembleInput(const char *ProgName, const Target *TheTarget,
SourceMgr &SrcMgr, MCContext &Ctx, MCStreamer &Str, SourceMgr &SrcMgr, MCContext &Ctx, MCStreamer &Str,
MCAsmInfo &MAI, MCSubtargetInfo &STI, MCInstrInfo &MCII) { MCAsmInfo &MAI, MCSubtargetInfo &STI, MCInstrInfo &MCII) {
std::unique_ptr<MCAsmParser> Parser(createMCAsmParser(SrcMgr, Ctx, Str, MAI)); std::unique_ptr<MCAsmParser> Parser(
createMCAsmParser(SrcMgr, Ctx, Str, MAI));
std::unique_ptr<MCTargetAsmParser> TAP( std::unique_ptr<MCTargetAsmParser> TAP(
TheTarget->createMCAsmParser(STI, *Parser, MCII)); TheTarget->createMCAsmParser(STI, *Parser, MCII,
InitMCTargetOptionsFromFlags()));
if (!TAP) { if (!TAP) {
errs() << ProgName errs() << ProgName
<< ": error: this target does not support assembly parsing.\n"; << ": error: this target does not support assembly parsing.\n";