com.sun.xml.wss.impl.c14n.EXC14nStAXReaderBasedCanonicalizer Maven / Gradle / Ivy
/*
* Copyright (c) 1997, 2018 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Distribution License v. 1.0, which is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
package com.sun.xml.wss.impl.c14n;
import com.sun.xml.wss.impl.dsig.NamespaceContextImpl;
import com.sun.xml.wss.impl.misc.UnsyncByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import static javax.xml.stream.XMLStreamConstants.*;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
/**
*
* @author [email protected]
*/
public class EXC14nStAXReaderBasedCanonicalizer extends BaseCanonicalizer {
private NamespaceContextImpl _exC14NContext = new NamespaceContextImpl();
private List _inclusivePrefixList = null;
private UnsyncByteArrayOutputStream _tmpBuffer = null;
private HashSet _visiblyUtilized = new HashSet();
private int _index =0;
/** Creates a new instance of EXC14nStAXReaderBasedCanonicalizer */
@SuppressWarnings("unchecked")
public EXC14nStAXReaderBasedCanonicalizer() {
_attrResult = new ArrayList();
for(int i=0;i<4;i++){
_attrs.add(new StAXAttr());
}
_tmpBuffer = new UnsyncByteArrayOutputStream();
}
public void canonicalize(XMLStreamReader reader,OutputStream stream,List inclusiveList) throws XMLStreamException, IOException{
if(reader.hasNext() && reader.getEventType() != START_ELEMENT){
throw new XMLStreamException("Reader should point to START_ELEMENT EVENT");
}
updatedNamespaceContext(reader);
this._stream = stream;
this._inclusivePrefixList = inclusiveList;
int eventType = reader.getEventType();
do{
switch(eventType){
case START_ELEMENT :{
_exC14NContext.push();
_index++;
writeStartElement(reader);
break;
}
case END_ELEMENT :{
_exC14NContext.pop();
_index--;
writeEndElement(reader);
break;
}
case CDATA:{
outputTextToWriter(reader.getTextCharacters(),reader.getTextStart(),reader.getTextLength(),_stream);
break;
}
case CHARACTERS :{
if(!reader.isWhiteSpace()){
outputTextToWriter(reader.getText(),_stream);
}
break;
}
case COMMENT :{
break;
}
case DTD :{
break;
}
case END_DOCUMENT :{
break;
}
case ENTITY_DECLARATION :{
break;
}
case ENTITY_REFERENCE :{
break;
}
case NOTATION_DECLARATION :{
break;
}
case PROCESSING_INSTRUCTION :{
break;
}
case SPACE :{
break;
}
case START_DOCUMENT :{
break;
}
default :{
break;
}
}
eventType = reader.next();
} while(reader.hasNext() && _index >0);
}
@SuppressWarnings("unchecked")
private void writeStartElement(XMLStreamReader reader) throws IOException {
final String localName = reader.getLocalName();
final String prefix = reader.getPrefix();
final String uri = reader.getNamespaceURI();
writeCharToUtf8('<',_stream);
if(prefix == null && uri == null){
writeStringToUtf8(localName,_stream);
}else if(prefix != null && prefix.length() > 0){
writeStringToUtf8(prefix,_stream);
writeStringToUtf8(":",_stream);
writeStringToUtf8(localName,_stream);
} else if(prefix == null){
writeStringToUtf8(localName,_stream);
}
updatedNamespaceContext(reader);
updateAttributes(reader);
if(prefix != null){
_visiblyUtilized.add(prefix);
}
if(_elementPrefix.length() >0){
AttributeNS eDecl = _exC14NContext.getNamespaceDeclaration(_elementPrefix);
if(eDecl !=null && !eDecl.isWritten()){
eDecl.setWritten(true);
_nsResult.add(eDecl);
}
}
if(_visiblyUtilized.size() > 0){
Iterator prefixItr = _visiblyUtilized.iterator();
populateNamespaceDecl(prefixItr);
}
if(_inclusivePrefixList != null){
populateNamespaceDecl(_inclusivePrefixList.iterator());
}
if ( _nsResult.size() > 0) {
BaseCanonicalizer.sort(_nsResult);
writeAttributesNS(_nsResult);
}
if ( _attrResult.size() > 0 ) {
BaseCanonicalizer.sort(_attrResult);
writeAttributes(_attrResult);
}
writeCharToUtf8('>',_stream);
_nsResult.clear();
_attrResult.clear();
_visiblyUtilized.clear();
}
@SuppressWarnings("unchecked")
private void populateNamespaceDecl(Iterator prefixItr){
AttributeNS nsDecl = null;
while(prefixItr.hasNext() ){
String prefix = (String)prefixItr.next();
nsDecl = _exC14NContext.getNamespaceDeclaration(prefix);
if(nsDecl !=null && !nsDecl.isWritten()){
nsDecl.setWritten(true);
_nsResult.add(nsDecl);
}
}
}
private void updatedNamespaceContext(XMLStreamReader reader){
if(reader.getEventType() != reader.START_ELEMENT){
return;
}
int count = reader.getNamespaceCount();
for(int i=0;i 0){
writeStringToUtf8(prefix,_stream);
writeStringToUtf8(":",_stream);
writeStringToUtf8(localName,_stream);
} else if(prefix == null){
writeStringToUtf8(localName,_stream);
}
writeCharToUtf8('>',_stream);
}
@SuppressWarnings("unchecked")
protected StAXAttr getAttribute(){
if(_attrPos < _attrs.size() ){
return (StAXAttr)_attrs.get(_attrPos++);
}else{
for(int i=0;i