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 499726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom @Override protected void setUp() throws Exception { 509726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom // Clear the interrupted bit to make sure an earlier test did 519726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom // not leave us in a bad state. 529726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread.interrupted(); 539726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom super.tearDown(); 549726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } 559726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom 569ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson @Override protected void tearDown() throws Exception { 579ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson if (sockets != null) { 589ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets[0].close(); 599ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets[1].close(); 609ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 611f3a0774e666448ceff2b8f2eebdee0e5e69a561Jesse Wilson Thread.interrupted(); // clear interrupted bit 629ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson super.tearDown(); 639ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 649ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 659ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedInputStream() throws Exception { 669ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedOutputStream out = new PipedOutputStream(); 679ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedInputStream in = new PipedInputStream(out); 689ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptInputStream(in); 699ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 709ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 719ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedOutputStream() throws Exception { 729ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedOutputStream out = new PipedOutputStream(); 739ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson new PipedInputStream(out); 749ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptOutputStream(out); 759ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 769ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 779ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedReader() throws Exception { 789ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedWriter writer = new PipedWriter(); 799ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson PipedReader reader = new PipedReader(writer); 809ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReader(reader); 819ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 829ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 839ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptPipedWriter() throws Exception { 849ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson final PipedWriter writer = new PipedWriter(); 859ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson new PipedReader(writer); 869ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptWriter(writer); 879ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 889ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 899ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptReadablePipeChannel() throws Exception { 909ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReadableChannel(Pipe.open().source()); 919ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 929ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 939ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptWritablePipeChannel() throws Exception { 949ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptWritableChannel(Pipe.open().sink()); 959ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 969ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 979ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptReadableSocketChannel() throws Exception { 989ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets = newSocketChannelPair(); 999ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson testInterruptReadableChannel(sockets[0].getChannel()); 1009ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1019ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1029ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson public void testInterruptWritableSocketChannel() throws Exception { 1039ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson sockets = newSocketChannelPair(); 1043ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino testInterruptWritableChannel(sockets[0].getChannel()); 1059ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1069ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1079ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson /** 1089ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson * Returns a pair of connected sockets backed by NIO socket channels. 1099ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson */ 1109ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private Socket[] newSocketChannelPair() throws IOException { 1119ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson ServerSocketChannel serverSocketChannel = ServerSocketChannel.open(); 1129ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson serverSocketChannel.socket().bind(new InetSocketAddress(0)); 1139ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson SocketChannel clientSocketChannel = SocketChannel.open(); 1149ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson clientSocketChannel.connect(serverSocketChannel.socket().getLocalSocketAddress()); 1159ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson SocketChannel server = serverSocketChannel.accept(); 1169ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson serverSocketChannel.close(); 1179ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson return new Socket[] { clientSocketChannel.socket(), server.socket() }; 1189ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1199ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1209ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptInputStream(final InputStream in) throws Exception { 1219726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1229ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1239ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson in.read(); 1249ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1259ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1269726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1279726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1289ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1299ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1309ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1319ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptReader(final PipedReader reader) throws Exception { 1329726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1339ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1349ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson reader.read(); 1359ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1369ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1379726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1389726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1399ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1409ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1419ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1429ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptReadableChannel(final ReadableByteChannel channel) throws Exception { 1439726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1449ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1459ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson channel.read(ByteBuffer.allocate(BUFFER_SIZE)); 1469ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson fail(); 1479ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (ClosedByInterruptException expected) { 1489726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1499726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1509ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1519ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1529ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1539ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptOutputStream(final OutputStream out) throws Exception { 1549726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1559ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1569ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1579ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1589ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson out.write(new byte[BUFFER_SIZE]); 1599ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1609ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1619726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1629726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1639ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1649ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1659ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1669ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptWriter(final PipedWriter writer) throws Exception { 1679726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1689ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1699ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1709ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1719ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson writer.write(new char[BUFFER_SIZE]); 1729ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1739ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (InterruptedIOException expected) { 1749726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1759726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1769ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1779ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1789ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1799ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson private void testInterruptWritableChannel(final WritableByteChannel channel) throws Exception { 1809726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = interruptMeLater(); 1819ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson try { 1829ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson // this will block when the receiving buffer fills up 1839ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson while (true) { 1849ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson channel.write(ByteBuffer.allocate(BUFFER_SIZE)); 1859ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1869ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } catch (ClosedByInterruptException expected) { 1873ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } catch (ClosedChannelException expected) { 1889726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } finally { 1899726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom confirmInterrupted(thread); 1909ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1919ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 1929ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson 1939726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom private Thread interruptMeLater() throws Exception { 1949ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson final Thread toInterrupt = Thread.currentThread(); 1959726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom Thread thread = new Thread(new Runnable () { 1969ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson @Override public void run() { 1973ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino try { 1983ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino Thread.sleep(1000); 1993ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } catch (InterruptedException ex) { 2003ec5e434af58fedddcb34e08dbd021bfb78bc69cMasanori Ogino } 2019ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson toInterrupt.interrupt(); 2029ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 2039726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom }); 2049726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom thread.start(); 2059726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom return thread; 2069726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom } 2079726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom 2089726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom private static void confirmInterrupted(Thread thread) throws InterruptedException { 2099726be51ed6f9695357bc8f11a547d9beaa79fceBrian Carlstrom // validate and clear interrupted bit before join 210f0d40d662d9dfdb04215c718961765837d2cf00cNeil Fuller try { 211f0d40d662d9dfdb04215c718961765837d2cf00cNeil Fuller assertTrue(Thread.interrupted()); 212f0d40d662d9dfdb04215c718961765837d2cf00cNeil Fuller } finally { 213f0d40d662d9dfdb04215c718961765837d2cf00cNeil Fuller thread.join(); 214f0d40d662d9dfdb04215c718961765837d2cf00cNeil Fuller } 2159ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson } 2169ad09702f665937ec727de97d41d7854ee352ce0Jesse Wilson} 217