File size: 3,396 Bytes
d94b56e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
# -*- coding: utf-8 -*-
#
# pyhwp : hwp file format parser in python
# Copyright (C) 2010-2023 mete0r <https://github.com/mete0r>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
''' Decode distribute doc stream.
Usage::
hwp5proc diststream
hwp5proc diststream sha1 [--raw]
hwp5proc diststream key [--raw]
hwp5proc diststream [--loglevel=<loglevel>] [--logfile=<logfile>]
hwp5proc diststream --help
Options::
-h --help Show this screen
--loglevel=<level> Set log level.
--logfile=<file> Set log file.
Example::
$ hwp5proc cat --ole samples/viewtext.hwp ViewText/Section0
| tee Section0.zraw.aes128ecb | hwp5proc diststream | tee Section0.zraw
| hwp5proc rawunz > Section0
$ hwp5proc diststream sha1 < Section0.zraw.aes128ecb
$ echo -n '12345' | sha1sum
'''
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals
from binascii import b2a_hex
from binascii import a2b_hex
import logging
import shutil
import sys
from ..distdoc import decode
from ..distdoc import decode_head_to_sha1
from ..distdoc import decode_head_to_key
from ..recordstream import read_record
PY2 = sys.version_info.major == 2
logger = logging.getLogger(__name__)
def main(args):
if PY2:
input_fp = sys.stdin
output_fp = sys.stdout
else:
input_fp = sys.stdin.buffer
output_fp = sys.stdout.buffer
if args.sha1:
head = read_record(input_fp, 0)
sha1ucs16le = decode_head_to_sha1(head['payload'])
sha1 = a2b_hex(sha1ucs16le.decode('utf-16le'))
if not args.raw:
sha1 = b2a_hex(sha1)
output_fp.write(sha1)
elif args.key:
head = read_record(input_fp, 0)
key = decode_head_to_key(head['payload'])
if not args.raw:
key = b2a_hex(key)
output_fp.write(key)
else:
result = decode(input_fp)
shutil.copyfileobj(result, output_fp)
def diststream_argparser(subparsers, _):
parser = subparsers.add_parser(
'diststream',
help=_(
'Decode a distribute document stream.'
),
description=_(
'Decode a distribute document stream.'
),
)
op_group = parser.add_mutually_exclusive_group()
op_group.add_argument(
'--sha1',
action='store_true',
help=_('Print SHA-1 value for decryption.'),
)
op_group.add_argument(
'--key',
action='store_true',
help=_('Print decrypted key.'),
)
parser.add_argument(
'--raw',
action='store_true',
help=_('Print raw binary objects as is.'),
)
parser.set_defaults(func=main)
return parser
|