1 files changed, 14 insertions, 3 deletions

diff --git a/libc/mem.c b/libc/mem.c
index 0c0ad4d..17703d5 100644
--- a/libc/mem.c
+++ b/libc/mem.c
@@ -347,13 +347,24 @@ void _free(void *ptr)
 
 void *malloc_debug(u32 size, const char *file, int line, const char *func, const char *inp)
 {
-	printf("MALLOC:\t%s:%d: %s: %dB (%s)\n", file, line, func, size, inp);
-	return _malloc(size);
+	void *ret = _malloc(size);
+#ifdef kernel
+	printf("K");
+#else
+	printf("U");
+#endif
+	printf("MALLOC\t%s:%d: %s: 0x%x %dB (%s)\n", file, line, func, ret, size, inp);
+	return ret;
 }
 
 void free_debug(void *ptr, const char *file, int line, const char *func, const char *inp)
 {
-	printf("FREE:\t%s:%d: %s: 0x%x (%s)\n", file, line, func, ptr, inp);
+#ifdef kernel
+	printf("K");
+#else
+	printf("U");
+#endif
+	printf("FREE\t%s:%d: %s: 0x%x (%s)\n", file, line, func, ptr, inp);
 	if (ptr)
 		_free(ptr);
 }