Testors.Net

#include "IAcceptor.h"
#include "XEnv.h"
#include "XUtil.h"
#include "XSyncStreamConnection.h"

#include <process.h>
#include <windows.h>

unsigned __stdcall PortProxy( void *pArg )
{
	// 새로 받은 연결
	XSyncStreamConnection source_conn( XSocket( (TSOCKET)pArg ) );

	// 목적지에 연결한다.
	XSyncStreamConnection target_conn;
	if( !target_conn.Connect( XAddr( ENV().GetString( "target_ip", "127.0.0.1" ).c_str(), ENV().GetInt( "target_port", 3389 ) ) ) ) return false;

	char buf[65535];
	while( true )
	{
		// 스트림 중계
		if( source_conn.IsReadable() )
		{
			int nLen = source_conn.Read( buf, sizeof(buf) );
			if( nLen < 1 ) break;

			if( target_conn.Write( buf, nLen ) != nLen ) break;
		}

		// 스트림 중계
		if( target_conn.IsReadable() )
		{
			int nLen = target_conn.Read( buf, sizeof(buf) );
			if( nLen < 1 ) break;

			if( source_conn.Write( buf, nLen ) != nLen ) break;
		}

		// 잠깐 쉬자.
		Sleep( 1 );
	}

	return 0;
}


int APIENTRY WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPTSTR    lpCmdLine,
                     int       nCmdShow)
{
	// 실행인수 해석
	ENV().ParseCmdLine();

	// 네트웍 초기화
	if( !XUtil::InitNetwork() )
	{
		return MessageBox( NULL, XUtil::GetWin32ErrorInfo( WSAGetLastError() ).c_str(), "Network init error!", MB_OK );
	}

	// Acceptor 초기화
	IAcceptor acceptor;
	if( !acceptor.StartAccept( ENV().GetInt( "listen_port", 6666 ) ) )
	{	
		return MessageBox( NULL, XUtil::GetWin32ErrorInfo( WSAGetLastError() ).c_str(), "Acceptor init error!", MB_OK );
	}

	XAddr peerAddr;
	XSocket newSocket;
	unsigned int nTmp;
	
	// 새 접속을 받는 메인 루프
	while( true )
	{
		// 새 접속을 받아서
		newSocket = acceptor.Accept( peerAddr );
		if( !newSocket.IsValidSocket() ) 
		{
			MessageBox( NULL, XUtil::GetWin32ErrorInfo( WSAGetLastError() ).c_str(), "Accept error!", MB_OK );
			break;
		}

		// 쓰레드에게 넘김
		_beginthreadex( NULL, 0, PortProxy, (void*)newSocket.GetSocketHandle(), 0, &nTmp );
	}

	XUtil::DeInitNetwork();

	return 0;
}