1 files changed, 21 insertions, 17 deletions

diff --git a/src/parser_impls.rs b/src/parser_impls.rs
index b2681bb..887a9a2 100644
--- a/src/parser_impls.rs
+++ b/src/parser_impls.rs
@@ -1,4 +1,7 @@
-use std::{fmt::Debug, mem::transmute};
+use std::{
+    fmt::Debug,
+    mem::{size_of, transmute},
+};
 
 use rustables_macros::nfnetlink_struct;
 
@@ -15,8 +18,8 @@ use crate::{
 };
 
 impl NfNetlinkAttribute for u8 {
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        *addr = *self;
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0] = *self;
     }
 }
 
@@ -27,8 +30,8 @@ impl NfNetlinkDeserializable for u8 {
 }
 
 impl NfNetlinkAttribute for u16 {
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        *(addr as *mut Self) = self.to_be();
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..size_of::<Self>()].copy_from_slice(&self.to_be_bytes());
     }
 }
 
@@ -39,8 +42,8 @@ impl NfNetlinkDeserializable for u16 {
 }
 
 impl NfNetlinkAttribute for i32 {
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        *(addr as *mut Self) = self.to_be();
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..size_of::<Self>()].copy_from_slice(&self.to_be_bytes());
     }
 }
 
@@ -54,8 +57,8 @@ impl NfNetlinkDeserializable for i32 {
 }
 
 impl NfNetlinkAttribute for u32 {
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        *(addr as *mut Self) = self.to_be();
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..size_of::<Self>()].copy_from_slice(&self.to_be_bytes());
     }
 }
 
@@ -69,8 +72,8 @@ impl NfNetlinkDeserializable for u32 {
 }
 
 impl NfNetlinkAttribute for u64 {
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        *(addr as *mut Self) = self.to_be();
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..size_of::<Self>()].copy_from_slice(&self.to_be_bytes());
     }
 }
 
@@ -90,8 +93,8 @@ impl NfNetlinkAttribute for String {
         self.len()
     }
 
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        std::ptr::copy_nonoverlapping(self.as_bytes().as_ptr(), addr, self.len());
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..self.len()].copy_from_slice(&self.as_bytes());
     }
 }
 
@@ -110,8 +113,8 @@ impl NfNetlinkAttribute for Vec<u8> {
         self.len()
     }
 
-    unsafe fn write_payload(&self, addr: *mut u8) {
-        std::ptr::copy_nonoverlapping(self.as_ptr(), addr, self.len());
+    fn write_payload(&self, addr: &mut [u8]) {
+        addr[0..self.len()].copy_from_slice(&self.as_slice());
     }
 }
 
@@ -170,10 +173,11 @@ where
         })
     }
 
-    unsafe fn write_payload(&self, mut addr: *mut u8) {
+    fn write_payload(&self, mut addr: &mut [u8]) {
         for item in &self.objs {
             write_attribute(NFTA_LIST_ELEM, item, addr);
-            addr = addr.offset((pad_netlink_object::<nlattr>() + item.get_size()) as isize);
+            let offset = pad_netlink_object::<nlattr>() + item.get_size();
+            addr = &mut addr[offset..];
         }
     }
 }