Add UnionStore

Author: lucic71

libcc2rs/src/lib.rs

@@ -7,6 +7,9 @@ pub use reinterpret::ByteRepr;
 mod rc;
 pub use rc::*;
 
+mod union;
+pub use union::*;
+
 mod fn_ptr;
 pub use fn_ptr::FnPtr;
 

libcc2rs/src/rc.rs

@@ -1246,6 +1246,15 @@ impl<T: ?Sized> AsPointerDyn<T> for Rc<RefCell<T>> {
 
 impl<T: 'static> ByteRepr for Ptr<T> {}
 
+impl<T> Ptr<T> {
+    pub(crate) fn reinterpreted(alloc: Rc<dyn OriginalAlloc>, byte_offset: usize) -> Self {
+        Ptr {
+            offset: byte_offset,
+            kind: PtrKind::Reinterpreted(alloc),
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

libcc2rs/src/union.rs

@@ -0,0 +1,34 @@
+// Copyright (c) 2022-present INESC-ID.
+// Distributed under the MIT license that can be found in the LICENSE file.
+
+use std::{cell::RefCell, rc::Rc};
+
+use crate::Ptr;
+use crate::reinterpret::{ByteRepr, OriginalAlloc, SliceOriginalAlloc};
+
+pub struct UnionStore {
+    bytes: Rc<RefCell<Vec<u8>>>,
+}
+
+impl UnionStore {
+    pub fn new(size: usize) -> Self {
+        UnionStore {
+            bytes: Rc::new(RefCell::new(vec![0u8; size])),
+        }
+    }
+
+    pub fn pod<T: ByteRepr>(&self, offset: usize) -> Ptr<T> {
+        let alloc: Rc<dyn OriginalAlloc> = Rc::new(SliceOriginalAlloc {
+            weak: Rc::downgrade(&self.bytes),
+        });
+        Ptr::reinterpreted(alloc, offset)
+    }
+}
+
+impl Clone for UnionStore {
+    fn clone(&self) -> Self {
+        UnionStore {
+            bytes: Rc::new(RefCell::new(self.bytes.borrow().clone())),
+        }
+    }
+}