Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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. */ package org.apache.cxf.systest.kerberos.wssec.spnego; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.net.URL; import javax.xml.namespace.QName; import javax.xml.ws.Service; import org.apache.commons.io.IOUtils; import org.apache.cxf.Bus; import org.apache.cxf.bus.spring.SpringBusFactory; import org.apache.cxf.systest.kerberos.common.SecurityTestUtil; import org.apache.cxf.testutil.common.AbstractBusClientServerTestBase; import org.apache.cxf.testutil.common.TestUtil; import org.apache.directory.server.annotations.CreateKdcServer; import org.apache.directory.server.annotations.CreateLdapServer; import org.apache.directory.server.annotations.CreateTransport; import org.apache.directory.server.core.annotations.ApplyLdifFiles; import org.apache.directory.server.core.annotations.CreateDS; import org.apache.directory.server.core.annotations.CreateIndex; import org.apache.directory.server.core.annotations.CreatePartition; import org.apache.directory.server.core.integ.AbstractLdapTestUnit; import org.apache.directory.server.core.integ.FrameworkRunner; import org.apache.directory.server.core.kerberos.KeyDerivationInterceptor; import org.apache.wss4j.dom.WSSConfig; import org.example.contract.doubleit.DoubleItPortType; import org.junit.Assert; import org.junit.Before; import org.junit.BeforeClass; import org.junit.runner.RunWith; /** * A set of tests for Spnego Tokens that use an Apache DS instance as the KDC. */ @RunWith(FrameworkRunner.class) //Define the DirectoryService @CreateDS(name = "AbstractKerberosTest-class", enableAccessControl = false, allowAnonAccess = false, enableChangeLog = true, partitions = { @CreatePartition(name = "example", suffix = "dc=example,dc=com", indexes = { @CreateIndex(attribute = "objectClass"), @CreateIndex(attribute = "dc"), @CreateIndex(attribute = "ou") }) }, additionalInterceptors = { KeyDerivationInterceptor.class }) @CreateLdapServer(transports = { @CreateTransport(protocol = "LDAP") }) @CreateKdcServer(transports = { // @CreateTransport(protocol = "TCP", address = "127.0.0.1", port=1024) @CreateTransport(protocol = "UDP", address = "127.0.0.1") }, primaryRealm = "service.ws.apache.org", kdcPrincipal = "krbtgt/service.ws.apache.org@service.ws.apache.org") //Inject an file containing entries @ApplyLdifFiles("kerberos.ldif") public class SpnegoTokenTest extends AbstractLdapTestUnit { static final String PORT = TestUtil.getPortNumber(Server.class); static final String STAX_PORT = TestUtil.getPortNumber(StaxServer.class); static final String PORT2 = TestUtil.getPortNumber(Server.class, 2); static final String STAX_PORT2 = TestUtil.getPortNumber(StaxServer.class, 2); private static final String NAMESPACE = "http://www.example.org/contract/DoubleIt"; private static final QName SERVICE_QNAME = new QName(NAMESPACE, "DoubleItService"); private static boolean unrestrictedPoliciesInstalled = SecurityTestUtil.checkUnrestrictedPoliciesInstalled(); private static boolean runTests; private static boolean portUpdated; @Before public void updatePort() throws Exception { if (!portUpdated) { String basedir = System.getProperty("basedir"); if (basedir == null) { basedir = new File(".").getCanonicalPath(); } // Read in krb5.conf and substitute in the correct port File f = new File(basedir + "/src/test/resources/krb5.conf"); FileInputStream inputStream = new FileInputStream(f); String content = IOUtils.toString(inputStream, "UTF-8"); inputStream.close(); content = content.replaceAll("port", "" + super.getKdcServer().getTransports()[0].getPort()); File f2 = new File(basedir + "/target/test-classes/krb5.conf"); FileOutputStream outputStream = new FileOutputStream(f2); IOUtils.write(content, outputStream, "UTF-8"); outputStream.close(); System.setProperty("java.security.krb5.conf", f2.getPath()); portUpdated = true; } } @BeforeClass public static void startServers() throws Exception { WSSConfig.init(); // // This test fails with the IBM JDK // if (!"IBM Corporation".equals(System.getProperty("java.vendor"))) { runTests = true; String basedir = System.getProperty("basedir"); if (basedir == null) { basedir = new File(".").getCanonicalPath(); } else { basedir += "/.."; } // System.setProperty("sun.security.krb5.debug", "true"); System.setProperty("java.security.auth.login.config", basedir + "/kerberos/src/test/resources/kerberos.jaas"); } // Launch servers org.junit.Assert.assertTrue("Server failed to launch", // run the server in the same process // set this to false to fork AbstractBusClientServerTestBase.launchServer(Server.class, true)); org.junit.Assert.assertTrue("Server failed to launch", // run the server in the same process // set this to false to fork AbstractBusClientServerTestBase.launchServer(StaxServer.class, true)); } @org.junit.AfterClass public static void cleanup() throws Exception { SecurityTestUtil.cleanup(); AbstractBusClientServerTestBase.stopAllServers(); } @org.junit.Test public void testSpnegoOverSymmetric() throws Exception { if (!runTests || !unrestrictedPoliciesInstalled) { return; } String portName = "DoubleItSpnegoSymmetricPort"; runKerberosTest(portName, false, PORT); runKerberosTest(portName, false, STAX_PORT); runKerberosTest(portName, true, PORT); runKerberosTest(portName, true, STAX_PORT); } @org.junit.Test public void testSpnegoOverSymmetricDerived() throws Exception { if (!runTests || !unrestrictedPoliciesInstalled) { return; } String portName = "DoubleItSpnegoSymmetricDerivedPort"; runKerberosTest(portName, false, PORT); runKerberosTest(portName, false, STAX_PORT); runKerberosTest(portName, true, PORT); runKerberosTest(portName, true, STAX_PORT); } @org.junit.Test public void testSpnegoOverSymmetricEncryptBeforeSigning() throws Exception { if (!runTests || !unrestrictedPoliciesInstalled) { return; } String portName = "DoubleItSpnegoSymmetricEncryptBeforeSigningPort"; runKerberosTest(portName, false, PORT); runKerberosTest(portName, false, STAX_PORT); runKerberosTest(portName, true, PORT); runKerberosTest(portName, true, STAX_PORT); } @org.junit.Test public void testSpnegoOverTransport() throws Exception { if (!runTests || !unrestrictedPoliciesInstalled) { return; } String portName = "DoubleItSpnegoTransportPort"; runKerberosTest(portName, false, PORT2); runKerberosTest(portName, false, STAX_PORT2); // // TODO Supporting streaming Snego outbound // runKerberosTest(portName, true, PORT2); // runKerberosTest(portName, true, STAX_PORT2); } @org.junit.Test public void testSpnegoOverTransportEndorsing() throws Exception { if (!runTests || !unrestrictedPoliciesInstalled) { return; } String portName = "DoubleItSpnegoTransportEndorsingPort"; runKerberosTest(portName, false, PORT2); runKerberosTest(portName, false, STAX_PORT2); // TODO Supporting streaming Spnego outbound // runKerberosTest(portName, true, PORT2); // runKerberosTest(portName, true, STAX_PORT2); } private void runKerberosTest(String portName, boolean streaming, String portNumber) throws Exception { SpringBusFactory bf = new SpringBusFactory(); URL busFile = SpnegoTokenTest.class.getResource("client.xml"); Bus bus = bf.createBus(busFile.toString()); SpringBusFactory.setDefaultBus(bus); SpringBusFactory.setThreadDefaultBus(bus); URL wsdl = SpnegoTokenTest.class.getResource("DoubleItSpnego.wsdl"); Service service = Service.create(wsdl, SERVICE_QNAME); QName portQName = new QName(NAMESPACE, portName); DoubleItPortType kerberosPort = service.getPort(portQName, DoubleItPortType.class); TestUtil.updateAddressPort(kerberosPort, portNumber); if (streaming) { SecurityTestUtil.enableStreaming(kerberosPort); } Assert.assertEquals(50, kerberosPort.doubleIt(25)); ((java.io.Closeable) kerberosPort).close(); bus.shutdown(true); } }