[PATCH] D139315: [NFC][LLVM-MT] Fix buffer-overflow when comparing buffers

[Juan Manuel Martinez Caamaño via Phabricator via llvm-commits]

jmmartinez created this revision.
Herald added a project: All.
jmmartinez requested review of this revision.
Herald added a project: LLVM.
Herald added a subscriber: llvm-commits.

Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D139315

Files:
  llvm/tools/llvm-mt/llvm-mt.cpp


Index: llvm/tools/llvm-mt/llvm-mt.cpp
===================================================================
--- llvm/tools/llvm-mt/llvm-mt.cpp
+++ llvm/tools/llvm-mt/llvm-mt.cpp
@@ -150,9 +150,11 @@
     bool Same = false;
     if (OutBuffOrErr) {
       const std::unique_ptr<MemoryBuffer> &FileBuffer = *OutBuffOrErr;
-      Same = std::equal(OutputBuffer->getBufferStart(),
-                        OutputBuffer->getBufferEnd(),
-                        FileBuffer->getBufferStart());
+      bool SameSize =
+          OutputBuffer->getBufferSize() == FileBuffer->getBufferSize();
+      Same = SameSize && std::equal(OutputBuffer->getBufferStart(),
+                                    OutputBuffer->getBufferEnd(),
+                                    FileBuffer->getBufferStart());
     }
     if (!Same) {
 #if LLVM_ON_UNIX


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D139315.480071.patch
Type: text/x-patch
Size: 829 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20221205/6f5a9849/attachment.bin>