1#
2# This file is part of pyasn1-modules software.
3#
4# Created by Russ Housley
5# Copyright (c) 2019, Vigil Security, LLC
6# License: http://snmplabs.com/pyasn1/license.html
7#
8
9import sys
10
11from pyasn1.codec.der.decoder import decode as der_decode
12from pyasn1.codec.der.encoder import encode as der_encode
13
14from pyasn1_modules import pem
15from pyasn1_modules import rfc5652
16from pyasn1_modules import rfc8018
17
18try:
19    import unittest2 as unittest
20except ImportError:
21    import unittest
22
23
24class PWRITestCase(unittest.TestCase):
25    rfc3211_ex1_pem_text = """\
26o1MCAQCgGgYJKoZIhvcNAQUMMA0ECBI0Vnh4VjQSAgEFMCAGCyqGSIb3DQEJEAMJMBEGBSsO
27AwIHBAjv5ZjvIbM9bQQQuBslZe43PKbe3KJqF4sMEA==
28"""
29
30    def setUp(self):
31        self.asn1Spec = rfc5652.RecipientInfo()
32
33    def testDerCodec(self):
34        substrate = pem.readBase64fromText(self.rfc3211_ex1_pem_text)
35        asn1Object, rest = der_decode(substrate, asn1Spec=self.asn1Spec)
36        assert not rest
37        assert asn1Object.prettyPrint()
38        assert der_encode(asn1Object) == substrate
39        alg_oid = asn1Object['pwri']['keyDerivationAlgorithm']['algorithm']
40        assert alg_oid == rfc8018.id_PBKDF2
41
42    def testOpenTypes(self):
43        substrate = pem.readBase64fromText(self.rfc3211_ex1_pem_text)
44        asn1Object, rest = der_decode(substrate,
45            asn1Spec=self.asn1Spec,
46            decodeOpenTypes=True)
47        assert not rest
48        assert asn1Object.prettyPrint()
49        assert der_encode(asn1Object) == substrate
50        icount = asn1Object['pwri']['keyDerivationAlgorithm']['parameters']['iterationCount']
51        assert icount == 5
52
53suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
54
55if __name__ == '__main__':
56    unittest.TextTestRunner(verbosity=2).run(suite)
57