Handle error when encrypted archive file is exctracted. (Closes: #904685)

from diffoscope.tools import tool_required

from diffoscope.difference import Difference

from diffoscope.exc import ContainerExtractionError

from .utils.file import File

from .directory import Directory

        # any weird character so we can get to the bytes.

        targetpath = os.path.join(dest_dir, os.path.basename(member_name)).encode(

            sys.getfilesystemencoding(), errors='replace')

        with self.archive.open(member_name) as source, open(targetpath, 'wb') as target:

        try:

            with self.archive.open(member_name) as source, \

                    open(targetpath, 'wb') as target:

                shutil.copyfileobj(source, target)

            return targetpath.decode(sys.getfilesystemencoding())

        except RuntimeError as exc:

            # Handle encrypted files see line 1292 of zipfile.py

            is_encrypted = self.archive.getinfo(member_name).flag_bits & 0x1

            if is_encrypted:

                raise ContainerExtractionError(member_name, exc)

            raise

    def get_member(self, member_name):

        zipinfo = self.archive.getinfo(member_name)

zip1 = load_fixture('test1.zip')

zip2 = load_fixture('test2.zip')

zip3 = load_fixture('test3.zip')

encrypted_zip1 = load_fixture('encrypted1.zip')

encrypted_zip2 = load_fixture('encrypted2.zip')

mozzip1 = load_fixture('test1.mozzip')

mozzip2 = load_fixture('test2.mozzip')

@skip_unless_tools_exist('zipinfo')

def test_mozzip_compare_non_existing(monkeypatch, mozzip1):

    assert_non_existing(monkeypatch, mozzip1)

def test_encrypted(encrypted_zip1, encrypted_zip2):

    difference = encrypted_zip1.compare(encrypted_zip2)

    assert difference is not None