Deal with frame removal by content scripts

extensions/renderer/extension_frame_helper.h

Index: extensions/renderer/extension_frame_helper.h
diff --git a/extensions/renderer/extension_frame_helper.h b/extensions/renderer/extension_frame_helper.h
index a710d4f970517610fe40be239176e685b9cf89ab..0189b11d8caaebe57d61afc195a8a6c9a548c402 100644
--- a/extensions/renderer/extension_frame_helper.h
+++ b/extensions/renderer/extension_frame_helper.h
@@ -8,6 +8,7 @@
 #include <vector>
 
 #include "base/macros.h"
+#include "base/memory/weak_ptr.h"
 #include "content/public/common/console_message_level.h"
 #include "content/public/renderer/render_frame_observer.h"
 #include "content/public/renderer/render_frame_observer_tracker.h"
@@ -61,6 +62,18 @@ class ExtensionFrameHelper
     return did_create_current_document_element_;
   }
 
+  // Called when the document element has been inserted. This method may invoke
+  // untrusted JavaScript code and invalidate the frame.
+  void AfterDidCreateDocumentElement(
+      const base::WeakPtr<content::RenderFrame>& render_frame_weak);
+
+  // Schedule a callback, to be run at the next AfterDidCreateDocumentElement
+  // notification. Only call this when you are certain that there will be such a
+  // notification, e.g. from RenderFrameObserver::DidCreateDocumentElement.
+  // Otherwise the callback is never invoked, or invoked for a document that you
+  // were not expecting.
+  void ScheduleAfterDidCreateDocumentElement(const base::Closure& callback);
+
  private:
   // RenderFrameObserver implementation.
   void DidCreateDocumentElement() override;
@@ -113,6 +126,9 @@ class ExtensionFrameHelper
   // Whether or not the current document element has been created.
   bool did_create_current_document_element_;
 
+  // Callbacks to be run at the next AfterDidCreateDocumentElement notification.
+  std::vector<base::Closure> document_element_created_callbacks_;
+
   DISALLOW_COPY_AND_ASSIGN(ExtensionFrameHelper);
 };