add

+/*
+ * Copyright 2008 Stephen Liu
+ * For license terms, see the file COPYING along with this library.
+ */
+
+#include <winsock2.h>
+#include <windows.h>
+#include <stdio.h>
+#include <assert.h>
+
+#define _CRTDBG_MAP_ALLOC
+#include <stdlib.h>
+#include <crtdbg.h>
+
+#include "spwin32iocp.hpp"
+#include "spiocpserver.hpp"
+#include "spiocplfserver.hpp"
+
+#include "spsession.hpp"
+#include "spbuffer.hpp"
+#include "spmsgdecoder.hpp"
+#include "sprequest.hpp"
+#include "sphandler.hpp"
+#include "sputils.hpp"
+#include "spgetopt.h"
+
+#pragma comment(lib,"ws2_32")
+#pragma comment(lib,"mswsock")
+#pragma comment(lib,"advapi32")
+
+class SP_EchoHandler : public SP_Handler {
+public:
+	SP_EchoHandler(){}
+	virtual ~SP_EchoHandler(){}
+
+	// return -1 : terminate session, 0 : continue
+	virtual int start( SP_Request * request, SP_Response * response ) {
+		request->setMsgDecoder( new SP_MultiLineMsgDecoder() );
+		response->getReply()->getMsg()->append(
+			"Welcome to line echo server, enter 'quit' to quit.\r\n" );
+
+		return 0;
+	}
+
+	// return -1 : terminate session, 0 : continue
+	virtual int handle( SP_Request * request, SP_Response * response ) {
+		SP_MultiLineMsgDecoder * decoder = (SP_MultiLineMsgDecoder*)request->getMsgDecoder();
+		SP_CircleQueue * queue = decoder->getQueue();
+
+		int ret = 0;
+		for( ; NULL != queue->top(); ) {
+			char * line = (char*)queue->pop();
+
+			if( 0 != strcasecmp( line, "quit" ) ) {
+				response->getReply()->getMsg()->append( line );
+				response->getReply()->getMsg()->append( "\r\n" );
+			} else {
+				response->getReply()->getMsg()->append( "Byebye\r\n" );
+				ret = -1;
+			}
+
+			free( line );
+		}
+
+		return ret;
+	}
+
+	virtual void error( SP_Response * response ) {}
+
+	virtual void timeout( SP_Response * response ) {}
+
+	virtual void close() {}
+};
+
+class SP_EchoHandlerFactory : public SP_HandlerFactory {
+public:
+	SP_EchoHandlerFactory() {}
+	virtual ~SP_EchoHandlerFactory() {}
+
+	virtual SP_Handler * create() const {
+		return new SP_EchoHandler();
+	}
+};
+
+void IncreaseConnections()
+{
+	SetProcessWorkingSetSize( GetCurrentProcess(),
+			10 * 1024 * 1024, 400 * 1024 * 1024 );
+
+	DWORD minSize = 0, maxSize = 0;
+	GetProcessWorkingSetSize( GetCurrentProcess(), &minSize, &maxSize );
+	printf( "WorkingSetSize min %d(%d), max %d(%d)\n",
+			minSize, minSize / 4096, maxSize, maxSize / 4096 );
+
+	HKEY hKey;
+
+	/* http://support.microsoft.com/default.aspx?scid=kb;EN-US;314053 */
+	printf("Writing to registry... \n");
+	if( RegOpenKeyEx( HKEY_LOCAL_MACHINE,
+			"SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Winsock", 0,
+			KEY_WRITE, &hKey ) != ERROR_SUCCESS ) {
+		printf( "RegOpenKeyEx fail, errno %d\n", GetLastError() );
+	} else {
+		DWORD dwCon = 0xfffffe;
+		if( RegSetValueEx( hKey, "TcpNumConnections", 0, REG_DWORD,
+				(const BYTE *) &dwCon, sizeof(dwCon) ) != ERROR_SUCCESS ) {
+			printf( "RegSetValueEx fail, errno %d\n", GetLastError() );
+		}
+		RegCloseKey( hKey );
+	}
+}
+
+int main( int argc, char * argv[] )
+{
+	int port = 3333, maxThreads = 4, maxConnections = 20000;
+	int timeout = 120, reqQueueSize = 10000;
+	const char * serverType = "lf";
+
+	extern char *optarg ;
+	int c ;
+
+	while( ( c = getopt ( argc, argv, "p:t:o:c:q:s:v" )) != EOF ) {
+		switch ( c ) {
+			case 'p' :
+				port = atoi( optarg );
+				break;
+			case 't':
+				maxThreads = atoi( optarg );
+				break;
+			case 'c':
+				maxConnections = atoi( optarg );
+				break;
+			case 'o':
+				timeout = atoi( optarg );
+				break;
+			case 'q':
+				reqQueueSize = atoi( optarg );
+				break;
+			case 's':
+				serverType = optarg;
+				break;
+			case '?' :
+			case 'v' :
+				printf( "Usage: %s [-p <port>] [-t <threads>] [-c <connections>] "
+						"[-o <timeout>] [-q <queue size>] [-s <hahs|lf>]\n", argv[0] );
+				exit( 0 );
+		}
+	}
+
+	if( 0 != sp_initsock() ) assert( 0 );
+
+	//Warning: This modifies your operating system. Use it at your own risk.
+	//IncreaseConnections();
+
+	sp_syslog( LOG_DEBUG, "server type %s", serverType );
+
+	if( 0 == strcasecmp( serverType, "hahs" ) ) {
+		SP_IocpServer server( "", port, new SP_EchoHandlerFactory() );
+		server.setTimeout( timeout );
+		server.setMaxThreads( maxThreads );
+		server.setReqQueueSize( reqQueueSize, "Byebye\r\n" );
+		server.setMaxConnections( maxConnections );
+		server.runForever();
+	} else {
+		SP_IocpLFServer server( "", port, new SP_EchoHandlerFactory() );
+		server.setTimeout( timeout );
+		server.setMaxThreads( maxThreads );
+		server.setReqQueueSize( reqQueueSize, "Byebye\r\n" );
+		server.setMaxConnections( maxConnections );
+		server.runForever();
+	}
+
+	return 0;
+}