--- /dev/null 2018-11-27 03:18:47.532777276 -0500 +++ new/test/jdk/jdk/net/RdmaSockets/rsocket/SocketChannel/CloseDuringWrite.java 2018-11-30 08:33:24.106538406 -0500 @@ -0,0 +1,144 @@ +/* + * 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 CloseDuringWrite + * @key randomness + */ + +import java.io.Closeable; +import java.io.IOException; +import java.net.InetAddress; +import java.net.InetSocketAddress; +import java.net.StandardProtocolFamily; +import java.net.SocketAddress; +import java.nio.ByteBuffer; +import java.nio.channels.ClosedChannelException; +import java.nio.channels.ServerSocketChannel; +import java.nio.channels.SocketChannel; +import java.util.concurrent.Callable; +import java.util.concurrent.Executors; +import java.util.concurrent.Future; +import java.util.concurrent.ScheduledExecutorService; +import java.util.concurrent.TimeUnit; +import java.util.Random; +import jdk.net.RdmaSockets; + +import jtreg.SkippedException; + +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()) + throw new SkippedException("rsocket is not available"); + + ScheduledExecutorService pool = Executors.newSingleThreadScheduledExecutor(); + try { + try { + ssc = RdmaSockets.openServerSocketChannel( + StandardProtocolFamily.INET); + 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 = RdmaSockets.openSocketChannel( + StandardProtocolFamily.INET); + 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(); + } + } +}