1 files changed, 97 insertions, 14 deletions
diff --git a/proxy/tests/t_impersonate.py b/proxy/tests/t_impersonate.py
index 43bb084..9bfd2cd 100644..100755
--- a/proxy/tests/t_impersonate.py
+++ b/proxy/tests/t_impersonate.py
@@ -3,21 +3,54 @@
 
 from testlib import *
 
-def run(testdir, env, conf, expected_failure=False):
-    print("Testing impersonate creds...", file=sys.stderr)
-    logfile = conf['logfile']
+IMPERSONATE_CONF_TEMPLATE = '''
+[gssproxy]
+  debug_level = 2
 
-    testenv = {'KRB5CCNAME': os.path.join(testdir, 't' + conf['prefix'] +
-                                                   '_impersonate.ccache'),
-               'KRB5_KTNAME': conf['keytab'],
-               'KRB5_TRACE': os.path.join(testdir, 't' + conf['prefix'] +
-                                                   '_impersonate.trace'),
-               'GSS_USE_PROXY': 'yes',
-               'GSSPROXY_BEHAVIOR': 'REMOTE_FIRST'}
-    testenv.update(env)
+[service/impersonate]
+  socket = ${TESTDIR}/impersonate.socket
+  mechs = krb5
+  cred_store = keytab:${GSSPROXY_KEYTAB}
+  cred_store = ccache:FILE:${GSSPROXY_CLIENT_CCACHE}
+  cred_store = client_keytab:${GSSPROXY_CLIENT_KEYTAB}
+  allow_protocol_transition = yes
+  allow_constrained_delegation = yes
+  euid = ${UIDNUMBER}
 
-    cmd = ["./tests/t_impersonate", USR_NAME, conf['svc_name']]
-    print("[COMMAND]\n%s\n[ENVIRONMENT]\n%s\n" % (cmd, env), file=logfile)
+[service/selfonly]
+  socket = ${TESTDIR}/impersonate-selfonly.socket
+  mechs = krb5
+  cred_store = keytab:${GSSPROXY_KEYTAB}
+  cred_store = ccache:FILE:${GSSPROXY_CLIENT_CCACHE}
+  cred_store = client_keytab:${GSSPROXY_CLIENT_KEYTAB}
+  allow_protocol_transition = yes
+  euid = ${UIDNUMBER}
+
+[service/proxyonly]
+  socket = ${TESTDIR}/impersonate-proxyonly.socket
+  mechs = krb5
+  cred_store = keytab:${GSSPROXY_KEYTAB}
+  cred_store = ccache:FILE:${GSSPROXY_CLIENT_CCACHE}
+  cred_store = client_keytab:${GSSPROXY_CLIENT_KEYTAB}
+  allow_constrained_delegation = yes
+  euid = ${UIDNUMBER}
+
+'''
+
+def run_cmd(testdir, env, conf, name, socket, cmd, expected_failure):
+
+    logfile = conf['logfile']
+    testenv = env.copy()
+    testenv.update({'KRB5CCNAME': os.path.join(testdir, 't' + conf['prefix'] +
+                                               '_impersonate.ccache'),
+                    'KRB5_KTNAME': os.path.join(testdir, PROXY_KTNAME),
+                    'KRB5_TRACE': os.path.join(testdir, 't' + conf['prefix'] +
+                                               '_impersonate.trace'),
+                    'GSS_USE_PROXY': 'yes',
+                    'GSSPROXY_SOCKET': socket,
+                    'GSSPROXY_BEHAVIOR': 'REMOTE_FIRST'})
+
+    print("[COMMAND]\n%s\n[ENVIRONMENT]\n%s\n" % (cmd, testenv), file=logfile)
     logfile.flush()
 
     p1 = subprocess.Popen(cmd, stderr=subprocess.STDOUT, stdout=logfile,
@@ -27,4 +60,54 @@ def run(testdir, env, conf, expected_failure=False):
     except subprocess.TimeoutExpired:
         # p1.returncode is set to None here
         pass
-    print_return(p1.returncode, "Impersonate", expected_failure)
+    print_return(p1.returncode, name, expected_failure)
+
+
+def run(testdir, env, conf, expected_failure=False):
+    print("Testing impersonate creds...", file=sys.stderr)
+    path_prefix = os.path.join(testdir, 't' + conf['prefix'] + '_')
+
+    # Change gssproxy conf for our test
+    keysenv = conf["keysenv"].copy()
+    keysenv['KRB5_KTNAME'] = os.path.join(testdir, PROXY_KTNAME)
+    update_gssproxy_conf(testdir, keysenv, IMPERSONATE_CONF_TEMPLATE)
+    os.kill(conf["gpid"], signal.SIGHUP)
+    time.sleep(1) #Let gssproxy reload everything
+
+    # Test all permitted
+    socket = os.path.join(testdir, 'impersonate.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, conf['svc_name']]
+    run_cmd(testdir, env, conf, "Impersonate", socket, cmd, False)
+
+    #Test fail
+    socket = os.path.join(testdir, 'impersonate-proxyonly.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, conf['svc_name']]
+    run_cmd(testdir, env, conf, "Impersonate fail self", socket, cmd, True)
+
+    #Test fail
+    socket = os.path.join(testdir, 'impersonate-selfonly.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, conf['svc_name']]
+    run_cmd(testdir, env, conf, "Impersonate fail proxy", socket, cmd, True)
+
+    #Test s4u2self half succeed
+    socket = os.path.join(testdir, 'impersonate-selfonly.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, conf['svc_name'], 's4u2self',
+           path_prefix + 'impersonate-proxy.ccache']
+    run_cmd(testdir, env, conf, "s4u2self delegation", socket, cmd, False)
+
+    #Test s4u2proxy half fail
+    socket = os.path.join(testdir, 'impersonate-selfonly.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, PROXY_GSS, 's4u2proxy',
+           path_prefix + 'impersonate-proxy.ccache']
+    run_cmd(testdir, env, conf, "s4u2proxy fail", socket, cmd, True)
+
+    #Test s4u2proxy half succeed
+    socket = os.path.join(testdir, 'impersonate-proxyonly.socket')
+    cmd = ["./tests/t_impersonate", USR_NAME, PROXY_GSS, 's4u2proxy',
+           path_prefix + 'impersonate-proxy.ccache']
+    run_cmd(testdir, env, conf, "s4u2proxy", socket, cmd, False)
+
+    # Reset back gssproxy conf
+    update_gssproxy_conf(testdir, keysenv, GSSPROXY_CONF_TEMPLATE)
+    os.kill(conf["gpid"], signal.SIGHUP)
+    time.sleep(1) #Let gssproxy reload everything