Use Scoped COM helpers for collecting GPU information

content/gpu/gpu_info_collector_win.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/gpu/gpu_info_collector.h"
 
 #include <windows.h>
 #include <d3d9.h>
 #include <setupapi.h>
 #include <winsatcominterfacei.h>
 
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/logging.h"
 #include "base/scoped_native_library.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
+#include "base/win/scoped_com_initializer.h"
+#include "base/win/scoped_comptr.h"
 #include "ui/gfx/gl/gl_implementation.h"
 #include "ui/gfx/gl/gl_surface_egl.h"
 
 // ANGLE seems to require that main.h be included before any other ANGLE header.
 #include "libEGL/Display.h"
 #include "libEGL/main.h"
 
 namespace {
 
 // The version number stores the major and minor version in the least 16 bits;
 // for example, 2.5 is 0x00000205.
 // Returned string is in the format of "major.minor".
 std::string VersionNumberToString(uint32 version_number) {
   int hi = (version_number >> 8) & 0xff;
   int low = version_number & 0xff;
   return base::IntToString(hi) + "." + base::IntToString(low);
 }
 
 float GetAssessmentScore(IProvideWinSATResultsInfo* results,
                          WINSAT_ASSESSMENT_TYPE type) {
-  IProvideWinSATAssessmentInfo* subcomponent = NULL;
-  if (FAILED(results->GetAssessmentInfo(type, &subcomponent)))
+  base::win::ScopedComPtr<IProvideWinSATAssessmentInfo> subcomponent;
+  if (FAILED(results->GetAssessmentInfo(type, subcomponent.Receive())))
     return 0.0;
 
   float score = 0.0;
   if (FAILED(subcomponent->get_Score(&score)))
     score = 0.0;
-  subcomponent->Release();
   return score;
 }
 
 content::GpuPerformanceStats RetrieveGpuPerformanceStats() {
-  IQueryRecentWinSATAssessment* assessment = NULL;
-  IProvideWinSATResultsInfo* results = NULL;
-
   content::GpuPerformanceStats stats;
 
-  do {
-    HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
-    if (FAILED(hr)) {
-      LOG(ERROR) << "CoInitializeEx() failed";
-      break;
-    }
+  base::win::ScopedCOMInitializer com_initializer;
+  if (!com_initializer.succeeded()) {
+    LOG(ERROR) << "CoInitializeEx() failed";
+    return stats;
+  }
 
-    hr = CoCreateInstance(__uuidof(CQueryWinSAT),
-                          NULL,
-                          CLSCTX_INPROC_SERVER,
-                          __uuidof(IQueryRecentWinSATAssessment),
-                          reinterpret_cast<void**>(&assessment));
-    if (FAILED(hr)) {
-      LOG(ERROR) << "CoCreateInstance() failed";
-      break;
-    }
+  base::win::ScopedComPtr<IQueryRecentWinSATAssessment> assessment;
+  HRESULT hr = assessment.CreateInstance(__uuidof(CQueryWinSAT), NULL,
+                                         CLSCTX_INPROC_SERVER);
+  if (FAILED(hr)) {
+    LOG(ERROR) << "CoCreateInstance() failed";
+    return stats;
+  }
 
-    hr = assessment->get_Info(&results);
-    if (FAILED(hr)) {
-      LOG(ERROR) << "get_Info() failed";
-      break;
-    }
+  base::win::ScopedComPtr<IProvideWinSATResultsInfo> results;
+  hr = assessment->get_Info(results.Receive());
+  if (FAILED(hr)) {
+    LOG(ERROR) << "get_Info() failed";
+    return stats;
+  }
 
-    WINSAT_ASSESSMENT_STATE state = WINSAT_ASSESSMENT_STATE_UNKNOWN;
-    hr = results->get_AssessmentState(&state);
-    if (FAILED(hr)) {
-      LOG(ERROR) << "get_AssessmentState() failed";
-      break;
-    }
-    if (state != WINSAT_ASSESSMENT_STATE_VALID &&
-        state != WINSAT_ASSESSMENT_STATE_INCOHERENT_WITH_HARDWARE) {
-      LOG(ERROR) << "Can't retrieve a valid assessment";
-      break;
-    }
+  WINSAT_ASSESSMENT_STATE state = WINSAT_ASSESSMENT_STATE_UNKNOWN;
+  hr = results->get_AssessmentState(&state);
+  if (FAILED(hr)) {
+    LOG(ERROR) << "get_AssessmentState() failed";
+    return stats;
+  }
 
-    hr = results->get_SystemRating(&stats.overall);
-    if (FAILED(hr))
-      LOG(ERROR) << "Get overall score failed";
+  if (state != WINSAT_ASSESSMENT_STATE_VALID &&
+      state != WINSAT_ASSESSMENT_STATE_INCOHERENT_WITH_HARDWARE) {
+    LOG(ERROR) << "Can't retrieve a valid assessment";
+    return stats;
+  }
 
-    stats.gaming = GetAssessmentScore(results, WINSAT_ASSESSMENT_D3D);
-    if (stats.gaming == 0.0)
-      LOG(ERROR) << "Get gaming score failed";
+  hr = results->get_SystemRating(&stats.overall);
+  if (FAILED(hr))
+    LOG(ERROR) << "Get overall score failed";
 
-    stats.graphics = GetAssessmentScore(results, WINSAT_ASSESSMENT_GRAPHICS);
-    if (stats.graphics == 0.0)
-      LOG(ERROR) << "Get graphics score failed";
-  } while (false);
+  stats.gaming = GetAssessmentScore(results, WINSAT_ASSESSMENT_D3D);
+  if (stats.gaming == 0.0)
+    LOG(ERROR) << "Get gaming score failed";
 
-  if (assessment)
-    assessment->Release();
-  if (results)
-    results->Release();
-  CoUninitialize();
+  stats.graphics = GetAssessmentScore(results, WINSAT_ASSESSMENT_GRAPHICS);
+  if (stats.graphics == 0.0)
+    LOG(ERROR) << "Get graphics score failed";
 
   return stats;
 }
 
 }  // namespace anonymous
 
 namespace gpu_info_collector {
 
 bool CollectGraphicsInfo(content::GPUInfo* gpu_info) {
   DCHECK(gpu_info);
@@ skipping 23 matching lines @@
     LOG(ERROR) << "gfx::BaseEGLContext::GetDisplay() failed";
     return false;
   }
 
   IDirect3DDevice9* device = display->getDevice();
   if (!device) {
     LOG(ERROR) << "display->getDevice() failed";
     return false;
   }
 
-  IDirect3D9* d3d = NULL;
-  if (FAILED(device->GetDirect3D(&d3d))) {
+  base::win::ScopedComPtr<IDirect3D9> d3d;
+  if (FAILED(device->GetDirect3D(d3d.Receive()))) {
     LOG(ERROR) << "device->GetDirect3D(&d3d) failed";
     return false;
   }
 
   if (!CollectGraphicsInfoD3D(d3d, gpu_info))
     return false;
 
   // DirectX diagnostics are collected asynchronously because it takes a
   // couple of seconds. Do not mark gpu_info as complete until that is done.
   return true;
@@ skipping 25 matching lines @@
     gpu_info->pixel_shader_version =
         VersionNumberToString(d3d_caps.PixelShaderVersion);
     gpu_info->vertex_shader_version =
         VersionNumberToString(d3d_caps.VertexShaderVersion);
   } else {
     LOG(ERROR) << "d3d->GetDeviceCaps() failed";
     succeed = false;
   }
 
   // Get can_lose_context
-  bool can_lose_context = false;
-  IDirect3D9Ex* d3dex = NULL;
-  if (SUCCEEDED(d3d->QueryInterface(__uuidof(IDirect3D9Ex),
-                                    reinterpret_cast<void**>(&d3dex)))) {
-    d3dex->Release();
-  } else {
-    can_lose_context = true;
-  }
-  gpu_info->can_lose_context = can_lose_context;
+  base::win::ScopedComPtr<IDirect3D9Ex> d3dex;
+  if (SUCCEEDED(d3dex.QueryFrom(d3d)))
+    gpu_info->can_lose_context = false;
+  else
+    gpu_info->can_lose_context = true;
 
-  d3d->Release();
   return true;
 }
 
 bool CollectVideoCardInfo(content::GPUInfo* gpu_info) {
   DCHECK(gpu_info);
 
   // nvd3d9wrap.dll is loaded into all processes when Optimus is enabled.
   HMODULE nvd3d9wrap = GetModuleHandleW(L"nvd3d9wrap.dll");
   gpu_info->optimus = nvd3d9wrap != NULL;
 
@@ skipping 20 matching lines @@
     // TODO(zmo): we only need to call CollectDriverInfoD3D() if we use ANGLE.
     return CollectDriverInfoD3D(id, gpu_info);
   }
   return false;
 }
 
 bool CollectDriverInfoD3D(const std::wstring& device_id,
                           content::GPUInfo* gpu_info) {
   // create device info for the display device
   HDEVINFO device_info = SetupDiGetClassDevsW(
-    NULL, device_id.c_str(), NULL,
-    DIGCF_PRESENT | DIGCF_PROFILE | DIGCF_ALLCLASSES);
+      NULL, device_id.c_str(), NULL,
+      DIGCF_PRESENT | DIGCF_PROFILE | DIGCF_ALLCLASSES);
   if (device_info == INVALID_HANDLE_VALUE) {
     LOG(ERROR) << "Creating device info failed";
     return false;
   }
 
   DWORD index = 0;
   bool found = false;
   SP_DEVINFO_DATA device_info_data;
   device_info_data.cbSize = sizeof(device_info_data);
   while (SetupDiEnumDeviceInfo(device_info, index++, &device_info_data)) {
@@ skipping 65 matching lines @@
 
   size_t pos = gl_version_string.find_last_not_of("0123456789.");
   if (pos != std::string::npos && pos < gl_version_string.length() - 1) {
     gpu_info->driver_version = gl_version_string.substr(pos + 1);
     return true;
   }
   return false;
 }
 
 }  // namespace gpu_info_collector