/*
* Copyright (c) 2014, 2017, 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 8036979 8072384 8044773
* @requires !vm.graal.enabled
* @run main/othervm -Xcheck:jni OptionsTest
* @run main/othervm -Xcheck:jni -Djava.net.preferIPv4Stack=true OptionsTest
* @run main/othervm --limit-modules=java.base OptionsTest
*/
package test.java.net.SocketOptions;
import java.lang.reflect.Method;
import java.net.*;
import java.util.*;
import org.testng.annotations.Test;
public class OptionsTest {
static class TestClass {
TestClass(SocketOption<?> option, Object testValue) {
this.option = option;
this.testValue = testValue;
}
static TestClass create(SocketOption<?> option, Object testValue) {
return new TestClass(option, testValue);
}
Object option;
Object testValue;
}
// The tests set the option using the new API, read back the set value
// which could be diferent, and then use the legacy get API to check
// these values are the same
static TestClass[] socketTests = new TestClass[] {
TestClass.create(StandardSocketOptions.SO_KEEPALIVE, Boolean.TRUE),
TestClass.create(StandardSocketOptions.SO_SNDBUF, Integer.valueOf(10 * 100)),
TestClass.create(StandardSocketOptions.SO_RCVBUF, Integer.valueOf(8 * 100)),
TestClass.create(StandardSocketOptions.SO_REUSEADDR, Boolean.FALSE),
TestClass.create(StandardSocketOptions.SO_REUSEPORT, Boolean.FALSE),
TestClass.create(StandardSocketOptions.SO_LINGER, Integer.valueOf(80)),
TestClass.create(StandardSocketOptions.IP_TOS, Integer.valueOf(100))
};
static TestClass[] serverSocketTests = new TestClass[] {
TestClass.create(StandardSocketOptions.SO_RCVBUF, Integer.valueOf(8 * 100)),
TestClass.create(StandardSocketOptions.SO_REUSEADDR, Boolean.FALSE),
TestClass.create(StandardSocketOptions.SO_REUSEPORT, Boolean.FALSE),
TestClass.create(StandardSocketOptions.IP_TOS, Integer.valueOf(100))
};
static TestClass[] dgSocketTests = new TestClass[] {
TestClass.create(StandardSocketOptions.SO_SNDBUF, Integer.valueOf(10 * 100)),
TestClass.create(StandardSocketOptions.SO_RCVBUF, Integer.valueOf(8 * 100)),
TestClass.create(StandardSocketOptions.SO_REUSEADDR, Boolean.FALSE),
TestClass.create(StandardSocketOptions.SO_REUSEPORT, Boolean.FALSE),
TestClass.create(StandardSocketOptions.IP_TOS, Integer.valueOf(100))
};
static TestClass[] mcSocketTests = new TestClass[] {
// Android-removed: unused in Android
// TestClass.create(StandardSocketOptions.IP_MULTICAST_IF, getNetworkInterface()),
TestClass.create(StandardSocketOptions.IP_MULTICAST_TTL, Integer.valueOf(10)),
TestClass.create(StandardSocketOptions.IP_MULTICAST_LOOP, Boolean.TRUE)
};
static void doSocketTests() throws Exception {
try (
ServerSocket srv = new ServerSocket(0);
Socket c = new Socket("127.0.0.1", srv.getLocalPort());
Socket s = srv.accept();
) {
Set<SocketOption<?>> options = c.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
for (int i=0; i<socketTests.length; i++) {
TestClass test = socketTests[i];
if (!(test.option == StandardSocketOptions.SO_REUSEPORT && !reuseport)) {
c.setOption((SocketOption)test.option, test.testValue);
Object getval = c.getOption((SocketOption)test.option);
Object legacyget = legacyGetOption(Socket.class, c,test.option);
if (!getval.equals(legacyget)) {
Formatter f = new Formatter();
f.format("S Err %d: %s/%s", i, getval, legacyget);
throw new RuntimeException(f.toString());
}
}
}
}
}
static void doDgSocketTests() throws Exception {
try (
DatagramSocket c = new DatagramSocket(0);
) {
Set<SocketOption<?>> options = c.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
for (int i=0; i<dgSocketTests.length; i++) {
TestClass test = dgSocketTests[i];
if (!(test.option == StandardSocketOptions.SO_REUSEPORT && !reuseport)) {
c.setOption((SocketOption)test.option, test.testValue);
Object getval = c.getOption((SocketOption)test.option);
Object legacyget = legacyGetOption(DatagramSocket.class, c,test.option);
if (!getval.equals(legacyget)) {
Formatter f = new Formatter();
f.format("DG Err %d: %s/%s", i, getval, legacyget);
throw new RuntimeException(f.toString());
}
}
}
}
}
static void doMcSocketTests() throws Exception {
try (
MulticastSocket c = new MulticastSocket(0);
) {
for (int i=0; i<mcSocketTests.length; i++) {
TestClass test = mcSocketTests[i];
c.setOption((SocketOption)test.option, test.testValue);
Object getval = c.getOption((SocketOption)test.option);
Object legacyget = legacyGetOption(MulticastSocket.class, c,test.option);
if (!getval.equals(legacyget)) {
Formatter f = new Formatter();
f.format("MC Err %d: %s/%s", i, getval, legacyget);
throw new RuntimeException(f.toString());
}
}
}
}
static void doServerSocketTests() throws Exception {
try (
ServerSocket c = new ServerSocket(0);
) {
Set<SocketOption<?>> options = c.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
for (int i=0; i<serverSocketTests.length; i++) {
TestClass test = serverSocketTests[i];
if (!(test.option == StandardSocketOptions.SO_REUSEPORT && !reuseport)) {
c.setOption((SocketOption)test.option, test.testValue);
Object getval = c.getOption((SocketOption)test.option);
Object legacyget = legacyGetOption(
ServerSocket.class, c, test.option
);
if (!getval.equals(legacyget)) {
Formatter f = new Formatter();
f.format("SS Err %d: %s/%s", i, getval, legacyget);
throw new RuntimeException(f.toString());
}
}
}
}
}
static Object legacyGetOption(
Class<?> type, Object s, Object option)
throws Exception
{
if (type.equals(Socket.class)) {
Socket socket = (Socket)s;
Set<SocketOption<?>> options = socket.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
if (option.equals(StandardSocketOptions.SO_KEEPALIVE)) {
return Boolean.valueOf(socket.getKeepAlive());
} else if (option.equals(StandardSocketOptions.SO_SNDBUF)) {
return Integer.valueOf(socket.getSendBufferSize());
} else if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
return Integer.valueOf(socket.getReceiveBufferSize());
} else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
return Boolean.valueOf(socket.getReuseAddress());
} else if (option.equals(StandardSocketOptions.SO_REUSEPORT) && reuseport) {
return Boolean.valueOf(socket.getOption(StandardSocketOptions.SO_REUSEPORT));
} else if (option.equals(StandardSocketOptions.SO_LINGER)) {
return Integer.valueOf(socket.getSoLinger());
} else if (option.equals(StandardSocketOptions.IP_TOS)) {
return Integer.valueOf(socket.getTrafficClass());
} else if (option.equals(StandardSocketOptions.TCP_NODELAY)) {
return Boolean.valueOf(socket.getTcpNoDelay());
} else {
throw new RuntimeException("unexecpted socket option");
}
} else if (type.equals(ServerSocket.class)) {
ServerSocket socket = (ServerSocket)s;
Set<SocketOption<?>> options = socket.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
return Integer.valueOf(socket.getReceiveBufferSize());
} else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
return Boolean.valueOf(socket.getReuseAddress());
} else if (option.equals(StandardSocketOptions.SO_REUSEPORT) && reuseport) {
return Boolean.valueOf(socket.getOption(StandardSocketOptions.SO_REUSEPORT));
} else if (option.equals(StandardSocketOptions.IP_TOS)) {
return getServerSocketTrafficClass(socket);
} else {
throw new RuntimeException("unexecpted socket option");
}
} else if (type.equals(DatagramSocket.class)) {
DatagramSocket socket = (DatagramSocket)s;
Set<SocketOption<?>> options = socket.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
if (option.equals(StandardSocketOptions.SO_SNDBUF)) {
return Integer.valueOf(socket.getSendBufferSize());
} else if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
return Integer.valueOf(socket.getReceiveBufferSize());
} else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
return Boolean.valueOf(socket.getReuseAddress());
} else if (option.equals(StandardSocketOptions.SO_REUSEPORT) && reuseport) {
return Boolean.valueOf(socket.getOption(StandardSocketOptions.SO_REUSEPORT));
} else if (option.equals(StandardSocketOptions.IP_TOS)) {
return Integer.valueOf(socket.getTrafficClass());
} else {
throw new RuntimeException("unexecpted socket option");
}
} else if (type.equals(MulticastSocket.class)) {
MulticastSocket socket = (MulticastSocket)s;
Set<SocketOption<?>> options = socket.supportedOptions();
boolean reuseport = options.contains(StandardSocketOptions.SO_REUSEPORT);
if (option.equals(StandardSocketOptions.SO_SNDBUF)) {
return Integer.valueOf(socket.getSendBufferSize());
} else if (option.equals(StandardSocketOptions.SO_RCVBUF)) {
return Integer.valueOf(socket.getReceiveBufferSize());
} else if (option.equals(StandardSocketOptions.SO_REUSEADDR)) {
return Boolean.valueOf(socket.getReuseAddress());
} else if (option.equals(StandardSocketOptions.SO_REUSEPORT) && reuseport) {
return Boolean.valueOf(socket.getOption(StandardSocketOptions.SO_REUSEPORT));
} else if (option.equals(StandardSocketOptions.IP_TOS)) {
return Integer.valueOf(socket.getTrafficClass());
} else if (option.equals(StandardSocketOptions.IP_MULTICAST_IF)) {
return socket.getNetworkInterface();
} else if (option.equals(StandardSocketOptions.IP_MULTICAST_TTL)) {
return Integer.valueOf(socket.getTimeToLive());
} else if (option.equals(StandardSocketOptions.IP_MULTICAST_LOOP)) {
return Boolean.valueOf(socket.getLoopbackMode());
} else {
throw new RuntimeException("unexecpted socket option");
}
}
throw new RuntimeException("unexecpted socket type");
}
@Test
public void testOptions() throws Exception {
doSocketTests();
doServerSocketTests();
doDgSocketTests();
doMcSocketTests();
}
// Reflectively access jdk.net.Sockets.getOption so that the test can run
// without the jdk.net module.
static Object getServerSocketTrafficClass(ServerSocket ss) throws Exception {
try {
Class<?> c = Class.forName("jdk.net.Sockets");
Method m = c.getDeclaredMethod("getOption", ServerSocket.class, SocketOption.class);
return m.invoke(null, ss, StandardSocketOptions.IP_TOS);
} catch (ClassNotFoundException e) {
// Ok, jdk.net module not present, just fall back
System.out.println("jdk.net module not present, falling back.");
return Integer.valueOf(ss.getOption(StandardSocketOptions.IP_TOS));
} catch (ReflectiveOperationException e) {
throw new AssertionError(e);
}
}
}