diff options
Diffstat (limited to 'tests/ui/allocator/auxiliary')
-rw-r--r-- | tests/ui/allocator/auxiliary/custom-as-global.rs | 16 | ||||
-rw-r--r-- | tests/ui/allocator/auxiliary/custom.rs | 21 | ||||
-rw-r--r-- | tests/ui/allocator/auxiliary/helper.rs | 11 | ||||
-rw-r--r-- | tests/ui/allocator/auxiliary/system-allocator.rs | 8 | ||||
-rw-r--r-- | tests/ui/allocator/auxiliary/system-allocator2.rs | 8 |
5 files changed, 64 insertions, 0 deletions
diff --git a/tests/ui/allocator/auxiliary/custom-as-global.rs b/tests/ui/allocator/auxiliary/custom-as-global.rs new file mode 100644 index 000000000..a5e96e775 --- /dev/null +++ b/tests/ui/allocator/auxiliary/custom-as-global.rs @@ -0,0 +1,16 @@ +// no-prefer-dynamic + +#![crate_type = "rlib"] + +extern crate custom; + +use std::sync::atomic::{AtomicUsize, Ordering}; + +use custom::A; + +#[global_allocator] +static ALLOCATOR: A = A(AtomicUsize::new(0)); + +pub fn get() -> usize { + ALLOCATOR.0.load(Ordering::SeqCst) +} diff --git a/tests/ui/allocator/auxiliary/custom.rs b/tests/ui/allocator/auxiliary/custom.rs new file mode 100644 index 000000000..b0ec9ab09 --- /dev/null +++ b/tests/ui/allocator/auxiliary/custom.rs @@ -0,0 +1,21 @@ +// no-prefer-dynamic + +#![feature(allocator_api)] +#![crate_type = "rlib"] + +use std::alloc::{GlobalAlloc, System, Layout}; +use std::sync::atomic::{AtomicUsize, Ordering}; + +pub struct A(pub AtomicUsize); + +unsafe impl GlobalAlloc for A { + unsafe fn alloc(&self, layout: Layout) -> *mut u8 { + self.0.fetch_add(1, Ordering::SeqCst); + System.alloc(layout) + } + + unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) { + self.0.fetch_add(1, Ordering::SeqCst); + System.dealloc(ptr, layout) + } +} diff --git a/tests/ui/allocator/auxiliary/helper.rs b/tests/ui/allocator/auxiliary/helper.rs new file mode 100644 index 000000000..008fb3501 --- /dev/null +++ b/tests/ui/allocator/auxiliary/helper.rs @@ -0,0 +1,11 @@ +// no-prefer-dynamic + +#![crate_type = "rlib"] +#![no_std] + +extern crate alloc; +use alloc::fmt; + +pub fn work_with(p: &fmt::Debug) { + drop(p); +} diff --git a/tests/ui/allocator/auxiliary/system-allocator.rs b/tests/ui/allocator/auxiliary/system-allocator.rs new file mode 100644 index 000000000..97b86bbc9 --- /dev/null +++ b/tests/ui/allocator/auxiliary/system-allocator.rs @@ -0,0 +1,8 @@ +// no-prefer-dynamic + +#![crate_type = "rlib"] + +use std::alloc::System; + +#[global_allocator] +static A: System = System; diff --git a/tests/ui/allocator/auxiliary/system-allocator2.rs b/tests/ui/allocator/auxiliary/system-allocator2.rs new file mode 100644 index 000000000..97b86bbc9 --- /dev/null +++ b/tests/ui/allocator/auxiliary/system-allocator2.rs @@ -0,0 +1,8 @@ +// no-prefer-dynamic + +#![crate_type = "rlib"] + +use std::alloc::System; + +#[global_allocator] +static A: System = System; |