Add socks proxy

envs/node/index.js

@@ -6,8 +6,8 @@ const Transport = require('./transport');
 const getRandomBytes = require('./get-random-bytes');
 const getLocalStorage = require('./get-local-storage');
 
-function createTransport(dc, crypto) {
-  return new Transport(dc, crypto);
+function createTransport(dc, crypto, proxy) {
+  return new Transport(dc, crypto, proxy);
 }
 
 const MTProto = makeMTProto({

envs/node/transport.js

@@ -1,14 +1,16 @@
 const net = require('net');
+const SocksClient = require('socks').SocksClient;
 const Obfuscated = require('../../src/transport/obfuscated');
 const baseDebug = require('../../src/utils/common/base-debug');
 
 class Transport extends Obfuscated {
-  constructor(dc, crypto) {
+  constructor(dc, crypto, proxy) {
     super();
 
     this.dc = dc;
     this.debug = baseDebug.extend(`transport-${this.dc.id}`);
     this.crypto = crypto;
+    this.proxy = proxy;
 
     this.connect();
   }
@@ -17,14 +19,32 @@ class Transport extends Obfuscated {
     return this.socket.writable;
   }
 
-  connect() {
+  async connect() {
     this.stream = new Uint8Array();
-
-    this.socket = net.connect(
-      this.dc.port,
-      this.dc.ip,
-      this.handleConnect.bind(this)
-    );
+    if (this.proxy && this.proxy.host && this.proxy.port && this.proxy.type) {
+      let options = {
+        proxy: {
+          host: this.proxy.host, // ipv4 or ipv6 or hostname
+          port: this.proxy.port,
+          type: this.proxy.type // Proxy version (4 or 5)
+        },
+        command: 'connect', // SOCKS command (createConnection factory function only supports the connect command
+        destination: {
+          host: this.dc.ip,
+          port: this.dc.port
+        }
+      };
+      const proxyClient = await SocksClient.createConnection(options)
+      this.socket = proxyClient.socket;
+      this.handleConnect()
+    } else {
+      this.socket = net.connect(
+        this.dc.port,
+        this.dc.ip,
+        this.handleConnect.bind(this)
+      );
+    }
+
 
     this.socket.on('data', this.handleData.bind(this));
     this.socket.on('error', this.handleError.bind(this));

package.json

@@ -49,7 +49,8 @@
     "events": "3.3.0",
     "leemon": "6.2.0",
     "lodash.debounce": "4.0.8",
-    "pako": "2.0.4"
+    "pako": "2.0.4",
+    "socks": "^2.6.1"
   },
   "devDependencies": {
     "jest": "27.4.5",

src/index.js

@@ -75,10 +75,11 @@ function makeMTProto(envMethods) {
 
   return class {
     constructor(options) {
-      const { api_id, api_hash, storageOptions } = options;
+      const { api_id, api_hash, storageOptions, proxy } = options;
 
       this.api_id = api_id;
       this.api_hash = api_hash;
+      this.proxy = proxy;
 
       this.initConnectionParams = {};
 
@@ -166,7 +167,7 @@ function makeMTProto(envMethods) {
         return;
       }
 
-      const transport = this.envMethods.createTransport(dc, this.crypto);
+      const transport = this.envMethods.createTransport(dc, this.crypto, this.proxy);
 
       const rpc = new RPC({
         dc,