+ * Copyright Oct 13, 2010 + *
+ */ +public class DosTestCase extends TestCase +{ + private static final Logger log =3D Logger.getLogger(DosTestCase.class); + private static final String CALLBACK_TEST =3D "callbackTest"; + = + private static boolean firstTime =3D true; + = + protected String host; + protected int port; + protected int secondaryPort; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected TestInvocationHandler invocationHandler; + protected Object lock =3D new Object(); + protected boolean dosAttackThreadRan; + protected boolean secondCallbackRan; + + = + 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); = + } + } + + = + public void tearDown() + { + } + = + public void testDosAttack() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + final Client client =3D new Client(serverLocator, clientConfig); + client.connect(); + assertEquals("abc", client.invoke("abc")); + log.info("client is connected"); + = + // Add callback handler. + TestCallbackHandler callbackHandler =3D new TestCallbackHandler(); + HashMap metadata =3D new HashMap(); + metadata.put(Bisocket.IS_CALLBACK_SERVER, "true"); + client.addListener(callbackHandler, metadata); + client.invoke(CALLBACK_TEST); + assertEquals(1, callbackHandler.counter); + log.info("callback handler is installed"); + = + // Test DOS attack. = + new Thread() + { + public void run() + { + try + { + Socket s =3D new Socket(host, secondaryPort); + log.info(this + " created socket " + s); + synchronized (lock) + { + dosAttackThreadRan =3D true; + } = + } + catch (IOException e) + { + log.error("unable to connect to secondaryPort: " + secondar= yPort, e); + } + finally + { + synchronized (lock) + { + lock.notifyAll(); + } = + } + } + }.start(); + = + Thread.sleep(2000); + = + synchronized (lock) + { + if (!dosAttackThreadRan) + { + long start =3D System.currentTimeMillis(); + long end =3D start + 10000; + while (end - System.currentTimeMillis() > 0) + { + try + { + lock.wait(end - System.currentTimeMillis()); + } + catch (InterruptedException e){ + + } + } + } + } = + = + if (!dosAttackThreadRan) + { + fail("DOS attack thread did not run"); + } + = + = + // DOS attack has occurred. Try to add another callback handler. + new Thread() + { + public void run() + { + TestCallbackHandler callbackHandler2 =3D new TestCallbackHandl= er(); + try + { + = + HashMap metadata =3D new HashMap(); + metadata.put(Bisocket.IS_CALLBACK_SERVER, "true"); + client.addListener(callbackHandler2, metadata); + secondCallbackRan =3D true; + log.info(this + " second callback handler installed after D= OS attack"); + } + catch (Throwable e) + { + log.info(this + " second callback failed", e); + } = + } + }.start(); + = + Thread.sleep(10000); + assertTrue(secondCallbackRan); + = + client.removeListener(callbackHandler); + client.disconnect(); + connector.stop(); + } + = + protected String getTransport() + { + return "bisocket"; + } + = + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + protected void setupServer() throws Exception + { + host =3D InetAddress.getLocalHost().getHostAddress(); + port =3D PortUtil.findFreePort(host); + secondaryPort =3D PortUtil.findFreePort(host); + locatorURI =3D getTransport() + "://" + host + ":" + port + "/?secon= daryBindPort=3D" + secondaryPort; + 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"); + addExtraServerConfig(config); + 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 Set listeners =3D new HashSet(); + = + public void addListener(InvokerCallbackHandler callbackHandler) + { + listeners.add(callbackHandler); + } + public Object invoke(final InvocationRequest invocation) throws Thro= wable + { + if (CALLBACK_TEST.equals(invocation.getParameter())) + { + Iterator it =3D listeners.iterator(); + while (it.hasNext()) + { + InvokerCallbackHandler handler =3D (InvokerCallbackHandler)= it.next(); + handler.handleCallback(new Callback("test")); + log.info(this + " sent callback"); + } + } + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + static class TestCallbackHandler implements InvokerCallbackHandler + { + public int counter; + = + public void handleCallback(Callback callback) throws HandleCallbackE= xception + { + log.info(this + " received callback"); + counter++; + } + } +} \ No newline at end of file --===============3613961645420395417==--