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;
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;