code

regionServer.js

@@ -0,0 +1,90 @@
+const fs = require("fs")
+const https = require("https")
+const axios = require("axios")
+const { decodeProto, encodeProto, debug } = require("./utils")
+
+const preparedRegions = {}
+let currentProxy = undefined
+
+const getModifiedRegionList = async (conf) => {
+    const d = await axios.get(`https://${conf.dispatchUrl}/query_region_list`, {
+        responseType: "text",
+        params: {
+            version: conf.version,
+            channel_id: conf.channel,
+            sub_channel_id: conf.subChannel
+        }
+    })
+    const regions = await decodeProto(Buffer.from(d.data,"base64"),"QueryRegionList")
+    regions.list = regions.list.map(item => {
+        const host = new URL(item.url).host
+        if (regions.list.length === 1) {
+            preparedRegions[host] = true
+        }
+        item.url = `https://localdispatch.yuanshen.com/query_cur_region/${host}`
+        return item
+    })
+    return (await encodeProto(regions,"QueryRegionList")).toString("base64")
+}
+
+const getModifiedRegionInfo = async (url, uc, hs) => {
+    const splitUrl = url.split("?")
+    const host = splitUrl[0].split("/")[2]
+    const noQueryRequest = splitUrl[1] === undefined
+    const query = noQueryRequest ? "" : `?${splitUrl[1]}`
+    const d = await axios.get(`https://${host}/query_cur_region${query}`, {
+        responseType: "text"
+    })
+    if (noQueryRequest) {
+        preparedRegions[host] = true
+        return d.data
+    } else {
+        const region = await decodeProto(Buffer.from(d.data,"base64"),"QueryCurRegion")
+        const info = region.info
+        if (preparedRegions[host]) {
+            if (currentProxy !== undefined) {
+                currentProxy.close()
+            }
+            debug("Create udp proxy: %s:%d", info.ip, info.port)
+            currentProxy = uc(info.ip, info.port, hs)
+        } else {
+            preparedRegions[host] = true
+        }
+        info.ip = "127.0.0.1"
+        info.port = 45678
+        return (await encodeProto(region,"QueryCurRegion")).toString("base64")
+    }
+}
+
+const agent = new https.Agent({
+    rejectUnauthorized: false
+})
+
+const create = async (conf, regionListLoadedCallback, regionSelectCallback) => {
+    const regions = await getModifiedRegionList(conf)
+    regionListLoadedCallback()
+    const hServer = https.createServer({
+        pfx: fs.readFileSync("./cert/root.p12"),
+        passphrase: ""
+    }, async (request, response) => {
+        const url = request.url
+        debug("HTTP请求: %s", url)
+        response.writeHead(200, { "Content-Type": "text/html" })
+        if (url.startsWith("/query_region_list")) {
+            response.end(regions)
+        } else if (url.startsWith("/query_cur_region")) {
+            const regionInfo = await getModifiedRegionInfo(url, regionSelectCallback, hServer)
+            response.end(regionInfo)
+        } else {
+            const frontResponse = await axios.get(`https://${conf.dispatchIP}${url}`, {
+                responseType: "arraybuffer",
+                httpsAgent: agent
+            })
+            response.end(frontResponse.data)
+        }
+    }).listen(443, "127.0.0.1")
+}
+
+module.exports = {
+    create
+}