1 /*
2 * Copyright (c) 2012, 2017, 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 package jdk.internal.util.xml;
27
28 import java.io.*;
29 import java.nio.charset.Charset;
30 import java.util.InvalidPropertiesFormatException;
31 import java.util.Map.Entry;
32 import java.util.Properties;
33 import jdk.internal.org.xml.sax.Attributes;
34 import jdk.internal.org.xml.sax.InputSource;
35 import jdk.internal.org.xml.sax.SAXException;
36 import jdk.internal.org.xml.sax.SAXParseException;
37 import jdk.internal.org.xml.sax.helpers.DefaultHandler;
38 import jdk.internal.util.xml.impl.SAXParserImpl;
39 import jdk.internal.util.xml.impl.XMLStreamWriterImpl;
40
41 /**
42 * A class used to aid in Properties load and save in XML. This class is
43 * re-implemented using a subset of SAX
44 *
45 * @author Joe Wang
46 * @since 1.8
47 */
48 public class PropertiesDefaultHandler extends DefaultHandler {
49
50 // Elements specified in the properties.dtd
51 private static final String ELEMENT_ROOT = "properties";
52 private static final String ELEMENT_COMMENT = "comment";
53 private static final String ELEMENT_ENTRY = "entry";
54 private static final String ATTR_KEY = "key";
55 // The required DTD URI for exported properties
56 private static final String PROPS_DTD_DECL =
57 "<!DOCTYPE properties SYSTEM \"http://java.sun.com/dtd/properties.dtd\">";
58 private static final String PROPS_DTD_URI =
59 "http://java.sun.com/dtd/properties.dtd";
60 private static final String PROPS_DTD =
61 "<?xml version=\"1.0\" encoding=\"UTF-8\"?>"
62 + "<!-- DTD for properties -->"
63 + "<!ELEMENT properties ( comment?, entry* ) >"
64 + "<!ATTLIST properties"
65 + " version CDATA #FIXED \"1.0\">"
66 + "<!ELEMENT comment (#PCDATA) >"
67 + "<!ELEMENT entry (#PCDATA) >"
68 + "<!ATTLIST entry "
69 + " key CDATA #REQUIRED>";
70 /**
71 * Version number for the format of exported properties files.
72 */
73 private static final String EXTERNAL_XML_VERSION = "1.0";
74 private Properties properties;
75
76 public void load(Properties props, InputStream in)
77 throws IOException, InvalidPropertiesFormatException, UnsupportedEncodingException
78 {
79 this.properties = props;
80
81 try {
82 SAXParser parser = new SAXParserImpl();
83 parser.parse(in, this);
84 } catch (SAXException saxe) {
85 throw new InvalidPropertiesFormatException(saxe);
86 }
87
88 /**
89 * String xmlVersion = propertiesElement.getAttribute("version"); if
90 * (xmlVersion.compareTo(EXTERNAL_XML_VERSION) > 0) throw new
91 * InvalidPropertiesFormatException( "Exported Properties file format
92 * version " + xmlVersion + " is not supported. This java installation
93 * can read" + " versions " + EXTERNAL_XML_VERSION + " or older. You" +
94 * " may need to install a newer version of JDK.");
95 */
96 }
97
98 public void store(Properties props, OutputStream os, String comment, Charset charset)
99 throws IOException
100 {
101 try {
102 XMLStreamWriter writer = new XMLStreamWriterImpl(os, charset);
103 writer.writeStartDocument();
104 writer.writeDTD(PROPS_DTD_DECL);
105 writer.writeStartElement(ELEMENT_ROOT);
106 if (comment != null && comment.length() > 0) {
107 writer.writeStartElement(ELEMENT_COMMENT);
108 writer.writeCharacters(comment);
109 writer.writeEndElement();
110 }
111
112 synchronized(props) {
113 for (Entry<Object, Object> e : props.entrySet()) {
114 final Object k = e.getKey();
115 final Object v = e.getValue();
116 if (k instanceof String && v instanceof String) {
117 writer.writeStartElement(ELEMENT_ENTRY);
118 writer.writeAttribute(ATTR_KEY, (String)k);
119 writer.writeCharacters((String)v);
120 writer.writeEndElement();
121 }
122 }
123 }
124
125 writer.writeEndElement();
126 writer.writeEndDocument();
127 writer.flush();
128 } catch (XMLStreamException e) {
129 if (e.getCause() instanceof UnsupportedEncodingException) {
130 throw (UnsupportedEncodingException) e.getCause();
131 }
132 throw new IOException(e);
133 }
134
135 }
136 ////////////////////////////////////////////////////////////////////
137 // Validate while parsing
138 ////////////////////////////////////////////////////////////////////
139 static final String ALLOWED_ELEMENTS = "properties, comment, entry";
140 static final String ALLOWED_COMMENT = "comment";
141 ////////////////////////////////////////////////////////////////////
142 // Handler methods
143 ////////////////////////////////////////////////////////////////////
144 StringBuffer buf = new StringBuffer();
145 boolean sawComment = false;
146 boolean validEntry = false;
147 int rootElem = 0;
148 String key;
149 String rootElm;
150
151 @Override
152 public void startElement(String uri, String localName, String qName, Attributes attributes)
153 throws SAXException
154 {
155 if (rootElem < 2) {
156 rootElem++;
157 }
158
159 if (rootElm == null) {
160 fatalError(new SAXParseException("An XML properties document must contain"
161 + " the DOCTYPE declaration as defined by java.util.Properties.", null));
162 }
163
164 if (rootElem == 1 && !rootElm.equals(qName)) {
165 fatalError(new SAXParseException("Document root element \"" + qName
166 + "\", must match DOCTYPE root \"" + rootElm + "\"", null));
167 }
168 if (!ALLOWED_ELEMENTS.contains(qName)) {
169 fatalError(new SAXParseException("Element type \"" + qName + "\" must be declared.", null));
170 }
171 if (qName.equals(ELEMENT_ENTRY)) {
172 validEntry = true;
173 key = attributes.getValue(ATTR_KEY);
174 if (key == null) {
175 fatalError(new SAXParseException("Attribute \"key\" is required and must be specified for element type \"entry\"", null));
176 }
177 } else if (qName.equals(ALLOWED_COMMENT)) {
178 if (sawComment) {
179 fatalError(new SAXParseException("Only one comment element may be allowed. "
180 + "The content of element type \"properties\" must match \"(comment?,entry*)\"", null));
181 }
182 sawComment = true;
183 }
184 }
185
186 @Override
187 public void characters(char[] ch, int start, int length) throws SAXException {
188 if (validEntry) {
189 buf.append(ch, start, length);
190 }
191 }
192
193 @Override
194 public void endElement(String uri, String localName, String qName) throws SAXException {
195 if (!ALLOWED_ELEMENTS.contains(qName)) {
196 fatalError(new SAXParseException("Element: " + qName + " is invalid, must match \"(comment?,entry*)\".", null));
197 }
198
199 if (validEntry) {
200 properties.setProperty(key, buf.toString());
201 buf.delete(0, buf.length());
202 validEntry = false;
203 }
204 }
205
206 @Override
207 public void notationDecl(String name, String publicId, String systemId) throws SAXException {
208 rootElm = name;
209 }
210
211 @Override
212 public InputSource resolveEntity(String pubid, String sysid)
213 throws SAXException, IOException {
214 {
215 if (sysid.equals(PROPS_DTD_URI)) {
216 InputSource is;
217 is = new InputSource(new StringReader(PROPS_DTD));
218 is.setSystemId(PROPS_DTD_URI);
219 return is;
220 }
221 throw new SAXException("Invalid system identifier: " + sysid);
222 }
223 }
224
225 @Override
226 public void error(SAXParseException x) throws SAXException {
227 throw x;
228 }
229
230 @Override
231 public void fatalError(SAXParseException x) throws SAXException {
232 throw x;
233 }
234
235 @Override
236 public void warning(SAXParseException x) throws SAXException {
237 throw x;
238 }
239 }