nodejs内网穿透

说明

本地服务注册,基于子域名->端口映射。公网测试请开启二级或三级域名泛解析

无心跳保活、无多线程并发处理

服务器端

  • 请求ID基于全局变量,不支持PM2多进程开服务端。(多开请修改uid函数,增加网卡以及进程信息)
  • 该代码包含HTTP代理和内网穿透服务端
const net = require('net');
var nat = {
    config: {servPort: 39001,natPort: 80,proxy:39002},
    registry: {},//服务注册表
    request: {},//请求连接
    uid(pre = '', howMany = 1, len = 4) {
        if (howMany > 9999) return false;
        var date = new Date();
        var str = date.getFullYear();
        str += ('0' + (date.getMonth() + 1)).slice(-2);
        str += ('0' + date.getDate()).slice(-2);
        str += ('0' + date.getHours()).slice(-2);
        str += ('0' + date.getMinutes()).slice(-2);
        str += ('0' + date.getSeconds()).slice(-2);
        if (!this.num || this.num.time !== str)
            this.num = { time: str, counter: 0 }
        let bd = [];
        for (var i = 0; i < howMany; i++) {
            this.num.counter++;
            var buffNum = (Array(len).join(0) + this.num.counter).slice(len * (-1));
            bd.push(pre + str + buffNum);
        }
        return bd.length > 1 ? bd : bd[0];
    },
    run(){
        this.nat()
        this.proxy()
    },
    /*************** TCP内网穿透 ***************/
    nat() {
        //创建主服务器
        var that = this;
        net.Server(async socket => {
            socket.on('data',data=>{
                socket.removeAllListeners('data');
                try {data=JSON.parse(data)} catch (err) {return false}
                if (data.act == 'regist') {
                    data.name.forEach(item => {
                        that.registry[item] = { client: socket }
                    })
                    socket.NatNames=data.name
                    socket.on('close',had_error=>{
                        socket.NatNames.forEach(item=>{
                            delete that.registry[item]
                        })
                    })
                } else if (data.act == 'connect') {
                    var request=that.request[data.key]
                    if(request&&request.data&&request.socket){
                        request.socket.pipe(socket).pipe(request.socket)
                        socket.write(request.data)
                        request.socket.on('error',err=>{request.socket.end();socket.end()})
                        socket.on('error',err=>{request.socket.end();socket.end()})
                    }
                }
            })
            socket.on('error',(err)=>{socket.end();console.log(`${new Date().toLocaleString()}: ${err.message}`)})
            
        }).listen(that.config.servPort,()=>{console.log(`NAR serv running at ${that.config.servPort}`)})
        //创建代理通信服务器
        net.Server(socket => {
            var key = this.uid('RC')
            that.request[key] = {socket}
            socket.on('data', data => {
                socket.removeAllListeners('data');
                that.request[key].data=data
                var domain = data.toString().match(/Host: (\S)+/i)[0].split(':')[1]
                var nat = domain.split('.')[0].trim()
                if (that.registry[nat]){
                    var str=JSON.stringify({ act: 'connect', key, nat})
                    that.registry[nat].client.write(str)
                }
            })
        }).listen(that.config.natPort,()=>{console.log(`NAR http running at ${that.config.natPort}`)})
    },
    /*************** http[s]代理 ***************/
    proxy(){
        var that=this
        const net = require("net")
        net.Server((client) => {
            client.on("data", (data) => {
                let req = that.reqParse(data)
                if (!req) return;
                let server = net.connect(req.port, req.host);
                client.removeAllListeners('data');
                client.pipe(server).pipe(client);
                server.on('error',err=>{client.end(),server.end()})
                client.on('error',err=>{client.end(),server.end()})
                if (req.method == 'CONNECT')
                    client.write("HTTP/1.1 200 Connection established\r\nConnection: close\r\n\r\n");
                else
                    server.write(req.buffer);
            })
        }).listen(this.config.proxy, () => {console.log("prox running at port: " + this.config.proxy)})
    },
    /*************** 格式化http请求头 ***************/
    reqParse(b) {
        let s = b.toString('utf8');
        let method = s.split('\n')[0].match(/^([A-Z]+)\s/)[1];
        if (method == 'CONNECT') {
            var arr = s.match(/^([A-Z]+)\s([^\:\s]+)\:(\d+)\sHTTP\/(\d\.\d)/);
            if (arr && arr[1] && arr[2] && arr[3] && arr[4])
                return {
                    method: arr[1],
                    host: arr[2],
                    port: arr[3],
                    httpVersion: arr[4],
                    buffer: b
                };
        } else {
            let arr = s.match(/^([A-Z]+)\s([^\s]+)\sHTTP\/(\d\.\d)/);
            let index = 0;
            //请求头,请求体处理
            for (var i = 0, len = b.length - 3; i < len; i++) {
                if (b[i] == 0x0d && b[i + 1] == 0x0a && b[i + 2] == 0x0d && b[i + 3] == 0x0a) {
                    index = i + 4;
                }
            }
            if (!index) return false;
            var header = b.slice(0, index).toString('utf8');
            //替换connection头
            header = header.replace(/(proxy\-)?connection\:.+\r\n/ig, '')
                .replace(/Keep\-Alive\:.+\r\n/i, '')
                .replace("\r\n", '\r\nConnection: close\r\n');
            //替换网址格式(去掉域名部分)
            if (arr[1] == '1.1') {
                var url = arr[2].replace(/http\:\/\/[^\/]+/, '');
                if (arr[2] != url) header = header.replace(arr[2], url);
            }
            b = Buffer.concat([Buffer.from(header), b.slice(index)]);

            if (arr && arr[1] && arr[2] && arr[3]) {
                var host = s.match(/Host\:\s+([^\n\s\r]+)/)[1];
                if (host) {
                    var _p = host.split(':', 2);
                    return {
                        method: arr[1],
                        host: _p[0],
                        port: _p[1] ? _p[1] : 80,
                        path: arr[2],
                        httpVersion: arr[3],
                        buffer: b
                    };
                }
            }
        }
        return false;
    }
}
nat.run()

本地客户端

const HTTP=require('http')
const FS = require('fs')
const net = require('net');
//内网穿透客户端
var nat = {
    config: {
        serv: {host: '47.92.29.253',port: 39001},//服务器配置
        local: {api: 80,md:81},//本地服务名=>端口映射
        retry: 300000,//异常重连
    },
    connServ() {
        var that = this
        let serv =this.serv= net.connect(that.config.serv, () => {
            console.log(`${new Date().toLocaleString()}: NAT已连接服务器`)
            var option = { act: 'regist', name: Object.keys(that.config.local) }
            serv.write(JSON.stringify(option))
        })
        serv.setKeepAlive(true)
        serv.on('data', data => {
            try { data = JSON.parse(data) } catch (err) { return false }
            if (data.act == 'connect') {
                var request = net.connect(that.config.serv, () => {
                    var response = net.connect({ host: '127.0.0.1', port: that.config.local[data.nat] },()=>{
                        response.pipe(request).pipe(response)
                        request.write(JSON.stringify({ act: 'connect', key: data.key }))
                    })
                    request.on('error',err=>{response.end();request.end()})
                    response.on('error',err=>{response.end();request.end()})
                })
            }
        })
        serv.on('error', err => {
            console.log(`${new Date().toLocaleString()}: NAT服务器连接异常:${err.message}`)
            serv.end();
            that.connServ()
        })
        serv.on('end', () => {
            console.log(`${new Date().toLocaleString()}: NAT服务器连接已断开`)
            serv.end();
            that.connServ()
        })
    },
    //定时监测连接状态&&断线重连
    run(){
        var that = this
        that.connServ()
        setInterval(() => {
            if (that.serv.readyState != 'open') {
                that.serv.end()
                that.connServ()
            }
        }, that.config.retry)
    }
}

你可能感兴趣的:(nodejs,proxy,socket,node.js)