|
| 1 | +# -*- coding: UTF-8 -*- |
| 2 | +"""Pkzip Codec - pkzip content compression. |
| 3 | +
|
| 4 | +NB: Not an encoding properly speaking. |
| 5 | +
|
| 6 | +This codec: |
| 7 | +- en/decodes strings from str to str |
| 8 | +- en/decodes strings from bytes to bytes |
| 9 | +- decodes file content to str (read) |
| 10 | +- encodes file content from str to bytes (write) |
| 11 | +""" |
| 12 | +import zipfile |
| 13 | + |
| 14 | +from ..__common__ import * |
| 15 | + |
| 16 | + |
| 17 | +_str = ["test", "This is a test", "@random{1024}"] |
| 18 | +__examples1__ = {'enc-dec(pkzip-deflate|deflate)': _str} |
| 19 | +__examples2__ = {'enc-dec(pkzip_bz2|bzip2)': _str} |
| 20 | +__examples3__ = {'enc-dec(pkzip-lzma|lzma)': _str} |
| 21 | + |
| 22 | + |
| 23 | +if PY3: |
| 24 | + def pkzip_encode(compression_type): |
| 25 | + def _encode(text, errors="strict"): |
| 26 | + c = zipfile._get_compressor(compression_type) |
| 27 | + return c.compress(b(text)) + c.flush(), len(text) |
| 28 | + return _encode |
| 29 | + |
| 30 | + |
| 31 | + def pkzip_decode(compression_type): |
| 32 | + def _decode(data, errors="strict"): |
| 33 | + d = zipfile._get_decompressor(compression_type) |
| 34 | + r = d.decompress(b(data)) |
| 35 | + return r, len(r) |
| 36 | + return _decode |
| 37 | + |
| 38 | + |
| 39 | + add("pkzip_deflate", pkzip_encode(8), pkzip_decode(8), r"(?:(?:pk)?zip[-_])?deflate", |
| 40 | + entropy=7.9, examples=__examples1__, guess=["deflate"]) |
| 41 | + |
| 42 | + add("pkzip_bzip2", pkzip_encode(12), pkzip_decode(12), r"(?:(?:pk)?zip[-_])?bz(?:ip)?2", |
| 43 | + entropy=7.9, examples=__examples2__, guess=["bz2"]) |
| 44 | + |
| 45 | + add("pkzip_lzma", pkzip_encode(14), pkzip_decode(14), r"(?:(?:pk)?zip[-_])?lzma", |
| 46 | + entropy=7.9, examples=__examples3__, guess=["lzma"]) |
| 47 | + |
0 commit comments