mirror of
https://github.com/BeamMP/BeamMP-Launcher.git
synced 2026-06-22 08:31:07 +00:00
Network, bug fixes, more patters, GELua and a lot more
This commit is contained in:
Anonymous275
@@ -0,0 +1,52 @@
|
||||
///
|
||||
/// Created by Anonymous275 on 1/30/22
|
||||
/// Copyright (c) 2021-present Anonymous275 read the LICENSE file for more info.
|
||||
///
|
||||
|
||||
#include "Compressor.h"
|
||||
#include <zlib.h>
|
||||
|
||||
#define Biggest 30000
|
||||
std::string Zlib::Comp(std::string Data){
|
||||
char*C = new char[Biggest];
|
||||
memset(C, 0, Biggest);
|
||||
z_stream defstream;
|
||||
defstream.zalloc = Z_NULL;
|
||||
defstream.zfree = Z_NULL;
|
||||
defstream.opaque = Z_NULL;
|
||||
defstream.avail_in = (uInt)Data.length();
|
||||
defstream.next_in = (Bytef *)&Data[0];
|
||||
defstream.avail_out = Biggest;
|
||||
defstream.next_out = reinterpret_cast<Bytef *>(C);
|
||||
deflateInit(&defstream, Z_BEST_COMPRESSION);
|
||||
deflate(&defstream, Z_SYNC_FLUSH);
|
||||
deflate(&defstream, Z_FINISH);
|
||||
deflateEnd(&defstream);
|
||||
uint32_t TO = defstream.total_out;
|
||||
std::string Ret(TO,0);
|
||||
memcpy_s(&Ret[0],TO,C,TO);
|
||||
delete [] C;
|
||||
return Ret;
|
||||
}
|
||||
|
||||
std::string Zlib::DeComp(std::string Compressed){
|
||||
char*C = new char[Biggest];
|
||||
memset(C, 0, Biggest);
|
||||
z_stream infstream;
|
||||
infstream.zalloc = Z_NULL;
|
||||
infstream.zfree = Z_NULL;
|
||||
infstream.opaque = Z_NULL;
|
||||
infstream.avail_in = Biggest;
|
||||
infstream.next_in = (Bytef *)(&Compressed[0]);
|
||||
infstream.avail_out = Biggest;
|
||||
infstream.next_out = (Bytef *)(C);
|
||||
inflateInit(&infstream);
|
||||
inflate(&infstream, Z_SYNC_FLUSH);
|
||||
inflate(&infstream, Z_FINISH);
|
||||
inflateEnd(&infstream);
|
||||
uint32_t TO = infstream.total_out;
|
||||
std::string Ret(TO,0);
|
||||
memcpy_s(&Ret[0],TO,C,TO);
|
||||
delete [] C;
|
||||
return Ret;
|
||||
}
|
||||
@@ -0,0 +1,317 @@
|
||||
///
|
||||
/// Created by Anonymous275 on 1/30/22
|
||||
/// Copyright (c) 2021-present Anonymous275 read the LICENSE file for more info.
|
||||
///
|
||||
|
||||
|
||||
#include <ws2tcpip.h>
|
||||
#include <filesystem>
|
||||
#include "Launcher.h"
|
||||
#include "Server.h"
|
||||
#include "Logger.h"
|
||||
#include <cstring>
|
||||
#include <fstream>
|
||||
#include <string>
|
||||
#include <thread>
|
||||
#include <atomic>
|
||||
#include <vector>
|
||||
#include <future>
|
||||
|
||||
namespace fs = std::filesystem;
|
||||
std::vector<std::string> Split(const std::string& String, const std::string& delimiter){
|
||||
std::vector<std::string> Val;
|
||||
size_t pos;
|
||||
std::string token,s = String;
|
||||
while ((pos = s.find(delimiter)) != std::string::npos) {
|
||||
token = s.substr(0, pos);
|
||||
if(!token.empty())Val.push_back(token);
|
||||
s.erase(0, pos + delimiter.length());
|
||||
}
|
||||
if(!s.empty())Val.push_back(s);
|
||||
return Val;
|
||||
}
|
||||
|
||||
void CheckForDir(){
|
||||
if(!fs::exists("Resources")){
|
||||
_wmkdir(L"Resources");
|
||||
}
|
||||
}
|
||||
|
||||
void Server::WaitForConfirm() {
|
||||
while(!Terminate && !ModLoaded){
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10));
|
||||
}
|
||||
ModLoaded = false;
|
||||
}
|
||||
|
||||
void Server::Abort() {
|
||||
Terminate = true;
|
||||
LOG(INFO) << "Terminated!";
|
||||
}
|
||||
|
||||
std::string Server::Auth() {
|
||||
TCPSend("VC" + LauncherInstance->getVersion());
|
||||
|
||||
auto Res = TCPRcv();
|
||||
|
||||
if(Res.empty() || Res[0] == 'E'){
|
||||
Abort();
|
||||
return "";
|
||||
}
|
||||
|
||||
TCPSend(LauncherInstance->getPublicKey());
|
||||
if(Terminate)return "";
|
||||
|
||||
Res = TCPRcv();
|
||||
if(Res.empty() || Res[0] != 'P'){
|
||||
Abort();
|
||||
return "";
|
||||
}
|
||||
|
||||
Res = Res.substr(1);
|
||||
if(Res.find_first_not_of("0123456789") == std::string::npos){
|
||||
ClientID = std::stoi(Res);
|
||||
}else{
|
||||
Abort();
|
||||
UUl("Authentication failed!");
|
||||
return "";
|
||||
}
|
||||
TCPSend("SR");
|
||||
if(Terminate)return "";
|
||||
|
||||
Res = TCPRcv();
|
||||
|
||||
if(Res[0] == 'E'){
|
||||
Abort();
|
||||
return "";
|
||||
}
|
||||
|
||||
if(Res.empty() || Res == "-"){
|
||||
LOG(INFO) << "Didn't Receive any mods...";
|
||||
ModList = "-";
|
||||
TCPSend("Done");
|
||||
LOG(INFO) << "Done!";
|
||||
return "";
|
||||
}
|
||||
return Res;
|
||||
}
|
||||
|
||||
void Server::UpdateUl(bool D, const std::string& msg) {
|
||||
if(D)UStatus = "UlDownloading Resource " + msg;
|
||||
else UStatus = "UlLoading Resource " + msg;
|
||||
}
|
||||
|
||||
void Server::AsyncUpdate(uint64_t& Rcv,uint64_t Size,const std::string& Name) {
|
||||
do {
|
||||
double pr = double(Rcv) / double(Size) * 100;
|
||||
std::string Per = std::to_string(trunc(pr * 10) / 10);
|
||||
UpdateUl(true, Name + " (" + Per.substr(0, Per.find('.') + 2) + "%)");
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
||||
}while(!Terminate && Rcv < Size);
|
||||
}
|
||||
|
||||
char* Server::TCPRcvRaw(uint64_t Sock, uint64_t& GRcv, uint64_t Size) {
|
||||
if(Sock == -1){
|
||||
Terminate = true;
|
||||
UUl("Invalid Socket");
|
||||
return nullptr;
|
||||
}
|
||||
char* File = new char[Size];
|
||||
uint64_t Rcv = 0;
|
||||
do{
|
||||
int Len = int(Size-Rcv);
|
||||
if(Len > 1000000)Len = 1000000;
|
||||
int32_t Temp = recv(Sock, &File[Rcv], Len, MSG_WAITALL);
|
||||
if(Temp < 1){
|
||||
UUl("Socket Closed Code 1");
|
||||
KillSocket(Sock);
|
||||
Terminate = true;
|
||||
delete[] File;
|
||||
return nullptr;
|
||||
}
|
||||
Rcv += Temp;
|
||||
GRcv += Temp;
|
||||
}while(Rcv < Size && !Terminate);
|
||||
return File;
|
||||
}
|
||||
|
||||
void Server::MultiKill(uint64_t Sock) {
|
||||
KillSocket(TCPSocket);
|
||||
KillSocket(Sock);
|
||||
Terminate = true;
|
||||
}
|
||||
|
||||
uint64_t Server::InitDSock() {
|
||||
SOCKET DSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||
SOCKADDR_IN ServerAddr;
|
||||
if(DSock < 1) {
|
||||
KillSocket(DSock);
|
||||
Terminate = true;
|
||||
return 0;
|
||||
}
|
||||
ServerAddr.sin_family = AF_INET;
|
||||
ServerAddr.sin_port = htons(Port);
|
||||
inet_pton(AF_INET, IP.c_str(), &ServerAddr.sin_addr);
|
||||
if(connect(DSock, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr)) != 0){
|
||||
KillSocket(DSock);
|
||||
Terminate = true;
|
||||
return 0;
|
||||
}
|
||||
char Code[2] = {'D', char(ClientID)};
|
||||
if(send(DSock,Code,2,0) != 2){
|
||||
KillSocket(DSock);
|
||||
Terminate = true;
|
||||
return 0;
|
||||
}
|
||||
return DSock;
|
||||
}
|
||||
|
||||
std::string Server::MultiDownload(uint64_t DSock, uint64_t Size, const std::string& Name) {
|
||||
|
||||
uint64_t GRcv = 0, MSize = Size/2, DSize = Size - MSize;
|
||||
|
||||
std::thread Au(&Server::AsyncUpdate, this, std::ref(GRcv), Size, Name);
|
||||
|
||||
std::packaged_task<char*()> task([&] { return TCPRcvRaw(TCPSocket,GRcv,MSize); });
|
||||
std::future<char*> f1 = task.get_future();
|
||||
std::thread Dt(std::move(task));
|
||||
Dt.detach();
|
||||
|
||||
char* DData = TCPRcvRaw(DSock,GRcv,DSize);
|
||||
|
||||
if(!DData){
|
||||
MultiKill(DSock);
|
||||
return "";
|
||||
}
|
||||
|
||||
f1.wait();
|
||||
char* MData = f1.get();
|
||||
|
||||
if(!MData){
|
||||
MultiKill(DSock);
|
||||
return "";
|
||||
}
|
||||
|
||||
if(Au.joinable())Au.join();
|
||||
|
||||
|
||||
///omg yes very ugly my god but i was in a rush will revisit
|
||||
std::string Ret(Size,0);
|
||||
memcpy_s(&Ret[0],MSize,MData,MSize);
|
||||
delete[]MData;
|
||||
|
||||
memcpy_s(&Ret[MSize],DSize,DData,DSize);
|
||||
delete[]DData;
|
||||
|
||||
return Ret;
|
||||
}
|
||||
|
||||
void Server::InvalidResource(const std::string& File) {
|
||||
UUl("Invalid mod \"" + File + "\"");
|
||||
LOG(WARNING) << "The server tried to sync \"" << File << "\" that is not a .zip file!";
|
||||
Terminate = true;
|
||||
}
|
||||
|
||||
void Server::SyncResources() {
|
||||
std::string Ret = Auth();
|
||||
if(Ret.empty())return;
|
||||
LOG(INFO) << "Checking Resources...";
|
||||
CheckForDir();
|
||||
|
||||
std::vector<std::string> list = Split(Ret, ";");
|
||||
std::vector<std::string> FNames(list.begin(), list.begin() + (list.size() / 2));
|
||||
std::vector<std::string> FSizes(list.begin() + (list.size() / 2), list.end());
|
||||
list.clear();
|
||||
Ret.clear();
|
||||
|
||||
int Amount = 0,Pos = 0;
|
||||
std::string a,t;
|
||||
for(const std::string&name : FNames){
|
||||
if(!name.empty()){
|
||||
t += name.substr(name.find_last_of('/') + 1) + ";";
|
||||
}
|
||||
}
|
||||
if(t.empty())ModList = "-";
|
||||
else ModList = t;
|
||||
t.clear();
|
||||
for(auto FN = FNames.begin(),FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN,++FS) {
|
||||
auto pos = FN->find_last_of('/');
|
||||
auto ZIP = FN->find(".zip");
|
||||
if (ZIP == std::string::npos || FN->length() - ZIP != 4) {
|
||||
InvalidResource(*FN);
|
||||
return;
|
||||
}
|
||||
if (pos == std::string::npos)continue;
|
||||
Amount++;
|
||||
}
|
||||
if(!FNames.empty())LOG(INFO) << "Syncing...";
|
||||
SOCKET DSock = InitDSock();
|
||||
for(auto FN = FNames.begin(),FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN,++FS) {
|
||||
auto pos = FN->find_last_of('/');
|
||||
if (pos != std::string::npos) {
|
||||
a = "Resources" + FN->substr(pos);
|
||||
} else continue;
|
||||
Pos++;
|
||||
if (fs::exists(a)) {
|
||||
if (FS->find_first_not_of("0123456789") != std::string::npos)continue;
|
||||
if (fs::file_size(a) == std::stoull(*FS)){
|
||||
UpdateUl(false,std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + a.substr(a.find_last_of('/')));
|
||||
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
||||
try {
|
||||
if(!fs::exists(LauncherInstance->getMPUserPath())){
|
||||
fs::create_directories(LauncherInstance->getMPUserPath());
|
||||
}
|
||||
fs::copy_file(a, LauncherInstance->getMPUserPath() + a.substr(a.find_last_of('/')),
|
||||
fs::copy_options::overwrite_existing);
|
||||
} catch (std::exception& e) {
|
||||
LOG(ERROR) << "Failed copy to the mods folder! " << e.what();
|
||||
Terminate = true;
|
||||
continue;
|
||||
}
|
||||
WaitForConfirm();
|
||||
continue;
|
||||
}else remove(a.c_str());
|
||||
}
|
||||
CheckForDir();
|
||||
std::string FName = a.substr(a.find_last_of('/'));
|
||||
do {
|
||||
TCPSend("f" + *FN);
|
||||
|
||||
std::string Data = TCPRcv();
|
||||
if (Data == "CO" || Terminate){
|
||||
Terminate = true;
|
||||
UUl("Server cannot find " + FName);
|
||||
break;
|
||||
}
|
||||
|
||||
std::string Name = std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + FName;
|
||||
|
||||
Data = MultiDownload(DSock,std::stoull(*FS), Name);
|
||||
|
||||
if(Terminate)break;
|
||||
UpdateUl(false,std::to_string(Pos)+"/"+std::to_string(Amount)+": "+FName);
|
||||
std::ofstream LFS;
|
||||
LFS.open(a.c_str(), std::ios_base::app | std::ios::binary);
|
||||
if (LFS.is_open()) {
|
||||
LFS.write(&Data[0], std::streamsize(Data.size()));
|
||||
LFS.close();
|
||||
}
|
||||
|
||||
}while(fs::file_size(a) != std::stoull(*FS) && !Terminate);
|
||||
if(!Terminate){
|
||||
if(!fs::exists(LauncherInstance->getMPUserPath())){
|
||||
fs::create_directories(LauncherInstance->getMPUserPath());
|
||||
}
|
||||
fs::copy_file(a,LauncherInstance->getMPUserPath() + FName, fs::copy_options::overwrite_existing);
|
||||
}
|
||||
WaitForConfirm();
|
||||
}
|
||||
KillSocket(DSock);
|
||||
if(!Terminate){
|
||||
TCPSend("Done");
|
||||
LOG(INFO) << "Done!";
|
||||
}else{
|
||||
UStatus = "start";
|
||||
LOG(INFO) << "Connection Terminated!";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,302 @@
|
||||
///
|
||||
/// Created by Anonymous275 on 1/30/22
|
||||
/// Copyright (c) 2021-present Anonymous275 read the LICENSE file for more info.
|
||||
///
|
||||
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#include "Compressor.h"
|
||||
#include "Server.h"
|
||||
#include "Launcher.h"
|
||||
#include <windows.h>
|
||||
#include <winsock2.h>
|
||||
#include <ws2tcpip.h>
|
||||
#include "Logger.h"
|
||||
|
||||
Server::Server(Launcher* Instance) : LauncherInstance(Instance) {
|
||||
WSADATA wsaData;
|
||||
int iRes = WSAStartup(514, &wsaData); //2.2
|
||||
if (iRes != 0) {
|
||||
LOG(ERROR) << "WSAStartup failed with error: " << iRes;
|
||||
}
|
||||
UDPSockAddress = std::make_unique<sockaddr_in>();
|
||||
}
|
||||
|
||||
Server::~Server() {
|
||||
Close();
|
||||
WSACleanup();
|
||||
}
|
||||
|
||||
void Server::TCPClientMain() {
|
||||
SOCKADDR_IN ServerAddr;
|
||||
TCPSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||
if(TCPSocket == -1) {
|
||||
LOG(ERROR) << "Socket failed! Error code: " << WSAGetLastError();
|
||||
return;
|
||||
}
|
||||
ServerAddr.sin_family = AF_INET;
|
||||
ServerAddr.sin_port = htons(Port);
|
||||
inet_pton(AF_INET, IP.c_str(), &ServerAddr.sin_addr);
|
||||
int status = connect(TCPSocket, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr));
|
||||
if(status != 0){
|
||||
UStatus = "Connection Failed!";
|
||||
LOG(ERROR) << "Connect failed! Error code: " << WSAGetLastError();
|
||||
Close();
|
||||
return;
|
||||
}
|
||||
LOG(INFO) << "Connected!";
|
||||
char Code = 'C';
|
||||
send(TCPSocket, &Code, 1, 0);
|
||||
SyncResources();
|
||||
UDPConnection = std::thread(&Server::UDPMain, this);
|
||||
while(!Terminate.load()) {
|
||||
ServerParser(TCPRcv());
|
||||
}
|
||||
LauncherInstance->SendIPC("T", false);
|
||||
KillSocket(TCPSocket);
|
||||
}
|
||||
|
||||
void Server::UDPSend(std::string Data) {
|
||||
if(ClientID == -1 || UDPSocket == -1)return;
|
||||
if(Data.length() > 400){
|
||||
std::string CMP(Zlib::Comp(Data));
|
||||
Data = "ABG:" + CMP;
|
||||
}
|
||||
std::string Packet = char(ClientID+1) + std::string(":") + Data;
|
||||
int sendOk = sendto(UDPSocket, Packet.c_str(), int(Packet.size()), 0, (sockaddr*)UDPSockAddress.get(), sizeof(sockaddr_in));
|
||||
if (sendOk == SOCKET_ERROR)LOG(ERROR) << "UDP Socket Error Code : " << WSAGetLastError();
|
||||
}
|
||||
|
||||
void Server::UDPParser(std::string Packet) {
|
||||
if(Packet.substr(0,4) == "ABG:"){
|
||||
Packet = Zlib::DeComp(Packet.substr(4));
|
||||
}
|
||||
ServerParser(Packet);
|
||||
}
|
||||
|
||||
void Server::UDPRcv() {
|
||||
sockaddr_in FromServer{};
|
||||
int clientLength = sizeof(FromServer);
|
||||
ZeroMemory(&FromServer, clientLength);
|
||||
std::string Ret(10240,0);
|
||||
if(UDPSocket == -1)return;
|
||||
int32_t Rcv = recvfrom(UDPSocket, &Ret[0], 10240, 0, (sockaddr*)&FromServer, &clientLength);
|
||||
if (Rcv == SOCKET_ERROR)return;
|
||||
UDPParser(Ret.substr(0,Rcv));
|
||||
}
|
||||
|
||||
void Server::UDPClient() {
|
||||
UDPSockAddress->sin_family = AF_INET;
|
||||
UDPSockAddress->sin_port = htons(Port);
|
||||
inet_pton(AF_INET, IP.c_str(), &UDPSockAddress->sin_addr);
|
||||
UDPSocket = socket(AF_INET, SOCK_DGRAM, 0);
|
||||
LauncherInstance->SendIPC("P"+std::to_string(ClientID), false);
|
||||
TCPSend("H");
|
||||
UDPSend("p");
|
||||
while(!Terminate)UDPRcv();
|
||||
KillSocket(UDPSocket);
|
||||
}
|
||||
|
||||
void Server::UDPMain() {
|
||||
AutoPing = std::thread(&Server::PingLoop, this);
|
||||
UDPClient();
|
||||
Terminate = true;
|
||||
LOG(INFO) << "Connection terminated";
|
||||
}
|
||||
|
||||
void Server::Connect(const std::string& Data) {
|
||||
ModList.clear();
|
||||
Terminate.store(false);
|
||||
IP = GetAddress(Data.substr(1,Data.find(':')-1));
|
||||
if(IP.find('.') == -1){
|
||||
if(IP == "DNS")UStatus ="Connection Failed! (DNS Lookup Failed)";
|
||||
else UStatus = "Connection Failed! (WSA failed to start)";
|
||||
ModList = "-";
|
||||
Terminate.store(true);
|
||||
return;
|
||||
}
|
||||
Port = std::stoi(Data.substr(Data.find(':')+1));
|
||||
LauncherInstance->CheckKey();
|
||||
UStatus = "Loading...";
|
||||
Ping = -1;
|
||||
TCPConnection = std::thread(&Server::TCPClientMain, this);
|
||||
LOG(INFO) << "Connecting to server";
|
||||
}
|
||||
|
||||
void Server::SendLarge(std::string Data) {
|
||||
if(Data.length() > 400) {
|
||||
std::string CMP(Zlib::Comp(Data));
|
||||
Data = "ABG:" + CMP;
|
||||
}
|
||||
TCPSend(Data);
|
||||
}
|
||||
|
||||
void Server::ServerSend(std::string Data, bool Rel) {
|
||||
if(Terminate || Data.empty())return;
|
||||
char C = 0;
|
||||
int DLen = int(Data.length());
|
||||
if(DLen > 3)C = Data.at(0);
|
||||
bool Ack = C == 'O' || C == 'T';
|
||||
if(C == 'N' || C == 'W' || C == 'Y' || C == 'V' || C == 'E' || C == 'C')Rel = true;
|
||||
if(Ack || Rel) {
|
||||
if(Ack || DLen > 1000)SendLarge(Data);
|
||||
else TCPSend(Data);
|
||||
}else UDPSend(Data);
|
||||
}
|
||||
|
||||
void Server::PingLoop() {
|
||||
while(!Terminate){
|
||||
ServerSend("p", false);
|
||||
PingStart = std::chrono::high_resolution_clock::now();
|
||||
std::this_thread::sleep_for(std::chrono::seconds (1));
|
||||
}
|
||||
}
|
||||
|
||||
void Server::Close() {
|
||||
Terminate.store(true);
|
||||
Ping = -1;
|
||||
if(TCPConnection.joinable()) {
|
||||
TCPConnection.join();
|
||||
}
|
||||
if(UDPConnection.joinable()) {
|
||||
UDPConnection.join();
|
||||
}
|
||||
if(AutoPing.joinable()) {
|
||||
AutoPing.join();
|
||||
}
|
||||
KillSocket(TCPSocket);
|
||||
KillSocket(UDPSocket);
|
||||
}
|
||||
|
||||
const std::string &Server::getMap() {
|
||||
return MStatus;
|
||||
}
|
||||
|
||||
bool Server::Terminated() {
|
||||
return Terminate;
|
||||
}
|
||||
|
||||
const std::string &Server::getModList() {
|
||||
return ModList;
|
||||
}
|
||||
|
||||
int Server::getPing() const {
|
||||
return Ping;
|
||||
}
|
||||
|
||||
const std::string& Server::getUIStatus() {
|
||||
return UStatus;
|
||||
}
|
||||
|
||||
std::string Server::GetAddress(const std::string& Data) {
|
||||
if(Data.find_first_not_of("0123456789.") == -1)return Data;
|
||||
hostent* host;
|
||||
host = gethostbyname(Data.c_str());
|
||||
if(!host){
|
||||
LOG(ERROR) << "DNS lookup failed! on " << Data;
|
||||
return "DNS";
|
||||
}
|
||||
std::string Ret = inet_ntoa(*((struct in_addr *)host->h_addr));
|
||||
return Ret;
|
||||
}
|
||||
|
||||
int Server::KillSocket(uint64_t Dead) {
|
||||
if(Dead == (SOCKET)-1)return 0;
|
||||
shutdown(Dead, SD_BOTH);
|
||||
return closesocket(Dead);
|
||||
}
|
||||
|
||||
void Server::setModLoaded() {
|
||||
ModLoaded.store(true);
|
||||
}
|
||||
|
||||
void Server::UUl(const std::string& R) {
|
||||
UStatus = "Disconnected: " + R;
|
||||
}
|
||||
|
||||
bool Server::CheckBytes(int32_t Bytes) {
|
||||
if (Bytes == 0){
|
||||
//debug("(TCP) Connection closing... CheckBytes(16)");
|
||||
Terminate = true;
|
||||
return false;
|
||||
}else if (Bytes < 0) {
|
||||
//debug("(TCP CB) recv failed with error: " + std::to_string(WSAGetLastError()));
|
||||
KillSocket(TCPSocket);
|
||||
Terminate = true;
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
void Server::TCPSend(const std::string& Data) {
|
||||
|
||||
if(TCPSocket == -1) {
|
||||
Terminate = true;
|
||||
UUl("Invalid Socket");
|
||||
return;
|
||||
}
|
||||
|
||||
int32_t Size,Sent,Temp;
|
||||
std::string Send(4,0);
|
||||
Size = int32_t(Data.size());
|
||||
memcpy(&Send[0],&Size,sizeof(Size));
|
||||
Send += Data;
|
||||
// Do not use Size before this point for anything but the header
|
||||
Sent = 0;
|
||||
Size += 4;
|
||||
do{
|
||||
if (size_t(Sent) >= Send.size()) {
|
||||
LOG(ERROR) << "string OOB in " << std::string(__func__);
|
||||
UUl("TCP Send OOB");
|
||||
return;
|
||||
}
|
||||
Temp = send(TCPSocket, &Send[Sent], Size - Sent, 0);
|
||||
if(!CheckBytes(Temp)){
|
||||
UUl("Socket Closed Code 2");
|
||||
return;
|
||||
}
|
||||
Sent += Temp;
|
||||
}while(Sent < Size);
|
||||
}
|
||||
|
||||
std::string Server::TCPRcv() {
|
||||
if(TCPSocket == -1){
|
||||
Terminate = true;
|
||||
UUl("Invalid Socket");
|
||||
return "";
|
||||
}
|
||||
int32_t Header,BytesRcv = 0,Temp;
|
||||
std::vector<char> Data(sizeof(Header));
|
||||
do{
|
||||
Temp = recv(TCPSocket, &Data[BytesRcv], 4-BytesRcv, 0);
|
||||
if(!CheckBytes(Temp)){
|
||||
UUl("Socket Closed Code 3");
|
||||
return "";
|
||||
}
|
||||
BytesRcv += Temp;
|
||||
}while(BytesRcv < 4);
|
||||
memcpy(&Header,&Data[0],sizeof(Header));
|
||||
|
||||
if(!CheckBytes(BytesRcv)){
|
||||
UUl("Socket Closed Code 4");
|
||||
return "";
|
||||
}
|
||||
Data.resize(Header);
|
||||
BytesRcv = 0;
|
||||
do{
|
||||
Temp = recv(TCPSocket, &Data[BytesRcv], Header-BytesRcv, 0);
|
||||
if(!CheckBytes(Temp)){
|
||||
UUl("Socket Closed Code 5");
|
||||
return "";
|
||||
}
|
||||
BytesRcv += Temp;
|
||||
}while(BytesRcv < Header);
|
||||
|
||||
std::string Ret(Data.data(),Header);
|
||||
|
||||
if (Ret.substr(0, 4) == "ABG:") {
|
||||
Ret = Zlib::DeComp(Ret.substr(4));
|
||||
}
|
||||
|
||||
if(Ret[0] == 'E')UUl(Ret.substr(1));
|
||||
return Ret;
|
||||
}
|
||||
Reference in New Issue
Block a user