1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
* Copyright (c) 2005, 2012, 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 4924577
* @summary When the current RMIFailureHandler (if any) is invoked
* because of a server socket accept failure, if it returns false,
* then (in addition to the accept loop terminating) the associated
* server socket should be closed. The server socket should also be
* closed if the accept loop terminates because of an unexpected
* exception for which it doesn't even consult the RMIFailureHandler.
* @author Peter Jones
*
* @run main/othervm CloseServerSocketOnTermination
*/
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.rmi.Remote;
import java.rmi.server.RMIFailureHandler;
import java.rmi.server.RMIServerSocketFactory;
import java.rmi.server.RMISocketFactory;
import java.rmi.server.UnicastRemoteObject;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
public class CloseServerSocketOnTermination {
private static long TIMEOUT = 5000;
public static void main(String[] args) throws Exception {
System.err.println("\nRegression test for bug 4924577\n");
RMISocketFactory.setFailureHandler(new RMIFailureHandler() {
public boolean failure(Exception e) { return false; }
});
tryWith(new IOException());
tryWith(new NullPointerException());
tryWith(new OutOfMemoryError());
tryWith(new NoClassDefFoundError());
tryWith(new InternalError());
tryWith(new Throwable());
System.err.println("TEST PASSED");
}
private static void tryWith(Throwable t) throws Exception {
Remote impl = new Remote() { };
try {
CountDownLatch latch = new CountDownLatch(1);
UnicastRemoteObject.exportObject(impl, 0, null, new SSF(t, latch));
if (!latch.await(TIMEOUT, TimeUnit.MILLISECONDS)) {
throw new Error("server socket not closed");
}
} finally {
UnicastRemoteObject.unexportObject(impl, true);
}
}
private static class SSF implements RMIServerSocketFactory {
private final Throwable acceptFailure;
private final CountDownLatch closedLatch;
SSF(Throwable acceptFailure, CountDownLatch closedLatch) {
this.acceptFailure = acceptFailure;
this.closedLatch = closedLatch;
}
public ServerSocket createServerSocket(int port) throws IOException {
return new ServerSocket(port) {
private int acceptInvocations = 0;
public synchronized Socket accept() throws IOException {
if (acceptInvocations++ == 0) {
throwException(acceptFailure);
}
return super.accept();
}
public void close() throws IOException {
closedLatch.countDown();
super.close();
}
};
}
// hack to throw an arbitrary (possibly checked) Throwable
private static void throwException(Throwable t) {
try {
toThrow.set(t);
Thrower.class.newInstance();
} catch (IllegalAccessException e) {
throw new AssertionError();
} catch (InstantiationException e) {
throw new AssertionError();
} finally {
toThrow.remove();
}
}
private static ThreadLocal<Throwable> toThrow =
new ThreadLocal<Throwable>();
private static class Thrower {
Thrower() throws Throwable { throw toThrow.get(); }
}
}
}
|
