// UDP_CE.cpp: implementation of the CUDP_CE class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "UDP.h"

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

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

CUDP_CE::CUDP_CE()
{
	
}

CUDP_CE::~CUDP_CE()
{
	
}


/*
*函數介紹：打開UDP通訊端口
*入口參數：pWnd: 指定父窗體指針 
           localPort: 指定遠程UDP端口
           romoteHost:指定遠程IP地址
           remotePort:指定遠程UDP端口
*出口參數：(無)
*返回值：1代表成功；-1，-2，-3等都代表失敗
*/
DWORD CUDP_CE::Open(CWnd* pWnd,int localPort,LPCTSTR remoteHost,int remotePort)
{
	
	WSADATA wsa;
	
	m_pOwnerWnd = pWnd;
	
	//加載winsock動態鏈接庫
	if (WSAStartup(MAKEWORD(1,1),&wsa) != 0)
	{
		return -1;//代表失敗
	}
	
	//創建UDP套接字
	m_UDPSocket = socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
	if (m_UDPSocket == INVALID_SOCKET)
	{
		return -2;
	}
	
	SOCKADDR_IN localAddr;
	localAddr.sin_family = AF_INET;
	localAddr.sin_port = htons(localPort);
	localAddr.sin_addr.s_addr=INADDR_ANY;
	
	//綁定地址
	if(bind(m_UDPSocket,(sockaddr*)&localAddr,sizeof(localAddr))!=0)
	{
		return -3;
	}
	
	
	//設置非堵塞通訊
	DWORD ul= 1;
	ioctlsocket(m_UDPSocket,FIONBIO,&ul);

	//創建一個線程退出事件
	m_ExitThreadEvent	= CreateEvent(NULL,TRUE,FALSE,NULL);
	
	//創建通訊線程
	AfxBeginThread(RecvThread,this);
	
	m_RemoteAddr.sin_family = AF_INET;
	m_RemoteAddr.sin_port = htons(remotePort);
	//此處要將雙字節轉換成單字節
	char ansiRemoteHost[255];
	ZeroMemory(ansiRemoteHost,255);
	WideCharToMultiByte(CP_ACP,WC_COMPOSITECHECK,remoteHost,wcslen(remoteHost)
		,ansiRemoteHost,wcslen(remoteHost),NULL,NULL);
	m_RemoteAddr.sin_addr.s_addr=inet_addr(ansiRemoteHost);
	return 1;
}

/*
*函數介紹：關閉UDP通訊端口
*入口參數：(無)
*出口參數：(無)
*返 回 值：1代表成功；-1，-2等都代表失敗
*/
DWORD CUDP_CE::Close(void)
{
	//設置通訊線程退出事件，通知線程退出
	SetEvent(m_ExitThreadEvent);
	Sleep(2000);
	//關閉線程句柄
	CloseHandle(m_ExitThreadEvent);
	//關閉socket
	if (closesocket(m_UDPSocket) == SOCKET_ERROR)
	{
		return -1;
	}
	
	//釋放socket資源
	if (WSACleanup() == SOCKET_ERROR)
	{
		return -2;
	}
	return 1;
}

/*
*函數介紹：發送數據
*入口參數：dataPriority: 數據優先級 
           dataType: 數據類型
           buf:緩沖區數據
           len:緩沖數據長度
*出口參數：(無)
*返回值：發送成功代表實際發送的字節數，否則返回-1
*/
bool CUDP_CE::SendData(const char *buf, int len)
{
	int nBytes = 0;
	int nErrorCode;
	nBytes = sendto(m_UDPSocket,buf,len,0,(sockaddr*)&m_RemoteAddr,sizeof(m_RemoteAddr));		

	return true;
}


/*
*函數介紹：接收線程函數
*入口參數：lparam : 指傳進線程的參數
*出口參數：(無)
*返 回 值：無意義。
*/

#define recv_size 1024*20//如果是WIN32，大

#if defined(_WIN32_WCE)//如果是WINCE
#define recv_size 1024//小
#endif

UINT CUDP_CE::RecvThread(LPVOID lparam)
{
	CUDP_CE *pSocket;
	pSocket = (CUDP_CE*)lparam;
	fd_set fdRead;
	int ret;
	TIMEVAL	aTime;
	char * recvBuf=NULL;
	aTime.tv_sec = 1;
	aTime.tv_usec = 0;
	SOCKADDR_IN tmpAddr;
	int tmpRecvLen;
	int recvLen;
	unsigned long recvBufLen;
	int iErrorCode;
	
	char * recvedBuf = NULL;
	int recvedBufLen;

	while (TRUE)
	{		
        //收到退出事件，結束線程
		if (WaitForSingleObject(pSocket->m_ExitThreadEvent,0) == WAIT_OBJECT_0)
		{
			break;
		}

		//將set初始化空集合
		FD_ZERO(&fdRead);
		
		//將pSocket->m_UDPSocket套接字添加到集合中
		FD_SET(pSocket->m_UDPSocket,&fdRead);
		
		//調用select函數，判斷套接字I/O狀態
		ret = select(0,&fdRead,NULL,NULL,&aTime);
		
		if (ret == SOCKET_ERROR)
			break;

		if (ret > 0)
		{
			if (FD_ISSET(pSocket->m_UDPSocket,&fdRead))
			{
                tmpAddr.sin_family=AF_INET;             
                tmpAddr.sin_port = htons(pSocket->m_RemoteAddr.sin_port);
				tmpAddr.sin_addr.s_addr =INADDR_ANY;
                tmpRecvLen = sizeof(tmpAddr);
				
				ioctlsocket(pSocket->m_UDPSocket,FIONREAD,&recvBufLen);
				if (recvBufLen <= 0)
					return 0;

				recvBuf = new char[recv_size];
				recvedBuf = new char[recv_size];
				ZeroMemory(recvBuf,recv_size);
				ZeroMemory(recvedBuf,recv_size);

				recvLen = recvfrom(pSocket->m_UDPSocket,recvBuf, (int)recvBufLen,0,(SOCKADDR*)&tmpAddr,&tmpRecvLen);

				
/*				if (recvLen == SOCKET_ERROR)//即使recvLen = SOCKET_ERROR，但收到的數據依然有效
					break;
				
				else if (recvLen == 0)
					break;
				 
				else
				{
*/				//此處添加解析程序，將接收到的數據解析后，



					//測試收到的數據大小
				if(recvBufLen>1)
					pSocket->m_OnUdpRecv(pSocket->m_pOwnerWnd,recvBuf,recvBufLen,(SOCKADDR*)&tmpAddr);
				delete []recvBuf;
				recvBuf = NULL;
				delete []recvedBuf;
				recvedBuf = NULL;
				//}

				recvBufLen = 0;
				ioctlsocket(pSocket->m_UDPSocket,FIONREAD,&recvBufLen);
				
			}
		}
	}
	return 0;
}