forked from openrs2/openrs2
parent
3be031e6b1
commit
bd005106a2
@ -1,12 +1,227 @@ |
|||||||
#include "jagex3_jagmisc_jagmisc.h" |
#include "jagex3_jagmisc_jagmisc.h" |
||||||
|
|
||||||
|
#include <stdbool.h> |
||||||
|
#include <stdint.h> |
||||||
|
#include <windows.h> |
||||||
|
#include <VersionHelpers.h> |
||||||
|
|
||||||
|
#define WAIT_OBJECT_1 (WAIT_OBJECT_0 + 1) |
||||||
|
|
||||||
|
#define JAGMISC_SPIN_COUNT 4000 |
||||||
|
#define JAGMISC_NANOS_PER_SEC 1000000000LL |
||||||
|
|
||||||
|
// TODO(gpe): do any of these need to be volatile?
|
||||||
|
static bool jagmisc_unpinned; |
||||||
|
static LARGE_INTEGER jagmisc_freq; |
||||||
|
static CRITICAL_SECTION jagmisc_lock; |
||||||
|
static HANDLE jagmisc_start_event; |
||||||
|
static HANDLE jagmisc_time_request_event; |
||||||
|
static HANDLE jagmisc_time_ready_event; |
||||||
|
static HANDLE jagmisc_stop_event; |
||||||
|
static HANDLE jagmisc_thread; |
||||||
|
static int64_t jagmisc_time; |
||||||
|
|
||||||
|
static inline int64_t jagmisc_to_nanos(LARGE_INTEGER count) { |
||||||
|
return count.QuadPart * JAGMISC_NANOS_PER_SEC / jagmisc_freq.QuadPart; |
||||||
|
} |
||||||
|
|
||||||
|
static DWORD WINAPI jagmisc_run(LPVOID parameter) { |
||||||
|
/* pin this thread to the system's first processor */ |
||||||
|
HANDLE thread = GetCurrentThread(); |
||||||
|
|
||||||
|
DWORD_PTR affinity_mask = 0x1; |
||||||
|
bool success = SetThreadAffinityMask(thread, affinity_mask); |
||||||
|
|
||||||
|
CloseHandle(thread); |
||||||
|
|
||||||
|
if (!success) { |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
/* notify Java_jagex3_jagmisc_jagmisc_init() that the thread has started successfully */ |
||||||
|
if (!SetEvent(jagmisc_start_event)) { |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
/* serve time request and stop events */ |
||||||
|
const HANDLE handles[] = { jagmisc_time_request_event, jagmisc_stop_event }; |
||||||
|
for (;;) { |
||||||
|
switch (WaitForMultipleObjects(2, handles, FALSE, INFINITE)) { |
||||||
|
case WAIT_OBJECT_0: |
||||||
|
LARGE_INTEGER count; |
||||||
|
if (QueryPerformanceCounter(&count)) { |
||||||
|
jagmisc_time = jagmisc_to_nanos(count); |
||||||
|
} else { |
||||||
|
jagmisc_time = 0; |
||||||
|
} |
||||||
|
|
||||||
|
if (!SetEvent(jagmisc_time_ready_event)) { |
||||||
|
return 1; |
||||||
|
} |
||||||
|
break; |
||||||
|
case WAIT_OBJECT_1: |
||||||
|
return 0; |
||||||
|
default: |
||||||
|
return 1; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static void jagmisc_stop(void) { |
||||||
|
// TODO(gpe): think about what to do if we fail to stop the thread. We
|
||||||
|
// could call TerminateThread (which sounds dangerous based on the
|
||||||
|
// documentation) or throw a Java exception. The current behaviour of
|
||||||
|
// ignoring the failure is okay as long as jagmisc.init() is never called
|
||||||
|
// again in the same JVM.
|
||||||
|
if (SetEvent(jagmisc_stop_event)) { |
||||||
|
WaitForSingleObject(jagmisc_thread, INFINITE); |
||||||
|
} |
||||||
|
|
||||||
|
CloseHandle(jagmisc_thread); |
||||||
|
jagmisc_thread = NULL; |
||||||
|
|
||||||
|
CloseHandle(jagmisc_stop_event); |
||||||
|
jagmisc_stop_event = NULL; |
||||||
|
|
||||||
|
CloseHandle(jagmisc_time_ready_event); |
||||||
|
jagmisc_time_ready_event = NULL; |
||||||
|
|
||||||
|
CloseHandle(jagmisc_time_request_event); |
||||||
|
jagmisc_time_request_event = NULL; |
||||||
|
|
||||||
|
CloseHandle(jagmisc_start_event); |
||||||
|
jagmisc_start_event = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) { |
||||||
|
if (fdwReason == DLL_PROCESS_ATTACH) { |
||||||
|
/*
|
||||||
|
* We only need to pin the timer thread to a single processor prior to |
||||||
|
* Windows Vista. |
||||||
|
*/ |
||||||
|
jagmisc_unpinned = IsWindowsVistaOrGreater(); |
||||||
|
if (jagmisc_unpinned) { |
||||||
|
return TRUE; |
||||||
|
} |
||||||
|
|
||||||
|
if (!InitializeCriticalSectionAndSpinCount(&jagmisc_lock, JAGMISC_SPIN_COUNT)) { |
||||||
|
return FALSE; |
||||||
|
} |
||||||
|
} |
||||||
|
return TRUE; |
||||||
|
} |
||||||
|
|
||||||
JNIEXPORT jboolean JNICALL Java_jagex3_jagmisc_jagmisc_init(JNIEnv *env, jclass cls) { |
JNIEXPORT jboolean JNICALL Java_jagex3_jagmisc_jagmisc_init(JNIEnv *env, jclass cls) { |
||||||
return JNI_FALSE; |
if (jagmisc_unpinned) { |
||||||
|
return (jboolean) QueryPerformanceFrequency(&jagmisc_freq); |
||||||
|
} |
||||||
|
|
||||||
|
EnterCriticalSection(&jagmisc_lock); |
||||||
|
jboolean success = JNI_FALSE; |
||||||
|
|
||||||
|
if (jagmisc_thread) { |
||||||
|
jagmisc_stop(); |
||||||
|
} |
||||||
|
|
||||||
|
if (!QueryPerformanceFrequency(&jagmisc_freq)) { |
||||||
|
goto unlock; |
||||||
|
} |
||||||
|
|
||||||
|
jagmisc_start_event = CreateEventA(NULL, FALSE, FALSE, NULL); |
||||||
|
if (!jagmisc_start_event) { |
||||||
|
goto unlock; |
||||||
|
} |
||||||
|
|
||||||
|
jagmisc_time_request_event = CreateEventA(NULL, FALSE, FALSE, NULL); |
||||||
|
if (!jagmisc_time_request_event) { |
||||||
|
goto create_time_request_event_failed; |
||||||
|
} |
||||||
|
|
||||||
|
jagmisc_time_ready_event = CreateEventA(NULL, FALSE, FALSE, NULL); |
||||||
|
if (!jagmisc_time_ready_event) { |
||||||
|
goto create_time_ready_event_failed; |
||||||
|
} |
||||||
|
|
||||||
|
jagmisc_stop_event = CreateEventA(NULL, FALSE, FALSE, NULL); |
||||||
|
if (!jagmisc_stop_event) { |
||||||
|
goto create_stop_event_failed; |
||||||
|
} |
||||||
|
|
||||||
|
jagmisc_thread = CreateThread(NULL, 0, &jagmisc_run, NULL, 0, NULL); |
||||||
|
if (!jagmisc_thread) { |
||||||
|
goto create_thread_failed; |
||||||
|
} |
||||||
|
|
||||||
|
const HANDLE handles[] = { jagmisc_start_event, jagmisc_thread }; |
||||||
|
DWORD result = WaitForMultipleObjects(2, handles, FALSE, INFINITE); |
||||||
|
if (result != WAIT_OBJECT_0) { |
||||||
|
goto start_failed; |
||||||
|
} |
||||||
|
|
||||||
|
success = JNI_TRUE; |
||||||
|
goto unlock; |
||||||
|
|
||||||
|
start_failed: |
||||||
|
CloseHandle(jagmisc_thread); |
||||||
|
jagmisc_thread = NULL; |
||||||
|
create_thread_failed: |
||||||
|
CloseHandle(jagmisc_stop_event); |
||||||
|
jagmisc_stop_event = NULL; |
||||||
|
create_stop_event_failed: |
||||||
|
CloseHandle(jagmisc_time_ready_event); |
||||||
|
jagmisc_time_ready_event = NULL; |
||||||
|
create_time_ready_event_failed: |
||||||
|
CloseHandle(jagmisc_time_request_event); |
||||||
|
jagmisc_time_request_event = NULL; |
||||||
|
create_time_request_event_failed: |
||||||
|
CloseHandle(jagmisc_start_event); |
||||||
|
jagmisc_start_event = NULL; |
||||||
|
unlock: |
||||||
|
LeaveCriticalSection(&jagmisc_lock); |
||||||
|
return success; |
||||||
} |
} |
||||||
|
|
||||||
JNIEXPORT void JNICALL Java_jagex3_jagmisc_jagmisc_Quit0(JNIEnv *env, jclass cls) { |
JNIEXPORT void JNICALL Java_jagex3_jagmisc_jagmisc_Quit0(JNIEnv *env, jclass cls) { |
||||||
|
if (jagmisc_unpinned) { |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
EnterCriticalSection(&jagmisc_lock); |
||||||
|
if (jagmisc_thread) { |
||||||
|
jagmisc_stop(); |
||||||
|
} |
||||||
|
LeaveCriticalSection(&jagmisc_lock); |
||||||
} |
} |
||||||
|
|
||||||
JNIEXPORT jlong JNICALL Java_jagex3_jagmisc_jagmisc_nanoTime(JNIEnv *env, jclass cls) { |
JNIEXPORT jlong JNICALL Java_jagex3_jagmisc_jagmisc_nanoTime(JNIEnv *env, jclass cls) { |
||||||
|
if (jagmisc_unpinned) { |
||||||
|
LARGE_INTEGER count; |
||||||
|
if (!QueryPerformanceCounter(&count)) { |
||||||
return 0; |
return 0; |
||||||
} |
} |
||||||
|
return jagmisc_to_nanos(count); |
||||||
|
} |
||||||
|
|
||||||
|
EnterCriticalSection(&jagmisc_lock); |
||||||
|
jlong time = 0; |
||||||
|
|
||||||
|
if (!jagmisc_thread) { |
||||||
|
goto unlock; |
||||||
|
} |
||||||
|
|
||||||
|
if (!SetEvent(jagmisc_time_request_event)) { |
||||||
|
goto unlock; |
||||||
|
} |
||||||
|
|
||||||
|
const HANDLE handles[] = { jagmisc_time_ready_event, jagmisc_thread }; |
||||||
|
DWORD result = WaitForMultipleObjects(2, handles, FALSE, INFINITE); |
||||||
|
if (result != WAIT_OBJECT_0) { |
||||||
|
goto unlock; |
||||||
|
} |
||||||
|
|
||||||
|
time = jagmisc_time; |
||||||
|
|
||||||
|
unlock: |
||||||
|
LeaveCriticalSection(&jagmisc_lock); |
||||||
|
return time; |
||||||
|
} |
||||||
|
Loading…
Reference in new issue