Add remove_dynamic_entries_by_tag and add_dynamic_entry

Author: yujincheng08

api/rust/cargo/lief/src/elf/binary.rs

@@ -7,7 +7,7 @@ use num_traits::{cast, Num};
 use lief_ffi as ffi;
 
 use super::builder::Config;
-use super::dynamic::{self, DynamicEntries, Library};
+use super::dynamic::{self, DynamicEntries, DynamicEntry, Library};
 use super::hash::{Gnu, Sysv};
 use super::header::Header;
 use super::note::ItNotes;
@@ -118,6 +118,16 @@ impl Binary {
         DynamicEntries::new(self.ptr.dynamic_entries())
     }
 
+    /// Remove **all** dynamic entries with the given tag
+    pub fn remove_dynamic_entries_by_tag(&mut self, tag: dynamic::Tag) {
+        self.ptr.as_mut().unwrap().remove_dynamic_entries_by_tag(tag.into())
+    }
+
+    /// Add the given dynamic entry and return the new entry
+    pub fn add_dynamic_entry(&mut self, entry: &dynamic::Entries) -> dynamic::Entries {
+        dynamic::Entries::from_ffi(self.ptr.as_mut().unwrap().add_dynamic_entry(entry.as_base()))
+    }
+
     /// Return an iterator over the dynamic [`crate::elf::Symbol`] of the binary
     pub fn dynamic_symbols(&self) -> DynamicSymbols {
         DynamicSymbols::new(self.ptr.dynamic_symbols())

api/rust/cargo/lief/src/elf/dynamic.rs

@@ -832,6 +832,14 @@ impl Flags<'_> {
     pub fn flags(&self) -> DtFlags {
         DtFlags::from(self.ptr.flags())
     }
+
+    pub fn create_dt_flag(value: u64) -> Self {
+        Self::from_ffi(ffi::ELF_DynamicEntryFlags::create_dt_flag(value))
+    }
+
+    pub fn create_dt_flag_1(value: u64) -> Self {
+        Self::from_ffi(ffi::ELF_DynamicEntryFlags::create_dt_flag_1(value))
+    }
 }
 
 impl FromFFI<ffi::ELF_DynamicEntryFlags> for Flags<'_> {

api/rust/include/LIEF/rust/ELF/Binary.hpp

@@ -242,6 +242,14 @@ class ELF_Binary : public AbstractBinary {
     return std::make_unique<it_dynamic_entries>(impl());
   }
 
+  void remove_dynamic_entries_by_tag(uint64_t tag) {
+    impl().remove(LIEF::ELF::DynamicEntry::TAG(tag));
+  }
+
+  auto add_dynamic_entry(const ELF_DynamicEntry& entry) {
+     return std::make_unique<ELF_DynamicEntry>(impl().add(entry.get()));
+  }
+
   auto dynamic_symbols() const {
     return std::make_unique<it_dynamic_symbols>(impl());
   }

api/rust/include/LIEF/rust/ELF/DynamicEntryFlags.hpp

@@ -19,7 +19,9 @@
 class ELF_DynamicEntryFlags : public ELF_DynamicEntry {
   public:
   using lief_t = LIEF::ELF::DynamicEntryFlags;
-
+  ELF_DynamicEntryFlags(const lief_t& impl) :
+    ELF_DynamicEntry(impl.clone())
+  {}
   auto flags() const {
     return impl().raw_flags();
   }
@@ -28,6 +30,14 @@ class ELF_DynamicEntryFlags : public ELF_DynamicEntry {
     return lief_t::classof(&entry.get());
   }
 
+  static auto create_dt_flag(uint64_t value) {
+    return std::make_unique<ELF_DynamicEntryFlags>(lief_t::create_dt_flag(value));
+  }
+
+  static auto create_dt_flag_1(uint64_t value) {
+    return std::make_unique<ELF_DynamicEntryFlags>(lief_t::create_dt_flag_1(value));
+  }
+
   private:
   const lief_t& impl() const { return as<lief_t>(this); }
 };