1 /*
2 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 /* @test
25 * @bug 8195160
26 * @summary Unit test for server-socket channels
27 * @requires (os.family == "linux")
28 * @library .. /test/lib /test/jdk/java/nio/channels
29 * @build RsocketTest
30 * @run main/othervm Basic
31 */
32
33 import java.io.IOException;
34 import java.io.PrintStream;
35 import java.net.InetAddress;
36 import java.net.InetSocketAddress;
37 import java.net.StandardProtocolFamily;
38 import java.nio.ByteBuffer;
39 import java.nio.channels.ServerSocketChannel;
40 import java.nio.channels.SocketChannel;
41 import java.util.Random;
42 import jdk.net.RdmaSockets;
43
44 import jtreg.SkippedException;
45
46 public class Basic {
47
48 static PrintStream log = System.err;
49
50 static class Server
51 extends TestThread
52 {
53 ServerSocketChannel ssc;
54 boolean block;
55
56 Server(ServerSocketChannel ssc, boolean block) {
57 super("Server", Basic.log);
58 this.ssc = ssc;
59 this.block = block;
60 }
61
62 void go() throws Exception {
63 log.println("Server: Listening "
64 + (block ? "(blocking)" : "(non-blocking)"));
65 if (!block)
66 ssc.configureBlocking(false);
67 log.println(" " + ssc);
68 SocketChannel sc = null;
69 for (;;) {
70 sc = ssc.accept();
71 if (sc != null) {
72 break;
73 }
74 log.println("Server: Sleeping...");
75 Thread.sleep(50);
76 }
77 log.println("Server: Accepted " + sc);
78 ByteBuffer bb = ByteBuffer.allocateDirect(100);
79 if (sc.read(bb) != 1)
80 throw new Exception("Read failed");
81 bb.flip();
82 byte b = bb.get();
83 log.println("Server: Read " + b + ", writing " + (b + 1));
84 bb.clear();
85 bb.put((byte)43);
86 bb.flip();
87 if (sc.write(bb) != 1)
88 throw new Exception("Write failed");
89 sc.close();
90 ssc.close();
91 log.println("Server: Finished");
92 }
93
94 }
95
96 static class Client
97 extends TestThread
98 {
99 int port;
100 boolean dally;
101
102 Client(int port, boolean block) {
103 super("Client", Basic.log);
104 this.port = port;
105 this.dally = !block;
106 }
107
108 public void go() throws Exception {
109 if (dally)
110 Thread.sleep(200);
111 InetSocketAddress isa
112 = new InetSocketAddress(InetAddress.getLocalHost(), port);
113 log.println("Client: Connecting to " + isa);
114 SocketChannel sc = RdmaSockets.openSocketChannel(
115 StandardProtocolFamily.INET);
116 sc.connect(isa);
117 log.println("Client: Connected");
118 ByteBuffer bb = ByteBuffer.allocateDirect(512);
119 bb.put((byte)42).flip();
120 log.println("Client: Writing " + bb.get(0));
121 if (sc.write(bb) != 1)
122 throw new Exception("Write failed");
123 bb.clear();
124 if (sc.read(bb) != 1)
125 throw new Exception("Read failed");
126 bb.flip();
127 if (bb.get() != 43)
128 throw new Exception("Read " + bb.get(bb.position() - 1));
129 log.println("Client: Read " + bb.get(0));
130 sc.close();
131 log.println("Client: Finished");
132 }
133
134 }
135
136 static void test(boolean block) throws Exception {
137 ServerSocketChannel ssc = RdmaSockets.openServerSocketChannel(
138 StandardProtocolFamily.INET);
139 ssc.socket().setReuseAddress(true);
140 InetAddress lh = InetAddress.getLocalHost();
141 int port;
142 Random r = new Random();
143 for (;;) {
144 port = r.nextInt((1 << 16) - 1024) + 1024;
145 InetSocketAddress isa = new InetSocketAddress(lh, port);
146 try {
147 ssc.socket().bind(isa);
148 } catch (IOException x) {
149 continue;
150 }
151 break;
152 }
153
154 Server server = new Server(ssc, block);
155 Client client = new Client(port, block);
156 server.start();
157 client.start();
158 if ((server.finish(0) & client.finish(0)) == 0)
159 throw new Exception("Failure");
160 log.println();
161 }
162
163 public static void main(String[] args) throws Exception {
164 if (!RsocketTest.isRsocketAvailable())
165 throw new SkippedException("rsocket is not available");
166
167 log.println();
168 test(true);
169 test(false);
170 }
171
172 }