diff --git a/{{project-name}}/src/main.rs b/{{project-name}}/src/main.rs index 47a929c..c25a66d 100644 --- a/{{project-name}}/src/main.rs +++ b/{{project-name}}/src/main.rs @@ -19,7 +19,7 @@ use aya::programs::{CgroupSkb, CgroupSkbAttachType}; {%- when "tracepoint" -%} use aya::programs::TracePoint; {%- when "lsm" -%} -use aya::programs::Lsm; +use aya::{programs::Lsm, Btf}; {%- endcase %} use std::{ convert::{TryFrom,TryInto}, @@ -94,8 +94,9 @@ fn try_main() -> Result<(), anyhow::Error> { program.load()?; program.attach("{{tracepoint_category}}", "{{tracepoint_name}}")?; {%- when "lsm" -%} + let btf = Btf::from_sys_fs()?; let program: &mut Lsm = bpf.program_mut("{{lsm_hook}}")?.try_into()?; - program.load("{{lsm_hook}}")?; + program.load("{{lsm_hook}}", &btf)?; program.attach()?; {%- endcase %}