Просмотр исходного кода

bpf: sk_skb: add ::cb() and ::cb_mut() to work with skb->cb

Alessandro Decina 3 лет назад
Родитель
Сommit
ef39e0ebd2
1 измененных файлов с 10 добавлено и 0 удалено
  1. 10 0
      bpf/aya-bpf/src/programs/sk_skb.rs

+ 10 - 0
bpf/aya-bpf/src/programs/sk_skb.rs

@@ -30,6 +30,16 @@ impl SkSkbContext {
         unsafe { *self.skb }.mark = mark;
     }
 
+    #[inline]
+    pub fn cb(&self) -> &[u32] {
+        unsafe { &(*self.skb).cb }
+    }
+
+    #[inline]
+    pub fn cb_mut(&self) -> &mut [u32] {
+        unsafe { &mut (*self.skb).cb }
+    }
+
     #[inline]
     pub fn load<T>(&self, offset: usize) -> Result<T, c_long> {
         unsafe {