8231506: Fix some instabilities in a few networking tests
Reviewed-by: alanb, chegar, msheppar
This commit is contained in:
parent
990ec34714
commit
8e98ce54bf
@ -50,6 +50,7 @@ import java.nio.file.Paths;
|
||||
import java.util.ArrayDeque;
|
||||
import java.util.List;
|
||||
import java.util.Optional;
|
||||
import java.util.concurrent.Phaser;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import jdk.test.lib.net.IPSupport;
|
||||
@ -72,11 +73,14 @@ public class UnreferencedMulticastSockets {
|
||||
static class Server implements Runnable {
|
||||
|
||||
MulticastSocket ss;
|
||||
|
||||
final int port;
|
||||
final Phaser phaser = new Phaser(2);
|
||||
Server() throws IOException {
|
||||
InetAddress loopback = InetAddress.getLoopbackAddress();
|
||||
InetSocketAddress serverAddress =
|
||||
new InetSocketAddress(InetAddress.getLoopbackAddress(), 0);
|
||||
new InetSocketAddress(loopback, 0);
|
||||
ss = new MulticastSocket(serverAddress);
|
||||
port = ss.getLocalPort();
|
||||
System.out.printf(" DatagramServer addr: %s: %d%n",
|
||||
this.getHost(), this.getPort());
|
||||
pendingSockets.add(new NamedWeak(ss, pendingQueue, "serverMulticastSocket"));
|
||||
@ -89,7 +93,7 @@ public class UnreferencedMulticastSockets {
|
||||
}
|
||||
|
||||
int getPort() {
|
||||
return ss.getLocalPort();
|
||||
return port;
|
||||
}
|
||||
|
||||
// Receive a byte and send back a byte
|
||||
@ -98,12 +102,18 @@ public class UnreferencedMulticastSockets {
|
||||
byte[] buffer = new byte[50];
|
||||
DatagramPacket p = new DatagramPacket(buffer, buffer.length);
|
||||
ss.receive(p);
|
||||
System.out.printf("Server: ping received from: %s%n", p.getSocketAddress());
|
||||
phaser.arriveAndAwaitAdvance(); // await the client...
|
||||
buffer[0] += 1;
|
||||
System.out.printf("Server: sending echo to: %s%n", p.getSocketAddress());
|
||||
ss.send(p); // send back +1
|
||||
|
||||
System.out.printf("Server: awaiting client%n");
|
||||
phaser.arriveAndAwaitAdvance(); // await the client...
|
||||
// do NOT close but 'forget' the socket reference
|
||||
System.out.printf("Server: forgetting socket...%n");
|
||||
ss = null;
|
||||
} catch (Exception ioe) {
|
||||
} catch (Throwable ioe) {
|
||||
ioe.printStackTrace();
|
||||
}
|
||||
}
|
||||
@ -112,8 +122,11 @@ public class UnreferencedMulticastSockets {
|
||||
public static void main(String args[]) throws Exception {
|
||||
IPSupport.throwSkippedExceptionIfNonOperational();
|
||||
|
||||
InetSocketAddress clientAddress =
|
||||
new InetSocketAddress(InetAddress.getLoopbackAddress(), 0);
|
||||
|
||||
// Create and close a MulticastSocket to warm up the FD count for side effects.
|
||||
try (MulticastSocket s = new MulticastSocket(0)) {
|
||||
try (MulticastSocket s = new MulticastSocket(clientAddress)) {
|
||||
// no-op; close immediately
|
||||
s.getLocalPort(); // no-op
|
||||
}
|
||||
@ -126,8 +139,33 @@ public class UnreferencedMulticastSockets {
|
||||
Thread thr = new Thread(svr);
|
||||
thr.start();
|
||||
|
||||
MulticastSocket client = new MulticastSocket(0);
|
||||
System.out.printf(" client bound port: %d%n", client.getLocalPort());
|
||||
// It is possible under some circumstances that the client
|
||||
// might get bound to the same port than the server: this
|
||||
// would make the test fail - so if this happen we try to
|
||||
// bind to a specific port by incrementing the server port.
|
||||
MulticastSocket client = null;
|
||||
int serverPort = svr.getPort();
|
||||
int maxtries = 20;
|
||||
for (int i = 0; i < maxtries; i++) {
|
||||
try {
|
||||
System.out.printf("Trying to bind client to: %s%n", clientAddress);
|
||||
client = new MulticastSocket(clientAddress);
|
||||
if (client.getLocalPort() != svr.getPort()) break;
|
||||
client.close();
|
||||
} catch (IOException x) {
|
||||
System.out.printf("Couldn't create client after %d attempts: %s%n", i, x);
|
||||
if (i == maxtries) throw x;
|
||||
}
|
||||
if (i == maxtries) {
|
||||
String msg = String.format("Couldn't create client after %d attempts", i);
|
||||
System.out.println(msg);
|
||||
throw new AssertionError(msg);
|
||||
}
|
||||
clientAddress = new InetSocketAddress(clientAddress.getAddress(), serverPort + i);
|
||||
}
|
||||
|
||||
System.out.printf(" client bound port: %s:%d%n",
|
||||
client.getLocalAddress(), client.getLocalPort());
|
||||
client.connect(svr.getHost(), svr.getPort());
|
||||
pendingSockets.add(new NamedWeak(client, pendingQueue, "clientMulticastSocket"));
|
||||
extractRefs(client, "clientMulticastSocket");
|
||||
@ -136,14 +174,17 @@ public class UnreferencedMulticastSockets {
|
||||
msg[0] = 1;
|
||||
DatagramPacket p = new DatagramPacket(msg, msg.length, svr.getHost(), svr.getPort());
|
||||
client.send(p);
|
||||
System.out.printf(" ping sent to: %s:%d%n", svr.getHost(), svr.getPort());
|
||||
svr.phaser.arriveAndAwaitAdvance(); // wait until the server has received its packet
|
||||
|
||||
p = new DatagramPacket(msg, msg.length);
|
||||
client.receive(p);
|
||||
|
||||
System.out.printf("echo received from: %s%n", p.getSocketAddress());
|
||||
System.out.printf(" echo received from: %s%n", p.getSocketAddress());
|
||||
if (msg[0] != 2) {
|
||||
throw new AssertionError("incorrect data received: expected: 2, actual: " + msg[0]);
|
||||
}
|
||||
svr.phaser.arriveAndAwaitAdvance(); // let the server null out its socket
|
||||
|
||||
// Do NOT close the MulticastSocket; forget it
|
||||
|
||||
|
@ -68,7 +68,7 @@ public class SocketImplCombinations {
|
||||
* Test creating a connected Socket, it should be created with a platform SocketImpl.
|
||||
*/
|
||||
public void testNewSocket2() throws IOException {
|
||||
try (ServerSocket ss = new ServerSocket(0)) {
|
||||
try (ServerSocket ss = boundServerSocket()) {
|
||||
try (Socket s = new Socket(ss.getInetAddress(), ss.getLocalPort())) {
|
||||
SocketImpl si = getSocketImpl(s);
|
||||
assertTrue(isSocksSocketImpl(si));
|
||||
@ -127,7 +127,7 @@ public class SocketImplCombinations {
|
||||
Socket s = new Socket((SocketImpl) null) { };
|
||||
try (s) {
|
||||
assertTrue(getSocketImpl(s) == null);
|
||||
s.bind(new InetSocketAddress(0)); // force SocketImpl to be created
|
||||
s.bind(loopbackSocketAddress()); // force SocketImpl to be created
|
||||
SocketImpl si = getSocketImpl(s);
|
||||
assertTrue(isSocksSocketImpl(si));
|
||||
SocketImpl delegate = getDelegate(si);
|
||||
@ -218,7 +218,7 @@ public class SocketImplCombinations {
|
||||
Socket s = new Socket((SocketImpl) null) { };
|
||||
try (s) {
|
||||
assertTrue(getSocketImpl(s) == null);
|
||||
s.bind(new InetSocketAddress(0)); // force SocketImpl to be created
|
||||
s.bind(loopbackSocketAddress()); // force SocketImpl to be created
|
||||
assertTrue(getSocketImpl(s) instanceof CustomSocketImpl);
|
||||
}
|
||||
} finally {
|
||||
@ -378,7 +378,7 @@ public class SocketImplCombinations {
|
||||
public void testServerSocketAccept5a() throws IOException {
|
||||
SocketImpl serverImpl = new CustomSocketImpl(true);
|
||||
try (ServerSocket ss = new ServerSocket(serverImpl) { }) {
|
||||
ss.bind(new InetSocketAddress(0));
|
||||
ss.bind(loopbackSocketAddress());
|
||||
expectThrows(IOException.class, ss::accept);
|
||||
}
|
||||
}
|
||||
@ -565,17 +565,37 @@ public class SocketImplCombinations {
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a new InetSocketAddress with the loopback interface
|
||||
* and port 0.
|
||||
*/
|
||||
static InetSocketAddress loopbackSocketAddress() {
|
||||
InetAddress loopback = InetAddress.getLoopbackAddress();
|
||||
return new InetSocketAddress(loopback, 0);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a ServerSocket bound to a port on the loopback address
|
||||
*/
|
||||
static ServerSocket boundServerSocket() throws IOException {
|
||||
ServerSocket ss = new ServerSocket();
|
||||
ss.bind(loopbackSocketAddress());
|
||||
return ss;
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a ServerSocket that returns the given Socket from accept.
|
||||
*/
|
||||
static ServerSocket serverSocketToAccept(Socket s) throws IOException {
|
||||
return new ServerSocket(0) {
|
||||
ServerSocket ss = new ServerSocket() {
|
||||
@Override
|
||||
public Socket accept() throws IOException {
|
||||
implAccept(s);
|
||||
return s;
|
||||
}
|
||||
};
|
||||
ss.bind(loopbackSocketAddress());
|
||||
return ss;
|
||||
}
|
||||
|
||||
/**
|
||||
@ -590,7 +610,7 @@ public class SocketImplCombinations {
|
||||
return s;
|
||||
}
|
||||
};
|
||||
ss.bind(new InetSocketAddress(0));
|
||||
ss.bind(loopbackSocketAddress());
|
||||
return ss;
|
||||
}
|
||||
|
||||
|
@ -26,6 +26,8 @@ import com.sun.net.httpserver.HttpServer;
|
||||
import com.sun.net.httpserver.HttpsConfigurator;
|
||||
import com.sun.net.httpserver.HttpsParameters;
|
||||
import com.sun.net.httpserver.HttpsServer;
|
||||
|
||||
import java.io.Closeable;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
@ -1568,8 +1570,8 @@ public abstract class DigestEchoServer implements HttpServerAdapters {
|
||||
@Override
|
||||
public void run() {
|
||||
try {
|
||||
int c = 0;
|
||||
try {
|
||||
int c;
|
||||
while ((c = is.read()) != -1) {
|
||||
os.write(c);
|
||||
os.flush();
|
||||
@ -1578,11 +1580,13 @@ public abstract class DigestEchoServer implements HttpServerAdapters {
|
||||
if (DEBUG) System.out.print(tag);
|
||||
}
|
||||
is.close();
|
||||
} catch (IOException ex) {
|
||||
if (DEBUG || !stopped && c > -1)
|
||||
ex.printStackTrace(System.out);
|
||||
end.completeExceptionally(ex);
|
||||
} finally {
|
||||
os.close();
|
||||
try {os.close();} catch (Throwable t) {}
|
||||
}
|
||||
} catch (IOException ex) {
|
||||
if (DEBUG) ex.printStackTrace(System.out);
|
||||
} finally {
|
||||
end.complete(null);
|
||||
}
|
||||
@ -1632,10 +1636,12 @@ public abstract class DigestEchoServer implements HttpServerAdapters {
|
||||
@Override
|
||||
public void run() {
|
||||
Socket clientConnection = null;
|
||||
Socket targetConnection = null;
|
||||
try {
|
||||
while (!stopped) {
|
||||
System.out.println(now() + "Tunnel: Waiting for client");
|
||||
Socket toClose;
|
||||
targetConnection = clientConnection = null;
|
||||
try {
|
||||
toClose = clientConnection = ss.accept();
|
||||
if (NO_LINGER) {
|
||||
@ -1649,7 +1655,6 @@ public abstract class DigestEchoServer implements HttpServerAdapters {
|
||||
}
|
||||
System.out.println(now() + "Tunnel: Client accepted");
|
||||
StringBuilder headers = new StringBuilder();
|
||||
Socket targetConnection = null;
|
||||
InputStream ccis = clientConnection.getInputStream();
|
||||
OutputStream ccos = clientConnection.getOutputStream();
|
||||
Writer w = new OutputStreamWriter(
|
||||
@ -1769,28 +1774,44 @@ public abstract class DigestEchoServer implements HttpServerAdapters {
|
||||
end1 = new CompletableFuture<>());
|
||||
Thread t2 = pipe(targetConnection.getInputStream(), ccos, '-',
|
||||
end2 = new CompletableFuture<>());
|
||||
end = CompletableFuture.allOf(end1, end2);
|
||||
var end11 = end1.whenComplete((r, t) -> exceptionally(end2, t));
|
||||
var end22 = end2.whenComplete((r, t) -> exceptionally(end1, t));
|
||||
end = CompletableFuture.allOf(end11, end22);
|
||||
Socket tc = targetConnection;
|
||||
end.whenComplete(
|
||||
(r,t) -> {
|
||||
try { toClose.close(); } catch (IOException x) { }
|
||||
try { tc.close(); } catch (IOException x) { }
|
||||
finally {connectionCFs.remove(end);}
|
||||
});
|
||||
connectionCFs.add(end);
|
||||
targetConnection = clientConnection = null;
|
||||
t1.start();
|
||||
t2.start();
|
||||
}
|
||||
} catch (Throwable ex) {
|
||||
try {
|
||||
ss.close();
|
||||
} catch (IOException ex1) {
|
||||
ex.addSuppressed(ex1);
|
||||
}
|
||||
close(clientConnection, ex);
|
||||
close(targetConnection, ex);
|
||||
close(ss, ex);
|
||||
ex.printStackTrace(System.err);
|
||||
} finally {
|
||||
System.out.println(now() + "Tunnel: exiting (stopped=" + stopped + ")");
|
||||
connectionCFs.forEach(cf -> cf.complete(null));
|
||||
}
|
||||
}
|
||||
|
||||
void exceptionally(CompletableFuture<?> cf, Throwable t) {
|
||||
if (t != null) cf.completeExceptionally(t);
|
||||
}
|
||||
|
||||
void close(Closeable c, Throwable e) {
|
||||
if (c == null) return;
|
||||
try {
|
||||
c.close();
|
||||
} catch (IOException x) {
|
||||
e.addSuppressed(x);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
Loading…
Reference in New Issue
Block a user