jdk/src/java.xml.crypto/share/classes/org/jcp/xml/dsig/internal/dom/ApacheNodeSetData.java
changeset 25859 3317bb8137f4
parent 24970 094bfaa699c3
child 27747 3a271dc8b758
--- /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;
+    }
+}