/*
* $Header: /home/cvs/jakarta-commons/httpclient/src/test/org/apache/commons/httpclient/server/SimpleHttpServerConnection.java,v 1.20 2004/11/13 22:38:27 mbecke Exp $
* $Revision: 1.20 $
* $Date: 2004/11/13 22:38:27 $
*
* ====================================================================
*
* Copyright 1999-2004 The Apache Software Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* <http://www.apache.org/>.
*
*/
package org.apache.commons.httpclient.server;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.Socket;
import java.util.Iterator;
import org.apache.commons.httpclient.ChunkedOutputStream;
import org.apache.commons.httpclient.Header;
import org.apache.commons.httpclient.HttpParser;
import org.apache.commons.httpclient.StatusLine;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* A connection to the SimpleHttpServer.
*
* @author Christian Kohlschuetter
* @author Oleg Kalnichevski
*/
public class SimpleHttpServerConnection {
private static final Log LOG = LogFactory.getLog(SimpleHttpServerConnection.class);
private static final String HTTP_ELEMENT_CHARSET = "US-ASCII";
private Socket socket = null;
private InputStream in = null;
private OutputStream out = null;
private boolean keepAlive = false;
public SimpleHttpServerConnection(final Socket socket)
throws IOException {
super();
if (socket == null) {
throw new IllegalArgumentException("Socket may not be null");
}
this.socket = socket;
this.socket.setSoTimeout(500);
this.in = socket.getInputStream();
this.out = socket.getOutputStream();
}
public synchronized void close() {
try {
if (socket != null) {
in.close();
out.close();
socket.close();
socket = null;
}
} catch (IOException e) {
}
}
public synchronized boolean isOpen() {
return this.socket != null;
}
public void setKeepAlive(boolean b) {
this.keepAlive = b;
}
public boolean isKeepAlive() {
return this.keepAlive;
}
public InputStream getInputStream() {
return this.in;
}
public OutputStream getOutputStream() {
return this.out;
}
/**
* Returns the ResponseWriter used to write the output to the socket.
*
* @return This connection's ResponseWriter
*/
public ResponseWriter getWriter() throws UnsupportedEncodingException {
return new ResponseWriter(out);
}
public SimpleRequest readRequest() throws IOException {
try {
String line = null;
do {
line = HttpParser.readLine(in, HTTP_ELEMENT_CHARSET);
} while (line != null && line.length() == 0);
if (line == null) {
setKeepAlive(false);
return null;
}
SimpleRequest request = new SimpleRequest(
RequestLine.parseLine(line),
HttpParser.parseHeaders(this.in, HTTP_ELEMENT_CHARSET),
this.in);
return request;
} catch (IOException e) {
close();
throw e;
}
}
public SimpleResponse readResponse() throws IOException {
try {
String line = null;
do {
line = HttpParser.readLine(in, HTTP_ELEMENT_CHARSET);
} while (line != null && line.length() == 0);
if (line == null) {
setKeepAlive(false);
return null;
}
SimpleResponse response = new SimpleResponse(
new StatusLine(line),
HttpParser.parseHeaders(this.in, HTTP_ELEMENT_CHARSET),
this.in);
return response;
} catch (IOException e) {
close();
throw e;
}
}
public void writeRequest(final SimpleRequest request) throws IOException {
if (request == null) {
return;
}
ResponseWriter writer = new ResponseWriter(this.out, HTTP_ELEMENT_CHARSET);
writer.println(request.getRequestLine().toString());
Iterator item = request.getHeaderIterator();
while (item.hasNext()) {
Header header = (Header) item.next();
writer.print(header.toExternalForm());
}
writer.println();
writer.flush();
OutputStream outsream = this.out;
InputStream content = request.getBody();
if (content != null) {
Header transferenc = request.getFirstHeader("Transfer-Encoding");
if (transferenc != null) {
request.removeHeaders("Content-Length");
if (transferenc.getValue().indexOf("chunked") != -1) {
outsream = new ChunkedOutputStream(outsream);
}
}
byte[] tmp = new byte[4096];
int i = 0;
while ((i = content.read(tmp)) >= 0) {
outsream.write(tmp, 0, i);
}
if (outsream instanceof ChunkedOutputStream) {
((ChunkedOutputStream)outsream).finish();
}
}
outsream.flush();
}
public void writeResponse(final SimpleResponse response) throws IOException {
if (response == null) {
return;
}
ResponseWriter writer = new ResponseWriter(this.out, HTTP_ELEMENT_CHARSET);
writer.println(response.getStatusLine());
Iterator item = response.getHeaderIterator();
while (item.hasNext()) {
Header header = (Header) item.next();
writer.print(header.toExternalForm());
}
writer.println();
writer.flush();
OutputStream outsream = this.out;
InputStream content = response.getBody();
if (content != null) {
Header transferenc = response.getFirstHeader("Transfer-Encoding");
if (transferenc != null) {
response.removeHeaders("Content-Length");
if (transferenc.getValue().indexOf("chunked") != -1) {
outsream = new ChunkedOutputStream(outsream);
}
}
byte[] tmp = new byte[1024];
int i = 0;
while ((i = content.read(tmp)) >= 0) {
outsream.write(tmp, 0, i);
}
if (outsream instanceof ChunkedOutputStream) {
((ChunkedOutputStream)outsream).finish();
}
}
outsream.flush();
}
}