Blame SOURCES/0010-Add-tests-for-certificate-mismatch-detection.patch

6457ff
From 94ca49d991b3f5bb404533f84970a1b2485d9cc8 Mon Sep 17 00:00:00 2001
6457ff
From: Antonio Torres <antorres@redhat.com>
6457ff
Date: Fri, 23 Apr 2021 17:48:14 +0200
6457ff
Subject: [PATCH] Add tests for certificate mismatch detection
6457ff
6457ff
Add tests for the IPACertMatchCheck and IPADogtagCertsMatchCheck plugins.
6457ff
6457ff
Related: https://bugzilla.redhat.com/show_bug.cgi?id=1886770
6457ff
Signed-off-by: Antonio Torres <antorres@redhat.com>
6457ff
---
6457ff
 tests/test_ipa_cert_match.py | 282 +++++++++++++++++++++++++++++++++++
6457ff
 1 file changed, 282 insertions(+)
6457ff
 create mode 100644 tests/test_ipa_cert_match.py
6457ff
6457ff
diff --git a/tests/test_ipa_cert_match.py b/tests/test_ipa_cert_match.py
6457ff
new file mode 100644
6457ff
index 0000000..460e61a
6457ff
--- /dev/null
6457ff
+++ b/tests/test_ipa_cert_match.py
6457ff
@@ -0,0 +1,282 @@
6457ff
+#
6457ff
+# Copyright (C) 2021 FreeIPA Contributors see COPYING for license
6457ff
+#
6457ff
+
6457ff
+from util import capture_results, m_api, CAInstance, KRAInstance
6457ff
+from base import BaseTest
6457ff
+from ipahealthcheck.core import config, constants
6457ff
+from ipahealthcheck.ipa.plugin import registry
6457ff
+from ipahealthcheck.ipa.certs import IPACertMatchCheck
6457ff
+from ipahealthcheck.ipa.certs import IPADogtagCertsMatchCheck
6457ff
+from unittest.mock import Mock, patch
6457ff
+
6457ff
+from ipalib import errors
6457ff
+from ipapython.dn import DN
6457ff
+from ipapython.ipaldap import LDAPClient, LDAPEntry
6457ff
+
6457ff
+
6457ff
+class IPACertificate:
6457ff
+    def __init__(self, serial_number=1):
6457ff
+        self.serial_number = serial_number
6457ff
+
6457ff
+    def __eq__(self, other):
6457ff
+        return self.serial_number == other.serial_number
6457ff
+
6457ff
+    def __hash__(self):
6457ff
+        return hash(self.serial_number)
6457ff
+
6457ff
+
6457ff
+class mock_ldap:
6457ff
+    SCOPE_BASE = 1
6457ff
+    SCOPE_ONELEVEL = 2
6457ff
+    SCOPE_SUBTREE = 4
6457ff
+
6457ff
+    def __init__(self, entries):
6457ff
+        """Initialize the results that we will return from get_entry"""
6457ff
+        self.results = {entry.dn: entry for entry in entries}
6457ff
+
6457ff
+    def get_entry(self, dn, attrs_list=None, time_limit=None,
6457ff
+                  size_limit=None, get_effective_rights=False):
6457ff
+        if self.results is None:
6457ff
+            raise errors.NotFound(reason='test')
6457ff
+        return self.results[dn]
6457ff
+
6457ff
+    def get_entries(self, base_dn, scope=SCOPE_SUBTREE, filter=None,
6457ff
+                    attrs_list=None, get_effective_rights=False, **kwargs):
6457ff
+        if self.results is None:
6457ff
+            raise errors.NotFound(reason='test')
6457ff
+        return self.results.values()
6457ff
+
6457ff
+
6457ff
+class mock_ldap_conn:
6457ff
+    def set_option(self, option, invalue):
6457ff
+        pass
6457ff
+
6457ff
+    def search_s(self, base, scope, filterstr=None,
6457ff
+                 attrlist=None, attrsonly=0):
6457ff
+        return tuple()
6457ff
+
6457ff
+
6457ff
+class mock_CertDB:
6457ff
+    def __init__(self, trust):
6457ff
+        """A dict of nickname + NSSdb trust flags"""
6457ff
+        self.trust = trust
6457ff
+        self.secdir = '/foo/bar/testdir'
6457ff
+
6457ff
+    def get_cert_from_db(self, nickname):
6457ff
+        if nickname not in self.trust.keys():
6457ff
+            raise errors.NotFound(reason='test')
6457ff
+        return IPACertificate()
6457ff
+
6457ff
+    def run_certutil(self, args, capture_output):
6457ff
+        class RunResult:
6457ff
+            def __init__(self, output):
6457ff
+                self.raw_output = output
6457ff
+
6457ff
+        return RunResult(b'test output')
6457ff
+
6457ff
+
6457ff
+class TestIPACertMatch(BaseTest):
6457ff
+    patches = {
6457ff
+        'ldap.initialize':
6457ff
+        Mock(return_value=mock_ldap_conn())
6457ff
+    }
6457ff
+
6457ff
+    @patch('ipalib.x509.load_certificate_list_from_file')
6457ff
+    @patch('ipaserver.install.certs.CertDB')
6457ff
+    def test_certs_match_ok(self, mock_certdb, mock_load_cert):
6457ff
+        """ Ensure match check is ok"""
6457ff
+        fake_conn = LDAPClient('ldap://localhost', no_schema=True)
6457ff
+        cacertentry = LDAPEntry(fake_conn,
6457ff
+                                DN('cn=%s IPA CA' % m_api.env.realm,
6457ff
+                                   'cn=certificates,cn=ipa,cn=etc',
6457ff
+                                    m_api.env.basedn),
6457ff
+                                CACertificate=[IPACertificate()])
6457ff
+        trust = {
6457ff
+            ('%s IPA CA' % m_api.env.realm): 'u,u,u'
6457ff
+        }
6457ff
+
6457ff
+        mock_certdb.return_value = mock_CertDB(trust)
6457ff
+        mock_load_cert.return_value = [IPACertificate()]
6457ff
+
6457ff
+        framework = object()
6457ff
+        registry.initialize(framework, config.Config())
6457ff
+        f = IPACertMatchCheck(registry)
6457ff
+        f.conn = mock_ldap([cacertentry])
6457ff
+        self.results = capture_results(f)
6457ff
+
6457ff
+        assert len(self.results) == 3
6457ff
+        for result in self.results.results:
6457ff
+            assert result.result == constants.SUCCESS
6457ff
+            assert result.source == 'ipahealthcheck.ipa.certs'
6457ff
+            assert result.check == 'IPACertMatchCheck'
6457ff
+
6457ff
+    @patch('ipalib.x509.load_certificate_list_from_file')
6457ff
+    @patch('ipaserver.install.certs.CertDB')
6457ff
+    def test_etc_cacert_mismatch(self, mock_certdb, mock_load_cert):
6457ff
+        """ Test mismatch with /etc/ipa/ca.crt """
6457ff
+        fake_conn = LDAPClient('ldap://localhost', no_schema=True)
6457ff
+        cacertentry = LDAPEntry(fake_conn,
6457ff
+                                DN('cn=%s IPA CA' % m_api.env.realm,
6457ff
+                                   'cn=certificates,cn=ipa,cn=etc',
6457ff
+                                    m_api.env.basedn),
6457ff
+                                CACertificate=[IPACertificate()])
6457ff
+        trust = {
6457ff
+            ('%s IPA CA' % m_api.env.realm): 'u,u,u'
6457ff
+        }
6457ff
+
6457ff
+        mock_certdb.return_value = mock_CertDB(trust)
6457ff
+        mock_load_cert.return_value = [IPACertificate(serial_number=2)]
6457ff
+
6457ff
+        framework = object()
6457ff
+        registry.initialize(framework, config.Config())
6457ff
+        f = IPACertMatchCheck(registry)
6457ff
+        f.conn = mock_ldap([cacertentry])
6457ff
+        self.results = capture_results(f)
6457ff
+
6457ff
+        assert len(self.results) == 3
6457ff
+        result = self.results.results[0]
6457ff
+        assert result.result == constants.ERROR
6457ff
+        assert result.source == 'ipahealthcheck.ipa.certs'
6457ff
+        assert result.check == 'IPACertMatchCheck'
6457ff
+
6457ff
+    @patch('ipaserver.install.cainstance.CAInstance')
6457ff
+    def test_cacert_caless(self, mock_cainstance):
6457ff
+        """Nothing to check if the master is CALess"""
6457ff
+
6457ff
+        mock_cainstance.return_value = CAInstance(False)
6457ff
+
6457ff
+        framework = object()
6457ff
+        registry.initialize(framework, config)
6457ff
+        f = IPACertMatchCheck(registry)
6457ff
+
6457ff
+        self.results = capture_results(f)
6457ff
+
6457ff
+        assert len(self.results) == 0
6457ff
+
6457ff
+
6457ff
+class TestIPADogtagCertMatch(BaseTest):
6457ff
+    patches = {
6457ff
+        'ipaserver.install.krainstance.KRAInstance':
6457ff
+        Mock(return_value=KRAInstance()),
6457ff
+    }
6457ff
+
6457ff
+    @patch('ipaserver.install.certs.CertDB')
6457ff
+    def test_certs_match_ok(self, mock_certdb):
6457ff
+        """ Ensure match check is ok"""
6457ff
+        fake_conn = LDAPClient('ldap://localhost', no_schema=True)
6457ff
+        pkidbentry = LDAPEntry(fake_conn,
6457ff
+                               DN('uid=pkidbuser,ou=people,o=ipaca'),
6457ff
+                               userCertificate=[IPACertificate()],
6457ff
+                               subjectName=['test'])
6457ff
+        casignentry = LDAPEntry(fake_conn,
6457ff
+                                DN('cn=%s IPA CA' % m_api.env.realm,
6457ff
+                                   'cn=certificates,cn=ipa,cn=etc',
6457ff
+                                    m_api.env.basedn),
6457ff
+                                CACertificate=[IPACertificate()],
6457ff
+                                userCertificate=[IPACertificate()],
6457ff
+                                subjectName=['test'])
6457ff
+        ldap_entries = [pkidbentry, casignentry]
6457ff
+        trust = {
6457ff
+            'ocspSigningCert cert-pki-ca': 'u,u,u',
6457ff
+            'caSigningCert cert-pki-ca': 'u,u,u',
6457ff
+            'subsystemCert cert-pki-ca': 'u,u,u',
6457ff
+            'auditSigningCert cert-pki-ca': 'u,u,Pu',
6457ff
+            'Server-Cert cert-pki-ca': 'u,u,u',
6457ff
+            'transportCert cert-pki-kra': 'u,u,u',
6457ff
+            'storageCert cert-pki-kra': 'u,u,u',
6457ff
+            'auditSigningCert cert-pki-kra': 'u,u,Pu',
6457ff
+        }
6457ff
+
6457ff
+        dogtag_entries_subjects = (
6457ff
+            'CN=OCSP Subsystem,O=%s' % m_api.env.realm,
6457ff
+            'CN=CA Subsystem,O=%s' % m_api.env.realm,
6457ff
+            'CN=CA Audit,O=%s' % m_api.env.realm,
6457ff
+            'CN=%s,O=%s' % (m_api.env.host, m_api.env.realm),
6457ff
+            'CN=KRA Transport Certificate,O=%s' % m_api.env.realm,
6457ff
+            'CN=KRA Storage Certificate,O=%s' % m_api.env.realm,
6457ff
+            'CN=KRA Audit,O=%s' % m_api.env.realm,
6457ff
+        )
6457ff
+
6457ff
+        for i, subject in enumerate(dogtag_entries_subjects):
6457ff
+            entry = LDAPEntry(fake_conn,
6457ff
+                              DN('cn=%i,ou=certificateRepository' % i,
6457ff
+                                 'ou=ca,o=ipaca'),
6457ff
+                              userCertificate=[IPACertificate()],
6457ff
+                              subjectName=[subject])
6457ff
+            ldap_entries.append(entry)
6457ff
+
6457ff
+        mock_certdb.return_value = mock_CertDB(trust)
6457ff
+
6457ff
+        framework = object()
6457ff
+        registry.initialize(framework, config.Config())
6457ff
+        f = IPADogtagCertsMatchCheck(registry)
6457ff
+        f.conn = mock_ldap(ldap_entries)
6457ff
+        self.results = capture_results(f)
6457ff
+
6457ff
+        assert len(self.results) == 3
6457ff
+        for result in self.results.results:
6457ff
+            assert result.result == constants.SUCCESS
6457ff
+            assert result.source == 'ipahealthcheck.ipa.certs'
6457ff
+            assert result.check == 'IPADogtagCertsMatchCheck'
6457ff
+
6457ff
+    @patch('ipaserver.install.certs.CertDB')
6457ff
+    def test_certs_mismatch(self, mock_certdb):
6457ff
+        """ Ensure mismatches are detected"""
6457ff
+        fake_conn = LDAPClient('ldap://localhost', no_schema=True)
6457ff
+        pkidbentry = LDAPEntry(fake_conn,
6457ff
+                               DN('uid=pkidbuser,ou=people,o=ipaca'),
6457ff
+                               userCertificate=[IPACertificate(
6457ff
+                                   serial_number=2
6457ff
+                               )],
6457ff
+                               subjectName=['test'])
6457ff
+        casignentry = LDAPEntry(fake_conn,
6457ff
+                                DN('cn=%s IPA CA' % m_api.env.realm,
6457ff
+                                   'cn=certificates,cn=ipa,cn=etc',
6457ff
+                                    m_api.env.basedn),
6457ff
+                                CACertificate=[IPACertificate()],
6457ff
+                                userCertificate=[IPACertificate()],
6457ff
+                                subjectName=['test'])
6457ff
+        ldap_entries = [pkidbentry, casignentry]
6457ff
+        trust = {
6457ff
+            'ocspSigningCert cert-pki-ca': 'u,u,u',
6457ff
+            'caSigningCert cert-pki-ca': 'u,u,u',
6457ff
+            'subsystemCert cert-pki-ca': 'u,u,u',
6457ff
+            'auditSigningCert cert-pki-ca': 'u,u,Pu',
6457ff
+            'Server-Cert cert-pki-ca': 'u,u,u',
6457ff
+            'transportCert cert-pki-kra': 'u,u,u',
6457ff
+            'storageCert cert-pki-kra': 'u,u,u',
6457ff
+            'auditSigningCert cert-pki-kra': 'u,u,Pu',
6457ff
+        }
6457ff
+
6457ff
+        dogtag_entries_subjects = (
6457ff
+            'CN=OCSP Subsystem,O=%s' % m_api.env.realm,
6457ff
+            'CN=CA Subsystem,O=%s' % m_api.env.realm,
6457ff
+            'CN=CA Audit,O=%s' % m_api.env.realm,
6457ff
+            'CN=%s,O=%s' % (m_api.env.host, m_api.env.realm),
6457ff
+            'CN=KRA Transport Certificate,O=%s' % m_api.env.realm,
6457ff
+            'CN=KRA Storage Certificate,O=%s' % m_api.env.realm,
6457ff
+            'CN=KRA Audit,O=%s' % m_api.env.realm,
6457ff
+        )
6457ff
+
6457ff
+        for i, subject in enumerate(dogtag_entries_subjects):
6457ff
+            entry = LDAPEntry(fake_conn,
6457ff
+                              DN('cn=%i,ou=certificateRepository' % i,
6457ff
+                                 'ou=ca,o=ipaca'),
6457ff
+                              userCertificate=[IPACertificate()],
6457ff
+                              subjectName=[subject])
6457ff
+            ldap_entries.append(entry)
6457ff
+
6457ff
+        mock_certdb.return_value = mock_CertDB(trust)
6457ff
+
6457ff
+        framework = object()
6457ff
+        registry.initialize(framework, config.Config())
6457ff
+        f = IPADogtagCertsMatchCheck(registry)
6457ff
+        f.conn = mock_ldap(ldap_entries)
6457ff
+        self.results = capture_results(f)
6457ff
+
6457ff
+        assert len(self.results) == 3
6457ff
+        result = self.results.results[0]
6457ff
+        assert result.result == constants.ERROR
6457ff
+        assert result.source == 'ipahealthcheck.ipa.certs'
6457ff
+        assert result.check == 'IPADogtagCertsMatchCheck'
6457ff
-- 
6457ff
2.26.3
6457ff