209 lines
5.6 KiB
C++
209 lines
5.6 KiB
C++
#ifndef CC_Socket_
|
||
#define CC_Socket_
|
||
#pragma once
|
||
|
||
#include "CC.h"
|
||
#include "CCString.h"
|
||
|
||
#define SockError -1
|
||
#define CC_Sock_Close 0
|
||
|
||
// 定义主机信息结构体
|
||
struct CCHostInfo
|
||
{
|
||
std::string IPAddress;
|
||
CCInt Port = -1;
|
||
};
|
||
|
||
// 定义套接字选项枚举
|
||
enum CCOpt
|
||
{
|
||
BROADCAST,
|
||
REUSEADDR,
|
||
REUSEPORT,
|
||
NOBLOCK,
|
||
LINGER_t,
|
||
SNDBUF,
|
||
RCVBUF,
|
||
TIMESTAMP,
|
||
};
|
||
|
||
// 定义地址类型枚举
|
||
enum CCAddress
|
||
{
|
||
Local,
|
||
Any
|
||
};
|
||
|
||
|
||
#ifdef _WIN32
|
||
|
||
#define MSL_N 0
|
||
|
||
// Windows平台下WSA初始化与清理类
|
||
class WinWSADWAInitAndClean
|
||
{
|
||
public:
|
||
WSADATA wsd{};
|
||
WinWSADWAInitAndClean(){
|
||
if (WSAStartup(MAKEWORD(2, 2), &this->wsd) != 0){
|
||
WSACleanup();
|
||
}
|
||
}
|
||
~WinWSADWAInitAndClean(){
|
||
WSACleanup();
|
||
}
|
||
};
|
||
static WinWSADWAInitAndClean winsc;
|
||
|
||
#elif __linux__
|
||
#define MSL_N MSG_NOSIGNAL
|
||
#define SOCKET_ERROR (-1)
|
||
#include <sys/socket.h>
|
||
#include <netinet/in.h>
|
||
#include <linux/if.h>
|
||
#include <unistd.h>
|
||
#include <arpa/inet.h>
|
||
#include <fcntl.h>
|
||
#include <ifaddrs.h>
|
||
#include <netdb.h>
|
||
#include <sys/select.h>
|
||
#include "poll.h"
|
||
typedef int SOCKET;
|
||
#endif
|
||
|
||
// 命名空间CTL，包含网络编程相关类和枚举
|
||
namespace CTL {
|
||
enum IPVX
|
||
{
|
||
IPV4 = AF_INET,
|
||
IPV6 = AF_INET6
|
||
};
|
||
enum TORU
|
||
{
|
||
TCP = IPPROTO_TCP,
|
||
UDP = IPPROTO_UDP
|
||
};
|
||
enum TYPE
|
||
{
|
||
STREAM = SOCK_STREAM,
|
||
DGRAM = SOCK_DGRAM
|
||
};
|
||
|
||
// Socket类，封装了网络编程的常见操作
|
||
class Socket{
|
||
public:
|
||
SOCKET Socketbit = -1;
|
||
sockaddr_in client = {}, server = {};
|
||
sockaddr_in6 client_6 = {}, server_6 = {};
|
||
short IPVx = 0;
|
||
// 默认构造函数
|
||
Socket() = default;
|
||
// 构造函数，初始化Socket
|
||
Socket(SOCKET sock);
|
||
// 初始化Socket并设置属性
|
||
bool Init(IPVX IPV4orIPV6, TORU TCPorUDP, TYPE Type = TYPE::STREAM);
|
||
// 获取Socket句柄
|
||
SOCKET GetSOCKET() const;
|
||
// 设置Socket选项
|
||
bool SetSockOpt(CCOpt opt) const;
|
||
// 设置Socket为非阻塞模式
|
||
void SetSocketNonBlocking() const;
|
||
// 检查Socket是否有可用数据
|
||
bool isDataAvailable() const;
|
||
// 获取本地IP地址
|
||
static std::vector<CTL::String> GetLocalIP(IPVX ipvx = IPVX::IPV4,int Number = 1);
|
||
// 获取客户端主机信息
|
||
CCHostInfo GetClientHost() const;
|
||
// 获取本地主机信息
|
||
CCHostInfo GetLocalHost() const;
|
||
// 连接到远程主机
|
||
bool Connect(const char* IP, unsigned short Port);
|
||
// 绑定Socket到指定地址和端口
|
||
bool Bind(const char* IP,unsigned short Port);
|
||
// 监听连接请求
|
||
bool Listen(unsigned short UserNum = 10);
|
||
// 接受连接请求
|
||
Socket Accept() const;
|
||
// 发送数据
|
||
bool Send(const char* str) const;
|
||
// 发送数据，带标志
|
||
bool Send(const char *__buf, size_t __n, int __flags) const;
|
||
// 发送数据到指定地址和端口
|
||
bool SendData(const char * str,const int length,sockaddr_in addr_in) const;
|
||
// 发送指定长度的数据
|
||
bool SendByte(const char* str,int len) const;
|
||
// 发送指定长度的数据
|
||
bool SendByte(const char* str,int len,int* ret) const;
|
||
// UDP发送数据
|
||
bool UDPSend(const char* str,const char* IP,int Port) const;
|
||
// UDP发送指定长度的数据
|
||
bool UDPSendByte(const char* str, ByteHander len, const char* IP, int Port) const;
|
||
// UDP发送指定长度的数据
|
||
bool UDPSendByte(void* str, ByteHander len, const char* IP, int Port) const;
|
||
// 接收数据
|
||
ByteHander RecvData(char* buffer,ByteHander lens) const;
|
||
// UDP接收数据
|
||
ByteHander UDPRecvData(char* buffer,ByteHander lens,CCHostInfo* info) const;
|
||
// 关闭Socket
|
||
void Close();
|
||
// 获取数据头部
|
||
static bool GetDateHead(char* data, ByteHander* size);
|
||
// 获取字符串头部
|
||
static bool GetStrHead(char* data,ByteHander size);
|
||
// 重载等号运算符，比较Socket对象是否相等
|
||
bool operator == (Socket socket) const;
|
||
// 重载不等号运算符，比较Socket对象是否不等
|
||
bool operator != (Socket socket) const;
|
||
// 检查连接是否有效
|
||
bool isConnectionAlive() const;
|
||
// 检查Socket是否可写
|
||
static bool IsSocketWritable(SOCKET sock);
|
||
static int GetLastError();
|
||
private:
|
||
int sock = -1;
|
||
bool Stop = false;
|
||
int opt = 1;
|
||
};
|
||
class InetAddress {
|
||
std::string hostAddress;
|
||
std::string hostName;
|
||
IPVX Ip_x = IPVX::IPV4;
|
||
public:
|
||
// 默认构造函数
|
||
InetAddress() = default;
|
||
// 根据IP地址构造
|
||
explicit InetAddress(const std::string& ip) : hostAddress(ip) {}
|
||
explicit InetAddress(IPVX ip_x,const std::string& ip= "");
|
||
// 根据IP地址和主机名构造
|
||
InetAddress(const std::string& ip, const std::string& hostname)
|
||
: hostAddress(ip), hostName(hostname) {}
|
||
// 获取本地主机InetAddress对象
|
||
static InetAddress getLocalHost(IPVX ipvx = IPVX::IPV4);
|
||
// 根据主机名获取InetAddress对象
|
||
static std::vector<InetAddress> getAllByName(const std::string& host);
|
||
// 根据IP地址获取InetAddress对象
|
||
static InetAddress getByName(const std::string& host);
|
||
// 获取IP地址字符串
|
||
std::string getHostAddress() const;
|
||
// 获取主机名
|
||
std::string getHostName() const;
|
||
// 获取任意地址 (相当于Java中的InetAddress.getByName("0.0.0.0"))
|
||
static InetAddress getAnyAddress();
|
||
// 获取回环地址 (相当于Java中的InetAddress.getByName("localhost"))
|
||
static InetAddress getLoopbackAddress();
|
||
// 获取广播地址
|
||
static InetAddress getBroadcastAddress(IPVX ipvx = IPVX::IPV4);
|
||
// 重载相等运算符
|
||
bool operator==(const InetAddress& other) const;
|
||
// 重载不等运算符
|
||
bool operator!=(const InetAddress& other) const;
|
||
// 转换为字符串
|
||
std::string toString() const;
|
||
IPVX getIPx() const;
|
||
};
|
||
}
|
||
|
||
|
||
#endif
|