ef390f5f54
Instead of relying on openssl code for certificate parsing, use the ASN.1 representation directly. All previous features are supported. Not all the extensions are full parsed yet, but the code doesn't require them for now. The code makes accessing and modifying the certificate structure simpler and requires less error checking than the original version. The code leaves few TODOs, but nothing that destroys previous behaviour. It still uses the cryptography.io backend for loading keys and producing signatures for the certificates. Implements: blueprint direct-asn1 Change-Id: Ic555d3d056ca8da7016e2d8b434506cf214d06a1
125 lines
5.0 KiB
Python
125 lines
5.0 KiB
Python
# -*- coding:utf-8 -*-
|
|
#
|
|
# Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import unittest
|
|
|
|
from anchor.X509 import errors as x509_errors
|
|
from anchor.X509 import name as x509_name
|
|
|
|
|
|
class TestX509Name(unittest.TestCase):
|
|
def setUp(self):
|
|
super(TestX509Name, self).setUp()
|
|
self.name = x509_name.X509Name()
|
|
self.name.add_name_entry(x509_name.OID_countryName,
|
|
"UK") # must be 2 chars
|
|
self.name.add_name_entry(x509_name.OID_stateOrProvinceName, "test_ST")
|
|
self.name.add_name_entry(x509_name.OID_localityName, "test_L")
|
|
self.name.add_name_entry(x509_name.OID_organizationName, "test_O")
|
|
self.name.add_name_entry(x509_name.OID_organizationalUnitName,
|
|
"test_OU")
|
|
self.name.add_name_entry(x509_name.OID_commonName, "test_CN")
|
|
self.name.add_name_entry(x509_name.OID_pkcs9_emailAddress,
|
|
"test_Email")
|
|
self.name.add_name_entry(x509_name.OID_surname, "test_SN")
|
|
self.name.add_name_entry(x509_name.OID_givenName, "test_GN")
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_add_bad_entry_throws(self):
|
|
self.assertRaises(x509_errors.X509Error,
|
|
self.name.add_name_entry,
|
|
-1, "BAD_WRONG")
|
|
|
|
def test_set_bad_c_throws(self):
|
|
self.assertRaises(x509_errors.X509Error,
|
|
self.name.add_name_entry,
|
|
x509_name.OID_countryName, "BAD_WRONG")
|
|
|
|
def test_name_to_string(self):
|
|
val = str(self.name)
|
|
self.assertEqual(val, ("/C=UK/ST=test_ST/L=test_L/O=test_O/OU=test_OU"
|
|
"/CN=test_CN/emailAddress=test_Email/"
|
|
"SN=test_SN/GN=test_GN"))
|
|
|
|
def test_get_countryName(self):
|
|
entries = self.name.get_entries_by_oid(x509_name.OID_countryName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "countryName")
|
|
self.assertEqual(entries[0].get_value(), "UK")
|
|
|
|
def test_get_stateOrProvinceName(self):
|
|
entries = self.name.get_entries_by_oid(
|
|
x509_name.OID_stateOrProvinceName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "stateOrProvinceName")
|
|
self.assertEqual(entries[0].get_value(), "test_ST")
|
|
|
|
def test_get_subject_localityName(self):
|
|
entries = self.name.get_entries_by_oid(x509_name.OID_localityName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "localityName")
|
|
self.assertEqual(entries[0].get_value(), "test_L")
|
|
|
|
def test_get_organizationName(self):
|
|
entries = self.name.get_entries_by_oid(x509_name.OID_organizationName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "organizationName")
|
|
self.assertEqual(entries[0].get_value(), "test_O")
|
|
|
|
def test_get_organizationUnitName(self):
|
|
entries = self.name.get_entries_by_oid(
|
|
x509_name.OID_organizationalUnitName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "organizationalUnitName")
|
|
self.assertEqual(entries[0].get_value(), "test_OU")
|
|
|
|
def test_get_commonName(self):
|
|
entries = self.name.get_entries_by_oid(x509_name.OID_commonName)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "commonName")
|
|
self.assertEqual(entries[0].get_value(), "test_CN")
|
|
|
|
def test_get_emailAddress(self):
|
|
entries = self.name.get_entries_by_oid(
|
|
x509_name.OID_pkcs9_emailAddress)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].get_name(), "emailAddress")
|
|
self.assertEqual(entries[0].get_value(), "test_Email")
|
|
|
|
def test_entry_to_string(self):
|
|
entries = self.name.get_entries_by_oid(
|
|
x509_name.OID_pkcs9_emailAddress)
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(str(entries[0]), "emailAddress: test_Email")
|
|
|
|
def test_entry_length(self):
|
|
num = len(self.name)
|
|
self.assertEqual(num, 9)
|
|
|
|
def test_entry_index_good(self):
|
|
self.assertEqual("givenName: test_GN", str(self.name[8]))
|
|
|
|
def test_entry_index_bad(self):
|
|
with self.assertRaises(IndexError):
|
|
self.name[9]
|
|
|
|
def test_entry_itter(self):
|
|
val = [str(e) for e in self.name]
|
|
self.assertEqual("countryName: UK", val[0])
|
|
self.assertEqual("givenName: test_GN", val[8])
|