/**
* Builds a SOAPEnvelope from DOM Document.
*
* @param doc - The dom document that contains a SOAP message
* @param useDoom
* @return
* @throws WSSecurityException
*/
public static SOAPEnvelope getSOAPEnvelopeFromDOMDocument(Document doc, boolean useDoom)
throws WSSecurityException {
if (useDoom) {
try {
// Get processed headers
SOAPEnvelope env = (SOAPEnvelope) doc.getDocumentElement();
ArrayList processedHeaderQNames = new ArrayList();
SOAPHeader soapHeader = env.getHeader();
if (soapHeader != null) {
Iterator headerBlocs = soapHeader.getChildElements();
while (headerBlocs.hasNext()) {
OMElement element = (OMElement) headerBlocs.next();
SOAPHeaderBlock header = null;
if (element instanceof SOAPHeaderBlock) {
header = (SOAPHeaderBlock) element;
// If a header block is not an instance of SOAPHeaderBlock, it means that
// it is a header we have added in rampart eg. EncryptedHeader and should
// be converted to SOAPHeaderBlock for processing
} else {
header = soapHeader.addHeaderBlock(element.getLocalName(), element.getNamespace());
Iterator attrIter = element.getAllAttributes();
while (attrIter.hasNext()) {
OMAttribute attr = (OMAttribute) attrIter.next();
header.addAttribute(
attr.getLocalName(), attr.getAttributeValue(), attr.getNamespace());
}
Iterator nsIter = element.getAllDeclaredNamespaces();
while (nsIter.hasNext()) {
OMNamespace ns = (OMNamespace) nsIter.next();
header.declareNamespace(ns);
}
// retrieve all child nodes (including any text nodes)
// and re-attach to header block
Iterator children = element.getChildren();
while (children.hasNext()) {
OMNode child = (OMNode) children.next();
child.detach();
header.addChild(child);
}
element.detach();
soapHeader.build();
header.setProcessed();
}
if (header.isProcessed()) {
processedHeaderQNames.add(element.getQName());
}
}
}
XMLStreamReader reader = ((OMElement) doc.getDocumentElement()).getXMLStreamReader();
StAXSOAPModelBuilder stAXSOAPModelBuilder = new StAXSOAPModelBuilder(reader, null);
SOAPEnvelope envelope = stAXSOAPModelBuilder.getSOAPEnvelope();
// Set the processed flag of the processed headers
SOAPHeader header = envelope.getHeader();
for (Iterator iter = processedHeaderQNames.iterator(); iter.hasNext(); ) {
QName name = (QName) iter.next();
Iterator omKids = header.getChildrenWithName(name);
if (omKids.hasNext()) {
((SOAPHeaderBlock) omKids.next()).setProcessed();
}
}
envelope.build();
return envelope;
} catch (FactoryConfigurationError e) {
throw new WSSecurityException(e.getMessage());
}
} else {
try {
ByteArrayOutputStream os = new ByteArrayOutputStream();
XMLUtils.outputDOM(doc.getDocumentElement(), os, true);
ByteArrayInputStream bais = new ByteArrayInputStream(os.toByteArray());
StAXSOAPModelBuilder stAXSOAPModelBuilder =
new StAXSOAPModelBuilder(
XMLInputFactory.newInstance().createXMLStreamReader(bais), null);
return stAXSOAPModelBuilder.getSOAPEnvelope();
} catch (Exception e) {
throw new WSSecurityException(e.getMessage());
}
}
}
public void toHeader(SOAPHeader header) throws SandeshaException {
SOAPHeaderBlock sequenceAcknowledgementHeaderBlock =
header.addHeaderBlock(Sandesha2Constants.WSRM_COMMON.SEQUENCE_ACK, omNamespace);
if (identifier == null)
throw new OMException(
SandeshaMessageHelper.getMessage(
SandeshaMessageKeys.invalidIdentifier, header.toString()));
// SequenceACK messages should always have the MustUnderstand flag set to true
// Unless it is a piggybacked ack
if (piggybacked) {
sequenceAcknowledgementHeaderBlock.setMustUnderstand(false);
} else {
sequenceAcknowledgementHeaderBlock.setMustUnderstand(true);
}
identifier.toOMElement(sequenceAcknowledgementHeaderBlock, omNamespace);
Iterator<Range> ackRangeIt = acknowledgementRangeList.iterator();
while (ackRangeIt.hasNext()) {
Range ackRange = (Range) ackRangeIt.next();
if (ackRange.upperValue <= 0
|| ackRange.lowerValue <= 0
|| ackRange.lowerValue > ackRange.upperValue)
throw new OMException(
SandeshaMessageHelper.getMessage(
SandeshaMessageKeys.ackRandDoesNotHaveCorrectValues,
ackRange.upperValue + ":" + ackRange.lowerValue));
OMFactory factory = sequenceAcknowledgementHeaderBlock.getOMFactory();
OMAttribute lowerAttrib =
factory.createOMAttribute(
Sandesha2Constants.WSRM_COMMON.LOWER, null, Long.toString(ackRange.lowerValue));
OMAttribute upperAttrib =
factory.createOMAttribute(
Sandesha2Constants.WSRM_COMMON.UPPER, null, Long.toString(ackRange.upperValue));
OMElement acknowledgementRangeElement =
factory.createOMElement(Sandesha2Constants.WSRM_COMMON.ACK_RANGE, omNamespace);
acknowledgementRangeElement.addAttribute(lowerAttrib);
acknowledgementRangeElement.addAttribute(upperAttrib);
sequenceAcknowledgementHeaderBlock.addChild(acknowledgementRangeElement);
}
Iterator<Long> nackIt = nackList.iterator();
while (nackIt.hasNext()) {
Long nack = (Long) nackIt.next();
if (nack.longValue() <= 0)
throw new OMException(
SandeshaMessageHelper.getMessage(SandeshaMessageKeys.nackDoesNotContainValidLongValue));
OMElement nackElement =
sequenceAcknowledgementHeaderBlock
.getOMFactory()
.createOMElement(Sandesha2Constants.WSRM_COMMON.NACK, omNamespace);
nackElement.setText(nack.toString());
sequenceAcknowledgementHeaderBlock.addChild(nackElement);
}
String rmSpecVersion = SpecSpecificConstants.getSpecVersionString(namespaceValue);
// setting a 'None' when nothing is there (for the correct RM version)
if (ackNone == false
&& acknowledgementRangeList.size() == 0
&& nackList.size() == 0
&& SpecSpecificConstants.isAckNoneAllowed(rmSpecVersion)) {
ackNone = true;
}
if (ackNone != false) {
if (!SpecSpecificConstants.isAckNoneAllowed(rmSpecVersion)) {
throw new SandeshaException(
SandeshaMessageHelper.getMessage(
SandeshaMessageKeys.noneNotAllowedNamespace, rmSpecVersion));
}
if (acknowledgementRangeList.size() > 0) {
throw new SandeshaException(
SandeshaMessageHelper.getMessage(SandeshaMessageKeys.noneNotAllowedAckRangesPresent));
}
if (nackList.size() > 0) {
throw new SandeshaException(
SandeshaMessageHelper.getMessage(SandeshaMessageKeys.noneNotAllowedNackPresent));
}
OMElement noneElement =
sequenceAcknowledgementHeaderBlock
.getOMFactory()
.createOMElement(Sandesha2Constants.WSRM_COMMON.NONE, omNamespace);
sequenceAcknowledgementHeaderBlock.addChild(noneElement);
}
if (ackFinal != false) {
if (!SpecSpecificConstants.isAckFinalAllowed(rmSpecVersion)) {
throw new SandeshaException(
SandeshaMessageHelper.getMessage(SandeshaMessageKeys.finalNotAllowedNamespace));
}
if (nackList.size() > 0) {
throw new SandeshaException(
SandeshaMessageHelper.getMessage(SandeshaMessageKeys.cannotHaveFinalWithNack));
}
OMElement finalElement =
sequenceAcknowledgementHeaderBlock
.getOMFactory()
.createOMElement(Sandesha2Constants.WSRM_COMMON.FINAL, omNamespace);
sequenceAcknowledgementHeaderBlock.addChild(finalElement);
}
}
