pygost/test_gost3411_2012.py

   1 # coding: utf-8
   2 # PyGOST -- Pure Python GOST cryptographic functions library
   3 # Copyright (C) 2015-2016 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, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 from unittest import TestCase
  19 import hmac
  20
  21 from pygost import gost3411_2012_256
  22 from pygost import gost3411_2012_512
  23 from pygost.gost3411_2012_256 import GOST34112012256
  24 from pygost.gost3411_2012_512 import GOST34112012512
  25 from pygost.utils import hexdec
  26
  27
  28 class TestCopy(TestCase):
  29     def runTest(self):
  30         m = GOST34112012256()
  31         c = m.copy()
  32         m.update(b'foobar')
  33         c.update(b'foo')
  34         c.update(b'bar')
  35         self.assertEqual(m.digest(), c.digest())
  36
  37
  38 class TestHMAC(TestCase):
  39     """http://tc26.ru/methods/recommendation/%D0%A2%D0%9A26%D0%90%D0%9B%D0%93.pdf test vectors
  40     """
  41     def test_256(self):
  42         for digestmod in (GOST34112012256, gost3411_2012_256):
  43             self.assertEqual(
  44                 hmac.new(
  45                     key=hexdec("000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f"),
  46                     msg=hexdec("0126bdb87800af214341456563780100"),
  47                     digestmod=digestmod,
  48                 ).hexdigest(),
  49                 "a1aa5f7de402d7b3d323f2991c8d4534013137010a83754fd0af6d7cd4922ed9",
  50             )
  51
  52     def test_512(self):
  53         for digestmod in (GOST34112012512, gost3411_2012_512):
  54             self.assertEqual(
  55                 hmac.new(
  56                     key=hexdec("000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f"),
  57                     msg=hexdec("0126bdb87800af214341456563780100"),
  58                     digestmod=digestmod,
  59                 ).hexdigest(),
  60                 "a59bab22ecae19c65fbde6e5f4e9f5d8549d31f037f9df9b905500e171923a773d5f1530f2ed7e964cb2eedc29e9ad2f3afe93b2814f79f5000ffc0366c251e6",
  61             )
  62
  63
  64 class TestVectors(TestCase):
  65     def test_m1(self):
  66         m = hexdec("323130393837363534333231303938373635343332313039383736353433323130393837363534333231303938373635343332313039383736353433323130")[::-1]
  67         self.assertEqual(
  68             GOST34112012512(m).digest(),
  69             hexdec("486f64c1917879417fef082b3381a4e211c324f074654c38823a7b76f830ad00fa1fbae42b1285c0352f227524bc9ab16254288dd6863dccd5b9f54a1ad0541b")[::-1]
  70         )
  71         self.assertEqual(
  72             GOST34112012256(m).digest(),
  73             hexdec("00557be5e584fd52a449b16b0251d05d27f94ab76cbaa6da890b59d8ef1e159d")[::-1]
  74         )
  75
  76     def test_m2(self):
  77         m = hexdec("fbe2e5f0eee3c820fbeafaebef20fffbf0e1e0f0f520e0ed20e8ece0ebe5f0f2f120fff0eeec20f120faf2fee5e2202ce8f6f3ede220e8e6eee1e8f0f2d1202ce8f0f2e5e220e5d1")[::-1]
  78         self.assertEqual(
  79             GOST34112012512(m).digest(),
  80             hexdec("28fbc9bada033b1460642bdcddb90c3fb3e56c497ccd0f62b8a2ad4935e85f037613966de4ee00531ae60f3b5a47f8dae06915d5f2f194996fcabf2622e6881e")[::-1]
  81         )
  82         self.assertEqual(
  83             GOST34112012256(m).digest(),
  84             hexdec("508f7e553c06501d749a66fc28c6cac0b005746d97537fa85d9e40904efed29d")[::-1]
  85         )