BeamMP/BeamMP-Launcher
1
0
Fork 0
mirror of https://github.com/BeamMP/BeamMP-Launcher.git synced 2025-07-02 07:56:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check port and timeout recv

Browse Source
This commit is contained in:
Tixx 2025-01-11 21:50:39 +01:00
parent d14b64c652
commit 333a95262b
No known key found for this signature in database
GPG Key ID: EC6E7A2BAABF0B8C
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/Network/Core.cpp
View File

@ -111,7 +111,17 @@ void GetServerInfo(std::string Data) {
return; return;
} }
ServerAddr.sin_family = AF_INET; ServerAddr.sin_family = AF_INET;
ServerAddr.sin_port = htons(std::stoi(Data.substr(Data.find(':') + 1)));
int port = std::stoi(Data.substr(Data.find(':') + 1));
if (port < 1 || port > 65535) {
debug("Invalid port number: " + std::to_string(port));
KillSocket(ISock);
CoreSend("I" + Data + ";");
return;
}
ServerAddr.sin_port = htons(port);
inet_pton(AF_INET, IP.c_str(), &ServerAddr.sin_addr); inet_pton(AF_INET, IP.c_str(), &ServerAddr.sin_addr);
if (connect(ISock, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)) != 0) { if (connect(ISock, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)) != 0) {
debug("Connection to server failed with error: " + std::to_string(WSAGetLastError())); debug("Connection to server failed with error: " + std::to_string(WSAGetLastError()));
@ -129,6 +139,12 @@ void GetServerInfo(std::string Data) {
std::string buffer; std::string buffer;
buffer.resize(1024); buffer.resize(1024);
struct timeval timeout;
timeout.tv_sec = 10;
timeout.tv_usec = 0;
setsockopt(ISock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&timeout, sizeof(timeout));
int bytesReceived = recv(ISock, &buffer[0], buffer.size() - 1, 0); int bytesReceived = recv(ISock, &buffer[0], buffer.size() - 1, 0);
if (bytesReceived > 0) { if (bytesReceived > 0) {