1 files changed, 152 insertions, 0 deletions
diff --git a/contrib/llvm/tools/llvm-pdbutil/StreamUtil.cpp b/contrib/llvm/tools/llvm-pdbutil/StreamUtil.cpp
new file mode 100644
index 0000000..4d35200
--- /dev/null
+++ b/contrib/llvm/tools/llvm-pdbutil/StreamUtil.cpp
@@ -0,0 +1,152 @@
+//===- StreamUtil.cpp - PDB stream utilities --------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "StreamUtil.h"
+#include "FormatUtil.h"
+
+#include "llvm/ADT/DenseMap.h"
+#include "llvm/ADT/DenseMapInfo.h"
+#include "llvm/DebugInfo/PDB/Native/DbiModuleDescriptor.h"
+#include "llvm/DebugInfo/PDB/Native/DbiModuleList.h"
+#include "llvm/DebugInfo/PDB/Native/DbiStream.h"
+#include "llvm/DebugInfo/PDB/Native/InfoStream.h"
+#include "llvm/DebugInfo/PDB/Native/PDBFile.h"
+#include "llvm/DebugInfo/PDB/Native/TpiStream.h"
+
+using namespace llvm;
+using namespace llvm::pdb;
+
+void llvm::pdb::discoverStreamPurposes(
+    PDBFile &File,
+    SmallVectorImpl<std::pair<StreamPurpose, std::string>> &Purposes) {
+  // It's OK if we fail to load some of these streams, we still attempt to print
+  // what we can.
+  auto Dbi = File.getPDBDbiStream();
+  auto Tpi = File.getPDBTpiStream();
+  auto Ipi = File.getPDBIpiStream();
+  auto Info = File.getPDBInfoStream();
+
+  uint32_t StreamCount = File.getNumStreams();
+  DenseMap<uint16_t, DbiModuleDescriptor> ModStreams;
+  DenseMap<uint16_t, std::string> NamedStreams;
+
+  if (Dbi) {
+    const DbiModuleList &Modules = Dbi->modules();
+    for (uint32_t I = 0; I < Modules.getModuleCount(); ++I) {
+      DbiModuleDescriptor Descriptor = Modules.getModuleDescriptor(I);
+      uint16_t SN = Descriptor.getModuleStreamIndex();
+      if (SN != kInvalidStreamIndex)
+        ModStreams[SN] = Descriptor;
+    }
+  }
+  if (Info) {
+    for (auto &NSE : Info->named_streams()) {
+      if (NSE.second != kInvalidStreamIndex)
+        NamedStreams[NSE.second] = NSE.first();
+    }
+  }
+
+  Purposes.resize(StreamCount);
+  for (uint16_t StreamIdx = 0; StreamIdx < StreamCount; ++StreamIdx) {
+    std::pair<StreamPurpose, std::string> Value;
+    if (StreamIdx == OldMSFDirectory)
+      Value = std::make_pair(StreamPurpose::Other, "Old MSF Directory");
+    else if (StreamIdx == StreamPDB)
+      Value = std::make_pair(StreamPurpose::Other, "PDB Stream");
+    else if (StreamIdx == StreamDBI)
+      Value = std::make_pair(StreamPurpose::Other, "DBI Stream");
+    else if (StreamIdx == StreamTPI)
+      Value = std::make_pair(StreamPurpose::Other, "TPI Stream");
+    else if (StreamIdx == StreamIPI)
+      Value = std::make_pair(StreamPurpose::Other, "IPI Stream");
+    else if (Dbi && StreamIdx == Dbi->getGlobalSymbolStreamIndex())
+      Value = std::make_pair(StreamPurpose::Other, "Global Symbol Hash");
+    else if (Dbi && StreamIdx == Dbi->getPublicSymbolStreamIndex())
+      Value = std::make_pair(StreamPurpose::Other, "Public Symbol Hash");
+    else if (Dbi && StreamIdx == Dbi->getSymRecordStreamIndex())
+      Value = std::make_pair(StreamPurpose::Other, "Public Symbol Records");
+    else if (Tpi && StreamIdx == Tpi->getTypeHashStreamIndex())
+      Value = std::make_pair(StreamPurpose::Other, "TPI Hash");
+    else if (Tpi && StreamIdx == Tpi->getTypeHashStreamAuxIndex())
+      Value = std::make_pair(StreamPurpose::Other, "TPI Aux Hash");
+    else if (Ipi && StreamIdx == Ipi->getTypeHashStreamIndex())
+      Value = std::make_pair(StreamPurpose::Other, "IPI Hash");
+    else if (Ipi && StreamIdx == Ipi->getTypeHashStreamAuxIndex())
+      Value = std::make_pair(StreamPurpose::Other, "IPI Aux Hash");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::Exception))
+      Value = std::make_pair(StreamPurpose::Other, "Exception Data");
+    else if (Dbi && StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::Fixup))
+      Value = std::make_pair(StreamPurpose::Other, "Fixup Data");
+    else if (Dbi && StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::FPO))
+      Value = std::make_pair(StreamPurpose::Other, "FPO Data");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::NewFPO))
+      Value = std::make_pair(StreamPurpose::Other, "New FPO Data");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::OmapFromSrc))
+      Value = std::make_pair(StreamPurpose::Other, "Omap From Source Data");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::OmapToSrc))
+      Value = std::make_pair(StreamPurpose::Other, "Omap To Source Data");
+    else if (Dbi && StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::Pdata))
+      Value = std::make_pair(StreamPurpose::Other, "Pdata");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::SectionHdr))
+      Value = std::make_pair(StreamPurpose::Other, "Section Header Data");
+    else if (Dbi &&
+             StreamIdx ==
+                 Dbi->getDebugStreamIndex(DbgHeaderType::SectionHdrOrig))
+      Value =
+          std::make_pair(StreamPurpose::Other, "Section Header Original Data");
+    else if (Dbi &&
+             StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::TokenRidMap))
+      Value = std::make_pair(StreamPurpose::Other, "Token Rid Data");
+    else if (Dbi && StreamIdx == Dbi->getDebugStreamIndex(DbgHeaderType::Xdata))
+      Value = std::make_pair(StreamPurpose::Other, "Xdata");
+    else {
+      auto ModIter = ModStreams.find(StreamIdx);
+      auto NSIter = NamedStreams.find(StreamIdx);
+      if (ModIter != ModStreams.end()) {
+        Value = std::make_pair(StreamPurpose::ModuleStream,
+                               ModIter->second.getModuleName());
+      } else if (NSIter != NamedStreams.end()) {
+        Value = std::make_pair(StreamPurpose::NamedStream, NSIter->second);
+      } else {
+        Value = std::make_pair(StreamPurpose::Other, "???");
+      }
+    }
+    Purposes[StreamIdx] = Value;
+  }
+
+  // Consume errors from missing streams.
+  if (!Dbi)
+    consumeError(Dbi.takeError());
+  if (!Tpi)
+    consumeError(Tpi.takeError());
+  if (!Ipi)
+    consumeError(Ipi.takeError());
+  if (!Info)
+    consumeError(Info.takeError());
+}
+
+void llvm::pdb::discoverStreamPurposes(PDBFile &File,
+                                       SmallVectorImpl<std::string> &Purposes) {
+  SmallVector<std::pair<StreamPurpose, std::string>, 24> SP;
+  discoverStreamPurposes(File, SP);
+  Purposes.reserve(SP.size());
+  for (const auto &P : SP) {
+    if (P.first == StreamPurpose::NamedStream)
+      Purposes.push_back(formatv("Named Stream \"{0}\"", P.second));
+    else if (P.first == StreamPurpose::ModuleStream)
+      Purposes.push_back(formatv("Module \"{0}\"", P.second));
+    else
+      Purposes.push_back(P.second);
+  }
+}