Mercurial > jdk8-mips64-public > jaxws / file revision

 /*

  * Copyright 2005-2006 Sun Microsystems, Inc.  All Rights Reserved.

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.  Sun designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Sun in the LICENSE file that accompanied this code.

  *

  * This code is distributed in the hope that it will be useful, but WITHOUT

  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

  * You should have received a copy of the GNU General Public License version

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

  * CA 95054 USA or visit www.sun.com if you need additional information or

  * have any questions.

  */

 package com.sun.xml.internal.messaging.saaj.util;

 import java.util.logging.Logger;

 import javax.xml.parsers.SAXParser;

 import javax.xml.soap.SOAPException;

 import org.xml.sax.*;

 import org.xml.sax.ext.LexicalHandler;

 import org.xml.sax.helpers.XMLFilterImpl;

 import com.sun.xml.internal.messaging.saaj.SOAPExceptionImpl;

 import org.xml.sax.helpers.AttributesImpl;

 /**

  * Users of this class see a SAX2 XMLReader (via XMLFilterImpl).  This

  * class creates a parent XMLReader via JAXP and installs itself as a SAX2

  * extension LexicalHandler which rejects document type declarations

  * because they are not legal in SOAP.  If the user of this class sets a

  * LexicalHandler, then it forwards events to that handler.

  *

  *

  * @author Edwin Goei

  */

 public class RejectDoctypeSaxFilter extends XMLFilterImpl implements XMLReader, LexicalHandler{

     protected static final Logger log =

     Logger.getLogger(LogDomainConstants.UTIL_DOMAIN,

     "com.sun.xml.internal.messaging.saaj.util.LocalStrings");

     /** Standard SAX 2.0 ext property */

     static final String LEXICAL_HANDLER_PROP =

     "http://xml.org/sax/properties/lexical-handler";

     static final String WSU_NS = "http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-utility-1.0.xsd".intern();

     static final String SIGNATURE_LNAME = "Signature".intern();

     static final String ENCRYPTED_DATA_LNAME = "EncryptedData".intern();

     static final String DSIG_NS = "http://www.w3.org/2000/09/xmldsig#".intern();

     static final String XENC_NS = "http://www.w3.org/2001/04/xmlenc#".intern();

     static final String ID_NAME = "ID".intern();

     /** LexicalHandler to forward events to, if any */

     private LexicalHandler lexicalHandler;

     public RejectDoctypeSaxFilter(SAXParser saxParser) throws SOAPException {

         XMLReader xmlReader;

         try {

             xmlReader = saxParser.getXMLReader();

         } catch (Exception e) {

             log.severe("SAAJ0602.util.getXMLReader.exception");

             throw new SOAPExceptionImpl(

             "Couldn't get an XMLReader while constructing a RejectDoctypeSaxFilter",

             e);

         }

         // Set ourselves up to be the SAX LexicalHandler

         try {

             xmlReader.setProperty(LEXICAL_HANDLER_PROP, this);

         } catch (Exception e) {

             log.severe("SAAJ0603.util.setProperty.exception");

             throw new SOAPExceptionImpl(

             "Couldn't set the lexical handler property while constructing a RejectDoctypeSaxFilter",

             e);

         }

         // Set the parent XMLReader of this SAX filter

         setParent(xmlReader);

     }

     /*

      * Override setProperty() to capture any LexicalHandler that is set for

      * forwarding of events.

      */

     public void setProperty(String name, Object value)

     throws SAXNotRecognizedException, SAXNotSupportedException {

         if (LEXICAL_HANDLER_PROP.equals(name)) {

             lexicalHandler = (LexicalHandler) value;

         } else {

             super.setProperty(name, value);

         }

     }

     //

     // Beginning of SAX LexicalHandler callbacks...

     //

     public void startDTD(String name, String publicId, String systemId)

     throws SAXException {

         throw new SAXException("Document Type Declaration is not allowed");

     }

     public void endDTD() throws SAXException {

     }

     public void startEntity(String name) throws SAXException {

         if (lexicalHandler != null) {

             lexicalHandler.startEntity(name);

         }

     }

     public void endEntity(String name) throws SAXException {

         if (lexicalHandler != null) {

             lexicalHandler.endEntity(name);

         }

     }

     public void startCDATA() throws SAXException {

         if (lexicalHandler != null) {

             lexicalHandler.startCDATA();

         }

     }

     public void endCDATA() throws SAXException {

         if (lexicalHandler != null) {

             lexicalHandler.endCDATA();

         }

     }

     public void comment(char[] ch, int start, int length) throws SAXException {

         if (lexicalHandler != null) {

             lexicalHandler.comment(ch, start, length);

         }

     }

     //

     // End of SAX LexicalHandler callbacks

     //

     public void startElement(String namespaceURI, String localName,

     String qName, Attributes atts)   throws SAXException{

         if(atts != null ){

             boolean eos = false;

             if(namespaceURI == DSIG_NS || XENC_NS == namespaceURI){

                 eos = true;

             }

             int length = atts.getLength();

             AttributesImpl attrImpl = new AttributesImpl();

             for(int i=0; i< length;i++){

                 String name = atts.getLocalName(i);

                 if(name!=null && (name.equals("Id"))){

                     if(eos || atts.getURI(i) == WSU_NS ){

                         attrImpl.addAttribute(atts.getURI(i), atts.getLocalName(i),

                         atts.getQName(i), ID_NAME, atts.getValue(i));

                     }else{

                          attrImpl.addAttribute(atts.getURI(i), atts.getLocalName(i), atts.getQName(i), atts.getType(i), atts.getValue(i));

                     }

                 }else{

                     attrImpl.addAttribute(atts.getURI(i), atts.getLocalName(i),

                     atts.getQName(i), atts.getType(i), atts.getValue(i));

                 }

             }

             super.startElement(namespaceURI,localName, qName,attrImpl);

         }else{

             super.startElement(namespaceURI,localName, qName, atts);

         }

     }

 }