LeoVasanko/pyaegis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add human-readable algorithm name as NAME constant.

Browse Source
This commit is contained in:
Leo Vasanko
2025-11-08 16:01:05 -06:00
parent 13445887e9
commit e58990a1c2
8 changed files with 29 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -95,7 +95,7 @@ Useful for creating pseudo random bytes as rapidly as possible. Reuse of the sam
### Miscellaneous
Constants (per module): KEYBYTES, NONCEBYTES, MACBYTES, MACBYTES_LONG, RATE, ALIGNMENT
Constants (per module): NAME, KEYBYTES, NONCEBYTES, MACBYTES, MACBYTES_LONG, RATE, ALIGNMENT
- random_key() -> bytearray (length KEYBYTES)
- random_nonce() -> bytearray (length NONCEBYTES)

2
pyaegis/aegis128l.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-128L" #: Algorithm name
KEYBYTES = 16 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 16 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

2
pyaegis/aegis128x2.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-128X2" #: Algorithm name
KEYBYTES = 16 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 16 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

2
pyaegis/aegis128x4.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-128X4" #: Algorithm name
KEYBYTES = 16 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 16 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

2
pyaegis/aegis256.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-256" #: Algorithm name
KEYBYTES = 32 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 32 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

2
pyaegis/aegis256x2.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-256X2" #: Algorithm name
KEYBYTES = 32 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 32 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

2
pyaegis/aegis256x4.py
View File

@@ -9,6 +9,7 @@ from ._loader import ffi
from ._loader import lib as _lib
from .util import Buffer, new_aligned_struct, nonce_increment, wipe
NAME = "AEGIS-256X4" #: Algorithm name
KEYBYTES = 32 #: Key size in bytes (varies by algorithm)
NONCEBYTES = 32 #: Nonce size in bytes (varies by algorithm)
MACBYTES = 16 #: Normal MAC size (always 16)
@@ -797,6 +798,7 @@ def new_mac_state():
__all__ = [
# constants
"NAME",
"KEYBYTES",
"NONCEBYTES",
"MACBYTES",

41
pyaegis/benchmark.py
View File

@@ -17,21 +17,12 @@ from pyaegis import aegis128l, aegis128x2, aegis128x4, aegis256, aegis256x2, aeg
MSG_LEN = 16384000 # 16 000 KiB
ITERATIONS = 100
ALGORITHMS = [
("AEGIS-128L", aegis128l),
("AEGIS-128X2", aegis128x2),
("AEGIS-128X4", aegis128x4),
("AEGIS-256", aegis256),
("AEGIS-256X2", aegis256x2),
("AEGIS-256X4", aegis256x4),
]
def _random_bytes(n: int) -> bytes:
return os.urandom(n)
def bench_encrypt(alg_name: str, ciph) -> None:
def bench_encrypt(ciph) -> None:
key = _random_bytes(ciph.KEYBYTES)
nonce = _random_bytes(ciph.NONCEBYTES)
@@ -56,10 +47,10 @@ def bench_encrypt(alg_name: str, ciph) -> None:
throughput_mbps = (
(bits / (elapsed_s * 1_000_000)) if elapsed_s > 0 else float("inf")
)
print(f"{alg_name}\t{throughput_mbps:10.2f} Mb/s")
print(f"{ciph.NAME}\t{throughput_mbps:10.2f} Mb/s")
def bench_mac(alg_name: str, ciph) -> None:
def bench_mac(ciph) -> None:
key = _random_bytes(ciph.KEYBYTES)
nonce = _random_bytes(ciph.NONCEBYTES)
@@ -83,23 +74,23 @@ def bench_mac(alg_name: str, ciph) -> None:
throughput_mbps = (
(bits / (elapsed_s * 1_000_000)) if elapsed_s > 0 else float("inf")
)
print(f"{alg_name} MAC\t{throughput_mbps:10.2f} Mb/s")
print(f"{ciph.NAME} MAC\t{throughput_mbps:10.2f} Mb/s")
if __name__ == "__main__":
# aegis_init() is called in the loader at import time already
# Run encrypt benchmarks in order: 256, 256x2, 256x4, 128l, 128x2, 128x4
bench_encrypt("AEGIS-256", aegis256)
bench_encrypt("AEGIS-256X2", aegis256x2)
bench_encrypt("AEGIS-256X4", aegis256x4)
bench_encrypt("AEGIS-128L", aegis128l)
bench_encrypt("AEGIS-128X2", aegis128x2)
bench_encrypt("AEGIS-128X4", aegis128x4)
bench_encrypt(aegis256)
bench_encrypt(aegis256x2)
bench_encrypt(aegis256x4)
bench_encrypt(aegis128l)
bench_encrypt(aegis128x2)
bench_encrypt(aegis128x4)
# Run MAC benchmarks in order: 128l, 128x2, 128x4, 256, 256x2, 256x4
bench_mac("AEGIS-128L", aegis128l)
bench_mac("AEGIS-128X2", aegis128x2)
bench_mac("AEGIS-128X4", aegis128x4)
bench_mac("AEGIS-256", aegis256)
bench_mac("AEGIS-256X2", aegis256x2)
bench_mac("AEGIS-256X4", aegis256x4)
bench_mac(aegis128l)
bench_mac(aegis128x2)
bench_mac(aegis128x4)
bench_mac(aegis256)
bench_mac(aegis256x2)
bench_mac(aegis256x4)