libbpf-rs: Add set_autoload to OpenProgram

Add bpf_program__set_autoload equivalent API to libbpf-rs. This allows user to decide which BPF program should be loaded automatically. Closes libbpf#175. Signed-off-by: Hengqi Chen <chenhengqi@outlook.com>
holyspectral · Mar 2, 2022 · 796f93e · 796f93e
1 parent 42b84c0
commit 796f93e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/libbpf-rs/src/object.rs b/libbpf-rs/src/object.rs
@@ -360,12 +360,6 @@ impl Object {
             let section = unsafe { libbpf_sys::bpf_program__section_name(next_ptr) };
             let section = util::c_ptr_to_string(section)?;
 
-            // Get the program fd
-            let fd = unsafe { libbpf_sys::bpf_program__fd(next_ptr) };
-            if fd < 0 {
-                return Err(Error::System(-fd));
-            }
-
             // Add the program to the hashmap
             obj.progs
                 .insert(name.clone(), Program::new(next_ptr, name, section));

diff --git a/libbpf-rs/src/program.rs b/libbpf-rs/src/program.rs
@@ -41,6 +41,11 @@ impl OpenProgram {
     pub fn section(&self) -> &str {
         &self.section
     }
+
+    pub fn set_autoload(&mut self, autoload: bool) -> Result<()> {
+        let ret = unsafe { libbpf_sys::bpf_program__set_autoload(self.ptr, autoload) };
+        util::parse_ret(ret)
+    }
 }
 
 /// Type of a [`Program`]. Maps to `enum bpf_prog_type` in kernel uapi.