1 files changed, 9 insertions, 5 deletions
diff --git a/Source/WebKit2/UIProcess/WebFrameListenerProxy.h b/Source/WebKit2/UIProcess/WebFrameListenerProxy.h
index 8b3ee3653..70380d6d1 100644
--- a/Source/WebKit2/UIProcess/WebFrameListenerProxy.h
+++ b/Source/WebKit2/UIProcess/WebFrameListenerProxy.h
@@ -23,17 +23,19 @@
  * THE POSSIBILITY OF SUCH DAMAGE.
  */
 
-#ifndef WebFrameListenerProxy_h
-#define WebFrameListenerProxy_h
+#pragma once
 
+#include "APINavigation.h"
 #include "APIObject.h"
 #include <WebCore/FrameLoaderTypes.h>
 #include <wtf/PassRefPtr.h>
+#include <wtf/Ref.h>
 #include <wtf/RefPtr.h>
 
 namespace WebKit {
 
 class WebFrameProxy;
+struct WebsitePolicies;
 
 class WebFrameListenerProxy : public API::Object {
 public:
@@ -42,16 +44,18 @@ public:
     void invalidate();
     uint64_t listenerID() const { return m_listenerID; }
 
+    API::Navigation* navigation() { return m_navigation.get(); }
+    void setNavigation(Ref<API::Navigation>&& navigation) { m_navigation = WTFMove(navigation); }
+
 protected:
     WebFrameListenerProxy(WebFrameProxy*, uint64_t listenerID);
 
-    void receivedPolicyDecision(WebCore::PolicyAction);
+    void receivedPolicyDecision(WebCore::PolicyAction, const WebsitePolicies&);
 
 private:
     RefPtr<WebFrameProxy> m_frame;
     uint64_t m_listenerID;
+    RefPtr<API::Navigation> m_navigation;
 };
 
 } // namespace WebKit
-
-#endif // WebFrameListenerProxy_h