mirror of
https://github.com/BeamMP/BeamMP-Launcher.git
synced 2026-08-27 12:41:32 +00:00
Compare commits
25 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 5295d5e710 | |||
| a891759d50 | |||
| 989cd3f0fc | |||
| 968c5d318e | |||
| f98e9a7746 | |||
| d7277151cc | |||
| 52e607cf2c | |||
| 92381aa048 | |||
| 30639abb88 | |||
| 23760da53b | |||
| 809080c9aa | |||
| f1ec753ca3 | |||
| f30a88e76e | |||
| 18a10dfe8f | |||
| 7ae273f550 | |||
| 13102552b7 | |||
| 8e4cfe55b9 | |||
| 1a7bd19bc6 | |||
| f79f05bc9d | |||
| b1e29a9295 | |||
| c6ae924346 | |||
| 6e26a28831 | |||
| e3049b3ba5 | |||
| 5cb4d79cdd | |||
| dbb0ef2a5c |
@@ -20,7 +20,7 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
vcpkgArguments: 'zlib nlohmann-json openssl cpp-httplib[openssl]'
|
vcpkgArguments: 'zlib nlohmann-json openssl cpp-httplib[openssl]'
|
||||||
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
|
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
|
||||||
vcpkgGitCommitId: '40616a5e954f7be1077ef37db3fbddbd5dcd1ca6'
|
vcpkgGitCommitId: 'b5d1a94fb7f88fd835e360fd23a45a09ceedbf48'
|
||||||
|
|
||||||
- name: Create Build Environment
|
- name: Create Build Environment
|
||||||
run: cmake -E make_directory ${{github.workspace}}/build-linux
|
run: cmake -E make_directory ${{github.workspace}}/build-linux
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
vcpkgArguments: 'discord-rpc zlib nlohmann-json openssl cpp-httplib[openssl]'
|
vcpkgArguments: 'discord-rpc zlib nlohmann-json openssl cpp-httplib[openssl]'
|
||||||
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
|
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
|
||||||
vcpkgGitCommitId: '40616a5e954f7be1077ef37db3fbddbd5dcd1ca6'
|
vcpkgGitCommitId: 'b5d1a94fb7f88fd835e360fd23a45a09ceedbf48'
|
||||||
vcpkgTriplet: 'x64-windows-static'
|
vcpkgTriplet: 'x64-windows-static'
|
||||||
|
|
||||||
- name: Create Build Environment
|
- name: Create Build Environment
|
||||||
|
|||||||
+1
-1
@@ -37,4 +37,4 @@ else(WIN32) #MINGW
|
|||||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Os -s --static")
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Os -s --static")
|
||||||
target_link_libraries(${PROJECT_NAME} ssl crypto ws2_32 ssp crypt32 z CURL::libcurl)
|
target_link_libraries(${PROJECT_NAME} ssl crypto ws2_32 ssp crypt32 z CURL::libcurl)
|
||||||
endif(WIN32)
|
endif(WIN32)
|
||||||
target_include_directories(${PROJECT_NAME} PRIVATE "include")
|
target_include_directories(${PROJECT_NAME} PRIVATE "include" "include/Network")
|
||||||
|
|||||||
+3
-3
@@ -11,9 +11,9 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
class HTTP {
|
class HTTP {
|
||||||
public:
|
public:
|
||||||
static bool Download(const std::string& IP, const beammp_fs_string& Path, const std::string& Hash);
|
static bool Download(const std::string& IP, const beammp_fs_string& Path, const std::string& Hash, const bool& redirect = true);
|
||||||
static std::string Post(const std::string& IP, const std::string& Fields);
|
static std::string Post(std::string IP, const std::string& Fields, const bool& redirect = true);
|
||||||
static std::string Get(const std::string& IP);
|
static std::string Get(std::string IP, const bool& redirect = true);
|
||||||
static bool ProgressBar(size_t c, size_t t);
|
static bool ProgressBar(size_t c, size_t t);
|
||||||
static void StartProxy();
|
static void StartProxy();
|
||||||
public:
|
public:
|
||||||
|
|||||||
@@ -0,0 +1,22 @@
|
|||||||
|
/*
|
||||||
|
Copyright (C) 2024 BeamMP Ltd., BeamMP team and contributors.
|
||||||
|
Licensed under AGPL-3.0 (or later), see <https://www.gnu.org/licenses/>.
|
||||||
|
SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
#include <string>
|
||||||
|
#include <array>
|
||||||
|
|
||||||
|
class RegionHandler final {
|
||||||
|
public:
|
||||||
|
RegionHandler() = delete;
|
||||||
|
static void TopLevelDomainFailed();
|
||||||
|
static std::string RegionToTopLevelDomain();
|
||||||
|
static std::string RedirectURL(const std::string& URL);
|
||||||
|
private:
|
||||||
|
static inline unsigned int mRegionIndex { 0 };
|
||||||
|
const static inline std::array<std::string, 2> mValidTLDs {"beammp.com", "beammp.net"};
|
||||||
|
};
|
||||||
@@ -42,7 +42,7 @@ extern std::string magic;
|
|||||||
int KillSocket(uint64_t Dead);
|
int KillSocket(uint64_t Dead);
|
||||||
void UUl(const std::string& R);
|
void UUl(const std::string& R);
|
||||||
void UDPSend(std::string Data);
|
void UDPSend(std::string Data);
|
||||||
bool CheckBytes(int32_t Bytes);
|
bool CheckBytes(int32_t Bytes, int32_t Expected = -1);
|
||||||
void GameSend(std::string_view Data);
|
void GameSend(std::string_view Data);
|
||||||
void SendLarge(std::string Data);
|
void SendLarge(std::string Data);
|
||||||
std::string TCPRcv(uint64_t Sock);
|
std::string TCPRcv(uint64_t Sock);
|
||||||
@@ -56,3 +56,4 @@ void UDPClientMain(const std::string& IP, int Port);
|
|||||||
void TCPGameServer(const std::string& IP, int Port);
|
void TCPGameServer(const std::string& IP, int Port);
|
||||||
bool SecurityWarning();
|
bool SecurityWarning();
|
||||||
void CoreSend(std::string data);
|
void CoreSend(std::string data);
|
||||||
|
int RecvWaitAll(int sockfd, char *buf, int len);
|
||||||
|
|||||||
@@ -20,6 +20,7 @@ struct Options {
|
|||||||
bool no_update = false;
|
bool no_update = false;
|
||||||
bool no_launch = false;
|
bool no_launch = false;
|
||||||
const char* user_path = nullptr;
|
const char* user_path = nullptr;
|
||||||
|
std::string region;
|
||||||
const char **game_arguments = nullptr;
|
const char **game_arguments = nullptr;
|
||||||
int game_arguments_length = 0;
|
int game_arguments_length = 0;
|
||||||
const char** argv = nullptr;
|
const char** argv = nullptr;
|
||||||
|
|||||||
@@ -49,6 +49,9 @@ void ParseConfig(const nlohmann::json& d) {
|
|||||||
deleteDuplicateMods = d["DeleteDuplicateMods"].get<bool>();
|
deleteDuplicateMods = d["DeleteDuplicateMods"].get<bool>();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (d.contains("Region") && d["Region"].is_string()) {
|
||||||
|
options.region = d["Region"].get<std::string>();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConfigInit() {
|
void ConfigInit() {
|
||||||
|
|||||||
+11
-10
@@ -5,6 +5,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include "Http.h"
|
#include "Http.h"
|
||||||
|
#include "RegionHandler.h"
|
||||||
#include "Network/network.hpp"
|
#include "Network/network.hpp"
|
||||||
#include "Security/Init.h"
|
#include "Security/Init.h"
|
||||||
#include "Utils.h"
|
#include "Utils.h"
|
||||||
@@ -143,29 +144,28 @@ void GetServerInfo(std::string Data) {
|
|||||||
const std::string buffer = ([&]() -> std::string {
|
const std::string buffer = ([&]() -> std::string {
|
||||||
int32_t Header;
|
int32_t Header;
|
||||||
std::vector<char> data(sizeof(Header));
|
std::vector<char> data(sizeof(Header));
|
||||||
int Temp = recv(ISock, data.data(), sizeof(Header), MSG_WAITALL);
|
int Temp = RecvWaitAll(ISock, data.data(), sizeof(Header));
|
||||||
|
|
||||||
auto checkBytes = ([&](const int32_t bytes) -> bool {
|
auto checkBytes = ([&](const int32_t bytes, const int32_t expected = -1) -> bool {
|
||||||
if (bytes == 0) {
|
if (bytes == 0) {
|
||||||
return false;
|
return false;
|
||||||
} else if (bytes < 0) {
|
} else if (bytes < 0) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
if (expected != -1 && bytes != expected) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
});
|
});
|
||||||
|
|
||||||
if (!checkBytes(Temp)) {
|
if (!checkBytes(Temp, sizeof(Header))) {
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
memcpy(&Header, data.data(), sizeof(Header));
|
memcpy(&Header, data.data(), sizeof(Header));
|
||||||
|
|
||||||
if (!checkBytes(Temp)) {
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
data.resize(Header, 0);
|
data.resize(Header, 0);
|
||||||
Temp = recv(ISock, data.data(), Header, MSG_WAITALL);
|
Temp = RecvWaitAll(ISock, data.data(), Header);
|
||||||
if (!checkBytes(Temp)) {
|
if (!checkBytes(Temp, Header)) {
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
return std::string(data.data(), Header);
|
return std::string(data.data(), Header);
|
||||||
@@ -227,7 +227,8 @@ void Parse(std::string Data, SOCKET CSocket) {
|
|||||||
TCPTerminate = true;
|
TCPTerminate = true;
|
||||||
Data.clear();
|
Data.clear();
|
||||||
futures.push_back(std::async(std::launch::async, []() {
|
futures.push_back(std::async(std::launch::async, []() {
|
||||||
CoreSend("B" + HTTP::Get("https://backend.beammp.com/servers-info"));
|
std::string resp = HTTP::Get("https://backend.beammp.com/servers-info", true);
|
||||||
|
CoreSend("B" + resp);
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
|||||||
@@ -50,17 +50,6 @@ int KillSocket(uint64_t Dead) {
|
|||||||
return a;
|
return a;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool CheckBytes(uint32_t Bytes) {
|
|
||||||
if (Bytes == 0) {
|
|
||||||
debug("(Proxy) Connection closing");
|
|
||||||
return false;
|
|
||||||
} else if (Bytes < 0) {
|
|
||||||
debug("(Proxy) send failed with error: " + std::to_string(WSAGetLastError()));
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void GameSend(std::string_view Data) {
|
void GameSend(std::string_view Data) {
|
||||||
static std::mutex Lock;
|
static std::mutex Lock;
|
||||||
std::scoped_lock Guard(Lock);
|
std::scoped_lock Guard(Lock);
|
||||||
|
|||||||
+36
-45
@@ -6,6 +6,8 @@
|
|||||||
|
|
||||||
|
|
||||||
#include "Http.h"
|
#include "Http.h"
|
||||||
|
#include "RegionHandler.h"
|
||||||
|
#include "Options.h"
|
||||||
#include <Logger.h>
|
#include <Logger.h>
|
||||||
#include <Network/network.hpp>
|
#include <Network/network.hpp>
|
||||||
#include <Startup.h>
|
#include <Startup.h>
|
||||||
@@ -20,45 +22,6 @@
|
|||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
void WriteHttpDebug(const httplib::Client& client, const std::string& method, const std::string& target, const httplib::Result& result) try {
|
|
||||||
const std::filesystem::path folder = ".https_debug";
|
|
||||||
std::filesystem::create_directories(folder);
|
|
||||||
if (!std::filesystem::exists(folder / "WHAT IS THIS FOLDER.txt")) {
|
|
||||||
std::ofstream ignore { folder / "WHAT IS THIS FOLDER.txt" };
|
|
||||||
ignore << "This folder exists to help debug current issues with the backend. Do not share this folder with anyone but BeamMP staff. It contains detailed logs of any failed http requests." << std::endl;
|
|
||||||
}
|
|
||||||
const auto file = folder / (method + ".json");
|
|
||||||
// 1 MB limit
|
|
||||||
if (std::filesystem::exists(file) && std::filesystem::file_size(file) > 1'000'000) {
|
|
||||||
std::filesystem::rename(file, file.generic_string() + ".bak");
|
|
||||||
}
|
|
||||||
|
|
||||||
std::ofstream of { file, std::ios::app };
|
|
||||||
nlohmann::json js {
|
|
||||||
{ "utc", std::chrono::system_clock::now().time_since_epoch().count() },
|
|
||||||
{ "target", target },
|
|
||||||
{ "client_info", {
|
|
||||||
{ "openssl_verify_result", client.get_openssl_verify_result() },
|
|
||||||
{ "host", client.host() },
|
|
||||||
{ "port", client.port() },
|
|
||||||
{ "socket_open", client.is_socket_open() },
|
|
||||||
{ "valid", client.is_valid() },
|
|
||||||
} },
|
|
||||||
};
|
|
||||||
if (result) {
|
|
||||||
auto value = result.value();
|
|
||||||
js["result"] = {};
|
|
||||||
js["result"]["body"] = value.body;
|
|
||||||
js["result"]["status"] = value.status;
|
|
||||||
js["result"]["headers"] = value.headers;
|
|
||||||
js["result"]["version"] = value.version;
|
|
||||||
js["result"]["location"] = value.location;
|
|
||||||
js["result"]["reason"] = value.reason;
|
|
||||||
}
|
|
||||||
of << js.dump();
|
|
||||||
} catch (const std::exception& e) {
|
|
||||||
error(e.what());
|
|
||||||
}
|
|
||||||
|
|
||||||
static size_t CurlWriteCallback(void* contents, size_t size, size_t nmemb, void* userp) {
|
static size_t CurlWriteCallback(void* contents, size_t size, size_t nmemb, void* userp) {
|
||||||
std::string* Result = reinterpret_cast<std::string*>(userp);
|
std::string* Result = reinterpret_cast<std::string*>(userp);
|
||||||
@@ -68,9 +31,12 @@ static size_t CurlWriteCallback(void* contents, size_t size, size_t nmemb, void*
|
|||||||
}
|
}
|
||||||
|
|
||||||
bool HTTP::isDownload = false;
|
bool HTTP::isDownload = false;
|
||||||
std::string HTTP::Get(const std::string& IP) {
|
std::string HTTP::Get(std::string IP, const bool& redirect) {
|
||||||
std::string Ret;
|
std::string Ret;
|
||||||
static thread_local CURL* curl = curl_easy_init();
|
static thread_local CURL* curl = curl_easy_init();
|
||||||
|
if (redirect) {
|
||||||
|
IP = RegionHandler::RedirectURL(IP);
|
||||||
|
}
|
||||||
if (curl) {
|
if (curl) {
|
||||||
CURLcode res;
|
CURLcode res;
|
||||||
char errbuf[CURL_ERROR_SIZE];
|
char errbuf[CURL_ERROR_SIZE];
|
||||||
@@ -85,7 +51,18 @@ std::string HTTP::Get(const std::string& IP) {
|
|||||||
if (res != CURLE_OK) {
|
if (res != CURLE_OK) {
|
||||||
error("GET to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
error("GET to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
||||||
error("Curl error: " + std::string(errbuf));
|
error("Curl error: " + std::string(errbuf));
|
||||||
return "";
|
if (!redirect) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
RegionHandler::TopLevelDomainFailed();
|
||||||
|
IP = RegionHandler::RedirectURL(IP);
|
||||||
|
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||||
|
res = curl_easy_perform(curl);
|
||||||
|
if (res != CURLE_OK) {
|
||||||
|
error("GET to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
||||||
|
error("Curl error: " + std::string(errbuf));
|
||||||
|
return "";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
error("Curl easy init failed");
|
error("Curl easy init failed");
|
||||||
@@ -94,9 +71,12 @@ std::string HTTP::Get(const std::string& IP) {
|
|||||||
return Ret;
|
return Ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string HTTP::Post(const std::string& IP, const std::string& Fields) {
|
std::string HTTP::Post(std::string IP, const std::string& Fields, const bool& redirect) {
|
||||||
std::string Ret;
|
std::string Ret;
|
||||||
static thread_local CURL* curl = curl_easy_init();
|
static thread_local CURL* curl = curl_easy_init();
|
||||||
|
if (redirect) {
|
||||||
|
IP = RegionHandler::RedirectURL(IP);
|
||||||
|
}
|
||||||
if (curl) {
|
if (curl) {
|
||||||
CURLcode res;
|
CURLcode res;
|
||||||
char errbuf[CURL_ERROR_SIZE];
|
char errbuf[CURL_ERROR_SIZE];
|
||||||
@@ -118,7 +98,18 @@ std::string HTTP::Post(const std::string& IP, const std::string& Fields) {
|
|||||||
if (res != CURLE_OK) {
|
if (res != CURLE_OK) {
|
||||||
error("POST to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
error("POST to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
||||||
error("Curl error: " + std::string(errbuf));
|
error("Curl error: " + std::string(errbuf));
|
||||||
return "";
|
if (!redirect) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
RegionHandler::TopLevelDomainFailed();
|
||||||
|
IP = RegionHandler::RedirectURL(IP);
|
||||||
|
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str());
|
||||||
|
res = curl_easy_perform(curl);
|
||||||
|
if (res != CURLE_OK) {
|
||||||
|
error("GET to " + IP + " failed: " + std::string(curl_easy_strerror(res)));
|
||||||
|
error("Curl error: " + std::string(errbuf));
|
||||||
|
return "";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
error("Curl easy init failed");
|
error("Curl easy init failed");
|
||||||
@@ -127,12 +118,12 @@ std::string HTTP::Post(const std::string& IP, const std::string& Fields) {
|
|||||||
return Ret;
|
return Ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool HTTP::Download(const std::string& IP, const beammp_fs_string& Path, const std::string& Hash) {
|
bool HTTP::Download(const std::string& IP, const beammp_fs_string& Path, const std::string& Hash, const bool& redirect) {
|
||||||
static std::mutex Lock;
|
static std::mutex Lock;
|
||||||
std::scoped_lock Guard(Lock);
|
std::scoped_lock Guard(Lock);
|
||||||
|
|
||||||
info("Downloading an update (this may take a while)");
|
info("Downloading an update (this may take a while)");
|
||||||
std::string Ret = Get(IP);
|
std::string Ret = Get(IP, redirect);
|
||||||
|
|
||||||
if (Ret.empty()) {
|
if (Ret.empty()) {
|
||||||
error("Download failed");
|
error("Download failed");
|
||||||
|
|||||||
@@ -0,0 +1,39 @@
|
|||||||
|
/*
|
||||||
|
Copyright (C) 2024 BeamMP Ltd., BeamMP team and contributors.
|
||||||
|
Licensed under AGPL-3.0 (or later), see <https://www.gnu.org/licenses/>.
|
||||||
|
SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "RegionHandler.h"
|
||||||
|
#include "Logger.h"
|
||||||
|
#include "Options.h"
|
||||||
|
|
||||||
|
#include <regex>
|
||||||
|
|
||||||
|
void RegionHandler::TopLevelDomainFailed()
|
||||||
|
{
|
||||||
|
info("Top level domain of " + mValidTLDs[mRegionIndex % mValidTLDs.size()] + " didn't respond correctly , changing domain to " + mValidTLDs[(mRegionIndex + 1) % mValidTLDs.size()]);
|
||||||
|
mRegionIndex++;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string RegionHandler::RegionToTopLevelDomain()
|
||||||
|
{
|
||||||
|
static bool isDeveloperRegion = options.region == "Developer";
|
||||||
|
if (isDeveloperRegion) {
|
||||||
|
return "beammp.dev";
|
||||||
|
}
|
||||||
|
return mValidTLDs[mRegionIndex % mValidTLDs.size()]; // Global
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string RegionHandler::RedirectURL(const std::string &URL)
|
||||||
|
{
|
||||||
|
std::regex link_pattern(R"(^(https:\/\/.*)beammp\.com(\/.*)?$)");
|
||||||
|
std::smatch link_match;
|
||||||
|
if (std::regex_search(URL, link_match, link_pattern) && link_match.position() == 0) {
|
||||||
|
//TLD matched beammp.com
|
||||||
|
std::string before = link_match[1].str(); // "https://..." up to beammp
|
||||||
|
std::string after = link_match[2].matched ? link_match[2].str() : ""; // "/path" or ""
|
||||||
|
return before + RegionToTopLevelDomain() + after;
|
||||||
|
}
|
||||||
|
return URL; //if it didn't match, just return the unmodified URL
|
||||||
|
}
|
||||||
@@ -164,7 +164,7 @@ std::vector<char> TCPRcvRaw(SOCKET Sock, uint64_t& GRcv, uint64_t Size) {
|
|||||||
do {
|
do {
|
||||||
// receive at most some MB at a time
|
// receive at most some MB at a time
|
||||||
int Len = std::min(int(Size - Rcv), 1 * 1024 * 1024);
|
int Len = std::min(int(Size - Rcv), 1 * 1024 * 1024);
|
||||||
int Temp = recv(Sock, &File[Rcv], Len, MSG_WAITALL);
|
int Temp = RecvWaitAll(Sock, &File[Rcv], Len);
|
||||||
if (Temp == -1 || Temp == 0) {
|
if (Temp == -1 || Temp == 0) {
|
||||||
debug("Recv returned: " + std::to_string(Temp));
|
debug("Recv returned: " + std::to_string(Temp));
|
||||||
if (Temp == -1) {
|
if (Temp == -1) {
|
||||||
|
|||||||
@@ -94,7 +94,8 @@ void UDPClientMain(const std::string& IP, int Port) {
|
|||||||
inet_pton(AF_INET, IP.c_str(), &ToServer->sin_addr);
|
inet_pton(AF_INET, IP.c_str(), &ToServer->sin_addr);
|
||||||
UDPSock = socket(AF_INET, SOCK_DGRAM, 0);
|
UDPSock = socket(AF_INET, SOCK_DGRAM, 0);
|
||||||
if (!magic.empty())
|
if (!magic.empty())
|
||||||
UDPSend(magic);
|
for (int i = 0; i < 10; i++)
|
||||||
|
UDPSend(magic);
|
||||||
GameSend("P" + std::to_string(ClientID));
|
GameSend("P" + std::to_string(ClientID));
|
||||||
TCPSend("H", TCPSock);
|
TCPSend("H", TCPSock);
|
||||||
UDPSend("p");
|
UDPSend("p");
|
||||||
|
|||||||
@@ -28,9 +28,9 @@ int LastPort;
|
|||||||
std::string LastIP;
|
std::string LastIP;
|
||||||
SOCKET TCPSock = -1;
|
SOCKET TCPSock = -1;
|
||||||
|
|
||||||
bool CheckBytes(int32_t Bytes) {
|
bool CheckBytes(int32_t Bytes, int32_t Expected) {
|
||||||
if (Bytes == 0) {
|
if (Bytes == 0) {
|
||||||
debug("(TCP) Connection closing... CheckBytes(16)");
|
debug("(TCP) Connection closing...");
|
||||||
Terminate = true;
|
Terminate = true;
|
||||||
return false;
|
return false;
|
||||||
} else if (Bytes < 0) {
|
} else if (Bytes < 0) {
|
||||||
@@ -39,6 +39,11 @@ bool CheckBytes(int32_t Bytes) {
|
|||||||
Terminate = true;
|
Terminate = true;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
if (Expected != -1 && Bytes != Expected) {
|
||||||
|
debug(std::format("(TCP) Short recv detected, expected {} bytes, got {} bytes", Expected, Bytes));
|
||||||
|
Terminate = true;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
void UUl(const std::string& R) {
|
void UUl(const std::string& R) {
|
||||||
@@ -75,6 +80,24 @@ void TCPSend(const std::string& Data, uint64_t Sock) {
|
|||||||
} while (Sent < Size);
|
} while (Sent < Size);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int RecvWaitAll(int sockfd, char *buf, int len) {
|
||||||
|
// handle MSG_WAITALL not actually filling the whole buffer
|
||||||
|
// happens frequently in wine, and can also happen natively when the OS pauses the execution for various reasons
|
||||||
|
int offset = 0;
|
||||||
|
while (offset < len) {
|
||||||
|
int recv_status = recv(sockfd, &buf[offset], len - offset, MSG_WAITALL);
|
||||||
|
if (recv_status == 0) {
|
||||||
|
// do not discard received data when the other side closes the socket cleanly
|
||||||
|
return offset;
|
||||||
|
}
|
||||||
|
if (recv_status == -1) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
offset += recv_status;
|
||||||
|
}
|
||||||
|
return offset;
|
||||||
|
}
|
||||||
|
|
||||||
std::string TCPRcv(SOCKET Sock) {
|
std::string TCPRcv(SOCKET Sock) {
|
||||||
if (Sock == -1) {
|
if (Sock == -1) {
|
||||||
Terminate = true;
|
Terminate = true;
|
||||||
@@ -84,8 +107,8 @@ std::string TCPRcv(SOCKET Sock) {
|
|||||||
int32_t Header;
|
int32_t Header;
|
||||||
int Temp;
|
int Temp;
|
||||||
std::vector<char> Data(sizeof(Header));
|
std::vector<char> Data(sizeof(Header));
|
||||||
Temp = recv(Sock, Data.data(), sizeof(Header), MSG_WAITALL);
|
Temp = RecvWaitAll(Sock, Data.data(), sizeof(Header));
|
||||||
if (!CheckBytes(Temp)) {
|
if (!CheckBytes(Temp, sizeof(Header))) {
|
||||||
UUl("Socket Closed Code 3");
|
UUl("Socket Closed Code 3");
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
@@ -97,8 +120,8 @@ std::string TCPRcv(SOCKET Sock) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
Data.resize(Header, 0);
|
Data.resize(Header, 0);
|
||||||
Temp = recv(Sock, Data.data(), Header, MSG_WAITALL);
|
Temp = RecvWaitAll(Sock, Data.data(), Header);
|
||||||
if (!CheckBytes(Temp)) {
|
if (!CheckBytes(Temp, Header)) {
|
||||||
UUl("Socket Closed Code 5");
|
UUl("Socket Closed Code 5");
|
||||||
return "";
|
return "";
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -92,6 +92,13 @@ void InitOptions(int argc, const char *argv[], Options &options) {
|
|||||||
}
|
}
|
||||||
options.user_path = argv[i + 1];
|
options.user_path = argv[i + 1];
|
||||||
i++;
|
i++;
|
||||||
|
} else if (argument == "--region") {
|
||||||
|
if (i + 1 >= argc) {
|
||||||
|
error("You must specify a region after the `--region` argument");
|
||||||
|
}
|
||||||
|
options.region = argv[i + 1];
|
||||||
|
info("Set the region to: " + options.region);
|
||||||
|
i++;
|
||||||
} else if (argument == "--" || argument == "--game") {
|
} else if (argument == "--" || argument == "--game") {
|
||||||
options.game_arguments = &argv[i + 1];
|
options.game_arguments = &argv[i + 1];
|
||||||
options.game_arguments_length = argc - i - 1;
|
options.game_arguments_length = argc - i - 1;
|
||||||
|
|||||||
@@ -268,7 +268,7 @@ void LegitimacyCheck() {
|
|||||||
std::ifstream libraryFolders(libraryFoldersPath);
|
std::ifstream libraryFolders(libraryFoldersPath);
|
||||||
auto root = tyti::vdf::read(libraryFolders);
|
auto root = tyti::vdf::read(libraryFolders);
|
||||||
for (auto folderInfo : root.childs) {
|
for (auto folderInfo : root.childs) {
|
||||||
if (std::filesystem::exists(folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/integrity.json")){
|
if ((folderInfo.second->childs["apps"]->attribs).contains("284160") && std::filesystem::exists(folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/integrity.json")){
|
||||||
GameDir = folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/";
|
GameDir = folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/";
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,6 +6,8 @@
|
|||||||
|
|
||||||
|
|
||||||
#include "Http.h"
|
#include "Http.h"
|
||||||
|
#include "Options.h"
|
||||||
|
#include "RegionHandler.h"
|
||||||
#include "Logger.h"
|
#include "Logger.h"
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
|
|||||||
+187
-13
@@ -4,7 +4,6 @@
|
|||||||
SPDX-License-Identifier: AGPL-3.0-or-later
|
SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
#include "zip_file.h"
|
#include "zip_file.h"
|
||||||
#include <charconv>
|
#include <charconv>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
@@ -14,9 +13,9 @@
|
|||||||
#if defined(_WIN32)
|
#if defined(_WIN32)
|
||||||
#elif defined(__linux__)
|
#elif defined(__linux__)
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#elif defined (__APPLE__)
|
#elif defined(__APPLE__)
|
||||||
#include <unistd.h>
|
|
||||||
#include <libproc.h>
|
#include <libproc.h>
|
||||||
|
#include <unistd.h>
|
||||||
#endif // __APPLE__
|
#endif // __APPLE__
|
||||||
#include "Http.h"
|
#include "Http.h"
|
||||||
#include "Logger.h"
|
#include "Logger.h"
|
||||||
@@ -25,6 +24,7 @@
|
|||||||
#include "Security/Init.h"
|
#include "Security/Init.h"
|
||||||
#include "Startup.h"
|
#include "Startup.h"
|
||||||
#include "Utils.h"
|
#include "Utils.h"
|
||||||
|
#include "RegionHandler.h"
|
||||||
#include "hashpp.h"
|
#include "hashpp.h"
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
@@ -84,7 +84,7 @@ beammp_fs_string GetEN() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
std::string GetVer() {
|
std::string GetVer() {
|
||||||
return "2.7";
|
return "2.8";
|
||||||
}
|
}
|
||||||
std::string GetPatch() {
|
std::string GetPatch() {
|
||||||
return ".0";
|
return ".0";
|
||||||
@@ -114,7 +114,7 @@ fs::path GetBP(const beammp_fs_char* P) {
|
|||||||
// should instead be placed in Application Support.
|
// should instead be placed in Application Support.
|
||||||
proc_pidpath(pid, path, sizeof(path));
|
proc_pidpath(pid, path, sizeof(path));
|
||||||
fspath = std::string(path);
|
fspath = std::string(path);
|
||||||
#else
|
#else
|
||||||
fspath = beammp_fs_string(P);
|
fspath = beammp_fs_string(P);
|
||||||
#endif
|
#endif
|
||||||
fspath = fs::weakly_canonical(fspath.string() + "/..");
|
fspath = fs::weakly_canonical(fspath.string() + "/..");
|
||||||
@@ -194,10 +194,155 @@ void CheckName() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#if defined(_WIN32)
|
||||||
|
#include <wincrypt.h>
|
||||||
|
#include <wintrust.h>
|
||||||
|
#include <softpub.h>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
#pragma comment(lib, "wintrust.lib")
|
||||||
|
#pragma comment(lib, "crypt32.lib")
|
||||||
|
|
||||||
|
bool CheckThumbprint(std::filesystem::path filepath)
|
||||||
|
{
|
||||||
|
HCERTSTORE hStore = NULL;
|
||||||
|
HCRYPTMSG hMsg = NULL;
|
||||||
|
|
||||||
|
if (!CryptQueryObject(
|
||||||
|
CERT_QUERY_OBJECT_FILE,
|
||||||
|
filepath.wstring().c_str(),
|
||||||
|
CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED_EMBED,
|
||||||
|
CERT_QUERY_FORMAT_FLAG_BINARY,
|
||||||
|
0,
|
||||||
|
NULL, NULL, NULL,
|
||||||
|
&hStore,
|
||||||
|
&hMsg,
|
||||||
|
NULL))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DWORD dwSignerInfo = 0;
|
||||||
|
if (!CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM, 0, NULL, &dwSignerInfo) || dwSignerInfo == 0)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
PCMSG_SIGNER_INFO pSignerInfo = (PCMSG_SIGNER_INFO)LocalAlloc(LPTR, dwSignerInfo);
|
||||||
|
if (!CryptMsgGetParam(hMsg, CMSG_SIGNER_INFO_PARAM, 0, pSignerInfo, &dwSignerInfo))
|
||||||
|
{
|
||||||
|
LocalFree(pSignerInfo);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
CERT_INFO certInfo = {};
|
||||||
|
certInfo.Issuer = pSignerInfo->Issuer;
|
||||||
|
certInfo.SerialNumber = pSignerInfo->SerialNumber;
|
||||||
|
|
||||||
|
PCCERT_CONTEXT pCertContext = CertFindCertificateInStore(
|
||||||
|
hStore,
|
||||||
|
X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
|
||||||
|
0,
|
||||||
|
CERT_FIND_SUBJECT_CERT,
|
||||||
|
&certInfo,
|
||||||
|
NULL);
|
||||||
|
|
||||||
|
if (!pCertContext)
|
||||||
|
{
|
||||||
|
LocalFree(pSignerInfo);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
BYTE hash[64];
|
||||||
|
DWORD hashSize = sizeof(hash);
|
||||||
|
|
||||||
|
bool match = false;
|
||||||
|
if (CertGetCertificateContextProperty(pCertContext, CERT_SHA256_HASH_PROP_ID, hash, &hashSize))
|
||||||
|
{
|
||||||
|
std::string pubKeyData(
|
||||||
|
reinterpret_cast<const char*>(pCertContext->pCertInfo->SubjectPublicKeyInfo.PublicKey.pbData),
|
||||||
|
pCertContext->pCertInfo->SubjectPublicKeyInfo.PublicKey.cbData
|
||||||
|
);
|
||||||
|
|
||||||
|
std::string pubKeyHash = Utils::GetSha256HashReallyFast(pubKeyData, L"PubKey");
|
||||||
|
debug("pub key hash: " + pubKeyHash);
|
||||||
|
|
||||||
|
std::string fileThumbprint;
|
||||||
|
for (DWORD i = 0; i < hashSize; i++)
|
||||||
|
{
|
||||||
|
char buf[3];
|
||||||
|
sprintf_s(buf, "%02x", hash[i]);
|
||||||
|
fileThumbprint += buf;
|
||||||
|
}
|
||||||
|
|
||||||
|
debug("File thumbprint: " +fileThumbprint);
|
||||||
|
debug(filepath);
|
||||||
|
|
||||||
|
if (fileThumbprint == "937f055b713de69416926ed4651d65219a0a0e77d7a78c1932c007e14326da33" && pubKeyHash == "2afad4a5773b0ac449f48350ce0d09c372be0d5bcbaa6d01332ce000baffde99"){
|
||||||
|
match = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
CertFreeCertificateContext(pCertContext);
|
||||||
|
CertCloseStore(hStore, 0);
|
||||||
|
LocalFree(pSignerInfo);
|
||||||
|
|
||||||
|
return match;
|
||||||
|
}
|
||||||
|
#include <windows.h>
|
||||||
|
#include <wintrust.h>
|
||||||
|
#include <Softpub.h>
|
||||||
|
#include <filesystem>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#pragma comment(lib, "wintrust")
|
||||||
|
|
||||||
|
bool VerifySignature(const std::filesystem::path& filePath)
|
||||||
|
{
|
||||||
|
std::wstring path = filePath.wstring();
|
||||||
|
|
||||||
|
WINTRUST_FILE_INFO fileInfo = {};
|
||||||
|
fileInfo.cbStruct = sizeof(WINTRUST_FILE_INFO);
|
||||||
|
fileInfo.pcwszFilePath = path.c_str();
|
||||||
|
fileInfo.hFile = NULL;
|
||||||
|
fileInfo.pgKnownSubject = NULL;
|
||||||
|
|
||||||
|
WINTRUST_DATA winTrustData = {};
|
||||||
|
winTrustData.cbStruct = sizeof(WINTRUST_DATA);
|
||||||
|
winTrustData.dwUIChoice = WTD_UI_NONE;
|
||||||
|
winTrustData.dwUnionChoice = WTD_CHOICE_FILE;
|
||||||
|
winTrustData.pFile = &fileInfo;
|
||||||
|
|
||||||
|
winTrustData.dwStateAction = WTD_STATEACTION_VERIFY;
|
||||||
|
|
||||||
|
GUID policyGUID = WINTRUST_ACTION_GENERIC_VERIFY_V2;
|
||||||
|
|
||||||
|
LONG status = WinVerifyTrust(
|
||||||
|
NULL,
|
||||||
|
&policyGUID,
|
||||||
|
&winTrustData
|
||||||
|
);
|
||||||
|
|
||||||
|
debug(filePath);
|
||||||
|
debug("Signature check code: " + std::to_string(status));
|
||||||
|
|
||||||
|
winTrustData.dwStateAction = WTD_STATEACTION_CLOSE;
|
||||||
|
WinVerifyTrust(NULL, &policyGUID, &winTrustData);
|
||||||
|
|
||||||
|
return (status == CERT_E_UNTRUSTEDROOT);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
void CheckForUpdates(const std::string& CV) {
|
void CheckForUpdates(const std::string& CV) {
|
||||||
std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/launcher?branch=" + Branch + "&pk=" + PublicKey);
|
std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/launcher?branch=" + Branch + "&pk=" + PublicKey);
|
||||||
std::string LatestVersion = HTTP::Get(
|
std::string LatestVersion = HTTP::Get("https://backend.beammp.com/version/launcher?branch=" + Branch + "&pk=" + PublicKey);
|
||||||
"https://backend.beammp.com/version/launcher?branch=" + Branch + "&pk=" + PublicKey);
|
|
||||||
|
std::regex sha256_pattern(R"(^[a-fA-F0-9]{64}$)");
|
||||||
|
std::smatch match;
|
||||||
|
|
||||||
|
if (LatestHash.length() != 64 || !std::regex_match(LatestHash, match, sha256_pattern)) {
|
||||||
|
error("Invalid hash from backend, skipping update check.");
|
||||||
|
debug("Launcher hash in question: " + LatestHash);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower);
|
transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower);
|
||||||
beammp_fs_string BP(GetBP() / GetEN()), Back(GetBP() / beammp_wide("BeamMP-Launcher.back"));
|
beammp_fs_string BP(GetBP() / GetEN()), Back(GetBP() / beammp_wide("BeamMP-Launcher.back"));
|
||||||
@@ -211,11 +356,25 @@ void CheckForUpdates(const std::string& CV) {
|
|||||||
error("Auto update is NOT implemented for the Linux version. Please update manually ASAP as updates contain security patches.");
|
error("Auto update is NOT implemented for the Linux version. Please update manually ASAP as updates contain security patches.");
|
||||||
#else
|
#else
|
||||||
info("Downloading Launcher update " + LatestHash);
|
info("Downloading Launcher update " + LatestHash);
|
||||||
if (HTTP::Download(
|
std::wstring DownloadLocation = GetBP() / (beammp_wide("new_") + GetEN());
|
||||||
"https://backend.beammp.com/builds/launcher?download=true"
|
bool downloadSuccess = false;
|
||||||
"&pk="
|
downloadSuccess = HTTP::Download(
|
||||||
+ PublicKey + "&branch=" + Branch,
|
"https://backend.beammp.com/builds/launcher?download=true"
|
||||||
GetBP() / (beammp_wide("new_") + GetEN()), LatestHash)) {
|
"&pk="
|
||||||
|
+ PublicKey + "&branch=" + Branch,
|
||||||
|
DownloadLocation, LatestHash);
|
||||||
|
if (downloadSuccess) {
|
||||||
|
if (!VerifySignature(DownloadLocation) || !CheckThumbprint(DownloadLocation)) {
|
||||||
|
std::error_code ec;
|
||||||
|
fs::remove(DownloadLocation, ec);
|
||||||
|
if (ec) {
|
||||||
|
error("Failed to remove broken launcher update");
|
||||||
|
}
|
||||||
|
throw std::runtime_error("The authenticity of the updated launcher could not be verified, it was corrupted or tampered with.");
|
||||||
|
}
|
||||||
|
|
||||||
|
info("Update signature is valid");
|
||||||
|
|
||||||
std::error_code ec;
|
std::error_code ec;
|
||||||
fs::remove(Back, ec);
|
fs::remove(Back, ec);
|
||||||
if (ec == std::errc::permission_denied) {
|
if (ec == std::errc::permission_denied) {
|
||||||
@@ -227,6 +386,13 @@ void CheckForUpdates(const std::string& CV) {
|
|||||||
fs::rename(GetBP() / (beammp_wide("new_") + GetEN()), BP);
|
fs::rename(GetBP() / (beammp_wide("new_") + GetEN()), BP);
|
||||||
URelaunch();
|
URelaunch();
|
||||||
} else {
|
} else {
|
||||||
|
if (fs::exists(DownloadLocation)) {
|
||||||
|
std::error_code error_code;
|
||||||
|
fs::remove(DownloadLocation, error_code);
|
||||||
|
if (error_code) {
|
||||||
|
error("Failed to remove broken launcher update");
|
||||||
|
}
|
||||||
|
}
|
||||||
throw std::runtime_error("Failed to download the launcher update! Please try manually updating it, https://docs.beammp.com/FAQ/Update-launcher/");
|
throw std::runtime_error("Failed to download the launcher update! Please try manually updating it, https://docs.beammp.com/FAQ/Update-launcher/");
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
@@ -238,7 +404,6 @@ void CheckForUpdates(const std::string& CV) {
|
|||||||
TraceBack++;
|
TraceBack++;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
void LinuxPatch() {
|
void LinuxPatch() {
|
||||||
HKEY hKey = nullptr;
|
HKEY hKey = nullptr;
|
||||||
@@ -357,6 +522,15 @@ void PreGame(const beammp_fs_string& GamePath) {
|
|||||||
[](auto const& c) -> bool { return !std::isalnum(c); }),
|
[](auto const& c) -> bool { return !std::isalnum(c); }),
|
||||||
LatestHash.end());
|
LatestHash.end());
|
||||||
|
|
||||||
|
std::regex sha256_pattern(R"(^[a-fA-F0-9]{64}$)");
|
||||||
|
std::smatch match;
|
||||||
|
|
||||||
|
if (LatestHash.length() != 64 || !std::regex_match(LatestHash, match, sha256_pattern)) {
|
||||||
|
error("Invalid hash from backend, skipping mod update check.");
|
||||||
|
debug("Mod hash in question: " + LatestHash);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
if (!fs::exists(GetGamePath() / beammp_wide("mods/multiplayer"))) {
|
if (!fs::exists(GetGamePath() / beammp_wide("mods/multiplayer"))) {
|
||||||
fs::create_directories(GetGamePath() / beammp_wide("mods/multiplayer"));
|
fs::create_directories(GetGamePath() / beammp_wide("mods/multiplayer"));
|
||||||
|
|||||||
Reference in New Issue
Block a user