diff -r 000000000000 -r 6474c204b198 uriloader/prefetch/nsPrefetchService.h
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/uriloader/prefetch/nsPrefetchService.h	Wed Dec 31 06:09:35 2014 +0100
@@ -0,0 +1,119 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef nsPrefetchService_h__
+#define nsPrefetchService_h__
+
+#include "nsCPrefetchService.h"
+#include "nsIObserver.h"
+#include "nsIInterfaceRequestor.h"
+#include "nsIChannelEventSink.h"
+#include "nsIRedirectResultListener.h"
+#include "nsIWebProgressListener.h"
+#include "nsIStreamListener.h"
+#include "nsIChannel.h"
+#include "nsIURI.h"
+#include "nsWeakReference.h"
+#include "nsCOMPtr.h"
+#include "nsAutoPtr.h"
+#include "mozilla/Attributes.h"
+
+class nsPrefetchService;
+class nsPrefetchListener;
+class nsPrefetchNode;
+
+//-----------------------------------------------------------------------------
+// nsPrefetchService
+//-----------------------------------------------------------------------------
+
+class nsPrefetchService MOZ_FINAL : public nsIPrefetchService
+                                  , public nsIWebProgressListener
+                                  , public nsIObserver
+                                  , public nsSupportsWeakReference
+{
+public:
+    NS_DECL_ISUPPORTS
+    NS_DECL_NSIPREFETCHSERVICE
+    NS_DECL_NSIWEBPROGRESSLISTENER
+    NS_DECL_NSIOBSERVER
+
+    nsPrefetchService();
+
+    nsresult Init();
+    void     ProcessNextURI();
+
+    nsPrefetchNode *GetCurrentNode() { return mCurrentNode.get(); }
+    nsPrefetchNode *GetQueueHead() { return mQueueHead; }
+
+    void NotifyLoadRequested(nsPrefetchNode *node);
+    void NotifyLoadCompleted(nsPrefetchNode *node);
+
+private:
+    ~nsPrefetchService();
+
+    nsresult Prefetch(nsIURI *aURI,
+                      nsIURI *aReferrerURI,
+                      nsIDOMNode *aSource,
+                      bool aExplicit);
+
+    void     AddProgressListener();
+    void     RemoveProgressListener();
+    nsresult EnqueueURI(nsIURI *aURI, nsIURI *aReferrerURI,
+                        nsIDOMNode *aSource, nsPrefetchNode **node);
+    nsresult EnqueueNode(nsPrefetchNode *node);
+    nsresult DequeueNode(nsPrefetchNode **node);
+    void     EmptyQueue();
+
+    void     StartPrefetching();
+    void     StopPrefetching();
+
+    nsPrefetchNode                   *mQueueHead;
+    nsPrefetchNode                   *mQueueTail;
+    nsRefPtr<nsPrefetchNode>          mCurrentNode;
+    int32_t                           mStopCount;
+    // true if pending document loads have ever reached zero.
+    int32_t                           mHaveProcessed;
+    bool                              mDisabled;
+};
+
+//-----------------------------------------------------------------------------
+// nsPrefetchNode
+//-----------------------------------------------------------------------------
+
+class nsPrefetchNode MOZ_FINAL : public nsIStreamListener
+                               , public nsIInterfaceRequestor
+                               , public nsIChannelEventSink
+                               , public nsIRedirectResultListener
+{
+public:
+    NS_DECL_ISUPPORTS
+    NS_DECL_NSIREQUESTOBSERVER
+    NS_DECL_NSISTREAMLISTENER
+    NS_DECL_NSIINTERFACEREQUESTOR
+    NS_DECL_NSICHANNELEVENTSINK
+    NS_DECL_NSIREDIRECTRESULTLISTENER
+
+    nsPrefetchNode(nsPrefetchService *aPrefetchService,
+                   nsIURI *aURI,
+                   nsIURI *aReferrerURI,
+                   nsIDOMNode *aSource);
+
+    ~nsPrefetchNode() {}
+
+    nsresult OpenChannel();
+    nsresult CancelChannel(nsresult error);
+
+    nsPrefetchNode             *mNext;
+    nsCOMPtr<nsIURI>            mURI;
+    nsCOMPtr<nsIURI>            mReferrerURI;
+    nsCOMPtr<nsIWeakReference>  mSource;
+
+private:
+    nsRefPtr<nsPrefetchService> mService;
+    nsCOMPtr<nsIChannel>        mChannel;
+    nsCOMPtr<nsIChannel>        mRedirectChannel;
+    int64_t                     mBytesRead;
+};
+
+#endif // !nsPrefetchService_h__