1 /*
2 * Copyright (c) 2008, 2010, 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. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 /*
27 * @bug 6622366
28 * @summary Utility class used by other jtreg tests
29 */
30
31 import com.sun.servicetag.RegistrationData;
32 import com.sun.servicetag.ServiceTag;
33 import com.sun.servicetag.Registry;
34
35 import java.util.Set;
36 import java.util.Date;
37 import java.util.Map;
38 import java.util.Properties;
39 import java.util.TimeZone;
40 import java.text.ParseException;
41 import java.text.SimpleDateFormat;
42 import java.io.*;
43
44 public class Util {
45 public static ServiceTag newServiceTag(File f)
46 throws FileNotFoundException, IOException, NumberFormatException {
47 return newServiceTag(f, false /* with instance_urn */);
48 }
49
50 public static ServiceTag newServiceTag(File f, boolean noInstanceURN)
51 throws FileNotFoundException, IOException, NumberFormatException {
52 Properties props = new Properties();
53 FileReader reader = new FileReader(f);
54 try {
55 props.load(reader);
56 } finally {
57 reader.close();
58 }
59 if (noInstanceURN) {
60 return ServiceTag.newInstance(
61 props.getProperty("product_name"),
62 props.getProperty("product_version"),
63 props.getProperty("product_urn"),
64 props.getProperty("product_parent"),
65 props.getProperty("product_parent_urn"),
66 props.getProperty("product_defined_inst_id"),
67 props.getProperty("product_vendor"),
68 props.getProperty("platform_arch"),
69 props.getProperty("container"),
70 props.getProperty("source"));
71 } else {
72 return ServiceTag.newInstance(
73 props.getProperty("instance_urn"),
74 props.getProperty("product_name"),
75 props.getProperty("product_version"),
76 props.getProperty("product_urn"),
77 props.getProperty("product_parent"),
78 props.getProperty("product_parent_urn"),
79 props.getProperty("product_defined_inst_id"),
80 props.getProperty("product_vendor"),
81 props.getProperty("platform_arch"),
82 props.getProperty("container"),
83 props.getProperty("source"));
84 }
85 }
86
87 public static boolean matches(ServiceTag st1, ServiceTag st2) {
88 if (!st1.getInstanceURN().equals(st2.getInstanceURN())) {
89 System.out.println("instance_urn: " + st1.getInstanceURN() +
90 " != " + st2.getInstanceURN());
91 return false;
92 }
93 return matchesNoInstanceUrn(st1, st2);
94 }
95
96 public static boolean matchesNoInstanceUrn(ServiceTag st1, ServiceTag st2) {
97 if (!st1.getProductName().equals(st2.getProductName())) {
98 System.out.println("product_name: " + st1.getProductName() +
99 " != " + st2.getProductName());
100 return false;
101 }
102
103 if (!st1.getProductVersion().equals(st2.getProductVersion())) {
104 System.out.println("product_version: " + st1.getProductVersion() +
105 " != " + st2.getProductVersion());
106 return false;
107 }
108 if (!st1.getProductURN().equals(st2.getProductURN())) {
109 System.out.println("product_urn: " + st1.getProductURN() +
110 " != " + st2.getProductURN());
111 return false;
112 }
113 if (!st1.getProductParentURN().equals(st2.getProductParentURN())) {
114 System.out.println("product_parent_urn: " + st1.getProductParentURN() +
115 " != " + st2.getProductParentURN());
116 return false;
117 }
118 if (!st1.getProductParent().equals(st2.getProductParent())) {
119 System.out.println("product_parent: " + st1.getProductParent() +
120 " != " + st2.getProductParent());
121 return false;
122 }
123 if (!st1.getProductDefinedInstanceID().equals(st2.getProductDefinedInstanceID())) {
124 System.out.println("product_defined_inst_id: " +
125 st1.getProductDefinedInstanceID() +
126 " != " + st2.getProductDefinedInstanceID());
127 return false;
128 }
129 if (!st1.getProductVendor().equals(st2.getProductVendor())) {
130 System.out.println("product_vendor: " + st1.getProductVendor() +
131 " != " + st2.getProductVendor());
132 return false;
133 }
134 if (!st1.getPlatformArch().equals(st2.getPlatformArch())) {
135 System.out.println("platform_arch: " + st1.getPlatformArch() +
136 " != " + st2.getPlatformArch());
137 return false;
138 }
139 if (!st1.getContainer().equals(st2.getContainer())) {
140 System.out.println("container: " + st1.getContainer() +
141 " != " + st2.getContainer());
142 return false;
143 }
144 if (!st1.getSource().equals(st2.getSource())) {
145 System.out.println("source: " + st1.getSource() +
146 " != " + st2.getSource());
147 return false;
148 }
149 return true;
150 }
151
152 public static void checkRegistrationData(String regFile,
153 Map<String, ServiceTag> stMap)
154 throws IOException {
155 BufferedInputStream in = new BufferedInputStream(new FileInputStream(regFile));
156 RegistrationData registration = RegistrationData.loadFromXML(in);
157 Set<ServiceTag> svcTags = registration.getServiceTags();
158 if (svcTags.size() != stMap.size()) {
159 throw new RuntimeException("Invalid service tag count= " +
160 svcTags.size() + " expected=" + stMap.size());
161 }
162 for (ServiceTag st : svcTags) {
163 ServiceTag st1 = stMap.get(st.getInstanceURN());
164 if (!matches(st, st1)) {
165 System.err.println(st);
166 System.err.println(st1);
167 throw new RuntimeException("ServiceTag in the registry " +
168 "does not match the one in the map");
169 }
170 }
171 }
172
173
174 /**
175 * Formats the Date into a timestamp string in YYYY-MM-dd HH:mm:ss GMT.
176 * @param timestamp Date
177 * @return a string representation of the timestamp in the YYYY-MM-dd HH:mm:ss GMT format.
178 */
179 static String formatTimestamp(Date timestamp) {
180 SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z");
181 df.setTimeZone(TimeZone.getTimeZone("GMT"));
182 return df.format(timestamp);
183 }
184
185 /**
186 * Parses a timestamp string in YYYY-MM-dd HH:mm:ss GMT format.
187 * @param timestamp Timestamp in the YYYY-MM-dd HH:mm:ss GMT format.
188 * @return Date
189 */
190 static Date parseTimestamp(String timestamp) {
191 SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss z");
192 df.setTimeZone(TimeZone.getTimeZone("GMT"));
193 try {
194 return df.parse(timestamp);
195 } catch (ParseException e) {
196 // should not reach here
197 e.printStackTrace();
198 return new Date();
199 }
200 }
201
202 /**
203 * Returns the command simulating stclient behavior.
204 */
205 static String getSvcClientCommand(String stclientRegistry) {
206 String regDir = System.getProperty("test.classes");
207
208 StringBuilder sb = new StringBuilder();
209 // wrap each argument to the command with double quotes
210 sb.append("\"");
211 sb.append(System.getProperty("java.home"));
212 sb.append(File.separator).append("bin");
213 sb.append(File.separator).append("java");
214 sb.append("\"");
215 sb.append(" -cp ");
216 sb.append("\"").append(regDir).append("\"");
217 sb.append(" \"-Dstclient.registry.path=");
218 sb.append(stclientRegistry).append("\"");
219 sb.append(" SvcTagClient");
220 return sb.toString();
221 }
222
223 private static Registry registry = null;
224 private static File registryFile = null;
225 /**
226 * Returns the Registry processed by SvcTagClient that simulates
227 * stclient.
228 */
229 static synchronized Registry getSvcTagClientRegistry() throws IOException {
230 String regDir = System.getProperty("test.classes");
231 File f = new File(regDir, "registry.xml");
232 if (registry != null) {
233 if (!f.equals(registryFile) && f.length() != 0) {
234 throw new AssertionError("Has to be empty registry.xml to run in samevm");
235 }
236 return registry;
237 }
238
239 // System.setProperty("servicetag.verbose", "true");
240 // enable the helper class
241 System.setProperty("servicetag.sthelper.supported", "true");
242 registryFile = f;
243
244 String stclientCmd = Util.getSvcClientCommand(registryFile.getCanonicalPath());
245 System.out.println("stclient cmd: " + stclientCmd);
246 System.setProperty("servicetag.stclient.cmd", stclientCmd);
247
248 // get the Registry object after the system properties are set
249 registry = Registry.getSystemRegistry();
250 return registry;
251 }
252
253 static void emptyRegistryFile() throws IOException {
254 if (registryFile.exists()) {
255 BufferedOutputStream out = new BufferedOutputStream(
256 new FileOutputStream(registryFile));
257 try {
258 RegistrationData data = new RegistrationData();
259 data.storeToXML(out);
260 } finally {
261 out.close();
262 }
263 }
264 }
265 }