python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-27 02:16:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix memory leak in JPEG2000 decoding, and JPEG decoding using PyPy

Browse Source
This commit is contained in:
Josh Ware 2014-08-26 22:22:35 +10:00 committed by wiredfool
parent 5a47b2bc84
commit 72e2a6cade
1 changed files with 4 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Tests/test_jp2k_leak.py → Tests/test_jp2k_leaks.py
View File

@ -1,18 +1,18 @@
from helper import unittest, PillowTestCase, tearDownModule from helper import unittest, PillowTestCase, tearDownModule
import sys import sys
from os import getpid from StringIO import StringIO
from PIL import Image from PIL import Image
# Limits for testing the leak # Limits for testing the leak
mem_limit = 512*1048576 mem_limit = 768*1048576
stack_size = 8*1048576 stack_size = 8*1048576
iterations = int(mem_limit/stack_size)*2 iterations = int((mem_limit/stack_size)*20)
codecs = dir(Image.core) codecs = dir(Image.core)
test_file = "Tests/images/rgb_trns_ycbc.jp2" test_file = "Tests/images/rgb_trns_ycbc.jp2"
@unittest.skipIf(sys.platform.startswith('win32'), "requires Unix or MacOS") @unittest.skipIf(sys.platform.startswith('win32'), "requires Unix or MacOS")
class TestJp2kLeak(PillowTestCase): class TestJpegLeaks(PillowTestCase):
def setUp(self): def setUp(self):
if "jpeg2k_encoder" not in codecs or "jpeg2k_decoder" not in codecs: if "jpeg2k_encoder" not in codecs or "jpeg2k_decoder" not in codecs:
self.skipTest('JPEG 2000 support not available') self.skipTest('JPEG 2000 support not available')
@ -24,20 +24,6 @@ class TestJp2kLeak(PillowTestCase):
for count in range(iterations): for count in range(iterations):
with Image.open(test_file) as im: with Image.open(test_file) as im:
im.load() im.load()
im.close()
def self_sanity_check(self):
# Arrange
j2k = Image.open('Tests/images/rgb_trns_ycbc.j2k')
jp2 = Image.open('Tests/images/rgb_trns_ycbc.jp2')
# Act
j2k.load()
jp2.load()
# Assert
self.assertEqual(j2k.mode, 'RGBA')
self.assertEqual(jp2.mode, 'RGBA')
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()