jdk/src/java.xml.crypto/share/classes/org/jcp/xml/dsig/internal/dom/ApacheNodeSetData.java
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/java.xml.crypto/share/classes/org/jcp/xml/dsig/internal/dom/ApacheNodeSetData.java Sun Aug 17 15:54:13 2014 +0100
@@ -0,0 +1,94 @@
+/*
+ * reserved comment block
+ * DO NOT REMOVE OR ALTER!
+ */
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+/*
+ * Copyright (c) 2005, 2014, Oracle and/or its affiliates. All rights reserved.
+ */
+/*
+ * $Id: ApacheNodeSetData.java 1203890 2011-11-18 22:47:56Z mullan $
+ */
+package org.jcp.xml.dsig.internal.dom;
+
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Set;
+import javax.xml.crypto.NodeSetData;
+import org.w3c.dom.Node;
+import com.sun.org.apache.xml.internal.security.signature.NodeFilter;
+import com.sun.org.apache.xml.internal.security.signature.XMLSignatureInput;
+import com.sun.org.apache.xml.internal.security.utils.XMLUtils;
+
+public class ApacheNodeSetData implements ApacheData, NodeSetData {
+
+ private XMLSignatureInput xi;
+
+ public ApacheNodeSetData(XMLSignatureInput xi) {
+ this.xi = xi;
+ }
+
+ public Iterator<Node> iterator() {
+ // If nodefilters are set, must execute them first to create node-set
+ if (xi.getNodeFilters() != null && !xi.getNodeFilters().isEmpty()) {
+ return Collections.unmodifiableSet
+ (getNodeSet(xi.getNodeFilters())).iterator();
+ }
+ try {
+ return Collections.unmodifiableSet(xi.getNodeSet()).iterator();
+ } catch (Exception e) {
+ // should not occur
+ throw new RuntimeException
+ ("unrecoverable error retrieving nodeset", e);
+ }
+ }
+
+ public XMLSignatureInput getXMLSignatureInput() {
+ return xi;
+ }
+
+ private Set<Node> getNodeSet(List<NodeFilter> nodeFilters) {
+ if (xi.isNeedsToBeExpanded()) {
+ XMLUtils.circumventBug2650
+ (XMLUtils.getOwnerDocument(xi.getSubNode()));
+ }
+
+ Set<Node> inputSet = new LinkedHashSet<Node>();
+ XMLUtils.getSet(xi.getSubNode(), inputSet,
+ null, !xi.isExcludeComments());
+ Set<Node> nodeSet = new LinkedHashSet<Node>();
+ for (Node currentNode : inputSet) {
+ Iterator<NodeFilter> it = nodeFilters.iterator();
+ boolean skipNode = false;
+ while (it.hasNext() && !skipNode) {
+ NodeFilter nf = it.next();
+ if (nf.isNodeInclude(currentNode) != 1) {
+ skipNode = true;
+ }
+ }
+ if (!skipNode) {
+ nodeSet.add(currentNode);
+ }
+ }
+ return nodeSet;
+ }
+}