]> Cypherpunks.ru repositories - pygost.git/blobdiff - pygost/test_gost3413.py
CTR-ACPKM, OMAC-ACPKM-Master modes of operation
[pygost.git] / pygost / test_gost3413.py
index 984bb76d17f00138a6e8551b5d2a6615fce65141..6bf2f885ec74b68d8ccc6bf9c9f1408bd0b83f89 100644 (file)
@@ -21,14 +21,19 @@ from unittest import TestCase
 from pygost.gost3412 import GOST3412Kuznechik
 from pygost.gost3412 import GOST3412Magma
 from pygost.gost3413 import _mac_ks
+from pygost.gost3413 import acpkm
+from pygost.gost3413 import acpkm_master
 from pygost.gost3413 import cbc_decrypt
 from pygost.gost3413 import cbc_encrypt
 from pygost.gost3413 import cfb_decrypt
 from pygost.gost3413 import cfb_encrypt
 from pygost.gost3413 import ctr
+from pygost.gost3413 import ctr_acpkm
 from pygost.gost3413 import ecb_decrypt
 from pygost.gost3413 import ecb_encrypt
+from pygost.gost3413 import KEY_SIZE
 from pygost.gost3413 import mac
+from pygost.gost3413 import mac_acpkm_master
 from pygost.gost3413 import ofb
 from pygost.gost3413 import pad2
 from pygost.gost3413 import unpad2
@@ -343,3 +348,233 @@ class GOST3412MagmaModesTest(TestCase):
             data = urandom(randint(0, 16 * 2))
             ciph = GOST3412Magma(urandom(32))
             mac(ciph.encrypt, 8, data)
+
+
+class TestVectorACPKM(TestCase):
+    """Test vectors from Р 1323565.1.017-2018
+    """
+    key = hexdec("8899AABBCCDDEEFF0011223344556677FEDCBA98765432100123456789ABCDEF")
+
+    def test_magma_ctr_acpkm(self):
+        key = acpkm(GOST3412Magma(self.key).encrypt, 8)
+        self.assertSequenceEqual(key, hexdec("863EA017842C3D372B18A85A28E2317D74BEFC107720DE0C9E8AB974ABD00CA0"))
+        key = acpkm(GOST3412Magma(key).encrypt, 8)
+        self.assertSequenceEqual(key, hexdec("49A5E2677DE555982B8AD5E826652D17EEC847BF5B3997A81CF7FE7F1187BD27"))
+        key = acpkm(GOST3412Magma(key).encrypt, 8)
+        self.assertSequenceEqual(key, hexdec("3256BF3F97B5667426A9FB1C5EAABE41893CCDD5A868F9B63B0AA90720FA43C4"))
+
+    def test_magma_ctr(self):
+        encrypter = GOST3412Magma(self.key).encrypt
+        plaintext = hexdec("""
+11 22 33 44 55 66 77 00 FF EE DD CC BB AA 99 88
+00 11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A
+11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A 00
+22 33 44 55 66 77 88 99
+        """.replace("\n", "").replace(" ", ""))
+        iv = hexdec("12345678")
+        ciphertext = hexdec("""
+2A B8 1D EE EB 1E 4C AB 68 E1 04 C4 BD 6B 94 EA
+C7 2C 67 AF 6C 2E 5B 6B 0E AF B6 17 70 F1 B3 2E
+A1 AE 71 14 9E ED 13 82 AB D4 67 18 06 72 EC 6F
+84 A2 F1 5B 3F CA 72 C1
+        """.replace("\n", "").replace(" ", ""))
+        self.assertSequenceEqual(
+            ctr_acpkm(GOST3412Magma, encrypter, bs=8, section_size=16, data=plaintext, iv=iv),
+            ciphertext,
+        )
+        self.assertSequenceEqual(
+            ctr_acpkm(GOST3412Magma, encrypter, bs=8, section_size=16, data=ciphertext, iv=iv),
+            plaintext,
+        )
+
+    def test_kuznechik_ctr_acpkm(self):
+        key = acpkm(GOST3412Kuznechik(self.key).encrypt, 16)
+        self.assertSequenceEqual(key, hexdec("2666ED40AE687811745CA0B448F57A7B390ADB5780307E8E9659AC403AE60C60"))
+        key = acpkm(GOST3412Kuznechik(key).encrypt, 16)
+        self.assertSequenceEqual(key, hexdec("BB3DD5402E999B7A3DEBB0DB45448EC530F07365DFEE3ABA8415F77AC8F34CE8"))
+        key = acpkm(GOST3412Kuznechik(key).encrypt, 16)
+        self.assertSequenceEqual(key, hexdec("23362FD553CAD2178299A5B5A2D4722E3BB83C730A8BF57CE2DD004017F8C565"))
+
+    def test_kuznechik_ctr(self):
+        encrypter = GOST3412Kuznechik(self.key).encrypt
+        iv = hexdec("1234567890ABCEF0")
+        plaintext = hexdec("""
+11 22 33 44 55 66 77 00 FF EE DD CC BB AA 99 88
+00 11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A
+11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A 00
+22 33 44 55 66 77 88 99 AA BB CC EE FF 0A 00 11
+33 44 55 66 77 88 99 AA BB CC EE FF 0A 00 11 22
+44 55 66 77 88 99 AA BB CC EE FF 0A 00 11 22 33
+55 66 77 88 99 AA BB CC EE FF 0A 00 11 22 33 44
+        """.replace("\n", "").replace(" ", ""))
+        ciphertext = hexdec("""
+F1 95 D8 BE C1 0E D1 DB D5 7B 5F A2 40 BD A1 B8
+85 EE E7 33 F6 A1 3E 5D F3 3C E4 B3 3C 45 DE E4
+4B CE EB 8F 64 6F 4C 55 00 17 06 27 5E 85 E8 00
+58 7C 4D F5 68 D0 94 39 3E 48 34 AF D0 80 50 46
+CF 30 F5 76 86 AE EC E1 1C FC 6C 31 6B 8A 89 6E
+DF FD 07 EC 81 36 36 46 0C 4F 3B 74 34 23 16 3E
+64 09 A9 C2 82 FA C8 D4 69 D2 21 E7 FB D6 DE 5D
+        """.replace("\n", "").replace(" ", ""))
+        self.assertSequenceEqual(
+            ctr_acpkm(
+                GOST3412Kuznechik,
+                encrypter,
+                bs=16,
+                section_size=32,
+                data=plaintext,
+                iv=iv,
+            ),
+            ciphertext,
+        )
+        self.assertSequenceEqual(
+            ctr_acpkm(
+                GOST3412Kuznechik,
+                encrypter,
+                bs=16,
+                section_size=32,
+                data=ciphertext,
+                iv=iv,
+            ),
+            plaintext,
+        )
+
+    def test_magma_omac_1_5_blocks(self):
+        encrypter = GOST3412Magma(self.key).encrypt
+        key_section_size = 640 // 8
+        self.assertSequenceEqual(
+            acpkm_master(
+                GOST3412Magma,
+                encrypter,
+                key_section_size=key_section_size,
+                bs=8,
+                keymat_len=KEY_SIZE + 8,
+            ),
+            hexdec("0DF2F5273DA328932AC49D81D36B2558A50DBF9BBCAC74A614B2CCB2F1CBCD8A70638E3DE8B3571E"),
+        )
+        text = hexdec("1122334455667700FFEEDDCC")
+        self.assertSequenceEqual(
+            mac_acpkm_master(
+                GOST3412Magma,
+                encrypter,
+                key_section_size,
+                section_size=16,
+                bs=8,
+                data=text,
+            ),
+            hexdec("A0540E3730ACBCF3"),
+        )
+
+    def test_magma_omac_5_blocks(self):
+        encrypter = GOST3412Magma(self.key).encrypt
+        key_section_size = 640 // 8
+        self.assertSequenceEqual(
+            acpkm_master(
+                GOST3412Magma,
+                encrypter,
+                key_section_size=key_section_size,
+                bs=8,
+                keymat_len=3 * (KEY_SIZE + 8),
+            ),
+            hexdec("""
+0D F2 F5 27 3D A3 28 93 2A C4 9D 81 D3 6B 25 58
+A5 0D BF 9B BC AC 74 A6 14 B2 CC B2 F1 CB CD 8A
+70 63 8E 3D E8 B3 57 1E 8D 38 26 D5 5E 63 A1 67
+E2 40 66 40 54 7B 9F 1F 5F 2B 43 61 2A AE AF DA
+18 0B AC 86 04 DF A6 FE 53 C2 CE 27 0E 9C 9F 52
+68 D0 FD BF E1 A3 BD D9 BE 5B 96 D0 A1 20 23 48
+6E F1 71 0F 92 4A E0 31 30 52 CB 5F CA 0B 79 1E
+1B AB E8 57 6D 0F E3 A8
+            """.replace("\n", "").replace(" ", "")),
+        )
+        text = hexdec("""
+11 22 33 44 55 66 77 00 FF EE DD CC BB AA 99 88
+00 11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A
+11 22 33 44 55 66 77 88
+        """.replace("\n", "").replace(" ", ""))
+        self.assertSequenceEqual(
+            mac_acpkm_master(
+                GOST3412Magma,
+                encrypter,
+                key_section_size,
+                section_size=16,
+                bs=8,
+                data=text,
+            ),
+            hexdec("34008DAD5496BB8E"),
+        )
+
+    def test_kuznechik_omac_1_5_blocks(self):
+        encrypter = GOST3412Kuznechik(self.key).encrypt
+        key_section_size = 768 // 8
+        self.assertSequenceEqual(
+            acpkm_master(
+                GOST3412Kuznechik,
+                encrypter,
+                key_section_size=key_section_size,
+                bs=16,
+                keymat_len=KEY_SIZE + 16,
+            ),
+            hexdec("""
+0C AB F1 F2 EF BC 4A C1 60 48 DF 1A 24 C6 05 B2
+C0 D1 67 3D 75 86 A8 EC 0D D4 2C 45 A4 F9 5B AE
+0F 2E 26 17 E4 71 48 68 0F C3 E6 17 8D F2 C1 37
+            """.replace("\n", "").replace(" ", ""))
+        )
+        text = hexdec("""
+11 22 33 44 55 66 77 00 FF EE DD CC BB AA 99 88
+00 11 22 33 44 55 66 77
+        """.replace("\n", "").replace(" ", ""))
+        self.assertSequenceEqual(
+            mac_acpkm_master(
+                GOST3412Kuznechik,
+                encrypter,
+                key_section_size,
+                section_size=32,
+                bs=16,
+                data=text,
+            ),
+            hexdec("B5367F47B62B995EEB2A648C5843145E"),
+        )
+
+    def test_kuznechik_omac_5_blocks(self):
+        encrypter = GOST3412Kuznechik(self.key).encrypt
+        key_section_size = 768 // 8
+        self.assertSequenceEqual(
+            acpkm_master(
+                GOST3412Kuznechik,
+                encrypter,
+                key_section_size=key_section_size,
+                bs=16,
+                keymat_len=3 * (KEY_SIZE + 16),
+            ),
+            hexdec("""
+0C AB F1 F2 EF BC 4A C1 60 48 DF 1A 24 C6 05 B2
+C0 D1 67 3D 75 86 A8 EC 0D D4 2C 45 A4 F9 5B AE
+0F 2E 26 17 E4 71 48 68 0F C3 E6 17 8D F2 C1 37
+C9 DD A8 9C FF A4 91 FE AD D9 B3 EA B7 03 BB 31
+BC 7E 92 7F 04 94 72 9F 51 B4 9D 3D F9 C9 46 08
+00 FB BC F5 ED EE 61 0E A0 2F 01 09 3C 7B C7 42
+D7 D6 27 15 01 B1 77 77 52 63 C2 A3 49 5A 83 18
+A8 1C 79 A0 4F 29 66 0E A3 FD A8 74 C6 30 79 9E
+14 2C 57 79 14 FE A9 0D 3B C2 50 2E 83 36 85 D9
+            """.replace("\n", "").replace(" ", "")),
+        )
+        text = hexdec("""
+11 22 33 44 55 66 77 00 FF EE DD CC BB AA 99 88
+00 11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A
+11 22 33 44 55 66 77 88 99 AA BB CC EE FF 0A 00
+22 33 44 55 66 77 88 99 AA BB CC EE FF 0A 00 11
+33 44 55 66 77 88 99 AA BB CC EE FF 0A 00 11 22
+        """.replace("\n", "").replace(" ", ""))
+        self.assertSequenceEqual(
+            mac_acpkm_master(
+                GOST3412Kuznechik,
+                encrypter,
+                key_section_size,
+                section_size=32,
+                bs=16,
+                data=text,
+            ),
+            hexdec("FBB8DCEE45BEA67C35F58C5700898E5D"),
+        )