steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
factor/vm/os-windows.cpp

419 lines
12 KiB
C++
Raw Blame History

#include "master.hpp"
namespace factor {
HMODULE hFactorDll;
bool set_memory_locked(cell base, cell size, bool locked) {
int prot = locked ? PAGE_NOACCESS : PAGE_READWRITE;
DWORD ignore;
int status = VirtualProtect((char*)base, size, prot, &ignore);
return status != 0;
}
void factor_vm::init_ffi() {
hFactorDll = GetModuleHandle(NULL);
if (!hFactorDll)
fatal_error("GetModuleHandle() failed", 0);
}
void factor_vm::ffi_dlopen(dll* dll) {
dll->handle = LoadLibraryEx((WCHAR*)alien_offset(dll->path), NULL, 0);
}
cell factor_vm::ffi_dlsym(dll* dll, symbol_char* symbol) {
return (cell)GetProcAddress(dll ? (HMODULE) dll->handle : hFactorDll,
symbol);
}
cell factor_vm::ffi_dlsym_raw(dll* dll, symbol_char* symbol) {
return ffi_dlsym(dll, symbol);
}
void factor_vm::ffi_dlclose(dll* dll) {
FreeLibrary((HMODULE) dll->handle);
dll->handle = NULL;
}
BOOL factor_vm::windows_stat(vm_char* path) {
BY_HANDLE_FILE_INFORMATION bhfi;
HANDLE h = CreateFileW(path, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL,
OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (h == INVALID_HANDLE_VALUE) {
// FindFirstFile is the only call that can stat c:\pagefile.sys
WIN32_FIND_DATA st;
HANDLE h;
if (INVALID_HANDLE_VALUE == (h = FindFirstFile(path, &st)))
return false;
FindClose(h);
return true;
}
BOOL ret = GetFileInformationByHandle(h, &bhfi);
CloseHandle(h);
return ret;
}
// You must free() this yourself.
const vm_char* factor_vm::default_image_path() {
vm_char full_path[MAX_UNICODE_PATH];
vm_char* ptr;
vm_char temp_path[MAX_UNICODE_PATH];
if (!GetModuleFileName(NULL, full_path, MAX_UNICODE_PATH))
fatal_error("GetModuleFileName() failed", 0);
if ((ptr = wcsrchr(full_path, '.')))
*ptr = 0;
wcsncpy(temp_path, full_path, MAX_UNICODE_PATH - 1);
size_t full_path_len = wcslen(full_path);
if (full_path_len < MAX_UNICODE_PATH - 1)
wcsncat(temp_path, L".image", MAX_UNICODE_PATH - full_path_len - 1);
temp_path[MAX_UNICODE_PATH - 1] = 0;
return safe_strdup(temp_path);
}
// You must free() this yourself.
const vm_char* factor_vm::vm_executable_path() {
vm_char full_path[MAX_UNICODE_PATH];
if (!GetModuleFileName(NULL, full_path, MAX_UNICODE_PATH))
fatal_error("GetModuleFileName() failed", 0);
return safe_strdup(full_path);
}
void factor_vm::primitive_existsp() {
vm_char* path = untag_check<byte_array>(ctx->pop())->data<vm_char>();
ctx->push(tag_boolean(windows_stat(path)));
}
segment::segment(cell size_, bool executable_p) {
size = size_;
char* mem;
cell alloc_size = getpagesize() * 2 + size;
if ((mem = (char*)VirtualAlloc(
NULL, alloc_size, MEM_COMMIT,
executable_p ? PAGE_EXECUTE_READWRITE : PAGE_READWRITE)) ==
0) {
fatal_error("Out of memory in VirtualAlloc", alloc_size);
}
start = (cell)mem + getpagesize();
end = start + size;
set_border_locked(true);
}
segment::~segment() {
SYSTEM_INFO si;
GetSystemInfo(&si);
if (!VirtualFree((void*)(start - si.dwPageSize), 0, MEM_RELEASE))
fatal_error("Segment deallocation failed", 0);
}
long getpagesize() {
static long g_pagesize = 0;
if (!g_pagesize) {
SYSTEM_INFO system_info;
GetSystemInfo(&system_info);
g_pagesize = system_info.dwPageSize;
}
return g_pagesize;
}
bool move_file(const vm_char* path1, const vm_char* path2) {
// MoveFileEx returns FALSE on fail.
BOOL val = MoveFileEx((path1), (path2), MOVEFILE_REPLACE_EXISTING);
if (val == FALSE) {
// MoveFileEx doesn't set errno, which primitive_save_image()
// reads the error code from. Instead of converting from
// GetLastError() to errno values, we ust set it to the generic
// EIO value.
errno = EIO;
}
return val == TRUE;
}
void factor_vm::init_signals() {}
THREADHANDLE start_thread(void* (*start_routine)(void*), void* args) {
return (void*)CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) start_routine,
args, 0, 0);
}
uint64_t nano_count() {
static double scale_factor;
static uint32_t hi = 0;
static uint32_t lo = 0;
// Note: on older systems QueryPerformanceCounter may be unreliable
// until you add /usepmtimer to Boot.ini. I had an issue where two
// nano_count calls would show a difference of about 1 second,
// while actually about 80 seconds have passed. The /usepmtimer
// switch cured the issue on that PC (WinXP Pro SP3 32-bit).
// See also http://www.virtualdub.org/blog/pivot/entry.php?id=106
LARGE_INTEGER count;
BOOL ret = QueryPerformanceCounter(&count);
if (ret == 0)
fatal_error("QueryPerformanceCounter", 0);
if (scale_factor == 0.0) {
LARGE_INTEGER frequency;
BOOL ret = QueryPerformanceFrequency(&frequency);
if (ret == 0)
fatal_error("QueryPerformanceFrequency", 0);
scale_factor = (1000000000.0 / frequency.QuadPart);
}
#ifdef FACTOR_64
hi = count.HighPart;
#else
// On VirtualBox, QueryPerformanceCounter does not increment
// the high part every time the low part overflows. Workaround.
if (lo > count.LowPart)
hi++;
#endif
lo = count.LowPart;
return (uint64_t)((((uint64_t)hi << 32) | (uint64_t)lo) * scale_factor);
}
void sleep_nanos(uint64_t nsec) { Sleep((DWORD)(nsec / 1000000)); }
#ifndef EXCEPTION_DISPOSITION
typedef enum _EXCEPTION_DISPOSITION {
ExceptionContinueExecution = 0,
ExceptionContinueSearch = 1,
ExceptionNestedException = 2,
ExceptionCollidedUnwind = 3
} EXCEPTION_DISPOSITION;
#endif
LONG factor_vm::exception_handler(PEXCEPTION_RECORD e, void* frame, PCONTEXT c,
void* dispatch) {
(void)frame;
(void)dispatch;
switch (e->ExceptionCode) {
case EXCEPTION_ACCESS_VIOLATION:
set_memory_protection_error(e->ExceptionInformation[1], c->EIP);
dispatch_signal_handler((cell*)&c->ESP, (cell*)&c->EIP,
(cell)factor::memory_signal_handler_impl);
break;
case STATUS_FLOAT_DENORMAL_OPERAND:
case STATUS_FLOAT_DIVIDE_BY_ZERO:
case STATUS_FLOAT_INEXACT_RESULT:
case STATUS_FLOAT_INVALID_OPERATION:
case STATUS_FLOAT_OVERFLOW:
case STATUS_FLOAT_STACK_CHECK:
case STATUS_FLOAT_UNDERFLOW:
case STATUS_FLOAT_MULTIPLE_FAULTS:
case STATUS_FLOAT_MULTIPLE_TRAPS:
#ifdef FACTOR_64
signal_fpu_status = fpu_status(MXCSR(c));
#else
signal_fpu_status = fpu_status(X87SW(c) | MXCSR(c));
// This seems to have no effect
X87SW(c) = 0;
#endif
MXCSR(c) &= 0xffffffc0;
dispatch_signal_handler((cell*)&c->ESP, (cell*)&c->EIP,
(cell)factor::fp_signal_handler_impl);
break;
default:
signal_number = e->ExceptionCode;
dispatch_signal_handler((cell*)&c->ESP, (cell*)&c->EIP,
(cell)factor::synchronous_signal_handler_impl);
break;
}
return ExceptionContinueExecution;
}
VM_C_API LONG exception_handler(PEXCEPTION_RECORD e, void* frame, PCONTEXT c,
void* dispatch) {
factor_vm* vm = current_vm_p();
if (factor_vm::fatal_erroring_p || !vm)
return ExceptionContinueSearch;
return vm->exception_handler(e, frame, c, dispatch);
}
// On Unix SIGINT (ctrl-c) automatically interrupts blocking io system
// calls. It doesn't on Windows, so we need to manually send some
// cancellation requests to unblock the thread.
VOID CALLBACK dummy_cb(ULONG_PTR dwParam) { (void)dwParam; }
// CancelSynchronousIo is not in Windows XP
#if _WIN32_WINNT >= 0x0600
static void wake_up_thread(HANDLE thread) {
if (!CancelSynchronousIo(thread)) {
DWORD err = GetLastError();
// CancelSynchronousIo() didn't find anything to cancel, let's try
// with QueueUserAPC() instead.
if (err == ERROR_NOT_FOUND) {
if (!QueueUserAPC(&dummy_cb, thread, NULL)) {
fatal_error("QueueUserAPC() failed", GetLastError());
}
} else {
fatal_error("CancelSynchronousIo() failed", err);
}
}
}
#else
static void wake_up_thread(HANDLE thread) { (void)thread; }
#endif
static BOOL WINAPI ctrl_handler(DWORD dwCtrlType) {
switch (dwCtrlType) {
case CTRL_C_EVENT: {
// The CtrlHandler runs in its own thread without stopping the main
// thread. Since in practice nobody uses the multi-VM stuff yet, we just
// grab the first VM we can get. This will not be a good idea when we
// actually support native threads.
FACTOR_ASSERT(thread_vms.size() == 1);
factor_vm* vm = thread_vms.begin()->second;
vm->enqueue_fep();
// Before leaving the ctrl_handler, try and wake up the main thread.
wake_up_thread(factor::boot_thread);
return TRUE;
}
default:
return FALSE;
}
}
void open_console() { handle_ctrl_c(); }
void ignore_ctrl_c() {
SetConsoleCtrlHandler(factor::ctrl_handler, FALSE);
}
void handle_ctrl_c() {
SetConsoleCtrlHandler(factor::ctrl_handler, TRUE);
}
const int ctrl_break_sleep = 10; /* msec */
static DWORD WINAPI ctrl_break_thread_proc(LPVOID parent_vm) {
bool ctrl_break_handled = false;
factor_vm* vm = static_cast<factor_vm*>(parent_vm);
while (vm->stop_on_ctrl_break) {
if (GetAsyncKeyState(VK_CANCEL) >= 0) { /* Ctrl-Break is released. */
ctrl_break_handled = false; /* Wait for the next press. */
} else if (!ctrl_break_handled) {
/* Check if the VM thread has the same Id as the thread Id of the
currently active window. Note that thread Id is not a handle. */
DWORD fg_thd_id = GetWindowThreadProcessId(GetForegroundWindow(), NULL);
if ((fg_thd_id == vm->thread_id) && !vm->fep_p) {
vm->enqueue_fep();
ctrl_break_handled = true;
}
}
Sleep(ctrl_break_sleep);
}
return 0;
}
void factor_vm::primitive_disable_ctrl_break() {
stop_on_ctrl_break = false;
if (ctrl_break_thread != NULL) {
DWORD wait_result = WaitForSingleObject(ctrl_break_thread,
2 * ctrl_break_sleep);
if (wait_result != WAIT_OBJECT_0)
TerminateThread(ctrl_break_thread, 0);
CloseHandle(ctrl_break_thread);
ctrl_break_thread = NULL;
}
}
void factor_vm::primitive_enable_ctrl_break() {
stop_on_ctrl_break = true;
if (ctrl_break_thread == NULL) {
DisableProcessWindowsGhosting();
ctrl_break_thread = CreateThread(NULL, 0, factor::ctrl_break_thread_proc,
static_cast<LPVOID>(this), 0, NULL);
SetThreadPriority(ctrl_break_thread, THREAD_PRIORITY_ABOVE_NORMAL);
}
}
void lock_console() {}
void unlock_console() {}
void close_console() {}
cell get_thread_pc(THREADHANDLE th) {
DWORD suscount = SuspendThread(th);
FACTOR_ASSERT(suscount == 0);
CONTEXT context;
memset((void*)&context, 0, sizeof(CONTEXT));
context.ContextFlags = CONTEXT_CONTROL;
BOOL context_ok = GetThreadContext(th, &context);
FACTOR_ASSERT(context_ok);
suscount = ResumeThread(th);
FACTOR_ASSERT(suscount == 1);
return context.EIP;
}
void factor_vm::sampler_thread_loop() {
LARGE_INTEGER counter, new_counter, units_per_second;
DWORD ok;
ok = QueryPerformanceFrequency(&units_per_second);
FACTOR_ASSERT(ok);
ok = QueryPerformanceCounter(&counter);
FACTOR_ASSERT(ok);
counter.QuadPart *= samples_per_second;
while (atomic::load(&sampling_profiler_p)) {
SwitchToThread();
ok = QueryPerformanceCounter(&new_counter);
FACTOR_ASSERT(ok);
new_counter.QuadPart *= samples_per_second;
cell samples = 0;
while (new_counter.QuadPart - counter.QuadPart >
units_per_second.QuadPart) {
++samples;
counter.QuadPart += units_per_second.QuadPart;
}
if (samples == 0)
continue;
cell pc = get_thread_pc(thread);
enqueue_samples(samples, pc, false);
}
}
static DWORD WINAPI sampler_thread_entry(LPVOID parent_vm) {
static_cast<factor_vm*>(parent_vm)->sampler_thread_loop();
return 0;
}
void factor_vm::start_sampling_profiler_timer() {
sampler_thread = CreateThread(NULL, 0, &sampler_thread_entry,
static_cast<LPVOID>(this), 0, NULL);
}
void factor_vm::end_sampling_profiler_timer() {
atomic::store(&sampling_profiler_p, false);
DWORD wait_result =
WaitForSingleObject(sampler_thread, 3000 * (DWORD) samples_per_second);
if (wait_result != WAIT_OBJECT_0)
TerminateThread(sampler_thread, 0);
CloseHandle(sampler_thread);
sampler_thread = NULL;
}
void abort() { ::abort(); }
}
Reference in New Issue View Git Blame Copy Permalink