|
@@ -3,8 +3,10 @@ use core::{marker::PhantomData, mem};
|
|
|
use aya_bpf_cty::c_void;
|
|
|
|
|
|
use crate::{
|
|
|
- bindings::{bpf_map_def, bpf_map_type::BPF_MAP_TYPE_SOCKHASH, bpf_sock_ops},
|
|
|
- helpers::bpf_sock_hash_update,
|
|
|
+ bindings::{bpf_map_def, bpf_map_type::BPF_MAP_TYPE_SOCKHASH, bpf_sock_ops, BPF_F_INGRESS},
|
|
|
+ helpers::{bpf_msg_redirect_hash, bpf_sock_hash_update},
|
|
|
+ programs::SkMsgContext,
|
|
|
+ BpfContext,
|
|
|
};
|
|
|
|
|
|
#[repr(transparent)]
|
|
@@ -45,4 +47,14 @@ impl<K> SockHash<K> {
|
|
|
Ok(())
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+ pub unsafe fn redirect(&mut self, ctx: &SkMsgContext, key: &mut K, flags: u64) -> i64 {
|
|
|
+ let ret = bpf_msg_redirect_hash(
|
|
|
+ ctx.as_ptr() as *mut _,
|
|
|
+ &mut self.def as *mut _ as *mut _,
|
|
|
+ key as *mut _ as *mut _,
|
|
|
+ flags,
|
|
|
+ );
|
|
|
+ ret
|
|
|
+ }
|
|
|
}
|