1#
2# This file is part of pyasn1-modules software.
3#
4# Created by Russ Housley
5# Copyright (c) 2018, 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 rfc5083
16from pyasn1_modules import rfc5084
17from pyasn1_modules import rfc5652
18
19try:
20    import unittest2 as unittest
21except ImportError:
22    import unittest
23
24
25class CCMParametersTestCase(unittest.TestCase):
26    ccm_pem_text = "MBEEDE2HVyIurFKUEX8MEgIBBA=="
27
28    def setUp(self):
29        self.asn1Spec = rfc5084.CCMParameters()
30
31    def testDerCodec(self):
32        substrate = pem.readBase64fromText(self.ccm_pem_text)
33        asn1Object, rest = der_decode(substrate, asn1Spec=self.asn1Spec)
34        assert not rest
35        assert asn1Object.prettyPrint()
36        assert der_encode(asn1Object) == substrate
37
38
39class GCMParametersTestCase(unittest.TestCase):
40    gcm_pem_text = "MBEEDE2HVyIurFKUEX8MEgIBEA=="
41
42    def setUp(self):
43        self.asn1Spec = rfc5084.GCMParameters()
44
45    def testDerCodec(self):
46        substrate = pem.readBase64fromText(self.gcm_pem_text)
47        asn1Object, rest = der_decode(substrate, asn1Spec=self.asn1Spec)
48        assert not rest
49        assert asn1Object.prettyPrint()
50        assert der_encode(asn1Object) == substrate
51
52
53class GCMOpenTypesTestCase(unittest.TestCase):
54    rfc8591_pem_pext = """\
55MIIHkAYLKoZIhvcNAQkQARegggd/MIIHewIBADGCAk8wggJLAgEAMDMwJjEUMBIGA1UECgwL
56ZXhhbXBsZS5jb20xDjAMBgNVBAMMBUFsaWNlAgkAg/ULtwvVxA4wDQYJKoZIhvcNAQEBBQAE
57ggIAdZphtN3x8a8kZoAFY15HYRD6JyPBueRUhLbTPoOH3pZ9xeDK+zVXGlahl1y1UOe+McEx
582oD7cxAkhFuruNZMrCYEBCTZMwVhyEOZlBXdZEs8rZUHL3FFE5PJnygsSIO9DMxd1UuTFGTg
59Cm5V5ZLFGmjeEGJRbsfTyo52S7iseJqIN3dl743DbApu0+yuUoXKxqKdUFlEVxmhvc+Qbg/z
60fiwu8PTsYiUQDMBi4cdIlju8iLjj389xQHNyndXHWD51is89GG8vpBe+IsN8mnbGtCcpqtJ/
61c65ErJhHTR7rSJSMEqQD0LPOCKIY1q9FaSSJfMXJZk9t/rPxgUEVjfw7hAkKpgOAqoZRN+Fp
62nFyBl0FnnXo8kLp55tfVyNibtUpmdCPkOwt9b3jAtKtnvDQ2YqY1/llfEUnFOVDKwuC6MYwi
63fm92qNlAQA/T0+ocjs6gA9zOLx+wD1zqM13hMD/L+T2OHL/WgvGb62JLrNHXuPWA8RShO4kI
64lPtARKXap2S3+MX/kpSUUrNa65Y5uK1jwFFclczG+CPCIBBn6iJiQT/vOX1I97YUP4Qq6OGk
65jK064Bq6o8+e5+NmIOBcygYRv6wA7vGkmPLSWbnw99qD728bBh84fC3EjItdusqGIwjzL0eS
66UWXJ5eu0Z3mYhJGN1pe0R/TEB5ibiJsMLpWAr3gwggUPBgkqhkiG9w0BBwEwHgYJYIZIAWUD
67BAEGMBEEDE2HVyIurFKUEX8MEgIBEICCBOD+L7PeC/BpmMOb9KlS+r+LD+49fi6FGBrs8aie
68Gi7ezZQEiFYS38aYQzTYYCt3SbJQTkX1fDsGZiaw/HRiNh7sJnxWATm+XNKGoq+Wls9RhSJ4
695Sw4GMqwpoxZjeT84UozOITk3l3fV+3XiGcCejHkp8DAKZFExd5rrjlpnnAOBX6w8NrXO4s2
70n0LrMhtBU4eB2YKhGgs5Q6wQyXtU7rc7OOwTGvxWEONzSHJ01pyvqVQZAohsZPaWLULrM/kE
71GkrhG4jcaVjVPfULi7Uqo14imYhdCq5Ba4bwqI0Ot6mB27KD6LlOnVC/YmXCNIoYoWmqy1o3
72pSm9ovnLEO/dzxQjEJXYeWRje9M/sTxotM/5oZBpYMHqIwHTJbehXFgp8+oDjyTfayMYA3fT
73cTH3XbGPQfnYW2U9+ka/JhcSYybM8cuDNFd1I1LIQXoJRITXtkvPUbJqm+s6DtS5yvG9I8aQ
74xlT365zphS4vbQaO74ujO8bE3dynrvTTV0c318TcHpN3DY9PIt6mHXMIPDLEA4wes90zg6ia
75h5XiQcLtfLaAdYwEEGlImGD8n0kOhSNgclSLMklpj5mVOs8exli3qoXlVMRJcBptSwOe0QPc
76RY30spywS4zt1UDIQ0jaecGGVtUYj586nkubhAxwZkuQKWxgt6yYTpGNSKCdvd+ygfyGJRDb
77Wdn6nck/EPnG1773KTHRhMrXrBPBpSlfyJ/ju3644CCFqCjFoTh4bmB63k9ejUEVkJIJuoeK
78eTBaUxbCIinkK4htBkgchHP51RJp4q9jQbziD3aOhg13hO1GFQ4E/1DNIJxbEnURNp/ga8Sq
79mnLY8f5Pzwhm1mSzZf+obowbQ+epISrswWyjUKKO+uJfrAVN2TS/5+X6T3U6pBWWjH6+xDng
80rAJwtIdKBo0iSEwJ2eir4X8TcrSy9l8RSOiTPtqS5dF3RWSWOzkcO72fHCf/42+DLgUVX8Oe
815mUvp7QYiXXsXGezLJ8hPIrGuOEypafDv3TwFkBc2MIB0QUhk+GG1ENY3jiNcyEbovF5Lzz+
82ubvechHSb1arBuEczJzN4riM2Dc3c+r8N/2Ft6eivK7HUuYX1uAcArhunZpA8yBGLF1m+DUX
83FtzWAUvfMKYPdfwGMckghF7YwLrTXd8ZhPIkHNO1KdwQKIRfgIlUPfTxRB7eNrG/Ma9a/Iwr
84cI1QtkXU59uIZIw+7+FHZRWPsOjTu1Pdy+JtcSTG4dmS+DIwqpUzdu6MaBCVaOhXHwybvaSP
85TfMG/nR/NxF1FI8xgydnzXZs8HtFDL9iytKnvXHx+IIz8Rahp/PK8S80vPQNIeef/JgnIhto
86sID/A614LW1tB4cWdveYlD5U8T/XXInAtCY78Q9WJD+ecu87OJmlOdmjrFvitpQAo8+NGWxc
877Wl7LtgDuYel7oXFCVtI2npbA7R+K5/kzUvDCY6GTgzn1Gfamc1/Op6Ue17qd/emvhbIx+ng
883swf8TJVnCNDIXucKVA4boXSlCEhCGzfoZZYGVvm1/hrypiBtpUIKWTxLnz4AQJdZ5LGiCQJ
89QU1wMyHsg6vWmNaJVhGHE6D/EnKsvJptFIkAx0wWkh35s48p7EbU8QBg//5eNru6yvLRutfd
90BX7T4w681pCD+dOiom75C3UdahrfoFkNsZ2hB88+qNsEEPb/xuGu8ZzSPZhakhl2NS0=
91"""
92
93    def setUp(self):
94        self.asn1Spec = rfc5652.ContentInfo()
95
96    def testDerCodec(self):
97        substrate = pem.readBase64fromText(self.rfc8591_pem_pext)
98        asn1Object, rest = der_decode(substrate, asn1Spec=self.asn1Spec)
99        assert not rest
100        assert asn1Object.prettyPrint()
101        assert der_encode(asn1Object) == substrate
102
103        assert asn1Object['contentType'] == rfc5083.id_ct_authEnvelopedData
104        aed, rest = der_decode(asn1Object['content'],
105            asn1Spec=rfc5083.AuthEnvelopedData(),
106            decodeOpenTypes=True)
107        assert not rest
108        assert aed.prettyPrint()
109        assert der_encode(aed) == asn1Object['content']
110
111        assert aed['version'] == 0
112        cea = aed['authEncryptedContentInfo']['contentEncryptionAlgorithm']
113        assert cea['algorithm'] == rfc5084.id_aes128_GCM
114        assert cea['parameters']['aes-ICVlen'] == 16
115
116
117suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])
118
119if __name__ == '__main__':
120    import sys
121
122    result = unittest.TextTestRunner(verbosity=2).run(suite)
123    sys.exit(not result.wasSuccessful())
124