TgCrypto benchmark

tgcrypto_bench.py

# https://github.com/pyrogram/tgcrypto

import os
import time

import tgcrypto


def fmt(d, s):
    return f"{len(d) / (time.time() - s) / 1024 / 1024 :.2f} MB/s"


data = os.urandom(50 * 1024 * 1024)
key = os.urandom(32)
iv = os.urandom(32)

start = time.time()
tgcrypto.ige256_encrypt(data, key, iv)
print(f"IGE Enc: {fmt(data, start)}")

start = time.time()
tgcrypto.ige256_decrypt(data, key, iv)
print(f"IGE Dec: {fmt(data, start)}")

print()

start = time.time()
tgcrypto.ctr256_encrypt(data, key, iv[:16], bytes(1))
print(f"CTR Enc: {fmt(data, start)}")

start = time.time()
tgcrypto.ctr256_decrypt(data, key, iv[:16], bytes(1))
print(f"CTR Dec: {fmt(data, start)}")

print()

start = time.time()
tgcrypto.cbc256_encrypt(data, key, iv[:16])
print(f"CBC Enc: {fmt(data, start)}")

start = time.time()
tgcrypto.cbc256_decrypt(data, key, iv[:16])
print(f"CBC Dec: {fmt(data, start)}")