from k5test import *

# Skip this test if we're missing proxy functionality or parts of the proxy.

if runenv.tls_impl == 'no':

    skip_rest('HTTP proxy tests', 'TLS build support not enabled')

try:

    import kdcproxy

except:

    skip_rest('HTTP proxy tests', 'Python kdcproxy module not found')

# Construct a krb5.conf fragment configuring the client to use a local proxy

# server.

proxysubjectpem = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs',

                               'proxy-subject.pem')

proxysanpem = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs',

                           'proxy-san.pem')

proxyidealpem = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs',

                             'proxy-ideal.pem')

proxywrongpem = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs',

                             'proxy-no-match.pem')

proxybadpem = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs',

                           'proxy-badsig.pem')

proxyca = os.path.join(srctop, 'tests', 'dejagnu', 'proxy-certs', 'ca.pem')

proxyurl = 'https://localhost:$port5/KdcProxy'

proxyurlupcase = 'https://LocalHost:$port5/KdcProxy'

proxyurl4 = 'https://127.0.0.1:$port5/KdcProxy'

proxyurl6 = 'https://[::1]:$port5/KdcProxy'

unanchored_krb5_conf = {'realms': {'$realm': {

                        'kdc': proxyurl,

                        'kpasswd_server': proxyurl}}}

anchored_name_krb5_conf = {'realms': {'$realm': {

                           'kdc': proxyurl,

                           'kpasswd_server': proxyurl,

                           'http_anchors': 'FILE:%s' % proxyca}}}

anchored_upcasename_krb5_conf = {'realms': {'$realm': {

                                 'kdc': proxyurlupcase,

                                 'kpasswd_server': proxyurlupcase,

                                 'http_anchors': 'FILE:%s' % proxyca}}}

anchored_kadmin_krb5_conf = {'realms': {'$realm': {

                             'kdc': proxyurl,

                             'admin_server': proxyurl,

                             'http_anchors': 'FILE:%s' % proxyca}}}

anchored_ipv4_krb5_conf = {'realms': {'$realm': {

                           'kdc': proxyurl4,

                           'kpasswd_server': proxyurl4,

                           'http_anchors': 'FILE:%s' % proxyca}}}

kpasswd_input = (password('user') + '\n' + password('user') + '\n' +

                 password('user') + '\n')

def start_proxy(realm, keycertpem):

    proxy_conf_path = os.path.join(realm.testdir, 'kdcproxy.conf')

    proxy_exec_path = os.path.join(srctop, 'util', 'wsgiref-kdcproxy.py')

    conf = open(proxy_conf_path, 'w')

    conf.write('[%s]\n' % realm.realm)

    conf.write('kerberos = kerberos://localhost:%d\n' % realm.portbase)

    conf.write('kpasswd = kpasswd://localhost:%d\n' % (realm.portbase + 2))

    conf.close()

    realm.env['KDCPROXY_CONFIG'] = proxy_conf_path

    cmd = [sys.executable, proxy_exec_path, str(realm.server_port()),

           keycertpem]

    return realm.start_server(cmd, sentinel='proxy server ready')

# Fail: untrusted issuer and hostname doesn't match.

mark('untrusted issuer, hostname mismatch')

output("running pass 1: issuer not trusted and hostname doesn't match\n")

realm = K5Realm(krb5_conf=unanchored_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxywrongpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: untrusted issuer, host name matches subject.

mark('untrusted issuer, hostname subject match')

output("running pass 2: subject matches, issuer not trusted\n")

realm = K5Realm(krb5_conf=unanchored_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxysubjectpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: untrusted issuer, host name matches subjectAltName.

mark('untrusted issuer, hostname SAN match')

output("running pass 3: subjectAltName matches, issuer not trusted\n")

realm = K5Realm(krb5_conf=unanchored_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxysanpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: untrusted issuer, certificate signature is bad.

mark('untrusted issuer, bad signature')

output("running pass 4: subject matches, issuer not trusted\n")

realm = K5Realm(krb5_conf=unanchored_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxybadpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: trusted issuer but hostname doesn't match.

mark('trusted issuer, hostname mismatch')

output("running pass 5: issuer trusted but hostname doesn't match\n")

realm = K5Realm(krb5_conf=anchored_name_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxywrongpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subject.

mark('trusted issuer, hostname subject match')

output("running pass 6: issuer trusted, subject matches\n")

realm = K5Realm(krb5_conf=anchored_name_krb5_conf, start_kadmind=True,

                get_creds=False)

proxy = start_proxy(realm, proxysubjectpem)

realm.kinit(realm.user_princ, password=password('user'))

realm.run([kvno, realm.host_princ])

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subjectAltName.

mark('trusted issuer, hostname SAN match')

output("running pass 7: issuer trusted, subjectAltName matches\n")

realm = K5Realm(krb5_conf=anchored_name_krb5_conf, start_kadmind=True,

                get_creds=False)

proxy = start_proxy(realm, proxysanpem)

realm.kinit(realm.user_princ, password=password('user'))

realm.run([kvno, realm.host_princ])

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

# Fail: certificate signature is bad.

mark('bad signature')

output("running pass 8: issuer trusted and subjectAltName matches, sig bad\n")

realm = K5Realm(krb5_conf=anchored_name_krb5_conf,

                get_creds=False,

		                create_host=False)

proxy = start_proxy(realm, proxybadpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: trusted issuer but IP doesn't match.

mark('trusted issuer, IP mismatch')

output("running pass 9: issuer trusted but no name matches IP\n")

realm = K5Realm(krb5_conf=anchored_ipv4_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxywrongpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Fail: trusted issuer, but subject does not match.

mark('trusted issuer, IP mismatch (hostname in subject)')

output("running pass 10: issuer trusted, but subject does not match IP\n")

realm = K5Realm(krb5_conf=anchored_ipv4_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxysubjectpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subjectAltName.

mark('trusted issuer, IP SAN match')

output("running pass 11: issuer trusted, subjectAltName matches IP\n")

realm = K5Realm(krb5_conf=anchored_ipv4_krb5_conf, start_kadmind=True,

                get_creds=False)

proxy = start_proxy(realm, proxysanpem)

realm.kinit(realm.user_princ, password=password('user'))

realm.run([kvno, realm.host_princ])

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

# Fail: certificate signature is bad.

mark('bad signature (IP hostname)')

output("running pass 12: issuer trusted, names don't match, signature bad\n")

realm = K5Realm(krb5_conf=anchored_ipv4_krb5_conf, get_creds=False,

                create_host=False)

proxy = start_proxy(realm, proxybadpem)

realm.kinit(realm.user_princ, password=password('user'), expected_code=1)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subject, using kadmin

# configuration to find kpasswdd.

mark('trusted issuer, hostname subject match (kadmin)')

output("running pass 13: issuer trusted, subject matches\n")

realm = K5Realm(krb5_conf=anchored_kadmin_krb5_conf, start_kadmind=True,

                get_creds=False, create_host=False)

proxy = start_proxy(realm, proxysubjectpem)

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subjectAltName, using

# kadmin configuration to find kpasswdd.

mark('trusted issuer, hostname SAN match (kadmin)')

output("running pass 14: issuer trusted, subjectAltName matches\n")

realm = K5Realm(krb5_conf=anchored_kadmin_krb5_conf, start_kadmind=True,

                get_creds=False, create_host=False)

proxy = start_proxy(realm, proxysanpem)

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

# Succeed: trusted issuer and host name matches subjectAltName (give or take

# case).

mark('trusted issuer, hostname SAN case-insensitive match')

output("running pass 15: issuer trusted, subjectAltName case-insensitive\n")

realm = K5Realm(krb5_conf=anchored_upcasename_krb5_conf, start_kadmind=True,

                get_creds=False, create_host=False)

proxy = start_proxy(realm, proxysanpem)

realm.run([kpasswd, realm.user_princ], input=kpasswd_input)

stop_daemon(proxy)

realm.stop()

success('MS-KKDCP proxy')