81 lines
2.2 KiB
C++
81 lines
2.2 KiB
C++
#include "PollManager.hpp"
|
|
|
|
#include <iostream>
|
|
#include <unistd.h>
|
|
#include <cstring>
|
|
#include <netinet/in.h>
|
|
#include <sys/socket.h>
|
|
#include <poll.h>
|
|
#include <cstdlib>
|
|
|
|
PollManager::PollManager() {}
|
|
|
|
PollManager::~PollManager() {
|
|
for (size_t i = 0; i < _fds.size(); ++i) {
|
|
close(_fds[i].fd);
|
|
}
|
|
}
|
|
|
|
void PollManager::pollLoop(int server_fd) {
|
|
|
|
struct pollfd server_pollfd;
|
|
server_pollfd.fd = server_fd;
|
|
server_pollfd.events = POLLIN;
|
|
_fds.push_back(server_pollfd);
|
|
|
|
std::cout << "Serveur prêt à accepter des connexions..." << std::endl;
|
|
|
|
while (true) {
|
|
int poll_count = poll(&_fds[0], _fds.size(), -1);
|
|
if (poll_count == -1) {
|
|
std::cerr << "poll error\n" << std::endl;
|
|
continue;
|
|
}
|
|
for (size_t i = 0; i < _fds.size(); ++i) {
|
|
int fd = _fds[i].fd;
|
|
|
|
if ((fd == server_fd) && (_fds[i].revents & POLLIN)) {
|
|
int client_fd = accept(server_fd, NULL, NULL);
|
|
if (client_fd == -1) {
|
|
std::cerr << "Error accept()" << std::endl;
|
|
continue;
|
|
}
|
|
addClient(client_fd);
|
|
}
|
|
else if (_fds[i].revents & POLLIN) {
|
|
char buffer[1024];
|
|
ssize_t bytes = recv(fd, buffer, sizeof(buffer) -1, 0);
|
|
if (bytes > 0) {
|
|
buffer[bytes] = '\0';
|
|
_buffers[fd] += buffer;
|
|
std::cout << "Client " << fd << " send : " << buffer;
|
|
} else {
|
|
std::cout << "Client " << fd << " disconected." << std::endl;
|
|
// removeClient(fd);
|
|
--i;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void PollManager::addClient(int fd) {
|
|
struct pollfd pfd;
|
|
pfd.fd = fd;
|
|
pfd.events = POLLIN;
|
|
_fds.push_back(pfd);
|
|
_buffers[fd] = "";
|
|
std::cout << "Client connecté (fd " << fd << ")" << std::endl;
|
|
}
|
|
|
|
void PollManager::removeClient(int fd) {
|
|
for (size_t i = 0; i < _fds.size(); ++i) {
|
|
if (_fds[i].fd == fd) {
|
|
_fds.erase(_fds.begin() + i);
|
|
break;
|
|
}
|
|
}
|
|
_buffers.erase(fd);
|
|
close(fd);
|
|
std::cout << "Client retiré (fd " << fd << ")" << std::endl;
|
|
}
|