From abc8d27440de76cf5ed2ca4aa56883bc07d3afc4 Mon Sep 17 00:00:00 2001
From: Dave Tucker <dave@dtucker.co.uk>
Date: Fri, 14 Jan 2022 16:53:52 +0000
Subject: [PATCH] aya: Retrieve program from pinned path

Signed-off-by: Dave Tucker <dave@dtucker.co.uk>
---
 aya/src/programs/mod.rs | 45 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/aya/src/programs/mod.rs b/aya/src/programs/mod.rs
index 6e269cb3..6410273c 100644
--- a/aya/src/programs/mod.rs
+++ b/aya/src/programs/mod.rs
@@ -95,7 +95,10 @@ use crate::{
     generated::{bpf_attach_type, bpf_prog_info, bpf_prog_type},
     maps::MapError,
     obj::{self, btf::BtfError, Function, KernelVersion},
-    sys::{bpf_load_program, bpf_pin_object, bpf_prog_detach, bpf_prog_query, BpfLoadProgramAttrs},
+    sys::{
+        bpf_get_object, bpf_load_program, bpf_obj_get_info_by_fd, bpf_pin_object, bpf_prog_detach,
+        bpf_prog_get_fd_by_id, bpf_prog_query, BpfLoadProgramAttrs,
+    },
     util::VerifierLog,
 };
 
@@ -702,4 +705,44 @@ impl ProgramInfo {
     pub fn id(&self) -> u32 {
         self.0.id
     }
+
+    /// Returns the fd associated with the program.
+    ///
+    /// The returned fd must be closed when no longer needed.
+    pub fn fd(&self) -> Result<RawFd, ProgramError> {
+        let fd =
+            bpf_prog_get_fd_by_id(self.0.id).map_err(|io_error| ProgramError::SyscallError {
+                call: "bpf_prog_get_fd_by_id".to_owned(),
+                io_error,
+            })?;
+        Ok(fd as RawFd)
+    }
+
+    /// Loads a program from a pinned path in bpffs.
+    pub fn from_pinned<P: AsRef<Path>>(path: P) -> Result<ProgramInfo, ProgramError> {
+        let path_string = match CString::new(path.as_ref().to_str().unwrap()) {
+            Ok(s) => s,
+            Err(e) => {
+                return Err(ProgramError::InvalidPinPath {
+                    error: e.to_string(),
+                })
+            }
+        };
+        let fd =
+            bpf_get_object(&path_string).map_err(|(_, io_error)| ProgramError::SyscallError {
+                call: "bpf_obj_get".to_owned(),
+                io_error,
+            })? as RawFd;
+
+        let info = bpf_obj_get_info_by_fd(fd).map_err(|io_error| ProgramError::SyscallError {
+            call: "bpf_obj_get_info_by_fd".to_owned(),
+            io_error,
+        })?;
+
+        unsafe {
+            libc::close(fd);
+        }
+
+        Ok(ProgramInfo(info))
+    }
 }