ddan-js 2.6.27 → 2.6.29

package/bin/lib/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.dNode = exports.dLogger = exports.dTracker = exports.dJoker = exports.dStore = exports.dCdn = exports.dMini = exports.dWeb = exports.dHook = exports.dUtil = void 0;
+exports.Mapping = exports.KValue = exports.dNode = exports.dLogger = exports.dTracker = exports.dJoker = exports.dStore = exports.dCdn = exports.dMini = exports.dWeb = exports.dHook = exports.dUtil = void 0;
 const gbk_1 = require("./modules/gbk");
 const time_1 = require("./modules/time");
 const hook_1 = require("./modules/hook");
@@ -31,6 +31,10 @@ const tracker_1 = require("./class/tracker");
 const rsa_1 = require("./modules/rsa");
 const node_1 = require("./modules/node");
 const browser_1 = require("./modules/browser");
+const kvalue_1 = require("./class/kvalue");
+exports.KValue = kvalue_1.default;
+const mapping_1 = require("./class/mapping");
+exports.Mapping = mapping_1.default;
 const dUtil = {
     gbk: gbk_1.default,
     math: math_1.default,
@@ -87,6 +91,8 @@ exports.default = {
     rule: rule_1.default,
     regex: regex_1.default,
     convert: convert_1.default,
+    KValue: kvalue_1.default,
+    Mapping: mapping_1.default,
     Event: event_1.default,
     Http: http_1.default,
     Store: store_1.default,
@@ -98,5 +104,5 @@ exports.default = {
     tracker: tracker_1.default,
     logger: logger_1.default,
     fetch: fetch_1.default,
-    rsa: rsa_1.default
+    rsa: rsa_1.default,
 };

package/bin/lib/modules/node/socks5.js

@@ -4,12 +4,16 @@ exports.Socks5 = void 0;
 const net_1 = require("net");
 const socks_1 = require("socks");
 const uuid_1 = require("../crypto/uuid");
+const mapping_1 = require("../../class/mapping");
+// interface IConnectionPool {
+//   [key: string]: Socket // Key为目标地址:端口组合
+// }
 class Socks5 {
     upstreamProxy;
     server = null;
     clientSockets = new Set();
     allowedDomains;
-    connectionPool = {};
+    connectionPool;
     systemProxy = null;
     useSystemProxy = false;
     __debug = false;
@@ -20,6 +24,7 @@ class Socks5 {
         this.allowedDomains = new Set(allowedDomains);
         this.__debug = debug;
         this.__uuid = (0, uuid_1.default)();
+        this.connectionPool = new mapping_1.default();
     }
     get id() {
         return this.__uuid;
@@ -112,6 +117,7 @@ class Socks5 {
             clientSocket?.destroy();
         });
         try {
+            this.connectionPool.clean(100, (kv) => kv?.destroy());
             await this.performHandshake(clientSocket);
             const destination = await this.parseClientRequest(clientSocket);
             if (this.isAllowedDomain(destination.addr)) {
@@ -178,10 +184,15 @@ class Socks5 {
     getPoolConnection(key) {
         if (!key)
             return undefined;
+        const connection = this.connectionPool.get(key);
         // 如果已有连接在池中，复用
         if (this.connectionPool[key] && !this.connectionPool[key].destroyed) {
             return this.connectionPool[key];
         }
+        // 如果已有连接在池中，复用
+        if (connection && !connection.destroyed) {
+            return connection;
+        }
         return undefined;
     }
     // 上游代理连接
@@ -200,9 +211,11 @@ class Socks5 {
         };
         try {
             const { socket: upstreamSocket } = await socks_1.SocksClient.createConnection(options);
-            this.connectionPool[destKey] = upstreamSocket;
+            // this.connectionPool[destKey] = upstreamSocket
+            this.connectionPool.set(destKey, upstreamSocket);
             upstreamSocket.on('close', () => {
-                delete this.connectionPool[destKey];
+                // delete this.connectionPool[destKey]
+                this.connectionPool.delete(destKey);
             });
             return upstreamSocket;
         }
@@ -226,9 +239,11 @@ class Socks5 {
         };
         try {
             const { socket: upstreamSocket } = await socks_1.SocksClient.createConnection(options);
-            this.connectionPool[destKey] = upstreamSocket;
+            // this.connectionPool[destKey] = upstreamSocket
+            this.connectionPool.set(destKey, upstreamSocket);
             upstreamSocket.on('close', () => {
-                delete this.connectionPool[destKey];
+                // delete this.connectionPool[destKey]
+                this.connectionPool.delete(destKey);
             });
             return upstreamSocket;
         }
@@ -246,6 +261,12 @@ class Socks5 {
             const localSocket = net_1.default.createConnection(destination.port, destination.addr, () => {
                 resolve(localSocket);
             });
+            // this.connectionPool[destKey] = localSocket
+            this.connectionPool.set(destKey, localSocket);
+            localSocket.on('close', () => {
+                // delete this.connectionPool[destKey]
+                this.connectionPool.delete(destKey);
+            });
             localSocket.on('error', (err) => {
                 reject(err);
             });
@@ -255,8 +276,11 @@ class Socks5 {
         clientSocket.write(new Uint8Array([0x05, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]));
         clientSocket.pipe(targetSocket);
         targetSocket.pipe(clientSocket);
+        this.__debug &&
+            console.log(`[socks] setupDataForwarding ${targetSocket.localAddress}:${targetSocket.localPort}`);
         clientSocket.on('close', () => {
-            this.__debug && console.error('[socks] client socket close', targetSocket?.remoteAddress || "");
+            this.__debug &&
+                console.error('[socks] client socket close', targetSocket?.remoteAddress || '');
             targetSocket.end();
         });
         targetSocket.on('close', () => {
@@ -295,14 +319,22 @@ class Socks5 {
                 }
             });
             // 销毁客户端 socket
-            this.clientSockets.forEach((socket) => socket.destroy());
+            this.clientSockets.forEach((socket) => socket?.destroy());
             // 销毁所有连接池中的 socket
-            Object.values(this.connectionPool).forEach((socket) => socket.destroy());
+            // Object.values(this.connectionPool).forEach((socket) => socket?.destroy())
+            Array.from(this.connectionPool.values).forEach((kv) => kv?.value?.destroy());
             this.server = null;
         }
         catch (error) {
             this.__debug && console.error('[socks] close errot:', error);
         }
     }
+    showPool() {
+        Array.from(this.connectionPool.values).forEach((kv) => {
+            if (kv && kv.value) {
+                console.log(`[socks] ${kv.value.remoteAddress} ${kv.value.localAddress}:${kv.value.localPort}`);
+            }
+        });
+    }
 }
 exports.Socks5 = Socks5;

package/bin/types/class/kvalue.d.ts

@@ -0,0 +1,10 @@
+export default class KValue<T = any> {
+    timestamp: number;
+    _key: string;
+    _value: T | undefined;
+    constructor(key: string, value?: T);
+    get key(): string;
+    get value(): T | undefined;
+    set(value?: T): void;
+    updateAt(): void;
+}

package/bin/types/class/mapping.d.ts

@@ -0,0 +1,12 @@
+import KValue from './kvalue';
+export default class Mapping<T = any> {
+    _map: Map<string, KValue<T>>;
+    constructor();
+    get keys(): IterableIterator<string>;
+    get values(): IterableIterator<KValue<T>>;
+    get(key: string): T | undefined;
+    set(key: string, value?: T): void;
+    delete(key: string, cb?: (value?: T, key?: string) => void): void;
+    clear(): void;
+    clean(max?: number, cb?: (value?: T, key?: string) => void): void;
+}

package/bin/types/index.d.ts

@@ -3,6 +3,8 @@ import Http from './modules/http';
 import Event from './class/event';
 import Store from './class/store';
 import Persist from './class/persist';
+import KValue from './class/kvalue';
+import Mapping from './class/mapping';
 declare const dUtil: {
     Event: typeof Event;
     includes: typeof import("./util/includes").default;
@@ -517,7 +519,7 @@ declare const dNode: {
     brotliCompress: typeof import("./modules/node/brotli").brotliCompress;
     brotliDecompress: typeof import("./modules/node/brotli").brotliDecompress;
 };
-export { dUtil, dHook, dWeb, dMini, dCdn, dStore, dJoker, dTracker, dLogger, dNode };
+export { dUtil, dHook, dWeb, dMini, dCdn, dStore, dJoker, dTracker, dLogger, dNode, KValue, Mapping };
 declare const _default: {
     gbk: {
         gbkLength: (str: string) => number;
@@ -851,6 +853,8 @@ declare const _default: {
         toUtf8Bytes: (content: string) => Uint8Array;
         fromUtf8Bytes: (utf8Bytes: Uint8Array) => string;
     };
+    KValue: typeof KValue;
+    Mapping: typeof Mapping;
     Event: typeof Event;
     Http: typeof Http;
     Store: typeof Store;

package/bin/types/modules/node/socks5.d.ts

@@ -35,5 +35,6 @@ export declare class Socks5 {
     private setupDataForwarding;
     private isAllowedDomain;
     close(): void;
+    showPool(): void;
 }
 export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ddan-js",
-  "version": "2.6.27",
+  "version": "2.6.29",
   "description": "",
   "keywords": [
     "ddan-js",