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 17e5fea3d504609d22337a5311d3ce0e72314bceeeNarayan Kamathpackage org.apache.harmony.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 { 31318d6de23bec3024a85eb211f5843d925774622aElliott Hughes ServerSocketChannel serverChannel = ServerSocketChannel.open(); 32318d6de23bec3024a85eb211f5843d925774622aElliott Hughes 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(); 40318d6de23bec3024a85eb211f5843d925774622aElliott Hughes socket.bind(null); 41dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 42dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 43dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void accept() { 44dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread serverThread = new Thread(new Runnable() { 45dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void run() { 46dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes try { 47dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes while (serverChannel.accept() == null) { 48dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(1000); 49dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 50dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } catch (Exception e) {} 51dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 52dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes }); 53dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverThread.start(); 54dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 55dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 56dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void close() throws Exception{ 57dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes serverChannel.close(); 58dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 59dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 60dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 61dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void testSelectorAcceptAndRead() throws Exception { 62dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel0 = Selector.open(); 63dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel1 = Selector.open(); 64dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Server server = new Server(); 65318d6de23bec3024a85eb211f5843d925774622aElliott Hughes SelectionKey mkey0 = server.serverChannel.register(sel0, SelectionKey.OP_ACCEPT); 66318d6de23bec3024a85eb211f5843d925774622aElliott Hughes server.serverChannel.register(sel1, SelectionKey.OP_ACCEPT); 67dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 68dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // HUP is treating as acceptable 69dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel0.select(100)); 70dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, sel0.selectedKeys().contains(mkey0)); 71dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.initialize(); 72dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // after bind can not accept 73dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(0, sel1.select(100)); 74dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.accept(); 75dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(1000); 76dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SocketChannel socketChannel = SocketChannel.open(); 77dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.configureBlocking(false); 78dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel2 = Selector.open(); 79dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.register(sel2, SelectionKey.OP_WRITE); 80318d6de23bec3024a85eb211f5843d925774622aElliott Hughes boolean isConnected = socketChannel.connect(server.socket.getLocalSocketAddress()); 81dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes if (!isConnected) { 82dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel.finishConnect(); 83dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 84dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 85dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, socketChannel.isConnected()); 86dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes server.close(); 87dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Thread.sleep(3000); 88dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, socketChannel.isConnected()); 89dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel2.select(100)); 90dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 91dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 92dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes public void testSelectUnConnectedChannel() throws Exception { 93dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SocketChannel socketChannel2 = SocketChannel.open(); 94dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes socketChannel2.configureBlocking(false); 95dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel3 = Selector.open(); 96dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey3 = socketChannel2.register(sel3, SelectionKey.OP_WRITE); 97dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // HUP is also treating as writable 98dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel3.select(100)); 99dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, mkey3.isConnectable()); 100dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes // even the channel is not connected, the selector could be writable 101dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, socketChannel2.isConnected()); 102dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey3.isWritable()); 103dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 104dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel4 = Selector.open(); 105dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey4 = socketChannel2.register(sel4, SelectionKey.OP_CONNECT); 106dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel4.select(100)); 107dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(false, mkey4.isWritable()); 108dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey4.isConnectable()); 109dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes 110dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes Selector sel5 = Selector.open(); 111dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes SelectionKey mkey5 = socketChannel2.register(sel5, SelectionKey.OP_CONNECT | SelectionKey.OP_WRITE); 112dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(1, sel5.select(100)); 113dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey5.isWritable()); 114dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes assertEquals(true, mkey5.isConnectable()); 115dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes } 116dce2b2fa9d6b26414a8d5a55918e4d7ca2ab1baaElliott Hughes} 117