moonlight-stream/Internet-Hosting-Tool
1
0
Fork 0
mirror of https://github.com/moonlight-stream/Internet-Hosting-Tool.git synced 2025-07-03 08:15:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Validate the port number of loopback traffic

Browse Source
This commit is contained in:
Cameron Gutman 2020-08-14 17:48:55 -07:00
parent 53246bd4c5
commit 41ef072c9b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
miss/relay.cpp
View File

@ -19,6 +19,7 @@ WINAPI
UdpRelayThreadProc(LPVOID Context) UdpRelayThreadProc(LPVOID Context)
{ {
PUDP_TUPLE tuple = (PUDP_TUPLE)Context; PUDP_TUPLE tuple = (PUDP_TUPLE)Context;
USHORT nboPort = htons(tuple->port);
SOCKADDR_IN lastRemoteAddr; SOCKADDR_IN lastRemoteAddr;
// Ensure the relay threads aren't preempted by games or other CPU intensive activity // Ensure the relay threads aren't preempted by games or other CPU intensive activity
@ -39,7 +40,7 @@ UdpRelayThreadProc(LPVOID Context)
} }
SOCKADDR_IN destinationAddr; SOCKADDR_IN destinationAddr;
if (RtlEqualMemory(&sourceAddr.sin_addr, &in4addr_loopback, sizeof(sourceAddr.sin_addr))) { if (RtlEqualMemory(&sourceAddr.sin_addr, &in4addr_loopback, sizeof(sourceAddr.sin_addr)) && sourceAddr.sin_port == nboPort) {
// Traffic incoming from loopback interface - send it to the last remote address // Traffic incoming from loopback interface - send it to the last remote address
destinationAddr = lastRemoteAddr; destinationAddr = lastRemoteAddr;
} }
@ -50,7 +51,7 @@ UdpRelayThreadProc(LPVOID Context)
// Send it to the normal port via the loopback adapter // Send it to the normal port via the loopback adapter
destinationAddr = sourceAddr; destinationAddr = sourceAddr;
destinationAddr.sin_addr = in4addr_loopback; destinationAddr.sin_addr = in4addr_loopback;
destinationAddr.sin_port = htons(tuple->port); destinationAddr.sin_port = nboPort;
} }
sendto(tuple->socket, buffer, recvLen, 0, (PSOCKADDR)&destinationAddr, sizeof(destinationAddr)); sendto(tuple->socket, buffer, recvLen, 0, (PSOCKADDR)&destinationAddr, sizeof(destinationAddr));