Adding return value checking for CoCreateInstance of IShellWindows

Fixes issue SeleniumHQ#5564.
LiangY · Mar 20, 2018 · 81371b4 · 81371b4
1 parent f454a5c
commit 81371b4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cpp/iedriver/BrowserFactory.cpp b/cpp/iedriver/BrowserFactory.cpp
@@ -494,10 +494,14 @@ bool BrowserFactory::AttachToBrowserUsingShellWindows(
   LOG(TRACE) << "Entering BrowserFactory::AttachToBrowserUsingShellWindows";
 
   CComPtr<IShellWindows> shell_windows;
-  shell_windows.CoCreateInstance(CLSID_ShellWindows);
+  HRESULT hr = shell_windows.CoCreateInstance(CLSID_ShellWindows);
+  if (FAILED(hr)) {
+    LOGHR(WARN, hr) << "Unable to create an object using the IShellWindows interface with CoCreateInstance";
+    return false;
+  }
 
   CComPtr<IUnknown> enumerator_unknown;
-  HRESULT hr = shell_windows->_NewEnum(&enumerator_unknown);
+  hr = shell_windows->_NewEnum(&enumerator_unknown);
   if (FAILED(hr)) {
     LOGHR(WARN, hr) << "Unable to get enumerator from IShellWindows interface";
     return false;