diff -r 88b85470e72c -r f50545b5e2f1 src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java Tue Mar 06 16:09:35 2012 -0800 @@ -0,0 +1,232 @@ +/* + * Copyright (c) 1997, 2010, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package com.sun.xml.internal.ws.wsdl.writer; + +import com.sun.istack.internal.NotNull; +import com.sun.xml.internal.ws.api.server.PortAddressResolver; +import com.sun.xml.internal.ws.util.xml.XMLStreamReaderToXMLStreamWriter; +import com.sun.xml.internal.ws.wsdl.parser.WSDLConstants; +import com.sun.xml.internal.ws.addressing.W3CAddressingConstants; +import com.sun.istack.internal.Nullable; + +import javax.xml.namespace.QName; +import javax.xml.stream.XMLStreamException; +import java.util.logging.Logger; + +/** + * Patches WSDL with the correct endpoint address and the relative paths + * to other documents. + * + * @author Jitendra Kotamraju + * @author Kohsuke Kawaguchi + */ +public final class WSDLPatcher extends XMLStreamReaderToXMLStreamWriter { + + private static final String NS_XSD = "http://www.w3.org/2001/XMLSchema"; + private static final QName SCHEMA_INCLUDE_QNAME = new QName(NS_XSD, "include"); + private static final QName SCHEMA_IMPORT_QNAME = new QName(NS_XSD, "import"); + private static final QName SCHEMA_REDEFINE_QNAME = new QName(NS_XSD, "redefine"); + + private static final Logger logger = Logger.getLogger( + com.sun.xml.internal.ws.util.Constants.LoggingDomain + ".wsdl.patcher"); + + private final DocumentLocationResolver docResolver; + private final PortAddressResolver portAddressResolver; + + // + // fields accumulated as we parse through documents + // + private String targetNamespace; + private QName serviceName; + private QName portName; + private String portAddress; + + // true inside // + private boolean inEpr; + // true inside /// + private boolean inEprAddress; + + /** + * Creates a {@link WSDLPatcher} for patching WSDL. + * + * @param portAddressResolver + * address of the endpoint is resolved using this docResolver. + * @param docResolver + * Consulted to get the import/include document locations. + * Must not be null. + */ + public WSDLPatcher(@NotNull PortAddressResolver portAddressResolver, + @NotNull DocumentLocationResolver docResolver) { + this.portAddressResolver = portAddressResolver; + this.docResolver = docResolver; + } + + @Override + protected void handleAttribute(int i) throws XMLStreamException { + QName name = in.getName(); + String attLocalName = in.getAttributeLocalName(i); + + if((name.equals(SCHEMA_INCLUDE_QNAME) && attLocalName.equals("schemaLocation")) + || (name.equals(SCHEMA_IMPORT_QNAME) && attLocalName.equals("schemaLocation")) + || (name.equals(SCHEMA_REDEFINE_QNAME) && attLocalName.equals("schemaLocation")) + || (name.equals(WSDLConstants.QNAME_IMPORT) && attLocalName.equals("location"))) { + // patch this attribute value. + + String relPath = in.getAttributeValue(i); + String actualPath = getPatchedImportLocation(relPath); + if (actualPath == null) { + return; // skip this attribute to leave it up to "implicit reference". + } + + logger.fine("Fixing the relative location:"+relPath + +" with absolute location:"+actualPath); + writeAttribute(i, actualPath); + return; + } + + if (name.equals(WSDLConstants.NS_SOAP_BINDING_ADDRESS) || + name.equals(WSDLConstants.NS_SOAP12_BINDING_ADDRESS)) { + + if(attLocalName.equals("location")) { + portAddress = in.getAttributeValue(i); + String value = getAddressLocation(); + if (value != null) { + logger.fine("Service:"+serviceName+ " port:"+portName + + " current address "+portAddress+" Patching it with "+value); + writeAttribute(i, value); + return; + } + } + } + + super.handleAttribute(i); + } + + /** + * Writes out an {@code i}-th attribute but with a different value. + * @param i attribute index + * @param value attribute value + * @throws XMLStreamException when an error encountered while writing attribute + */ + private void writeAttribute(int i, String value) throws XMLStreamException { + String nsUri = in.getAttributeNamespace(i); + if(nsUri!=null) + out.writeAttribute( in.getAttributePrefix(i), nsUri, in.getAttributeLocalName(i), value ); + else + out.writeAttribute( in.getAttributeLocalName(i), value ); + } + + @Override + protected void handleStartElement() throws XMLStreamException { + QName name = in.getName(); + + if (name.equals(WSDLConstants.QNAME_DEFINITIONS)) { + String value = in.getAttributeValue(null,"targetNamespace"); + if (value != null) { + targetNamespace = value; + } + } else if (name.equals(WSDLConstants.QNAME_SERVICE)) { + String value = in.getAttributeValue(null,"name"); + if (value != null) { + serviceName = new QName(targetNamespace, value); + } + } else if (name.equals(WSDLConstants.QNAME_PORT)) { + String value = in.getAttributeValue(null,"name"); + if (value != null) { + portName = new QName(targetNamespace,value); + } + } else if (name.equals(W3CAddressingConstants.WSA_EPR_QNAME)) { + if (serviceName != null && portName != null) { + inEpr = true; + } + } else if (name.equals(W3CAddressingConstants.WSA_ADDRESS_QNAME)) { + if (inEpr) { + inEprAddress = true; + } + } + super.handleStartElement(); + } + + @Override + protected void handleEndElement() throws XMLStreamException { + QName name = in.getName(); + if (name.equals(WSDLConstants.QNAME_SERVICE)) { + serviceName = null; + } else if (name.equals(WSDLConstants.QNAME_PORT)) { + portName = null; + } else if (name.equals(W3CAddressingConstants.WSA_EPR_QNAME)) { + if (inEpr) { + inEpr = false; + } + } else if (name.equals(W3CAddressingConstants.WSA_ADDRESS_QNAME)) { + if (inEprAddress) { + String value = getAddressLocation(); + if (value != null) { + logger.fine("Fixing EPR Address for service:"+serviceName+ " port:"+portName + + " address with "+value); + out.writeCharacters(value); + } + inEprAddress = false; + } + } + super.handleEndElement(); + } + + @Override + protected void handleCharacters() throws XMLStreamException { + // handleCharacters() may be called multiple times. + if (inEprAddress) { + String value = getAddressLocation(); + if (value != null) { + // will write the address with end element + return; + } + } + super.handleCharacters(); + } + + /** + * Returns the location to be placed into the generated document. + * + * @param relPath relative URI to be resolved + * @return + * null to leave it to the "implicit reference". + */ + private @Nullable String getPatchedImportLocation(String relPath) { + return docResolver.getLocationFor(null, relPath); + } + + /** + * For the given service, port names it matches the correct endpoint and + * reutrns its endpoint address + * + * @return returns the resolved endpoint address + */ + private String getAddressLocation() { + return (portAddressResolver == null || portName == null) + ? null : portAddressResolver.getAddressFor(serviceName, portName.getLocalPart(), portAddress); + } +}