blob: 72609ee1c34a29088cec10267ad96dd8329b9d69 [file] [log] [blame]
/*
* Copyright (C) 2010 The Android Open Source Project
*
* Licensed 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 libcore.java.nio.channels;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.net.BindException;
import java.net.ConnectException;
import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketImpl;
import java.net.StandardSocketOptions;
import java.nio.ByteBuffer;
import java.nio.channels.AlreadyBoundException;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.NotYetConnectedException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
public class SocketChannelTest extends junit.framework.TestCase {
public void test_read_intoReadOnlyByteArrays() throws Exception {
ByteBuffer readOnly = ByteBuffer.allocate(1).asReadOnlyBuffer();
ServerSocket ss = new ServerSocket(0);
ss.setReuseAddress(true);
SocketChannel sc = SocketChannel.open(ss.getLocalSocketAddress());
try {
sc.read(readOnly);
fail();
} catch (IllegalArgumentException expected) {
}
try {
sc.read(new ByteBuffer[] { readOnly });
fail();
} catch (IllegalArgumentException expected) {
}
try {
sc.read(new ByteBuffer[] { readOnly }, 0, 1);
fail();
} catch (IllegalArgumentException expected) {
}
}
// https://code.google.com/p/android/issues/detail?id=56684
public void test_56684() throws Exception {
SocketChannel sc = SocketChannel.open();
sc.configureBlocking(false);
Selector selector = Selector.open();
SelectionKey selectionKey = sc.register(selector, SelectionKey.OP_CONNECT);
try {
// This test originally mocked the connect syscall to return ENETUNREACH.
// This is not easily doable in openJdk libcore, but a connect to broadcast
// address (255.255.255.255) for a TCP connection produces ENETUNREACH
// Kernel code that does it is at
// http://lxr.free-electrons.com/source/net/ipv4/tcp_ipv4.c?v=3.18#L182
sc.connect(new InetSocketAddress(InetAddress.getByAddress(new byte[] {
(byte) 255, (byte) 255, (byte) 255, (byte) 255 }), 0));
fail();
} catch (ConnectException ex) {
}
try {
sc.finishConnect();
fail();
} catch (ClosedChannelException expected) {
}
}
/** Checks that closing a Socket's output stream also closes the Socket and SocketChannel. */
public void test_channelSocketOutputStreamClosureState() throws Exception {
ServerSocket ss = new ServerSocket(0);
SocketChannel sc = SocketChannel.open(ss.getLocalSocketAddress());
sc.configureBlocking(true);
Socket scSocket = sc.socket();
OutputStream os = scSocket.getOutputStream();
assertTrue(sc.isOpen());
assertFalse(scSocket.isClosed());
os.close();
assertFalse(sc.isOpen());
assertTrue(scSocket.isClosed());
ss.close();
}
/** Checks that closing a Socket's input stream also closes the Socket and SocketChannel. */
public void test_channelSocketInputStreamClosureState() throws Exception {
ServerSocket ss = new ServerSocket(0);
SocketChannel sc = SocketChannel.open(ss.getLocalSocketAddress());
sc.configureBlocking(true);
Socket scSocket = sc.socket();
InputStream is = scSocket.getInputStream();
assertTrue(sc.isOpen());
assertFalse(scSocket.isClosed());
is.close();
assertFalse(sc.isOpen());
assertTrue(scSocket.isClosed());
ss.close();
}
/** Checks the state of the SocketChannel and associated Socket after open() */
public void test_open_initialState() throws Exception {
SocketChannel sc = SocketChannel.open();
try {
assertNull(sc.socket().getLocalSocketAddress());
Socket socket = sc.socket();
assertFalse(socket.isBound());
assertFalse(socket.isClosed());
assertFalse(socket.isConnected());
assertEquals(-1, socket.getLocalPort());
assertTrue(socket.getLocalAddress().isAnyLocalAddress());
assertNull(socket.getLocalSocketAddress());
assertNull(socket.getInetAddress());
assertEquals(0, socket.getPort());
assertNull(socket.getRemoteSocketAddress());
assertFalse(socket.getReuseAddress());
assertSame(sc, socket.getChannel());
} finally {
sc.close();
}
}
public void test_bind_unresolvedAddress() throws IOException {
SocketChannel sc = SocketChannel.open();
try {
sc.socket().bind(new InetSocketAddress("unresolvedname", 31415));
fail();
} catch (IOException expected) {
}
assertNull(sc.socket().getLocalSocketAddress());
assertTrue(sc.isOpen());
assertFalse(sc.isConnected());
sc.close();
}
/** Checks that the SocketChannel and associated Socket agree on the socket state. */
public void test_bind_socketStateSync() throws IOException {
SocketChannel sc = SocketChannel.open();
assertNull(sc.socket().getLocalSocketAddress());
Socket socket = sc.socket();
assertNull(socket.getLocalSocketAddress());
assertFalse(socket.isBound());
InetSocketAddress bindAddr = new InetSocketAddress("localhost", 0);
sc.socket().bind(bindAddr);
InetSocketAddress actualAddr = (InetSocketAddress) sc.socket().getLocalSocketAddress();
assertEquals(actualAddr, socket.getLocalSocketAddress());
assertEquals(bindAddr.getHostName(), actualAddr.getHostName());
assertTrue(socket.isBound());
assertFalse(socket.isConnected());
assertFalse(socket.isClosed());
sc.close();
assertFalse(sc.isOpen());
assertTrue(socket.isClosed());
}
/**
* Checks that the SocketChannel and associated Socket agree on the socket state, even if
* the Socket object is requested/created after bind().
*/
public void test_bind_socketObjectCreationAfterBind() throws IOException {
SocketChannel sc = SocketChannel.open();
assertNull(sc.socket().getLocalSocketAddress());
InetSocketAddress bindAddr = new InetSocketAddress("localhost", 0);
sc.socket().bind(bindAddr);
// Socket object creation after bind().
Socket socket = sc.socket();
InetSocketAddress actualAddr = (InetSocketAddress) sc.socket().getLocalSocketAddress();
assertEquals(actualAddr, socket.getLocalSocketAddress());
assertEquals(bindAddr.getHostName(), actualAddr.getHostName());
assertTrue(socket.isBound());
assertFalse(socket.isConnected());
assertFalse(socket.isClosed());
sc.close();
assertFalse(sc.isOpen());
assertTrue(socket.isClosed());
}
/**
* Tests connect() and object state for a blocking SocketChannel. Blocking mode is the default.
*/
public void test_connect_blocking() throws Exception {
ServerSocket ss = new ServerSocket(0);
SocketChannel sc = SocketChannel.open();
assertTrue(sc.isBlocking());
assertTrue(sc.connect(ss.getLocalSocketAddress()));
assertTrue(sc.socket().isBound());
assertTrue(sc.isConnected());
assertTrue(sc.socket().isConnected());
assertFalse(sc.socket().isClosed());
assertTrue(sc.isBlocking());
ss.close();
sc.close();
}
/** Tests connect() and object state for a non-blocking SocketChannel. */
public void test_connect_nonBlocking() throws Exception {
ServerSocket ss = new ServerSocket(0);
SocketChannel sc = SocketChannel.open();
assertTrue(sc.isBlocking());
sc.configureBlocking(false);
assertFalse(sc.isBlocking());
if (!sc.connect(ss.getLocalSocketAddress())) {
do {
assertTrue(sc.socket().isBound());
assertFalse(sc.isConnected());
assertFalse(sc.socket().isConnected());
assertFalse(sc.socket().isClosed());
} while (!sc.finishConnect());
}
assertTrue(sc.socket().isBound());
assertTrue(sc.isConnected());
assertTrue(sc.socket().isConnected());
assertFalse(sc.socket().isClosed());
assertFalse(sc.isBlocking());
ss.close();
sc.close();
}
public void test_Socket_impl_notNull() throws Exception {
SocketChannel sc = SocketChannel.open();
Socket socket = sc.socket();
Field f_impl = Socket.class.getDeclaredField("impl");
f_impl.setAccessible(true);
Object implFieldValue = f_impl.get(socket);
assertNotNull(implFieldValue);
assertTrue(implFieldValue instanceof SocketImpl);
}
public void test_setOption() throws Exception {
SocketChannel sc = SocketChannel.open();
sc.setOption(StandardSocketOptions.SO_LINGER, 1000);
// Assert that we can read back the option from the channel...
assertEquals(1000, (int) sc.<Integer>getOption(StandardSocketOptions.SO_LINGER));
// ... and its socket adaptor.
assertEquals(1000, sc.socket().getSoLinger());
sc.close();
try {
sc.setOption(StandardSocketOptions.SO_LINGER, 2000);
fail();
} catch (ClosedChannelException expected) {
}
}
public void test_bind() throws IOException {
InetSocketAddress socketAddress = new InetSocketAddress(Inet4Address.LOOPBACK, 0);
SocketChannel sc = SocketChannel.open();
sc.bind(socketAddress);
assertEquals(socketAddress.getAddress(),
((InetSocketAddress) (sc.getLocalAddress())).getAddress());
assertTrue(((InetSocketAddress) (sc.getLocalAddress())).getPort() > 0);
try {
sc.bind(socketAddress);
fail();
} catch (AlreadyBoundException expected) {
}
socketAddress = new InetSocketAddress(Inet4Address.LOOPBACK,
((InetSocketAddress) (sc.getLocalAddress())).getPort());
try {
SocketChannel.open().bind(socketAddress);
fail();
} catch (BindException expected) {
}
sc.close();
socketAddress = new InetSocketAddress(Inet4Address.LOOPBACK, 0);
try {
sc.bind(socketAddress);
} catch (ClosedChannelException expected) {
}
}
public void test_getRemoteAddress() throws IOException {
SocketChannel sc = SocketChannel.open();
ServerSocket ss = new ServerSocket(0);
assertNull(sc.getRemoteAddress());
sc.connect(ss.getLocalSocketAddress());
assertEquals(sc.getRemoteAddress(), ss.getLocalSocketAddress());
}
public void test_shutdownInput() throws IOException {
SocketChannel channel1 = SocketChannel.open();
ServerSocket server1 = new ServerSocket(0);
InetSocketAddress localAddr1 = new InetSocketAddress("127.0.0.1", server1.getLocalPort());
// initialize write content
byte[] writeContent = new byte[10];
for (int i = 0; i < writeContent.length; i++) {
writeContent[i] = (byte) i;
}
// establish connection
channel1.connect(localAddr1);
Socket acceptedSocket = server1.accept();
// use OutputStream.write to write bytes data.
OutputStream out = acceptedSocket.getOutputStream();
out.write(writeContent);
// use close to guarantee all data is sent
acceptedSocket.close();
channel1.configureBlocking(false);
ByteBuffer readContent = ByteBuffer.allocate(10 + 1);
channel1.shutdownInput();
assertEquals(-1, channel1.read(readContent));
}
public void test_shutdownOutput() throws IOException {
SocketChannel channel1 = SocketChannel.open();
ServerSocket server1 = new ServerSocket(0);
InetSocketAddress localAddr1 = new InetSocketAddress("127.0.0.1", server1.getLocalPort());
// initialize write content
ByteBuffer writeContent = ByteBuffer.allocate(10);
for (int i = 0; i < 10; i++) {
writeContent.put((byte) i);
}
writeContent.flip();
try {
channel1.shutdownOutput();
fail();
} catch (NotYetConnectedException expected) {}
// establish connection
channel1.connect(localAddr1);
channel1.shutdownOutput();
try {
channel1.write(writeContent);
fail();
} catch (ClosedChannelException expected) {}
channel1.close();
try {
channel1.shutdownOutput();
fail();
} catch(ClosedChannelException expected) {}
}
}