類名：TSocket
文件包含：TSocket.cpp、TSocket.h
使用方法：
1、在對話框頭文件包含
TSocket.h
#include "TSocket.h"
2、定義變量
TSocket m_TSocket;
3、TSocket類函數功能介紹：
LoadSocket()：加載SOCKET庫
CreateSocketSer（TRUE）：創建SOCKET，TRUE為TCP，FALSE為UDP
BingSocket()：綁定IP和端口，有默認值
Start(precvPar)：precvPar為一個RECVPARAM結構體指針
SendData(strSend)：發送數據函數
4、服務端處理接收到的數據（消息映射、路由）
afx_msg void OnRecvData(WPARAM wParam,LPARAM lParam);
ON_MESSAGE(WM_RECVDATA,OnRecvData)
void CXXXDlg::OnRecvData( WPARAM wParam,LPARAM lParam )
{
CString strrecv = (char*)lParam;//得到消息內容
}
5、穩定性不能保證，僅供參考socket通信原理。。。。。。
****************************************
TSocket.h*****************************************
// TSocket.h: interface for the TSocket class.
//
//////////////////////////////////////////////////////////////////////
#if !defined(AFX_TSOCKET_H__ECFF7A02_DCAF_455D_97C3_0C1D465D977B__INCLUDED_)
#define AFX_TSOCKET_H__ECFF7A02_DCAF_455D_97C3_0C1D465D977B__INCLUDED_
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include
#pragma comment(lib,"ws2_32.lib")
#define WM_RECVDATA WM_USER+500 //定義接收到數據發出的消息號
#define MAX_MSG_LEN 1500 //最大的消息長度(MTU)
#define DEFAULTPORT 1991 //定義默認服務端口
class TSocket;
struct RECVPARAM
{
TSocket* psocket;//指定創建的socket
HWND hwnd;//指定需要處理接收消息的窗口句柄
};
class TSocket
{
public:
TSocket();
virtual ~TSocket();
public:
//加載Winsock庫
BOOL LoadSocket();
BOOL CreateSocketSer(BOOL bMode);//創建socket用於服務器端
SOCKET CreateSocket(BOOL bMode);//創建socket用於客戶端
BOOL BingSocket(u_long ulIP=0,u_short usPort=DEFAULTPORT);//綁定端口、IP(默認所有IP)
BOOL Start(RECVPARAM* recvPar);
BOOL Stop();
SOCKET GetSocket();//得到當前socket
CString m_strData;//發出去的數據
BOOL m_bIsRun; //是否繼續運行
u_long m_ulLocalIP;//本地IP
u_short m_usLocalPort;//本地端口
u_long m_ulRemoteIP;//遠程IP
u_short m_usRemotePort;//遠程端口
//發送數據
DWORD SendData(CString strSend);
//TCP發送線程
static DWORD WINAPI TcpSendProc(LPVOID lpParameter);
//UDP發送線程
static DWORD WINAPI UdpSendProc(LPVOID lpParameter);
//TCP發送數據
DWORD TcpSend();
//UDP發送數據
DWORD UdpSend();
private:
BOOL m_bMode; //通信方式 1--TCP 0--UDP
SOCKET m_socket;//套接字
HANDLE m_hRecv; //接收線程的句柄
//TCP接收線程
static DWORD WINAPI TcpRecvProc(LPVOID lpParameter);
//UDP接收線程
static DWORD WINAPI UdpRecvProc(LPVOID lpParameter);
};
#endif // !defined(AFX_TSOCKET_H__ECFF7A02_DCAF_455D_97C3_0C1D465D977B__INCLUDED_)
****************************************
TSocket.h
*****************************************
****************************************
TSocket.
cpp
*****************************************
// TSocket.cpp: implementation of the TSocket class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "TSocket.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
TSocket::TSocket()
{
m_bMode = FALSE;//UDP
m_ulRemoteIP = 0;
m_ulLocalIP = 0;
}
TSocket::~TSocket()
{
//釋放Winsock庫
WSACleanup();
}
BOOL TSocket::LoadSocket()
{
//加載Winsock庫
WSADATA wsa;
if (WSAStartup(MAKEWORD(1,1),&wsa) != 0)
{
return FALSE;
//加載失敗
}
return TRUE;
}
BOOL TSocket::CreateSocketSer(BOOL bMode)
{
m_bMode = bMode;
//創建SOCKET
if (m_bMode)
{
m_socket = socket(AF_INET,SOCK_STREAM,0);//TCP流式
}else
{
m_socket = socket(AF_INET,SOCK_DGRAM,0);//UDP方式
}
return TRUE;
}
SOCKET TSocket::CreateSocket( BOOL bMode )
{
m_bMode = bMode;
SOCKET socketCr;
//創建SOCKET
if (m_bMode)
{
socketCr = socket(AF_INET,SOCK_STREAM,0);//TCP方式
}else
{
socketCr = socket(AF_INET,SOCK_DGRAM,0);//UDP方式
}
return socketCr;
}
BOOL TSocket::BingSocket(u_long ulIP,u_short usPort)
{
//配置監聽地址、綁定監聽端口
SOCKADDR_IN Sersock;//用於服務器的監聽SOCKET
ZeroMemory(&Sersock,sizeof(Sersock));
if(ulIP==0)
{
Sersock.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
}else
{
Sersock.sin_addr.S_un.S_addr = htonl(ulIP);//IP
}
Sersock.sin_family = AF_INET;
Sersock.sin_port = htons(usPort);//端口號
int nbind = bind(m_socket,(SOCKADDR *)&Sersock,sizeof(SOCKADDR));
if (SOCKET_ERROR==nbind)
{
closesocket(m_socket);
return FALSE;
}
return TRUE;
}
DWORD WINAPI TSocket::TcpRecvProc( LPVOID lpParameter )
{
TSocket* psocket = ((RECVPARAM *)lpParameter)->psocket;
HWND hwnd = ((RECVPARAM *)lpParameter)->hwnd;
delete lpParameter;//釋放內存
listen(psocket->m_socket,5);
SOCKADDR_IN addrClient;
int len=sizeof(SOCKADDR);
char szrecvbuf[MAX_MSG_LEN];
char szsendBuf[50];
ZeroMemory(szrecvbuf,sizeof(szrecvbuf));
ZeroMemory(szsendBuf,sizeof(szsendBuf));
SOCKET sockSer;
while(psocket->m_bIsRun)
{
//用accept函數接收客戶方的連接
sockSer=accept(psocket->m_socket,(SOCKADDR*)&addrClient,&len);
if (INVALID_SOCKET==sockSer)
continue;
sprintf(szsendBuf,"Welcome %s to here!",inet_ntoa(addrClient.sin_addr));
//在客戶方套接字clientSocket上使用send發送數據
send(sockSer,szsendBuf,strlen(szsendBuf)+1,0);
//接收的數據
recv(sockSer,szrecvbuf,50,0);
psocket->m_strData.Format(_T("IP:%s,端口:%d 說：%s"),
inet_ntoa(addrClient.sin_addr),addrClient.sin_port,szrecvbuf);
::PostMessage(hwnd,WM_RECVDATA,0,(LPARAM)psocket->m_strData.GetBuffer(0));
//關閉本次連接
closesocket(sockSer);
//休息100S
Sleep(100);
}
//關閉socket
closesocket(psocket->m_socket);
return 0;
}
DWORD WINAPI TSocket::UdpRecvProc( LPVOID lpParameter )
{
TSocket* psocket = ((RECVPARAM *)lpParameter)->psocket;
HWND hwnd = ((RECVPARAM *)lpParameter)->hwnd;
delete lpParameter;//釋放內存
SOCKADDR_IN sockRecv;
int nlen = sizeof(SOCKADDR);
char szrecvbuf[MAX_MSG_LEN];
ZeroMemory(szrecvbuf,sizeof(szrecvbuf));
int nrecv;
while (psocket->m_bIsRun)
{
nrecv=recvfrom(psocket->m_socket,szrecvbuf,sizeof(szrecvbuf),0,(SOCKADDR *)&sockRecv,&nlen);
if (SOCKET_ERROR == nrecv)
break;
psocket->m_strData.Format(_T("IP:%s,端口:%d 說：%s"),
inet_ntoa(sockRecv.sin_addr),sockRecv.sin_port,szrecvbuf);
::PostMessage(hwnd,WM_RECVDATA,0,(LPARAM)psocket->m_strData.GetBuffer(0));
//休息100S
Sleep(1000);
}
//關閉socket
closesocket(psocket->m_socket);
CloseHandle(psocket->m_hRecv);
return 0;
}
BOOL TSocket::Start( RECVPARAM* recvPar )
{
m_bIsRun = TRUE;
//創建接收消息的線程
if (m_bMode)
{
m_hRecv = CreateThread(NULL,0,TcpRecvProc,(LPVOID)recvPar,0,NULL);
}else
{
m_hRecv = CreateThread(NULL,0,UdpRecvProc,(LPVOID)recvPar,0,NULL);
}
return TRUE;
}
BOOL TSocket::Stop()
{
//Kill線程？？？
if (m_hRecv!=NULL)
{
TerminateThread(m_hRecv,0);
//關閉socket
closesocket(m_socket);
}
m_bIsRun = FALSE;
return TRUE;
}
DWORD TSocket::TcpSend()
{
if (m_ulRemoteIP==0)
return -1;
SOCKET SocketSend;
SocketSend=CreateSocket(TRUE);
if (SocketSend==INVALID_SOCKET)
{
return -1;
}
SOCKADDR_IN SocketSendIn;
SocketSendIn.sin_family = AF_INET;
SocketSendIn.sin_addr.S_un.S_addr = htonl(m_ulRemoteIP);
SocketSendIn.sin_port = htons(m_usRemotePort);
char szBuf[50];
//先建立連接
int nRet = connect(SocketSend,(SOCKADDR*)&SocketSendIn,sizeof(SOCKADDR));
if (nRet==SOCKET_ERROR)
{
return -1;
}
nRet = recv(SocketSend,szBuf,sizeof(szBuf),0);
if (nRet==0)//超時
{
return -2;
}
//開始發送數據
send(SocketSend,m_strData.GetBuffer(0),m_strData.GetLength(),0);
closesocket(SocketSend);
return 0;
}
DWORD TSocket::UdpSend()
{
if (m_ulRemoteIP==0)
return -1;
SOCKET SocketSend;
SocketSend=CreateSocket(FALSE);
if (SocketSend==INVALID_SOCKET)
{
return -1;
}
SOCKADDR_IN SocketSendIn;
SocketSendIn.sin_family = AF_INET;
SocketSendIn.sin_addr.S_un.S_addr = htonl(m_ulRemoteIP);
SocketSendIn.sin_port = htons(m_usRemotePort);
int nSenlen=m_strData.GetLength()+1;
sendto(SocketSend,m_strData,nSenlen,0,(SOCKADDR *)&SocketSendIn,sizeof(SOCKADDR));
closesocket(SocketSend);
return 0;
}
DWORD TSocket::SendData( CString strSend )
{
//保存數據
m_strData = strSend;
if (m_bMode)
{
CreateThread(NULL,0,TcpSendProc,(LPVOID)this,0,NULL);
}else
{
CreateThread(NULL,0,UdpSendProc,(LPVOID)this,0,NULL);
}
return 0;
}
SOCKET TSocket::GetSocket()
{
return m_socket;
}
DWORD WINAPI TSocket::TcpSendProc( LPVOID lpParameter )
{
TSocket* pThis = (TSocket*)lpParameter;
return pThis->TcpSend();
}
DWORD WINAPI TSocket::UdpSendProc( LPVOID lpParameter )
{
TSocket* pThis = (TSocket*)lpParameter;
return pThis->UdpSend();
}
****************************************
TSocket.
cpp
*************