io.rubrica.sign.ooxml.relprovider.RelationshipTransformService Maven / Gradle / Ivy
/*
* Copyright 2009-2017 Rubrica
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see sourceIds;
/** Crea el servicio de la transformación RelationshipTransform. */
public RelationshipTransformService() {
super();
this.sourceIds = new LinkedList<>();
}
/** {@inheritDoc} */
@Override
public void init(TransformParameterSpec params) throws InvalidAlgorithmParameterException {
if (!(params instanceof RelationshipTransformParameterSpec)) {
throw new InvalidAlgorithmParameterException();
}
RelationshipTransformParameterSpec relParams = (RelationshipTransformParameterSpec) params;
for (String sourceId : relParams.getSourceIds()) {
this.sourceIds.add(sourceId);
}
}
/** {@inheritDoc} */
@Override
public void init(XMLStructure parent, XMLCryptoContext context) throws InvalidAlgorithmParameterException {
DOMStructure domParent = (DOMStructure) parent;
Node parentNode = domParent.getNode();
try {
toString(parentNode);
} catch (TransformerException e) {
throw new InvalidAlgorithmParameterException(e);
}
NodeList nodeList;
try {
XPath xpath = XPathFactory.newInstance().newXPath();
xpath.setNamespaceContext(new NamespaceContext() {
@Override
public Iterator getPrefixes(String namespaceURI) {
throw new UnsupportedOperationException();
}
@Override
public String getPrefix(String namespaceURI) {
throw new UnsupportedOperationException();
}
@Override
public String getNamespaceURI(String prefix) {
if (prefix == null) {
throw new IllegalArgumentException("El prefijo no puede ser nulo");
}
if ("xml".equals(prefix)) { //$NON-NLS-1$
return XMLConstants.XML_NS_URI;
}
if ("ds".equals(prefix)) { //$NON-NLS-1$
return SIGNATURE_SPEC_NS;
}
if ("mdssi".equals(prefix)) { //$NON-NLS-1$
return "http://schemas.openxmlformats.org/package/2006/digital-signature";
}
return XMLConstants.NULL_NS_URI;
}
});
XPathExpression exp = xpath.compile("mdssi:RelationshipReference/@SourceId");
nodeList = (NodeList) exp.evaluate(parentNode, XPathConstants.NODESET);
} catch (Exception e) {
logger.severe("Error en la transformacion XPath: " + e);
throw new InvalidAlgorithmParameterException(e);
}
if (0 == nodeList.getLength()) {
logger.warning("no RelationshipReference/@SourceId parameters present");
}
for (int nodeIdx = 0; nodeIdx < nodeList.getLength(); nodeIdx++) {
Node node = nodeList.item(nodeIdx);
String sourceId = node.getTextContent();
this.sourceIds.add(sourceId);
}
}
/** {@inheritDoc} */
@Override
public void marshalParams(XMLStructure parent, XMLCryptoContext context) {
DOMStructure domParent = (DOMStructure) parent;
Node parentNode = domParent.getNode();
Element parentElement = (Element) parentNode;
parentElement.setAttributeNS(NAMESPACE_SPEC_NS, "xmlns:mdssi",
"http://schemas.openxmlformats.org/package/2006/digital-signature");
Document document = parentNode.getOwnerDocument();
for (String sourceId : this.sourceIds) {
Element relationshipReferenceElement = document.createElementNS(
"http://schemas.openxmlformats.org/package/2006/digital-signature", "mdssi:RelationshipReference");
relationshipReferenceElement.setAttribute("SourceId", sourceId);
parentElement.appendChild(relationshipReferenceElement);
}
}
/** {@inheritDoc} */
@Override
public AlgorithmParameterSpec getParameterSpec() {
return null;
}
/** {@inheritDoc} */
@Override
public Data transform(Data data, XMLCryptoContext context) throws TransformException {
OctetStreamData octetStreamData = (OctetStreamData) data;
Document relationshipsDocument;
try (InputStream octetStream = octetStreamData.getOctetStream();) {
relationshipsDocument = loadDocument(octetStream);
octetStream.close();
} catch (Exception e) {
throw new TransformException(e.getMessage(), e);
}
try {
toString(relationshipsDocument);
} catch (TransformerException e) {
throw new TransformException(e.getMessage(), e);
}
Element nsElement = relationshipsDocument.createElement("ns");
nsElement.setAttributeNS(NAMESPACE_SPEC_NS, "xmlns:tns",
"http://schemas.openxmlformats.org/package/2006/relationships");
Element relationshipsElement = relationshipsDocument.getDocumentElement();
NodeList childNodes = relationshipsElement.getChildNodes();
for (int nodeIdx = 0; nodeIdx < childNodes.getLength(); nodeIdx++) {
Node childNode = childNodes.item(nodeIdx);
if (Node.ELEMENT_NODE != childNode.getNodeType()) {
relationshipsElement.removeChild(childNode);
nodeIdx--;
continue;
}
Element childElement = (Element) childNode;
String idAttribute = childElement.getAttribute("Id");
if (!this.sourceIds.contains(idAttribute)) {
relationshipsElement.removeChild(childNode);
nodeIdx--;
}
/*
* See: ISO/IEC 29500-2:2008(E) - 13.2.4.24 Relationships Transform
* Algorithm.
*/
if (null == childElement.getAttributeNode("TargetMode")) {
childElement.setAttribute("TargetMode", "Internal");
}
}
sortRelationshipElements(relationshipsElement);
try {
return toOctetStreamData(relationshipsDocument);
} catch (TransformerException e) {
throw new TransformException(e.getMessage(), e);
}
}
private static void sortRelationshipElements(Element relationshipsElement) {
List relationshipElements = new LinkedList<>();
NodeList relationshipNodes = relationshipsElement.getElementsByTagName("*");
int nodeCount = relationshipNodes.getLength();
for (int nodeIdx = 0; nodeIdx < nodeCount; nodeIdx++) {
Node relationshipNode = relationshipNodes.item(0);
Element relationshipElement = (Element) relationshipNode;
relationshipElements.add(relationshipElement);
relationshipsElement.removeChild(relationshipNode);
}
Collections.sort(relationshipElements, new RelationshipComparator());
for (Element relationshipElement : relationshipElements) {
relationshipsElement.appendChild(relationshipElement);
}
}
private static String toString(Node dom) throws TransformerException {
Source source = new DOMSource(dom);
StringWriter stringWriter = new StringWriter();
Result result = new StreamResult(stringWriter);
Transformer transformer = TransformerFactory.newInstance().newTransformer();
/*
* We have to omit the ?xml declaration if we want to embed the
* document.
*/
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
transformer.transform(source, result);
return stringWriter.getBuffer().toString();
}
private static OctetStreamData toOctetStreamData(Node node) throws TransformerException {
Source source = new DOMSource(node);
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
Result result = new StreamResult(outputStream);
TransformerFactory transformerFactory = TransformerFactory.newInstance();
Transformer transformer = transformerFactory.newTransformer();
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
transformer.transform(source, result);
return new OctetStreamData(new ByteArrayInputStream(outputStream.toByteArray()));
}
private static Document loadDocument(InputStream documentInputStream)
throws ParserConfigurationException, SAXException, IOException {
InputSource inputSource = new InputSource(documentInputStream);
DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
documentBuilderFactory.setNamespaceAware(true);
return documentBuilderFactory.newDocumentBuilder().parse(inputSource);
}
/** {@inheritDoc} */
@Override
public Data transform(Data data, XMLCryptoContext context, OutputStream os) {
return null;
}
/** {@inheritDoc} */
@Override
public boolean isFeatureSupported(String feature) {
return false;
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy