Pillow/Tests/test_imagepath.py

import array
import struct

import pytest

from PIL import Image, ImagePath


class TestImagePath:
    def test_path(self):

        p = ImagePath.Path(list(range(10)))

        # sequence interface
        assert len(p) == 5
        assert p[0] == (0.0, 1.0)
        assert p[-1] == (8.0, 9.0)
        assert list(p[:1]) == [(0.0, 1.0)]
        with pytest.raises(TypeError) as cm:
            p["foo"]
        assert str(cm.value) == "Path indices must be integers, not str"
        assert list(p) == [(0.0, 1.0), (2.0, 3.0), (4.0, 5.0), (6.0, 7.0), (8.0, 9.0)]

        # method sanity check
        assert p.tolist() == [
            (0.0, 1.0),
            (2.0, 3.0),
            (4.0, 5.0),
            (6.0, 7.0),
            (8.0, 9.0),
        ]
        assert p.tolist(1) == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]

        assert p.getbbox() == (0.0, 1.0, 8.0, 9.0)

        assert p.compact(5) == 2
        assert list(p) == [(0.0, 1.0), (4.0, 5.0), (8.0, 9.0)]

        p.transform((1, 0, 1, 0, 1, 1))
        assert list(p) == [(1.0, 2.0), (5.0, 6.0), (9.0, 10.0)]

        # alternative constructors
        p = ImagePath.Path([0, 1])
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path([0.0, 1.0])
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path([0, 1])
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path([(0, 1)])
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path(p)
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path(p.tolist(0))
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path(p.tolist(1))
        assert list(p) == [(0.0, 1.0)]
        p = ImagePath.Path(array.array("f", [0, 1]))
        assert list(p) == [(0.0, 1.0)]

        arr = array.array("f", [0, 1])
        if hasattr(arr, "tobytes"):
            p = ImagePath.Path(arr.tobytes())
        else:
            p = ImagePath.Path(arr.tostring())
        assert list(p) == [(0.0, 1.0)]

    def test_overflow_segfault(self):
        # Some Pythons fail getting the argument as an integer, and it falls
        # through to the sequence. Seeing this on 32-bit Windows.
        with pytest.raises((TypeError, MemoryError)):
            # post patch, this fails with a memory error
            x = evil()

            # This fails due to the invalid malloc above,
            # and segfaults
            for i in range(200000):
                x[i] = b"0" * 16


class evil:
    def __init__(self):
        self.corrupt = Image.core.path(0x4000000000000000)

    def __getitem__(self, i):
        x = self.corrupt[i]
        return struct.pack("dd", x[0], x[1])

    def __setitem__(self, i, x):
        self.corrupt[i] = struct.unpack("dd", x)