parent
37213eb76c
commit
1100b04b59
@ -73,9 +73,9 @@ function(link_dependencies TARGET)
|
|||||||
tolualib
|
tolualib
|
||||||
)
|
)
|
||||||
|
|
||||||
# Link process information library:
|
# Link process information, multimedia (for sleep resolution) libraries:
|
||||||
if (WIN32)
|
if (WIN32)
|
||||||
target_link_libraries(${TARGET} PRIVATE Psapi.lib)
|
target_link_libraries(${TARGET} PRIVATE Psapi.lib Winmm.lib)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
# Special case handling for libevent pthreads:
|
# Special case handling for libevent pthreads:
|
||||||
|
@ -78,12 +78,14 @@
|
|||||||
|
|
||||||
// OS-dependent stuff:
|
// OS-dependent stuff:
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
|
#include <sdkddkver.h>
|
||||||
|
|
||||||
#define NOMINMAX // Windows SDK defines min and max macros, messing up with our std::min and std::max usage.
|
#define NOMINMAX // Windows SDK defines min and max macros, messing up with our std::min and std::max usage.
|
||||||
#define WIN32_LEAN_AND_MEAN
|
#define WIN32_LEAN_AND_MEAN
|
||||||
#define _WIN32_WINNT 0x0501 // We want to target Windows XP with Service Pack 2 & Windows Server 2003 with Service Pack 1 and higher.
|
#define _WIN32_WINNT _WIN32_WINNT_WINXP // We want to target Windows XP with Service Pack 2 & Windows Server 2003 with Service Pack 1 and higher.
|
||||||
|
|
||||||
// Use CryptoAPI primitives when targeting a version that supports encrypting with AES-CFB8 smaller than a full block at a time.
|
// Use CryptoAPI primitives when targeting a version that supports encrypting with AES-CFB8 smaller than a full block at a time.
|
||||||
#define PLATFORM_CRYPTOGRAPHY (_WIN32_WINNT >= 0x0602)
|
#define PLATFORM_CRYPTOGRAPHY (_WIN32_WINNT >= _WIN32_WINNT_WIN8)
|
||||||
|
|
||||||
#include <Windows.h>
|
#include <Windows.h>
|
||||||
#include <winsock2.h>
|
#include <winsock2.h>
|
||||||
|
@ -2,7 +2,6 @@ target_sources(
|
|||||||
${CMAKE_PROJECT_NAME} PRIVATE
|
${CMAKE_PROJECT_NAME} PRIVATE
|
||||||
|
|
||||||
CriticalSection.cpp
|
CriticalSection.cpp
|
||||||
Errors.cpp
|
|
||||||
Event.cpp
|
Event.cpp
|
||||||
File.cpp
|
File.cpp
|
||||||
GZipFile.cpp
|
GZipFile.cpp
|
||||||
@ -19,8 +18,8 @@ target_sources(
|
|||||||
WinStackWalker.cpp
|
WinStackWalker.cpp
|
||||||
|
|
||||||
AtomicUniquePtr.h
|
AtomicUniquePtr.h
|
||||||
|
ConsoleSignalHandler.h
|
||||||
CriticalSection.h
|
CriticalSection.h
|
||||||
Errors.h
|
|
||||||
Event.h
|
Event.h
|
||||||
File.h
|
File.h
|
||||||
GetAddressInfoError.h
|
GetAddressInfoError.h
|
||||||
@ -34,6 +33,7 @@ target_sources(
|
|||||||
NetworkSingleton.h
|
NetworkSingleton.h
|
||||||
Queue.h
|
Queue.h
|
||||||
ServerHandleImpl.h
|
ServerHandleImpl.h
|
||||||
|
SleepResolutionBooster.h
|
||||||
StackTrace.h
|
StackTrace.h
|
||||||
StartAsService.h
|
StartAsService.h
|
||||||
TCPLinkImpl.h
|
TCPLinkImpl.h
|
||||||
|
130
src/OSSupport/ConsoleSignalHandler.h
Normal file
130
src/OSSupport/ConsoleSignalHandler.h
Normal file
@ -0,0 +1,130 @@
|
|||||||
|
|
||||||
|
// ConsoleSignalHandler.h
|
||||||
|
|
||||||
|
// Intercepts signals for graceful CTRL-C (and others) handling.
|
||||||
|
|
||||||
|
// This file MUST NOT be included from anywhere other than main.cpp.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#include <csignal>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// Because SIG_DFL or SIG_IGN could be NULL instead of nullptr, we need to disable the Clang warning here:
|
||||||
|
#ifdef __clang__
|
||||||
|
#pragma clang diagnostic push
|
||||||
|
#pragma clang diagnostic ignored "-Wunknown-warning-option"
|
||||||
|
#pragma clang diagnostic ignored "-Wunknown-pragmas"
|
||||||
|
#pragma clang diagnostic ignored "-Wzero-as-null-pointer-constant"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
static void NonCtrlHandler(int a_Signal)
|
||||||
|
{
|
||||||
|
LOGD("Terminate event raised from std::signal");
|
||||||
|
|
||||||
|
switch (a_Signal)
|
||||||
|
{
|
||||||
|
case SIGSEGV:
|
||||||
|
{
|
||||||
|
PrintStackTrace();
|
||||||
|
|
||||||
|
LOGERROR(
|
||||||
|
"Failure report: \n\n"
|
||||||
|
" :( | Cuberite has encountered an error and needs to close\n"
|
||||||
|
" | SIGSEGV: Segmentation fault\n"
|
||||||
|
" |\n"
|
||||||
|
#ifdef BUILD_ID
|
||||||
|
" | Cuberite " BUILD_SERIES_NAME " (id: " BUILD_ID ")\n"
|
||||||
|
" | from commit " BUILD_COMMIT_ID "\n"
|
||||||
|
#endif
|
||||||
|
);
|
||||||
|
|
||||||
|
std::signal(SIGSEGV, SIG_DFL);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
case SIGABRT:
|
||||||
|
#ifdef SIGABRT_COMPAT
|
||||||
|
case SIGABRT_COMPAT:
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
PrintStackTrace();
|
||||||
|
|
||||||
|
LOGERROR(
|
||||||
|
"Failure report: \n\n"
|
||||||
|
" :( | Cuberite has encountered an error and needs to close\n"
|
||||||
|
" | SIGABRT: Server self-terminated due to an internal fault\n"
|
||||||
|
" |\n"
|
||||||
|
#ifdef BUILD_ID
|
||||||
|
" | Cuberite " BUILD_SERIES_NAME " (id: " BUILD_ID ")\n"
|
||||||
|
" | from commit " BUILD_COMMIT_ID "\n"
|
||||||
|
#endif
|
||||||
|
);
|
||||||
|
|
||||||
|
std::signal(SIGSEGV, SIG_DFL);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
case SIGINT:
|
||||||
|
case SIGTERM:
|
||||||
|
{
|
||||||
|
// Server is shutting down, wait for it...
|
||||||
|
cRoot::Stop();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef __clang__
|
||||||
|
#pragma clang diagnostic pop
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
|
||||||
|
/** Handle CTRL events in windows, including console window close. */
|
||||||
|
static BOOL CtrlHandler(DWORD fdwCtrlType)
|
||||||
|
{
|
||||||
|
cRoot::Stop();
|
||||||
|
LOGD("Terminate event raised from the Windows CtrlHandler");
|
||||||
|
|
||||||
|
// Delay as much as possible to try to get the server to shut down cleanly - 10 seconds given by Windows:
|
||||||
|
std::this_thread::sleep_for(std::chrono::seconds(10));
|
||||||
|
|
||||||
|
// Returning from main() automatically aborts this handler thread.
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
namespace ConsoleSignalHandler
|
||||||
|
{
|
||||||
|
static void Register()
|
||||||
|
{
|
||||||
|
std::signal(SIGSEGV, NonCtrlHandler);
|
||||||
|
std::signal(SIGTERM, NonCtrlHandler);
|
||||||
|
std::signal(SIGINT, NonCtrlHandler);
|
||||||
|
std::signal(SIGABRT, NonCtrlHandler);
|
||||||
|
#ifdef SIGABRT_COMPAT
|
||||||
|
std::signal(SIGABRT_COMPAT, NonCtrlHandler);
|
||||||
|
#endif
|
||||||
|
#ifdef SIGPIPE
|
||||||
|
std::signal(SIGPIPE, SIG_IGN); // Ignore (PR #2487).
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
SetConsoleCtrlHandler(reinterpret_cast<PHANDLER_ROUTINE>(CtrlHandler), TRUE);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
};
|
@ -1,53 +0,0 @@
|
|||||||
|
|
||||||
#include "Globals.h"
|
|
||||||
|
|
||||||
#include "Errors.h"
|
|
||||||
|
|
||||||
AString GetOSErrorString( int a_ErrNo)
|
|
||||||
{
|
|
||||||
char buffer[ 1024 ];
|
|
||||||
AString Out;
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM, nullptr, a_ErrNo, 0, buffer, ARRAYCOUNT(buffer), nullptr);
|
|
||||||
Printf(Out, "%d: %s", a_ErrNo, buffer);
|
|
||||||
if (!Out.empty() && (Out[Out.length() - 1] == '\n'))
|
|
||||||
{
|
|
||||||
Out.erase(Out.length() - 2);
|
|
||||||
}
|
|
||||||
return Out;
|
|
||||||
|
|
||||||
#else // _WIN32
|
|
||||||
|
|
||||||
// According to https://linux.die.net/man/3/strerror_r there are two versions of strerror_r():
|
|
||||||
|
|
||||||
#if defined(__GLIBC__) && defined( _GNU_SOURCE) // GNU version of strerror_r()
|
|
||||||
|
|
||||||
char * res = strerror_r( errno, buffer, ARRAYCOUNT(buffer));
|
|
||||||
if (res != nullptr)
|
|
||||||
{
|
|
||||||
Printf(Out, "%d: %s", a_ErrNo, res);
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else // XSI version of strerror_r():
|
|
||||||
|
|
||||||
int res = strerror_r( errno, buffer, ARRAYCOUNT(buffer));
|
|
||||||
if (res == 0)
|
|
||||||
{
|
|
||||||
Printf(Out, "%d: %s", a_ErrNo, buffer);
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // strerror_r() version
|
|
||||||
|
|
||||||
else
|
|
||||||
{
|
|
||||||
Printf(Out, "Error %d while getting error string for error #%d!", errno, a_ErrNo);
|
|
||||||
return Out;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // else _WIN32
|
|
||||||
}
|
|
||||||
|
|
@ -1,5 +0,0 @@
|
|||||||
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
AString GetOSErrorString(int a_ErrNo);
|
|
||||||
|
|
@ -7,7 +7,6 @@
|
|||||||
#include "Globals.h" // NOTE: MSVC stupidness requires this to be the same across all modules
|
#include "Globals.h" // NOTE: MSVC stupidness requires this to be the same across all modules
|
||||||
|
|
||||||
#include "Event.h"
|
#include "Event.h"
|
||||||
#include "Errors.h"
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,5 +1,11 @@
|
|||||||
|
|
||||||
#pragma once
|
// MiniDumpWriter.h
|
||||||
|
|
||||||
|
// 32-bit only:
|
||||||
|
// When the server crashes, create a "dump file" containing the callstack of each thread and some variables;
|
||||||
|
// let the user send us that crash file for analysis.
|
||||||
|
|
||||||
|
// This file MUST NOT be included from anywhere other than main.cpp.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -18,82 +24,32 @@ enum class MiniDumpFlags
|
|||||||
|
|
||||||
#if defined(_WIN32) && !defined(_WIN64) && defined(_MSC_VER) // 32-bit Windows app compiled in MSVC
|
#if defined(_WIN32) && !defined(_WIN64) && defined(_MSC_VER) // 32-bit Windows app compiled in MSVC
|
||||||
|
|
||||||
#include <dbghelp.h>
|
#include <DbgHelp.h>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/** Windows 32-bit stuff:
|
using MiniDumpWriteDumpFunction = decltype(&MiniDumpWriteDump);
|
||||||
When the server crashes, create a "dump file" containing the callstack of each thread and some variables;
|
|
||||||
let the user send us that crash file for analysis */
|
static HINSTANCE m_DbgHelp;
|
||||||
class MiniDumpWriter
|
static MiniDumpWriteDumpFunction s_WriteMiniDump; // The function in dbghlp DLL that creates dump files
|
||||||
|
static wchar_t s_DumpFileName[MAX_PATH]; // Filename of the dump file; hes to be created before the dump handler kicks in
|
||||||
|
static char s_ExceptionStack[128 * 1024]; // Substitute stack, just in case the handler kicks in because of "insufficient stack space"
|
||||||
|
static MINIDUMP_TYPE s_DumpFlags = MiniDumpNormal; // By default dump only the stack and some helpers
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/** This function gets called just before the "program executed an illegal instruction and will be terminated" or similar.
|
||||||
|
Its purpose is to create the crashdump using the dbghlp DLLs */
|
||||||
|
static LONG WINAPI LastChanceExceptionFilter(__in struct _EXCEPTION_POINTERS * a_ExceptionInfo)
|
||||||
{
|
{
|
||||||
typedef BOOL(WINAPI *pMiniDumpWriteDump)(
|
|
||||||
HANDLE hProcess,
|
|
||||||
DWORD ProcessId,
|
|
||||||
HANDLE hFile,
|
|
||||||
MINIDUMP_TYPE DumpType,
|
|
||||||
PMINIDUMP_EXCEPTION_INFORMATION ExceptionParam,
|
|
||||||
PMINIDUMP_USER_STREAM_INFORMATION UserStreamParam,
|
|
||||||
PMINIDUMP_CALLBACK_INFORMATION CallbackParam
|
|
||||||
);
|
|
||||||
|
|
||||||
public:
|
|
||||||
|
|
||||||
MiniDumpWriter()
|
|
||||||
{
|
|
||||||
// Magic code to produce dump-files on Windows if the server crashes:
|
|
||||||
|
|
||||||
m_DbgHelp = LoadLibrary(L"DBGHELP.DLL");
|
|
||||||
if (m_DbgHelp == INVALID_HANDLE_VALUE)
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
s_WriteMiniDump = (pMiniDumpWriteDump)GetProcAddress(m_DbgHelp, "MiniDumpWriteDump");
|
|
||||||
if (s_WriteMiniDump != nullptr)
|
|
||||||
{
|
|
||||||
ASSERT(swprintf(s_DumpFileName, ARRAYCOUNT(s_DumpFileName), L"crash_mcs_%x.dmp", GetCurrentProcessId()) > 0);
|
|
||||||
SetUnhandledExceptionFilter(LastChanceExceptionFilter);
|
|
||||||
}
|
|
||||||
|
|
||||||
// End of dump-file magic
|
|
||||||
}
|
|
||||||
|
|
||||||
void AddDumpFlags(const MiniDumpFlags a_Flags)
|
|
||||||
{
|
|
||||||
switch (a_Flags)
|
|
||||||
{
|
|
||||||
case MiniDumpFlags::WithDataSegments:
|
|
||||||
{
|
|
||||||
s_DumpFlags = static_cast<MINIDUMP_TYPE>(s_DumpFlags | MINIDUMP_TYPE::MiniDumpWithDataSegs);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
case MiniDumpFlags::WithFullMemory:
|
|
||||||
{
|
|
||||||
s_DumpFlags = static_cast<MINIDUMP_TYPE>(s_DumpFlags | MINIDUMP_TYPE::MiniDumpWithFullMemory);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
~MiniDumpWriter()
|
|
||||||
{
|
|
||||||
FreeLibrary(m_DbgHelp);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
/** This function gets called just before the "program executed an illegal instruction and will be terminated" or similar.
|
|
||||||
Its purpose is to create the crashdump using the dbghlp DLLs */
|
|
||||||
static LONG WINAPI LastChanceExceptionFilter(__in struct _EXCEPTION_POINTERS * a_ExceptionInfo)
|
|
||||||
{
|
|
||||||
char * newStack = &s_ExceptionStack[sizeof(s_ExceptionStack) - 1];
|
char * newStack = &s_ExceptionStack[sizeof(s_ExceptionStack) - 1];
|
||||||
char * oldStack;
|
char * oldStack;
|
||||||
|
|
||||||
// Use the substitute stack:
|
// Use the substitute stack:
|
||||||
// This code is the reason why we don't support 64-bit (yet)
|
|
||||||
_asm
|
_asm
|
||||||
{
|
{
|
||||||
mov oldStack, esp
|
mov oldStack, esp
|
||||||
@ -117,21 +73,70 @@ private:
|
|||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
namespace MiniDumpWriter
|
||||||
|
{
|
||||||
|
static void Register()
|
||||||
|
{
|
||||||
|
// Magic code to produce dump-files on Windows if the server crashes:
|
||||||
|
|
||||||
|
m_DbgHelp = LoadLibrary(L"DBGHELP.DLL");
|
||||||
|
if (m_DbgHelp == INVALID_HANDLE_VALUE)
|
||||||
|
{
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
HINSTANCE m_DbgHelp;
|
s_WriteMiniDump = (MiniDumpWriteDumpFunction)GetProcAddress(m_DbgHelp, "MiniDumpWriteDump");
|
||||||
|
if (s_WriteMiniDump != nullptr)
|
||||||
|
{
|
||||||
|
ASSERT(swprintf(s_DumpFileName, ARRAYCOUNT(s_DumpFileName), L"crash_mcs_%x.dmp", GetCurrentProcessId()) > 0);
|
||||||
|
SetUnhandledExceptionFilter(LastChanceExceptionFilter);
|
||||||
|
}
|
||||||
|
|
||||||
static inline pMiniDumpWriteDump s_WriteMiniDump; // The function in dbghlp DLL that creates dump files
|
// End of dump-file magic
|
||||||
static inline wchar_t s_DumpFileName[MAX_PATH]; // Filename of the dump file; hes to be created before the dump handler kicks in
|
}
|
||||||
static inline char s_ExceptionStack[128 * 1024]; // Substitute stack, just in case the handler kicks in because of "insufficient stack space"
|
|
||||||
static inline MINIDUMP_TYPE s_DumpFlags = MiniDumpNormal; // By default dump only the stack and some helpers
|
static void AddDumpFlags(const MiniDumpFlags a_Flags)
|
||||||
|
{
|
||||||
|
switch (a_Flags)
|
||||||
|
{
|
||||||
|
case MiniDumpFlags::WithDataSegments:
|
||||||
|
{
|
||||||
|
s_DumpFlags = static_cast<MINIDUMP_TYPE>(s_DumpFlags | MINIDUMP_TYPE::MiniDumpWithDataSegs);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case MiniDumpFlags::WithFullMemory:
|
||||||
|
{
|
||||||
|
s_DumpFlags = static_cast<MINIDUMP_TYPE>(s_DumpFlags | MINIDUMP_TYPE::MiniDumpWithFullMemory);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Unregister()
|
||||||
|
{
|
||||||
|
FreeLibrary(m_DbgHelp);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
struct MiniDumpWriter
|
namespace MiniDumpWriter
|
||||||
{
|
{
|
||||||
void AddDumpFlags(const MiniDumpFlags)
|
static void Register()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
static void AddDumpFlags(const MiniDumpFlags)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Unregister()
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
66
src/OSSupport/SleepResolutionBooster.h
Normal file
66
src/OSSupport/SleepResolutionBooster.h
Normal file
@ -0,0 +1,66 @@
|
|||||||
|
|
||||||
|
// SleepResolutionBooster.h
|
||||||
|
|
||||||
|
// Increases the accuracy of Sleep on Windows (GH #5140).
|
||||||
|
|
||||||
|
// This file MUST NOT be included from anywhere other than main.cpp.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
|
||||||
|
#include <timeapi.h>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
static TIMECAPS g_Resolution;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
namespace SleepResolutionBooster
|
||||||
|
{
|
||||||
|
static void Register()
|
||||||
|
{
|
||||||
|
// Default sleep resolution on Windows isn't accurate enough (GH #5140) so try to boost it:
|
||||||
|
if (
|
||||||
|
(timeGetDevCaps(&g_Resolution, sizeof(g_Resolution)) == MMSYSERR_NOERROR) &&
|
||||||
|
(timeBeginPeriod(g_Resolution.wPeriodMin) == MMSYSERR_NOERROR)
|
||||||
|
)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Max < Min sentinel for failure, to prevent bogus timeEndPeriod calls:
|
||||||
|
g_Resolution.wPeriodMax = 0;
|
||||||
|
g_Resolution.wPeriodMin = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Unregister()
|
||||||
|
{
|
||||||
|
if (g_Resolution.wPeriodMax >= g_Resolution.wPeriodMin)
|
||||||
|
{
|
||||||
|
timeEndPeriod(g_Resolution.wPeriodMin);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
namespace SleepResolutionBooster
|
||||||
|
{
|
||||||
|
static void Register()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
static void Unregister()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
@ -1,5 +1,9 @@
|
|||||||
|
|
||||||
#pragma once
|
// StartAsService.h
|
||||||
|
|
||||||
|
// Handles startup as a Windows Service or UNIX daemon.
|
||||||
|
|
||||||
|
// This file MUST NOT be included from anywhere other than main.cpp.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -7,13 +11,7 @@
|
|||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
|
|
||||||
#include <csignal>
|
class StartAsService
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class cStartAsService
|
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
|
||||||
@ -79,7 +77,7 @@ private:
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const auto LastComponent = wcsrchr(applicationFilename, L'\\');
|
const auto LastComponent = std::wcsrchr(applicationFilename, L'\\');
|
||||||
if (LastComponent == nullptr)
|
if (LastComponent == nullptr)
|
||||||
{
|
{
|
||||||
serviceSetState(0, SERVICE_STOPPED, E_UNEXPECTED);
|
serviceSetState(0, SERVICE_STOPPED, E_UNEXPECTED);
|
||||||
@ -89,7 +87,7 @@ private:
|
|||||||
const auto LengthToLastComponent = LastComponent - applicationFilename;
|
const auto LengthToLastComponent = LastComponent - applicationFilename;
|
||||||
|
|
||||||
// Strip off the filename, keep only the path:
|
// Strip off the filename, keep only the path:
|
||||||
wcsncpy(applicationDirectory, applicationFilename, LengthToLastComponent);
|
std::wcsncpy(applicationDirectory, applicationFilename, LengthToLastComponent);
|
||||||
applicationDirectory[LengthToLastComponent] = L'\0'; // Make sure new path is null terminated
|
applicationDirectory[LengthToLastComponent] = L'\0'; // Make sure new path is null terminated
|
||||||
|
|
||||||
// Services are run by the SCM, and inherit its working directory - usually System32.
|
// Services are run by the SCM, and inherit its working directory - usually System32.
|
||||||
@ -115,7 +113,7 @@ private:
|
|||||||
char * MultibyteArgV[] = { MultibyteArgV0 };
|
char * MultibyteArgV[] = { MultibyteArgV0 };
|
||||||
|
|
||||||
const auto OutputSize = std::size(MultibyteArgV0);
|
const auto OutputSize = std::size(MultibyteArgV0);
|
||||||
const auto TranslateResult = wcstombs(MultibyteArgV0, argv[0], OutputSize);
|
const auto TranslateResult = std::wcstombs(MultibyteArgV0, argv[0], OutputSize);
|
||||||
|
|
||||||
if (TranslateResult == static_cast<size_t>(-1))
|
if (TranslateResult == static_cast<size_t>(-1))
|
||||||
{
|
{
|
||||||
@ -141,7 +139,7 @@ private:
|
|||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
struct cStartAsService
|
struct StartAsService
|
||||||
{
|
{
|
||||||
/** Make a UNIX daemon. */
|
/** Make a UNIX daemon. */
|
||||||
template <auto>
|
template <auto>
|
||||||
|
215
src/main.cpp
215
src/main.cpp
@ -5,14 +5,14 @@
|
|||||||
#include "BuildInfo.h"
|
#include "BuildInfo.h"
|
||||||
#include "Logger.h"
|
#include "Logger.h"
|
||||||
#include "MemorySettingsRepository.h"
|
#include "MemorySettingsRepository.h"
|
||||||
#include "OSSupport/NetworkSingleton.h"
|
|
||||||
#include "OSSupport/MiniDumpWriter.h"
|
|
||||||
#include "OSSupport/StartAsService.h"
|
|
||||||
#include "Root.h"
|
#include "Root.h"
|
||||||
#include "tclap/CmdLine.h"
|
#include "tclap/CmdLine.h"
|
||||||
|
|
||||||
#include <csignal>
|
#include "OSSupport/ConsoleSignalHandler.h"
|
||||||
#include <cstdlib>
|
#include "OSSupport/NetworkSingleton.h"
|
||||||
|
#include "OSSupport/MiniDumpWriter.h"
|
||||||
|
#include "OSSupport/SleepResolutionBooster.h"
|
||||||
|
#include "OSSupport/StartAsService.h"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -22,107 +22,6 @@ bool g_ShouldLogCommIn;
|
|||||||
bool g_ShouldLogCommOut;
|
bool g_ShouldLogCommOut;
|
||||||
bool g_RunAsService;
|
bool g_RunAsService;
|
||||||
|
|
||||||
/** Global that registers itself as a last chance exception handler to write a minidump on crash. */
|
|
||||||
static MiniDumpWriter g_MiniDumpWriter;
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// Because SIG_DFL or SIG_IGN could be NULL instead of nullptr, we need to disable the Clang warning here
|
|
||||||
#ifdef __clang__
|
|
||||||
#pragma clang diagnostic push
|
|
||||||
#pragma clang diagnostic ignored "-Wunknown-warning-option"
|
|
||||||
#pragma clang diagnostic ignored "-Wunknown-pragmas"
|
|
||||||
#pragma clang diagnostic ignored "-Wzero-as-null-pointer-constant"
|
|
||||||
#endif // __clang__
|
|
||||||
|
|
||||||
static void NonCtrlHandler(int a_Signal)
|
|
||||||
{
|
|
||||||
LOGD("Terminate event raised from std::signal");
|
|
||||||
|
|
||||||
switch (a_Signal)
|
|
||||||
{
|
|
||||||
case SIGSEGV:
|
|
||||||
{
|
|
||||||
PrintStackTrace();
|
|
||||||
|
|
||||||
LOGERROR(
|
|
||||||
"Failure report: \n\n"
|
|
||||||
" :( | Cuberite has encountered an error and needs to close\n"
|
|
||||||
" | SIGSEGV: Segmentation fault\n"
|
|
||||||
" |\n"
|
|
||||||
#ifdef BUILD_ID
|
|
||||||
" | Cuberite " BUILD_SERIES_NAME " (id: " BUILD_ID ")\n"
|
|
||||||
" | from commit " BUILD_COMMIT_ID "\n"
|
|
||||||
#endif
|
|
||||||
);
|
|
||||||
|
|
||||||
std::signal(SIGSEGV, SIG_DFL);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
case SIGABRT:
|
|
||||||
#ifdef SIGABRT_COMPAT
|
|
||||||
case SIGABRT_COMPAT:
|
|
||||||
#endif
|
|
||||||
{
|
|
||||||
PrintStackTrace();
|
|
||||||
|
|
||||||
LOGERROR(
|
|
||||||
"Failure report: \n\n"
|
|
||||||
" :( | Cuberite has encountered an error and needs to close\n"
|
|
||||||
" | SIGABRT: Server self-terminated due to an internal fault\n"
|
|
||||||
" |\n"
|
|
||||||
#ifdef BUILD_ID
|
|
||||||
" | Cuberite " BUILD_SERIES_NAME " (id: " BUILD_ID ")\n"
|
|
||||||
" | from commit " BUILD_COMMIT_ID "\n"
|
|
||||||
#endif
|
|
||||||
);
|
|
||||||
|
|
||||||
std::signal(SIGSEGV, SIG_DFL);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
case SIGINT:
|
|
||||||
case SIGTERM:
|
|
||||||
{
|
|
||||||
// Server is shutting down, wait for it...
|
|
||||||
cRoot::Stop();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
#ifdef SIGPIPE
|
|
||||||
case SIGPIPE:
|
|
||||||
{
|
|
||||||
// Ignore (PR #2487)
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#ifdef __clang__
|
|
||||||
#pragma clang diagnostic pop
|
|
||||||
#endif // __clang__
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
// Handle CTRL events in windows, including console window close
|
|
||||||
static BOOL CtrlHandler(DWORD fdwCtrlType)
|
|
||||||
{
|
|
||||||
cRoot::Stop();
|
|
||||||
LOGD("Terminate event raised from the Windows CtrlHandler");
|
|
||||||
|
|
||||||
// Delay as much as possible to try to get the server to shut down cleanly - 10 seconds given by Windows
|
|
||||||
std::this_thread::sleep_for(std::chrono::seconds(10));
|
|
||||||
|
|
||||||
// Returning from main() automatically aborts this handler thread
|
|
||||||
|
|
||||||
return TRUE;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -130,7 +29,7 @@ static BOOL CtrlHandler(DWORD fdwCtrlType)
|
|||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// ParseArguments - Read the startup arguments and store into a settings object
|
// ParseArguments - Read the startup arguments and store into a settings object
|
||||||
|
|
||||||
static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & Settings)
|
static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & a_Settings)
|
||||||
{
|
{
|
||||||
// Parse the comand line args:
|
// Parse the comand line args:
|
||||||
TCLAP::CmdLine cmd("Cuberite");
|
TCLAP::CmdLine cmd("Cuberite");
|
||||||
@ -151,23 +50,23 @@ static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & S
|
|||||||
if (confArg.isSet())
|
if (confArg.isSet())
|
||||||
{
|
{
|
||||||
AString conf_file = confArg.getValue();
|
AString conf_file = confArg.getValue();
|
||||||
Settings.AddValue("Server", "ConfigFile", conf_file);
|
a_Settings.AddValue("Server", "ConfigFile", conf_file);
|
||||||
}
|
}
|
||||||
if (slotsArg.isSet())
|
if (slotsArg.isSet())
|
||||||
{
|
{
|
||||||
int slots = slotsArg.getValue();
|
int slots = slotsArg.getValue();
|
||||||
Settings.AddValue("Server", "MaxPlayers", static_cast<Int64>(slots));
|
a_Settings.AddValue("Server", "MaxPlayers", static_cast<Int64>(slots));
|
||||||
}
|
}
|
||||||
if (portsArg.isSet())
|
if (portsArg.isSet())
|
||||||
{
|
{
|
||||||
for (auto port: portsArg.getValue())
|
for (auto port: portsArg.getValue())
|
||||||
{
|
{
|
||||||
Settings.AddValue("Server", "Ports", std::to_string(port));
|
a_Settings.AddValue("Server", "Ports", std::to_string(port));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (noFileLogArg.getValue())
|
if (noFileLogArg.getValue())
|
||||||
{
|
{
|
||||||
Settings.AddValue("Server", "DisableLogFile", true);
|
a_Settings.AddValue("Server", "DisableLogFile", true);
|
||||||
}
|
}
|
||||||
if (commLogArg.getValue())
|
if (commLogArg.getValue())
|
||||||
{
|
{
|
||||||
@ -183,7 +82,7 @@ static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & S
|
|||||||
{
|
{
|
||||||
setvbuf(stdout, nullptr, _IONBF, 0);
|
setvbuf(stdout, nullptr, _IONBF, 0);
|
||||||
}
|
}
|
||||||
Settings.SetReadOnly();
|
a_Settings.SetReadOnly();
|
||||||
|
|
||||||
if (runAsServiceArg.getValue())
|
if (runAsServiceArg.getValue())
|
||||||
{
|
{
|
||||||
@ -193,11 +92,11 @@ static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & S
|
|||||||
// Apply the CrashDump flags for platforms that support them:
|
// Apply the CrashDump flags for platforms that support them:
|
||||||
if (crashDumpGlobals.getValue())
|
if (crashDumpGlobals.getValue())
|
||||||
{
|
{
|
||||||
g_MiniDumpWriter.AddDumpFlags(MiniDumpFlags::WithDataSegments);
|
MiniDumpWriter::AddDumpFlags(MiniDumpFlags::WithDataSegments);
|
||||||
}
|
}
|
||||||
if (crashDumpFull.getValue())
|
if (crashDumpFull.getValue())
|
||||||
{
|
{
|
||||||
g_MiniDumpWriter.AddDumpFlags(MiniDumpFlags::WithFullMemory);
|
MiniDumpWriter::AddDumpFlags(MiniDumpFlags::WithFullMemory);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -208,12 +107,60 @@ static void ParseArguments(int argc, char ** argv, cMemorySettingsRepository & S
|
|||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// UniversalMain - Main startup logic for both standard running and as a service
|
// UniversalMain - Main startup logic for both standard running and as a service
|
||||||
|
|
||||||
static int UniversalMain(int argc, char * argv[], bool RunningAsService)
|
static int UniversalMain(int argc, char * argv[], const bool a_RunningAsService)
|
||||||
{
|
{
|
||||||
|
const struct MiniDumpWriterRAII
|
||||||
|
{
|
||||||
|
MiniDumpWriterRAII()
|
||||||
|
{
|
||||||
|
// Registers a last chance exception handler to write a minidump on crash:
|
||||||
|
MiniDumpWriter::Register();
|
||||||
|
}
|
||||||
|
|
||||||
|
~MiniDumpWriterRAII()
|
||||||
|
{
|
||||||
|
MiniDumpWriter::Unregister();
|
||||||
|
}
|
||||||
|
} MiniDumpWriter;
|
||||||
|
|
||||||
|
const struct SleepResolutionBoosterRAII
|
||||||
|
{
|
||||||
|
SleepResolutionBoosterRAII()
|
||||||
|
{
|
||||||
|
// Boost timer resolution to keep TPS high:
|
||||||
|
SleepResolutionBooster::Register();
|
||||||
|
}
|
||||||
|
|
||||||
|
~SleepResolutionBoosterRAII()
|
||||||
|
{
|
||||||
|
SleepResolutionBooster::Unregister();
|
||||||
|
}
|
||||||
|
} SleepResolutionBooster;
|
||||||
|
|
||||||
|
// Register signal handlers, enabling graceful shutdown from the terminal:
|
||||||
|
ConsoleSignalHandler::Register();
|
||||||
|
|
||||||
// Initialize logging subsystem:
|
// Initialize logging subsystem:
|
||||||
cLogger::InitiateMultithreading();
|
cLogger::InitiateMultithreading();
|
||||||
|
|
||||||
struct NetworkRAII
|
try
|
||||||
|
{
|
||||||
|
cMemorySettingsRepository Settings;
|
||||||
|
ParseArguments(argc, argv, Settings); // Make sure g_RunAsService is set correctly before checking its value.
|
||||||
|
|
||||||
|
// Attempt to run as a service:
|
||||||
|
if (g_RunAsService && !a_RunningAsService)
|
||||||
|
{
|
||||||
|
// This will either fork or call UniversalMain again:
|
||||||
|
if (StartAsService::MakeIntoService<&UniversalMain>())
|
||||||
|
{
|
||||||
|
return EXIT_SUCCESS;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while (true)
|
||||||
|
{
|
||||||
|
const struct NetworkRAII
|
||||||
{
|
{
|
||||||
NetworkRAII()
|
NetworkRAII()
|
||||||
{
|
{
|
||||||
@ -226,28 +173,9 @@ static int UniversalMain(int argc, char * argv[], bool RunningAsService)
|
|||||||
// Shutdown all of LibEvent:
|
// Shutdown all of LibEvent:
|
||||||
cNetworkSingleton::Get().Terminate();
|
cNetworkSingleton::Get().Terminate();
|
||||||
}
|
}
|
||||||
};
|
} LibEvent;
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
cMemorySettingsRepository Settings;
|
|
||||||
ParseArguments(argc, argv, Settings); // Make sure g_RunAsService is set correctly before checking it's value
|
|
||||||
|
|
||||||
// Attempt to run as a service:
|
|
||||||
if (!RunningAsService && g_RunAsService)
|
|
||||||
{
|
|
||||||
// This will either fork or call UniversalMain again:
|
|
||||||
if (cStartAsService::MakeIntoService<&UniversalMain>())
|
|
||||||
{
|
|
||||||
return EXIT_SUCCESS;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
while (true)
|
|
||||||
{
|
|
||||||
NetworkRAII LibEvent;
|
|
||||||
cRoot Root;
|
cRoot Root;
|
||||||
|
|
||||||
if (!Root.Run(Settings))
|
if (!Root.Run(Settings))
|
||||||
{
|
{
|
||||||
break;
|
break;
|
||||||
@ -291,20 +219,5 @@ int main(int argc, char ** argv)
|
|||||||
|
|
||||||
#endif // !NDEBUG && _MSC_VER
|
#endif // !NDEBUG && _MSC_VER
|
||||||
|
|
||||||
std::signal(SIGSEGV, NonCtrlHandler);
|
|
||||||
std::signal(SIGTERM, NonCtrlHandler);
|
|
||||||
std::signal(SIGINT, NonCtrlHandler);
|
|
||||||
std::signal(SIGABRT, NonCtrlHandler);
|
|
||||||
#ifdef SIGABRT_COMPAT
|
|
||||||
std::signal(SIGABRT_COMPAT, NonCtrlHandler);
|
|
||||||
#endif
|
|
||||||
#ifdef SIGPIPE
|
|
||||||
std::signal(SIGPIPE, SIG_IGN);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
VERIFY(SetConsoleCtrlHandler(reinterpret_cast<PHANDLER_ROUTINE>(CtrlHandler), TRUE) == TRUE);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return UniversalMain(argc, argv, false);
|
return UniversalMain(argc, argv, false);
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user