mirror of
https://github.com/OpenRCT2/OpenRCT2.git
synced 2024-08-31 06:26:07 +02:00
586 lines
15 KiB
C++
586 lines
15 KiB
C++
#pragma region Copyright (c) 2014-2016 OpenRCT2 Developers
|
|
/*****************************************************************************
|
|
* OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
|
|
*
|
|
* OpenRCT2 is the work of many authors, a full list can be found in contributors.md
|
|
* For more information, visit https://github.com/OpenRCT2/OpenRCT2
|
|
*
|
|
* OpenRCT2 is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* A full copy of the GNU General Public License can be found in licence.txt
|
|
*****************************************************************************/
|
|
#pragma endregion
|
|
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#if defined(__unix__)
|
|
#include <unistd.h>
|
|
#include <sys/mman.h>
|
|
#endif // defined(__unix__)
|
|
|
|
#include "PaintIntercept.hpp"
|
|
#include "TestTrack.hpp"
|
|
#include "Utils.hpp"
|
|
|
|
extern "C" {
|
|
#include "data.h"
|
|
#include <openrct2/rct2.h>
|
|
#include <openrct2/ride/ride.h>
|
|
#include <openrct2/ride/ride_data.h>
|
|
#include <openrct2/ride/track.h>
|
|
#include <openrct2/ride/track_data.h>
|
|
}
|
|
|
|
typedef struct {
|
|
uint8 rideType;
|
|
std::vector<uint8> trackTypes;
|
|
} TestCase;
|
|
|
|
enum CLIColour {
|
|
DEFAULT,
|
|
RED,
|
|
YELLOW,
|
|
GREEN,
|
|
};
|
|
|
|
bool gTestColor = true;
|
|
Verbosity _verbosity = NORMAL;
|
|
|
|
static bool CStringEquals(const char *lhs, const char *rhs) {
|
|
if (lhs == NULL) return rhs == NULL;
|
|
|
|
if (rhs == NULL) return false;
|
|
|
|
return strcmp(lhs, rhs) == 0;
|
|
}
|
|
|
|
enum COLOUR_METHOD {
|
|
COLOUR_METHOD_NONE,
|
|
COLOUR_METHOD_ANSI,
|
|
COLOUR_METHOD_WINDOWS,
|
|
};
|
|
|
|
static COLOUR_METHOD GetColourMethod()
|
|
{
|
|
if (!gTestColor) {
|
|
return COLOUR_METHOD_NONE;
|
|
}
|
|
|
|
const char* const term = getenv("TERM");
|
|
const bool term_supports_color =
|
|
CStringEquals(term, "xterm") ||
|
|
CStringEquals(term, "xterm-color") ||
|
|
CStringEquals(term, "xterm-256color") ||
|
|
CStringEquals(term, "screen") ||
|
|
CStringEquals(term, "screen-256color") ||
|
|
CStringEquals(term, "tmux") ||
|
|
CStringEquals(term, "tmux-256color") ||
|
|
CStringEquals(term, "rxvt-unicode") ||
|
|
CStringEquals(term, "rxvt-unicode-256color") ||
|
|
CStringEquals(term, "linux") ||
|
|
CStringEquals(term, "cygwin");
|
|
|
|
if (term_supports_color) {
|
|
return COLOUR_METHOD_ANSI;
|
|
}
|
|
|
|
#ifdef __WINDOWS__
|
|
return COLOUR_METHOD_WINDOWS;
|
|
#else
|
|
return COLOUR_METHOD_NONE;
|
|
#endif
|
|
}
|
|
|
|
static const char* GetAnsiColorCode(CLIColour color) {
|
|
switch (color) {
|
|
case RED: return "1";
|
|
case GREEN: return "2";
|
|
case YELLOW:
|
|
return "3";
|
|
default: return NULL;
|
|
};
|
|
}
|
|
|
|
#ifdef __WINDOWS__
|
|
|
|
static WORD GetCurrentWindowsConsoleAttribute(HANDLE hConsoleOutput)
|
|
{
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
|
GetConsoleScreenBufferInfo(hConsoleOutput, &csbi);
|
|
return csbi.wAttributes;
|
|
}
|
|
|
|
static WORD GetWindowsConsoleAttribute(CLIColour color, WORD defaultAttr)
|
|
{
|
|
switch (color) {
|
|
case RED: return FOREGROUND_RED;
|
|
case GREEN: return FOREGROUND_GREEN;
|
|
case YELLOW: return FOREGROUND_RED | FOREGROUND_GREEN;
|
|
default: return defaultAttr;
|
|
};
|
|
}
|
|
|
|
#endif
|
|
|
|
static void Write_VA(Verbosity verbosity, CLIColour colour, const char *fmt, va_list args)
|
|
{
|
|
if (_verbosity < verbosity) return;
|
|
|
|
COLOUR_METHOD colourMethod = GetColourMethod();
|
|
|
|
if (colour == CLIColour::DEFAULT || colourMethod == COLOUR_METHOD_NONE) {
|
|
vprintf(fmt, args);
|
|
} else if (colourMethod == COLOUR_METHOD_ANSI) {
|
|
printf("\033[0;3%sm", GetAnsiColorCode(colour));
|
|
vprintf(fmt, args);
|
|
printf("\033[m");
|
|
} else if (colourMethod == COLOUR_METHOD_WINDOWS) {
|
|
#ifdef __WINDOWS__
|
|
HANDLE hStdOut = GetStdHandle(STD_OUTPUT_HANDLE);
|
|
WORD defaultAttr = GetCurrentWindowsConsoleAttribute(hStdOut);
|
|
SetConsoleTextAttribute(hStdOut, GetWindowsConsoleAttribute(colour, defaultAttr));
|
|
vprintf(fmt, args);
|
|
SetConsoleTextAttribute(hStdOut, defaultAttr);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
static void Write(Verbosity verbosity, CLIColour colour, const char *fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
Write_VA(verbosity, colour, fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
static void Write(Verbosity verbosity, const char * fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
Write_VA(verbosity, DEFAULT, fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
static void Write(CLIColour colour, const char * fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
Write_VA(NORMAL, colour, fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
static void Write(const char * fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
Write_VA(NORMAL, DEFAULT, fmt, args);
|
|
va_end(args);
|
|
}
|
|
|
|
#if defined(__WINDOWS__)
|
|
|
|
#include <shellapi.h>
|
|
|
|
int main(int argc, char *argv[]);
|
|
|
|
#define OPENRCT2_DLL_MODULE_NAME "openrct2.dll"
|
|
|
|
static HMODULE _dllModule = NULL;
|
|
|
|
utf8 *utf8_write_codepoint(utf8 *dst, uint32 codepoint)
|
|
{
|
|
if (codepoint <= 0x7F) {
|
|
dst[0] = (utf8)codepoint;
|
|
return dst + 1;
|
|
} else if (codepoint <= 0x7FF) {
|
|
dst[0] = 0xC0 | ((codepoint >> 6) & 0x1F);
|
|
dst[1] = 0x80 | (codepoint & 0x3F);
|
|
return dst + 2;
|
|
} else if (codepoint <= 0xFFFF) {
|
|
dst[0] = 0xE0 | ((codepoint >> 12) & 0x0F);
|
|
dst[1] = 0x80 | ((codepoint >> 6) & 0x3F);
|
|
dst[2] = 0x80 | (codepoint & 0x3F);
|
|
return dst + 3;
|
|
} else {
|
|
dst[0] = 0xF0 | ((codepoint >> 18) & 0x07);
|
|
dst[1] = 0x80 | ((codepoint >> 12) & 0x3F);
|
|
dst[2] = 0x80 | ((codepoint >> 6) & 0x3F);
|
|
dst[3] = 0x80 | (codepoint & 0x3F);
|
|
return dst + 4;
|
|
}
|
|
}
|
|
|
|
utf8 *widechar_to_utf8(const wchar_t *src)
|
|
{
|
|
utf8 *result = (utf8 *)malloc((wcslen(src) * 4) + 1);
|
|
utf8 *dst = result;
|
|
|
|
for (; *src != 0; src++) {
|
|
dst = utf8_write_codepoint(dst, *src);
|
|
}
|
|
*dst++ = 0;
|
|
|
|
size_t size = (size_t)(dst - result);
|
|
return (utf8 *)realloc(result, size);
|
|
}
|
|
|
|
utf8 **windows_get_command_line_args(int *outNumArgs)
|
|
{
|
|
int argc;
|
|
|
|
// Get command line arguments as widechar
|
|
LPWSTR commandLine = GetCommandLineW();
|
|
LPWSTR *argvW = CommandLineToArgvW(commandLine, &argc);
|
|
|
|
// Convert to UTF-8
|
|
utf8 **argvUtf8 = (utf8**)malloc(argc * sizeof(utf8*));
|
|
for (int i = 0; i < argc; i++) {
|
|
argvUtf8[i] = widechar_to_utf8(argvW[i]);
|
|
}
|
|
LocalFree(argvW);
|
|
|
|
*outNumArgs = argc;
|
|
return argvUtf8;
|
|
}
|
|
|
|
BOOL APIENTRY DllMain(HANDLE hModule, DWORD dwReason, LPVOID lpReserved)
|
|
{
|
|
_dllModule = (HMODULE)hModule;
|
|
return TRUE;
|
|
}
|
|
|
|
__declspec(dllexport) int StartOpenRCT(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
|
|
{
|
|
|
|
if (_dllModule == NULL) {
|
|
_dllModule = GetModuleHandleA(OPENRCT2_DLL_MODULE_NAME);
|
|
}
|
|
|
|
int argc;
|
|
char ** argv = (char**)windows_get_command_line_args(&argc);
|
|
|
|
int gExitCode = main(argc, argv);
|
|
|
|
// Free argv
|
|
for (int i = 0; i < argc; i++) {
|
|
free(argv[i]);
|
|
}
|
|
free(argv);
|
|
|
|
exit(gExitCode);
|
|
return gExitCode;
|
|
}
|
|
|
|
#endif
|
|
|
|
char *segments = (char *)(GOOD_PLACE_FOR_DATA_SEGMENT);
|
|
|
|
static uint32 sawyercoding_calculate_checksum(const uint8* buffer, size_t length)
|
|
{
|
|
size_t i;
|
|
uint32 checksum = 0;
|
|
for (i = 0; i < length; i++)
|
|
checksum += buffer[i];
|
|
|
|
return checksum;
|
|
}
|
|
|
|
/**
|
|
* Loads RCT2's data model and remaps the addresses.
|
|
* @returns true if the data integrity check succeeded, otherwise false.
|
|
*/
|
|
static bool openrct2_setup_rct2_segment()
|
|
{
|
|
// OpenRCT2 on Linux and macOS is wired to have the original Windows PE sections loaded
|
|
// necessary. Windows does not need to do this as OpenRCT2 runs as a DLL loaded from the Windows PE.
|
|
int len = 0x01429000 - 0x8a4000; // 0xB85000, 12079104 bytes or around 11.5MB
|
|
int err = 0;
|
|
// in some configurations err and len may be unused
|
|
UNUSED(err);
|
|
UNUSED(len);
|
|
#if defined(__unix__)
|
|
int pageSize = getpagesize();
|
|
int numPages = (len + pageSize - 1) / pageSize;
|
|
unsigned char *dummy = (unsigned char *)malloc(numPages);
|
|
|
|
err = mincore((void *)segments, len, dummy);
|
|
bool pagesMissing = false;
|
|
if (err != 0)
|
|
{
|
|
err = errno;
|
|
#ifdef __LINUX__
|
|
// On Linux ENOMEM means all requested range is unmapped
|
|
if (err != ENOMEM)
|
|
{
|
|
pagesMissing = true;
|
|
perror("mincore");
|
|
}
|
|
#else
|
|
pagesMissing = true;
|
|
perror("mincore");
|
|
#endif // __LINUX__
|
|
} else {
|
|
for (int i = 0; i < numPages; i++)
|
|
{
|
|
if (dummy[i] != 1)
|
|
{
|
|
pagesMissing = true;
|
|
void *start = (void *)(segments + i * pageSize);
|
|
void *end = (void *)(segments + (i + 1) * pageSize - 1);
|
|
log_warning("required page %p - %p is not in memory!", start, end);
|
|
}
|
|
}
|
|
}
|
|
free(dummy);
|
|
if (pagesMissing)
|
|
{
|
|
log_error("At least one of required pages was not found in memory. This can cause segfaults later on.");
|
|
}
|
|
|
|
// section: text
|
|
err = mprotect((void *)0x401000, 0x8a4000 - 0x401000, PROT_READ | PROT_EXEC | PROT_WRITE);
|
|
if (err != 0)
|
|
{
|
|
perror("mprotect");
|
|
}
|
|
|
|
// section: rw data
|
|
err = mprotect((void *)segments, 0x01429000 - 0x8a4000, PROT_READ | PROT_WRITE);
|
|
if (err != 0)
|
|
{
|
|
perror("mprotect");
|
|
}
|
|
#endif // defined(__unix__)
|
|
|
|
// Check that the expected data is at various addresses.
|
|
// Start at 0x9a6000, which is start of .data, to skip the region containing addresses to DLL
|
|
// calls, which can be changed by windows/wine loader.
|
|
const uint32 c1 = sawyercoding_calculate_checksum((const uint8*)(segments + (uintptr_t)(0x009A6000 - 0x8a4000)), 0x009E0000 - 0x009A6000);
|
|
const uint32 c2 = sawyercoding_calculate_checksum((const uint8*)(segments + (uintptr_t)(0x01428000 - 0x8a4000)), 0x014282BC - 0x01428000);
|
|
const uint32 exp_c1 = 10114815;
|
|
const uint32 exp_c2 = 23564;
|
|
if (c1 != exp_c1 || c2 != exp_c2) {
|
|
log_warning("c1 = %u, expected %u, match %d", c1, exp_c1, c1 == exp_c1);
|
|
log_warning("c2 = %u, expected %u, match %d", c2, exp_c2, c2 == exp_c2);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static void PrintRideTypes()
|
|
{
|
|
for (uint8 rideType = 0; rideType < RIDE_TYPE_COUNT; rideType++) {
|
|
CLIColour colour = CLIColour::DEFAULT;
|
|
bool implemented = Utils::rideIsImplemented(rideType);
|
|
const char * rideName = RideNames[rideType];
|
|
const char * status = "";
|
|
if (implemented) {
|
|
status = " [IMPLEMENTED]";
|
|
colour = CLIColour::GREEN;
|
|
}
|
|
|
|
Write(colour, "%2d: %-30s%s\n", rideType, rideName, status);
|
|
}
|
|
}
|
|
|
|
#include "GeneralSupportHeightCall.hpp"
|
|
|
|
static void TestGeneralSupportHeightCall() {
|
|
SupportCall callA = {16, 0x20};
|
|
SupportCall callB = {32, 0x20};
|
|
SupportCall callC = {48, 0x20};
|
|
SupportCall callD = {48, 0x1F};
|
|
|
|
SupportCall out = {0,0};
|
|
bool success;
|
|
|
|
SupportCall groupA[4] = {callA, callA, callA, callA};
|
|
success = GeneralSupportHeightCall::FindMostCommonSupportCall(groupA, &out);
|
|
assert(success);
|
|
assert(out == callA);
|
|
|
|
SupportCall groupB[4] = {callB, callA, callA, callA};
|
|
success = GeneralSupportHeightCall::FindMostCommonSupportCall(groupB, &out);
|
|
assert(success);
|
|
assert(out == callA);
|
|
|
|
SupportCall groupC[4] = {callB, callA, callB, callA};
|
|
success = GeneralSupportHeightCall::FindMostCommonSupportCall(groupC, &out);
|
|
assert(!success);
|
|
|
|
SupportCall groupD[4] = {callB, callC, callB, callA};
|
|
success = GeneralSupportHeightCall::FindMostCommonSupportCall(groupD, &out);
|
|
assert(!success);
|
|
|
|
SupportCall groupE[4] = {callD, callC, callB, callA};
|
|
success = GeneralSupportHeightCall::FindMostCommonSupportCall(groupE, &out);
|
|
assert(!success);
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
TestGeneralSupportHeightCall();
|
|
|
|
std::vector<TestCase> testCases;
|
|
|
|
bool generate = false;
|
|
uint8 specificRideType = 0xFF;
|
|
for (int i = 0; i < argc; ++i) {
|
|
char *arg = argv[i];
|
|
if (strcmp(arg, "--gtest_color=no") == 0) {
|
|
gTestColor = false;
|
|
}
|
|
else if (strcmp(arg, "--quiet") == 0) {
|
|
_verbosity = Verbosity::QUIET;
|
|
}
|
|
else if (strcmp(arg, "--ride-type") == 0) {
|
|
if (i + 1 < argc) {
|
|
i++;
|
|
specificRideType = atoi(argv[i]);
|
|
} else {
|
|
PrintRideTypes();
|
|
return 2;
|
|
}
|
|
}
|
|
else if (strcmp(arg, "--generate") == 0) {
|
|
generate = true;
|
|
}
|
|
}
|
|
|
|
if (generate) {
|
|
if (specificRideType > 90) {
|
|
fprintf(stderr, "No ride or invalid ride specified.\n");
|
|
return 1;
|
|
}
|
|
|
|
openrct2_setup_rct2_segment();
|
|
PaintIntercept::InitHooks();
|
|
|
|
return generatePaintCode(specificRideType);
|
|
}
|
|
|
|
for (uint8 rideType = 0; rideType < RIDE_TYPE_COUNT; rideType++) {
|
|
if (specificRideType != 0xFF && rideType != specificRideType) {
|
|
continue;
|
|
}
|
|
|
|
if (!Utils::rideIsImplemented(rideType)) {
|
|
continue;
|
|
}
|
|
|
|
TestCase testCase = {0};
|
|
testCase.rideType = rideType;
|
|
|
|
if (ride_type_has_flag(rideType, RIDE_TYPE_FLAG_FLAT_RIDE)) {
|
|
testCase.trackTypes.push_back(RideConstructionDefaultTrackType[rideType]);
|
|
} else {
|
|
for (int trackType = 0; trackType < 256; trackType++) {
|
|
if (Utils::rideSupportsTrackType(rideType, trackType)) {
|
|
testCase.trackTypes.push_back(trackType);
|
|
}
|
|
}
|
|
}
|
|
|
|
testCases.push_back(testCase);
|
|
}
|
|
|
|
int testCaseCount = (int) testCases.size();
|
|
int testCount = 0;
|
|
for (auto &&tc : testCases) {
|
|
testCount += tc.trackTypes.size();
|
|
}
|
|
|
|
Write(CLIColour::GREEN, "[==========] ");
|
|
Write("Running %d tests from %d test cases.\n", testCount, testCaseCount);
|
|
|
|
Write(CLIColour::GREEN, "[----------] ");
|
|
Write("Global test environment set-up.\n");
|
|
openrct2_setup_rct2_segment();
|
|
PaintIntercept::InitHooks();
|
|
|
|
int successCount = 0;
|
|
std::vector<utf8string> failures;
|
|
for (auto &&tc : testCases) {
|
|
const utf8string rideTypeName = RideNames[tc.rideType];
|
|
Write(CLIColour::GREEN, "[----------] ");
|
|
Write("%d tests from %s\n", (int)tc.trackTypes.size(), rideTypeName);
|
|
|
|
for (auto &&trackType : tc.trackTypes) {
|
|
utf8string trackTypeName;
|
|
if (ride_type_has_flag(tc.rideType, RIDE_TYPE_FLAG_FLAT_RIDE)) {
|
|
trackTypeName = FlatTrackNames[trackType];
|
|
} else {
|
|
trackTypeName = TrackNames[trackType];
|
|
}
|
|
|
|
Write(CLIColour::GREEN, "[ RUN ] ");
|
|
Write("%s.%s\n", rideTypeName, trackTypeName);
|
|
|
|
std::string out;
|
|
int retVal = TestTrack::TestPaintTrackElement(tc.rideType, trackType, &out);
|
|
Write("%s", out.c_str());
|
|
switch (retVal) {
|
|
case TEST_SUCCESS:
|
|
Write(CLIColour::GREEN, "[ OK ] ");
|
|
Write("%s.%s (0 ms)\n", rideTypeName, trackTypeName);
|
|
successCount++;
|
|
break;
|
|
|
|
case TEST_SKIPPED:
|
|
Write("Skipped\n");
|
|
// Outputting this as OK because CLion only allows FAILED or OK
|
|
Write(CLIColour::YELLOW, "[ OK ] ");
|
|
Write("%s.%s (0 ms)\n", rideTypeName, trackTypeName);
|
|
successCount++;
|
|
break;
|
|
|
|
case TEST_FAILED:
|
|
utf8string testCaseName = new utf8[64];
|
|
snprintf(testCaseName, 64, "%s.%s", rideTypeName, trackTypeName);
|
|
|
|
Write(CLIColour::RED, "[ FAILED ] ");
|
|
Write("%s (0 ms)\n", testCaseName);
|
|
failures.push_back(testCaseName);
|
|
break;
|
|
}
|
|
}
|
|
|
|
Write(CLIColour::GREEN, "[----------] ");
|
|
Write("%d tests from %s (0 ms total)\n", (int)tc.trackTypes.size(), rideTypeName);
|
|
}
|
|
Write("\n");
|
|
|
|
Write(CLIColour::GREEN, "[----------] ");
|
|
Write("Global test environment tear-down\n");
|
|
|
|
Write(CLIColour::GREEN, "[==========] ");
|
|
Write("%d tests from %d test cases ran. (0 ms total).\n", testCount, testCaseCount);
|
|
|
|
Write(Verbosity::QUIET, CLIColour::GREEN, "[ PASSED ] ");
|
|
Write(Verbosity::QUIET, "%d tests.\n", successCount);
|
|
|
|
if (failures.size() > 0) {
|
|
Write(Verbosity::QUIET, CLIColour::RED, "[ FAILED ] ");
|
|
Write(Verbosity::QUIET, "%d tests, listed below:\n", (int)failures.size());
|
|
|
|
for (auto &&failure : failures) {
|
|
Write(Verbosity::QUIET, CLIColour::RED, "[ FAILED ] ");
|
|
Write(Verbosity::QUIET, "%s\n", failure);
|
|
delete [] failure;
|
|
}
|
|
|
|
Write(Verbosity::QUIET, "\n");
|
|
|
|
Write(Verbosity::QUIET, "%d FAILED TESTS\n", (int)failures.size());
|
|
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|