| | |
| | from __future__ import absolute_import |
| | from __future__ import print_function |
| | from __future__ import unicode_literals |
| | from hashlib import sha1 |
| | from io import BytesIO |
| | import zlib |
| |
|
| | from hwp5.filestructure import Hwp5DistDoc |
| | from hwp5.distdoc import decode_head_to_sha1 |
| | from hwp5.distdoc import decode_head_to_key |
| | from hwp5.distdoc import decrypt_tail |
| | from hwp5.recordstream import read_record |
| | from hwp5.tagids import HWPTAG_PARA_HEADER |
| | import hwp5.distdoc |
| | import hwp5.compressed |
| |
|
| | from .test_filestructure import TestBase |
| |
|
| |
|
| | class TestHwp5DistDocFunctions(TestBase): |
| |
|
| | hwp5file_name = 'viewtext.hwp' |
| | password_sha1 = sha1(b'12345').hexdigest() |
| |
|
| | @property |
| | def hwp5distdoc(self): |
| | return Hwp5DistDoc(self.olestg) |
| |
|
| | @property |
| | def section(self): |
| | return self.hwp5distdoc['ViewText']['Section0'] |
| |
|
| | def test_distdoc_decode_head_to_sha1(self): |
| | password_sha1 = self.password_sha1 |
| | password_sha1 = password_sha1.upper() |
| | password_sha1 = password_sha1.encode('utf-16le') |
| | expected = password_sha1 |
| | section_head = self.section.head() |
| | decoded = decode_head_to_sha1(section_head) |
| | self.assertEqual(expected, decoded) |
| |
|
| | def test_distdoc_decode_head_to_key(self): |
| | section = self.section |
| | expected = self.password_sha1.upper().encode('utf-16le')[:16] |
| | self.assertEqual(expected, decode_head_to_key(section.head())) |
| | self.assertEqual(expected, section.head_key()) |
| |
|
| | def test_distdoc_decrypt_tail(self): |
| | section = self.section |
| |
|
| | key = section.head_key() |
| | tail = section.tail() |
| | decrypted = decrypt_tail(key, tail) |
| | decompressed = zlib.decompress(decrypted, -15) |
| | record = read_record(BytesIO(decompressed), 0) |
| | self.assertEqual(0, record['level']) |
| | self.assertEqual(HWPTAG_PARA_HEADER, record['tagid']) |
| | self.assertEqual(22, record['size']) |
| |
|
| | self.assertEqual(390, len(decompressed)) |
| |
|
| | def test_distdoc_decode(self): |
| | section = self.section |
| |
|
| | stream = hwp5.distdoc.decode(section.wrapped.open()) |
| | stream = hwp5.compressed.decompress(stream) |
| | record = read_record(stream, 0) |
| | self.assertEqual(0, record['level']) |
| | self.assertEqual(HWPTAG_PARA_HEADER, record['tagid']) |
| | self.assertEqual(22, record['size']) |
| |
|
