pygost/test_gost34112012.py

   1 # coding: utf-8
   2 # PyGOST -- Pure Python GOST cryptographic functions library
   3 # Copyright (C) 2015-2018 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 gost34112012256
  22 from pygost import gost34112012512
  23 from pygost.gost34112012256 import GOST34112012256
  24 from pygost.gost34112012512 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     """RFC 7836
  40     """
  41     def test_256(self):
  42         for digestmod in (GOST34112012256, gost34112012256):
  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, gost34112012512):
  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         )