1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
|
#
# This file is part of pyasn1-modules software.
#
# Created by Russ Housley
# Copyright (c) 2019, Vigil Security, LLC
# License: http://snmplabs.com/pyasn1/license.html
#
import sys
import unittest
from pyasn1.codec.der.decoder import decode as der_decoder
from pyasn1.codec.der.encoder import encode as der_encoder
from pyasn1_modules import pem
from pyasn1_modules import rfc5280
from pyasn1_modules import rfc8419
class Ed25519TestCase(unittest.TestCase):
alg_id_1_pem_text = "MAUGAytlcA=="
def setUp(self):
self.asn1Spec = rfc5280.AlgorithmIdentifier()
def testDerCodec(self):
substrate = pem.readBase64fromText(self.alg_id_1_pem_text)
asn1Object, rest = der_decoder(substrate, asn1Spec=self.asn1Spec)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_Ed25519, asn1Object['algorithm'])
self.assertFalse(asn1Object['parameters'].isValue)
self.assertEqual(substrate, der_encoder(asn1Object))
class Ed448TestCase(unittest.TestCase):
alg_id_2_pem_text = "MAUGAytlcQ=="
def setUp(self):
self.asn1Spec = rfc5280.AlgorithmIdentifier()
def testDerCodec(self):
substrate = pem.readBase64fromText(self.alg_id_2_pem_text)
asn1Object, rest = der_decoder(
substrate, asn1Spec=self.asn1Spec)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_Ed448, asn1Object['algorithm'])
self.assertFalse(asn1Object['parameters'].isValue)
self.assertEqual(substrate, der_encoder(asn1Object))
class SHA512TestCase(unittest.TestCase):
alg_id_3_pem_text = "MAsGCWCGSAFlAwQCAw=="
def setUp(self):
self.asn1Spec = rfc5280.AlgorithmIdentifier()
def testDerCodec(self):
substrate = pem.readBase64fromText(self.alg_id_3_pem_text)
asn1Object, rest = der_decoder(
substrate, asn1Spec=self.asn1Spec)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_sha512, asn1Object['algorithm'])
self.assertFalse(asn1Object['parameters'].isValue)
self.assertEqual(substrate, der_encoder(asn1Object))
class SHAKE256TestCase(unittest.TestCase):
alg_id_4_pem_text = "MAsGCWCGSAFlAwQCDA=="
def setUp(self):
self.asn1Spec = rfc5280.AlgorithmIdentifier()
def testDerCodec(self):
substrate = pem.readBase64fromText(self.alg_id_4_pem_text)
asn1Object, rest = der_decoder(substrate, asn1Spec=self.asn1Spec)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_shake256, asn1Object['algorithm'])
self.assertFalse(asn1Object['parameters'].isValue)
self.assertEqual(substrate, der_encoder(asn1Object))
class SHAKE256LENTestCase(unittest.TestCase):
alg_id_5_pem_text = "MA8GCWCGSAFlAwQCEgICAgA="
def setUp(self):
self.asn1Spec = rfc5280.AlgorithmIdentifier()
def testDerCodec(self):
substrate = pem.readBase64fromText(self.alg_id_5_pem_text)
asn1Object, rest = der_decoder(substrate, asn1Spec=self.asn1Spec)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_shake256_len, asn1Object['algorithm'])
self.assertTrue(asn1Object['parameters'].isValue)
self.assertEqual(substrate, der_encoder(asn1Object))
param, rest = der_decoder(
asn1Object['parameters'],
asn1Spec=rfc5280.algorithmIdentifierMap[asn1Object['algorithm']])
self.assertFalse(rest)
self.assertTrue(param.prettyPrint())
self.assertEqual(asn1Object['parameters'], der_encoder(param))
self.assertEqual(512, param)
def testOpenTypes(self):
substrate = pem.readBase64fromText(self.alg_id_5_pem_text)
asn1Object, rest = der_decoder(
substrate, asn1Spec=self.asn1Spec, decodeOpenTypes=True)
self.assertFalse(rest)
self.assertTrue(asn1Object.prettyPrint())
self.assertEqual(rfc8419.id_shake256_len, asn1Object['algorithm'])
self.assertEqual(512, asn1Object['parameters'])
self.assertEqual(substrate, der_encoder(asn1Object))
suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
if __name__ == '__main__':
result = unittest.TextTestRunner(verbosity=2).run(suite)
sys.exit(not result.wasSuccessful())
|