/* * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ /* @test * @bug 8195160 * @summary Test asynchronous close during a blocking write * @requires (os.family == "linux") * @library .. /test/lib * @build RsocketTest * @run main/othervm -Djava.net.preferIPv4Stack=true CloseDuringWrite * @key randomness */ import java.io.Closeable; import java.io.IOException; import java.nio.ByteBuffer; import java.nio.channels.*; import java.net.*; import java.util.concurrent.*; import java.util.Random; import jdk.net.Sockets; public class CloseDuringWrite { static final Random rand = new Random(); static ServerSocketChannel ssc; static SocketChannel source; static SocketAddress sa; /** * A task that closes a Closeable */ static class Closer implements Callable { final Closeable c; Closer(Closeable c) { this.c = c; } public Void call() throws IOException { c.close(); return null; } } public static void main(String[] args) throws Exception { if (!RsocketTest.isRsocketAvailable()) return; ScheduledExecutorService pool = Executors.newSingleThreadScheduledExecutor(); try { try { ssc = Sockets.openRdmaServerSocketChannel(); InetAddress lh = InetAddress.getLocalHost(); ssc.bind(new InetSocketAddress(lh, 0)); int port = ssc.socket().getLocalPort(); sa = new InetSocketAddress(lh, port); ByteBuffer bb = ByteBuffer.allocate(2 * 1024 * 1024); for (int i = 0; i < 20; i++) { try { source = Sockets.openRdmaSocketChannel(); Runnable runnable = new Runnable() { @Override public void run() { try { source.connect(sa); } catch (Exception e) { e.printStackTrace(); throw new RuntimeException("Test Failed"); } } }; Thread t = new Thread(runnable); t.start(); SocketChannel sink = ssc.accept(); // schedule channel to be closed Closer c = new Closer(source); int when = 1000 + rand.nextInt(2000); Future result = pool.schedule(c, when, TimeUnit.MILLISECONDS); // the write should either succeed or else throw a // ClosedChannelException (more likely an // AsynchronousCloseException) try { for (;;) { int limit = rand.nextInt(bb.capacity()); bb.position(0); bb.limit(limit); int n = source.write(bb); System.out.format("wrote %d, expected %d%n", n, limit); } } catch (ClosedChannelException expected) { System.out.println(expected + " (expected)"); } finally { result.get(); } } catch (Exception e) { e.printStackTrace(); throw new RuntimeException("Test Failed"); } } } catch (Exception e) { e.printStackTrace(); throw new RuntimeException("Test Failed"); } } finally { pool.shutdown(); } } }