aoqi@0: /*
aoqi@0:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: package com.sun.xml.internal.ws.transport.http.server;
aoqi@0: 
aoqi@0: import com.sun.istack.internal.NotNull;
aoqi@0: import com.sun.xml.internal.ws.api.message.Packet;
aoqi@0: import com.sun.xml.internal.ws.api.server.WSEndpoint;
aoqi@0: import com.sun.xml.internal.ws.api.server.WebServiceContextDelegate;
aoqi@0: import com.sun.xml.internal.ws.api.server.PortAddressResolver;
aoqi@0: import com.sun.xml.internal.ws.transport.http.HttpAdapter;
aoqi@0: import com.sun.xml.internal.ws.transport.http.WSHTTPConnection;
aoqi@0: import com.sun.xml.internal.ws.developer.JAXWSProperties;
aoqi@0: import com.sun.xml.internal.ws.resources.WsservletMessages;
aoqi@0: 
aoqi@0: import javax.xml.ws.handler.MessageContext;
aoqi@0: import javax.xml.ws.WebServiceException;
aoqi@0: import javax.xml.ws.spi.http.HttpExchange;
aoqi@0: import java.io.IOException;
aoqi@0: import java.io.InputStream;
aoqi@0: import java.io.OutputStream;
aoqi@0: import java.security.Principal;
aoqi@0: import java.util.ArrayList;
aoqi@0: import java.util.List;
aoqi@0: import java.util.Map;
aoqi@0: import java.util.Set;
aoqi@0: 
aoqi@0: /**
aoqi@0:  * {@link WSHTTPConnection} used with Java SE endpoints. It provides connection
aoqi@0:  * implementation using {@link HttpExchange} object.
aoqi@0:  *
aoqi@0:  * @author Jitendra Kotamraju
aoqi@0:  */
aoqi@0: final class PortableConnectionImpl extends WSHTTPConnection implements WebServiceContextDelegate {
aoqi@0: 
aoqi@0:     private final HttpExchange httpExchange;
aoqi@0:     private int status;
aoqi@0:     private final HttpAdapter adapter;
aoqi@0:     private boolean outputWritten;
aoqi@0: 
aoqi@0:     public PortableConnectionImpl(@NotNull HttpAdapter adapter, @NotNull HttpExchange httpExchange) {
aoqi@0:         this.adapter = adapter;
aoqi@0:         this.httpExchange = httpExchange;
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property(value = {MessageContext.HTTP_REQUEST_HEADERS, Packet.INBOUND_TRANSPORT_HEADERS})
aoqi@0:     public @NotNull Map<String,List<String>> getRequestHeaders() {
aoqi@0:         return httpExchange.getRequestHeaders();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getRequestHeader(String headerName) {
aoqi@0:         return httpExchange.getRequestHeader(headerName);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public void setResponseHeaders(Map<String,List<String>> headers) {
aoqi@0:         Map<String, List<String>> r = httpExchange.getResponseHeaders();
aoqi@0:         r.clear();
aoqi@0:         for(Map.Entry <String, List<String>> entry : headers.entrySet()) {
aoqi@0:             String name = entry.getKey();
aoqi@0:             List<String> values = entry.getValue();
aoqi@0:             // ignore headers that interfere with our correct operations
aoqi@0:             if (!name.equalsIgnoreCase("Content-Length") && !name.equalsIgnoreCase("Content-Type")) {
aoqi@0:                 r.put(name,new ArrayList<String>(values));
aoqi@0:             }
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public void setResponseHeader(String key, List<String> value) {
aoqi@0:         httpExchange.getResponseHeaders().put(key, value);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public Set<String> getRequestHeaderNames() {
aoqi@0:         return httpExchange.getRequestHeaders().keySet();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public List<String> getRequestHeaderValues(String headerName) {
aoqi@0:         return httpExchange.getRequestHeaders().get(headerName);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property({MessageContext.HTTP_RESPONSE_HEADERS,Packet.OUTBOUND_TRANSPORT_HEADERS})
aoqi@0:     public Map<String,List<String>> getResponseHeaders() {
aoqi@0:         return httpExchange.getResponseHeaders();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public void setContentTypeResponseHeader(@NotNull String value) {
aoqi@0:         httpExchange.addResponseHeader("Content-Type", value);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public void setStatus(int status) {
aoqi@0:         this.status = status;
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property(MessageContext.HTTP_RESPONSE_CODE)
aoqi@0:     public int getStatus() {
aoqi@0:         return status;
aoqi@0:     }
aoqi@0: 
aoqi@0:     public @Override @NotNull InputStream getInput() throws IOException {
aoqi@0:         return httpExchange.getRequestBody();
aoqi@0:     }
aoqi@0: 
aoqi@0:     public @Override @NotNull OutputStream getOutput() throws IOException {
aoqi@0:         assert !outputWritten;
aoqi@0:         outputWritten = true;
aoqi@0: 
aoqi@0:         httpExchange.setStatus(getStatus());
aoqi@0:         return httpExchange.getResponseBody();
aoqi@0:     }
aoqi@0: 
aoqi@0:     public @Override @NotNull WebServiceContextDelegate getWebServiceContextDelegate() {
aoqi@0:         return this;
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public Principal getUserPrincipal(Packet request) {
aoqi@0:         return httpExchange.getUserPrincipal();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public boolean isUserInRole(Packet request, String role) {
aoqi@0:         return httpExchange.isUserInRole(role);
aoqi@0:     }
aoqi@0: 
aoqi@0:     public @Override @NotNull String getEPRAddress(Packet request, WSEndpoint endpoint) {
aoqi@0:         PortAddressResolver resolver = adapter.owner.createPortAddressResolver(getBaseAddress(), endpoint.getImplementationClass());
aoqi@0:         String address = resolver.getAddressFor(endpoint.getServiceName(), endpoint.getPortName().getLocalPart());
aoqi@0:         if(address==null) {
aoqi@0:             throw new WebServiceException(WsservletMessages.SERVLET_NO_ADDRESS_AVAILABLE(endpoint.getPortName()));
aoqi@0:         }
aoqi@0:         return address;
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Property(MessageContext.SERVLET_CONTEXT)
aoqi@0:     public Object getServletContext() {
aoqi@0:         return httpExchange.getAttribute(MessageContext.SERVLET_CONTEXT);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Property(MessageContext.SERVLET_RESPONSE)
aoqi@0:     public Object getServletResponse() {
aoqi@0:         return httpExchange.getAttribute(MessageContext.SERVLET_RESPONSE);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Property(MessageContext.SERVLET_REQUEST)
aoqi@0:     public Object getServletRequest() {
aoqi@0:         return httpExchange.getAttribute(MessageContext.SERVLET_REQUEST);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getWSDLAddress(@NotNull Packet request, @NotNull WSEndpoint endpoint) {
aoqi@0:         String eprAddress = getEPRAddress(request,endpoint);
aoqi@0:         if(adapter.getEndpoint().getPort() != null) {
aoqi@0:             return eprAddress+"?wsdl";
aoqi@0:         } else {
aoqi@0:             return null;
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public boolean isSecure() {
aoqi@0:         return httpExchange.getScheme().equals("https");
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property(MessageContext.HTTP_REQUEST_METHOD)
aoqi@0:     public @NotNull String getRequestMethod() {
aoqi@0:         return httpExchange.getRequestMethod();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property(MessageContext.QUERY_STRING)
aoqi@0:     public String getQueryString() {
aoqi@0:         return httpExchange.getQueryString();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     @Property(MessageContext.PATH_INFO)
aoqi@0:     public String getPathInfo() {
aoqi@0:         return httpExchange.getPathInfo();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Property(JAXWSProperties.HTTP_EXCHANGE)
aoqi@0:     public HttpExchange getExchange() {
aoqi@0:         return httpExchange;
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override @NotNull
aoqi@0:     public String getBaseAddress() {
aoqi@0:         StringBuilder sb = new StringBuilder();
aoqi@0:         sb.append(httpExchange.getScheme());
aoqi@0:         sb.append("://");
aoqi@0:         sb.append(httpExchange.getLocalAddress().getHostName());
aoqi@0:         sb.append(":");
aoqi@0:         sb.append(httpExchange.getLocalAddress().getPort());
aoqi@0:         sb.append(httpExchange.getContextPath());
aoqi@0:         return sb.toString();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getProtocol() {
aoqi@0:         return httpExchange.getProtocol();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public void setContentLengthResponseHeader(int value) {
aoqi@0:         httpExchange.addResponseHeader("Content-Length", ""+value);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getRequestURI() {
aoqi@0:         return httpExchange.getRequestURI().toString();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getRequestScheme() {
aoqi@0:         return httpExchange.getScheme();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public String getServerName() {
aoqi@0:         return httpExchange.getLocalAddress().getHostName();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     public int getServerPort() {
aoqi@0:         return httpExchange.getLocalAddress().getPort();
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     protected PropertyMap getPropertyMap() {
aoqi@0:         return model;
aoqi@0:     }
aoqi@0: 
aoqi@0:     private static final PropertyMap model;
aoqi@0: 
aoqi@0:     static {
aoqi@0:         model = parse(PortableConnectionImpl.class);
aoqi@0:     }
aoqi@0: }