1 /*
2  * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */
25 
26 package com.sun.xml.internal.ws.api.addressing;
27 
28 import java.io.ByteArrayInputStream;
29 import java.io.ByteArrayOutputStream;
30 import java.io.Writer;
31 
32 import com.sun.istack.internal.NotNull;
33 import com.sun.istack.internal.Nullable;
34 import com.sun.xml.internal.ws.message.AbstractHeaderImpl;
35 import com.sun.xml.internal.ws.util.xml.XmlUtil;
36 
37 import org.w3c.dom.Document;
38 import org.w3c.dom.Node;
39 import org.xml.sax.ContentHandler;
40 import org.xml.sax.ErrorHandler;
41 import org.xml.sax.SAXException;
42 
43 import javax.xml.namespace.QName;
44 import javax.xml.parsers.DocumentBuilderFactory;
45 import javax.xml.soap.SOAPException;
46 import javax.xml.soap.SOAPMessage;
47 import javax.xml.soap.SOAPHeader;
48 import javax.xml.stream.XMLOutputFactory;
49 import javax.xml.stream.XMLStreamConstants;
50 import javax.xml.stream.XMLStreamException;
51 import javax.xml.stream.XMLStreamReader;
52 import javax.xml.stream.XMLStreamWriter;
53 import javax.xml.transform.Transformer;
54 import javax.xml.transform.dom.DOMResult;
55 
56 /**
57  * Used to represent outbound endpoint reference header,
58  * such as <ReplyTo> and <FaultTo>.
59  * Used from {@link WSEndpointReference}.
60  *
61  * @author Kohsuke Kawaguchi
62  * @see WSEndpointReference
63  */
64 final class EPRHeader extends AbstractHeaderImpl {
65 
66     private final String nsUri,localName;
67     private final WSEndpointReference epr;
68 
EPRHeader(QName tagName, WSEndpointReference epr)69     EPRHeader(QName tagName, WSEndpointReference epr) {
70         this.nsUri = tagName.getNamespaceURI();
71         this.localName = tagName.getLocalPart();
72         this.epr = epr;
73     }
74 
getNamespaceURI()75     public @NotNull String getNamespaceURI() {
76         return nsUri;
77     }
78 
getLocalPart()79     public @NotNull String getLocalPart() {
80         return localName;
81     }
82 
83     @Nullable
getAttribute(@otNull String nsUri, @NotNull String localName)84     public String getAttribute(@NotNull String nsUri, @NotNull String localName) {
85         try {
86             XMLStreamReader sr = epr.read("EndpointReference"/*doesn't matter*/);
87             while(sr.getEventType()!= XMLStreamConstants.START_ELEMENT)
88                 sr.next();
89 
90             return sr.getAttributeValue(nsUri,localName);
91         } catch (XMLStreamException e) {
92             // since we are reading from buffer, this can't happen.
93             throw new AssertionError(e);
94         }
95     }
96 
readHeader()97     public XMLStreamReader readHeader() throws XMLStreamException {
98         return epr.read(localName);
99     }
100 
writeTo(XMLStreamWriter w)101     public void writeTo(XMLStreamWriter w) throws XMLStreamException {
102         epr.writeTo(localName, w);
103     }
104 
writeTo(SOAPMessage saaj)105     public void writeTo(SOAPMessage saaj) throws SOAPException {
106         try {
107             // TODO what about in-scope namespaces
108             // Not very efficient consider implementing a stream buffer
109             // processor that produces a DOM node from the buffer.
110             Transformer t = XmlUtil.newTransformer();
111             SOAPHeader header = saaj.getSOAPHeader();
112             if (header == null)
113                 header = saaj.getSOAPPart().getEnvelope().addHeader();
114 // TODO workaround for oracle xdk bug 16555545, when this bug is fixed the line below can be
115 // uncommented and all lines below, except the catch block, can be removed.
116 //            t.transform(epr.asSource(localName), new DOMResult(header));
117             ByteArrayOutputStream baos = new ByteArrayOutputStream();
118             XMLStreamWriter w = XMLOutputFactory.newFactory().createXMLStreamWriter(baos);
119             epr.writeTo(localName, w);
120             w.flush();
121             ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
122             DocumentBuilderFactory fac = DocumentBuilderFactory.newInstance();
123             fac.setNamespaceAware(true);
124             Node eprNode = fac.newDocumentBuilder().parse(bais).getDocumentElement();
125             Node eprNodeToAdd = header.getOwnerDocument().importNode(eprNode, true);
126             header.appendChild(eprNodeToAdd);
127         } catch (Exception e) {
128             throw new SOAPException(e);
129         }
130     }
131 
writeTo(ContentHandler contentHandler, ErrorHandler errorHandler)132     public void writeTo(ContentHandler contentHandler, ErrorHandler errorHandler) throws SAXException {
133         epr.writeTo(localName,contentHandler,errorHandler,true);
134     }
135 }
136