From 884ba7609344b7a986894792226f1f03472c0fda Mon Sep 17 00:00:00 2001
From: Joanne Koong <joannekoong@gmail.com>
Date: Tue, 26 Jul 2022 15:18:47 -0700
Subject: [PATCH] libbpf-rs: Add setter functionalities on open maps

As part of the initiative to reach feature-parity with libbpf-1.0, this
commit adds:

* set_map_flags
* set_numa_node
* set_map_extra
* set_autocreate
* set_pin_path

Signed-off-by: Joanne Koong <joannekoong@gmail.com>
---
 libbpf-rs/src/map.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/libbpf-rs/src/map.rs b/libbpf-rs/src/map.rs
index e25ef30a..aeb82e95 100644
--- a/libbpf-rs/src/map.rs
+++ b/libbpf-rs/src/map.rs
@@ -62,10 +62,38 @@ impl OpenMap {
         util::parse_ret(ret)
     }
 
+    pub fn set_map_flags(&mut self, flags: u32) -> Result<()> {
+        let ret = unsafe { libbpf_sys::bpf_map__set_map_flags(self.ptr, flags) };
+        util::parse_ret(ret)
+    }
+
+    pub fn set_numa_node(&mut self, numa_node: u32) -> Result<()> {
+        let ret = unsafe { libbpf_sys::bpf_map__set_numa_node(self.ptr, numa_node) };
+        util::parse_ret(ret)
+    }
+
     pub fn set_inner_map_fd(&mut self, inner: &Map) {
         unsafe { libbpf_sys::bpf_map__set_inner_map_fd(self.ptr, inner.fd()) };
     }
 
+    pub fn set_map_extra(&mut self, map_extra: u64) -> Result<()> {
+        let ret = unsafe { libbpf_sys::bpf_map__set_map_extra(self.ptr, map_extra) };
+        util::parse_ret(ret)
+    }
+
+    pub fn set_autocreate(&mut self, autocreate: bool) -> Result<()> {
+        let ret = unsafe { libbpf_sys::bpf_map__set_autocreate(self.ptr, autocreate) };
+        util::parse_ret(ret)
+    }
+
+    pub fn set_pin_path<P: AsRef<Path>>(&mut self, path: P) -> Result<()> {
+        let path_c = util::path_to_cstring(path)?;
+        let path_ptr = path_c.as_ptr();
+
+        let ret = unsafe { libbpf_sys::bpf_map__set_pin_path(self.ptr, path_ptr) };
+        util::parse_ret(ret)
+    }
+
     /// Reuse an fd for a BPF map
     pub fn reuse_fd(&self, fd: i32) -> Result<()> {
         let ret = unsafe { libbpf_sys::bpf_map__reuse_fd(self.ptr, fd) };