From 1da10120d9cf3de5dca8dbf50ba7c67ecd20626b Mon Sep 17 00:00:00 2001
From: Michael Wu <mwu@mozilla.com>
Date: Mon, 23 May 2011 17:23:13 -0700
Subject: [PATCH] Bug 600241 - Strip zips during optimization, r=taras

---
 browser/installer/Makefile.in         |  3 +-
 config/optimizejars.py                | 46 +++++++++++++++++----------
 toolkit/mozapps/installer/packager.mk |  3 +-
 3 files changed, 32 insertions(+), 20 deletions(-)

diff --git a/browser/installer/Makefile.in b/browser/installer/Makefile.in
index 47138c44ba55..6cfb170a85ac 100644
--- a/browser/installer/Makefile.in
+++ b/browser/installer/Makefile.in
@@ -118,8 +118,7 @@ GENERATE_CACHE = \
   rm -rf jsloader && \
   $(UNZIP) startupCache.zip && \
   rm startupCache.zip && \
-  find jsloader | xargs touch -t 201001010000 && \
-  $(ZIP) -r9mX omni.jar jsloader
+  $(ZIP) -r9m omni.jar jsloader
 endif
 
 include $(topsrcdir)/toolkit/mozapps/installer/packager.mk
diff --git a/config/optimizejars.py b/config/optimizejars.py
index 234d82bf0392..42fbfdee49dd 100644
--- a/config/optimizejars.py
+++ b/config/optimizejars.py
@@ -201,10 +201,15 @@ def optimizejar(jar, outjar, inlog = None):
         readahead = struct.unpack("<I", jarblob.readAt(0, 4))[0]
         print("%s: startup data ends at byte %d" % (outjar, readahead));
 
+    total_stripped = 0;
     jarblob.offset = cdir_offset
     central_directory = []
-    for i  in range(0, dirend.cdir_entries):
+    for i in range(0, dirend.cdir_entries):
         entry = jarblob.read_struct(cdir_entry)
+        if entry.filename[-1:] == "/":
+            total_stripped += len(entry.pack())
+        else:
+            total_stripped += entry.extrafield_size
         central_directory.append(entry)
         
     reordered_count = 0
@@ -235,7 +240,7 @@ def optimizejar(jar, outjar, inlog = None):
         # This also lets us specify how many entries should be preread
         dirend.cdir_offset = 4
         # make room for central dir + end of dir + 4 extra bytes at front
-        out_offset = dirend.cdir_offset + dirend.cdir_size + size_of(cdir_end) 
+        out_offset = dirend.cdir_offset + dirend.cdir_size + size_of(cdir_end) - total_stripped
         outfd.seek(out_offset)
 
     cdir_data = ""
@@ -245,6 +250,19 @@ def optimizejar(jar, outjar, inlog = None):
         # read in the header twice..first for comparison, second time for convenience when writing out
         jarfile = jarblob.read_struct(local_file_header, entry.offset)
         assert_true(jarfile.filename == entry.filename, "Directory/Localheader mismatch")
+        # drop directory entries
+        if entry.filename[-1:] == "/":
+            total_stripped += len(jarfile.pack())
+            dirend.cdir_entries -= 1
+            continue
+        # drop extra field data
+        else:
+            total_stripped += jarfile.extra_field_size;
+        entry.extrafield = jarfile.extra_field = ""
+        entry.extrafield_size = jarfile.extra_field_size = 0
+        # January 1st, 2010
+        entry.lastmod_date = jarfile.lastmod_date = ((2010 - 1980) << 9) | (1 << 5) | 1
+        entry.lastmod_time = jarfile.lastmod_time = 0
         data = jarfile.pack()
         outfd.write(data)
         old_entry_offset = entry.offset
@@ -270,11 +288,11 @@ def optimizejar(jar, outjar, inlog = None):
     if inlog is None:
         dirend.cdir_offset = out_offset
 
+    dirend.cdir_size = len(cdir_data)
     dirend_data = dirend.pack()
     assert_true(size_of(cdir_end) == len(dirend_data), "Failed to serialize directory end correctly. Serialized size;%d, expected:%d"%(len(dirend_data), size_of(cdir_end)));
 
     outfd.seek(dirend.cdir_offset)
-    assert_true(len(cdir_data) == dirend.cdir_size, "Failed to serialize central directory correctly. Serialized size;%d, expected:%d expected-size:%d" % (len(cdir_data), dirend.cdir_size, dirend.cdir_size - len(cdir_data)));
     outfd.write(cdir_data)
     outfd.write(dirend_data)
 
@@ -286,6 +304,7 @@ def optimizejar(jar, outjar, inlog = None):
         outfd.seek(out_offset)
         outfd.write(dirend_data)
 
+    print "Stripped %d bytes" % total_stripped
     print "%s %d/%d in %s" % (("Ordered" if inlog is not None else "Deoptimized"),
                               reordered_count, len(central_directory), outjar)
     outfd.close()
@@ -296,20 +315,15 @@ if len(sys.argv) != 5:
     exit(1)
 
 def optimize(JAR_LOG_DIR, IN_JAR_DIR, OUT_JAR_DIR):
-    if not os.path.exists(JAR_LOG_DIR):
-        print("No jar logs found in %s. No jars to optimize." % JAR_LOG_DIR)
-        exit(0)
-
-    ls = os.listdir(JAR_LOG_DIR)
-    for logfile in ls:
-        if not logfile.endswith(".jar.log"):
+    ls = os.listdir(IN_JAR_DIR)
+    for jarfile in ls:
+        if not jarfile.endswith(".jar"):
             continue
-        injarfile = os.path.join(IN_JAR_DIR, logfile[:-4])
-        outjarfile = os.path.join(OUT_JAR_DIR, logfile[:-4]) 
-        if not os.path.exists(injarfile):
-            print "Warning: Skipping %s, %s doesn't exist" % (logfile, injarfile)
-            continue
-        logfile = os.path.join(JAR_LOG_DIR, logfile)
+        injarfile = os.path.join(IN_JAR_DIR, jarfile)
+        outjarfile = os.path.join(OUT_JAR_DIR, jarfile) 
+        logfile = os.path.join(JAR_LOG_DIR, jarfile + ".log")
+        if not os.path.isfile(logfile):
+            logfile = None
         optimizejar(injarfile, outjarfile, logfile)
 
 def deoptimize(JAR_LOG_DIR, IN_JAR_DIR, OUT_JAR_DIR):
diff --git a/toolkit/mozapps/installer/packager.mk b/toolkit/mozapps/installer/packager.mk
index 93d6a36d60bc..7038e14a3d2d 100644
--- a/toolkit/mozapps/installer/packager.mk
+++ b/toolkit/mozapps/installer/packager.mk
@@ -390,8 +390,7 @@ PACK_OMNIJAR	= \
   grep -h '^binary-component' components/*.manifest > binary.manifest ; \
   sed -e 's/^binary-component/\#binary-component/' components/components.manifest > components.manifest && \
   mv components.manifest components && \
-  find . | xargs touch -t 201001010000 && \
-  zip -r9mX omni.jar $(OMNIJAR_FILES) -x $(NON_OMNIJAR_FILES) && \
+  zip -r9m omni.jar $(OMNIJAR_FILES) -x $(NON_OMNIJAR_FILES) && \
   $(GENERATE_CACHE) && \
   $(OPTIMIZE_JARS_CMD) --optimize $(JARLOG_DIR) ./ ./ && \
   mv binary.manifest components && \