fix: Close protocol response streams when aborted (#24657)

Co-authored-by: Paul Frazee <pfrazee@gmail.com>
This commit is contained in:
trop[bot]
2020-07-20 19:08:55 -07:00
committed by GitHub
parent 88d54af01e
commit 00be927ad6
2 changed files with 30 additions and 0 deletions

View File

@@ -42,6 +42,12 @@ NodeStreamLoader::~NodeStreamLoader() {
node::MakeCallback(isolate_, emitter_.Get(isolate_), "removeListener",
node::arraysize(args), args, {0, 0});
}
// Destroy the stream if not already ended
if (!ended_) {
node::MakeCallback(isolate_, emitter_.Get(isolate_), "destroy", 0, nullptr,
{0, 0});
}
}
void NodeStreamLoader::Start(network::mojom::URLResponseHeadPtr head) {

View File

@@ -443,6 +443,30 @@ describe('protocol module', () => {
ajax(protocolName + '://fake-host');
await hasEndedPromise;
});
it('destroys response streams when aborted before completion', async () => {
const events = new EventEmitter();
registerStreamProtocol(protocolName, (request, callback) => {
const responseStream = new stream.PassThrough();
responseStream.push('data\r\n');
responseStream.on('close', () => {
events.emit('close');
});
callback({
statusCode: 200,
headers: { 'Content-Type': 'text/plain' },
data: responseStream
});
events.emit('respond');
});
const hasRespondedPromise = emittedOnce(events, 'respond');
const hasClosedPromise = emittedOnce(events, 'close');
ajax(protocolName + '://fake-host');
await hasRespondedPromise;
await contents.loadFile(path.join(__dirname, 'fixtures', 'pages', 'jquery.html'));
await hasClosedPromise;
});
});
describe('protocol.isProtocolHandled', () => {