From 2f78d2eb94484c787d78ba100024359f43b18fc4 Mon Sep 17 00:00:00 2001
From: Rui Ueyama <ruiu@bluewhale.systems>
Date: Mon, 30 May 2022 13:04:37 +0800
Subject: [PATCH] [Mach-O] Parallelize a part of merge_cstring_sections()

---
 macho/main.cc | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/macho/main.cc b/macho/main.cc
index 8ffe3946..526813ff 100644
--- a/macho/main.cc
+++ b/macho/main.cc
@@ -264,12 +264,13 @@ static void merge_cstring_sections(Context<E> &ctx) {
     }
   }
 
-  for (ObjectFile<E> *file : ctx.objs)
+  tbb::parallel_for_each(ctx.objs, [&](ObjectFile<E> *file) {
     for (std::unique_ptr<InputSection<E>> &isec : file->sections)
       if (isec)
         for (Relocation<E> &r : isec->rels)
           if (r.subsec && r.subsec->is_coalesced)
             r.subsec = r.subsec->replacer;
+  });
 
   auto replace = [&](InputFile<E> *file) {
     for (Symbol<E> *sym : file->syms)
@@ -277,16 +278,14 @@ static void merge_cstring_sections(Context<E> &ctx) {
         sym->subsec = sym->subsec->replacer;
   };
 
-  for (InputFile<E> *file : ctx.objs)
-    replace(file);
-  for (InputFile<E> *file : ctx.dylibs)
-    replace(file);
+  tbb::parallel_for_each(ctx.objs, replace);
+  tbb::parallel_for_each(ctx.dylibs, replace);
 
-  for (ObjectFile<E> *file : ctx.objs) {
+  tbb::parallel_for_each(ctx.objs, [&](ObjectFile<E> *file) {
     std::erase_if(file->subsections, [](Subsection<E> *subsec) {
       return subsec->is_coalesced;
     });
-  }
+  });
 }
 
 template <typename E>