Loading...
src/quarantine_malloc.h /dev/null libmalloc-374.100.5
--- /dev/null
+++ libmalloc/libmalloc-374.100.5/src/quarantine_malloc.h
@@ -0,0 +1,93 @@
+/*
+ * Copyright (c) 2021 Apple Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ *
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+#ifndef _QUARANTINE_MALLOC_H_
+#define _QUARANTINE_MALLOC_H_
+
+#include "base.h"
+#include "malloc/malloc.h"
+#include <stdbool.h>
+
+MALLOC_NOEXPORT
+bool
+quarantine_should_enable(void);
+
+MALLOC_NOEXPORT
+void
+quarantine_reset_environment(void);
+
+MALLOC_NOEXPORT
+malloc_zone_t *
+quarantine_create_zone(malloc_zone_t *wrapped_zone);
+
+static inline uint16_t
+_malloc_read_uint16_via_rsp(void *ptr)
+{
+#if TARGET_CPU_X86_64
+    __asm__ (
+        "subq  %%rsp,       %0  \n"
+        "movw (%%rsp, %0),  %w0 \n"
+        : "+r" (ptr)            // outputs, ptr = %0 read-write
+        :                       // inputs, empty
+        :                       // clobbers, empty
+    );
+    return (uint16_t)(uintptr_t)ptr;
+#else
+    return *(uint16_t *)ptr;
+#endif
+}
+
+static inline uint64_t
+_malloc_read_uint64_via_rsp(void *ptr)
+{
+#if TARGET_CPU_X86_64
+    __asm__ (
+        "subq  %%rsp,       %0  \n"
+        "movq (%%rsp, %0),  %0  \n"
+        : "+r" (ptr)            // outputs, ptr = %0 read-write
+        :                       // inputs, empty
+        :                       // clobbers, empty
+    );
+    return (uint64_t)ptr;
+#else
+    return *(uint64_t *)ptr;
+#endif
+}
+
+static inline void
+_malloc_write_uint64_via_rsp(void *ptr, uint64_t value)
+{
+#if TARGET_CPU_X86_64
+    __asm__ (
+        "subq  %%rsp,  %0         \n"
+        "movq  %1,    (%%rsp, %0) \n"
+        :                         // outputs, empty
+        : "r" (ptr), "r" (value)  // inputs, ptr = %0, value = %1
+        :                         // clobbers, empty
+    );
+#else
+    *(uint64_t *)ptr = value;
+#endif
+}
+
+#endif // _QUARANTINE_MALLOC_H_