[llvm] r271103 - Make sure the directory contains info for all streams

[David Majnemer via llvm-commits]

Author: majnemer
Date: Sat May 28 00:59:19 2016
New Revision: 271103

URL: http://llvm.org/viewvc/llvm-project?rev=271103&view=rev
Log:
Make sure the directory contains info for all streams

Modified:
    llvm/trunk/lib/DebugInfo/PDB/Raw/PDBFile.cpp

Modified: llvm/trunk/lib/DebugInfo/PDB/Raw/PDBFile.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/DebugInfo/PDB/Raw/PDBFile.cpp?rev=271103&r1=271102&r2=271103&view=diff
==============================================================================
--- llvm/trunk/lib/DebugInfo/PDB/Raw/PDBFile.cpp (original)
+++ llvm/trunk/lib/DebugInfo/PDB/Raw/PDBFile.cpp Sat May 28 00:59:19 2016
@@ -269,10 +269,15 @@ Error PDBFile::parseStreamData() {
     }
   }
 
-  for (uint32_t SI = 0; SI != NumStreams; ++SI) {
+  if (Context->StreamSizes.size() != NumStreams)
+    return make_error<RawError>(
+        raw_error_code::corrupt_file,
+        "The directory has fewer streams then expected");
+
+  for (uint32_t I = 0; I != NumStreams; ++I) {
     uint64_t NumExpectedStreamBlocks =
-        bytesToBlocks(getStreamByteSize(SI), getBlockSize());
-    size_t NumStreamBlocks = getStreamBlockList(SI).size();
+        bytesToBlocks(getStreamByteSize(I), getBlockSize());
+    size_t NumStreamBlocks = getStreamBlockList(I).size();
     if (NumExpectedStreamBlocks != NumStreamBlocks)
       return make_error<RawError>(raw_error_code::corrupt_file,
                                   "The number of stream blocks is not "