Make simple files for testing compresion

diracdeltas · diracdeltas · commit 59a92fa6e344 · 2013-10-02T17:48:41.000-07:00
diff --git a/makexpi_test.sh b/makexpi_test.sh
@@ -0,0 +1,20 @@
+#!/bin/sh
+cd "`dirname $0`"
+XPI_NAME="pkg/test.xpi"
+
+[ -d pkg ] || mkdir pkg
+
+# The name/version of the XPI we're building comes from src/install.rdf
+cd src
+
+# Build the XPI!
+rm -f "../$XPI_NAME"
+#zip -q -X -9r "../$XPI_NAME" . "-x@../.build_exclusions"
+
+python ../utils/create_xpi_test.py "../$XPI_NAME" "../.build_exclusions"
+
+ret="$?"
+if [ "$ret" != 0 ]; then
+    rm -f "../$XPI_NAME"
+    exit "$?"
+fi
diff --git a/src/testfile.txt b/src/testfile.txt
diff --git a/utils/create_xpi_test.py b/utils/create_xpi_test.py
@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+
+# Uses the Python zip implementation to create deterministic XPI's
+# Author: Yan Zhu, yan@mit.edu
+
+# ZipFile: infolist, write, writestr,
+
+import os
+import zipfile2_6 as zipfile
+import sys
+import time
+import glob
+
+xpiName = sys.argv[1]
+exclusionsFile = sys.argv[2]
+exclusions = []
+tmpfile = '../pkg/tmp.xpi'
+compress = zipfile.ZIP_STORED
+testfile = 'testfile.txt'
+
+with open(exclusionsFile) as f:
+    for line in f:
+        exclusions.extend(glob.glob(line.strip()))
+
+xpiFile = zipfile.ZipFile(xpiName, mode='w', compression=compress)
+
+def createTmpZipInfo():
+    """
+    Create a non-deterministic zip in order to use the file info
+    generated to create a deterministic zip
+    """
+    xpiFileTmp = zipfile.ZipFile(tmpfile, mode='w', compression=compress)
+    xpiFileTmp.write(testfile, compress_type=compress)
+    xpiFileTmp.close()
+    xpiFileTmp.infolist().sort(key = lambda x: x.filename)
+    return xpiFileTmp.infolist()
+
+def constructZipDet():
+    """
+    Create a deterministic zip by setting timestamps,
+    operating system, and pkzip version info to hard-coded
+    values. See the pkzip specification at
+    https://www.pkware.com/documents/casestudies/APPNOTE.TXT
+    """
+    tmpInfo = createTmpZipInfo()
+    for info in tmpInfo:
+        info.date_time = time.gmtime(1378343307)
+        info.create_system = 3 # aka, UNIX
+        info.create_version = 20
+        info.extract_version = 20
+        xpiFile.writestr(info, open(info.filename).read())
+
+constructZipDet()
+xpiFile.close()
+os.remove(tmpfile)
