Added support for running reliability tests for ChromeFrame on similar lines ...

chrome_frame/utils.cc

 // Copyright (c) 2009 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 <htiframe.h>
 #include <mshtml.h>
 #include <shlobj.h>
 #include <wininet.h>
 
 #include "chrome_frame/html_utils.h"
@@ skipping 29 matching lines @@
 static const wchar_t kChromeFrameNPAPIKey[] =
     L"Software\\MozillaPlugins\\@google.com/ChromeFrame,version=1.0";
 static const wchar_t kChromeFramePersistNPAPIReg[] = L"PersistNPAPIReg";
 
 // Used to isolate chrome frame builds from google chrome release channels.
 const wchar_t kChromeFrameOmahaSuffix[] = L"-cf";
 const wchar_t kDevChannelName[] = L"-dev";
 
 const wchar_t kChromeAttachExternalTabPrefix[] = L"attach_external_tab";
 
+// Indicates that we are running in a test environment, where execptions, etc
+// are handled by the chrome test crash server.
+const wchar_t kChromeFrameHeadlessMode[] = L"ChromeFrameHeadlessMode";
+
 HRESULT UtilRegisterTypeLib(HINSTANCE tlb_instance,
                             LPCOLESTR index,
                             bool for_current_user_only) {
   CComBSTR path;
   CComPtr<ITypeLib> type_lib;
   HRESULT hr = AtlLoadTypeLib(tlb_instance, index, &path, &type_lib);
   if (SUCCEEDED(hr)) {
     hr = UtilRegisterTypeLib(type_lib, path, NULL, for_current_user_only);
   }
   return hr;
@@ skipping 479 matching lines @@
   }
 
   return ret;
 }
 
 bool GetConfigBool(bool default_value, const wchar_t* value_name) {
   DWORD value = GetConfigInt(default_value, value_name);
   return (value != FALSE);
 }
 
+bool SetConfigInt(const wchar_t* value_name, int value) {
+  RegKey config_key;
+  if (config_key.Open(HKEY_CURRENT_USER, kChromeFrameConfigKey,
+                      KEY_SET_VALUE)) {
+    if (config_key.WriteValue(value_name, value)) {
+      return true;
+    }
+  }
+
+  return false;
+}
+
+bool SetConfigBool(const wchar_t* value_name, bool value) {
+  return SetConfigInt(value_name, value);
+}
+
+bool DeleteConfigValue(const wchar_t* value_name) {
+  RegKey config_key;
+  if (config_key.Open(HKEY_CURRENT_USER, kChromeFrameConfigKey,
+                      KEY_WRITE)) {
+    return config_key.DeleteValue(value_name);
+  }
+  return false;
+}
+
 bool IsOptInUrl(const wchar_t* url) {
   RegKey config_key;
   if (!config_key.Open(HKEY_CURRENT_USER, kChromeFrameConfigKey, KEY_READ))
     return false;
 
   RegistryValueIterator optin_urls_list(config_key.Handle(),
                                         kChromeFrameOptinUrlsKey);
   while (optin_urls_list.Valid()) {
     if (MatchPatternWide(url, optin_urls_list.Name()))
       return true;
@@ skipping 101 matching lines @@
         DLOG(INFO) << "Sub frame detected";
         is_non_top_level_request = true;
       }
     }
   } else {
     is_non_top_level_request = true;
   }
 
   return is_non_top_level_request;
 }
+
+bool IsHeadlessMode() {
+  bool headless = GetConfigBool(false, kChromeFrameHeadlessMode);
+  return headless;
+}
+