View Javadoc
1   /**
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements. See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership. The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License. You may obtain a copy of the License at
9    *
10   * http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied. See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.wss4j.stax.impl.resourceResolvers;
20  
21  import org.apache.xml.security.stax.ext.ResourceResolver;
22  import org.apache.xml.security.stax.ext.XMLSecurityConstants;
23  import org.apache.xml.security.stax.ext.stax.XMLSecStartElement;
24  import org.apache.wss4j.stax.ext.WSSConstants;
25  
26  import javax.xml.namespace.QName;
27  import javax.xml.stream.events.Attribute;
28  
29  public class ResolverSameDocument extends org.apache.xml.security.stax.impl.resourceResolvers.ResolverSameDocument {
30  
31      public ResolverSameDocument() {
32          super();
33      }
34  
35      public ResolverSameDocument(String uri) {
36          super(uri);
37      }
38  
39      @Override
40      public ResourceResolver newInstance(String uri, String baseURI) {
41          return new ResolverSameDocument(uri);
42      }
43  
44      @Override
45      public boolean matches(XMLSecStartElement xmlSecStartElement) {
46          return matches(xmlSecStartElement, XMLSecurityConstants.ATT_NULL_Id);
47      }
48  
49      public boolean matches(XMLSecStartElement xmlSecStartElement, QName idAttributeNS) {
50          Attribute attribute = xmlSecStartElement.getAttributeByName(WSSConstants.ATT_WSU_ID);
51          if (attribute != null && attribute.getValue().equals(getId())) {
52              return true;
53          }
54  
55          attribute = xmlSecStartElement.getAttributeByName(WSSConstants.ATT_NULL_ASSERTION_ID);
56          if (attribute != null && attribute.getValue().equals(getId())) {
57              return true;
58          }
59  
60          attribute = xmlSecStartElement.getAttributeByName(WSSConstants.ATT_NULL_ID);
61          if (attribute != null && attribute.getValue().equals(getId())) {
62              return true;
63          }
64          return super.matches(xmlSecStartElement, idAttributeNS);
65      }
66  
67  }