Compare commits

..

3 Commits

Author SHA1 Message Date
Lion Kortlepel 103d2dc030 update vcpkg in workflows 2023-12-03 17:50:45 +01:00
Lion Kortlepel 7f206fd0d4 bump version to 3.1.2 2023-12-03 17:49:12 +01:00
Lion Kortlepel 22805af716 fix crash in authentication 2023-12-03 17:43:41 +01:00
45 changed files with 771 additions and 1487 deletions
+2 -2
View File
@@ -18,9 +18,9 @@ jobs:
uses: lukka/run-vcpkg@v7 uses: lukka/run-vcpkg@v7
id: runvcpkg id: runvcpkg
with: with:
vcpkgArguments: 'lua zlib rapidjson openssl websocketpp curl boost-variant boost-spirit boost-phoenix boost-core boost-system boost-asio boost-uuid' vcpkgArguments: 'lua zlib rapidjson openssl websocketpp curl'
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg' vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
vcpkgGitCommitId: "06b5f4a769d848d1a20fa0acd556019728b56273" vcpkgGitCommitId: "16ee2ecb31788c336ace8bb14c21801efb6836e4"
vcpkgTriplet: 'x64-windows-static' vcpkgTriplet: 'x64-windows-static'
- name: Create Build Environment - name: Create Build Environment
+2 -2
View File
@@ -83,9 +83,9 @@ jobs:
uses: lukka/run-vcpkg@v7 uses: lukka/run-vcpkg@v7
id: runvcpkg id: runvcpkg
with: with:
vcpkgArguments: 'lua zlib rapidjson openssl websocketpp curl boost-variant boost-spirit boost-phoenix boost-core boost-system boost-asio' vcpkgArguments: 'lua zlib rapidjson openssl websocketpp curl'
vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg' vcpkgDirectory: '${{ runner.workspace }}/b/vcpkg'
vcpkgGitCommitId: '06b5f4a769d848d1a20fa0acd556019728b56273' vcpkgGitCommitId: '16ee2ecb31788c336ace8bb14c21801efb6836e4'
vcpkgTriplet: 'x64-windows-static' vcpkgTriplet: 'x64-windows-static'
- name: Create Build Environment - name: Create Build Environment
-3
View File
@@ -31,6 +31,3 @@
[submodule "deps/doctest"] [submodule "deps/doctest"]
path = deps/doctest path = deps/doctest
url = https://github.com/doctest/doctest url = https://github.com/doctest/doctest
[submodule "deps/lk-result"]
path = deps/lk-result
url = https://gitlab.com/lionkor/lk-result.git
+7 -31
View File
@@ -4,32 +4,11 @@ cmake_minimum_required(VERSION 3.4 FATAL_ERROR)
message(STATUS "You can find build instructions and a list of dependencies in the README at \ message(STATUS "You can find build instructions and a list of dependencies in the README at \
https://github.com/BeamMP/BeamMP-Server") https://github.com/BeamMP/BeamMP-Server")
include(FetchContent)
FetchContent_Declare(
cmake-tools
GIT_REPOSITORY https://github.com/brobeson/cmake-tools.git
GIT_TAG main
)
FetchContent_MakeAvailable(cmake-tools)
list(APPEND CMAKE_MODULE_PATH "${cmake-tools_SOURCE_DIR}")
include(CMakeToolsVersionFromGit)
execute_process(
COMMAND git log -1 --format=%h
WORKING_DIRECTORY ${CMAKE_CURRENT_LIST_DIR}
OUTPUT_VARIABLE GIT_HASH
OUTPUT_STRIP_TRAILING_WHITESPACE
)
project(BeamMP-Server project(BeamMP-Server
DESCRIPTION "Server for BeamMP - The Multiplayer Mod for BeamNG.drive" DESCRIPTION "Server for BeamMP - The Multiplayer Mod for BeamNG.drive"
HOMEPAGE_URL https://beammp.com HOMEPAGE_URL https://beammp.com
LANGUAGES CXX C) LANGUAGES CXX C)
# quality of life stuff
set(CMAKE_COLOR_DIAGNOSTICS ON)
find_package(Git REQUIRED) find_package(Git REQUIRED)
# Update submodules as needed # Update submodules as needed
option(GIT_SUBMODULE "Check submodules during build" ON) option(GIT_SUBMODULE "Check submodules during build" ON)
@@ -43,6 +22,7 @@ if(GIT_SUBMODULE)
endif() endif()
endif() endif()
set(HTTPLIB_REQUIRE_OPENSSL ON) set(HTTPLIB_REQUIRE_OPENSSL ON)
set(SENTRY_BUILD_SHARED_LIBS OFF) set(SENTRY_BUILD_SHARED_LIBS OFF)
@@ -99,7 +79,7 @@ endif()
add_subdirectory("deps/sentry-native") add_subdirectory("deps/sentry-native")
# ------------------------ C++ SETUP --------------------------------- # ------------------------ C++ SETUP ---------------------------------
set(CMAKE_CXX_STANDARD 20) set(CMAKE_CXX_STANDARD 17)
# ------------------------ DEPENDENCIES ------------------------------ # ------------------------ DEPENDENCIES ------------------------------
message(STATUS "Adding local source dependencies") message(STATUS "Adding local source dependencies")
@@ -138,7 +118,6 @@ set(BeamMP_Sources
include/TPluginMonitor.h src/TPluginMonitor.cpp include/TPluginMonitor.h src/TPluginMonitor.cpp
include/Environment.h include/Environment.h
include/BoostAliases.h include/BoostAliases.h
include/Uuid.h src/Uuid.cpp
) )
set(BeamMP_Includes set(BeamMP_Includes
@@ -150,15 +129,11 @@ set(BeamMP_Includes
"${CMAKE_CURRENT_SOURCE_DIR}/deps/sol2/include" "${CMAKE_CURRENT_SOURCE_DIR}/deps/sol2/include"
"${CMAKE_CURRENT_SOURCE_DIR}/deps/rapidjson/include" "${CMAKE_CURRENT_SOURCE_DIR}/deps/rapidjson/include"
"${CMAKE_CURRENT_SOURCE_DIR}/deps/asio/asio/include" "${CMAKE_CURRENT_SOURCE_DIR}/deps/asio/asio/include"
"${CMAKE_CURRENT_SOURCE_DIR}/deps/lk-result/include"
"${CMAKE_CURRENT_SOURCE_DIR}/deps" "${CMAKE_CURRENT_SOURCE_DIR}/deps"
"${CMAKE_CURRENT_SOURCE_DIR}/deps/fmt/include"
) )
set(BeamMP_Definitions set(BeamMP_Definitions
SECRET_SENTRY_URL="${BEAMMP_SECRET_SENTRY_URL}" SECRET_SENTRY_URL="${BEAMMP_SECRET_SENTRY_URL}"
BEAMMP_GIT_HASH="${GIT_HASH}"
LK_RESULT_USE_FMT
) )
if (WIN32) if (WIN32)
@@ -186,8 +161,8 @@ if (UNIX)
-Werror=missing-field-initializers -Werror=missing-field-initializers
-Werror=write-strings -Werror=write-strings
-Werror=ctor-dtor-privacy -Werror=ctor-dtor-privacy
-Wswitch-enum -Werror=switch-enum
-Wswitch-default -Werror=switch-default
-Werror=old-style-cast -Werror=old-style-cast
-Werror=overloaded-virtual -Werror=overloaded-virtual
-Werror=overloaded-virtual -Werror=overloaded-virtual
@@ -198,6 +173,7 @@ if (UNIX)
-Wzero-as-null-pointer-constant -Wzero-as-null-pointer-constant
) )
else() else()
set(BeamMP_CompileOptions set(BeamMP_CompileOptions
/bigobj /bigobj
/INCREMENTAL:NO /NODEFAULTLIB:MSVCRT /NODEFAULTLIB:LIBCMT /INCREMENTAL:NO /NODEFAULTLIB:MSVCRT /NODEFAULTLIB:LIBCMT
@@ -211,12 +187,12 @@ set(BeamMP_Libraries
OpenSSL::SSL OpenSSL::SSL
OpenSSL::Crypto OpenSSL::Crypto
sol2::sol2 sol2::sol2
fmt::fmt
Threads::Threads Threads::Threads
ZLIB::ZLIB ZLIB::ZLIB
${LUA_LIBRARIES} ${LUA_LIBRARIES}
commandline commandline
sentry sentry
fmt::fmt
) )
if (WIN32) if (WIN32)
@@ -247,7 +223,7 @@ target_include_directories(BeamMP-Server SYSTEM PRIVATE
${BeamMP_Includes} ${BeamMP_Includes}
) )
target_link_libraries(BeamMP-Server target_link_libraries(BeamMP-Server
${BeamMP_Libraries} ${BeamMP_Libraries}
${BeamMP_PlatformLibs} ${BeamMP_PlatformLibs}
) )
-9
View File
@@ -1,12 +1,3 @@
# v3.2.0
- ADDED `settings` command, which lets you `get`, `list`, and `set` config options from the console
- ADDED `debug` command, which shows info about connected clients & networking (for developers)
- ADDED `Util.GenerateUUID()`, which generates an RFC4122 UUID (universally unique identifier)
- ADDED `version` command, which shows version information
- ADDED `onPlayerRequestMods` event, letting Lua disallow individual mods from being sent to clients
- CHANGED `onShutdown` to be called before all players are kicked
# v3.1.1 # v3.1.1
- FIXED bug which caused GetPlayerIdentifiers, GetPlayerName, etc not to work in `onPlayerDisconnect` - FIXED bug which caused GetPlayerIdentifiers, GetPlayerName, etc not to work in `onPlayerDisconnect`
+1 -1
View File
@@ -66,7 +66,7 @@ Please use the prepackaged binaries in [Releases](https://github.com/BeamMP/Beam
Dependencies for **Windows** can be installed with `vcpkg`. Dependencies for **Windows** can be installed with `vcpkg`.
These are: These are:
``` ```
lua zlib rapidjson openssl websocketpp curl boost lua zlib rapidjson openssl websocketpp curl
``` ```
The triplet we use for releases is `x64-windows-static`. The triplet we use for releases is `x64-windows-static`.
-1
Submodule deps/lk-result deleted from 4a4eda8092
+2 -24
View File
@@ -10,7 +10,6 @@
#include "BoostAliases.h" #include "BoostAliases.h"
#include "Common.h" #include "Common.h"
#include "Compat.h" #include "Compat.h"
#include "TResourceManager.h"
#include "VehicleData.h" #include "VehicleData.h"
class TServer; class TServer;
@@ -71,7 +70,6 @@ public:
void ClearCars(); void ClearCars();
[[nodiscard]] int GetID() const { return mID; } [[nodiscard]] int GetID() const { return mID; }
[[nodiscard]] int GetUnicycleID() const { return mUnicycleID; } [[nodiscard]] int GetUnicycleID() const { return mUnicycleID; }
bool mUDPCONNECTED = false;
[[nodiscard]] bool IsConnected() const { return mIsConnected; } [[nodiscard]] bool IsConnected() const { return mIsConnected; }
[[nodiscard]] bool IsSynced() const { return mIsSynced; } [[nodiscard]] bool IsSynced() const { return mIsSynced; }
[[nodiscard]] bool IsSyncing() const { return mIsSyncing; } [[nodiscard]] bool IsSyncing() const { return mIsSyncing; }
@@ -89,24 +87,6 @@ public:
void UpdatePingTime(); void UpdatePingTime();
int SecondsSinceLastPing(); int SecondsSinceLastPing();
// bytes received on UDP
std::atomic_size_t UdpReceived = 0;
// number of packets received on UDP
std::atomic_size_t UdpPacketsReceived = 0;
// bytes sent on UDP
std::atomic_size_t UdpSent = 0;
// number of packets sent on UDP
std::atomic_size_t UdpPacketsSent = 0;
// bytes received on TCP
std::atomic_size_t TcpReceived = 0;
// bytes sent on TCP
std::atomic_size_t TcpSent = 0;
TimeType::time_point ConnectionTime {};
ModMap AllowedMods;
private: private:
void InsertVehicle(int ID, const std::string& Data); void InsertVehicle(int ID, const std::string& Data);
@@ -130,9 +110,7 @@ private:
std::string mRole; std::string mRole;
std::string mDID; std::string mDID;
int mID = -1; int mID = -1;
std::chrono::time_point<TimeType> mLastPingTime; std::chrono::time_point<std::chrono::high_resolution_clock> mLastPingTime;
std::mutex mDisconnectMtx;
}; };
// Returns a valid client, or nullptr if no such client exists std::optional<std::weak_ptr<TClient>> GetClient(class TServer& Server, int ID);
std::shared_ptr<TClient> GetClient(class TServer& Server, int ID);
+55 -151
View File
@@ -1,6 +1,5 @@
#pragma once #pragma once
#include "CustomAssert.h"
#include "TSentry.h" #include "TSentry.h"
extern TSentry Sentry; extern TSentry Sentry;
@@ -24,50 +23,6 @@ namespace fs = std::filesystem;
#include "TConsole.h" #include "TConsole.h"
#include <boost/container/flat_map.hpp>
#include <boost/variant.hpp>
#include <chrono>
#include <variant>
using TimeType = std::chrono::system_clock;
#include <lk/Result.h>
template <typename T>
using Result = lk::Result<T>;
using Error = lk::Error;
// General
constexpr std::string_view StrDebug = "Debug";
constexpr std::string_view StrPrivate = "Private";
constexpr std::string_view StrPort = "Port";
constexpr std::string_view StrMaxCars = "MaxCars";
constexpr std::string_view StrMaxPlayers = "MaxPlayers";
constexpr std::string_view StrMap = "Map";
constexpr std::string_view StrName = "Name";
constexpr std::string_view StrDescription = "Description";
constexpr std::string_view StrResourceFolder = "ResourceFolder";
constexpr std::string_view StrAuthKey = "AuthKey";
constexpr std::string_view StrLogChat = "LogChat";
// Misc
constexpr std::string_view StrSendErrors = "SendErrors";
constexpr std::string_view StrSendErrorsMessageEnabled = "SendErrorsShowMessage";
constexpr std::string_view StrHideUpdateMessages = "ImScaredOfUpdates";
constexpr std::string_view StrIncludeSubdirectories = "IncludeSubdirectories";
// HTTP
constexpr std::string_view StrHTTPServerEnabled = "HTTPServerEnabled";
constexpr std::string_view StrHTTPServerUseSSL = "UseSSL";
constexpr std::string_view StrSSLKeyPath = "SSLKeyPath";
constexpr std::string_view StrSSLCertPath = "SSLCertPath";
constexpr std::string_view StrHTTPServerPort = "HTTPServerPort";
constexpr std::string_view StrHTTPServerIP = "HTTPServerIP";
// Unused
constexpr std::string_view StrCustomIP = "Unused.CustomIP";
struct Version { struct Version {
uint8_t major; uint8_t major;
uint8_t minor; uint8_t minor;
@@ -80,12 +35,6 @@ struct Version {
template <typename T> template <typename T>
using SparseArray = std::unordered_map<size_t, T>; using SparseArray = std::unordered_map<size_t, T>;
template <typename K, typename V>
using HashMap = std::unordered_map<K, V>;
using boost::variant;
using boost::container::flat_map;
// static class handling application start, shutdown, etc. // static class handling application start, shutdown, etc.
// yes, static classes, singletons, globals are all pretty // yes, static classes, singletons, globals are all pretty
// bad idioms. In this case we need a central way to access // bad idioms. In this case we need a central way to access
@@ -94,17 +43,30 @@ using boost::container::flat_map;
class Application final { class Application final {
public: public:
// types // types
struct TSettings {
using SettingValue = std::variant<std::string, bool, int>; std::string ServerName { "BeamMP Server" };
using SettingsMap = flat_map<std::string_view, SettingValue>; std::string ServerDesc { "BeamMP Default Description" };
std::string Resource { "Resources" };
static SettingsMap mSettings; std::string MapName { "/levels/gridmap_v2/info.json" };
std::string Key {};
static std::string GetSettingString(std::string_view Name); std::string SSLKeyPath { "./.ssl/HttpServer/key.pem" };
static int GetSettingInt(std::string_view Name); std::string SSLCertPath { "./.ssl/HttpServer/cert.pem" };
static bool GetSettingBool(std::string_view Name); bool HTTPServerEnabled { false };
int MaxPlayers { 8 };
static void SetSetting(std::string_view Name, const SettingValue& value); bool Private { true };
int MaxCars { 1 };
bool DebugModeEnabled { false };
int Port { 30814 };
std::string CustomIP {};
bool LogChat { true };
bool SendErrors { true };
bool SendErrorsMessageEnabled { true };
int HTTPServerPort { 8080 };
std::string HTTPServerIP { "127.0.0.1" };
bool HTTPServerUseSSL { false };
bool HideUpdateMessages { false };
[[nodiscard]] bool HasCustomIP() const { return !CustomIP.empty(); }
};
using TShutdownHandler = std::function<void()>; using TShutdownHandler = std::function<void()>;
@@ -122,6 +84,8 @@ public:
static std::string PPS() { return mPPS; } static std::string PPS() { return mPPS; }
static void SetPPS(const std::string& NewPPS) { mPPS = NewPPS; } static void SetPPS(const std::string& NewPPS) { mPPS = NewPPS; }
static TSettings Settings;
static std::vector<std::string> GetBackendUrlsInOrder() { static std::vector<std::string> GetBackendUrlsInOrder() {
return { return {
"backend.beammp.com", "backend.beammp.com",
@@ -161,16 +125,6 @@ public:
static void SetSubsystemStatus(const std::string& Subsystem, Status status); static void SetSubsystemStatus(const std::string& Subsystem, Status status);
static std::string SettingToString(const SettingValue& Value);
// Keeps track of how many packets we dropped on UDP due to fundamentally being malformed
static inline std::atomic_size_t MalformedUdpPackets { 0 };
// Keeps track of how many packets we dropped on UDP due to
// 1) not having a valid (known) player id
// 2) player disconnecting
// 3) packet failing to parse
static inline std::atomic_size_t InvalidUdpPackets { 0 };
private: private:
static void SetShutdown(bool Val); static void SetShutdown(bool Val);
@@ -183,7 +137,7 @@ private:
static inline std::mutex mShutdownHandlersMutex {}; static inline std::mutex mShutdownHandlersMutex {};
static inline std::deque<TShutdownHandler> mShutdownHandlers {}; static inline std::deque<TShutdownHandler> mShutdownHandlers {};
static inline Version mVersion { 3, 2, 0 }; static inline Version mVersion { 3, 1, 2 };
}; };
std::string ThreadName(bool DebugModeOverride = false); std::string ThreadName(bool DebugModeOverride = false);
@@ -193,7 +147,6 @@ void RegisterThread(const std::string& str);
#define KB 1024llu #define KB 1024llu
#define MB (KB * 1024llu) #define MB (KB * 1024llu)
#define GB (MB * 1024llu) #define GB (MB * 1024llu)
#define TB (GB * 1024llu)
#define SSU_UNRAW SECRET_SENTRY_URL #define SSU_UNRAW SECRET_SENTRY_URL
#define _file_basename std::filesystem::path(__FILE__).filename().string() #define _file_basename std::filesystem::path(__FILE__).filename().string()
@@ -234,88 +187,41 @@ void RegisterThread(const std::string& str);
#endif // defined(DEBUG) #endif // defined(DEBUG)
#define beammp_internal_error(x) Application::Console().Write(_this_location + std::string("[INTERNAL ERROR] ") + (x)) #define beammp_warn(x) Application::Console().Write(_this_location + std::string("[WARN] ") + (x))
#define beammp_info(x) Application::Console().Write(_this_location + std::string("[INFO] ") + (x))
#define beammp_warn(x) \ #define beammp_error(x) \
do { \ do { \
try{ \ Application::Console().Write(_this_location + std::string("[ERROR] ") + (x)); \
Application::Console().Write(_this_location + std::string("[WARN] ") + (x)); \ Sentry.AddErrorBreadcrumb((x), _file_basename, _line); \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false) } while (false)
#define beammp_info(x) \ #define beammp_lua_error(x) \
do { \ do { \
try{ \ Application::Console().Write(_this_location + std::string("[LUA ERROR] ") + (x)); \
Application::Console().Write(_this_location + std::string("[INFO] ") + (x)); \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false) } while (false)
#define beammp_error(x) \ #define beammp_lua_warn(x) \
do { \ do { \
try{ \ Application::Console().Write(_this_location + std::string("[LUA WARN] ") + (x)); \
Application::Console().Write(_this_location + std::string("[ERROR] ") + (x)); \
Sentry.AddErrorBreadcrumb((x), _file_basename, _line); \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false) } while (false)
#define beammp_lua_error(x) \ #define luaprint(x) Application::Console().Write(_this_location + std::string("[LUA] ") + (x))
do { \ #define beammp_debug(x) \
try{ \ do { \
Application::Console().Write(_this_location + std::string("[LUA ERROR] ") + (x)); \ if (Application::Settings.DebugModeEnabled) { \
} catch (const std::exception& e) { \ Application::Console().Write(_this_location + std::string("[DEBUG] ") + (x)); \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\ } \
} \
} while (false) } while (false)
#define beammp_lua_warn(x) \ #define beammp_event(x) \
do { \ do { \
try{ \ if (Application::Settings.DebugModeEnabled) { \
Application::Console().Write(_this_location + std::string("[LUA WARN] ") + (x)); \ Application::Console().Write(_this_location + std::string("[EVENT] ") + (x)); \
} catch (const std::exception& e) { \ } \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false)
#define luaprint(x) \
do { \
try{ \
Application::Console().Write(_this_location + std::string("[LUA] ") + (x)); \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false)
#define beammp_debug(x) \
do { \
try{ \
if (Application::GetSettingBool("Debug")) { \
Application::Console().Write(_this_location + std::string("[DEBUG] ") + (x)); \
} \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false)
#define beammp_event(x) \
do { \
try{ \
if (Application::GetSettingBool("Debug")) { \
Application::Console().Write(_this_location + std::string("[EVENT] ") + (x)); \
} \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}': {}", #x, e.what()));\
} \
} while (false) } while (false)
// trace() is a debug-build debug() // trace() is a debug-build debug()
#if defined(DEBUG) #if defined(DEBUG)
#define beammp_trace(x) \ #define beammp_trace(x) \
do { \ do { \
try{ \ if (Application::Settings.DebugModeEnabled) { \
if (Application::GetSettingBool("Debug")) { \ Application::Console().Write(_this_location + std::string("[TRACE] ") + (x)); \
Application::Console().Write(_this_location + std::string("[TRACE] ") + (x)); \ } \
} \
} catch (const std::exception& e) { \
beammp_internal_error(fmt::format("Exception in logging function, failed to print '{}'", #x)); \
} \
} while (false) } while (false)
#else #else
#define beammp_trace(x) #define beammp_trace(x)
@@ -415,5 +321,3 @@ inline T DeComp(const T& Compressed) {
std::string GetPlatformAgnosticErrorString(); std::string GetPlatformAgnosticErrorString();
#define S_DSN SU_RAW #define S_DSN SU_RAW
std::string ToHumanReadableSize(size_t Size);
+3 -3
View File
@@ -66,9 +66,9 @@ inline void _assert([[maybe_unused]] const char* file, [[maybe_unused]] const ch
Sentry.LogAssert(#cond, _file_basename, _line, __func__); \ Sentry.LogAssert(#cond, _file_basename, _line, __func__); \
} \ } \
} while (false) } while (false)
#define beammp_assert_not_reachable() \ #define beammp_assert_not_reachable() \
do { \ do { \
beammp_errorf("Assertion failed in '{}:{}': Unreachable code reached. This may result in a crash or undefined state of the program.", __func__, _line); \ beammp_errorf("Assertion failed in '{}:{}': Unreachable code reached. This may result in a crash or undefined state of the program.", __func__, _line); \
Sentry.LogAssert("code is unreachable", _file_basename, _line, __func__); \ Sentry.LogAssert("code is unreachable", _file_basename, _line, __func__); \
} while (false) } while (false)
#endif // DEBUG #endif // DEBUG
-6
View File
@@ -1,6 +0,0 @@
#pragma once
enum IterationDecision {
Continue,
Break,
};
+2 -2
View File
@@ -3,7 +3,7 @@
// //
#pragma once #pragma once
#include "rapidjson/document.h"
#include "rapidjson/prettywriter.h"
#include "rapidjson/stringbuffer.h" #include "rapidjson/stringbuffer.h"
#include "rapidjson/prettywriter.h"
#include "rapidjson/document.h"
#include "rapidjson/writer.h" #include "rapidjson/writer.h"
+1 -1
View File
@@ -6,8 +6,8 @@
* and write locks and read locks are mutually exclusive. * and write locks and read locks are mutually exclusive.
*/ */
#include <mutex>
#include <shared_mutex> #include <shared_mutex>
#include <mutex>
// Use ReadLock(m) and WriteLock(m) to lock it. // Use ReadLock(m) and WriteLock(m) to lock it.
using RWMutex = std::shared_mutex; using RWMutex = std::shared_mutex;
+3 -1
View File
@@ -22,7 +22,9 @@ private:
void CreateConfigFile(); void CreateConfigFile();
void ParseFromFile(std::string_view name); void ParseFromFile(std::string_view name);
void PrintDebug(); void PrintDebug();
void TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key); void TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, std::string& OutValue);
void TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, bool& OutValue);
void TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, int& OutValue);
void ParseOldFormat(); void ParseOldFormat();
bool IsDefault(); bool IsDefault();
-14
View File
@@ -36,12 +36,6 @@ private:
void Command_Status(const std::string& cmd, const std::vector<std::string>& args); void Command_Status(const std::string& cmd, const std::vector<std::string>& args);
void Command_Settings(const std::string& cmd, const std::vector<std::string>& args); void Command_Settings(const std::string& cmd, const std::vector<std::string>& args);
void Command_Clear(const std::string&, const std::vector<std::string>& args); void Command_Clear(const std::string&, const std::vector<std::string>& args);
void Command_Debug(const std::string&, const std::vector<std::string>& args);
void Command_Version(const std::string&, const std::vector<std::string>& args);
void Autocomplete_Lua(const std::string&, std::vector<std::string>& suggestions);
void Autocomplete_Kick(const std::string&, std::vector<std::string>& suggestions);
void Autocomplete_Settings(const std::string&, std::vector<std::string>& suggestions);
void Command_Say(const std::string& FullCommand); void Command_Say(const std::string& FullCommand);
bool EnsureArgsCount(const std::vector<std::string>& args, size_t n); bool EnsureArgsCount(const std::vector<std::string>& args, size_t n);
@@ -58,17 +52,9 @@ private:
{ "status", [this](const auto& a, const auto& b) { Command_Status(a, b); } }, { "status", [this](const auto& a, const auto& b) { Command_Status(a, b); } },
{ "settings", [this](const auto& a, const auto& b) { Command_Settings(a, b); } }, { "settings", [this](const auto& a, const auto& b) { Command_Settings(a, b); } },
{ "clear", [this](const auto& a, const auto& b) { Command_Clear(a, b); } }, { "clear", [this](const auto& a, const auto& b) { Command_Clear(a, b); } },
{ "debug", [this](const auto& a, const auto& b) { Command_Debug(a, b); } },
{ "version", [this](const auto& a, const auto& b) { Command_Version(a, b); } },
{ "say", [this](const auto&, const auto&) { Command_Say(""); } }, // shouldn't actually be called { "say", [this](const auto&, const auto&) { Command_Say(""); } }, // shouldn't actually be called
}; };
std::unordered_map<std::string, std::function<void(const std::string&, std::vector<std::string>&)>> mCommandAutocompleteMap = {
{ "lua", [this](const auto& a, auto& b) { Autocomplete_Lua(a, b); } },
{ "kick", [this](const auto& a, auto& b) { Autocomplete_Kick(a, b); } },
{ "settings", [this](const auto& a, auto& b) { Autocomplete_Settings(a, b); } },
};
Commandline mCommandline; Commandline mCommandline;
std::vector<std::string> mCachedLuaHistory; std::vector<std::string> mCachedLuaHistory;
std::vector<std::string> mCachedRegularHistory; std::vector<std::string> mCachedRegularHistory;
+17 -40
View File
@@ -10,7 +10,6 @@
#include <lua.hpp> #include <lua.hpp>
#include <memory> #include <memory>
#include <mutex> #include <mutex>
#include <nlohmann/json.hpp>
#include <queue> #include <queue>
#include <random> #include <random>
#include <set> #include <set>
@@ -19,40 +18,19 @@
#include <vector> #include <vector>
#define SOL_ALL_SAFETIES_ON 1 #define SOL_ALL_SAFETIES_ON 1
#define SOL_USER_C_ASSERT SOL_ON
#define SOL_C_ASSERT(...) \
do { \
if (!(__VA_ARGS__)) { \
beammp_lua_errorf("SOL2 assertion failure: Assertion `{}` failed in {}:{}. This *should* be a fatal error, but BeamMP Server overrides it to not be fatal. This may cause the Lua Engine to crash, or cause other issues.", #__VA_ARGS__, __FILE__, __LINE__); \
std::abort(); \
} \
} while (0)
#include <sol/sol.hpp> #include <sol/sol.hpp>
struct JsonString {
std::string value;
};
// value used to keep nils in a table or array, across serialization boundaries like
// JsonEncode, so that the nil stays at the same index and isn't treated like a special
// value (e.g. one that can be ignored or discarded).
const inline std::string BEAMMP_INTERNAL_NIL = "BEAMMP_SERVER_INTERNAL_NIL_VALUE";
using TLuaStateId = std::string; using TLuaStateId = std::string;
namespace fs = std::filesystem; namespace fs = std::filesystem;
/** /**
* std::variant means, that TLuaArgTypes may be one of the Types listed as template args * std::variant means, that TLuaArgTypes may be one of the Types listed as template args
*/ */
using TLuaValue = std::variant<std::string, int, JsonString, bool, std::unordered_map<std::string, std::string>, std::unordered_map<std::string, size_t>, float>; using TLuaArgTypes = std::variant<std::string, int, sol::variadic_args, bool, std::unordered_map<std::string, std::string>>;
enum TLuaType { static constexpr size_t TLuaArgTypes_String = 0;
String = 0, static constexpr size_t TLuaArgTypes_Int = 1;
Int = 1, static constexpr size_t TLuaArgTypes_VariadicArgs = 2;
Json = 2, static constexpr size_t TLuaArgTypes_Bool = 3;
Bool = 3, static constexpr size_t TLuaArgTypes_StringStringMap = 4;
StringStringMap = 4,
StringSizeTMap = 5,
Float = 6,
};
class TLuaPlugin; class TLuaPlugin;
@@ -93,7 +71,7 @@ public:
struct QueuedFunction { struct QueuedFunction {
std::string FunctionName; std::string FunctionName;
std::shared_ptr<TLuaResult> Result; std::shared_ptr<TLuaResult> Result;
std::vector<TLuaValue> Args; std::vector<TLuaArgTypes> Args;
std::string EventName; // optional, may be empty std::string EventName; // optional, may be empty
}; };
@@ -120,8 +98,8 @@ public:
return mLuaStates.size(); return mLuaStates.size();
} }
std::vector<std::string> GetLuaStateNames() { std::vector<std::string> GetLuaStateNames() {
std::vector<std::string> names {}; std::vector<std::string> names{};
for (auto const& [stateId, _] : mLuaStates) { for(auto const& [stateId, _ ] : mLuaStates) {
names.push_back(stateId); names.push_back(stateId);
} }
return names; return names;
@@ -142,11 +120,11 @@ public:
} }
static void WaitForAll(std::vector<std::shared_ptr<TLuaResult>>& Results, static void WaitForAll(std::vector<std::shared_ptr<TLuaResult>>& Results,
const std::optional<TimeType::duration>& Max = std::nullopt); const std::optional<std::chrono::high_resolution_clock::duration>& Max = std::nullopt);
void ReportErrors(const std::vector<std::shared_ptr<TLuaResult>>& Results); void ReportErrors(const std::vector<std::shared_ptr<TLuaResult>>& Results);
bool HasState(TLuaStateId StateId); bool HasState(TLuaStateId StateId);
[[nodiscard]] std::shared_ptr<TLuaResult> EnqueueScript(TLuaStateId StateID, const TLuaChunk& Script); [[nodiscard]] std::shared_ptr<TLuaResult> EnqueueScript(TLuaStateId StateID, const TLuaChunk& Script);
[[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCall(TLuaStateId StateID, const std::string& FunctionName, const std::vector<TLuaValue>& Args); [[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCall(TLuaStateId StateID, const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args);
void EnsureStateExists(TLuaStateId StateId, const std::string& Name, bool DontCallOnInit = false); void EnsureStateExists(TLuaStateId StateId, const std::string& Name, bool DontCallOnInit = false);
void RegisterEvent(const std::string& EventName, TLuaStateId StateId, const std::string& FunctionName); void RegisterEvent(const std::string& EventName, TLuaStateId StateId, const std::string& FunctionName);
/** /**
@@ -166,7 +144,7 @@ public:
} }
std::vector<std::shared_ptr<TLuaResult>> Results; std::vector<std::shared_ptr<TLuaResult>> Results;
std::vector<TLuaValue> Arguments { TLuaValue { std::forward<ArgsT>(Args) }... }; std::vector<TLuaArgTypes> Arguments { TLuaArgTypes { std::forward<ArgsT>(Args) }... };
for (const auto& Event : mLuaEvents.at(EventName)) { for (const auto& Event : mLuaEvents.at(EventName)) {
for (const auto& Function : Event.second) { for (const auto& Function : Event.second) {
@@ -185,7 +163,7 @@ public:
return {}; return {};
} }
std::vector<std::shared_ptr<TLuaResult>> Results; std::vector<std::shared_ptr<TLuaResult>> Results;
std::vector<TLuaValue> Arguments { TLuaValue { std::forward<ArgsT>(Args) }... }; std::vector<TLuaArgTypes> Arguments { TLuaArgTypes { std::forward<ArgsT>(Args) }... };
const auto Handlers = GetEventHandlersForState(EventName, StateId); const auto Handlers = GetEventHandlersForState(EventName, StateId);
for (const auto& Handler : Handlers) { for (const auto& Handler : Handlers) {
Results.push_back(EnqueueFunctionCall(StateId, Handler, Arguments)); Results.push_back(EnqueueFunctionCall(StateId, Handler, Arguments));
@@ -222,8 +200,8 @@ private:
StateThreadData(const StateThreadData&) = delete; StateThreadData(const StateThreadData&) = delete;
virtual ~StateThreadData() noexcept { beammp_debug("\"" + mStateId + "\" destroyed"); } virtual ~StateThreadData() noexcept { beammp_debug("\"" + mStateId + "\" destroyed"); }
[[nodiscard]] std::shared_ptr<TLuaResult> EnqueueScript(const TLuaChunk& Script); [[nodiscard]] std::shared_ptr<TLuaResult> EnqueueScript(const TLuaChunk& Script);
[[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCall(const std::string& FunctionName, const std::vector<TLuaValue>& Args); [[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCall(const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args);
[[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCallFromCustomEvent(const std::string& FunctionName, const std::vector<TLuaValue>& Args, const std::string& EventName, CallStrategy Strategy); [[nodiscard]] std::shared_ptr<TLuaResult> EnqueueFunctionCallFromCustomEvent(const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args, const std::string& EventName, CallStrategy Strategy);
void RegisterEvent(const std::string& EventName, const std::string& FunctionName); void RegisterEvent(const std::string& EventName, const std::string& FunctionName);
void AddPath(const fs::path& Path); // to be added to path and cpath void AddPath(const fs::path& Path); // to be added to path and cpath
void operator()() override; void operator()() override;
@@ -265,12 +243,11 @@ private:
std::recursive_mutex mPathsMutex; std::recursive_mutex mPathsMutex;
std::mt19937 mMersenneTwister; std::mt19937 mMersenneTwister;
std::uniform_real_distribution<double> mUniformRealDistribution01; std::uniform_real_distribution<double> mUniformRealDistribution01;
std::vector<sol::object> JsonStringToArray(JsonString Str);
}; };
struct TimedEvent { struct TimedEvent {
TimeType::duration Duration {}; std::chrono::high_resolution_clock::duration Duration {};
TimeType::time_point LastCompletion {}; std::chrono::high_resolution_clock::time_point LastCompletion {};
std::string EventName; std::string EventName;
TLuaStateId StateId; TLuaStateId StateId;
CallStrategy Strategy; CallStrategy Strategy;
+1 -3
View File
@@ -37,7 +37,6 @@ private:
TResourceManager& mResourceManager; TResourceManager& mResourceManager;
std::thread mUDPThread; std::thread mUDPThread;
std::thread mTCPThread; std::thread mTCPThread;
std::mutex mOpenIDMutex;
std::vector<uint8_t> UDPRcvFromClient(ip::udp::endpoint& ClientEndpoint); std::vector<uint8_t> UDPRcvFromClient(ip::udp::endpoint& ClientEndpoint);
void HandleDownload(TConnection&& TCPSock); void HandleDownload(TConnection&& TCPSock);
@@ -46,8 +45,7 @@ private:
void Looper(const std::weak_ptr<TClient>& c); void Looper(const std::weak_ptr<TClient>& c);
int OpenID(); int OpenID();
void OnDisconnect(const std::weak_ptr<TClient>& ClientPtr); void OnDisconnect(const std::weak_ptr<TClient>& ClientPtr);
ModMap GetClientMods(TClient& Client); void Parse(TClient& c, const std::vector<uint8_t>& Packet);
void HandleResourcePackets(TClient& c, const std::vector<uint8_t>& Packet);
void SendFile(TClient& c, const std::string& Name); void SendFile(TClient& c, const std::string& Name);
static bool TCPSendRaw(TClient& C, ip::tcp::socket& socket, const uint8_t* Data, size_t Size); static bool TCPSendRaw(TClient& C, ip::tcp::socket& socket, const uint8_t* Data, size_t Size);
static void SplitLoad(TClient& c, size_t Sent, size_t Size, bool D, const std::string& Name); static void SplitLoad(TClient& c, size_t Sent, size_t Size, bool D, const std::string& Name);
+10 -11
View File
@@ -1,22 +1,21 @@
#pragma once #pragma once
#include "Common.h" #include "Common.h"
#include <optional>
using ModMap = HashMap<std::string, size_t>;
class TResourceManager { class TResourceManager {
public: public:
TResourceManager(); TResourceManager();
[[nodiscard]] size_t TotalModsSize() const { return mTotalModSize; } [[nodiscard]] size_t MaxModSize() const { return mMaxModSize; }
[[nodiscard]] ModMap FileMap() const { return mMods; } [[nodiscard]] std::string FileList() const { return mFileList; }
[[nodiscard]] static std::string FormatForBackend(const ModMap& mods); [[nodiscard]] std::string TrimmedList() const { return mTrimmedList; }
[[nodiscard]] static std::string FormatForClient(const ModMap& mods); [[nodiscard]] std::string FileSizes() const { return mFileSizes; }
[[nodiscard]] static std::optional<std::string> IsModValid(std::string& pathString, const ModMap& mods); [[nodiscard]] int ModsLoaded() const { return mModsLoaded; }
[[nodiscard]] int LoadedModCount() const { return mMods.size(); }
private: private:
size_t mTotalModSize = 0; // size of all mods size_t mMaxModSize = 0;
ModMap mMods; // map of mod names std::string mFileSizes;
std::string mFileList;
std::string mTrimmedList;
int mModsLoaded = 0;
}; };
+2 -4
View File
@@ -4,8 +4,6 @@
#include <functional> #include <functional>
#include <string> #include <string>
#include "Common.h"
class TScopedTimer { class TScopedTimer {
public: public:
TScopedTimer(); TScopedTimer();
@@ -13,7 +11,7 @@ public:
TScopedTimer(std::function<void(size_t)> OnDestroy); TScopedTimer(std::function<void(size_t)> OnDestroy);
~TScopedTimer(); ~TScopedTimer();
auto GetElapsedTime() const { auto GetElapsedTime() const {
auto EndTime = TimeType::now(); auto EndTime = std::chrono::high_resolution_clock::now();
auto Delta = EndTime - mStartTime; auto Delta = EndTime - mStartTime;
size_t TimeDelta = Delta / std::chrono::milliseconds(1); size_t TimeDelta = Delta / std::chrono::milliseconds(1);
return TimeDelta; return TimeDelta;
@@ -22,6 +20,6 @@ public:
std::function<void(size_t /* time_ms */)> OnDestroy { nullptr }; std::function<void(size_t /* time_ms */)> OnDestroy { nullptr };
private: private:
TimeType::time_point mStartTime; std::chrono::high_resolution_clock::time_point mStartTime;
std::string Name; std::string Name;
}; };
+3 -57
View File
@@ -1,14 +1,11 @@
#pragma once #pragma once
#include "IThreaded.h" #include "IThreaded.h"
#include "IterationDecision.h"
#include "RWMutex.h" #include "RWMutex.h"
#include "TScopedTimer.h" #include "TScopedTimer.h"
#include <concepts>
#include <functional> #include <functional>
#include <memory> #include <memory>
#include <mutex> #include <mutex>
#include <type_traits>
#include <unordered_set> #include <unordered_set>
#include "BoostAliases.h" #include "BoostAliases.h"
@@ -17,21 +14,6 @@ class TClient;
class TNetwork; class TNetwork;
class TPPSMonitor; class TPPSMonitor;
// clang-format doesn't know how to deal with concepts
// clang-format off
template <typename FnT>
concept ForEachHandlerWithDecision = requires(FnT Fn, const std::shared_ptr<TClient>& Ptr) {
requires std::invocable<FnT, const std::shared_ptr<TClient>&> ;
{ std::invoke(Fn, Ptr) } -> std::convertible_to<IterationDecision>;
};
template <typename FnT>
concept ForEachHandler = requires(FnT Fn, const std::shared_ptr<TClient>& Ptr) {
requires std::invocable <FnT, const std::shared_ptr<TClient>&> ;
{ std::invoke(Fn, Ptr) } -> std::same_as<void>;
};
// clang-format on
class TServer final { class TServer final {
public: public:
using TClientSet = std::unordered_set<std::shared_ptr<TClient>>; using TClientSet = std::unordered_set<std::shared_ptr<TClient>>;
@@ -40,46 +22,11 @@ public:
void InsertClient(const std::shared_ptr<TClient>& Ptr); void InsertClient(const std::shared_ptr<TClient>& Ptr);
void RemoveClient(const std::weak_ptr<TClient>&); void RemoveClient(const std::weak_ptr<TClient>&);
void RemoveClientById(int Id);
// in Fn, return true to continue, return false to break // in Fn, return true to continue, return false to break
[[deprecated("use ForEachClient instead")]] void ForEachClientWeak(const std::function<bool(std::weak_ptr<TClient>)>& Fn); void ForEachClient(const std::function<bool(std::weak_ptr<TClient>)>& Fn);
// in Fn, return Break or Continue
template <ForEachHandlerWithDecision FnT>
void ForEachClient(FnT Fn) {
decltype(mClients) Clients;
{
ReadLock lock(mClientsMutex);
Clients = mClients;
}
for (auto& Client : Clients) {
if (Client) [[likely]] {
IterationDecision Decision = std::invoke(Fn, Client);
if (Decision == IterationDecision::Break) {
break;
}
} else {
beammp_assert_not_reachable();
}
}
}
template <ForEachHandler FnT>
void ForEachClient(FnT Fn) {
decltype(mClients) Clients;
{
ReadLock lock(mClientsMutex);
Clients = mClients;
}
for (auto& Client : Clients) {
if (Client) [[likely]] {
std::invoke(Fn, Client);
} else {
beammp_assert_not_reachable();
}
}
}
size_t ClientCount() const; size_t ClientCount() const;
static void GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uint8_t>&& Packet, TNetwork& Network); static void GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uint8_t>&& Packet, TPPSMonitor& PPSMonitor, TNetwork& Network);
static void HandleEvent(TClient& c, const std::string& Data); static void HandleEvent(TClient& c, const std::string& Data);
RWMutex& GetClientMutex() const { return mClientsMutex; } RWMutex& GetClientMutex() const { return mClientsMutex; }
@@ -96,8 +43,7 @@ private:
static bool ShouldSpawn(TClient& c, const std::string& CarJson, int ID); static bool ShouldSpawn(TClient& c, const std::string& CarJson, int ID);
static bool IsUnicycle(TClient& c, const std::string& CarJson); static bool IsUnicycle(TClient& c, const std::string& CarJson);
static void Apply(TClient& c, int VID, const std::string& pckt); static void Apply(TClient& c, int VID, const std::string& pckt);
static bool HandlePosition(TClient& c, const std::string& PacketStr); static void HandlePosition(TClient& c, const std::string& Packet);
static bool HandleVehicleUpdate(const std::string& PacketStr, const int playerID);
}; };
struct BufferView { struct BufferView {
-9
View File
@@ -1,9 +0,0 @@
#pragma once
#include <string>
namespace uuid {
std::string GenerateUuid();
}
-3
View File
@@ -1,3 +0,0 @@
FROM archlinux
RUN pacman -Syu --noconfirm lua53 openssl curl git cmake gcc make zlib boost websocketpp
-9
View File
@@ -1,9 +0,0 @@
FROM debian:11
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update -y
RUN apt-get install -y git cmake g++-10 curl libboost1.74-all-dev libssl-dev libz-dev rapidjson-dev liblua5.3 libssl-dev libwebsocketpp-dev libcurl4-openssl-dev
ENV CXX=g++-10
-9
View File
@@ -1,9 +0,0 @@
FROM ubuntu:20.04
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update -y
RUN apt-get install -y libz-dev rapidjson-dev liblua5.3 libssl-dev libwebsocketpp-dev libcurl4-openssl-dev git make cmake g++
RUN apt-get install -y libboost1.71-all-dev
-9
View File
@@ -1,9 +0,0 @@
FROM ubuntu:22.04
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update -y
RUN apt-get install -y git libz-dev rapidjson-dev liblua5.3 libssl-dev libwebsocketpp-dev libcurl4-openssl-dev cmake g++-10 libboost1.74-all-dev libssl3 curl
ENV CXX=g++-10
-18
View File
@@ -1,18 +0,0 @@
#!/bin/sh
set -e
printf "enter DSN (optional): "
read DSN
docker build -f Ubuntu-20.04-Dockerfile . -t beammp-server-build:Ubuntu-20.04
docker build -f Ubuntu-22.04-Dockerfile . -t beammp-server-build:Ubuntu-22.04
docker build -f ArchLinux-Dockerfile . -t beammp-server-build:ArchLinux
docker build -f Debian-11-Dockerfile . -t beammp-server-build:Debian-11
CMD="cd /beammp; cmake . -DGIT_SUBMODULE=OFF -DCMAKE_BUILD_TYPE=Release -DBEAMMP_SECRET_SENTRY_URL=\"${DSN}\" -B /build && make -j -C /build BeamMP-Server"
docker run -v $(pwd)/..:/beammp -v $(pwd)/../build-ubuntu-20.04:/build -it --rm beammp-server-build:Ubuntu-20.04 bash -c "${CMD}"
docker run -v $(pwd)/..:/beammp -v $(pwd)/../build-ubuntu-22.04:/build -it --rm beammp-server-build:Ubuntu-22.04 bash -c "${CMD}"
docker run -v $(pwd)/..:/beammp -v $(pwd)/../build-archlinux:/build -it --rm beammp-server-build:ArchLinux bash -c "${CMD}"
docker run -v $(pwd)/..:/beammp -v $(pwd)/../build-debian-11:/build -it --rm beammp-server-build:Debian-11 bash -c "${CMD}"
-5
View File
@@ -1,5 +0,0 @@
#!/bin/sh
cmake . -B build -DCMAKE_BUILD_TYPE=Release -DGIT_SUBMODULE=OFF
make -j -C build BeamMP-Server
+6 -6
View File
@@ -13,7 +13,7 @@ void ArgsParser::Parse(const std::vector<std::string_view>& ArgList) {
ConsumeLongFlag(std::string(Arg)); ConsumeLongFlag(std::string(Arg));
} }
} else { } else {
fmt::print(stderr, "Error parsing commandline arguments: Supplied argument '{}' is not a valid argument and was ignored.", Arg); beammp_errorf("Error parsing commandline arguments: Supplied argument '{}' is not a valid argument and was ignored.", Arg);
} }
} }
} }
@@ -22,17 +22,17 @@ bool ArgsParser::Verify() {
bool Ok = true; bool Ok = true;
for (const auto& RegisteredArg : mRegisteredArguments) { for (const auto& RegisteredArg : mRegisteredArguments) {
if (RegisteredArg.Flags & Flags::REQUIRED && !FoundArgument(RegisteredArg.Names)) { if (RegisteredArg.Flags & Flags::REQUIRED && !FoundArgument(RegisteredArg.Names)) {
fmt::print(stderr, "Error in commandline arguments: Argument '{}' is required but wasn't found.", RegisteredArg.Names.at(0)); beammp_errorf("Error in commandline arguments: Argument '{}' is required but wasn't found.", RegisteredArg.Names.at(0));
Ok = false; Ok = false;
continue; continue;
} else if (FoundArgument(RegisteredArg.Names)) { } else if (FoundArgument(RegisteredArg.Names)) {
if (RegisteredArg.Flags & Flags::HAS_VALUE) { if (RegisteredArg.Flags & Flags::HAS_VALUE) {
if (!GetValueOfArgument(RegisteredArg.Names).has_value()) { if (!GetValueOfArgument(RegisteredArg.Names).has_value()) {
fmt::print(stderr, "Error in commandline arguments: Argument '{}' expects a value, but no value was given.", RegisteredArg.Names.at(0)); beammp_error("Error in commandline arguments: Argument '" + std::string(RegisteredArg.Names.at(0)) + "' expects a value, but no value was given.");
Ok = false; Ok = false;
} }
} else if (GetValueOfArgument(RegisteredArg.Names).has_value()) { } else if (GetValueOfArgument(RegisteredArg.Names).has_value()) {
fmt::print(stderr, "Error in commandline arguments: Argument '{}' does not expect a value, but one was given.", RegisteredArg.Names.at(0)); beammp_error("Error in commandline arguments: Argument '" + std::string(RegisteredArg.Names.at(0)) + "' does not expect a value, but one was given.");
Ok = false; Ok = false;
} }
} }
@@ -81,7 +81,7 @@ void ArgsParser::ConsumeLongAssignment(const std::string& Arg) {
auto Value = Arg.substr(Arg.rfind("=") + 1); auto Value = Arg.substr(Arg.rfind("=") + 1);
auto Name = Arg.substr(2, Arg.rfind("=") - 2); auto Name = Arg.substr(2, Arg.rfind("=") - 2);
if (!IsRegistered(Name)) { if (!IsRegistered(Name)) {
fmt::print(stdout, "Argument '{}' was supplied but isn't a known argument, so it is likely being ignored.", Name); beammp_warn("Argument '" + Name + "' was supplied but isn't a known argument, so it is likely being ignored.");
} }
mFoundArgs.push_back({ Name, Value }); mFoundArgs.push_back({ Name, Value });
} }
@@ -90,7 +90,7 @@ void ArgsParser::ConsumeLongFlag(const std::string& Arg) {
auto Name = Arg.substr(2, Arg.rfind("=") - 2); auto Name = Arg.substr(2, Arg.rfind("=") - 2);
mFoundArgs.push_back({ Name, std::nullopt }); mFoundArgs.push_back({ Name, std::nullopt });
if (!IsRegistered(Name)) { if (!IsRegistered(Name)) {
fmt::print(stdout, "Argument '{}' was supplied but isn't a known argument, so it is likely being ignored.", Name); beammp_warn("Argument '" + Name + "' was supplied but isn't a known argument, so it is likely being ignored.");
} }
} }
+15 -24
View File
@@ -1,7 +1,6 @@
#include "Client.h" #include "Client.h"
#include "CustomAssert.h" #include "CustomAssert.h"
#include "LuaAPI.h"
#include "TServer.h" #include "TServer.h"
#include <memory> #include <memory>
#include <optional> #include <optional>
@@ -13,8 +12,6 @@ void TClient::DeleteCar(int Ident) {
return Ident == elem.ID(); return Ident == elem.ID();
}); });
if (iter != mVehicleData.end()) { if (iter != mVehicleData.end()) {
std::string Destroy = "Od:" + std::to_string(GetID()) + "-" + std::to_string(iter->ID());
LuaAPI::MP::Engine->Network().SendToAll(nullptr, StringToVector(Destroy), true, true);
mVehicleData.erase(iter); mVehicleData.erase(iter);
} else { } else {
beammp_debug("tried to erase a vehicle that doesn't exist (not an error)"); beammp_debug("tried to erase a vehicle that doesn't exist (not an error)");
@@ -23,10 +20,6 @@ void TClient::DeleteCar(int Ident) {
void TClient::ClearCars() { void TClient::ClearCars() {
std::unique_lock lock(mVehicleDataMutex); std::unique_lock lock(mVehicleDataMutex);
for (const auto& Car : mVehicleData) {
std::string Destroy = "Od:" + std::to_string(GetID()) + "-" + std::to_string(Car.ID());
LuaAPI::MP::Engine->Network().SendToAll(this, StringToVector(Destroy), false, true);
}
mVehicleData.clear(); mVehicleData.clear();
} }
@@ -66,12 +59,7 @@ std::string TClient::GetCarPositionRaw(int Ident) {
} }
void TClient::Disconnect(std::string_view Reason) { void TClient::Disconnect(std::string_view Reason) {
// we need exclusivity here in case we can't
std::unique_lock Lock(mDisconnectMtx);
beammp_debugf("Disconnecting client {} for reason: {}", GetID(), Reason); beammp_debugf("Disconnecting client {} for reason: {}", GetID(), Reason);
if (mSocket.is_open()) {
beammp_debugf("Socket for client {} already closed", GetID());
}
boost::system::error_code ec; boost::system::error_code ec;
mSocket.shutdown(socket_base::shutdown_both, ec); mSocket.shutdown(socket_base::shutdown_both, ec);
if (ec) { if (ec) {
@@ -81,7 +69,6 @@ void TClient::Disconnect(std::string_view Reason) {
if (ec) { if (ec) {
beammp_debugf("Failed to close client socket: {}", ec.message()); beammp_debugf("Failed to close client socket: {}", ec.message());
} }
mServer.RemoveClientById(GetID());
} }
void TClient::SetCarPosition(int Ident, const std::string& Data) { void TClient::SetCarPosition(int Ident, const std::string& Data) {
@@ -132,7 +119,7 @@ TClient::TClient(TServer& Server, ip::tcp::socket&& Socket)
: mServer(Server) : mServer(Server)
, mSocket(std::move(Socket)) , mSocket(std::move(Socket))
, mDownSocket(ip::tcp::socket(Server.IoCtx())) , mDownSocket(ip::tcp::socket(Server.IoCtx()))
, mLastPingTime(TimeType::now()) { , mLastPingTime(std::chrono::high_resolution_clock::now()) {
} }
TClient::~TClient() { TClient::~TClient() {
@@ -140,23 +127,27 @@ TClient::~TClient() {
} }
void TClient::UpdatePingTime() { void TClient::UpdatePingTime() {
mLastPingTime = TimeType::now(); mLastPingTime = std::chrono::high_resolution_clock::now();
} }
int TClient::SecondsSinceLastPing() { int TClient::SecondsSinceLastPing() {
auto seconds = std::chrono::duration_cast<std::chrono::seconds>( auto seconds = std::chrono::duration_cast<std::chrono::seconds>(
TimeType::now() - mLastPingTime) std::chrono::high_resolution_clock::now() - mLastPingTime)
.count(); .count();
return int(seconds); return int(seconds);
} }
std::shared_ptr<TClient> GetClient(TServer& Server, int ID) { std::optional<std::weak_ptr<TClient>> GetClient(TServer& Server, int ID) {
std::shared_ptr<TClient> Result {}; std::optional<std::weak_ptr<TClient>> MaybeClient { std::nullopt };
Server.ForEachClient([&](const auto& Client) { Server.ForEachClient([&](std::weak_ptr<TClient> CPtr) -> bool {
if (Client->GetID() == ID) { ReadLock Lock(Server.GetClientMutex());
Result = Client; if (!CPtr.expired()) {
return Break; auto C = CPtr.lock();
if (C->GetID() == ID) {
MaybeClient = CPtr;
return false;
}
} }
return Continue; return true;
}); });
return Result; return MaybeClient;
} }
+8 -97
View File
@@ -13,85 +13,10 @@
#include "CustomAssert.h" #include "CustomAssert.h"
#include "Http.h" #include "Http.h"
Application::SettingsMap Application::mSettings = {
{ StrName, std::string("BeamMP Server") },
{ StrDescription, std::string("No description") },
{ StrResourceFolder, std::string("Resources") },
{ StrMap, std::string("/levels/gridmap_v2/info.json") },
{ StrSSLKeyPath, std::string("./.ssl/HttpServer/key.pem") },
{ StrSSLCertPath, std::string("./.ssl/HttpServer/cert.pem") },
{ StrHTTPServerEnabled, false },
{ StrMaxPlayers, int(8) },
{ StrPrivate, true },
{ StrMaxCars, int(1) },
{ StrDebug, false },
{ StrPort, int(30814) },
{ StrCustomIP, std::string("") },
{ StrLogChat, true },
{ StrSendErrors, true },
{ StrSendErrorsMessageEnabled, true },
{ StrHTTPServerPort, int(8080) },
{ StrHTTPServerIP, std::string("127.0.0.1") },
{ StrHTTPServerUseSSL, false },
{ StrHideUpdateMessages, false },
{ StrAuthKey, std::string("") },
{ StrIncludeSubdirectories, false },
};
// global, yes, this is ugly, no, it cant be done another way // global, yes, this is ugly, no, it cant be done another way
TSentry Sentry {}; TSentry Sentry {};
std::string Application::SettingToString(const Application::SettingValue& Value) { Application::TSettings Application::Settings = {};
switch (Value.index()) {
case 0:
return fmt::format("{}", std::get<std::string>(Value));
case 1:
return fmt::format("{}", std::get<bool>(Value));
case 2:
return fmt::format("{}", std::get<int>(Value));
default:
return "<unknown type>";
}
}
std::string Application::GetSettingString(std::string_view Name) {
try {
return std::get<std::string>(Application::mSettings.at(Name));
} catch (const std::exception& e) {
beammp_errorf("Failed to get string setting '{}': {}", Name, e.what());
return "";
}
}
int Application::GetSettingInt(std::string_view Name) {
try {
return std::get<int>(Application::mSettings.at(Name));
} catch (const std::exception& e) {
beammp_errorf("Failed to get int setting '{}': {}", Name, e.what());
return 0;
}
}
bool Application::GetSettingBool(std::string_view Name) {
try {
return std::get<bool>(Application::mSettings.at(Name));
} catch (const std::exception& e) {
beammp_errorf("Failed to get bool setting '{}': {}", Name, e.what());
return false;
}
}
void Application::SetSetting(std::string_view Name, const Application::SettingValue& Value) {
if (mSettings.contains(Name)) {
if (mSettings[Name].index() == Value.index()) {
mSettings[Name] = Value;
} else {
beammp_errorf("Could not change value of setting '{}', because it has a different type.", Name);
}
} else {
mSettings[Name] = Value;
}
}
void Application::RegisterShutdownHandler(const TShutdownHandler& Handler) { void Application::RegisterShutdownHandler(const TShutdownHandler& Handler) {
std::unique_lock Lock(mShutdownHandlersMutex); std::unique_lock Lock(mShutdownHandlersMutex);
@@ -314,7 +239,7 @@ static std::mutex ThreadNameMapMutex {};
std::string ThreadName(bool DebugModeOverride) { std::string ThreadName(bool DebugModeOverride) {
auto Lock = std::unique_lock(ThreadNameMapMutex); auto Lock = std::unique_lock(ThreadNameMapMutex);
if (DebugModeOverride || Application::GetSettingBool(StrDebug)) { if (DebugModeOverride || Application::Settings.DebugModeEnabled) {
auto id = std::this_thread::get_id(); auto id = std::this_thread::get_id();
if (threadNameMap.find(id) != threadNameMap.end()) { if (threadNameMap.find(id) != threadNameMap.end()) {
// found // found
@@ -326,21 +251,21 @@ std::string ThreadName(bool DebugModeOverride) {
TEST_CASE("ThreadName") { TEST_CASE("ThreadName") {
RegisterThread("MyThread"); RegisterThread("MyThread");
auto OrigDebug = Application::GetSettingBool(StrDebug); auto OrigDebug = Application::Settings.DebugModeEnabled;
// ThreadName adds a space at the end, legacy but we need it still // ThreadName adds a space at the end, legacy but we need it still
SUBCASE("Debug mode enabled") { SUBCASE("Debug mode enabled") {
Application::SetSetting(StrDebug, true); Application::Settings.DebugModeEnabled = true;
CHECK(ThreadName(true) == "MyThread "); CHECK(ThreadName(true) == "MyThread ");
CHECK(ThreadName(false) == "MyThread "); CHECK(ThreadName(false) == "MyThread ");
} }
SUBCASE("Debug mode disabled") { SUBCASE("Debug mode disabled") {
Application::SetSetting(StrDebug, false); Application::Settings.DebugModeEnabled = false;
CHECK(ThreadName(true) == "MyThread "); CHECK(ThreadName(true) == "MyThread ");
CHECK(ThreadName(false) == ""); CHECK(ThreadName(false) == "");
} }
// cleanup // cleanup
Application::SetSetting(StrDebug, OrigDebug); Application::Settings.DebugModeEnabled = OrigDebug;
} }
void RegisterThread(const std::string& str) { void RegisterThread(const std::string& str) {
@@ -352,7 +277,7 @@ void RegisterThread(const std::string& str) {
#elif defined(BEAMMP_LINUX) #elif defined(BEAMMP_LINUX)
ThreadId = std::to_string(gettid()); ThreadId = std::to_string(gettid());
#endif #endif
if (Application::GetSettingBool(StrDebug)) { if (Application::Settings.DebugModeEnabled) {
std::ofstream ThreadFile(".Threads.log", std::ios::app); std::ofstream ThreadFile(".Threads.log", std::ios::app);
ThreadFile << ("Thread \"" + str + "\" is TID " + ThreadId) << std::endl; ThreadFile << ("Thread \"" + str + "\" is TID " + ThreadId) << std::endl;
} }
@@ -385,7 +310,7 @@ TEST_CASE("Version::AsString") {
} }
void LogChatMessage(const std::string& name, int id, const std::string& msg) { void LogChatMessage(const std::string& name, int id, const std::string& msg) {
if (Application::GetSettingBool(StrLogChat)) { if (Application::Settings.LogChat) {
std::stringstream ss; std::stringstream ss;
ss << ThreadName(); ss << ThreadName();
ss << "[CHAT] "; ss << "[CHAT] ";
@@ -429,17 +354,3 @@ std::string GetPlatformAgnosticErrorString() {
return "(no human-readable errors on this platform)"; return "(no human-readable errors on this platform)";
#endif #endif
} }
std::string ToHumanReadableSize(size_t Size) {
if (Size > TB) {
return fmt::format("{:.2f} TiB", double(Size) / TB);
} else if (Size > GB) {
return fmt::format("{:.2f} GiB", double(Size) / GB);
} else if (Size > MB) {
return fmt::format("{:.2f} MiB", double(Size) / MB);
} else if (Size > KB) {
return fmt::format("{:.2f} KiB", double(Size) / KB);
} else {
return fmt::format("{} B", Size);
}
}
+2 -2
View File
@@ -154,7 +154,7 @@ Http::Server::THttpServerInstance::THttpServerInstance() {
} }
void Http::Server::THttpServerInstance::operator()() try { void Http::Server::THttpServerInstance::operator()() try {
beammp_info("HTTP(S) Server started on port " + std::to_string(Application::GetSettingInt(StrHTTPServerPort))); beammp_info("HTTP(S) Server started on port " + std::to_string(Application::Settings.HTTPServerPort));
std::unique_ptr<httplib::Server> HttpLibServerInstance; std::unique_ptr<httplib::Server> HttpLibServerInstance;
HttpLibServerInstance = std::make_unique<httplib::Server>(); HttpLibServerInstance = std::make_unique<httplib::Server>();
// todo: make this IP agnostic so people can set their own IP // todo: make this IP agnostic so people can set their own IP
@@ -196,7 +196,7 @@ void Http::Server::THttpServerInstance::operator()() try {
beammp_debug("Http Server: " + Req.method + " " + Req.target + " -> " + std::to_string(Res.status)); beammp_debug("Http Server: " + Req.method + " " + Req.target + " -> " + std::to_string(Res.status));
}); });
Application::SetSubsystemStatus("HTTPServer", Application::Status::Good); Application::SetSubsystemStatus("HTTPServer", Application::Status::Good);
auto ret = HttpLibServerInstance->listen(Application::GetSettingString(StrHTTPServerIP).c_str(), Application::GetSettingInt(StrHTTPServerPort)); auto ret = HttpLibServerInstance->listen(Application::Settings.HTTPServerIP.c_str(), Application::Settings.HTTPServerPort);
if (!ret) { if (!ret) {
beammp_error("Failed to start http server (failed to listen). Please ensure the http server is configured properly in the ServerConfig.toml, or turn it off if you don't need it."); beammp_error("Failed to start http server (failed to listen). Please ensure the http server is configured properly in the ServerConfig.toml, or turn it off if you don't need it.");
} }
+43 -46
View File
@@ -119,14 +119,15 @@ static inline std::pair<bool, std::string> InternalTriggerClientEvent(int Player
LuaAPI::MP::Engine->Network().SendToAll(nullptr, StringToVector(Packet), true, true); LuaAPI::MP::Engine->Network().SendToAll(nullptr, StringToVector(Packet), true, true);
return { true, "" }; return { true, "" };
} else { } else {
auto Client = GetClient(LuaAPI::MP::Engine->Server(), PlayerID); auto MaybeClient = GetClient(LuaAPI::MP::Engine->Server(), PlayerID);
if (!Client) { if (!MaybeClient || MaybeClient.value().expired()) {
beammp_lua_errorf("TriggerClientEvent invalid Player ID '{}'", PlayerID); beammp_lua_errorf("TriggerClientEvent invalid Player ID '{}'", PlayerID);
return { false, "Invalid Player ID" }; return { false, "Invalid Player ID" };
} }
if (!LuaAPI::MP::Engine->Network().Respond(*Client, StringToVector(Packet), true)) { auto c = MaybeClient.value().lock();
if (!LuaAPI::MP::Engine->Network().Respond(*c, StringToVector(Packet), true)) {
beammp_lua_errorf("Respond failed, dropping client {}", PlayerID); beammp_lua_errorf("Respond failed, dropping client {}", PlayerID);
LuaAPI::MP::Engine->Network().ClientKick(*Client, "Disconnected after failing to receive packets"); LuaAPI::MP::Engine->Network().ClientKick(*c, "Disconnected after failing to receive packets");
return { false, "Respond failed, dropping client" }; return { false, "Respond failed, dropping client" };
} }
return { true, "" }; return { true, "" };
@@ -139,12 +140,13 @@ std::pair<bool, std::string> LuaAPI::MP::TriggerClientEvent(int PlayerID, const
} }
std::pair<bool, std::string> LuaAPI::MP::DropPlayer(int ID, std::optional<std::string> MaybeReason) { std::pair<bool, std::string> LuaAPI::MP::DropPlayer(int ID, std::optional<std::string> MaybeReason) {
auto Client = GetClient(Engine->Server(), ID); auto MaybeClient = GetClient(Engine->Server(), ID);
if (!Client) { if (!MaybeClient || MaybeClient.value().expired()) {
beammp_lua_errorf("Tried to drop client with id {}, who doesn't exist", ID); beammp_lua_errorf("Tried to drop client with id {}, who doesn't exist", ID);
return { false, "Player does not exist" }; return { false, "Player does not exist" };
} }
LuaAPI::MP::Engine->Network().ClientKick(*Client, MaybeReason.value_or("No reason")); auto c = MaybeClient.value().lock();
LuaAPI::MP::Engine->Network().ClientKick(*c, MaybeReason.value_or("No reason"));
return { true, "" }; return { true, "" };
} }
@@ -156,15 +158,16 @@ std::pair<bool, std::string> LuaAPI::MP::SendChatMessage(int ID, const std::stri
Engine->Network().SendToAll(nullptr, StringToVector(Packet), true, true); Engine->Network().SendToAll(nullptr, StringToVector(Packet), true, true);
Result.first = true; Result.first = true;
} else { } else {
auto Client = GetClient(Engine->Server(), ID); auto MaybeClient = GetClient(Engine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
if (!Client->IsSynced()) { auto c = MaybeClient.value().lock();
if (!c->IsSynced()) {
Result.first = false; Result.first = false;
Result.second = "Player still syncing data"; Result.second = "Player still syncing data";
return Result; return Result;
} }
LogChatMessage("<Server> (to \"" + Client->GetName() + "\")", -1, Message); LogChatMessage("<Server> (to \"" + c->GetName() + "\")", -1, Message);
if (!Engine->Network().Respond(*Client, StringToVector(Packet), true)) { if (!Engine->Network().Respond(*c, StringToVector(Packet), true)) {
beammp_errorf("Failed to send chat message back to sender (id {}) - did the sender disconnect?", ID); beammp_errorf("Failed to send chat message back to sender (id {}) - did the sender disconnect?", ID);
// TODO: should we return an error here? // TODO: should we return an error here?
} }
@@ -181,17 +184,18 @@ std::pair<bool, std::string> LuaAPI::MP::SendChatMessage(int ID, const std::stri
std::pair<bool, std::string> LuaAPI::MP::RemoveVehicle(int PID, int VID) { std::pair<bool, std::string> LuaAPI::MP::RemoveVehicle(int PID, int VID) {
std::pair<bool, std::string> Result; std::pair<bool, std::string> Result;
auto Client = GetClient(Engine->Server(), PID); auto MaybeClient = GetClient(Engine->Server(), PID);
if (!Client) { if (!MaybeClient || MaybeClient.value().expired()) {
beammp_lua_error("RemoveVehicle invalid Player ID"); beammp_lua_error("RemoveVehicle invalid Player ID");
Result.first = false; Result.first = false;
Result.second = "Invalid Player ID"; Result.second = "Invalid Player ID";
return Result; return Result;
} }
if (!Client->GetCarData(VID).empty()) { auto c = MaybeClient.value().lock();
if (!c->GetCarData(VID).empty()) {
std::string Destroy = "Od:" + std::to_string(PID) + "-" + std::to_string(VID); std::string Destroy = "Od:" + std::to_string(PID) + "-" + std::to_string(VID);
Engine->Network().SendToAll(nullptr, StringToVector(Destroy), true, true); Engine->Network().SendToAll(nullptr, StringToVector(Destroy), true, true);
Client->DeleteCar(VID); c->DeleteCar(VID);
Result.first = true; Result.first = true;
} else { } else {
Result.first = false; Result.first = false;
@@ -204,56 +208,56 @@ void LuaAPI::MP::Set(int ConfigID, sol::object NewValue) {
switch (ConfigID) { switch (ConfigID) {
case 0: // debug case 0: // debug
if (NewValue.is<bool>()) { if (NewValue.is<bool>()) {
Application::SetSetting(StrDebug, NewValue.as<bool>()); Application::Settings.DebugModeEnabled = NewValue.as<bool>();
beammp_info(std::string("Set `Debug` to ") + (Application::GetSettingBool(StrDebug) ? "true" : "false")); beammp_info(std::string("Set `Debug` to ") + (Application::Settings.DebugModeEnabled ? "true" : "false"));
} else { } else {
beammp_lua_error("set invalid argument [2] expected boolean"); beammp_lua_error("set invalid argument [2] expected boolean");
} }
break; break;
case 1: // private case 1: // private
if (NewValue.is<bool>()) { if (NewValue.is<bool>()) {
Application::SetSetting(StrPrivate, NewValue.as<bool>()); Application::Settings.Private = NewValue.as<bool>();
beammp_info(std::string("Set `Private` to ") + (Application::GetSettingBool(StrPrivate) ? "true" : "false")); beammp_info(std::string("Set `Private` to ") + (Application::Settings.Private ? "true" : "false"));
} else { } else {
beammp_lua_error("set invalid argument [2] expected boolean"); beammp_lua_error("set invalid argument [2] expected boolean");
} }
break; break;
case 2: // max cars case 2: // max cars
if (NewValue.is<int>()) { if (NewValue.is<int>()) {
Application::SetSetting(StrMaxCars, NewValue.as<int>()); Application::Settings.MaxCars = NewValue.as<int>();
beammp_info(std::string("Set `MaxCars` to ") + std::to_string(Application::GetSettingInt(StrMaxCars))); beammp_info(std::string("Set `MaxCars` to ") + std::to_string(Application::Settings.MaxCars));
} else { } else {
beammp_lua_error("set invalid argument [2] expected integer"); beammp_lua_error("set invalid argument [2] expected integer");
} }
break; break;
case 3: // max players case 3: // max players
if (NewValue.is<int>()) { if (NewValue.is<int>()) {
Application::SetSetting(StrMaxPlayers, NewValue.as<int>()); Application::Settings.MaxPlayers = NewValue.as<int>();
beammp_info(std::string("Set `MaxPlayers` to ") + std::to_string(Application::GetSettingInt(StrMaxPlayers))); beammp_info(std::string("Set `MaxPlayers` to ") + std::to_string(Application::Settings.MaxPlayers));
} else { } else {
beammp_lua_error("set invalid argument [2] expected integer"); beammp_lua_error("set invalid argument [2] expected integer");
} }
break; break;
case 4: // Map case 4: // Map
if (NewValue.is<std::string>()) { if (NewValue.is<std::string>()) {
Application::SetSetting(StrMap, NewValue.as<std::string>()); Application::Settings.MapName = NewValue.as<std::string>();
beammp_info(std::string("Set `Map` to ") + Application::GetSettingString(StrMap)); beammp_info(std::string("Set `Map` to ") + Application::Settings.MapName);
} else { } else {
beammp_lua_error("set invalid argument [2] expected string"); beammp_lua_error("set invalid argument [2] expected string");
} }
break; break;
case 5: // Name case 5: // Name
if (NewValue.is<std::string>()) { if (NewValue.is<std::string>()) {
Application::SetSetting(StrName, NewValue.as<std::string>()); Application::Settings.ServerName = NewValue.as<std::string>();
beammp_info(std::string("Set `Name` to ") + Application::GetSettingString(StrName)); beammp_info(std::string("Set `Name` to ") + Application::Settings.ServerName);
} else { } else {
beammp_lua_error("set invalid argument [2] expected string"); beammp_lua_error("set invalid argument [2] expected string");
} }
break; break;
case 6: // Desc case 6: // Desc
if (NewValue.is<std::string>()) { if (NewValue.is<std::string>()) {
Application::SetSetting(StrDescription, NewValue.as<std::string>()); Application::Settings.ServerDesc = NewValue.as<std::string>();
beammp_info(std::string("Set `Description` to ") + Application::GetSettingString(StrDescription)); beammp_info(std::string("Set `Description` to ") + Application::Settings.ServerDesc);
} else { } else {
beammp_lua_error("set invalid argument [2] expected string"); beammp_lua_error("set invalid argument [2] expected string");
} }
@@ -269,18 +273,18 @@ void LuaAPI::MP::Sleep(size_t Ms) {
} }
bool LuaAPI::MP::IsPlayerConnected(int ID) { bool LuaAPI::MP::IsPlayerConnected(int ID) {
auto Client = GetClient(Engine->Server(), ID); auto MaybeClient = GetClient(Engine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
return Client->IsConnected(); return MaybeClient.value().lock()->IsConnected();
} else { } else {
return false; return false;
} }
} }
bool LuaAPI::MP::IsPlayerGuest(int ID) { bool LuaAPI::MP::IsPlayerGuest(int ID) {
auto Client = GetClient(Engine->Server(), ID); auto MaybeClient = GetClient(Engine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
return Client->IsGuest(); return MaybeClient.value().lock()->IsGuest();
} else { } else {
return false; return false;
} }
@@ -298,7 +302,6 @@ void LuaAPI::MP::PrintRaw(sol::variadic_args Args) {
} }
int LuaAPI::PanicHandler(lua_State* State) { int LuaAPI::PanicHandler(lua_State* State) {
// FIXME: unsafe operation, can cause stack overflow
beammp_lua_error("PANIC: " + sol::stack::get<std::string>(State, 1)); beammp_lua_error("PANIC: " + sol::stack::get<std::string>(State, 1));
return 0; return 0;
} }
@@ -569,11 +572,7 @@ static void JsonEncodeRecursive(nlohmann::json& json, const sol::object& left, c
value = right.as<std::string>(); value = right.as<std::string>();
break; break;
case sol::type::number: case sol::type::number:
if (right.is<int>()) { value = right.as<double>();
value = right.as<int>();
} else {
value = right.as<float>();
}
break; break;
case sol::type::function: case sol::type::function:
beammp_lua_warn("unsure what to do with function in JsonEncode, ignoring"); beammp_lua_warn("unsure what to do with function in JsonEncode, ignoring");
@@ -583,7 +582,6 @@ static void JsonEncodeRecursive(nlohmann::json& json, const sol::object& left, c
for (const auto& pair : right.as<sol::table>()) { for (const auto& pair : right.as<sol::table>()) {
if (pair.first.get_type() != sol::type::number) { if (pair.first.get_type() != sol::type::number) {
local_is_array = false; local_is_array = false;
break;
} }
} }
for (const auto& pair : right.as<sol::table>()) { for (const auto& pair : right.as<sol::table>()) {
@@ -608,11 +606,10 @@ std::string LuaAPI::MP::JsonEncode(const sol::table& object) {
for (const auto& pair : object.as<sol::table>()) { for (const auto& pair : object.as<sol::table>()) {
if (pair.first.get_type() != sol::type::number) { if (pair.first.get_type() != sol::type::number) {
is_array = false; is_array = false;
break;
} }
} }
for (const auto& [key, value] : object) { for (const auto& entry : object) {
JsonEncodeRecursive(json, key, value, is_array); JsonEncodeRecursive(json, entry.first, entry.second, is_array);
} }
return json.dump(); return json.dump();
} }
-2
View File
@@ -1,8 +1,6 @@
#include "SignalHandling.h" #include "SignalHandling.h"
#include "Common.h" #include "Common.h"
#include "Compat.h"
#if defined(BEAMMP_LINUX) || defined(BEAMMP_APPLE) #if defined(BEAMMP_LINUX) || defined(BEAMMP_APPLE)
#include <csignal> #include <csignal>
static void UnixSignalHandler(int sig) { static void UnixSignalHandler(int sig) {
+150 -60
View File
@@ -6,6 +6,32 @@
#include <istream> #include <istream>
#include <sstream> #include <sstream>
// General
static constexpr std::string_view StrDebug = "Debug";
static constexpr std::string_view StrPrivate = "Private";
static constexpr std::string_view StrPort = "Port";
static constexpr std::string_view StrMaxCars = "MaxCars";
static constexpr std::string_view StrMaxPlayers = "MaxPlayers";
static constexpr std::string_view StrMap = "Map";
static constexpr std::string_view StrName = "Name";
static constexpr std::string_view StrDescription = "Description";
static constexpr std::string_view StrResourceFolder = "ResourceFolder";
static constexpr std::string_view StrAuthKey = "AuthKey";
static constexpr std::string_view StrLogChat = "LogChat";
// Misc
static constexpr std::string_view StrSendErrors = "SendErrors";
static constexpr std::string_view StrSendErrorsMessageEnabled = "SendErrorsShowMessage";
static constexpr std::string_view StrHideUpdateMessages = "ImScaredOfUpdates";
// HTTP
static constexpr std::string_view StrHTTPServerEnabled = "HTTPServerEnabled";
static constexpr std::string_view StrHTTPServerUseSSL = "UseSSL";
static constexpr std::string_view StrSSLKeyPath = "SSLKeyPath";
static constexpr std::string_view StrSSLCertPath = "SSLCertPath";
static constexpr std::string_view StrHTTPServerPort = "HTTPServerPort";
static constexpr std::string_view StrHTTPServerIP = "HTTPServerIP";
TEST_CASE("TConfig::TConfig") { TEST_CASE("TConfig::TConfig") {
const std::string CfgFile = "beammp_server_testconfig.toml"; const std::string CfgFile = "beammp_server_testconfig.toml";
fs::remove(CfgFile); fs::remove(CfgFile);
@@ -67,37 +93,35 @@ void SetComment(CommentsT& Comments, const std::string& Comment) {
void TConfig::FlushToFile() { void TConfig::FlushToFile() {
// auto data = toml::parse<toml::preserve_comments>(mConfigFileName); // auto data = toml::parse<toml::preserve_comments>(mConfigFileName);
auto data = toml::value {}; auto data = toml::value {};
data["General"][StrAuthKey.data()] = Application::GetSettingString(StrAuthKey.data()); data["General"][StrAuthKey.data()] = Application::Settings.Key;
SetComment(data["General"][StrAuthKey.data()].comments(), " AuthKey has to be filled out in order to run the server"); SetComment(data["General"][StrAuthKey.data()].comments(), " AuthKey has to be filled out in order to run the server");
data["General"][StrLogChat.data()] = Application::GetSettingBool(StrLogChat.data()); data["General"][StrLogChat.data()] = Application::Settings.LogChat;
SetComment(data["General"][StrLogChat.data()].comments(), " Whether to log chat messages in the console / log"); SetComment(data["General"][StrLogChat.data()].comments(), " Whether to log chat messages in the console / log");
data["General"][StrDebug.data()] = Application::GetSettingBool(StrDebug.data()); data["General"][StrDebug.data()] = Application::Settings.DebugModeEnabled;
data["General"][StrPrivate.data()] = Application::GetSettingBool(StrPrivate.data()); data["General"][StrPrivate.data()] = Application::Settings.Private;
data["General"][StrPort.data()] = Application::GetSettingInt(StrPort.data()); data["General"][StrPort.data()] = Application::Settings.Port;
data["General"][StrName.data()] = Application::GetSettingString(StrName.data()); data["General"][StrName.data()] = Application::Settings.ServerName;
data["General"][StrMaxCars.data()] = Application::GetSettingInt(StrMaxCars.data()); data["General"][StrMaxCars.data()] = Application::Settings.MaxCars;
data["General"][StrMaxPlayers.data()] = Application::GetSettingInt(StrMaxPlayers.data()); data["General"][StrMaxPlayers.data()] = Application::Settings.MaxPlayers;
data["General"][StrMap.data()] = Application::GetSettingString(StrMap.data()); data["General"][StrMap.data()] = Application::Settings.MapName;
data["General"][StrDescription.data()] = Application::GetSettingString(StrDescription.data()); data["General"][StrDescription.data()] = Application::Settings.ServerDesc;
data["General"][StrResourceFolder.data()] = Application::GetSettingString(StrResourceFolder.data()); data["General"][StrResourceFolder.data()] = Application::Settings.Resource;
// Misc // Misc
data["Misc"][StrHideUpdateMessages.data()] = Application::GetSettingBool(StrHideUpdateMessages.data()); data["Misc"][StrHideUpdateMessages.data()] = Application::Settings.HideUpdateMessages;
SetComment(data["Misc"][StrHideUpdateMessages.data()].comments(), " Hides the periodic update message which notifies you of a new server version. You should really keep this on and always update as soon as possible. For more information visit https://wiki.beammp.com/en/home/server-maintenance#updating-the-server. An update message will always appear at startup regardless."); SetComment(data["Misc"][StrHideUpdateMessages.data()].comments(), " Hides the periodic update message which notifies you of a new server version. You should really keep this on and always update as soon as possible. For more information visit https://wiki.beammp.com/en/home/server-maintenance#updating-the-server. An update message will always appear at startup regardless.");
data["Misc"][StrSendErrors.data()] = Application::GetSettingBool(StrSendErrors.data()); data["Misc"][StrSendErrors.data()] = Application::Settings.SendErrors;
SetComment(data["Misc"][StrSendErrors.data()].comments(), " You can turn on/off the SendErrors message you get on startup here"); SetComment(data["Misc"][StrSendErrors.data()].comments(), " You can turn on/off the SendErrors message you get on startup here");
data["Misc"][StrSendErrorsMessageEnabled.data()] = Application::GetSettingBool(StrSendErrorsMessageEnabled.data()); data["Misc"][StrSendErrorsMessageEnabled.data()] = Application::Settings.SendErrorsMessageEnabled;
SetComment(data["Misc"][StrSendErrorsMessageEnabled.data()].comments(), " If SendErrors is `true`, the server will send helpful info about crashes and other issues back to the BeamMP developers. This info may include your config, who is on your server at the time of the error, and similar general information. This kind of data is vital in helping us diagnose and fix issues faster. This has no impact on server performance. You can opt-out of this system by setting this to `false`"); SetComment(data["Misc"][StrSendErrorsMessageEnabled.data()].comments(), " If SendErrors is `true`, the server will send helpful info about crashes and other issues back to the BeamMP developers. This info may include your config, who is on your server at the time of the error, and similar general information. This kind of data is vital in helping us diagnose and fix issues faster. This has no impact on server performance. You can opt-out of this system by setting this to `false`");
data["Misc"][StrIncludeSubdirectories.data()] = Application::GetSettingBool(StrIncludeSubdirectories.data());
SetComment(data["Misc"][StrIncludeSubdirectories.data()].comments(), " Whether or not to include subdirectories in General.ResourceFolder/Client when searching for `.zip`s");
// HTTP // HTTP
data["HTTP"][StrSSLKeyPath.data()] = Application::GetSettingString(StrSSLKeyPath.data()); data["HTTP"][StrSSLKeyPath.data()] = Application::Settings.SSLKeyPath;
data["HTTP"][StrSSLCertPath.data()] = Application::GetSettingString(StrSSLCertPath.data()); data["HTTP"][StrSSLCertPath.data()] = Application::Settings.SSLCertPath;
data["HTTP"][StrHTTPServerPort.data()] = Application::GetSettingInt(StrHTTPServerPort.data()); data["HTTP"][StrHTTPServerPort.data()] = Application::Settings.HTTPServerPort;
SetComment(data["HTTP"][StrHTTPServerIP.data()].comments(), " Which IP to listen on. Pick 0.0.0.0 for a public-facing server with no specific IP, and 127.0.0.1 or 'localhost' for a local server."); SetComment(data["HTTP"][StrHTTPServerIP.data()].comments(), " Which IP to listen on. Pick 0.0.0.0 for a public-facing server with no specific IP, and 127.0.0.1 or 'localhost' for a local server.");
data["HTTP"][StrHTTPServerIP.data()] = Application::GetSettingString(StrHTTPServerIP.data()); data["HTTP"][StrHTTPServerIP.data()] = Application::Settings.HTTPServerIP;
data["HTTP"][StrHTTPServerUseSSL.data()] = Application::GetSettingBool(StrHTTPServerUseSSL.data()); data["HTTP"][StrHTTPServerUseSSL.data()] = Application::Settings.HTTPServerUseSSL;
SetComment(data["HTTP"][StrHTTPServerUseSSL.data()].comments(), " Recommended to have enabled for servers which face the internet. With SSL the server will serve https and requires valid key and cert files"); SetComment(data["HTTP"][StrHTTPServerUseSSL.data()].comments(), " Recommended to have enabled for servers which face the internet. With SSL the server will serve https and requires valid key and cert files");
data["HTTP"][StrHTTPServerEnabled.data()] = Application::GetSettingBool(StrHTTPServerEnabled.data()); data["HTTP"][StrHTTPServerEnabled.data()] = Application::Settings.HTTPServerEnabled;
SetComment(data["HTTP"][StrHTTPServerEnabled.data()].comments(), " Enables the internal HTTP server"); SetComment(data["HTTP"][StrHTTPServerEnabled.data()].comments(), " Enables the internal HTTP server");
std::stringstream Ss; std::stringstream Ss;
Ss << "# This is the BeamMP-Server config file.\n" Ss << "# This is the BeamMP-Server config file.\n"
@@ -132,42 +156,50 @@ void TConfig::CreateConfigFile() {
FlushToFile(); FlushToFile();
} }
void TConfig::TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key) { void TConfig::TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, std::string& OutValue) {
if (Table[Category.c_str()][Key.data()].is_string()) { if (Table[Category.c_str()][Key.data()].is_string()) {
Application::SetSetting(Key, std::string(Table[Category.c_str()][Key.data()].as_string())); OutValue = Table[Category.c_str()][Key.data()].as_string();
} else if (Table[Category.c_str()][Key.data()].is_boolean()) {
Application::SetSetting(Key, bool(Table[Category.c_str()][Key.data()].as_boolean()));
} else if (Table[Category.c_str()][Key.data()].is_integer()) {
Application::SetSetting(Key, int(Table[Category.c_str()][Key.data()].as_integer()));
} }
} }
void TConfig::TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, bool& OutValue) {
if (Table[Category.c_str()][Key.data()].is_boolean()) {
OutValue = Table[Category.c_str()][Key.data()].as_boolean();
}
}
void TConfig::TryReadValue(toml::value& Table, const std::string& Category, const std::string_view& Key, int& OutValue) {
if (Table[Category.c_str()][Key.data()].is_integer()) {
OutValue = int(Table[Category.c_str()][Key.data()].as_integer());
}
}
void TConfig::ParseFromFile(std::string_view name) { void TConfig::ParseFromFile(std::string_view name) {
try { try {
toml::value data = toml::parse<toml::preserve_comments>(name.data()); toml::value data = toml::parse<toml::preserve_comments>(name.data());
// GENERAL // GENERAL
TryReadValue(data, "General", StrDebug); TryReadValue(data, "General", StrDebug, Application::Settings.DebugModeEnabled);
TryReadValue(data, "General", StrPrivate); TryReadValue(data, "General", StrPrivate, Application::Settings.Private);
TryReadValue(data, "General", StrPort); TryReadValue(data, "General", StrPort, Application::Settings.Port);
TryReadValue(data, "General", StrMaxCars); TryReadValue(data, "General", StrMaxCars, Application::Settings.MaxCars);
TryReadValue(data, "General", StrMaxPlayers); TryReadValue(data, "General", StrMaxPlayers, Application::Settings.MaxPlayers);
TryReadValue(data, "General", StrMap); TryReadValue(data, "General", StrMap, Application::Settings.MapName);
TryReadValue(data, "General", StrName); TryReadValue(data, "General", StrName, Application::Settings.ServerName);
TryReadValue(data, "General", StrDescription); TryReadValue(data, "General", StrDescription, Application::Settings.ServerDesc);
TryReadValue(data, "General", StrResourceFolder); TryReadValue(data, "General", StrResourceFolder, Application::Settings.Resource);
TryReadValue(data, "General", StrAuthKey); TryReadValue(data, "General", StrAuthKey, Application::Settings.Key);
TryReadValue(data, "General", StrLogChat); TryReadValue(data, "General", StrLogChat, Application::Settings.LogChat);
// Misc // Misc
TryReadValue(data, "Misc", StrSendErrors); TryReadValue(data, "Misc", StrSendErrors, Application::Settings.SendErrors);
TryReadValue(data, "Misc", StrHideUpdateMessages); TryReadValue(data, "Misc", StrHideUpdateMessages, Application::Settings.HideUpdateMessages);
TryReadValue(data, "Misc", StrSendErrorsMessageEnabled); TryReadValue(data, "Misc", StrSendErrorsMessageEnabled, Application::Settings.SendErrorsMessageEnabled);
TryReadValue(data, "Misc", StrIncludeSubdirectories);
// HTTP // HTTP
TryReadValue(data, "HTTP", StrSSLKeyPath); TryReadValue(data, "HTTP", StrSSLKeyPath, Application::Settings.SSLKeyPath);
TryReadValue(data, "HTTP", StrSSLCertPath); TryReadValue(data, "HTTP", StrSSLCertPath, Application::Settings.SSLCertPath);
TryReadValue(data, "HTTP", StrHTTPServerPort); TryReadValue(data, "HTTP", StrHTTPServerPort, Application::Settings.HTTPServerPort);
TryReadValue(data, "HTTP", StrHTTPServerIP); TryReadValue(data, "HTTP", StrHTTPServerIP, Application::Settings.HTTPServerIP);
TryReadValue(data, "HTTP", StrHTTPServerEnabled); TryReadValue(data, "HTTP", StrHTTPServerEnabled, Application::Settings.HTTPServerEnabled);
TryReadValue(data, "HTTP", StrHTTPServerUseSSL); TryReadValue(data, "HTTP", StrHTTPServerUseSSL, Application::Settings.HTTPServerUseSSL);
} catch (const std::exception& err) { } catch (const std::exception& err) {
beammp_error("Error parsing config file value: " + std::string(err.what())); beammp_error("Error parsing config file value: " + std::string(err.what()));
mFailed = true; mFailed = true;
@@ -179,28 +211,86 @@ void TConfig::ParseFromFile(std::string_view name) {
// Update in any case // Update in any case
FlushToFile(); FlushToFile();
// all good so far, let's check if there's a key // all good so far, let's check if there's a key
if (Application::GetSettingString(StrAuthKey).empty()) { if (Application::Settings.Key.empty()) {
beammp_error("No AuthKey specified in the \"" + std::string(mConfigFileName) + "\" file. Please get an AuthKey, enter it into the config file, and restart this server. Alternatively, if you only want a private server (one that doesn't show up in the server list), you may put anything as your AuthKey, like \"hello\"."); beammp_error("No AuthKey specified in the \"" + std::string(mConfigFileName) + "\" file. Please get an AuthKey, enter it into the config file, and restart this server.");
Application::SetSubsystemStatus("Config", Application::Status::Bad); Application::SetSubsystemStatus("Config", Application::Status::Bad);
mFailed = true; mFailed = true;
return; return;
} }
Application::SetSubsystemStatus("Config", Application::Status::Good); Application::SetSubsystemStatus("Config", Application::Status::Good);
if (Application::GetSettingString(StrAuthKey).size() != 36) { if (Application::Settings.Key.size() != 36) {
beammp_warn("AuthKey specified is the wrong length and likely isn't valid."); beammp_warn("AuthKey specified is the wrong length and likely isn't valid.");
} }
} }
void TConfig::PrintDebug() { void TConfig::PrintDebug() {
for (const auto& [k, v] : Application::mSettings) { beammp_debug(std::string(StrDebug) + ": " + std::string(Application::Settings.DebugModeEnabled ? "true" : "false"));
if (k == StrAuthKey) { beammp_debug(std::string(StrPrivate) + ": " + std::string(Application::Settings.Private ? "true" : "false"));
beammp_debugf("AuthKey: length {}", std::get<std::string>(v).size()); beammp_debug(std::string(StrPort) + ": " + std::to_string(Application::Settings.Port));
continue; beammp_debug(std::string(StrMaxCars) + ": " + std::to_string(Application::Settings.MaxCars));
} beammp_debug(std::string(StrMaxPlayers) + ": " + std::to_string(Application::Settings.MaxPlayers));
beammp_debugf("{}: {}", k, Application::SettingToString(v)); beammp_debug(std::string(StrMap) + ": \"" + Application::Settings.MapName + "\"");
} beammp_debug(std::string(StrName) + ": \"" + Application::Settings.ServerName + "\"");
beammp_debug(std::string(StrDescription) + ": \"" + Application::Settings.ServerDesc + "\"");
beammp_debug(std::string(StrLogChat) + ": \"" + (Application::Settings.LogChat ? "true" : "false") + "\"");
beammp_debug(std::string(StrResourceFolder) + ": \"" + Application::Settings.Resource + "\"");
beammp_debug(std::string(StrSSLKeyPath) + ": \"" + Application::Settings.SSLKeyPath + "\"");
beammp_debug(std::string(StrSSLCertPath) + ": \"" + Application::Settings.SSLCertPath + "\"");
beammp_debug(std::string(StrHTTPServerPort) + ": \"" + std::to_string(Application::Settings.HTTPServerPort) + "\"");
beammp_debug(std::string(StrHTTPServerIP) + ": \"" + Application::Settings.HTTPServerIP + "\"");
// special!
beammp_debug("Key Length: " + std::to_string(Application::Settings.Key.length()) + "");
} }
void TConfig::ParseOldFormat() { void TConfig::ParseOldFormat() {
beammp_warnf("You still have a 'Server.cfg' - this will not be used (this server uses 'ServerConfig.toml'. Since v3.0.2 we no longer parse and import these old settings. Remove the file to avoid confusion and disable this message."); std::ifstream File("Server.cfg");
// read all, strip comments
std::string Content;
for (;;) {
std::string Line;
std::getline(File, Line);
if (!Line.empty() && Line.at(0) != '#') {
Line = Line.substr(0, Line.find_first_of('#'));
Content += Line + "\n";
}
if (!File.good()) {
break;
}
}
std::stringstream Str(Content);
std::string Key, Ignore, Value;
for (;;) {
Str >> Key >> std::ws >> Ignore >> std::ws;
std::getline(Str, Value);
if (Str.eof()) {
break;
}
std::stringstream ValueStream(Value);
ValueStream >> std::ws; // strip leading whitespace if any
Value = ValueStream.str();
if (Key == "Debug") {
Application::Settings.DebugModeEnabled = Value.find("true") != std::string::npos;
} else if (Key == "Private") {
Application::Settings.Private = Value.find("true") != std::string::npos;
} else if (Key == "Port") {
ValueStream >> Application::Settings.Port;
} else if (Key == "Cars") {
ValueStream >> Application::Settings.MaxCars;
} else if (Key == "MaxPlayers") {
ValueStream >> Application::Settings.MaxPlayers;
} else if (Key == "Map") {
Application::Settings.MapName = Value.substr(1, Value.size() - 3);
} else if (Key == "Name") {
Application::Settings.ServerName = Value.substr(1, Value.size() - 3);
} else if (Key == "Desc") {
Application::Settings.ServerDesc = Value.substr(1, Value.size() - 3);
} else if (Key == "use") {
Application::Settings.Resource = Value.substr(1, Value.size() - 3);
} else if (Key == "AuthKey") {
Application::Settings.Key = Value.substr(1, Value.size() - 3);
} else {
beammp_warn("unknown key in old auth file (ignored): " + Key);
}
Str >> std::ws;
}
} }
+58 -265
View File
@@ -7,34 +7,13 @@
#include "LuaAPI.h" #include "LuaAPI.h"
#include "TLuaEngine.h" #include "TLuaEngine.h"
#include <boost/asio/ip/address.hpp>
#include <boost/spirit/home/qi/directive/lexeme.hpp>
#include <boost/spirit/home/qi/parse.hpp>
#include <chrono>
#include <ctime> #include <ctime>
#include <fmt/chrono.h>
#include <sstream> #include <sstream>
#include <boost/phoenix.hpp>
#include <boost/spirit/include/qi.hpp>
#include <boost/algorithm/string.hpp>
static inline bool StringStartsWith(const std::string& What, const std::string& StartsWith) { static inline bool StringStartsWith(const std::string& What, const std::string& StartsWith) {
return What.size() >= StartsWith.size() && What.substr(0, StartsWith.size()) == StartsWith; return What.size() >= StartsWith.size() && What.substr(0, StartsWith.size()) == StartsWith;
} }
static inline bool StringStartsWithLower(const std::string& Name1, const std::string& Name2) {
std::string Name1Lower = boost::algorithm::to_lower_copy(Name1);
return StringStartsWith(Name1Lower, Name2) || StringStartsWith(Name2, Name1Lower);
};
static inline bool StringStartsWithLowerBoth(const std::string& Name1, const std::string& Name2) {
std::string Name1Lower = boost::algorithm::to_lower_copy(Name1);
std::string Name2Lower = boost::algorithm::to_lower_copy(Name2);
return StringStartsWith(Name1Lower, Name2Lower) || StringStartsWith(Name2Lower, Name1Lower);
};
TEST_CASE("StringStartsWith") { TEST_CASE("StringStartsWith") {
CHECK(StringStartsWith("Hello, World", "Hello")); CHECK(StringStartsWith("Hello, World", "Hello"));
CHECK(StringStartsWith("Hello, World", "H")); CHECK(StringStartsWith("Hello, World", "H"));
@@ -84,13 +63,13 @@ static inline void SplitString(std::string const& str, const char delim, std::ve
} }
static std::string GetDate() { static std::string GetDate() {
TimeType::time_point now = TimeType::now(); std::chrono::system_clock::time_point now = std::chrono::system_clock::now();
time_t tt = TimeType::to_time_t(now); time_t tt = std::chrono::system_clock::to_time_t(now);
auto local_tm = std::localtime(&tt); auto local_tm = std::localtime(&tt);
char buf[30]; char buf[30];
std::string date; std::string date;
if (Application::GetSettingBool(StrDebug)) { if (Application::Settings.DebugModeEnabled) {
std::strftime(buf, sizeof(buf), "[%Y/%m/%d %T.", local_tm); std::strftime(buf, sizeof(buf), "[%d/%m/%y %T.", local_tm);
date += buf; date += buf;
auto seconds = std::chrono::time_point_cast<std::chrono::seconds>(now); auto seconds = std::chrono::time_point_cast<std::chrono::seconds>(now);
auto fraction = now - seconds; auto fraction = now - seconds;
@@ -100,7 +79,7 @@ static std::string GetDate() {
date += fracstr; date += fracstr;
date += "] "; date += "] ";
} else { } else {
std::strftime(buf, sizeof(buf), "[%Y/%m/%d %T] ", local_tm); std::strftime(buf, sizeof(buf), "[%d/%m/%y %T] ", local_tm);
date += buf; date += buf;
} }
@@ -253,17 +232,14 @@ void TConsole::Command_Help(const std::string&, const std::vector<std::string>&
static constexpr const char* sHelpString = R"( static constexpr const char* sHelpString = R"(
Commands: Commands:
help displays this help help displays this help
exit exit shuts down the server
quit shuts down the server
kick <name> [reason] kicks specified player with an optional reason kick <name> [reason] kicks specified player with an optional reason
list lists all players and info about them list lists all players and info about them
say <message> sends the message to all players in chat say <message> sends the message to all players in chat
lua [state id] switches to lua, optionally into a specific state id's lua lua [state id] switches to lua, optionally into a specific state id's lua
settings [command] sets or gets settings for the server, run `settings help` for more info settings [command] sets or gets settings for the server, run `settings help` for more info
status how the server is doing and what it's up to status how the server is doing and what it's up to
debug internal error and debug state of the server (for development) clear clears the console window)";
clear clears the console window
version version info)";
Application::Console().WriteRaw("BeamMP-Server Console: " + std::string(sHelpString)); Application::Console().WriteRaw("BeamMP-Server Console: " + std::string(sHelpString));
} }
@@ -283,114 +259,33 @@ void TConsole::Command_Clear(const std::string&, const std::vector<std::string>&
mCommandline.write("\x1b[;H\x1b[2J"); mCommandline.write("\x1b[;H\x1b[2J");
} }
void TConsole::Command_Debug(const std::string&, const std::vector<std::string>& args) {
if (!EnsureArgsCount(args, 0)) {
return;
}
Application::Console().WriteRaw(fmt::format(R"(Debug info (for developers):
UDP:
Malformed packets: {}
Invalid packets: {})",
Application::MalformedUdpPackets,
Application::InvalidUdpPackets));
Application::Console().WriteRaw(fmt::format(R"( Clients:
Note: All data/second rates are an average across the total time since
connection and do not necessarily reflect the *current* data rate
of that client.
)"));
mLuaEngine->Server().ForEachClient([&](const auto& Client) {
std::string State = "";
if (Client->IsSyncing()) {
State += "Syncing";
}
if (Client->IsSynced()) {
if (!State.empty()) {
State += " & ";
}
State += "Synced";
}
if (Client->IsConnected()) {
if (!State.empty()) {
State += " & ";
}
State += "Connected";
}
if (Client->IsDisconnected()) {
if (!State.empty()) {
State += " & ";
}
State += "Disconnected";
}
auto Now = TimeType::now();
auto Seconds = std::chrono::duration_cast<std::chrono::seconds>(Now - Client->ConnectionTime);
std::string ConnectedSince = fmt::format("{:%Y/%m/%d %H:%M:%S}, {:%H:%M:%S} ago ({} seconds)",
fmt::localtime(TimeType::to_time_t(Client->ConnectionTime)),
Seconds,
Seconds.count());
Application::Console().WriteRaw(fmt::format(
R"( {} ('{}'):
Roles: {}
Cars: {}
Is guest: {}
Has unicycle: {}
TCP: {} (on port {})
UDP: {} (on port {})
Sent via TCP: {}
Received via TCP: {}
Sent via UDP: {} ({} packets)
Received via UDP: {} ({} packets)
Status: {}
Queued packets: {}
Latest packet: {}s ago
Connected since: {}
Average send: {}/s
Average receive: {}/s)",
Client->GetID(), Client->GetName(),
Client->GetRoles(),
Client->GetCarCount(),
Client->IsGuest() ? "yes" : "no",
Client->GetUnicycleID() == -1 ? "no" : "yes",
Client->GetTCPSock().remote_endpoint().address() == ip::address::from_string("0.0.0.0") ? "not connected" : "connected", Client->GetTCPSock().remote_endpoint().port(),
Client->GetUDPAddr().address() == ip::address::from_string("0.0.0.0") ? "NOT connected" : "connected", Client->GetUDPAddr().port(),
ToHumanReadableSize(Client->TcpSent),
ToHumanReadableSize(Client->TcpReceived),
ToHumanReadableSize(Client->UdpSent), Client->UdpPacketsSent,
ToHumanReadableSize(Client->UdpReceived), Client->UdpPacketsReceived,
State.empty() ? "None (likely pre-sync)" : State,
Client->MissedPacketQueueSize(),
Client->SecondsSinceLastPing(),
ConnectedSince,
ToHumanReadableSize((Client->TcpSent + Client->UdpSent) / Seconds.count()),
ToHumanReadableSize((Client->TcpReceived + Client->UdpReceived) / Seconds.count())));
});
}
void TConsole::Command_Version(const std::string&, const std::vector<std::string>& args) {
if (!EnsureArgsCount(args, 0)) {
return;
}
Application::Console().WriteRaw(fmt::format("BeamMP Server v{} ({})", Application::ServerVersionString(), BEAMMP_GIT_HASH));
}
void TConsole::Command_Kick(const std::string&, const std::vector<std::string>& args) { void TConsole::Command_Kick(const std::string&, const std::vector<std::string>& args) {
if (!EnsureArgsCount(args, 1, size_t(-1))) { if (!EnsureArgsCount(args, 1, size_t(-1))) {
return; return;
} }
auto Name = boost::algorithm::to_lower_copy(args.at(0)); auto Name = args.at(0);
std::string Reason = "Kicked by server console"; std::string Reason = "Kicked by server console";
if (args.size() > 1) { if (args.size() > 1) {
Reason = ConcatArgs({ args.begin() + 1, args.end() }); Reason = ConcatArgs({ args.begin() + 1, args.end() });
} }
beammp_trace("attempt to kick '" + Name + "' for '" + Reason + "'"); beammp_trace("attempt to kick '" + Name + "' for '" + Reason + "'");
bool Kicked = false; bool Kicked = false;
// TODO: this sucks, tolower is locale-dependent.
mLuaEngine->Server().ForEachClient([&](const auto& Client) -> IterationDecision { auto NameCompare = [](std::string Name1, std::string Name2) -> bool {
if (StringStartsWithLower(Client->GetName(), Name)) { std::for_each(Name1.begin(), Name1.end(), [](char& c) { c = char(std::tolower(char(c))); });
mLuaEngine->Network().ClientKick(*Client, Reason); std::for_each(Name2.begin(), Name2.end(), [](char& c) { c = char(std::tolower(char(c))); });
Kicked = true; return StringStartsWith(Name1, Name2) || StringStartsWith(Name2, Name1);
return Break; };
mLuaEngine->Server().ForEachClient([&](std::weak_ptr<TClient> Client) -> bool {
if (!Client.expired()) {
auto locked = Client.lock();
if (NameCompare(locked->GetName(), Name)) {
mLuaEngine->Network().ClientKick(*locked, Reason);
Kicked = true;
return false;
}
} }
return Continue; return true;
}); });
if (!Kicked) { if (!Kicked) {
Application::Console().WriteRaw("Error: No player with name matching '" + Name + "' was found."); Application::Console().WriteRaw("Error: No player with name matching '" + Name + "' was found.");
@@ -455,80 +350,17 @@ std::tuple<std::string, std::vector<std::string>> TConsole::ParseCommand(const s
} }
void TConsole::Command_Settings(const std::string&, const std::vector<std::string>& args) { void TConsole::Command_Settings(const std::string&, const std::vector<std::string>& args) {
if (!EnsureArgsCount(args, 1, 100)) { if (!EnsureArgsCount(args, 1, 2)) {
return; return;
} }
static const char* SETTINGS_HELP = R"(Settings:
settings help Displays this help
settings list Lists all settings
settings get <setting> Prints the current value of the specified setting
settings set <setting> <value> Sets the specified setting to the value)";
if (args.at(0) == "help") {
Application::Console().WriteRaw(SETTINGS_HELP);
} else if (args.at(0) == "list") {
Application::Console().WriteRaw("Available settings:");
Application::Console().WriteRaw(fmt::format("\t{:<25} {}", "<NAME>", "<CURRENT VALUE>"));
for (const auto& [k, v] : Application::mSettings) {
if (k == StrAuthKey) {
Application::Console().WriteRaw(fmt::format("\t{:<25} <key of length {}>", k, Application::SettingToString(v).size()));
} else {
Application::Console().WriteRaw(fmt::format("\t{:<25} {}", k, Application::SettingToString(v)));
}
}
} else if (args.at(0) == "get") {
if (args.size() < 2) {
Application::Console().WriteRaw("Not enough arguments: `settings get` requires a setting name.");
} else {
if (Application::mSettings.contains(args.at(1))) {
if (args.at(1) != StrAuthKey) {
Application::Console().WriteRaw(fmt::format("{} = {}", args.at(1), Application::SettingToString(Application::mSettings.at(args.at(1)))));
} else {
Application::Console().WriteRaw(fmt::format("{} = <key of length {}>", args.at(1), Application::SettingToString(Application::mSettings.at(args.at(1))).size()));
}
} else {
Application::Console().WriteRaw(fmt::format("Setting '{}' doesn't exist.", args.at(1)));
}
}
} else if (args.at(0) == "set") {
if (args.size() < 3) {
Application::Console().WriteRaw("Not enough arguments: `settings set` requires a setting name and value.");
} else {
if (args.at(1) == StrAuthKey) {
Application::Console().WriteRaw("It's not allowed to set the AuthKey during runtime.");
} else {
using namespace boost::spirit;
using qi::_1;
std::string ValueString = args.at(2);
Application::SettingValue Value;
qi::rule<std::string::iterator, std::string()> StringRule;
StringRule
%= qi::lexeme['"' >> *(qi::char_ - '"') >> '"']
| +(qi::char_ - '"');
qi::rule<std::string::iterator, Application::SettingValue()> ValueRule
= qi::bool_
| qi::int_
| StringRule;
auto It = ValueString.begin();
if (qi::phrase_parse(It, ValueString.end(), ValueRule[boost::phoenix::ref(Value) = _1], ascii::space)
&& It == ValueString.end()) {
Application::SetSetting(args.at(1), Value);
Application::Console().WriteRaw(fmt::format("{} := {}", args.at(1), Application::SettingToString(Application::mSettings.at(args.at(1)))));
} else {
Application::Console().WriteRaw(fmt::format("New value '{}' did not parse as a valid value.", ValueString));
}
}
}
} else {
Application::Console().WriteRaw(fmt::format("Unknown argument '{}' - 'settings {}' is not a valid command.", args.at(0), args.at(0)));
}
} }
void TConsole::Command_Say(const std::string& FullCmd) { void TConsole::Command_Say(const std::string& FullCmd) {
if (FullCmd.size() > 3) { if (FullCmd.size() > 3) {
auto Message = FullCmd.substr(4); auto Message = FullCmd.substr(4);
LuaAPI::MP::SendChatMessage(-1, Message); LuaAPI::MP::SendChatMessage(-1, Message);
if (!Application::GetSettingBool(StrLogChat)) { if (!Application::Settings.LogChat) {
Application::Console().WriteRaw("Chat message sent!"); Application::Console().WriteRaw("Chat message sent!");
} }
} }
@@ -543,10 +375,14 @@ void TConsole::Command_List(const std::string&, const std::vector<std::string>&
} else { } else {
std::stringstream ss; std::stringstream ss;
ss << std::left << std::setw(25) << "Name" << std::setw(6) << "ID" << std::setw(6) << "Cars" << std::endl; ss << std::left << std::setw(25) << "Name" << std::setw(6) << "ID" << std::setw(6) << "Cars" << std::endl;
mLuaEngine->Server().ForEachClient([&](const auto& Client) { mLuaEngine->Server().ForEachClient([&](std::weak_ptr<TClient> Client) -> bool {
ss << std::left << std::setw(25) << Client->GetName() if (!Client.expired()) {
<< std::setw(6) << Client->GetID() auto locked = Client.lock();
<< std::setw(6) << Client->GetCarCount() << "\n"; ss << std::left << std::setw(25) << locked->GetName()
<< std::setw(6) << locked->GetID()
<< std::setw(6) << locked->GetCarCount() << "\n";
}
return true;
}); });
auto Str = ss.str(); auto Str = ss.str();
Application::Console().WriteRaw(Str.substr(0, Str.size() - 1)); Application::Console().WriteRaw(Str.substr(0, Str.size() - 1));
@@ -566,16 +402,20 @@ void TConsole::Command_Status(const std::string&, const std::vector<std::string>
size_t SyncingCount = 0; size_t SyncingCount = 0;
size_t MissedPacketQueueSum = 0; size_t MissedPacketQueueSum = 0;
int LargestSecondsSinceLastPing = 0; int LargestSecondsSinceLastPing = 0;
mLuaEngine->Server().ForEachClient([&](const auto& Client) { mLuaEngine->Server().ForEachClient([&](std::weak_ptr<TClient> Client) -> bool {
CarCount += Client->GetCarCount(); if (!Client.expired()) {
ConnectedCount += Client->IsConnected() ? 1 : 0; auto Locked = Client.lock();
GuestCount += Client->IsGuest() ? 1 : 0; CarCount += Locked->GetCarCount();
SyncedCount += Client->IsSynced() ? 1 : 0; ConnectedCount += Locked->IsConnected() ? 1 : 0;
SyncingCount += Client->IsSyncing() ? 1 : 0; GuestCount += Locked->IsGuest() ? 1 : 0;
MissedPacketQueueSum += Client->MissedPacketQueueSize(); SyncedCount += Locked->IsSynced() ? 1 : 0;
if (Client->SecondsSinceLastPing() < LargestSecondsSinceLastPing) { SyncingCount += Locked->IsSyncing() ? 1 : 0;
LargestSecondsSinceLastPing = Client->SecondsSinceLastPing(); MissedPacketQueueSum += Locked->MissedPacketQueueSize();
if (Locked->SecondsSinceLastPing() < LargestSecondsSinceLastPing) {
LargestSecondsSinceLastPing = Locked->SecondsSinceLastPing();
}
} }
return true;
}); });
size_t SystemsStarting = 0; size_t SystemsStarting = 0;
@@ -650,55 +490,6 @@ void TConsole::Command_Status(const std::string&, const std::vector<std::string>
Application::Console().WriteRaw(Status.str()); Application::Console().WriteRaw(Status.str());
} }
void TConsole::Autocomplete_Lua(const std::string& stub, std::vector<std::string>& suggestions) {
auto stateNames = mLuaEngine->GetLuaStateNames();
for (const auto& name : stateNames) {
if (name.find(stub) == 0) {
suggestions.push_back("lua " + name);
}
}
}
void TConsole::Autocomplete_Kick(const std::string& stub, std::vector<std::string>& suggestions) {
std::string stub_lower = boost::algorithm::to_lower_copy(stub);
mLuaEngine->Server().ForEachClient([&](const auto& Client) {
if (StringStartsWithLower(Client->GetName(), stub_lower)) {
suggestions.push_back("kick " + Client->GetName());
}
});
}
void TConsole::Autocomplete_Settings(const std::string& stub, std::vector<std::string>& suggestions) {
const std::string subcommands[] = { "help", "list", "set", "get" };
auto [command, args] = ParseCommand(stub);
std::string arg;
if (!args.empty())
arg = boost::algorithm::to_lower_copy(args.at(0));
// suggest setting names
if (command == "set" || command == "get") {
for (const auto& [k, v] : Application::mSettings) {
std::string key = std::string(k);
if (StringStartsWithLower(key, arg)) {
suggestions.push_back("settings " + command + " " + key);
}
}
return;
}
// suggest subcommands
for (const auto& cmd : subcommands) {
if (cmd.find(command) == 0) {
suggestions.push_back("settings " + cmd);
}
}
}
void TConsole::RunAsCommand(const std::string& cmd, bool IgnoreNotACommand) { void TConsole::RunAsCommand(const std::string& cmd, bool IgnoreNotACommand) {
auto FutureIsNonNil = auto FutureIsNonNil =
[](const std::shared_ptr<TLuaResult>& Future) { [](const std::shared_ptr<TLuaResult>& Future) {
@@ -800,6 +591,7 @@ Commands
TConsole::TConsole() { TConsole::TConsole() {
mCommandline.enable_history(); mCommandline.enable_history();
mCommandline.set_history_limit(20); mCommandline.set_history_limit(20);
mCommandline.set_prompt("> ");
BackupOldLog(); BackupOldLog();
mCommandline.on_command = [this](Commandline& c) { mCommandline.on_command = [this](Commandline& c) {
try { try {
@@ -824,7 +616,7 @@ TConsole::TConsole() {
} else { } else {
if (!mLuaEngine) { if (!mLuaEngine) {
beammp_error("Attempted to run a command before Lua engine started. Please wait and try again."); beammp_error("Attempted to run a command before Lua engine started. Please wait and try again.");
} else if (cmd == "exit" || cmd == "quit") { } else if (cmd == "exit") {
beammp_info("gracefully shutting down"); beammp_info("gracefully shutting down");
Application::GracefullyShutdown(); Application::GracefullyShutdown();
} else if (cmd == "say") { } else if (cmd == "say") {
@@ -886,15 +678,16 @@ TConsole::TConsole() {
} }
} }
} else { // if not lua } else { // if not lua
for (const auto& [cmd_name, autocomplete_fn] : mCommandAutocompleteMap) { if (stub.find("lua") == 0) { // starts with "lua" means we should suggest state names
if (stub.find(cmd_name) == 0) { // input starts with a full command (that has autocomplete) std::string after_prefix = TrimString(stub.substr(3));
std::size_t cmd_len = cmd_name.length(); auto stateNames = mLuaEngine->GetLuaStateNames();
std::string trimmed = TrimString(stub.substr(cmd_len));
autocomplete_fn(trimmed, suggestions); for (const auto& name : stateNames) {
break; if (name.find(after_prefix) == 0) {
suggestions.push_back("lua " + name);
}
} }
} } else {
if (suggestions.empty()) {
for (const auto& [cmd_name, cmd_fn] : mCommandMap) { for (const auto& [cmd_name, cmd_fn] : mCommandMap) {
if (cmd_name.find(stub) == 0) { if (cmd_name.find(stub) == 0) {
suggestions.push_back(cmd_name); suggestions.push_back(cmd_name);
+25 -21
View File
@@ -17,14 +17,14 @@ void THeartbeatThread::operator()() {
// these are "hot-change" related variables // these are "hot-change" related variables
static std::string Last; static std::string Last;
static TimeType::time_point LastNormalUpdateTime = TimeType::now(); static std::chrono::high_resolution_clock::time_point LastNormalUpdateTime = std::chrono::high_resolution_clock::now();
bool isAuth = false; bool isAuth = false;
size_t UpdateReminderCounter = 0; size_t UpdateReminderCounter = 0;
while (!Application::IsShuttingDown()) { while (!Application::IsShuttingDown()) {
++UpdateReminderCounter; ++UpdateReminderCounter;
Body = GenerateCall(); Body = GenerateCall();
// a hot-change occurs when a setting has changed, to update the backend of that change. // a hot-change occurs when a setting has changed, to update the backend of that change.
auto Now = TimeType::now(); auto Now = std::chrono::high_resolution_clock::now();
bool Unchanged = Last == Body; bool Unchanged = Last == Body;
auto TimePassed = (Now - LastNormalUpdateTime); auto TimePassed = (Now - LastNormalUpdateTime);
auto Threshold = Unchanged ? 30 : 5; auto Threshold = Unchanged ? 30 : 5;
@@ -36,8 +36,8 @@ void THeartbeatThread::operator()() {
Last = Body; Last = Body;
LastNormalUpdateTime = Now; LastNormalUpdateTime = Now;
if (!Application::GetSettingString(StrCustomIP).empty()) { if (!Application::Settings.CustomIP.empty()) {
Body += "&ip=" + Application::GetSettingString(StrCustomIP); Body += "&ip=" + Application::Settings.CustomIP;
} }
auto SentryReportError = [&](const std::string& transaction, int status) { auto SentryReportError = [&](const std::string& transaction, int status) {
@@ -59,7 +59,7 @@ void THeartbeatThread::operator()() {
T = Http::POST(Url, 443, Target, Body, "application/x-www-form-urlencoded", &ResponseCode, { { "api-v", "2" } }); T = Http::POST(Url, 443, Target, Body, "application/x-www-form-urlencoded", &ResponseCode, { { "api-v", "2" } });
Doc.Parse(T.data(), T.size()); Doc.Parse(T.data(), T.size());
if (Doc.HasParseError() || !Doc.IsObject()) { if (Doc.HasParseError() || !Doc.IsObject()) {
if (!Application::GetSettingBool(StrPrivate)) { if (!Application::Settings.Private) {
beammp_trace("Backend response failed to parse as valid json"); beammp_trace("Backend response failed to parse as valid json");
beammp_trace("Response was: `" + T + "`"); beammp_trace("Response was: `" + T + "`");
} }
@@ -105,12 +105,12 @@ void THeartbeatThread::operator()() {
Sentry.LogError("Missing/invalid json members in backend response", __FILE__, std::to_string(__LINE__)); Sentry.LogError("Missing/invalid json members in backend response", __FILE__, std::to_string(__LINE__));
} }
} else { } else {
if (!Application::GetSettingBool(StrPrivate)) { if (!Application::Settings.Private) {
beammp_warn("Backend failed to respond to a heartbeat. Your server may temporarily disappear from the server list. This is not an error, and will likely resolve itself soon. Direct connect will still work."); beammp_warn("Backend failed to respond to a heartbeat. Your server may temporarily disappear from the server list. This is not an error, and will likely resolve itself soon. Direct connect will still work.");
} }
} }
if (Ok && !isAuth && !Application::GetSettingBool(StrPrivate)) { if (Ok && !isAuth && !Application::Settings.Private) {
if (Status == "2000") { if (Status == "2000") {
beammp_info(("Authenticated! " + Message)); beammp_info(("Authenticated! " + Message));
isAuth = true; isAuth = true;
@@ -124,10 +124,10 @@ void THeartbeatThread::operator()() {
beammp_error("Backend REFUSED the auth key. Reason: " + Message); beammp_error("Backend REFUSED the auth key. Reason: " + Message);
} }
} }
if (isAuth || Application::GetSettingBool(StrPrivate)) { if (isAuth || Application::Settings.Private) {
Application::SetSubsystemStatus("Heartbeat", Application::Status::Good); Application::SetSubsystemStatus("Heartbeat", Application::Status::Good);
} }
if (!Application::GetSettingBool(StrHideUpdateMessages) && UpdateReminderCounter % 5) { if (!Application::Settings.HideUpdateMessages && UpdateReminderCounter % 5) {
Application::CheckForUpdates(); Application::CheckForUpdates();
} }
} }
@@ -136,20 +136,20 @@ void THeartbeatThread::operator()() {
std::string THeartbeatThread::GenerateCall() { std::string THeartbeatThread::GenerateCall() {
std::stringstream Ret; std::stringstream Ret;
Ret << "uuid=" << Application::GetSettingString(StrAuthKey) Ret << "uuid=" << Application::Settings.Key
<< "&players=" << mServer.ClientCount() << "&players=" << mServer.ClientCount()
<< "&maxplayers=" << Application::GetSettingInt(StrMaxPlayers) << "&maxplayers=" << Application::Settings.MaxPlayers
<< "&port=" << Application::GetSettingInt(StrPort) << "&port=" << Application::Settings.Port
<< "&map=" << Application::GetSettingString(StrMap) << "&map=" << Application::Settings.MapName
<< "&private=" << (Application::GetSettingBool(StrPrivate) ? "true" : "false") << "&private=" << (Application::Settings.Private ? "true" : "false")
<< "&version=" << Application::ServerVersionString() << "&version=" << Application::ServerVersionString()
<< "&clientversion=" << std::to_string(Application::ClientMajorVersion()) + ".0" // FIXME: Wtf. << "&clientversion=" << std::to_string(Application::ClientMajorVersion()) + ".0" // FIXME: Wtf.
<< "&name=" << Application::GetSettingString(StrName) << "&name=" << Application::Settings.ServerName
<< "&modlist=" << TResourceManager::FormatForBackend(mResourceManager.FileMap()) << "&modlist=" << mResourceManager.TrimmedList()
<< "&modstotalsize=" << mResourceManager.TotalModsSize() << "&modstotalsize=" << mResourceManager.MaxModSize()
<< "&modstotal=" << mResourceManager.LoadedModCount() << "&modstotal=" << mResourceManager.ModsLoaded()
<< "&playerslist=" << GetPlayers() << "&playerslist=" << GetPlayers()
<< "&desc=" << Application::GetSettingString(StrDescription); << "&desc=" << Application::Settings.ServerDesc;
return Ret.str(); return Ret.str();
} }
THeartbeatThread::THeartbeatThread(TResourceManager& ResourceManager, TServer& Server) THeartbeatThread::THeartbeatThread(TResourceManager& ResourceManager, TServer& Server)
@@ -167,8 +167,12 @@ THeartbeatThread::THeartbeatThread(TResourceManager& ResourceManager, TServer& S
} }
std::string THeartbeatThread::GetPlayers() { std::string THeartbeatThread::GetPlayers() {
std::string Return; std::string Return;
mServer.ForEachClient([&](const auto& Client) { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
Return += Client->GetName() + ";"; ReadLock Lock(mServer.GetClientMutex());
if (!ClientPtr.expired()) {
Return += ClientPtr.lock()->GetName() + ";";
}
return true;
}); });
return Return; return Return;
} }
+65 -131
View File
@@ -4,8 +4,7 @@
#include "Http.h" #include "Http.h"
#include "LuaAPI.h" #include "LuaAPI.h"
#include "TLuaPlugin.h" #include "TLuaPlugin.h"
#include "Uuid.h" #include "sol/object.hpp"
#include "sol/types.hpp"
#include <chrono> #include <chrono>
#include <condition_variable> #include <condition_variable>
@@ -17,11 +16,11 @@
TLuaEngine* LuaAPI::MP::Engine; TLuaEngine* LuaAPI::MP::Engine;
TLuaEngine::TLuaEngine() TLuaEngine::TLuaEngine()
: mResourceServerPath(fs::path(Application::GetSettingString(StrResourceFolder)) / "Server") { : mResourceServerPath(fs::path(Application::Settings.Resource) / "Server") {
Application::SetSubsystemStatus("LuaEngine", Application::Status::Starting); Application::SetSubsystemStatus("LuaEngine", Application::Status::Starting);
LuaAPI::MP::Engine = this; LuaAPI::MP::Engine = this;
if (!fs::exists(Application::GetSettingString(StrResourceFolder))) { if (!fs::exists(Application::Settings.Resource)) {
fs::create_directory(Application::GetSettingString(StrResourceFolder)); fs::create_directory(Application::Settings.Resource);
} }
if (!fs::exists(mResourceServerPath)) { if (!fs::exists(mResourceServerPath)) {
fs::create_directory(mResourceServerPath); fs::create_directory(mResourceServerPath);
@@ -37,7 +36,7 @@ TLuaEngine::TLuaEngine()
} }
TEST_CASE("TLuaEngine ctor & dtor") { TEST_CASE("TLuaEngine ctor & dtor") {
Application::SetSetting(StrResourceFolder, "beammp_server_test_resources"); Application::Settings.Resource = "beammp_server_test_resources";
TLuaEngine engine; TLuaEngine engine;
Application::GracefullyShutdown(); Application::GracefullyShutdown();
} }
@@ -72,12 +71,13 @@ void TLuaEngine::operator()() {
} }
return false; return false;
}); });
} else {
mResultsToCheckCond.wait_for(Lock, std::chrono::milliseconds(20));
} }
mResultsToCheckCond.wait_for(Lock, std::chrono::milliseconds(10));
} }
}); });
// event loop // event loop
auto Before = TimeType::now(); auto Before = std::chrono::high_resolution_clock::now();
while (!Application::IsShuttingDown()) { while (!Application::IsShuttingDown()) {
{ // Timed Events Scope { // Timed Events Scope
std::unique_lock Lock(mTimedEventsMutex); std::unique_lock Lock(mTimedEventsMutex);
@@ -110,14 +110,14 @@ void TLuaEngine::operator()() {
} else { } else {
constexpr double NsFactor = 1000000.0; constexpr double NsFactor = 1000000.0;
constexpr double Expected = 10.0; // ms constexpr double Expected = 10.0; // ms
const auto Diff = (TimeType::now() - Before).count() / NsFactor; const auto Diff = (std::chrono::high_resolution_clock::now() - Before).count() / NsFactor;
if (Diff < Expected) { if (Diff < Expected) {
std::this_thread::sleep_for(std::chrono::nanoseconds(size_t((Expected - Diff) * NsFactor))); std::this_thread::sleep_for(std::chrono::nanoseconds(size_t((Expected - Diff) * NsFactor)));
} else { } else {
beammp_tracef("Event loop cannot keep up! Running {}ms behind", Diff); beammp_tracef("Event loop cannot keep up! Running {}ms behind", Diff);
} }
} }
Before = TimeType::now(); Before = std::chrono::high_resolution_clock::now();
} }
if (ResultCheckThread.joinable()) { if (ResultCheckThread.joinable()) {
@@ -253,15 +253,11 @@ std::vector<std::string> TLuaEngine::StateThreadData::GetStateTableKeys(const st
} else if (i == keys.size() - 1) { } else if (i == keys.size() - 1) {
if (obj.get_type() == sol::type::table) { if (obj.get_type() == sol::type::table) {
for (const auto& [key, value] : obj.as<sol::table>()) { for (const auto& [key, value] : obj.as<sol::table>()) {
if (key.get_type() == sol::type::string) { std::string s = key.as<std::string>();
std::string s = key.as<std::string>(); if (value.get_type() == sol::type::function) {
s += "(";
if (value.get_type() == sol::type::function) {
s += "(";
}
Result.push_back(s);
} }
Result.push_back(s);
} }
} else { } else {
Result = { obj.as<std::string>() }; Result = { obj.as<std::string>() };
@@ -285,7 +281,7 @@ std::vector<std::string> TLuaEngine::StateThreadData::GetStateTableKeys(const st
*/ */
void TLuaEngine::WaitForAll(std::vector<std::shared_ptr<TLuaResult>>& Results, const std::optional<TimeType::duration>& Max) { void TLuaEngine::WaitForAll(std::vector<std::shared_ptr<TLuaResult>>& Results, const std::optional<std::chrono::high_resolution_clock::duration>& Max) {
for (const auto& Result : Results) { for (const auto& Result : Results) {
bool Cancelled = false; bool Cancelled = false;
size_t ms = 0; size_t ms = 0;
@@ -334,7 +330,7 @@ std::shared_ptr<TLuaResult> TLuaEngine::EnqueueScript(TLuaStateId StateID, const
return mLuaStates.at(StateID)->EnqueueScript(Script); return mLuaStates.at(StateID)->EnqueueScript(Script);
} }
std::shared_ptr<TLuaResult> TLuaEngine::EnqueueFunctionCall(TLuaStateId StateID, const std::string& FunctionName, const std::vector<TLuaValue>& Args) { std::shared_ptr<TLuaResult> TLuaEngine::EnqueueFunctionCall(TLuaStateId StateID, const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args) {
std::unique_lock Lock(mLuaStatesMutex); std::unique_lock Lock(mLuaStatesMutex);
return mLuaStates.at(StateID)->EnqueueFunctionCall(FunctionName, Args); return mLuaStates.at(StateID)->EnqueueFunctionCall(FunctionName, Args);
} }
@@ -414,55 +410,13 @@ std::set<std::string> TLuaEngine::GetEventHandlersForState(const std::string& Ev
return mLuaEvents[EventName][StateId]; return mLuaEvents[EventName][StateId];
} }
std::vector<sol::object> TLuaEngine::StateThreadData::JsonStringToArray(JsonString Str) {
auto LocalTable = Lua_JsonDecode(Str.value).as<std::vector<sol::object>>();
for (auto& value : LocalTable) {
if (value.is<std::string>() && value.as<std::string>() == BEAMMP_INTERNAL_NIL) {
value = sol::object {};
}
}
return LocalTable;
}
sol::table TLuaEngine::StateThreadData::Lua_TriggerGlobalEvent(const std::string& EventName, sol::variadic_args EventArgs) { sol::table TLuaEngine::StateThreadData::Lua_TriggerGlobalEvent(const std::string& EventName, sol::variadic_args EventArgs) {
auto Table = mStateView.create_table(); auto Return = mEngine->TriggerEvent(EventName, mStateId, EventArgs);
for (const sol::stack_proxy& Arg : EventArgs) {
switch (Arg.get_type()) {
case sol::type::none:
case sol::type::userdata:
case sol::type::lightuserdata:
case sol::type::thread:
case sol::type::function:
case sol::type::poly:
Table.add(BEAMMP_INTERNAL_NIL);
beammp_warnf("Passed a value of type '{}' to TriggerGlobalEvent(\"{}\", ...). This type can not be serialized, and cannot be passed between states. It will arrive as <nil> in handlers.", sol::type_name(EventArgs.lua_state(), Arg.get_type()), EventName);
break;
case sol::type::lua_nil:
Table.add(BEAMMP_INTERNAL_NIL);
break;
case sol::type::string:
case sol::type::number:
case sol::type::boolean:
case sol::type::table:
Table.add(Arg);
break;
}
}
JsonString Str { "" };
if (!Table.empty()) {
Str.value = LuaAPI::MP::JsonEncode(Table);
}
auto Return = mEngine->TriggerEvent(EventName, mStateId, Str);
mEngine->ReportErrors(Return);
auto MyHandlers = mEngine->GetEventHandlersForState(EventName, mStateId); auto MyHandlers = mEngine->GetEventHandlersForState(EventName, mStateId);
sol::variadic_results LocalArgs = Str.value.empty() ? sol::variadic_results {} : JsonStringToArray(Str);
for (const auto& Handler : MyHandlers) { for (const auto& Handler : MyHandlers) {
auto Fn = mStateView[Handler]; auto Fn = mStateView[Handler];
if (Fn.valid()) { if (Fn.valid()) {
auto LuaResult = LocalArgs.empty() ? Fn() : Fn(LocalArgs); auto LuaResult = Fn(EventArgs);
auto Result = std::make_shared<TLuaResult>(); auto Result = std::make_shared<TLuaResult>();
if (LuaResult.valid()) { if (LuaResult.valid()) {
Result->Error = false; Result->Error = false;
@@ -501,44 +455,31 @@ sol::table TLuaEngine::StateThreadData::Lua_TriggerGlobalEvent(const std::string
} }
return Result; return Result;
}); });
AsyncEventReturn.set_function("Wait",
[](const sol::table& Self, std::optional<float> TimeoutS) {
auto Vector = Self.get<std::vector<std::shared_ptr<TLuaResult>>>("ReturnValueImpl");
if (TimeoutS.has_value()) {
TLuaEngine::WaitForAll(Vector, std::chrono::milliseconds(size_t(TimeoutS.value() * 1000.0f)));
} else {
TLuaEngine::WaitForAll(Vector);
}
});
return AsyncEventReturn; return AsyncEventReturn;
} }
sol::table TLuaEngine::StateThreadData::Lua_TriggerLocalEvent(const std::string& EventName, sol::variadic_args EventArgs) { sol::table TLuaEngine::StateThreadData::Lua_TriggerLocalEvent(const std::string& EventName, sol::variadic_args EventArgs) {
// TODO: make asynchronous? // TODO: make asynchronous?
auto Result = mStateView.create_table(); sol::table Result = mStateView.create_table();
for (const auto& Handler : mEngine->GetEventHandlersForState(EventName, mStateId)) { for (const auto& Handler : mEngine->GetEventHandlersForState(EventName, mStateId)) {
auto Fn = mStateView[Handler]; auto Fn = mStateView[Handler];
if (Fn.valid() && Fn.get_type() == sol::type::function) { if (Fn.valid() && Fn.get_type() == sol::type::function) {
auto FnRet = Fn(EventArgs); auto FnRet = Fn(EventArgs);
if (FnRet.valid()) { if (FnRet.valid()) {
for (const auto& Res : FnRet) { Result.add(FnRet);
Result.add(Res);
}
} else { } else {
sol::error Err = FnRet; sol::error Err = FnRet;
beammp_lua_error(std::string("TriggerLocalEvent: ") + Err.what()); beammp_lua_error(std::string("TriggerLocalEvent: ") + Err.what());
} }
} else {
beammp_lua_errorf("Handler '{}' for event '{}' in state '{}' is NOT a function, and will be ignored.", Handler, EventName, mStateId);
} }
} }
return Result; return Result;
} }
sol::table TLuaEngine::StateThreadData::Lua_GetPlayerIdentifiers(int ID) { sol::table TLuaEngine::StateThreadData::Lua_GetPlayerIdentifiers(int ID) {
auto Client = GetClient(mEngine->Server(), ID); auto MaybeClient = GetClient(mEngine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
auto IDs = Client->GetIdentifiers(); auto IDs = MaybeClient.value().lock()->GetIdentifiers();
if (IDs.empty()) { if (IDs.empty()) {
return sol::lua_nil; return sol::lua_nil;
} }
@@ -554,20 +495,27 @@ sol::table TLuaEngine::StateThreadData::Lua_GetPlayerIdentifiers(int ID) {
sol::table TLuaEngine::StateThreadData::Lua_GetPlayers() { sol::table TLuaEngine::StateThreadData::Lua_GetPlayers() {
sol::table Result = mStateView.create_table(); sol::table Result = mStateView.create_table();
mEngine->Server().ForEachClient([&](const auto& Client) { mEngine->Server().ForEachClient([&](std::weak_ptr<TClient> Client) -> bool {
Result[Client->GetID()] = Client->GetName(); if (!Client.expired()) {
auto locked = Client.lock();
Result[locked->GetID()] = locked->GetName();
}
return true;
}); });
return Result; return Result;
} }
int TLuaEngine::StateThreadData::Lua_GetPlayerIDByName(const std::string& Name) { int TLuaEngine::StateThreadData::Lua_GetPlayerIDByName(const std::string& Name) {
int Id = -1; int Id = -1;
mEngine->mServer->ForEachClient([&Id, &Name](const auto& Client) -> IterationDecision { mEngine->mServer->ForEachClient([&Id, &Name](std::weak_ptr<TClient> Client) -> bool {
if (Client->GetName() == Name) { if (!Client.expired()) {
Id = Client->GetID(); auto locked = Client.lock();
return Break; if (locked->GetName() == Name) {
Id = locked->GetID();
return false;
}
} }
return Continue; return true;
}); });
return Id; return Id;
} }
@@ -599,17 +547,18 @@ sol::table TLuaEngine::StateThreadData::Lua_FS_ListDirectories(const std::string
} }
std::string TLuaEngine::StateThreadData::Lua_GetPlayerName(int ID) { std::string TLuaEngine::StateThreadData::Lua_GetPlayerName(int ID) {
auto Client = GetClient(mEngine->Server(), ID); auto MaybeClient = GetClient(mEngine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
return Client->GetName(); return MaybeClient.value().lock()->GetName();
} else { } else {
return ""; return "";
} }
} }
sol::table TLuaEngine::StateThreadData::Lua_GetPlayerVehicles(int ID) { sol::table TLuaEngine::StateThreadData::Lua_GetPlayerVehicles(int ID) {
auto Client = GetClient(mEngine->Server(), ID); auto MaybeClient = GetClient(mEngine->Server(), ID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
auto Client = MaybeClient.value().lock();
TClient::TSetOfVehicleData VehicleData; TClient::TSetOfVehicleData VehicleData;
{ // Vehicle Data Lock Scope { // Vehicle Data Lock Scope
auto LockedData = Client->GetAllCars(); auto LockedData = Client->GetAllCars();
@@ -630,8 +579,9 @@ sol::table TLuaEngine::StateThreadData::Lua_GetPlayerVehicles(int ID) {
std::pair<sol::table, std::string> TLuaEngine::StateThreadData::Lua_GetPositionRaw(int PID, int VID) { std::pair<sol::table, std::string> TLuaEngine::StateThreadData::Lua_GetPositionRaw(int PID, int VID) {
std::pair<sol::table, std::string> Result; std::pair<sol::table, std::string> Result;
auto Client = GetClient(mEngine->Server(), PID); auto MaybeClient = GetClient(mEngine->Server(), PID);
if (Client) { if (MaybeClient && !MaybeClient.value().expired()) {
auto Client = MaybeClient.value().lock();
std::string VehiclePos = Client->GetCarPositionRaw(VID); std::string VehiclePos = Client->GetCarPositionRaw(VID);
if (VehiclePos.empty()) { if (VehiclePos.empty()) {
@@ -776,14 +726,14 @@ TLuaEngine::StateThreadData::StateThreadData(const std::string& Name, TLuaStateI
MPTable.set_function("CreateTimer", [&]() -> sol::table { MPTable.set_function("CreateTimer", [&]() -> sol::table {
sol::state_view StateView(mState); sol::state_view StateView(mState);
sol::table Result = StateView.create_table(); sol::table Result = StateView.create_table();
Result["__StartTime"] = TimeType::now(); Result["__StartTime"] = std::chrono::high_resolution_clock::now();
Result.set_function("GetCurrent", [&](const sol::table& Table) -> float { Result.set_function("GetCurrent", [&](const sol::table& Table) -> float {
auto End = TimeType::now(); auto End = std::chrono::high_resolution_clock::now();
auto Start = Table.get<TimeType::time_point>("__StartTime"); auto Start = Table.get<std::chrono::high_resolution_clock::time_point>("__StartTime");
return std::chrono::duration_cast<std::chrono::microseconds>(End - Start).count() / 1000000.0f; return std::chrono::duration_cast<std::chrono::microseconds>(End - Start).count() / 1000000.0f;
}); });
Result.set_function("Start", [&](sol::table Table) { Result.set_function("Start", [&](sol::table Table) {
Table["__StartTime"] = TimeType::now(); Table["__StartTime"] = std::chrono::high_resolution_clock::now();
}); });
return Result; return Result;
}); });
@@ -795,10 +745,9 @@ TLuaEngine::StateThreadData::StateThreadData(const std::string& Name, TLuaStateI
MPTable.set_function("TriggerGlobalEvent", [&](const std::string& EventName, sol::variadic_args EventArgs) -> sol::table { MPTable.set_function("TriggerGlobalEvent", [&](const std::string& EventName, sol::variadic_args EventArgs) -> sol::table {
return Lua_TriggerGlobalEvent(EventName, EventArgs); return Lua_TriggerGlobalEvent(EventName, EventArgs);
}); });
MPTable.set_function( MPTable.set_function("TriggerLocalEvent", [&](const std::string& EventName, sol::variadic_args EventArgs) -> sol::table {
"TriggerLocalEvent", [&](const std::string& EventName, sol::variadic_args EventArgs) -> auto{ return Lua_TriggerLocalEvent(EventName, EventArgs);
return Lua_TriggerLocalEvent(EventName, EventArgs); });
});
MPTable.set_function("TriggerClientEvent", &LuaAPI::MP::TriggerClientEvent); MPTable.set_function("TriggerClientEvent", &LuaAPI::MP::TriggerClientEvent);
MPTable.set_function("TriggerClientEventJson", &LuaAPI::MP::TriggerClientEventJson); MPTable.set_function("TriggerClientEventJson", &LuaAPI::MP::TriggerClientEventJson);
MPTable.set_function("GetPlayerCount", &LuaAPI::MP::GetPlayerCount); MPTable.set_function("GetPlayerCount", &LuaAPI::MP::GetPlayerCount);
@@ -869,7 +818,6 @@ TLuaEngine::StateThreadData::StateThreadData(const std::string& Name, TLuaStateI
UtilTable.set_function("JsonUnflatten", &LuaAPI::MP::JsonUnflatten); UtilTable.set_function("JsonUnflatten", &LuaAPI::MP::JsonUnflatten);
UtilTable.set_function("JsonPrettify", &LuaAPI::MP::JsonPrettify); UtilTable.set_function("JsonPrettify", &LuaAPI::MP::JsonPrettify);
UtilTable.set_function("JsonMinify", &LuaAPI::MP::JsonMinify); UtilTable.set_function("JsonMinify", &LuaAPI::MP::JsonMinify);
UtilTable.set_function("GenerateUUID", &uuid::GenerateUuid);
UtilTable.set_function("Random", [this] { UtilTable.set_function("Random", [this] {
return mUniformRealDistribution01(mMersenneTwister); return mUniformRealDistribution01(mMersenneTwister);
}); });
@@ -926,7 +874,7 @@ std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueScript(const TLu
return Result; return Result;
} }
std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueFunctionCallFromCustomEvent(const std::string& FunctionName, const std::vector<TLuaValue>& Args, const std::string& EventName, CallStrategy Strategy) { std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueFunctionCallFromCustomEvent(const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args, const std::string& EventName, CallStrategy Strategy) {
// TODO: Document all this // TODO: Document all this
decltype(mStateFunctionQueue)::iterator Iter = mStateFunctionQueue.end(); decltype(mStateFunctionQueue)::iterator Iter = mStateFunctionQueue.end();
if (Strategy == CallStrategy::BestEffort) { if (Strategy == CallStrategy::BestEffort) {
@@ -948,7 +896,7 @@ std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueFunctionCallFrom
} }
} }
std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueFunctionCall(const std::string& FunctionName, const std::vector<TLuaValue>& Args) { std::shared_ptr<TLuaResult> TLuaEngine::StateThreadData::EnqueueFunctionCall(const std::string& FunctionName, const std::vector<TLuaArgTypes>& Args) {
auto Result = std::make_shared<TLuaResult>(); auto Result = std::make_shared<TLuaResult>();
Result->StateId = mStateId; Result->StateId = mStateId;
Result->Function = FunctionName; Result->Function = FunctionName;
@@ -1033,24 +981,19 @@ void TLuaEngine::StateThreadData::operator()() {
continue; continue;
} }
switch (Arg.index()) { switch (Arg.index()) {
case TLuaType::String: case TLuaArgTypes_String:
LuaArgs.push_back(sol::make_object(StateView, std::get<std::string>(Arg))); LuaArgs.push_back(sol::make_object(StateView, std::get<std::string>(Arg)));
break; break;
case TLuaType::Int: case TLuaArgTypes_Int:
LuaArgs.push_back(sol::make_object(StateView, std::get<int>(Arg))); LuaArgs.push_back(sol::make_object(StateView, std::get<int>(Arg)));
break; break;
case TLuaType::Json: { case TLuaArgTypes_VariadicArgs:
auto Str = std::get<JsonString>(Arg); LuaArgs.push_back(sol::make_object(StateView, std::get<sol::variadic_args>(Arg)));
if (!Str.value.empty()) {
auto LocalArgs = JsonStringToArray(Str);
LuaArgs.insert(LuaArgs.end(), LocalArgs.begin(), LocalArgs.end());
}
break; break;
} case TLuaArgTypes_Bool:
case TLuaType::Bool:
LuaArgs.push_back(sol::make_object(StateView, std::get<bool>(Arg))); LuaArgs.push_back(sol::make_object(StateView, std::get<bool>(Arg)));
break; break;
case TLuaType::StringStringMap: { case TLuaArgTypes_StringStringMap: {
auto Map = std::get<std::unordered_map<std::string, std::string>>(Arg); auto Map = std::get<std::unordered_map<std::string, std::string>>(Arg);
auto Table = StateView.create_table(); auto Table = StateView.create_table();
for (const auto& [k, v] : Map) { for (const auto& [k, v] : Map) {
@@ -1059,15 +1002,6 @@ void TLuaEngine::StateThreadData::operator()() {
LuaArgs.push_back(sol::make_object(StateView, Table)); LuaArgs.push_back(sol::make_object(StateView, Table));
break; break;
} }
case TLuaType::StringSizeTMap: {
auto Map = std::get<std::unordered_map<std::string, size_t>>(Arg);
auto Table = StateView.create_table();
for (const auto& [k, v] : Map) {
Table[k] = v;
}
LuaArgs.push_back(sol::make_object(StateView, Table));
break;
}
default: default:
beammp_error("Unknown argument type, passed as nil"); beammp_error("Unknown argument type, passed as nil");
break; break;
@@ -1106,8 +1040,8 @@ std::vector<TLuaEngine::QueuedFunction> TLuaEngine::StateThreadData::Debug_GetSt
void TLuaEngine::CreateEventTimer(const std::string& EventName, TLuaStateId StateId, size_t IntervalMS, CallStrategy Strategy) { void TLuaEngine::CreateEventTimer(const std::string& EventName, TLuaStateId StateId, size_t IntervalMS, CallStrategy Strategy) {
std::unique_lock Lock(mTimedEventsMutex); std::unique_lock Lock(mTimedEventsMutex);
TimedEvent Event { TimedEvent Event {
TimeType::duration { std::chrono::milliseconds(IntervalMS) }, std::chrono::high_resolution_clock::duration { std::chrono::milliseconds(IntervalMS) },
TimeType::now(), std::chrono::high_resolution_clock::now(),
EventName, EventName,
StateId, StateId,
Strategy Strategy
@@ -1150,10 +1084,10 @@ TLuaChunk::TLuaChunk(std::shared_ptr<std::string> Content, std::string FileName,
} }
bool TLuaEngine::TimedEvent::Expired() { bool TLuaEngine::TimedEvent::Expired() {
auto Waited = (TimeType::now() - LastCompletion); auto Waited = (std::chrono::high_resolution_clock::now() - LastCompletion);
return Waited >= Duration; return Waited >= Duration;
} }
void TLuaEngine::TimedEvent::Reset() { void TLuaEngine::TimedEvent::Reset() {
LastCompletion = TimeType::now(); LastCompletion = std::chrono::high_resolution_clock::now();
} }
+172 -159
View File
@@ -1,7 +1,6 @@
#include "TNetwork.h" #include "TNetwork.h"
#include "Client.h" #include "Client.h"
#include "Common.h" #include "Common.h"
#include "IterationDecision.h"
#include "LuaAPI.h" #include "LuaAPI.h"
#include "TLuaEngine.h" #include "TLuaEngine.h"
#include "nlohmann/json.hpp" #include "nlohmann/json.hpp"
@@ -10,7 +9,6 @@
#include <array> #include <array>
#include <boost/asio/ip/address.hpp> #include <boost/asio/ip/address.hpp>
#include <boost/asio/ip/address_v4.hpp> #include <boost/asio/ip/address_v4.hpp>
#include <chrono>
#include <cstring> #include <cstring>
std::vector<uint8_t> StringToVector(const std::string& Str) { std::vector<uint8_t> StringToVector(const std::string& Str) {
@@ -33,6 +31,15 @@ TNetwork::TNetwork(TServer& Server, TPPSMonitor& PPSMonitor, TResourceManager& R
, mResourceManager(ResourceManager) { , mResourceManager(ResourceManager) {
Application::SetSubsystemStatus("TCPNetwork", Application::Status::Starting); Application::SetSubsystemStatus("TCPNetwork", Application::Status::Starting);
Application::SetSubsystemStatus("UDPNetwork", Application::Status::Starting); Application::SetSubsystemStatus("UDPNetwork", Application::Status::Starting);
Application::RegisterShutdownHandler([&] {
beammp_debug("Kicking all players due to shutdown");
Server.ForEachClient([&](std::weak_ptr<TClient> client) -> bool {
if (!client.expired()) {
ClientKick(*client.lock(), "Server shutdown");
}
return true;
});
});
Application::RegisterShutdownHandler([&] { Application::RegisterShutdownHandler([&] {
Application::SetSubsystemStatus("UDPNetwork", Application::Status::ShuttingDown); Application::SetSubsystemStatus("UDPNetwork", Application::Status::ShuttingDown);
if (mUDPThread.joinable()) { if (mUDPThread.joinable()) {
@@ -49,19 +56,11 @@ TNetwork::TNetwork(TServer& Server, TPPSMonitor& PPSMonitor, TResourceManager& R
}); });
mTCPThread = std::thread(&TNetwork::TCPServerMain, this); mTCPThread = std::thread(&TNetwork::TCPServerMain, this);
mUDPThread = std::thread(&TNetwork::UDPServerMain, this); mUDPThread = std::thread(&TNetwork::UDPServerMain, this);
Application::RegisterShutdownHandler([&] {
auto Futures = LuaAPI::MP::Engine->TriggerEvent("onShutdown", "");
TLuaEngine::WaitForAll(Futures, std::chrono::seconds(60));
beammp_debug("Kicking all players due to shutdown");
Server.ForEachClient([&](const auto& Client) {
ClientKick(*Client, "Server shutdown");
});
});
} }
void TNetwork::UDPServerMain() { void TNetwork::UDPServerMain() {
RegisterThread("UDPServer"); RegisterThread("UDPServer");
ip::udp::endpoint UdpListenEndpoint(ip::address::from_string("0.0.0.0"), Application::GetSettingInt(StrPort)); ip::udp::endpoint UdpListenEndpoint(ip::address::from_string("0.0.0.0"), Application::Settings.Port);
boost::system::error_code ec; boost::system::error_code ec;
mUDPSock.open(UdpListenEndpoint.protocol(), ec); mUDPSock.open(UdpListenEndpoint.protocol(), ec);
if (ec) { if (ec) {
@@ -76,45 +75,35 @@ void TNetwork::UDPServerMain() {
Application::GracefullyShutdown(); Application::GracefullyShutdown();
} }
Application::SetSubsystemStatus("UDPNetwork", Application::Status::Good); Application::SetSubsystemStatus("UDPNetwork", Application::Status::Good);
beammp_info(("Vehicle data network online on port ") + std::to_string(Application::GetSettingInt(StrPort)) + (" with a Max of ") beammp_info(("Vehicle data network online on port ") + std::to_string(Application::Settings.Port) + (" with a Max of ")
+ std::to_string(Application::GetSettingInt(StrMaxPlayers)) + (" Clients")); + std::to_string(Application::Settings.MaxPlayers) + (" Clients"));
while (!Application::IsShuttingDown()) { while (!Application::IsShuttingDown()) {
try { try {
ip::udp::endpoint client {}; ip::udp::endpoint client {};
std::vector<uint8_t> Data = UDPRcvFromClient(client); // Receives any data from Socket std::vector<uint8_t> Data = UDPRcvFromClient(client); // Receives any data from Socket
// It has to be size>=2, because each UDP packet has a header of auto Pos = std::find(Data.begin(), Data.end(), ':');
// <id+1>: if (Data.empty() || Pos > Data.begin() + 2)
// where id+1 is one byte.
// We discard a UDP packet if it doesn't start like this.
if (Data.size() < 2 || Data.at(1) != ':') {
++Application::MalformedUdpPackets;
continue; continue;
}
// We do -1 here, because the launcher does +1
// because it uses (or used to use) null-terminated strings
// to represent packets. This would mean that player 0 would
// cause empty packets.
uint8_t ID = uint8_t(Data.at(0)) - 1; uint8_t ID = uint8_t(Data.at(0)) - 1;
auto Client = GetClient(mServer, ID); mServer.ForEachClient([&](std::weak_ptr<TClient> ClientPtr) -> bool {
if (Client) { std::shared_ptr<TClient> Client;
try { {
if (Client->IsConnected() && Client->GetUDPAddr() != client) { ReadLock Lock(mServer.GetClientMutex());
beammp_debugf("Client at {}:{} tried to send UDP for client {}", client.address().to_string(), client.port(), Client->GetID()); if (!ClientPtr.expired()) {
++Application::InvalidUdpPackets; Client = ClientPtr.lock();
continue; } else
} else { return true;
Client->SetUDPAddr(client);
Client->SetIsConnected(true);
Client->mUDPCONNECTED = true;
}
Client->UdpReceived += Data.size();
++Client->UdpPacketsReceived;
Data.erase(Data.begin() + 0, Data.begin() + 2);
TServer::GlobalParser(Client, std::move(Data), *this);
} catch (const std::exception&) {
++Application::InvalidUdpPackets;
} }
}
if (Client->GetID() == ID) {
Client->SetUDPAddr(client);
Client->SetIsConnected(true);
Data.erase(Data.begin(), Data.begin() + 2);
TServer::GlobalParser(ClientPtr, std::move(Data), mPPSMonitor, *this);
}
return true;
});
} catch (const std::exception& e) { } catch (const std::exception& e) {
beammp_error(("fatal: ") + std::string(e.what())); beammp_error(("fatal: ") + std::string(e.what()));
} }
@@ -124,7 +113,7 @@ void TNetwork::UDPServerMain() {
void TNetwork::TCPServerMain() { void TNetwork::TCPServerMain() {
RegisterThread("TCPServer"); RegisterThread("TCPServer");
ip::tcp::endpoint ListenEp(ip::address::from_string("0.0.0.0"), Application::GetSettingInt(StrPort)); ip::tcp::endpoint ListenEp(ip::address::from_string("0.0.0.0"), Application::Settings.Port);
ip::tcp::socket Listener(mServer.IoCtx()); ip::tcp::socket Listener(mServer.IoCtx());
boost::system::error_code ec; boost::system::error_code ec;
Listener.open(ListenEp.protocol(), ec); Listener.open(ListenEp.protocol(), ec);
@@ -188,16 +177,29 @@ void TNetwork::Identify(TConnection&& RawConnection) {
return; return;
} }
std::shared_ptr<TClient> Client { nullptr }; std::shared_ptr<TClient> Client { nullptr };
if (Code == 'C') { try {
Client = Authentication(std::move(RawConnection)); if (Code == 'C') {
} else if (Code == 'D') { Client = Authentication(std::move(RawConnection));
HandleDownload(std::move(RawConnection)); } else if (Code == 'D') {
} else if (Code == 'P') { HandleDownload(std::move(RawConnection));
} else if (Code == 'P') {
boost::system::error_code ec;
write(RawConnection.Socket, buffer("P"), ec);
return;
} else {
beammp_errorf("Invalid code got in Identify: '{}'", Code);
}
} catch(const std::exception& e) {
beammp_errorf("Error during handling of code {} - client left in invalid state, closing socket", Code);
boost::system::error_code ec; boost::system::error_code ec;
write(RawConnection.Socket, buffer("P"), ec); RawConnection.Socket.shutdown(socket_base::shutdown_both, ec);
return; if (ec) {
} else { beammp_debugf("Failed to shutdown client socket: {}", ec.message());
beammp_errorf("Invalid code got in Identify: '{}'", Code); }
RawConnection.Socket.close(ec);
if (ec) {
beammp_debugf("Failed to close client socket: {}", ec.message());
}
} }
} }
@@ -211,18 +213,20 @@ void TNetwork::HandleDownload(TConnection&& Conn) {
return; return;
} }
auto ID = uint8_t(D); auto ID = uint8_t(D);
mServer.ForEachClient([&](const auto& Client) -> IterationDecision { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
if (Client->GetID() == ID) { ReadLock Lock(mServer.GetClientMutex());
Client->SetDownSock(std::move(Conn.Socket)); if (!ClientPtr.expired()) {
return Break; auto c = ClientPtr.lock();
if (c->GetID() == ID) {
c->SetDownSock(std::move(Conn.Socket));
}
} }
return Continue; return true;
}); });
} }
std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) { std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) {
auto Client = CreateClient(std::move(RawConnection.Socket)); auto Client = CreateClient(std::move(RawConnection.Socket));
Client->ConnectionTime = TimeType::now();
Client->SetIdentifier("ip", RawConnection.SockAddr.address().to_string()); Client->SetIdentifier("ip", RawConnection.SockAddr.address().to_string());
beammp_tracef("This thread is ip {}", RawConnection.SockAddr.address().to_string()); beammp_tracef("This thread is ip {}", RawConnection.SockAddr.address().to_string());
@@ -255,12 +259,24 @@ std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) {
return nullptr; return nullptr;
} }
nlohmann::json AuthReq { std::string key(reinterpret_cast<const char*>(Data.data()), Data.size());
{ "key", std::string(reinterpret_cast<const char*>(Data.data()), Data.size()) }
}; nlohmann::json AuthReq{};
auto Target = "/pkToUser"; std::string AuthResStr{};
unsigned int ResponseCode = 0; try {
const auto AuthResStr = Http::POST(Application::GetBackendUrlForAuth(), 443, Target, AuthReq.dump(), "application/json", &ResponseCode); AuthReq = nlohmann::json {
{ "key", key }
};
auto Target = "/pkToUser";
unsigned int ResponseCode = 0;
AuthResStr = Http::POST(Application::GetBackendUrlForAuth(), 443, Target, AuthReq.dump(), "application/json", &ResponseCode);
} catch (const std::exception& e) {
beammp_debugf("Invalid json sent by client, kicking: {}", e.what());
ClientKick(*Client, "Invalid Key (invalid UTF8 string)!");
return nullptr;
}
try { try {
nlohmann::json AuthRes = nlohmann::json::parse(AuthResStr); nlohmann::json AuthRes = nlohmann::json::parse(AuthResStr);
@@ -274,11 +290,7 @@ std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) {
for (const auto& ID : AuthRes["identifiers"]) { for (const auto& ID : AuthRes["identifiers"]) {
auto Raw = std::string(ID); auto Raw = std::string(ID);
auto SepIndex = Raw.find(':'); auto SepIndex = Raw.find(':');
if (SepIndex != std::string::npos) { Client->SetIdentifier(Raw.substr(0, SepIndex), Raw.substr(SepIndex + 1));
Client->SetIdentifier(Raw.substr(0, SepIndex), Raw.substr(SepIndex + 1));
} else {
beammp_errorf("Invalid response from auth servers: No ':' in identifier '{}', ignoring it", Raw);
}
} }
} else { } else {
beammp_error("Invalid authentication data received from authentication backend"); beammp_error("Invalid authentication data received from authentication backend");
@@ -293,21 +305,24 @@ std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) {
} }
beammp_debug("Name -> " + Client->GetName() + ", Guest -> " + std::to_string(Client->IsGuest()) + ", Roles -> " + Client->GetRoles()); beammp_debug("Name -> " + Client->GetName() + ", Guest -> " + std::to_string(Client->IsGuest()) + ", Roles -> " + Client->GetRoles());
mServer.ForEachClient([&](const auto& ThisClient) -> IterationDecision { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
// FIXME: This doesn't respect forum ID, and it should :^) std::shared_ptr<TClient> Cl;
if (ThisClient->GetName() == Client->GetName() && ThisClient->IsGuest() == Client->IsGuest()) { {
ThisClient->Disconnect("Stale Client (replaced by new client)"); ReadLock Lock(mServer.GetClientMutex());
return Break; if (!ClientPtr.expired()) {
Cl = ClientPtr.lock();
} else
return true;
} }
return Continue; if (Cl->GetName() == Client->GetName() && Cl->IsGuest() == Client->IsGuest()) {
Cl->Disconnect("Stale Client (not a real player)");
return false;
}
return true;
}); });
Client->SetID(OpenID()); auto Futures = LuaAPI::MP::Engine->TriggerEvent("onPlayerAuth", "", Client->GetName(), Client->GetRoles(), Client->IsGuest(), Client->GetIdentifiers());
beammp_info("Assigned ID " + std::to_string(Client->GetID()) + " to " + Client->GetName());
mServer.InsertClient(Client);
auto Futures = LuaAPI::MP::Engine->TriggerEvent("onPlayerAuth", "", Client->GetName(), Client->GetRoles(), Client->IsGuest(), Client->GetIdentifiers(), Client->GetID());
TLuaEngine::WaitForAll(Futures); TLuaEngine::WaitForAll(Futures);
bool NotAllowed = std::any_of(Futures.begin(), Futures.end(), bool NotAllowed = std::any_of(Futures.begin(), Futures.end(),
[](const std::shared_ptr<TLuaResult>& Result) { [](const std::shared_ptr<TLuaResult>& Result) {
@@ -323,24 +338,22 @@ std::shared_ptr<TClient> TNetwork::Authentication(TConnection&& RawConnection) {
return false; return false;
}); });
bool fitsOnServer = mServer.ClientCount() < size_t(Application::GetSettingInt(StrMaxPlayers)); // || luaplayercountbypass;
if (!NotAllowed && !NotAllowedWithReason && fitsOnServer) {
beammp_info("Identification success");
TCPClient(Client);
return Client;
}
if (NotAllowed) { if (NotAllowed) {
ClientKick(*Client, "You are not allowed on the server!"); ClientKick(*Client, "you are not allowed on the server!");
return {};
} else if (NotAllowedWithReason) { } else if (NotAllowedWithReason) {
ClientKick(*Client, Reason); ClientKick(*Client, Reason);
return {};
}
if (mServer.ClientCount() < size_t(Application::Settings.MaxPlayers)) {
beammp_info("Identification success");
mServer.InsertClient(Client);
TCPClient(Client);
} else { } else {
ClientKick(*Client, "Server full!"); ClientKick(*Client, "Server full!");
} }
return Client;
mServer.RemoveClient(Client);
return {};
} }
std::shared_ptr<TClient> TNetwork::CreateClient(ip::tcp::socket&& TCPSock) { std::shared_ptr<TClient> TNetwork::CreateClient(ip::tcp::socket&& TCPSock) {
@@ -383,12 +396,11 @@ bool TNetwork::TCPSend(TClient& c, const std::vector<uint8_t>& Data, bool IsSync
return false; return false;
} }
c.UpdatePingTime(); c.UpdatePingTime();
c.TcpSent += ToSend.size();
return true; return true;
} }
std::vector<uint8_t> TNetwork::TCPRcv(TClient& c) { std::vector<uint8_t> TNetwork::TCPRcv(TClient& c) {
if (c.IsDisconnected() || (c.mUDPCONNECTED && !c.IsConnected())) { if (c.IsDisconnected()) {
beammp_error("Client disconnected, cancelling TCPRcv"); beammp_error("Client disconnected, cancelling TCPRcv");
return {}; return {};
} }
@@ -432,7 +444,6 @@ std::vector<uint8_t> TNetwork::TCPRcv(TClient& c) {
beammp_errorf("Expected to read {} bytes, instead got {}", Header, N); beammp_errorf("Expected to read {} bytes, instead got {}", Header, N);
} }
c.TcpReceived += N + HeaderData.size();
constexpr std::string_view ABG = "ABG:"; constexpr std::string_view ABG = "ABG:";
if (Data.size() >= ABG.size() && std::equal(Data.begin(), Data.begin() + ABG.size(), ABG.begin(), ABG.end())) { if (Data.size() >= ABG.size() && std::equal(Data.begin(), Data.begin() + ABG.size(), ABG.begin(), ABG.end())) {
Data.erase(Data.begin(), Data.begin() + ABG.size()); Data.erase(Data.begin(), Data.begin() + ABG.size());
@@ -512,7 +523,7 @@ void TNetwork::TCPClient(const std::weak_ptr<TClient>& c) {
Client->Disconnect("TCPRcv failed"); Client->Disconnect("TCPRcv failed");
break; break;
} }
TServer::GlobalParser(c, std::move(res), *this); TServer::GlobalParser(c, std::move(res), mPPSMonitor, *this);
} }
if (QueueSync.joinable()) if (QueueSync.joinable())
@@ -527,9 +538,14 @@ void TNetwork::TCPClient(const std::weak_ptr<TClient>& c) {
} }
void TNetwork::UpdatePlayer(TClient& Client) { void TNetwork::UpdatePlayer(TClient& Client) {
std::string Packet = ("Ss") + std::to_string(mServer.ClientCount()) + "/" + std::to_string(Application::GetSettingInt(StrMaxPlayers)) + ":"; std::string Packet = ("Ss") + std::to_string(mServer.ClientCount()) + "/" + std::to_string(Application::Settings.MaxPlayers) + ":";
mServer.ForEachClient([&](const auto& ThisClient) { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
Packet += ThisClient->GetName() + ","; ReadLock Lock(mServer.GetClientMutex());
if (!ClientPtr.expired()) {
auto c = ClientPtr.lock();
Packet += c->GetName() + ",";
}
return true;
}); });
Packet = Packet.substr(0, Packet.length() - 1); Packet = Packet.substr(0, Packet.length() - 1);
Client.EnqueuePacket(StringToVector(Packet)); Client.EnqueuePacket(StringToVector(Packet));
@@ -567,19 +583,20 @@ void TNetwork::OnDisconnect(const std::weak_ptr<TClient>& ClientPtr) {
} }
int TNetwork::OpenID() { int TNetwork::OpenID() {
// This lock ensures that each call to OpenID is exclusive.
// If we didn't have this, two concurrent calls to this function may result
// in the same ID.
std::unique_lock Lock(mOpenIDMutex);
int ID = 0; int ID = 0;
bool found; bool found;
do { do {
found = true; found = true;
mServer.ForEachClient([&](const auto& Client) { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
if (Client->GetID() == ID) { ReadLock Lock(mServer.GetClientMutex());
found = false; if (!ClientPtr.expired()) {
ID++; auto c = ClientPtr.lock();
if (c->GetID() == ID) {
found = false;
ID++;
}
} }
return true;
}); });
} while (!found); } while (!found);
return ID; return ID;
@@ -589,11 +606,13 @@ void TNetwork::OnConnect(const std::weak_ptr<TClient>& c) {
beammp_assert(!c.expired()); beammp_assert(!c.expired());
beammp_info("Client connected"); beammp_info("Client connected");
auto LockedClient = c.lock(); auto LockedClient = c.lock();
LockedClient->SetID(OpenID());
beammp_info("Assigned ID " + std::to_string(LockedClient->GetID()) + " to " + LockedClient->GetName());
LuaAPI::MP::Engine->ReportErrors(LuaAPI::MP::Engine->TriggerEvent("onPlayerConnecting", "", LockedClient->GetID())); LuaAPI::MP::Engine->ReportErrors(LuaAPI::MP::Engine->TriggerEvent("onPlayerConnecting", "", LockedClient->GetID()));
SyncResources(*LockedClient); SyncResources(*LockedClient);
if (LockedClient->IsDisconnected()) if (LockedClient->IsDisconnected())
return; return;
(void)Respond(*LockedClient, StringToVector("M" + Application::GetSettingString(StrMap)), true); // Send the Map on connect (void)Respond(*LockedClient, StringToVector("M" + Application::Settings.MapName), true); // Send the Map on connect
beammp_info(LockedClient->GetName() + " : Connected"); beammp_info(LockedClient->GetName() + " : Connected");
LuaAPI::MP::Engine->ReportErrors(LuaAPI::MP::Engine->TriggerEvent("onPlayerJoining", "", LockedClient->GetID())); LuaAPI::MP::Engine->ReportErrors(LuaAPI::MP::Engine->TriggerEvent("onPlayerJoining", "", LockedClient->GetID()));
} }
@@ -611,39 +630,11 @@ void TNetwork::SyncResources(TClient& c) {
constexpr std::string_view Done = "Done"; constexpr std::string_view Done = "Done";
if (std::equal(Data.begin(), Data.end(), Done.begin(), Done.end())) if (std::equal(Data.begin(), Data.end(), Done.begin(), Done.end()))
break; break;
HandleResourcePackets(c, Data); Parse(c, Data);
} }
} }
ModMap TNetwork::GetClientMods(TClient& Client) { void TNetwork::Parse(TClient& c, const std::vector<uint8_t>& Packet) {
auto AllMods = mResourceManager.FileMap();
auto Futures = LuaAPI::MP::Engine->TriggerEvent("onPlayerRequestMods", "", Client.GetName(), Client.GetRoles(), Client.IsGuest(), Client.GetIdentifiers(), Client.GetID(), AllMods);
TLuaEngine::WaitForAll(Futures);
ModMap AllowedMods = AllMods;
for (const std::shared_ptr<TLuaResult>& Future : Futures) {
sol::table Result;
if (!Future->Error && Future->Result.is<sol::table>()) {
Result = Future->Result.as<sol::table>();
for (const auto& [name, size] : AllMods) {
auto val = Result.get<sol::optional<int>>(name);
if (!val.has_value()) {
AllowedMods.erase(name);
beammp_debugf("Not sending mod '{}' to player '{}' (from state '{}')", name, Client.GetName(), Future->StateId);
}
}
}
}
Client.AllowedMods = AllowedMods;
return AllowedMods;
}
void TNetwork::HandleResourcePackets(TClient& c, const std::vector<uint8_t>& Packet) {
if (Packet.empty()) if (Packet.empty())
return; return;
char Code = Packet.at(0), SubCode = 0; char Code = Packet.at(0), SubCode = 0;
@@ -656,7 +647,7 @@ void TNetwork::HandleResourcePackets(TClient& c, const std::vector<uint8_t>& Pac
case 'S': case 'S':
if (SubCode == 'R') { if (SubCode == 'R') {
beammp_debug("Sending Mod Info"); beammp_debug("Sending Mod Info");
std::string ToSend = TResourceManager::FormatForClient(GetClientMods(c)); std::string ToSend = mResourceManager.FileList() + mResourceManager.FileSizes();
if (ToSend.empty()) if (ToSend.empty())
ToSend = "-"; ToSend = "-";
if (!TCPSend(c, StringToVector(ToSend))) { if (!TCPSend(c, StringToVector(ToSend))) {
@@ -670,22 +661,28 @@ void TNetwork::HandleResourcePackets(TClient& c, const std::vector<uint8_t>& Pac
} }
void TNetwork::SendFile(TClient& c, const std::string& UnsafeName) { void TNetwork::SendFile(TClient& c, const std::string& UnsafeName) {
beammp_infof("{} ({}) requesting mod: '{}'", c.GetName(), c.GetID(), UnsafeName); beammp_info(c.GetName() + " requesting : " + UnsafeName.substr(UnsafeName.find_last_of('/')));
auto FileName = UnsafeName; if (!fs::path(UnsafeName).has_filename()) {
auto res = TResourceManager::IsModValid(FileName, c.AllowedMods);
if (res.has_value()) {
if (!TCPSend(c, StringToVector("CO"))) { if (!TCPSend(c, StringToVector("CO"))) {
c.Disconnect("TCP send failed in SendFile, when trying to cancel file transfer because " + res.value()); // TODO: handle
} }
beammp_warn("File " + UnsafeName + " is not a file!");
return;
}
auto FileName = fs::path(UnsafeName).filename().string();
FileName = Application::Settings.Resource + "/Client/" + FileName;
if (!std::filesystem::exists(FileName)) {
if (!TCPSend(c, StringToVector("CO"))) {
// TODO: handle
}
beammp_warn("File " + UnsafeName + " could not be accessed!");
return; return;
} }
if (!TCPSend(c, StringToVector("AG"))) { if (!TCPSend(c, StringToVector("AG"))) {
c.Disconnect("TCP send failed in SendFile, when trying to send "); // TODO: handle
return;
} }
/// Wait for connections /// Wait for connections
@@ -815,7 +812,6 @@ bool TNetwork::TCPSendRaw(TClient& C, ip::tcp::socket& socket, const uint8_t* Da
return false; return false;
} }
C.UpdatePingTime(); C.UpdatePingTime();
C.TcpSent += Size;
return true; return true;
} }
@@ -858,23 +854,32 @@ bool TNetwork::SyncClient(const std::weak_ptr<TClient>& c) {
LockedClient->SetIsSyncing(true); LockedClient->SetIsSyncing(true);
bool Return = false; bool Return = false;
bool res = true; bool res = true;
mServer.ForEachClient([&](const auto& Client) -> IterationDecision { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
std::shared_ptr<TClient> client;
{
ReadLock Lock(mServer.GetClientMutex());
if (!ClientPtr.expired()) {
client = ClientPtr.lock();
} else
return true;
}
TClient::TSetOfVehicleData VehicleData; TClient::TSetOfVehicleData VehicleData;
{ // Vehicle Data Lock Scope { // Vehicle Data Lock Scope
auto LockedData = Client->GetAllCars(); auto LockedData = client->GetAllCars();
VehicleData = *LockedData.VehicleData; VehicleData = *LockedData.VehicleData;
} // End Vehicle Data Lock Scope } // End Vehicle Data Lock Scope
if (Client != LockedClient) { if (client != LockedClient) {
for (auto& v : VehicleData) { for (auto& v : VehicleData) {
if (LockedClient->IsDisconnected()) { if (LockedClient->IsDisconnected()) {
Return = true; Return = true;
res = false; res = false;
return Break; return false;
} }
res = Respond(*LockedClient, StringToVector(v.Data()), true, true); res = Respond(*LockedClient, StringToVector(v.Data()), true, true);
} }
} }
return Continue;
return true;
}); });
LockedClient->SetIsSyncing(false); LockedClient->SetIsSyncing(false);
if (Return) { if (Return) {
@@ -890,7 +895,16 @@ void TNetwork::SendToAll(TClient* c, const std::vector<uint8_t>& Data, bool Self
beammp_assert(c); beammp_assert(c);
char C = Data.at(0); char C = Data.at(0);
bool ret = true; bool ret = true;
mServer.ForEachClient([&](const auto& Client) { mServer.ForEachClient([&](std::weak_ptr<TClient> ClientPtr) -> bool {
std::shared_ptr<TClient> Client;
try {
ReadLock Lock(mServer.GetClientMutex());
Client = ClientPtr.lock();
} catch (const std::exception&) {
// continue
beammp_warn("Client expired, shouldn't happen - if a client disconnected recently, you can ignore this");
return true;
}
if (Self || Client.get() != c) { if (Self || Client.get() != c) {
if (Client->IsSynced() || Client->IsSyncing()) { if (Client->IsSynced() || Client->IsSyncing()) {
if (Rel || C == 'W' || C == 'Y' || C == 'V' || C == 'E') { if (Rel || C == 'W' || C == 'Y' || C == 'V' || C == 'E') {
@@ -912,6 +926,7 @@ void TNetwork::SendToAll(TClient* c, const std::vector<uint8_t>& Data, bool Self
} }
} }
} }
return true;
}); });
if (!ret) { if (!ret) {
// TODO: handle // TODO: handle
@@ -939,8 +954,6 @@ bool TNetwork::UDPSend(TClient& Client, std::vector<uint8_t> Data) {
Client.Disconnect("UDP send failed"); Client.Disconnect("UDP send failed");
return false; return false;
} }
Client.UdpSent += Data.size();
++Client.UdpPacketsSent;
return true; return true;
} }
+16 -6
View File
@@ -33,16 +33,26 @@ void TPPSMonitor::operator()() {
Application::SetPPS("-"); Application::SetPPS("-");
continue; continue;
} }
mServer.ForEachClient([&](const auto& Client) { mServer.ForEachClient([&](const std::weak_ptr<TClient>& ClientPtr) -> bool {
if (Client->GetCarCount() > 0) { std::shared_ptr<TClient> c;
{
ReadLock Lock(mServer.GetClientMutex());
if (!ClientPtr.expired()) {
c = ClientPtr.lock();
} else
return true;
}
if (c->GetCarCount() > 0) {
C++; C++;
V += Client->GetCarCount(); V += c->GetCarCount();
} }
// kick on "no ping" // kick on "no ping"
if (Client->SecondsSinceLastPing() > (20 * 60)) { if (c->SecondsSinceLastPing() > (20 * 60)) {
beammp_debugf("Client {} ({}) timing out: {}s since last contact", Client->GetName(), Client->GetID(), Client->SecondsSinceLastPing()); beammp_debug("client " + std::string("(") + std::to_string(c->GetID()) + ")" + c->GetName() + " timing out: " + std::to_string(c->SecondsSinceLastPing()) + ", pps: " + Application::PPS());
TimedOutClients.push_back(Client); TimedOutClients.push_back(c);
} }
return true;
}); });
for (auto& ClientToKick : TimedOutClients) { for (auto& ClientToKick : TimedOutClients) {
Network().ClientKick(*ClientToKick, "Timeout (no ping for way too long)"); Network().ClientKick(*ClientToKick, "Timeout (no ping for way too long)");
+20 -68
View File
@@ -5,79 +5,31 @@
namespace fs = std::filesystem; namespace fs = std::filesystem;
std::string TResourceManager::FormatForBackend(const ModMap& mods) {
std::string monkey;
for (const auto& [name, size] : mods) {
monkey += fs::path(name).filename().string() + ';';
}
return monkey;
}
std::string TResourceManager::FormatForClient(const ModMap& mods) {
std::string monkey;
for (const auto& [name, size] : mods) {
monkey += '/' + name + ';';
}
for (const auto& [name, size] : mods) {
monkey += std::to_string(size) + ';';
}
return monkey;
}
/// @brief Sanitizes a requested mod string
/// @param pathString Raw mod path string
/// @param mods List of allowed mods for this client
/// @return Error, if any
std::optional<std::string> TResourceManager::IsModValid(std::string& pathString, const ModMap& mods) {
auto path = fs::path(pathString);
if (!path.has_filename()) {
beammp_warn("File " + pathString + " is not a file!");
return { "the requested file doesn't contain a valid filename" };
}
auto BasePath = fs::path(Application::GetSettingString(StrResourceFolder) + "/Client");
auto CombinedPath = fs::path(BasePath.string() + pathString).lexically_normal();
// beammp_infof("path: {}, base: {}, combined: {}", pathString, BasePath.string(), CombinedPath.string());
if (!std::filesystem::exists(CombinedPath)) {
beammp_warn("File " + pathString + " could not be accessed!");
return { "the requested file doesn't exist or couldn't be accessed" };
}
auto relative = fs::relative(CombinedPath, BasePath);
if (mods.count(relative.string()) == 0) {
beammp_warn("File " + pathString + " is disallowed for this player!");
return { "the requested file is disallowed for this player" };
}
pathString = CombinedPath.string();
return {};
}
TResourceManager::TResourceManager() { TResourceManager::TResourceManager() {
Application::SetSubsystemStatus("ResourceManager", Application::Status::Starting); Application::SetSubsystemStatus("ResourceManager", Application::Status::Starting);
std::string BasePath = Application::GetSettingString(StrResourceFolder) + "/Client"; std::string Path = Application::Settings.Resource + "/Client";
if (!fs::exists(BasePath)) if (!fs::exists(Path))
fs::create_directories(BasePath); fs::create_directories(Path);
std::vector<std::string> modNames; for (const auto& entry : fs::directory_iterator(Path)) {
std::string File(entry.path().string());
auto iterator = fs::recursive_directory_iterator(BasePath, fs::directory_options::follow_directory_symlink | fs::directory_options::skip_permission_denied); if (auto pos = File.find(".zip"); pos != std::string::npos) {
for (const auto& entry : iterator) { if (File.length() - pos == 4) {
if (iterator.depth() > 0 && !Application::GetSettingBool(StrIncludeSubdirectories)) std::replace(File.begin(), File.end(), '\\', '/');
continue; mFileList += File + ';';
if ((entry.is_regular_file() || entry.is_symlink()) && entry.path().extension() == ".zip") { if (auto i = File.find_last_of('/'); i != std::string::npos) {
auto relativePath = fs::relative(entry.path(), BasePath); ++i;
File = File.substr(i, pos - i);
mMods[relativePath.string()] = entry.file_size(); }
mTotalModSize += entry.file_size(); mTrimmedList += "/" + fs::path(File).filename().string() + ';';
mFileSizes += std::to_string(size_t(fs::file_size(entry.path()))) + ';';
mMaxModSize += size_t(fs::file_size(entry.path()));
mModsLoaded++;
}
} }
} }
if (!mMods.empty()) { if (mModsLoaded) {
beammp_infof("Loaded {} mod{}", mMods.size(), mMods.size() != 1 ? 's' : ' '); beammp_info("Loaded " + std::to_string(mModsLoaded) + " Mods");
} }
Application::SetSubsystemStatus("ResourceManager", Application::Status::Good); Application::SetSubsystemStatus("ResourceManager", Application::Status::Good);
+4 -4
View File
@@ -2,21 +2,21 @@
#include "Common.h" #include "Common.h"
TScopedTimer::TScopedTimer() TScopedTimer::TScopedTimer()
: mStartTime(TimeType::now()) { : mStartTime(std::chrono::high_resolution_clock::now()) {
} }
TScopedTimer::TScopedTimer(const std::string& mName) TScopedTimer::TScopedTimer(const std::string& mName)
: mStartTime(TimeType::now()) : mStartTime(std::chrono::high_resolution_clock::now())
, Name(mName) { , Name(mName) {
} }
TScopedTimer::TScopedTimer(std::function<void(size_t)> OnDestroy) TScopedTimer::TScopedTimer(std::function<void(size_t)> OnDestroy)
: OnDestroy(OnDestroy) : OnDestroy(OnDestroy)
, mStartTime(TimeType::now()) { , mStartTime(std::chrono::high_resolution_clock::now()) {
} }
TScopedTimer::~TScopedTimer() { TScopedTimer::~TScopedTimer() {
auto EndTime = TimeType::now(); auto EndTime = std::chrono::high_resolution_clock::now();
auto Delta = EndTime - mStartTime; auto Delta = EndTime - mStartTime;
size_t TimeDelta = Delta / std::chrono::milliseconds(1); size_t TimeDelta = Delta / std::chrono::milliseconds(1);
if (OnDestroy) { if (OnDestroy) {
+6 -6
View File
@@ -28,22 +28,22 @@ TSentry::~TSentry() {
void TSentry::PrintWelcome() { void TSentry::PrintWelcome() {
if (mValid) { if (mValid) {
if (!Application::GetSettingBool("SendErrors")) { if (!Application::Settings.SendErrors) {
mValid = false; mValid = false;
if (Application::GetSettingBool(StrSendErrors)) { if (Application::Settings.SendErrorsMessageEnabled) {
beammp_info("Opted out of error reporting (SendErrors), Sentry disabled."); beammp_info("Opted out of error reporting (SendErrors), Sentry disabled.");
} else { } else {
beammp_info("Sentry disabled"); beammp_info("Sentry disabled");
} }
} else { } else {
if (Application::GetSettingBool(StrSendErrors)) { if (Application::Settings.SendErrorsMessageEnabled) {
beammp_info("Sentry started! Reporting errors automatically. This sends data to the developers in case of errors and crashes. You can learn more, turn this message off or opt-out of this in the ServerConfig.toml."); beammp_info("Sentry started! Reporting errors automatically. This sends data to the developers in case of errors and crashes. You can learn more, turn this message off or opt-out of this in the ServerConfig.toml.");
} else { } else {
beammp_info("Sentry started"); beammp_info("Sentry started");
} }
} }
} else { } else {
if (Application::GetSettingBool(StrSendErrors)) { if (Application::Settings.SendErrorsMessageEnabled) {
beammp_info("Sentry disabled in unofficial build. Automatic error reporting disabled."); beammp_info("Sentry disabled in unofficial build. Automatic error reporting disabled.");
} else { } else {
beammp_info("Sentry disabled in unofficial build"); beammp_info("Sentry disabled in unofficial build");
@@ -57,8 +57,8 @@ void TSentry::SetupUser() {
} }
Application::SetSubsystemStatus("Sentry", Application::Status::Good); Application::SetSubsystemStatus("Sentry", Application::Status::Good);
sentry_value_t user = sentry_value_new_object(); sentry_value_t user = sentry_value_new_object();
if (Application::GetSettingString(StrAuthKey).size() == 36) { if (Application::Settings.Key.size() == 36) {
sentry_value_set_by_key(user, "id", sentry_value_new_string(Application::GetSettingString(StrAuthKey).c_str())); sentry_value_set_by_key(user, "id", sentry_value_new_string(Application::Settings.Key.c_str()));
} else { } else {
sentry_value_set_by_key(user, "id", sentry_value_new_string("unauthenticated")); sentry_value_set_by_key(user, "id", sentry_value_new_string("unauthenticated"));
} }
+48 -125
View File
@@ -2,81 +2,35 @@
#include "Client.h" #include "Client.h"
#include "Common.h" #include "Common.h"
#include "CustomAssert.h" #include "CustomAssert.h"
#include "IterationDecision.h"
#include "TNetwork.h" #include "TNetwork.h"
#include "TPPSMonitor.h" #include "TPPSMonitor.h"
#include <TLuaPlugin.h> #include <TLuaPlugin.h>
#include <algorithm> #include <algorithm>
#include <any> #include <any>
#include <nlohmann/json.hpp>
#include <sstream> #include <sstream>
#include <nlohmann/json.hpp>
#include "LuaAPI.h" #include "LuaAPI.h"
#undef GetObject // Fixes Windows #undef GetObject // Fixes Windows
#include "Json.h" #include "Json.h"
struct VehiclePacket { static std::optional<std::pair<int, int>> GetPidVid(const std::string& str) {
std::string Data;
int Pid;
int Vid;
};
static Result<std::pair<int, int>> GetPidVid(const std::string& str) {
auto IDSep = str.find('-'); auto IDSep = str.find('-');
if (IDSep == std::string::npos) {
return Error("Invalid packet: Could not parse pid/vid from packet, as there is no '-' separator: '{}'", str);
}
std::string pid = str.substr(0, IDSep); std::string pid = str.substr(0, IDSep);
std::string vid = str.substr(IDSep + 1); std::string vid = str.substr(IDSep + 1);
if (pid.find_first_not_of("0123456789") == std::string::npos && vid.find_first_not_of("0123456789") == std::string::npos) { if (pid.find_first_not_of("0123456789") == std::string::npos && vid.find_first_not_of("0123456789") == std::string::npos) {
try { try {
int PID = std::stoi(pid); int PID = stoi(pid);
int VID = std::stoi(vid); int VID = stoi(vid);
return { { PID, VID } }; return { { PID, VID } };
} catch (const std::exception&) { } catch (const std::exception&) {
return Error("Invalid packet: Could not parse pid/vid from packet, as one or both are not valid numbers: '{}'", str); return std::nullopt;
} }
} }
return Error("Invalid packet: Could not parse pid/vid from packet: '{}'", str);
}
static std::optional<VehiclePacket> ParseVehiclePacket(const std::string& Packet, const int playerID) {
if (Packet.size() < 8) { // 2 for code, 3<= for pidvid, 1<= for data, 2 for dividers
// invalid packet
return std::nullopt;
}
// Zp:serverVehicleID:data
// 0-0:data
std::string withoutCode = Packet.substr(3);
auto NameDataSep = withoutCode.find(':', 3);
if (NameDataSep == std::string::npos) {
beammp_debugf("Invalid packet from {} in ParseVehiclePacket: No ':' separator, assuming corrupted packet. Packet: '{}'", playerID, Packet);
return std::nullopt;
}
if (NameDataSep + 1 > withoutCode.size()) {
beammp_debugf("Invalid packet from {} in ParseVehiclePacket: Separator in unexpected place, assuming corrupted packet. Packet: '{}'", playerID, Packet);
return std::nullopt;
}
std::string ServerVehicleID = withoutCode.substr(0, NameDataSep);
std::string Data = withoutCode.substr(NameDataSep + 1);
// parse veh ID
auto MaybePidVid = GetPidVid(ServerVehicleID);
if (MaybePidVid) {
int PID, VID;
std::tie(PID, VID) = MaybePidVid.value();
if (PID == playerID) {
return { { Data, PID, VID } }; // std::vector<char>(Data.begin(), Data.end())
}
}
beammp_debugf("Failed to parse packet from player {}", playerID);
return std::nullopt; return std::nullopt;
} }
@@ -123,48 +77,15 @@ TEST_CASE("GetPidVid") {
} }
} }
TEST_CASE("ParseVehiclePacket") {
SUBCASE("Valid packet") {
const auto valid = ParseVehiclePacket("Zp:0-0:{jsonstring}", 0);
CHECK(valid.has_value());
}
SUBCASE("Valid packet 2") {
const auto packet = ParseVehiclePacket("Zp:12-3:{jsonstring}", 12);
CHECK(packet.has_value());
CHECK_EQ(packet.value().Pid, 12);
CHECK_EQ(packet.value().Vid, 3);
}
SUBCASE("Missing packet") {
const auto valid = ParseVehiclePacket("", 0);
CHECK(!valid.has_value());
}
SUBCASE("Missing ServerVehicleID") {
const auto valid = ParseVehiclePacket("Zp:{jsonstring}", 0);
CHECK(!valid.has_value());
}
SUBCASE("Missing data") {
const auto valid = ParseVehiclePacket("Zp:0-0:", 0);
CHECK(!valid.has_value());
}
SUBCASE("Incorrect Pid") {
const auto valid = ParseVehiclePacket("Zp:0-0:{jsonstring}", 1);
CHECK(!valid.has_value());
}
SUBCASE("Incorrect Pid 2") {
const auto valid = ParseVehiclePacket("Zp:12-0:{jsonstring}", 13);
CHECK(!valid.has_value());
}
}
TServer::TServer(const std::vector<std::string_view>& Arguments) { TServer::TServer(const std::vector<std::string_view>& Arguments) {
beammp_info("BeamMP Server v" + Application::ServerVersionString());
Application::SetSubsystemStatus("Server", Application::Status::Starting); Application::SetSubsystemStatus("Server", Application::Status::Starting);
if (Arguments.size() > 1) { if (Arguments.size() > 1) {
Application::SetSetting(StrCustomIP, std::string(Arguments[0])); Application::Settings.CustomIP = Arguments[0];
auto CustomIP = Application::GetSettingString(StrCustomIP); size_t n = std::count(Application::Settings.CustomIP.begin(), Application::Settings.CustomIP.end(), '.');
size_t n = std::count(CustomIP.begin(), CustomIP.end(), '.'); auto p = Application::Settings.CustomIP.find_first_not_of(".0123456789");
auto p = Application::GetSettingString(StrCustomIP).find_first_not_of(".0123456789"); if (p != std::string::npos || n != 3 || Application::Settings.CustomIP.substr(0, 3) == "127") {
if (p != std::string::npos || n != 3 || CustomIP.substr(0, 3) == "127") { Application::Settings.CustomIP.clear();
Application::SetSetting(StrCustomIP, "");
beammp_warn("IP Specified is invalid! Ignoring"); beammp_warn("IP Specified is invalid! Ignoring");
} else { } else {
beammp_info("server started with custom IP"); beammp_info("server started with custom IP");
@@ -190,16 +111,7 @@ void TServer::RemoveClient(const std::weak_ptr<TClient>& WeakClientPtr) {
mClients.erase(WeakClientPtr.lock()); mClients.erase(WeakClientPtr.lock());
} }
void TServer::RemoveClientById(int Id) { void TServer::ForEachClient(const std::function<bool(std::weak_ptr<TClient>)>& Fn) {
auto Client = GetClient(*this, Id);
if (Client) {
Client->ClearCars();
WriteLock Lock(mClientsMutex);
mClients.erase(Client);
}
}
void TServer::ForEachClientWeak(const std::function<bool(std::weak_ptr<TClient>)>& Fn) {
decltype(mClients) Clients; decltype(mClients) Clients;
{ {
ReadLock lock(mClientsMutex); ReadLock lock(mClientsMutex);
@@ -217,7 +129,7 @@ size_t TServer::ClientCount() const {
return mClients.size(); return mClients.size();
} }
void TServer::GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uint8_t>&& Packet, TNetwork& Network) { void TServer::GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uint8_t>&& Packet, TPPSMonitor& PPSMonitor, TNetwork& Network) {
constexpr std::string_view ABG = "ABG:"; constexpr std::string_view ABG = "ABG:";
if (Packet.size() >= ABG.size() && std::equal(Packet.begin(), Packet.begin() + ABG.size(), ABG.begin(), ABG.end())) { if (Packet.size() >= ABG.size() && std::equal(Packet.begin(), Packet.begin() + ABG.size(), ABG.begin(), ABG.end())) {
Packet.erase(Packet.begin(), Packet.begin() + ABG.size()); Packet.erase(Packet.begin(), Packet.begin() + ABG.size());
@@ -239,11 +151,8 @@ void TServer::GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uin
// V to Y // V to Y
if (Code <= 89 && Code >= 86) { if (Code <= 89 && Code >= 86) {
if (HandleVehicleUpdate(StringPacket, LockedClient->GetID())) { PPSMonitor.IncrementInternalPPS();
Network.SendToAll(LockedClient.get(), Packet, false, false); Network.SendToAll(LockedClient.get(), Packet, false, false);
} else {
beammp_debugf("Invalid vehicle update packet received from '{}' ({}), ignoring it", LockedClient->GetName(), LockedClient->GetID());
}
return; return;
} }
switch (Code) { switch (Code) {
@@ -302,11 +211,9 @@ void TServer::GlobalParser(const std::weak_ptr<TClient>& Client, std::vector<uin
Network.SendToAll(LockedClient.get(), Packet, false, true); Network.SendToAll(LockedClient.get(), Packet, false, true);
return; return;
case 'Z': // position packet case 'Z': // position packet
if (HandlePosition(*LockedClient, StringPacket)) { PPSMonitor.IncrementInternalPPS();
Network.SendToAll(LockedClient.get(), Packet, false, false); Network.SendToAll(LockedClient.get(), Packet, false, false);
} else { HandlePosition(*LockedClient, StringPacket);
beammp_debugf("Invalid vehicle position packet received from '{}' ({}), ignoring it", LockedClient->GetName(), LockedClient->GetID());
}
default: default:
return; return;
} }
@@ -346,7 +253,7 @@ bool TServer::ShouldSpawn(TClient& c, const std::string& CarJson, int ID) {
c.SetUnicycleID(ID); c.SetUnicycleID(ID);
return true; return true;
} else { } else {
return c.GetCarCount() < Application::GetSettingInt(StrMaxCars); return c.GetCarCount() < Application::Settings.MaxCars;
} }
} }
@@ -525,19 +432,35 @@ void TServer::InsertClient(const std::shared_ptr<TClient>& NewClient) {
(void)mClients.insert(NewClient); (void)mClients.insert(NewClient);
} }
bool TServer::HandlePosition(TClient& c, const std::string& PacketStr) { void TServer::HandlePosition(TClient& c, const std::string& Packet) {
auto MaybePacket = ParseVehiclePacket(PacketStr, c.GetID()); if (Packet.size() < 3) {
// invalid packet
if (MaybePacket) { return;
auto packet = MaybePacket.value();
c.SetCarPosition(packet.Vid, packet.Data);
return true;
} }
return false; // Zp:serverVehicleID:data
} // Zp:0:data
std::string withoutCode = Packet.substr(3);
auto NameDataSep = withoutCode.find(':', 2);
if (NameDataSep == std::string::npos || NameDataSep < 2) {
// invalid packet
return;
}
// FIXME: ensure that -2 does what it should... it seems weird.
std::string ServerVehicleID = withoutCode.substr(2, NameDataSep - 2);
if (NameDataSep + 1 > withoutCode.size()) {
// invalid packet
return;
}
std::string Data = withoutCode.substr(NameDataSep + 1);
bool TServer::HandleVehicleUpdate(const std::string& PacketStr, const int playerID) { // parse veh ID
auto MaybePacket = ParseVehiclePacket(PacketStr, playerID); auto MaybePidVid = GetPidVid(ServerVehicleID);
if (MaybePidVid) {
int PID = -1;
int VID = -1;
// FIXME: check that the VID and PID are valid, so that we don't waste memory
std::tie(PID, VID) = MaybePidVid.value();
return MaybePacket.has_value(); c.SetCarPosition(VID, Data);
}
} }
-15
View File
@@ -1,15 +0,0 @@
#include "Uuid.h"
#include <boost/uuid/uuid.hpp>
#include <boost/uuid/random_generator.hpp>
#include <boost/uuid/uuid_generators.hpp>
#include <boost/uuid/uuid_io.hpp>
#include <fmt/format.h>
#include <fmt/ostream.h>
std::string uuid::GenerateUuid() {
static thread_local boost::uuids::random_generator Generator {};
boost::uuids::uuid Id { Generator() };
return boost::uuids::to_string(Id);
}
+22 -22
View File
@@ -73,7 +73,6 @@ int main(int argc, char** argv) {
int BeamMPServerMain(MainArguments Arguments) { int BeamMPServerMain(MainArguments Arguments) {
setlocale(LC_ALL, "C"); setlocale(LC_ALL, "C");
Application::InitializeConsole(); Application::InitializeConsole();
Application::Console().Internal().set_prompt("");
ArgsParser Parser; ArgsParser Parser;
Parser.RegisterArgument({ "help" }, ArgsParser::NONE); Parser.RegisterArgument({ "help" }, ArgsParser::NONE);
Parser.RegisterArgument({ "version" }, ArgsParser::NONE); Parser.RegisterArgument({ "version" }, ArgsParser::NONE);
@@ -84,14 +83,16 @@ int BeamMPServerMain(MainArguments Arguments) {
return 1; return 1;
} }
if (Parser.FoundArgument({ "help" })) { if (Parser.FoundArgument({ "help" })) {
Application::Console().Internal().set_prompt("");
Application::Console().WriteRaw(sCommandlineArguments); Application::Console().WriteRaw(sCommandlineArguments);
return 0; return 0;
} }
if (Parser.FoundArgument({ "version" })) { if (Parser.FoundArgument({ "version" })) {
Application::Console().WriteRaw(fmt::format("BeamMP Server v{} ({})", Application::ServerVersionString(), BEAMMP_GIT_HASH)); Application::Console().Internal().set_prompt("");
Application::Console().WriteRaw("BeamMP-Server v" + Application::ServerVersionString());
return 0; return 0;
} }
std::string ConfigPath = "ServerConfig.toml"; std::string ConfigPath = "ServerConfig.toml";
if (Parser.FoundArgument({ "config" })) { if (Parser.FoundArgument({ "config" })) {
auto MaybeConfigPath = Parser.GetValueOfArgument({ "config" }); auto MaybeConfigPath = Parser.GetValueOfArgument({ "config" });
@@ -114,37 +115,36 @@ int BeamMPServerMain(MainArguments Arguments) {
Application::SetSubsystemStatus("Main", Application::Status::Starting); Application::SetSubsystemStatus("Main", Application::Status::Starting);
SetupSignalHandlers();
beammp_infof("BeamMP Server v{} ({})", Application::ServerVersionString(), BEAMMP_GIT_HASH);
TConfig Config(ConfigPath);
if (Config.Failed()) {
beammp_info("Closing in 10 seconds");
// loop to make it possible to ctrl+c instead
Application::SleepSafeSeconds(5);
beammp_info("Closing in 5 seconds");
Application::SleepSafeSeconds(5);
Application::GracefullyShutdown();
return 1;
}
Application::Console().Internal().set_prompt("> ");
Application::Console().StartLoggingToFile(); Application::Console().StartLoggingToFile();
SetupSignalHandlers();
bool Shutdown = false; bool Shutdown = false;
Application::RegisterShutdownHandler([&Shutdown] { Application::RegisterShutdownHandler([&Shutdown] {
beammp_info("If this takes too long, you can press Ctrl+C repeatedly to force a shutdown."); beammp_info("If this takes too long, you can press Ctrl+C repeatedly to force a shutdown.");
Application::SetSubsystemStatus("Main", Application::Status::ShuttingDown); Application::SetSubsystemStatus("Main", Application::Status::ShuttingDown);
Shutdown = true; Shutdown = true;
}); });
Application::RegisterShutdownHandler([] {
auto Futures = LuaAPI::MP::Engine->TriggerEvent("onShutdown", "");
TLuaEngine::WaitForAll(Futures, std::chrono::seconds(5));
});
TServer Server(Arguments.List); TServer Server(Arguments.List);
TConfig Config(ConfigPath);
auto LuaEngine = std::make_shared<TLuaEngine>(); auto LuaEngine = std::make_shared<TLuaEngine>();
LuaEngine->SetServer(&Server); LuaEngine->SetServer(&Server);
Application::Console().InitializeLuaConsole(*LuaEngine); Application::Console().InitializeLuaConsole(*LuaEngine);
if (Config.Failed()) {
beammp_info("Closing in 10 seconds");
// loop to make it possible to ctrl+c instead
for (size_t i = 0; i < 20; ++i) {
std::this_thread::sleep_for(std::chrono::milliseconds(500));
}
return 1;
}
RegisterThread("Main"); RegisterThread("Main");
beammp_trace("Running in debug mode on a debug build"); beammp_trace("Running in debug mode on a debug build");
@@ -158,9 +158,9 @@ int BeamMPServerMain(MainArguments Arguments) {
PPSMonitor.SetNetwork(Network); PPSMonitor.SetNetwork(Network);
Application::CheckForUpdates(); Application::CheckForUpdates();
TPluginMonitor PluginMonitor(fs::path(Application::GetSettingString(StrResourceFolder)) / "Server", LuaEngine); TPluginMonitor PluginMonitor(fs::path(Application::Settings.Resource) / "Server", LuaEngine);
if (Application::GetSettingBool(StrHTTPServerEnabled)) { if (Application::Settings.HTTPServerEnabled) {
Http::Server::THttpServerInstance HttpServerInstance {}; Http::Server::THttpServerInstance HttpServerInstance {};
} }