replica
/
hanchenye-llvm-project
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

ptx: bug fix: use after free 

llvm-svn: 120571
This commit is contained in:
Che-Liang Chiou 2010-12-01 11:45:53 +00:00
parent 25a5e4ca1f
commit b2f77f6206
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
llvm/lib/Target/PTX/PTXAsmPrinter.cpp
View File

@ -130,20 +130,21 @@ void PTXAsmPrinter::EmitFunctionBodyStart() {
}
void PTXAsmPrinter::EmitInstruction(const MachineInstr *MI) {
SmallString<128> sstr;
raw_svector_ostream OS(sstr);
std::string str;
str.reserve(64);
// Write instruction to str
raw_string_ostream OS(str);
printInstruction(MI, OS);
OS << ';';
OS.flush();
// Replace "%type" if found
StringRef strref = OS.str();
size_t pos;
if ((pos = strref.find("%type")) != StringRef::npos) {
std::string str = strref;
if ((pos = str.find("%type")) != std::string::npos)
str.replace(pos, /*strlen("%type")==*/5, getInstructionTypeName(MI));
strref = StringRef(str);
}
StringRef strref = StringRef(str);
OutStreamer.EmitRawText(strref);
}