1 files changed, 24 insertions, 2 deletions
diff --git a/rustables/src/expr/ct.rs b/rustables/src/expr/ct.rs
index c0349ab..001aef8 100644
--- a/rustables/src/expr/ct.rs
+++ b/rustables/src/expr/ct.rs
@@ -1,4 +1,4 @@
-use super::{Expression, Rule};
+use super::{DeserializationError, Expression, Rule};
 use rustables_sys::{self as sys, libc};
 use std::os::raw::c_char;
 
@@ -27,9 +27,31 @@ impl Conntrack {
 }
 
 impl Expression for Conntrack {
+    fn get_raw_name() -> *const c_char {
+        b"ct\0" as *const _ as *const c_char
+    }
+
+    fn from_expr(expr: *const sys::nftnl_expr) -> Result<Self, DeserializationError>
+    where
+        Self: Sized,
+    {
+        unsafe {
+            let ct_key = sys::nftnl_expr_get_u32(expr, sys::NFTNL_EXPR_CT_KEY as u16);
+            let ct_sreg_is_set = sys::nftnl_expr_is_set(expr, sys::NFTNL_EXPR_CT_SREG as u16);
+
+            match ct_key as i32 {
+                libc::NFT_CT_STATE => Ok(Conntrack::State),
+                libc::NFT_CT_MARK => Ok(Conntrack::Mark {
+                    set: ct_sreg_is_set,
+                }),
+                _ => Err(DeserializationError::InvalidValue),
+            }
+        }
+    }
+
     fn to_expr(&self, _rule: &Rule) -> *mut sys::nftnl_expr {
         unsafe {
-            let expr = try_alloc!(sys::nftnl_expr_alloc(b"ct\0" as *const _ as *const c_char));
+            let expr = try_alloc!(sys::nftnl_expr_alloc(Self::get_raw_name()));
 
             if let Conntrack::Mark { set: true } = self {
                 sys::nftnl_expr_set_u32(