A new way of hooking internet protocols.

chrome_frame/bho.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/bho.h"
 
 #include <shlguid.h>
 
 #include "base/file_path.h"
 #include "base/logging.h"
@@ skipping 105 matching lines @@
   if (!browser_service || !CheckForCFNavigation(browser_service, false)) {
     // TODO(tommi): Remove? Isn't this done below by calling set_referrer("")?
     referrer_.clear();
   }
 
   VARIANT_BOOL is_top_level = VARIANT_FALSE;
   web_browser2->get_TopLevelContainer(&is_top_level);
   if (is_top_level) {
     set_url(url->bstrVal);
     set_referrer("");
-    // The moniker patch checks opt-in URLs in SniffData::DetermineRendererType.
-    if (!MonikerPatchEnabled()) {
+    if (IsIBrowserServicePatchEnabled()) {
       ProcessOptInUrls(web_browser2, url->bstrVal);
     }
   }
   return S_OK;
 }
 
 STDMETHODIMP_(void) Bho::NavigateComplete2(IDispatch* dispatch, VARIANT* url) {
   DLOG(INFO) << __FUNCTION__;
 }
 
@@ skipping 175 matching lines @@
   if (state_ == UNKNOWN) {
     // If we're going to start patching things for reals, we'd better make sure
     // that we stick around for ever more:
     if (!IsUnpinnedMode())
       PinModule();
 
     HttpNegotiatePatch::Initialize();
 
     ProtocolPatchMethod patch_method = GetPatchMethod();
     if (patch_method == PATCH_METHOD_INET_PROTOCOL) {
-      ProtocolSinkWrap::PatchProtocolHandlers();
+      g_trans_hooks.InstallHooks();
       state_ = PATCH_PROTOCOL;
     } else if (patch_method == PATCH_METHOD_IBROWSER) {
-        state_ =  PATCH_IBROWSER;
+      state_ =  PATCH_IBROWSER;
     } else {
       DCHECK(patch_method == PATCH_METHOD_MONIKER);
       state_ = PATCH_MONIKER;
       MonikerPatch::Initialize();
     }
 
     ret = true;
   }
 
   _pAtlModule->m_csStaticDataInitAndTypeInfo.Unlock();
 
   return ret;
 }
 
 void PatchHelper::PatchBrowserService(IBrowserService* browser_service) {
   DCHECK(state_ == PATCH_IBROWSER);
   if (!IS_PATCHED(IBrowserService)) {
     vtable_patch::PatchInterfaceMethods(browser_service,
                                         IBrowserService_PatchInfo);
   }
 }
 
 void PatchHelper::UnpatchIfNeeded() {
   if (state_ == PATCH_PROTOCOL) {
-    ProtocolSinkWrap::UnpatchProtocolHandlers();
+    g_trans_hooks.RevertHooks();
   } else if (state_ == PATCH_IBROWSER) {
     vtable_patch::UnpatchInterfaceMethods(IBrowserService_PatchInfo);
     MonikerPatch::Uninitialize();
   }
 
   HttpNegotiatePatch::Uninitialize();
 
   state_ = UNKNOWN;
 }