刚学c++第10天 封装了一个socket类 有待改进大神勿喷
头文件>=================================================================================#pragma once
#include<winsock2.h>
class Sockets
{
protected:
SOCKET m_hSocket;//核心成员变量
public:
BOOL Create(UINT nSocketProt = 0, int nSocketType = SOCK_STREAM,
LPCTSTR lpszSocketAddress = NULL);//端口，tcp /udp数据流或者数据包 以及socket地址
int SendTo(const void*lpBuf, int lpBuflen,UINT rSockPort, char *rSocketAddress = NULL, int nFlags = 0);//缓冲区 长度 地址
int ReceiveFrom(void*lpBuf, int lpBuflen, char *rSocketAddRess, UINT&rSockaPort, int nFlags = 0);
int Receive(void*lpBuf, int lpBuflen,int nFlage = 0)
{
return recv(m_hSocket, (LPTSTR)lpBuf, lpBuflen, nFlage);
}
void Close()
{
closesocket(m_hSocket);
m_hSocket = INVALID_SOCKET;
}
Sockets();
virtual ~Sockets();
};
源文件>=================================================================================
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include "Sockets.h"
#pragma comment(lib,"ws2_32.lib")
BOOL Sockets::Create(UINT nSocketProt, int nSocketType, LPCTSTR lpszSocketAddress)
{
m_hSocket = socket(AF_INET, nSocketType, 0);
if (m_hSocket == INVALID_SOCKET)
return FALSE;
sockaddr_in sa = {AF_INET, htons(nSocketProt)};
if(lpszSocketAddress)
sa.sin_addr.s_addr = inet_addr(lpszSocketAddress);
return !bind(m_hSocket, (sockaddr*)&sa, sizeof(sa));
}
int Sockets::SendTo(const void * lpBuf, int lpBuflen, UINTrSockPort, char * rSocketAddress, int nFlags)
{
sockaddr_in sa{ AF_INET,htons(rSockPort) };
sa.sin_addr.s_addr = inet_addr(rSocketAddress);
sendto(m_hSocket, (const char*)lpBuf, lpBuflen, 0, (sockaddr*)&sa, sizeof(sa));
return 0;
}
int Sockets::ReceiveFrom(void * lpBuf, int lpBuflen, char * rSocketAddRess, UINT & rSockaPort, int nFlags)
{
sockaddr_in sa{ AF_INET};
int nLen = sizeof(sa);
int n=recvfrom(m_hSocket, (char*)lpBuf,lpBuflen,0,(sockaddr*)&sa,&nLen);
if (n >= 0)
{
rSockaPort = htons(sa.sin_port);
strcpy(rSocketAddRess, inet_ntoa(sa.sin_addr));
}
return n;
}
Sockets::Sockets()
{
WSADATA wd;
WSAStartup(0x0202, &wd);
m_hSocket = INVALID_SOCKET;
}
Sockets::~Sockets()
{
Close();
}
页:
[1]