57 lines
2.1 KiB
C++
57 lines
2.1 KiB
C++
#include "wums_thread_specific.h"
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
extern "C" void OSFatal(const char *);
|
|
extern "C" void OSReport(const char *, ...);
|
|
|
|
typedef struct OSThread {
|
|
uint8_t other[0x68c];
|
|
void *reserved[5];
|
|
} OSThread;
|
|
|
|
static_assert(offsetof(OSThread, reserved) == 0x68c, "OSThread: \"reserved\" at wrong offset");
|
|
static_assert(sizeof(OSThread) == 0x6a0, "OSThread: wrong size");
|
|
|
|
extern "C" const char wums_meta_info_dump[];
|
|
|
|
void wums_set_thread_specific(__wut_thread_specific_id id, void *value) {
|
|
OSThread *thread;
|
|
asm volatile("lwz %0, -0x20(0)"
|
|
: "=r"(thread)); // OSGetCurrentThread()
|
|
if (thread != nullptr) {
|
|
if (id == WUT_THREAD_SPECIFIC_0) {
|
|
thread->reserved[3] = value;
|
|
} else if (id == WUT_THREAD_SPECIFIC_1) {
|
|
thread->reserved[4] = value;
|
|
} else {
|
|
OSReport("[%s] wums_set_thread_specific: invalid id\n", wums_meta_info_dump);
|
|
OSFatal("wut_set_thread_specific: invalid id");
|
|
}
|
|
} else {
|
|
OSReport("[%s] wums_set_thread_specific: invalid thread\n", wums_meta_info_dump);
|
|
OSFatal("wums_set_thread_specific: invalid thread");
|
|
}
|
|
}
|
|
|
|
void *wums_get_thread_specific(__wut_thread_specific_id id) {
|
|
OSThread *thread;
|
|
asm volatile("lwz %0, -0x20(0)"
|
|
: "=r"(thread)); // OSGetCurrentThread()
|
|
if (thread != nullptr) {
|
|
if (id == WUT_THREAD_SPECIFIC_0) {
|
|
return thread->reserved[3];
|
|
} else if (id == WUT_THREAD_SPECIFIC_1) {
|
|
return thread->reserved[4];
|
|
} else if ((uint32_t) id == 0x13371337) { // Mechanism to detect if the function was overridden properly
|
|
return (void *) 0x42424242;
|
|
} else {
|
|
OSReport("[%s] wums_get_thread_specific: invalid id\n", wums_meta_info_dump);
|
|
OSFatal("wums_get_thread_specific: invalid id");
|
|
}
|
|
} else {
|
|
OSReport("[%s] wums_get_thread_specific: invalid thread\n", wums_meta_info_dump);
|
|
OSFatal("wums_get_thread_specific: invalid thread\n");
|
|
}
|
|
return nullptr;
|
|
} |