Merge pull request #90 from aya-rs/fix-tests

Fix sk_msg tests
pull/92/head
Alessandro Decina 1 year ago committed by GitHub
commit 69ea57ac54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -8,7 +8,7 @@ default = []
user = ["aya"]
[dependencies]
aya = { version = ">=0.11", optional = true }
aya = { git = "https://github.com/aya-rs/aya", optional = true }
[lib]
path = "src/lib.rs"

@ -10,7 +10,7 @@ use aya::programs::UProbe;
{%- when "sock_ops" -%}
use aya::programs::SockOps;
{%- when "sk_msg" -%}
use aya::maps::{MapRefMut,SockHash};
use aya::maps::SockHash;
use aya::programs::SkMsg;
use {{crate_name}}_common::SockKey;
{%- when "xdp" -%}
@ -112,10 +112,12 @@ async fn main() -> Result<(), anyhow::Error> {
program.load()?;
program.attach(cgroup)?;
{%- when "sk_msg" -%}
let sock_map = SockHash::<MapRefMut, SockKey>::try_from(bpf.map_mut("{{sock_map}}")?)?;
let sock_map: SockHash::<_, SockKey> = bpf.map("{{sock_map}}").unwrap().try_into()?;
let map_fd = sock_map.fd()?;
let prog: &mut SkMsg = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;
prog.load()?;
prog.attach(&sock_map)?;
prog.attach(map_fd)?;
// insert sockets to the map using sock_map.insert here, or from a sock_ops program
{%- when "xdp" -%}
let program: &mut Xdp = bpf.program_mut("{{crate_name}}").unwrap().try_into()?;

Loading…
Cancel
Save