pygost/test_gost34112012.py

   1 # coding: utf-8
   2 # PyGOST -- Pure Python GOST cryptographic functions library
   3 # Copyright (C) 2015-2024 Sergey Matveev <stargrave@stargrave.org>
   4 #
   5 # This program is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, version 3 of the License.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 from os import urandom
  18 from random import randint
  19 from unittest import skip
  20 from unittest import TestCase
  21 import hmac
  22
  23 from pygost import gost34112012256
  24 from pygost import gost34112012512
  25 from pygost.gost34112012256 import GOST34112012256
  26 from pygost.gost34112012512 import GOST34112012512
  27 from pygost.gost34112012512 import pbkdf2
  28 from pygost.utils import hexdec
  29 from pygost.utils import hexenc
  30
  31
  32 class TestCopy(TestCase):
  33     def runTest(self):
  34         m = GOST34112012256()
  35         c = m.copy()
  36         m.update(b"foobar")
  37         c.update(b"foo")
  38         c.update(b"bar")
  39         self.assertSequenceEqual(m.digest(), c.digest())
  40
  41
  42 class TestSymmetric(TestCase):
  43     def runTest(self):
  44         chunks = []
  45         for _ in range(randint(1, 10)):
  46             chunks.append(urandom(randint(20, 80)))
  47         m = GOST34112012256()
  48         for chunk in chunks:
  49             m.update(chunk)
  50         self.assertSequenceEqual(
  51             m.hexdigest(),
  52             GOST34112012256(b"".join(chunks)).hexdigest(),
  53         )
  54
  55
  56 class TestHMAC(TestCase):
  57     """RFC 7836
  58     """
  59     def test_256(self):
  60         for digestmod in (GOST34112012256, gost34112012256):
  61             self.assertSequenceEqual(
  62                 hmac.new(
  63                     key=hexdec("000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f"),
  64                     msg=hexdec("0126bdb87800af214341456563780100"),
  65                     digestmod=digestmod,
  66                 ).hexdigest(),
  67                 "a1aa5f7de402d7b3d323f2991c8d4534013137010a83754fd0af6d7cd4922ed9",
  68             )
  69
  70     def test_512(self):
  71         for digestmod in (GOST34112012512, gost34112012512):
  72             self.assertSequenceEqual(
  73                 hmac.new(
  74                     key=hexdec("000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f"),
  75                     msg=hexdec("0126bdb87800af214341456563780100"),
  76                     digestmod=digestmod,
  77                 ).hexdigest(),
  78                 "a59bab22ecae19c65fbde6e5f4e9f5d8549d31f037f9df9b905500e171923a773d5f1530f2ed7e964cb2eedc29e9ad2f3afe93b2814f79f5000ffc0366c251e6",
  79             )
  80
  81
  82 class TestVectors(TestCase):
  83     def test_m1(self):
  84         m = hexdec("323130393837363534333231303938373635343332313039383736353433323130393837363534333231303938373635343332313039383736353433323130")[::-1]
  85         self.assertSequenceEqual(
  86             GOST34112012512(m).digest(),
  87             hexdec("486f64c1917879417fef082b3381a4e211c324f074654c38823a7b76f830ad00fa1fbae42b1285c0352f227524bc9ab16254288dd6863dccd5b9f54a1ad0541b")[::-1]
  88         )
  89         self.assertSequenceEqual(
  90             GOST34112012256(m).digest(),
  91             hexdec("00557be5e584fd52a449b16b0251d05d27f94ab76cbaa6da890b59d8ef1e159d")[::-1]
  92         )
  93
  94     def test_m2(self):
  95         m = u"Се ветри, Стрибожи внуци, веютъ с моря стрелами на храбрыя плъкы Игоревы".encode("cp1251")
  96         self.assertSequenceEqual(m, hexdec("fbe2e5f0eee3c820fbeafaebef20fffbf0e1e0f0f520e0ed20e8ece0ebe5f0f2f120fff0eeec20f120faf2fee5e2202ce8f6f3ede220e8e6eee1e8f0f2d1202ce8f0f2e5e220e5d1")[::-1])
  97         self.assertSequenceEqual(
  98             GOST34112012512(m).digest(),
  99             hexdec("28fbc9bada033b1460642bdcddb90c3fb3e56c497ccd0f62b8a2ad4935e85f037613966de4ee00531ae60f3b5a47f8dae06915d5f2f194996fcabf2622e6881e")[::-1]
 100         )
 101         self.assertSequenceEqual(
 102             GOST34112012256(m).digest(),
 103             hexdec("508f7e553c06501d749a66fc28c6cac0b005746d97537fa85d9e40904efed29d")[::-1]
 104         )
 105
 106     def test_habr144(self):
 107         """Test vector from https://habr.com/ru/post/452200/
 108         """
 109         m = hexdec("d0cf11e0a1b11ae1000000000000000000000000000000003e000300feff0900060000000000000000000000010000000100000000000000001000002400000001000000feffffff0000000000000000ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff")
 110         self.assertSequenceEqual(
 111             GOST34112012256(m).hexdigest(),
 112             "c766085540caaa8953bfcf7a1ba220619cee50d65dc242f82f23ba4b180b18e0",
 113         )
 114
 115
 116 class TestPBKDF2(TestCase):
 117     """http://tc26.ru/.../R_50.1.111-2016.pdf
 118     """
 119     def test_1(self):
 120         self.assertSequenceEqual(
 121             hexenc(pbkdf2(b"password", b"salt", 1, 64)),
 122             "64770af7f748c3b1c9ac831dbcfd85c26111b30a8a657ddc3056b80ca73e040d2854fd36811f6d825cc4ab66ec0a68a490a9e5cf5156b3a2b7eecddbf9a16b47",
 123         )
 124
 125     def test_2(self):
 126         self.assertSequenceEqual(
 127             hexenc(pbkdf2(b"password", b"salt", 2, 64)),
 128             "5a585bafdfbb6e8830d6d68aa3b43ac00d2e4aebce01c9b31c2caed56f0236d4d34b2b8fbd2c4e89d54d46f50e47d45bbac301571743119e8d3c42ba66d348de",
 129         )
 130
 131     def test_3(self):
 132         self.assertSequenceEqual(
 133             hexenc(pbkdf2(b"password", b"salt", 4096, 64)),
 134             "e52deb9a2d2aaff4e2ac9d47a41f34c20376591c67807f0477e32549dc341bc7867c09841b6d58e29d0347c996301d55df0d34e47cf68f4e3c2cdaf1d9ab86c3",
 135         )
 136
 137     @skip("it takes too long")
 138     def test_4(self):
 139         self.assertSequenceEqual(
 140             hexenc(pbkdf2(b"password", b"salt", 16777216, 64)),
 141             "49e4843bba76e300afe24c4d23dc7392def12f2c0e244172367cd70a8982ac361adb601c7e2a314e8cb7b1e9df840e36ab5615be5d742b6cf203fb55fdc48071",
 142         )
 143
 144     def test_5(self):
 145         self.assertSequenceEqual(
 146             hexenc(pbkdf2(
 147                 b"passwordPASSWORDpassword",
 148                 b"saltSALTsaltSALTsaltSALTsaltSALTsalt",
 149                 4096,
 150                 100,
 151             )),
 152             "b2d8f1245fc4d29274802057e4b54e0a0753aa22fc53760b301cf008679e58fe4bee9addcae99ba2b0b20f431a9c5e50f395c89387d0945aedeca6eb4015dfc2bd2421ee9bb71183ba882ceebfef259f33f9e27dc6178cb89dc37428cf9cc52a2baa2d3a",
 153         )
 154
 155     def test_6(self):
 156         self.assertSequenceEqual(
 157             hexenc(pbkdf2(b"pass\x00word", b"sa\x00lt", 4096, 64)),
 158             "50df062885b69801a3c10248eb0a27ab6e522ffeb20c991c660f001475d73a4e167f782c18e97e92976d9c1d970831ea78ccb879f67068cdac1910740844e830",
 159         )