InterruptedStreamTest.java revision 3ec5e434af58fedddcb34e08dbd021bfb78bc69c
19ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson/* 29ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Copyright (C) 2011 The Android Open Source Project 39ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * 49ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Licensed under the Apache License, Version 2.0 (the "License"); 59ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * you may not use this file except in compliance with the License. 69ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * You may obtain a copy of the License at 79ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * 89ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * http://www.apache.org/licenses/LICENSE-2.0 99ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * 109ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Unless required by applicable law or agreed to in writing, software 119ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * distributed under the License is distributed on an "AS IS" BASIS, 129ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 139ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * See the License for the specific language governing permissions and 149ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * limitations under the License. 159ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson */ 169ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 179ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonpackage libcore.java.io; 189ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 199ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.IOException; 209ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.InputStream; 219ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.InterruptedIOException; 229ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.OutputStream; 239ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.PipedInputStream; 249ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.PipedOutputStream; 259ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.PipedReader; 269ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.io.PipedWriter; 279ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.net.InetSocketAddress; 289ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.net.Socket; 299ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.ByteBuffer; 309ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.ClosedByInterruptException; 313ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Oginoimport java.nio.channels.ClosedChannelException; 329ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.Pipe; 339ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.ReadableByteChannel; 349ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.ServerSocketChannel; 359ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.SocketChannel; 369ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport java.nio.channels.WritableByteChannel; 379ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonimport junit.framework.TestCase; 389ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 399ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson/** 409ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Test that interrupting a thread blocked on I/O causes that thread to throw 419ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * an InterruptedIOException. 429ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson */ 439ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilsonpublic final class InterruptedStreamTest extends TestCase { 449ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 459ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private static final int BUFFER_SIZE = 1024 * 1024; 469ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 479ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private Socket[] sockets; 489ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 499ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson @Override protected void tearDown() throws Exception { 509ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson if (sockets != null) { 519ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets[0].close(); 529ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets[1].close(); 539ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 541f3a0774e666448ceff2b8f2eebdee0e5e69a561Jesse Wilson Thread.interrupted(); // clear interrupted bit 559ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson super.tearDown(); 569ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 579ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 589ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedInputStream() throws Exception { 599ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedOutputStream out = new PipedOutputStream(); 609ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedInputStream in = new PipedInputStream(out); 619ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptInputStream(in); 629ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 639ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 649ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedOutputStream() throws Exception { 659ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedOutputStream out = new PipedOutputStream(); 669ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson new PipedInputStream(out); 679ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptOutputStream(out); 689ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 699ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 709ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedReader() throws Exception { 719ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedWriter writer = new PipedWriter(); 729ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedReader reader = new PipedReader(writer); 739ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReader(reader); 749ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 759ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 769ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedWriter() throws Exception { 779ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson final PipedWriter writer = new PipedWriter(); 789ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson new PipedReader(writer); 799ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptWriter(writer); 809ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 819ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 829ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptReadablePipeChannel() throws Exception { 839ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReadableChannel(Pipe.open().source()); 849ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 859ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 869ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptWritablePipeChannel() throws Exception { 879ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptWritableChannel(Pipe.open().sink()); 889ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 899ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 909ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptReadableSocketChannel() throws Exception { 919ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets = newSocketChannelPair(); 929ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReadableChannel(sockets[0].getChannel()); 939ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 949ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 959ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptWritableSocketChannel() throws Exception { 969ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets = newSocketChannelPair(); 973ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino testInterruptWritableChannel(sockets[0].getChannel()); 989ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 999ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1009ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson /** 1019ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Returns a pair of connected sockets backed by NIO socket channels. 1029ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson */ 1039ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private Socket[] newSocketChannelPair() throws IOException { 1049ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson ServerSocketChannel serverSocketChannel = ServerSocketChannel.open(); 1059ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson serverSocketChannel.socket().bind(new InetSocketAddress(0)); 1069ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson SocketChannel clientSocketChannel = SocketChannel.open(); 1079ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson clientSocketChannel.connect(serverSocketChannel.socket().getLocalSocketAddress()); 1089ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson SocketChannel server = serverSocketChannel.accept(); 1099ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson serverSocketChannel.close(); 1109ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson return new Socket[] { clientSocketChannel.socket(), server.socket() }; 1119ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1129ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1139ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptInputStream(final InputStream in) throws Exception { 1149ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1159ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1169ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson in.read(); 1179ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1189ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1199ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1209ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1219ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1229ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptReader(final PipedReader reader) throws Exception { 1239ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1249ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1259ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson reader.read(); 1269ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1279ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1289ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1299ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1309ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1319ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptReadableChannel(final ReadableByteChannel channel) throws Exception { 1329ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1339ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1349ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson channel.read(ByteBuffer.allocate(BUFFER_SIZE)); 1359ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1369ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (ClosedByInterruptException expected) { 1379ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1389ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1399ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1409ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptOutputStream(final OutputStream out) throws Exception { 1419ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1429ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1439ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1449ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1459ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson out.write(new byte[BUFFER_SIZE]); 1469ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1479ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1489ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1499ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1509ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1519ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptWriter(final PipedWriter writer) throws Exception { 1529ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1539ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1549ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1559ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1569ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson writer.write(new char[BUFFER_SIZE]); 1579ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1589ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1599ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1609ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1619ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1629ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptWritableChannel(final WritableByteChannel channel) throws Exception { 1639ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson interruptMeLater(); 1649ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1659ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1669ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1679ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson channel.write(ByteBuffer.allocate(BUFFER_SIZE)); 1689ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1699ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (ClosedByInterruptException expected) { 1703ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } catch (ClosedChannelException expected) { 1719ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1729ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1739ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1749ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void interruptMeLater() throws Exception { 1759ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson final Thread toInterrupt = Thread.currentThread(); 1769ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson new Thread(new Runnable () { 1779ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson @Override public void run() { 1783ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino try { 1793ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino Thread.sleep(1000); 1803ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } catch (InterruptedException ex) { 1813ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } 1829ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson toInterrupt.interrupt(); 1839ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1849ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson }).start(); 1859ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1869ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson} 187