1 Commits

Author SHA1 Message Date
Starystars67
7217cb1cab Update Startup.cpp 2023-12-16 03:01:28 +00:00
74 changed files with 1802 additions and 3113 deletions

View File

@@ -1,5 +0,0 @@
---
BasedOnStyle: WebKit
BreakBeforeBraces: Attach
SpaceAfterTemplateKeyword: false
...

View File

@@ -1,42 +0,0 @@
name: CMake Linux Build
on: [push, pull_request, workflow_dispatch]
env:
BUILD_TYPE: Release
jobs:
linux-build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
with:
submodules: 'true'
- name: Restore artifacts, or run vcpkg, build and cache artifacts
uses: lukka/run-vcpkg@v7
id: runvcpkg
with:
vcpkgArguments: 'zlib nlohmann-json openssl cpp-httplib[openssl]'
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
vcpkgGitCommitId: '40616a5e954f7be1077ef37db3fbddbd5dcd1ca6'
- name: Create Build Environment
run: cmake -E make_directory ${{github.workspace}}/build-linux
- name: Configure CMake
shell: bash
working-directory: ${{github.workspace}}/build-linux
run: cmake $GITHUB_WORKSPACE -DCMAKE_BUILD_TYPE=$BUILD_TYPE -DCMAKE_TOOLCHAIN_FILE='${{ runner.workspace }}/b/vcpkg/scripts/buildsystems/vcpkg.cmake'
- name: Build
working-directory: ${{github.workspace}}/build-linux
shell: bash
run: cmake --build . --config $BUILD_TYPE
- name: Archive artifacts
uses: actions/upload-artifact@v4
with:
name: BeamMP-Launcher
path: ${{github.workspace}}/build-linux/BeamMP-Launcher

View File

@@ -1,6 +1,6 @@
name: CMake Windows Build name: CMake Windows Build
on: [push, pull_request, workflow_dispatch] on: [push, pull_request]
env: env:
BUILD_TYPE: Release BUILD_TYPE: Release
@@ -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: '16ee2ecb31788c336ace8bb14c21801efb6836e4'
vcpkgTriplet: 'x64-windows-static' vcpkgTriplet: 'x64-windows-static'
- name: Create Build Environment - name: Create Build Environment
@@ -37,7 +37,7 @@ jobs:
run: cmake --build . --config $BUILD_TYPE run: cmake --build . --config $BUILD_TYPE
- name: Archive artifacts - name: Archive artifacts
uses: actions/upload-artifact@v4 uses: actions/upload-artifact@v2
with: with:
name: BeamMP-Launcher.exe name: BeamMP-Launcher.exe
path: ${{github.workspace}}/build-windows/Release/BeamMP-Launcher.exe path: ${{github.workspace}}/build-windows/Release/BeamMP-Launcher.exe

72
.github/workflows/release-build.yml vendored Normal file
View File

@@ -0,0 +1,72 @@
name: Release Create & Build
on:
push:
# Sequence of patterns matched against refs/tags
tags:
- 'v*' # Push events to matching v*, i.e. v1.0, v20.15.10
env:
BUILD_TYPE: Release
jobs:
create-release:
runs-on: ubuntu-latest
name: Create Release
outputs:
upload_url: ${{ steps.create_release.outputs.upload_url }}
steps:
- name: Create Release
id: create_release
uses: actions/create-release@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
tag_name: ${{ github.ref }}
release_name: Release ${{ github.ref }}
draft: false
prerelease: false
body: |
Files included in this release:
- `BeamMP-Launcher.exe` windows build
upload-release-files-windows:
name: Upload Windows Release Files
runs-on: windows-latest
needs: create-release
steps:
- uses: actions/checkout@v2
with:
submodules: 'true'
- name: Restore artifacts, or run vcpkg, build and cache artifacts
uses: lukka/run-vcpkg@main
id: runvcpkg
with:
vcpkgArguments: 'discord-rpc zlib nlohmann-json openssl cpp-httplib[openssl]'
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
vcpkgGitCommitId: '16ee2ecb31788c336ace8bb14c21801efb6836e4'
vcpkgTriplet: 'x64-windows-static'
- name: Create Build Environment
run: cmake -E make_directory ${{github.workspace}}/build-windows
- name: Configure CMake
shell: bash
working-directory: ${{github.workspace}}/build-windows
run: cmake $GITHUB_WORKSPACE -DCMAKE_BUILD_TYPE=$BUILD_TYPE -DCMAKE_TOOLCHAIN_FILE='${{ runner.workspace }}/b/vcpkg/scripts/buildsystems/vcpkg.cmake' -DVCPKG_TARGET_TRIPLET=x64-windows-static
- name: Build
working-directory: ${{github.workspace}}/build-windows
shell: bash
run: cmake --build . --config $BUILD_TYPE
- name: Upload Release Asset
id: upload-release-asset
uses: actions/upload-release-asset@v1
env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
with:
upload_url: ${{ needs.create-release.outputs.upload_url }} # This pulls from the CREATE RELEASE step above, referencing it's ID to get its outputs object, which include a `upload_url`. See this blog post for more info: https://jasonet.co/posts/new-features-of-github-actions/#passing-data-to-future-steps
asset_path: ${{github.workspace}}/build-windows/Release/BeamMP-Launcher.exe
asset_name: BeamMP-Launcher.exe
asset_content_type: application/vnd.microsoft.portable-executable

9
.gitignore vendored
View File

@@ -4,11 +4,4 @@ cmake-build-release
*.log *.log
/*.sh /*.sh
/*.obj /*.obj
/*.exe /*.exe
.cache/
.https_debug/
Launcher.cfg
Resources/
bin/
compile_commands.json
key

View File

@@ -1,5 +1,4 @@
cmake_minimum_required(VERSION 3.10) cmake_minimum_required(VERSION 3.10)
project(Launcher) project(Launcher)
if (WIN32) if (WIN32)
@@ -12,29 +11,32 @@ set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DDEBUG") set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -DDEBUG")
add_compile_definitions(CPPHTTPLIB_OPENSSL_SUPPORT) file(GLOB source_files "src/*.cpp" "src/*/*.cpp" "src/*/*.hpp" "include/*.h" "include/*/*.h" "include/*/*/*.h")
file(GLOB source_files "src/*.cpp" "src/*/*.cpp" "src/*/*.hpp" "include/*.h" "include/*/*.h" "include/*/*/*.h" "include/*.hpp" "include/*/*.hpp" "include/*/*/*.hpp")
find_package(httplib CONFIG REQUIRED) find_package(httplib CONFIG REQUIRED)
find_package(nlohmann_json CONFIG REQUIRED) find_package(nlohmann_json CONFIG REQUIRED)
find_package(CURL REQUIRED)
add_executable(${PROJECT_NAME} ${source_files}) add_executable(${PROJECT_NAME} ${source_files})
set_target_properties(${PROJECT_NAME} PROPERTIES OUTPUT_NAME "BeamMP-Launcher") set_target_properties(${PROJECT_NAME} PROPERTIES OUTPUT_NAME "BeamMP-Launcher")
if (WIN32) if (WIN32)
find_package(ZLIB REQUIRED) find_package(ZLIB REQUIRED)
find_package(OpenSSL REQUIRED) find_package(OpenSSL REQUIRED)
#-DCMAKE_TOOLCHAIN_FILE=C:/vcpkg/scripts/buildsystems/vcpkg.cmake -DVCPKG_TARGET_TRIPLET=x64-windows-static
set(VcpkgRoot ${_VCPKG_INSTALLED_DIR}/${VCPKG_TARGET_TRIPLET})
include_directories(${VcpkgRoot}/include)
link_directories(${VcpkgRoot}/lib)
target_link_libraries(${PROJECT_NAME} PRIVATE target_link_libraries(${PROJECT_NAME} PRIVATE
ZLIB::ZLIB OpenSSL::SSL OpenSSL::Crypto ws2_32 httplib::httplib nlohmann_json::nlohmann_json CURL::libcurl) ZLIB::ZLIB OpenSSL::SSL OpenSSL::Crypto ws2_32 httplib::httplib nlohmann_json::nlohmann_json)
elseif (LINUX)
find_package(ZLIB REQUIRED) if (NOT CMAKE_BUILD_TYPE STREQUAL "Debug")
find_package(OpenSSL REQUIRED) target_link_libraries(${PROJECT_NAME} PRIVATE ${VcpkgRoot}/lib/discord-rpc.lib)
target_link_libraries(${PROJECT_NAME} PRIVATE else ()
ZLIB::ZLIB OpenSSL::SSL OpenSSL::Crypto CURL::libcurl) target_link_libraries(${PROJECT_NAME} PRIVATE ${VcpkgRoot}/debug/lib/discord-rpc.lib)
endif()
else(WIN32) #MINGW else(WIN32) #MINGW
add_definitions("-D_WIN32_WINNT=0x0600") add_definitions("-D_WIN32_WINNT=0x0600")
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} discord-rpc ssl crypto ws2_32 ssp crypt32 z)
endif(WIN32) endif(WIN32)
target_include_directories(${PROJECT_NAME} PRIVATE "include") target_include_directories(${PROJECT_NAME} PRIVATE "include")

View File

@@ -2,23 +2,6 @@
The launcher is the way we communitcate to outside the game, it does a few automated actions such as but not limited to: downloading the mod, launching the game, and create a connection to a server. The launcher is the way we communitcate to outside the game, it does a few automated actions such as but not limited to: downloading the mod, launching the game, and create a connection to a server.
**To clone this repository**: `git clone --recurse-submodules https://github.com/BeamMP/BeamMP-Launcher.git`
## How to build - Release
In the root directory of the project,
1. `cmake -DCMAKE_BUILD_TYPE=Release . -B bin -DCMAKE_TOOLCHAIN_FILE=C:/vcpkg/scripts/buildsystems/vcpkg.cmake -DVCPKG_TARGET_TRIPLET=x64-windows-static`
2. `cmake --build bin --parallel --config Release`
Remember to change `C:/vcpkg` to wherever you have vcpkg installed.
## How to build - Debug
In the root directory of the project,
1. `cmake . -B bin -DCMAKE_TOOLCHAIN_FILE=C:/vcpkg/scripts/buildsystems/vcpkg.cmake -DVCPKG_TARGET_TRIPLET=x64-windows-static`
2. `cmake --build bin --parallel`
Remember to change `C:/vcpkg` to wherever you have vcpkg installed.
Copyright (c) 2019-present Anonymous275. Copyright (c) 2019-present Anonymous275.
BeamMP Launcher code is not in the public domain and is not free software. BeamMP Launcher code is not in the public domain and is not free software.

15
include/Discord/discord_info.h Executable file
View File

@@ -0,0 +1,15 @@
// Copyright (c) 2019-present Anonymous275.
// BeamMP Launcher code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
///
/// Created by Anonymous275 on 7/16/2020
///
#pragma once
#include <string>
void Discord_Main();
std::string GetDName();
std::string GetDTag();
std::string GetDID();
void DAboard();

View File

@@ -0,0 +1,26 @@
#pragma once
#if defined(DISCORD_DYNAMIC_LIB)
#if defined(_WIN32)
#if defined(DISCORD_BUILDING_SDK)
#define DISCORD_EXPORT __declspec(dllexport)
#else
#define DISCORD_EXPORT __declspec(dllimport)
#endif
#else
#define DISCORD_EXPORT __attribute__((visibility("default")))
#endif
#else
#define DISCORD_EXPORT
#endif
#ifdef __cplusplus
extern "C" {
#endif
DISCORD_EXPORT void Discord_Register(const char* applicationId, const char* command);
DISCORD_EXPORT void Discord_RegisterSteamGame(const char* applicationId, const char* steamId);
#ifdef __cplusplus
}
#endif

View File

@@ -0,0 +1,87 @@
#pragma once
#include <cstdint>
// clang-format off
#if defined(DISCORD_DYNAMIC_LIB)
# if defined(_WIN32)
# if defined(DISCORD_BUILDING_SDK)
# define DISCORD_EXPORT __declspec(dllexport)
# else
# define DISCORD_EXPORT __declspec(dllimport)
# endif
# else
# define DISCORD_EXPORT __attribute__((visibility("default")))
# endif
#else
# define DISCORD_EXPORT
#endif
// clang-format on
#ifdef __cplusplus
extern "C" {
#endif
typedef struct DiscordRichPresence {
const char* state; /* max 128 bytes */
const char* details; /* max 128 bytes */
int64_t startTimestamp;
int64_t endTimestamp;
const char* largeImageKey; /* max 32 bytes */
const char* largeImageText; /* max 128 bytes */
const char* smallImageKey; /* max 32 bytes */
const char* smallImageText; /* max 128 bytes */
const char* partyId; /* max 128 bytes */
int partySize;
int partyMax;
const char* matchSecret; /* max 128 bytes */
const char* joinSecret; /* max 128 bytes */
const char* spectateSecret; /* max 128 bytes */
int8_t instance;
} DiscordRichPresence;
typedef struct DiscordUser {
const char* userId;
const char* username;
const char* discriminator;
const char* avatar;
} DiscordUser;
typedef struct DiscordEventHandlers {
void (*ready)(const DiscordUser* request);
void (*disconnected)(int errorCode, const char* message);
void (*errored)(int errorCode, const char* message);
void (*joinGame)(const char* joinSecret);
void (*spectateGame)(const char* spectateSecret);
void (*joinRequest)(const DiscordUser* request);
} DiscordEventHandlers;
#define DISCORD_REPLY_NO 0
#define DISCORD_REPLY_YES 1
#define DISCORD_REPLY_IGNORE 2
DISCORD_EXPORT void Discord_Initialize(const char* applicationId,
DiscordEventHandlers* handlers,
int autoRegister,
const char* optionalSteamId);
DISCORD_EXPORT void Discord_Shutdown(void);
/* checks for incoming messages, dispatches callbacks */
DISCORD_EXPORT void Discord_RunCallbacks(void);
/* If you disable the lib starting its own io thread, you'll need to call this from your own */
#ifdef DISCORD_DISABLE_IO_THREAD
DISCORD_EXPORT void Discord_UpdateConnection(void);
#endif
DISCORD_EXPORT void Discord_UpdatePresence(const DiscordRichPresence* presence);
DISCORD_EXPORT void Discord_ClearPresence(void);
DISCORD_EXPORT void Discord_Respond(const char* userid, /* DISCORD_REPLY_ */ int reply);
DISCORD_EXPORT void Discord_UpdateHandlers(DiscordEventHandlers* handlers);
#ifdef __cplusplus
} /* extern "C" */
#endif

10
include/Http.h Normal file → Executable file
View File

@@ -6,16 +6,14 @@
/// Created by Anonymous275 on 7/18/2020 /// Created by Anonymous275 on 7/18/2020
/// ///
#pragma once #pragma once
#include "Logger.h"
#include <string> #include <string>
#include "Logger.h"
class HTTP { class HTTP {
public: public:
static bool Download(const std::string& IP, const std::string& Path); static bool Download(const std::string &IP, const std::string &Path);
static std::string Post(const std::string& IP, const std::string& Fields); static std::string Post(const std::string& IP, const std::string& Fields);
static std::string Get(const std::string& IP); static std::string Get(const std::string &IP);
static bool ProgressBar(size_t c, size_t t); static bool ProgressBar(size_t c, size_t t);
static void StartProxy();
public: public:
static bool isDownload; static bool isDownload;
static inline bool SkipSslVerify = false; };
};

3
include/Logger.h Normal file → Executable file
View File

@@ -6,8 +6,8 @@
/// Created by Anonymous275 on 4/2/2020. /// Created by Anonymous275 on 4/2/2020.
/// ///
#pragma once #pragma once
#include <iostream>
#include <string> #include <string>
#include <iostream>
void InitLog(); void InitLog();
void except(const std::string& toPrint); void except(const std::string& toPrint);
void fatal(const std::string& toPrint); void fatal(const std::string& toPrint);
@@ -15,4 +15,3 @@ void debug(const std::string& toPrint);
void error(const std::string& toPrint); void error(const std::string& toPrint);
void info(const std::string& toPrint); void info(const std::string& toPrint);
void warn(const std::string& toPrint); void warn(const std::string& toPrint);
std::string getDate();

View File

@@ -6,16 +6,9 @@
/// Created by Anonymous275 on 7/18/2020 /// Created by Anonymous275 on 7/18/2020
/// ///
#pragma once #pragma once
#include <string> #include <string>
#ifdef __linux__
#include "linuxfixes.h"
#include <bits/types/siginfo_t.h>
#include <cstdint>
#include <sys/ucontext.h>
#endif
void NetReset(); void NetReset();
extern bool Dev; extern bool Dev;
extern int ping; extern int ping;
@@ -30,27 +23,26 @@ extern int DEFAULT_PORT;
extern uint64_t UDPSock; extern uint64_t UDPSock;
extern uint64_t TCPSock; extern uint64_t TCPSock;
extern std::string Branch; extern std::string Branch;
extern std::string CachingDirectory;
extern bool TCPTerminate; extern bool TCPTerminate;
extern std::string LastIP; extern std::string LastIP;
extern std::string MStatus; extern std::string MStatus;
extern std::string UlStatus; extern std::string UlStatus;
extern std::string PublicKey; extern std::string PublicKey;
extern std::string PrivateKey; extern std::string PrivateKey;
extern std::string ListOfMods;
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);
void GameSend(std::string_view Data); void GameSend(std::string Data);
void SendLarge(std::string Data); void SendLarge(std::string Data);
std::string TCPRcv(uint64_t Sock); std::string TCPRcv(uint64_t Sock);
void SyncResources(uint64_t TCPSock); void SyncResources(uint64_t TCPSock);
std::string GetAddr(const std::string& IP); std::string GetAddr(const std::string&IP);
void ServerParser(std::string_view Data); void ServerParser(const std::string& Data);
std::string Login(const std::string& fields); std::string Login(const std::string& fields);
void TCPSend(const std::string& Data, uint64_t Sock); void TCPSend(const std::string&Data,uint64_t Sock);
void TCPClientMain(const std::string& IP, int Port); void TCPClientMain(const std::string& IP,int Port);
void UDPClientMain(const std::string& IP, int Port); 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();
void CoreSend(std::string data);

0
include/Security/Game.h Normal file → Executable file
View File

2
include/Security/Init.h Normal file → Executable file
View File

@@ -8,7 +8,7 @@
#pragma once #pragma once
#include <string> #include <string>
void PreGame(const std::string& GamePath); void PreGame(const std::string& GamePath);
std::string CheckVer(const std::string& path); std::string CheckVer(const std::string &path);
void InitGame(const std::string& Dir); void InitGame(const std::string& Dir);
std::string GetGameDir(); std::string GetGameDir();
void LegitimacyCheck(); void LegitimacyCheck();

13
include/Startup.h Normal file → Executable file
View File

@@ -6,16 +6,23 @@
/// Created by Anonymous275 on 7/18/2020 /// Created by Anonymous275 on 7/18/2020
/// ///
#pragma once #pragma once
#include <compare>
#include <string> #include <string>
#include <compare>
#include <vector> #include <vector>
void InitLauncher(int argc, char* argv[]); void InitLauncher(int argc, char* argv[]);
std::string GetEP(char* P = nullptr); std::string GetEP(char*P = nullptr);
std::string GetGamePath(); std::string GetGamePath();
std::string GetVer(); std::string GetVer();
std::string GetPatch();
std::string GetEN(); std::string GetEN();
void StartProxy();
void ConfigInit(); void ConfigInit();
extern bool Dev; extern bool Dev;
struct VersionParser {
explicit VersionParser(const std::string& from_string);
std::strong_ordering operator<=>(VersionParser const& rhs) const noexcept;
bool operator==(VersionParser const& rhs) const noexcept;
std::vector<std::string> split;
std::vector<size_t> data;
};

View File

@@ -1,20 +0,0 @@
#pragma once
#include <string>
#include <vector>
namespace Utils {
inline 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;
};
};

8
include/Zlib/Compressor.h Normal file → Executable file
View File

@@ -6,8 +6,6 @@
/// Created by Anonymous275 on 7/24/2020 /// Created by Anonymous275 on 7/24/2020
/// ///
#pragma once #pragma once
#include <span> #include <string>
#include <vector> std::string Comp(std::string Data);
std::string DeComp(std::string Compressed);
std::vector<char> Comp(std::span<const char> input);
std::vector<char> DeComp(std::span<const char> input);

View File

@@ -1,19 +0,0 @@
#ifndef _LINUXFIXES_H
#define _LINUXFIXES_H
#include <stdint.h>
// Translate windows sockets stuff to linux sockets
#define SOCKET uint64_t
#define SOCKADDR sockaddr
#define SOCKADDR_IN sockaddr_in
#define WSAGetLastError() errno
#define closesocket close
#define SD_BOTH SHUT_RDWR
// We dont need wsacleanup
#define WSACleanup()
#define SOCKET_ERROR -1
#define ZeroMemory(mem, len) memset(mem, 0, len)
#endif

0
include/rapidjson/allocators.h Normal file → Executable file
View File

0
include/rapidjson/cursorstreamwrapper.h Normal file → Executable file
View File

0
include/rapidjson/document.h Normal file → Executable file
View File

0
include/rapidjson/encodedstream.h Normal file → Executable file
View File

0
include/rapidjson/encodings.h Normal file → Executable file
View File

0
include/rapidjson/error/en.h Normal file → Executable file
View File

0
include/rapidjson/error/error.h Normal file → Executable file
View File

0
include/rapidjson/filereadstream.h Normal file → Executable file
View File

0
include/rapidjson/filewritestream.h Normal file → Executable file
View File

0
include/rapidjson/fwd.h Normal file → Executable file
View File

0
include/rapidjson/internal/biginteger.h Normal file → Executable file
View File

0
include/rapidjson/internal/clzll.h Normal file → Executable file
View File

0
include/rapidjson/internal/diyfp.h Normal file → Executable file
View File

0
include/rapidjson/internal/dtoa.h Normal file → Executable file
View File

0
include/rapidjson/internal/ieee754.h Normal file → Executable file
View File

0
include/rapidjson/internal/itoa.h Normal file → Executable file
View File

0
include/rapidjson/internal/meta.h Normal file → Executable file
View File

0
include/rapidjson/internal/pow10.h Normal file → Executable file
View File

0
include/rapidjson/internal/regex.h Normal file → Executable file
View File

0
include/rapidjson/internal/stack.h Normal file → Executable file
View File

0
include/rapidjson/internal/strfunc.h Normal file → Executable file
View File

0
include/rapidjson/internal/strtod.h Normal file → Executable file
View File

0
include/rapidjson/internal/swap.h Normal file → Executable file
View File

0
include/rapidjson/istreamwrapper.h Normal file → Executable file
View File

0
include/rapidjson/memorybuffer.h Normal file → Executable file
View File

0
include/rapidjson/memorystream.h Normal file → Executable file
View File

0
include/rapidjson/msinttypes/inttypes.h Normal file → Executable file
View File

0
include/rapidjson/msinttypes/stdint.h Normal file → Executable file
View File

0
include/rapidjson/ostreamwrapper.h Normal file → Executable file
View File

0
include/rapidjson/pointer.h Normal file → Executable file
View File

0
include/rapidjson/prettywriter.h Normal file → Executable file
View File

0
include/rapidjson/rapidjson.h Normal file → Executable file
View File

0
include/rapidjson/reader.h Normal file → Executable file
View File

0
include/rapidjson/schema.h Normal file → Executable file
View File

0
include/rapidjson/stream.h Normal file → Executable file
View File

0
include/rapidjson/stringbuffer.h Normal file → Executable file
View File

0
include/rapidjson/writer.h Normal file → Executable file
View File

View File

@@ -1,730 +0,0 @@
//MIT License
//
//Copyright(c) 2016 Matthias Moeller
//
//Permission is hereby granted, free of charge, to any person obtaining a copy
//of this software and associated documentation files(the "Software"), to deal
//in the Software without restriction, including without limitation the rights
//to use, copy, modify, merge, publish, distribute, sublicense, and / or sell
//copies of the Software, and to permit persons to whom the Software is
//furnished to do so, subject to the following conditions :
//
//The above copyright notice and this permission notice shall be included in all
//copies or substantial portions of the Software.
//
//THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE
//AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
//SOFTWARE.
#ifndef __TYTI_STEAM_VDF_PARSER_H__
#define __TYTI_STEAM_VDF_PARSER_H__
#include <map>
#include <vector>
#include <unordered_map>
#include <utility>
#include <fstream>
#include <memory>
#include <unordered_set>
#include <algorithm>
#include <iterator>
#include <functional>
#include <system_error>
#include <exception>
//for wstring support
#include <locale>
#include <string>
// internal
#include <stack>
//VS < 2015 has only partial C++11 support
#if defined(_MSC_VER) && _MSC_VER < 1900
#ifndef CONSTEXPR
#define CONSTEXPR
#endif
#ifndef NOEXCEPT
#define NOEXCEPT
#endif
#else
#ifndef CONSTEXPR
#define CONSTEXPR constexpr
#define TYTI_UNDEF_CONSTEXPR
#endif
#ifndef NOEXCEPT
#define NOEXCEPT noexcept
#define TYTI_UNDEF_NOEXCEPT
#endif
#endif
namespace tyti
{
namespace vdf
{
namespace detail
{
///////////////////////////////////////////////////////////////////////////
// Helper functions selecting the right encoding (char/wchar_T)
///////////////////////////////////////////////////////////////////////////
template <typename T>
struct literal_macro_help
{
static CONSTEXPR const char* result(const char* c, const wchar_t*) NOEXCEPT
{
return c;
}
static CONSTEXPR const char result(const char c, const wchar_t) NOEXCEPT
{
return c;
}
};
template <>
struct literal_macro_help<wchar_t>
{
static CONSTEXPR const wchar_t* result(const char*, const wchar_t* wc) NOEXCEPT
{
return wc;
}
static CONSTEXPR const wchar_t result(const char, const wchar_t wc) NOEXCEPT
{
return wc;
}
};
#define TYTI_L(type, text) vdf::detail::literal_macro_help<type>::result(text, L##text)
inline std::string string_converter(const std::string& w) NOEXCEPT
{
return w;
}
// utility wrapper to adapt locale-bound facets for wstring/wbuffer convert
// from cppreference
template <class Facet>
struct deletable_facet : Facet
{
template <class... Args>
deletable_facet(Args &&... args) : Facet(std::forward<Args>(args)...) {}
~deletable_facet() {}
};
inline std::string string_converter(const std::wstring& w) //todo: use us-locale
{
std::wstring_convert<deletable_facet<std::codecvt<wchar_t, char, std::mbstate_t>>> conv1;
return conv1.to_bytes(w);
}
///////////////////////////////////////////////////////////////////////////
// Writer helper functions
///////////////////////////////////////////////////////////////////////////
template <typename charT>
class tabs
{
const size_t t;
public:
explicit CONSTEXPR tabs(size_t i) NOEXCEPT : t(i) {}
std::basic_string<charT> print() const { return std::basic_string<charT>(t, TYTI_L(charT, '\t')); }
inline CONSTEXPR tabs operator+(size_t i) const NOEXCEPT
{
return tabs(t + i);
}
};
template <typename oStreamT>
oStreamT& operator<<(oStreamT& s, const tabs<typename oStreamT::char_type> t)
{
s << t.print();
return s;
}
} // end namespace detail
///////////////////////////////////////////////////////////////////////////
// Interface
///////////////////////////////////////////////////////////////////////////
/// custom objects and their corresponding write functions
/// basic object node. Every object has a name and can contains attributes saved as key_value pairs or childrens
template <typename CharT>
struct basic_object
{
typedef CharT char_type;
std::basic_string<char_type> name;
std::unordered_map<std::basic_string<char_type>, std::basic_string<char_type>> attribs;
std::unordered_map<std::basic_string<char_type>, std::shared_ptr<basic_object<char_type>>> childs;
void add_attribute(std::basic_string<char_type> key, std::basic_string<char_type> value)
{
attribs.emplace(std::move(key), std::move(value));
}
void add_child(std::unique_ptr<basic_object<char_type>> child)
{
std::shared_ptr<basic_object<char_type>> obj{ child.release() };
childs.emplace(obj->name, obj);
}
void set_name(std::basic_string<char_type> n)
{
name = std::move(n);
}
};
template <typename CharT>
struct basic_multikey_object
{
typedef CharT char_type;
std::basic_string<char_type> name;
std::unordered_multimap<std::basic_string<char_type>, std::basic_string<char_type>> attribs;
std::unordered_multimap<std::basic_string<char_type>, std::shared_ptr<basic_multikey_object<char_type>>> childs;
void add_attribute(std::basic_string<char_type> key, std::basic_string<char_type> value)
{
attribs.emplace(std::move(key), std::move(value));
}
void add_child(std::unique_ptr<basic_multikey_object<char_type>> child)
{
std::shared_ptr<basic_multikey_object<char_type>> obj{ child.release() };
childs.emplace(obj->name, obj);
}
void set_name(std::basic_string<char_type> n)
{
name = std::move(n);
}
};
typedef basic_object<char> object;
typedef basic_object<wchar_t> wobject;
typedef basic_multikey_object<char> multikey_object;
typedef basic_multikey_object<wchar_t> wmultikey_object;
struct Options
{
bool strip_escape_symbols;
bool ignore_all_platform_conditionals;
bool ignore_includes;
Options() : strip_escape_symbols(true), ignore_all_platform_conditionals(false), ignore_includes(false) {}
};
//forward decls
//forward decl
template <typename OutputT, typename iStreamT>
OutputT read(iStreamT& inStream, const Options& opt = Options{});
/** \brief writes given object tree in vdf format to given stream.
Output is prettyfied, using tabs
*/
template <typename oStreamT, typename T>
void write(oStreamT& s, const T& r,
const detail::tabs<typename oStreamT::char_type> tab = detail::tabs<typename oStreamT::char_type>(0))
{
typedef typename oStreamT::char_type charT;
using namespace detail;
s << tab << TYTI_L(charT, '"') << r.name << TYTI_L(charT, "\"\n") << tab << TYTI_L(charT, "{\n");
for (const auto& i : r.attribs)
s << tab + 1 << TYTI_L(charT, '"') << i.first << TYTI_L(charT, "\"\t\t\"") << i.second << TYTI_L(charT, "\"\n");
for (const auto& i : r.childs)
if (i.second)
write(s, *i.second, tab + 1);
s << tab << TYTI_L(charT, "}\n");
}
namespace detail
{
template <typename iStreamT>
std::basic_string<typename iStreamT::char_type> read_file(iStreamT& inStream)
{
// cache the file
typedef typename iStreamT::char_type charT;
std::basic_string<charT> str;
inStream.seekg(0, std::ios::end);
str.resize(static_cast<size_t>(inStream.tellg()));
if (str.empty())
return str;
inStream.seekg(0, std::ios::beg);
inStream.read(&str[0], str.size());
return str;
}
/** \brief Read VDF formatted sequences defined by the range [first, last).
If the file is mailformatted, parser will try to read it until it can.
@param first begin iterator
@param end end iterator
@param exclude_files list of files which cant be included anymore.
prevents circular includes
can thow:
- "std::runtime_error" if a parsing error occured
- "std::bad_alloc" if not enough memory coup be allocated
*/
template <typename OutputT, typename IterT>
std::vector<std::unique_ptr<OutputT>> read_internal(IterT first, const IterT last,
std::unordered_set<std::basic_string<typename std::iterator_traits<IterT>::value_type>>& exclude_files,
const Options& opt)
{
static_assert(std::is_default_constructible<OutputT>::value,
"Output Type must be default constructible (provide constructor without arguments)");
static_assert(std::is_move_constructible<OutputT>::value,
"Output Type must be move constructible");
typedef typename std::iterator_traits<IterT>::value_type charT;
const std::basic_string<charT> comment_end_str = TYTI_L(charT, "*/");
const std::basic_string<charT> whitespaces = TYTI_L(charT, " \n\v\f\r\t");
#ifdef WIN32
std::function<bool(const std::basic_string<charT>&)> is_platform_str = [](const std::basic_string<charT>& in) {
return in == TYTI_L(charT, "$WIN32") || in == TYTI_L(charT, "$WINDOWS");
};
#elif __APPLE__
// WIN32 stands for pc in general
std::function<bool(const std::basic_string<charT>&)> is_platform_str = [](const std::basic_string<charT>& in) {
return in == TYTI_L(charT, "$WIN32") || in == TYTI_L(charT, "$POSIX") || in == TYTI_L(charT, "$OSX");
};
#elif __linux__
// WIN32 stands for pc in general
std::function<bool(const std::basic_string<charT>&)> is_platform_str = [](const std::basic_string<charT>& in) {
return in == TYTI_L(charT, "$WIN32") || in == TYTI_L(charT, "$POSIX") || in == TYTI_L(charT, "$LINUX");
};
#else
std::function<bool(const std::basic_string<charT>&)> is_platform_str = [](const std::basic_string<charT>& in) {
return false;
};
#endif
if (opt.ignore_all_platform_conditionals)
is_platform_str = [](const std::basic_string<charT>&) {
return false;
};
// function for skipping a comment block
// iter: iterator poition to the position after a '/'
auto skip_comments = [&comment_end_str](IterT iter, const IterT& last) -> IterT {
++iter;
if (iter != last)
{
if (*iter == TYTI_L(charT, '/'))
{
// line comment, skip whole line
iter = std::find(iter + 1, last, TYTI_L(charT, '\n'));
}
if (*iter == '*')
{
// block comment, skip until next occurance of "*\"
iter = std::search(iter + 1, last, std::begin(comment_end_str), std::end(comment_end_str));
iter += 2;
}
}
return iter;
};
auto end_quote = [](IterT iter, const IterT& last) -> IterT {
const auto begin = iter;
auto last_esc = iter;
do
{
++iter;
iter = std::find(iter, last, TYTI_L(charT, '\"'));
if (iter == last)
break;
last_esc = std::prev(iter);
while (last_esc != begin && *last_esc == '\\')
--last_esc;
} while (!(std::distance(last_esc, iter) % 2));
if (iter == last)
throw std::runtime_error{ "quote was opened but not closed." };
return iter;
};
auto end_word = [&whitespaces](IterT iter, const IterT& last) -> IterT {
const auto begin = iter;
auto last_esc = iter;
do
{
++iter;
iter = std::find_first_of(iter, last, std::begin(whitespaces), std::end(whitespaces));
if (iter == last)
break;
last_esc = std::prev(iter);
while (last_esc != begin && *last_esc == '\\')
--last_esc;
} while (!(std::distance(last_esc, iter) % 2));
//if (iter == last)
// throw std::runtime_error{ "word wasnt properly ended" };
return iter;
};
auto skip_whitespaces = [&whitespaces](IterT iter, const IterT& last) -> IterT {
iter = std::find_if_not(iter, last, [&whitespaces](charT c) {
// return true if whitespace
return std::any_of(std::begin(whitespaces), std::end(whitespaces), [c](charT pc) { return pc == c; });
});
return iter;
};
std::function<void(std::basic_string<charT>&)> strip_escape_symbols = [](std::basic_string<charT>& s) {
auto quote_searcher = [&s](size_t pos) { return s.find(TYTI_L(charT, "\\\""), pos); };
auto p = quote_searcher(0);
while (p != s.npos)
{
s.replace(p, 2, TYTI_L(charT, "\""));
p = quote_searcher(p);
}
auto searcher = [&s](size_t pos) { return s.find(TYTI_L(charT, "\\\\"), pos); };
p = searcher(0);
while (p != s.npos)
{
s.replace(p, 2, TYTI_L(charT, "\\"));
p = searcher(p);
}
};
if (!opt.strip_escape_symbols)
strip_escape_symbols = [](std::basic_string<charT>&) {};
auto conditional_fullfilled = [&skip_whitespaces, &is_platform_str](IterT& iter, const IterT& last) {
iter = skip_whitespaces(iter, last);
if (*iter == '[')
{
++iter;
const auto end = std::find(iter, last, ']');
const bool negate = *iter == '!';
if (negate)
++iter;
auto conditional = std::basic_string<charT>(iter, end);
const bool is_platform = is_platform_str(conditional);
iter = end + 1;
return static_cast<bool>(is_platform ^ negate);
}
return true;
};
//read header
// first, quoted name
std::unique_ptr<OutputT> curObj = nullptr;
std::vector<std::unique_ptr<OutputT>> roots;
std::stack<std::unique_ptr<OutputT>> lvls;
auto curIter = first;
while (curIter != last && *curIter != '\0')
{
//find first starting attrib/child, or ending
curIter = skip_whitespaces(curIter, last);
if (curIter == last || *curIter == '\0')
break;
if (*curIter == TYTI_L(charT, '/'))
{
curIter = skip_comments(curIter, last);
}
else if (*curIter != TYTI_L(charT, '}'))
{
// get key
const auto keyEnd = (*curIter == TYTI_L(charT, '\"')) ? end_quote(curIter, last) : end_word(curIter, last);
if (*curIter == TYTI_L(charT, '\"'))
++curIter;
std::basic_string<charT> key(curIter, keyEnd);
strip_escape_symbols(key);
curIter = keyEnd + ((*keyEnd == TYTI_L(charT, '\"')) ? 1 : 0);
curIter = skip_whitespaces(curIter, last);
auto conditional = conditional_fullfilled(curIter, last);
if (!conditional)
continue;
while (*curIter == TYTI_L(charT, '/'))
{
curIter = skip_comments(curIter, last);
if (curIter == last || *curIter == '}')
throw std::runtime_error{ "key declared, but no value" };
curIter = skip_whitespaces(curIter, last);
if (curIter == last || *curIter == '}')
throw std::runtime_error{ "key declared, but no value" };
}
// get value
if (*curIter != '{')
{
const auto valueEnd = (*curIter == TYTI_L(charT, '\"')) ? end_quote(curIter, last) : end_word(curIter, last);
if (*curIter == TYTI_L(charT, '\"'))
++curIter;
auto value = std::basic_string<charT>(curIter, valueEnd);
strip_escape_symbols(value);
curIter = valueEnd + ((*valueEnd == TYTI_L(charT, '\"')) ? 1 : 0);
auto conditional = conditional_fullfilled(curIter, last);
if (!conditional)
continue;
// process value
if (key != TYTI_L(charT, "#include") && key != TYTI_L(charT, "#base"))
{
if (curObj)
{
curObj->add_attribute(std::move(key), std::move(value));
}
else
{
throw std::runtime_error{ "unexpected key without object" };
}
}
else
{
if (!opt.ignore_includes && exclude_files.find(value) == exclude_files.end())
{
exclude_files.insert(value);
std::basic_ifstream<charT> i(detail::string_converter(value));
auto str = read_file(i);
auto file_objs = read_internal<OutputT>(str.begin(), str.end(), exclude_files, opt);
for (auto& n : file_objs)
{
if (curObj)
curObj->add_child(std::move(n));
else
roots.push_back(std::move(n));
}
exclude_files.erase(value);
}
}
}
else if (*curIter == '{')
{
if (curObj)
lvls.push(std::move(curObj));
curObj = std::make_unique<OutputT>();
curObj->set_name(std::move(key));
++curIter;
}
}
//end of new object
else if (curObj && *curIter == TYTI_L(charT, '}'))
{
if (!lvls.empty())
{
//get object before
std::unique_ptr<OutputT> prev{ std::move(lvls.top()) };
lvls.pop();
// add finished obj to obj before and release it from processing
prev->add_child(std::move(curObj));
curObj = std::move(prev);
}
else
{
roots.push_back(std::move(curObj));
curObj.reset();
}
++curIter;
}
else
{
throw std::runtime_error{ "unexpected '}'" };
}
}
if (curObj != nullptr || !lvls.empty())
{
throw std::runtime_error{ "object is not closed with '}'" };
}
return roots;
}
} // namespace detail
/** \brief Read VDF formatted sequences defined by the range [first, last).
If the file is mailformatted, parser will try to read it until it can.
@param first begin iterator
@param end end iterator
can thow:
- "std::runtime_error" if a parsing error occured
- "std::bad_alloc" if not enough memory coup be allocated
*/
template <typename OutputT, typename IterT>
OutputT read(IterT first, const IterT last, const Options& opt = Options{})
{
auto exclude_files = std::unordered_set<std::basic_string<typename std::iterator_traits<IterT>::value_type>>{};
auto roots = detail::read_internal<OutputT>(first, last, exclude_files, opt);
OutputT result;
if (roots.size() > 1)
{
for (auto& i : roots)
result.add_child(std::move(i));
}
else if (roots.size() == 1)
result = std::move(*roots[0]);
return result;
}
/** \brief Read VDF formatted sequences defined by the range [first, last).
If the file is mailformatted, parser will try to read it until it can.
@param first begin iterator
@param end end iterator
@param ec output bool. 0 if ok, otherwise, holds an system error code
Possible error codes:
std::errc::protocol_error: file is mailformatted
std::errc::not_enough_memory: not enough space
std::errc::invalid_argument: iterators throws e.g. out of range
*/
template <typename OutputT, typename IterT>
OutputT read(IterT first, IterT last, std::error_code& ec, const Options& opt = Options{}) NOEXCEPT
{
ec.clear();
OutputT r{};
try
{
r = read<OutputT>(first, last, opt);
}
catch (std::runtime_error&)
{
ec = std::make_error_code(std::errc::protocol_error);
}
catch (std::bad_alloc&)
{
ec = std::make_error_code(std::errc::not_enough_memory);
}
catch (...)
{
ec = std::make_error_code(std::errc::invalid_argument);
}
return r;
}
/** \brief Read VDF formatted sequences defined by the range [first, last).
If the file is mailformatted, parser will try to read it until it can.
@param first begin iterator
@param end end iterator
@param ok output bool. true, if parser successed, false, if parser failed
*/
template <typename OutputT, typename IterT>
OutputT read(IterT first, const IterT last, bool* ok, const Options& opt = Options{}) NOEXCEPT
{
std::error_code ec;
auto r = read<OutputT>(first, last, ec, opt);
if (ok)
*ok = !ec;
return r;
}
template <typename IterT>
inline auto read(IterT first, const IterT last, bool* ok, const Options& opt = Options{}) NOEXCEPT -> basic_object<typename std::iterator_traits<IterT>::value_type>
{
return read<basic_object<typename std::iterator_traits<IterT>::value_type>>(first, last, ok, opt);
}
template <typename IterT>
inline auto read(IterT first, IterT last, std::error_code& ec, const Options& opt = Options{}) NOEXCEPT
-> basic_object<typename std::iterator_traits<IterT>::value_type>
{
return read<basic_object<typename std::iterator_traits<IterT>::value_type>>(first, last, ec, opt);
}
template <typename IterT>
inline auto read(IterT first, const IterT last, const Options& opt = Options{})
-> basic_object<typename std::iterator_traits<IterT>::value_type>
{
return read<basic_object<typename std::iterator_traits<IterT>::value_type>>(first, last, opt);
}
/** \brief Loads a stream (e.g. filestream) into the memory and parses the vdf formatted data.
throws "std::bad_alloc" if file buffer could not be allocated
*/
template <typename OutputT, typename iStreamT>
OutputT read(iStreamT& inStream, std::error_code& ec, const Options& opt = Options{})
{
// cache the file
typedef typename iStreamT::char_type charT;
std::basic_string<charT> str = detail::read_file(inStream);
// parse it
return read<OutputT>(str.begin(), str.end(), ec, opt);
}
template <typename iStreamT>
inline basic_object<typename iStreamT::char_type> read(iStreamT& inStream, std::error_code& ec, const Options& opt = Options{})
{
return read<basic_object<typename iStreamT::char_type>>(inStream, ec, opt);
}
/** \brief Loads a stream (e.g. filestream) into the memory and parses the vdf formatted data.
throws "std::bad_alloc" if file buffer could not be allocated
ok == false, if a parsing error occured
*/
template <typename OutputT, typename iStreamT>
OutputT read(iStreamT& inStream, bool* ok, const Options& opt = Options{})
{
std::error_code ec;
const auto r = read<OutputT>(inStream, ec, opt);
if (ok)
*ok = !ec;
return r;
}
template <typename iStreamT>
inline basic_object<typename iStreamT::char_type> read(iStreamT& inStream, bool* ok, const Options& opt = Options{})
{
return read<basic_object<typename iStreamT::char_type>>(inStream, ok, opt);
}
/** \brief Loads a stream (e.g. filestream) into the memory and parses the vdf formatted data.
throws "std::bad_alloc" if file buffer could not be allocated
throws "std::runtime_error" if a parsing error occured
*/
template <typename OutputT, typename iStreamT>
OutputT read(iStreamT& inStream, const Options& opt)
{
// cache the file
typedef typename iStreamT::char_type charT;
std::basic_string<charT> str = detail::read_file(inStream);
// parse it
return read<OutputT>(str.begin(), str.end(), opt);
}
template <typename iStreamT>
inline basic_object<typename iStreamT::char_type> read(iStreamT& inStream, const Options& opt = Options{})
{
return read<basic_object<typename iStreamT::char_type>>(inStream, opt);
}
} // namespace vdf
} // namespace tyti
#ifndef TYTI_NO_L_UNDEF
#undef TYTI_L
#endif
#ifdef TYTI_UNDEF_CONSTEXPR
#undef CONSTEXPR
#undef TYTI_NO_L_UNDEF
#endif
#ifdef TYTI_UNDEF_NOTHROW
#undef NOTHROW
#undef TYTI_UNDEF_NOTHROW
#endif
#endif //__TYTI_STEAM_VDF_PARSER_H__

94
src/Compressor.cpp Normal file → Executable file
View File

@@ -6,57 +6,49 @@
/// Created by Anonymous275 on 7/15/2020 /// Created by Anonymous275 on 7/15/2020
/// ///
#include "Logger.h" #include <iostream>
#include <span>
#include <vector>
#include <zconf.h>
#include <zlib.h> #include <zlib.h>
#ifdef __linux__
#include <cstring>
#endif
std::vector<char> Comp(std::span<const char> input) { #define Biggest 30000
auto max_size = compressBound(input.size()); std::string Comp(std::string Data){
std::vector<char> output(max_size); char*C = new char[Biggest];
uLongf output_size = output.size(); memset(C, 0, Biggest);
int res = compress( z_stream defstream;
reinterpret_cast<Bytef*>(output.data()), defstream.zalloc = Z_NULL;
&output_size, defstream.zfree = Z_NULL;
reinterpret_cast<const Bytef*>(input.data()), defstream.opaque = Z_NULL;
static_cast<uLongf>(input.size())); defstream.avail_in = (uInt)Data.length();
if (res != Z_OK) { defstream.next_in = (Bytef *)&Data[0];
error("zlib compress() failed: " + std::to_string(res)); defstream.avail_out = Biggest;
throw std::runtime_error("zlib compress() failed"); defstream.next_out = reinterpret_cast<Bytef *>(C);
} deflateInit(&defstream, Z_BEST_COMPRESSION);
debug("zlib compressed " + std::to_string(input.size()) + " B to " + std::to_string(output_size) + " B"); deflate(&defstream, Z_SYNC_FLUSH);
output.resize(output_size); deflate(&defstream, Z_FINISH);
return output; deflateEnd(&defstream);
} int TO = defstream.total_out;
std::string Ret(TO,0);
std::vector<char> DeComp(std::span<const char> input) { memcpy_s(&Ret[0],TO,C,TO);
std::vector<char> output_buffer(std::min<size_t>(input.size() * 5, 15 * 1024 * 1024)); delete [] C;
return Ret;
uLongf output_size = output_buffer.size();
while (true) {
int res = uncompress(
reinterpret_cast<Bytef*>(output_buffer.data()),
&output_size,
reinterpret_cast<const Bytef*>(input.data()),
static_cast<uLongf>(input.size()));
if (res == Z_BUF_ERROR) {
if (output_buffer.size() > 30 * 1024 * 1024) {
throw std::runtime_error("decompressed packet size of 30 MB exceeded");
}
debug("zlib uncompress() failed, trying with 2x buffer size of " + std::to_string(output_buffer.size() * 2));
output_buffer.resize(output_buffer.size() * 2);
output_size = output_buffer.size();
} else if (res != Z_OK) {
error("zlib uncompress() failed: " + std::to_string(res));
throw std::runtime_error("zlib uncompress() failed");
} else if (res == Z_OK) {
break;
}
} output_buffer.resize(output_size);
return output_buffer;
} }
std::string 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);
int TO = infstream.total_out;
std::string Ret(TO,0);
memcpy_s(&Ret[0],TO,C,TO);
delete [] C;
return Ret;
}

123
src/Config.cpp Normal file → Executable file
View File

@@ -1,65 +1,58 @@
/// ///
/// Created by Anonymous275 on 2/23/2021 /// Created by Anonymous275 on 2/23/2021
/// ///
#include "Logger.h" #include <nlohmann/json.hpp>
#include "Network/network.hpp" #include "Network/network.h"
#include <cstdint> #include <filesystem>
#include <filesystem> #include "Logger.h"
#include <fstream> #include <fstream>
#include <nlohmann/json.hpp> #include <cstdint>
namespace fs = std::filesystem; namespace fs = std::filesystem;
std::string Branch; std::string Branch;
std::string CachingDirectory = "./Resources"; void ParseConfig(const nlohmann::json& d){
if(d["Port"].is_number()){
void ParseConfig(const nlohmann::json& d) { DEFAULT_PORT = d["Port"].get<int>();
if (d["Port"].is_number()) { }
DEFAULT_PORT = d["Port"].get<int>(); //Default -1
} //Release 1
// Default -1 //EA 2
// Release 1 //Dev 3
// EA 2 //Custom 3
// Dev 3
// Custom 3 if(d["Build"].is_string()){
if (d["Build"].is_string()) { Branch = d["Build"].get<std::string>();
Branch = d["Build"].get<std::string>(); for(char& c : Branch)c = char(tolower(c));
for (char& c : Branch) }
c = char(tolower(c)); }
}
if (d.contains("CachingDirectory") && d["CachingDirectory"].is_string()) { void ConfigInit(){
CachingDirectory = d["CachingDirectory"].get<std::string>(); if(fs::exists("Launcher.cfg")){
info("Mod caching directory: " + CachingDirectory); std::ifstream cfg("Launcher.cfg");
} if(cfg.is_open()){
} auto Size = fs::file_size("Launcher.cfg");
std::string Buffer(Size, 0);
void ConfigInit() { cfg.read(&Buffer[0], Size);
if (fs::exists("Launcher.cfg")) { cfg.close();
std::ifstream cfg("Launcher.cfg"); nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false);
if (cfg.is_open()) { if(d.is_discarded()){
auto Size = fs::file_size("Launcher.cfg"); fatal("Config failed to parse make sure it's valid JSON!");
std::string Buffer(Size, 0); }
cfg.read(&Buffer[0], Size); ParseConfig(d);
cfg.close(); }else fatal("Failed to open Launcher.cfg!");
nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false); }else{
if (d.is_discarded()) { std::ofstream cfg("Launcher.cfg");
fatal("Config failed to parse make sure it's valid JSON!"); if(cfg.is_open()){
} cfg <<
ParseConfig(d); R"({
} else "Port": 4444,
fatal("Failed to open Launcher.cfg!"); "Build": "Default"
} else { })";
std::ofstream cfg("Launcher.cfg"); cfg.close();
if (cfg.is_open()) { }else{
cfg << fatal("Failed to write config on disk!");
R"({ }
"Port": 4444, }
"Build": "Default", }
"CachingDirectory": "./Resources"
})";
cfg.close();
} else {
fatal("Failed to write config on disk!");
}
}
}

111
src/Discord.cpp Executable file
View File

@@ -0,0 +1,111 @@
// Copyright (c) 2019-present Anonymous275.
// BeamMP Launcher code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
///
/// Created by Anonymous275 on 7/16/2020
///
#include "Discord/discord_rpc.h"
#include "Logger.h"
#include <cstring>
#include <thread>
#include <ctime>
struct DInfo{
std::string Name;
std::string Tag;
std::string DID;
};
DInfo* DiscordInfo = nullptr;
int64_t StartTime;
void updateDiscordPresence(){
//if (SendPresence) {
//char buffer[256];
DiscordRichPresence discordPresence;
memset(&discordPresence, 0, sizeof(discordPresence));
std::string P = "Playing with friends!"; ///to be revisited
discordPresence.state = P.c_str();
//sprintf(buffer, "Frustration level: %d", FrustrationLevel);
//discordPresence.details = buffer;
discordPresence.startTimestamp = StartTime;
//discordPresence.endTimestamp = time(0) + 5 * 60;
discordPresence.largeImageKey = "mainlogo";
//discordPresence.smallImageKey = "logo";
//discordPresence.partyId = "party1234";
//discordPresence.partySize = 1;
//discordPresence.partyMax = 6;
//discordPresence.matchSecret = "xyzzy";
//discordPresence.joinSecret = "join";
//discordPresence.spectateSecret = "look";
//discordPresence.instance = 0;
Discord_UpdatePresence(&discordPresence);
//}
//else {
// Discord_ClearPresence();
//}
}
void handleDiscordReady(const DiscordUser* User){
DiscordInfo = new DInfo{
User->username,
User->discriminator,
User->userId
};
}
void discordInit(){
DiscordEventHandlers handlers;
memset(&handlers, 0, sizeof(handlers));
handlers.ready = handleDiscordReady;
/*handlers.disconnected = handleDiscordDisconnected;
handlers.errored = handleDiscordError;
handlers.joinGame = handleDiscordJoin;
handlers.spectateGame = handleDiscordSpectate;
handlers.joinRequest = handleDiscordJoinRequest;*/
Discord_Initialize("629743237988352010", &handlers, 1,nullptr);
}
[[noreturn]] void Loop(){
StartTime = time(nullptr);
while (true) {
updateDiscordPresence();
#ifdef DISCORD_DISABLE_IO_THREAD
Discord_UpdateConnection();
#endif
Discord_RunCallbacks();
if(DiscordInfo == nullptr){
std::this_thread::sleep_for(std::chrono::milliseconds(250));
}else std::this_thread::sleep_for(std::chrono::seconds(2));
}
}
void DMain(){
discordInit();
Loop();
}
std::string GetDName(){
return DiscordInfo->Name;
}
std::string GetDTag(){
return DiscordInfo->Tag;
}
std::string GetDID(){
return DiscordInfo->DID;
}
void DAboard(){
DiscordInfo = nullptr;
}
void ErrorAboard(){
error("Discord timeout! please start the discord app and try again after 30 secs");
std::this_thread::sleep_for(std::chrono::seconds(5));
exit(6);
}
void Discord_Main(){
/*std::thread t1(DMain);
t1.detach();*/
/*info("Connecting to discord client...");
int C = 0;
while(DiscordInfo == nullptr && C < 80){
std::this_thread::sleep_for(std::chrono::milliseconds(300));
C++;
}
if(DiscordInfo == nullptr)ErrorAboard();*/
}

102
src/GameStart.cpp Normal file → Executable file
View File

@@ -6,122 +6,64 @@
/// Created by Anonymous275 on 7/19/2020 /// Created by Anonymous275 on 7/19/2020
/// ///
#if defined(_WIN32)
#include <windows.h>
#include <shlobj.h>
#elif defined(__linux__)
#include "vdf_parser.hpp"
#include <pwd.h>
#include <spawn.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#endif
#include "Logger.h"
#include "Startup.h"
#include <Security/Init.h> #include <Security/Init.h>
#include <filesystem> #include <windows.h>
#include "Startup.h"
#include "Logger.h"
#include <thread> #include <thread>
unsigned long GamePID = 0; unsigned long GamePID = 0;
#if defined(_WIN32) std::string QueryKey(HKEY hKey,int ID);
std::string QueryKey(HKEY hKey, int ID); std::string GetGamePath(){
std::string GetGamePath() {
static std::string Path; static std::string Path;
if (!Path.empty()) if(!Path.empty())return Path;
return Path;
HKEY hKey; HKEY hKey;
LPCTSTR sk = "Software\\BeamNG\\BeamNG.drive"; LPCTSTR sk = "Software\\BeamNG\\BeamNG.drive";
LONG openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey); LONG openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
if (openRes != ERROR_SUCCESS) { if (openRes != ERROR_SUCCESS){
fatal("Please launch the game at least once!"); fatal("Please launch the game at least once!");
} }
Path = QueryKey(hKey, 4); Path = QueryKey(hKey,4);
if (Path.empty()) { if(Path.empty()){
Path = ""; sk = R"(SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders)";
char appDataPath[MAX_PATH]; openRes = RegOpenKeyEx(HKEY_CURRENT_USER, sk, 0, KEY_ALL_ACCESS, &hKey);
HRESULT result = SHGetFolderPathA(NULL, CSIDL_LOCAL_APPDATA, NULL, 0, appDataPath); if (openRes != ERROR_SUCCESS){
if (SUCCEEDED(result)) { fatal("Cannot get Local Appdata directory!");
Path = appDataPath;
} }
Path = QueryKey(hKey,5);
if (Path.empty()) {
fatal("Cannot get Local Appdata directory");
}
Path += "\\BeamNG.drive\\"; Path += "\\BeamNG.drive\\";
} }
std::string Ver = CheckVer(GetGameDir()); std::string Ver = CheckVer(GetGameDir());
Ver = Ver.substr(0, Ver.find('.', Ver.find('.') + 1)); Ver = Ver.substr(0,Ver.find('.',Ver.find('.')+1));
Path += Ver + "\\"; Path += Ver + "\\";
return Path; return Path;
} }
#elif defined(__linux__)
std::string GetGamePath() {
// Right now only steam is supported
struct passwd* pw = getpwuid(getuid());
std::string homeDir = pw->pw_dir;
std::string Path = homeDir + "/.local/share/BeamNG.drive/"; void StartGame(std::string Dir){
std::string Ver = CheckVer(GetGameDir());
Ver = Ver.substr(0, Ver.find('.', Ver.find('.') + 1));
Path += Ver + "/";
return Path;
}
#endif
#if defined(_WIN32)
void StartGame(std::string Dir) {
BOOL bSuccess = FALSE; BOOL bSuccess = FALSE;
PROCESS_INFORMATION pi; PROCESS_INFORMATION pi;
STARTUPINFO si = { 0 }; STARTUPINFO si = {0};
si.cb = sizeof(si); si.cb = sizeof(si);
std::string BaseDir = Dir; //+"\\Bin64"; std::string BaseDir = Dir; //+"\\Bin64";
// Dir += R"(\Bin64\BeamNG.drive.x64.exe)"; //Dir += R"(\Bin64\BeamNG.drive.x64.exe)";
Dir += "\\BeamNG.drive.exe"; Dir += "\\BeamNG.drive.exe";
bSuccess = CreateProcessA(Dir.c_str(), nullptr, nullptr, nullptr, TRUE, 0, nullptr, BaseDir.c_str(), &si, &pi); bSuccess = CreateProcessA(Dir.c_str(), nullptr, nullptr, nullptr, TRUE, 0, nullptr, BaseDir.c_str(), &si, &pi);
if (bSuccess) { if (bSuccess){
info("Game Launched!"); info("Game Launched!");
GamePID = pi.dwProcessId; GamePID = pi.dwProcessId;
WaitForSingleObject(pi.hProcess, INFINITE); WaitForSingleObject(pi.hProcess, INFINITE);
error("Game Closed! launcher closing soon"); error("Game Closed! launcher closing soon");
} else { }else{
error("Failed to Launch the game! launcher closing soon"); error("Failed to Launch the game! launcher closing soon");
} }
std::this_thread::sleep_for(std::chrono::seconds(5)); std::this_thread::sleep_for(std::chrono::seconds(5));
exit(2); exit(2);
} }
#elif defined(__linux__)
void StartGame(std::string Dir) {
int status;
std::string filename = (Dir + "/BinLinux/BeamNG.drive.x64");
char* argv[] = { filename.data(), NULL };
pid_t pid;
posix_spawn_file_actions_t spawn_actions;
posix_spawn_file_actions_init(&spawn_actions);
posix_spawn_file_actions_addclose(&spawn_actions, STDOUT_FILENO);
posix_spawn_file_actions_addclose(&spawn_actions, STDERR_FILENO);
int result = posix_spawn(&pid, filename.c_str(), &spawn_actions, nullptr, argv, environ);
if (result != 0) { void InitGame(const std::string& Dir){
error("Failed to Launch the game! launcher closing soon"); if(!Dev){
return;
} else {
waitpid(pid, &status, 0);
error("Game Closed! launcher closing soon");
}
std::this_thread::sleep_for(std::chrono::seconds(5));
exit(2);
}
#endif
void InitGame(const std::string& Dir) {
if (!Dev) {
std::thread Game(StartGame, Dir); std::thread Game(StartGame, Dir);
Game.detach(); Game.detach();
} }

38
src/Logger.cpp Normal file → Executable file
View File

@@ -6,11 +6,11 @@
/// Created by Anonymous275 on 7/17/2020 /// Created by Anonymous275 on 7/17/2020
/// ///
#include "Logger.h"
#include "Startup.h" #include "Startup.h"
#include <chrono> #include "Logger.h"
#include <fstream> #include <fstream>
#include <sstream> #include <sstream>
#include <chrono>
#include <thread> #include <thread>
std::string getDate() { std::string getDate() {
@@ -24,25 +24,24 @@ std::string getDate() {
std::string Min = (M > 9 ? std::to_string(M) : "0" + std::to_string(M)); std::string Min = (M > 9 ? std::to_string(M) : "0" + std::to_string(M));
std::string Hour = (H > 9 ? std::to_string(H) : "0" + std::to_string(H)); std::string Hour = (H > 9 ? std::to_string(H) : "0" + std::to_string(H));
date date
<< "[" << "["
<< local_tm.tm_mday << "/" << local_tm.tm_mday << "/"
<< local_tm.tm_mon + 1 << "/" << local_tm.tm_mon + 1 << "/"
<< local_tm.tm_year + 1900 << " " << local_tm.tm_year + 1900 << " "
<< Hour << ":" << Hour << ":"
<< Min << ":" << Min << ":"
<< Secs << Secs
<< "] "; << "] ";
return date.str(); return date.str();
} }
void InitLog() { void InitLog(){
std::ofstream LFS; std::ofstream LFS;
LFS.open(GetEP() + "Launcher.log"); LFS.open(GetEP() + "Launcher.log");
if (!LFS.is_open()) { if(!LFS.is_open()){
error("logger file init failed!"); error("logger file init failed!");
} else }else LFS.close();
LFS.close();
} }
void addToLog(const std::string& Line) { void addToLog(const std::string& Line){
std::ofstream LFS; std::ofstream LFS;
LFS.open(GetEP() + "Launcher.log", std::ios_base::app); LFS.open(GetEP() + "Launcher.log", std::ios_base::app);
LFS << Line.c_str(); LFS << Line.c_str();
@@ -54,13 +53,12 @@ void info(const std::string& toPrint) {
addToLog(Print); addToLog(Print);
} }
void debug(const std::string& toPrint) { void debug(const std::string& toPrint) {
if(!Dev)return;
std::string Print = getDate() + "[DEBUG] " + toPrint + "\n"; std::string Print = getDate() + "[DEBUG] " + toPrint + "\n";
if (Dev) { std::cout << Print;
std::cout << Print;
}
addToLog(Print); addToLog(Print);
} }
void warn(const std::string& toPrint) { void warn(const std::string& toPrint){
std::string Print = getDate() + "[WARN] " + toPrint + "\n"; std::string Print = getDate() + "[WARN] " + toPrint + "\n";
std::cout << Print; std::cout << Print;
addToLog(Print); addToLog(Print);
@@ -75,7 +73,7 @@ void fatal(const std::string& toPrint) {
std::cout << Print; std::cout << Print;
addToLog(Print); addToLog(Print);
std::this_thread::sleep_for(std::chrono::seconds(5)); std::this_thread::sleep_for(std::chrono::seconds(5));
std::exit(1); _Exit(-1);
} }
void except(const std::string& toPrint) { void except(const std::string& toPrint) {
std::string Print = getDate() + "[EXCEP] " + toPrint + "\n"; std::string Print = getDate() + "[EXCEP] " + toPrint + "\n";

378
src/Network/Core.cpp Normal file → Executable file
View File

@@ -5,31 +5,18 @@
/// ///
/// Created by Anonymous275 on 7/20/2020 /// Created by Anonymous275 on 7/20/2020
/// ///
#include "Http.h"
#include "Network/network.hpp" #include "Network/network.h"
#include "Security/Init.h" #include "Security/Init.h"
#include <cstdlib>
#include <regex> #include <regex>
#if defined(_WIN32) #include "Http.h"
#include <winsock2.h> #include <winsock2.h>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#elif defined(__linux__)
#include <cstring>
#include <errno.h>
#include <netdb.h>
#include <spawn.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#endif
#include "Logger.h"
#include "Startup.h" #include "Startup.h"
#include "Logger.h"
#include <charconv> #include <charconv>
#include <nlohmann/json.hpp>
#include <set>
#include <thread> #include <thread>
#include <set>
extern int TraceBack; extern int TraceBack;
std::set<std::string>* ConfList = nullptr; std::set<std::string>* ConfList = nullptr;
@@ -37,43 +24,18 @@ bool TCPTerminate = false;
int DEFAULT_PORT = 4444; int DEFAULT_PORT = 4444;
bool Terminate = false; bool Terminate = false;
bool LoginAuth = false; bool LoginAuth = false;
std::string Username = "";
std::string UserRole = "";
int UserID = -1;
std::string UlStatus; std::string UlStatus;
std::string MStatus; std::string MStatus;
bool ModLoaded; bool ModLoaded;
int ping = -1; int ping = -1;
SOCKET CoreSocket = -1;
signed char confirmed = -1;
bool SecurityWarning() { void StartSync(const std::string &Data){
confirmed = -1; std::string IP = GetAddr(Data.substr(1,Data.find(':')-1));
CoreSend("WMODS_FOUND"); if(IP.find('.') == -1){
if(IP == "DNS")UlStatus ="UlConnection Failed! (DNS Lookup Failed)";
while (confirmed == -1) else UlStatus = "UlConnection Failed! (WSA failed to start)";
std::this_thread::sleep_for(std::chrono::milliseconds(10)); ListOfMods = "-";
if (confirmed == 1)
return true;
NetReset();
Terminate = true;
TCPTerminate = true;
ping = -1;
return false;
}
void StartSync(const std::string& Data) {
std::string IP = GetAddr(Data.substr(1, Data.find(':') - 1));
if (IP.find('.') == -1) {
if (IP == "DNS")
UlStatus = "UlConnection Failed! (DNS Lookup Failed)";
else
UlStatus = "UlConnection Failed! (WSA failed to start)";
Terminate = true; Terminate = true;
CoreSend("L");
return; return;
} }
CheckLocalKey(); CheckLocalKey();
@@ -82,193 +44,137 @@ void StartSync(const std::string& Data) {
Terminate = false; Terminate = false;
ConfList->clear(); ConfList->clear();
ping = -1; ping = -1;
std::thread GS(TCPGameServer, IP, std::stoi(Data.substr(Data.find(':') + 1))); std::thread GS(TCPGameServer,IP,std::stoi(Data.substr(Data.find(':')+1)));
GS.detach(); GS.detach();
info("Connecting to server"); info("Connecting to server");
} }
void CoreSend(std::string data) {
if (CoreSocket != -1) {
int res = send(CoreSocket, (data + "\n").c_str(), int(data.size()) + 1, 0);
if (res < 0) {
debug("(Core) send failed with error: " + std::to_string(WSAGetLastError()));
}
}
}
bool IsAllowedLink(const std::string& Link) { bool IsAllowedLink(const std::string& Link) {
std::regex link_pattern(R"(https:\/\/(?:\w+)?(?:\.)?(?:beammp\.com|discord\.gg|patreon\.com\/BeamMP))"); std::regex link_pattern(R"(https:\/\/(?:\w+)?(?:\.)?(?:beammp\.com|discord\.gg))");
std::smatch link_match; std::smatch link_match;
return std::regex_search(Link, link_match, link_pattern) && link_match.position() == 0; return std::regex_search(Link,link_match, link_pattern) && link_match.position() == 0;
} }
void Parse(std::string Data, SOCKET CSocket) { void Parse(std::string Data,SOCKET CSocket){
char Code = Data.at(0), SubCode = 0; char Code = Data.at(0), SubCode = 0;
if (Data.length() > 1) if(Data.length() > 1)SubCode = Data.at(1);
SubCode = Data.at(1); switch (Code){
switch (Code) { case 'A':
case 'A': Data = Data.substr(0,1);
Data = Data.substr(0, 1); break;
break; case 'B':
case 'B':
NetReset();
Terminate = true;
TCPTerminate = true;
Data = Code + HTTP::Get("https://backend.beammp.com/servers-info");
break;
case 'C':
StartSync(Data);
Data.clear();
break;
case 'O': // open default browser with URL
if (IsAllowedLink(Data.substr(1))) {
#if defined(__linux)
if (char* browser = getenv("BROWSER"); browser != nullptr && !std::string_view(browser).empty()) {
pid_t pid;
auto arg = Data.substr(1);
char* argv[] = { browser, arg.data() };
auto status = posix_spawn(&pid, browser, nullptr, nullptr, argv, environ);
if (status == 0) {
debug("Browser PID: " + std::to_string(pid));
// we don't wait for it to exit, because we just don't care.
// typically, you'd waitpid() here.
} else {
error("Failed to open the following link in the browser (error follows below): " + arg);
error(std::string("posix_spawn: ") + strerror(status));
}
} else {
error("Failed to open the following link in the browser because the $BROWSER environment variable is not set: " + Data.substr(1));
}
#elif defined(WIN32)
ShellExecuteA(nullptr, "open", Data.substr(1).c_str(), nullptr, nullptr, SW_SHOW); /// TODO: Look at when working on linux port
#endif
info("Opening Link \"" + Data.substr(1) + "\"");
}
Data.clear();
break;
case 'P':
Data = Code + std::to_string(ProxyPort);
break;
case 'U':
if (SubCode == 'l')
Data = UlStatus;
if (SubCode == 'p') {
if (ping > 800) {
Data = "Up-2";
} else
Data = "Up" + std::to_string(ping);
}
if (!SubCode) {
std::string Ping;
if (ping > 800)
Ping = "-2";
else
Ping = std::to_string(ping);
Data = std::string(UlStatus) + "\n" + "Up" + Ping;
}
break;
case 'M':
Data = MStatus;
break;
case 'Q':
if (SubCode == 'S') {
NetReset(); NetReset();
Terminate = true; Terminate = true;
TCPTerminate = true; TCPTerminate = true;
ping = -1; Data = Code + HTTP::Get("https://backend.beammp.com/servers-info");
} break;
if (SubCode == 'G') { case 'C':
debug("Closing via 'G' packet"); ListOfMods.clear();
exit(2); StartSync(Data);
} while(ListOfMods.empty() && !Terminate){
Data.clear(); std::this_thread::sleep_for(std::chrono::seconds(1));
break;
case 'R': // will send mod name
if (ConfList->find(Data) == ConfList->end()) {
ConfList->insert(Data);
ModLoaded = true;
}
Data.clear();
break;
case 'Z':
Data = "Z" + GetVer();
break;
case 'N':
if (SubCode == 'c') {
nlohmann::json Auth = {
{ "Auth", LoginAuth ? 1 : 0 },
};
if (!Username.empty()) {
Auth["username"] = Username;
} }
if (!UserRole.empty()) { if(ListOfMods == "-")Data = "L";
Auth["role"] = UserRole; else Data = "L"+ListOfMods;
break;
case 'O': //open default browser with URL
if(IsAllowedLink(Data.substr(1))) {
ShellExecuteA(nullptr, "open", Data.substr(1).c_str(), nullptr, nullptr,SW_SHOW); ///TODO: Look at when working on linux port
info("Opening Link \"" + Data.substr(1) + "\"");
} }
if (UserID != -1) { Data.clear();
Auth["id"] = UserID; break;
case 'P':
Data = Code + std::to_string(ProxyPort);
break;
case 'U':
if(SubCode == 'l')Data = UlStatus;
if(SubCode == 'p'){
if(ping > 800){
Data = "Up-2";
}else Data = "Up" + std::to_string(ping);
} }
Data = "N" + Auth.dump(); if(!SubCode){
} else { std::string Ping;
Data = "N" + Login(Data.substr(Data.find(':') + 1)); if(ping > 800)Ping = "-2";
} else Ping = std::to_string(ping);
break; Data = std::string(UlStatus) + "\n" + "Up" + Ping;
case 'W': }
if (SubCode == 'Y') { break;
confirmed = 1; case 'M':
} else if (SubCode == 'N') { Data = MStatus;
confirmed = 0; break;
} case 'Q':
if(SubCode == 'S'){
Data.clear(); NetReset();
break; Terminate = true;
default: TCPTerminate = true;
Data.clear(); ping = -1;
break; }
if(SubCode == 'G')exit(2);
Data.clear();
break;
case 'R': //will send mod name
if(ConfList->find(Data) == ConfList->end()){
ConfList->insert(Data);
ModLoaded = true;
}
Data.clear();
break;
case 'Z':
Data = "Z" + GetVer();
break;
case 'N':
if (SubCode == 'c'){
Data = "N{\"Auth\":"+std::to_string(LoginAuth)+"}";
}else{
Data = "N" + Login(Data.substr(Data.find(':') + 1));
}
break;
default:
Data.clear();
break;
} }
if (!Data.empty() && CSocket != -1) { if(!Data.empty() && CSocket != -1){
int res = send(CSocket, (Data + "\n").c_str(), int(Data.size()) + 1, 0); int res = send(CSocket, (Data+"\n").c_str(), int(Data.size())+1, 0);
if (res < 0) { if(res < 0){
debug("(Core) send failed with error: " + std::to_string(WSAGetLastError())); debug("(Core) send failed with error: " + std::to_string(WSAGetLastError()));
} }
} }
} }
void GameHandler(SOCKET Client) { void GameHandler(SOCKET Client){
CoreSocket = Client;
int32_t Size, Temp, Rcv; int32_t Size,Temp,Rcv;
char Header[10] = { 0 }; char Header[10] = {0};
do { do{
Rcv = 0; Rcv = 0;
do { do{
Temp = recv(Client, &Header[Rcv], 1, 0); Temp = recv(Client,&Header[Rcv],1,0);
if (Temp < 1) if(Temp < 1)break;
break; if(!isdigit(Header[Rcv]) && Header[Rcv] != '>') {
if (!isdigit(Header[Rcv]) && Header[Rcv] != '>') {
error("(Core) Invalid lua communication"); error("(Core) Invalid lua communication");
KillSocket(Client); KillSocket(Client);
return; return;
} }
} while (Header[Rcv++] != '>'); }while(Header[Rcv++] != '>');
if (Temp < 1) if(Temp < 1)break;
break; if(std::from_chars(Header,&Header[Rcv],Size).ptr[0] != '>'){
if (std::from_chars(Header, &Header[Rcv], Size).ptr[0] != '>') { debug("(Core) Invalid lua Header -> " + std::string(Header,Rcv));
debug("(Core) Invalid lua Header -> " + std::string(Header, Rcv));
break; break;
} }
std::string Ret(Size, 0); std::string Ret(Size,0);
Rcv = 0; Rcv = 0;
do { do{
Temp = recv(Client, &Ret[Rcv], Size - Rcv, 0); Temp = recv(Client,&Ret[Rcv],Size-Rcv,0);
if (Temp < 1) if(Temp < 1)break;
break;
Rcv += Temp; Rcv += Temp;
} while (Rcv < Size); }while(Rcv < Size);
if (Temp < 1) if(Temp < 1)break;
break;
Parse(Ret, Client); std::thread Respond(Parse, Ret, Client);
} while (Temp > 0); Respond.detach();
}while(Temp > 0);
if (Temp == 0) { if (Temp == 0) {
debug("(Core) Connection closing"); debug("(Core) Connection closing");
} else { } else {
@@ -277,10 +183,10 @@ void GameHandler(SOCKET Client) {
NetReset(); NetReset();
KillSocket(Client); KillSocket(Client);
} }
void localRes() { void localRes(){
MStatus = " "; MStatus = " ";
UlStatus = "Ulstart"; UlStatus = "Ulstart";
if (ConfList != nullptr) { if(ConfList != nullptr){
ConfList->clear(); ConfList->clear();
delete ConfList; delete ConfList;
ConfList = nullptr; ConfList = nullptr;
@@ -289,31 +195,25 @@ void localRes() {
} }
void CoreMain() { void CoreMain() {
debug("Core Network on start!"); debug("Core Network on start!");
SOCKET LSocket, CSocket;
struct addrinfo* res = nullptr;
struct addrinfo hints { };
int iRes;
#ifdef _WIN32
WSADATA wsaData; WSADATA wsaData;
iRes = WSAStartup(514, &wsaData); // 2.2 SOCKET LSocket,CSocket;
if (iRes) struct addrinfo *res = nullptr;
debug("WSAStartup failed with error: " + std::to_string(iRes)); struct addrinfo hints{};
#endif int iRes = WSAStartup(514, &wsaData); //2.2
if (iRes)debug("WSAStartup failed with error: " + std::to_string(iRes));
ZeroMemory(&hints, sizeof(hints)); ZeroMemory(&hints, sizeof(hints));
hints.ai_family = AF_INET; hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM; hints.ai_socktype = SOCK_STREAM;
hints.ai_protocol = IPPROTO_TCP; hints.ai_protocol = IPPROTO_TCP;
hints.ai_flags = AI_PASSIVE; hints.ai_flags = AI_PASSIVE;
iRes = getaddrinfo(nullptr, std::to_string(DEFAULT_PORT).c_str(), &hints, &res); iRes = getaddrinfo(nullptr, std::to_string(DEFAULT_PORT).c_str(), &hints, &res);
if (iRes) { if (iRes){
debug("(Core) addr info failed with error: " + std::to_string(iRes)); debug("(Core) addr info failed with error: " + std::to_string(iRes));
WSACleanup(); WSACleanup();
return; return;
} }
LSocket = socket(res->ai_family, res->ai_socktype, res->ai_protocol); LSocket = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (LSocket == -1) { if (LSocket == -1){
debug("(Core) socket failed with error: " + std::to_string(WSAGetLastError())); debug("(Core) socket failed with error: " + std::to_string(WSAGetLastError()));
freeaddrinfo(res); freeaddrinfo(res);
WSACleanup(); WSACleanup();
@@ -335,7 +235,7 @@ void CoreMain() {
WSACleanup(); WSACleanup();
return; return;
} }
do { do{
CSocket = accept(LSocket, nullptr, nullptr); CSocket = accept(LSocket, nullptr, nullptr);
if (CSocket == -1) { if (CSocket == -1) {
error("(Core) accept failed with error: " + std::to_string(WSAGetLastError())); error("(Core) accept failed with error: " + std::to_string(WSAGetLastError()));
@@ -345,38 +245,28 @@ void CoreMain() {
info("Game Connected!"); info("Game Connected!");
GameHandler(CSocket); GameHandler(CSocket);
warn("Game Reconnecting..."); warn("Game Reconnecting...");
} while (CSocket); }while(CSocket);
KillSocket(LSocket); KillSocket(LSocket);
WSACleanup(); WSACleanup();
} }
int Handle(EXCEPTION_POINTERS *ep){
#if defined(_WIN32)
int Handle(EXCEPTION_POINTERS* ep) {
char* hex = new char[100]; char* hex = new char[100];
sprintf_s(hex, 100, "%lX", ep->ExceptionRecord->ExceptionCode); sprintf_s(hex,100, "%lX", ep->ExceptionRecord->ExceptionCode);
except("(Core) Code : " + std::string(hex)); except("(Core) Code : " + std::string(hex));
delete[] hex; delete [] hex;
return 1; return 1;
} }
[[noreturn]] void CoreNetwork(){
while(true) {
#ifndef __MINGW32__
__try{
#endif #endif
CoreMain();
[[noreturn]] void CoreNetwork() { #ifndef __MINGW32__
while (true) { }__except(Handle(GetExceptionInformation())){}
#if not defined(__MINGW32__)
__try {
#endif #endif
CoreMain();
#if not defined(__MINGW32__) and not defined(__linux__)
} __except (Handle(GetExceptionInformation())) { }
#elif not defined(__MINGW32__) and defined(__linux__)
}
catch (...) {
except("(Core) Code : " + std::string(strerror(errno)));
}
#endif
std::this_thread::sleep_for(std::chrono::seconds(1)); std::this_thread::sleep_for(std::chrono::seconds(1));
} }
} }

24
src/Network/DNS.cpp Normal file → Executable file
View File

@@ -7,37 +7,25 @@
/// ///
#include <string> #include <string>
#if defined(_WIN32)
#include <winsock2.h> #include <winsock2.h>
#elif defined(__linux__)
#include "linuxfixes.h"
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include "Logger.h" #include "Logger.h"
std::string GetAddr(const std::string& IP) { std::string GetAddr(const std::string&IP){
if (IP.find_first_not_of("0123456789.") == -1) if(IP.find_first_not_of("0123456789.") == -1)return IP;
return IP;
hostent* host;
#ifdef _WIN32
WSADATA wsaData; WSADATA wsaData;
if (WSAStartup(514, &wsaData) != 0) { hostent *host;
if(WSAStartup(514, &wsaData) != 0){
error("WSA Startup Failed!"); error("WSA Startup Failed!");
WSACleanup(); WSACleanup();
return ""; return "";
} }
#endif
host = gethostbyname(IP.c_str()); host = gethostbyname(IP.c_str());
if (!host) { if(!host){
error("DNS lookup failed! on " + IP); error("DNS lookup failed! on " + IP);
WSACleanup(); WSACleanup();
return "DNS"; return "DNS";
} }
std::string Ret = inet_ntoa(*((struct in_addr*)host->h_addr)); std::string Ret = inet_ntoa(*((struct in_addr *)host->h_addr));
WSACleanup(); WSACleanup();
return Ret; return Ret;
} }

266
src/Network/GlobalHandler.cpp Normal file → Executable file
View File

@@ -5,163 +5,130 @@
/// ///
/// Created by Anonymous275 on 7/25/2020 /// Created by Anonymous275 on 7/25/2020
/// ///
#include "Network/network.hpp" #include "Network/network.h"
#include <memory>
#include <zlib.h>
#if defined(_WIN32)
#include <winsock2.h> #include <winsock2.h>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#elif defined(__linux__)
#include "linuxfixes.h"
#include <arpa/inet.h>
#include <cstring>
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#endif
#include "Logger.h" #include "Logger.h"
#include <charconv> #include <charconv>
#include <mutex>
#include <string> #include <string>
#include <thread> #include <thread>
#include <mutex>
std::chrono::time_point<std::chrono::high_resolution_clock> PingStart, PingEnd; std::chrono::time_point<std::chrono::high_resolution_clock> PingStart,PingEnd;
bool GConnected = false; bool GConnected = false;
bool CServer = true; bool CServer = true;
SOCKET CSocket = -1; SOCKET CSocket = -1;
SOCKET GSocket = -1; SOCKET GSocket = -1;
int KillSocket(uint64_t Dead) { int KillSocket(uint64_t Dead){
if (Dead == (SOCKET)-1) { if(Dead == (SOCKET)-1){
debug("Kill socket got -1 returning..."); debug("Kill socket got -1 returning...");
return 0; return 0;
} }
shutdown(Dead, SD_BOTH); shutdown(Dead,SD_BOTH);
int a = closesocket(Dead); int a = closesocket(Dead);
if (a != 0) { if(a != 0){
warn("Failed to close socket!"); warn("Failed to close socket!");
} }
return a; return a;
} }
bool CheckBytes(uint32_t Bytes) { bool CheckBytes(uint32_t Bytes){
if (Bytes == 0) { if(Bytes == 0){
debug("(Proxy) Connection closing"); debug("(Proxy) Connection closing");
return false; return false;
} else if (Bytes < 0) { }else if(Bytes < 0){
debug("(Proxy) send failed with error: " + std::to_string(WSAGetLastError())); debug("(Proxy) send failed with error: " + std::to_string(WSAGetLastError()));
return false; return false;
} }
return true; return true;
} }
void GameSend(std::string_view Data) { void GameSend(std::string Data){
static std::mutex Lock; static std::mutex Lock;
std::scoped_lock Guard(Lock); std::scoped_lock Guard(Lock);
if (TCPTerminate || !GConnected || CSocket == -1) if(TCPTerminate || !GConnected || CSocket == -1)return;
return; int32_t Size,Temp,Sent;
int32_t Size, Temp, Sent; Data += '\n';
Size = int32_t(Data.size()); Size = int32_t(Data.size());
Sent = 0; Sent = 0;
#ifdef DEBUG #ifdef DEBUG
if (Size > 1000) { if(Size > 1000){
debug("Launcher -> game (" + std::to_string(Size) + ")"); debug("Launcher -> game (" +std::to_string(Size)+")");
}
#endif
do {
if (Sent > -1) {
Temp = send(CSocket, &Data[Sent], Size - Sent, 0);
} }
if (!CheckBytes(Temp)) #endif
return; do{
if(Sent > -1){
Temp = send(CSocket, &Data[Sent], Size - Sent, 0);
}
if(!CheckBytes(Temp))return;
Sent += Temp; Sent += Temp;
} while (Sent < Size); }while(Sent < Size);
// send separately to avoid an allocation for += "\n"
Temp = send(CSocket, "\n", 1, 0);
if (!CheckBytes(Temp)) {
return;
}
} }
void ServerSend(std::string Data, bool Rel) { void ServerSend(std::string Data, bool Rel){
if (Terminate || Data.empty()) if(Terminate || Data.empty())return;
return; if(Data.find("Zp") != std::string::npos && Data.size() > 500){
if (Data.find("Zp") != std::string::npos && Data.size() > 500) {
abort(); abort();
} }
char C = 0; char C = 0;
bool Ack = false; bool Ack = false;
int DLen = int(Data.length()); int DLen = int(Data.length());
if (DLen > 3) if(DLen > 3)C = Data.at(0);
C = Data.at(0); if (C == 'O' || C == 'T')Ack = true;
if (C == 'O' || C == 'T') if(C == 'N' || C == 'W' || C == 'Y' || C == 'V' || C == 'E' || C == 'C')Rel = true;
Ack = true; if(Ack || Rel){
if (C == 'N' || C == 'W' || C == 'Y' || C == 'V' || C == 'E' || C == 'C') if(Ack || DLen > 1000)SendLarge(Data);
Rel = true; else TCPSend(Data,TCPSock);
if (compressBound(Data.size()) > 1024) }else UDPSend(Data);
Rel = true;
if (Ack || Rel) {
if (Ack || DLen > 1000)
SendLarge(Data);
else
TCPSend(Data, TCPSock);
} else
UDPSend(Data);
if (DLen > 1000) { if (DLen > 1000) {
debug("(Launcher->Server) Bytes sent: " + std::to_string(Data.length()) + " : " debug("(Launcher->Server) Bytes sent: " + std::to_string(Data.length()) + " : "
+ Data.substr(0, 10) + Data.substr(0, 10)
+ Data.substr(Data.length() - 10)); + Data.substr(Data.length() - 10));
} else if (C == 'Z') { }else if(C == 'Z'){
// debug("(Game->Launcher) : " + Data); //debug("(Game->Launcher) : " + Data);
} }
} }
void NetReset() { void NetReset(){
TCPTerminate = false; TCPTerminate = false;
GConnected = false; GConnected = false;
Terminate = false; Terminate = false;
UlStatus = "Ulstart"; UlStatus = "Ulstart";
MStatus = " "; MStatus = " ";
if (UDPSock != (SOCKET)(-1)) { if(UDPSock != (SOCKET)(-1)){
debug("Terminating UDP Socket: " + std::to_string(TCPSock)); debug("Terminating UDP Socket : " + std::to_string(TCPSock));
KillSocket(UDPSock); KillSocket(UDPSock);
} }
UDPSock = -1; UDPSock = -1;
if (TCPSock != (SOCKET)(-1)) { if(TCPSock != (SOCKET)(-1)){
debug("Terminating TCP Socket: " + std::to_string(TCPSock)); debug("Terminating TCP Socket : " + std::to_string(TCPSock));
KillSocket(TCPSock); KillSocket(TCPSock);
} }
TCPSock = -1; TCPSock = -1;
if (GSocket != (SOCKET)(-1)) { if(GSocket != (SOCKET)(-1)){
debug("Terminating GTCP Socket: " + std::to_string(GSocket)); debug("Terminating GTCP Socket : " + std::to_string(GSocket));
KillSocket(GSocket); KillSocket(GSocket);
} }
GSocket = -1; GSocket = -1;
} }
SOCKET SetupListener() { SOCKET SetupListener(){
if (GSocket != -1) if(GSocket != -1)return GSocket;
return GSocket; struct addrinfo *result = nullptr;
struct addrinfo* result = nullptr; struct addrinfo hints{};
struct addrinfo hints { };
int iRes;
#ifdef _WIN32
WSADATA wsaData; WSADATA wsaData;
iRes = WSAStartup(514, &wsaData); // 2.2 int iRes = WSAStartup(514, &wsaData); //2.2
if (iRes != 0) { if (iRes != 0) {
error("(Proxy) WSAStartup failed with error: " + std::to_string(iRes)); error("(Proxy) WSAStartup failed with error: " + std::to_string(iRes));
return -1; return -1;
} }
#endif
ZeroMemory(&hints, sizeof(hints)); ZeroMemory(&hints, sizeof(hints));
hints.ai_family = AF_INET; hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM; hints.ai_socktype = SOCK_STREAM;
hints.ai_protocol = IPPROTO_TCP; hints.ai_protocol = IPPROTO_TCP;
hints.ai_flags = AI_PASSIVE; hints.ai_flags = AI_PASSIVE;
iRes = getaddrinfo(nullptr, std::to_string(DEFAULT_PORT + 1).c_str(), &hints, &result); iRes = getaddrinfo(nullptr, std::to_string(DEFAULT_PORT+1).c_str(), &hints, &result);
if (iRes != 0) { if (iRes != 0) {
error("(Proxy) info failed with error: " + std::to_string(iRes)); error("(Proxy) info failed with error: " + std::to_string(iRes));
WSACleanup(); WSACleanup();
@@ -173,7 +140,7 @@ SOCKET SetupListener() {
WSACleanup(); WSACleanup();
return -1; return -1;
} }
iRes = bind(GSocket, result->ai_addr, (int)result->ai_addrlen); iRes = bind(GSocket, result->ai_addr, (int) result->ai_addrlen);
if (iRes == SOCKET_ERROR) { if (iRes == SOCKET_ERROR) {
error("(Proxy) bind failed with error: " + std::to_string(WSAGetLastError())); error("(Proxy) bind failed with error: " + std::to_string(WSAGetLastError()));
freeaddrinfo(result); freeaddrinfo(result);
@@ -191,64 +158,59 @@ SOCKET SetupListener() {
} }
return GSocket; return GSocket;
} }
void AutoPing() { void AutoPing(){
while (!Terminate) { while(!Terminate){
ServerSend("p", false); ServerSend("p",false);
PingStart = std::chrono::high_resolution_clock::now(); PingStart = std::chrono::high_resolution_clock::now();
std::this_thread::sleep_for(std::chrono::seconds(1)); std::this_thread::sleep_for(std::chrono::seconds (1));
} }
} }
int ClientID = -1; int ClientID = -1;
void ParserAsync(std::string_view Data) { void ParserAsync(const std::string& Data){
if (Data.empty()) if(Data.empty())return;
return; char Code = Data.at(0),SubCode = 0;
char Code = Data.at(0), SubCode = 0; if(Data.length() > 1)SubCode = Data.at(1);
if (Data.length() > 1)
SubCode = Data.at(1);
switch (Code) { switch (Code) {
case 'p': case 'p':
PingEnd = std::chrono::high_resolution_clock::now(); PingEnd = std::chrono::high_resolution_clock::now();
if (PingStart > PingEnd) if(PingStart > PingEnd)ping = 0;
ping = 0; else ping = int(std::chrono::duration_cast<std::chrono::milliseconds>(PingEnd-PingStart).count());
else return;
ping = int(std::chrono::duration_cast<std::chrono::milliseconds>(PingEnd - PingStart).count()); case 'M':
return; MStatus = Data;
case 'M': UlStatus = "Uldone";
MStatus = Data; return;
UlStatus = "Uldone"; default:
return; break;
default:
break;
} }
GameSend(Data); GameSend(Data);
} }
void ServerParser(std::string_view Data) { void ServerParser(const std::string& Data){
ParserAsync(Data); ParserAsync(Data);
} }
void NetMain(const std::string& IP, int Port) { void NetMain(const std::string& IP, int Port){
std::thread Ping(AutoPing); std::thread Ping(AutoPing);
Ping.detach(); Ping.detach();
UDPClientMain(IP, Port); UDPClientMain(IP,Port);
CServer = true; CServer = true;
Terminate = true; Terminate = true;
info("Connection Terminated!"); info("Connection Terminated!");
} }
void TCPGameServer(const std::string& IP, int Port) { void TCPGameServer(const std::string& IP, int Port){
GSocket = SetupListener(); GSocket = SetupListener();
std::unique_ptr<std::thread> ClientThread {}; while (!TCPTerminate && GSocket != -1){
std::unique_ptr<std::thread> NetMainThread {};
while (!TCPTerminate && GSocket != -1) {
debug("MAIN LOOP OF GAME SERVER"); debug("MAIN LOOP OF GAME SERVER");
GConnected = false; GConnected = false;
if (!CServer) { if(!CServer){
warn("Connection still alive terminating"); warn("Connection still alive terminating");
NetReset(); NetReset();
TCPTerminate = true; TCPTerminate = true;
Terminate = true; Terminate = true;
break; break;
} }
if (CServer) { if(CServer) {
ClientThread = std::make_unique<std::thread>(TCPClientMain, IP, Port); std::thread Client(TCPClientMain, IP, Port);
Client.detach();
} }
CSocket = accept(GSocket, nullptr, nullptr); CSocket = accept(GSocket, nullptr, nullptr);
if (CSocket == -1) { if (CSocket == -1) {
@@ -257,61 +219,45 @@ void TCPGameServer(const std::string& IP, int Port) {
} }
debug("(Proxy) Game Connected!"); debug("(Proxy) Game Connected!");
GConnected = true; GConnected = true;
if (CServer) { if(CServer){
NetMainThread = std::make_unique<std::thread>(NetMain, IP, Port); std::thread t1(NetMain, IP, Port);
t1.detach();
CServer = false; CServer = false;
} }
int32_t Size, Temp, Rcv; int32_t Size,Temp,Rcv;
char Header[10] = { 0 }; char Header[10] = {0};
// Read byte by byte until '>' is rcved then get the size and read based on it //Read byte by byte until '>' is rcved then get the size and read based on it
do { do{
Rcv = 0; Rcv = 0;
do { do{
Temp = recv(CSocket, &Header[Rcv], 1, 0); Temp = recv(CSocket,&Header[Rcv],1,0);
if (Temp < 1 || TCPTerminate) if(Temp < 1 || TCPTerminate)break;
break; }while(Header[Rcv++] != '>');
} while (Header[Rcv++] != '>'); if(Temp < 1 || TCPTerminate)break;
if (Temp < 1 || TCPTerminate) if(std::from_chars(Header,&Header[Rcv],Size).ptr[0] != '>'){
break; debug("(Game) Invalid lua Header -> " + std::string(Header,Rcv));
if (std::from_chars(Header, &Header[Rcv], Size).ptr[0] != '>') {
debug("(Game) Invalid lua Header -> " + std::string(Header, Rcv));
break; break;
} }
std::string Ret(Size, 0); std::string Ret(Size,0);
Rcv = 0; Rcv = 0;
do { do{
Temp = recv(CSocket, &Ret[Rcv], Size - Rcv, 0); Temp = recv(CSocket,&Ret[Rcv],Size-Rcv,0);
if (Temp < 1) if(Temp < 1)break;
break;
Rcv += Temp; Rcv += Temp;
} while (Rcv < Size && !TCPTerminate); }while(Rcv < Size && !TCPTerminate);
if (Temp < 1 || TCPTerminate) if(Temp < 1 || TCPTerminate)break;
break;
ServerSend(Ret, false); ServerSend(Ret,false);
} while (Temp > 0 && !TCPTerminate); }while(Temp > 0 && !TCPTerminate);
if (Temp == 0) if(Temp == 0)debug("(Proxy) Connection closing");
debug("(Proxy) Connection closing"); else debug("(Proxy) recv failed error : " + std::to_string(WSAGetLastError()));
else
debug("(Proxy) recv failed error : " + std::to_string(WSAGetLastError()));
} }
TCPTerminate = true; TCPTerminate = true;
GConnected = false; GConnected = false;
Terminate = true; Terminate = true;
if (ClientThread) { if(CSocket != SOCKET_ERROR)KillSocket(CSocket);
debug("Waiting for client thread");
ClientThread->join();
debug("Client thread done");
}
if (NetMainThread) {
debug("Waiting for net main thread");
NetMainThread->join();
debug("Net main thread done");
}
if (CSocket != SOCKET_ERROR)
KillSocket(CSocket);
debug("END OF GAME SERVER"); debug("END OF GAME SERVER");
} }

390
src/Network/Http.cpp Normal file → Executable file
View File

@@ -1,268 +1,122 @@
// Copyright (c) 2019-present Anonymous275. // Copyright (c) 2019-present Anonymous275.
// BeamMP Launcher code is not in the public domain and is not free software. // BeamMP Launcher code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries. // One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository. // Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
/// ///
/// Created by Anonymous275 on 7/18/2020 /// Created by Anonymous275 on 7/18/2020
/// ///
#define CPPHTTPLIB_OPENSSL_SUPPORT
#include "Http.h"
#include <Logger.h> #include <iostream>
#include <Network/network.hpp> #include <Logger.h>
#include <Startup.h> #include <fstream>
#include <Utils.h> #include "Http.h"
#include <cmath> #include <mutex>
#include <curl/curl.h> #include <cmath>
#include <curl/easy.h> #include <httplib.h>
#include <filesystem>
#include <fstream> bool HTTP::isDownload = false;
#include <httplib.h> std::string HTTP::Get(const std::string &IP) {
#include <iostream> static std::mutex Lock;
#include <mutex> std::scoped_lock Guard(Lock);
#include <nlohmann/json.hpp>
auto pos = IP.find('/',10);
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"; httplib::Client cli(IP.substr(0, pos).c_str());
std::filesystem::create_directories(folder); cli.set_connection_timeout(std::chrono::seconds(10));
if (!std::filesystem::exists(folder / "WHAT IS THIS FOLDER.txt")) { cli.set_follow_location(true);
std::ofstream ignore { folder / "WHAT IS THIS FOLDER.txt" }; auto res = cli.Get(IP.substr(pos).c_str(), ProgressBar);
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; std::string Ret;
}
const auto file = folder / (method + ".json"); if(res){
// 1 MB limit if(res->status == 200){
if (std::filesystem::exists(file) && std::filesystem::file_size(file) > 1'000'000) { Ret = res->body;
std::filesystem::rename(file, file.generic_string() + ".bak"); }else error(res->reason);
}
}else{
std::ofstream of { file, std::ios::app }; if(isDownload) {
nlohmann::json js { std::cout << "\n";
{ "utc", std::chrono::system_clock::now().time_since_epoch().count() }, }
{ "target", target }, error("HTTP Get failed on " + to_string(res.error()));
{ "client_info", { }
{ "openssl_verify_result", client.get_openssl_verify_result() },
{ "host", client.host() }, return Ret;
{ "port", client.port() }, }
{ "socket_open", client.is_socket_open() },
{ "valid", client.is_valid() }, std::string HTTP::Post(const std::string& IP, const std::string& Fields) {
} }, static std::mutex Lock;
}; std::scoped_lock Guard(Lock);
if (result) {
auto value = result.value(); auto pos = IP.find('/',10);
js["result"] = {};
js["result"]["body"] = value.body; httplib::Client cli(IP.substr(0, pos).c_str());
js["result"]["status"] = value.status; cli.set_connection_timeout(std::chrono::seconds(10));
js["result"]["headers"] = value.headers; std::string Ret;
js["result"]["version"] = value.version;
js["result"]["location"] = value.location; if(!Fields.empty()) {
js["result"]["reason"] = value.reason; httplib::Result res = cli.Post(IP.substr(pos).c_str(), Fields, "application/json");
}
of << js.dump(); if(res) {
} catch (const std::exception& e) { if (res->status != 200) {
error(e.what()); error(res->reason);
} }
Ret = res->body;
static size_t CurlWriteCallback(void* contents, size_t size, size_t nmemb, void* userp) { }else{
std::string* Result = reinterpret_cast<std::string*>(userp); error("HTTP Post failed on " + to_string(res.error()));
std::string NewContents(reinterpret_cast<char*>(contents), size * nmemb); }
*Result += NewContents; }else{
return size * nmemb; httplib::Result res = cli.Post(IP.substr(pos).c_str());
} if(res) {
if (res->status != 200) {
bool HTTP::isDownload = false; error(res->reason);
std::string HTTP::Get(const std::string& IP) { }
std::string Ret; Ret = res->body;
static thread_local CURL* curl = curl_easy_init(); }else{
if (curl) { error("HTTP Post failed on " + to_string(res.error()));
CURLcode res; }
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str()); }
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, CurlWriteCallback);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void*)&Ret); if(Ret.empty())return "-1";
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10); // seconds else return Ret;
res = curl_easy_perform(curl); }
if (res != CURLE_OK) {
error("GET to " + IP + " failed: " + std::string(curl_easy_strerror(res))); bool HTTP::ProgressBar(size_t c, size_t t){
return ""; if(isDownload) {
} static double last_progress, progress_bar_adv;
} else { progress_bar_adv = round(c / double(t) * 25);
error("Curl easy init failed"); std::cout << "\r";
return ""; std::cout << "Progress : [ ";
} std::cout << round(c / double(t) * 100);
return Ret; std::cout << "% ] [";
} int i;
for (i = 0; i <= progress_bar_adv; i++)std::cout << "#";
std::string HTTP::Post(const std::string& IP, const std::string& Fields) { for (i = 0; i < 25 - progress_bar_adv; i++)std::cout << ".";
std::string Ret; std::cout << "]";
static thread_local CURL* curl = curl_easy_init(); last_progress = round(c / double(t) * 100);
if (curl) { }
CURLcode res; return true;
curl_easy_setopt(curl, CURLOPT_URL, IP.c_str()); }
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, CurlWriteCallback);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void*)&Ret); bool HTTP::Download(const std::string &IP, const std::string &Path) {
curl_easy_setopt(curl, CURLOPT_POST, 1); static std::mutex Lock;
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, Fields.c_str()); std::scoped_lock Guard(Lock);
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, Fields.size());
struct curl_slist* list = nullptr; isDownload = true;
list = curl_slist_append(list, "Content-Type: application/json"); std::string Ret = Get(IP);
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, list); isDownload = false;
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 10); // seconds
res = curl_easy_perform(curl); if(Ret.empty())return false;
curl_slist_free_all(list);
if (res != CURLE_OK) { std::ofstream File(Path, std::ios::binary);
error("POST to " + IP + " failed: " + std::string(curl_easy_strerror(res))); if(File.is_open()) {
return ""; File << Ret;
} File.close();
} else { std::cout << "\n";
error("Curl easy init failed"); info("Download Complete!");
return ""; }else{
} error("Failed to open file directory: " + Path);
return Ret; return false;
} }
bool HTTP::Download(const std::string& IP, const std::string& Path) { return true;
static std::mutex Lock; }
std::scoped_lock Guard(Lock);
info("Downloading an update (this may take a while)");
std::string Ret = Get(IP);
if (Ret.empty()) {
error("Download failed");
return false;
}
std::ofstream File(Path, std::ios::binary);
if (File.is_open()) {
File << Ret;
File.close();
std::cout << "\n";
info("Download Complete!");
} else {
error("Failed to open file directory: " + Path);
return false;
}
return true;
}
void set_headers(httplib::Response& res) {
res.set_header("Access-Control-Allow-Origin", "*");
res.set_header("Access-Control-Request-Method", "POST, OPTIONS, GET");
res.set_header("Access-Control-Request-Headers", "X-API-Version");
}
void HTTP::StartProxy() {
std::thread proxy([&]() {
httplib::Server HTTPProxy;
httplib::Headers headers = {
{ "User-Agent", "BeamMP-Launcher/" + GetVer() + GetPatch() },
{ "Accept", "*/*" }
};
httplib::Client backend("https://backend.beammp.com");
httplib::Client forum("https://forum.beammp.com");
const std::string pattern = ".*";
auto handle_request = [&](const httplib::Request& req, httplib::Response& res) {
set_headers(res);
if (req.has_header("X-BMP-Authentication")) {
headers.emplace("X-BMP-Authentication", PrivateKey);
}
if (req.has_header("X-API-Version")) {
headers.emplace("X-API-Version", req.get_header_value("X-API-Version"));
}
const std::vector<std::string> path = Utils::Split(req.path, "/");
httplib::Result cli_res;
const std::string method = req.method;
std::string host = "";
if (!path.empty())
host = path[0];
if (host == "backend") {
std::string remaining_path = req.path.substr(std::strlen("/backend"));
if (method == "GET")
cli_res = backend.Get(remaining_path, headers);
else if (method == "POST")
cli_res = backend.Post(remaining_path, headers);
} else if (host == "avatar") {
bool error = false;
std::string username;
std::string avatar_size = "100";
if (path.size() > 1) {
username = path[1];
} else {
error = true;
}
if (path.size() > 2) {
try {
if (std::stoi(path[2]) > 0)
avatar_size = path[2];
} catch (std::exception&) { }
}
httplib::Result summary_res;
if (!error) {
summary_res = forum.Get("/u/" + username + ".json", headers);
if (!summary_res || summary_res->status != 200) {
error = true;
}
}
if (!error) {
try {
nlohmann::json d = nlohmann::json::parse(summary_res->body, nullptr, false); // can fail with parse_error
auto user = d.at("user"); // can fail with out_of_range
auto avatar_link_json = user.at("avatar_template"); // can fail with out_of_range
auto avatar_link = avatar_link_json.get<std::string>();
size_t start_pos = avatar_link.find("{size}");
if (start_pos != std::string::npos)
avatar_link.replace(start_pos, std::strlen("{size}"), avatar_size);
cli_res = forum.Get(avatar_link, headers);
} catch (std::exception&) {
error = true;
}
}
if (error) {
cli_res = forum.Get("/user_avatar/forum.beammp.com/user/0/0.png", headers);
}
} else {
res.set_content("Host not found", "text/plain");
return;
}
if (cli_res) {
res.set_content(cli_res->body, cli_res->get_header_value("Content-Type"));
} else {
res.set_content(to_string(cli_res.error()), "text/plain");
}
};
HTTPProxy.Get(pattern, [&](const httplib::Request& req, httplib::Response& res) {
handle_request(req, res);
});
HTTPProxy.Post(pattern, [&](const httplib::Request& req, httplib::Response& res) {
handle_request(req, res);
});
ProxyPort = HTTPProxy.bind_to_any_port("0.0.0.0");
debug("HTTP Proxy listening on port " + std::to_string(ProxyPort));
HTTPProxy.listen_after_bind();
});
proxy.detach();
}

620
src/Network/Resources.cpp Normal file → Executable file
View File

@@ -6,198 +6,150 @@
/// Created by Anonymous275 on 4/11/2020 /// Created by Anonymous275 on 4/11/2020
/// ///
#include "Network/network.hpp" #include "Network/network.h"
#include <chrono>
#include <iomanip>
#include <ios>
#include <mutex>
#include <nlohmann/json.hpp>
#include <openssl/err.h>
#include <openssl/evp.h>
#if defined(_WIN32)
#include <ws2tcpip.h> #include <ws2tcpip.h>
#elif defined(__linux__)
#include <arpa/inet.h>
#include <cstring>
#include <errno.h>
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
#endif
#include "Logger.h"
#include "Startup.h"
#include <Utils.h>
#include <atomic>
#include <cmath>
#include <cstring>
#include <filesystem> #include <filesystem>
#include <fstream> #include "Startup.h"
#include <future> #include "Logger.h"
#include <iostream> #include <iostream>
#include <cstring>
#include <fstream>
#include <string>
#include <thread> #include <thread>
#include <atomic>
#include "hashpp.h" #include <vector>
#include <future>
#include <cmath>
namespace fs = std::filesystem; namespace fs = std::filesystem;
std::string ListOfMods;
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() { void CheckForDir(){
if (!fs::exists(CachingDirectory)) { if(!fs::exists("Resources")){
try { _wmkdir(L"Resources");
fs::create_directories(CachingDirectory);
} catch (const std::exception& e) {
error(std::string("Failed to create caching directory: ") + e.what() + ". This is a fatal error. Please make sure to configure a directory which you have permission to create, read and write from/to.");
std::this_thread::sleep_for(std::chrono::seconds(3));
std::exit(1);
}
} }
} }
void WaitForConfirm() { void WaitForConfirm(){
while (!Terminate && !ModLoaded) { while(!Terminate && !ModLoaded){
std::this_thread::sleep_for(std::chrono::milliseconds(10)); std::this_thread::sleep_for(std::chrono::milliseconds(10));
} }
ModLoaded = false; ModLoaded = false;
} }
void Abord() {
void Abord(){
Terminate = true; Terminate = true;
TCPTerminate = true; TCPTerminate = true;
info("Terminated!"); info("Terminated!");
} }
std::string Auth(SOCKET Sock) { std::string Auth(SOCKET Sock){
TCPSend("VC" + GetVer(), Sock); TCPSend("VC" + GetVer(),Sock);
auto Res = TCPRcv(Sock); auto Res = TCPRcv(Sock);
if (Res.empty() || Res[0] == 'E' || Res[0] == 'K') { if(Res.empty() || Res[0] == 'E' || Res[0] == 'K'){
Abord(); Abord();
CoreSend("L");
return ""; return "";
} }
TCPSend(PublicKey, Sock); TCPSend(PublicKey,Sock);
if (Terminate) { if(Terminate)return "";
CoreSend("L");
return "";
}
Res = TCPRcv(Sock); Res = TCPRcv(Sock);
if (Res.empty() || Res[0] != 'P') { if(Res.empty() || Res[0] != 'P'){
Abord(); Abord();
CoreSend("L");
return ""; return "";
} }
Res = Res.substr(1); Res = Res.substr(1);
if (Res.find_first_not_of("0123456789") == std::string::npos) { if(Res.find_first_not_of("0123456789") == std::string::npos){
ClientID = std::stoi(Res); ClientID = std::stoi(Res);
} else { }else{
Abord(); Abord();
CoreSend("L");
UUl("Authentication failed!"); UUl("Authentication failed!");
return ""; return "";
} }
TCPSend("SR", Sock); TCPSend("SR",Sock);
if (Terminate) { if(Terminate)return "";
CoreSend("L");
return "";
}
Res = TCPRcv(Sock); Res = TCPRcv(Sock);
if (Res[0] == 'E' || Res[0] == 'K') { if(Res[0] == 'E' || Res[0] == 'K'){
Abord(); Abord();
CoreSend("L");
return ""; return "";
} }
if (Res.empty() || Res == "-") { if(Res.empty() || Res == "-"){
info("Didn't Receive any mods..."); info("Didn't Receive any mods...");
CoreSend("L"); ListOfMods = "-";
TCPSend("Done", Sock); TCPSend("Done",Sock);
info("Done!"); info("Done!");
return ""; return "";
} }
return Res; return Res;
} }
void UpdateUl(bool D, const std::string& msg) { void UpdateUl(bool D,const std::string& msg){
if (D) if(D)UlStatus = "UlDownloading Resource " + msg;
UlStatus = "UlDownloading Resource " + msg; else UlStatus = "UlLoading Resource " + msg;
else
UlStatus = "UlLoading Resource " + msg;
} }
float DownloadSpeed = 0; void AsyncUpdate(uint64_t& Rcv,uint64_t Size,const std::string& Name){
void AsyncUpdate(uint64_t& Rcv, uint64_t Size, const std::string& Name) {
do { do {
double pr = double(Rcv) / double(Size) * 100; double pr = double(Rcv) / double(Size) * 100;
std::string Per = std::to_string(trunc(pr * 10) / 10); std::string Per = std::to_string(trunc(pr * 10) / 10);
std::string SpeedString = ""; UpdateUl(true, Name + " (" + Per.substr(0, Per.find('.') + 2) + "%)");
if (DownloadSpeed > 0.01) {
std::stringstream ss;
ss << " at " << std::setprecision(1) << std::fixed << DownloadSpeed << " Mbit/s";
SpeedString = ss.str();
}
UpdateUl(true, Name + " (" + Per.substr(0, Per.find('.') + 2) + "%)" + SpeedString);
std::this_thread::sleep_for(std::chrono::milliseconds(100)); std::this_thread::sleep_for(std::chrono::milliseconds(100));
} while (!Terminate && Rcv < Size); }while(!Terminate && Rcv < Size);
} }
// MICROSOFT, I DONT CARE, WRITE BETTER CODE char* TCPRcvRaw(SOCKET Sock,uint64_t& GRcv, uint64_t Size){
#undef min if(Sock == -1){
std::vector<char> TCPRcvRaw(SOCKET Sock, uint64_t& GRcv, uint64_t Size) {
if (Sock == -1) {
Terminate = true; Terminate = true;
UUl("Invalid Socket"); UUl("Invalid Socket");
return {}; return nullptr;
} }
std::vector<char> File(Size); char* File = new char[Size];
uint64_t Rcv = 0; uint64_t Rcv = 0;
do{
auto start = std::chrono::high_resolution_clock::now(); int Len = int(Size-Rcv);
if(Len > 1000000)Len = 1000000;
int i = 0;
do {
// receive at most some MB at a time
int Len = std::min(int(Size - Rcv), 1 * 1024 * 1024);
int32_t Temp = recv(Sock, &File[Rcv], Len, MSG_WAITALL); int32_t Temp = recv(Sock, &File[Rcv], Len, MSG_WAITALL);
if (Temp < 1) { if(Temp < 1){
info(std::to_string(Temp)); info(std::to_string(Temp));
UUl("Socket Closed Code 1"); UUl("Socket Closed Code 1");
KillSocket(Sock); KillSocket(Sock);
Terminate = true; Terminate = true;
return {}; delete[] File;
return nullptr;
} }
Rcv += Temp; Rcv += Temp;
GRcv += Temp; GRcv += Temp;
}while(Rcv < Size && !Terminate);
auto end = std::chrono::high_resolution_clock::now();
auto difference = end - start;
float bits_per_s = float(Rcv * 8) / float(std::chrono::duration_cast<std::chrono::milliseconds>(difference).count());
float megabits_per_s = bits_per_s / 1000;
DownloadSpeed = megabits_per_s;
// every 8th iteration print the speed
if (i % 8 == 0) {
debug("Download speed: " + std::to_string(uint32_t(megabits_per_s)) + "Mbit/s");
}
++i;
} while (Rcv < Size && !Terminate);
return File; return File;
} }
void MultiKill(SOCKET Sock, SOCKET Sock1) { void MultiKill(SOCKET Sock,SOCKET Sock1){
KillSocket(Sock1); KillSocket(Sock1);
KillSocket(Sock); KillSocket(Sock);
Terminate = true; Terminate = true;
} }
SOCKET InitDSock() { SOCKET InitDSock(){
SOCKET DSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); SOCKET DSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
SOCKADDR_IN ServerAddr; SOCKADDR_IN ServerAddr;
if (DSock < 1) { if(DSock < 1){
KillSocket(DSock); KillSocket(DSock);
Terminate = true; Terminate = true;
return 0; return 0;
@@ -205,13 +157,13 @@ SOCKET InitDSock() {
ServerAddr.sin_family = AF_INET; ServerAddr.sin_family = AF_INET;
ServerAddr.sin_port = htons(LastPort); ServerAddr.sin_port = htons(LastPort);
inet_pton(AF_INET, LastIP.c_str(), &ServerAddr.sin_addr); inet_pton(AF_INET, LastIP.c_str(), &ServerAddr.sin_addr);
if (connect(DSock, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)) != 0) { if(connect(DSock, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr)) != 0){
KillSocket(DSock); KillSocket(DSock);
Terminate = true; Terminate = true;
return 0; return 0;
} }
char Code[2] = { 'D', char(ClientID) }; char Code[2] = {'D',char(ClientID)};
if (send(DSock, Code, 2, 0) != 2) { if(send(DSock,Code,2,0) != 2){
KillSocket(DSock); KillSocket(DSock);
Terminate = true; Terminate = true;
return 0; return 0;
@@ -219,379 +171,105 @@ SOCKET InitDSock() {
return DSock; return DSock;
} }
std::vector<char> SingleNormalDownload(SOCKET MSock, uint64_t Size, const std::string& Name) { std::string MultiDownload(SOCKET MSock,SOCKET DSock, uint64_t Size, const std::string& Name){
DownloadSpeed = 0;
uint64_t GRcv = 0; uint64_t GRcv = 0, MSize = Size/2, DSize = Size - MSize;
std::thread Au([&] { AsyncUpdate(GRcv, Size, Name); }); std::thread Au(AsyncUpdate,std::ref(GRcv), Size, Name);
const std::vector<char> MData = TCPRcvRaw(MSock, GRcv, Size); std::packaged_task<char*()> task([&] { return TCPRcvRaw(MSock,GRcv,MSize); });
std::future<char*> f1 = task.get_future();
std::thread Dt(std::move(task));
Dt.detach();
if (MData.empty()) { char* DData = TCPRcvRaw(DSock,GRcv,DSize);
KillSocket(MSock);
Terminate = true; if(!DData){
Au.join(); MultiKill(MSock,DSock);
return {}; return "";
} }
// ensure that GRcv is good before joining the async update thread f1.wait();
GRcv = MData.size(); char* MData = f1.get();
if (GRcv != Size) {
error("Something went wrong during download; didn't get enough data. Expected " + std::to_string(Size) + " bytes, got " + std::to_string(GRcv) + " bytes instead"); if(!MData){
Terminate = true; MultiKill(MSock,DSock);
Au.join(); return "";
return {};
} }
Au.join(); if(Au.joinable())Au.join();
return MData;
///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;
} }
std::vector<char> MultiDownload(SOCKET MSock, SOCKET DSock, uint64_t Size, const std::string& Name) { void InvalidResource(const std::string& File){
DownloadSpeed = 0;
uint64_t GRcv = 0;
uint64_t MSize = Size / 2;
uint64_t DSize = Size - MSize;
std::thread Au([&] { AsyncUpdate(GRcv, Size, Name); });
const std::vector<char> MData = TCPRcvRaw(MSock, GRcv, MSize);
if (MData.empty()) {
MultiKill(MSock, DSock);
Terminate = true;
Au.join();
return {};
}
const std::vector<char> DData = TCPRcvRaw(DSock, GRcv, DSize);
if (DData.empty()) {
MultiKill(MSock, DSock);
Terminate = true;
Au.join();
return {};
}
// ensure that GRcv is good before joining the async update thread
GRcv = MData.size() + DData.size();
if (GRcv != Size) {
error("Something went wrong during download; didn't get enough data. Expected " + std::to_string(Size) + " bytes, got " + std::to_string(GRcv) + " bytes instead");
Terminate = true;
Au.join();
return {};
}
Au.join();
std::vector<char> Result {};
Result.insert(Result.begin(), MData.begin(), MData.end());
Result.insert(Result.end(), DData.begin(), DData.end());
return Result;
}
void InvalidResource(const std::string& File) {
UUl("Invalid mod \"" + File + "\""); UUl("Invalid mod \"" + File + "\"");
warn("The server tried to sync \"" + File + "\" that is not a .zip file!"); warn("The server tried to sync \"" + File + "\" that is not a .zip file!");
Terminate = true; Terminate = true;
} }
std::string GetSha256HashReallyFast(const std::string& filename) { void SyncResources(SOCKET Sock){
try {
EVP_MD_CTX* mdctx;
const EVP_MD* md;
uint8_t sha256_value[EVP_MAX_MD_SIZE];
md = EVP_sha256();
if (md == nullptr) {
throw std::runtime_error("EVP_sha256() failed");
}
mdctx = EVP_MD_CTX_new();
if (mdctx == nullptr) {
throw std::runtime_error("EVP_MD_CTX_new() failed");
}
if (!EVP_DigestInit_ex2(mdctx, md, NULL)) {
EVP_MD_CTX_free(mdctx);
throw std::runtime_error("EVP_DigestInit_ex2() failed");
}
std::ifstream stream(filename, std::ios::binary);
const size_t FileSize = std::filesystem::file_size(filename);
size_t Read = 0;
std::vector<char> Data;
while (Read < FileSize) {
Data.resize(size_t(std::min<size_t>(FileSize - Read, 4096)));
size_t RealDataSize = Data.size();
stream.read(Data.data(), std::streamsize(Data.size()));
if (stream.eof() || stream.fail()) {
RealDataSize = size_t(stream.gcount());
}
Data.resize(RealDataSize);
if (RealDataSize == 0) {
break;
}
if (RealDataSize > 0 && !EVP_DigestUpdate(mdctx, Data.data(), Data.size())) {
EVP_MD_CTX_free(mdctx);
throw std::runtime_error("EVP_DigestUpdate() failed");
}
Read += RealDataSize;
}
unsigned int sha256_len = 0;
if (!EVP_DigestFinal_ex(mdctx, sha256_value, &sha256_len)) {
EVP_MD_CTX_free(mdctx);
throw std::runtime_error("EVP_DigestFinal_ex() failed");
}
EVP_MD_CTX_free(mdctx);
std::string result;
for (size_t i = 0; i < sha256_len; i++) {
char buf[3];
sprintf(buf, "%02x", sha256_value[i]);
buf[2] = 0;
result += buf;
}
return result;
} catch (const std::exception& e) {
error("Sha256 hashing of '" + filename + "' failed: " + e.what());
return "";
}
}
struct ModInfo {
static std::vector<ModInfo> ParseModInfosFromPacket(const std::string& packet) {
std::vector<ModInfo> modInfos;
try {
auto json = nlohmann::json::parse(packet);
for (const auto& entry : json) {
ModInfo modInfo {
.FileName = entry["file_name"],
.FileSize = entry["file_size"],
.Hash = entry["hash"],
.HashAlgorithm = entry["hash_algorithm"],
};
modInfos.push_back(modInfo);
}
} catch (const std::exception& e) {
debug(std::string("Failed to receive mod list: ") + e.what());
warn("Failed to receive new mod list format! This server may be outdated, but everything should still work as expected.");
return {};
}
return modInfos;
}
std::string FileName;
size_t FileSize;
std::string Hash;
std::string HashAlgorithm;
};
void NewSyncResources(SOCKET Sock, const std::string& Mods, const std::vector<ModInfo> ModInfos) {
if (!SecurityWarning())
return;
info("Checking Resources...");
CheckForDir();
std::string t;
for (const auto& mod : ModInfos) {
t += mod.FileName + ";";
}
if (t.empty())
CoreSend("L");
else
CoreSend("L" + t);
t.clear();
info("Syncing...");
int ModNo = 1;
int TotalMods = ModInfos.size();
for (auto ModInfoIter = ModInfos.begin(), AlsoModInfoIter = ModInfos.begin(); ModInfoIter != ModInfos.end() && !Terminate; ++ModInfoIter, ++AlsoModInfoIter) {
if (ModInfoIter->Hash.length() < 8 || ModInfoIter->HashAlgorithm != "sha256") {
error("Unsupported hash algorithm or invalid hash for '" + ModInfoIter->FileName + "'");
Terminate = true;
return;
}
auto FileName = std::filesystem::path(ModInfoIter->FileName).stem().string() + "-" + ModInfoIter->Hash.substr(0, 8) + std::filesystem::path(ModInfoIter->FileName).extension().string();
auto PathToSaveTo = (fs::path(CachingDirectory) / FileName).string();
if (fs::exists(PathToSaveTo) && GetSha256HashReallyFast(PathToSaveTo) == ModInfoIter->Hash) {
debug("Mod '" + FileName + "' found in cache");
std::this_thread::sleep_for(std::chrono::milliseconds(50));
try {
if (!fs::exists(GetGamePath() + "mods/multiplayer")) {
fs::create_directories(GetGamePath() + "mods/multiplayer");
}
auto modname = ModInfoIter->FileName;
#if defined(__linux__)
// Linux version of the game doesnt support uppercase letters in mod names
for (char& c : modname) {
c = ::tolower(c);
}
#endif
debug("Mod name: " + modname);
auto name = std::filesystem::path(GetGamePath()) / "mods/multiplayer" / modname;
std::string tmp_name = name.string();
tmp_name += ".tmp";
fs::copy_file(PathToSaveTo, tmp_name, fs::copy_options::overwrite_existing);
fs::rename(tmp_name, name);
} catch (std::exception& e) {
error("Failed copy to the mods folder! " + std::string(e.what()));
Terminate = true;
continue;
}
WaitForConfirm();
continue;
}
CheckForDir();
std::string FName = ModInfoIter->FileName;
do {
debug("Loading file '" + FName + "' to '" + PathToSaveTo + "'");
TCPSend("f" + ModInfoIter->FileName, Sock);
std::string Data = TCPRcv(Sock);
if (Data == "CO" || Terminate) {
Terminate = true;
UUl("Server cannot find " + FName);
break;
}
std::string Name = std::to_string(ModNo) + "/" + std::to_string(TotalMods) + ": " + FName;
std::vector<char> DownloadedFile = SingleNormalDownload(Sock, ModInfoIter->FileSize, Name);
if (Terminate)
break;
UpdateUl(false, std::to_string(ModNo) + "/" + std::to_string(TotalMods) + ": " + FName);
// 1. write downloaded file to disk
{
std::ofstream OutFile(PathToSaveTo, std::ios::binary | std::ios::trunc);
OutFile.write(DownloadedFile.data(), DownloadedFile.size());
OutFile.flush();
}
// 2. verify size
if (std::filesystem::file_size(PathToSaveTo) != DownloadedFile.size()) {
error("Failed to write the entire file '" + PathToSaveTo + "' correctly (file size mismatch)");
Terminate = true;
}
} while (fs::file_size(PathToSaveTo) != ModInfoIter->FileSize && !Terminate);
if (!Terminate) {
if (!fs::exists(GetGamePath() + "mods/multiplayer")) {
fs::create_directories(GetGamePath() + "mods/multiplayer");
}
// Linux version of the game doesnt support uppercase letters in mod names
#if defined(__linux__)
for (char& c : FName) {
c = ::tolower(c);
}
#endif
fs::copy_file(PathToSaveTo, std::filesystem::path(GetGamePath()) / "mods/multiplayer" / FName, fs::copy_options::overwrite_existing);
}
WaitForConfirm();
++ModNo;
}
if (!Terminate) {
TCPSend("Done", Sock);
info("Done!");
} else {
UlStatus = "Ulstart";
info("Connection Terminated!");
}
}
void SyncResources(SOCKET Sock) {
std::string Ret = Auth(Sock); std::string Ret = Auth(Sock);
if(Ret.empty())return;
if (Ret.starts_with("R")) {
debug("This server is likely outdated, not trying to parse new mod info format");
} else {
auto ModInfos = ModInfo::ParseModInfosFromPacket(Ret);
if (!ModInfos.empty()) {
NewSyncResources(Sock, Ret, ModInfos);
return;
}
}
if (Ret.empty())
return;
if (!SecurityWarning())
return;
info("Checking Resources..."); info("Checking Resources...");
CheckForDir(); CheckForDir();
std::vector<std::string> list = Utils::Split(Ret, ";"); std::vector<std::string> list = Split(Ret, ";");
std::vector<std::string> FNames(list.begin(), list.begin() + (list.size() / 2)); std::vector<std::string> FNames(list.begin(), list.begin() + (list.size() / 2));
std::vector<std::string> FSizes(list.begin() + (list.size() / 2), list.end()); std::vector<std::string> FSizes(list.begin() + (list.size() / 2), list.end());
list.clear(); list.clear();
Ret.clear(); Ret.clear();
int Amount = 0, Pos = 0; int Amount = 0,Pos = 0;
std::string PathToSaveTo, t; std::string a,t;
for (const std::string& name : FNames) { for(const std::string&name : FNames){
if (!name.empty()) { if(!name.empty()){
t += name.substr(name.find_last_of('/') + 1) + ";"; t += name.substr(name.find_last_of('/') + 1) + ";";
} }
} }
if (t.empty()) if(t.empty())ListOfMods = "-";
CoreSend("L"); else ListOfMods = t;
else
CoreSend("L" + t);
t.clear(); t.clear();
for (auto FN = FNames.begin(), FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN, ++FS) { for(auto FN = FNames.begin(),FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN,++FS) {
auto pos = FN->find_last_of('/'); auto pos = FN->find_last_of('/');
auto ZIP = FN->find(".zip"); auto ZIP = FN->find(".zip");
if (ZIP == std::string::npos || FN->length() - ZIP != 4) { if (ZIP == std::string::npos || FN->length() - ZIP != 4) {
InvalidResource(*FN); InvalidResource(*FN);
return; return;
} }
if (pos == std::string::npos) if (pos == std::string::npos)continue;
continue;
Amount++; Amount++;
} }
if (!FNames.empty()) if(!FNames.empty())info("Syncing...");
info("Syncing...");
SOCKET DSock = InitDSock(); SOCKET DSock = InitDSock();
for (auto FN = FNames.begin(), FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN, ++FS) { for(auto FN = FNames.begin(),FS = FSizes.begin(); FN != FNames.end() && !Terminate; ++FN,++FS) {
auto pos = FN->find_last_of('/'); auto pos = FN->find_last_of('/');
if (pos != std::string::npos) { if (pos != std::string::npos) {
PathToSaveTo = CachingDirectory + FN->substr(pos); a = "Resources" + FN->substr(pos);
} else { } else continue;
continue;
}
Pos++; Pos++;
auto FileSize = std::stoull(*FS); if (fs::exists(a)) {
if (fs::exists(PathToSaveTo)) { if (FS->find_first_not_of("0123456789") != std::string::npos)continue;
if (FS->find_first_not_of("0123456789") != std::string::npos) if (fs::file_size(a) == std::stoull(*FS)){
continue; UpdateUl(false,std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + a.substr(a.find_last_of('/')));
if (fs::file_size(PathToSaveTo) == FileSize) {
UpdateUl(false, std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + PathToSaveTo.substr(PathToSaveTo.find_last_of('/')));
std::this_thread::sleep_for(std::chrono::milliseconds(50)); std::this_thread::sleep_for(std::chrono::milliseconds(50));
try { try {
if (!fs::exists(GetGamePath() + "mods/multiplayer")) { if(!fs::exists(GetGamePath() + "mods/multiplayer")){
fs::create_directories(GetGamePath() + "mods/multiplayer"); fs::create_directories(GetGamePath() + "mods/multiplayer");
} }
auto modname = PathToSaveTo.substr(PathToSaveTo.find_last_of('/')); auto name = GetGamePath() + "mods/multiplayer" + a.substr(a.find_last_of('/'));
#if defined(__linux__)
// Linux version of the game doesnt support uppercase letters in mod names
for (char& c : modname) {
c = ::tolower(c);
}
#endif
auto name = GetGamePath() + "mods/multiplayer" + modname;
auto tmp_name = name + ".tmp"; auto tmp_name = name + ".tmp";
fs::copy_file(PathToSaveTo, tmp_name, fs::copy_options::overwrite_existing); fs::copy_file(a,tmp_name,fs::copy_options::overwrite_existing);
fs::rename(tmp_name, name); fs::rename(tmp_name, name);
} catch (std::exception& e) { } catch (std::exception& e) {
error("Failed copy to the mods folder! " + std::string(e.what())); error("Failed copy to the mods folder! " + std::string(e.what()));
@@ -600,17 +278,15 @@ void SyncResources(SOCKET Sock) {
} }
WaitForConfirm(); WaitForConfirm();
continue; continue;
} else }else remove(a.c_str());
remove(PathToSaveTo.c_str());
} }
CheckForDir(); CheckForDir();
std::string FName = PathToSaveTo.substr(PathToSaveTo.find_last_of('/')); std::string FName = a.substr(a.find_last_of('/'));
do { do {
debug("Loading file '" + FName + "' to '" + PathToSaveTo + "'"); TCPSend("f" + *FN,Sock);
TCPSend("f" + *FN, Sock);
std::string Data = TCPRcv(Sock); std::string Data = TCPRcv(Sock);
if (Data == "CO" || Terminate) { if (Data == "CO" || Terminate){
Terminate = true; Terminate = true;
UUl("Server cannot find " + FName); UUl("Server cannot find " + FName);
break; break;
@@ -618,46 +294,32 @@ void SyncResources(SOCKET Sock) {
std::string Name = std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + FName; std::string Name = std::to_string(Pos) + "/" + std::to_string(Amount) + ": " + FName;
std::vector<char> DownloadedFile = MultiDownload(Sock, DSock, FileSize, Name); Data = MultiDownload(Sock,DSock,std::stoull(*FS), Name);
if (Terminate) if(Terminate)break;
break; UpdateUl(false,std::to_string(Pos)+"/"+std::to_string(Amount)+": "+FName);
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], Data.size());
LFS.close();
}
// 1. write downloaded file to disk }while(fs::file_size(a) != std::stoull(*FS) && !Terminate);
{ if(!Terminate){
std::ofstream OutFile(PathToSaveTo, std::ios::binary | std::ios::trunc); if(!fs::exists(GetGamePath() + "mods/multiplayer")){
OutFile.write(DownloadedFile.data(), DownloadedFile.size());
}
// 2. verify size
if (std::filesystem::file_size(PathToSaveTo) != DownloadedFile.size()) {
error("Failed to write the entire file '" + PathToSaveTo + "' correctly (file size mismatch)");
Terminate = true;
}
} while (fs::file_size(PathToSaveTo) != std::stoull(*FS) && !Terminate);
if (!Terminate) {
if (!fs::exists(GetGamePath() + "mods/multiplayer")) {
fs::create_directories(GetGamePath() + "mods/multiplayer"); fs::create_directories(GetGamePath() + "mods/multiplayer");
} }
fs::copy_file(a,GetGamePath() + "mods/multiplayer" + FName, fs::copy_options::overwrite_existing);
// Linux version of the game doesnt support uppercase letters in mod names
#if defined(__linux__)
for (char& c : FName) {
c = ::tolower(c);
}
#endif
fs::copy_file(PathToSaveTo, GetGamePath() + "mods/multiplayer" + FName, fs::copy_options::overwrite_existing);
} }
WaitForConfirm(); WaitForConfirm();
} }
KillSocket(DSock); KillSocket(DSock);
if (!Terminate) { if(!Terminate){
TCPSend("Done", Sock); TCPSend("Done",Sock);
info("Done!"); info("Done!");
} else { }else{
UlStatus = "Ulstart"; UlStatus = "Ulstart";
info("Connection Terminated!"); info("Connection Terminated!");
} }
} }

98
src/Network/VehicleData.cpp Normal file → Executable file
View File

@@ -5,103 +5,69 @@
/// ///
/// Created by Anonymous275 on 5/8/2020 /// Created by Anonymous275 on 5/8/2020
/// ///
#include "Network/network.hpp"
#include "Zlib/Compressor.h" #include "Zlib/Compressor.h"
#include <stdexcept> #include "Network/network.h"
#if defined(_WIN32)
#include <ws2tcpip.h> #include <ws2tcpip.h>
#elif defined(__linux__)
#include "linuxfixes.h"
#include <arpa/inet.h>
#include <cstring>
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
#endif
#include "Logger.h" #include "Logger.h"
#include <array>
#include <string> #include <string>
#include <set>
SOCKET UDPSock = -1; SOCKET UDPSock = -1;
sockaddr_in* ToServer = nullptr; sockaddr_in* ToServer = nullptr;
void UDPSend(std::string Data) { void UDPSend(std::string Data){
if (ClientID == -1 || UDPSock == -1) if(ClientID == -1 || UDPSock == -1)return;
return; if(Data.length() > 400){
if (Data.length() > 400) { std::string CMP(Comp(Data));
auto res = Comp(std::span<char>(Data.data(), Data.size())); Data = "ABG:" + CMP;
Data = "ABG:" + std::string(res.data(), res.size());
} }
std::string Packet = char(ClientID + 1) + std::string(":") + Data; std::string Packet = char(ClientID+1) + std::string(":") + Data;
int sendOk = sendto(UDPSock, Packet.c_str(), int(Packet.size()), 0, (sockaddr*)ToServer, sizeof(*ToServer)); int sendOk = sendto(UDPSock, Packet.c_str(), int(Packet.size()), 0, (sockaddr*)ToServer, sizeof(*ToServer));
if (sendOk == SOCKET_ERROR) if (sendOk == SOCKET_ERROR)error("Error Code : " + std::to_string(WSAGetLastError()));
error("Error Code : " + std::to_string(WSAGetLastError()));
} }
void SendLarge(std::string Data) {
if (Data.length() > 400) { void SendLarge(std::string Data){
auto res = Comp(std::span<char>(Data.data(), Data.size())); if(Data.length() > 400){
Data = "ABG:" + std::string(res.data(), res.size()); std::string CMP(Comp(Data));
Data = "ABG:" + CMP;
} }
TCPSend(Data, TCPSock); TCPSend(Data,TCPSock);
} }
void UDPParser(std::string_view Packet) { void UDPParser(std::string Packet){
if (Packet.substr(0, 4) == "ABG:") { if(Packet.substr(0,4) == "ABG:"){
auto substr = Packet.substr(4); Packet = DeComp(Packet.substr(4));
try {
auto res = DeComp(std::span<const char>(substr.data(), substr.size()));
std::string DeCompPacket = std::string(res.data(), res.size());
ServerParser(DeCompPacket);
} catch (const std::runtime_error& err) {
error("Error in decompression of UDP, ignoring");
}
} else {
ServerParser(Packet);
} }
ServerParser(Packet);
} }
void UDPRcv() { void UDPRcv(){
sockaddr_in FromServer {}; sockaddr_in FromServer{};
#if defined(_WIN32)
int clientLength = sizeof(FromServer); int clientLength = sizeof(FromServer);
#elif defined(__linux__)
socklen_t clientLength = sizeof(FromServer);
#endif
ZeroMemory(&FromServer, clientLength); ZeroMemory(&FromServer, clientLength);
static thread_local std::array<char, 10240> Ret {}; std::string Ret(10240,0);
if (UDPSock == -1) if(UDPSock == -1)return;
return; int32_t Rcv = recvfrom(UDPSock, &Ret[0], 10240, 0, (sockaddr*)&FromServer, &clientLength);
int32_t Rcv = recvfrom(UDPSock, Ret.data(), Ret.size() - 1, 0, (sockaddr*)&FromServer, &clientLength); if (Rcv == SOCKET_ERROR)return;
if (Rcv == SOCKET_ERROR) UDPParser(Ret.substr(0,Rcv));
return;
Ret[Rcv] = 0;
UDPParser(std::string_view(Ret.data(), Rcv));
} }
void UDPClientMain(const std::string& IP, int Port) { void UDPClientMain(const std::string& IP,int Port){
#ifdef _WIN32
WSADATA data; WSADATA data;
if (WSAStartup(514, &data)) { if (WSAStartup(514, &data)){
error("Can't start Winsock!"); error("Can't start Winsock!");
return; return;
} }
#endif
delete ToServer; delete ToServer;
ToServer = new sockaddr_in; ToServer = new sockaddr_in;
ToServer->sin_family = AF_INET; ToServer->sin_family = AF_INET;
ToServer->sin_port = htons(Port); ToServer->sin_port = htons(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);
GameSend("P" + std::to_string(ClientID)); GameSend("P"+std::to_string(ClientID));
TCPSend("H", TCPSock); TCPSend("H",TCPSock);
UDPSend("p"); UDPSend("p");
debug("Starting UDP receive loop"); while(!Terminate)UDPRcv();
while (!Terminate) {
UDPRcv();
}
debug("UDP receive loop done");
KillSocket(UDPSock); KillSocket(UDPSock);
WSACleanup(); WSACleanup();
} }

164
src/Network/VehicleEvent.cpp Normal file → Executable file
View File

@@ -6,35 +6,26 @@
/// Created by Anonymous275 on 5/8/2020 /// Created by Anonymous275 on 5/8/2020
/// ///
#include "Logger.h"
#include <Zlib/Compressor.h>
#include <chrono> #include <chrono>
#include <iostream>
#include <vector> #include <vector>
#include "Logger.h"
#if defined(_WIN32) #include <iostream>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#elif defined(__linux__) #include <Zlib/Compressor.h>
#include <arpa/inet.h>
#include <cstring> #include "Network/network.h"
#include <errno.h>
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
#endif
#include "Network/network.hpp"
int LastPort; int LastPort;
std::string LastIP; std::string LastIP;
SOCKET TCPSock = -1; SOCKET TCPSock = -1;
bool CheckBytes(int32_t Bytes) { bool CheckBytes(int32_t Bytes){
if (Bytes == 0) { if (Bytes == 0){
debug("(TCP) Connection closing... CheckBytes(16)"); debug("(TCP) Connection closing... CheckBytes(16)");
Terminate = true; Terminate = true;
return false; return false;
} else if (Bytes < 0) { }else if (Bytes < 0) {
debug("(TCP CB) recv failed with error: " + std::to_string(WSAGetLastError())); debug("(TCP CB) recv failed with error: " + std::to_string(WSAGetLastError()));
KillSocket(TCPSock); KillSocket(TCPSock);
Terminate = true; Terminate = true;
@@ -42,117 +33,110 @@ bool CheckBytes(int32_t Bytes) {
} }
return true; return true;
} }
void UUl(const std::string& R) { void UUl(const std::string& R){
UlStatus = "UlDisconnected: " + R; UlStatus = "UlDisconnected: " + R;
} }
void TCPSend(const std::string& Data, uint64_t Sock) { void TCPSend(const std::string&Data,uint64_t Sock){
if (Sock == -1) { if(Sock == -1){
Terminate = true; Terminate = true;
UUl("Invalid Socket"); UUl("Invalid Socket");
return; return;
} }
int32_t Size, Sent, Temp; int32_t Size,Sent,Temp;
std::string Send(4, 0); std::string Send(4,0);
Size = int32_t(Data.size()); Size = int32_t(Data.size());
memcpy(&Send[0], &Size, sizeof(Size)); memcpy(&Send[0],&Size,sizeof(Size));
Send += Data; Send += Data;
// Do not use Size before this point for anything but the header // Do not use Size before this point for anything but the header
Sent = 0; Sent = 0;
Size += 4; Size += 4;
do { do{
if (size_t(Sent) >= Send.size()) { if (size_t(Sent) >= Send.size()) {
error("string OOB in " + std::string(__func__)); error("string OOB in " + std::string(__func__));
UUl("TCP Send OOB"); UUl("TCP Send OOB");
return; return;
} }
Temp = send(Sock, &Send[Sent], Size - Sent, 0); Temp = send(Sock, &Send[Sent], Size - Sent, 0);
if (!CheckBytes(Temp)) { if(!CheckBytes(Temp)){
UUl("Socket Closed Code 2"); UUl("Socket Closed Code 2");
return; return;
} }
Sent += Temp; Sent += Temp;
} while (Sent < Size); }while(Sent < Size);
} }
std::string TCPRcv(SOCKET Sock) { std::string TCPRcv(SOCKET Sock){
if (Sock == -1) { if(Sock == -1){
Terminate = true; Terminate = true;
UUl("Invalid Socket"); UUl("Invalid Socket");
return ""; return "";
} }
int32_t Header, Temp; int32_t Header,BytesRcv = 0,Temp;
std::vector<char> Data(sizeof(Header)); std::vector<char> Data(sizeof(Header));
Temp = recv(Sock, Data.data(), sizeof(Header), MSG_WAITALL); do{
if (!CheckBytes(Temp)) { Temp = recv(Sock,&Data[BytesRcv],4-BytesRcv,0);
UUl("Socket Closed Code 3"); if(!CheckBytes(Temp)){
return ""; UUl("Socket Closed Code 3");
} return "";
memcpy(&Header, Data.data(), sizeof(Header)); }
BytesRcv += Temp;
}while(BytesRcv < 4);
memcpy(&Header,&Data[0],sizeof(Header));
if (!CheckBytes(Temp)) { if(!CheckBytes(BytesRcv)){
UUl("Socket Closed Code 4"); UUl("Socket Closed Code 4");
return ""; return "";
} }
Data.resize(Header);
Data.resize(Header, 0); BytesRcv = 0;
Temp = recv(Sock, Data.data(), Header, MSG_WAITALL); do{
if (!CheckBytes(Temp)) { Temp = recv(Sock,&Data[BytesRcv],Header-BytesRcv,0);
UUl("Socket Closed Code 5"); if(!CheckBytes(Temp)){
return ""; UUl("Socket Closed Code 5");
}
std::string Ret(Data.data(), Header);
if (Ret.substr(0, 4) == "ABG:") {
auto substr = Ret.substr(4);
try {
auto res = DeComp(std::span<char>(substr.data(), substr.size()));
Ret = std::string(res.data(), res.size());
} catch (const std::runtime_error& err) {
// this happens e.g. when we're out of memory, or when we get incomplete data
error("Decompression failed");
return ""; return "";
} }
BytesRcv += Temp;
}while(BytesRcv < Header);
std::string Ret(Data.data(),Header);
if (Ret.substr(0, 4) == "ABG:") {
Ret = DeComp(Ret.substr(4));
} }
#ifdef DEBUG #ifdef DEBUG
// debug("Parsing from server -> " + std::to_string(Ret.size())); //debug("Parsing from server -> " + std::to_string(Ret.size()));
#endif #endif
if (Ret[0] == 'E' || Ret[0] == 'K') if(Ret[0] == 'E' || Ret[0] == 'K')UUl(Ret.substr(1));
UUl(Ret.substr(1));
return Ret; return Ret;
} }
void TCPClientMain(const std::string& IP, int Port) { void TCPClientMain(const std::string& IP,int Port){
LastIP = IP; LastIP = IP;
LastPort = Port; LastPort = Port;
WSADATA wsaData;
SOCKADDR_IN ServerAddr; SOCKADDR_IN ServerAddr;
int RetCode; int RetCode;
#ifdef _WIN32 WSAStartup(514, &wsaData); //2.2
WSADATA wsaData;
WSAStartup(514, &wsaData); // 2.2
#endif
TCPSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); TCPSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (TCPSock == -1) { if(TCPSock == -1){
printf("Client: socket failed! Error code: %d\n", WSAGetLastError()); printf("Client: socket failed! Error code: %d\n", WSAGetLastError());
WSACleanup(); WSACleanup();
return; return;
} }
ServerAddr.sin_family = AF_INET; ServerAddr.sin_family = AF_INET;
ServerAddr.sin_port = htons(Port); 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);
RetCode = connect(TCPSock, (SOCKADDR*)&ServerAddr, sizeof(ServerAddr)); RetCode = connect(TCPSock, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr));
if (RetCode != 0) { if(RetCode != 0){
UlStatus = "UlConnection Failed!"; UlStatus = "UlConnection Failed!";
error("Client: connect failed! Error code: " + std::to_string(WSAGetLastError())); error("Client: connect failed! Error code: " + std::to_string(WSAGetLastError()));
KillSocket(TCPSock); KillSocket(TCPSock);
WSACleanup(); WSACleanup();
Terminate = true; Terminate = true;
CoreSend("L");
return; return;
} }
info("Connected!"); info("Connected!");
@@ -160,16 +144,14 @@ void TCPClientMain(const std::string& IP, int Port) {
char Code = 'C'; char Code = 'C';
send(TCPSock, &Code, 1, 0); send(TCPSock, &Code, 1, 0);
SyncResources(TCPSock); SyncResources(TCPSock);
while (!Terminate) { while(!Terminate){
ServerParser(TCPRcv(TCPSock)); ServerParser(TCPRcv(TCPSock));
} }
GameSend("T"); GameSend("T");
////Game Send Terminate ////Game Send Terminate
if (KillSocket(TCPSock) != 0) if(KillSocket(TCPSock) != 0)
debug("(TCP) Cannot close socket. Error code: " + std::to_string(WSAGetLastError())); debug("(TCP) Cannot close socket. Error code: " + std::to_string(WSAGetLastError()));
#ifdef _WIN32 if(WSACleanup() != 0)
if (WSACleanup() != 0)
debug("(TCP) Client: WSACleanup() failed!..."); debug("(TCP) Client: WSACleanup() failed!...");
#endif
} }

354
src/Security/BeamNG.cpp Normal file → Executable file
View File

@@ -7,16 +7,10 @@
/// ///
#include <filesystem> #include <filesystem>
#if defined(_WIN32)
#include <windows.h> #include <windows.h>
#elif defined(__linux__)
#include "vdf_parser.hpp"
#include <pwd.h>
#include <unistd.h>
#include <vector>
#endif
#include "Logger.h" #include "Logger.h"
#include <fstream> #include <fstream>
#include <sstream>
#include <string> #include <string>
#include <thread> #include <thread>
@@ -26,189 +20,299 @@
int TraceBack = 0; int TraceBack = 0;
std::string GameDir; std::string GameDir;
void lowExit(int code) { void lowExit(int code){
TraceBack = 0; TraceBack = 0;
std::string msg = "Failed to find the game please launch it. Report this if the issue persists code "; std::string msg =
error(msg + std::to_string(code)); "Failed to find the game please launch it. Report this if the issue persists code ";
error(msg+std::to_string(code));
std::this_thread::sleep_for(std::chrono::seconds(10)); std::this_thread::sleep_for(std::chrono::seconds(10));
exit(2); exit(2);
} }
/*void Exit(int code){
std::string GetGameDir() { TraceBack = 0;
#if defined(_WIN32) std::string msg =
return GameDir.substr(0, GameDir.find_last_of('\\')); "Sorry. We do not support cracked copies report this if you believe this is a mistake code ";
#elif defined(__linux__) error(msg+std::to_string(code));
return GameDir.substr(0, GameDir.find_last_of('/')); std::this_thread::sleep_for(std::chrono::seconds(10));
#endif exit(3);
} }
#ifdef _WIN32 void SteamExit(int code){
LONG OpenKey(HKEY root, const char* path, PHKEY hKey) { TraceBack = 0;
std::string msg =
"Illegal steam modifications detected report this if you believe this is a mistake code ";
error(msg+std::to_string(code));
std::this_thread::sleep_for(std::chrono::seconds(10));
exit(4);
}*/
std::string GetGameDir(){
//if(TraceBack != 4)Exit(0);
return GameDir.substr(0,GameDir.find_last_of('\\'));
}
LONG OpenKey(HKEY root,const char* path,PHKEY hKey){
return RegOpenKeyEx(root, reinterpret_cast<LPCSTR>(path), 0, KEY_READ, hKey); return RegOpenKeyEx(root, reinterpret_cast<LPCSTR>(path), 0, KEY_READ, hKey);
} }
std::string QueryKey(HKEY hKey, int ID) { std::string QueryKey(HKEY hKey,int ID){
TCHAR achKey[MAX_KEY_LENGTH]; // buffer for subkey name TCHAR achKey[MAX_KEY_LENGTH]; // buffer for subkey name
DWORD cbName; // size of name string DWORD cbName; // size of name string
TCHAR achClass[MAX_PATH] = TEXT(""); // buffer for class name TCHAR achClass[MAX_PATH] = TEXT(""); // buffer for class name
DWORD cchClassName = MAX_PATH; // size of class string DWORD cchClassName = MAX_PATH; // size of class string
DWORD cSubKeys = 0; // number of subkeys DWORD cSubKeys=0; // number of subkeys
DWORD cbMaxSubKey; // longest subkey size DWORD cbMaxSubKey; // longest subkey size
DWORD cchMaxClass; // longest class string DWORD cchMaxClass; // longest class string
DWORD cValues; // number of values for key DWORD cValues; // number of values for key
DWORD cchMaxValue; // longest value name DWORD cchMaxValue; // longest value name
DWORD cbMaxValueData; // longest value data DWORD cbMaxValueData; // longest value data
DWORD cbSecurityDescriptor; // size of security descriptor DWORD cbSecurityDescriptor; // size of security descriptor
FILETIME ftLastWriteTime; // last write time FILETIME ftLastWriteTime; // last write time
DWORD i, retCode; DWORD i, retCode;
TCHAR achValue[MAX_VALUE_NAME]; TCHAR achValue[MAX_VALUE_NAME];
DWORD cchValue = MAX_VALUE_NAME; DWORD cchValue = MAX_VALUE_NAME;
retCode = RegQueryInfoKey( retCode = RegQueryInfoKey(
hKey, // key handle hKey, // key handle
achClass, // buffer for class name achClass, // buffer for class name
&cchClassName, // size of class string &cchClassName, // size of class string
nullptr, // reserved nullptr, // reserved
&cSubKeys, // number of subkeys &cSubKeys, // number of subkeys
&cbMaxSubKey, // longest subkey size &cbMaxSubKey, // longest subkey size
&cchMaxClass, // longest class string &cchMaxClass, // longest class string
&cValues, // number of values for this key &cValues, // number of values for this key
&cchMaxValue, // longest value name &cchMaxValue, // longest value name
&cbMaxValueData, // longest value data &cbMaxValueData, // longest value data
&cbSecurityDescriptor, // security descriptor &cbSecurityDescriptor, // security descriptor
&ftLastWriteTime); // last write time &ftLastWriteTime); // last write time
BYTE* buffer = new BYTE[cbMaxValueData]; BYTE* buffer = new BYTE[cbMaxValueData];
ZeroMemory(buffer, cbMaxValueData); ZeroMemory(buffer, cbMaxValueData);
if (cSubKeys) { if (cSubKeys){
for (i = 0; i < cSubKeys; i++) { for (i=0; i<cSubKeys; i++){
cbName = MAX_KEY_LENGTH; cbName = MAX_KEY_LENGTH;
retCode = RegEnumKeyEx(hKey, i, achKey, &cbName, nullptr, nullptr, nullptr, &ftLastWriteTime); retCode = RegEnumKeyEx(hKey, i,achKey,&cbName,nullptr,nullptr,nullptr,&ftLastWriteTime);
if (retCode == ERROR_SUCCESS) { if (retCode == ERROR_SUCCESS){
if (strcmp(achKey, "Steam App 284160") == 0) { if(strcmp(achKey,"Steam App 284160") == 0){
return achKey; return achKey;
} }
} }
} }
} }
if (cValues) { if (cValues){
for (i = 0, retCode = ERROR_SUCCESS; i < cValues; i++) { for (i=0, retCode = ERROR_SUCCESS; i<cValues; i++){
cchValue = MAX_VALUE_NAME; cchValue = MAX_VALUE_NAME;
achValue[0] = '\0'; achValue[0] = '\0';
retCode = RegEnumValue(hKey, i, achValue, &cchValue, nullptr, nullptr, nullptr, nullptr); retCode = RegEnumValue(hKey, i,achValue,&cchValue,nullptr,nullptr,nullptr,nullptr);
if (retCode == ERROR_SUCCESS) { if (retCode == ERROR_SUCCESS ){
DWORD lpData = cbMaxValueData; DWORD lpData = cbMaxValueData;
buffer[0] = '\0'; buffer[0] = '\0';
LONG dwRes = RegQueryValueEx(hKey, achValue, nullptr, nullptr, buffer, &lpData); LONG dwRes = RegQueryValueEx(hKey, achValue, nullptr, nullptr, buffer, &lpData);
std::string data = (char*)(buffer); std::string data = (char *)(buffer);
std::string key = achValue; std::string key = achValue;
switch (ID) { switch (ID){
case 1: case 1: if(key == "SteamExe"){
if (key == "SteamExe") { auto p = data.find_last_of("/\\");
auto p = data.find_last_of("/\\"); if(p != std::string::npos){
if (p != std::string::npos) { return data.substr(0,p);
return data.substr(0, p); }
} }
} break;
break; case 2: if(key == "Name" && data == "BeamNG.drive")return data;break;
case 2: case 3: if(key == "rootpath")return data;break;
if (key == "Name" && data == "BeamNG.drive") case 4: if(key == "userpath_override")return data;
return data; case 5: if(key == "Local AppData")return data;
break; default: break;
case 3:
if (key == "rootpath")
return data;
break;
case 4:
if (key == "userpath_override")
return data;
case 5:
if (key == "Local AppData")
return data;
default:
break;
} }
} }
} }
} }
delete[] buffer; delete [] buffer;
return ""; return "";
} }
#endif
namespace fs = std::filesystem; namespace fs = std::filesystem;
bool NameValid(const std::string& N) { bool NameValid(const std::string& N){
if (N == "config" || N == "librarycache") { if(N == "config" || N == "librarycache"){
return true; return true;
} }
if (N.find_first_not_of("0123456789") == std::string::npos) { if(N.find_first_not_of("0123456789") == std::string::npos){
return true; return true;
} }
return false; return false;
} }
void FileList(std::vector<std::string>& a, const std::string& Path) { void FileList(std::vector<std::string>&a,const std::string& Path){
for (const auto& entry : fs::directory_iterator(Path)) { for (const auto &entry : fs::directory_iterator(Path)) {
const auto& DPath = entry.path(); const auto& DPath = entry.path();
if (!entry.is_directory()) { if (!entry.is_directory()) {
a.emplace_back(DPath.string()); a.emplace_back(DPath.string());
} else if (NameValid(DPath.filename().string())) { }else if(NameValid(DPath.filename().string())){
FileList(a, DPath.string()); FileList(a, DPath.string());
} }
} }
} }
void LegitimacyCheck() { bool Find(const std::string& FName,const std::string& Path){
#if defined(_WIN32) std::vector<std::string> FS;
FileList(FS,Path+"\\userdata");
for(std::string&a : FS){
if(a.find(FName) != std::string::npos){
FS.clear();
return true;
}
}
FS.clear();
return false;
}
bool FindHack(const std::string& Path){
bool s = true;
for (const auto &entry : fs::directory_iterator(Path)) {
std::string Name = entry.path().filename().string();
for(char&c : Name)c = char(tolower(c));
if(Name == "steam.exe")s = false;
if(Name.find("greenluma") != -1){
error("Found malicious file/folder \"" + Name+"\"");
return true;
}
Name.clear();
}
return s;
}
std::vector<std::string> GetID(const std::string& log){
std::string vec,t,r;
std::vector<std::string> Ret;
std::ifstream f(log.c_str(), std::ios::binary);
f.seekg(0, std::ios_base::end);
std::streampos fileSize = f.tellg();
vec.resize(size_t(fileSize) + 1);
f.seekg(0, std::ios_base::beg);
f.read(&vec[0], fileSize);
f.close();
std::stringstream ss(vec);
bool S = false;
while (std::getline(ss, t, '{')) {
if(!S)S = true;
else{
for(char& c : t){
if(isdigit(c))r += c;
}
break;
}
}
Ret.emplace_back(r);
r.clear();
S = false;
bool L = true;
while (std::getline(ss, t, '}')) {
if(L){
L = false;
continue;
}
for(char& c : t){
if(c == '"'){
if(!S)S = true;
else{
if(r.length() > 10) {
Ret.emplace_back(r);
}
r.clear();
S = false;
continue;
}
}
if(isdigit(c))r += c;
}
}
vec.clear();
return Ret;
}
std::string GetManifest(const std::string& Man){
std::string vec;
std::ifstream f(Man.c_str(), std::ios::binary);
f.seekg(0, std::ios_base::end);
std::streampos fileSize = f.tellg();
vec.resize(size_t(fileSize) + 1);
f.seekg(0, std::ios_base::beg);
f.read(&vec[0], fileSize);
f.close();
std::string ToFind = "\"LastOwner\"\t\t\"";
int pos = int(vec.find(ToFind));
if(pos != -1){
pos += int(ToFind.length());
vec = vec.substr(pos);
return vec.substr(0,vec.find('\"'));
}else return "";
}
bool IDCheck(std::string Man, std::string steam){
bool a = false,b = true;
int pos = int(Man.rfind("steamapps"));
// if(pos == -1)Exit(5);
Man = Man.substr(0,pos+9) + "\\appmanifest_284160.acf";
steam += "\\config\\loginusers.vdf";
if(fs::exists(Man) && fs::exists(steam)){
for(const std::string&ID : GetID(steam)){
if(ID == GetManifest(Man))b = false;
}
//if(b)Exit(6);
}else a = true;
return a;
}
void LegitimacyCheck(){
//std::string K1 = R"(Software\Valve\Steam)";
//std::string K2 = R"(Software\Valve\Steam\Apps\284160)";
/*LONG dwRegOPenKey = OpenKey(HKEY_CURRENT_USER, K1.c_str(), &hKey);
if(dwRegOPenKey == ERROR_SUCCESS) {
Result = QueryKey(hKey, 1);
if(Result.empty())Exit(1);
if(fs::exists(Result)){
if(!Find("284160.json",Result))Exit(2);
if(FindHack(Result))SteamExit(1);
}else Exit(3);
T = Result;
Result.clear();
TraceBack++;
}else Exit(4);
K1.clear();
RegCloseKey(hKey);
dwRegOPenKey = OpenKey(HKEY_CURRENT_USER, K2.c_str(), &hKey);
if(dwRegOPenKey == ERROR_SUCCESS) {
Result = QueryKey(hKey, 2);
if(Result.empty())lowExit(1);
TraceBack++;
}else lowExit(2);
K2.clear();
RegCloseKey(hKey);*/
std::string Result; std::string Result;
std::string K3 = R"(Software\BeamNG\BeamNG.drive)"; std::string K3 = R"(Software\BeamNG\BeamNG.drive)";
HKEY hKey; HKEY hKey;
LONG dwRegOPenKey = OpenKey(HKEY_CURRENT_USER, K3.c_str(), &hKey); LONG dwRegOPenKey = OpenKey(HKEY_CURRENT_USER, K3.c_str(), &hKey);
if (dwRegOPenKey == ERROR_SUCCESS) { if(dwRegOPenKey == ERROR_SUCCESS) {
Result = QueryKey(hKey, 3); Result = QueryKey(hKey, 3);
if (Result.empty()) { if(Result.empty())lowExit(3);
debug("Failed to QUERY key HKEY_CURRENT_USER\\Software\\BeamNG\\BeamNG.drive"); //if(IDCheck(Result,T))lowExit(5);
lowExit(3);
}
GameDir = Result; GameDir = Result;
} else { //TraceBack++;
debug("Failed to OPEN key HKEY_CURRENT_USER\\Software\\BeamNG\\BeamNG.drive"); }else lowExit(4);
lowExit(4);
}
K3.clear(); K3.clear();
Result.clear(); Result.clear();
RegCloseKey(hKey); RegCloseKey(hKey);
#elif defined(__linux__) //if(TraceBack < 3)exit(-1);
struct passwd* pw = getpwuid(getuid());
std::string homeDir = pw->pw_dir;
// Right now only steam is supported
std::ifstream libraryFolders(homeDir + "/.steam/root/steamapps/libraryfolders.vdf");
auto root = tyti::vdf::read(libraryFolders);
for (auto folderInfo : root.childs) {
if (std::filesystem::exists(folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/")) {
GameDir = folderInfo.second->attribs["path"] + "/steamapps/common/BeamNG.drive/";
break;
}
}
#endif
} }
std::string CheckVer(const std::string& dir) { std::string CheckVer(const std::string &dir){
#if defined(_WIN32) std::string temp,Path = dir + "\\integrity.json";
std::string temp, Path = dir + "\\integrity.json";
#elif defined(__linux__)
std::string temp, Path = dir + "/integrity.json";
#endif
std::ifstream f(Path.c_str(), std::ios::binary); std::ifstream f(Path.c_str(), std::ios::binary);
int Size = int(std::filesystem::file_size(Path)); int Size = int(std::filesystem::file_size(Path));
std::string vec(Size, 0); std::string vec(Size,0);
f.read(&vec[0], Size); f.read(&vec[0], Size);
f.close(); f.close();
vec = vec.substr(vec.find_last_of("version"), vec.find_last_of('"')); vec = vec.substr(vec.find_last_of("version"),vec.find_last_of('"'));
for (const char& a : vec) { for(const char &a : vec){
if (isdigit(a) || a == '.') if(isdigit(a) || a == '.')temp+=a;
temp += a;
} }
return temp; return temp;
} }

273
src/Security/Login.cpp Normal file → Executable file
View File

@@ -1,150 +1,123 @@
// Copyright (c) 2019-present Anonymous275. // Copyright (c) 2019-present Anonymous275.
// BeamMP Launcher code is not in the public domain and is not free software. // BeamMP Launcher code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries. // One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository. // Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
/// ///
/// Created by Anonymous275 on 11/26/2020 /// Created by Anonymous275 on 11/26/2020
/// ///
#include "Http.h" #include <nlohmann/json.hpp>
#include "Logger.h" #include "Http.h"
#include <filesystem> #include <filesystem>
#include <fstream> #include "Logger.h"
#include <nlohmann/json.hpp> #include <fstream>
namespace fs = std::filesystem;
std::string PublicKey; namespace fs = std::filesystem;
std::string PrivateKey; std::string PublicKey;
extern bool LoginAuth; std::string PrivateKey;
extern std::string Username; extern bool LoginAuth;
extern std::string UserRole; std::string Role;
extern int UserID;
void UpdateKey(const char* newKey){
void UpdateKey(const char* newKey) { if(newKey && std::isalnum(newKey[0])){
if (newKey && std::isalnum(newKey[0])) { PrivateKey = newKey;
PrivateKey = newKey; std::ofstream Key("key");
std::ofstream Key("key"); if(Key.is_open()){
if (Key.is_open()) { Key << newKey;
Key << newKey; Key.close();
Key.close(); }else fatal("Cannot write to disk!");
} else }else if(fs::exists("key")){
fatal("Cannot write to disk!"); remove("key");
} else if (fs::exists("key")) { }
remove("key"); }
}
} /// "username":"value","password":"value"
/// "Guest":"Name"
/// "username":"value","password":"value" /// "pk":"private_key"
/// "Guest":"Name"
/// "pk":"private_key" std::string GetFail(const std::string& R){
std::string DRet = R"({"success":false,"message":)";
std::string GetFail(const std::string& R) { DRet += "\""+R+"\"}";
std::string DRet = R"({"success":false,"message":)"; error(R);
DRet += "\"" + R + "\"}"; return DRet;
error(R); }
return DRet;
} std::string Login(const std::string& fields){
if(fields == "LO"){
std::string Login(const std::string& fields) { LoginAuth = false;
if (fields == "LO") { UpdateKey(nullptr);
Username = ""; return "";
UserRole = ""; }
UserID = -1; info("Attempting to authenticate...");
LoginAuth = false; std::string Buffer = HTTP::Post("https://auth.beammp.com/userlogin", fields);
UpdateKey(nullptr);
return ""; if(Buffer == "-1"){
} return GetFail("Failed to communicate with the auth system!");
info("Attempting to authenticate..."); }
try {
std::string Buffer = HTTP::Post("https://auth.beammp.com/userlogin", fields); nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false);
if (Buffer.empty()) { if (Buffer.at(0) != '{' || d.is_discarded()) {
return GetFail("Failed to communicate with the auth system!"); error(Buffer);
} return GetFail("Invalid answer from authentication servers, please try again later!");
}
nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false); if(d.contains("success") && d["success"].get<bool>()){
LoginAuth = true;
if (Buffer.at(0) != '{' || d.is_discarded()) { if(!d.contains("private_key")) {
error(Buffer); UpdateKey(d["private_key"].get<std::string>().c_str());
return GetFail("Invalid answer from authentication servers, please try again later!"); }
} if(!d.contains("public_key")){
if (d.contains("success") && d["success"].get<bool>()) { PublicKey = d["public_key"].get<std::string>();
LoginAuth = true; }
if (d.contains("username")) { info("Authentication successful!");
Username = d["username"].get<std::string>(); }else info("Authentication failed!");
} if(!d.contains("message")){
if (d.contains("role")) { d.erase("private_key");
UserRole = d["role"].get<std::string>(); d.erase("public_key");
} return d.dump();
if (d.contains("id")) { }
UserID = d["id"].get<int>(); return GetFail("Invalid message parsing!");
} }
if (d.contains("private_key")) {
UpdateKey(d["private_key"].get<std::string>().c_str()); void CheckLocalKey(){
} if(fs::exists("key") && fs::file_size("key") < 100){
if (d.contains("public_key")) { std::ifstream Key("key");
PublicKey = d["public_key"].get<std::string>(); if(Key.is_open()) {
} auto Size = fs::file_size("key");
info("Authentication successful!"); std::string Buffer(Size, 0);
} else Key.read(&Buffer[0], Size);
info("Authentication failed!"); Key.close();
if (d.contains("message")) {
d.erase("private_key"); for (char& c : Buffer) {
d.erase("public_key"); if (!std::isalnum(c) && c != '-') {
return d.dump(); UpdateKey(nullptr);
} return;
return GetFail("Invalid message parsing!"); }
} catch (const std::exception& e) { }
return GetFail(e.what());
} Buffer = HTTP::Post("https://auth.beammp.com/userlogin", R"({"pk":")" + Buffer + "\"}");
}
nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false);
void CheckLocalKey() {
if (fs::exists("key") && fs::file_size("key") < 100) { if (Buffer == "-1" || Buffer.at(0) != '{' || d.is_discarded()) {
std::ifstream Key("key"); error(Buffer);
if (Key.is_open()) { info("Invalid answer from authentication servers.");
auto Size = fs::file_size("key"); UpdateKey(nullptr);
std::string Buffer(Size, 0); }
Key.read(&Buffer[0], Size); if(d["success"].get<bool>()){
Key.close(); LoginAuth = true;
UpdateKey(d["private_key"].get<std::string>().c_str());
for (char& c : Buffer) { PublicKey = d["public_key"].get<std::string>();
if (!std::isalnum(c) && c != '-') { Role = d["role"].get<std::string>();
UpdateKey(nullptr); //info(Role);
return; }else{
} info("Auto-Authentication unsuccessful please re-login!");
} UpdateKey(nullptr);
}
Buffer = HTTP::Post("https://auth.beammp.com/userlogin", R"({"pk":")" + Buffer + "\"}"); }else{
warn("Could not open saved key!");
nlohmann::json d = nlohmann::json::parse(Buffer, nullptr, false); UpdateKey(nullptr);
}
if (Buffer.empty() || Buffer.at(0) != '{' || d.is_discarded()) { }else UpdateKey(nullptr);
error(Buffer); }
info("Invalid answer from authentication servers.");
UpdateKey(nullptr);
}
if (d["success"].get<bool>()) {
LoginAuth = true;
UpdateKey(d["private_key"].get<std::string>().c_str());
PublicKey = d["public_key"].get<std::string>();
if (d.contains("username")) {
Username = d["username"].get<std::string>();
}
if (d.contains("role")) {
UserRole = d["role"].get<std::string>();
}
if (d.contains("id")) {
UserID = d["id"].get<int>();
}
} else {
info("Auto-Authentication unsuccessful please re-login!");
UpdateKey(nullptr);
}
} else {
warn("Could not open saved key!");
UpdateKey(nullptr);
}
} else
UpdateKey(nullptr);
}

683
src/Startup.cpp Normal file → Executable file
View File

@@ -1,357 +1,326 @@
// Copyright (c) 2019-present Anonymous275. // Copyright (c) 2019-present Anonymous275.
// BeamMP Launcher code is not in the public domain and is not free software. // BeamMP Launcher code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries. // One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository. // Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
/// ///
/// Created by Anonymous275 on 7/16/2020 /// Created by Anonymous275 on 7/16/2020
/// ///
#include "zip_file.h" #include <nlohmann/json.hpp>
#include <charconv> #include <httplib.h>
#include <httplib.h> #include "zip_file.h"
#include <nlohmann/json.hpp> #include <windows.h>
#include <string> #include "Discord/discord_info.h"
#if defined(_WIN32) #include "Network/network.h"
#include <windows.h> #include "Security/Init.h"
#elif defined(__linux__) #include <filesystem>
#include <unistd.h> #include "Startup.h"
#endif #include "hashpp.h"
#include "Http.h" #include "Logger.h"
#include "Logger.h" #include <fstream>
#include "Network/network.hpp" #include <thread>
#include "Security/Init.h" #include "Http.h"
#include "Startup.h"
#include "hashpp.h"
#include <filesystem> extern int TraceBack;
#include <fstream> bool Dev = false;
#include <thread> int ProxyPort = 0;
extern int TraceBack; namespace fs = std::filesystem;
bool Dev = false;
int ProxyPort = 0; VersionParser::VersionParser(const std::string& from_string) {
std::string token;
namespace fs = std::filesystem; std::istringstream tokenStream(from_string);
while (std::getline(tokenStream, token, '.')) {
struct Version { data.emplace_back(std::stol(token));
uint8_t major; split.emplace_back(token);
uint8_t minor; }
uint8_t patch; }
Version(uint8_t major, uint8_t minor, uint8_t patch);
Version(const std::array<uint8_t, 3>& v); std::strong_ordering VersionParser::operator<=>(
}; const VersionParser& rhs) const noexcept {
size_t const fields = std::min(data.size(), rhs.data.size());
std::array<uint8_t, 3> VersionStrToInts(const std::string& str) { for (size_t i = 0; i != fields; ++i) {
std::array<uint8_t, 3> Version; if (data[i] == rhs.data[i]) continue;
std::stringstream ss(str); else if (data[i] < rhs.data[i]) return std::strong_ordering::less;
for (uint8_t& i : Version) { else return std::strong_ordering::greater;
std::string Part; }
std::getline(ss, Part, '.'); if (data.size() == rhs.data.size()) return std::strong_ordering::equal;
std::from_chars(&*Part.begin(), &*Part.begin() + Part.size(), i); else if (data.size() > rhs.data.size()) return std::strong_ordering::greater;
} else return std::strong_ordering::less;
return Version; }
}
bool VersionParser::operator==(const VersionParser& rhs) const noexcept {
bool IsOutdated(const Version& Current, const Version& Newest) { return std::is_eq(*this <=> rhs);
if (Newest.major > Current.major) { }
return true;
} else if (Newest.major == Current.major && Newest.minor > Current.minor) {
return true; std::string GetEN(){
} else if (Newest.major == Current.major && Newest.minor == Current.minor && Newest.patch > Current.patch) { return "BeamMP-Launcher.exe";
return true; }
} else { std::string GetVer(){
return false; return "2.0";
} }
} std::string GetPatch(){
return ".84";
Version::Version(uint8_t major, uint8_t minor, uint8_t patch) }
: major(major)
, minor(minor) std::string GetEP(char*P){
, patch(patch) { } static std::string Ret = [&](){
std::string path(P);
Version::Version(const std::array<uint8_t, 3>& v) return path.substr(0, path.find_last_of("\\/") + 1);
: Version(v[0], v[1], v[2]) { } ();
} return Ret;
}
std::string GetEN() { void ReLaunch(int argc,char*args[]){
#if defined(_WIN32) std::string Arg;
return "BeamMP-Launcher.exe"; for(int c = 2; c <= argc; c++){
#elif defined(__linux__) Arg += " ";
return "BeamMP-Launcher"; Arg += args[c-1];
#endif }
} system("cls");
ShellExecute(nullptr,"runas",(GetEP() + GetEN()).c_str(),Arg.c_str(),nullptr,SW_SHOWNORMAL);
std::string GetVer() { ShowWindow(GetConsoleWindow(),0);
return "2.2"; std::this_thread::sleep_for(std::chrono::seconds(1));
} exit(1);
std::string GetPatch() { }
return ".0"; void URelaunch(int argc,char* args[]){
} std::string Arg;
for(int c = 2; c <= argc; c++){
std::string GetEP(char* P) { Arg += " ";
static std::string Ret = [&]() { Arg += args[c-1];
std::string path(P); }
return path.substr(0, path.find_last_of("\\/") + 1); ShellExecute(nullptr,"open",(GetEP() + GetEN()).c_str(),Arg.c_str(),nullptr,SW_SHOWNORMAL);
}(); ShowWindow(GetConsoleWindow(),0);
return Ret; std::this_thread::sleep_for(std::chrono::seconds(1));
} exit(1);
#if defined(_WIN32) }
void ReLaunch(int argc, char* args[]) { void CheckName(int argc,char* args[]){
std::string Arg; std::string DN = GetEN(),CDir = args[0],FN = CDir.substr(CDir.find_last_of('\\')+1);
for (int c = 2; c <= argc; c++) { if(FN != DN){
Arg += " "; if(fs::exists(DN))remove(DN.c_str());
Arg += args[c - 1]; if(fs::exists(DN))ReLaunch(argc,args);
} std::rename(FN.c_str(), DN.c_str());
info("Relaunch!"); URelaunch(argc,args);
system("cls"); }
ShellExecute(nullptr, "runas", (GetEP() + GetEN()).c_str(), Arg.c_str(), nullptr, SW_SHOWNORMAL); }
ShowWindow(GetConsoleWindow(), 0);
std::this_thread::sleep_for(std::chrono::seconds(1)); void CheckForUpdates(int argc, char* args[], const std::string& CV) {
exit(1); std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/launcher?branch=" + Branch + "&pk=" + PublicKey);
} std::string LatestVersion = HTTP::Get(
void URelaunch(int argc, char* args[]) { "https://backend.beammp.com/version/launcher?branch=" + Branch + "&pk=" + PublicKey);
std::string Arg;
for (int c = 2; c <= argc; c++) { transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower);
Arg += " "; std::string EP(GetEP() + GetEN()), Back(GetEP() + "BeamMP-Launcher.back");
Arg += args[c - 1];
} std::string FileHash = hashpp::get::getFileHash(hashpp::ALGORITHMS::SHA2_256, EP);
ShellExecute(nullptr, "open", (GetEP() + GetEN()).c_str(), Arg.c_str(), nullptr, SW_SHOWNORMAL);
ShowWindow(GetConsoleWindow(), 0); if (FileHash != LatestHash && VersionParser(LatestVersion) > VersionParser(GetVer()+GetPatch())) {
std::this_thread::sleep_for(std::chrono::seconds(1)); info("Launcher update found!");
exit(1); fs::remove(Back);
} fs::rename(EP, Back);
#elif defined(__linux__) info("Downloading Launcher update " + LatestHash);
void ReLaunch(int argc, char* args[]) { HTTP::Download(
std::string Arg; "https://backend.beammp.com/builds/launcher?download=true"
for (int c = 2; c <= argc; c++) { "&pk=" +
Arg += " "; PublicKey + "&branch=" + Branch,
Arg += args[c - 1]; EP);
} URelaunch(argc, args);
info("Relaunch!"); } else info("Launcher version is up to date");
system("clear"); TraceBack++;
execl((GetEP() + GetEN()).c_str(), Arg.c_str(), NULL); }
std::this_thread::sleep_for(std::chrono::seconds(1));
exit(1); void CustomPort(int argc, char* argv[]){
} if(argc > 1){
void URelaunch(int argc, char* args[]) { std::string Port = argv[1];
std::string Arg; if(Port.find_first_not_of("0123456789") == std::string::npos){
for (int c = 2; c <= argc; c++) { if(std::stoi(Port) > 1000){
Arg += " "; DEFAULT_PORT = std::stoi(Port);
Arg += args[c - 1]; warn("Running on custom port : " + std::to_string(DEFAULT_PORT));
} }
execl((GetEP() + GetEN()).c_str(), Arg.c_str(), NULL); }
std::this_thread::sleep_for(std::chrono::seconds(1)); if(argc > 2)Dev = true;
exit(1); }
} }
#endif
void LinuxPatch(){
void CheckName(int argc, char* args[]) { HKEY hKey = nullptr;
#if defined(_WIN32) LONG result = RegOpenKeyEx(HKEY_CURRENT_USER, R"(Software\Wine)", 0, KEY_READ, &hKey);
std::string DN = GetEN(), CDir = args[0], FN = CDir.substr(CDir.find_last_of('\\') + 1); if (result != ERROR_SUCCESS || getenv("USER") == nullptr)return;
#elif defined(__linux__) RegCloseKey(hKey);
std::string DN = GetEN(), CDir = args[0], FN = CDir.substr(CDir.find_last_of('/') + 1); info("Wine/Proton Detected! If you are on windows delete HKEY_CURRENT_USER\\Software\\Wine in regedit");
#endif info("Applying patches...");
if (FN != DN) {
if (fs::exists(DN)) result = RegCreateKey(HKEY_CURRENT_USER, R"(Software\Valve\Steam\Apps\284160)", &hKey);
remove(DN.c_str());
if (fs::exists(DN)) if (result != ERROR_SUCCESS){
ReLaunch(argc, args); fatal(R"(failed to create HKEY_CURRENT_USER\Software\Valve\Steam\Apps\284160)");
std::rename(FN.c_str(), DN.c_str()); return;
URelaunch(argc, args); }
}
} result = RegSetValueEx(hKey, "Name", 0, REG_SZ, (BYTE*)"BeamNG.drive", 12);
void CheckForUpdates(int argc, char* args[], const std::string& CV) { if (result != ERROR_SUCCESS){
std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/launcher?branch=" + Branch + "&pk=" + PublicKey); fatal(R"(failed to create the value "Name" under HKEY_CURRENT_USER\Software\Valve\Steam\Apps\284160)");
std::string LatestVersion = HTTP::Get( return;
"https://backend.beammp.com/version/launcher?branch=" + Branch + "&pk=" + PublicKey); }
RegCloseKey(hKey);
transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower);
std::string EP(GetEP() + GetEN()), Back(GetEP() + "BeamMP-Launcher.back"); info("Patched!");
}
std::string FileHash = hashpp::get::getFileHash(hashpp::ALGORITHMS::SHA2_256, EP);
void InitLauncher(int argc, char* argv[]) {
if (FileHash != LatestHash && IsOutdated(Version(VersionStrToInts(GetVer() + GetPatch())), Version(VersionStrToInts(LatestVersion))) && !Dev) { system("cls");
info("Launcher update found!"); SetConsoleTitleA(("BeamMP Launcher v" + std::string(GetVer()) + GetPatch()).c_str());
#if defined(__linux__) InitLog();
error("Auto update is NOT implemented for the Linux version. Please update manually ASAP as updates contain security patches."); CheckName(argc, argv);
#else LinuxPatch();
fs::remove(Back); CheckLocalKey();
fs::rename(EP, Back); ConfigInit();
info("Downloading Launcher update " + LatestHash); CustomPort(argc, argv);
HTTP::Download( Discord_Main();
"https://backend.beammp.com/builds/launcher?download=true" CheckForUpdates(argc, argv, std::string(GetVer()) + GetPatch());
"&pk=" }
+ PublicKey + "&branch=" + Branch,
EP); size_t DirCount(const std::filesystem::path& path){
URelaunch(argc, args); return (size_t)std::distance(std::filesystem::directory_iterator{path}, std::filesystem::directory_iterator{});
#endif }
} else
info("Launcher version is up to date"); void CheckMP(const std::string& Path) {
TraceBack++; if (!fs::exists(Path))return;
} size_t c = DirCount(fs::path(Path));
try {
void CustomPort(int argc, char* argv[]) { for (auto& p : fs::directory_iterator(Path)){
if (argc > 1) { if(p.exists() && !p.is_directory()){
std::string Port = argv[1]; std::string Name = p.path().filename().string();
if (Port.find_first_not_of("0123456789") == std::string::npos) { for(char&Ch : Name)Ch = char(tolower(Ch));
if (std::stoi(Port) > 1000) { if(Name != "beammp.zip")fs::remove(p.path());
DEFAULT_PORT = std::stoi(Port); }
warn("Running on custom port : " + std::to_string(DEFAULT_PORT)); }
} } catch (...) {
} fatal("We were unable to clean the multiplayer mods folder! Is the game still running or do you have something open in that folder?");
if (argc > 2) }
Dev = true;
} }
}
void EnableMP(){
#ifdef _WIN32 std::string File(GetGamePath() + "mods/db.json");
void LinuxPatch() { if(!fs::exists(File))return;
HKEY hKey = nullptr; auto Size = fs::file_size(File);
LONG result = RegOpenKeyEx(HKEY_CURRENT_USER, R"(Software\Wine)", 0, KEY_READ, &hKey); if(Size < 2)return;
if (result != ERROR_SUCCESS || getenv("USER") == nullptr) std::ifstream db(File);
return; if(db.is_open()) {
RegCloseKey(hKey); std::string Data(Size, 0);
info("Wine/Proton Detected! If you are on windows delete HKEY_CURRENT_USER\\Software\\Wine in regedit"); db.read(&Data[0], Size);
info("Applying patches..."); db.close();
nlohmann::json d = nlohmann::json::parse(Data, nullptr, false);
result = RegCreateKey(HKEY_CURRENT_USER, R"(Software\Valve\Steam\Apps\284160)", &hKey); if(Data.at(0) != '{' || d.is_discarded()) {
//error("Failed to parse " + File); //TODO illegal formatting
if (result != ERROR_SUCCESS) { return;
fatal(R"(failed to create HKEY_CURRENT_USER\Software\Valve\Steam\Apps\284160)"); }
return; if(d.contains("mods") && d["mods"].contains("multiplayerbeammp")){
} d["mods"]["multiplayerbeammp"]["active"] = true;
std::ofstream ofs(File);
result = RegSetValueEx(hKey, "Name", 0, REG_SZ, (BYTE*)"BeamNG.drive", 12); if(ofs.is_open()){
ofs << d.dump();
if (result != ERROR_SUCCESS) { ofs.close();
fatal(R"(failed to create the value "Name" under HKEY_CURRENT_USER\Software\Valve\Steam\Apps\284160)"); }else{
return; error("Failed to write " + File);
} }
RegCloseKey(hKey); }
}
info("Patched!"); }
}
#endif void PreGame(const std::string& GamePath){
std::string GameVer = CheckVer(GamePath);
#if defined(_WIN32) info("Game Version : " + GameVer);
void InitLauncher(int argc, char* argv[]) {
SetConsoleTitleA(("BeamMP Launcher v" + std::string(GetVer()) + GetPatch()).c_str()); CheckMP(GetGamePath() + "mods/multiplayer");
InitLog();
CheckName(argc, argv); if(!Dev) {
LinuxPatch(); std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/mod?branch=" + Branch + "&pk=" + PublicKey);
CheckLocalKey(); transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower);
ConfigInit(); LatestHash.erase(std::remove_if(LatestHash.begin(), LatestHash.end(),
CustomPort(argc, argv); [](auto const& c ) -> bool { return !std::isalnum(c); } ), LatestHash.end());
CheckForUpdates(argc, argv, std::string(GetVer()) + GetPatch());
} try {
#elif defined(__linux__) if (!fs::exists(GetGamePath() + "mods/multiplayer")) {
void InitLauncher(int argc, char* argv[]) { fs::create_directories(GetGamePath() + "mods/multiplayer");
InitLog(); }
info("BeamMP Launcher v" + GetVer() + GetPatch()); EnableMP();
CheckName(argc, argv); }catch(std::exception&e){
CheckLocalKey(); fatal(e.what());
ConfigInit(); }
CustomPort(argc, argv);
CheckForUpdates(argc, argv, std::string(GetVer()) + GetPatch()); std::string ZipPath(GetGamePath() + R"(mods\multiplayer\BeamMP.zip)");
}
#endif std::string FileHash = hashpp::get::getFileHash(hashpp::ALGORITHMS::SHA2_256, ZipPath);
size_t DirCount(const std::filesystem::path& path) { if (FileHash != LatestHash) {
return (size_t)std::distance(std::filesystem::directory_iterator { path }, std::filesystem::directory_iterator {}); info("Downloading BeamMP Update " + LatestHash);
} HTTP::Download("https://backend.beammp.com/builds/client?download=true"
"&pk=" + PublicKey + "&branch=" + Branch, ZipPath);
void CheckMP(const std::string& Path) { }
if (!fs::exists(Path))
return; std::string Target(GetGamePath() + "mods/unpacked/beammp");
size_t c = DirCount(fs::path(Path));
try { if(fs::is_directory(Target)) {
for (auto& p : fs::directory_iterator(Path)) { fs::remove_all(Target);
if (p.exists() && !p.is_directory()) { }
std::string Name = p.path().filename().string(); }
for (char& Ch : Name) }
Ch = char(tolower(Ch));
if (Name != "beammp.zip") void set_headers(httplib::Response& res) {
fs::remove(p.path()); res.set_header("Access-Control-Allow-Origin", "*");
} res.set_header("Access-Control-Allow-Headers", "X-API-Version");
} res.set_header("Access-Control-Allow-Method", "GET,POST,OPTIONS");
} catch (...) { }
fatal("We were unable to clean the multiplayer mods folder! Is the game still running or do you have something open in that folder?");
} void StartProxy() {
} std::thread proxy([&](){
httplib::Server HTTPProxy;
void EnableMP() { httplib::Headers headers = {
std::string File(GetGamePath() + "mods/db.json"); {"User-Agent", "BeamMP-Launcher/" + GetVer() + GetPatch()},
if (!fs::exists(File)) {"Accept", "*/*"}
return; };
auto Size = fs::file_size(File); std::string pattern = "/:any1";
if (Size < 2) for (int i = 2; i <= 4; i++) {
return; HTTPProxy.Get(pattern, [&](const httplib::Request &req, httplib::Response &res) {
std::ifstream db(File); httplib::Client cli("https://backend.beammp.com");
if (db.is_open()) { set_headers(res);
std::string Data(Size, 0); if (req.has_header("X-BMP-Authentication")) {
db.read(&Data[0], Size); headers.emplace("X-BMP-Authentication", PrivateKey);
db.close(); }
nlohmann::json d = nlohmann::json::parse(Data, nullptr, false); if (req.has_header("X-API-Version")) {
if (Data.at(0) != '{' || d.is_discarded()) { headers.emplace("X-API-Version", req.get_header_value("X-API-Version"));
// error("Failed to parse " + File); //TODO illegal formatting }
return; if (auto cli_res = cli.Get(req.path, headers); cli_res) {
} res.set_content(cli_res->body, cli_res->get_header_value("Content-Type"));
if (d.contains("mods") && d["mods"].contains("multiplayerbeammp")) { } else {
d["mods"]["multiplayerbeammp"]["active"] = true; res.set_content(to_string(cli_res.error()), "text/plain");
std::ofstream ofs(File); }
if (ofs.is_open()) { });
ofs << d.dump();
ofs.close(); HTTPProxy.Post(pattern, [&](const httplib::Request &req, httplib::Response &res) {
} else { httplib::Client cli("https://backend.beammp.com");
error("Failed to write " + File); set_headers(res);
} if (req.has_header("X-BMP-Authentication")) {
} headers.emplace("X-BMP-Authentication", PrivateKey);
} }
} if (req.has_header("X-API-Version")) {
headers.emplace("X-API-Version", req.get_header_value("X-API-Version"));
void PreGame(const std::string& GamePath) { }
std::string GameVer = CheckVer(GamePath); if (auto cli_res = cli.Post(req.path, headers, req.body,
info("Game Version : " + GameVer); req.get_header_value("Content-Type")); cli_res) {
res.set_content(cli_res->body, cli_res->get_header_value("Content-Type"));
CheckMP(GetGamePath() + "mods/multiplayer"); } else {
info("Game user path: " + GetGamePath()); res.set_content(to_string(cli_res.error()), "text/plain");
}
if (!Dev) { });
std::string LatestHash = HTTP::Get("https://backend.beammp.com/sha/mod?branch=" + Branch + "&pk=" + PublicKey); pattern += "/:any" + std::to_string(i);
transform(LatestHash.begin(), LatestHash.end(), LatestHash.begin(), ::tolower); }
LatestHash.erase(std::remove_if(LatestHash.begin(), LatestHash.end(), ProxyPort = HTTPProxy.bind_to_any_port("0.0.0.0");
[](auto const& c) -> bool { return !std::isalnum(c); }), HTTPProxy.listen_after_bind();
LatestHash.end()); });
proxy.detach();
try { }
if (!fs::exists(GetGamePath() + "mods/multiplayer")) {
fs::create_directories(GetGamePath() + "mods/multiplayer");
}
EnableMP();
} catch (std::exception& e) {
fatal(e.what());
}
#if defined(_WIN32)
std::string ZipPath(GetGamePath() + R"(mods\multiplayer\BeamMP.zip)");
#elif defined(__linux__)
// Linux version of the game cant handle mods with uppercase names
std::string ZipPath(GetGamePath() + R"(mods/multiplayer/beammp.zip)");
#endif
std::string FileHash = hashpp::get::getFileHash(hashpp::ALGORITHMS::SHA2_256, ZipPath);
if (FileHash != LatestHash) {
info("Downloading BeamMP Update " + LatestHash);
HTTP::Download("https://backend.beammp.com/builds/client?download=true"
"&pk="
+ PublicKey + "&branch=" + Branch,
ZipPath);
}
std::string Target(GetGamePath() + "mods/unpacked/beammp");
if (fs::is_directory(Target)) {
fs::remove_all(Target);
}
}
}

50
src/main.cpp Normal file → Executable file
View File

@@ -5,68 +5,40 @@
/// ///
/// Created by Anonymous275 on 7/16/2020 /// Created by Anonymous275 on 7/16/2020
/// ///
#include "Http.h" #include "Network/network.h"
#include "Logger.h"
#include "Network/network.hpp"
#include "Security/Init.h" #include "Security/Init.h"
#include "Startup.h" #include "Startup.h"
#include <curl/curl.h>
#include <iostream> #include <iostream>
#include "Logger.h"
#include <thread> #include <thread>
#include "Http.h"
[[noreturn]] void flush() { [[noreturn]] void flush(){
while (true) { while(true){
std::cout.flush(); std::cout.flush();
std::this_thread::sleep_for(std::chrono::milliseconds(100)); std::this_thread::sleep_for(std::chrono::milliseconds(100));
} }
} }
int main(int argc, char** argv) try { int main(int argc, char* argv[]) {
#ifdef DEBUG #ifdef DEBUG
std::thread th(flush); std::thread th(flush);
th.detach(); th.detach();
#endif #endif
curl_global_init(CURL_GLOBAL_ALL);
#if defined(_WIN32)
system("cls");
#elif defined(__linux__)
system("clear");
#endif
GetEP(argv[0]); GetEP(argv[0]);
for (int i = 0; i < argc; ++i) {
if (std::string_view(argv[i]) == "--skip-ssl-verify") {
info("SSL verification skip enabled");
HTTP::SkipSslVerify = true;
}
}
InitLauncher(argc, argv); InitLauncher(argc, argv);
info("IMPORTANT: You MUST keep this window open to play BeamMP!");
try { try {
LegitimacyCheck(); LegitimacyCheck();
} catch (std::exception& e) { } catch (std::exception &e) {
error("Failure in LegitimacyCheck: " + std::string(e.what())); fatal("Main 1 : " + std::string(e.what()));
throw;
} }
try { StartProxy();
HTTP::StartProxy();
} catch (const std::exception& e) {
error(std::string("Failed to start HTTP proxy: Some in-game functions may not work. Error: ") + e.what());
}
PreGame(GetGameDir()); PreGame(GetGameDir());
InitGame(GetGameDir()); InitGame(GetGameDir());
CoreNetwork(); CoreNetwork();
} catch (const std::exception& e) {
error(std::string("Exception in main(): ") + e.what()); ///TODO: make sure to use argv[0] for everything that should be in the same dir (mod down ect...)
info("Closing in 5 seconds");
info("If this keeps happening, contact us on either: Forum: https://forum.beammp.com, Discord: https://discord.gg/beammp");
std::this_thread::sleep_for(std::chrono::seconds(5));
} }

View File

@@ -1,9 +0,0 @@
{
"dependencies": [
"cpp-httplib",
"nlohmann-json",
"zlib",
"openssl",
"curl"
]
}