From 1555be40b7dd9cd03992bed864e46665da65dc4d Mon Sep 17 00:00:00 2001
From: John Reiser <jreiser@BitWagon.com>
Date: Fri, 2 Apr 2004 04:37:03 +0000
Subject: [PATCH] fix alignment of loader after packing multiple blocks 
 p_lx_elf.h p_lx_elf.cpp

committer: jreiser <jreiser> 1080880623 +0000
---
 src/p_lx_elf.cpp | 22 ++++++++++++++++++++++
 src/p_lx_elf.h   |  1 +
 2 files changed, 23 insertions(+)

diff --git a/src/p_lx_elf.cpp b/src/p_lx_elf.cpp
index 3a450b1c..b4405c4f 100644
--- a/src/p_lx_elf.cpp
+++ b/src/p_lx_elf.cpp
@@ -340,6 +340,28 @@ void PackLinuxI386elf::pack2(OutputFile *fo, Filter &ft)
         throwEOFException();
 }
 
+void PackLinuxI386elf::pack3(OutputFile *fo, Filter &ft)
+{
+    // We have packed multiple blocks, so PackLinuxI386::buildLinuxLoader
+    // needs an adjusted ph.c_len in order to get alignment correct.
+    unsigned const save_c_len = ph.c_len;
+    ph.c_len = fo->getBytesWritten() - (
+            // Elf32_Edhr
+        sizeof(elfout.ehdr) +
+            // Elf32_Phdr: 1 for exec86, 2 for sh86, 3 for elf86
+        (elfout.ehdr.e_phentsize * elfout.ehdr.e_phnum) +
+            // checksum UPX! lsize version format
+        sizeof(l_info) +
+            // PT_DYNAMIC with DT_NEEDED "forwarded" from original file
+        ((elfout.ehdr.e_phnum==3) ? elfout.phdr[2].p_memsz : 0) +
+            // p_progid, p_filesize, p_blocksize
+        sizeof(p_info) +
+            // compressed data
+        b_len );
+    buildLoader(&ft);  /* redo for final .align constraints */
+    ph.c_len = save_c_len;
+    super::pack3(fo, ft);
+}
 
 void PackLinuxI386elf::unpackExtent(unsigned wanted, OutputFile *fo,
     unsigned &total_in, unsigned &total_out,
diff --git a/src/p_lx_elf.h b/src/p_lx_elf.h
index cde58eb1..7e32f563 100644
--- a/src/p_lx_elf.h
+++ b/src/p_lx_elf.h
@@ -68,6 +68,7 @@ protected:
 protected:
     virtual void pack1(OutputFile *, Filter &);  // generate executable header
     virtual void pack2(OutputFile *, Filter &);  // append compressed data
+    virtual void pack3(OutputFile *, Filter &);  // append loader
 
     virtual void patchLoader();