github/socket.io
1
1
Fork 0
mirror of https://github.com/socketio/socket.io.git synced 2026-04-30 03:00:39 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(webtransport): properly handle abruptly closed connections

Browse Source
This commit is contained in:
Damien Arrachequesne
2023-06-27 14:25:52 +02:00
parent 01804d59c3
commit cf6aa1f43c
1 changed files with 31 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
lib/transports/webtransport.ts
View File

@@ -39,7 +39,15 @@ export class WT extends Transport {
this.opts.transportOptions[this.name]
);
this.transport.closed.then(() => this.onClose());
this.transport.closed
.then(() => {
debug("transport closed gracefully");
this.onClose();
})
.catch((err) => {
debug("transport closed due to %s", err);
this.onError("webtransport error", err);
});
// note: we could have used async/await, but that would require some additional polyfills
this.transport.ready.then(() => {
@@ -50,23 +58,28 @@ export class WT extends Transport {
let binaryFlag;
const read = () => {
reader.read().then(({ done, value }) => {
if (done) {
debug("session is closed");
return;
}
debug("received chunk: %o", value);
if (!binaryFlag && value.byteLength === 1 && value[0] === 54) {
binaryFlag = true;
} else {
// TODO expose binarytype
this.onPacket(
decodePacketFromBinary(value, binaryFlag, "arraybuffer")
);
binaryFlag = false;
}
read();
});
reader
.read()
.then(({ done, value }) => {
if (done) {
debug("session is closed");
return;
}
debug("received chunk: %o", value);
if (!binaryFlag && value.byteLength === 1 && value[0] === 54) {
binaryFlag = true;
} else {
// TODO expose binarytype
this.onPacket(
decodePacketFromBinary(value, binaryFlag, "arraybuffer")
);
binaryFlag = false;
}
read();
})
.catch((err) => {
debug("an error occurred while reading: %s", err);
});
};
read();