Module refinery.units.crypto.cipher.xxtea

Classes

class XXTEA (key, mode, big_endian=False, block_size=8)

Helper class that provides a standard way to create an ABC using inheritance.

Expand source code Browse git

class XXTEA(TEABase):

    def __init__(
        self,
        key: BufferType,
        mode: Optional[CipherMode],
        big_endian: bool = False,
        block_size: int = TEABase.block_size
    ):
        self.block_size = block_size
        super().__init__(key, mode, big_endian)

    @tea_block_operation
    def block_encrypt(key: Sequence[int], v: Sequence[int]) -> Sequence[int]:
        n = len(v)
        s = 0
        r = 6 + 52 // n
        z = v[n - 1]
        for _ in range(r):
            s = s + 0x9E3779B9 & 0xFFFFFFFF
            e = (s >> 2) & 3
            for p in range(n):
                y = v[(p + 1) % n]
                k = (p & 3) ^ e
                x = ((z >> 5) ^ (y << 2)) + ((y >> 3) ^ (z << 4)) ^ (s ^ y) + (key[k] ^ z)
                z = v[p] = v[p] + x & 0xFFFFFFFF
        return v

    @tea_block_operation
    def block_decrypt(key: Sequence[int], v: Sequence[int]) -> Sequence[int]:
        n = len(v)
        r = 6 + 52 // n
        s = r * 0x9E3779B9 & 0xFFFFFFFF
        y = v[0]
        for _ in range(r):
            e = (s >> 2) & 3
            for p in range(n - 1, -1, -1):
                z = v[(p - 1) % n]
                k = (p & 3) ^ e
                x = ((z >> 5) ^ (y << 2)) + ((y >> 3) ^ (z << 4)) ^ (s ^ y) + (key[k] ^ z)
                y = v[p] = v[p] - x & 0xFFFFFFFF
            s = s - 0x9E3779B9 & 0xFFFFFFFF
        return v

Ancestors

• TEABase
• BlockCipher
• CipherInterface
• abc.ABC

Class variables

var derived_key

var big_endian

Methods

def block_encrypt(self, data)

Expand source code Browse git

def wrapped(self: TEABase, data: ByteString) -> ByteString:
    be = self.big_endian
    blocks = list(unpack(data, 4, be))
    blocks = blk(self.derived_key, blocks)
    return pack(blocks, 4, be)

def block_decrypt(self, data)

Expand source code Browse git

def wrapped(self: TEABase, data: ByteString) -> ByteString:
    be = self.big_endian
    blocks = list(unpack(data, 4, be))
    blocks = blk(self.derived_key, blocks)
    return pack(blocks, 4, be)

class xxtea (key, iv=b'', padding=None, mode=None, raw=False, swap=False, block_size=1)

Expand source code Browse git

class xxtea(TEAUnit, cipher=BlockCipherFactory(XXTEA)):

    block_size: int = 8

    def __init__(
        self, key, iv=b'', padding=None, mode=None, raw=False, swap=False,
        block_size: Arg.Number('-b', help=(
            'Cipher block size in 32-bit words. The default value {default} implies that the input '
            'is treated as a single block, which is common behaviour of many implementations.')) = 1
    ):
        super().__init__(key, iv, padding, mode, raw, swap=swap, block_size=block_size)

    def _prepare_block(self, data: bytes):
        if self.args.block_size < 2:
            blocks, remainder = divmod(len(data), 4)
            if remainder:
                blocks += 1
            self.block_size = blocks * 4
        else:
            self.block_size = self.args.block_size * 4

    def encrypt(self, data: bytes) -> bytes:
        self._prepare_block(data)
        return super().encrypt(data)

    def decrypt(self, data: bytes) -> bytes:
        self._prepare_block(data)
        return super().decrypt(data)

    def _new_cipher(self, **optionals) -> CipherInterface:
        return super()._new_cipher(block_size=self.block_size, **optionals)

Ancestors

• TEAUnit
• StandardBlockCipherUnit
• BlockCipherUnitBase
• StandardCipherUnit
• CipherUnit
• Unit
• UnitBase
• Entry

Class variables

var block_size

var key_size

Methods

def encrypt(self, data)

Expand source code Browse git

def encrypt(self, data: bytes) -> bytes:
    self._prepare_block(data)
    return super().encrypt(data)

def decrypt(self, data)

Expand source code Browse git

def decrypt(self, data: bytes) -> bytes:
    self._prepare_block(data)
    return super().decrypt(data)

Inherited members

• TEAUnit:
  • Arg
  • assemble
  • filter
  • finish
  • handles
  • labelled
  • log_debug
  • log_detach
  • log_fail
  • log_info
  • log_level
  • log_warn
  • nozzle
  • process
  • read
  • read1
  • reverse
  • run
  • source
  • superinit