add auto query server

index.d.ts

@@ -35,9 +35,10 @@ export interface ConfBot {
     //即使不用相关API(使用`no_cache=true`)强制刷新数据，超过这个时间后内部也会自动刷新
     internal_cache_life?: number,
 
-    //手动指定ip和port
-    //默认使用msfwifi.3g.qq.com:8080进行连接，若要修改建议优先更改该域名hosts指向而不是手动指定ip
-    //@link https://site.ip138.com/msfwifi.3g.qq.com/ 端口通常以下四个都会开放：80,443,8080,14000
+    /** 自动选择最优服务器(默认开启)，关闭后会一直使用`msfwifi.3g.qq.com`进行连接 */
+    auto_server?: boolean;
+
+    /** 手动指定ip和port，不推荐使用，大多数情况下你应该使用auto_server */
     remote_ip?: string,
     remote_port?: number,
 }

lib/client.js

@@ -117,6 +117,7 @@ class Client extends EventEmitter {
             resend: true,
             reconn_interval: 5,
             internal_cache_life: 3600,
+            auto_server: true,
             data_dir: path.join(require.main ? require.main.path : process.cwd(), "data"),
             ...config
         };

lib/network.js

@@ -1,9 +1,20 @@
 "use strict";
 const net = require("net");
-const { BUF0 } = require("./common");
+const https = require("https");
+const { BUF0, timestamp } = require("./common");
+const tea = require("./algo/tea");
+const jce = require("./algo/jce");
+
+const default_host = "msfwifi.3g.qq.com";
+const default_port = 8080;
+let update_time = 0;
+let host_port = { };
 
 class Network extends net.Socket {
 
+    closed_time = timestamp();
+    host = default_host;
+    port = default_port;
     _data = BUF0;
 
     /**
@@ -21,6 +32,9 @@ class Network extends net.Socket {
             this._data = BUF0;
             if (this.remoteAddress)
                 this.c.logger.mark(`${this.remoteAddress}:${this.remotePort} closed`);
+            if (timestamp() - this.closed_time <= 300)
+                delete host_port[this.remoteAddress];
+            this.closed_time = timestamp();
             this.c.emit("internal.offline");
         });
 
@@ -39,19 +53,85 @@ class Network extends net.Socket {
         });
     }
 
-    join(cb) {
-        let ip = "msfwifi.3g.qq.com", port = 8080;
-        if (net.isIP(this.c.config.remote_ip))
-            ip = this.c.config.remote_ip;
-        if (this.c.config.remote_port > 0 && this.c.config.remote_port < 65536)
-            port = this.c.config.remote_port;
-        this.c.logger.mark(`connecting to ${ip}:${port}`);
+    async join(cb) {
+        if (timestamp() - this.closed_time > 300) {
+            //
+        } else if (net.isIP(this.c.config.remote_ip)) {
+            this.host = this.c.config.remote_ip;
+            this.port = 8080;
+            if (this.c.config.remote_port > 0 && this.c.config.remote_port < 65536)
+                this.host = this.c.config.remote_port;
+        } else if (this.c.config.auto_server) {
+            if (!Object.keys(host_port).length || timestamp() - update_time >= 3600) {
+                this.c.logger.mark("正在探索可用服务器...");
+                try {
+                    await this._queryServerList();
+                    this.host = Object.keys(host_port)[0];
+                    this.port = host_port[this.host];
+                } catch (err) {
+                    this.c.logger.warn("探索服务器失败: " + (err ? err.message : "timeout"));
+                }
+            } else {
+                this.host = Object.keys(host_port)[0];
+                this.port = host_port[this.host];
+            }
+        }
+        this.c.logger.mark(`connecting to ${this.host}:${this.port}`);
         this.removeAllListeners("connect");
-        this.connect(port, ip, () => {
+        this.connect(this.port, this.host, () => {
             this.c.logger.mark(`${this.remoteAddress}:${this.remotePort} connected`);
             cb();
         });
     }
+
+    async _queryServerList() {
+        const key = Buffer.from("F0441F5FF42DA58FDCF7949ABA62D411", "hex");
+        const HttpServerListReq = jce.encodeStruct([
+            null,
+            0, 0, 1, "00000", 100, this.c.apk.subid, this.c.device.imei, 0, 0, 0,
+            0, 0, 0, 1
+        ]);
+        const extra = {
+            service: "ConfigHttp",
+            method: "HttpServerListReq",
+        };
+        let body = jce.encodeWrapper({ HttpServerListReq }, extra);
+        const len = Buffer.alloc(4);
+        len.writeUInt32BE(body.length + 4);
+        body = Buffer.concat([len, body]);
+        body = tea.encrypt(body, key);
+        return await new Promise((resolve, reject) => {
+            const id = setTimeout(reject, 3000);
+            https.request("https://configsvr.msf.3g.qq.com/configsvr/serverlist.jsp", { method: "POST" }, (res) => {
+                let data = [];
+                res.on("error", reject);
+                res.on("data", (chunk) => data.push(chunk));
+                res.on("end", () => {
+                    try {
+                        clearTimeout(id);
+                        data = Buffer.concat(data);
+                        data = tea.decrypt(data, key).slice(4);
+                        const nested = jce.decode(data);
+                        host_port = { };
+                        for (let v of nested[2]) {
+                            if (Object.keys(host_port).length >= 3)
+                                continue;
+                            host_port[v[1]] = v[2];
+                        }
+                        if (Object.keys(host_port).length > 0) {
+                            update_time = timestamp();
+                            this.c.logger.debug(host_port);
+                            resolve();
+                        } else {
+                            reject(new Error("no aliveable server"));
+                        }
+                    } catch (err) {
+                        reject(err);
+                    }
+                });
+            }).on("error", reject).end(body);
+        });
+    }
 }
 
 module.exports = Network;

lib/wtlogin/wt.js

@@ -464,7 +464,7 @@ async function register(logout = false) {
         const blob = await this.send(pkt);
         const rsp = jce.decode(blob);
         const result = rsp[9] ? true : false;
-        if (!result)
+        if (!result && !logout)
             fs.unlink(path.join(this.dir, "token"), NOOP);
         return result;
     } catch {