Initial commit

Minecraft.Client/PSVita/PSVitaExtras/user_malloc_for_tls.c

@@ -0,0 +1,103 @@
+/* SCE CONFIDENTIAL
+ PlayStation(R)Vita Programmer Tool Runtime Library Release 03.000.061
+ * Copyright (C) 2012 Sony Computer Entertainment Inc.
+ * All Rights Reserved.
+ */
+
+#include <stdlib.h>
+#include <mspace.h>
+#include <kernel.h>
+//#include <libdbg.h>
+
+#define HEAP_SIZE (1024 * 1024)
+#define HEAP_ERROR1 1
+#define HEAP_ERROR2 2
+#define HEAP_ERROR3 3
+
+static SceUID s_heapUid;
+static mspace s_mspace;
+
+void user_malloc_for_tls_init(void);
+void user_malloc_for_tls_finalize(void);
+void *user_malloc_for_tls(size_t size);
+void user_free_for_tls(void *ptr);
+
+/**E Replace _malloc_for_tls_init function. */
+/**J _malloc_for_tls_init 関数と置き換わる */
+void user_malloc_for_tls_init(void)
+{
+	int res;
+	void *base = NULL;
+
+	/**E Allocate a memory block from the kernel */
+	/**J カーネルからメモリブロックを確保する */
+	s_heapUid = sceKernelAllocMemBlock("UserAllocatorForTLS", SCE_KERNEL_MEMBLOCK_TYPE_USER_RWDATA, HEAP_SIZE, SCE_NULL);
+	if (s_heapUid < SCE_OK) {
+		/**E Error handling */
+		/**J エラー処理 */
+		sceLibcSetHeapInitError(HEAP_ERROR1);
+	} else {
+		/**E Obtain the address of the allocated memory block */
+		/**J 確保したメモリブロックのアドレスを取得する */
+		res = sceKernelGetMemBlockBase(s_heapUid, &base);
+		if (res < SCE_OK) {
+			/**E Error handling */
+			/**J エラー処理 */
+			sceLibcSetHeapInitError(HEAP_ERROR2);
+		} else {
+			/**E Generate mspace */
+			/**J mspace を生成する */
+			s_mspace = mspace_create(base, HEAP_SIZE);
+			if (s_mspace == NULL) {
+				/**E Error handling */
+				/**J エラー処理 */
+				sceLibcSetHeapInitError(HEAP_ERROR3);
+			}
+		}
+	}
+}
+
+/**E Replace _malloc_for_tls_finalize function. */
+/**J _malloc_for_tls_finalize 関数と置き換わる */
+void user_malloc_for_tls_finalize(void)
+{
+	int res;
+
+	if (s_mspace != NULL) {
+		/**E Free mspace */
+		/**J mspace を解放する */
+		res = mspace_destroy(s_mspace);
+		if (res != 0) {
+			/**E Error handling */
+			/**J エラー処理 */
+			__breakpoint(0);
+		}
+	}
+
+	if (SCE_OK <= s_heapUid) {
+		/**E Free the memory block */
+		/**J メモリブロックを解放する */
+		res = sceKernelFreeMemBlock(s_heapUid);
+		if (res < SCE_OK) {
+			/**E Error handling */
+			/**J エラー処理 */
+			__breakpoint(0);
+		}
+	}
+}
+
+/**E Replace _malloc_for_tls function. */
+/**J _malloc_for_tls 関数と置き換わる */
+void *user_malloc_for_tls(size_t size)
+{
+//	SCE_DBG_LOG_TRACE("Called malloc_for_tls(%u)", size);
+	return mspace_malloc(s_mspace, size);
+}
+
+/**E Replace _free_for_tls function. */
+/**J _free_for_tls 関数と置き換わる */
+void user_free_for_tls(void *ptr)
+{
+//	SCE_DBG_LOG_TRACE("Called free_for_tls(%p)", ptr);
+	mspace_free(s_mspace, ptr);
+}