Inform the WebFrameClient of BeforeUnload/Unload handlers

Source/core/frame/LocalDOMWindow.cpp

 /*
  * Copyright (C) 2006, 2007, 2008, 2010 Apple Inc. All rights reserved.
  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 213 matching lines @@
     DEFINE_STATIC_LOCAL(DOMWindowSet, windowsWithBeforeUnloadEventListeners, ());
     return windowsWithBeforeUnloadEventListeners;
 }
 
 static void addUnloadEventListener(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithUnloadEventListeners();
     if (set.isEmpty())
         disableSuddenTermination();
     set.add(domWindow);
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            1, FrameLoaderClient::UnloadHandler);
+    }
 }
 
 static void removeUnloadEventListener(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithUnloadEventListeners();
     DOMWindowSet::iterator it = set.find(domWindow);
     if (it == set.end())
         return;
     set.remove(it);
     if (set.isEmpty())
         enableSuddenTermination();
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            -1, FrameLoaderClient::UnloadHandler);
+    }
 }
 
 static void removeAllUnloadEventListeners(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithUnloadEventListeners();
     DOMWindowSet::iterator it = set.find(domWindow);
     if (it == set.end())
         return;
+    int numHandlers = set.count(domWindow);
     set.removeAll(it);
     if (set.isEmpty())
         enableSuddenTermination();
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            -numHandlers, FrameLoaderClient::UnloadHandler);
+    }
 }
 
 static void addBeforeUnloadEventListener(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithBeforeUnloadEventListeners();
     if (set.isEmpty())
         disableSuddenTermination();
     set.add(domWindow);
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            1, FrameLoaderClient::BeforeUnloadHandler);
+    }
 }
 
 static void removeBeforeUnloadEventListener(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithBeforeUnloadEventListeners();
     DOMWindowSet::iterator it = set.find(domWindow);
     if (it == set.end())
         return;
     set.remove(it);
     if (set.isEmpty())
         enableSuddenTermination();
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            -1, FrameLoaderClient::BeforeUnloadHandler);
+    }
 }
 
 static void removeAllBeforeUnloadEventListeners(LocalDOMWindow* domWindow)
 {
     DOMWindowSet& set = windowsWithBeforeUnloadEventListeners();
     DOMWindowSet::iterator it = set.find(domWindow);
     if (it == set.end())
         return;
+    int numHandlers = set.count(domWindow);
     set.removeAll(it);
     if (set.isEmpty())
         enableSuddenTermination();
+    if (domWindow->frame()) {
+        domWindow->frame()->loader().client()->suddenTerminationDisablerChanged(
+            -numHandlers, FrameLoaderClient::BeforeUnloadHandler);
+    }
 }
 
 static bool allowsBeforeUnloadListeners(LocalDOMWindow* window)
 {
     ASSERT_ARG(window, window);
     LocalFrame* frame = window->frame();
     if (!frame)
         return false;
     return frame->isMainFrame();
 }
@@ skipping 1530 matching lines @@
     return m_frameObserver->frame();
 }
 
 v8::Handle<v8::Object> LocalDOMWindow::wrap(v8::Handle<v8::Object> creationContext, v8::Isolate* isolate)
 {
     ASSERT_NOT_REACHED(); // LocalDOMWindow has [Custom=ToV8].
     return v8::Handle<v8::Object>();
 }
 
 } // namespace blink