1 /*
2 * Copyright (c) 1998, 2012, 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 4155894
26 *
27 * @summary remote references can't be downloaded
28 * @author Ann Wollrath
29 *
30 * test currently needs RMISecurityManager because of
31 * 4180392
32 *
33 * @library ../../testlibrary
34 * @build UseCustomRef
35 * @build Ping
36 * @build UseCustomRef_Stub
37 * @build UseCustomRef_Skel
38 * @build TestLibrary
39 * @run main/othervm/policy=security.policy/secure=java.rmi.RMISecurityManager/timeout=120 UseCustomRef
40 *
41 * This test was failing to run because the synthetic access
42 * control context used by the application class loader to find and define
43 * CustomServerRef does not have accessClassInPackage.sun.rmi.server runtime
44 * permission necessary to load its superclass sun.rmi.server.UnicastServerRef,
45 * even though this test's code is granted that permission in its policy file.
46 * That bug number is 4256530
47 */
48
49 import java.io.*;
50 import java.rmi.*;
51 import java.rmi.server.*;
52 import java.rmi.registry.*;
53 import sun.rmi.transport.LiveRef;
54
55 public class UseCustomRef
56 extends RemoteServer
57 implements Ping
58 {
59
60 public UseCustomRef() throws RemoteException {
61 exportObject();
62 }
63
64 public void exportObject() throws RemoteException {
65 ref = new CustomServerRef(new LiveRef(0));
66 ((ServerRef) ref).exportObject(this, null);
67 }
68
69 public RemoteRef getRef() { return ref; }
70
71 public void ping() {}
72
73 public void receiveAndPing(Ping p) throws RemoteException {
74 p.ping();
75 }
76
77 public static void main(String[] args) {
78 Ping obj = null;
79 Registry registry = null;
80
81 try {
82 /*
83 * create registry
84 */
85 TestLibrary.suggestSecurityManager("java.rmi.RMISecurityManager");
86
87 System.err.println("creating Registry...");
88
89 registry = TestLibrary.createRegistryOnUnusedPort();
90 int port = TestLibrary.getRegistryPort(registry);
91 /*
92 * create object with custom ref and bind in registry
93 */
94 System.err.println("creating UseCustomRef...");
95 UseCustomRef cr = new UseCustomRef();
96 RemoteRef ref = cr.getRef();
97 if (!(ref instanceof CustomServerRef)) {
98 TestLibrary.bomb("test failed: reference not " +
99 "instanceof CustomServerRef");
100 }
101
102 String name = "//:" + port + "/UseCustomRef";
103 // String name = "UseCustomRef";
104 System.err.println("binding object in registry...");
105 Naming.rebind(name, cr);
106
107 /*
108 * look up object and invoke its ping method
109 */
110 System.err.println("ping object...");
111 obj = (Ping) Naming.lookup(name);
112 obj.ping();
113
114 /*
115 * pass object with custom ref in remote call
116 */
117 System.err.println("pass object in remote call...");
118 obj.receiveAndPing(cr);
119
120 /*
121 * write remote object with custom ref to output stream
122 */
123 System.err.println("writing remote object to stream...");
124 ByteArrayOutputStream bout = new ByteArrayOutputStream();
125 ObjectOutputStream out = new ObjectOutputStream(bout);
126 out.writeObject(cr);
127 out.flush();
128 out.close();
129
130 /*
131 * read back remote object from output stream
132 */
133 System.err.println("reading remote object from stream...");
134 ObjectInputStream in = new ObjectInputStream(
135 new ByteArrayInputStream(bout.toByteArray()));
136 cr = (UseCustomRef) in.readObject();
137
138 /*
139 * re-export object and ping
140 */
141 System.err.println("re-export object read...");
142 cr.exportObject();
143 System.err.println("look up object again...");
144 Naming.rebind(name, cr);
145 System.err.println("ping object read...");
146 obj = (Ping) Naming.lookup(name);
147 obj.ping();
148 System.err.println("TEST PASSED");
149 Naming.unbind(name);
150 cr = null;
151
152 } catch (Exception e) {
153 TestLibrary.bomb("test failed with exception: ", e);
154 } finally {
155 TestLibrary.unexport(obj);
156 TestLibrary.unexport(registry);
157
158 registry = null;
159 obj = null;
160 }
161 }
162
163 public static class CustomServerRef
164 extends sun.rmi.server.UnicastServerRef
165 {
166 public CustomServerRef() {}
167
168 public CustomServerRef(LiveRef ref) {
169 super(ref);
170 }
171 /*****
172 public CustomServerRef(int port,
173 RMIClientSocketFactory csf,
174 RMIServerSocketFactory ssf)
175 {
176 super (new LiveRef(port, csf, ssf));
177 }
178 *****/
179
180 public String getRefClass(ObjectOutput out) {
181 return "";
182 }
183
184 protected void unmarshalCustomCallData(ObjectInput in)
185 throws IOException, ClassNotFoundException
186 {
187 System.err.println("unmarshalling call data...");
188 String s = (String) (in.readObject());
189 System.err.println(s);
190 }
191
192 protected RemoteRef getClientRef() {
193 return new CustomRef(ref);
194 }
195 }
196
197 public static class CustomRef extends sun.rmi.server.UnicastRef {
198
199 public CustomRef() {
200 }
201
202 public CustomRef(sun.rmi.transport.LiveRef ref) {
203 super(ref);
204 }
205
206 protected void marshalCustomCallData(ObjectOutput out)
207 throws IOException
208 {
209 // this custom data ensures that a custom server
210 // ref has written the relevant information.
211 System.err.println("marshalling call data...");
212 out.writeObject("hello there.");
213 }
214
215 public String getRefClass(ObjectOutput out) {
216 return "";
217 }
218
219 }
220 }