pygost/test_x509.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 base64 import b64decode
  19 from unittest import skipIf
  20 from unittest import TestCase
  21
  22 from pygost.gost3410 import CURVE_PARAMS
  23 from pygost.gost3410 import GOST3410Curve
  24 from pygost.gost3410 import prv_unmarshal
  25 from pygost.gost3410 import pub_unmarshal
  26 from pygost.gost3410 import public_key
  27 from pygost.gost3410 import verify
  28 from pygost.gost34112012256 import GOST34112012256
  29 from pygost.gost34112012512 import GOST34112012512
  30 from pygost.utils import hexdec
  31
  32 try:
  33     from pyderasn import OctetString
  34
  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)
  52         self.assertSequenceEqual(tail, b"")
  53         curve = GOST3410Curve(*CURVE_PARAMS[curve_name])
  54         prv_key = prv_unmarshal(prv_key_raw)
  55         pub_key_raw, tail = OctetString().decode(
  56             bytes(cert["tbsCertificate"]["subjectPublicKeyInfo"]["subjectPublicKey"])
  57         )
  58         pub_key = pub_unmarshal(bytes(pub_key_raw), mode=mode)
  59         self.assertSequenceEqual(tail, b"")
  60         self.assertSequenceEqual(pub_key, public_key(curve, prv_key))
  61         self.assertTrue(verify(
  62             curve,
  63             pub_key,
  64             hasher(cert["tbsCertificate"].encode()).digest()[::-1],
  65             bytes(cert["signatureValue"]),
  66             mode=mode,
  67         ))
  68
  69     def test_256(self):
  70         cert_raw = b64decode("""
  71 MIICYjCCAg+gAwIBAgIBATAKBggqhQMHAQEDAjBWMSkwJwYJKoZIhvcNAQkBFhpH
  72 b3N0UjM0MTAtMjAxMkBleGFtcGxlLmNvbTEpMCcGA1UEAxMgR29zdFIzNDEwLTIw
  73 MTIgKDI1NiBiaXQpIGV4YW1wbGUwHhcNMTMxMTA1MTQwMjM3WhcNMzAxMTAxMTQw
  74 MjM3WjBWMSkwJwYJKoZIhvcNAQkBFhpHb3N0UjM0MTAtMjAxMkBleGFtcGxlLmNv
  75 bTEpMCcGA1UEAxMgR29zdFIzNDEwLTIwMTIgKDI1NiBiaXQpIGV4YW1wbGUwZjAf
  76 BggqhQMHAQEBATATBgcqhQMCAiQABggqhQMHAQECAgNDAARAut/Qw1MUq9KPqkdH
  77 C2xAF3K7TugHfo9n525D2s5mFZdD5pwf90/i4vF0mFmr9nfRwMYP4o0Pg1mOn5Rl
  78 aXNYraOBwDCBvTAdBgNVHQ4EFgQU1fIeN1HaPbw+XWUzbkJ+kHJUT0AwCwYDVR0P
  79 BAQDAgHGMA8GA1UdEwQIMAYBAf8CAQEwfgYDVR0BBHcwdYAU1fIeN1HaPbw+XWUz
  80 bkJ+kHJUT0ChWqRYMFYxKTAnBgkqhkiG9w0BCQEWGkdvc3RSMzQxMC0yMDEyQGV4
  81 YW1wbGUuY29tMSkwJwYDVQQDEyBHb3N0UjM0MTAtMjAxMiAoMjU2IGJpdCkgZXhh
  82 bXBsZYIBATAKBggqhQMHAQEDAgNBAF5bm4BbARR6hJLEoWJkOsYV3Hd7kXQQjz3C
  83 dqQfmHrz6TI6Xojdh/t8ckODv/587NS5/6KsM77vc6Wh90NAT2s=
  84         """)
  85         prv_key_raw = hexdec("BFCF1D623E5CDD3032A7C6EABB4A923C46E43D640FFEAAF2C3ED39A8FA399924")[::-1]
  86         self.process_cert(
  87             "GostR3410_2001_CryptoPro_XchA_ParamSet",
  88             2001,
  89             GOST34112012256,
  90             prv_key_raw,
  91             cert_raw,
  92         )
  93
  94     def test_512(self):
  95         cert_raw = b64decode("""
  96 MIIC6DCCAlSgAwIBAgIBATAKBggqhQMHAQEDAzBWMSkwJwYJKoZIhvcNAQkBFhpH
  97 b3N0UjM0MTAtMjAxMkBleGFtcGxlLmNvbTEpMCcGA1UEAxMgR29zdFIzNDEwLTIw
  98 MTIgKDUxMiBiaXQpIGV4YW1wbGUwHhcNMTMxMDA0MDczNjA0WhcNMzAxMDAxMDcz
  99 NjA0WjBWMSkwJwYJKoZIhvcNAQkBFhpHb3N0UjM0MTAtMjAxMkBleGFtcGxlLmNv
 100 bTEpMCcGA1UEAxMgR29zdFIzNDEwLTIwMTIgKDUxMiBiaXQpIGV4YW1wbGUwgaow
 101 IQYIKoUDBwEBAQIwFQYJKoUDBwECAQICBggqhQMHAQECAwOBhAAEgYATGQ9VCiM5
 102 FRGCQ8MEz2F1dANqhaEuywa8CbxOnTvaGJpFQVXQwkwvLFAKh7hk542vOEtxpKtT
 103 CXfGf84nRhMH/Q9bZeAc2eO/yhxrsQhTBufa1Fuou2oe/jUOaG6RAtUUvRzhNTpp
 104 RGGl1+EIY2vzzUua9j9Ol/gAoy/LNKQIfqOBwDCBvTAdBgNVHQ4EFgQUPcbTRXJZ
 105 nHtjj+eBP7b5lcTMekIwCwYDVR0PBAQDAgHGMA8GA1UdEwQIMAYBAf8CAQEwfgYD
 106 VR0BBHcwdYAUPcbTRXJZnHtjj+eBP7b5lcTMekKhWqRYMFYxKTAnBgkqhkiG9w0B
 107 CQEWGkdvc3RSMzQxMC0yMDEyQGV4YW1wbGUuY29tMSkwJwYDVQQDEyBHb3N0UjM0
 108 MTAtMjAxMiAoNTEyIGJpdCkgZXhhbXBsZYIBATAKBggqhQMHAQEDAwOBgQBObS7o
 109 ppPTXzHyVR1DtPa8b57nudJzI4czhsfeX5HDntOq45t9B/qSs8dC6eGxbhHZ9zCO
 110 SFtxWYdmg0au8XI9Xb8vTC1qdwWID7FFjMWDNQZb6lYh/J+8F2xKylvB5nIlRZqO
 111 o3eUNFkNyHJwQCk2WoOlO16zwGk2tdKH4KmD5w==
 112         """)
 113         prv_key_raw = hexdec("3FC01CDCD4EC5F972EB482774C41E66DB7F380528DFE9E67992BA05AEE462435757530E641077CE587B976C8EEB48C48FD33FD175F0C7DE6A44E014E6BCB074B")[::-1]
 114         self.process_cert(
 115             "GostR3410_2012_TC26_ParamSetB",
 116             2012,
 117             GOST34112012512,
 118             prv_key_raw,
 119             cert_raw,
 120         )