Refactor host network

chrome_frame/chrome_frame_activex_base.h

 // 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.
 
 #ifndef CHROME_FRAME_CHROME_FRAME_ACTIVEX_BASE_H_
 #define CHROME_FRAME_CHROME_FRAME_ACTIVEX_BASE_H_
 
 #include <atlbase.h>
 #include <atlcom.h>
 #include <atlctl.h>
@@ skipping 147 matching lines @@
   public IPropertyNotifySinkCP<T>,
   public CComCoClass<T, &class_id>,
   public CComControl<T>,
   public ChromeFramePlugin<T> {
  protected:
   typedef std::set<ScopedComPtr<IDispatch> > EventHandlers;
   typedef ChromeFrameActivexBase<T, class_id> Base;
 
  public:
   ChromeFrameActivexBase()
-      : ready_state_(READYSTATE_UNINITIALIZED),
-        worker_thread_("ChromeFrameWorker_Thread") {
+      : ready_state_(READYSTATE_UNINITIALIZED) {
     m_bWindowOnly = TRUE;
   }
 
   ~ChromeFrameActivexBase() {
   }
 
 DECLARE_OLEMISC_STATUS(OLEMISC_RECOMPOSEONRESIZE | OLEMISC_CANTLINKINSIDE |
                        OLEMISC_INSIDEOUT | OLEMISC_ACTIVATEWHENVISIBLE |
                        OLEMISC_SETCLIENTSITEFIRST)
 
@@ skipping 278 matching lines @@
       referrer_header += UTF8ToWide(referrer.spec());
       referrer_header += L"\r\n\r\n";
       http_headers.Set(referrer_header.c_str());
     }
 
     web_browser2->Navigate2(url.AsInput(), &flags, &empty, &empty,
                             http_headers.AsInput());
     web_browser2->put_Visible(VARIANT_TRUE);
   }
 
-  virtual void OnRequestStart(int tab_handle, int request_id,
-                              const IPC::AutomationURLRequest& request_info) {
-    // The worker thread may have been stopped. This could happen if the
-    // ActiveX instance was reused.
-    if (!worker_thread_.message_loop()) {
-      base::Thread::Options options;
-      options.message_loop_type = MessageLoop::TYPE_UI;
-      worker_thread_.StartWithOptions(options);
-      worker_thread_.message_loop()->PostTask(
-          FROM_HERE, NewRunnableMethod(this, &Base::OnWorkerStart));
-    }
-
-    scoped_refptr<CComObject<UrlmonUrlRequest> > request;
-    if (base_url_request_.get() &&
-        GURL(base_url_request_->url()) == GURL(request_info.url)) {
-      request.swap(base_url_request_);
-    } else {
-      CComObject<UrlmonUrlRequest>* new_request = NULL;
-      CComObject<UrlmonUrlRequest>::CreateInstance(&new_request);
-      request = new_request;
-    }
-
-    DCHECK(request.get() != NULL);
-
-    if (request->Initialize(
-        automation_client_.get(), tab_handle, request_id, request_info.url,
-        request_info.method, request_info.referrer,
-        request_info.extra_request_headers, request_info.upload_data.get(),
-        static_cast<T*>(this)->is_frame_busting_enabled())) {
-      request->set_worker_thread(&worker_thread_);
-      // If Start is successful, it will add a self reference.
-      request->Start();
-      request->set_parent_window(m_hWnd);
-    }
-  }
-
-  virtual void OnRequestRead(int tab_handle, int request_id,
-                             int bytes_to_read) {
-    automation_client_->ReadRequest(request_id, bytes_to_read);
-  }
-
-  virtual void OnRequestEnd(int tab_handle, int request_id,
-                            const URLRequestStatus& status) {
-    automation_client_->RemoveRequest(request_id, true);
-  }
-
   virtual void OnDownloadRequestInHost(int tab_handle, int request_id) {
     DLOG(INFO) << "TODO: Let the host browser handle this download";
-    PluginUrlRequest* request = automation_client_->LookupRequest(request_id);
-    if (request) {
-      static_cast<UrlmonUrlRequest*>(request)->TransferToHost(doc_site_);
+    ScopedComPtr<IBindCtx> bind_context;
+    ScopedComPtr<IMoniker> moniker;
+    url_fetcher_.StealMonikerFromRequest(request_id, moniker.Receive());
+    if (moniker) {
+      ::CreateBindCtx(0, bind_context.Receive());
+      DCHECK(bind_context);
+      NavigateBrowserToMoniker(doc_site_, moniker, NULL, bind_context, NULL);
     }
-    automation_client_->RemoveRequest(request_id, false);
   }
 
   virtual void OnSetCookieAsync(int tab_handle, const GURL& url,
                                 const std::string& cookie) {
     std::string name;
     std::string data;
 
     size_t name_end = cookie.find('=');
     if (std::string::npos != name_end) {
       net::CookieMonster::ParsedCookie parsed_cookie = cookie;
@@ skipping 29 matching lines @@
     // Only fire the 'interactive' ready state if we aren't there already.
     if (ready_state_ < READYSTATE_INTERACTIVE) {
       ready_state_ = READYSTATE_INTERACTIVE;
       FireOnChanged(DISPID_READYSTATE);
     }
     return 0;
   }
 
   LRESULT OnDestroy(UINT message, WPARAM wparam, LPARAM lparam,
                     BOOL& handled) {  // NO_LINT
-    if (worker_thread_.message_loop()) {
-      if (automation_client_.get())
-        automation_client_->CleanupRequests();
-
-      worker_thread_.message_loop()->PostTask(
-          FROM_HERE, NewRunnableMethod(this, &Base::OnWorkerStop));
-
-      MSG msg = {0};
-      while (GetMessage(&msg, NULL, WM_USER,
-                        WM_WORKER_THREAD_UNINITIALIZED_MSG)) {
-        if (msg.hwnd == m_hWnd &&
-            msg.message == WM_WORKER_THREAD_UNINITIALIZED_MSG) {
-          break;
-        }
-
-        TranslateMessage(&msg);
-        DispatchMessage(&msg);
-      }
-      worker_thread_.Stop();
-    }
+    DLOG(INFO) << __FUNCTION__;
     return 0;
   }
 
   // ChromeFrameDelegate override
   virtual void OnAutomationServerReady() {
     ChromeFramePlugin<T>::OnAutomationServerReady();
 
     ready_state_ = READYSTATE_COMPLETE;
     FireOnChanged(DISPID_READYSTATE);
   }
@@ skipping 502 matching lines @@
     // should have first dibs anyway.
     if (hr != S_OK && automation_client_.get()) {
       TabProxy* tab = automation_client_->tab();
       if (tab) {
         tab->ProcessUnhandledAccelerator(accel_message);
       }
     }
   }
 
  protected:
-  // The following functions are called to initialize and uninitialize the
-  // worker thread.
-  void OnWorkerStart() {
-    CoInitialize(NULL);
-  }
-
-  void OnWorkerStop() {
-    CoUninitialize();
-    PostMessage(WM_WORKER_THREAD_UNINITIALIZED_MSG, 0, 0);
-  }
-
   ScopedBstr url_;
   ScopedComPtr<IOleDocumentSite> doc_site_;
 
   // For more information on the ready_state_ property see:
   // http://msdn.microsoft.com/en-us/library/aa768179(VS.85).aspx#
   READYSTATE ready_state_;
 
   // The following members contain IDispatch interfaces representing the
   // onload/onerror/onmessage handlers on the page.
   ScopedVariant onload_handler_;
   ScopedVariant onerror_handler_;
   ScopedVariant onmessage_handler_;
 
   EventHandlers onmessage_;
   EventHandlers onloaderror_;
   EventHandlers onload_;
   EventHandlers onreadystatechanged_;
   EventHandlers onprivatemessage_;
   EventHandlers onextensionready_;
 
-  // The UrlmonUrlRequest instance instantiated for downloading the base URL.
-  scoped_refptr<CComObject<UrlmonUrlRequest> > base_url_request_;
-
-  base::Thread worker_thread_;
+  // Handle network requests when host network stack is used. Passed to the
+  // automation client on initialization.
+  UrlmonUrlRequestManager url_fetcher_;
 };
 
 #endif  // CHROME_FRAME_CHROME_FRAME_ACTIVEX_BASE_H_