Introduce Program::attach_iter_with_opts()

Add the Program::attach_iter_with_opts() method to allow for iterator
attachment with support for providing additional options.

Signed-off-by: Daniel Müller <[email protected]>

Author: d-e-s-o

libbpf-rs/CHANGELOG.md

@@ -1,3 +1,9 @@
+Unreleased
+----------
+- Added `Program::attach_iter_with_opts` for attaching to iterators with
+  additional options
+
+
 0.26.0-beta.0
 -------------
 - Added `target_obj_id` and `target_btf_id` fields to `TracingLinkInfo`

libbpf-rs/src/lib.rs

@@ -130,7 +130,9 @@ pub use crate::print::get_print;
 pub use crate::print::set_print;
 pub use crate::print::PrintCallback;
 pub use crate::print::PrintLevel;
+pub use crate::program::CgroupIterOrder;
 pub use crate::program::Input as ProgramInput;
+pub use crate::program::IterOpts;
 pub use crate::program::KprobeMultiOpts;
 pub use crate::program::KprobeOpts;
 pub use crate::program::OpenProgram;

libbpf-rs/src/program.rs

@@ -159,6 +159,38 @@ impl From<PerfEventOpts> for libbpf_sys::bpf_perf_event_opts {
     }
 }
 
+
+/// Iteration order for cgroups.
+#[non_exhaustive]
+#[repr(u32)]
+#[derive(Clone, Debug, Default)]
+pub enum CgroupIterOrder {
+    /// Use the default iteration order.
+    #[default]
+    Default = libbpf_sys::BPF_CGROUP_ITER_ORDER_UNSPEC,
+    /// Process only a single object.
+    SelfOnly = libbpf_sys::BPF_CGROUP_ITER_SELF_ONLY,
+    /// Walk descendants in pre-order.
+    DescendantsPre = libbpf_sys::BPF_CGROUP_ITER_DESCENDANTS_PRE,
+    /// Walk descendants in post-order.
+    DescendantsPost = libbpf_sys::BPF_CGROUP_ITER_DESCENDANTS_POST,
+    /// Walk ancestors upward.
+    AncestorsUp = libbpf_sys::BPF_CGROUP_ITER_ANCESTORS_UP,
+}
+
+
+/// Options to optionally be provided when attaching to an iterator.
+#[derive(Clone, Debug, Default)]
+pub struct IterOpts<'fd> {
+    /// The file descriptor of the cgroup.
+    pub cgroup_fd: Option<BorrowedFd<'fd>>,
+    /// The iteration to use on the cgroup.
+    pub cgroup_order: CgroupIterOrder,
+    #[doc(hidden)]
+    pub _non_exhaustive: (),
+}
+
+
 /// An immutable parsed but not yet loaded BPF program.
 pub type OpenProgram<'obj> = OpenProgramImpl<'obj>;
 /// A mutable parsed but not yet loaded BPF program.
@@ -1319,14 +1351,9 @@ impl<'obj> ProgramMut<'obj> {
         )
     }
 
-    /// Attach this program to a
-    /// [BPF Iterator](https://www.kernel.org/doc/html/latest/bpf/bpf_iterators.html).
-    /// The entry point of the program must be defined with `SEC("iter")` or `SEC("iter.s")`.
-    pub fn attach_iter(&self, map_fd: BorrowedFd<'_>) -> Result<Link> {
-        let mut linkinfo = libbpf_sys::bpf_iter_link_info::default();
-        linkinfo.map.map_fd = map_fd.as_raw_fd() as _;
+    fn attach_iter_impl(&self, linkinfo: &mut libbpf_sys::bpf_iter_link_info) -> Result<Link> {
         let attach_opt = libbpf_sys::bpf_iter_attach_opts {
-            link_info: &mut linkinfo as *mut libbpf_sys::bpf_iter_link_info,
+            link_info: linkinfo as *mut libbpf_sys::bpf_iter_link_info,
             link_info_len: size_of::<libbpf_sys::bpf_iter_link_info>() as _,
             sz: size_of::<libbpf_sys::bpf_iter_attach_opts>() as _,
             ..Default::default()
@@ -1344,6 +1371,43 @@ impl<'obj> ProgramMut<'obj> {
         Ok(link)
     }
 
+
+    /// Attach this program to a
+    /// [BPF Iterator](https://www.kernel.org/doc/html/latest/bpf/bpf_iterators.html).
+    /// The entry point of the program must be defined with `SEC("iter")` or `SEC("iter.s")`.
+    pub fn attach_iter(&self, map_fd: BorrowedFd<'_>) -> Result<Link> {
+        let mut linkinfo = libbpf_sys::bpf_iter_link_info::default();
+        linkinfo.map.map_fd = map_fd.as_raw_fd() as _;
+        self.attach_iter_impl(&mut linkinfo)
+    }
+
+    /// Attach this program to a
+    /// [BPF Iterator](https://www.kernel.org/doc/html/latest/bpf/bpf_iterators.html),
+    /// providing additional options.
+    ///
+    /// The entry point of the program must be defined with `SEC("iter")` or `SEC("iter.s")`.
+    pub fn attach_iter_with_opts(
+        &self,
+        map_fd: BorrowedFd<'_>,
+        opts: IterOpts<'_>,
+    ) -> Result<Link> {
+        let IterOpts {
+            cgroup_fd,
+            cgroup_order,
+            _non_exhaustive: (),
+        } = opts;
+
+        let mut linkinfo = libbpf_sys::bpf_iter_link_info::default();
+        linkinfo.map.map_fd = map_fd.as_raw_fd() as _;
+        linkinfo.cgroup.order = cgroup_order as libbpf_sys::bpf_cgroup_iter_order;
+
+        if let Some(cgroup_fd) = cgroup_fd {
+            linkinfo.cgroup.cgroup_fd = cgroup_fd.as_raw_fd() as _;
+        }
+
+        self.attach_iter_impl(&mut linkinfo)
+    }
+
     /// Test run the program with the given input data.
     ///
     /// This function uses the