Add support for UA spoofing, and spoof Safari's UA string when loading URLs...

webkit/glue/webkit_glue.cc

 // Copyright (c) 2006-2008 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 "base/compiler_specific.h"
 #include "build/build_config.h"
 
 #if defined(OS_WIN)
 #include <objidl.h>
 #include <mlang.h>
@@ skipping 298 matching lines @@
     return;
   delegate->OnNavStateChanged(webview);
 }
 
 std::string GetWebKitVersion() {
   return StringPrintf("%d.%d", WEBKIT_VERSION_MAJOR, WEBKIT_VERSION_MINOR);
 }
 
 namespace {
 
-std::string* user_agent = NULL;
+const std::string* user_agent = NULL;
 bool user_agent_requested = false;
+bool user_agent_is_overridden = false;
 
-void SetUserAgentToDefault() {
-  static std::string default_user_agent;
+void BuildUserAgent(bool mimic_safari, std::string* result) {
 #if defined(OS_WIN) || defined(OS_MACOSX)
   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;
 #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));
 #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;
 
-  scoped_ptr<FileVersionInfo> version_info(
-      FileVersionInfo::CreateFileVersionInfoForCurrentModule());
-  if (version_info.get())
-    product = "Chrome/" + WideToASCII(version_info->product_version());
+  if (!mimic_safari) {
+    scoped_ptr<FileVersionInfo> version_info(
+        FileVersionInfo::CreateFileVersionInfoForCurrentModule());
+    if (version_info.get())
+      product = "Chrome/" + WideToASCII(version_info->product_version());
+  }
 
   if (product.empty())
-    product = "Version/3.1";
+    product = "Version/3.2.1";

[wtc, 2009/01/27 19:19:24]

Nit: We should move the hardcoded Safari version 3.2.1 to a
more discoverable header file.

[darin (slow to review), 2009/01/27 19:52:06]

Agreed, but I'll defer that to a subsequent patch.

 
   // Derived from Safari's UA string.
   StringAppendF(
-      &default_user_agent,
+      result,
 #if defined(OS_WIN)
       "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"
 #endif
       " (KHTML, like Gecko) %s Safari/%d.%d",
       os_major_version,
       os_minor_version,
 #if defined(OS_MACOSX)
       os_bugfix_version,
 #endif
       WEBKIT_VERSION_MAJOR,
       WEBKIT_VERSION_MINOR,
       product.c_str(),
       WEBKIT_VERSION_MAJOR,
       WEBKIT_VERSION_MINOR
       );
 #elif defined(OS_LINUX)
   // TODO(agl): We don't have version information embedded in files under Linux
   // so we use the following string which is based off the UA string for
   // Windows. Some solution for embedding the Chrome version number needs to be
   // found here.
   StringAppendF(
-      &default_user_agent,
+      result,
       "Mozilla/5.0 (Linux; U; en-US) AppleWebKit/525.13 "
       "(KHTML, like Gecko) Chrome/0.2.149.27 Safari/525.13");
 #else
   // TODO(port): we need something like FileVersionInfo for our UA string.
   NOTIMPLEMENTED();
 #endif
+}
+
+void SetUserAgentToDefault() {
+  static std::string default_user_agent;
+  BuildUserAgent(false, &default_user_agent);
   user_agent = &default_user_agent;
 }
 
-};
+}  // namespace
 
 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_is_overridden = true;
   user_agent = &overridden_user_agent;
 }
 
-const std::string& GetUserAgent() {
+const std::string& GetUserAgent(const GURL& url) {
   if (!user_agent)
     SetUserAgentToDefault();
   user_agent_requested = true;
+  if (!user_agent_is_overridden) {
+    static std::string mimic_safari_user_agent;

[wtc, 2009/01/27 19:19:24]

Because you're using a static object here, please make
sure this method is only called by a single thread.

[darin (slow to review), 2009/01/27 19:52:06]

hmm... good point.  i believe it is.

+        // For hotmail, we need to spoof as Safari (bug 4111).

[wtc, 2009/01/27 19:19:24]

Nit: is this comment over-indented?

[darin (slow to review), 2009/01/27 19:52:06]

oops.. a <TAB> slipped in there.

+    if (MatchPattern(url.host(), "*.mail.live.com")) {
+      if (mimic_safari_user_agent.empty()) 
+        BuildUserAgent(true, &mimic_safari_user_agent);
+      return mimic_safari_user_agent;
+    }
+  }
   return *user_agent;
 }
 
 void NotifyJSOutOfMemory(WebCore::Frame* frame) {
   if (!frame)
     return;
 
   // Dispatch to the delegate of the view that owns the frame.
   WebFrame* webframe = WebFrameImpl::FromFrame(frame);
   WebView* webview = webframe->GetView();
@@ skipping 12 matching lines @@
   }
 
   g_forcefully_terminate_plugin_process = value;
 }
 
 bool ShouldForcefullyTerminatePluginProcess() {
   return g_forcefully_terminate_plugin_process;
 }
 
 } // namespace webkit_glue