from k5test import *
# Skip this test if pkinit wasn't built.
if not os.path.exists(os.path.join(plugins, 'preauth', 'pkinit.so')):
skip_rest('certauth tests', 'PKINIT module not built')
certs = os.path.join(srctop, 'tests', 'dejagnu', 'pkinit-certs')
ca_pem = os.path.join(certs, 'ca.pem')
kdc_pem = os.path.join(certs, 'kdc.pem')
privkey_pem = os.path.join(certs, 'privkey.pem')
user_pem = os.path.join(certs, 'user.pem')
modpath = os.path.join(buildtop, 'plugins', 'certauth', 'test',
'certauth_test.so')
pkinit_krb5_conf = {'realms': {'$realm': {
'pkinit_anchors': 'FILE:%s' % ca_pem}},
'plugins': {'certauth': {'module': ['test1:' + modpath,
'test2:' + modpath],
'enable_only': ['test1', 'test2']}}}
pkinit_kdc_conf = {'realms': {'$realm': {
'default_principal_flags': '+preauth',
'pkinit_eku_checking': 'none',
'pkinit_identity': 'FILE:%s,%s' % (kdc_pem, privkey_pem),
'pkinit_indicator': ['indpkinit1', 'indpkinit2']}}}
file_identity = 'FILE:%s,%s' % (user_pem, privkey_pem)
realm = K5Realm(krb5_conf=pkinit_krb5_conf, kdc_conf=pkinit_kdc_conf,
get_creds=False)
# Let the test module match user to CN=user, with indicators.
realm.kinit(realm.user_princ,
flags=['-X', 'X509_user_identity=%s' % file_identity])
realm.klist(realm.user_princ)
realm.run([kvno, realm.host_princ])
realm.run(['./adata', realm.host_princ],
expected_msg='+97: [test1, test2, user, indpkinit1, indpkinit2]')
# Let the test module mismatch with user2 to CN=user.
realm.addprinc("user2@KRBTEST.COM")
out = realm.kinit("user2@KRBTEST.COM",
flags=['-X', 'X509_user_identity=%s' % file_identity],
expected_code=1,
expected_msg='kinit: Certificate mismatch')
# Test the KRB5_CERTAUTH_HWAUTH return code.
mark('hw-authent flag tests')
# First test +requires_hwauth without causing the hw-authent ticket
# flag to be set. This currently results in a preauth loop.
realm.run([kadminl, 'modprinc', '+requires_hwauth', realm.user_princ])
realm.kinit(realm.user_princ,
flags=['-X', 'X509_user_identity=%s' % file_identity],
expected_code=1, expected_msg='Looping detected')
# Cause the test2 module to return KRB5_CERTAUTH_HWAUTH and try again.
realm.run([kadminl, 'setstr', realm.user_princ, 'hwauth', 'x'])
realm.kinit(realm.user_princ,
flags=['-X', 'X509_user_identity=%s' % file_identity])
success("certauth tests")