commit 9b6fe632bbbf233336cb4dd4c35ace4e71fef6fe Author: numzero Date: Fri Aug 23 01:04:15 2024 +0300 It... works? diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..2c2f614 --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,7 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "stack-dyn" +version = "0.1.0" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..490ae99 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,7 @@ +[package] +name = "stack-dyn" +version = "0.1.0" +edition = "2021" +# rust-version = "nightly" + +[dependencies] diff --git a/rust-toolchain.toml b/rust-toolchain.toml new file mode 100644 index 0000000..5d56faf --- /dev/null +++ b/rust-toolchain.toml @@ -0,0 +1,2 @@ +[toolchain] +channel = "nightly" diff --git a/rustfmt.toml b/rustfmt.toml new file mode 100644 index 0000000..218e203 --- /dev/null +++ b/rustfmt.toml @@ -0,0 +1 @@ +hard_tabs = true diff --git a/src/lib.rs b/src/lib.rs new file mode 100644 index 0000000..d046d3d --- /dev/null +++ b/src/lib.rs @@ -0,0 +1,87 @@ +#![feature( + ptr_metadata, + ptr_as_ref_unchecked, + generic_const_exprs, + inherent_associated_types +)] + +use std::{ + marker::PhantomData, + mem::MaybeUninit, + ops, + ptr::{self, addr_of_mut}, +}; + +#[repr(C, align(16))] +struct Data([u8; 48]); + +pub struct Dyn<'a, Trait: ?Sized + 'a> { + _phantom1: PhantomData, + _phantom2: PhantomData<&'a ()>, + meta: ::Metadata, + data: Data, +} + +impl<'a, Trait: ?Sized + 'a> Dyn<'a, Trait> { + pub type Target = Trait; + + fn as_ptr(&self) -> *const Self::Target { + ptr::from_raw_parts::(&self.data, self.meta) + } +} + +impl<'a, Trait: ?Sized + 'a> ops::Deref for Dyn<'a, Trait> { + type Target = Trait; + + fn deref(&self) -> &Self::Target { + unsafe { self.as_ptr().as_ref_unchecked() } + } +} + +impl<'a, Trait: ?Sized + 'a> ops::DerefMut for Dyn<'a, Trait> { + fn deref_mut(&mut self) -> &mut Self::Target { + unsafe { self.as_ptr().cast_mut().as_mut_unchecked() } + } +} + +impl<'a, Iter: Iterator + 'a> From for Dyn<'a, dyn Iterator + 'a> +where + [(); std::mem::size_of::() - std::mem::size_of::()]:, + [(); std::mem::align_of::() - std::mem::align_of::()]:, +{ + fn from(value: Iter) -> Self { + let (_, meta) = (&value as &Self::Target as *const Self::Target).to_raw_parts(); + unsafe { + let mut ret = MaybeUninit::::zeroed(); + let pret = ret.as_mut_ptr(); + (addr_of_mut!((*pret).data) as *mut Iter).write(value); + addr_of_mut!((*pret).meta).write(meta); + ret.assume_init() + } + } +} + +#[cfg(test)] +mod tests { + use super::*; + + struct Small([u8; 48]); + + impl Iterator for Small { + type Item = u8; + fn next(&mut self) -> Option { + let ret = self.0.get(self.0[0] as usize).copied(); + self.0[0] += 1; + ret + } + } + + #[test] + fn it_works() { + let mut it = Small([42; 48]); + assert_eq!(it.next(), Some(42)); + + let mut it: Dyn> = Small([42; 48]).into(); + assert_eq!(it.next(), Some(42)); + } +}