1 /*
2 * Copyright (c) 2008, 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 */
29
30 import com.sun.servicetag.*;
31 import java.io.*;
32 import java.util.*;
33
34 /**
35 * A utility class simulating stclient for testing purpose.
36 */
37 public class SvcTagClient {
38 private static File xmlFile;
39 private static RegistrationData registration;
40 private static String instanceURN;
41 private static String productName;
42 private static String productVersion;
43 private static String productURN;
44 private static String productParent;
45 private static String productParentURN = ""; // optional
46 private static String productDefinedInstanceID = ""; //optional
47 private static String productVendor;
48 private static String platformArch;
49 private static String container;
50 private static String source;
51
52 // stclient exit value
53 private static final int ST_ERR_REC_NOT_FOUND = 225;
54
55 public static void main(String[] args) throws Exception {
56 String path = System.getProperty("stclient.registry.path");
57 if (path == null) {
58 System.err.println("stclient registry path missing");
59 System.exit(-1);
60 }
61 xmlFile = new File(path);
62 if (xmlFile.exists()) {
63 BufferedInputStream in = new BufferedInputStream(new FileInputStream(xmlFile));
64 registration = RegistrationData.loadFromXML(in);
65 } else {
66 registration = new RegistrationData();
67 }
68 boolean add = false;
69 boolean delete = false;
70 boolean update = false;
71 boolean get = false;
72 boolean find = false;
73
74 int count = 0;
75 while (count < args.length) {
76 String arg = args[count];
77 if (!arg.startsWith("-")) {
78 System.err.println("Invalid option:" + arg);
79 System.exit(-1);
80 }
81 if (arg.equals("-a")) {
82 add = true;
83 } else if (arg.equals("-d")) {
84 delete = true;
85 } else if (arg.equals("-u")) {
86 update = true;
87 } else if (arg.equals("-g")) {
88 get = true;
89 } else if (arg.equals("-f")) {
90 find = true;
91 productURN = "";
92 } else if (arg.equals("-t")) {
93 productURN = args[++count];
94 } else if (arg.equals("-i")) {
95 instanceURN = args[++count];
96 } else if (arg.equals("-p")) {
97 productName = args[++count];
98 } else if (arg.equals("-e")) {
99 productVersion = args[++count];
100 } else if (arg.equals("-t")) {
101 productURN = args[++count];
102 } else if (arg.equals("-F")) {
103 productParentURN = args[++count];
104 } else if (arg.equals("-P")) {
105 productParent = args[++count];
106 } else if (arg.equals("-I")) {
107 productDefinedInstanceID = args[++count];
108 } else if (arg.equals("-m")) {
109 productVendor = args[++count];
110 } else if (arg.equals("-A")) {
111 platformArch = args[++count];
112 } else if (arg.equals("-z")) {
113 container = args[++count];
114 } else if (arg.equals("-S")) {
115 source = args[++count];
116 } else {
117 System.err.println("Invalid option:" + arg);
118 System.exit(-1);
119 }
120 count++;
121 }
122
123 if (add) {
124 addServiceTag();
125 } else if (delete) {
126 deleteServiceTag();
127 } else if (update) {
128 updateServiceTag();
129 } else if (get) {
130 getServiceTag();
131 } else if (find) {
132 findServiceTags();
133 } else {
134 System.err.println("Error");
135 System.exit(-1);
136 }
137 updateXmlFile();
138 }
139 private static String OUTPUT = "Product instance URN=";
140
141 private static void addServiceTag() {
142 if (instanceURN == null) {
143 instanceURN = ServiceTag.generateInstanceURN();
144 }
145 ServiceTag st = ServiceTag.newInstance(instanceURN,
146 productName,
147 productVersion,
148 productURN,
149 productParent,
150 productParentURN,
151 productDefinedInstanceID,
152 productVendor,
153 platformArch,
154 container,
155 source);
156 registration.addServiceTag(st);
157 System.out.println(OUTPUT + st.getInstanceURN());
158 }
159
160 private static void deleteServiceTag() {
161 registration.removeServiceTag(instanceURN);
162 System.out.println("instance_urn=" + instanceURN + " deleted");
163 }
164
165 private static void updateServiceTag() {
166 registration.updateServiceTag(instanceURN, productDefinedInstanceID);
167 System.out.println("instance_urn=" + instanceURN + " updated");
168 }
169
170 private static void getServiceTag() {
171 ServiceTag st = registration.getServiceTag(instanceURN);
172 if (st == null) {
173 System.err.println("instance_urn=" + instanceURN + " not found");
174 System.exit(ST_ERR_REC_NOT_FOUND);
175 } else {
176 System.out.println(st);
177 }
178 }
179
180 private static void findServiceTags() {
181 Set<ServiceTag> set = registration.getServiceTags();
182 for (ServiceTag st : set) {
183 if (st.getProductURN().equals(productURN)) {
184 System.out.println(st.getInstanceURN());
185 }
186 }
187 if (set.size() == 0) {
188 System.out.println("No records found");
189 System.exit(ST_ERR_REC_NOT_FOUND);
190 }
191 }
192 private static void updateXmlFile() throws IOException {
193 BufferedOutputStream out = new BufferedOutputStream(new FileOutputStream(xmlFile));
194 try {
195 registration.storeToXML(out);
196 } finally {
197 out.close();
198 }
199 }
200 }