Chromeframe user-agent without need of HttpNegotiate patch.

chrome_frame/http_negotiate.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 "chrome_frame/http_negotiate.h"
 
 #include <atlbase.h>
 #include <atlcom.h>
 #include <htiframe.h>
 
@@ skipping 75 matching lines @@
   }
 
   STDMETHOD(OnDataAvailable)(DWORD flags, DWORD size, FORMATETC* formatetc,
     STGMEDIUM* storage) {
     return E_NOTIMPL;
   }
   STDMETHOD(OnObjectAvailable)(REFIID iid, IUnknown* object) {
     return E_NOTIMPL;
   }
 };
+}  // end namespace
 
-}  // end namespace
+std::string AppendCFUserAgentString(LPCWSTR headers,
+                                    LPCWSTR additional_headers) {
+  static const char kLowerCaseUserAgent[] = "user-agent";
+  using net::HttpUtil;
+
+  std::string ascii_headers;
+  if (additional_headers) {
+    ascii_headers = WideToASCII(additional_headers);
+  }
+
+  // Extract "User-Agent" from |additiona_headers| or |headers|.
+  HttpUtil::HeadersIterator headers_iterator(ascii_headers.begin(),
+                                             ascii_headers.end(), "\r\n");
+  std::string user_agent_value;
+  if (headers_iterator.AdvanceTo(kLowerCaseUserAgent)) {
+    user_agent_value = headers_iterator.values();
+  } else if (headers != NULL) {
+    // See if there's a user-agent header specified in the original headers.
+    std::string original_headers(WideToASCII(headers));
+    HttpUtil::HeadersIterator original_it(original_headers.begin(),
+      original_headers.end(), "\r\n");
+    if (original_it.AdvanceTo(kLowerCaseUserAgent))
+      user_agent_value = original_it.values();
+  }
+
+  // Use the default "User-Agent" if none was provided.
+  if (user_agent_value.empty())
+    user_agent_value = http_utils::GetDefaultUserAgent();
+
+  // Now add chromeframe to it.
+  user_agent_value = http_utils::AddChromeFrameToUserAgentValue(
+      user_agent_value);
+
+  // Build new headers, skip the existing user agent value from
+  // existing headers.
+  std::string new_headers;
+  headers_iterator.Reset();
+  while (headers_iterator.GetNext()) {
+    std::string name(headers_iterator.name());
+    if (!LowerCaseEqualsASCII(name, kLowerCaseUserAgent)) {
+      new_headers += name + ": " + headers_iterator.values() + "\r\n";
+    }
+  }
+
+  new_headers += "User-Agent: " + user_agent_value;
+  new_headers += "\r\n";
+  return new_headers;
+}
 
 HRESULT GetBrowserServiceFromProtocolSink(IInternetProtocolSink* sink,
                                           IBrowserService** browser_service) {
   DCHECK(browser_service);
   // When fetching a page for the first time (not cached), we can query the
   // sink directly for IID_IShellBrowser to get the browser service.
   HRESULT hr = DoQueryService(IID_IShellBrowser, sink, browser_service);
   if (FAILED(hr)) {
     // When the request is being served up from the cache, we have to take
     // a different route via IID_ITargetFrame2.
@@ skipping 58 matching lines @@
 
   if (http) {
     hr = vtable_patch::PatchInterfaceMethods(http, IHttpNegotiate_PatchInfo);
     DLOG_IF(ERROR, FAILED(hr))
         << StringPrintf("HttpNegotiate patch failed 0x%08X", hr);
   } else {
     DLOG(WARNING)
         << StringPrintf("IHttpNegotiate not supported 0x%08X", hr);
   }
 
-  // CTransaction patch supports sniffing HTTP headers, so no need of
-  // sniffing inside HttpNegotiatePatch::ReportProgress.
-  if (GetPatchMethod() != PATCH_METHOD_INET_PROTOCOL) {
-    ScopedComPtr<IBindStatusCallback> bscb;
-    hr = bscb.QueryFrom(to_patch);
+  ScopedComPtr<IBindStatusCallback> bscb;
+  hr = bscb.QueryFrom(to_patch);
 
-    if (bscb) {
-      hr = vtable_patch::PatchInterfaceMethods(bscb,
-                                               IBindStatusCallback_PatchInfo);
-      DLOG_IF(ERROR, FAILED(hr))
-          << StringPrintf("BindStatusCallback patch failed 0x%08X", hr);
-    } else {
-      DLOG(WARNING) << StringPrintf("IBindStatusCallback not supported 0x%08X",
-                                    hr);
-    }
+  if (bscb) {
+    hr = vtable_patch::PatchInterfaceMethods(bscb,
+                                             IBindStatusCallback_PatchInfo);
+    DLOG_IF(ERROR, FAILED(hr))
+        << StringPrintf("BindStatusCallback patch failed 0x%08X", hr);
+  } else {
+    DLOG(WARNING) << StringPrintf("IBindStatusCallback not supported 0x%08X",
+                                  hr);
   }
   return hr;
 }
 
 // static
 HRESULT HttpNegotiatePatch::BeginningTransaction(
     IHttpNegotiate_BeginningTransaction_Fn original, IHttpNegotiate* me,
     LPCWSTR url, LPCWSTR headers, DWORD reserved, LPWSTR* additional_headers) {
   DLOG(INFO) << __FUNCTION__ << " " << url << " headers:\n" << headers;
 
@@ skipping 25 matching lines @@
           mgr->set_referrer(referrer);
         }
       }
     } else {
       DLOG(INFO) << "No IWebBrowser2";
     }
   } else {
     DLOG(INFO) << "No NavigationManager";
   }
 
-  static const char kLowerCaseUserAgent[] = "user-agent";
-
-  using net::HttpUtil;
-
-  std::string ascii_headers;
-  if (*additional_headers) {
-    ascii_headers = WideToASCII(*additional_headers);
-    DLOG(INFO) << __FUNCTION__ << " additional headers: " << ascii_headers;
-  }
-
-  HttpUtil::HeadersIterator headers_iterator(ascii_headers.begin(),
-                                             ascii_headers.end(), "\r\n");
-  std::string user_agent_value;
-  if (headers_iterator.AdvanceTo(kLowerCaseUserAgent)) {
-    user_agent_value = headers_iterator.values();
-  } else if (headers != NULL) {
-    // See if there's a user-agent header specified in the original headers.
-    std::string original_headers(WideToASCII(headers));
-    HttpUtil::HeadersIterator original_it(original_headers.begin(),
-                                          original_headers.end(), "\r\n");
-    if (original_it.AdvanceTo(kLowerCaseUserAgent))
-      user_agent_value = original_it.values();
-  }
-
-  // Use the default one if none was provided.
-  if (user_agent_value.empty())
-    user_agent_value = http_utils::GetDefaultUserAgent();
-
-  // Now add chromeframe to it.
-  user_agent_value = http_utils::AddChromeFrameToUserAgentValue(
-      user_agent_value);
-
-  // Build new headers, skip the existing user agent value from
-  // existing headers.
-  std::string new_headers;
-  headers_iterator.Reset();
-  while (headers_iterator.GetNext()) {
-    std::string name(headers_iterator.name());
-    if (!LowerCaseEqualsASCII(name, kLowerCaseUserAgent)) {
-      new_headers += name + ": " + headers_iterator.values() + "\r\n";
-    }
-  }
-
-  new_headers += "User-Agent: " + user_agent_value;
-  new_headers += "\r\n\r\n";
-
-  if (*additional_headers)
-    ::CoTaskMemFree(*additional_headers);
-  *additional_headers = reinterpret_cast<wchar_t*>(::CoTaskMemAlloc(
-      (new_headers.length() + 1) * sizeof(wchar_t)));
-  lstrcpyW(*additional_headers, ASCIIToWide(new_headers).c_str());
-
-  return hr;
+  std::string updated = AppendCFUserAgentString(headers, *additional_headers);
+  *additional_headers = reinterpret_cast<wchar_t*>(::CoTaskMemRealloc(
+      *additional_headers, (updated.length() + 1) * sizeof(wchar_t)));
+  lstrcpyW(*additional_headers, ASCIIToWide(updated).c_str());
+  return S_OK;
 }
 
 // static
 HRESULT HttpNegotiatePatch::StartBinding(
     IBindStatusCallback_StartBinding_Fn original,
     IBindStatusCallback* me, DWORD reserved, IBinding* binding) {
   ScopedComPtr<IBinding> local_binding(binding);
   ScopedComPtr<IInternetProtocolSink> protocol_sink;
 
   HRESULT hr = protocol_sink.QueryFrom(local_binding);
@@ skipping 102 matching lines @@
     // Note that this avoids having the VEH report a crash if an SEH earlier in
     // the chain handles the exception.
     ExceptionBarrier barrier;
     return original(me, status_code, status_text);
   } else {
     // Only report exceptions caused within ChromeFrame in this context.
     ExceptionBarrierReportOnlyModule barrier;
     return original(me, status_code, status_text);
   }
 }
+
+STDMETHODIMP UserAgentAddOn::BeginningTransaction(LPCWSTR url, LPCWSTR headers,
+                                                  DWORD reserved,
+                                                  LPWSTR* additional_headers) {
+  HRESULT hr = S_OK;
+  if (delegate_) {
+    hr = delegate_->BeginningTransaction(url, headers, reserved,
+                                         additional_headers);
+  }
+
+  if (hr == S_OK) {
+    // Add "chromeframe" user-agent string.
+    std::string updated_headers = AppendCFUserAgentString(headers,
+                                                          *additional_headers);
+    *additional_headers = reinterpret_cast<wchar_t*>(::CoTaskMemRealloc(
+        *additional_headers, (updated_headers.length() + 1) * sizeof(wchar_t)));
+    lstrcpyW(*additional_headers, ASCIIToWide(updated_headers).c_str());
+  }
+  return hr;
+}
+
+STDMETHODIMP UserAgentAddOn::OnResponse(DWORD response_code,
+    LPCWSTR response_headers, LPCWSTR request_headers,
+    LPWSTR* additional_headers) {
+  HRESULT hr = S_OK;
+  if (delegate_) {
+    hr = delegate_->OnResponse(response_code, response_headers, request_headers,
+                               additional_headers);
+  }
+  return hr;
+}