Handle automation server crashes.

chrome_frame/chrome_frame_activex_base.h

 // Copyright (c) 2010 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 157 matching lines @@
   public CComCoClass<T, &class_id>,
   public CComControl<T>,
   public ChromeFramePlugin<T> {
  protected:
   typedef std::set<ScopedComPtr<IDispatch> > EventHandlers;
   typedef ChromeFrameActivexBase<T, class_id> BasePlugin;
 
  public:
   ChromeFrameActivexBase()
       : ready_state_(READYSTATE_UNINITIALIZED),
-        failed_to_fetch_in_place_frame_(false) {
+      url_fetcher_(new UrlmonUrlRequestManager()),
+      failed_to_fetch_in_place_frame_(false),
+      draw_sad_tab_(false) {
     m_bWindowOnly = TRUE;
-    url_fetcher_.set_container(static_cast<IDispatch*>(this));
+    url_fetcher_->set_container(static_cast<IDispatch*>(this));
   }
 
   ~ChromeFrameActivexBase() {
-    url_fetcher_.set_container(NULL);
+    url_fetcher_->set_container(NULL);
   }
 
 DECLARE_OLEMISC_STATUS(OLEMISC_RECOMPOSEONRESIZE | OLEMISC_CANTLINKINSIDE |
                        OLEMISC_INSIDEOUT | OLEMISC_ACTIVATEWHENVISIBLE |
                        OLEMISC_SETCLIENTSITEFIRST)
 
 DECLARE_NOT_AGGREGATABLE(T)
 
 BEGIN_COM_MAP(ChromeFrameActivexBase)
   COM_INTERFACE_ENTRY(IChromeFrame)
   COM_INTERFACE_ENTRY(IDispatch)
   COM_INTERFACE_ENTRY(IViewObjectEx)
   COM_INTERFACE_ENTRY(IViewObject2)
   COM_INTERFACE_ENTRY(IViewObject)
   COM_INTERFACE_ENTRY(IOleInPlaceObjectWindowless)
   COM_INTERFACE_ENTRY(IOleInPlaceObject)
   COM_INTERFACE_ENTRY2(IOleWindow, IOleInPlaceObjectWindowless)
   COM_INTERFACE_ENTRY(IOleInPlaceActiveObject)
   COM_INTERFACE_ENTRY(IOleControl)
   COM_INTERFACE_ENTRY(IOleObject)
   COM_INTERFACE_ENTRY(ISupportErrorInfo)
   COM_INTERFACE_ENTRY(IQuickActivate)
   COM_INTERFACE_ENTRY(IProvideClassInfo)
   COM_INTERFACE_ENTRY(IProvideClassInfo2)
+  COM_INTERFACE_ENTRY(IConnectionPointContainer)
   COM_INTERFACE_ENTRY_FUNC_BLIND(0, InterfaceNotSupported)
 END_COM_MAP()
 
 BEGIN_CONNECTION_POINT_MAP(T)
   CONNECTION_POINT_ENTRY(IID_IPropertyNotifySink)
   CONNECTION_POINT_ENTRY(DIID_DIChromeFrameEvents)
 END_CONNECTION_POINT_MAP()
 
 BEGIN_MSG_MAP(ChromeFrameActivexBase)
   MESSAGE_HANDLER(WM_CREATE, OnCreate)
@@ skipping 44 matching lines @@
                                               IE_8,
                                               IE_8 + 1);
     }
     return S_OK;
   }
 
   void FinalRelease() {
     Uninitialize();
   }
 
+  void ResetUrlRequestManager() {
+    url_fetcher_.reset(new UrlmonUrlRequestManager());
+  }
+
   static HRESULT WINAPI InterfaceNotSupported(void* pv, REFIID riid, void** ppv,
                                               DWORD dw) {
 #ifndef NDEBUG
     wchar_t buffer[64] = {0};
     ::StringFromGUID2(riid, buffer, arraysize(buffer));
     DLOG(INFO) << "E_NOINTERFACE: " << buffer;
 #endif
     return E_NOINTERFACE;
   }
 
@@ skipping 10 matching lines @@
     }
     return S_FALSE;
   }
 
   // Called to draw our control when chrome hasn't been initialized.
   virtual HRESULT OnDraw(ATL_DRAWINFO& draw_info) {  // NOLINT
     if (NULL == draw_info.prcBounds) {
       NOTREACHED();
       return E_FAIL;
     }
-    // Don't draw anything.
+
+    if (draw_sad_tab_) {
+      // TODO(tommi): Draw a proper sad tab.
+      RECT rc = {0};
+      if (draw_info.prcBounds) {
+        rc.top = draw_info.prcBounds->top;
+        rc.bottom = draw_info.prcBounds->bottom;
+        rc.left = draw_info.prcBounds->left;
+        rc.right = draw_info.prcBounds->right;
+      } else {
+        GetClientRect(&rc);
+      }
+      ::DrawTextA(draw_info.hdcDraw, ":-(", -1, &rc,
+          DT_CENTER | DT_VCENTER | DT_SINGLELINE);
+    } else {
+      // Don't draw anything.
+    }
     return S_OK;
   }
 
 
   // Used to setup the document_url_ member needed for completing navigation.
   // Create external tab (possibly in incognito mode).
   HRESULT IOleObject_SetClientSite(IOleClientSite* client_site) {
     // If we currently have a document site pointer, release it.
     doc_site_.Release();
     if (client_site) {
@@ skipping 191 matching lines @@
                                    int align_flags,
                                    const IPC::ContextMenuParams& params) {
     scoped_refptr<BasePlugin> ref(this);
     ChromeFramePlugin<T>::OnHandleContextMenu(tab_handle, menu_handle,
                                               align_flags, params);
   }
 
   LRESULT OnCreate(UINT message, WPARAM wparam, LPARAM lparam,
                    BOOL& handled) {  // NO_LINT
     ModifyStyle(0, WS_CLIPCHILDREN | WS_CLIPSIBLINGS, 0);
-    url_fetcher_.put_notification_window(m_hWnd);
+    url_fetcher_->put_notification_window(m_hWnd);
     if (automation_client_.get()) {
       automation_client_->SetParentWindow(m_hWnd);
     } else {
       NOTREACHED() << "No automation server";
       return -1;
     }
     // 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
     DLOG(INFO) << __FUNCTION__;
     return 0;
   }
 
   // ChromeFrameDelegate override
   virtual void OnAutomationServerReady() {
+    draw_sad_tab_ = false;
     ChromeFramePlugin<T>::OnAutomationServerReady();
 
     ready_state_ = READYSTATE_COMPLETE;
     FireOnChanged(DISPID_READYSTATE);
   }
 
   // ChromeFrameDelegate override
   virtual void OnAutomationServerLaunchFailed(
       AutomationLaunchResult reason, const std::string& server_version) {
+    DLOG(INFO) << __FUNCTION__;
+    if (reason == AUTOMATION_SERVER_CRASHED)
+      draw_sad_tab_ = true;
+
     ready_state_ = READYSTATE_UNINITIALIZED;
     FireOnChanged(DISPID_READYSTATE);
   }
 
   // Overridden to take advantage of readystate prop changes and send those
   // to potential listeners.
   HRESULT FireOnChanged(DISPID dispid) {
     if (dispid == DISPID_READYSTATE) {
       Fire_onreadystatechanged(ready_state_);
     }
@@ skipping 637 matching lines @@
     web_browser2->Navigate2(url.AsInput(), &flags, &empty, &empty,
                             http_headers.AsInput());
   }
 
   ScopedBstr url_;
   ScopedComPtr<IOleDocumentSite> doc_site_;
 
   // If false, we tried but failed to fetch an IOleInPlaceFrame from our host.
   // Cached here so we don't try to fetch it every time if we keep failing.
   bool failed_to_fetch_in_place_frame_;
+  bool draw_sad_tab_;
+
   ScopedComPtr<IOleInPlaceFrame> in_place_frame_;
 
   // 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_;
 
   // Handle network requests when host network stack is used. Passed to the
   // automation client on initialization.
-  UrlmonUrlRequestManager url_fetcher_;
+  scoped_ptr<UrlmonUrlRequestManager> url_fetcher_;
 };
 
 #endif  // CHROME_FRAME_CHROME_FRAME_ACTIVEX_BASE_H_