add

+/*
+ * Copyright 2007 Stephen Liu
+ * For license terms, see the file COPYING along with this library.
+ */
+
+#include <stdio.h>
+#include <time.h>
+#include <string.h>
+
+#include "spthread.hpp"
+
+#include "sphttp.hpp"
+#include "sphttpmsg.hpp"
+#include "spbuffer.hpp"
+#include "sprequest.hpp"
+#include "spresponse.hpp"
+#include "spmsgblock.hpp"
+
+SP_HttpHandler :: ~SP_HttpHandler()
+{
+}
+
+void SP_HttpHandler :: error()
+{
+}
+
+void SP_HttpHandler :: timeout()
+{
+}
+
+//---------------------------------------------------------
+
+SP_HttpHandlerFactory :: ~SP_HttpHandlerFactory()
+{
+}
+
+//---------------------------------------------------------
+
+class SP_HttpRequestDecoder : public SP_MsgDecoder {
+public:
+	SP_HttpRequestDecoder();
+
+	virtual ~SP_HttpRequestDecoder();
+
+	virtual int decode( SP_Buffer * inBuffer );
+
+	SP_HttpRequest * getMsg();
+
+private:
+	SP_HttpMsgParser * mParser;
+};
+
+SP_HttpRequestDecoder :: SP_HttpRequestDecoder()
+{
+	mParser = new SP_HttpMsgParser();
+}
+
+SP_HttpRequestDecoder :: ~SP_HttpRequestDecoder()
+{
+	delete mParser;
+}
+
+int SP_HttpRequestDecoder :: decode( SP_Buffer * inBuffer )
+{
+	if( inBuffer->getSize() > 0 ) {
+		int len = mParser->append( inBuffer->getBuffer(), inBuffer->getSize() );
+
+		inBuffer->erase( len );
+
+		return mParser->isCompleted() ? eOK : eMoreData;
+	} else {
+		return eMoreData;
+	}
+}
+
+SP_HttpRequest * SP_HttpRequestDecoder :: getMsg()
+{
+	return mParser->getRequest();
+}
+
+//---------------------------------------------------------
+
+class SP_HttpResponseMsgBlock : public SP_MsgBlock {
+public:
+	SP_HttpResponseMsgBlock( SP_HttpResponse * response );
+	virtual ~SP_HttpResponseMsgBlock();
+
+	virtual const void * getData() const;
+	virtual size_t getSize() const;
+
+private:
+	SP_HttpResponse * mResponse;
+};
+
+
+SP_HttpResponseMsgBlock :: SP_HttpResponseMsgBlock( SP_HttpResponse * response )
+{
+	mResponse = response;
+}
+
+SP_HttpResponseMsgBlock :: ~SP_HttpResponseMsgBlock()
+{
+	if( NULL != mResponse ) delete mResponse;
+	mResponse = NULL;
+}
+
+const void * SP_HttpResponseMsgBlock :: getData() const
+{
+	return mResponse->getContent();
+}
+
+size_t SP_HttpResponseMsgBlock :: getSize() const
+{
+	return mResponse->getContentLength();
+}
+
+//---------------------------------------------------------
+
+class SP_HttpHandlerAdapter : public SP_Handler {
+public:
+	SP_HttpHandlerAdapter( SP_HttpHandler * handler );
+
+	virtual ~SP_HttpHandlerAdapter();
+
+	// return -1 : terminate session, 0 : continue
+	virtual int start( SP_Request * request, SP_Response * response );
+
+	// return -1 : terminate session, 0 : continue
+	virtual int handle( SP_Request * request, SP_Response * response );
+
+	virtual void error( SP_Response * response );
+
+	virtual void timeout( SP_Response * response );
+
+	virtual void close();
+
+private:
+	SP_HttpHandler * mHandler;
+};
+
+SP_HttpHandlerAdapter :: SP_HttpHandlerAdapter( SP_HttpHandler * handler )
+{
+	mHandler = handler;
+}
+
+SP_HttpHandlerAdapter :: ~SP_HttpHandlerAdapter()
+{
+	delete mHandler;
+}
+
+int SP_HttpHandlerAdapter :: start( SP_Request * request, SP_Response * response )
+{
+	request->setMsgDecoder( new SP_HttpRequestDecoder() );
+
+	return 0;
+}
+
+int SP_HttpHandlerAdapter :: handle( SP_Request * request, SP_Response * response )
+{
+	SP_HttpRequestDecoder * decoder = ( SP_HttpRequestDecoder * ) request->getMsgDecoder();
+	SP_HttpRequest * httpRequest = ( SP_HttpRequest * ) decoder->getMsg();
+
+	httpRequest->setClinetIP( request->getClientIP() );
+
+	SP_HttpResponse * httpResponse = new SP_HttpResponse();
+	httpResponse->setVersion( httpRequest->getVersion() );
+
+	mHandler->handle( httpRequest, httpResponse );
+
+	SP_Buffer * reply = response->getReply()->getMsg();
+
+	char buffer[ 512 ] = { 0 };
+	snprintf( buffer, sizeof( buffer ), "%s %i %s\r\n", httpResponse->getVersion(),
+		httpResponse->getStatusCode(), httpResponse->getReasonPhrase() );
+	reply->append( buffer );
+
+	// check keep alive header
+	if( httpRequest->isKeepAlive() ) {
+		if( NULL == httpResponse->getHeaderValue( SP_HttpMessage::HEADER_CONNECTION ) ) {
+			httpResponse->addHeader( SP_HttpMessage::HEADER_CONNECTION, "Keep-Alive" );
+		}
+	}
+
+	if( 0 != strcasecmp( httpRequest->getMethod(), "head" ) ) {
+		// check Content-Length header
+		httpResponse->removeHeader( SP_HttpMessage::HEADER_CONTENT_LENGTH );
+		if( httpResponse->getContentLength() >= 0 ) {
+				snprintf( buffer, sizeof( buffer ), "%d", httpResponse->getContentLength() );
+				httpResponse->addHeader( SP_HttpMessage::HEADER_CONTENT_LENGTH, buffer );
+		}
+	}
+
+	// check date header
+	httpResponse->removeHeader( SP_HttpMessage::HEADER_DATE );
+	time_t tTime = time( NULL );
+	struct tm tmTime;
+	gmtime_r( &tTime, &tmTime );
+	strftime( buffer, sizeof( buffer ), "%a, %d %b %Y %H:%M:%S %Z", &tmTime );
+	httpResponse->addHeader( SP_HttpMessage::HEADER_DATE, buffer );
+
+	// check Content-Type header
+	if( NULL == httpResponse->getHeaderValue( SP_HttpMessage::HEADER_CONTENT_TYPE ) ) {
+		httpResponse->addHeader( SP_HttpMessage::HEADER_CONTENT_TYPE,
+			"text/html; charset=ISO-8859-1" );
+	}
+
+	// check Server header
+	httpResponse->removeHeader( SP_HttpMessage::HEADER_SERVER );
+	httpResponse->addHeader( SP_HttpMessage::HEADER_SERVER, "sphttp/spserver" );
+
+	for( int i = 0; i < httpResponse->getHeaderCount(); i++ ) {
+		snprintf( buffer, sizeof( buffer ), "%s: %s\r\n",
+			httpResponse->getHeaderName( i ), httpResponse->getHeaderValue( i ) );
+		reply->append( buffer );
+	}
+
+	reply->append( "\r\n" );	
+
+	char keepAlive[ 32 ] = { 0 };
+	if( NULL != httpResponse->getHeaderValue( SP_HttpMessage::HEADER_CONNECTION ) ) {
+		strncpy( keepAlive, httpResponse->getHeaderValue(
+				SP_HttpMessage::HEADER_CONNECTION ), sizeof( keepAlive ) - 1 );
+	}
+
+	if( NULL != httpResponse->getContent() ) {
+		response->getReply()->getFollowBlockList()->append(
+				new SP_HttpResponseMsgBlock( httpResponse ) );
+	} else {
+		delete httpResponse;
+	}
+
+	request->setMsgDecoder( new SP_HttpRequestDecoder() );
+
+	return 0 == strcasecmp( keepAlive, "Keep-Alive" ) ? 0 : -1;
+}
+
+void SP_HttpHandlerAdapter :: error( SP_Response * response )
+{
+	mHandler->error();
+}
+
+void SP_HttpHandlerAdapter :: timeout( SP_Response * response )
+{
+	mHandler->timeout();
+}
+
+void SP_HttpHandlerAdapter :: close()
+{
+}
+
+//---------------------------------------------------------
+
+SP_HttpHandlerAdapterFactory :: SP_HttpHandlerAdapterFactory( SP_HttpHandlerFactory * factory )
+{
+	mFactory = factory;
+}
+
+SP_HttpHandlerAdapterFactory :: ~SP_HttpHandlerAdapterFactory()
+{
+	delete mFactory;
+}
+
+SP_Handler * SP_HttpHandlerAdapterFactory :: create() const
+{
+	return new SP_HttpHandlerAdapter( mFactory->create() );
+}
+