mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-20 16:55:40 +00:00
3f2298c2f6
MozReview-Commit-ID: 666LNaHyiuQ --HG-- extra : rebase_source : f3a24e0342d8dcdf9020dd473bc6b1045cbc533b
204 lines
6.1 KiB
C++
204 lines
6.1 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set expandtab ts=2 sw=2 sts=2 cin: */
|
|
/* 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 InterceptedChannel_h
|
|
#define InterceptedChannel_h
|
|
|
|
#include "nsINetworkInterceptController.h"
|
|
#include "mozilla/RefPtr.h"
|
|
#include "mozilla/Maybe.h"
|
|
|
|
class nsICacheEntry;
|
|
class nsInputStreamPump;
|
|
class nsIStreamListener;
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
class nsHttpChannel;
|
|
class HttpChannelChild;
|
|
class nsHttpResponseHead;
|
|
class InterceptStreamListener;
|
|
|
|
// An object representing a channel that has been intercepted. This avoids complicating
|
|
// the actual channel implementation with the details of synthesizing responses.
|
|
class InterceptedChannelBase : public nsIInterceptedChannel {
|
|
protected:
|
|
// The interception controller to notify about the successful channel interception
|
|
nsCOMPtr<nsINetworkInterceptController> mController;
|
|
|
|
// Response head for use when synthesizing
|
|
Maybe<nsAutoPtr<nsHttpResponseHead>> mSynthesizedResponseHead;
|
|
|
|
nsCOMPtr<nsIConsoleReportCollector> mReportCollector;
|
|
nsCOMPtr<nsISupports> mReleaseHandle;
|
|
|
|
bool mClosed;
|
|
|
|
void EnsureSynthesizedResponse();
|
|
void DoNotifyController();
|
|
MOZ_MUST_USE nsresult DoSynthesizeStatus(uint16_t aStatus,
|
|
const nsACString& aReason);
|
|
MOZ_MUST_USE nsresult DoSynthesizeHeader(const nsACString& aName,
|
|
const nsACString& aValue);
|
|
|
|
TimeStamp mLaunchServiceWorkerStart;
|
|
TimeStamp mLaunchServiceWorkerEnd;
|
|
TimeStamp mDispatchFetchEventStart;
|
|
TimeStamp mDispatchFetchEventEnd;
|
|
TimeStamp mHandleFetchEventStart;
|
|
TimeStamp mHandleFetchEventEnd;
|
|
|
|
TimeStamp mFinishResponseStart;
|
|
TimeStamp mFinishResponseEnd;
|
|
enum {
|
|
Invalid = 0,
|
|
Synthesized,
|
|
Reset
|
|
} mSynthesizedOrReset;
|
|
|
|
virtual ~InterceptedChannelBase() = default;
|
|
public:
|
|
explicit InterceptedChannelBase(nsINetworkInterceptController* aController);
|
|
|
|
// Notify the interception controller that the channel has been intercepted
|
|
// and prepare the response body output stream.
|
|
virtual void NotifyController() = 0;
|
|
|
|
NS_DECL_ISUPPORTS
|
|
|
|
NS_IMETHOD GetConsoleReportCollector(nsIConsoleReportCollector** aCollectorOut) override;
|
|
NS_IMETHOD SetReleaseHandle(nsISupports* aHandle) override;
|
|
|
|
NS_IMETHODIMP
|
|
SetLaunchServiceWorkerStart(TimeStamp aTimeStamp) override
|
|
{
|
|
mLaunchServiceWorkerStart = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
GetLaunchServiceWorkerStart(TimeStamp* aTimeStamp) override
|
|
{
|
|
MOZ_DIAGNOSTIC_ASSERT(aTimeStamp);
|
|
*aTimeStamp = mLaunchServiceWorkerStart;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetLaunchServiceWorkerEnd(TimeStamp aTimeStamp) override
|
|
{
|
|
mLaunchServiceWorkerEnd = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
GetLaunchServiceWorkerEnd(TimeStamp* aTimeStamp) override
|
|
{
|
|
MOZ_DIAGNOSTIC_ASSERT(aTimeStamp);
|
|
*aTimeStamp = mLaunchServiceWorkerEnd;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetDispatchFetchEventStart(TimeStamp aTimeStamp) override
|
|
{
|
|
mDispatchFetchEventStart = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetDispatchFetchEventEnd(TimeStamp aTimeStamp) override
|
|
{
|
|
mDispatchFetchEventEnd = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetHandleFetchEventStart(TimeStamp aTimeStamp) override
|
|
{
|
|
mHandleFetchEventStart = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetHandleFetchEventEnd(TimeStamp aTimeStamp) override
|
|
{
|
|
mHandleFetchEventEnd = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetFinishResponseStart(TimeStamp aTimeStamp) override
|
|
{
|
|
mFinishResponseStart = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetFinishSynthesizedResponseEnd(TimeStamp aTimeStamp) override
|
|
{
|
|
MOZ_ASSERT(mSynthesizedOrReset == Invalid);
|
|
mSynthesizedOrReset = Synthesized;
|
|
mFinishResponseEnd = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
SetChannelResetEnd(TimeStamp aTimeStamp) override
|
|
{
|
|
MOZ_ASSERT(mSynthesizedOrReset == Invalid);
|
|
mSynthesizedOrReset = Reset;
|
|
mFinishResponseEnd = aTimeStamp;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP SaveTimeStamps() override;
|
|
|
|
static already_AddRefed<nsIURI>
|
|
SecureUpgradeChannelURI(nsIChannel* aChannel);
|
|
};
|
|
|
|
class InterceptedChannelContent : public InterceptedChannelBase
|
|
{
|
|
// The actual channel being intercepted.
|
|
RefPtr<HttpChannelChild> mChannel;
|
|
|
|
// Listener for the synthesized response to fix up the notifications before they reach
|
|
// the actual channel.
|
|
RefPtr<InterceptStreamListener> mStreamListener;
|
|
|
|
// Set for intercepted channels that have gone through a secure upgrade.
|
|
bool mSecureUpgrade;
|
|
public:
|
|
InterceptedChannelContent(HttpChannelChild* aChannel,
|
|
nsINetworkInterceptController* aController,
|
|
InterceptStreamListener* aListener,
|
|
bool aSecureUpgrade);
|
|
|
|
NS_IMETHOD ResetInterception() override;
|
|
NS_IMETHOD StartSynthesizedResponse(nsIInputStream* aBody,
|
|
nsIInterceptedBodyCallback* aBodyCallback,
|
|
nsICacheInfoChannel* aChannel,
|
|
const nsACString& aFinalURLSpec,
|
|
bool aResponseRedirected) override;
|
|
NS_IMETHOD FinishSynthesizedResponse() override;
|
|
NS_IMETHOD GetChannel(nsIChannel** aChannel) override;
|
|
NS_IMETHOD GetSecureUpgradedChannelURI(nsIURI** aURI) override;
|
|
NS_IMETHOD SynthesizeStatus(uint16_t aStatus, const nsACString& aReason) override;
|
|
NS_IMETHOD SynthesizeHeader(const nsACString& aName, const nsACString& aValue) override;
|
|
NS_IMETHOD CancelInterception(nsresult aStatus) override;
|
|
NS_IMETHOD SetChannelInfo(mozilla::dom::ChannelInfo* aChannelInfo) override;
|
|
NS_IMETHOD GetInternalContentPolicyType(nsContentPolicyType *aInternalContentPolicyType) override;
|
|
|
|
virtual void NotifyController() override;
|
|
};
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|
|
|
|
#endif // InterceptedChannel_h
|