Cancelling the loadgroup after we've run JS but before we've started pumping

data should still work.  Bug 389151, r+sr=biesi
This commit is contained in:
bzbarsky@mit.edu 2007-07-24 17:08:58 -07:00
parent 5412d34bc5
commit 6a6777c6e5

View File

@ -522,7 +522,12 @@ nsJSChannel::IsPending(PRBool *aResult)
NS_IMETHODIMP
nsJSChannel::GetStatus(nsresult *aResult)
{
if (NS_SUCCEEDED(mStatus) && mOpenedStreamChannel) {
return mStreamChannel->GetStatus(aResult);
}
*aResult = mStatus;
return NS_OK;
}
@ -623,7 +628,12 @@ nsJSChannel::AsyncOpen(nsIStreamListener *aListener, nsISupports *aContext)
nsCOMPtr<nsILoadGroup> loadGroup;
mStreamChannel->GetLoadGroup(getter_AddRefs(loadGroup));
if (loadGroup) {
loadGroup->AddRequest(this, nsnull);
nsresult rv = loadGroup->AddRequest(this, nsnull);
if (NS_FAILED(rv)) {
mIsActive = PR_FALSE;
CleanupStrongRefs();
return rv;
}
}
mDocumentOnloadBlockedOn =
@ -784,6 +794,19 @@ nsJSChannel::EvaluateScript()
// mStreamChannel will call OnStartRequest and OnStopRequest on
// us, so we'll be sure to call them on our listener.
mOpenedStreamChannel = PR_TRUE;
// Now readd ourselves to the loadgroup so we can receive
// cancellation notifications.
mIsActive = PR_TRUE;
if (loadGroup) {
mStatus = loadGroup->AddRequest(this, nsnull);
// If AddRequest failed, that's OK. The key is to make sure we get
// cancelled if needed, and that call just canceled us if it
// failed. We'll still get notified by the stream channel when it
// finishes.
}
} else if (mIsAsync) {
NotifyListener();
}
@ -951,8 +974,23 @@ nsJSChannel::OnStopRequest(nsIRequest* aRequest,
nsCOMPtr<nsIStreamListener> listener = mListener;
CleanupStrongRefs();
// Make sure aStatus matches what GetStatus() returns
if (NS_FAILED(mStatus)) {
aStatus = mStatus;
}
return listener->OnStopRequest(this, aContext, aStatus);
nsresult rv = listener->OnStopRequest(this, aContext, aStatus);
nsCOMPtr<nsILoadGroup> loadGroup;
mStreamChannel->GetLoadGroup(getter_AddRefs(loadGroup));
if (loadGroup) {
loadGroup->RemoveRequest(this, nsnull, mStatus);
}
mIsActive = PR_FALSE;
return rv;
}
NS_IMETHODIMP