123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- exports.Client = void 0;
- const socket_io_parser_1 = require("socket.io-parser");
- const debugModule = require("debug");
- const url = require("url");
- const debug = debugModule("socket.io:client");
- class Client {
- /**
- * Client constructor.
- *
- * @param server instance
- * @param conn
- * @package
- */
- constructor(server, conn) {
- this.sockets = new Map();
- this.nsps = new Map();
- this.server = server;
- this.conn = conn;
- this.encoder = server.encoder;
- this.decoder = new server._parser.Decoder();
- this.id = conn.id;
- this.setup();
- }
- /**
- * @return the reference to the request that originated the Engine.IO connection
- *
- * @public
- */
- get request() {
- return this.conn.request;
- }
- /**
- * Sets up event listeners.
- *
- * @private
- */
- setup() {
- this.onclose = this.onclose.bind(this);
- this.ondata = this.ondata.bind(this);
- this.onerror = this.onerror.bind(this);
- this.ondecoded = this.ondecoded.bind(this);
- // @ts-ignore
- this.decoder.on("decoded", this.ondecoded);
- this.conn.on("data", this.ondata);
- this.conn.on("error", this.onerror);
- this.conn.on("close", this.onclose);
- this.connectTimeout = setTimeout(() => {
- if (this.nsps.size === 0) {
- debug("no namespace joined yet, close the client");
- this.close();
- }
- else {
- debug("the client has already joined a namespace, nothing to do");
- }
- }, this.server._connectTimeout);
- }
- /**
- * Connects a client to a namespace.
- *
- * @param {String} name - the namespace
- * @param {Object} auth - the auth parameters
- * @private
- */
- connect(name, auth = {}) {
- if (this.server._nsps.has(name)) {
- debug("connecting to namespace %s", name);
- return this.doConnect(name, auth);
- }
- this.server._checkNamespace(name, auth, (dynamicNspName) => {
- if (dynamicNspName) {
- this.doConnect(name, auth);
- }
- else {
- debug("creation of namespace %s was denied", name);
- this._packet({
- type: socket_io_parser_1.PacketType.CONNECT_ERROR,
- nsp: name,
- data: {
- message: "Invalid namespace",
- },
- });
- }
- });
- }
- /**
- * Connects a client to a namespace.
- *
- * @param name - the namespace
- * @param {Object} auth - the auth parameters
- *
- * @private
- */
- doConnect(name, auth) {
- const nsp = this.server.of(name);
- const socket = nsp._add(this, auth, () => {
- this.sockets.set(socket.id, socket);
- this.nsps.set(nsp.name, socket);
- if (this.connectTimeout) {
- clearTimeout(this.connectTimeout);
- this.connectTimeout = undefined;
- }
- });
- }
- /**
- * Disconnects from all namespaces and closes transport.
- *
- * @private
- */
- _disconnect() {
- for (const socket of this.sockets.values()) {
- socket.disconnect();
- }
- this.sockets.clear();
- this.close();
- }
- /**
- * Removes a socket. Called by each `Socket`.
- *
- * @private
- */
- _remove(socket) {
- if (this.sockets.has(socket.id)) {
- const nsp = this.sockets.get(socket.id).nsp.name;
- this.sockets.delete(socket.id);
- this.nsps.delete(nsp);
- }
- else {
- debug("ignoring remove for %s", socket.id);
- }
- }
- /**
- * Closes the underlying connection.
- *
- * @private
- */
- close() {
- if ("open" === this.conn.readyState) {
- debug("forcing transport close");
- this.conn.close();
- this.onclose("forced server close");
- }
- }
- /**
- * Writes a packet to the transport.
- *
- * @param {Object} packet object
- * @param {Object} opts
- * @private
- */
- _packet(packet, opts = {}) {
- if (this.conn.readyState !== "open") {
- debug("ignoring packet write %j", packet);
- return;
- }
- const encodedPackets = opts.preEncoded
- ? packet // previous versions of the adapter incorrectly used socket.packet() instead of writeToEngine()
- : this.encoder.encode(packet);
- this.writeToEngine(encodedPackets, opts);
- }
- writeToEngine(encodedPackets, opts) {
- if (opts.volatile && !this.conn.transport.writable) {
- debug("volatile packet is discarded since the transport is not currently writable");
- return;
- }
- const packets = Array.isArray(encodedPackets)
- ? encodedPackets
- : [encodedPackets];
- for (const encodedPacket of packets) {
- this.conn.write(encodedPacket, opts);
- }
- }
- /**
- * Called with incoming transport data.
- *
- * @private
- */
- ondata(data) {
- // try/catch is needed for protocol violations (GH-1880)
- try {
- this.decoder.add(data);
- }
- catch (e) {
- debug("invalid packet format");
- this.onerror(e);
- }
- }
- /**
- * Called when parser fully decodes a packet.
- *
- * @private
- */
- ondecoded(packet) {
- let namespace;
- let authPayload;
- if (this.conn.protocol === 3) {
- const parsed = url.parse(packet.nsp, true);
- namespace = parsed.pathname;
- authPayload = parsed.query;
- }
- else {
- namespace = packet.nsp;
- authPayload = packet.data;
- }
- const socket = this.nsps.get(namespace);
- if (!socket && packet.type === socket_io_parser_1.PacketType.CONNECT) {
- this.connect(namespace, authPayload);
- }
- else if (socket &&
- packet.type !== socket_io_parser_1.PacketType.CONNECT &&
- packet.type !== socket_io_parser_1.PacketType.CONNECT_ERROR) {
- process.nextTick(function () {
- socket._onpacket(packet);
- });
- }
- else {
- debug("invalid state (packet type: %s)", packet.type);
- this.close();
- }
- }
- /**
- * Handles an error.
- *
- * @param {Object} err object
- * @private
- */
- onerror(err) {
- for (const socket of this.sockets.values()) {
- socket._onerror(err);
- }
- this.conn.close();
- }
- /**
- * Called upon transport close.
- *
- * @param reason
- * @private
- */
- onclose(reason) {
- debug("client close with reason %s", reason);
- // ignore a potential subsequent `close` event
- this.destroy();
- // `nsps` and `sockets` are cleaned up seamlessly
- for (const socket of this.sockets.values()) {
- socket._onclose(reason);
- }
- this.sockets.clear();
- this.decoder.destroy(); // clean up decoder
- }
- /**
- * Cleans up event listeners.
- * @private
- */
- destroy() {
- this.conn.removeListener("data", this.ondata);
- this.conn.removeListener("error", this.onerror);
- this.conn.removeListener("close", this.onclose);
- // @ts-ignore
- this.decoder.removeListener("decoded", this.ondecoded);
- if (this.connectTimeout) {
- clearTimeout(this.connectTimeout);
- this.connectTimeout = undefined;
- }
- }
- }
- exports.Client = Client;
|