95 lines · 3.0 KB · Java
RawDownload
/* * reserved comment block * DO NOT REMOVE OR ALTER! *//* * Copyright 2005 The Apache Software Foundation. * *  Licensed 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, 2008, Oracle and/or its affiliates. All rights reserved. *//* * $Id: ApacheNodeSetData.java,v 1.2 2008/07/24 15:20:32 mullan Exp $ */package org.jcp.xml.dsig.internal.dom; import java.util.Collections;import java.util.HashSet;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 iterator() {        // If nodefilters are set, must execute them first to create node-set        if (xi.getNodeFilters() != null) {            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 getNodeSet(List nodeFilters) {        if (xi.isNeedsToBeExpanded()) {            XMLUtils.circumventBug2650                (XMLUtils.getOwnerDocument(xi.getSubNode()));        }         Set inputSet = new LinkedHashSet();        XMLUtils.getSet          (xi.getSubNode(), inputSet, null, !xi.isExcludeComments());        Set nodeSet = new LinkedHashSet();        Iterator i = inputSet.iterator();        while (i.hasNext()) {            Node currentNode = (Node) i.next();            Iterator it = nodeFilters.iterator();            boolean skipNode = false;            while (it.hasNext() && !skipNode) {                NodeFilter nf = (NodeFilter) it.next();                if (nf.isNodeInclude(currentNode)!=1) {                    skipNode = true;                }            }            if (!skipNode) {                nodeSet.add(currentNode);            }        }        return nodeSet;    }}