fix _crop and tests

This commit is contained in:
dinko 2018-08-23 15:40:46 +02:00 committed by Dinko Lubina
parent 43f860fce6
commit 66207b47bc
3 changed files with 25 additions and 6 deletions

View File

@ -6,6 +6,7 @@ TEST_FILE = "Tests/images/hopper.ppm"
ORIGINAL_LIMIT = Image.MAX_IMAGE_PIXELS ORIGINAL_LIMIT = Image.MAX_IMAGE_PIXELS
from PIL.Image import DecompressionBombWarning, DecompressionBombError
class TestDecompressionBomb(PillowTestCase): class TestDecompressionBomb(PillowTestCase):
@ -61,6 +62,28 @@ class TestDecompressionCrop(PillowTestCase):
self.assert_warning(Image.DecompressionBombWarning, self.assert_warning(Image.DecompressionBombWarning,
self.src.crop, box) self.src.crop, box)
def test_crop_decompression_checks(self):
im = Image.new("RGB", (100, 100))
good_values = ((-9999, -9999, -9990, -9990),
(-999, -999, -990, -990))
warning_values = ((-160, -160, 99, 99),
(160, 160, -99, -99))
error_values = ((-99909, -99990, 99999, 99999),
(99909, 99990, -99999, -99999))
for value in good_values:
self.assertEqual(im.crop(value).size, (9,9))
for value in warning_values:
self.assert_warning(DecompressionBombWarning, im.crop, value)
for value in error_values:
with self.assertRaises(DecompressionBombError):
im.crop(value)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -2,7 +2,6 @@ from helper import unittest, PillowTestCase, hopper
from PIL import Image from PIL import Image
class TestImageCrop(PillowTestCase): class TestImageCrop(PillowTestCase):
def test_crop(self): def test_crop(self):

View File

@ -1104,12 +1104,9 @@ class Image(object):
x0, y0, x1, y1 = map(int, map(round, box)) x0, y0, x1, y1 = map(int, map(round, box))
if x1 < x0: absolute_values = (abs(x1 - x0), abs(y1 - y0))
x1 = x0
if y1 < y0:
y1 = y0
_decompression_bomb_check((x1, y1)) _decompression_bomb_check(absolute_values)
return im.crop((x0, y0, x1, y1)) return im.crop((x0, y0, x1, y1))