Ivan I. Ovchinnikov
125 lines
4.5 KiB
C++
125 lines
4.5 KiB
C++
#include "../include/iovi_space/network/server_socket/ServerSocket.h"
|
|
|
|
#include <system_error>
|
|
|
|
#ifdef _WIN32
|
|
#include <ws2tcpip.h>
|
|
// Инициализация Winsock должна быть выполнена один раз в main()
|
|
// через WSAStartup(). Здесь предполагаем, что это уже сделано.
|
|
#define SOCKET_INIT() do {} while(0)
|
|
#else
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <sys/select.h>
|
|
#endif
|
|
|
|
namespace iovi_space::network {
|
|
|
|
ServerSocket::ServerSocket(ServerSocketListener* listener, const std::string& name, int port, int timeout_ms)
|
|
: listener_(listener), name_(name), port_(port), timeout_ms_(timeout_ms)
|
|
{
|
|
if (!listener_) {
|
|
throw std::invalid_argument("Listener cannot be null");
|
|
}
|
|
worker_ = std::thread(&ServerSocket::run, this);
|
|
}
|
|
|
|
ServerSocket::~ServerSocket() {
|
|
stop();
|
|
if (worker_.joinable()) {
|
|
worker_.join();
|
|
}
|
|
}
|
|
|
|
void ServerSocket::stop() {
|
|
stop_requested_.store(true);
|
|
running_.store(false);
|
|
}
|
|
|
|
void ServerSocket::run() {
|
|
#ifdef _WIN32
|
|
// Примечание: в реальном проекте инициализация Winsock должна быть в main()
|
|
// WSADATA wsaData; WSAStartup(MAKEWORD(2,2), &wsaData);
|
|
#endif
|
|
|
|
listener_->onServerStart(this);
|
|
|
|
try {
|
|
// Создаём сокет
|
|
SocketHandle sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
|
if (sock == INVALID_SOCKET_HANDLE) {
|
|
throw std::system_error(GET_ERROR, std::system_category(), "socket() failed");
|
|
}
|
|
|
|
// Опция SO_REUSEADDR для быстрого перезапуска
|
|
int opt = 1;
|
|
#ifdef _WIN32
|
|
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, reinterpret_cast<const char*>(&opt), sizeof(opt));
|
|
#else
|
|
setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
|
|
#endif
|
|
|
|
// Привязка к адресу
|
|
sockaddr_in addr{};
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
addr.sin_port = htons(static_cast<uint16_t>(port_));
|
|
|
|
if (bind(sock, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) == -1) {
|
|
CLOSE_SOCKET(sock);
|
|
throw std::system_error(GET_ERROR, std::system_category(), "bind() failed on port " + std::to_string(port_));
|
|
}
|
|
|
|
// Переход в режим прослушивания
|
|
if (listen(sock, SOMAXCONN) == -1) {
|
|
CLOSE_SOCKET(sock);
|
|
throw std::system_error(GET_ERROR, std::system_category(), "listen() failed");
|
|
}
|
|
|
|
listenSocket_ = SocketWrapper(sock);
|
|
listener_->onServerSocketCreated(this, listenSocket_);
|
|
|
|
running_.store(true);
|
|
|
|
// Главный цикл
|
|
while (!stop_requested_.load()) {
|
|
// Используем select для реализации таймаута accept()
|
|
fd_set readfds;
|
|
FD_ZERO(&readfds);
|
|
FD_SET(listenSocket_.get(), &readfds);
|
|
|
|
struct timeval timeout{};
|
|
timeout.tv_sec = timeout_ms_ / 1000;
|
|
timeout.tv_usec = (timeout_ms_ % 1000) * 1000;
|
|
|
|
int result = select(static_cast<int>(listenSocket_.get()) + 1, &readfds, nullptr, nullptr, &timeout);
|
|
|
|
if (stop_requested_.load()) break;
|
|
|
|
if (result == 0) {
|
|
// Таймаут
|
|
listener_->onServerTimeout(this, listenSocket_);
|
|
continue;
|
|
} else if (result > 0 && FD_ISSET(listenSocket_.get(), &readfds)) {
|
|
// Есть новое подключение
|
|
SocketHandle clientSock = accept(listenSocket_.get(), nullptr, nullptr);
|
|
if (clientSock != INVALID_SOCKET_HANDLE) {
|
|
auto clientWrapper = std::make_unique<SocketWrapper>(clientSock);
|
|
listener_->onSocketAccepted(this, listenSocket_, std::move(clientWrapper));
|
|
}
|
|
// Если accept не удался - игнорируем и продолжаем (как в оригинале)
|
|
}
|
|
// result < 0 означает ошибку select, но для простоты продолжаем цикл
|
|
}
|
|
|
|
} catch (const std::system_error& e) {
|
|
listener_->onServerException(this, e);
|
|
} catch (...) {
|
|
listener_->onServerException(this, std::system_error(std::error_code(), std::generic_category(), "Unknown exception"));
|
|
}
|
|
|
|
listener_->onServerStop(this);
|
|
running_.store(false);
|
|
}
|
|
|
|
} // namespace iovi_space::network
|