// SerMain.cpp : Defines the entry point for the console application.

//

#include "Socket.h"

using namespace mdk;

#include <map>

using namespace std;

typedef struct THREAD_CONTROL

{

DWORD dwID;

bool bCheck;

}THREAD_CONTROL;

map<DWORD,THREAD_CONTROL*> g_activeThread;

DWORD TMainThread( LPVOID pParam );//主线程入口

UINT uPort;

char strIP[256];

time_t g_lastdata = 0;

DWORD Client( LPVOID pParam )

{

DWORD dwThreadID;

Socket sock;

sock.Init( Socket::tcp );

printf( "connect(%s:%d)\n", strIP, uPort );

if ( !sock.Connect( strIP, uPort ) )

{

printf( "connect error\n" );

return 0;

}

HANDLE hThread = CreateThread(

NULL, 0,

(LPTHREAD_START_ROUTINE)TMainThread,

(LPVOID)&sock, 0, &dwThreadID );

if ( NULL == hThread )

{

sock.Close();

printf( "create thread error\n" );

return 0;

}

short nSendSize;

char buf[256];

short len = 20;

memcpy( buf, &len, 2 );

memcpy( &buf[2], "01234567890123456789", len );

len += 2;

DWORD dwID = GetCurrentThreadId();

while ( 1 )

{

nSendSize = sock.Send( buf, len );

if ( -1 == nSendSize )

{

printf( "线程(%d)退出:发送失败\n", dwID );

return 0;

}

Sleep( 100 );

continue;

}

sock.Close();

return 0;

}

DWORD TMainThread( LPVOID pParam )//主线程入口

{

static int nID = 0;

DWORD dwID = GetCurrentThreadId();

THREAD_CONTROL *pThread = new THREAD_CONTROL;

pThread->dwID = dwID;

pThread->bCheck = false;

g_activeThread.insert( map<DWORD,THREAD_CONTROL*>::value_type( dwID, pThread ) );

char buf[256];

short nLen;

Socket *pSock = (Socket*)pParam;

while ( 1 )

{

nLen = pSock->Receive( buf, 2 );

if ( nLen <= 0 )

{

map<DWORD,THREAD_CONTROL*>::iterator it = g_activeThread.find( dwID );

if ( it != g_activeThread.end() ) g_activeThread.erase( it );

printf( "线程(%d)退出:服务器断开连接\n", dwID );

return 0;

}

memcpy( &nLen, buf, 2 );

nLen = pSock->Receive( buf, nLen );

if ( nLen <= 0 )

{

map<DWORD,THREAD_CONTROL*>::iterator it = g_activeThread.find( dwID );

if ( it != g_activeThread.end() ) g_activeThread.erase( it );

printf( "线程(%d)退出:服务器断开连接\n", dwID );

return 0;

}

g_lastdata = time(NULL);

if ( pThread->bCheck )

{

buf[nLen] = 0;

printf( buf );

}

}

}

int main(int argc, char* argv[])

{

uPort = GetPrivateProfileInt( "Server", "Port", 8080, ".\\cli.ini" );

DWORD dwSize = 255;

GetPrivateProfileString( "Server", "IP", "127.0.0.1", strIP, dwSize, ".\\cli.ini" );

setvbuf( stdout, (char*)NULL, _IONBF, 0 ); //added

Socket::SocketInit();

// Client( NULL );

DWORD dwThreadID;

int i = 1;

int nThreadCount = 0;

char sCount[5];

sCount[0]= getchar();

sCount[1]= getchar();

sCount[2]= getchar();

getchar();

sCount[3]= 0;

nThreadCount = atoi( sCount );

// nThreadCount = 999;

printf( "创建%d个client 按任意键开始\n", nThreadCount );

for ( i = 1; i <= nThreadCount; i++ ){

HANDLE hThread = CreateThread(

NULL, 0,

(LPTHREAD_START_ROUTINE)Client,

(LPVOID)NULL, 0, &dwThreadID );

if ( NULL == hThread )

{

printf( "create client(%d) error\n", i );

}

if ( 0 == i % 200 )

{

printf( "create%d\n", i );

// getchar();

}

}

char c[100];

int nPos;

THREAD_CONTROL* pThead;

map<DWORD, THREAD_CONTROL*>::iterator it;

int nCount;

while ( 1 ){

c[0] = getchar();

c[1] = getchar();

c[2] = getchar();

c[3] = getchar();

c[3] = 0;

nPos = atoi( c );

nCount = g_activeThread.size();

if ( nPos > nCount )

{

printf( "创建线程%d/%d\n", nPos, nCount );

HANDLE hThread = CreateThread(

NULL, 0,

(LPTHREAD_START_ROUTINE)Client,

(LPVOID)NULL, 0, &dwThreadID );

}

else printf( "检查线程%d/%d\n", nPos, nCount );

tm *last = localtime(&g_lastdata);

char stime[256];

strftime( stime, 256, "%Y-%m-%d %H:%M:%S", last );

printf( "last data:%s\n", stime );

Sleep( 1000 );

it = g_activeThread.begin();

for ( i = 1; it != g_activeThread.end(); it++, i++ )

{

pThead = it->second;

if ( i == nPos ) pThead->bCheck = true;

else pThead->bCheck = false;

}

Sleep( 1000 );

}

return 0;

}