maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Fix potential buffer overflow on win32. 

llvm-svn: 129214
This commit is contained in:
Benjamin Kramer 2011-04-09 14:06:12 +00:00
parent 92c07c27be
commit 26e7768f38
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
llvm/lib/MC/MCDisassembler/Disassembler.cpp
View File

@ -128,9 +128,9 @@ public:
} // namespace
//
// LLVMDisasmInstruction() disassmbles a single instruction using the
// LLVMDisasmInstruction() disassembles a single instruction using the
// disassembler context specified in the parameter DC. The bytes of the
// instuction are specified in the parameter Bytes, and contains at least
// instruction are specified in the parameter Bytes, and contains at least
// BytesSize number of bytes. The instruction is at the address specified by
// the PC parameter. If a valid instruction can be disassembled its string is
// returned indirectly in OutString which whos size is specified in the
@ -155,16 +155,13 @@ size_t LLVMDisasmInstruction(LLVMDisasmContextRef DCR, uint8_t *Bytes,
std::string InsnStr;
raw_string_ostream OS(InsnStr);
raw_ostream &Out = OS;
IP->printInst(&Inst, Out);
IP->printInst(&Inst, OS);
OS.flush();
size_t OutputSize = std::min(OutStringSize-1, InsnStr.size());
std::memcpy(OutString, InsnStr.data(), OutputSize);
OutString[OutputSize] = '\0'; // Terminate string.
std::string p;
p = OS.str();
#ifdef LLVM_ON_WIN32
sprintf(OutString, "%s", p.c_str());
#else
snprintf(OutString, OutStringSize, "%s", p.c_str());
#endif
return Size;
}