67 lines
1.7 KiB
C++
67 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include "ASIO_Utils.h"
|
|
|
|
#include <deque>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
namespace Psc::asio_socket {
|
|
|
|
class UDP_Server {
|
|
public:
|
|
void set_bind_address(const Sockaddr_In& addr) { bind_address = addr; }
|
|
void set_bind_address(const std::string& ip, std::uint32_t port) { bind_address = Sockaddr_In(ip, port); }
|
|
[[nodiscard]] std::string to_string() const { return bind_address.to_string(); }
|
|
|
|
void create();
|
|
bool bind();
|
|
void tick();
|
|
void close();
|
|
void reply_last_peer(const std::string& data);
|
|
void send_to(const std::string& ip, uint16_t port, const std::string& data);
|
|
void write_to_all_clients(const std::string& msg);
|
|
|
|
struct Read_Info {
|
|
Sockaddr_In peer;
|
|
std::string data;
|
|
};
|
|
|
|
std::vector<Read_Info> read();
|
|
|
|
enum State {
|
|
Not_Created,
|
|
Not_Set_Field,
|
|
Not_Bind,
|
|
Working,
|
|
};
|
|
|
|
State state = Not_Created;
|
|
Sockaddr_In bind_address;
|
|
std::vector<Sockaddr_In> clients;
|
|
Socket_FD socket_fd = static_cast<Socket_FD>(-1);
|
|
bool bound = false;
|
|
int read_chunk_size = 1024 * 1024;
|
|
Sockaddr_In last_peer{};
|
|
bool has_last_peer = false;
|
|
size_t buffer_size = 4096 * 10;
|
|
|
|
private:
|
|
asio::io_context io_context;
|
|
std::unique_ptr<asio::ip::udp::socket> socket;
|
|
asio::ip::udp::endpoint recv_endpoint;
|
|
ASIO_PacketRingBuffer recv_buffer;
|
|
ASIO_PacketRingBuffer send_buffer;
|
|
std::vector<Sockaddr_In> recv_peers;
|
|
std::vector<asio::ip::udp::endpoint> send_endpoints;
|
|
std::vector<char> recv_storage;
|
|
std::vector<char> send_storage;
|
|
bool read_pending = false;
|
|
bool write_pending = false;
|
|
|
|
void start_read();
|
|
void start_write();
|
|
};
|
|
|
|
} // namespace Psc::asio_socket
|