1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/message/source/SourceUtils.java Tue Mar 06 16:09:35 2012 -0800 1.3 @@ -0,0 +1,253 @@ 1.4 +/* 1.5 + * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved. 1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 1.7 + * 1.8 + * This code is free software; you can redistribute it and/or modify it 1.9 + * under the terms of the GNU General Public License version 2 only, as 1.10 + * published by the Free Software Foundation. Oracle designates this 1.11 + * particular file as subject to the "Classpath" exception as provided 1.12 + * by Oracle in the LICENSE file that accompanied this code. 1.13 + * 1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT 1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 1.17 + * version 2 for more details (a copy is included in the LICENSE file that 1.18 + * accompanied this code). 1.19 + * 1.20 + * You should have received a copy of the GNU General Public License version 1.21 + * 2 along with this work; if not, write to the Free Software Foundation, 1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 1.23 + * 1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 1.25 + * or visit www.oracle.com if you need additional information or have any 1.26 + * questions. 1.27 + */ 1.28 + 1.29 +package com.sun.xml.internal.ws.message.source; 1.30 + 1.31 +import com.sun.xml.internal.ws.message.RootElementSniffer; 1.32 +import com.sun.xml.internal.ws.streaming.SourceReaderFactory; 1.33 +import com.sun.xml.internal.ws.util.xml.XmlUtil; 1.34 +import org.w3c.dom.Document; 1.35 +import org.w3c.dom.Node; 1.36 + 1.37 +import javax.xml.namespace.QName; 1.38 +import javax.xml.stream.XMLStreamConstants; 1.39 +import javax.xml.stream.XMLStreamException; 1.40 +import javax.xml.stream.XMLStreamReader; 1.41 +import javax.xml.stream.XMLStreamWriter; 1.42 +import javax.xml.transform.Source; 1.43 +import javax.xml.transform.Transformer; 1.44 +import javax.xml.transform.TransformerConfigurationException; 1.45 +import javax.xml.transform.TransformerException; 1.46 +import javax.xml.transform.dom.DOMSource; 1.47 +import javax.xml.transform.sax.SAXResult; 1.48 +import javax.xml.transform.sax.SAXSource; 1.49 +import javax.xml.transform.stream.StreamSource; 1.50 +import javax.xml.ws.WebServiceException; 1.51 + 1.52 +/** 1.53 + * 1.54 + * @author Vivek Pandey 1.55 + */ 1.56 +final class SourceUtils { 1.57 + 1.58 + int srcType; 1.59 + 1.60 + private final int domSource = 1; 1.61 + private final int streamSource = 2; 1.62 + private final int saxSource=4; 1.63 + 1.64 + public SourceUtils(Source src) { 1.65 + if(src instanceof StreamSource){ 1.66 + srcType = streamSource; 1.67 + }else if(src instanceof DOMSource){ 1.68 + srcType = domSource; 1.69 + }else if(src instanceof SAXSource){ 1.70 + srcType = saxSource; 1.71 + } 1.72 + } 1.73 + 1.74 + public boolean isDOMSource(){ 1.75 + return (srcType&domSource) == domSource; 1.76 + } 1.77 + 1.78 + public boolean isStreamSource(){ 1.79 + return (srcType&streamSource) == streamSource; 1.80 + } 1.81 + 1.82 + public boolean isSaxSource(){ 1.83 + return (srcType&saxSource) == saxSource; 1.84 + } 1.85 + 1.86 + /** 1.87 + * This would peek into the Source (DOMSource and SAXSource) for the localName and NamespaceURI 1.88 + * of the top-level element. 1.89 + * @param src 1.90 + * @return QName of the payload 1.91 + */ 1.92 + public QName sniff(Source src) { 1.93 + return sniff(src, new RootElementSniffer()); 1.94 + } 1.95 + 1.96 + public QName sniff(Source src, RootElementSniffer sniffer){ 1.97 + String localName = null; 1.98 + String namespaceUri = null; 1.99 + 1.100 + if(isDOMSource()){ 1.101 + DOMSource domSource = (DOMSource)src; 1.102 + Node n = domSource.getNode(); 1.103 + if(n.getNodeType()== Node.DOCUMENT_NODE) { 1.104 + n = ((Document)n).getDocumentElement(); 1.105 + } 1.106 + localName = n.getLocalName(); 1.107 + namespaceUri = n.getNamespaceURI(); 1.108 + }else if(isSaxSource()){ 1.109 + SAXSource saxSrc = (SAXSource)src; 1.110 + SAXResult saxResult = new SAXResult(sniffer); 1.111 + try { 1.112 + Transformer tr = XmlUtil.newTransformer(); 1.113 + tr.transform(saxSrc, saxResult); 1.114 + } catch (TransformerConfigurationException e) { 1.115 + throw new WebServiceException(e); 1.116 + } catch (TransformerException e) { 1.117 + // if it's due to aborting the processing after the first element, 1.118 + // we can safely ignore this exception. 1.119 + // 1.120 + // if it's due to error in the object, the same error will be reported 1.121 + // when the readHeader() method is used, so we don't have to report 1.122 + // an error right now. 1.123 + localName = sniffer.getLocalName(); 1.124 + namespaceUri = sniffer.getNsUri(); 1.125 + } 1.126 + } 1.127 + return new QName(namespaceUri, localName); 1.128 + } 1.129 + 1.130 + public static void serializeSource(Source src, XMLStreamWriter writer) throws XMLStreamException { 1.131 + XMLStreamReader reader = SourceReaderFactory.createSourceReader(src, true); 1.132 + int state; 1.133 + do { 1.134 + state = reader.next(); 1.135 + switch (state) { 1.136 + case XMLStreamConstants.START_ELEMENT: 1.137 + /* 1.138 + * TODO: Is this necessary, shouldn't zephyr return "" instead of 1.139 + * null for getNamespaceURI() and getPrefix()? 1.140 + */ 1.141 + String uri = reader.getNamespaceURI(); 1.142 + String prefix = reader.getPrefix(); 1.143 + String localName = reader.getLocalName(); 1.144 + 1.145 + if (prefix == null) { 1.146 + if (uri == null) { 1.147 + writer.writeStartElement(localName); 1.148 + } else { 1.149 + writer.writeStartElement(uri, localName); 1.150 + } 1.151 + } else { 1.152 + assert uri != null; 1.153 + 1.154 + if(prefix.length() > 0){ 1.155 + /** 1.156 + * Before we write the 1.157 + */ 1.158 + String writerURI = null; 1.159 + if (writer.getNamespaceContext() != null) 1.160 + writerURI = writer.getNamespaceContext().getNamespaceURI(prefix); 1.161 + String writerPrefix = writer.getPrefix(uri); 1.162 + if(declarePrefix(prefix, uri, writerPrefix, writerURI)){ 1.163 + writer.writeStartElement(prefix, localName, uri); 1.164 + writer.setPrefix(prefix, uri != null ? uri : ""); 1.165 + writer.writeNamespace(prefix, uri); 1.166 + }else{ 1.167 + writer.writeStartElement(prefix, localName, uri); 1.168 + } 1.169 + }else{ 1.170 + writer.writeStartElement(prefix, localName, uri); 1.171 + } 1.172 + } 1.173 + 1.174 + int n = reader.getNamespaceCount(); 1.175 + // Write namespace declarations 1.176 + for (int i = 0; i < n; i++) { 1.177 + String nsPrefix = reader.getNamespacePrefix(i); 1.178 + if (nsPrefix == null) nsPrefix = ""; 1.179 + // StAX returns null for default ns 1.180 + String writerURI = null; 1.181 + if (writer.getNamespaceContext() != null) 1.182 + writerURI = writer.getNamespaceContext().getNamespaceURI(nsPrefix); 1.183 + 1.184 + // Zephyr: Why is this returning null? 1.185 + // Compare nsPrefix with prefix because of [1] (above) 1.186 + String readerURI = reader.getNamespaceURI(i); 1.187 + 1.188 + /** 1.189 + * write the namespace in 3 conditions 1.190 + * - when the namespace URI is not bound to the prefix in writer(writerURI == 0) 1.191 + * - when the readerPrefix and writerPrefix are "" 1.192 + * - when readerPrefix and writerPrefix are not equal and the URI bound to them 1.193 + * are different 1.194 + */ 1.195 + if (writerURI == null || ((nsPrefix.length() == 0) || (prefix.length() == 0)) || 1.196 + (!nsPrefix.equals(prefix) && !writerURI.equals(readerURI))) { 1.197 + writer.setPrefix(nsPrefix, readerURI != null ? readerURI : ""); 1.198 + writer.writeNamespace(nsPrefix, readerURI != null ? readerURI : ""); 1.199 + } 1.200 + } 1.201 + 1.202 + // Write attributes 1.203 + n = reader.getAttributeCount(); 1.204 + for (int i = 0; i < n; i++) { 1.205 + String attrPrefix = reader.getAttributePrefix(i); 1.206 + String attrURI = reader.getAttributeNamespace(i); 1.207 + 1.208 + writer.writeAttribute(attrPrefix != null ? attrPrefix : "", 1.209 + attrURI != null ? attrURI : "", 1.210 + reader.getAttributeLocalName(i), 1.211 + reader.getAttributeValue(i)); 1.212 + // if the attribute prefix is undeclared in current writer scope then declare it 1.213 + setUndeclaredPrefix(attrPrefix, attrURI, writer); 1.214 + } 1.215 + break; 1.216 + case XMLStreamConstants.END_ELEMENT: 1.217 + writer.writeEndElement(); 1.218 + break; 1.219 + case XMLStreamConstants.CHARACTERS: 1.220 + writer.writeCharacters(reader.getText()); 1.221 + } 1.222 + } while (state != XMLStreamConstants.END_DOCUMENT); 1.223 + reader.close(); 1.224 + } 1.225 + 1.226 + /** 1.227 + * sets undeclared prefixes on the writer 1.228 + * @param prefix 1.229 + * @param writer 1.230 + * @throws XMLStreamException 1.231 + */ 1.232 + private static void setUndeclaredPrefix(String prefix, String readerURI, XMLStreamWriter writer) throws XMLStreamException { 1.233 + String writerURI = null; 1.234 + if (writer.getNamespaceContext() != null) 1.235 + writerURI = writer.getNamespaceContext().getNamespaceURI(prefix); 1.236 + 1.237 + if (writerURI == null) { 1.238 + writer.setPrefix(prefix, readerURI != null ? readerURI : ""); 1.239 + writer.writeNamespace(prefix, readerURI != null ? readerURI : ""); 1.240 + } 1.241 + } 1.242 + 1.243 + /** 1.244 + * check if we need to declare 1.245 + * @param rPrefix 1.246 + * @param rUri 1.247 + * @param wPrefix 1.248 + * @param wUri 1.249 + */ 1.250 + private static boolean declarePrefix(String rPrefix, String rUri, String wPrefix, String wUri){ 1.251 + if (wUri == null ||((wPrefix != null) && !rPrefix.equals(wPrefix))|| 1.252 + (rUri != null && !wUri.equals(rUri))) 1.253 + return true; 1.254 + return false; 1.255 + } 1.256 +}