tcpcustom_ce.cpp

EVC4.0入门程序 关于TCP测试使用工具

// TCPCustom_CE.cpp: implementation of the CTCPCustom_CE class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Step3_tcptest.h"
#include "Step3_tcptestDlg.h"
#include "TCPCustom_CE.h"
#include "TCPServer_CE.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

//构造函数
CTCPCustom_CE::CTCPCustom_CE()
{
   //创建线程退出事件
   m_exitThreadEvent = CreateEvent(NULL,FALSE,FALSE,NULL);
   m_bSocketEnable = FALSE;
}

//析构函数
CTCPCustom_CE::~CTCPCustom_CE()
{
   //关闭线程退出事件
   CloseHandle(m_exitThreadEvent);
}

/*--------------------------------------------------------------------
【函数介绍】:  此线程用于监听与客户端连接的socket通讯的事件，例如当接收到数据、
			   连接断开和通讯过程发生错误等事件
【入口参数】:  lparam:无类型指针，可以通过此参数，向线程中传入需要用到的资源。
			   在这里我们将CTCPCustom_CE类实例指针传进来
【出口参数】:  (无)
【返回  值】:  返回值没有特别的意义，在此我们将返回值设为0。
---------------------------------------------------------------------*/
DWORD CTCPCustom_CE::SocketDataThread(PVOID lparam)
{
	//CTCPCustom_CE *pSocket;
	CTCPServer_CE *pTCPServer;

	//得到CTCPCustom类实例指针
	//pSocket = (CTCPCustom_CE*)lparam;
	pTCPServer = (CTCPServer_CE*)lparam;

	//定义读事件集合
	fd_set fdRead;  
	int ret;
	TIMEVAL	aTime;
	aTime.tv_sec = 1;
	aTime.tv_usec = 0;

	//CStep3_tcptestDlg * pDlg = (CStep3_tcptestDlg*)(pSocket->m_pTCPServer_CE->m_pOwnerWnd);
	CStep3_tcptestDlg *pDlg = (CStep3_tcptestDlg*)(pTCPServer->m_pOwnerWnd);

	while (TRUE)
	{
        //收到退出事件，结束线程
		//if (WaitForSingleObject(pSocket->m_exitThreadEvent,0) == WAIT_OBJECT_0)
		if (WaitForSingleObject(pTCPServer->m_pTCPConnect->m_exitThreadEvent,0) == WAIT_OBJECT_0)
		{
			break;
		}
		//置空读事件集合
		FD_ZERO(&fdRead);
		//给pSocket设置读事件
		//FD_SET(pSocket->m_socket,&fdRead);
		FD_SET(pTCPServer->m_pTCPConnect->m_socket,&fdRead);
		//调用select函数，判断是否有读事件发生
		ret = select(0,&fdRead,NULL,NULL,&aTime);
		
		if (ret == SOCKET_ERROR)
		{
			//触发错误事件
			//pSocket->m_pTCPServer_CE->OnError(pSocket->m_pTCPServer_CE->m_pOwnerWnd, 1);
			pTCPServer->OnError(pTCPServer->m_pOwnerWnd, 1);
			//关闭socket
			//closesocket(pSocket->m_socket);
			closesocket(pTCPServer->m_pTCPConnect->m_socket);
			break;
		}
		
		if (ret > 0)
		{
			//判断是否读事件
			if (FD_ISSET(pTCPServer->m_pTCPConnect->m_socket,&fdRead))
			{
				ZeroMemory(pDlg->RecvBuf,1024);
				pDlg->RecvLen = recv(pTCPServer->m_pTCPConnect->m_socket, pDlg->RecvBuf, 1024,0); 
				if (pDlg->RecvLen == SOCKET_ERROR)
				{
					int nErrorCode = WSAGetLastError();
					//触发与客户端端连接的Socket错误
					pTCPServer->OnError(pTCPServer->m_pOwnerWnd, nErrorCode);
					//触发与客户端端连接的Socket关闭事件
					pTCPServer->OnClose(pTCPServer->m_pOwnerWnd );
					//关闭socket
					closesocket(pTCPServer->m_pTCPConnect->m_socket);
					break;

				}
				//表示连接已经从容关闭
				else if (pDlg->RecvLen == 0)
				{
					pTCPServer->OnClose(pTCPServer->m_pOwnerWnd);
					//关闭socket
					closesocket(pTCPServer->m_pTCPConnect->m_socket);
					pTCPServer->m_pTCPConnect->m_bSocketEnable = FALSE;
					break;
				}
				else
				{
				   //触发与客户端端连接的Socket读事件
                   pTCPServer->OnRead(pTCPServer->m_pOwnerWnd );    //,recvBuf,recvLen);
				}
			}
		}
	}
	return 0;
}

/*--------------------------------------------------------------------
【函数介绍】: 打开socket，创建通讯线程
【入口参数】:  pTCPServer指向服务器端监听socket
【出口参数】:  (无)
【返回  值】:  TRUE:打开成功;FALSE:打开失败
---------------------------------------------------------------------*/
//bool CTCPCustom_CE::Open(CTCPServer_CE *pTCPServer)
bool CTCPCustom_CE::Open(PVOID lparam)
{
	CTCPServer_CE *pTCPServer;

	pTCPServer = (CTCPServer_CE*)lparam;

	//清结束事件
	ResetEvent(m_exitThreadEvent);

	//创建通讯线程, 把侦听服务器类指针带入线程
	//m_tcpThreadHandle = CreateThread(NULL, 0, SocketDataThread, this,0,NULL);
	m_tcpThreadHandle = CreateThread(NULL, 0, SocketDataThread, pTCPServer, 0, NULL);
	if (m_tcpThreadHandle == NULL)
	{
	   closesocket(m_socket);
	   return FALSE;
	}
	//设置通讯模式为异步模式
	DWORD ul= 1;
	ioctlsocket(m_socket,FIONBIO,&ul);
	//m_pTCPServer_CE = pTCPServer;
	m_bSocketEnable = TRUE;
	return TRUE;
}

/*--------------------------------------------------------------------
【函数介绍】: 关闭socket，关闭线程，释放Socket资源
【入口参数】:  (无)
【出口参数】:  (无)
【返回  值】:  TRUE:成功关闭;FALSE:关闭失败
---------------------------------------------------------------------*/
bool CTCPCustom_CE::Close()
{
   //生成结束事件，通知数据线程结束
   SetEvent(m_exitThreadEvent);
   //等待1s，确保数据线程结束
   Sleep(1000);
	
   //关闭Socket，释放资源
   int err = closesocket(m_socket);
   if (err == SOCKET_ERROR)
   {
	   return FALSE;
   }
   return TRUE;
}


/*-----------------------------------------------------------------
【函数介绍】: 向客户端发送数据
【入口参数】: buf: 待发送的数据
              len: 待发送的数据长度
【出口参数】: (无)
【返回  值】: >0: 发送数据成功, 返回所发送的数据长度; 
              <0: 发送数据失败。
------------------------------------------------------------------*/
int CTCPCustom_CE::SendData( const char *buf , int len )
{
	int nBytes = 0;
	int nSendBytes=0;

	while (nSendBytes < len)
	{
	    nBytes = send(m_socket,buf+nSendBytes,len-nSendBytes,0);
		if (nBytes==SOCKET_ERROR )
		{
			return nBytes;
		}

		nSendBytes = nSendBytes + nBytes;
		
		if (nSendBytes < len)
		{
		    Sleep(1000);
		}
	} 
	return nSendBytes; 
}