CSocket

CSocket.h头文件

#pragma once

#include<WinSock2.h>

#define zjlUINT unsigned int
#define zjlINT  int
#define zjlPCHAR char*
#define zjlPCCHAR const char*
#define zjlPVOID  void*
#define zjlPCVOID const void*

#define zjlNO         -1
#define zjlYES        0
#define zjlSOCK_FALSE 1    //创建套接字失败
#define zjlBIND_FALSE 2    //绑定端口失败
#define zjlCON_FALSE  3    //连接失败

class Socket
{
private:
	SOCKET m_sock;
public:
	Socket();
	~Socket();
	//创建socket套接字 
	BOOL Create(zjlUINT nSocketPort = 0, zjlINT nSocketType = SOCK_STREAM,
		zjlPCHAR SocketAddress = NULL); 
	BOOL Accept(Socket &sock, zjlPCHAR IP = NULL, zjlUINT *nPort = NULL);
	BOOL Connect(zjlPCHAR SocketAddress, zjlUINT nHostPort);
	zjlINT Listen(zjlINT nConnectBacklog = 5);
	//发送 
	zjlINT SendTo(zjlPCVOID Buff, zjlINT nBuffLen, zjlUINT nPort,
		zjlPCHAR HostAddress = NULL, int flag = 0);
	zjlINT Send(zjlPCVOID Buff, zjlINT nBuffLen, zjlINT flags = 0);
	//接收 
	zjlINT RecvFrom(zjlPCVOID Buff, zjlINT nBuffLen, zjlPCHAR SocketAddress,
		zjlUINT SocketPort, int nFlags = 0);
	zjlINT Recv(zjlPVOID Buff, zjlINT nLen, zjlINT flag = 0);
	BOOL GetPeerName(zjlPCHAR SocketAddress, zjlUINT &SocketPort);
	BOOL GetSocketName(zjlPCHAR SocketAddress, zjlUINT &SocketPort);
	void close();
};

CSocket.cpp文件

#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include "Socket.h"

Socket::Socket()
{
	WSADATA data;
	WSAStartup(0x0202, &data);
	m_sock = INVALID_SOCKET;
}

Socket::~Socket()
{
	close();
}

BOOL Socket::Create(zjlUINT nSocketPort, zjlINT nSocketType, zjlPCHAR SocketAddress)
{
	m_sock = socket(AF_INET, nSocketType, 0);
	if (INVALID_SOCKET == m_sock)
		return zjlSOCK_FALSE;
	sockaddr_in sa = { AF_INET,htons(nSocketPort) };
	if (SocketAddress)
		sa.sin_addr.S_un.S_addr = inet_addr(SocketAddress);
	int b = bind(m_sock, (sockaddr*)&sa, sizeof(sa));
	if (SOCKET_ERROR == b)
		return zjlBIND_FALSE;
	return zjlYES;
}

BOOL Socket::Accept(Socket & sock, zjlPCHAR IP, zjlUINT * nPort)
{
	sockaddr_in sa = { AF_INET };
	zjlINT nLen = sizeof(sa);
	sock.m_sock = accept(m_sock, (sockaddr*)&sa, &nLen);
	if (INVALID_SOCKET == sock.m_sock)
		return zjlSOCK_FALSE;
	if (IP)
		sa.sin_addr.S_un.S_addr = inet_addr(IP);
	if (nPort)
		*nPort = htons(sa.sin_port);
	return zjlYES;
}

BOOL Socket::Connect(zjlPCHAR SocketAddress, zjlUINT nHostPort)
{
	sockaddr_in sa = { AF_INET,htons(nHostPort) };
	sa.sin_addr.S_un.S_addr = inet_addr(SocketAddress);
	int n = connect(m_sock, (sockaddr*)&sa, sizeof(sa));
	if (!n)
		return zjlCON_FALSE;
	return zjlYES;
}

zjlINT Socket::Listen(zjlINT nConnectBacklog)
{
	return listen(m_sock, nConnectBacklog);
}

zjlINT Socket::SendTo(zjlPCVOID Buff, zjlINT nBuffLen, zjlUINT nPort, 
	zjlPCHAR HostAddress, int flag)
{
	sockaddr_in sa = { AF_INET };
	sendto(m_sock, (const char*)Buff, nBuffLen, 0, (sockaddr*)&sa, sizeof(sa));
	strcpy(HostAddress, inet_ntoa(sa.sin_addr));
	nPort = htons(sa.sin_port);
	return zjlYES;
}

zjlINT Socket::RecvFrom(zjlPCVOID Buff, zjlINT nBuffLen, zjlPCHAR SocketAddress,
	zjlUINT SocketPort, int nFlags)
{
	sockaddr_in sa = { AF_INET };
	zjlINT nLen = sizeof(sa);
	int nResult = recvfrom(m_sock, (char*)Buff, nBuffLen, 0, (sockaddr*)&sa, &nLen);
	if (0 >= nResult)
		return nResult;
	SocketPort = htons(sa.sin_port);
	strcpy(SocketAddress, inet_ntoa(sa.sin_addr));
	return nResult;
}

zjlINT Socket::Send(zjlPCVOID Buff, zjlINT nBuffLen, zjlINT flags)
{
	return send(m_sock, (zjlPCHAR)Buff, nBuffLen, flags);
}

zjlINT Socket::Recv(zjlPVOID Buff, zjlINT nLen, zjlINT flag)
{
	return recv(m_sock, (zjlPCHAR)Buff, nLen, flag);
}

BOOL Socket::GetPeerName(zjlPCHAR SocketAddress, zjlUINT & SocketPort)
{
	sockaddr_in sa = { AF_INET };
	zjlINT nLen = sizeof(sa);
	if (0 > getpeername(m_sock, (sockaddr*)&sa, &nLen))
		return zjlNO;
	SocketPort = htons(sa.sin_port);
	strcpy(SocketAddress, inet_ntoa(sa.sin_addr));
	return zjlYES;
}

BOOL Socket::GetSocketName(zjlPCHAR SocketAddress, zjlUINT & SocketPort)
{
	sockaddr_in sa = { AF_INET };
	zjlINT nLen = sizeof(sa);
	if (0 > getsockname(m_sock, (sockaddr*)&sa, &nLen))
		return zjlNO;
	SocketPort = htons(sa.sin_port);
	strcpy(SocketAddress, inet_ntoa(sa.sin_addr));
	return zjlYES;
}

void Socket::close()
{
	closesocket(m_sock);
	m_sock = INVALID_SOCKET;
}