/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2012 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.xml.ws.config.management.server;
import com.sun.istack.logging.Logger;
import com.sun.xml.txw2.output.StaxSerializer;
import com.sun.xml.ws.api.policy.ModelGenerator;
import com.sun.xml.ws.config.management.ManagementMessages;
import com.sun.xml.ws.policy.Policy;
import com.sun.xml.ws.policy.PolicyException;
import com.sun.xml.ws.policy.sourcemodel.PolicyModelGenerator;
import com.sun.xml.ws.policy.sourcemodel.PolicyModelMarshaller;
import com.sun.xml.ws.policy.sourcemodel.PolicySourceModel;
import com.sun.xml.ws.policy.sourcemodel.attach.ExternalAttachmentsUnmarshaller;
import com.sun.xml.ws.policy.sourcemodel.wspolicy.NamespaceVersion;
import com.sun.xml.ws.policy.sourcemodel.wspolicy.XmlToken;
import com.sun.xml.ws.util.xml.XMLStreamReaderToXMLStreamWriter;
import com.sun.xml.ws.wsdl.parser.WSDLConstants;
import java.net.URI;
import java.util.Map;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.ws.WebServiceException;
/**
* Remove all existing policies and policy references from the given XML document
* and insert the new effective policies.
*
* @author Fabian Ritzmann
*/
public class ManagementWSDLPatcher extends XMLStreamReaderToXMLStreamWriter {
private static final Logger LOGGER = Logger.getLogger(ManagementWSDLPatcher.class);
private static final PolicyModelMarshaller POLICY_MARSHALLER = PolicyModelMarshaller.getXmlMarshaller(true);
private static final PolicyModelGenerator POLICY_GENERATOR = ModelGenerator.getGenerator();
private final Map<URI, Policy> urnToPolicy;
// Skip element if this value is 0 or positive
private long skipDepth = -1L;
private boolean inBinding = false;
public ManagementWSDLPatcher(Map<URI, Policy> urnToPolicy) {
this.urnToPolicy = urnToPolicy;
}
/**
* If we find a policy element, skip it. If we find a binding element,
* marshal any policies onto it.
*
* @throws XMLStreamException If a parsing error occured
*/
@Override
protected void handleStartElement() throws XMLStreamException {
if (this.skipDepth >= 0L) {
this.skipDepth++;
return;
}
final QName elementName = this.in.getName();
final XmlToken policyToken = NamespaceVersion.resolveAsToken(elementName);
if (policyToken != XmlToken.UNKNOWN) {
this.skipDepth++;
return;
}
else if (elementName.equals(WSDLConstants.QNAME_BINDING)) {
this.inBinding = true;
super.handleStartElement();
final Policy bindingPolicy = urnToPolicy.get(ExternalAttachmentsUnmarshaller.BINDING_ID);
if (bindingPolicy != null) {
writePolicy(bindingPolicy);
}
}
else if (this.inBinding && elementName.equals(WSDLConstants.QNAME_OPERATION)) {
super.handleStartElement();
final Policy operationPolicy = urnToPolicy.get(ExternalAttachmentsUnmarshaller.BINDING_OPERATION_ID);
if (operationPolicy != null) {
writePolicy(operationPolicy);
}
}
else if (this.inBinding && elementName.equals(WSDLConstants.QNAME_INPUT)) {
super.handleStartElement();
final Policy inputPolicy = urnToPolicy.get(ExternalAttachmentsUnmarshaller.BINDING_OPERATION_INPUT_ID);
if (inputPolicy != null) {
writePolicy(inputPolicy);
}
}
else if (this.inBinding && elementName.equals(WSDLConstants.QNAME_OUTPUT)) {
super.handleStartElement();
final Policy outputPolicy = urnToPolicy.get(ExternalAttachmentsUnmarshaller.BINDING_OPERATION_OUTPUT_ID);
if (outputPolicy != null) {
writePolicy(outputPolicy);
}
}
else if (this.inBinding && elementName.equals(WSDLConstants.QNAME_FAULT)) {
super.handleStartElement();
final Policy faultPolicy = urnToPolicy.get(ExternalAttachmentsUnmarshaller.BINDING_OPERATION_FAULT_ID);
if (faultPolicy != null) {
writePolicy(faultPolicy);
}
}
else {
super.handleStartElement();
}
}
/**
* Skip all policy expressions.
*
* @throws XMLStreamException If a parsing error occured
*/
@Override
protected void handleEndElement() throws XMLStreamException {
final QName elementName = this.in.getName();
if (this.inBinding) {
this.inBinding = !elementName.equals(WSDLConstants.QNAME_BINDING);
}
if (this.skipDepth < 0L) {
super.handleEndElement();
return;
}
else {
this.skipDepth--;
return;
}
}
/**
* Skip all policy attributes
*
* @param i The i-th attribute of the current element
* @throws XMLStreamException If a parsing error occured
*/
@Override
protected void handleAttribute(int i) throws XMLStreamException {
final QName attributeName = this.in.getAttributeName(i);
final XmlToken policyToken = NamespaceVersion.resolveAsToken(attributeName);
switch (policyToken) {
case PolicyUris:
return;
}
super.handleAttribute(i);
}
@Override
protected void handleCharacters() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleCharacters();
}
}
@Override
protected void handleComment() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleComment();
}
}
@Override
protected void handlePI() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handlePI();
}
}
@Override
protected void handleDTD() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleDTD();
}
}
@Override
protected void handleEntityReference() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleEntityReference();
}
}
@Override
protected void handleSpace() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleSpace();
}
}
@Override
protected void handleCDATA() throws XMLStreamException {
if (this.skipDepth < 0L) {
super.handleCDATA();
}
}
private void writePolicy(final Policy policy) {
try {
final PolicySourceModel policyModel = POLICY_GENERATOR.translate(policy);
this.out.writeCharacters("\n");
final StaxSerializer serializer = new FragmentSerializer(this.out);
POLICY_MARSHALLER.marshal(policyModel, serializer);
} catch (PolicyException e) {
throw LOGGER.logSevereException(new WebServiceException(
ManagementMessages.WSM_5096_CANNOT_MARSHAL(this.out)), e);
} catch (XMLStreamException e) {
throw LOGGER.logSevereException(new WebServiceException(
ManagementMessages.WSM_5096_CANNOT_MARSHAL(this.out)), e);
}
}
class FragmentSerializer extends StaxSerializer {
public FragmentSerializer(XMLStreamWriter writer) {
super(writer);
}
@Override
public void endDocument() {
return;
}
@Override
public void startDocument() {
return;
}
}
}