purepython-aes 0.5.0__tar.gz → 0.6.0__tar.gz

{purepython_aes-0.5.0/src/purepython_aes.egg-info → purepython_aes-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: purepython-aes
-Version: 0.5.0
+Version: 0.6.0
 Summary: Pure-Python implementation of AES encryption algorithm.
 Author-email: Emelianov Artem <penguin5726@proton.me>
 License-Expression: MIT

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes/__init__.py

@@ -10,6 +10,7 @@ from purepython_aes.aes import (
     CbcMode,
     CfbMode,
     CipherMode,
+    CtrMode,
     EcbMode,
     Iso7816Padding,
     Iso10126Padding,
@@ -36,6 +37,7 @@ __all__: list[str] = [
     'CbcMode',
     'CfbMode',
     'CipherMode',
+    'CtrMode',
     'EcbMode',
     'Iso7816Padding',
     'Iso10126Padding',

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes/aes/__init__.py

@@ -13,6 +13,7 @@ from purepython_aes.aes.modes import (
     CbcMode,
     CfbMode,
     CipherMode,
+    CtrMode,
     EcbMode,
     OfbMode,
     PcbcMode,
@@ -41,6 +42,7 @@ __all__: list[str] = [
     'CbcMode',
     'CfbMode',
     'CipherMode',
+    'CtrMode',
     'EcbMode',
     'OfbMode',
     'PcbcMode',

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes/aes/modes/__init__.py

@@ -1,6 +1,6 @@
 from purepython_aes.aes.modes._base import AesMode, CipherMode
 from purepython_aes.aes.modes.block import BlockCipherMode, CbcMode, EcbMode, PcbcMode
-from purepython_aes.aes.modes.stream import CfbMode, OfbMode, StreamCipherMode
+from purepython_aes.aes.modes.stream import CfbMode, CtrMode, OfbMode, StreamCipherMode
 
 __all__: list[str] = [
     'AesMode',
@@ -10,6 +10,7 @@ __all__: list[str] = [
     'EcbMode',
     'PcbcMode',
     'CfbMode',
+    'CtrMode',
     'OfbMode',
     'StreamCipherMode',
 ]

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes/aes/modes/stream/__init__.py

@@ -1,5 +1,6 @@
 from purepython_aes.aes.modes.stream._base import StreamCipherMode
 from purepython_aes.aes.modes.stream.cfb import CfbMode
+from purepython_aes.aes.modes.stream.ctr import CtrMode
 from purepython_aes.aes.modes.stream.ofb import OfbMode
 
-__all__: list[str] = ['StreamCipherMode', 'CfbMode', 'OfbMode']
+__all__: list[str] = ['StreamCipherMode', 'CfbMode', 'CtrMode', 'OfbMode']

purepython_aes-0.6.0/src/purepython_aes/aes/modes/stream/ctr.py

@@ -0,0 +1,50 @@
+# src/purepython_aes/aes/modes/stream/ctr.py
+from dataclasses import dataclass
+from typing import final
+
+from purepython_aes.aes.modes.operations import xor
+from purepython_aes.aes.modes.stream._base import StreamCipherMode
+from purepython_aes.const import AES_BLOCK_SIZE
+
+
+@final
+@dataclass(slots=True)
+class CtrMode(StreamCipherMode):
+    """Counter mode of operation."""
+
+    def __encrypt_stream__(
+        self,
+        initialization_value: bytes,
+        plaintext: bytes,
+    ) -> bytes:
+        return self.transform(initialization_value, plaintext)
+
+    def __decrypt_stream__(
+        self,
+        initialization_value: bytes,
+        ciphertext: bytes,
+    ) -> bytes:
+        return self.transform(initialization_value, ciphertext)
+
+    def transform(
+        self,
+        initialization_value: bytes,
+        data: bytes,
+    ) -> bytes:
+        """XOR data with the counter-generated keystream."""
+
+        block_count: int = (len(data) + AES_BLOCK_SIZE - 1) // AES_BLOCK_SIZE
+        initial_counter: int = int.from_bytes(initialization_value, byteorder='big')
+        transformed_data: bytearray = bytearray()
+        for block_index in range(block_count):
+            block_start: int = block_index * AES_BLOCK_SIZE
+            data_block: bytes = data[block_start : (block_start + AES_BLOCK_SIZE)]
+            counter: int = initial_counter + block_index
+            counter_block: bytes = counter.to_bytes(AES_BLOCK_SIZE, byteorder='big')
+            keystream_block: bytes = self.algorithm.encrypt_block(counter_block)
+            transformed_block: bytes = xor(
+                data_block,
+                keystream_block[: len(data_block)],
+            )
+            transformed_data.extend(transformed_block)
+        return bytes(transformed_data)

{purepython_aes-0.5.0 → purepython_aes-0.6.0/src/purepython_aes.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: purepython-aes
-Version: 0.5.0
+Version: 0.6.0
 Summary: Pure-Python implementation of AES encryption algorithm.
 Author-email: Emelianov Artem <penguin5726@proton.me>
 License-Expression: MIT

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes.egg-info/SOURCES.txt

@@ -47,6 +47,7 @@ src/purepython_aes/aes/modes/block/pcbc.py
 src/purepython_aes/aes/modes/stream/__init__.py
 src/purepython_aes/aes/modes/stream/_base.py
 src/purepython_aes/aes/modes/stream/cfb.py
+src/purepython_aes/aes/modes/stream/ctr.py
 src/purepython_aes/aes/modes/stream/ofb.py
 src/purepython_aes/aes/padding/__init__.py
 src/purepython_aes/aes/padding/_base.py
@@ -99,6 +100,7 @@ tests/unit/aes/modes/block/test_ecb.py
 tests/unit/aes/modes/block/test_pcbc.py
 tests/unit/aes/modes/stream/__init__.py
 tests/unit/aes/modes/stream/test_cfb.py
+tests/unit/aes/modes/stream/test_ctr.py
 tests/unit/aes/modes/stream/test_ofb.py
 tests/unit/aes/padding/__init__.py
 tests/unit/aes/padding/conftest.py

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes.egg-info/scm_file_list.json

@@ -48,6 +48,7 @@
     "src/purepython_aes/aes/modes/__init__.py",
     "src/purepython_aes/aes/modes/operations.py",
     "src/purepython_aes/aes/modes/stream/_base.py",
+    "src/purepython_aes/aes/modes/stream/ctr.py",
     "src/purepython_aes/aes/modes/stream/cfb.py",
     "src/purepython_aes/aes/modes/stream/__init__.py",
     "src/purepython_aes/aes/modes/stream/ofb.py",
@@ -100,6 +101,7 @@
     "tests/unit/aes/modes/stream/test_cfb.py",
     "tests/unit/aes/modes/stream/__init__.py",
     "tests/unit/aes/modes/stream/test_ofb.py",
+    "tests/unit/aes/modes/stream/test_ctr.py",
     "tests/unit/aes/modes/block/__init__.py",
     "tests/unit/aes/modes/block/test_pcbc.py",
     "tests/unit/aes/modes/block/test_ecb.py",

{purepython_aes-0.5.0 → purepython_aes-0.6.0}/src/purepython_aes.egg-info/scm_version.json

@@ -1,7 +1,7 @@
 {
-  "tag": "0.5.0",
+  "tag": "0.6.0",
   "distance": 0,
-  "node": "g393fbd90ca44525a385004d5d26622a21b782356",
+  "node": "g0b685541a28a77cde6c306ae0d3077cfeb784035",
   "dirty": false,
   "branch": "HEAD",
   "node_date": "2026-07-14"

purepython_aes-0.6.0/tests/unit/aes/modes/stream/test_ctr.py

@@ -0,0 +1,28 @@
+from hypothesis import given
+from hypothesis.strategies import binary
+from pytest import mark
+
+from purepython_aes import (
+    Aes,
+    Aes128,
+    Aes192,
+    Aes256,
+    CtrMode,
+    ReferenceAes128,
+    ReferenceAes192,
+    ReferenceAes256,
+)
+from tests.unit.aes.strategies import aes256key
+
+
+@mark.quick
+class TestCtrMode:
+    @staticmethod
+    @mark.parametrize(
+        'aes',
+        [Aes128, Aes192, Aes256, ReferenceAes128, ReferenceAes192, ReferenceAes256],
+    )
+    @given(key=aes256key, data=binary())
+    def test_decrypt_inverts_encrypt(aes: type[Aes], key: bytes, data: bytes) -> None:
+        ctr: CtrMode = CtrMode(aes(key[:aes.__key_size__]))  # fmt: skip
+        assert ctr.decrypt(ctr.encrypt(data)) == data