Add a command-line switch to change the user agent.BUG=757345

Review URL: http://codereview.chromium.org/4059

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@2571 0039d316-1c4b-4281-b951-d872f2087c98

7 files changed, 77 insertions, 58 deletions

diff --git a/webkit/glue/plugins/plugin_host.cc b/webkit/glue/plugins/plugin_host.cc
index 273219d..7993048 100644
--- a/webkit/glue/plugins/plugin_host.cc
+++ b/webkit/glue/plugins/plugin_host.cc
@@ -631,7 +631,7 @@ const char* NPN_UserAgent(NPP id) {
   if (id) {
     scoped_refptr<NPAPI::PluginInstance> plugin = FindInstance(id);
     if (plugin.get() && !plugin->use_mozilla_user_agent())
-      return webkit_glue::GetDefaultUserAgent().c_str();
+      return webkit_glue::GetUserAgent().c_str();
   }
 
   static const char *UA = "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.9a1) Gecko/20061103 Firefox/2.0a1";
diff --git a/webkit/glue/webframeloaderclient_impl.cc b/webkit/glue/webframeloaderclient_impl.cc
index 3250144..7d3011c 100644
--- a/webkit/glue/webframeloaderclient_impl.cc
+++ b/webkit/glue/webframeloaderclient_impl.cc
@@ -1257,8 +1257,7 @@ void WebFrameLoaderClient::setTitle(const String& title, const KURL& url) {
 }
 
 String WebFrameLoaderClient::userAgent(const KURL& url) {
-  return webkit_glue::StdStringToString(
-      webframe_->webview_impl()->GetPreferences().user_agent);
+  return webkit_glue::StdStringToString(webkit_glue::GetUserAgent());
 }
 
 void WebFrameLoaderClient::savePlatformDataToCachedPage(WebCore::CachedPage*) {
diff --git a/webkit/glue/webkit_glue.cc b/webkit/glue/webkit_glue.cc
index 904834b..9e40a85 100644
--- a/webkit/glue/webkit_glue.cc
+++ b/webkit/glue/webkit_glue.cc
@@ -339,74 +339,91 @@ std::string GetWebKitVersion() {
   return StringPrintf("%d.%d", WEBKIT_VERSION_MAJOR, WEBKIT_VERSION_MINOR);
 }
 
-const std::string& GetDefaultUserAgent() {
+namespace {
+
+std::string* user_agent = NULL;
+bool user_agent_requested = false;
+
+void SetUserAgentToDefault() {
+  static std::string default_user_agent;
 #if defined(OS_WIN) || defined(OS_MACOSX)
-  static std::string user_agent;
-  static bool generated_user_agent;
-  if (!generated_user_agent) {
-    int32 os_major_version = 0;
-    int32 os_minor_version = 0;
-    int32 os_bugfix_version = 0;
+  int32 os_major_version = 0;
+  int32 os_minor_version = 0;
+  int32 os_bugfix_version = 0;
 #if defined(OS_WIN)
-    OSVERSIONINFO info = {0};
-    info.dwOSVersionInfoSize = sizeof(info);
-    GetVersionEx(&info);
-    os_major_version = info.dwMajorVersion;
-    os_minor_version = info.dwMinorVersion;
+  OSVERSIONINFO info = {0};
+  info.dwOSVersionInfoSize = sizeof(info);
+  GetVersionEx(&info);
+  os_major_version = info.dwMajorVersion;
+  os_minor_version = info.dwMinorVersion;
 #elif defined(OS_MACOSX)
-    Gestalt(gestaltSystemVersionMajor, 
-        reinterpret_cast<SInt32*>(&os_major_version));
-    Gestalt(gestaltSystemVersionMinor, 
-        reinterpret_cast<SInt32*>(&os_minor_version));
-    Gestalt(gestaltSystemVersionBugFix, 
-        reinterpret_cast<SInt32*>(&os_bugfix_version));
+  Gestalt(gestaltSystemVersionMajor, 
+      reinterpret_cast<SInt32*>(&os_major_version));
+  Gestalt(gestaltSystemVersionMinor, 
+      reinterpret_cast<SInt32*>(&os_minor_version));
+  Gestalt(gestaltSystemVersionBugFix, 
+      reinterpret_cast<SInt32*>(&os_bugfix_version));
 #endif
 
-    // Get the product name and version, and replace Safari's Version/X string
-    // with it.  This is done to expose our product name in a manner that is
-    // maximally compatible with Safari, we hope!!
-    std::string product;
+  // Get the product name and version, and replace Safari's Version/X string
+  // with it.  This is done to expose our product name in a manner that is
+  // maximally compatible with Safari, we hope!!
+  std::string product;
 
-    FileVersionInfo* version_info =
-        FileVersionInfo::CreateFileVersionInfoForCurrentModule();
-    if (version_info)
-      product = "Chrome/" + WideToASCII(version_info->product_version());
+  FileVersionInfo* version_info =
+      FileVersionInfo::CreateFileVersionInfoForCurrentModule();
+  if (version_info)
+    product = "Chrome/" + WideToASCII(version_info->product_version());
 
-    if (product.empty())
-      product = "Version/3.1";
+  if (product.empty())
+    product = "Version/3.1";
 
-    // Derived from Safari's UA string.
-    StringAppendF(
-        &user_agent,
+  // Derived from Safari's UA string.
+  StringAppendF(
+      &default_user_agent,
 #if defined(OS_WIN)
-        "Mozilla/5.0 (Windows; U; Windows NT %d.%d; en-US) AppleWebKit/%d.%d"
+      "Mozilla/5.0 (Windows; U; Windows NT %d.%d; en-US) AppleWebKit/%d.%d"
 #elif defined(OS_MACOSX)
-        "Mozilla/5.0 (Macintosh; U; Intel Mac OS X %d_%d_%d; en-US) AppleWebKit/%d.%d"
+      "Mozilla/5.0 (Macintosh; U; Intel Mac OS X %d_%d_%d; en-US) "
+      "AppleWebKit/%d.%d"
 #endif
-        " (KHTML, like Gecko) %s Safari/%d.%d",
-        os_major_version,
-        os_minor_version,
+      " (KHTML, like Gecko) %s Safari/%d.%d",
+      os_major_version,
+      os_minor_version,
 #if defined(OS_MACOSX)
-        os_bugfix_version,
+      os_bugfix_version,
 #endif
-        WEBKIT_VERSION_MAJOR,
-        WEBKIT_VERSION_MINOR,
-        product.c_str(),
-        WEBKIT_VERSION_MAJOR,
-        WEBKIT_VERSION_MINOR
-        );
-
-    generated_user_agent = true;
-  }
-
-  return user_agent;
+      WEBKIT_VERSION_MAJOR,
+      WEBKIT_VERSION_MINOR,
+      product.c_str(),
+      WEBKIT_VERSION_MAJOR,
+      WEBKIT_VERSION_MINOR
+      );
 #else
   // TODO(port): we need something like FileVersionInfo for our UA string.
   NOTIMPLEMENTED();
-  return EmptyString();
 #endif
+  user_agent = &default_user_agent;
 }
 
+};
+
+void SetUserAgent(const std::string& new_user_agent) {
+  DCHECK(!user_agent_requested) << "Setting the user agent after someone has "
+      "already requested it can result in unexpected behavior.";
+  static std::string overridden_user_agent;
+  overridden_user_agent = new_user_agent;  // If you combine this with the
+                                           // previous line, the function only
+                                           // works the first time.
+  user_agent = &overridden_user_agent;
+}
+
+const std::string& GetUserAgent() {
+  if (!user_agent)
+    SetUserAgentToDefault();
+  user_agent_requested = true;
+  return *user_agent;
+}
 
 void NotifyJSOutOfMemory(WebCore::Frame* frame) {
   if (!frame)
diff --git a/webkit/glue/webkit_glue.h b/webkit/glue/webkit_glue.h
index 411ad20..78c953a 100644
--- a/webkit/glue/webkit_glue.h
+++ b/webkit/glue/webkit_glue.h
@@ -88,8 +88,14 @@ void ResetBeforeTestRun(WebView* view);
 // Returns the WebKit version (major.minor).
 std::string GetWebKitVersion();
 
-// Returns the user agent.
-const std::string& GetDefaultUserAgent();
+// Called to override the default user agent with a custom one.  Call this
+// before anyone actually asks for the user agent in order to prevent
+// inconsistent behavior.
+void SetUserAgent(const std::string& new_user_agent);
+
+// Returns the user agent, which is usually the default user agent but may be
+// overriden by a call to SetUserAgent() (which should be done at startup).
+const std::string& GetUserAgent();
 
 // Creates serialized state for the specified URL. This is a variant of
 // HistoryItemToString (in glue_serialize) that is used during session restore
diff --git a/webkit/glue/webpreferences.h b/webkit/glue/webpreferences.h
index 524e38e..4a08fa0 100644
--- a/webkit/glue/webpreferences.h
+++ b/webkit/glue/webpreferences.h
@@ -44,8 +44,6 @@ struct WebPreferences {
   bool user_style_sheet_enabled;
   GURL user_style_sheet_location;
 
-  std::string user_agent;
-
   // We try to keep the default values the same as the default values in
   // chrome, except for the cases where it would require lots of extra work for
   // the embedder to use the same default value.
diff --git a/webkit/tools/test_shell/test_shell.cc b/webkit/tools/test_shell/test_shell.cc
index 283e730..f11d21a 100644
--- a/webkit/tools/test_shell/test_shell.cc
+++ b/webkit/tools/test_shell/test_shell.cc
@@ -247,7 +247,6 @@ void TestShell::ResetWebPreferences() {
         web_prefs_->shrinks_standalone_images_to_fit = false;
         web_prefs_->uses_universal_detector = false;
         web_prefs_->text_areas_are_resizable = false;
-        web_prefs_->user_agent = webkit_glue::GetDefaultUserAgent();
         web_prefs_->dashboard_compatibility_mode = false;
         web_prefs_->java_enabled = true;
         web_prefs_->allow_scripts_to_close_windows = false;
diff --git a/webkit/tools/test_shell/test_shell_request_context.cc b/webkit/tools/test_shell/test_shell_request_context.cc
index 2fdf164..9767b83 100644
--- a/webkit/tools/test_shell/test_shell_request_context.cc
+++ b/webkit/tools/test_shell/test_shell_request_context.cc
@@ -22,7 +22,7 @@ void TestShellRequestContext::Init(
     net::HttpCache::Mode cache_mode) {
   cookie_store_ = new net::CookieMonster();
 
-  user_agent_ = webkit_glue::GetDefaultUserAgent();
+  user_agent_ = webkit_glue::GetUserAgent();
 
   // hard-code A-L and A-C for test shells
   accept_language_ = "en-us,en";