org.apache.xml.security.utils.resolver.ResourceResolverException Java Examples

public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
   String id = context.attr.getNodeValue();
   if (id.startsWith("#")) {
      id = id.substring(1);
   }

   if (LOG.isDebugEnabled()) {
      try {
         LOG.debug("Selected document: " + ConnectorXmlUtils.flatten(ConnectorXmlUtils.toString((Source)(new DOMSource(this.doc)))));
      } catch (TechnicalConnectorException var5) {
         LOG.error(var5.getMessage());
      }
   }

   Node selectedElem = this.doc.getElementById(id);
   if (LOG.isDebugEnabled()) {
      LOG.debug("Try to catch an Element with ID " + id + " and Element was " + selectedElem);
   }

   this.processElement(context.attr, context.baseUri, selectedElem, id);
   XMLSignatureInput result = new XMLSignatureInput(selectedElem);
   result.setExcludeComments(true);
   result.setMIMEType("text/xml");
   result.setSourceURI(context.baseUri != null ? context.baseUri.concat(context.attr.getNodeValue()) : context.attr.getNodeValue());
   return result;
}
 

@Test
public void nullAttributeTwoDocs() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(
				Arrays.<DSSDocument> asList(new InMemoryDocument(new byte[] { 1, 2, 3 }), new InMemoryDocument(new byte[] { 2, 3 })), DigestAlgorithm.SHA256);

		Attr attr = null;

		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertTrue(resolver.engineCanResolveURI(context));

		// 2 docs + no name -> exception
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document (detached signature)", exception.getMessage());
}
 

@Test
public void emptyAttribute() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(Collections.<DSSDocument>emptyList(), DigestAlgorithm.SHA256);

		Attr attr = mock(Attr.class);

		// Empty
		when(attr.getNodeValue()).thenReturn("");
		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertFalse(resolver.engineCanResolveURI(context));

		// will throw ResourceResolverException
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document (detached signature)", exception.getMessage());
}
 

@Test
public void nullListAndNullAttribute() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(null, DigestAlgorithm.SHA256);

		Attr attr = null;

		// Empty
		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertTrue(resolver.engineCanResolveURI(context));

		// will throw ResourceResolverException
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document (detached signature)", exception.getMessage());
}
 

@Test
public void nullAttribute() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(Collections.<DSSDocument>emptyList(), DigestAlgorithm.SHA256);

		Attr attr = null;

		// Empty
		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertTrue(resolver.engineCanResolveURI(context));

		// will throw ResourceResolverException
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document (detached signature)", exception.getMessage());
}
 

public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
   String id = context.attr.getNodeValue();
   if (id.startsWith("#")) {
      id = id.substring(1);
   }

   if (LOG.isDebugEnabled()) {
      try {
         LOG.debug("Selected document: " + ConnectorXmlUtils.flatten(ConnectorXmlUtils.toString((Source)(new DOMSource(this.doc)))));
      } catch (TechnicalConnectorException var5) {
         LOG.error(var5.getMessage());
      }
   }

   Node selectedElem = this.doc.getElementById(id);
   if (LOG.isDebugEnabled()) {
      LOG.debug("Try to catch an Element with ID " + id + " and Element was " + selectedElem);
   }

   this.processElement(context.attr, context.baseUri, selectedElem, id);
   XMLSignatureInput result = new XMLSignatureInput(selectedElem);
   result.setExcludeComments(true);
   result.setMIMEType("text/xml");
   result.setSourceURI(context.baseUri != null ? context.baseUri.concat(context.attr.getNodeValue()) : context.attr.getNodeValue());
   return result;
}
 

@Test
public void documentNameWithEmptyList() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(Collections.<DSSDocument>emptyList(), DigestAlgorithm.SHA256);

		Attr attr = mock(Attr.class);

		// document name + no document in the list
		when(attr.getNodeValue()).thenReturn("sample.xml");
		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertTrue(resolver.engineCanResolveURI(context));

		// will throw ResourceResolverException
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document (detached signature)", exception.getMessage());
}
 

@Test
public void engineCanResolveURIWithWrongDocumentNameInList() throws ResourceResolverException {
	Exception exception = assertThrows(ResourceResolverException.class, () -> {
		DetachedSignatureResolver resolver = new DetachedSignatureResolver(
				Arrays.<DSSDocument>asList(new InMemoryDocument(new byte[] { 1, 2, 3 }, "toto.xml", MimeType.XML)),
				DigestAlgorithm.SHA256);

		Attr attr = mock(Attr.class);

		// document name + wrong document in the list
		when(attr.getNodeValue()).thenReturn("sample.xml");
		ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
		assertTrue(resolver.engineCanResolveURI(context));

		// doc not found -> exception
		resolver.engineResolveURI(context);
	});
	assertEquals("Unable to find document 'sample.xml' (detached signature)", exception.getMessage());
}
 

public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
   String id = context.attr.getNodeValue();
   if (id.startsWith("#")) {
      id = id.substring(1);
   }

   if (LOG.isDebugEnabled()) {
      try {
         LOG.debug("Selected document: " + ConnectorXmlUtils.flatten(ConnectorXmlUtils.toString((Source)(new DOMSource(this.doc)))));
      } catch (TechnicalConnectorException var5) {
         LOG.error(var5.getMessage());
      }
   }

   Node selectedElem = this.doc.getElementById(id);
   if (LOG.isDebugEnabled()) {
      LOG.debug("Try to catch an Element with ID " + id + " and Element was " + selectedElem);
   }

   this.processElement(context.attr, context.baseUri, selectedElem, id);
   XMLSignatureInput result = new XMLSignatureInput(selectedElem);
   result.setExcludeComments(true);
   result.setMIMEType("text/xml");
   result.setSourceURI(context.baseUri != null ? context.baseUri.concat(context.attr.getNodeValue()) : context.attr.getNodeValue());
   return result;
}
 

public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
   String id = context.attr.getNodeValue();
   if (id.startsWith("#")) {
      id = id.substring(1);
   }

   if (LOG.isDebugEnabled()) {
      try {
         LOG.debug("Selected document: " + ConnectorXmlUtils.flatten(ConnectorXmlUtils.toString((Source)(new DOMSource(this.doc)))));
      } catch (TechnicalConnectorException var5) {
         LOG.error(var5.getMessage());
      }
   }

   Node selectedElem = this.doc.getElementById(id);
   if (LOG.isDebugEnabled()) {
      LOG.debug("Try to catch an Element with ID " + id + " and Element was " + selectedElem);
   }

   this.processElement(context.attr, context.baseUri, selectedElem, id);
   XMLSignatureInput result = new XMLSignatureInput(selectedElem);
   result.setExcludeComments(true);
   result.setMIMEType("text/xml");
   result.setSourceURI(context.baseUri != null ? context.baseUri.concat(context.attr.getNodeValue()) : context.attr.getNodeValue());
   return result;
}
 

public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
   String id = context.attr.getNodeValue();
   if (id.startsWith("#")) {
      id = id.substring(1);
   }

   if (LOG.isDebugEnabled()) {
      try {
         LOG.debug("Selected document: " + ConnectorXmlUtils.flatten(ConnectorXmlUtils.toString((Source)(new DOMSource(this.doc)))));
      } catch (TechnicalConnectorException var5) {
         LOG.error(var5.getMessage());
      }
   }

   Node selectedElem = this.doc.getElementById(id);
   if (LOG.isDebugEnabled()) {
      LOG.debug("Try to catch an Element with ID " + id + " and Element was " + selectedElem);
   }

   this.processElement(context.attr, context.baseUri, selectedElem, id);
   XMLSignatureInput result = new XMLSignatureInput(selectedElem);
   result.setExcludeComments(true);
   result.setMIMEType("text/xml");
   result.setSourceURI(context.baseUri != null ? context.baseUri.concat(context.attr.getNodeValue()) : context.attr.getNodeValue());
   return result;
}
 

@Override
public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException {
	DSSDocument document = getCurrentDocument(context);
	if (document instanceof DigestDocument) {
		DigestDocument digestDoc = (DigestDocument) document;
		return new XMLSignatureInput(digestDoc.getDigest(digestAlgorithm));
	} else {
		return createFromCommonDocument(document);
	}
}
 

@Test
public void engineCanResolveURIWithDocumentNoNameInList() throws ResourceResolverException {
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(Arrays.<DSSDocument>asList(new InMemoryDocument(new byte[] { 1, 2, 3 })),
			DigestAlgorithm.SHA256);

	Attr attr = mock(Attr.class);

	// document name + only one document
	when(attr.getNodeValue()).thenReturn("sample.xml");
	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Test
public void engineCanResolveURIWithDocumentNameInList() throws ResourceResolverException {
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(
			Arrays.<DSSDocument>asList(new InMemoryDocument(new byte[] { 1, 2, 3 }, "sample.xml", MimeType.XML)),
			DigestAlgorithm.SHA256);

	Attr attr = mock(Attr.class);

	when(attr.getNodeValue()).thenReturn("sample.xml");
	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Test
public void engineCanResolveURIWithDocumentNameInListOfMultiples() throws ResourceResolverException {
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(
			Arrays.<DSSDocument>asList(new InMemoryDocument(new byte[] { 1, 2, 3 }, "sample.xml", MimeType.XML),
			new InMemoryDocument(new byte[] { 2, 3 }, "sample2.xml", MimeType.XML)), DigestAlgorithm.SHA256);

	Attr attr = mock(Attr.class);

	when(attr.getNodeValue()).thenReturn("sample.xml");
	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Test
public void engineCanResolveURIWithDigestDocument() throws ResourceResolverException {
	DigestDocument doc = new DigestDocument(DigestAlgorithm.SHA256, "abcdef");
	doc.setName("sample.xml");
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(Arrays.<DSSDocument>asList(doc), DigestAlgorithm.SHA256);

	Attr attr = mock(Attr.class);

	when(attr.getNodeValue()).thenReturn("sample.xml");
	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Test
public void engineCanResolveURIWithDigestDocumentNoName() throws ResourceResolverException {
	DigestDocument doc = new DigestDocument(DigestAlgorithm.SHA256, "abcdef");
	// doc.setName("sample.xml");
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(Arrays.<DSSDocument>asList(doc), DigestAlgorithm.SHA256);

	Attr attr = mock(Attr.class);

	when(attr.getNodeValue()).thenReturn("sample.xml");
	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Test
public void nullAttributeOneDoc() throws ResourceResolverException {
	DetachedSignatureResolver resolver = new DetachedSignatureResolver(Arrays.<DSSDocument>asList(new InMemoryDocument(new byte[] { 1, 2, 3 })),
			DigestAlgorithm.SHA256);

	Attr attr = null;

	ResourceResolverContext context = new ResourceResolverContext(attr, null, false);
	assertTrue(resolver.engineCanResolveURI(context));

	assertNotNull(resolver.engineResolveURI(context));
}
 

@Override
public XMLSignatureInput engineResolveURI(ResourceResolverContext context) throws ResourceResolverException
{
    XMLSignatureInput input = new XMLSignatureInput(context.attr.getValue().getBytes());
    resolveCount++;
    return input;
}
 

private <TObj> TObj getReferencedElement(
        String objReference,
        Map<Element, TObj> mapper)
{
    if (!objReference.startsWith("#"))
        throw new IllegalArgumentException();

    if (mapper.isEmpty())
        return null;

    // A little workaround to be able to use the ResourceResolver.
    Attr refAttr = this.signatureDoc.createAttributeNS(null, "ref");
    refAttr.setNodeValue(objReference);
    this.signatureDoc.createElementNS(null, "dummy").setAttributeNodeNS(refAttr);

    try
    {
        XMLSignatureInput refData = ResourceResolver.getInstance(refAttr, "", true).resolve(refAttr, "", true);
        // This has to be a NodeSet data because it is a same-document reference.
        Node refNode = refData.getSubNode();
        if (refNode.getNodeType() != Node.ELEMENT_NODE)
            return null;
        // May return null.
        return mapper.get((Element)refNode);
    } catch (ResourceResolverException ex)
    {
        // Maybe an exception should be thrown...
        return null;
    }
}
 

private void processElement(Attr uri, String baseURI, Node selectedElem, String id) throws ResourceResolverException {
   if (selectedElem == null) {
      Object[] exArgs = new Object[]{id};
      throw new ResourceResolverException("signature.Verification.MissingID", exArgs, uri.getBaseURI(), baseURI);
   }
}
 

private void processElement(Attr uri, String baseURI, Node selectedElem, String id) throws ResourceResolverException {
   if (selectedElem == null) {
      Object[] exArgs = new Object[]{id};
      throw new ResourceResolverException("signature.Verification.MissingID", exArgs, uri.getBaseURI(), baseURI);
   }
}
 

private void processElement(Attr uri, String baseURI, Node selectedElem, String id) throws ResourceResolverException {
   if (selectedElem == null) {
      Object[] exArgs = new Object[]{id};
      throw new ResourceResolverException("signature.Verification.MissingID", exArgs, uri.getBaseURI(), baseURI);
   }
}
 

private void processElement(Attr uri, String baseURI, Node selectedElem, String id) throws ResourceResolverException {
   if (selectedElem == null) {
      Object[] exArgs = new Object[]{id};
      throw new ResourceResolverException("signature.Verification.MissingID", exArgs, uri.getBaseURI(), baseURI);
   }
}
 

private void processElement(Attr uri, String baseURI, Node selectedElem, String id) throws ResourceResolverException {
   if (selectedElem == null) {
      Object[] exArgs = new Object[]{id};
      throw new ResourceResolverException("signature.Verification.MissingID", exArgs, uri.getBaseURI(), baseURI);
   }
}