socket网络编程/CSocket的简单封装




MyCSocket.h


#pragma once
#include<WinSock2.h>
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS

class MyCSocket
{
protected:
	SOCKET m_Socket;
public:
	MyCSocket();
	~MyCSocket();
	BOOL Create(UINT nSocketPort = 0, int nSocketType = SOCK_STREAM, LPCTSTR lpszSocketAddress = NULL);//创建Socket
	BOOL Bind(UINT nSocketPort, LPCTSTR lpszSocketAddress = NULL);//绑定
	BOOL Listen(int nConnectionBacklog = 5);//TCP监听
	BOOL Connect(LPCTSTR lpszHostAddress,UINT nHostPort);//TCP客户端需要连接
	BOOL Accept(MyCSocket& rConnectedSocket, LPSTR lpSockAddr = NULL, UINT *nPort = NULL);//TCP服务端需要接受连接
	int Send(const void* lpBuf,int nBufLen,int nFlags = 0);//TCP发送消息
	int SendTo(const void * lpBuff, int nBuffLen, UINT nSendPort, LPCTSTR lpszSendIPAddress = NULL, int nFlags = 0);//UDP发送消息
	int Receive(void* lpBuf,int nBufLen,int nFlags = 0);//TCP接收消息
	int ReceiveFrom(void* lpBuff, int nBuffLen, char* rRecvIPAddress, UINT& nRecvPort,int nFlags=0);//UDP接收消息
	BOOL GetSockName(char*  rSocketAddress, UINT& rSocketPort);
	BOOL GetPeerName(char* rPeerAddress, UINT& rPeerPort);
	void close();

};

MyCSocket.cpp

#include "MyCSocket.h"
#pragma comment(lib,"ws2_32.lib")

/*****************************************************************
	函数名:		MyCSocket()
	参数:		无
	返回值:		无
	功能:		启动socket，默认使用版本2.2
	调用函数:	WSAStartup()
******************************************************************/
MyCSocket::MyCSocket()
{
	WSAData wsaData;
	WSAStartup(0x0202, &wsaData);
}

/*****************************************************************
	函数名:		~MyCSocket()
	参数:		无
	返回值:		无
	功能:		
	调用函数:	close()
******************************************************************/
MyCSocket::~MyCSocket()
{
	close();
}

/*****************************************************************
	函数名:		Create()
	参数:		nSocketPort
			nSocketType
			lpszSocketAddress
	返回值:		BOOL
	功能:		创建socket，默认协议使用TCP
	调用函数:	socket()
******************************************************************/
BOOL MyCSocket::Create(UINT nSocketPort, int nSocketType, LPCTSTR lpszSocketAddress)
{
	//创建socket
	m_Socket = socket(AF_INET, nSocketType, 0);
	if (INVALID_SOCKET == m_Socket)
		return FALSE;
	return TRUE;
}

/*****************************************************************
	函数名:		Bind()
	参数:		nSocketPort
			lpszSocketAddress
	返回值:		BOOL
	功能:		初始化协议簇、绑定socket
	调用函数:	bind()
******************************************************************/
BOOL MyCSocket::Bind(UINT nSocketPort, LPCTSTR lpszSocketAddress)
{
	//初始化sockaddr_in
	sockaddr_in MySockAddr = { AF_INET,htons(nSocketPort) };
	if (lpszSocketAddress)
		MySockAddr.sin_addr.S_un.S_addr = inet_addr(lpszSocketAddress);
	else
		MySockAddr.sin_addr.S_un.S_addr = INADDR_ANY;
	//绑定bind
	return !(bind(m_Socket, (sockaddr*)&MySockAddr, sizeof(MySockAddr)));
}

/*****************************************************************
	函数名:		Listen()
	参数:		nConnectionBacklog
	返回值:		BOOL
	功能:		监听连接，默认个数为5
	调用函数:	listen()
******************************************************************/
BOOL MyCSocket::Listen(int nConnectionBacklog)
{
	return !(listen(m_Socket, nConnectionBacklog));
}

/*****************************************************************
	函数名:		Connect()
	参数:		lpszHostAddress
			nHostPort
	返回值:		BOOL
	功能:		服务器需要接受连接，而客户端需要连接,该函数为客户端使用
	调用函数:	connect()
******************************************************************/
BOOL MyCSocket::Connect(LPCTSTR lpszHostAddress, UINT nHostPort)
{
	sockaddr_in sockAddr = {};
	sockAddr.sin_family = AF_INET;
	sockAddr.sin_addr.S_un.S_addr = inet_addr(lpszHostAddress);
	sockAddr.sin_port = htons(nHostPort);
	return !connect(m_Socket, (sockaddr*)&sockAddr, sizeof(sockAddr));
}

/*****************************************************************
	函数名:		Accept()
	参数:		rConnectedSocket
			lpSockAddr
			nPort
	返回值:		BOOL
	功能:		服务器需要接受连接，而客户端需要连接，该函数为服务器使用
	调用函数:	accept()
******************************************************************/
BOOL MyCSocket::Accept(MyCSocket & rConnectedSocket, LPSTR lpSockAddr, UINT *nPort)
{
	sockaddr_in AcceptSockAddr = { AF_INET };
	int AcceptSockAddrLen = sizeof(AcceptSockAddr);
	rConnectedSocket.m_Socket = accept(this->m_Socket, (sockaddr*)&AcceptSockAddr, &AcceptSockAddrLen);
	if (INVALID_SOCKET == rConnectedSocket.m_Socket)
		return FALSE;
	if (lpSockAddr)
		strcpy(lpSockAddr, inet_ntoa(AcceptSockAddr.sin_addr));
	if (nPort)
		*nPort = htons(AcceptSockAddr.sin_port);
	return TRUE;
}

/*****************************************************************
	函数名:		Send()
	参数:		lpBuf
			nBufLen
			nFlags = 0
	返回值:		int
	功能:		TCP发送消息
	调用函数:	send()
******************************************************************/
int MyCSocket::Send(const void * lpBuf, int nBufLen, int nFlags)
{
	return send(m_Socket, (LPCSTR)lpBuf, nBufLen, nFlags);
}

/*****************************************************************
	函数名:		SendTo()
	参数:		lpBuf
			nBufLen
			nSendPort
			lpszSendIPAddress
			nFlags = 0
	返回值:		int
	功能:		UDP发送消息
	调用函数:	sendto()
******************************************************************/
int MyCSocket::SendTo(const void * lpBuff, int nBuffLen, UINT nSendPort, LPCTSTR lpszSendIPAddress, int nFlags)
{
	sockaddr_in SendToSockAddr = { AF_INET,htons(nSendPort) };
	SendToSockAddr.sin_addr.S_un.S_addr = inet_addr(lpszSendIPAddress);
	return sendto(m_Socket, (const char*)lpBuff, nBuffLen, nFlags, (sockaddr*)&SendToSockAddr, sizeof(SendToSockAddr));
}

/*****************************************************************
	函数名:		Receive()
	参数:		lpBuf
			nBufLen
			nFlags = 0
	返回值:		int
	功能:		TCP接收消息
	调用函数:	recv()
******************************************************************/
int MyCSocket::Receive(void * lpBuf, int nBufLen, int nFlags)
{
	return recv(m_Socket, (LPSTR)lpBuf, nBufLen, nFlags);
}

/*****************************************************************
	函数名:		ReceiveFrom()
	参数:		lpBuf
			nBufLen
			rRecvIPAddress
			nRecvPort
			nFlags = 0
	返回值:		int
	功能:		UDP接收消息
	调用函数:	recvfrom()
******************************************************************/
int MyCSocket::ReceiveFrom(void * lpBuff, int nBuffLen, char * rRecvIPAddress, UINT & nRecvPort, int nFlags)
{
	sockaddr_in RecvFromSockAddr = { AF_INET };
	int nSockAddrLen = sizeof(RecvFromSockAddr);
	int nDataLen = recvfrom(m_Socket, (char*)lpBuff, nBuffLen, nFlags, (sockaddr*)&RecvFromSockAddr, &nSockAddrLen);
	if (nDataLen > 0)
	{
		strcpy(rRecvIPAddress, inet_ntoa(RecvFromSockAddr.sin_addr));
		nRecvPort = htons(RecvFromSockAddr.sin_port);
	}
	return 0;
}

/*****************************************************************
	函数名:		Close()
	参数:		无
	返回值:		void
	功能:		关闭socket
	调用函数:	closesocket()
******************************************************************/
void MyCSocket::close()
{
	closesocket(m_Socket);
	WSACleanup();
}