commit fcd7af40529f4e74c27429606b5cd2e3aa5166dd
author Aaron Smith <aaron.smith@microsoft.com> Thu Mar 22 03:57:06 2018 +0000
committer Aaron Smith <aaron.smith@microsoft.com> Thu Mar 22 03:57:06 2018 +0000
tree 5d50804fbe807e5f17ad14ced7afa15d9a3a3641
parent 164c112ac4acdc48d14e72db3621d2eef46ec181

[PDB] Get more DIA table enumerators

Rename the original function and make it a static template.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@328177 91177308-0d34-0410-b5e6-96231b3b80d8

lib/DebugInfo/PDB/DIA/DIASession.cpp

diff --git a/lib/DebugInfo/PDB/DIA/DIASession.cpp b/lib/DebugInfo/PDB/DIA/DIASession.cpp
index 3a4ecdb..489184a 100644
--- a/lib/DebugInfo/PDB/DIA/DIASession.cpp
+++ b/lib/DebugInfo/PDB/DIA/DIASession.cpp
@@ -327,9 +327,10 @@
   return llvm::make_unique<DIAEnumTables>(DiaEnumerator);
 }
 
-static CComPtr<IDiaEnumInjectedSources>
-getEnumInjectedSources(IDiaSession &Session) {
-  CComPtr<IDiaEnumInjectedSources> EIS;
+template <class T>
+static CComPtr<T>
+getTableEnumerator(IDiaSession &Session) {
+  CComPtr<T> Enumerator;
   CComPtr<IDiaEnumTables> ET;
   CComPtr<IDiaTable> Table;
   ULONG Count = 0;
@@ -340,15 +341,16 @@
   while (ET->Next(1, &Table, &Count) == S_OK && Count == 1) {
     // There is only one table that matches the given iid
     if (S_OK ==
-        Table->QueryInterface(__uuidof(IDiaEnumInjectedSources), (void **)&EIS))
+        Table->QueryInterface(__uuidof(T), (void **)&Enumerator))
       break;
     Table.Release();
   }
-  return EIS;
+  return Enumerator;
 }
 std::unique_ptr<IPDBEnumInjectedSources>
 DIASession::getInjectedSources() const {
-  CComPtr<IDiaEnumInjectedSources> Files = getEnumInjectedSources(*Session);
+  CComPtr<IDiaEnumInjectedSources> Files =
+      getTableEnumerator<IDiaEnumInjectedSources>(*Session);
   if (!Files)
     return nullptr;