Feat: use accumulate address on sections with SHT_ALLOC

VitalyAnkh · Feb 3, 2024 · 8c99dd0 · 8c99dd0
1 parent 2535137
commit 8c99dd0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/elf/passes.cc b/elf/passes.cc
@@ -2457,6 +2457,8 @@ static i64 set_file_offsets(Context<E> &ctx) {
     }
 
     if (first.shdr.sh_type == SHT_NOBITS) {
+      fileoff = align_to(fileoff, first.shdr.sh_addralign);
+      first.shdr.sh_offset = fileoff;
       i++;
       continue;
     }
@@ -2497,8 +2499,11 @@ static i64 set_file_offsets(Context<E> &ctx) {
 
     while (i < chunks.size() &&
            (chunks[i]->shdr.sh_flags & SHF_ALLOC) &&
-           chunks[i]->shdr.sh_type == SHT_NOBITS)
+           chunks[i]->shdr.sh_type == SHT_NOBITS) {
+      fileoff = align_to(fileoff, chunks[i]->shdr.sh_addralign);
+      chunks[i]->shdr.sh_offset = fileoff;
       i++;
+	}
   }
 
   return fileoff;