mika/libnx
1
0
Fork 0
mirror of https://github.com/switchbrew/libnx.git synced 2025-06-21 12:32:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

thread: Allow user override of automatic stack memory allocation

Browse Source
This commit is contained in:
fincs 2021-01-17 17:48:08 +01:00
parent 109bd683df
commit d53150f855
No known key found for this signature in database
GPG Key ID: 62C7609ADA219C60
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
nx/source/kernel/thread.c
View File

@ -90,6 +90,14 @@ void __libnx_init_thread(void) {
getThreadVars()->thread_ptr = &g_mainThread; getThreadVars()->thread_ptr = &g_mainThread;
} }
void* __attribute__((weak)) __libnx_thread_alloc(size_t size) {
return memalign(0x1000, size);
}
void __attribute__((weak)) __libnx_thread_free(void* mem) {
free(mem);
}
Result threadCreate( Result threadCreate(
Thread* t, ThreadFunc entry, void* arg, void* stack_mem, size_t stack_sz, Thread* t, ThreadFunc entry, void* arg, void* stack_mem, size_t stack_sz,
int prio, int cpuid) int prio, int cpuid)
@ -101,7 +109,7 @@ Result threadCreate(
bool owns_stack_mem; bool owns_stack_mem;
if (stack_mem == NULL) { if (stack_mem == NULL) {
// Allocate new memory, stack then reent then tls. // Allocate new memory, stack then reent then tls.
stack_mem = memalign(0x1000, ((stack_sz + reent_sz + tls_sz) + 0xFFF) & ~0xFFF); stack_mem = __libnx_thread_alloc(((stack_sz + reent_sz + tls_sz) + 0xFFF) & ~0xFFF);
owns_stack_mem = true; owns_stack_mem = true;
} else { } else {
@ -182,7 +190,7 @@ Result threadCreate(
if (R_FAILED(rc)) { if (R_FAILED(rc)) {
if (owns_stack_mem) { if (owns_stack_mem) {
free(stack_mem); __libnx_thread_free(stack_mem);
} }
} }
@ -240,7 +248,7 @@ Result threadClose(Thread* t) {
if (R_SUCCEEDED(rc)) { if (R_SUCCEEDED(rc)) {
if (t->owns_stack_mem) { if (t->owns_stack_mem) {
free(t->stack_mem); __libnx_thread_free(t->stack_mem);
} }
svcCloseHandle(t->handle); svcCloseHandle(t->handle);
} }