refactor: transform request info

This commit is contained in:
avwo
2019-03-21 16:31:49 +08:00
parent 08c9f6062a
commit 5599fbc1fb
+3 -2
View File
@@ -543,7 +543,8 @@ function resolveWebsocket(socket, wss) {
function addReqInfo(req) {
var clientPort = req.socket.remotePort;
var remoteData = tunnelTmplData.get(clientPort);
var serverPort = req.socket.localPort;
var remoteData = tunnelTmplData.get(clientPort + ':' + serverPort);
req.headers[config.HTTPS_FIELD] = 1;
if (remoteData) {
req.headers[config.CLIENT_IP_HEAD] = remoteData.clientIp || LOCALHOST;
@@ -643,7 +644,7 @@ module.exports = function(socket, hostname, next, isWebPort) {
return destroy(err);
}
reqSocket = s;
tunnelTmplData.set(reqSocket.localPort, {
tunnelTmplData.set(reqSocket.localPort + ':' + reqSocket.remotePort, {
clientIp: clientIp,
clientPort: socket.remotePort,
tunnelHost: tunnelHost