/*
* Copyright (c) 1997, 2010, 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.ws.transport.http.server;
import com.sun.istack.internal.NotNull;
import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpsExchange;
import com.sun.xml.internal.ws.api.message.Packet;
import com.sun.xml.internal.ws.api.server.WSEndpoint;
import com.sun.xml.internal.ws.api.server.WebServiceContextDelegate;
import com.sun.xml.internal.ws.api.server.PortAddressResolver;
import com.sun.xml.internal.ws.transport.http.HttpAdapter;
import com.sun.xml.internal.ws.transport.http.WSHTTPConnection;
import com.sun.xml.internal.ws.developer.JAXWSProperties;
import com.sun.xml.internal.ws.resources.WsservletMessages;
import com.sun.xml.internal.ws.util.ReadAllStream;
import javax.xml.ws.handler.MessageContext;
import javax.xml.ws.WebServiceException;
import java.io.FilterInputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.security.Principal;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
/**
* {@link WSHTTPConnection} used with Java SE endpoints. It provides connection
* implementation using {@link HttpExchange} object.
*
* @author Jitendra Kotamraju
*/
final class ServerConnectionImpl extends WSHTTPConnection implements WebServiceContextDelegate {
private final HttpExchange httpExchange;
private int status;
private final HttpAdapter adapter;
private LWHSInputStream in;
private OutputStream out;
public ServerConnectionImpl(@NotNull HttpAdapter adapter, @NotNull HttpExchange httpExchange) {
this.adapter = adapter;
this.httpExchange = httpExchange;
}
@Override
@Property(value = {MessageContext.HTTP_REQUEST_HEADERS, Packet.INBOUND_TRANSPORT_HEADERS})
public @NotNull Map<String,List<String>> getRequestHeaders() {
return httpExchange.getRequestHeaders();
}
@Override
public String getRequestHeader(String headerName) {
return httpExchange.getRequestHeaders().getFirst(headerName);
}
@Override
public void setResponseHeaders(Map<String,List<String>> headers) {
Headers r = httpExchange.getResponseHeaders();
r.clear();
for(Map.Entry <String, List<String>> entry : headers.entrySet()) {
String name = entry.getKey();
List<String> values = entry.getValue();
// ignore headers that interfere with our correct operations
if (!name.equalsIgnoreCase("Content-Length") && !name.equalsIgnoreCase("Content-Type")) {
r.put(name,new ArrayList<String>(values));
}
}
}
@Override
@Property({MessageContext.HTTP_RESPONSE_HEADERS,Packet.OUTBOUND_TRANSPORT_HEADERS})
public Map<String,List<String>> getResponseHeaders() {
return httpExchange.getResponseHeaders();
}
@Override
public void setContentTypeResponseHeader(@NotNull String value) {
httpExchange.getResponseHeaders().set("Content-Type",value);
}
@Override
public void setStatus(int status) {
this.status = status;
}
@Override
@Property(MessageContext.HTTP_RESPONSE_CODE)
public int getStatus() {
return status;
}
public @NotNull InputStream getInput() {
if (in == null) {
in = new LWHSInputStream(httpExchange.getRequestBody());
}
return in;
}
// Light weight http server's InputStream.close() throws exception if
// all the bytes are not read. Work around until it is fixed.
private static class LWHSInputStream extends FilterInputStream {
// Workaround for "SJSXP XMLStreamReader.next() closes stream".
boolean closed;
boolean readAll;
LWHSInputStream(InputStream in) {
super(in);
}
void readAll() throws IOException {
if (!closed && !readAll) {
ReadAllStream all = new ReadAllStream();
all.readAll(in, 4000000);
in.close();
in = all;
readAll = true;
}
}
@Override
public void close() throws IOException {
if (!closed) {
readAll();
super.close();
closed = true;
}
}
}
public @NotNull OutputStream getOutput() throws IOException {
if (out == null) {
String lenHeader = httpExchange.getResponseHeaders().getFirst("Content-Length");
int length = (lenHeader != null) ? Integer.parseInt(lenHeader) : 0;
httpExchange.sendResponseHeaders(getStatus(), length);
// Light weight http server's OutputStream.close() throws exception if
// all the bytes are not read on the client side(StreamMessage on the client
// side doesn't read all bytes.
out = new FilterOutputStream(httpExchange.getResponseBody()) {
boolean closed;
@Override
public void close() throws IOException {
if (!closed) {
closed = true;
// lwhs closes input stream, when you close the output stream
// This causes problems for streaming in one-way cases
in.readAll();
try {
super.close();
} catch(IOException ioe) {
// Ignoring purposefully.
}
}
}
// Otherwise, FilterOutpuStream writes byte by byte
@Override
public void write(byte[] buf, int start, int len) throws IOException {
out.write(buf, start, len);
}
};
}
return out;
}
public @NotNull WebServiceContextDelegate getWebServiceContextDelegate() {
return this;
}
public Principal getUserPrincipal(Packet request) {
return httpExchange.getPrincipal();
}
public boolean isUserInRole(Packet request, String role) {
return false;
}
public @NotNull String getEPRAddress(Packet request, WSEndpoint endpoint) {
//return WSHttpHandler.getRequestAddress(httpExchange);
PortAddressResolver resolver = adapter.owner.createPortAddressResolver(getBaseAddress());
String address = resolver.getAddressFor(endpoint.getServiceName(), endpoint.getPortName().getLocalPart());
if(address==null)
throw new WebServiceException(WsservletMessages.SERVLET_NO_ADDRESS_AVAILABLE(endpoint.getPortName()));
return address;
}
public String getWSDLAddress(@NotNull Packet request, @NotNull WSEndpoint endpoint) {
String eprAddress = getEPRAddress(request,endpoint);
if(adapter.getEndpoint().getPort() != null)
return eprAddress+"?wsdl";
else
return null;
}
@Override
public boolean isSecure() {
return (httpExchange instanceof HttpsExchange);
}
@Override
@Property(MessageContext.HTTP_REQUEST_METHOD)
public @NotNull String getRequestMethod() {
return httpExchange.getRequestMethod();
}
@Override
@Property(MessageContext.QUERY_STRING)
public String getQueryString() {
URI requestUri = httpExchange.getRequestURI();
String query = requestUri.getQuery();
if (query != null)
return query;
return null;
}
@Override
@Property(MessageContext.PATH_INFO)
public String getPathInfo() {
URI requestUri = httpExchange.getRequestURI();
String reqPath = requestUri.getPath();
String ctxtPath = httpExchange.getHttpContext().getPath();
if (reqPath.length() > ctxtPath.length()) {
return reqPath.substring(ctxtPath.length());
}
return null;
}
@Property(JAXWSProperties.HTTP_EXCHANGE)
public HttpExchange getExchange() {
return httpExchange;
}
@Override @NotNull
public String getBaseAddress() {
/*
* Computes the Endpoint's address from the request. Use "Host" header
* so that it has correct address(IP address or someother hostname)
* through which the application reached the endpoint.
*
*/
StringBuilder strBuf = new StringBuilder();
strBuf.append((httpExchange instanceof HttpsExchange) ? "https" : "http");
strBuf.append("://");
String hostHeader = httpExchange.getRequestHeaders().getFirst("Host");
if (hostHeader != null) {
strBuf.append(hostHeader); // Uses Host header
} else {
strBuf.append(httpExchange.getLocalAddress().getHostName());
strBuf.append(":");
strBuf.append(httpExchange.getLocalAddress().getPort());
}
//Do not include URL pattern here
//strBuf.append(httpExchange.getRequestURI().getPath());
return strBuf.toString();
}
@Override
public String getProtocol() {
return httpExchange.getProtocol();
}
@Override
public void setContentLengthResponseHeader(int value) {
httpExchange.getResponseHeaders().set("Content-Length", ""+value);
}
protected PropertyMap getPropertyMap() {
return model;
}
private static final PropertyMap model;
static {
model = parse(ServerConnectionImpl.class);
}
}