C++ 简单封装socket类包含TCP和UDP基本通信能力

takeiteasy

类名：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*****************************************