mirror of
https://github.com/denoland/deno.git
synced 2024-11-23 15:16:54 -05:00
577 lines
16 KiB
TypeScript
577 lines
16 KiB
TypeScript
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
import { BufReader, BufWriter } from "../io/bufio.ts";
|
|
import { assert, assertEquals, assertThrowsAsync } from "../testing/asserts.ts";
|
|
const { test } = Deno;
|
|
import { TextProtoReader } from "../textproto/mod.ts";
|
|
import * as bytes from "../bytes/mod.ts";
|
|
import {
|
|
acceptable,
|
|
connectWebSocket,
|
|
createSecAccept,
|
|
createSecKey,
|
|
handshake,
|
|
OpCode,
|
|
readFrame,
|
|
unmask,
|
|
writeFrame,
|
|
createWebSocket,
|
|
} from "./mod.ts";
|
|
import { encode, decode } from "../encoding/utf8.ts";
|
|
import Writer = Deno.Writer;
|
|
import Reader = Deno.Reader;
|
|
import Conn = Deno.Conn;
|
|
import Buffer = Deno.Buffer;
|
|
import { delay } from "../util/async.ts";
|
|
|
|
test("[ws] read unmasked text frame", async () => {
|
|
// unmasked single text frame with payload "Hello"
|
|
const buf = new BufReader(
|
|
new Buffer(new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]))
|
|
);
|
|
const frame = await readFrame(buf);
|
|
assertEquals(frame.opcode, OpCode.TextFrame);
|
|
assertEquals(frame.mask, undefined);
|
|
const actual = new TextDecoder().decode(new Buffer(frame.payload).bytes());
|
|
assertEquals(actual, "Hello");
|
|
assertEquals(frame.isLastFrame, true);
|
|
});
|
|
|
|
test("[ws] read masked text frame", async () => {
|
|
// a masked single text frame with payload "Hello"
|
|
const buf = new BufReader(
|
|
new Buffer(
|
|
new Uint8Array([
|
|
0x81,
|
|
0x85,
|
|
0x37,
|
|
0xfa,
|
|
0x21,
|
|
0x3d,
|
|
0x7f,
|
|
0x9f,
|
|
0x4d,
|
|
0x51,
|
|
0x58,
|
|
])
|
|
)
|
|
);
|
|
const frame = await readFrame(buf);
|
|
assertEquals(frame.opcode, OpCode.TextFrame);
|
|
unmask(frame.payload, frame.mask);
|
|
const actual = new TextDecoder().decode(new Buffer(frame.payload).bytes());
|
|
assertEquals(actual, "Hello");
|
|
assertEquals(frame.isLastFrame, true);
|
|
});
|
|
|
|
test("[ws] read unmasked split text frames", async () => {
|
|
const buf1 = new BufReader(
|
|
new Buffer(new Uint8Array([0x01, 0x03, 0x48, 0x65, 0x6c]))
|
|
);
|
|
const buf2 = new BufReader(
|
|
new Buffer(new Uint8Array([0x80, 0x02, 0x6c, 0x6f]))
|
|
);
|
|
const [f1, f2] = await Promise.all([readFrame(buf1), readFrame(buf2)]);
|
|
assertEquals(f1.isLastFrame, false);
|
|
assertEquals(f1.mask, undefined);
|
|
assertEquals(f1.opcode, OpCode.TextFrame);
|
|
const actual1 = new TextDecoder().decode(new Buffer(f1.payload).bytes());
|
|
assertEquals(actual1, "Hel");
|
|
|
|
assertEquals(f2.isLastFrame, true);
|
|
assertEquals(f2.mask, undefined);
|
|
assertEquals(f2.opcode, OpCode.Continue);
|
|
const actual2 = new TextDecoder().decode(new Buffer(f2.payload).bytes());
|
|
assertEquals(actual2, "lo");
|
|
});
|
|
|
|
test("[ws] read unmasked ping / pong frame", async () => {
|
|
// unmasked ping with payload "Hello"
|
|
const buf = new BufReader(
|
|
new Buffer(new Uint8Array([0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]))
|
|
);
|
|
const ping = await readFrame(buf);
|
|
assertEquals(ping.opcode, OpCode.Ping);
|
|
const actual1 = new TextDecoder().decode(new Buffer(ping.payload).bytes());
|
|
assertEquals(actual1, "Hello");
|
|
// prettier-ignore
|
|
const pongFrame= [0x8a, 0x85, 0x37, 0xfa, 0x21, 0x3d, 0x7f, 0x9f, 0x4d, 0x51, 0x58]
|
|
const buf2 = new BufReader(new Buffer(new Uint8Array(pongFrame)));
|
|
const pong = await readFrame(buf2);
|
|
assertEquals(pong.opcode, OpCode.Pong);
|
|
assert(pong.mask !== undefined);
|
|
unmask(pong.payload, pong.mask);
|
|
const actual2 = new TextDecoder().decode(new Buffer(pong.payload).bytes());
|
|
assertEquals(actual2, "Hello");
|
|
});
|
|
|
|
test("[ws] read unmasked big binary frame", async () => {
|
|
const payloadLength = 0x100;
|
|
const a = [0x82, 0x7e, 0x01, 0x00];
|
|
for (let i = 0; i < payloadLength; i++) {
|
|
a.push(i);
|
|
}
|
|
const buf = new BufReader(new Buffer(new Uint8Array(a)));
|
|
const bin = await readFrame(buf);
|
|
assertEquals(bin.opcode, OpCode.BinaryFrame);
|
|
assertEquals(bin.isLastFrame, true);
|
|
assertEquals(bin.mask, undefined);
|
|
assertEquals(bin.payload.length, payloadLength);
|
|
});
|
|
|
|
test("[ws] read unmasked bigger binary frame", async () => {
|
|
const payloadLength = 0x10000;
|
|
const a = [0x82, 0x7f, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00];
|
|
for (let i = 0; i < payloadLength; i++) {
|
|
a.push(i);
|
|
}
|
|
const buf = new BufReader(new Buffer(new Uint8Array(a)));
|
|
const bin = await readFrame(buf);
|
|
assertEquals(bin.opcode, OpCode.BinaryFrame);
|
|
assertEquals(bin.isLastFrame, true);
|
|
assertEquals(bin.mask, undefined);
|
|
assertEquals(bin.payload.length, payloadLength);
|
|
});
|
|
|
|
test("[ws] createSecAccept", () => {
|
|
const nonce = "dGhlIHNhbXBsZSBub25jZQ==";
|
|
const d = createSecAccept(nonce);
|
|
assertEquals(d, "s3pPLMBiTxaQ9kYGzzhZRbK+xOo=");
|
|
});
|
|
|
|
test("[ws] acceptable", () => {
|
|
const ret = acceptable({
|
|
headers: new Headers({
|
|
upgrade: "websocket",
|
|
"sec-websocket-key": "aaa",
|
|
}),
|
|
});
|
|
assertEquals(ret, true);
|
|
|
|
assert(
|
|
acceptable({
|
|
headers: new Headers([
|
|
["connection", "Upgrade"],
|
|
["host", "127.0.0.1:9229"],
|
|
[
|
|
"sec-websocket-extensions",
|
|
"permessage-deflate; client_max_window_bits",
|
|
],
|
|
["sec-websocket-key", "dGhlIHNhbXBsZSBub25jZQ=="],
|
|
["sec-websocket-version", "13"],
|
|
["upgrade", "WebSocket"],
|
|
]),
|
|
})
|
|
);
|
|
});
|
|
|
|
test("[ws] acceptable should return false when headers invalid", () => {
|
|
assertEquals(
|
|
acceptable({
|
|
headers: new Headers({ "sec-websocket-key": "aaa" }),
|
|
}),
|
|
false
|
|
);
|
|
assertEquals(
|
|
acceptable({
|
|
headers: new Headers({ upgrade: "websocket" }),
|
|
}),
|
|
false
|
|
);
|
|
assertEquals(
|
|
acceptable({
|
|
headers: new Headers({ upgrade: "invalid", "sec-websocket-key": "aaa" }),
|
|
}),
|
|
false
|
|
);
|
|
assertEquals(
|
|
acceptable({
|
|
headers: new Headers({ upgrade: "websocket", "sec-websocket-ky": "" }),
|
|
}),
|
|
false
|
|
);
|
|
});
|
|
|
|
test("[ws] connectWebSocket should throw invalid scheme of url", async (): Promise<
|
|
void
|
|
> => {
|
|
await assertThrowsAsync(
|
|
async (): Promise<void> => {
|
|
await connectWebSocket("file://hoge/hoge");
|
|
}
|
|
);
|
|
});
|
|
|
|
test("[ws] write and read masked frame", async () => {
|
|
const mask = new Uint8Array([0, 1, 2, 3]);
|
|
const msg = "hello";
|
|
const buf = new Buffer();
|
|
const r = new BufReader(buf);
|
|
await writeFrame(
|
|
{
|
|
isLastFrame: true,
|
|
mask,
|
|
opcode: OpCode.TextFrame,
|
|
payload: encode(msg),
|
|
},
|
|
buf
|
|
);
|
|
const frame = await readFrame(r);
|
|
assertEquals(frame.opcode, OpCode.TextFrame);
|
|
assertEquals(frame.isLastFrame, true);
|
|
assertEquals(frame.mask, mask);
|
|
unmask(frame.payload, frame.mask);
|
|
assertEquals(frame.payload, encode(msg));
|
|
});
|
|
|
|
test("[ws] handshake should not send search when it's empty", async () => {
|
|
const writer = new Buffer();
|
|
const reader = new Buffer(encode("HTTP/1.1 400\r\n"));
|
|
|
|
await assertThrowsAsync(
|
|
async (): Promise<void> => {
|
|
await handshake(
|
|
new URL("ws://example.com"),
|
|
new Headers(),
|
|
new BufReader(reader),
|
|
new BufWriter(writer)
|
|
);
|
|
}
|
|
);
|
|
|
|
const tpReader = new TextProtoReader(new BufReader(writer));
|
|
const statusLine = await tpReader.readLine();
|
|
|
|
assertEquals(statusLine, "GET / HTTP/1.1");
|
|
});
|
|
|
|
test("[ws] handshake should send search correctly", async function wsHandshakeWithSearch(): Promise<
|
|
void
|
|
> {
|
|
const writer = new Buffer();
|
|
const reader = new Buffer(encode("HTTP/1.1 400\r\n"));
|
|
|
|
await assertThrowsAsync(
|
|
async (): Promise<void> => {
|
|
await handshake(
|
|
new URL("ws://example.com?a=1"),
|
|
new Headers(),
|
|
new BufReader(reader),
|
|
new BufWriter(writer)
|
|
);
|
|
}
|
|
);
|
|
|
|
const tpReader = new TextProtoReader(new BufReader(writer));
|
|
const statusLine = await tpReader.readLine();
|
|
|
|
assertEquals(statusLine, "GET /?a=1 HTTP/1.1");
|
|
});
|
|
|
|
test("[ws] ws.close() should use 1000 as close code", async () => {
|
|
const buf = new Buffer();
|
|
const bufr = new BufReader(buf);
|
|
const conn = dummyConn(buf, buf);
|
|
const ws = createWebSocket({ conn });
|
|
await ws.close();
|
|
const frame = await readFrame(bufr);
|
|
assertEquals(frame.opcode, OpCode.Close);
|
|
const code = (frame.payload[0] << 8) | frame.payload[1];
|
|
assertEquals(code, 1000);
|
|
});
|
|
|
|
function dummyConn(r: Reader, w: Writer): Conn {
|
|
return {
|
|
rid: -1,
|
|
closeWrite: (): void => {},
|
|
read: (x): Promise<number | null> => r.read(x),
|
|
write: (x): Promise<number> => w.write(x),
|
|
close: (): void => {},
|
|
localAddr: { transport: "tcp", hostname: "0.0.0.0", port: 0 },
|
|
remoteAddr: { transport: "tcp", hostname: "0.0.0.0", port: 0 },
|
|
};
|
|
}
|
|
|
|
function delayedWriter(ms: number, dest: Writer): Writer {
|
|
return {
|
|
write(p: Uint8Array): Promise<number> {
|
|
return new Promise<number>((resolve) => {
|
|
setTimeout(async (): Promise<void> => {
|
|
resolve(await dest.write(p));
|
|
}, ms);
|
|
});
|
|
},
|
|
};
|
|
}
|
|
test({
|
|
name: "[ws] WebSocket.send(), WebSocket.ping() should be exclusive",
|
|
fn: async (): Promise<void> => {
|
|
const buf = new Buffer();
|
|
const conn = dummyConn(new Buffer(), delayedWriter(1, buf));
|
|
const sock = createWebSocket({ conn });
|
|
// Ensure send call
|
|
await Promise.all([
|
|
sock.send("first"),
|
|
sock.send("second"),
|
|
sock.ping(),
|
|
sock.send(new Uint8Array([3])),
|
|
]);
|
|
const bufr = new BufReader(buf);
|
|
const first = await readFrame(bufr);
|
|
const second = await readFrame(bufr);
|
|
const ping = await readFrame(bufr);
|
|
const third = await readFrame(bufr);
|
|
assertEquals(first.opcode, OpCode.TextFrame);
|
|
assertEquals(decode(first.payload), "first");
|
|
assertEquals(first.opcode, OpCode.TextFrame);
|
|
assertEquals(decode(second.payload), "second");
|
|
assertEquals(ping.opcode, OpCode.Ping);
|
|
assertEquals(third.opcode, OpCode.BinaryFrame);
|
|
assertEquals(bytes.equal(third.payload, new Uint8Array([3])), true);
|
|
},
|
|
});
|
|
|
|
test("[ws] createSecKeyHasCorrectLength", () => {
|
|
// Note: relies on --seed=86 being passed to deno to reproduce failure in
|
|
// #4063.
|
|
const secKey = createSecKey();
|
|
assertEquals(atob(secKey).length, 16);
|
|
});
|
|
|
|
test("[ws] WebSocket should throw `Deno.errors.ConnectionReset` when peer closed connection without close frame", async () => {
|
|
const buf = new Buffer();
|
|
const eofReader: Deno.Reader = {
|
|
read(_: Uint8Array): Promise<number | null> {
|
|
return Promise.resolve(null);
|
|
},
|
|
};
|
|
const conn = dummyConn(eofReader, buf);
|
|
const sock = createWebSocket({ conn });
|
|
sock.closeForce();
|
|
await assertThrowsAsync(
|
|
() => sock.send("hello"),
|
|
Deno.errors.ConnectionReset
|
|
);
|
|
await assertThrowsAsync(() => sock.ping(), Deno.errors.ConnectionReset);
|
|
await assertThrowsAsync(() => sock.close(0), Deno.errors.ConnectionReset);
|
|
});
|
|
|
|
test("[ws] WebSocket shouldn't throw `Deno.errors.UnexpectedEof`", async () => {
|
|
const buf = new Buffer();
|
|
const eofReader: Deno.Reader = {
|
|
read(_: Uint8Array): Promise<number | null> {
|
|
return Promise.resolve(null);
|
|
},
|
|
};
|
|
const conn = dummyConn(eofReader, buf);
|
|
const sock = createWebSocket({ conn });
|
|
const it = sock[Symbol.asyncIterator]();
|
|
const { value, done } = await it.next();
|
|
assertEquals(value, undefined);
|
|
assertEquals(done, true);
|
|
});
|
|
|
|
test({
|
|
name:
|
|
"[ws] WebSocket should reject sending promise when connection reset forcely",
|
|
fn: async () => {
|
|
const buf = new Buffer();
|
|
let timer: number | undefined;
|
|
const lazyWriter: Deno.Writer = {
|
|
write(_: Uint8Array): Promise<number> {
|
|
return new Promise((resolve) => {
|
|
timer = setTimeout(() => resolve(0), 1000);
|
|
});
|
|
},
|
|
};
|
|
const conn = dummyConn(buf, lazyWriter);
|
|
const sock = createWebSocket({ conn });
|
|
const onError = (e: unknown): unknown => e;
|
|
const p = Promise.all([
|
|
sock.send("hello").catch(onError),
|
|
sock.send(new Uint8Array([1, 2])).catch(onError),
|
|
sock.ping().catch(onError),
|
|
]);
|
|
sock.closeForce();
|
|
assertEquals(sock.isClosed, true);
|
|
const [a, b, c] = await p;
|
|
assert(a instanceof Deno.errors.ConnectionReset);
|
|
assert(b instanceof Deno.errors.ConnectionReset);
|
|
assert(c instanceof Deno.errors.ConnectionReset);
|
|
clearTimeout(timer);
|
|
// Wait for another event loop turn for `timeout` op promise
|
|
// to resolve, otherwise we'll get "op leak".
|
|
await delay(10);
|
|
},
|
|
});
|
|
|
|
test("[ws] WebSocket should implement Writer", async () => {
|
|
const buf = new Buffer();
|
|
|
|
const conn = dummyConn(buf, buf);
|
|
const sock = createWebSocket({ conn });
|
|
|
|
const [write0, write1, write2] = await Promise.all([
|
|
sock.write(new Uint8Array([1, 2, 3])),
|
|
sock.write(new Uint8Array([])),
|
|
sock.write(new Uint8Array([0])),
|
|
]);
|
|
|
|
assertEquals(write0, 3);
|
|
assertEquals(write1, 0);
|
|
assertEquals(write2, 1);
|
|
});
|
|
|
|
test("[ws] WebSocket should implement Reader", async () => {
|
|
const hello = new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
|
|
|
|
const bufHello = new Buffer(hello);
|
|
|
|
const conn = dummyConn(bufHello, new Buffer());
|
|
const sock = createWebSocket({ conn });
|
|
|
|
const p = new Uint8Array(100);
|
|
const read = await sock.read(p);
|
|
const readLast = await sock.read(p);
|
|
|
|
const helloLength = "Hello".length;
|
|
|
|
assertEquals(read, helloLength);
|
|
assertEquals(decode(new Buffer(p.subarray(0, helloLength)).bytes()), "Hello");
|
|
assertEquals(readLast, null);
|
|
});
|
|
|
|
test("[ws] WebSocket Reader should ignore non-message frames", async () => {
|
|
const pong = new Uint8Array([
|
|
0x8a,
|
|
0x85,
|
|
0x37,
|
|
0xfa,
|
|
0x21,
|
|
0x3d,
|
|
0x7f,
|
|
0x9f,
|
|
0x4d,
|
|
0x51,
|
|
0x58,
|
|
]);
|
|
const pingHello = new Uint8Array([0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
|
|
const hello = new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
|
|
const close = new Uint8Array([0x88, 0x02, 0x03, 0xe8]);
|
|
|
|
const dataPayloadLength = 0x100;
|
|
const dataArr = [0x82, 0x7e, 0x01, 0x00];
|
|
for (let i = 0; i < dataPayloadLength; i++) {
|
|
dataArr.push(i);
|
|
}
|
|
const data = new Uint8Array(dataArr);
|
|
|
|
enum Frames {
|
|
ping,
|
|
text,
|
|
pong,
|
|
data,
|
|
close,
|
|
end,
|
|
}
|
|
|
|
let frame = Frames.ping;
|
|
|
|
const reader: Reader = {
|
|
read(p: Uint8Array): Promise<number | null> {
|
|
if (frame === Frames.ping) {
|
|
frame = Frames.text;
|
|
p.set(pingHello);
|
|
return Promise.resolve(pingHello.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.text) {
|
|
frame = Frames.pong;
|
|
p.set(hello);
|
|
return Promise.resolve(hello.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.pong) {
|
|
frame = Frames.data;
|
|
p.set(pong);
|
|
return Promise.resolve(pong.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.data) {
|
|
frame = Frames.close;
|
|
p.set(data);
|
|
return Promise.resolve(data.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.close) {
|
|
frame = Frames.end;
|
|
p.set(close);
|
|
return Promise.resolve(close.byteLength);
|
|
}
|
|
|
|
return Promise.resolve(null);
|
|
},
|
|
};
|
|
|
|
const conn = dummyConn(reader, new Buffer());
|
|
const sock = createWebSocket({ conn });
|
|
|
|
const p = await Deno.readAll(sock);
|
|
|
|
const helloLength = "Hello".length;
|
|
|
|
assertEquals(p.byteLength, helloLength + dataPayloadLength);
|
|
assertEquals(decode(new Buffer(p.subarray(0, helloLength)).bytes()), "Hello");
|
|
assertEquals(p.subarray(helloLength), data.subarray(4));
|
|
});
|
|
|
|
test("[ws] WebSocket should act as asyncIterator", async () => {
|
|
const pingHello = new Uint8Array([0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
|
|
const hello = new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
|
|
const close = new Uint8Array([0x88, 0x04, 0x03, 0xf3, 0x34, 0x32]);
|
|
|
|
enum Frames {
|
|
ping,
|
|
hello,
|
|
close,
|
|
end,
|
|
}
|
|
|
|
let frame = Frames.ping;
|
|
|
|
const reader: Reader = {
|
|
read(p: Uint8Array): Promise<number | null> {
|
|
if (frame === Frames.ping) {
|
|
frame = Frames.hello;
|
|
p.set(pingHello);
|
|
return Promise.resolve(pingHello.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.hello) {
|
|
frame = Frames.close;
|
|
p.set(hello);
|
|
return Promise.resolve(hello.byteLength);
|
|
}
|
|
|
|
if (frame === Frames.close) {
|
|
frame = Frames.end;
|
|
p.set(close);
|
|
return Promise.resolve(close.byteLength);
|
|
}
|
|
|
|
return Promise.resolve(null);
|
|
},
|
|
};
|
|
|
|
const conn = dummyConn(reader, new Buffer());
|
|
const sock = createWebSocket({ conn });
|
|
|
|
const events = [];
|
|
for await (const wsEvent of sock) {
|
|
events.push(wsEvent);
|
|
}
|
|
|
|
assertEquals(events.length, 3);
|
|
assertEquals(events[0], ["ping", encode("Hello")]);
|
|
assertEquals(events[1], "Hello");
|
|
assertEquals(events[2], { code: 1011, reason: "42" });
|
|
});
|