From 7908e8593d4603b71fdb217582368d456f771b46 Mon Sep 17 00:00:00 2001 From: William Findlay Date: Fri, 29 Oct 2021 00:45:34 -0400 Subject: [PATCH] lsm: update program.load() to use new signature with &Btf --- {{project-name}}/src/main.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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 %}