Add Fibre local storage (Fls*) functions for armcc

decompals · Oct 16, 2023 · 00412a0 · 00412a0
1 parent 90101d8
commit 00412a0
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/dll/kernel32.cpp b/dll/kernel32.cpp
@@ -1992,6 +1992,63 @@ namespace kernel32 {
 		*Target = Value;
 		return initial;
 	}
+
+	// These are effectively a copy/paste of the Tls* functions
+	enum { MAX_FLS_VALUES = 100 };
+	static bool flsValuesUsed[MAX_FLS_VALUES] = { false };
+	static void *flsValues[MAX_FLS_VALUES];
+	int WIN_FUNC FlsAlloc(void *lpCallback) {
+		DEBUG_LOG("FlsAlloc (lpCallback: %x)\n", lpCallback);
+		// If the function succeeds, the return value is an FLS index initialized to zero.
+		for (size_t i = 0; i < MAX_FLS_VALUES; i++) {
+			if (flsValuesUsed[i] == false) {
+				flsValuesUsed[i] = true;
+				flsValues[i] = 0;
+				DEBUG_LOG("...returning %d\n", i);
+				return i;
+			}
+		}
+		DEBUG_LOG("...returning nothing\n");
+		wibo::lastError = 1;
+		return 0xFFFFFFFF; // FLS_OUT_OF_INDEXES
+	}
+
+	unsigned int WIN_FUNC FlsFree(unsigned int dwFlsIndex) {
+		DEBUG_LOG("FlsFree(%u)\n", dwFlsIndex);
+		if (dwFlsIndex >= 0 && dwFlsIndex < MAX_TLS_VALUES && flsValuesUsed[dwFlsIndex]) {
+			flsValuesUsed[dwFlsIndex] = false;
+			return 1;
+		} else {
+			wibo::lastError = 1;
+			return 0;
+		}
+	}
+
+	void *WIN_FUNC FlsGetValue(unsigned int dwFlsIndex) {
+		// DEBUG_LOG("FlsGetValue(%u)", dwTlsIndex);
+		void *result = nullptr;
+		if (dwFlsIndex >= 0 && dwFlsIndex < MAX_TLS_VALUES && flsValuesUsed[dwFlsIndex]) {
+			result = flsValues[dwFlsIndex];
+			// See https://learn.microsoft.com/en-us/windows/win32/api/fibersapi/nf-fibersapi-flsgetvalue
+			wibo::lastError = ERROR_SUCCESS;
+		} else {
+			wibo::lastError = 1;
+		}
+		// DEBUG_LOG(" -> %p\n", result);
+		return result;
+	}
+
+	unsigned int WIN_FUNC FlsSetValue(unsigned int dwFlsIndex, void *lpFlsData) {
+		// DEBUG_LOG("TlsSetValue(%u, %p)\n", dwTlsIndex, lpTlsValue);
+		if (dwFlsIndex >= 0 && dwFlsIndex < MAX_FLS_VALUES && flsValuesUsed[dwFlsIndex]) {
+			flsValues[dwFlsIndex] = lpFlsData;
+			return 1;
+		} else {
+			wibo::lastError = 1;
+			return 0;
+		}
+	}
+
 }
 
 static void *resolveByName(const char *name) {
@@ -2180,6 +2237,12 @@ static void *resolveByName(const char *name) {
 	if (strcmp(name, "InterlockedDecrement") == 0) return (void *) kernel32::InterlockedDecrement;
 	if (strcmp(name, "InterlockedExchange") == 0) return (void *) kernel32::InterlockedExchange;
 
+	// fibersapi.h
+	if (strcmp(name, "FlsAlloc") == 0) return (void *) kernel32::FlsAlloc;
+	if (strcmp(name, "FlsFree") == 0) return (void *) kernel32::FlsFree;
+	if (strcmp(name, "FlsSetValue") == 0) return (void *) kernel32::FlsSetValue;
+	if (strcmp(name, "FlsGetValue") == 0) return (void *) kernel32::FlsGetValue;
+
 	return 0;
 }