diff -r 000000000000 -r 373ffda63c9a src/share/jaxws_classes/com/sun/xml/internal/bind/v2/runtime/unmarshaller/DomLoader.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/share/jaxws_classes/com/sun/xml/internal/bind/v2/runtime/unmarshaller/DomLoader.java Wed Apr 27 01:27:09 2016 +0800 @@ -0,0 +1,164 @@ +/* + * Copyright (c) 1997, 2012, Oracle and/or its affiliates. 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. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package com.sun.xml.internal.bind.v2.runtime.unmarshaller; + +import javax.xml.bind.annotation.DomHandler; +import javax.xml.transform.Result; +import javax.xml.transform.sax.TransformerHandler; +import com.sun.xml.internal.bind.v2.runtime.JAXBContextImpl; +import org.xml.sax.SAXException; + +/** + * Loads a DOM. + * + * @author Kohsuke Kawaguchi + */ +public class DomLoader extends Loader { + + private final DomHandler dom; + + /** + * Used to capture the state. + * + * This instance is created for each unmarshalling episode. + */ + private final class State { + + /** This handler will receive SAX events. */ + private TransformerHandler handler = null; + + /** {@link #handler} will produce this result. */ + private final ResultT result; + + // nest level of elements. + int depth = 1; + + public State( UnmarshallingContext context ) throws SAXException { + handler = JAXBContextImpl.createTransformerHandler(context.getJAXBContext().disableSecurityProcessing); + result = dom.createUnmarshaller(context); + + handler.setResult(result); + + // emulate the start of documents + try { + handler.setDocumentLocator(context.getLocator()); + handler.startDocument(); + declarePrefixes( context, context.getAllDeclaredPrefixes() ); + } catch( SAXException e ) { + context.handleError(e); + throw e; + } + } + + public Object getElement() { + return dom.getElement(result); + } + + private void declarePrefixes( UnmarshallingContext context, String[] prefixes ) throws SAXException { + for( int i=prefixes.length-1; i>=0; i-- ) { + String nsUri = context.getNamespaceURI(prefixes[i]); + if(nsUri==null) throw new IllegalStateException("prefix \'"+prefixes[i]+"\' isn't bound"); + handler.startPrefixMapping(prefixes[i],nsUri ); + } + } + + private void undeclarePrefixes( String[] prefixes ) throws SAXException { + for( int i=prefixes.length-1; i>=0; i-- ) + handler.endPrefixMapping( prefixes[i] ); + } + } + + public DomLoader(DomHandler dom) { + super(true); + this.dom = dom; + } + + @Override + public void startElement(UnmarshallingContext.State state, TagName ea) throws SAXException { + UnmarshallingContext context = state.getContext(); + if (state.target == null) + state.target = new State(context); + + State s = (State) state.target; + try { + s.declarePrefixes(context, context.getNewlyDeclaredPrefixes()); + s.handler.startElement(ea.uri, ea.local, ea.getQname(), ea.atts); + } catch (SAXException e) { + context.handleError(e); + throw e; + } + } + + @Override + public void childElement(UnmarshallingContext.State state, TagName ea) throws SAXException { + state.loader = this; + State s = (State) state.prev.target; + s.depth++; + state.target = s; + } + + @Override + public void text(UnmarshallingContext.State state, CharSequence text) throws SAXException { + if(text.length()==0) + return; // there's no point in creating an empty Text node in DOM. + try { + State s = (State) state.target; + s.handler.characters(text.toString().toCharArray(),0,text.length()); + } catch( SAXException e ) { + state.getContext().handleError(e); + throw e; + } + } + + @Override + public void leaveElement(UnmarshallingContext.State state, TagName ea) throws SAXException { + State s = (State) state.target; + UnmarshallingContext context = state.getContext(); + + try { + s.handler.endElement(ea.uri, ea.local, ea.getQname()); + s.undeclarePrefixes(context.getNewlyDeclaredPrefixes()); + } catch( SAXException e ) { + context.handleError(e); + throw e; + } + + if((--s.depth)==0) { + // emulate the end of the document + try { + s.undeclarePrefixes(context.getAllDeclaredPrefixes()); + s.handler.endDocument(); + } catch( SAXException e ) { + context.handleError(e); + throw e; + } + + // we are done + state.target = s.getElement(); + } + } + +}