diff --git a/lib/Transforms/Instrumentation/GCOVProfiling.cpp b/lib/Transforms/Instrumentation/GCOVProfiling.cpp index 236d27cfaff97..7a40fe6bbd64b 100644 --- a/lib/Transforms/Instrumentation/GCOVProfiling.cpp +++ b/lib/Transforms/Instrumentation/GCOVProfiling.cpp @@ -58,28 +58,40 @@ namespace { private: // Create the GCNO files for the Module based on DebugInfo. - void EmitGCNO(DebugInfoFinder &DIF); + void emitGCNO(DebugInfoFinder &DIF); // Modify the program to track transitions along edges and call into the // profiling runtime to emit .gcda files when run. - bool EmitProfileArcs(DebugInfoFinder &DIF); + bool emitProfileArcs(DebugInfoFinder &DIF); // Get pointers to the functions in the runtime library. Constant *getStartFileFunc(); + Constant *getIncrementIndirectCounterFunc(); Constant *getEmitFunctionFunc(); Constant *getEmitArcsFunc(); Constant *getEndFileFunc(); + // Create or retrieve an i32 state value that is used to represent the + // pred block number for certain non-trivial edges. + GlobalVariable *getEdgeStateValue(); + + // Produce a table of pointers to counters, by predecessor and successor + // block number. + GlobalVariable *buildEdgeLookupTable(Function *F, + GlobalVariable *Counter, + const UniqueVector &Preds, + const UniqueVector &Succs); + // Add the function to write out all our counters to the global destructor // list. - void InsertCounterWriteout(DebugInfoFinder &, + void insertCounterWriteout(DebugInfoFinder &, SmallVector, 8> &); bool EmitNotes; bool EmitData; - Module *Mod; + Module *M; LLVMContext *Ctx; }; } @@ -92,59 +104,59 @@ ModulePass *llvm::createGCOVProfilerPass(bool EmitNotes, bool EmitData) { return new GCOVProfiler(EmitNotes, EmitData); } -static DISubprogram FindSubprogram(DIScope scope) { - while (!scope.isSubprogram()) { - assert(scope.isLexicalBlock() && +static DISubprogram findSubprogram(DIScope Scope) { + while (!Scope.isSubprogram()) { + assert(Scope.isLexicalBlock() && "Debug location not lexical block or subprogram"); - scope = DILexicalBlock(scope).getContext(); + Scope = DILexicalBlock(Scope).getContext(); } - return DISubprogram(scope); + return DISubprogram(Scope); } namespace { class GCOVRecord { protected: - static const char *lines_tag; - static const char *function_tag; - static const char *block_tag; - static const char *edge_tag; + static const char *LinesTag; + static const char *FunctionTag; + static const char *BlockTag; + static const char *EdgeTag; GCOVRecord() {} - void WriteBytes(const char *b, int size) { - os->write(b, size); + void writeBytes(const char *Bytes, int Size) { + os->write(Bytes, Size); } - void Write(uint32_t i) { - WriteBytes(reinterpret_cast(&i), 4); + void write(uint32_t i) { + writeBytes(reinterpret_cast(&i), 4); } // Returns the length measured in 4-byte blocks that will be used to // represent this string in a GCOV file - unsigned LengthOfGCOVString(StringRef s) { + unsigned lengthOfGCOVString(StringRef s) { // A GCOV string is a length, followed by a NUL, then between 0 and 3 NULs // padding out to the next 4-byte word. The length is measured in 4-byte // words including padding, not bytes of actual string. return (s.size() + 5) / 4; } - void WriteGCOVString(StringRef s) { - uint32_t len = LengthOfGCOVString(s); - Write(len); - WriteBytes(s.data(), s.size()); + void writeGCOVString(StringRef s) { + uint32_t Len = lengthOfGCOVString(s); + write(Len); + writeBytes(s.data(), s.size()); // Write 1 to 4 bytes of NUL padding. assert((unsigned)(5 - ((s.size() + 1) % 4)) > 0); assert((unsigned)(5 - ((s.size() + 1) % 4)) <= 4); - WriteBytes("\0\0\0\0", 5 - ((s.size() + 1) % 4)); + writeBytes("\0\0\0\0", 5 - ((s.size() + 1) % 4)); } raw_ostream *os; }; - const char *GCOVRecord::lines_tag = "\0\0\x45\x01"; - const char *GCOVRecord::function_tag = "\0\0\0\1"; - const char *GCOVRecord::block_tag = "\0\0\x41\x01"; - const char *GCOVRecord::edge_tag = "\0\0\x43\x01"; + const char *GCOVRecord::LinesTag = "\0\0\x45\x01"; + const char *GCOVRecord::FunctionTag = "\0\0\0\1"; + const char *GCOVRecord::BlockTag = "\0\0\x41\x01"; + const char *GCOVRecord::EdgeTag = "\0\0\x43\x01"; class GCOVFunction; class GCOVBlock; @@ -154,24 +166,24 @@ namespace { // to the block. class GCOVLines : public GCOVRecord { public: - void AddLine(uint32_t line) { - lines.push_back(line); + void addLine(uint32_t Line) { + Lines.push_back(Line); } - uint32_t Length() { - return LengthOfGCOVString(filename) + 2 + lines.size(); + uint32_t length() { + return lengthOfGCOVString(Filename) + 2 + Lines.size(); } private: friend class GCOVBlock; - GCOVLines(std::string filename, raw_ostream *os) - : filename(filename) { + GCOVLines(std::string Filename, raw_ostream *os) + : Filename(Filename) { this->os = os; } - std::string filename; - SmallVector lines; + std::string Filename; + SmallVector Lines; }; // Represent a basic block in GCOV. Each block has a unique number in the @@ -179,56 +191,55 @@ namespace { // other blocks. class GCOVBlock : public GCOVRecord { public: - GCOVLines &GetFile(std::string filename) { - GCOVLines *&lines = lines_by_file[filename]; - if (!lines) { - lines = new GCOVLines(filename, os); + GCOVLines &getFile(std::string Filename) { + GCOVLines *&Lines = LinesByFile[Filename]; + if (!Lines) { + Lines = new GCOVLines(Filename, os); } - return *lines; + return *Lines; } - void AddEdge(GCOVBlock &successor) { - out_edges.push_back(&successor); + void addEdge(GCOVBlock &Successor) { + OutEdges.push_back(&Successor); } - void WriteOut() { - uint32_t len = 3; - for (StringMap::iterator I = lines_by_file.begin(), - E = lines_by_file.end(); I != E; ++I) { - len += I->second->Length(); + void writeOut() { + uint32_t Len = 3; + for (StringMap::iterator I = LinesByFile.begin(), + E = LinesByFile.end(); I != E; ++I) { + Len += I->second->length(); } - WriteBytes(lines_tag, 4); - Write(len); - Write(number); - for (StringMap::iterator I = lines_by_file.begin(), - E = lines_by_file.end(); I != E; ++I) { - Write(0); - WriteGCOVString(I->second->filename); - for (int i = 0, e = I->second->lines.size(); i != e; ++i) { - Write(I->second->lines[i]); + writeBytes(LinesTag, 4); + write(Len); + write(Number); + for (StringMap::iterator I = LinesByFile.begin(), + E = LinesByFile.end(); I != E; ++I) { + write(0); + writeGCOVString(I->second->Filename); + for (int i = 0, e = I->second->Lines.size(); i != e; ++i) { + write(I->second->Lines[i]); } } - Write(0); - Write(0); + write(0); + write(0); } ~GCOVBlock() { - DeleteContainerSeconds(lines_by_file); + DeleteContainerSeconds(LinesByFile); } private: friend class GCOVFunction; - GCOVBlock(uint32_t number, raw_ostream *os) - : number(number) { + GCOVBlock(uint32_t Number, raw_ostream *os) + : Number(Number) { this->os = os; } - uint32_t number; - BasicBlock *block; - StringMap lines_by_file; - SmallVector out_edges; + uint32_t Number; + StringMap LinesByFile; + SmallVector OutEdges; }; // A function has a unique identifier, a checksum (we leave as zero) and a @@ -242,90 +253,90 @@ namespace { Function *F = SP.getFunction(); uint32_t i = 0; for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) { - blocks[BB] = new GCOVBlock(i++, os); + Blocks[BB] = new GCOVBlock(i++, os); } - return_block = new GCOVBlock(i++, os); - - WriteBytes(function_tag, 4); - uint32_t block_len = 1 + 1 + 1 + LengthOfGCOVString(SP.getName()) + - 1 + LengthOfGCOVString(SP.getFilename()) + 1; - Write(block_len); - uint32_t ident = reinterpret_cast((MDNode*)SP); - Write(ident); - Write(0); // checksum - WriteGCOVString(SP.getName()); - WriteGCOVString(SP.getFilename()); - Write(SP.getLineNumber()); + ReturnBlock = new GCOVBlock(i++, os); + + writeBytes(FunctionTag, 4); + uint32_t BlockLen = 1 + 1 + 1 + lengthOfGCOVString(SP.getName()) + + 1 + lengthOfGCOVString(SP.getFilename()) + 1; + write(BlockLen); + uint32_t Ident = reinterpret_cast((MDNode*)SP); + write(Ident); + write(0); // checksum + writeGCOVString(SP.getName()); + writeGCOVString(SP.getFilename()); + write(SP.getLineNumber()); } ~GCOVFunction() { - DeleteContainerSeconds(blocks); - delete return_block; + DeleteContainerSeconds(Blocks); + delete ReturnBlock; } - GCOVBlock &GetBlock(BasicBlock *BB) { - return *blocks[BB]; + GCOVBlock &getBlock(BasicBlock *BB) { + return *Blocks[BB]; } - GCOVBlock &GetReturnBlock() { - return *return_block; + GCOVBlock &getReturnBlock() { + return *ReturnBlock; } - void WriteOut() { + void writeOut() { // Emit count of blocks. - WriteBytes(block_tag, 4); - Write(blocks.size() + 1); - for (int i = 0, e = blocks.size() + 1; i != e; ++i) { - Write(0); // No flags on our blocks. + writeBytes(BlockTag, 4); + write(Blocks.size() + 1); + for (int i = 0, e = Blocks.size() + 1; i != e; ++i) { + write(0); // No flags on our blocks. } // Emit edges between blocks. - for (DenseMap::iterator I = blocks.begin(), - E = blocks.end(); I != E; ++I) { - GCOVBlock &block = *I->second; - if (block.out_edges.empty()) continue; - - WriteBytes(edge_tag, 4); - Write(block.out_edges.size() * 2 + 1); - Write(block.number); - for (int i = 0, e = block.out_edges.size(); i != e; ++i) { - Write(block.out_edges[i]->number); - Write(0); // no flags + for (DenseMap::iterator I = Blocks.begin(), + E = Blocks.end(); I != E; ++I) { + GCOVBlock &Block = *I->second; + if (Block.OutEdges.empty()) continue; + + writeBytes(EdgeTag, 4); + write(Block.OutEdges.size() * 2 + 1); + write(Block.Number); + for (int i = 0, e = Block.OutEdges.size(); i != e; ++i) { + write(Block.OutEdges[i]->Number); + write(0); // no flags } } // Emit lines for each block. - for (DenseMap::iterator I = blocks.begin(), - E = blocks.end(); I != E; ++I) { - I->second->WriteOut(); + for (DenseMap::iterator I = Blocks.begin(), + E = Blocks.end(); I != E; ++I) { + I->second->writeOut(); } } private: - DenseMap blocks; - GCOVBlock *return_block; + DenseMap Blocks; + GCOVBlock *ReturnBlock; }; } // Replace the stem of a file, or add one if missing. -static std::string ReplaceStem(std::string orig_filename, std::string new_stem){ - return (sys::path::stem(orig_filename) + "." + new_stem).str(); +static std::string replaceStem(std::string OrigFilename, std::string NewStem) { + return (sys::path::stem(OrigFilename) + "." + NewStem).str(); } bool GCOVProfiler::runOnModule(Module &M) { - Mod = &M; + this->M = &M; Ctx = &M.getContext(); DebugInfoFinder DIF; - DIF.processModule(*Mod); + DIF.processModule(M); - if (EmitNotes) EmitGCNO(DIF); - if (EmitData) return EmitProfileArcs(DIF); + if (EmitNotes) emitGCNO(DIF); + if (EmitData) return emitProfileArcs(DIF); return false; } -void GCOVProfiler::EmitGCNO(DebugInfoFinder &DIF) { - DenseMap gcno_files; +void GCOVProfiler::emitGCNO(DebugInfoFinder &DIF) { + DenseMap GcnoFiles; for (DebugInfoFinder::iterator I = DIF.compile_unit_begin(), E = DIF.compile_unit_end(); I != E; ++I) { // Each compile unit gets its own .gcno file. This means that whether we run @@ -333,153 +344,224 @@ void GCOVProfiler::EmitGCNO(DebugInfoFinder &DIF) { // LTO, we'll generate the same .gcno files. DICompileUnit CU(*I); - raw_fd_ostream *&Out = gcno_files[CU]; + raw_fd_ostream *&out = GcnoFiles[CU]; std::string ErrorInfo; - Out = new raw_fd_ostream(ReplaceStem(CU.getFilename(), "gcno").c_str(), + out = new raw_fd_ostream(replaceStem(CU.getFilename(), "gcno").c_str(), ErrorInfo, raw_fd_ostream::F_Binary); - Out->write("oncg*404MVLL", 12); + out->write("oncg*404MVLL", 12); } for (DebugInfoFinder::iterator SPI = DIF.subprogram_begin(), SPE = DIF.subprogram_end(); SPI != SPE; ++SPI) { DISubprogram SP(*SPI); - raw_fd_ostream *&os = gcno_files[SP.getCompileUnit()]; + raw_fd_ostream *&os = GcnoFiles[SP.getCompileUnit()]; - GCOVFunction function(SP, os); + GCOVFunction Func(SP, os); Function *F = SP.getFunction(); for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) { - GCOVBlock &block = function.GetBlock(BB); + GCOVBlock &Block = Func.getBlock(BB); TerminatorInst *TI = BB->getTerminator(); if (int successors = TI->getNumSuccessors()) { for (int i = 0; i != successors; ++i) { - block.AddEdge(function.GetBlock(TI->getSuccessor(i))); + Block.addEdge(Func.getBlock(TI->getSuccessor(i))); } } else if (isa(TI)) { - block.AddEdge(function.GetReturnBlock()); + Block.addEdge(Func.getReturnBlock()); } - uint32_t line = 0; + uint32_t Line = 0; for (BasicBlock::iterator I = BB->begin(), IE = BB->end(); I != IE; ++I) { - const DebugLoc &loc = I->getDebugLoc(); - if (loc.isUnknown()) continue; - if (line == loc.getLine()) continue; - line = loc.getLine(); - if (SP != FindSubprogram(DIScope(loc.getScope(*Ctx)))) continue; - - GCOVLines &lines = block.GetFile(SP.getFilename()); - lines.AddLine(loc.getLine()); + const DebugLoc &Loc = I->getDebugLoc(); + if (Loc.isUnknown()) continue; + if (Line == Loc.getLine()) continue; + Line = Loc.getLine(); + if (SP != findSubprogram(DIScope(Loc.getScope(*Ctx)))) continue; + + GCOVLines &Lines = Block.getFile(SP.getFilename()); + Lines.addLine(Loc.getLine()); } } - function.WriteOut(); + Func.writeOut(); } for (DenseMap::iterator - I = gcno_files.begin(), E = gcno_files.end(); I != E; ++I) { - raw_fd_ostream *&Out = I->second; - Out->write("\0\0\0\0\0\0\0\0", 8); // EOF - Out->close(); - delete Out; + I = GcnoFiles.begin(), E = GcnoFiles.end(); I != E; ++I) { + raw_fd_ostream *&out = I->second; + out->write("\0\0\0\0\0\0\0\0", 8); // EOF + out->close(); + delete out; } } -bool GCOVProfiler::EmitProfileArcs(DebugInfoFinder &DIF) { +bool GCOVProfiler::emitProfileArcs(DebugInfoFinder &DIF) { if (DIF.subprogram_begin() == DIF.subprogram_end()) return false; - SmallVector, 8> counters_by_ident; + SmallVector, 8> CountersByIdent; for (DebugInfoFinder::iterator SPI = DIF.subprogram_begin(), SPE = DIF.subprogram_end(); SPI != SPE; ++SPI) { DISubprogram SP(*SPI); Function *F = SP.getFunction(); - unsigned edges = 0; + unsigned Edges = 0; for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) { TerminatorInst *TI = BB->getTerminator(); if (isa(TI)) - ++edges; + ++Edges; else - edges += TI->getNumSuccessors(); + Edges += TI->getNumSuccessors(); } - const ArrayType *counter_type = - ArrayType::get(Type::getInt64Ty(*Ctx), edges); - GlobalVariable *counter = - new GlobalVariable(*Mod, counter_type, false, + const ArrayType *CounterTy = + ArrayType::get(Type::getInt64Ty(*Ctx), Edges); + GlobalVariable *Counters = + new GlobalVariable(*M, CounterTy, false, GlobalValue::InternalLinkage, - Constant::getNullValue(counter_type), + Constant::getNullValue(CounterTy), "__llvm_gcov_ctr", 0, false, 0); - counters_by_ident.push_back( - std::make_pair(counter, reinterpret_cast((MDNode*)SP))); + CountersByIdent.push_back( + std::make_pair(Counters, reinterpret_cast((MDNode*)SP))); - UniqueVector complex_edge_preds; - UniqueVector complex_edge_succs; + UniqueVector ComplexEdgePreds; + UniqueVector ComplexEdgeSuccs; - unsigned edge_num = 0; + unsigned Edge = 0; for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) { TerminatorInst *TI = BB->getTerminator(); - int successors = isa(TI) ? 1 : TI->getNumSuccessors(); - if (successors) { - IRBuilder<> builder(TI); - - if (successors == 1) { - Value *ctr = builder.CreateConstInBoundsGEP2_64(counter, 0, edge_num); - Value *count = builder.CreateLoad(ctr); - count = builder.CreateAdd(count, + int Successors = isa(TI) ? 1 : TI->getNumSuccessors(); + if (Successors) { + IRBuilder<> Builder(TI); + + if (Successors == 1) { + Value *Counter = Builder.CreateConstInBoundsGEP2_64(Counters, 0, + Edge); + Value *Count = Builder.CreateLoad(Counter); + Count = Builder.CreateAdd(Count, ConstantInt::get(Type::getInt64Ty(*Ctx),1)); - builder.CreateStore(count, ctr); + Builder.CreateStore(Count, Counter); } else if (BranchInst *BI = dyn_cast(TI)) { - Value *sel = builder.CreateSelect( + Value *Sel = Builder.CreateSelect( BI->getCondition(), - ConstantInt::get(Type::getInt64Ty(*Ctx), edge_num), - ConstantInt::get(Type::getInt64Ty(*Ctx), edge_num + 1)); - SmallVector idx; - idx.push_back(Constant::getNullValue(Type::getInt64Ty(*Ctx))); - idx.push_back(sel); - Value *ctr = builder.CreateInBoundsGEP(counter, - idx.begin(), idx.end()); - Value *count = builder.CreateLoad(ctr); - count = builder.CreateAdd(count, + ConstantInt::get(Type::getInt64Ty(*Ctx), Edge), + ConstantInt::get(Type::getInt64Ty(*Ctx), Edge + 1)); + SmallVector Idx; + Idx.push_back(Constant::getNullValue(Type::getInt64Ty(*Ctx))); + Idx.push_back(Sel); + Value *Counter = Builder.CreateInBoundsGEP(Counters, + Idx.begin(), Idx.end()); + Value *Count = Builder.CreateLoad(Counter); + Count = Builder.CreateAdd(Count, ConstantInt::get(Type::getInt64Ty(*Ctx),1)); - builder.CreateStore(count, ctr); + Builder.CreateStore(Count, Counter); } else { - complex_edge_preds.insert(BB); - for (int i = 0; i != successors; ++i) { - complex_edge_succs.insert(TI->getSuccessor(i)); - } + ComplexEdgePreds.insert(BB); + for (int i = 0; i != Successors; ++i) + ComplexEdgeSuccs.insert(TI->getSuccessor(i)); } - edge_num += successors; + Edge += Successors; } } - // TODO: support switch, invoke, indirectbr - if (!complex_edge_preds.empty()) { - // emit a [preds x [succs x i64*]]. - for (int i = 0, e = complex_edge_preds.size(); i != e; ++i) { - // call runtime to state save + if (!ComplexEdgePreds.empty()) { + GlobalVariable *EdgeTable = + buildEdgeLookupTable(F, Counters, + ComplexEdgePreds, ComplexEdgeSuccs); + GlobalVariable *EdgeState = getEdgeStateValue(); + + const Type *Int32Ty = Type::getInt32Ty(*Ctx); + for (int i = 0, e = ComplexEdgePreds.size(); i != e; ++i) { + IRBuilder<> Builder(ComplexEdgePreds[i+1]->getTerminator()); + Builder.CreateStore(ConstantInt::get(Int32Ty, i+1), EdgeState); } - for (int i = 0, e = complex_edge_succs.size(); i != e; ++i) { + for (int i = 0, e = ComplexEdgeSuccs.size(); i != e; ++i) { // call runtime to perform increment + IRBuilder<> Builder(ComplexEdgeSuccs[i+1]->getFirstNonPHI()); + Value *CounterPtrArray = + Builder.CreateConstInBoundsGEP2_64(EdgeTable, 0, + i * ComplexEdgePreds.size()); + Builder.CreateCall2(getIncrementIndirectCounterFunc(), + EdgeState, CounterPtrArray); + // clear the predecessor number + Builder.CreateStore(ConstantInt::get(Int32Ty, 0xffffffff), EdgeState); } } } - InsertCounterWriteout(DIF, counters_by_ident); + insertCounterWriteout(DIF, CountersByIdent); return true; } +// All edges with successors that aren't branches are "complex", because it +// requires complex logic to pick which counter to update. +GlobalVariable *GCOVProfiler::buildEdgeLookupTable( + Function *F, + GlobalVariable *Counters, + const UniqueVector &Preds, + const UniqueVector &Succs) { + // TODO: support invoke, threads. We rely on the fact that nothing can modify + // the whole-Module pred edge# between the time we set it and the time we next + // read it. Threads and invoke make this untrue. + + // emit [(succs * preds) x i64*], logically [succ x [pred x i64*]]. + const Type *Int64PtrTy = Type::getInt64PtrTy(*Ctx); + const ArrayType *EdgeTableTy = ArrayType::get( + Int64PtrTy, Succs.size() * Preds.size()); + + Constant **EdgeTable = new Constant*[Succs.size() * Preds.size()]; + Constant *NullValue = Constant::getNullValue(Int64PtrTy); + for (int i = 0, ie = Succs.size() * Preds.size(); i != ie; ++i) + EdgeTable[i] = NullValue; + + unsigned Edge = 0; + for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) { + TerminatorInst *TI = BB->getTerminator(); + int Successors = isa(TI) ? 1 : TI->getNumSuccessors(); + if (Successors && !isa(TI) && !isa(TI)) { + for (int i = 0; i != Successors; ++i) { + BasicBlock *Succ = TI->getSuccessor(i); + IRBuilder<> builder(Succ); + Value *Counter = builder.CreateConstInBoundsGEP2_64(Counters, 0, + Edge + i); + EdgeTable[((Succs.idFor(Succ)-1) * Preds.size()) + + (Preds.idFor(BB)-1)] = cast(Counter); + } + } + Edge += Successors; + } + + GlobalVariable *EdgeTableGV = + new GlobalVariable( + *M, EdgeTableTy, true, GlobalValue::InternalLinkage, + ConstantArray::get(EdgeTableTy, + &EdgeTable[0], Succs.size() * Preds.size()), + "__llvm_gcda_edge_table"); + EdgeTableGV->setUnnamedAddr(true); + return EdgeTableGV; +} + Constant *GCOVProfiler::getStartFileFunc() { - const Type *Args[1] = { Type::getInt8PtrTy(*Ctx) }; + const Type *Args[] = { Type::getInt8PtrTy(*Ctx) }; const FunctionType *FTy = FunctionType::get(Type::getVoidTy(*Ctx), Args, false); - return Mod->getOrInsertFunction("llvm_gcda_start_file", FTy); + return M->getOrInsertFunction("llvm_gcda_start_file", FTy); +} + +Constant *GCOVProfiler::getIncrementIndirectCounterFunc() { + const Type *Args[] = { + Type::getInt32PtrTy(*Ctx), // uint32_t *predecessor + Type::getInt64PtrTy(*Ctx)->getPointerTo(), // uint64_t **state_table_row + }; + const FunctionType *FTy = FunctionType::get(Type::getVoidTy(*Ctx), + Args, false); + return M->getOrInsertFunction("llvm_gcda_increment_indirect_counter", FTy); } Constant *GCOVProfiler::getEmitFunctionFunc() { - const Type *Args[1] = { Type::getInt32Ty(*Ctx) }; + const Type *Args[] = { Type::getInt32Ty(*Ctx) }; const FunctionType *FTy = FunctionType::get(Type::getVoidTy(*Ctx), Args, false); - return Mod->getOrInsertFunction("llvm_gcda_emit_function", FTy); + return M->getOrInsertFunction("llvm_gcda_emit_function", FTy); } Constant *GCOVProfiler::getEmitArcsFunc() { @@ -489,26 +571,38 @@ Constant *GCOVProfiler::getEmitArcsFunc() { }; const FunctionType *FTy = FunctionType::get(Type::getVoidTy(*Ctx), Args, false); - return Mod->getOrInsertFunction("llvm_gcda_emit_arcs", FTy); + return M->getOrInsertFunction("llvm_gcda_emit_arcs", FTy); } Constant *GCOVProfiler::getEndFileFunc() { const FunctionType *FTy = FunctionType::get(Type::getVoidTy(*Ctx), false); - return Mod->getOrInsertFunction("llvm_gcda_end_file", FTy); + return M->getOrInsertFunction("llvm_gcda_end_file", FTy); } -void GCOVProfiler::InsertCounterWriteout( - DebugInfoFinder &DIF, - SmallVector, 8> &counters_by_ident) { +GlobalVariable *GCOVProfiler::getEdgeStateValue() { + GlobalVariable *GV = M->getGlobalVariable("__llvm_gcov_global_state_pred"); + if (!GV) { + GV = new GlobalVariable(*M, Type::getInt32Ty(*Ctx), false, + GlobalValue::InternalLinkage, + ConstantInt::get(Type::getInt32Ty(*Ctx), + 0xffffffff), + "__llvm_gcov_global_state_pred"); + GV->setUnnamedAddr(true); + } + return GV; +} +void GCOVProfiler::insertCounterWriteout( + DebugInfoFinder &DIF, + SmallVector, 8> &CountersByIdent) { const FunctionType *WriteoutFTy = FunctionType::get(Type::getVoidTy(*Ctx), false); Function *WriteoutF = Function::Create(WriteoutFTy, GlobalValue::InternalLinkage, - "__llvm_gcda_writeout", Mod); + "__llvm_gcov_writeout", M); WriteoutF->setUnnamedAddr(true); BasicBlock *BB = BasicBlock::Create(*Ctx, "", WriteoutF); - IRBuilder<> builder(BB); + IRBuilder<> Builder(BB); Constant *StartFile = getStartFileFunc(); Constant *EmitFunction = getEmitFunctionFunc(); @@ -518,25 +612,24 @@ void GCOVProfiler::InsertCounterWriteout( for (DebugInfoFinder::iterator CUI = DIF.compile_unit_begin(), CUE = DIF.compile_unit_end(); CUI != CUE; ++CUI) { DICompileUnit compile_unit(*CUI); - std::string filename_gcda = ReplaceStem(compile_unit.getFilename(), "gcda"); - builder.CreateCall(StartFile, - builder.CreateGlobalStringPtr(filename_gcda)); + std::string FilenameGcda = replaceStem(compile_unit.getFilename(), "gcda"); + Builder.CreateCall(StartFile, + Builder.CreateGlobalStringPtr(FilenameGcda)); for (SmallVector, 8>::iterator - I = counters_by_ident.begin(), E = counters_by_ident.end(); + I = CountersByIdent.begin(), E = CountersByIdent.end(); I != E; ++I) { - builder.CreateCall(EmitFunction, ConstantInt::get(Type::getInt32Ty(*Ctx), + Builder.CreateCall(EmitFunction, ConstantInt::get(Type::getInt32Ty(*Ctx), I->second)); GlobalVariable *GV = I->first; - unsigned num_arcs = + unsigned Arcs = cast(GV->getType()->getElementType())->getNumElements(); - builder.CreateCall2( - EmitArcs, - ConstantInt::get(Type::getInt32Ty(*Ctx), num_arcs), - builder.CreateConstGEP2_64(GV, 0, 0)); + Builder.CreateCall2(EmitArcs, + ConstantInt::get(Type::getInt32Ty(*Ctx), Arcs), + Builder.CreateConstGEP2_64(GV, 0, 0)); } - builder.CreateCall(EndFile); + Builder.CreateCall(EndFile); } - builder.CreateRetVoid(); + Builder.CreateRetVoid(); - InsertProfilingShutdownCall(WriteoutF, Mod); + InsertProfilingShutdownCall(WriteoutF, M); } diff --git a/runtime/libprofile/GCDAProfiling.c b/runtime/libprofile/GCDAProfiling.c index 0bd445c01f591..c8161d44b07ff 100644 --- a/runtime/libprofile/GCDAProfiling.c +++ b/runtime/libprofile/GCDAProfiling.c @@ -15,6 +15,9 @@ |* are only close enough that LCOV will happily parse them. Anything that lcov |* ignores is missing. |* +|* TODO: gcov is multi-process safe by having each exit open the existing file +|* and append to it. We'd like to achieve that and be thread-safe too. +|* \*===----------------------------------------------------------------------===*/ #include "llvm/Support/DataTypes.h" @@ -58,13 +61,33 @@ void llvm_gcda_start_file(const char *filename) { fwrite("adcg*404MVLL", 12, 1, output_file); #ifdef DEBUG_GCDAPROFILING - printf("[%s]\n", filename); + printf("llvmgcda: [%s]\n", filename); +#endif +} + +/* Given an array of pointers to counters (counters), increment the n-th one, + * where we're also given a pointer to n (predecessor). + */ +void llvm_gcda_increment_indirect_counter(uint32_t *predecessor, + uint64_t **counters) { + uint64_t *counter; + if (*predecessor == 0xffffffff) + return; + + /* Don't crash if the pred# is out of sync. This can happen due to threads, + or because of a TODO in GCOVProfiling.cpp buildEdgeLookupTable(). */ + if ((counter = counters[*predecessor])) + ++*counter; +#ifdef DEBUG_GCDAPROFILING + else + printf("llvmgcda: increment_indirect_counter counters=%x, pred=%u\n", + state_table_row, *predecessor); #endif } void llvm_gcda_emit_function(uint32_t ident) { #ifdef DEBUG_GCDAPROFILING - printf("function id=%x\n", ident); + printf("llvmgcda: function id=%x\n", ident); #endif /* function tag */ @@ -84,9 +107,9 @@ void llvm_gcda_emit_arcs(uint32_t num_counters, uint64_t *counters) { } #ifdef DEBUG_GCDAPROFILING - printf(" %u arcs\n", num_counters); + printf("llvmgcda: %u arcs\n", num_counters); for (i = 0; i < num_counters; ++i) { - printf(" %llu\n", (unsigned long long)counters[i]); + printf("llvmgcda: %llu\n", (unsigned long long)counters[i]); } #endif } @@ -98,6 +121,6 @@ void llvm_gcda_end_file() { output_file = NULL; #ifdef DEBUG_GCDAPROFILING - printf("-----\n"); + printf("llvmgcda: -----\n"); #endif } diff --git a/runtime/libprofile/libprofile.exports b/runtime/libprofile/libprofile.exports index aca563fc5ca27..2f25be692047a 100644 --- a/runtime/libprofile/libprofile.exports +++ b/runtime/libprofile/libprofile.exports @@ -6,6 +6,7 @@ llvm_trace_basic_block llvm_increment_path_count llvm_decrement_path_count llvm_gcda_start_file +llvm_gcda_increment_indirect_counter llvm_gcda_emit_function llvm_gcda_emit_arcs llvm_gcda_end_file