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 SocketOption
27 * @requires (os.family == "linux")
28 * @requires !vm.graal.enabled
29 * @library .. /test/lib
30 * @build RsocketTest
31 * @run main/othervm OptionsTest
32 */
33
34 import java.net.StandardProtocolFamily;
35 import java.net.ServerSocket;
36 import java.net.Socket;
37 import java.net.SocketOption;
38 import java.net.StandardSocketOptions;
39 import java.util.Formatter;
40 import java.util.Set;
41 import jdk.net.RdmaSockets;
42 import jdk.net.RdmaSocketOptions;
43
44 import jtreg.SkippedException;
45
46 public class OptionsTest {
47
48 static class Test {
49 Test(SocketOption<?> option, Object testValue) {
50 this.option = option;
51 this.testValue = testValue;
52 }
53 static Test create (SocketOption<?> option, Object testValue) {
54 return new Test(option, testValue);
55 }
56 Object option;
57 Object testValue;
58 }
59
60 // The tests set the option using the new API, read back the set value
61 // which could be diferent, and then use the legacy get API to check
62 // these values are the same
63
64 static Test[] socketTests = new Test[] {
65 Test.create(StandardSocketOptions.SO_SNDBUF, Integer.valueOf(10 * 100)),
66 Test.create(StandardSocketOptions.SO_RCVBUF, Integer.valueOf(8 * 100)),
67 Test.create(StandardSocketOptions.SO_REUSEADDR, Boolean.FALSE),
68 };
69
70 static Test[] serverSocketTests = new Test[] {
71 Test.create(StandardSocketOptions.SO_RCVBUF, Integer.valueOf(8 * 100)),
72 Test.create(StandardSocketOptions.SO_REUSEADDR, Boolean.FALSE),
73 };
74
75 static Test[] rdmaSocketOptionTests = new Test[] {
76 Test.create(RdmaSocketOptions.RDMA_SQSIZE, Integer.valueOf(8 * 100)),
77 Test.create(RdmaSocketOptions.RDMA_RQSIZE, Integer.valueOf(10 * 100)),
78 Test.create(RdmaSocketOptions.RDMA_INLINE, Integer.valueOf(20 * 100)),
79 };
80
81 static void doSocketTests() throws Exception {
82 try {
83 Socket c = RdmaSockets.openSocket(StandardProtocolFamily.INET);
84
85 for (int i = 0; i < socketTests.length; i++) {
86 Test test = socketTests[i];
87 c.setOption((SocketOption)test.option, test.testValue);
88 Object getval = c.getOption((SocketOption)test.option);
89 Object legacyget = legacyGetOption(Socket.class, c, test.option);
90 if (!getval.equals(legacyget)) {
91 Formatter f = new Formatter();
92 f.format("S Err %d: %s/%s", i, getval, legacyget);
93 throw new RuntimeException(f.toString());
94 }
95 }
96
97 for (int i = 0; i < rdmaSocketOptionTests.length; i++) {
98 Test test = rdmaSocketOptionTests[i];
99 c.setOption((SocketOption)test.option, test.testValue);
100 Object getval = c.getOption((SocketOption)test.option);
101 if (((Integer)getval).intValue() !=
102 ((Integer)test.testValue).intValue())
103 throw new RuntimeException("Test Failed");
104 }
105 } catch (Exception e) {
106 e.printStackTrace();
107 throw new RuntimeException("Test Failed");
108 }
109 }
110
111 static void doServerSocketTests() throws Exception {
112 try {
113 ServerSocket srv = RdmaSockets.openServerSocket(
114 StandardProtocolFamily.INET);
115 for (int i = 0; i < serverSocketTests.length; i++) {
116 Test test = serverSocketTests[i];
117 srv.setOption((SocketOption)test.option, test.testValue);
118 Object getval = srv.getOption((SocketOption)test.option);
119 Object legacyget = legacyGetOption(
120 ServerSocket.class, srv, test.option
121 );
122 if (!getval.equals(legacyget)) {
123 Formatter f = new Formatter();
124 f.format("SS Err %d: %s/%s", i, getval, legacyget);
125 throw new RuntimeException(f.toString());
126 }
127 }
128
129 for (int i = 0; i < rdmaSocketOptionTests.length; i++) {
130 Test test = rdmaSocketOptionTests[i];
131 srv.setOption((SocketOption)test.option, test.testValue);
132 Object getval = srv.getOption((SocketOption)test.option);
133 if (((Integer)getval).intValue() !=
134 ((Integer)test.testValue).intValue())
135 throw new RuntimeException("Test Failed");
136 }
137 } catch (Exception e) {
138 e.printStackTrace();
139 throw new RuntimeException("Test Failed");
140 }
141 }
142
143 static Object legacyGetOption(
144 Class<?> type, Object s, Object option)
145
146 throws Exception
147 {
148 if (type.equals(Socket.class)) {
149 Socket socket = (Socket)s;
150 Set<SocketOption<?>> options = socket.supportedOptions();
151
152 if (option.equals(StandardSocketOptions.SO_SNDBUF)) {
153 return Integer.valueOf(socket.getSendBufferSize());
154 } else if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
155 return Integer.valueOf(socket.getReceiveBufferSize());
156 } else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
157 return Boolean.valueOf(socket.getReuseAddress());
158 } else if (option.equals(StandardSocketOptions.TCP_NODELAY)) {
159 return Boolean.valueOf(socket.getTcpNoDelay());
160 } else {
161 throw new RuntimeException("unexecpted socket option");
162 }
163 } else if (type.equals(ServerSocket.class)) {
164 ServerSocket socket = (ServerSocket)s;
165 Set<SocketOption<?>> options = socket.supportedOptions();
166
167 if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
168 return Integer.valueOf(socket.getReceiveBufferSize());
169 } else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
170 return Boolean.valueOf(socket.getReuseAddress());
171 } else {
172 throw new RuntimeException("unexecpted socket option");
173 }
174 }
175 throw new RuntimeException("unexecpted socket type");
176 }
177
178 public static void main(String args[]) throws Exception {
179 if (!RsocketTest.isRsocketAvailable())
180 throw new SkippedException("rsocket is not available");
181
182 doSocketTests();
183 doServerSocketTests();
184 }
185 }