1 /*
2 * Copyright (c) 1997, 2014, 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.messaging.saaj.soap;
27
28 import com.sun.xml.internal.messaging.saaj.LazyEnvelopeSource;
29 import com.sun.xml.internal.messaging.saaj.SOAPExceptionImpl;
30 import com.sun.xml.internal.messaging.saaj.util.JAXMStreamSource;
31 import com.sun.xml.internal.messaging.saaj.util.LogDomainConstants;
32 import com.sun.xml.internal.messaging.saaj.util.ParserPool;
33 import com.sun.xml.internal.messaging.saaj.util.RejectDoctypeSaxFilter;
34 import com.sun.xml.internal.messaging.saaj.util.transform.EfficientStreamingTransformer;
35 import org.xml.sax.InputSource;
36 import org.xml.sax.XMLReader;
37
38 import javax.xml.parsers.SAXParser;
39 import javax.xml.soap.SOAPException;
40 import javax.xml.stream.XMLInputFactory;
41 import javax.xml.stream.XMLStreamException;
42 import javax.xml.stream.XMLStreamReader;
43 import javax.xml.transform.Source;
44 import javax.xml.transform.Transformer;
45 import javax.xml.transform.dom.DOMResult;
46 import javax.xml.transform.sax.SAXSource;
47 import javax.xml.transform.stax.StAXSource;
48 import javax.xml.transform.stream.StreamSource;
49 import java.util.logging.Logger;
50
51 /**
52 * EnvelopeFactory creates SOAP Envelope objects using different
53 * underlying implementations.
54 */
55 public class EnvelopeFactory {
56
57 protected static final Logger
58 log = Logger.getLogger(LogDomainConstants.SOAP_DOMAIN,
59 "com.sun.xml.internal.messaging.saaj.soap.LocalStrings");
60
61 private static ContextClassloaderLocal<ParserPool> parserPool =
62 new ContextClassloaderLocal<ParserPool>() {
63 @Override
64 protected ParserPool initialValue() throws Exception {
65 return new ParserPool(5);
66 }
67 };
68
69 public static Envelope createEnvelope(Source src, SOAPPartImpl soapPart)
70 throws SOAPException
71 {
72 if (src instanceof JAXMStreamSource) {
73 try {
74 if (!SOAPPartImpl.lazyContentLength) {
75 ((JAXMStreamSource) src).reset();
76 }
77 } catch (java.io.IOException ioe) {
78 log.severe("SAAJ0515.source.reset.exception");
79 throw new SOAPExceptionImpl(ioe);
80 }
81 }
82 if (src instanceof LazyEnvelopeSource) {
83 return lazy((LazyEnvelopeSource)src, soapPart);
84 }
85 if (soapPart.message.isLazySoapBodyParsing()) {
86 return parseEnvelopeStax(src, soapPart);
87 } else {
88 return parseEnvelopeSax(src, soapPart);
89 }
90 }
91
92 private static Envelope lazy(LazyEnvelopeSource src, SOAPPartImpl soapPart) throws SOAPException {
93 try {
94 StaxBridge staxBridge = new StaxLazySourceBridge(src, soapPart);
95 staxBridge.bridgeEnvelopeAndHeaders();
96 Envelope env = (Envelope) soapPart.getEnvelope();
97 env.setStaxBridge(staxBridge);
98 return env;
99 } catch (XMLStreamException e) {
100 throw new SOAPException(e);
101 }
102 }
103
104 static private XMLInputFactory xmlInputFactory = null;
105
106 private static Envelope parseEnvelopeStax(Source src, SOAPPartImpl soapPart)
107 throws SOAPException {
108 XMLStreamReader streamReader = null;
109 if (src instanceof StAXSource) {
110 streamReader = ((StAXSource) src).getXMLStreamReader();
111 }
112 try {
113 if (streamReader == null) {
114 if (xmlInputFactory == null) xmlInputFactory = XMLInputFactory.newInstance();
115 streamReader = xmlInputFactory.createXMLStreamReader(src);
116 }
117 // SaajStaxWriter saajWriter = new SaajStaxWriter(soapPart.message, soapPart.document);
118 // XMLStreamReaderToXMLStreamWriter readerWriterBridge = new XMLStreamReaderToXMLStreamWriter(
119 // streamReader, saajWriter, soapPart.getSOAPNamespace());
120
121 StaxBridge readerWriterBridge = new StaxReaderBridge(streamReader, soapPart);
122 //bridge will stop reading at body element, and parse upon request, so save it
123 //on the envelope
124 readerWriterBridge.bridgeEnvelopeAndHeaders();
125
126 Envelope env = (Envelope) soapPart.getEnvelope();
127 env.setStaxBridge(readerWriterBridge);
128 return env;
129 } catch (Exception e) {
130 throw new SOAPException(e);
131 }
132 }
133 private static Envelope parseEnvelopeSax(Source src, SOAPPartImpl soapPart)
134 throws SOAPException {
135 // Insert SAX filter to disallow Document Type Declarations since
136 // they are not legal in SOAP
137 SAXParser saxParser = null;
138 if (src instanceof StreamSource) {
139 try {
140 saxParser = parserPool.get().get();
141 } catch (Exception e) {
142 log.severe("SAAJ0601.util.newSAXParser.exception");
143 throw new SOAPExceptionImpl(
144 "Couldn't get a SAX parser while constructing a envelope",
145 e);
146 }
147 InputSource is = SAXSource.sourceToInputSource(src);
148 if (is.getEncoding()== null && soapPart.getSourceCharsetEncoding() != null) {
149 is.setEncoding(soapPart.getSourceCharsetEncoding());
150 }
151 XMLReader rejectFilter;
152 try {
153 rejectFilter = new RejectDoctypeSaxFilter(saxParser);
154 } catch (Exception ex) {
155 log.severe("SAAJ0510.soap.cannot.create.envelope");
156 throw new SOAPExceptionImpl(
157 "Unable to create envelope from given source: ",
158 ex);
159 }
160 src = new SAXSource(rejectFilter, is);
161 }
162
163 try {
164 Transformer transformer =
165 EfficientStreamingTransformer.newTransformer();
166 DOMResult result = new DOMResult(soapPart);
167 transformer.transform(src, result);
168
169 Envelope env = (Envelope) soapPart.getEnvelope();
170 return env;
171 } catch (Exception ex) {
172 if (ex instanceof SOAPVersionMismatchException) {
173 throw (SOAPVersionMismatchException) ex;
174 }
175 log.severe("SAAJ0511.soap.cannot.create.envelope");
176 throw new SOAPExceptionImpl(
177 "Unable to create envelope from given source: ",
178 ex);
179 } finally {
180 if (saxParser != null) {
181 parserPool.get().returnParser(saxParser);
182 }
183 }
184 }
185 }