@mtcute/core 0.1.0 → 0.1.2

package/esm/network/transports/intermediate.js.map

@@ -1 +1 @@
-{"version":3,"file":"intermediate.js","sourceRoot":"","sources":["../../../../src/network/transports/intermediate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAA;AACtE,OAAO,EAAgB,cAAc,EAAE,MAAM,eAAe,CAAA;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAE7C,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;AACpD,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;AAE3D;;;GAGG;AACH,MAAM,OAAO,uBAAwB,SAAQ,aAAa;IACtD,GAAG;QACC,OAAO,GAAG,CAAA;IACd,CAAC;IAED,MAAM,CAAC,MAAkB;QACrB,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAC7C,MAAM,EAAE,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAClC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QACpC,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAElB,OAAO,GAAG,CAAA;IACd,CAAC;IAES,gBAAgB;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAA;IACnC,CAAC;IAES,aAAa;QACnB,MAAM,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;QAE3C,IAAI,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1C,IAAI,aAAa,KAAK,CAAC,EAAE;gBACrB,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;gBACtC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC,CAAA;aAC/C;iBAAM;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,CAAA;gBAC3D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;aAC/B;YAED,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,GAAG,CAAC,CAAC,CAAA;YAEvD,OAAO,IAAI,CAAA;SACd;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;CACJ;AAED;;;GAGG;AACH,MAAM,OAAO,6BAA8B,SAAQ,uBAAuB;IACtE,GAAG;QACC,OAAO,UAAU,CAAA;IACrB,CAAC;IAED,MAAM,CAAC,MAAkB;QACrB,qBAAqB;QACrB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC9C,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,CAAA;QAEpC,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,CAAA;QACvD,MAAM,EAAE,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAClC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,GAAG,OAAO,EAAE,IAAI,CAAC,CAAA;QAC9C,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAClB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAA;QAEnC,OAAO,GAAG,CAAA;IACd,CAAC;CACJ","sourcesContent":["import { dataViewFromBuffer, randomBytes } from '../../utils/index.js'\nimport { IPacketCodec, TransportError } from './abstract.js'\nimport { StreamedCodec } from './streamed.js'\n\nconst TAG = new Uint8Array([0xee, 0xee, 0xee, 0xee])\nconst PADDED_TAG = new Uint8Array([0xdd, 0xdd, 0xdd, 0xdd])\n\n/**\n * Intermediate packet codec.\n * See https://core.telegram.org/mtproto/mtproto-transports#intermediate\n */\nexport class IntermediatePacketCodec extends StreamedCodec implements IPacketCodec {\n    tag(): Uint8Array {\n        return TAG\n    }\n\n    encode(packet: Uint8Array): Uint8Array {\n        const ret = new Uint8Array(packet.length + 4)\n        const dv = dataViewFromBuffer(ret)\n        dv.setUint32(0, packet.length, true)\n        ret.set(packet, 4)\n\n        return ret\n    }\n\n    protected _packetAvailable(): boolean {\n        return this._stream.length >= 8\n    }\n\n    protected _handlePacket(): boolean {\n        const dv = dataViewFromBuffer(this._stream)\n        const payloadLength = dv.getUint32(0, true)\n\n        if (payloadLength <= this._stream.length - 4) {\n            if (payloadLength === 4) {\n                const code = dv.getInt32(4, true) * -1\n                this.emit('error', new TransportError(code))\n            } else {\n                const payload = this._stream.subarray(4, payloadLength + 4)\n                this.emit('packet', payload)\n            }\n\n            this._stream = this._stream.subarray(payloadLength + 4)\n\n            return true\n        }\n\n        return false\n    }\n}\n\n/**\n * Padded intermediate packet codec.\n * See https://core.telegram.org/mtproto/mtproto-transports#padded-intermediate\n */\nexport class PaddedIntermediatePacketCodec extends IntermediatePacketCodec {\n    tag(): Uint8Array {\n        return PADDED_TAG\n    }\n\n    encode(packet: Uint8Array): Uint8Array {\n        // padding size, 0-15\n        const padSize = Math.floor(Math.random() * 16)\n        const padding = randomBytes(padSize)\n\n        const ret = new Uint8Array(packet.length + 4 + padSize)\n        const dv = dataViewFromBuffer(ret)\n        dv.setUint32(0, packet.length + padSize, true)\n        ret.set(packet, 4)\n        ret.set(padding, 4 + packet.length)\n\n        return ret\n    }\n}\n"]}
+{"version":3,"file":"intermediate.js","sourceRoot":"","sources":["../../../../src/network/transports/intermediate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAmB,MAAM,sBAAsB,CAAA;AACxF,OAAO,EAAgB,cAAc,EAAE,MAAM,eAAe,CAAA;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAA;AAE7C,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;AACpD,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;AAE3D;;;GAGG;AACH,MAAM,OAAO,uBAAwB,SAAQ,aAAa;IACtD,GAAG;QACC,OAAO,GAAG,CAAA;IACd,CAAC;IAED,MAAM,CAAC,MAAkB;QACrB,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QAC7C,MAAM,EAAE,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAClC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAA;QACpC,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAElB,OAAO,GAAG,CAAA;IACd,CAAC;IAES,gBAAgB;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAA;IACnC,CAAC;IAES,aAAa;QACnB,MAAM,EAAE,GAAG,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;QAE3C,IAAI,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1C,IAAI,aAAa,KAAK,CAAC,EAAE;gBACrB,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAA;gBACtC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC,CAAA;aAC/C;iBAAM;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,CAAA;gBAC3D,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;aAC/B;YAED,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,GAAG,CAAC,CAAC,CAAA;YAEvD,OAAO,IAAI,CAAA;SACd;QAED,OAAO,KAAK,CAAA;IAChB,CAAC;CACJ;AAED;;;GAGG;AACH,MAAM,OAAO,6BAA8B,SAAQ,uBAAuB;IACtE,GAAG;QACC,OAAO,UAAU,CAAA;IACrB,CAAC;IAGD,KAAK,CAAE,MAAuB;QAC1B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;IACzB,CAAC;IAED,MAAM,CAAC,MAAkB;QACrB,qBAAqB;QACrB,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,CAAC,CAAA;QAEhC,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,CAAA;QACvD,MAAM,EAAE,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAClC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,GAAG,OAAO,EAAE,IAAI,CAAC,CAAA;QAC9C,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;QAClB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;QAExD,OAAO,GAAG,CAAA;IACd,CAAC;CACJ","sourcesContent":["import { dataViewFromBuffer, getRandomInt, ICryptoProvider } from '../../utils/index.js'\nimport { IPacketCodec, TransportError } from './abstract.js'\nimport { StreamedCodec } from './streamed.js'\n\nconst TAG = new Uint8Array([0xee, 0xee, 0xee, 0xee])\nconst PADDED_TAG = new Uint8Array([0xdd, 0xdd, 0xdd, 0xdd])\n\n/**\n * Intermediate packet codec.\n * See https://core.telegram.org/mtproto/mtproto-transports#intermediate\n */\nexport class IntermediatePacketCodec extends StreamedCodec implements IPacketCodec {\n    tag(): Uint8Array {\n        return TAG\n    }\n\n    encode(packet: Uint8Array): Uint8Array {\n        const ret = new Uint8Array(packet.length + 4)\n        const dv = dataViewFromBuffer(ret)\n        dv.setUint32(0, packet.length, true)\n        ret.set(packet, 4)\n\n        return ret\n    }\n\n    protected _packetAvailable(): boolean {\n        return this._stream.length >= 8\n    }\n\n    protected _handlePacket(): boolean {\n        const dv = dataViewFromBuffer(this._stream)\n        const payloadLength = dv.getUint32(0, true)\n\n        if (payloadLength <= this._stream.length - 4) {\n            if (payloadLength === 4) {\n                const code = dv.getInt32(4, true) * -1\n                this.emit('error', new TransportError(code))\n            } else {\n                const payload = this._stream.subarray(4, payloadLength + 4)\n                this.emit('packet', payload)\n            }\n\n            this._stream = this._stream.subarray(payloadLength + 4)\n\n            return true\n        }\n\n        return false\n    }\n}\n\n/**\n * Padded intermediate packet codec.\n * See https://core.telegram.org/mtproto/mtproto-transports#padded-intermediate\n */\nexport class PaddedIntermediatePacketCodec extends IntermediatePacketCodec {\n    tag(): Uint8Array {\n        return PADDED_TAG\n    }\n\n    private _crypto!: ICryptoProvider\n    setup?(crypto: ICryptoProvider) {\n        this._crypto = crypto\n    }\n\n    encode(packet: Uint8Array): Uint8Array {\n        // padding size, 0-15\n        const padSize = getRandomInt(16)\n\n        const ret = new Uint8Array(packet.length + 4 + padSize)\n        const dv = dataViewFromBuffer(ret)\n        dv.setUint32(0, packet.length + padSize, true)\n        ret.set(packet, 4)\n        this._crypto.randomFill(ret.subarray(4 + packet.length))\n\n        return ret\n    }\n}\n"]}

package/esm/network/transports/intermediate.test.d.ts

@@ -0,0 +1 @@
+export {};

package/esm/network/transports/intermediate.test.js

@@ -0,0 +1,84 @@
+import { describe, expect, it } from 'vitest';
+import { defaultTestCryptoProvider, useFakeMathRandom } from '@mtcute/test';
+import { hexDecodeToBuffer, hexEncode } from '@mtcute/tl-runtime';
+import { IntermediatePacketCodec, PaddedIntermediatePacketCodec, TransportError } from '../../index.js';
+import { concatBuffers } from '../../utils/index.js';
+describe('IntermediatePacketCodec', () => {
+    it('should return correct tag', () => {
+        expect(hexEncode(new IntermediatePacketCodec().tag())).eq('eeeeeeee');
+    });
+    it('should correctly parse immediate framing', () => new Promise((done) => {
+        const codec = new IntermediatePacketCodec();
+        codec.on('packet', (data) => {
+            expect([...data]).eql([5, 1, 2, 3, 4]);
+            done();
+        });
+        codec.feed(hexDecodeToBuffer('050000000501020304'));
+    }));
+    it('should correctly parse incomplete framing', () => new Promise((done) => {
+        const codec = new IntermediatePacketCodec();
+        codec.on('packet', (data) => {
+            expect([...data]).eql([5, 1, 2, 3, 4]);
+            done();
+        });
+        codec.feed(hexDecodeToBuffer('050000000501'));
+        codec.feed(hexDecodeToBuffer('020304'));
+    }));
+    it('should correctly parse multiple streamed packets', () => new Promise((done) => {
+        const codec = new IntermediatePacketCodec();
+        let number = 0;
+        codec.on('packet', (data) => {
+            if (number === 0) {
+                expect([...data]).eql([5, 1, 2, 3, 4]);
+                number = 1;
+            }
+            else {
+                expect([...data]).eql([3, 1, 2, 3, 1]);
+                done();
+            }
+        });
+        codec.feed(hexDecodeToBuffer('050000000501'));
+        codec.feed(hexDecodeToBuffer('020304050000'));
+        codec.feed(hexDecodeToBuffer('000301020301'));
+    }));
+    it('should correctly parse transport errors', () => new Promise((done) => {
+        const codec = new IntermediatePacketCodec();
+        codec.on('error', (err) => {
+            expect(err).to.have.instanceOf(TransportError);
+            expect(err.code).eq(404);
+            done();
+        });
+        codec.feed(hexDecodeToBuffer('040000006cfeffff'));
+    }));
+    it('should reset when called reset()', () => new Promise((done) => {
+        const codec = new IntermediatePacketCodec();
+        codec.on('packet', (data) => {
+            expect([...data]).eql([1, 2, 3, 4, 5]);
+            done();
+        });
+        codec.feed(hexDecodeToBuffer('ff0000001234567812345678'));
+        codec.reset();
+        codec.feed(hexDecodeToBuffer('050000000102030405'));
+    }));
+    it('should correctly frame packets', () => {
+        const data = hexDecodeToBuffer('6cfeffff');
+        // eslint-disable-next-line no-restricted-globals
+        expect(Buffer.from(new IntermediatePacketCodec().encode(data))).toEqual(concatBuffers([new Uint8Array([0x04, 0x00, 0x00, 0x00]), data]));
+    });
+});
+describe('PaddedIntermediatePacketCodec', () => {
+    useFakeMathRandom();
+    const create = async () => {
+        const codec = new PaddedIntermediatePacketCodec();
+        codec.setup(await defaultTestCryptoProvider());
+        return codec;
+    };
+    it('should return correct tag', async () => {
+        expect(hexEncode((await create()).tag())).eq('dddddddd');
+    });
+    it('should correctly frame packets', async () => {
+        const data = hexDecodeToBuffer('6cfeffff');
+        expect(hexEncode((await create()).encode(data))).toEqual('0a0000006cfeffff29afd26df40f');
+    });
+});
+//# sourceMappingURL=intermediate.test.js.map

package/esm/network/transports/intermediate.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"intermediate.test.js","sourceRoot":"","sources":["../../../../src/network/transports/intermediate.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7C,OAAO,EAAE,yBAAyB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAA;AAC3E,OAAO,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAA;AAEjE,OAAO,EAAE,uBAAuB,EAAE,6BAA6B,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAA;AACvG,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAA;AAEpD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACjC,MAAM,CAAC,SAAS,CAAC,IAAI,uBAAuB,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAA;IACzE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE,CAChD,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE;QACvB,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAC3C,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAgB,EAAE,EAAE;YACpC,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;YACtC,IAAI,EAAE,CAAA;QACV,CAAC,CAAC,CAAA;QACF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,CAAC,CAAA;IACvD,CAAC,CAAC,CAAC,CAAA;IAEP,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE,CACjD,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE;QACvB,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAC3C,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAgB,EAAE,EAAE;YACpC,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;YACtC,IAAI,EAAE,CAAA;QACV,CAAC,CAAC,CAAA;QACF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAC,CAAA;IAEP,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE,CACxD,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE;QACvB,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAE3C,IAAI,MAAM,GAAG,CAAC,CAAA;QAEd,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAgB,EAAE,EAAE;YACpC,IAAI,MAAM,KAAK,CAAC,EAAE;gBACd,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;gBACtC,MAAM,GAAG,CAAC,CAAA;aACb;iBAAM;gBACH,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;gBACtC,IAAI,EAAE,CAAA;aACT;QACL,CAAC,CAAC,CAAA;QACF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAA;IACjD,CAAC,CAAC,CAAC,CAAA;IAEP,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAC/C,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE;QACvB,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAE3C,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAmB,EAAE,EAAE;YACtC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAA;YAC9C,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAA;YACxB,IAAI,EAAE,CAAA;QACV,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC,CAAA;IACrD,CAAC,CAAC,CAAC,CAAA;IAEP,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE,CACxC,IAAI,OAAO,CAAO,CAAC,IAAI,EAAE,EAAE;QACvB,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAE3C,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAgB,EAAE,EAAE;YACpC,MAAM,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;YACtC,IAAI,EAAE,CAAA;QACV,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,0BAA0B,CAAC,CAAC,CAAA;QACzD,KAAK,CAAC,KAAK,EAAE,CAAA;QACb,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,CAAC,CAAA;IACvD,CAAC,CAAC,CAAC,CAAA;IAEP,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACtC,MAAM,IAAI,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAA;QAE1C,iDAAiD;QACjD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,uBAAuB,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CACnE,aAAa,CAAC,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAClE,CAAA;IACL,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC3C,iBAAiB,EAAE,CAAA;IAEnB,MAAM,MAAM,GAAG,KAAK,IAAI,EAAE;QACtB,MAAM,KAAK,GAAG,IAAI,6BAA6B,EAAE,CAAA;QACjD,KAAK,CAAC,KAAM,CAAC,MAAM,yBAAyB,EAAE,CAAC,CAAA;QAE/C,OAAO,KAAK,CAAA;IAChB,CAAC,CAAA;IAED,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,MAAM,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAA;IAC5D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,IAAI,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAA;QAE1C,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAA;IAC5F,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { describe, expect, it } from 'vitest'\n\nimport { defaultTestCryptoProvider, useFakeMathRandom } from '@mtcute/test'\nimport { hexDecodeToBuffer, hexEncode } from '@mtcute/tl-runtime'\n\nimport { IntermediatePacketCodec, PaddedIntermediatePacketCodec, TransportError } from '../../index.js'\nimport { concatBuffers } from '../../utils/index.js'\n\ndescribe('IntermediatePacketCodec', () => {\n    it('should return correct tag', () => {\n        expect(hexEncode(new IntermediatePacketCodec().tag())).eq('eeeeeeee')\n    })\n\n    it('should correctly parse immediate framing', () =>\n        new Promise<void>((done) => {\n            const codec = new IntermediatePacketCodec()\n            codec.on('packet', (data: Uint8Array) => {\n                expect([...data]).eql([5, 1, 2, 3, 4])\n                done()\n            })\n            codec.feed(hexDecodeToBuffer('050000000501020304'))\n        }))\n\n    it('should correctly parse incomplete framing', () =>\n        new Promise<void>((done) => {\n            const codec = new IntermediatePacketCodec()\n            codec.on('packet', (data: Uint8Array) => {\n                expect([...data]).eql([5, 1, 2, 3, 4])\n                done()\n            })\n            codec.feed(hexDecodeToBuffer('050000000501'))\n            codec.feed(hexDecodeToBuffer('020304'))\n        }))\n\n    it('should correctly parse multiple streamed packets', () =>\n        new Promise<void>((done) => {\n            const codec = new IntermediatePacketCodec()\n\n            let number = 0\n\n            codec.on('packet', (data: Uint8Array) => {\n                if (number === 0) {\n                    expect([...data]).eql([5, 1, 2, 3, 4])\n                    number = 1\n                } else {\n                    expect([...data]).eql([3, 1, 2, 3, 1])\n                    done()\n                }\n            })\n            codec.feed(hexDecodeToBuffer('050000000501'))\n            codec.feed(hexDecodeToBuffer('020304050000'))\n            codec.feed(hexDecodeToBuffer('000301020301'))\n        }))\n\n    it('should correctly parse transport errors', () =>\n        new Promise<void>((done) => {\n            const codec = new IntermediatePacketCodec()\n\n            codec.on('error', (err: TransportError) => {\n                expect(err).to.have.instanceOf(TransportError)\n                expect(err.code).eq(404)\n                done()\n            })\n\n            codec.feed(hexDecodeToBuffer('040000006cfeffff'))\n        }))\n\n    it('should reset when called reset()', () =>\n        new Promise<void>((done) => {\n            const codec = new IntermediatePacketCodec()\n\n            codec.on('packet', (data: Uint8Array) => {\n                expect([...data]).eql([1, 2, 3, 4, 5])\n                done()\n            })\n\n            codec.feed(hexDecodeToBuffer('ff0000001234567812345678'))\n            codec.reset()\n            codec.feed(hexDecodeToBuffer('050000000102030405'))\n        }))\n\n    it('should correctly frame packets', () => {\n        const data = hexDecodeToBuffer('6cfeffff')\n\n        // eslint-disable-next-line no-restricted-globals\n        expect(Buffer.from(new IntermediatePacketCodec().encode(data))).toEqual(\n            concatBuffers([new Uint8Array([0x04, 0x00, 0x00, 0x00]), data]),\n        )\n    })\n})\n\ndescribe('PaddedIntermediatePacketCodec', () => {\n    useFakeMathRandom()\n\n    const create = async () => {\n        const codec = new PaddedIntermediatePacketCodec()\n        codec.setup!(await defaultTestCryptoProvider())\n\n        return codec\n    }\n\n    it('should return correct tag', async () => {\n        expect(hexEncode((await create()).tag())).eq('dddddddd')\n    })\n\n    it('should correctly frame packets', async () => {\n        const data = hexDecodeToBuffer('6cfeffff')\n\n        expect(hexEncode((await create()).encode(data))).toEqual('0a0000006cfeffff29afd26df40f')\n    })\n})\n"]}

package/esm/network/transports/obfuscated.js

@@ -1,5 +1,4 @@
-import { concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js';
-import { randomBytes } from '../../utils/index.js';
+import { bufferToReversed, concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js';
 import { WrappedCodec } from './wrapped.js';
 export class ObfuscatedPacketCodec extends WrappedCodec {
     constructor(inner, proxy) {
@@ -10,11 +9,11 @@ export class ObfuscatedPacketCodec extends WrappedCodec {
         let random;
         let dv;
         for (;;) {
-            random = randomBytes(64);
+            random = this._crypto.randomBytes(64);
             if (random[0] === 0xef)
                 continue;
             dv = dataViewFromBuffer(random);
-            const firstInt = dv.getInt32(0, true);
+            const firstInt = dv.getUint32(0, true);
             if (firstInt === 0x44414548 || // HEAD
                 firstInt === 0x54534f50 || // POST
                 firstInt === 0x20544547 || // GET(space)
@@ -43,37 +42,34 @@ export class ObfuscatedPacketCodec extends WrappedCodec {
                 dcId = -dcId;
             dv.setInt16(60, dcId, true);
         }
-        // randomBytes may return a Buffer in Node.js, whose .slice() doesn't copy
-        const randomRev = Uint8Array.prototype.slice.call(random, 8, 56).reverse();
+        const randomRev = bufferToReversed(random, 8, 56);
         let encryptKey = random.subarray(8, 40);
         const encryptIv = random.subarray(40, 56);
         let decryptKey = randomRev.subarray(0, 32);
         const decryptIv = randomRev.subarray(32, 48);
         if (this._proxy) {
-            encryptKey = await this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]));
-            decryptKey = await this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]));
+            encryptKey = this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]));
+            decryptKey = this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]));
         }
-        this._encryptor = await this._crypto.createAesCtr(encryptKey, encryptIv, true);
-        this._decryptor = await this._crypto.createAesCtr(decryptKey, decryptIv, false);
-        const encrypted = await this._encryptor.encrypt(random);
+        this._encryptor = this._crypto.createAesCtr(encryptKey, encryptIv, true);
+        this._decryptor = this._crypto.createAesCtr(decryptKey, decryptIv, false);
+        const encrypted = this._encryptor.process(random);
         random.set(encrypted.subarray(56, 64), 56);
         return random;
     }
     async encode(packet) {
-        return this._encryptor.encrypt(await this._inner.encode(packet));
+        return this._encryptor.process(await this._inner.encode(packet));
     }
     feed(data) {
-        const dec = this._decryptor.decrypt(data);
-        if (ArrayBuffer.isView(dec))
-            this._inner.feed(dec);
-        else {
-            dec.then((dec) => this._inner.feed(dec)).catch((err) => this.emit('error', err));
-        }
+        const dec = this._decryptor.process(data);
+        this._inner.feed(dec);
     }
     reset() {
         this._inner.reset();
-        delete this._encryptor;
-        delete this._decryptor;
+        this._encryptor?.close?.();
+        this._decryptor?.close?.();
+        this._encryptor = undefined;
+        this._decryptor = undefined;
     }
 }
 //# sourceMappingURL=obfuscated.js.map

package/esm/network/transports/obfuscated.js.map

@@ -1 +1 @@
-{"version":3,"file":"obfuscated.js","sourceRoot":"","sources":["../../../../src/network/transports/obfuscated.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAA;AAC/E,OAAO,EAAqB,WAAW,EAAE,MAAM,sBAAsB,CAAA;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAA;AAS3C,MAAM,OAAO,qBAAsB,SAAQ,YAAY;IAMnD,YAAY,KAAmB,EAAE,KAAmB;QAChD,KAAK,CAAC,KAAK,CAAC,CAAA;QACZ,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,GAAG;QACL,IAAI,MAAkB,CAAA;QACtB,IAAI,EAAY,CAAA;QAEhB,SAAS;YACL,MAAM,GAAG,WAAW,CAAC,EAAE,CAAC,CAAA;YACxB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI;gBAAE,SAAQ;YAEhC,EAAE,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAA;YAC/B,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAErC,IACI,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,aAAa;gBACxC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,sBAAsB;gBACjD,QAAQ,KAAK,UAAU,IAAI,eAAe;gBAC1C,QAAQ,KAAK,UAAU,CAAC,4BAA4B;cACtD;gBACE,SAAQ;aACX;YACD,IAAI,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC;gBAAE,SAAQ;YAExC,MAAK;SACR;QAED,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAA;QAEtC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YACvB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;YACrB,QAAQ,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;SAC1C;QACD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;QAExB,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAA;YAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAAE,IAAI,IAAI,KAAK,CAAA;YACnC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK;gBAAE,IAAI,GAAG,CAAC,IAAI,CAAA;YAEnC,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAA;SAC9B;QAED,0EAA0E;QAC1E,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAA;QAE1E,IAAI,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvC,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAEzC,IAAI,UAAU,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAC1C,MAAM,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAE5C,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;YACvF,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;SAC1F;QAED,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAA;QAC9E,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;QAE/E,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACvD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QAE1C,OAAO,MAAM,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAkB;QAC3B,OAAO,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;IACrE,CAAC;IAED,IAAI,CAAC,IAAgB;QACjB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAE1C,IAAI,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;aAC7C;YACD,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;SACnF;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACnB,OAAO,IAAI,CAAC,UAAU,CAAA;QACtB,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;CACJ","sourcesContent":["import { concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js'\nimport { IEncryptionScheme, randomBytes } from '../../utils/index.js'\nimport { IPacketCodec } from './abstract.js'\nimport { WrappedCodec } from './wrapped.js'\n\nexport interface MtProxyInfo {\n    dcId: number\n    secret: Uint8Array\n    test: boolean\n    media: boolean\n}\n\nexport class ObfuscatedPacketCodec extends WrappedCodec implements IPacketCodec {\n    private _encryptor?: IEncryptionScheme\n    private _decryptor?: IEncryptionScheme\n\n    private _proxy?: MtProxyInfo\n\n    constructor(inner: IPacketCodec, proxy?: MtProxyInfo) {\n        super(inner)\n        this._proxy = proxy\n    }\n\n    async tag(): Promise<Uint8Array> {\n        let random: Uint8Array\n        let dv: DataView\n\n        for (;;) {\n            random = randomBytes(64)\n            if (random[0] === 0xef) continue\n\n            dv = dataViewFromBuffer(random)\n            const firstInt = dv.getInt32(0, true)\n\n            if (\n                firstInt === 0x44414548 || // HEAD\n                firstInt === 0x54534f50 || // POST\n                firstInt === 0x20544547 || // GET(space)\n                firstInt === 0x4954504f || // OPTI\n                firstInt === 0xdddddddd || // padded intermediate\n                firstInt === 0xeeeeeeee || // intermediate\n                firstInt === 0x02010316 // no idea, taken from tdlib\n            ) {\n                continue\n            }\n            if (dv.getInt32(4, true) === 0) continue\n\n            break\n        }\n\n        let innerTag = await this._inner.tag()\n\n        if (innerTag.length !== 4) {\n            const b = innerTag[0]\n            innerTag = new Uint8Array([b, b, b, b])\n        }\n        random.set(innerTag, 56)\n\n        if (this._proxy) {\n            let dcId = this._proxy.dcId\n            if (this._proxy.test) dcId += 10000\n            if (this._proxy.media) dcId = -dcId\n\n            dv.setInt16(60, dcId, true)\n        }\n\n        // randomBytes may return a Buffer in Node.js, whose .slice() doesn't copy\n        const randomRev = Uint8Array.prototype.slice.call(random, 8, 56).reverse()\n\n        let encryptKey = random.subarray(8, 40)\n        const encryptIv = random.subarray(40, 56)\n\n        let decryptKey = randomRev.subarray(0, 32)\n        const decryptIv = randomRev.subarray(32, 48)\n\n        if (this._proxy) {\n            encryptKey = await this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]))\n            decryptKey = await this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]))\n        }\n\n        this._encryptor = await this._crypto.createAesCtr(encryptKey, encryptIv, true)\n        this._decryptor = await this._crypto.createAesCtr(decryptKey, decryptIv, false)\n\n        const encrypted = await this._encryptor.encrypt(random)\n        random.set(encrypted.subarray(56, 64), 56)\n\n        return random\n    }\n\n    async encode(packet: Uint8Array): Promise<Uint8Array> {\n        return this._encryptor!.encrypt(await this._inner.encode(packet))\n    }\n\n    feed(data: Uint8Array): void {\n        const dec = this._decryptor!.decrypt(data)\n\n        if (ArrayBuffer.isView(dec)) this._inner.feed(dec)\n        else {\n            dec.then((dec) => this._inner.feed(dec)).catch((err) => this.emit('error', err))\n        }\n    }\n\n    reset(): void {\n        this._inner.reset()\n        delete this._encryptor\n        delete this._decryptor\n    }\n}\n"]}
+{"version":3,"file":"obfuscated.js","sourceRoot":"","sources":["../../../../src/network/transports/obfuscated.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAA;AAGjG,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAA;AAS3C,MAAM,OAAO,qBAAsB,SAAQ,YAAY;IAMnD,YAAY,KAAmB,EAAE,KAAmB;QAChD,KAAK,CAAC,KAAK,CAAC,CAAA;QACZ,IAAI,CAAC,MAAM,GAAG,KAAK,CAAA;IACvB,CAAC;IAED,KAAK,CAAC,GAAG;QACL,IAAI,MAAkB,CAAA;QACtB,IAAI,EAAY,CAAA;QAEhB,SAAS;YACL,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;YACrC,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI;gBAAE,SAAQ;YAEhC,EAAE,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAA;YAC/B,MAAM,QAAQ,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;YAEtC,IACI,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,aAAa;gBACxC,QAAQ,KAAK,UAAU,IAAI,OAAO;gBAClC,QAAQ,KAAK,UAAU,IAAI,sBAAsB;gBACjD,QAAQ,KAAK,UAAU,IAAI,eAAe;gBAC1C,QAAQ,KAAK,UAAU,CAAC,4BAA4B;cACtD;gBACE,SAAQ;aACX;YACD,IAAI,EAAE,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC;gBAAE,SAAQ;YAExC,MAAK;SACR;QAED,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAA;QAEtC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YACvB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;YACrB,QAAQ,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;SAC1C;QACD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAA;QAExB,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,IAAI,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAA;YAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBAAE,IAAI,IAAI,KAAK,CAAA;YACnC,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK;gBAAE,IAAI,GAAG,CAAC,IAAI,CAAA;YAEnC,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAA;SAC9B;QAED,MAAM,SAAS,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QAEjD,IAAI,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QACvC,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAEzC,IAAI,UAAU,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAC1C,MAAM,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAA;QAE5C,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;YACjF,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAA;SACpF;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAA;QACxE,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;QAEzE,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;QACjD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;QAE1C,OAAO,MAAM,CAAA;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAkB;QAC3B,OAAO,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAA;IACrE,CAAC;IAED,IAAI,CAAC,IAAgB;QACjB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAE1C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACzB,CAAC;IAED,KAAK;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAA;QACnB,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAA;QAC1B,IAAI,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAA;QAE1B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;QAC3B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAA;IAC/B,CAAC;CACJ","sourcesContent":["import { bufferToReversed, concatBuffers, dataViewFromBuffer } from '../../utils/buffer-utils.js'\nimport { IAesCtr } from '../../utils/index.js'\nimport { IPacketCodec } from './abstract.js'\nimport { WrappedCodec } from './wrapped.js'\n\nexport interface MtProxyInfo {\n    dcId: number\n    secret: Uint8Array\n    test: boolean\n    media: boolean\n}\n\nexport class ObfuscatedPacketCodec extends WrappedCodec implements IPacketCodec {\n    private _encryptor?: IAesCtr\n    private _decryptor?: IAesCtr\n\n    private _proxy?: MtProxyInfo\n\n    constructor(inner: IPacketCodec, proxy?: MtProxyInfo) {\n        super(inner)\n        this._proxy = proxy\n    }\n\n    async tag(): Promise<Uint8Array> {\n        let random: Uint8Array\n        let dv: DataView\n\n        for (;;) {\n            random = this._crypto.randomBytes(64)\n            if (random[0] === 0xef) continue\n\n            dv = dataViewFromBuffer(random)\n            const firstInt = dv.getUint32(0, true)\n\n            if (\n                firstInt === 0x44414548 || // HEAD\n                firstInt === 0x54534f50 || // POST\n                firstInt === 0x20544547 || // GET(space)\n                firstInt === 0x4954504f || // OPTI\n                firstInt === 0xdddddddd || // padded intermediate\n                firstInt === 0xeeeeeeee || // intermediate\n                firstInt === 0x02010316 // no idea, taken from tdlib\n            ) {\n                continue\n            }\n            if (dv.getInt32(4, true) === 0) continue\n\n            break\n        }\n\n        let innerTag = await this._inner.tag()\n\n        if (innerTag.length !== 4) {\n            const b = innerTag[0]\n            innerTag = new Uint8Array([b, b, b, b])\n        }\n        random.set(innerTag, 56)\n\n        if (this._proxy) {\n            let dcId = this._proxy.dcId\n            if (this._proxy.test) dcId += 10000\n            if (this._proxy.media) dcId = -dcId\n\n            dv.setInt16(60, dcId, true)\n        }\n\n        const randomRev = bufferToReversed(random, 8, 56)\n\n        let encryptKey = random.subarray(8, 40)\n        const encryptIv = random.subarray(40, 56)\n\n        let decryptKey = randomRev.subarray(0, 32)\n        const decryptIv = randomRev.subarray(32, 48)\n\n        if (this._proxy) {\n            encryptKey = this._crypto.sha256(concatBuffers([encryptKey, this._proxy.secret]))\n            decryptKey = this._crypto.sha256(concatBuffers([decryptKey, this._proxy.secret]))\n        }\n\n        this._encryptor = this._crypto.createAesCtr(encryptKey, encryptIv, true)\n        this._decryptor = this._crypto.createAesCtr(decryptKey, decryptIv, false)\n\n        const encrypted = this._encryptor.process(random)\n        random.set(encrypted.subarray(56, 64), 56)\n\n        return random\n    }\n\n    async encode(packet: Uint8Array): Promise<Uint8Array> {\n        return this._encryptor!.process(await this._inner.encode(packet))\n    }\n\n    feed(data: Uint8Array): void {\n        const dec = this._decryptor!.process(data)\n\n        this._inner.feed(dec)\n    }\n\n    reset(): void {\n        this._inner.reset()\n        this._encryptor?.close?.()\n        this._decryptor?.close?.()\n\n        this._encryptor = undefined\n        this._decryptor = undefined\n    }\n}\n"]}

package/esm/network/transports/obfuscated.test.d.ts

@@ -0,0 +1 @@
+export {};

package/esm/network/transports/obfuscated.test.js

@@ -0,0 +1,129 @@
+import { describe, expect, it, vi } from 'vitest';
+import { defaultTestCryptoProvider, u8HexDecode } from '@mtcute/test';
+import { hexDecodeToBuffer, hexEncode, LogManager } from '../../utils/index.js';
+import { IntermediatePacketCodec } from './intermediate.js';
+import { ObfuscatedPacketCodec } from './obfuscated.js';
+describe('ObfuscatedPacketCodec', () => {
+    const create = async (randomSource, proxy) => {
+        const codec = new ObfuscatedPacketCodec(new IntermediatePacketCodec(), proxy);
+        const crypto = await defaultTestCryptoProvider(randomSource);
+        codec.setup(crypto, new LogManager());
+        return [codec, crypto];
+    };
+    describe('tag', () => {
+        it('should correctly generate random initial payload', async () => {
+            const random = 'ff'.repeat(64);
+            const [codec] = await create(random);
+            const tag = await codec.tag();
+            expect(hexEncode(tag)).toEqual('ff'.repeat(56) + 'fce8ab2203db2bff');
+        });
+        describe('mtproxy', () => {
+            it('should correctly generate random initial payload for prod dc', async () => {
+                const random = 'ff'.repeat(64);
+                const proxy = {
+                    dcId: 1,
+                    secret: new Uint8Array(16),
+                    test: false,
+                    media: false,
+                };
+                const [codec] = await create(random, proxy);
+                const tag = await codec.tag();
+                expect(hexEncode(tag)).toEqual('ff'.repeat(56) + 'ecec4cbda8bb188b');
+            });
+            it('should correctly generate random initial payload for test dc', async () => {
+                const random = 'ff'.repeat(64);
+                const proxy = {
+                    dcId: 1,
+                    secret: new Uint8Array(16),
+                    test: true,
+                    media: false,
+                };
+                const [codec] = await create(random, proxy);
+                const tag = await codec.tag();
+                expect(hexEncode(tag)).toEqual('ff'.repeat(56) + 'ecec4cbdb89c188b');
+            });
+            it('should correctly generate random initial payload for media dc', async () => {
+                const random = 'ff'.repeat(64);
+                const proxy = {
+                    dcId: 1,
+                    secret: new Uint8Array(16),
+                    test: false,
+                    media: true,
+                };
+                const [codec] = await create(random, proxy);
+                const tag = await codec.tag();
+                expect(hexEncode(tag)).toEqual('ff'.repeat(56) + 'ecec4cbd5644188b');
+            });
+        });
+        it.each([
+            ['ef'],
+            ['48454144'],
+            ['504f5354'],
+            ['47455420'],
+            ['4f505449'],
+            ['dddddddd'],
+            ['eeeeeeee'],
+            ['16030102'],
+        ])('should correctly retry for %s prefix', async (prefix) => {
+            const random = prefix + 'ff'.repeat(64 - prefix.length / 2);
+            const [codec] = await create(random);
+            // generating random payload requires 64 bytes of entropy, so
+            // if it asks for more, it means it tried to generate it again
+            await expect(() => codec.tag()).rejects.toThrow('not enough entropy');
+        });
+    });
+    it('should correctly create aes ctr', async () => {
+        const [codec, crypto] = await create();
+        const spyCreateAesCtr = vi.spyOn(crypto, 'createAesCtr');
+        await codec.tag();
+        expect(spyCreateAesCtr).toHaveBeenCalledTimes(2);
+        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(1, u8HexDecode('10b6b4ad6d56ef5df9453f88e6ee6adb6e0544ba635dc6a8a990c9b8b980c343'), u8HexDecode('936b33fa7f97bae025102532233abb26'), true);
+        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(2, u8HexDecode('26bb3a2332251025e0ba977ffa336b9343c380b9b8c990a9a8c65d63ba44056e'), u8HexDecode('db6aeee6883f45f95def566dadb4b610'), false);
+    });
+    it('should correctly create aes ctr for mtproxy', async () => {
+        const proxy = {
+            dcId: 1,
+            secret: hexDecodeToBuffer('00112233445566778899aabbccddeeff'),
+            test: true,
+            media: false,
+        };
+        const [codec, crypto] = await create(undefined, proxy);
+        const spyCreateAesCtr = vi.spyOn(crypto, 'createAesCtr');
+        await codec.tag();
+        expect(spyCreateAesCtr).toHaveBeenCalledTimes(2);
+        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(1, hexDecodeToBuffer('dd03188944590983e28dad14d97d0952389d118af4ffcbdb28d56a6a612ef7a6'), u8HexDecode('936b33fa7f97bae025102532233abb26'), true);
+        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(2, hexDecodeToBuffer('413b8e08021fbb08a2962b6d7187194fe46565c6b329d3bbdfcffd4870c16119'), u8HexDecode('db6aeee6883f45f95def566dadb4b610'), false);
+    });
+    it('should correctly encrypt the underlying codec', async () => {
+        const data = hexDecodeToBuffer('6cfeffff');
+        const msg1 = 'a1020630a410e940';
+        const msg2 = 'f53ff53f371db495';
+        const [codec] = await create();
+        await codec.tag();
+        expect(hexEncode(await codec.encode(data))).toEqual(msg1);
+        expect(hexEncode(await codec.encode(data))).toEqual(msg2);
+    });
+    it('should correctly decrypt the underlying codec', async () => {
+        const msg1 = 'e8027df708ab3b5c';
+        const msg2 = '1854be76d2df4949';
+        const [codec] = await create();
+        await codec.tag();
+        const log = [];
+        codec.on('error', (e) => {
+            log.push(e.toString());
+        });
+        codec.feed(hexDecodeToBuffer(msg1));
+        codec.feed(hexDecodeToBuffer(msg2));
+        await vi.waitFor(() => expect(log).toEqual(['Error: Transport error: 404', 'Error: Transport error: 404']));
+    });
+    it('should correctly reset', async () => {
+        const inner = new IntermediatePacketCodec();
+        const spyInnerReset = vi.spyOn(inner, 'reset');
+        const codec = new ObfuscatedPacketCodec(inner);
+        codec.setup(await defaultTestCryptoProvider(), new LogManager());
+        await codec.tag();
+        codec.reset();
+        expect(spyInnerReset).toHaveBeenCalledTimes(1);
+    });
+});
+//# sourceMappingURL=obfuscated.test.js.map

package/esm/network/transports/obfuscated.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"obfuscated.test.js","sourceRoot":"","sources":["../../../../src/network/transports/obfuscated.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,yBAAyB,EAAE,WAAW,EAAE,MAAM,cAAc,CAAA;AAErE,OAAO,EAAE,iBAAiB,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AAC/E,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAC3D,OAAO,EAAe,qBAAqB,EAAE,MAAM,iBAAiB,CAAA;AAEpE,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACnC,MAAM,MAAM,GAAG,KAAK,EAAE,YAAqB,EAAE,KAAmB,EAAE,EAAE;QAChE,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,uBAAuB,EAAE,EAAE,KAAK,CAAC,CAAA;QAC7E,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAAC,YAAY,CAAC,CAAA;QAC5D,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,UAAU,EAAE,CAAC,CAAA;QAErC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAU,CAAA;IACnC,CAAC,CAAA;IAED,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACjB,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAA;YAEpC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;YAE7B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAC1B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,kBAAkB,CACvC,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;YACrB,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;gBAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;gBAC9B,MAAM,KAAK,GAAgB;oBACvB,IAAI,EAAE,CAAC;oBACP,MAAM,EAAE,IAAI,UAAU,CAAC,EAAE,CAAC;oBAC1B,IAAI,EAAE,KAAK;oBACX,KAAK,EAAE,KAAK;iBACf,CAAA;gBACD,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;gBAE3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;gBAE7B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAC1B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,kBAAkB,CACvC,CAAA;YACL,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;gBAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;gBAC9B,MAAM,KAAK,GAAgB;oBACvB,IAAI,EAAE,CAAC;oBACP,MAAM,EAAE,IAAI,UAAU,CAAC,EAAE,CAAC;oBAC1B,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,KAAK;iBACf,CAAA;gBACD,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;gBAE3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;gBAE7B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAC1B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,kBAAkB,CACvC,CAAA;YACL,CAAC,CAAC,CAAA;YAEF,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;gBAC3E,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;gBAC9B,MAAM,KAAK,GAAgB;oBACvB,IAAI,EAAE,CAAC;oBACP,MAAM,EAAE,IAAI,UAAU,CAAC,EAAE,CAAC;oBAC1B,IAAI,EAAE,KAAK;oBACX,KAAK,EAAE,IAAI;iBACd,CAAA;gBACD,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;gBAE3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;gBAE7B,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAC1B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,kBAAkB,CACvC,CAAA;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,IAAI,CAAC;YACJ,CAAC,IAAI,CAAC;YACN,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;YACZ,CAAC,UAAU,CAAC;SACf,CAAC,CAAC,sCAAsC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACxD,MAAM,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;YAC3D,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,CAAA;YAEpC,6DAA6D;YAC7D,8DAA8D;YAC9D,MAAM,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAA;QACzE,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAEtC,MAAM,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAExD,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAEjB,MAAM,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAChD,MAAM,CAAC,eAAe,CAAC,CAAC,uBAAuB,CAC3C,CAAC,EACD,WAAW,CAAC,kEAAkE,CAAC,EAC/E,WAAW,CAAC,kCAAkC,CAAC,EAC/C,IAAI,CACP,CAAA;QACD,MAAM,CAAC,eAAe,CAAC,CAAC,uBAAuB,CAC3C,CAAC,EACD,WAAW,CAAC,kEAAkE,CAAC,EAC/E,WAAW,CAAC,kCAAkC,CAAC,EAC/C,KAAK,CACR,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,KAAK,GAAgB;YACvB,IAAI,EAAE,CAAC;YACP,MAAM,EAAE,iBAAiB,CAAC,kCAAkC,CAAC;YAC7D,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,KAAK;SACf,CAAA;QACD,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAA;QAEtD,MAAM,eAAe,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;QAExD,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAEjB,MAAM,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAChD,MAAM,CAAC,eAAe,CAAC,CAAC,uBAAuB,CAC3C,CAAC,EACD,iBAAiB,CAAC,kEAAkE,CAAC,EACrF,WAAW,CAAC,kCAAkC,CAAC,EAC/C,IAAI,CACP,CAAA;QACD,MAAM,CAAC,eAAe,CAAC,CAAC,uBAAuB,CAC3C,CAAC,EACD,iBAAiB,CAAC,kEAAkE,CAAC,EACrF,WAAW,CAAC,kCAAkC,CAAC,EAC/C,KAAK,CACR,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,IAAI,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAA;QAC1C,MAAM,IAAI,GAAG,kBAAkB,CAAA;QAC/B,MAAM,IAAI,GAAG,kBAAkB,CAAA;QAE/B,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAE9B,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAEjB,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QACzD,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IAC7D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,IAAI,GAAG,kBAAkB,CAAA;QAC/B,MAAM,IAAI,GAAG,kBAAkB,CAAA;QAE/B,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAE9B,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAEjB,MAAM,GAAG,GAAa,EAAE,CAAA;QAExB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAQ,EAAE,EAAE;YAC3B,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAA;QAC1B,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAA;QACnC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAA;QAEnC,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,6BAA6B,EAAE,6BAA6B,CAAC,CAAC,CAAC,CAAA;IAC/G,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,KAAK,GAAG,IAAI,uBAAuB,EAAE,CAAA;QAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QAE9C,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAA;QAC9C,KAAK,CAAC,KAAK,CAAC,MAAM,yBAAyB,EAAE,EAAE,IAAI,UAAU,EAAE,CAAC,CAAA;QAEhE,MAAM,KAAK,CAAC,GAAG,EAAE,CAAA;QAEjB,KAAK,CAAC,KAAK,EAAE,CAAA;QAEb,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { describe, expect, it, vi } from 'vitest'\n\nimport { defaultTestCryptoProvider, u8HexDecode } from '@mtcute/test'\n\nimport { hexDecodeToBuffer, hexEncode, LogManager } from '../../utils/index.js'\nimport { IntermediatePacketCodec } from './intermediate.js'\nimport { MtProxyInfo, ObfuscatedPacketCodec } from './obfuscated.js'\n\ndescribe('ObfuscatedPacketCodec', () => {\n    const create = async (randomSource?: string, proxy?: MtProxyInfo) => {\n        const codec = new ObfuscatedPacketCodec(new IntermediatePacketCodec(), proxy)\n        const crypto = await defaultTestCryptoProvider(randomSource)\n        codec.setup(crypto, new LogManager())\n\n        return [codec, crypto] as const\n    }\n\n    describe('tag', () => {\n        it('should correctly generate random initial payload', async () => {\n            const random = 'ff'.repeat(64)\n            const [codec] = await create(random)\n\n            const tag = await codec.tag()\n\n            expect(hexEncode(tag)).toEqual(\n                'ff'.repeat(56) + 'fce8ab2203db2bff', // encrypted part\n            )\n        })\n\n        describe('mtproxy', () => {\n            it('should correctly generate random initial payload for prod dc', async () => {\n                const random = 'ff'.repeat(64)\n                const proxy: MtProxyInfo = {\n                    dcId: 1,\n                    secret: new Uint8Array(16),\n                    test: false,\n                    media: false,\n                }\n                const [codec] = await create(random, proxy)\n\n                const tag = await codec.tag()\n\n                expect(hexEncode(tag)).toEqual(\n                    'ff'.repeat(56) + 'ecec4cbda8bb188b', // encrypted part with dcId = 1\n                )\n            })\n\n            it('should correctly generate random initial payload for test dc', async () => {\n                const random = 'ff'.repeat(64)\n                const proxy: MtProxyInfo = {\n                    dcId: 1,\n                    secret: new Uint8Array(16),\n                    test: true,\n                    media: false,\n                }\n                const [codec] = await create(random, proxy)\n\n                const tag = await codec.tag()\n\n                expect(hexEncode(tag)).toEqual(\n                    'ff'.repeat(56) + 'ecec4cbdb89c188b', // encrypted part with dcId = 10001\n                )\n            })\n\n            it('should correctly generate random initial payload for media dc', async () => {\n                const random = 'ff'.repeat(64)\n                const proxy: MtProxyInfo = {\n                    dcId: 1,\n                    secret: new Uint8Array(16),\n                    test: false,\n                    media: true,\n                }\n                const [codec] = await create(random, proxy)\n\n                const tag = await codec.tag()\n\n                expect(hexEncode(tag)).toEqual(\n                    'ff'.repeat(56) + 'ecec4cbd5644188b', // encrypted part with dcId = -1\n                )\n            })\n        })\n\n        it.each([\n            ['ef'],\n            ['48454144'],\n            ['504f5354'],\n            ['47455420'],\n            ['4f505449'],\n            ['dddddddd'],\n            ['eeeeeeee'],\n            ['16030102'],\n        ])('should correctly retry for %s prefix', async (prefix) => {\n            const random = prefix + 'ff'.repeat(64 - prefix.length / 2)\n            const [codec] = await create(random)\n\n            // generating random payload requires 64 bytes of entropy, so\n            // if it asks for more, it means it tried to generate it again\n            await expect(() => codec.tag()).rejects.toThrow('not enough entropy')\n        })\n    })\n\n    it('should correctly create aes ctr', async () => {\n        const [codec, crypto] = await create()\n\n        const spyCreateAesCtr = vi.spyOn(crypto, 'createAesCtr')\n\n        await codec.tag()\n\n        expect(spyCreateAesCtr).toHaveBeenCalledTimes(2)\n        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(\n            1,\n            u8HexDecode('10b6b4ad6d56ef5df9453f88e6ee6adb6e0544ba635dc6a8a990c9b8b980c343'),\n            u8HexDecode('936b33fa7f97bae025102532233abb26'),\n            true,\n        )\n        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(\n            2,\n            u8HexDecode('26bb3a2332251025e0ba977ffa336b9343c380b9b8c990a9a8c65d63ba44056e'),\n            u8HexDecode('db6aeee6883f45f95def566dadb4b610'),\n            false,\n        )\n    })\n\n    it('should correctly create aes ctr for mtproxy', async () => {\n        const proxy: MtProxyInfo = {\n            dcId: 1,\n            secret: hexDecodeToBuffer('00112233445566778899aabbccddeeff'),\n            test: true,\n            media: false,\n        }\n        const [codec, crypto] = await create(undefined, proxy)\n\n        const spyCreateAesCtr = vi.spyOn(crypto, 'createAesCtr')\n\n        await codec.tag()\n\n        expect(spyCreateAesCtr).toHaveBeenCalledTimes(2)\n        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(\n            1,\n            hexDecodeToBuffer('dd03188944590983e28dad14d97d0952389d118af4ffcbdb28d56a6a612ef7a6'),\n            u8HexDecode('936b33fa7f97bae025102532233abb26'),\n            true,\n        )\n        expect(spyCreateAesCtr).toHaveBeenNthCalledWith(\n            2,\n            hexDecodeToBuffer('413b8e08021fbb08a2962b6d7187194fe46565c6b329d3bbdfcffd4870c16119'),\n            u8HexDecode('db6aeee6883f45f95def566dadb4b610'),\n            false,\n        )\n    })\n\n    it('should correctly encrypt the underlying codec', async () => {\n        const data = hexDecodeToBuffer('6cfeffff')\n        const msg1 = 'a1020630a410e940'\n        const msg2 = 'f53ff53f371db495'\n\n        const [codec] = await create()\n\n        await codec.tag()\n\n        expect(hexEncode(await codec.encode(data))).toEqual(msg1)\n        expect(hexEncode(await codec.encode(data))).toEqual(msg2)\n    })\n\n    it('should correctly decrypt the underlying codec', async () => {\n        const msg1 = 'e8027df708ab3b5c'\n        const msg2 = '1854be76d2df4949'\n\n        const [codec] = await create()\n\n        await codec.tag()\n\n        const log: string[] = []\n\n        codec.on('error', (e: Error) => {\n            log.push(e.toString())\n        })\n\n        codec.feed(hexDecodeToBuffer(msg1))\n        codec.feed(hexDecodeToBuffer(msg2))\n\n        await vi.waitFor(() => expect(log).toEqual(['Error: Transport error: 404', 'Error: Transport error: 404']))\n    })\n\n    it('should correctly reset', async () => {\n        const inner = new IntermediatePacketCodec()\n        const spyInnerReset = vi.spyOn(inner, 'reset')\n\n        const codec = new ObfuscatedPacketCodec(inner)\n        codec.setup(await defaultTestCryptoProvider(), new LogManager())\n\n        await codec.tag()\n\n        codec.reset()\n\n        expect(spyInnerReset).toHaveBeenCalledTimes(1)\n    })\n})\n"]}

package/esm/network/transports/tcp.js

@@ -97,10 +97,10 @@ export class BaseTcpTransport extends EventEmitter {
             .catch((err) => this.emit('error', err));
     }
     async send(bytes) {
+        const framed = await this._packetCodec.encode(bytes);
         if (this._state !== TransportState.Ready) {
             throw new MtcuteError('Transport is not READY');
         }
-        const framed = await this._packetCodec.encode(bytes);
         return new Promise((resolve, reject) => {
             this._socket.write(framed, (error) => {
                 if (error) {

package/esm/network/transports/tcp.js.map

@@ -1 +1 @@
-{"version":3,"file":"tcp.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AACjC,OAAO,EAAE,OAAO,EAAU,MAAM,KAAK,CAAA;AAIrC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAA;AAEnD,OAAO,EAAoC,cAAc,EAAE,MAAM,eAAe,CAAA;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAE3D;;;GAGG;AACH,MAAM,OAAgB,gBAAiB,SAAQ,YAAY;IAA3D;;QACc,eAAU,GAA0B,IAAI,CAAA;QACxC,WAAM,GAAmB,cAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAkB,IAAI,CAAA;QAMvC,2BAAsB,GAAG,KAAK,CAAA;IA+GlC,CAAC;IA7GW,gBAAgB;QACpB,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,QAAQ,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAA;SAClF;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,qBAAqB,CAAA;SAC1C;IACL,CAAC;IAED,KAAK,CAAC,MAAuB,EAAE,GAAW;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC5B,IAAI,CAAC,gBAAgB,EAAE,CAAA;IAC3B,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,MAAM,CAAA;IACtB,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,EAAkB,EAAE,QAAiB;QACzC,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,WAAW,CAAC,uBAAuB,CAAC,CAAA;SACjD;QAED,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAA;YACjD,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;YAC/D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAA;YAClE,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAA;SACrC;QAED,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,UAAU,CAAA;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAEvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAA;QAEtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAE5E,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACrD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IACnD,CAAC;IAED,KAAK;QACD,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI;YAAE,OAAM;QAC/C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAElC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAClB,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,IAAI,CAAA;QACjC,IAAI,CAAC,OAAQ,CAAC,kBAAkB,EAAE,CAAA;QAClC,IAAI,CAAC,OAAQ,CAAC,OAAO,EAAE,CAAA;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;IAC7B,CAAC;IAED,WAAW,CAAC,KAAY;QACpB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,CAAC;IAED,aAAa;QACT,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAE1B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;aACnC,IAAI,CAAC,CAAC,cAAc,EAAE,EAAE;YACrB,IAAI,cAAc,CAAC,MAAM,EAAE;gBACvB,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,GAAG,EAAE,EAAE;oBACxC,IAAI,GAAG,EAAE;wBACL,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;wBAChE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;wBAClB,IAAI,CAAC,KAAK,EAAE,CAAA;qBACf;yBAAM;wBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;wBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;qBACrB;gBACL,CAAC,CAAC,CAAA;aACL;iBAAM;gBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;gBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;aACrB;QACL,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IAChD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAiB;QACxB,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,WAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAClC,IAAI,KAAK,EAAE;oBACP,MAAM,CAAC,KAAK,CAAC,CAAA;iBAChB;qBAAM;oBACH,OAAO,EAAE,CAAA;iBACZ;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;CACJ;AAED,MAAM,OAAO,YAAa,SAAQ,gBAAgB;IAAlD;;QACI,iBAAY,GAAG,IAAI,uBAAuB,EAAE,CAAA;IAChD,CAAC;CAAA","sourcesContent":["import EventEmitter from 'events'\nimport { connect, Socket } from 'net'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MtcuteError } from '../../types/errors.js'\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\nimport { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js'\nimport { IntermediatePacketCodec } from './intermediate.js'\n\n/**\n * Base for TCP transports.\n * Subclasses must provide packet codec in `_packetCodec` property\n */\nexport abstract class BaseTcpTransport extends EventEmitter implements ITelegramTransport {\n    protected _currentDc: tl.RawDcOption | null = null\n    protected _state: TransportState = TransportState.Idle\n    protected _socket: Socket | null = null\n\n    abstract _packetCodec: IPacketCodec\n    protected _crypto!: ICryptoProvider\n    protected log!: Logger\n\n    packetCodecInitialized = false\n\n    private _updateLogPrefix() {\n        if (this._currentDc) {\n            this.log.prefix = `[TCP:${this._currentDc.ipAddress}:${this._currentDc.port}] `\n        } else {\n            this.log.prefix = '[TCP:disconnected] '\n        }\n    }\n\n    setup(crypto: ICryptoProvider, log: Logger): void {\n        this._crypto = crypto\n        this.log = log.create('tcp')\n        this._updateLogPrefix()\n    }\n\n    state(): TransportState {\n        return this._state\n    }\n\n    currentDc(): tl.RawDcOption | null {\n        return this._currentDc\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-unused-vars\n    connect(dc: tl.RawDcOption, testMode: boolean): void {\n        if (this._state !== TransportState.Idle) {\n            throw new MtcuteError('Transport is not IDLE')\n        }\n\n        if (!this.packetCodecInitialized) {\n            this._packetCodec.setup?.(this._crypto, this.log)\n            this._packetCodec.on('error', (err) => this.emit('error', err))\n            this._packetCodec.on('packet', (buf) => this.emit('message', buf))\n            this.packetCodecInitialized = true\n        }\n\n        this._state = TransportState.Connecting\n        this._currentDc = dc\n        this._updateLogPrefix()\n\n        this.log.debug('connecting to %j', dc)\n\n        this._socket = connect(dc.port, dc.ipAddress, this.handleConnect.bind(this))\n\n        this._socket.on('data', (data) => {\n            this._packetCodec.feed(data)\n        })\n        this._socket.on('error', this.handleError.bind(this))\n        this._socket.on('close', this.close.bind(this))\n    }\n\n    close(): void {\n        if (this._state === TransportState.Idle) return\n        this.log.info('connection closed')\n\n        this.emit('close')\n        this._state = TransportState.Idle\n        this._socket!.removeAllListeners()\n        this._socket!.destroy()\n        this._socket = null\n        this._currentDc = null\n        this._packetCodec.reset()\n    }\n\n    handleError(error: Error): void {\n        this.log.error('error: %s', error.stack)\n        this.emit('error', error)\n    }\n\n    handleConnect(): void {\n        this.log.info('connected')\n\n        Promise.resolve(this._packetCodec.tag())\n            .then((initialMessage) => {\n                if (initialMessage.length) {\n                    this._socket!.write(initialMessage, (err) => {\n                        if (err) {\n                            this.log.error('failed to write initial message: %s', err.stack)\n                            this.emit('error')\n                            this.close()\n                        } else {\n                            this._state = TransportState.Ready\n                            this.emit('ready')\n                        }\n                    })\n                } else {\n                    this._state = TransportState.Ready\n                    this.emit('ready')\n                }\n            })\n            .catch((err) => this.emit('error', err))\n    }\n\n    async send(bytes: Uint8Array): Promise<void> {\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\n\n        const framed = await this._packetCodec.encode(bytes)\n\n        return new Promise((resolve, reject) => {\n            this._socket!.write(framed, (error) => {\n                if (error) {\n                    reject(error)\n                } else {\n                    resolve()\n                }\n            })\n        })\n    }\n}\n\nexport class TcpTransport extends BaseTcpTransport {\n    _packetCodec = new IntermediatePacketCodec()\n}\n"]}
+{"version":3,"file":"tcp.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.ts"],"names":[],"mappings":"AAAA,OAAO,YAAY,MAAM,QAAQ,CAAA;AACjC,OAAO,EAAE,OAAO,EAAU,MAAM,KAAK,CAAA;AAIrC,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAA;AAEnD,OAAO,EAAoC,cAAc,EAAE,MAAM,eAAe,CAAA;AAChF,OAAO,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAA;AAE3D;;;GAGG;AACH,MAAM,OAAgB,gBAAiB,SAAQ,YAAY;IAA3D;;QACc,eAAU,GAA0B,IAAI,CAAA;QACxC,WAAM,GAAmB,cAAc,CAAC,IAAI,CAAA;QAC5C,YAAO,GAAkB,IAAI,CAAA;QAMvC,2BAAsB,GAAG,KAAK,CAAA;IA+GlC,CAAC;IA7GW,gBAAgB;QACpB,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,QAAQ,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAA;SAClF;aAAM;YACH,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,qBAAqB,CAAA;SAC1C;IACL,CAAC;IAED,KAAK,CAAC,MAAuB,EAAE,GAAW;QACtC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC5B,IAAI,CAAC,gBAAgB,EAAE,CAAA;IAC3B,CAAC;IAED,KAAK;QACD,OAAO,IAAI,CAAC,MAAM,CAAA;IACtB,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,UAAU,CAAA;IAC1B,CAAC;IAED,6DAA6D;IAC7D,OAAO,CAAC,EAAkB,EAAE,QAAiB;QACzC,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI,EAAE;YACrC,MAAM,IAAI,WAAW,CAAC,uBAAuB,CAAC,CAAA;SACjD;QAED,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,CAAA;YACjD,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;YAC/D,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAA;YAClE,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAA;SACrC;QAED,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,UAAU,CAAA;QACvC,IAAI,CAAC,UAAU,GAAG,EAAE,CAAA;QACpB,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAEvB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAA;QAEtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAE5E,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QACrD,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IACnD,CAAC;IAED,KAAK;QACD,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,IAAI;YAAE,OAAM;QAC/C,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAElC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAClB,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,IAAI,CAAA;QACjC,IAAI,CAAC,OAAQ,CAAC,kBAAkB,EAAE,CAAA;QAClC,IAAI,CAAC,OAAQ,CAAC,OAAO,EAAE,CAAA;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAA;QACtB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;IAC7B,CAAC;IAED,WAAW,CAAC,KAAY;QACpB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,CAAA;QACxC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IAC7B,CAAC;IAED,aAAa;QACT,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAE1B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;aACnC,IAAI,CAAC,CAAC,cAAc,EAAE,EAAE;YACrB,IAAI,cAAc,CAAC,MAAM,EAAE;gBACvB,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,GAAG,EAAE,EAAE;oBACxC,IAAI,GAAG,EAAE;wBACL,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAA;wBAChE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;wBAClB,IAAI,CAAC,KAAK,EAAE,CAAA;qBACf;yBAAM;wBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;wBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;qBACrB;gBACL,CAAC,CAAC,CAAA;aACL;iBAAM;gBACH,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,KAAK,CAAA;gBAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;aACrB;QACL,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAA;IAChD,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAiB;QACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAEpD,IAAI,IAAI,CAAC,MAAM,KAAK,cAAc,CAAC,KAAK,EAAE;YACtC,MAAM,IAAI,WAAW,CAAC,wBAAwB,CAAC,CAAA;SAClD;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnC,IAAI,CAAC,OAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAClC,IAAI,KAAK,EAAE;oBACP,MAAM,CAAC,KAAK,CAAC,CAAA;iBAChB;qBAAM;oBACH,OAAO,EAAE,CAAA;iBACZ;YACL,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;IACN,CAAC;CACJ;AAED,MAAM,OAAO,YAAa,SAAQ,gBAAgB;IAAlD;;QACI,iBAAY,GAAG,IAAI,uBAAuB,EAAE,CAAA;IAChD,CAAC;CAAA","sourcesContent":["import EventEmitter from 'events'\nimport { connect, Socket } from 'net'\n\nimport { tl } from '@mtcute/tl'\n\nimport { MtcuteError } from '../../types/errors.js'\nimport { ICryptoProvider, Logger } from '../../utils/index.js'\nimport { IPacketCodec, ITelegramTransport, TransportState } from './abstract.js'\nimport { IntermediatePacketCodec } from './intermediate.js'\n\n/**\n * Base for TCP transports.\n * Subclasses must provide packet codec in `_packetCodec` property\n */\nexport abstract class BaseTcpTransport extends EventEmitter implements ITelegramTransport {\n    protected _currentDc: tl.RawDcOption | null = null\n    protected _state: TransportState = TransportState.Idle\n    protected _socket: Socket | null = null\n\n    abstract _packetCodec: IPacketCodec\n    protected _crypto!: ICryptoProvider\n    protected log!: Logger\n\n    packetCodecInitialized = false\n\n    private _updateLogPrefix() {\n        if (this._currentDc) {\n            this.log.prefix = `[TCP:${this._currentDc.ipAddress}:${this._currentDc.port}] `\n        } else {\n            this.log.prefix = '[TCP:disconnected] '\n        }\n    }\n\n    setup(crypto: ICryptoProvider, log: Logger): void {\n        this._crypto = crypto\n        this.log = log.create('tcp')\n        this._updateLogPrefix()\n    }\n\n    state(): TransportState {\n        return this._state\n    }\n\n    currentDc(): tl.RawDcOption | null {\n        return this._currentDc\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-unused-vars\n    connect(dc: tl.RawDcOption, testMode: boolean): void {\n        if (this._state !== TransportState.Idle) {\n            throw new MtcuteError('Transport is not IDLE')\n        }\n\n        if (!this.packetCodecInitialized) {\n            this._packetCodec.setup?.(this._crypto, this.log)\n            this._packetCodec.on('error', (err) => this.emit('error', err))\n            this._packetCodec.on('packet', (buf) => this.emit('message', buf))\n            this.packetCodecInitialized = true\n        }\n\n        this._state = TransportState.Connecting\n        this._currentDc = dc\n        this._updateLogPrefix()\n\n        this.log.debug('connecting to %j', dc)\n\n        this._socket = connect(dc.port, dc.ipAddress, this.handleConnect.bind(this))\n\n        this._socket.on('data', (data) => {\n            this._packetCodec.feed(data)\n        })\n        this._socket.on('error', this.handleError.bind(this))\n        this._socket.on('close', this.close.bind(this))\n    }\n\n    close(): void {\n        if (this._state === TransportState.Idle) return\n        this.log.info('connection closed')\n\n        this.emit('close')\n        this._state = TransportState.Idle\n        this._socket!.removeAllListeners()\n        this._socket!.destroy()\n        this._socket = null\n        this._currentDc = null\n        this._packetCodec.reset()\n    }\n\n    handleError(error: Error): void {\n        this.log.error('error: %s', error.stack)\n        this.emit('error', error)\n    }\n\n    handleConnect(): void {\n        this.log.info('connected')\n\n        Promise.resolve(this._packetCodec.tag())\n            .then((initialMessage) => {\n                if (initialMessage.length) {\n                    this._socket!.write(initialMessage, (err) => {\n                        if (err) {\n                            this.log.error('failed to write initial message: %s', err.stack)\n                            this.emit('error')\n                            this.close()\n                        } else {\n                            this._state = TransportState.Ready\n                            this.emit('ready')\n                        }\n                    })\n                } else {\n                    this._state = TransportState.Ready\n                    this.emit('ready')\n                }\n            })\n            .catch((err) => this.emit('error', err))\n    }\n\n    async send(bytes: Uint8Array): Promise<void> {\n        const framed = await this._packetCodec.encode(bytes)\n\n        if (this._state !== TransportState.Ready) {\n            throw new MtcuteError('Transport is not READY')\n        }\n\n        return new Promise((resolve, reject) => {\n            this._socket!.write(framed, (error) => {\n                if (error) {\n                    reject(error)\n                } else {\n                    resolve()\n                }\n            })\n        })\n    }\n}\n\nexport class TcpTransport extends BaseTcpTransport {\n    _packetCodec = new IntermediatePacketCodec()\n}\n"]}

package/esm/network/transports/tcp.test.d.ts

@@ -0,0 +1 @@
+export {};

package/esm/network/transports/tcp.test.js

@@ -0,0 +1,95 @@
+import { describe, expect, it, vi } from 'vitest';
+import { defaultTestCryptoProvider, u8HexDecode } from '@mtcute/test';
+import { defaultProductionDc, hexDecodeToBuffer, LogManager } from '../../utils/index.js';
+import { TransportState } from './abstract.js';
+import { TcpTransport } from './tcp.js';
+vi.mock('net', () => ({
+    connect: vi.fn().mockImplementation((port, ip, cb) => {
+        cb();
+        return {
+            on: vi.fn(),
+            write: vi.fn().mockImplementation((data, cb) => {
+                cb();
+            }),
+            end: vi.fn(),
+            removeAllListeners: vi.fn(),
+            destroy: vi.fn(),
+        };
+    }),
+}));
+describe('TcpTransport', async () => {
+    const net = await import('net');
+    const connect = vi.mocked(net.connect);
+    const getLastSocket = () => {
+        return connect.mock.results[connect.mock.results.length - 1].value;
+    };
+    const create = async () => {
+        const transport = new TcpTransport();
+        const logger = new LogManager();
+        logger.level = 0;
+        transport.setup(await defaultTestCryptoProvider(), logger);
+        return transport;
+    };
+    it('should initiate a tcp connection to the given dc', async () => {
+        const t = await create();
+        t.connect(defaultProductionDc.main, false);
+        expect(connect).toHaveBeenCalledOnce();
+        expect(connect).toHaveBeenCalledWith(defaultProductionDc.main.port, defaultProductionDc.main.ipAddress, expect.any(Function));
+        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready));
+    });
+    it('should set up event handlers', async () => {
+        const t = await create();
+        t.connect(defaultProductionDc.main, false);
+        const socket = getLastSocket();
+        expect(socket.on).toHaveBeenCalledTimes(3);
+        expect(socket.on).toHaveBeenCalledWith('data', expect.any(Function));
+        expect(socket.on).toHaveBeenCalledWith('error', expect.any(Function));
+        expect(socket.on).toHaveBeenCalledWith('close', expect.any(Function));
+    });
+    it('should write packet codec tag once connected', async () => {
+        const t = await create();
+        t.connect(defaultProductionDc.main, false);
+        const socket = getLastSocket();
+        await vi.waitFor(() => expect(socket.write).toHaveBeenCalledWith(u8HexDecode('eeeeeeee'), // intermediate
+        expect.any(Function)));
+    });
+    it('should write to the underlying socket', async () => {
+        const t = await create();
+        t.connect(defaultProductionDc.main, false);
+        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready));
+        await t.send(hexDecodeToBuffer('00010203040506070809'));
+        const socket = getLastSocket();
+        expect(socket.write).toHaveBeenCalledWith(u8HexDecode('0a00000000010203040506070809'), expect.any(Function));
+    });
+    it('should correctly close', async () => {
+        const t = await create();
+        t.connect(defaultProductionDc.main, false);
+        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready));
+        t.close();
+        const socket = getLastSocket();
+        expect(socket.removeAllListeners).toHaveBeenCalledOnce();
+        expect(socket.destroy).toHaveBeenCalledOnce();
+    });
+    it('should feed data to the packet codec', async () => {
+        const t = await create();
+        const codec = t._packetCodec;
+        const spyFeed = vi.spyOn(codec, 'feed');
+        t.connect(defaultProductionDc.main, false);
+        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready));
+        const socket = getLastSocket();
+        const onDataCall = socket.on.mock.calls.find((c) => c[0] === 'data');
+        onDataCall[1](u8HexDecode('00010203040506070809'));
+        expect(spyFeed).toHaveBeenCalledWith(u8HexDecode('00010203040506070809'));
+    });
+    it('should propagate errors', async () => {
+        const t = await create();
+        const spyEmit = vi.spyOn(t, 'emit').mockImplementation(() => true);
+        t.connect(defaultProductionDc.main, false);
+        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready));
+        const socket = getLastSocket();
+        const onErrorCall = socket.on.mock.calls.find((c) => c[0] === 'error');
+        onErrorCall[1](new Error('test error'));
+        expect(spyEmit).toHaveBeenCalledWith('error', new Error('test error'));
+    });
+});
+//# sourceMappingURL=tcp.test.js.map

package/esm/network/transports/tcp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tcp.test.js","sourceRoot":"","sources":["../../../../src/network/transports/tcp.test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAgB,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE/D,OAAO,EAAE,yBAAyB,EAAE,WAAW,EAAE,MAAM,cAAc,CAAA;AAErE,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AACzF,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAA;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAA;AAEvC,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAClB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAY,EAAE,EAAU,EAAE,EAAc,EAAE,EAAE;QAC7E,EAAE,EAAE,CAAA;QAEJ,OAAO;YACH,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;YACX,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAgB,EAAE,EAAc,EAAE,EAAE;gBACnE,EAAE,EAAE,CAAA;YACR,CAAC,CAAC;YACF,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACnB,CAAA;IACL,CAAC,CAAC;CACL,CAAC,CAAC,CAAA;AAEH,QAAQ,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;IAChC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAA;IAC/B,MAAM,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;IAEtC,MAAM,aAAa,GAAG,GAAG,EAAE;QACvB,OAAO,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAA6B,CAAA;IAC9F,CAAC,CAAA;IAED,MAAM,MAAM,GAAG,KAAK,IAAI,EAAE;QACtB,MAAM,SAAS,GAAG,IAAI,YAAY,EAAE,CAAA;QACpC,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAA;QAC/B,MAAM,CAAC,KAAK,GAAG,CAAC,CAAA;QAChB,SAAS,CAAC,KAAK,CAAC,MAAM,yBAAyB,EAAE,EAAE,MAAM,CAAC,CAAA;QAE1D,OAAO,SAAS,CAAA;IACpB,CAAC,CAAA;IAED,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAE1C,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAA;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAChC,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAC7B,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAClC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACvB,CAAA;QACD,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAE1C,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAC1C,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QACpE,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QACrE,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;IACzE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAE1C,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAClB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACrC,WAAW,CAAC,UAAU,CAAC,EAAE,eAAe;QACxC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACvB,CACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;QAEvE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,CAAC,CAAA;QAEvD,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,8BAA8B,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;IAChH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;QAEvE,CAAC,CAAC,KAAK,EAAE,CAAA;QAET,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,EAAE,CAAA;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,EAAE,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QACxB,MAAM,KAAK,GAAG,CAAC,CAAC,YAAY,CAAA;QAE5B,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QAEvC,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;QAEvE,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,UAAU,GAAG,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAc,CAAC,CAAC,CAAC,KAAK,MAAM,CAGhF,CAAA;QACD,UAAU,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAA;QAElD,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,CAAC,GAAG,MAAM,MAAM,EAAE,CAAA;QAExB,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAA;QAElE,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAA;QAEvE,MAAM,MAAM,GAAG,aAAa,EAAE,CAAA;QAE9B,MAAM,WAAW,GAAG,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAc,CAAC,CAAC,CAAC,KAAK,OAAO,CAGlF,CAAA;QACD,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;QAEvC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA","sourcesContent":["import { Socket } from 'net'\nimport { describe, expect, it, MockedObject, vi } from 'vitest'\n\nimport { defaultTestCryptoProvider, u8HexDecode } from '@mtcute/test'\n\nimport { defaultProductionDc, hexDecodeToBuffer, LogManager } from '../../utils/index.js'\nimport { TransportState } from './abstract.js'\nimport { TcpTransport } from './tcp.js'\n\nvi.mock('net', () => ({\n    connect: vi.fn().mockImplementation((port: number, ip: string, cb: () => void) => {\n        cb()\n\n        return {\n            on: vi.fn(),\n            write: vi.fn().mockImplementation((data: Uint8Array, cb: () => void) => {\n                cb()\n            }),\n            end: vi.fn(),\n            removeAllListeners: vi.fn(),\n            destroy: vi.fn(),\n        }\n    }),\n}))\n\ndescribe('TcpTransport', async () => {\n    const net = await import('net')\n    const connect = vi.mocked(net.connect)\n\n    const getLastSocket = () => {\n        return connect.mock.results[connect.mock.results.length - 1].value as MockedObject<Socket>\n    }\n\n    const create = async () => {\n        const transport = new TcpTransport()\n        const logger = new LogManager()\n        logger.level = 0\n        transport.setup(await defaultTestCryptoProvider(), logger)\n\n        return transport\n    }\n\n    it('should initiate a tcp connection to the given dc', async () => {\n        const t = await create()\n\n        t.connect(defaultProductionDc.main, false)\n\n        expect(connect).toHaveBeenCalledOnce()\n        expect(connect).toHaveBeenCalledWith(\n            defaultProductionDc.main.port,\n            defaultProductionDc.main.ipAddress,\n            expect.any(Function),\n        )\n        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready))\n    })\n\n    it('should set up event handlers', async () => {\n        const t = await create()\n\n        t.connect(defaultProductionDc.main, false)\n\n        const socket = getLastSocket()\n\n        expect(socket.on).toHaveBeenCalledTimes(3)\n        expect(socket.on).toHaveBeenCalledWith('data', expect.any(Function))\n        expect(socket.on).toHaveBeenCalledWith('error', expect.any(Function))\n        expect(socket.on).toHaveBeenCalledWith('close', expect.any(Function))\n    })\n\n    it('should write packet codec tag once connected', async () => {\n        const t = await create()\n\n        t.connect(defaultProductionDc.main, false)\n\n        const socket = getLastSocket()\n\n        await vi.waitFor(() =>\n            expect(socket.write).toHaveBeenCalledWith(\n                u8HexDecode('eeeeeeee'), // intermediate\n                expect.any(Function),\n            ),\n        )\n    })\n\n    it('should write to the underlying socket', async () => {\n        const t = await create()\n\n        t.connect(defaultProductionDc.main, false)\n        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready))\n\n        await t.send(hexDecodeToBuffer('00010203040506070809'))\n\n        const socket = getLastSocket()\n\n        expect(socket.write).toHaveBeenCalledWith(u8HexDecode('0a00000000010203040506070809'), expect.any(Function))\n    })\n\n    it('should correctly close', async () => {\n        const t = await create()\n\n        t.connect(defaultProductionDc.main, false)\n        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready))\n\n        t.close()\n\n        const socket = getLastSocket()\n\n        expect(socket.removeAllListeners).toHaveBeenCalledOnce()\n        expect(socket.destroy).toHaveBeenCalledOnce()\n    })\n\n    it('should feed data to the packet codec', async () => {\n        const t = await create()\n        const codec = t._packetCodec\n\n        const spyFeed = vi.spyOn(codec, 'feed')\n\n        t.connect(defaultProductionDc.main, false)\n        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready))\n\n        const socket = getLastSocket()\n\n        const onDataCall = socket.on.mock.calls.find((c) => (c as string[])[0] === 'data') as unknown as [\n            string,\n            (data: Uint8Array) => void,\n        ]\n        onDataCall[1](u8HexDecode('00010203040506070809'))\n\n        expect(spyFeed).toHaveBeenCalledWith(u8HexDecode('00010203040506070809'))\n    })\n\n    it('should propagate errors', async () => {\n        const t = await create()\n\n        const spyEmit = vi.spyOn(t, 'emit').mockImplementation(() => true)\n\n        t.connect(defaultProductionDc.main, false)\n        await vi.waitFor(() => expect(t.state()).toEqual(TransportState.Ready))\n\n        const socket = getLastSocket()\n\n        const onErrorCall = socket.on.mock.calls.find((c) => (c as string[])[0] === 'error') as unknown as [\n            string,\n            (error: Error) => void,\n        ]\n        onErrorCall[1](new Error('test error'))\n\n        expect(spyEmit).toHaveBeenCalledWith('error', new Error('test error'))\n    })\n})\n"]}

package/esm/network/transports/websocket.test.d.ts

@@ -0,0 +1 @@
+export {};