MyLib
This commit is contained in:
51
Socket/NetworkRelated.h
Normal file
51
Socket/NetworkRelated.h
Normal file
@ -0,0 +1,51 @@
|
||||
#ifndef LXZL_NETWORKRELATED_H
|
||||
#define LXZL_NETWORKRELATED_H
|
||||
|
||||
#include <iostream>
|
||||
// 平台检测
|
||||
#ifdef _WIN32
|
||||
#include <winsock2.h>
|
||||
#include <ws2tcpip.h>
|
||||
#else
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <sys/ioctl.h>
|
||||
#endif
|
||||
|
||||
// 平台兼容类型定义
|
||||
#ifdef _WIN32
|
||||
using socklen_t = int;
|
||||
using socket_t = SOCKET;
|
||||
constexpr socket_t INVALID_SOCKET_VALUE = INVALID_SOCKET;
|
||||
#else
|
||||
using socket_t = int;
|
||||
constexpr socket_t INVALID_SOCKET_VALUE = -1;
|
||||
#define SOCKET_ERROR (-1)
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
// Windows平台初始化
|
||||
#ifdef _WIN32
|
||||
|
||||
class WSAInitializer {
|
||||
public:
|
||||
WSAInitializer() {
|
||||
WSADATA wsaData;
|
||||
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
|
||||
throw std::runtime_error("WSAStartup failed");
|
||||
}
|
||||
}
|
||||
|
||||
~WSAInitializer() {
|
||||
WSACleanup();
|
||||
}
|
||||
};
|
||||
|
||||
static WSAInitializer wsa_initializer;
|
||||
#endif
|
||||
|
||||
#endif //LXZL_NETWORKRELATED_H
|
||||
156
Socket/TCP.cpp
Normal file
156
Socket/TCP.cpp
Normal file
@ -0,0 +1,156 @@
|
||||
#include "TCP.h"
|
||||
|
||||
void TCPSocket::create_socket() {
|
||||
sockfd_ = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (sockfd_ == INVALID_SOCKET_VALUE) {
|
||||
throw std::runtime_error("Socket creation failed");
|
||||
}
|
||||
}
|
||||
|
||||
void TCPSocket::close_socket() {
|
||||
if (sockfd_ != INVALID_SOCKET_VALUE) {
|
||||
#ifdef _WIN32
|
||||
closesocket(sockfd_);
|
||||
#else
|
||||
close(sockfd_);
|
||||
#endif
|
||||
sockfd_ = INVALID_SOCKET_VALUE;
|
||||
}
|
||||
connected_.store(false);
|
||||
}
|
||||
|
||||
TCPSocket::~TCPSocket() {
|
||||
close_socket();
|
||||
}
|
||||
|
||||
bool TCPSocket::is_connected() const { return connected_.load(); }
|
||||
|
||||
void TCPClient::connect(const std::string &ip, uint16_t port) {
|
||||
create_socket();
|
||||
|
||||
sockaddr_in addr{};
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons(port);
|
||||
inet_pton(AF_INET, ip.c_str(), &addr.sin_addr);
|
||||
|
||||
if (::connect(sockfd_, (sockaddr *) &addr, sizeof(addr)) != 0) {
|
||||
close_socket();
|
||||
throw std::runtime_error("Connection failed");
|
||||
}
|
||||
connected_.store(true);
|
||||
}
|
||||
|
||||
size_t TCPClient::send(const void *data, size_t size) {
|
||||
#ifdef _WIN32
|
||||
auto sent = ::send(sockfd_, (const char *) data, (int) size, 0);
|
||||
#else
|
||||
auto sent = ::send(sockfd_, data, size, 0);
|
||||
#endif
|
||||
|
||||
if (sent == SOCKET_ERROR) {
|
||||
connected_.store(false);
|
||||
throw std::runtime_error("Send failed");
|
||||
}
|
||||
return static_cast<size_t>(sent);
|
||||
}
|
||||
|
||||
size_t TCPClient::send(const std::string &data) {
|
||||
#ifdef _WIN32
|
||||
auto sent = ::send(sockfd_, data.c_str(), static_cast<int>(data.size()), 0);
|
||||
#else
|
||||
auto sent = ::send(sockfd_, data.c_str(), data.size(), 0);
|
||||
#endif
|
||||
|
||||
if (sent == SOCKET_ERROR) {
|
||||
connected_.store(false);
|
||||
throw std::runtime_error("Send failed");
|
||||
}
|
||||
return static_cast<size_t>(sent);
|
||||
}
|
||||
|
||||
size_t TCPClient::receive(void *buffer, size_t buffer_size) {
|
||||
#ifdef _WIN32
|
||||
auto received = ::recv(sockfd_, (char *) buffer, (int) buffer_size, 0);
|
||||
#else
|
||||
auto received = ::recv(sockfd_, buffer, buffer_size, 0);
|
||||
#endif
|
||||
|
||||
if (received == 0) {
|
||||
connected_.store(false);
|
||||
return 0;
|
||||
}
|
||||
if (received == SOCKET_ERROR) {
|
||||
connected_.store(false);
|
||||
throw std::runtime_error("Receive failed");
|
||||
}
|
||||
return static_cast<size_t>(received);
|
||||
}
|
||||
|
||||
void TCPServer::listen(uint16_t port, int backlog) {
|
||||
create_socket();
|
||||
|
||||
sockaddr_in addr{};
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons(port);
|
||||
addr.sin_addr.s_addr = INADDR_ANY;
|
||||
|
||||
if (bind(sockfd_, (sockaddr *) &addr, sizeof(addr)) != 0) {
|
||||
close_socket();
|
||||
throw std::runtime_error("Bind failed");
|
||||
}
|
||||
|
||||
if (::listen(sockfd_, backlog) != 0) {
|
||||
close_socket();
|
||||
throw std::runtime_error("Listen failed");
|
||||
}
|
||||
}
|
||||
|
||||
std::shared_ptr<TCPClient> TCPServer::accept() {
|
||||
sockaddr_in client_addr{};
|
||||
socklen_t addr_len = sizeof(client_addr);
|
||||
|
||||
auto client_sock = ::accept(sockfd_, (sockaddr *) &client_addr, &addr_len);
|
||||
if (client_sock == INVALID_SOCKET_VALUE) {
|
||||
throw std::runtime_error("Accept failed");
|
||||
}
|
||||
|
||||
auto client = std::make_shared<TCPClient>();
|
||||
client->sockfd_ = client_sock;
|
||||
client->connected_.store(true);
|
||||
return client;
|
||||
}
|
||||
|
||||
void AsyncTCPClient::start_async() {
|
||||
running_.store(true);
|
||||
recv_thread_ = std::thread([this]() {
|
||||
std::vector<uint8_t> buffer(4096);
|
||||
while (running_.load()) {
|
||||
try {
|
||||
auto size = receive(buffer.data(), buffer.size());
|
||||
if (size > 0) {
|
||||
std::lock_guard<std::mutex> lock(queue_mutex_);
|
||||
recv_queue_.emplace(buffer.begin(), buffer.begin() + static_cast<long long>(size));
|
||||
}
|
||||
}
|
||||
catch (...) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
bool AsyncTCPClient::try_pop(std::vector<uint8_t> &data) {
|
||||
std::lock_guard<std::mutex> lock(queue_mutex_);
|
||||
if (recv_queue_.empty()) return false;
|
||||
data = std::move(recv_queue_.front());
|
||||
recv_queue_.pop();
|
||||
return true;
|
||||
}
|
||||
|
||||
AsyncTCPClient::~AsyncTCPClient() {
|
||||
running_.store(false);
|
||||
if (recv_thread_.joinable()) {
|
||||
recv_thread_.join();
|
||||
}
|
||||
close_socket();
|
||||
}
|
||||
79
Socket/TCP.h
Normal file
79
Socket/TCP.h
Normal file
@ -0,0 +1,79 @@
|
||||
#ifndef LXZL_TCP_H
|
||||
#define LXZL_TCP_H
|
||||
|
||||
#include "NetworkRelated.h"
|
||||
|
||||
#include <thread>
|
||||
#include <mutex>
|
||||
#include <queue>
|
||||
#include <atomic>
|
||||
#include <memory>
|
||||
#include <condition_variable>
|
||||
|
||||
/**
|
||||
* TCP套接字
|
||||
*/
|
||||
class TCPSocket {
|
||||
public:
|
||||
/** 创建套接字 */
|
||||
void create_socket();
|
||||
|
||||
/** 关闭套接字 */
|
||||
void close_socket();
|
||||
|
||||
virtual ~TCPSocket();
|
||||
|
||||
/** 套接字状态 */
|
||||
bool is_connected() const;
|
||||
|
||||
public:
|
||||
socket_t sockfd_ = INVALID_SOCKET_VALUE;
|
||||
std::atomic<bool> connected_{false};
|
||||
};
|
||||
|
||||
/**
|
||||
* TCP客户端
|
||||
*/
|
||||
class TCPClient : public TCPSocket {
|
||||
public:
|
||||
/** 建立链接 */
|
||||
void connect(const std::string &ip, uint16_t port);
|
||||
|
||||
/** 发送数据 */
|
||||
size_t send(const void *data, size_t size);
|
||||
|
||||
size_t send(const std::string &data);
|
||||
|
||||
/** 接收数据，返回数据长度 */
|
||||
size_t receive(void *buffer, size_t buffer_size);
|
||||
};
|
||||
|
||||
/**
|
||||
* TCP服务端
|
||||
*/
|
||||
class TCPServer : public TCPSocket {
|
||||
public:
|
||||
void listen(uint16_t port, int backlog = SOMAXCONN);
|
||||
|
||||
std::shared_ptr<TCPClient> accept();
|
||||
};
|
||||
|
||||
// 异步客户端（生产者-消费者模式）
|
||||
class AsyncTCPClient : public TCPClient {
|
||||
public:
|
||||
void start_async();
|
||||
|
||||
bool try_pop(std::vector<uint8_t> &data);
|
||||
|
||||
~AsyncTCPClient() override;
|
||||
|
||||
private:
|
||||
std::thread recv_thread_;
|
||||
std::mutex queue_mutex_;
|
||||
std::condition_variable queue_cv_;
|
||||
std::queue<std::vector<uint8_t>> recv_queue_;
|
||||
std::atomic<bool> running_{false};
|
||||
};
|
||||
|
||||
|
||||
#endif //LXZL_TCP_H
|
||||
207
Socket/UDP.cpp
Normal file
207
Socket/UDP.cpp
Normal file
@ -0,0 +1,207 @@
|
||||
#include "UDP.h"
|
||||
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
|
||||
void UDPSocketBase::create_socket() {
|
||||
while (true) {
|
||||
sockfd_ = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||
#if defined(_WIN32)
|
||||
if (sockfd_ != INVALID_SOCKET_VALUE) {
|
||||
break;
|
||||
}
|
||||
#else
|
||||
if (sockfd_ >= 0) {
|
||||
break;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
void UDPSocketBase::create_socket(socket_t socket) {
|
||||
sockfd_ = socket;
|
||||
}
|
||||
|
||||
void UDPSocketBase::set_nonblock(bool enable) {
|
||||
u_long ctl = enable ? 1 : 0;
|
||||
#if defined(_WIN32)
|
||||
ioctlsocket(sockfd_, FIONBIO, &ctl) == 0;
|
||||
#else
|
||||
ioctl(sockfd_, FIONBIO, &ctl) == 0;
|
||||
#endif
|
||||
is_nonblock_ = enable;
|
||||
}
|
||||
|
||||
void UDPSocketBase::bind_socket(const sockaddr_in &addr) const {
|
||||
if (bind(sockfd_, (struct sockaddr *) &addr, sizeof(addr)) != 0) {
|
||||
throw std::runtime_error("Bind failed");
|
||||
}
|
||||
}
|
||||
|
||||
UDPSocketBase::~UDPSocketBase() {
|
||||
if (sockfd_ != INVALID_SOCKET_VALUE) {
|
||||
#ifdef _WIN32
|
||||
closesocket(sockfd_);
|
||||
#else
|
||||
close(sockfd_);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
UDPSender::UDPSender(socket_t socket, bool nonblock) {
|
||||
if (socket != INVALID_SOCKET_VALUE) {
|
||||
create_socket(socket);
|
||||
}
|
||||
else {
|
||||
create_socket();
|
||||
}
|
||||
set_nonblock(nonblock);
|
||||
}
|
||||
|
||||
void UDPSender::set_destination(const std::string &ip, uint16_t port) {
|
||||
dstIp = ip;
|
||||
dstPort = port;
|
||||
dest_addr_.sin_family = AF_INET;
|
||||
dest_addr_.sin_port = htons(port);
|
||||
#if defined(_WIN32_WINNT) && _WIN32_WINNT < 0x600
|
||||
dest_addr_.sin_addr.S_un.S_addr = inet_addr(ip.c_str());
|
||||
#else
|
||||
inet_pton(AF_INET, ip.c_str(), &dest_addr_.sin_addr);
|
||||
#endif
|
||||
}
|
||||
|
||||
void UDPSender::set_source(const std::string &ip, uint16_t port) {
|
||||
sockaddr_in addr{};
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons(port);
|
||||
#if defined(_WIN32_WINNT) && _WIN32_WINNT < 0x600
|
||||
addr.sin_addr.S_un.S_addr = inet_addr(ip.c_str());
|
||||
#else
|
||||
inet_pton(AF_INET, ip.c_str(), &addr.sin_addr);
|
||||
#endif
|
||||
bind_socket(addr);
|
||||
}
|
||||
|
||||
size_t UDPSender::send(const std::vector<uint8_t> &data) {
|
||||
auto sent = sendto(sockfd_,
|
||||
reinterpret_cast<const char *>(data.data()),
|
||||
static_cast<int>(data.size()),
|
||||
0,
|
||||
(struct sockaddr *) &dest_addr_,
|
||||
sizeof(dest_addr_));
|
||||
|
||||
if (sent == SOCKET_ERROR) {
|
||||
throw std::runtime_error("Send failed");
|
||||
}
|
||||
return static_cast<size_t>(sent);
|
||||
}
|
||||
|
||||
size_t UDPSender::send(const std::string &data) {
|
||||
auto sent = sendto(sockfd_,
|
||||
reinterpret_cast<const char *>(data.data()),
|
||||
static_cast<int>(data.size()),
|
||||
0,
|
||||
(struct sockaddr *) &dest_addr_,
|
||||
sizeof(dest_addr_));
|
||||
|
||||
if (sent == SOCKET_ERROR) {
|
||||
throw std::runtime_error("Send failed");
|
||||
}
|
||||
return static_cast<size_t>(sent);
|
||||
}
|
||||
|
||||
size_t UDPSender::send(uint8_t *data, size_t size) {
|
||||
int nleft, nwritten;
|
||||
const char *ptr;
|
||||
|
||||
ptr = (char *) data; /* can't do pointer arithmetic on void* */
|
||||
nleft = size;
|
||||
while (nleft > 0) {
|
||||
#if defined(_WIN32)
|
||||
if ((nwritten = sendto(sockfd_, ptr, nleft, 0, (const struct sockaddr *) &dest_addr_, sizeof(dest_addr_))) <= 0)
|
||||
#else
|
||||
if ((nwritten = sendto(sockfd_, ptr, nleft, MSG_NOSIGNAL, (const struct sockaddr *) &dest_addr_,
|
||||
sizeof(struct sockaddr))) <= 0)
|
||||
#endif
|
||||
{
|
||||
if (!(errno == EWOULDBLOCK || errno == EAGAIN || errno == EINTR))
|
||||
return -1; /* error */
|
||||
else
|
||||
continue;
|
||||
}
|
||||
|
||||
nleft -= nwritten;
|
||||
ptr += nwritten;
|
||||
}
|
||||
|
||||
return (size);
|
||||
}
|
||||
|
||||
|
||||
UDPReceiver::UDPReceiver(uint16_t port, bool nonblock, size_t buffer_size)
|
||||
: buffer_(buffer_size), running_(false), srcPort(port) {
|
||||
create_socket();
|
||||
|
||||
sockaddr_in addr{};
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons(port);
|
||||
addr.sin_addr.s_addr = INADDR_ANY;
|
||||
|
||||
bind_socket(addr);
|
||||
set_nonblock(nonblock);
|
||||
}
|
||||
|
||||
void UDPReceiver::start_receiving(const UDPReceiver::ReceiveCallback &callback) {
|
||||
running_ = true;
|
||||
receiver_thread_ = std::thread([this, callback]() {
|
||||
while (running_.load(std::memory_order_relaxed)) {
|
||||
try {
|
||||
auto result = receive();
|
||||
ssize_t size = result.first;
|
||||
auto &from = result.second;
|
||||
if (size > 0) {
|
||||
std::vector<uint8_t> data(buffer_.begin(),
|
||||
buffer_.begin() + size);
|
||||
callback(data, from.first, from.second);
|
||||
}
|
||||
}
|
||||
catch (const std::runtime_error &e) {
|
||||
if (is_nonblock_) continue;
|
||||
throw;
|
||||
}
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
void UDPReceiver::stop_receiving() {
|
||||
running_.store(false);
|
||||
if (receiver_thread_.joinable()) {
|
||||
receiver_thread_.join();
|
||||
}
|
||||
}
|
||||
|
||||
std::pair<ssize_t, std::pair<sockaddr_in, socklen_t>> UDPReceiver::receive() {
|
||||
sockaddr_in from_addr{};
|
||||
socklen_t from_len = sizeof(from_addr);
|
||||
#ifdef _WIN32
|
||||
DWORD timeout = UDPTimeout * 1000; // 3秒(单位：毫秒)
|
||||
setsockopt(sockfd_, SOL_SOCKET, SO_RCVTIMEO, (const char *) &timeout, sizeof(timeout));
|
||||
#else
|
||||
struct timeval tv;
|
||||
tv.tv_sec = UDPTimeout; // 秒
|
||||
tv.tv_usec = 0; // 微秒
|
||||
setsockopt(sockfd_, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
|
||||
#endif
|
||||
auto size = recvfrom(sockfd_,
|
||||
reinterpret_cast<char *>(buffer_.data()),
|
||||
static_cast<int>(buffer_.size()),
|
||||
0,
|
||||
(struct sockaddr *) &from_addr,
|
||||
&from_len);
|
||||
|
||||
if (size == SOCKET_ERROR) {
|
||||
throw std::runtime_error("Receive Timeout");
|
||||
}
|
||||
|
||||
return {size, {from_addr, from_len}};
|
||||
}
|
||||
95
Socket/UDP.h
Normal file
95
Socket/UDP.h
Normal file
@ -0,0 +1,95 @@
|
||||
#ifndef LXZL_UDP_H
|
||||
#define LXZL_UDP_H
|
||||
|
||||
#include "NetworkRelated.h"
|
||||
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <functional>
|
||||
#include <thread>
|
||||
#include <atomic>
|
||||
|
||||
//定义UDP接收线程超时时间(秒)
|
||||
#define UDPTimeout (3)
|
||||
|
||||
/**
|
||||
* UDP基类
|
||||
*/
|
||||
class UDPSocketBase {
|
||||
protected:
|
||||
UDPSocketBase() : sockfd_(INVALID_SOCKET_VALUE), is_nonblock_(false) {}
|
||||
|
||||
/** 创建套接字 */
|
||||
void create_socket();
|
||||
void create_socket(socket_t socket);
|
||||
|
||||
/** 设置是否阻塞 */
|
||||
void set_nonblock(bool enable);
|
||||
|
||||
/** 建立链接 */
|
||||
void bind_socket(const sockaddr_in &addr) const;
|
||||
|
||||
public:
|
||||
virtual ~UDPSocketBase();
|
||||
|
||||
socket_t sockfd_;
|
||||
protected:
|
||||
bool is_nonblock_;
|
||||
};
|
||||
|
||||
/**
|
||||
* UDP发送端
|
||||
*/
|
||||
class UDPSender : public UDPSocketBase {
|
||||
public:
|
||||
explicit UDPSender(socket_t socket=INVALID_SOCKET_VALUE,bool nonblock = false);
|
||||
|
||||
/** 设置目的地址 */
|
||||
void set_destination(const std::string &ip, uint16_t port);
|
||||
|
||||
/** 设置源地址 */
|
||||
void set_source(const std::string &ip, uint16_t port);
|
||||
|
||||
/** 发送 */
|
||||
size_t send(const std::vector<uint8_t> &data);
|
||||
size_t send(uint8_t *data, size_t size);
|
||||
size_t send(const std::string &data);
|
||||
|
||||
std::string dstIp;
|
||||
uint16_t dstPort;
|
||||
|
||||
private:
|
||||
sockaddr_in dest_addr_{};
|
||||
};
|
||||
|
||||
/**
|
||||
* UDP接收端
|
||||
*/
|
||||
class UDPReceiver : public UDPSocketBase {
|
||||
public:
|
||||
/** 处理接收数据函数模板 */
|
||||
using ReceiveCallback = std::function<void(const std::vector<uint8_t> &, sockaddr_in, socklen_t)>;
|
||||
|
||||
/** 设置接收端口 */
|
||||
explicit UDPReceiver(uint16_t port, bool nonblock = false, size_t buffer_size = 4096);
|
||||
|
||||
/** 开始接收线程 */
|
||||
void start_receiving(const ReceiveCallback &callback);
|
||||
|
||||
/** 停止接收线程 */
|
||||
void stop_receiving();
|
||||
|
||||
uint16_t srcPort;
|
||||
|
||||
private:
|
||||
/** 接收函数 */
|
||||
std::pair<ssize_t, std::pair<sockaddr_in, socklen_t>> receive();
|
||||
|
||||
private:
|
||||
std::vector<uint8_t> buffer_;
|
||||
std::atomic<bool> running_;
|
||||
std::thread receiver_thread_;
|
||||
};
|
||||
|
||||
|
||||
#endif //LXZL_UDP_H
|
||||
Reference in New Issue
Block a user