/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2013 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.security.opt.impl.incoming;
import com.sun.istack.NotNull;
import com.sun.istack.Nullable;
import com.sun.istack.XMLStreamReaderToContentHandler;
import com.sun.xml.bind.api.Bridge;
import com.sun.xml.stream.buffer.stax.StreamReaderBufferCreator;
import com.sun.xml.ws.api.SOAPVersion;
import com.sun.xml.ws.api.message.AttachmentSet;
import com.sun.xml.ws.api.message.Header;
import com.sun.xml.ws.api.message.HeaderList;
import com.sun.xml.ws.api.message.Message;
import com.sun.xml.ws.api.message.MessageHeaders;
import com.sun.xml.ws.api.streaming.XMLStreamReaderFactory;
import com.sun.xml.ws.encoding.TagInfoset;
import com.sun.xml.ws.message.AbstractMessageImpl;
import com.sun.xml.ws.message.AttachmentUnmarshallerImpl;
import com.sun.xml.ws.streaming.XMLStreamReaderUtil;
import com.sun.xml.ws.util.xml.DummyLocation;
import com.sun.xml.ws.util.xml.StAXSource;
import com.sun.xml.ws.util.xml.XMLStreamReaderToXMLStreamWriter;
import java.security.PrivilegedActionException;
import java.util.Map;
import org.xml.sax.ContentHandler;
import org.xml.sax.ErrorHandler;
import org.xml.sax.SAXException;
import org.xml.sax.SAXParseException;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.stream.Location;
import javax.xml.stream.XMLStreamConstants;
import static javax.xml.stream.XMLStreamConstants.START_DOCUMENT;
import static javax.xml.stream.XMLStreamConstants.START_ELEMENT;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.stream.XMLStreamWriter;
import javax.xml.transform.Source;
import javax.xml.ws.WebServiceException;
import com.sun.xml.stream.buffer.MutableXMLStreamBuffer;
import com.sun.xml.ws.message.stream.StreamMessage;
import com.sun.xml.ws.security.message.stream.LazyStreamBasedMessage;
import com.sun.xml.wss.logging.LogDomainConstants;
import java.util.logging.Level;
import java.util.logging.Logger;
import com.sun.xml.wss.logging.impl.opt.LogStringsMessages;
import com.sun.xml.ws.security.opt.impl.util.VerifiedMessageXMLStreamReader;
import com.sun.xml.wss.impl.XWSSecurityRuntimeException;
import java.security.AccessController;
import java.security.PrivilegedExceptionAction;
import javax.xml.soap.SOAPException;
import javax.xml.soap.SOAPMessage;
/**
* {@link Message} implementation backed by {@link XMLStreamReader}.
*
* TODO: we need another message class that keeps {@link XMLStreamReader} that points
* at the start of the envelope element.
*/
public final class VerifiedStreamMessage extends AbstractMessageImpl {
/**
* The reader will be positioned at
* the first child of the SOAP body
*/
private
@NotNull
XMLStreamReader reader;
private MutableXMLStreamBuffer buffer = null;
// lazily created
private
@Nullable
MessageHeaders headers;
private final String payloadLocalName;
private final String payloadNamespaceURI;
private static final Logger logger = Logger.getLogger(LogDomainConstants.IMPL_OPT_DOMAIN,
LogDomainConstants.IMPL_OPT_DOMAIN_BUNDLE);
private Map<String, String> bodyEnvNs;
/**
* infoset about the SOAP envelope, header, and body.
*
* <p>
* If the creater of this object didn't care about those,
* we use stock values.
*/
private /*almost final*/
@NotNull
TagInfoset envelopeTag, headerTag, bodyTag;
/**
* Creates a {@link StreamMessage} from a {@link XMLStreamReader}
* that points at the start element of the payload, and headers.
*
* <p>
* This method creaets a {@link Message} from a payload.
*
* @param headers
* if null, it means no headers. if non-null,
* it will be owned by this message.
* @param reader
* points at the start element/document of the payload (or the end element of the <s:Body>
* if there's no payload)
*/
public VerifiedStreamMessage(
@Nullable MessageHeaders headers,
@NotNull AttachmentSet attachmentSet,
@NotNull XMLStreamReader reader,
@NotNull SOAPVersion soapVersion, Map<String, String> bodyEnvNs) {
super(soapVersion);
this.headers = headers;
this.attachmentSet = attachmentSet;
this.reader = reader;
this.bodyEnvNs = bodyEnvNs;
if (reader.getEventType() == START_DOCUMENT) {
XMLStreamReaderUtil.nextElementContent(reader);
}
//if the reader is pointing to the end element </soapenv:Body> then its empty message
// or no payload
if (reader.getEventType() == XMLStreamConstants.END_ELEMENT) {
String body = reader.getLocalName();
String nsUri = reader.getNamespaceURI();
assert body != null;
assert nsUri != null;
//if its not soapenv:Body then throw exception, we received malformed stream
if (body.equals("Body") && nsUri.equals(soapVersion.nsUri)) {
this.payloadLocalName = null;
this.payloadNamespaceURI = null;
} else { //TODO: i18n and also we should be throwing better message that this
throw new WebServiceException("Malformed stream: {" + nsUri + "}" + body);
}
} else {
this.payloadLocalName = reader.getLocalName();
this.payloadNamespaceURI = reader.getNamespaceURI();
}
// use the default infoset representation for headers
int base = soapVersion.ordinal() * 3;
this.envelopeTag = DEFAULT_TAGS[base];
this.headerTag = DEFAULT_TAGS[base + 1];
this.bodyTag = DEFAULT_TAGS[base + 2];
}
/**
* Creates a {@link StreamMessage} from a {@link XMLStreamReader}
* and the complete infoset of the SOAP envelope.
*
* <p>
* See {@link #StreamMessage(MessageHeaders, AttachmentSet, XMLStreamReader, SOAPVersion)} for
* the description of the basic parameters.
*
* @param headerTag
* Null if the message didn't have a header tag.
*
*/
public VerifiedStreamMessage(
@NotNull TagInfoset envelopeTag,
@Nullable TagInfoset headerTag,
@NotNull AttachmentSet attachmentSet,
@Nullable MessageHeaders headers,
@NotNull TagInfoset bodyTag,
@NotNull XMLStreamReader reader,
@NotNull SOAPVersion soapVersion, Map<String, String> bodyEnvNs) {
this(headers, attachmentSet, reader, soapVersion, bodyEnvNs);
assert envelopeTag != null && bodyTag != null;
this.envelopeTag = envelopeTag;
this.headerTag = headerTag != null ? headerTag : new TagInfoset(envelopeTag.nsUri, "Header", envelopeTag.prefix, EMPTY_ATTS);
this.bodyTag = bodyTag;
}
public boolean hasHeaders() {
// FIXME: RJE -- remove cast when MessageHeaders supports hasHeaders
return headers != null && !((HeaderList)headers).isEmpty();
}
public HeaderList getHeaders() {
if (headers == null) {
headers = new HeaderList();
}
// FIXME: RJE -- remove cast once getHeaders returns MessageContext
return (HeaderList) headers;
}
@Override
public
@NotNull
AttachmentSet getAttachments() {
return attachmentSet;
}
public String getPayloadLocalPart() {
return payloadLocalName;
}
public String getPayloadNamespaceURI() {
return payloadNamespaceURI;
}
public boolean hasPayload() {
return payloadLocalName != null;
}
public Source readPayloadAsSource() {
cacheMessage();
if (hasPayload()) {
assert unconsumed();
return new StAXSource(reader, true);
} else {
return null;
}
}
@SuppressWarnings("unchecked")
@Override
public <T> T readPayloadAsJAXB(final Unmarshaller unmarshaller) throws JAXBException {
try {
cacheMessage();
if (!hasPayload()) {
return null;
}
assert unconsumed();
// TODO: How can the unmarshaller process this as a fragment?
final Object ret = AccessController.doPrivileged(new PrivilegedExceptionAction() {
public Object run() throws Exception {
if (hasAttachments()) {
unmarshaller.setAttachmentUnmarshaller(new AttachmentUnmarshallerImpl(getAttachments()));
}
try {
if (reader.END_DOCUMENT == reader.getEventType() && buffer != null) {
try {
reader = buffer.readAsXMLStreamReader();
reader = new VerifiedMessageXMLStreamReader(reader, bodyEnvNs);
reader.next();
} catch (XMLStreamException ex) {
logger.log(Level.SEVERE, LogStringsMessages.WSS_1612_ERROR_READING_BUFFER(), ex);
throw new com.sun.xml.wss.impl.XWSSecurityRuntimeException(ex);
}
}
return unmarshaller.unmarshal(reader);
} finally {
unmarshaller.setAttachmentUnmarshaller(null);
XMLStreamReaderUtil.close(reader);
XMLStreamReaderFactory.recycle(reader);
}
}
});
return (T)ret;
} catch (PrivilegedActionException ex) {
throw new XWSSecurityRuntimeException(ex);
}
}
@Override
public <T> T readPayloadAsJAXB(Bridge<T> bridge) throws JAXBException {
cacheMessage();
if (!hasPayload()) {
return null;
}
assert unconsumed();
T r = bridge.unmarshal(reader,
hasAttachments() ? new AttachmentUnmarshallerImpl(getAttachments()) : null);
XMLStreamReaderUtil.close(reader);
XMLStreamReaderFactory.recycle(reader);
return r;
}
@Override
public void consume() {
assert unconsumed();
XMLStreamReaderFactory.recycle(reader);
}
public XMLStreamReader readPayload() {
cacheMessage();
// TODO: What about access at and beyond </soap:Body>
assert unconsumed();
return this.reader;
}
public void writePayloadTo(XMLStreamWriter writer) throws XMLStreamException {
if (payloadLocalName == null) {
return;
} // no body
assert unconsumed();
XMLStreamReaderToXMLStreamWriter conv = new XMLStreamReaderToXMLStreamWriter();
while (reader.getEventType() != XMLStreamConstants.END_DOCUMENT) {
String name = reader.getLocalName();
String nsUri = reader.getNamespaceURI();
//after previous conv.bridge() call the cursor will be at
//END_ELEMENT. Check if its not soapenv:Body then move to next
// ELEMENT
if (reader.getEventType() == XMLStreamConstants.END_ELEMENT) {
if (!name.equals("Body") || !nsUri.equals(soapVersion.nsUri)) {
XMLStreamReaderUtil.nextElementContent(reader);
if (reader.getEventType() == XMLStreamConstants.END_DOCUMENT) {
break;
}
name = reader.getLocalName();
nsUri = reader.getNamespaceURI();
}
}
if (name.equals("Body") && nsUri.equals(soapVersion.nsUri) || (reader.getEventType() == XMLStreamConstants.END_DOCUMENT)) {
break;
}
conv.bridge(reader, writer);
}
reader.close();
XMLStreamReaderFactory.recycle(reader);
}
@Override
public void writeTo(XMLStreamWriter sw) throws XMLStreamException {
writeEnvelope(sw);
}
/**
* This method should be called when the StreamMessage is created with a payload
* @param writer
*/
private void writeEnvelope(XMLStreamWriter writer) throws XMLStreamException {
writer.writeStartDocument();
envelopeTag.writeStart(writer);
//write headers
// FIXME: RJE -- remove cast
HeaderList hl = (HeaderList) getHeaders();
if (hl.size() > 0) {
headerTag.writeStart(writer);
for (Header h : hl) {
h.writeTo(writer);
}
writer.writeEndElement();
}
bodyTag.writeStart(writer);
if (hasPayload()) {
writePayloadTo(writer);
}
writer.writeEndElement();
writer.writeEndElement();
writer.writeEndDocument();
}
public void writePayloadTo(ContentHandler contentHandler, ErrorHandler errorHandler, boolean fragment) throws SAXException {
assert unconsumed();
try {
if (payloadLocalName == null) {
return;
} // no body
XMLStreamReaderToContentHandler conv =
new XMLStreamReaderToContentHandler(reader, contentHandler, true, fragment);
while (reader.getEventType() != XMLStreamConstants.END_DOCUMENT) {
String name = reader.getLocalName();
String nsUri = reader.getNamespaceURI();
//after previous conv.bridge() call the cursor will be at
//END_ELEMENT. Check if its not soapenv:Body then move to next
// ELEMENT
if (reader.getEventType() == XMLStreamConstants.END_ELEMENT) {
if (!name.equals("Body") || !nsUri.equals(soapVersion.nsUri)) {
XMLStreamReaderUtil.nextElementContent(reader);
if (reader.getEventType() == XMLStreamConstants.END_DOCUMENT) {
break;
}
name = reader.getLocalName();
nsUri = reader.getNamespaceURI();
}
}
if (name.equals("Body") && nsUri.equals(soapVersion.nsUri) || (reader.getEventType() == XMLStreamConstants.END_DOCUMENT)) {
break;
}
conv.bridge();
}
reader.close();
XMLStreamReaderFactory.recycle(reader);
} catch (XMLStreamException e) {
Location loc = e.getLocation();
if (loc == null) {
loc = DummyLocation.INSTANCE;
}
SAXParseException x = new SAXParseException(
e.getMessage(), loc.getPublicId(), loc.getSystemId(), loc.getLineNumber(), loc.getColumnNumber(), e);
errorHandler.error(x);
}
}
public Message copy() {
try {
// copy the payload
XMLStreamReader clone;
XMLStreamReader clonedReader;
if (hasPayload()) {
assert unconsumed();
consumedAt = null; // but we don't want to mark it as consumed
MutableXMLStreamBuffer xsb = new MutableXMLStreamBuffer();
//the boolean value tells the first body part is written.
//based on this we do the right thing
StreamReaderBufferCreator c = new StreamReaderBufferCreator(xsb);
while (reader.getEventType() != XMLStreamConstants.END_DOCUMENT) {
String name = reader.getLocalName();
String nsUri = reader.getNamespaceURI();
if (name.equals("Body") && nsUri.equals(soapVersion.nsUri) || (reader.getEventType() == XMLStreamConstants.END_DOCUMENT)) {
break;
}
c.create(reader);
}
XMLStreamReaderFactory.recycle(reader);
reader = xsb.readAsXMLStreamReader();
reader = new VerifiedMessageXMLStreamReader(reader, bodyEnvNs);
clone = xsb.readAsXMLStreamReader();
clonedReader = new VerifiedMessageXMLStreamReader(clone, bodyEnvNs);
// advance to the start tag of the first element
proceedToRootElement(reader);
proceedToRootElement(clonedReader);
} else {
// it's tempting to use EmptyMessageImpl, but it doesn't presere the infoset
// of <envelope>,<header>, and <body>, so we need to stick to StreamMessage.
clone = reader;
clonedReader = reader;
}
// FIXME: RJE -- remove cast once HeaderList.copy() can handle MessageContext
return new VerifiedStreamMessage(envelopeTag, headerTag, attachmentSet, HeaderList.copy((HeaderList)headers), bodyTag, clone, soapVersion, this.bodyEnvNs);
} catch (XMLStreamException e) {
throw new WebServiceException("Failed to copy a message", e);
}
}
private void proceedToRootElement(XMLStreamReader xsr) throws XMLStreamException {
assert xsr.getEventType() == START_DOCUMENT;
xsr.nextTag();
assert xsr.getEventType() == START_ELEMENT;
}
@Override
public SOAPMessage readAsSOAPMessage() throws SOAPException {
cacheMessage();
return super.readAsSOAPMessage();
}
@Override
public void writeTo(ContentHandler contentHandler, ErrorHandler errorHandler) throws SAXException {
contentHandler.setDocumentLocator(NULL_LOCATOR);
contentHandler.startDocument();
envelopeTag.writeStart(contentHandler);
headerTag.writeStart(contentHandler);
if (hasHeaders()) {
// FIXME: remove cast
HeaderList headerList = (HeaderList) getHeaders();
int len = headerList.size();
for (int i = 0; i < len; i++) {
// shouldn't JDK be smart enough to use array-style indexing for this foreach!?
headerList.get(i).writeTo(contentHandler, errorHandler);
}
}
headerTag.writeEnd(contentHandler);
bodyTag.writeStart(contentHandler);
writePayloadTo(contentHandler, errorHandler, true);
bodyTag.writeEnd(contentHandler);
envelopeTag.writeEnd(contentHandler);
}
/**
* Used for an assertion. Returns true when the message is unconsumed,
* or otherwise throw an exception.
*
* <p>
* Calling this method also marks the stream as 'consumed'
*/
private boolean unconsumed() {
if (payloadLocalName == null) {
return true;
} // no payload. can be consumed multiple times.
if (reader.getEventType() != XMLStreamReader.START_ELEMENT) {
System.out.append("Event Type=" + reader.getEventType() + " name=" + reader.getLocalName());
System.out.append("START " + XMLStreamReader.START_ELEMENT);
System.out.append("END =" + XMLStreamReader.END_ELEMENT);
AssertionError error = new AssertionError("StreamMessage has been already consumed. See the nested exception for where it's consumed");
error.initCause(consumedAt);
throw error;
}
consumedAt = new Exception().fillInStackTrace();
return true;
}
/**
* Used only for debugging. This records where the message was consumed.
*/
private Throwable consumedAt;
/**
* Default s:Envelope, s:Header, and s:Body tag infoset definitions.
*
* We need 3 for SOAP 1.1, 3 for SOAP 1.2.
*/
private static final TagInfoset[] DEFAULT_TAGS;
static {
DEFAULT_TAGS = new TagInfoset[6];
create(SOAPVersion.SOAP_11);
create(SOAPVersion.SOAP_12);
}
private static void create(SOAPVersion v) {
int base = v.ordinal() * 3;
DEFAULT_TAGS[base] = new TagInfoset(v.nsUri, "Envelope", "S", EMPTY_ATTS, "S", v.nsUri);
DEFAULT_TAGS[base + 1] = new TagInfoset(v.nsUri, "Header", "S", EMPTY_ATTS);
DEFAULT_TAGS[base + 2] = new TagInfoset(v.nsUri, "Body", "S", EMPTY_ATTS);
}
private void cacheMessage() {
//TODO 1081: need to verify this
if (LazyStreamBasedMessage.mtomLargeData()) {
return;
}
if (buffer == null) {
try {
buffer = new com.sun.xml.stream.buffer.MutableXMLStreamBuffer();
buffer.createFromXMLStreamReader(reader);
} catch (javax.xml.stream.XMLStreamException ex) {
logger.log(Level.SEVERE, LogStringsMessages.WSS_1611_PROBLEM_CACHING(), ex);
throw new com.sun.xml.wss.impl.XWSSecurityRuntimeException(ex);
}
try {
reader = buffer.readAsXMLStreamReader();
reader = new VerifiedMessageXMLStreamReader(reader, bodyEnvNs);
reader.next();
} catch (XMLStreamException ex) {
logger.log(Level.SEVERE, LogStringsMessages.WSS_1612_ERROR_READING_BUFFER(), ex);
throw new com.sun.xml.wss.impl.XWSSecurityRuntimeException(ex);
}
}
if (reader.getEventType() == XMLStreamReader.END_DOCUMENT) {
if (buffer != null) {
try {
reader = buffer.readAsXMLStreamReader();
reader = new VerifiedMessageXMLStreamReader(reader, bodyEnvNs);
reader.next();
} catch (XMLStreamException ex) {
logger.log(Level.SEVERE, LogStringsMessages.WSS_1612_ERROR_READING_BUFFER(), ex);
throw new com.sun.xml.wss.impl.XWSSecurityRuntimeException(ex);
}
}
}
}
}