Add module::for_each_section

Author: ZeroMemes

README.md

@@ -115,6 +115,7 @@ Below is a summary of the current support for libhat's platform-dependent APIs:
 | `hp::is_executable`            |    ✅    |   ✅   |       |    ✅    |
 | `hp::module::get_module_data`  |    ✅    |   ✅   |       |    ✅    |
 | `hp::module::get_section_data` |    ✅    |       |       |         |
+| `hp::module::for_each_section` |    ✅    |       |       |         |
 | `hp::module::for_each_segment` |    ✅    |   ✅   |   ✅   |    ✅    |
 
 ## Quick start

include/libhat/process.hpp

@@ -28,6 +28,10 @@ LIBHAT_EXPORT namespace hat::process {
         /// Returns the memory region for a named section
         [[nodiscard]] std::span<std::byte> get_section_data(std::string_view name) const;
 
+        /// Invokes the callback for each named linker section defined by this module as long as it returns true. The
+        /// returned byte range is not guaranteed to have page aligned begin and end addresses.
+        void for_each_section(const std::function<bool(std::string_view name, std::span<std::byte>, hat::protection)>& callback) const;
+
         /// Invokes the callback for each memory segment defined by this module as long as it returns true. Depending on
         /// the platform, a segment may be represented by multiple linker sections. The returned byte range is not
         /// guaranteed to have page aligned begin and end addresses.

src/os/win32/Process.cpp

@@ -126,8 +126,7 @@ namespace hat::process {
         return {scanBytes, sizeOfImage};
     }
 
-    std::span<std::byte> hat::process::module::get_section_data(const std::string_view name) const {
-        auto* bytes = reinterpret_cast<std::byte*>(this->address());
+    std::span<std::byte> module::get_section_data(const std::string_view name) const {
         const auto& ntHeaders = getNTHeaders(*this);
 
         const auto* sectionHeader = IMAGE_FIRST_SECTION(&ntHeaders);
@@ -138,15 +137,40 @@ namespace hat::process {
             };
             if (sectionName == name) {
                 return {
-                    bytes + sectionHeader->VirtualAddress,
+                    reinterpret_cast<std::byte*>(this->address() + sectionHeader->VirtualAddress),
                     static_cast<size_t>(sectionHeader->Misc.VirtualSize)
                 };
             }
         }
         return {};
     }
 
-    void hat::process::module::for_each_segment(const std::function<bool(std::span<std::byte>, hat::protection)>& callback) const {
+    void module::for_each_section(const std::function<bool(std::string_view, std::span<std::byte>, hat::protection)>& callback) const {
+        const auto& ntHeaders = getNTHeaders(*this);
+
+        const auto* sectionHeader = IMAGE_FIRST_SECTION(&ntHeaders);
+        for (WORD i = 0; i < ntHeaders.FileHeader.NumberOfSections; i++, sectionHeader++) {
+            const std::string_view name{
+                reinterpret_cast<const char*>(sectionHeader->Name),
+                strnlen_s(reinterpret_cast<const char*>(sectionHeader->Name), IMAGE_SIZEOF_SHORT_NAME)
+            };
+            const std::span data{
+                reinterpret_cast<std::byte*>(this->address() + sectionHeader->VirtualAddress),
+                sectionHeader->Misc.VirtualSize
+            };
+
+            hat::protection prot{};
+            if (sectionHeader->Characteristics & IMAGE_SCN_MEM_READ) prot |= hat::protection::Read;
+            if (sectionHeader->Characteristics & IMAGE_SCN_MEM_WRITE) prot |= hat::protection::Write;
+            if (sectionHeader->Characteristics & IMAGE_SCN_MEM_EXECUTE) prot |= hat::protection::Execute;
+
+            if (!callback(name, data, prot)) {
+                break;
+            }
+        }
+    }
+
+    void module::for_each_segment(const std::function<bool(std::span<std::byte>, hat::protection)>& callback) const {
         const auto& ntHeaders = getNTHeaders(*this);
 
         const auto* sectionHeader = IMAGE_FIRST_SECTION(&ntHeaders);