/**
* Copyright 2005-2008 Noelios Technologies.
*
* The contents of this file are subject to the terms of the following open
* source licenses: LGPL 3.0 or LGPL 2.1 or CDDL 1.0 (the "Licenses"). You can
* select the license that you prefer but you may not use this file except in
* compliance with one of these Licenses.
*
* You can obtain a copy of the LGPL 3.0 license at
* http://www.gnu.org/licenses/lgpl-3.0.html
*
* You can obtain a copy of the LGPL 2.1 license at
* http://www.gnu.org/licenses/lgpl-2.1.html
*
* You can obtain a copy of the CDDL 1.0 license at
* http://www.sun.com/cddl/cddl.html
*
* See the Licenses for the specific language governing permissions and
* limitations under the Licenses.
*
* Alternatively, you can obtain a royaltee free commercial license with less
* limitations, transferable or non-transferable, directly at
* http://www.noelios.com/products/restlet-engine
*
* Restlet is a registered trademark of Noelios Technologies.
*/
package com.noelios.restlet.http;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.WritableByteChannel;
import java.security.cert.Certificate;
import java.util.List;
import java.util.logging.Level;
import org.restlet.Server;
import org.restlet.data.Digest;
import org.restlet.data.Encoding;
import org.restlet.data.Language;
import org.restlet.data.Parameter;
import org.restlet.data.Response;
import org.restlet.resource.InputRepresentation;
import org.restlet.resource.ReadableRepresentation;
import org.restlet.resource.Representation;
import org.restlet.service.ConnectorService;
import com.noelios.restlet.util.Base64;
import com.noelios.restlet.util.RangeUtils;
/**
* Abstract HTTP server connector call.
*
* @author Jerome Louvel
*/
public abstract class HttpServerCall extends HttpCall {
/**
* Format {@code fileName} as a Content-Disposition header value
*
* @param fileName
* Filename to format
* @return {@code fileName} formatted
*/
public static String formatContentDisposition(String fileName) {
final StringBuilder b = new StringBuilder("attachment; filename=\"");
if (fileName != null) {
b.append(fileName);
}
b.append('"');
return b.toString();
}
/** Indicates if the "host" header was already parsed. */
private volatile boolean hostParsed;
/**
* Constructor.
*
* @param serverAddress
* The server IP address.
* @param serverPort
* The server port.
*/
public HttpServerCall(String serverAddress, int serverPort) {
setServerAddress(serverAddress);
setServerPort(serverPort);
this.hostParsed = false;
}
/**
* Constructor.
*
* @param server
* The parent server connector.
*/
public HttpServerCall(Server server) {
this(server.getAddress(), server.getPort());
}
/**
* Complete the response
*/
public void complete() {
}
/**
* Returns the content length of the request entity if know,
* {@link Representation#UNKNOWN_SIZE} otherwise.
*
* @return The request content length.
*/
protected long getContentLength() {
return getContentLength(getRequestHeaders());
}
/**
* Returns the host domain name.
*
* @return The host domain name.
*/
@Override
public String getHostDomain() {
if (!this.hostParsed) {
parseHost();
}
return super.getHostDomain();
}
/**
* Returns the host port.
*
* @return The host port.
*/
@Override
public int getHostPort() {
if (!this.hostParsed) {
parseHost();
}
return super.getHostPort();
}
/**
* Returns the request entity if available.
*
* @return The request entity if available.
*/
public Representation getRequestEntity() {
Representation result = null;
final long contentLength = getContentLength();
// Create the result representation
final InputStream requestStream = getRequestEntityStream(contentLength);
final ReadableByteChannel requestChannel = getRequestEntityChannel(contentLength);
if (requestStream != null) {
result = new InputRepresentation(requestStream, null, contentLength);
} else if (requestChannel != null) {
result = new ReadableRepresentation(requestChannel, null,
contentLength);
}
// TODO result may be null
result.setSize(contentLength);
// Extract some interesting header values
for (final Parameter header : getRequestHeaders()) {
if (header.getName().equalsIgnoreCase(
HttpConstants.HEADER_CONTENT_ENCODING)) {
final HeaderReader hr = new HeaderReader(header.getValue());
String value = hr.readValue();
while (value != null) {
final Encoding encoding = Encoding.valueOf(value);
if (!encoding.equals(Encoding.IDENTITY)) {
result.getEncodings().add(encoding);
}
value = hr.readValue();
}
} else if (header.getName().equalsIgnoreCase(
HttpConstants.HEADER_CONTENT_LANGUAGE)) {
final HeaderReader hr = new HeaderReader(header.getValue());
String value = hr.readValue();
while (value != null) {
result.getLanguages().add(Language.valueOf(value));
value = hr.readValue();
}
} else if (header.getName().equalsIgnoreCase(
HttpConstants.HEADER_CONTENT_TYPE)) {
final ContentType contentType = new ContentType(header
.getValue());
result.setMediaType(contentType.getMediaType());
result.setCharacterSet(contentType.getCharacterSet());
} else if (header.getName().equalsIgnoreCase(
HttpConstants.HEADER_CONTENT_RANGE)) {
RangeUtils.parseContentRange(header.getValue(), result);
} else if (header.getName().equalsIgnoreCase(
HttpConstants.HEADER_CONTENT_MD5)) {
result.setDigest(new Digest(Digest.ALGORITHM_MD5, Base64
.decode(header.getValue())));
}
}
return result;
}
/**
* Returns the request entity channel if it exists.
*
* @param size
* The expected entity size or -1 if unknown.
*
* @return The request entity channel if it exists.
*/
public abstract ReadableByteChannel getRequestEntityChannel(long size);
/**
* Returns the request entity stream if it exists.
*
* @param size
* The expected entity size or -1 if unknown.
*
* @return The request entity stream if it exists.
*/
public abstract InputStream getRequestEntityStream(long size);
/**
* Returns the request head channel if it exists.
*
* @return The request head channel if it exists.
*/
public abstract ReadableByteChannel getRequestHeadChannel();
/**
* Returns the request head stream if it exists.
*
* @return The request head stream if it exists.
*/
public abstract InputStream getRequestHeadStream();
/**
* Returns the response channel if it exists.
*
* @return The response channel if it exists.
*/
public abstract WritableByteChannel getResponseEntityChannel();
/**
* Returns the response entity stream if it exists.
*
* @return The response entity stream if it exists.
*/
public abstract OutputStream getResponseEntityStream();
/**
* Returns the SSL Cipher Suite, if available and accessible.
*
* @return The SSL Cipher Suite, if available and accessible.
*/
public String getSslCipherSuite() {
return null;
}
/**
* Returns the chain of client certificates, if available and accessible.
*
* @return The chain of client certificates, if available and accessible.
*/
public List<Certificate> getSslClientCertificates() {
return null;
}
/**
* Returns the SSL key size, if available and accessible.
*
* @return The SSL key size, if available and accessible.
*/
public Integer getSslKeySize() {
Integer keySize = null;
final String sslCipherSuite = getSslCipherSuite();
if (sslCipherSuite != null) {
keySize = HttpsUtils.extractKeySize(sslCipherSuite);
}
return keySize;
}
@Override
protected boolean isClientKeepAlive() {
final String header = getRequestHeaders().getFirstValue(
HttpConstants.HEADER_CONNECTION, true);
return (header == null) || !header.equalsIgnoreCase("close");
}
@Override
protected boolean isServerKeepAlive() {
return true;
}
/**
* Parses the "host" header to set the server host and port properties.
*/
private void parseHost() {
final String host = getRequestHeaders().getFirstValue(
HttpConstants.HEADER_HOST, true);
if (host != null) {
final int colonIndex = host.indexOf(':');
if (colonIndex != -1) {
super.setHostDomain(host.substring(0, colonIndex));
super.setHostPort(Integer.valueOf(host
.substring(colonIndex + 1)));
} else {
super.setHostDomain(host);
super.setHostPort(getProtocol().getDefaultPort());
}
} else {
getLogger().info(
"Couldn't find the mandatory \"Host\" HTTP header.");
}
this.hostParsed = true;
}
/**
* Reads the HTTP request head (request line and headers).
*
* @throws IOException
*/
protected void readRequestHead(InputStream headStream) throws IOException {
final StringBuilder sb = new StringBuilder();
// Parse the request method
int next = headStream.read();
while ((next != -1) && !HttpUtils.isSpace(next)) {
sb.append((char) next);
next = headStream.read();
}
if (next == -1) {
throw new IOException(
"Unable to parse the request method. End of stream reached too early.");
} else {
setMethod(sb.toString());
sb.delete(0, sb.length());
// Parse the request URI
next = headStream.read();
while ((next != -1) && !HttpUtils.isSpace(next)) {
sb.append((char) next);
next = headStream.read();
}
if (next == -1) {
throw new IOException(
"Unable to parse the request URI. End of stream reached too early.");
} else {
setRequestUri(sb.toString());
sb.delete(0, sb.length());
// Parse the HTTP version
next = headStream.read();
while ((next != -1) && !HttpUtils.isCarriageReturn(next)) {
sb.append((char) next);
next = headStream.read();
}
if (next == -1) {
throw new IOException(
"Unable to parse the HTTP version. End of stream reached too early.");
} else {
next = headStream.read();
if (HttpUtils.isLineFeed(next)) {
setVersion(sb.toString());
sb.delete(0, sb.length());
// Parse the headers
Parameter header = HttpUtils.readHeader(headStream, sb);
while (header != null) {
getRequestHeaders().add(header);
header = HttpUtils.readHeader(headStream, sb);
}
} else {
throw new IOException(
"Unable to parse the HTTP version. The carriage return must be followed by a line feed.");
}
}
}
}
}
/**
* Sends the response back to the client. Commits the status, headers and
* optional entity and send them over the network. The default
* implementation only writes the response entity on the reponse stream or
* channel. Subclasses will probably also copy the response headers and
* status.
*
* @param response
* The high-level response.
* @throws IOException
* if the Response could not be written to the network.
*/
public void sendResponse(Response response) throws IOException {
if (response != null) {
try {
writeResponseHead(response);
final Representation entity = response.getEntity();
if (entity != null) {
// Get the connector service to callback
final ConnectorService connectorService = getConnectorService(response
.getRequest());
if (connectorService != null) {
connectorService.beforeSend(entity);
}
final WritableByteChannel responseEntityChannel = getResponseEntityChannel();
final OutputStream responseEntityStream = getResponseEntityStream();
writeResponseBody(entity, responseEntityChannel,
responseEntityStream);
if (connectorService != null) {
connectorService.afterSend(entity);
}
if (responseEntityStream != null) {
try {
responseEntityStream.flush();
responseEntityStream.close();
} catch (IOException ioe) {
// The stream was probably already closed by the
// connector. Probably ok, low message priority.
getLogger()
.log(
Level.FINE,
"Exception while flushing and closing the entity stream.",
ioe);
}
}
}
} finally {
final Representation entity = response.getEntity();
if (entity != null) {
entity.release();
}
}
}
}
/**
* Indicates if the response should be chunked because its length is
* unknown.
*
* @param response
* The response to analyze.
* @return True if the response should be chunked.
*/
protected boolean shouldResponseBeChunked(Response response) {
return (response.getEntity() != null)
&& (response.getEntity().getSize() == Representation.UNKNOWN_SIZE);
}
/**
* Effectively writes the response body. The entity to write is guaranteed
* to be non null. Attempts to write the entity on the response channel or
* response stream by default.
*
* @param entity
* The representation to write as entity of the body.
* @param responseEntityChannel
* The response entity channel or null if a stream is used.
* @param responseEntityStream
* The response entity stream or null if a channel is used.
* @throws IOException
*/
public void writeResponseBody(Representation entity,
WritableByteChannel responseEntityChannel,
OutputStream responseEntityStream) throws IOException {
// Send the entity to the client
if (responseEntityChannel != null) {
entity.write(responseEntityChannel);
} else if (responseEntityStream != null) {
entity.write(responseEntityStream);
responseEntityStream.flush();
}
}
/**
* Writes the response status line and headers. Does nothing by default.
*
* @param response
* The response.
* @throws IOException
*/
public void writeResponseHead(Response response) throws IOException {
// Do nothing by default
}
/**
* Writes the response head to the given output stream.
*
* @param response
* The response.
* @param headStream
* The output stream to write to.
* @throws IOException
*/
protected void writeResponseHead(Response response, OutputStream headStream)
throws IOException {
// Write the status line
final String version = (getVersion() == null) ? "1.1" : getVersion();
headStream.write(version.getBytes());
headStream.write(' ');
headStream.write(Integer.toString(getStatusCode()).getBytes());
headStream.write(' ');
if (getReasonPhrase() != null) {
headStream.write(getReasonPhrase().getBytes());
} else {
headStream.write(("Status " + getStatusCode()).getBytes());
}
headStream.write(13); // CR
headStream.write(10); // LF
// We don't support persistent connections yet
getResponseHeaders().set(HttpConstants.HEADER_CONNECTION, "close",
isServerKeepAlive());
// Check if 'Transfer-Encoding' header should be set
if (shouldResponseBeChunked(response)) {
getResponseHeaders().add(HttpConstants.HEADER_TRANSFER_ENCODING,
"chunked");
}
// Write the response headers
for (final Parameter header : getResponseHeaders()) {
HttpUtils.writeHeader(header, headStream);
}
// Write the end of the headers section
headStream.write(13); // CR
headStream.write(10); // LF
headStream.flush();
}
}