+ * Copyright May 30, 2012 + *
+ */ +public class DefaultSocketFactoryTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(DefaultSocketFactoryTest= Case.class); + = + private static boolean firstTime =3D true; + = + protected String host; + protected int port; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected TestInvocationHandler invocationHandler; + + = + public void setUp() throws Exception + { + if (firstTime) + { + firstTime =3D false; + Logger.getLogger("org.jboss.remoting").setLevel(Level.INFO); + Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO); + String pattern =3D "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n"; + PatternLayout layout =3D new PatternLayout(pattern); + ConsoleAppender consoleAppender =3D new ConsoleAppender(layout); + Logger.getRootLogger().addAppender(consoleAppender); = + } + = + String trustStoreFilePath =3D this.getClass().getResource("../.trust= store").getFile(); + System.setProperty("javax.net.ssl.trustStore", trustStoreFilePath); + System.setProperty("javax.net.ssl.trustStorePassword", "unit-tests-c= lient"); + String keyStoreFilePath =3D this.getClass().getResource("../.keystor= e").getFile(); + System.setProperty("javax.net.ssl.keyStore", keyStoreFilePath); + System.setProperty("javax.net.ssl.keyStorePassword", "unit-tests-ser= ver"); + System.setProperty("org.jboss.security.ignoreHttpsHost", "true"); + } + + = + public void tearDown() + { + } + = + = + public void testCustomSocketFactory() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSSLSocketFa= ctory()); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Verify custom SSLSocketFactory is being used. + assertTrue(client.getInvoker().getSocketFactory() instanceof TestSSL= SocketFactory); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testDefaultSocketFactory() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSSLSocketFa= ctory()); + clientConfig.put(HTTPSClientInvoker.USE_DEFAULT_SSL_SOCKET_FACTORY, = "true"); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Verify custom SSLSocketFactory is not being used. + assertFalse(client.getInvoker().getSocketFactory() instanceof TestSS= LSocketFactory); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testDefaultSocketFactoryWithParameterFalse() throws Throwab= le + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSSLSocketFa= ctory()); + clientConfig.put(HTTPSClientInvoker.USE_DEFAULT_SSL_SOCKET_FACTORY, = "false"); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Verify custom SSLSocketFactory is being used. + assertTrue(client.getInvoker().getSocketFactory() instanceof TestSSL= SocketFactory); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "https"; + } + + = + protected void setupServer() throws Exception + { + host =3D InetAddress.getLocalHost().getHostAddress(); + port =3D PortUtil.findFreePort(host); + locatorURI =3D getTransport() + "://" + host + ":" + port; + String metadata =3D System.getProperty("remoting.metadata"); + if (metadata !=3D null) + { + locatorURI +=3D "/?" + metadata; + } + serverLocator =3D new InvokerLocator(locatorURI); + log.info("Starting remoting server with locator uri of: " + locatorU= RI); + HashMap config =3D new HashMap(); + config.put(InvokerLocator.FORCE_REMOTE, "true"); + connector =3D new Connector(serverLocator, config); + connector.create(); + invocationHandler =3D new TestInvocationHandler(); + connector.addInvocationHandler("test", invocationHandler); + connector.start(); + } + = + = + protected void shutdownServer() throws Exception + { + if (connector !=3D null) + connector.stop(); + } + = + = + static class TestInvocationHandler implements ServerInvocationHandler + { + public void addListener(InvokerCallbackHandler callbackHandler) {} + public Object invoke(final InvocationRequest invocation) throws Thro= wable + { + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + = + public static class TestSSLSocketFactory extends SocketFactory + { + SocketFactory sf =3D SSLSocketFactory.getDefault(); + = + public TestSSLSocketFactory() + { + } + + public Socket createSocket() throws IOException, UnknownHostException + { + return sf.createSocket(); + } + = + public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException + { + return sf.createSocket(arg0, arg1); + } + + public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption + { + return sf.createSocket(arg0, arg1); + } + + public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, + UnknownHostException + { + return sf.createSocket(arg0, arg1, arg2, arg3); + } + + public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException + { + return sf.createSocket(arg0, arg1, arg2, arg3); + } = + } +} \ No newline at end of file --===============0392922705208535715==--