// license:BSD-3-Clause // copyright-holders:Olivier Galibert, R. Belmont //============================================================ // // sdlos_*.c - OS specific low level code // // SDLMAME by Olivier Galibert and R. Belmont // //============================================================ #define WIN32_LEAN_AND_MEAN #include #include #include #ifndef _MSC_VER #include #endif #include #include // MAME headers #include "osdlib.h" #include "osdcomm.h" #include "osdcore.h" #include "strconv.h" #include #include #include "strconv.h" using namespace Platform; using namespace Windows::ApplicationModel::DataTransfer; using namespace Windows::Foundation; #include //============================================================ // MACROS //============================================================ // presumed size of a page of memory #define PAGE_SIZE 4096 // align allocations to start or end of the page? #define GUARD_ALIGN_START 0 #if defined(__BIGGEST_ALIGNMENT__) #define MAX_ALIGNMENT __BIGGEST_ALIGNMENT__ #elif defined(__AVX__) #define MAX_ALIGNMENT 32 #elif defined(__SSE__) || defined(__x86_64__) || defined(_M_X64) #define MAX_ALIGNMENT 16 #else #define MAX_ALIGNMENT sizeof(int64_t) #endif //============================================================ // GLOBAL VARIABLES //============================================================ std::map> g_runtime_environment; //============================================================ // osd_getenv //============================================================ const char *osd_getenv(const char *name) { for (auto iter = g_runtime_environment.begin(); iter != g_runtime_environment.end(); iter++) { if (stricmp(iter->first, name) == 0) { osd_printf_debug("ENVIRONMENT: Get %s = value: '%s'", name, iter->second.get()); return iter->second.get(); } } return nullptr; } //============================================================ // osd_setenv //============================================================ int osd_setenv(const char *name, const char *value, int overwrite) { if (!overwrite) { if (osd_getenv(name) != nullptr) return 0; } auto buf = std::make_unique(strlen(name) + strlen(value) + 2); sprintf(buf.get(), "%s=%s", name, value); g_runtime_environment[name] = std::move(buf); osd_printf_debug("ENVIRONMENT: Set %s to value: '%s'", name, buf.get()); return 0; } //============================================================ // osd_process_kill //============================================================ void osd_process_kill() { std::fflush(stdout); std::fflush(stderr); TerminateProcess(GetCurrentProcess(), -1); } //============================================================ // osd_malloc //============================================================ void *osd_malloc(size_t size) { #ifndef MALLOC_DEBUG return malloc(size); #else // add in space for the size and offset size += MAX_ALIGNMENT + sizeof(size_t) + 2; size &= ~size_t(1); // basic objects just come from the heap uint8_t *const block = reinterpret_cast(HeapAlloc(GetProcessHeap(), 0, size)); if (block == nullptr) return nullptr; uint8_t *const result = reinterpret_cast(reinterpret_cast(block + sizeof(size_t) + MAX_ALIGNMENT) & ~(uintptr_t(MAX_ALIGNMENT) - 1)); // store the size and return and pointer to the data afterward *reinterpret_cast(block) = size; *(result - 1) = result - block; return result; #endif } //============================================================ // osd_malloc_array //============================================================ void *osd_malloc_array(size_t size) { #ifndef MALLOC_DEBUG return malloc(size); #else // add in space for the size and offset size += MAX_ALIGNMENT + sizeof(size_t) + 2; size &= ~size_t(1); // round the size up to a page boundary size_t const rounded_size = ((size + sizeof(void *) + PAGE_SIZE - 1) / PAGE_SIZE) * PAGE_SIZE; // reserve that much memory, plus two guard pages void *page_base = VirtualAlloc(nullptr, rounded_size + 2 * PAGE_SIZE, MEM_RESERVE, PAGE_NOACCESS); if (page_base == nullptr) return nullptr; // now allow access to everything but the first and last pages page_base = VirtualAlloc(reinterpret_cast(page_base) + PAGE_SIZE, rounded_size, MEM_COMMIT, PAGE_READWRITE); if (page_base == nullptr) return nullptr; // work backwards from the page base to get to the block base uint8_t *const block = GUARD_ALIGN_START ? reinterpret_cast(page_base) : (reinterpret_cast(page_base) + rounded_size - size); uint8_t *const result = reinterpret_cast(reinterpret_cast(block + sizeof(size_t) + MAX_ALIGNMENT) & ~(uintptr_t(MAX_ALIGNMENT) - 1)); // store the size at the start with a flag indicating it has a guard page *reinterpret_cast(block) = size | 1; *(result - 1) = result - block; return result; #endif } //============================================================ // osd_free //============================================================ void osd_free(void *ptr) { #ifndef MALLOC_DEBUG free(ptr); #else uint8_t const offset = *(reinterpret_cast(ptr) - 1); uint8_t *const block = reinterpret_cast(ptr) - offset; size_t const size = *reinterpret_cast(block); if ((size & 0x1) == 0) { // if no guard page, just free the pointer HeapFree(GetProcessHeap(), 0, block); } else { // large items need more care ULONG_PTR const page_base = reinterpret_cast(block) & ~(PAGE_SIZE - 1); VirtualFree(reinterpret_cast(page_base - PAGE_SIZE), 0, MEM_RELEASE); } #endif } //============================================================ // osd_alloc_executable // // allocates "size" bytes of executable memory. this must take // things like NX support into account. //============================================================ void *osd_alloc_executable(size_t size) { return nullptr; } //============================================================ // osd_free_executable // // frees memory allocated with osd_alloc_executable //============================================================ void osd_free_executable(void *ptr, size_t size) { } //============================================================ // osd_break_into_debugger //============================================================ void osd_break_into_debugger(const char *message) { #ifdef OSD_WINDOWS if (IsDebuggerPresent()) { win_output_debug_string_utf8(message); DebugBreak(); } else if (s_debugger_stack_crawler != nullptr) (*s_debugger_stack_crawler)(); #else if (IsDebuggerPresent()) { OutputDebugStringA(message); __debugbreak(); } #endif } //============================================================ // get_clipboard_text_by_format //============================================================ static char *get_clipboard_text_by_format(UINT format, std::string (*convert)(LPCVOID data)) { /*DataPackageView^ dataPackageView; IAsyncOperation^ getTextOp; String^ clipboardText; dataPackageView = Clipboard::GetContent(); getTextOp = dataPackageView->GetTextAsync(); clipboardText = getTextOp->GetResults(); return osd::text::from_wstring(clipboardText->Data()).c_str();*/ return nullptr; } //============================================================ // convert_wide //============================================================ static std::string convert_wide(LPCVOID data) { return osd::text::from_wstring((LPCWSTR) data); } //============================================================ // convert_ansi //============================================================ static std::string convert_ansi(LPCVOID data) { return osd::text::from_astring((LPCSTR) data); } //============================================================ // osd_get_clipboard_text //============================================================ char *osd_get_clipboard_text(void) { // try to access unicode text char *result = get_clipboard_text_by_format(CF_UNICODETEXT, convert_wide); // try to access ANSI text if (result == nullptr) result = get_clipboard_text_by_format(CF_TEXT, convert_ansi); return result; } //============================================================ // osd_dynamic_bind //============================================================ #if WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP) // for classic desktop applications #define load_library(filename) LoadLibrary(filename) #else // for Windows Store universal applications // This needs to change ASAP as it won't be allowed in the store typedef HMODULE __stdcall t_LLA(const char *); typedef FARPROC __stdcall t_GPA(HMODULE H, const char *); PIMAGE_NT_HEADERS WINAPI ImageNtHeader(PVOID Base) { return (PIMAGE_NT_HEADERS) ((LPBYTE)Base + ((PIMAGE_DOS_HEADER)Base)->e_lfanew); } PIMAGE_SECTION_HEADER WINAPI RtlImageRvaToSection(const IMAGE_NT_HEADERS *nt, HMODULE module, DWORD_PTR rva) { int i; const IMAGE_SECTION_HEADER *sec; sec = (const IMAGE_SECTION_HEADER*)((const char*)&nt->OptionalHeader + nt->FileHeader.SizeOfOptionalHeader); for (i = 0; i < nt->FileHeader.NumberOfSections; i++, sec++) { if ((sec->VirtualAddress <= rva) && (sec->VirtualAddress + sec->SizeOfRawData > rva)) return (PIMAGE_SECTION_HEADER)sec; } return NULL; } PVOID WINAPI RtlImageRvaToVa(const IMAGE_NT_HEADERS *nt, HMODULE module, DWORD_PTR rva, IMAGE_SECTION_HEADER **section) { IMAGE_SECTION_HEADER *sec; if (section && *section) /* try this section first */ { sec = *section; if ((sec->VirtualAddress <= rva) && (sec->VirtualAddress + sec->SizeOfRawData > rva)) goto found; } if (!(sec = RtlImageRvaToSection(nt, module, rva))) return NULL; found: if (section) *section = sec; return (char *)module + sec->PointerToRawData + (rva - sec->VirtualAddress); } PVOID WINAPI ImageDirectoryEntryToDataEx(PVOID base, BOOLEAN image, USHORT dir, PULONG size, PIMAGE_SECTION_HEADER *section) { const IMAGE_NT_HEADERS *nt; DWORD_PTR addr; *size = 0; if (section) *section = NULL; if (!(nt = ImageNtHeader(base))) return NULL; if (dir >= nt->OptionalHeader.NumberOfRvaAndSizes) return NULL; if (!(addr = nt->OptionalHeader.DataDirectory[dir].VirtualAddress)) return NULL; *size = nt->OptionalHeader.DataDirectory[dir].Size; if (image || addr < nt->OptionalHeader.SizeOfHeaders) return (char *)base + addr; return RtlImageRvaToVa(nt, (HMODULE)base, addr, section); } // == Windows API GetProcAddress void *PeGetProcAddressA(void *Base, LPCSTR Name) { DWORD Tmp; IMAGE_NT_HEADERS *NT = ImageNtHeader(Base); IMAGE_EXPORT_DIRECTORY *Exp = (IMAGE_EXPORT_DIRECTORY*)ImageDirectoryEntryToDataEx(Base, TRUE, IMAGE_DIRECTORY_ENTRY_EXPORT, &Tmp, 0); if (Exp == 0 || Exp->NumberOfFunctions == 0) { SetLastError(ERROR_NOT_FOUND); return 0; } DWORD *Names = (DWORD*)(Exp->AddressOfNames + (DWORD_PTR)Base); WORD *Ordinals = (WORD*)(Exp->AddressOfNameOrdinals + (DWORD_PTR)Base); DWORD *Functions = (DWORD*)(Exp->AddressOfFunctions + (DWORD_PTR)Base); FARPROC Ret = 0; if ((DWORD_PTR)Name<65536) { if ((DWORD_PTR)Name - Exp->BaseNumberOfFunctions) Ret = (FARPROC)(Functions[(DWORD_PTR)Name - Exp->Base] + (DWORD_PTR)Base); } else { for (DWORD i = 0; iNumberOfNames && Ret == 0; i++) { char *Func = (char*)(Names[i] + (DWORD_PTR)Base); if (Func && strcmp(Func, Name) == 0) Ret = (FARPROC)(Functions[Ordinals[i]] + (DWORD_PTR)Base); } } if (Ret) { DWORD ExpStart = NT->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress + (DWORD)Base; DWORD ExpSize = NT->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].Size; if ((DWORD)Ret >= ExpStart && (DWORD)Ret <= ExpStart + ExpSize) { // Forwarder return 0; } return Ret; } return 0; } t_LLA* g_LoadLibraryA = 0; t_GPA* g_GetProcAddressA = 0; void find_load_exports() { char *Tmp = (char*)GetTickCount64; Tmp = (char*)((~0xFFF)&(DWORD_PTR)Tmp); while (Tmp) { __try { if (Tmp[0] == 'M' && Tmp[1] == 'Z') break; } __except (EXCEPTION_EXECUTE_HANDLER) { } Tmp -= 0x1000; } if (Tmp == 0) return; g_LoadLibraryA = (t_LLA*)PeGetProcAddressA(Tmp, "LoadLibraryA"); g_GetProcAddressA = (t_GPA*)PeGetProcAddressA(Tmp, "GetProcAddress"); } #define load_library(filename) g_LoadLibraryA(osd::text::from_wstring(filename).c_str()) #define get_proc_address(mod, proc) g_GetProcAddressA(mod, proc) #endif namespace osd { class dynamic_module_win32_impl : public dynamic_module { public: dynamic_module_win32_impl(std::vector &libraries) : m_module(nullptr) { m_libraries = libraries; #if defined(OSD_UWP) find_load_exports(); #endif } virtual ~dynamic_module_win32_impl() override { if (m_module != nullptr) FreeLibrary(m_module); }; protected: virtual generic_fptr_t get_symbol_address(char const *symbol) override { /* * given a list of libraries, if a first symbol is successfully loaded from * one of them, all additional symbols will be loaded from the same library */ if (m_module) { return reinterpret_cast(GetProcAddress(m_module, symbol)); } for (auto const &library : m_libraries) { osd::text::tstring tempstr = osd::text::to_tstring(library); HMODULE module = load_library(tempstr.c_str()); if (module != nullptr) { generic_fptr_t function = reinterpret_cast(GetProcAddress(module, symbol)); if (function != nullptr) { m_module = module; return function; } else { FreeLibrary(module); } } } return nullptr; } private: std::vector m_libraries; HMODULE m_module; }; dynamic_module::ptr dynamic_module::open(std::vector &&names) { return std::make_unique(names); } } // namespace osd