137 lines
3.1 KiB
JavaScript
137 lines
3.1 KiB
JavaScript
const net = require("net");
|
|
const { PomeloClient } = require("./lib/boot");
|
|
const { Stick, MaxBodyLen } = require("./lib/stick");
|
|
const { Message } = require("./Message");
|
|
|
|
const process = require("process");
|
|
|
|
const config = process.argv[2] || 'default';
|
|
const proxyList = require(`./config/${config}.json`);
|
|
|
|
for (const channel of proxyList) {
|
|
proxy(channel);
|
|
}
|
|
|
|
function proxy(config) {
|
|
console.log("proxy:", config);
|
|
|
|
const host = config.host;
|
|
const port = config.port;
|
|
const srcHost = config.srcHost;
|
|
const srcPort = config.srcPort;
|
|
|
|
const server = net.createServer((socket) => {
|
|
console.log("listening " + host + ":" + port);
|
|
|
|
const pomelo = new PomeloClient();
|
|
const stick = new Stick(1024);
|
|
stick.setMaxBodyLen(MaxBodyLen["32K"], true);
|
|
const connectBuffer = [];
|
|
let isReady = false;
|
|
|
|
pomelo.init(
|
|
{ host: srcHost, port: srcPort, log: true },
|
|
() => {
|
|
isReady = true;
|
|
connectBuffer.forEach((data) => {
|
|
stick.putData(data);
|
|
});
|
|
},
|
|
(route, body) => {
|
|
repPush(route, body);
|
|
}
|
|
);
|
|
|
|
pomelo.on("close", ()=> {
|
|
try {
|
|
socket.destroy();
|
|
} catch(e) {
|
|
console.log(e);
|
|
}
|
|
});
|
|
|
|
socket.on("connect", () => {
|
|
console.log("connect");
|
|
});
|
|
|
|
socket.on("error", (error) => {
|
|
console.log(error);
|
|
});
|
|
|
|
socket.on("close", () => {
|
|
console.log("close");
|
|
try {
|
|
pomelo.disconnect();
|
|
}
|
|
catch(err) {
|
|
console.log(err);
|
|
}
|
|
});
|
|
|
|
socket.on("drop", () => {
|
|
console.log("drop");
|
|
});
|
|
|
|
socket.on("data", (data) => {
|
|
try {
|
|
if (isReady) {
|
|
stick.putData(data);
|
|
} else {
|
|
connectBuffer.push(data);
|
|
}
|
|
} catch (e) {
|
|
console.log(e);
|
|
socket.destroy();
|
|
}
|
|
});
|
|
|
|
stick.onBody((msgId, data) => {
|
|
try {
|
|
const len = data.readUInt16LE(0);
|
|
const buf1 = data.subarray(2, 2 + len);
|
|
const msg = JSON.parse(buf1);
|
|
console.log(msgId, msg);
|
|
|
|
const account = "6516_2006_0xef59f6cc4d190a0ae576c46d4583e92b61174340";
|
|
switch (msgId) {
|
|
case Message.TYPE_REQUEST:
|
|
pomelo.request(msg.route, msg.params, (data) => {
|
|
repRequest(msg.serial, msg.route, data);
|
|
});
|
|
break;
|
|
case Message.TYPE_NOTIFY:
|
|
pomelo.notify(data.toString(), { uid: account });
|
|
break;
|
|
}
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
});
|
|
|
|
const repRequest = (serial, route, data) => {
|
|
data.serial = serial;
|
|
data.route = route;
|
|
try {
|
|
console.log("repRequest", data);
|
|
socket.write(
|
|
stick.makeData(Message.TYPE_RESPONSE, JSON.stringify(data))
|
|
);
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
};
|
|
|
|
const repPush = (route, data) => {
|
|
try {
|
|
console.log("repPush", data);
|
|
data.route = route;
|
|
socket.write(stick.makeData(Message.TYPE_PUSH, JSON.stringify(data)));
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
};
|
|
});
|
|
|
|
server.listen(port, host);
|
|
}
|