UnixSelectorTest.java revision dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baa
1dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes/* Licensed to the Apache Software Foundation (ASF) under one or more 2dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * contributor license agreements. See the NOTICE file distributed with 3dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * this work for additional information regarding copyright ownership. 4dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * The ASF licenses this file to You under the Apache License, Version 2.0 5dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * (the "License"); you may not use this file except in compliance with 6dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * the License. You may obtain a copy of the License at 7dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * 8dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * http://www.apache.org/licenses/LICENSE-2.0 9dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * 10dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * Unless required by applicable law or agreed to in writing, software 11dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * distributed under the License is distributed on an "AS IS" BASIS, 12dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * See the License for the specific language governing permissions and 14dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes * limitations under the License. 15dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes */ 16dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 17dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughespackage org.apache.harmony.nio.tests.java.nio.channels; 18dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 19dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.net.InetSocketAddress; 20dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.net.ServerSocket; 21dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.nio.channels.SelectableChannel; 22dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.nio.channels.SelectionKey; 23dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.nio.channels.Selector; 24dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.nio.channels.ServerSocketChannel; 25dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport java.nio.channels.SocketChannel; 26dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 27dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughesimport junit.framework.TestCase; 28dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 29dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughespublic class UnixSelectorTest extends TestCase { 30dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes static class Server { 31dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes private ServerSocketChannel serverChannel = ServerSocketChannel.open(); 32dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes private ServerSocket socket = null; 33dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 34dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Server() throws Exception { 35dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverChannel.configureBlocking(false); 36dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 37dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 38dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void initialize() throws Exception { 39dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes this.socket = serverChannel.socket(); 40dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socket.bind(new InetSocketAddress("localhost", 0)); 41dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 42dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 43dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public int getPort() { 44dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes return socket.getLocalPort(); 45dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 46dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 47dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public boolean isOpen() { 48dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes return !socket.isClosed(); 49dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 50dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 51dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public ServerSocketChannel getServerChannel() { 52dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes return serverChannel; 53dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 54dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 55dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void accept() { 56dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread serverThread = new Thread(new Runnable() { 57dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void run() { 58dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes try { 59dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes while (serverChannel.accept() == null) { 60dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(1000); 61dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 62dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } catch (Exception e) {} 63dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 64dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes }); 65dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverThread.start(); 66dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 67dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 68dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void close() throws Exception{ 69dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverChannel.close(); 70dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 71dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 72dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 73dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void testSelectorAcceptAndRead() throws Exception { 74dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel0 = Selector.open(); 75dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel1 = Selector.open(); 76dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Server server = new Server(); 77dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectableChannel serverChannel = server.getServerChannel(); 78dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey0 = serverChannel.register(sel0, SelectionKey.OP_ACCEPT); 79dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverChannel.register(sel1, SelectionKey.OP_ACCEPT); 80dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 81dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // HUP is treating as acceptable 82dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel0.select(100)); 83dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, sel0.selectedKeys().contains(mkey0)); 84dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.initialize(); 85dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // after bind can not accept 86dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(0, sel1.select(100)); 87dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.accept(); 88dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(1000); 89dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes int port = server.getPort(); 90dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SocketChannel socketChannel = SocketChannel.open(); 91dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.configureBlocking(false); 92dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel2 = Selector.open(); 93dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.register(sel2, SelectionKey.OP_WRITE); 94dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes boolean isConnected = socketChannel.connect(new InetSocketAddress("localhost", port)); 95dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes if (!isConnected) { 96dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.finishConnect(); 97dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 98dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 99dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, socketChannel.isConnected()); 100dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.close(); 101dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(3000); 102dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, socketChannel.isConnected()); 103dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel2.select(100)); 104dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 105dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 106dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void testSelectUnConnectedChannel() throws Exception { 107dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SocketChannel socketChannel2 = SocketChannel.open(); 108dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel2.configureBlocking(false); 109dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel3 = Selector.open(); 110dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey3 = socketChannel2.register(sel3, SelectionKey.OP_WRITE); 111dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // HUP is also treating as writable 112dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel3.select(100)); 113dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, mkey3.isConnectable()); 114dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // even the channel is not connected, the selector could be writable 115dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, socketChannel2.isConnected()); 116dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey3.isWritable()); 117dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 118dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel4 = Selector.open(); 119dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey4 = socketChannel2.register(sel4, SelectionKey.OP_CONNECT); 120dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel4.select(100)); 121dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, mkey4.isWritable()); 122dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey4.isConnectable()); 123dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 124dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel5 = Selector.open(); 125dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey5 = socketChannel2.register(sel5, SelectionKey.OP_CONNECT | SelectionKey.OP_WRITE); 126dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel5.select(100)); 127dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey5.isWritable()); 128dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey5.isConnectable()); 129dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 130dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes} 131