mirror of
https://github.com/TeamPGM/tgcrypto.git
synced 2024-11-22 15:38:12 +00:00
90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
import unittest
|
|
|
|
import tgcrypto
|
|
|
|
|
|
# https://csrc.nist.gov/CSRC/media/Projects/Cryptographic-Standards-and-Guidelines/documents/examples/AES_CTR.pdf
|
|
|
|
class TestCTR256(unittest.TestCase):
|
|
def test_encrypt(self):
|
|
key = bytes.fromhex("""
|
|
603DEB10 15CA71BE 2B73AEF0 857D7781
|
|
1F352C07 3B6108D7 2D9810A3 0914DFF4
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
iv = bytes.fromhex("""
|
|
F0F1F2F3 F4F5F6F7 F8F9FAFB FCFDFEFF
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
plaintext = bytes.fromhex("""
|
|
6BC1BEE2 2E409F96 E93D7E11 7393172A
|
|
AE2D8A57 1E03AC9C 9EB76FAC 45AF8E51
|
|
30C81C46 A35CE411 E5FBC119 1A0A52EF
|
|
F69F2445 DF4F9B17 AD2B417B E66C3710
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
ciphertext = bytes.fromhex("""
|
|
601EC313 775789A5 B7A7F504 BBF3D228
|
|
F443E3CA 4D62B59A CA84E990 CACAF5C5
|
|
2B0930DA A23DE94C E87017BA 2D84988D
|
|
DFC9C58D B67AADA6 13C2DD08 457941A6
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
self.assertEqual(tgcrypto.ctr256_encrypt(plaintext, key, iv, bytes(1)), ciphertext)
|
|
|
|
def test_decrypt(self):
|
|
key = bytes.fromhex("""
|
|
603DEB10 15CA71BE 2B73AEF0 857D7781
|
|
1F352C07 3B6108D7 2D9810A3 0914DFF4
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
iv = bytes.fromhex("""
|
|
F0F1F2F3 F4F5F6F7 F8F9FAFB FCFDFEFF
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
ciphertext = bytes.fromhex("""
|
|
601EC313 775789A5 B7A7F504 BBF3D228
|
|
F443E3CA 4D62B59A CA84E990 CACAF5C5
|
|
2B0930DA A23DE94C E87017BA 2D84988D
|
|
DFC9C58D B67AADA6 13C2DD08 457941A6
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
plaintext = bytes.fromhex("""
|
|
6BC1BEE2 2E409F96 E93D7E11 7393172A
|
|
AE2D8A57 1E03AC9C 9EB76FAC 45AF8E51
|
|
30C81C46 A35CE411 E5FBC119 1A0A52EF
|
|
F69F2445 DF4F9B17 AD2B417B E66C3710
|
|
""".replace(" ", "").replace("\n", ""))
|
|
|
|
self.assertEqual(tgcrypto.ctr256_decrypt(ciphertext, key, iv, bytes(1)), plaintext)
|
|
|
|
# https://github.com/pyca/cryptography/blob/cd4de3ce6dc2a0dd4171b869e187857e4125853b/vectors/cryptography_vectors/ciphers/AES/CTR/aes-256-ctr.txt
|
|
|
|
def test_encrypt_extra1(self):
|
|
key = bytes.fromhex("776BEFF2851DB06F4C8A0542C8696F6C6A81AF1EEC96B4D37FC1D689E6C1C104")
|
|
iv = bytes.fromhex("00000060DB5672C97AA8F0B200000001")
|
|
plaintext = bytes.fromhex("53696E676C6520626C6F636B206D7367")
|
|
ciphertext = bytes.fromhex("145AD01DBF824EC7560863DC71E3E0C0")
|
|
|
|
self.assertEqual(tgcrypto.ctr256_encrypt(plaintext, key, iv, bytes(1)), ciphertext)
|
|
|
|
def test_encrypt_extra2(self):
|
|
key = bytes.fromhex("F6D66D6BD52D59BB0796365879EFF886C66DD51A5B6A99744B50590C87A23884")
|
|
iv = bytes.fromhex("00FAAC24C1585EF15A43D87500000001")
|
|
plaintext = bytes.fromhex("000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F")
|
|
ciphertext = bytes.fromhex("F05E231B3894612C49EE000B804EB2A9B8306B508F839D6A5530831D9344AF1C")
|
|
|
|
self.assertEqual(tgcrypto.ctr256_encrypt(plaintext, key, iv, bytes(1)), ciphertext)
|
|
|
|
def test_encrypt_extra3(self):
|
|
key = bytes.fromhex("FF7A617CE69148E4F1726E2F43581DE2AA62D9F805532EDFF1EED687FB54153D")
|
|
iv = bytes.fromhex("001CC5B751A51D70A1C1114800000001")
|
|
plaintext = bytes.fromhex("000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F20212223")
|
|
ciphertext = bytes.fromhex("EB6C52821D0BBBF7CE7594462ACA4FAAB407DF866569FD07F48CC0B583D6071F1EC0E6B8")
|
|
|
|
self.assertEqual(tgcrypto.ctr256_encrypt(plaintext, key, iv, bytes(1)), ciphertext)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|