1 /*
2 * Copyright (c) 2014, 2016, 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 package stream;
25
26 import static jaxp.library.JAXPTestUtilities.getSystemProperty;
27 import static jaxp.library.JAXPTestUtilities.runWithAllPerm;
28
29 import java.io.File;
30 import java.io.FileInputStream;
31 import java.io.FileNotFoundException;
32 import java.io.FileOutputStream;
33 import java.io.IOException;
34 import java.net.URL;
35 import java.net.URLClassLoader;
36 import java.util.Properties;
37
38 import javax.xml.stream.XMLInputFactory;
39 import javax.xml.stream.XMLOutputFactory;
40
41 import org.testng.Assert;
42 import org.testng.annotations.Listeners;
43 import org.testng.annotations.Test;
44
45 /*
46 * @test
47 * @library /javax/xml/jaxp/libs /javax/xml/jaxp/unittest
48 * @run testng/othervm -DrunSecMngr=true stream.FactoryFindTest
49 * @run testng/othervm stream.FactoryFindTest
50 * @summary Test SaTX factory using factory property and using ContextClassLoader.
51 */
52 @Listeners({jaxp.library.FilePolicy.class})
53 public class FactoryFindTest {
54
55 boolean myClassLoaderUsed = false;
56
57 final static String FACTORY_KEY = "javax.xml.stream.XMLInputFactory";
58
59 // @BeforeClass
60 // public void setup(){
61 // policy.PolicyUtil.changePolicy(getClass().getResource("FactoryFindTest.policy").getFile());
62 // }
63
64 @Test(enabled=false) // due to 8156508
65 public void testFactoryFindUsingStaxProperties() {
66 // If property is defined, will take precendence so this test
67 // is ignored :(
68 if (getSystemProperty(FACTORY_KEY) != null) {
69 return;
70 }
71
72 Properties props = new Properties();
73 String configFile = getSystemProperty("java.home") + File.separator + "lib" + File.separator + "stax.properties";
74
75 File f = new File(configFile);
76 if (f.exists()) {
77 try {
78 FileInputStream fis = new FileInputStream(f);
79 props.load(fis);
80 fis.close();
81 } catch (FileNotFoundException e) {
82 return;
83 } catch (IOException e) {
84 return;
85 }
86 } else {
87 props.setProperty(FACTORY_KEY, "com.sun.xml.internal.stream.XMLInputFactoryImpl");
88 try {
89 FileOutputStream fos = new FileOutputStream(f);
90 props.store(fos, null);
91 fos.close();
92 f.deleteOnExit();
93 } catch (FileNotFoundException e) {
94 return;
95 } catch (IOException e) {
96 return;
97 }
98 }
99
100 XMLInputFactory factory = XMLInputFactory.newInstance();
101 Assert.assertTrue(factory.getClass().getName().equals(props.getProperty(FACTORY_KEY)));
102 }
103
104 @Test
105 public void testFactoryFind() {
106 try {
107 // setSystemProperty("jaxp.debug", "true");
108
109 XMLInputFactory factory = XMLInputFactory.newInstance();
110 Assert.assertTrue(factory.getClass().getClassLoader() == null);
111
112 runWithAllPerm(() -> Thread.currentThread().setContextClassLoader(null));
113 factory = XMLInputFactory.newInstance();
114 Assert.assertTrue(factory.getClass().getClassLoader() == null);
115
116 runWithAllPerm(() -> Thread.currentThread().setContextClassLoader(new MyClassLoader()));
117 factory = XMLInputFactory.newInstance();
118 // because it's decided by having sm or not in FactoryFind code
119 if (System.getSecurityManager() == null)
120 Assert.assertTrue(myClassLoaderUsed);
121 else
122 Assert.assertFalse(myClassLoaderUsed);
123
124 XMLOutputFactory ofactory = XMLOutputFactory.newInstance();
125 Assert.assertTrue(ofactory.getClass().getClassLoader() == null);
126
127 runWithAllPerm(() -> Thread.currentThread().setContextClassLoader(null));
128 ofactory = XMLOutputFactory.newInstance();
129 Assert.assertTrue(ofactory.getClass().getClassLoader() == null);
130
131 runWithAllPerm(() -> Thread.currentThread().setContextClassLoader(new MyClassLoader()));
132 ofactory = XMLOutputFactory.newInstance();
133 if (System.getSecurityManager() == null)
134 Assert.assertTrue(myClassLoaderUsed);
135 else
136 Assert.assertFalse(myClassLoaderUsed);
137 } catch (Exception ex) {
138 throw new RuntimeException(ex);
139 }
140 }
141
142 class MyClassLoader extends URLClassLoader {
143
144 public MyClassLoader() {
145 super(new URL[0]);
146 }
147
148 public Class loadClass(String name) throws ClassNotFoundException {
149 myClassLoaderUsed = true;
150 return super.loadClass(name);
151 }
152 }
153 }
154