aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/Support/MemoryBuffer.cpp
diff options
context:
space:
mode:
authorDmitry Vassiliev <dvassiliev@accesssoftek.com>2022-03-11 13:16:58 +0400
committerDmitry Vassiliev <dvassiliev@accesssoftek.com>2022-03-11 13:16:58 +0400
commitdfeb9781557ca655bb99a31f4bcf4d0187d7dd38 (patch)
treebd3c34dca1011c86fee51a58888c9e8bfef3cad7 /llvm/lib/Support/MemoryBuffer.cpp
parent374bb6dd8090ffaca5aad86b2d1cd9acad401f3a (diff)
downloadllvm-dfeb9781557ca655bb99a31f4bcf4d0187d7dd38.zip
llvm-dfeb9781557ca655bb99a31f4bcf4d0187d7dd38.tar.gz
llvm-dfeb9781557ca655bb99a31f4bcf4d0187d7dd38.tar.bz2
Fixed a roll-over on size_t in getNewUninitMemBuffer()
Reviewed By: serge-sans-paille Differential Revision: https://reviews.llvm.org/D121399
Diffstat (limited to 'llvm/lib/Support/MemoryBuffer.cpp')
-rw-r--r--llvm/lib/Support/MemoryBuffer.cpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/llvm/lib/Support/MemoryBuffer.cpp b/llvm/lib/Support/MemoryBuffer.cpp
index 5003822..872b874 100644
--- a/llvm/lib/Support/MemoryBuffer.cpp
+++ b/llvm/lib/Support/MemoryBuffer.cpp
@@ -286,6 +286,8 @@ WritableMemoryBuffer::getNewUninitMemBuffer(size_t Size, const Twine &BufferName
StringRef NameRef = BufferName.toStringRef(NameBuf);
size_t AlignedStringLen = alignTo(sizeof(MemBuffer) + NameRef.size() + 1, 16);
size_t RealLen = AlignedStringLen + Size + 1;
+ if (RealLen <= Size) // Check for rollover.
+ return nullptr;
char *Mem = static_cast<char*>(operator new(RealLen, std::nothrow));
if (!Mem)
return nullptr;