mirror of
https://github.com/avwo/whistle.git
synced 2026-10-27 04:19:59 +08:00
refactor: transform request info
This commit is contained in:
+3
-2
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user