Fix test errors

wenyuzhao · Mar 13, 2024 · d7a7401 · d7a7401
1 parent a9e2541
commit d7a7401
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/mallockit/src/malloc.rs b/mallockit/src/malloc.rs
@@ -54,7 +54,7 @@ impl<P: Plan> MallocAPI<P> {
     pub unsafe fn malloc_size(&self, ptr: Address) -> usize {
         let ptr = Address::from(ptr);
         #[cfg(target_os = "macos")]
-        if Self::is_in_mallockit_heap(ptr.into()) {
+        if !Self::is_in_mallockit_heap(ptr.into()) {
             return crate::util::macos_malloc_zone::external_memory_size(ptr);
         }
         P::get_layout(ptr).size()
@@ -85,7 +85,7 @@ impl<P: Plan> MallocAPI<P> {
             return;
         }
         #[cfg(target_os = "macos")]
-        if Self::is_in_mallockit_heap(ptr.into()) {
+        if !Self::is_in_mallockit_heap(ptr.into()) {
             return;
         }
         self.mutator().dealloc(ptr.into());
@@ -108,7 +108,7 @@ impl<P: Plan> MallocAPI<P> {
         let new_size = Self::align_up(new_size, Self::MIN_ALIGNMENT);
 
         #[cfg(target_os = "macos")]
-        if Self::is_in_mallockit_heap(ptr.into()) {
+        if !Self::is_in_mallockit_heap(ptr.into()) {
             let ptr = Address::from(ptr);
             let old_size = crate::util::macos_malloc_zone::external_memory_size(ptr);
             let new_layout =

diff --git a/mallockit/src/mutator.rs b/mallockit/src/mutator.rs
@@ -91,6 +91,10 @@ pub trait TLS: Sized {
     }
 }
 
+impl TLS for u8 {
+    const NEW: Self = 0;
+}
+
 #[cfg(target_os = "macos")]
 mod macos_tls {
     use spin::{mutex::Mutex, Yield};
@@ -111,9 +115,6 @@ mod macos_tls {
     #[cfg(test)]
     #[no_mangle]
     extern "C" fn mallockit_initialize_macos_tls() -> *mut u8 {
-        impl TLS for u8 {
-            const NEW: Self = 0;
-        }
         get_tls::<u8>()
     }