From 556463a85ff7cf921049cb0679a787389a87d7ec Mon Sep 17 00:00:00 2001
From: Alessandro Decina <alessandro.d@gmail.com>
Date: Wed, 25 Jan 2023 22:25:00 +0000
Subject: [PATCH] ebpf: SkbBuff: add some accessors

---
 bpf/aya-bpf/src/programs/sk_buff.rs | 40 +++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/bpf/aya-bpf/src/programs/sk_buff.rs b/bpf/aya-bpf/src/programs/sk_buff.rs
index 3ae6cf3d..669c2c95 100644
--- a/bpf/aya-bpf/src/programs/sk_buff.rs
+++ b/bpf/aya-bpf/src/programs/sk_buff.rs
@@ -217,6 +217,46 @@ impl SkBuff {
     pub(crate) fn as_ptr(&self) -> *mut c_void {
         self.skb as *mut _
     }
+
+    #[inline]
+    pub fn protocol(&self) -> u32 {
+        unsafe { (*self.skb).protocol }
+    }
+
+    #[inline]
+    pub fn family(&self) -> u32 {
+        unsafe { (*self.skb).family }
+    }
+
+    #[inline]
+    pub fn local_ipv4(&self) -> u32 {
+        unsafe { (*self.skb).local_ip4 }
+    }
+
+    #[inline]
+    pub fn local_ipv6(&self) -> &[u32; 4] {
+        unsafe { &(*self.skb).local_ip6 }
+    }
+
+    #[inline]
+    pub fn remote_ipv4(&self) -> u32 {
+        unsafe { (*self.skb).remote_ip4 }
+    }
+
+    #[inline]
+    pub fn remote_ipv6(&self) -> &[u32; 4] {
+        unsafe { &(*self.skb).remote_ip6 }
+    }
+
+    #[inline]
+    pub fn local_port(&self) -> u32 {
+        unsafe { (*self.skb).local_port }
+    }
+
+    #[inline]
+    pub fn remote_port(&self) -> u32 {
+        unsafe { (*self.skb).remote_port }
+    }
 }
 
 pub struct SkBuffContext {