pygost/test_x509.py

   1 # coding: utf-8
   2 # PyGOST -- Pure Python GOST cryptographic functions library
   3 # Copyright (C) 2015-2019 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 base64 import b64decode
  18 from unittest import skipIf
  19 from unittest import TestCase
  20
  21 from pygost.gost3410 import CURVES
  22 from pygost.gost3410 import prv_unmarshal
  23 from pygost.gost3410 import pub_unmarshal
  24 from pygost.gost3410 import public_key
  25 from pygost.gost3410 import verify
  26 from pygost.gost34112012256 import GOST34112012256
  27 from pygost.gost34112012512 import GOST34112012512
  28 from pygost.utils import hexdec
  29
  30 try:
  31     from pyderasn import OctetString
  32
  33     from pygost.asn1schemas.oids import id_tc26_gost3410_2012_256
  34     from pygost.asn1schemas.oids import id_tc26_gost3410_2012_512
  35     from pygost.asn1schemas.x509 import Certificate
  36 except ImportError:
  37     pyderasn_exists = False
  38 else:
  39     pyderasn_exists = True
  40
  41
  42 @skipIf(not pyderasn_exists, "PyDERASN dependency is required")
  43 class TestCertificate(TestCase):
  44     """Certificate test vectors from "Использования алгоритмов ГОСТ Р
  45     34.10, ГОСТ Р 34.11 в профиле сертификата и списке отзыва
  46     сертификатов (CRL) инфраструктуры открытых ключей X.509"
  47     (TK26IOK.pdf)
  48     """
  49
  50     def process_cert(self, curve_name, mode, hasher, prv_key_raw, cert_raw):
  51         cert, tail = Certificate().decode(cert_raw, ctx={
  52             "defines_by_path": (
  53                 (
  54                     (
  55                         "tbsCertificate",
  56                         "subjectPublicKeyInfo",
  57                         "algorithm",
  58                         "algorithm",
  59                     ),
  60                     (
  61                         (
  62                             ("..", "subjectPublicKey"),
  63                             {
  64                                 id_tc26_gost3410_2012_256: OctetString(),
  65                                 id_tc26_gost3410_2012_512: OctetString(),
  66                             },
  67                         ),
  68                     ),
  69                 ),
  70             ),
  71         })
  72         self.assertSequenceEqual(tail, b"")
  73         curve = CURVES[curve_name]
  74         prv_key = prv_unmarshal(prv_key_raw)
  75         spk = cert["tbsCertificate"]["subjectPublicKeyInfo"]["subjectPublicKey"]
  76         self.assertIsNotNone(spk.defined)
  77         _, pub_key_raw = spk.defined
  78         pub_key = pub_unmarshal(bytes(pub_key_raw), mode=mode)
  79         self.assertSequenceEqual(pub_key, public_key(curve, prv_key))
  80         self.assertTrue(verify(
  81             curve,
  82             pub_key,
  83             hasher(cert["tbsCertificate"].encode()).digest()[::-1],
  84             bytes(cert["signatureValue"]),
  85             mode=mode,
  86         ))
  87
  88     def test_256(self):
  89         cert_raw = b64decode("""
  90 MIICYjCCAg+gAwIBAgIBATAKBggqhQMHAQEDAjBWMSkwJwYJKoZIhvcNAQkBFhpH
  91 b3N0UjM0MTAtMjAxMkBleGFtcGxlLmNvbTEpMCcGA1UEAxMgR29zdFIzNDEwLTIw
  92 MTIgKDI1NiBiaXQpIGV4YW1wbGUwHhcNMTMxMTA1MTQwMjM3WhcNMzAxMTAxMTQw
  93 MjM3WjBWMSkwJwYJKoZIhvcNAQkBFhpHb3N0UjM0MTAtMjAxMkBleGFtcGxlLmNv
  94 bTEpMCcGA1UEAxMgR29zdFIzNDEwLTIwMTIgKDI1NiBiaXQpIGV4YW1wbGUwZjAf
  95 BggqhQMHAQEBATATBgcqhQMCAiQABggqhQMHAQECAgNDAARAut/Qw1MUq9KPqkdH
  96 C2xAF3K7TugHfo9n525D2s5mFZdD5pwf90/i4vF0mFmr9nfRwMYP4o0Pg1mOn5Rl
  97 aXNYraOBwDCBvTAdBgNVHQ4EFgQU1fIeN1HaPbw+XWUzbkJ+kHJUT0AwCwYDVR0P
  98 BAQDAgHGMA8GA1UdEwQIMAYBAf8CAQEwfgYDVR0BBHcwdYAU1fIeN1HaPbw+XWUz
  99 bkJ+kHJUT0ChWqRYMFYxKTAnBgkqhkiG9w0BCQEWGkdvc3RSMzQxMC0yMDEyQGV4
 100 YW1wbGUuY29tMSkwJwYDVQQDEyBHb3N0UjM0MTAtMjAxMiAoMjU2IGJpdCkgZXhh
 101 bXBsZYIBATAKBggqhQMHAQEDAgNBAF5bm4BbARR6hJLEoWJkOsYV3Hd7kXQQjz3C
 102 dqQfmHrz6TI6Xojdh/t8ckODv/587NS5/6KsM77vc6Wh90NAT2s=
 103         """)
 104         prv_key_raw = hexdec("BFCF1D623E5CDD3032A7C6EABB4A923C46E43D640FFEAAF2C3ED39A8FA399924")[::-1]
 105         self.process_cert(
 106             "id-GostR3410-2001-CryptoPro-XchA-ParamSet",
 107             2001,
 108             GOST34112012256,
 109             prv_key_raw,
 110             cert_raw,
 111         )
 112
 113     def test_512(self):
 114         cert_raw = b64decode("""
 115 MIIC6DCCAlSgAwIBAgIBATAKBggqhQMHAQEDAzBWMSkwJwYJKoZIhvcNAQkBFhpH
 116 b3N0UjM0MTAtMjAxMkBleGFtcGxlLmNvbTEpMCcGA1UEAxMgR29zdFIzNDEwLTIw
 117 MTIgKDUxMiBiaXQpIGV4YW1wbGUwHhcNMTMxMDA0MDczNjA0WhcNMzAxMDAxMDcz
 118 NjA0WjBWMSkwJwYJKoZIhvcNAQkBFhpHb3N0UjM0MTAtMjAxMkBleGFtcGxlLmNv
 119 bTEpMCcGA1UEAxMgR29zdFIzNDEwLTIwMTIgKDUxMiBiaXQpIGV4YW1wbGUwgaow
 120 IQYIKoUDBwEBAQIwFQYJKoUDBwECAQICBggqhQMHAQECAwOBhAAEgYATGQ9VCiM5
 121 FRGCQ8MEz2F1dANqhaEuywa8CbxOnTvaGJpFQVXQwkwvLFAKh7hk542vOEtxpKtT
 122 CXfGf84nRhMH/Q9bZeAc2eO/yhxrsQhTBufa1Fuou2oe/jUOaG6RAtUUvRzhNTpp
 123 RGGl1+EIY2vzzUua9j9Ol/gAoy/LNKQIfqOBwDCBvTAdBgNVHQ4EFgQUPcbTRXJZ
 124 nHtjj+eBP7b5lcTMekIwCwYDVR0PBAQDAgHGMA8GA1UdEwQIMAYBAf8CAQEwfgYD
 125 VR0BBHcwdYAUPcbTRXJZnHtjj+eBP7b5lcTMekKhWqRYMFYxKTAnBgkqhkiG9w0B
 126 CQEWGkdvc3RSMzQxMC0yMDEyQGV4YW1wbGUuY29tMSkwJwYDVQQDEyBHb3N0UjM0
 127 MTAtMjAxMiAoNTEyIGJpdCkgZXhhbXBsZYIBATAKBggqhQMHAQEDAwOBgQBObS7o
 128 ppPTXzHyVR1DtPa8b57nudJzI4czhsfeX5HDntOq45t9B/qSs8dC6eGxbhHZ9zCO
 129 SFtxWYdmg0au8XI9Xb8vTC1qdwWID7FFjMWDNQZb6lYh/J+8F2xKylvB5nIlRZqO
 130 o3eUNFkNyHJwQCk2WoOlO16zwGk2tdKH4KmD5w==
 131         """)
 132         prv_key_raw = hexdec("3FC01CDCD4EC5F972EB482774C41E66DB7F380528DFE9E67992BA05AEE462435757530E641077CE587B976C8EEB48C48FD33FD175F0C7DE6A44E014E6BCB074B")[::-1]
 133         self.process_cert(
 134             "id-tc26-gost-3410-12-512-paramSetB",
 135             2012,
 136             GOST34112012512,
 137             prv_key_raw,
 138             cert_raw,
 139         )