--- /dev/null 2018-06-30 13:49:24.634979840 -0700 +++ new/test/jdk/jdk/net/Sockets/rsocket/SocketChannel/CloseDuringWrite.java 2018-06-30 14:07:48.734165903 -0700 @@ -0,0 +1,130 @@ +/* + * 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(); + } + } +}