[llvm] [llvm-objcopy] MachO: Fix section finding policy for object files (PR #127604)

[via llvm-commits]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-llvm-binary-utilities

Author: Dmitry Nechitaev (Nechda)

<details>
<summary>Changes</summary>

llvm-objcopy is unable to find the specified segment when the `--update-section` flag is used. The input file is object.

This PR fixes #<!-- -->127495 

---
Full diff: https://github.com/llvm/llvm-project/pull/127604.diff


1 Files Affected:

- (modified) llvm/lib/ObjCopy/MachO/MachOObjcopy.cpp (+6) 


``````````diff
diff --git a/llvm/lib/ObjCopy/MachO/MachOObjcopy.cpp b/llvm/lib/ObjCopy/MachO/MachOObjcopy.cpp
index 682edffc84f34..b5ddaa5666229 100644
--- a/llvm/lib/ObjCopy/MachO/MachOObjcopy.cpp
+++ b/llvm/lib/ObjCopy/MachO/MachOObjcopy.cpp
@@ -360,6 +360,12 @@ static Error addSection(const NewSectionInfo &NewSection, Object &Obj) {
 static Expected<Section &> findSection(StringRef SecName, Object &O) {
   StringRef SegName;
   std::tie(SegName, SecName) = SecName.split(",");
+  if (O.Header.FileType == MachO::HeaderFileType::MH_OBJECT) {
+    for (const auto& LC : O.LoadCommands)
+      for (const auto& Sec : LC.Sections)
+        if (Sec->Segname == SegName && Sec->Sectname == SecName)
+          return *Sec;
+  }
   auto FoundSeg =
       llvm::find_if(O.LoadCommands, [SegName](const LoadCommand &LC) {
         return LC.getSegmentName() == SegName;

``````````

</details>


https://github.com/llvm/llvm-project/pull/127604