| 123456789101112131415161718192021222324252627282930313233343536373839 |
- #-----------------------------------------------------------------------------
- # Copyright (c) 2005-2021, PyInstaller Development Team.
- #
- # Distributed under the terms of the GNU General Public License (version 2
- # or later) with exception for distributing the bootloader.
- #
- # The full license is in the file COPYING.txt, distributed with this software.
- #
- # SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception)
- #-----------------------------------------------------------------------------
- import os
- BLOCK_SIZE = 16
- class PyiBlockCipher(object):
- """
- This class is used only to encrypt Python modules.
- """
- def __init__(self, key=None):
- assert type(key) is str
- if len(key) > BLOCK_SIZE:
- self.key = key[0:BLOCK_SIZE]
- else:
- self.key = key.zfill(BLOCK_SIZE)
- assert len(self.key) == BLOCK_SIZE
- import tinyaes
- self._aesmod = tinyaes
- def encrypt(self, data):
- iv = os.urandom(BLOCK_SIZE)
- return iv + self.__create_cipher(iv).CTR_xcrypt_buffer(data)
- def __create_cipher(self, iv):
- # The 'AES' class is stateful, and this factory method is used to re-initialize the block cipher class with
- # each call to xcrypt().
- return self._aesmod.AES(self.key.encode(), iv)
|
