dulwich-users team mailing list archive

[dborowitz@xxxxxxxxxx]

From: Dave Borowitz <dborowitz@xxxxxxxxxx>

Change-Id: I1ce460d3d2a4f1b897e3c1af1c93812ebb216a4f
---
 dulwich/pack.py            |   16 ++++++++++------
 dulwich/tests/test_pack.py |    8 ++++++--
 2 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/dulwich/pack.py b/dulwich/pack.py
index e71a2d2..2d0946f 100644
--- a/dulwich/pack.py
+++ b/dulwich/pack.py
@@ -1661,12 +1661,7 @@ class Pack(object):
         if self._data is None:
             self._data = self._data_load()
             self._data.pack = self
-            assert len(self.index) == len(self._data)
-            idx_stored_checksum = self.index.get_pack_checksum()
-            data_stored_checksum = self._data.get_stored_checksum()
-            if idx_stored_checksum != data_stored_checksum:
-                raise ChecksumMismatch(sha_to_hex(idx_stored_checksum),
-                                       sha_to_hex(data_stored_checksum))
+            self.check_length_and_checksum()
         return self._data
 
     @property
@@ -1698,6 +1693,15 @@ class Pack(object):
         """Iterate over all the sha1s of the objects in this pack."""
         return iter(self.index)
 
+    def check_length_and_checksum(self):
+        """Sanity check the length and checksum of the pack index and data."""
+        assert len(self.index) == len(self.data)
+        idx_stored_checksum = self.index.get_pack_checksum()
+        data_stored_checksum = self.data.get_stored_checksum()
+        if idx_stored_checksum != data_stored_checksum:
+            raise ChecksumMismatch(sha_to_hex(idx_stored_checksum),
+                                   sha_to_hex(data_stored_checksum))
+
     def check(self):
         """Check the integrity of this pack.
 
diff --git a/dulwich/tests/test_pack.py b/dulwich/tests/test_pack.py
index 59445f2..550c01e 100644
--- a/dulwich/tests/test_pack.py
+++ b/dulwich/tests/test_pack.py
@@ -381,7 +381,7 @@ class TestPack(PackTests):
     def test_length_mismatch(self):
         data = self.get_pack_data(pack1_sha)
         index = self.get_pack_index(pack1_sha)
-        self.assertTrue(Pack.from_objects(data, index).data)
+        Pack.from_objects(data, index).check_length_and_checksum()
 
         data._file.seek(12)
         bad_file = StringIO()
@@ -391,17 +391,21 @@ class TestPack(PackTests):
         bad_data = PackData('', file=bad_file)
         bad_pack = Pack.from_lazy_objects(lambda: bad_data, lambda: index)
         self.assertRaises(AssertionError, lambda: bad_pack.data)
+        self.assertRaises(AssertionError,
+                          lambda: bad_pack.check_length_and_checksum())
 
     def test_checksum_mismatch(self):
         data = self.get_pack_data(pack1_sha)
         index = self.get_pack_index(pack1_sha)
-        self.assertTrue(Pack.from_objects(data, index).data)
+        Pack.from_objects(data, index).check_length_and_checksum()
 
         data._file.seek(0)
         bad_file = StringIO(data._file.read()[:-20] + ('\xff' * 20))
         bad_data = PackData('', file=bad_file)
         bad_pack = Pack.from_lazy_objects(lambda: bad_data, lambda: index)
         self.assertRaises(ChecksumMismatch, lambda: bad_pack.data)
+        self.assertRaises(ChecksumMismatch, lambda:
+                          bad_pack.check_length_and_checksum())
 
 
 class WritePackTests(TestCase):
-- 
1.7.3.1