/* * reserved comment block * DO NOT REMOVE OR ALTER! */ /* * Copyright 1999-2004 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. * */ package com.sun.org.apache.xml.internal.security.utils; import java.math.BigInteger; import java.util.HashMap; import com.sun.org.apache.xml.internal.security.exceptions.Base64DecodingException; import com.sun.org.apache.xml.internal.security.exceptions.XMLSecurityException; import org.w3c.dom.Attr; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.Node; import org.w3c.dom.NodeList; import org.w3c.dom.Text; /** * This is the base class to all Objects which have a direct 1:1 mapping to an * Element in a particular namespace. * * @author $Author: mullan $ */ public abstract class ElementProxy { /** {@link java.util.logging} logging facility */ static java.util.logging.Logger log = java.util.logging.Logger.getLogger(ElementProxy.class.getName()); //J- /** The element has been created by the code **/ public static final int MODE_CREATE = 0; /** The element has been readed from a DOM tree by the code **/ public static final int MODE_PROCESS = 1; /** The element isn't known if it is readen or created **/ public static final int MODE_UNKNOWN = 2; /** The element is going to be signed **/ public static final int MODE_SIGN = MODE_CREATE; /** The element is going to be verified **/ public static final int MODE_VERIFY = MODE_PROCESS; /** The element is going to be encrypted **/ public static final int MODE_ENCRYPT = MODE_CREATE; /** The element is going to be decrypted **/ public static final int MODE_DECRYPT = MODE_PROCESS; protected int _state = MODE_UNKNOWN; //J+ /** * Returns the namespace of the Elements of the sub-class. * * @return the namespace of the Elements of the sub-class. */ public abstract String getBaseNamespace(); /** * Returns the localname of the Elements of the sub-class. * * @return the localname of the Elements of the sub-class. */ public abstract String getBaseLocalName(); /** Field _constructionElement */ protected Element _constructionElement = null; /** Field _baseURI */ protected String _baseURI = null; /** Field _doc */ protected Document _doc = null; /** * Constructor ElementProxy * */ public ElementProxy() { this._doc = null; this._state = ElementProxy.MODE_UNKNOWN; this._baseURI = null; this._constructionElement = null; } /** * Constructor ElementProxy * * @param doc */ public ElementProxy(Document doc) { this(); if (doc == null) { throw new RuntimeException("Document is null"); } this._doc = doc; this._state = ElementProxy.MODE_CREATE; this._constructionElement = ElementProxy.createElementForFamily(this._doc, this.getBaseNamespace(), this.getBaseLocalName()); } /** * This method creates an Element in a given namespace with a given localname. * It uses the {@link ElementProxy#getDefaultPrefix} method to decide whether * a particular prefix is bound to that namespace. * <BR /> * This method was refactored out of the constructor. * * @param doc * @param namespace * @param localName * @return The element created. */ public static Element createElementForFamily(Document doc, String namespace, String localName) { //Element nscontext = XMLUtils.createDSctx(doc, "x", namespace); Element result = null; String prefix = ElementProxy.getDefaultPrefix(namespace); if (namespace == null) { result = doc.createElementNS(null, localName); } else { if ((prefix == null) || (prefix.length() == 0)) { result = doc.createElementNS(namespace, localName); result.setAttributeNS(Constants.NamespaceSpecNS, "xmlns", namespace); } else { result = doc.createElementNS(namespace, prefix + ":" + localName); result.setAttributeNS(Constants.NamespaceSpecNS, "xmlns:" + prefix, namespace); } } return result; } /** * Method setElement * * @param element * @param BaseURI * @throws XMLSecurityException */ public void setElement(Element element, String BaseURI) throws XMLSecurityException { if (element == null) { throw new XMLSecurityException("ElementProxy.nullElement"); } if (true) { } if (true) { if (log.isLoggable(java.util.logging.Level.FINE)) log.log(java.util.logging.Level.FINE, "setElement(" + element.getTagName() + ", \"" + BaseURI + "\""); } this._doc = element.getOwnerDocument(); this._state = ElementProxy.MODE_PROCESS; this._constructionElement = element; this._baseURI = BaseURI; } /** * Constructor ElementProxy * * @param element * @param BaseURI * @throws XMLSecurityException */ public ElementProxy(Element element, String BaseURI) throws XMLSecurityException { this(); if (element == null) { throw new XMLSecurityException("ElementProxy.nullElement"); } if (true) { if (log.isLoggable(java.util.logging.Level.FINE)) log.log(java.util.logging.Level.FINE, "setElement(\"" + element.getTagName() + "\", \"" + BaseURI + "\")"); } this._doc = element.getOwnerDocument(); this._state = ElementProxy.MODE_PROCESS; this._constructionElement = element; this._baseURI = BaseURI; this.guaranteeThatElementInCorrectSpace(); } /** * Returns the Element which was constructed by the Object. * * @return the Element which was constructed by the Object. */ public final Element getElement() { return this._constructionElement; } /** * Returns the Element plus a leading and a trailing CarriageReturn Text node. * * @return the Element which was constructed by the Object. */ public final NodeList getElementPlusReturns() { HelperNodeList nl = new HelperNodeList(); nl.appendChild(this._doc.createTextNode("\n")); nl.appendChild(this.getElement()); nl.appendChild(this._doc.createTextNode("\n")); return nl; } /** * Method getDocument * * @return the Document where this element is contained. */ public Document getDocument() { return this._doc; } /** * Method getBaseURI * * @return the base uri of the namespace of this element */ public String getBaseURI() { return this._baseURI; } /** * Method guaranteeThatElementInCorrectSpace * * @throws XMLSecurityException */ public void guaranteeThatElementInCorrectSpace() throws XMLSecurityException { String localnameSHOULDBE = this.getBaseLocalName(); String namespaceSHOULDBE = this.getBaseNamespace(); String localnameIS = this._constructionElement.getLocalName(); String namespaceIS = this._constructionElement.getNamespaceURI(); if ( !localnameSHOULDBE.equals(localnameIS) || !namespaceSHOULDBE.equals(namespaceIS)) { Object exArgs[] = { namespaceIS +":"+ localnameIS, namespaceSHOULDBE +":"+ localnameSHOULDBE}; throw new XMLSecurityException("xml.WrongElement", exArgs); } } /** * Method setVal * * @param bi * @param localname */ public void addBigIntegerElement(BigInteger bi, String localname) { if (bi != null) { Element e = XMLUtils.createElementInSignatureSpace(this._doc, localname); Base64.fillElementWithBigInteger(e, bi); this._constructionElement.appendChild(e); XMLUtils.addReturnToElement(this._constructionElement); } } /** * Method addBase64Element * * @param bytes * @param localname */ public void addBase64Element(byte[] bytes, String localname) { if (bytes != null) { Element e = Base64.encodeToElement(this._doc, localname, bytes); this._constructionElement.appendChild(e); this._constructionElement.appendChild(this._doc.createTextNode("\n")); } } /** * Method addTextElement * * @param text * @param localname */ public void addTextElement(String text, String localname) { Element e = XMLUtils.createElementInSignatureSpace(this._doc, localname); Text t = this._doc.createTextNode(text); e.appendChild(t); this._constructionElement.appendChild(e); XMLUtils.addReturnToElement(this._constructionElement); } /** * Method addBase64Text * * @param bytes */ public void addBase64Text(byte[] bytes) { if (bytes != null) { Text t = this._doc.createTextNode("\n" + Base64.encode(bytes) + "\n"); this._constructionElement.appendChild(t); } } /** * Method addText * * @param text */ public void addText(String text) { if (text != null) { Text t = this._doc.createTextNode(text); this._constructionElement.appendChild(t); } } /** * Method getVal * * @param localname * @param namespace * @return The biginter contained in the given element * @throws Base64DecodingException */ public BigInteger getBigIntegerFromChildElement( String localname, String namespace) throws Base64DecodingException { return Base64.decodeBigIntegerFromText( XMLUtils.selectNodeText(this._constructionElement.getFirstChild(), namespace,localname,0)); } /** * Method getBytesFromChildElement * * @param localname * @param namespace * @return the bytes * @throws XMLSecurityException */ public byte[] getBytesFromChildElement(String localname, String namespace) throws XMLSecurityException { Element e = XMLUtils.selectNode( this._constructionElement.getFirstChild(), namespace, localname, 0); return Base64.decode(e); } /** * Method getTextFromChildElement * * @param localname * @param namespace * @return the Text of the textNode */ public String getTextFromChildElement(String localname, String namespace) { Text t = (Text) XMLUtils.selectNode( this._constructionElement.getFirstChild(), namespace, localname, 0).getFirstChild(); return t.getData(); } /** * Method getBytesFromTextChild * * @return The base64 bytes from the first text child of this element * @throws XMLSecurityException */ public byte[] getBytesFromTextChild() throws XMLSecurityException { Text t = (Text)this._constructionElement.getFirstChild(); return Base64.decode(t.getData()); } /** * Method getTextFromTextChild * * @return the Text obtained concatening all the the text nodes of this element */ public String getTextFromTextChild() { return XMLUtils.getFullTextChildrenFromElement(this._constructionElement); } /** * Method length * * @param namespace * @param localname * @return the number of elements {namespace}:localname under this element */ public int length(String namespace, String localname) { int number=0; Node sibling=this._constructionElement.getFirstChild(); while (sibling!=null) { if (localname.equals(sibling.getLocalName()) && namespace.equals(sibling.getNamespaceURI())) { number++; } sibling=sibling.getNextSibling(); } return number; } /** * Adds an xmlns: definition to the Element. This can be called as follows: * * <PRE> * // set namespace with ds prefix * xpathContainer.setXPathNamespaceContext("ds", "http://www.w3.org/2000/09/xmldsig#"); * xpathContainer.setXPathNamespaceContext("xmlns:ds", "http://www.w3.org/2000/09/xmldsig#"); * </PRE> * * @param prefix * @param uri * @throws XMLSecurityException */ public void setXPathNamespaceContext(String prefix, String uri) throws XMLSecurityException { String ns; if ((prefix == null) || (prefix.length() == 0)) { throw new XMLSecurityException("defaultNamespaceCannotBeSetHere"); } else if (prefix.equals("xmlns")) { throw new XMLSecurityException("defaultNamespaceCannotBeSetHere"); } else if (prefix.startsWith("xmlns:")) { ns = prefix;//"xmlns:" + prefix.substring("xmlns:".length()); } else { ns = "xmlns:" + prefix; } Attr a = this._constructionElement.getAttributeNodeNS(Constants.NamespaceSpecNS, ns); if (a != null) { if (!a.getNodeValue().equals(uri)) { Object exArgs[] = { ns, this._constructionElement.getAttributeNS(null, ns) }; throw new XMLSecurityException("namespacePrefixAlreadyUsedByOtherURI", exArgs); } return; } this._constructionElement.setAttributeNS(Constants.NamespaceSpecNS, ns, uri); } /** Field _prefixMappings */ static HashMap _prefixMappings = new HashMap(); /** * Method setDefaultPrefix * * @param namespace * @param prefix * @throws XMLSecurityException */ public static void setDefaultPrefix(String namespace, String prefix) throws XMLSecurityException { if (ElementProxy._prefixMappings.containsValue(prefix)) { Object storedNamespace=ElementProxy._prefixMappings.get(namespace); if (!storedNamespace.equals(prefix)) { Object exArgs[] = { prefix, namespace, storedNamespace }; throw new XMLSecurityException("prefix.AlreadyAssigned", exArgs); } } ElementProxy._prefixMappings.put(namespace, prefix); } /** * Method getDefaultPrefix * * @param namespace * @return the default prefix bind to this element. */ public static String getDefaultPrefix(String namespace) { String prefix = (String) ElementProxy._prefixMappings.get(namespace); return prefix; } }