mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-04 02:57:38 +00:00
d8bf3acf6e
Similar to how DocumentLoadListener redirects the document to the content process. This fixes the race conditions in early hint architecture fixing intermittent early hint test cases. See also: https://bugzilla.mozilla.org/show_bug.cgi?id=1753730#c11 Connecting back from the Preloader in the content process to the EarlyHintPreloader in parent process is done in previous patches. Differential Revision: https://phabricator.services.mozilla.com/D149643
71 lines
2.3 KiB
C++
71 lines
2.3 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set sw=2 ts=8 et tw=80 : */
|
|
|
|
/* 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/. */
|
|
|
|
#include "NeckoCommon.h"
|
|
|
|
#include "nsIInputStream.h"
|
|
#include "nsIMultiPartChannel.h"
|
|
#include "nsIParentChannel.h"
|
|
#include "nsStringStream.h"
|
|
|
|
namespace mozilla::net {
|
|
|
|
nsresult ForwardStreamListenerFunctions(
|
|
nsTArray<StreamListenerFunction>& aCalls, nsIStreamListener* aParent) {
|
|
nsresult rv = NS_OK;
|
|
for (auto& variant : aCalls) {
|
|
variant.match(
|
|
[&](const OnStartRequestParams& aParams) {
|
|
rv = aParent->OnStartRequest(aParams.request);
|
|
if (NS_FAILED(rv)) {
|
|
aParams.request->Cancel(rv);
|
|
}
|
|
},
|
|
[&](const OnDataAvailableParams& aParams) {
|
|
// Don't deliver OnDataAvailable if we've
|
|
// already failed.
|
|
if (NS_FAILED(rv)) {
|
|
return;
|
|
}
|
|
nsCOMPtr<nsIInputStream> stringStream;
|
|
rv = NS_NewByteInputStream(
|
|
getter_AddRefs(stringStream),
|
|
Span<const char>(aParams.data.get(), aParams.count),
|
|
NS_ASSIGNMENT_DEPEND);
|
|
if (NS_SUCCEEDED(rv)) {
|
|
rv = aParent->OnDataAvailable(aParams.request, stringStream,
|
|
aParams.offset, aParams.count);
|
|
}
|
|
if (NS_FAILED(rv)) {
|
|
aParams.request->Cancel(rv);
|
|
}
|
|
},
|
|
[&](const OnStopRequestParams& aParams) {
|
|
if (NS_SUCCEEDED(rv)) {
|
|
aParent->OnStopRequest(aParams.request, aParams.status);
|
|
} else {
|
|
aParent->OnStopRequest(aParams.request, rv);
|
|
}
|
|
rv = NS_OK;
|
|
},
|
|
[&](const OnAfterLastPartParams& aParams) {
|
|
nsCOMPtr<nsIMultiPartChannelListener> multiListener =
|
|
do_QueryInterface(aParent);
|
|
if (multiListener) {
|
|
if (NS_SUCCEEDED(rv)) {
|
|
multiListener->OnAfterLastPart(aParams.status);
|
|
} else {
|
|
multiListener->OnAfterLastPart(rv);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
} // namespace mozilla::net
|