Added PollManager class

This commit is contained in:
ouafabulous 2025-05-19 20:23:22 +02:00
parent 53a83308de
commit 4a17d88d80
7 changed files with 179 additions and 37 deletions

View file

@ -1,4 +1,4 @@
# **************************************************************************** #
#******************************************************************************#
# #
# ::: :::::::: #
# Makefile :+: :+: :+: #
@ -6,9 +6,9 @@
# By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ #
# +#+#+#+#+#+ +#+ #
# Created: 2025/05/02 15:40:00 by rparodi #+# #+# #
# Updated: 2025/05/19 15:13:05 by omoudni ### ########.fr #
# Updated: 2025/05/19 20:14:35 by omoudni ### ########.fr #
# #
# **************************************************************************** #
#******************************************************************************#
# Name
@ -28,6 +28,7 @@ SRC = sources/core/main.cpp \
sources/core/server.cpp \
sources/core/check.cpp \
sources/core/parser.cpp \
sources/core/pollManager.cpp \
INC_DIR = include/core \

View file

@ -1,13 +1,16 @@
@startuml
' ========================
' CLASS: Server
' ========================
class Server {
- _port : int
- _password : string
- _clients : List<Client>
- _users : List<User>
- _server_fd : int
- _state_msg : std::pair<User, e_state>
- _pollManager : PollManager
+ showInfo() : void
+ getPort() : int
@ -15,9 +18,9 @@ class Server {
}
' ========================
' CLASS: Client
' CLASS: User
' ========================
class Client {
class User {
- _id : int
- _name : string
- _registered : bool
@ -29,7 +32,7 @@ class Client {
+ isRegistered() : bool
+ joinChannel(channel : Channel) : void
+ leaveChannel(channel : Channel) : void
+ sendMessage(to : Client, message : string) : void
+ sendMessage(to : User, message : string) : void
}
' ========================
@ -38,23 +41,23 @@ class Client {
class Channel {
- _name : string
- _password : string
- _owner : Client
- _operators : List<Client>
- _owner : User
- _operators : List<User>
- _socket_id : int
- _topic : string
+ getName() : string
+ setTopic(newTopic : string) : void
+ addOperator(client : Client) : void
+ removeOperator(client : Client) : void
+ addOperator(User : User) : void
+ removeOperator(User : User) : void
}
' ========================
' CLASS: Commandes
' ========================
class Commandes {
- _form_user : Client
- _to_user : Client
- _form_user : User
- _to_user : User
- _channel : Channel
+ MODE() : void
@ -85,14 +88,29 @@ class Parser {
+ getErrorMsg() : string
}
' ========================
' CLASS: PollManager
' ========================
class PollManager {
- _fds : vector<pollfd>
- _fd_events : map<int, short>
+ addFd(fd : int, events : short) : void
+ removeFd(fd : int) : void
+ updateFd(fd : int, events : short) : void
+ pollEvents(timeout : int) : int
+ getFds() : vector<pollfd>
}
' ========================
' RELATIONS
' ========================
Server "1" o-- "*" Client : _clients
Client "1" o-- "*" Channel : _channels
Channel "1" *-- "1" Client : _owner
Channel "1" o-- "*" Client : _operators
Server "1" o-- "*" User : _users
User "1" o-- "*" Channel : _channels
Channel "1" *-- "1" User : _owner
Channel "1" o-- "*" User : _operators
main ..> Parser : uses
Server "1" *-- "1" PollManager : pollManager
@enduml

View file

@ -1,4 +1,4 @@
/* ************************************************************************** */
/******************************************************************************/
/* */
/* ::: :::::::: */
/* core.hpp :+: :+: :+: */
@ -6,9 +6,9 @@
/* By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/05/12 14:16:03 by rparodi #+# #+# */
/* Updated: 2025/05/19 15:07:26 by omoudni ### ########.fr */
/* Updated: 2025/05/19 20:15:14 by omoudni ### ########.fr */
/* */
/* ************************************************************************** */
/******************************************************************************/
#pragma once
@ -16,11 +16,21 @@
#define LOG std::endl << CLR_CYAN << "Debug: " << __FILE__ << ":" << __LINE__ << std::endl << CLR_RESET
#endif
#ifndef DEBUG
#define DEBUG 0
#define LOG ""
#endif
enum e_state {
ERROR = 0,
CMD,
MSG
};
#include "pollManager.hpp"
#include "color.hpp"
#include "server.hpp"
#include "parser.hpp"

View file

@ -0,0 +1,30 @@
/******************************************************************************/
/* */
/* ::: :::::::: */
/* pollManager.hpp :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/05/19 19:15:13 by omoudni #+# #+# */
/* Updated: 2025/05/19 19:25:45 by omoudni ### ########.fr */
/* */
/******************************************************************************/
#pragma once
#include <vector>
#include <poll.h>
#include <map>
class PollManager {
private:
std::vector<struct pollfd> _fds;
std::map<int, short> _fd_events; // fd -> event
public:
PollManager();
void addFd(int fd, short events);
void removeFd(int fd);
void updateFd(int fd, short events);
int pollEvents(int timeout = -1);
const std::vector<struct pollfd>& getFds() const;
};

View file

@ -1,17 +1,18 @@
/* ************************************************************************** */
/******************************************************************************/
/* */
/* ::: :::::::: */
/* server.hpp :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: omoudni <omoudni@student.42.fr> +#+ +:+ +#+ */
/* By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/05/13 11:06:56 by rparodi #+# #+# */
/* Updated: 2025/05/14 23:23:13 by omoudni ### ########.fr */
/* Updated: 2025/05/19 20:12:46 by omoudni ### ########.fr */
/* */
/* ************************************************************************** */
/******************************************************************************/
#pragma once
#include "core.hpp"
#include <string>
class Server {
@ -19,6 +20,7 @@ class Server {
unsigned short int _port;
std::string _password;
int server_fd;
PollManager _pollManager;
public:
Server();
Server(int port, const std::string &password);

View file

@ -0,0 +1,50 @@
/******************************************************************************/
/* */
/* ::: :::::::: */
/* pollManager.cpp :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/05/19 19:15:48 by omoudni #+# #+# */
/* Updated: 2025/05/19 20:21:11 by omoudni ### ########.fr */
/* */
/******************************************************************************/
#include "pollManager.hpp"
#include <algorithm>
PollManager::PollManager() {}
void PollManager::addFd(int fd, short events) {
struct pollfd pfd = {fd, events, 0};
_fds.push_back(pfd);
_fd_events[fd] = events;
}
void PollManager::removeFd(int fd) {
for (std::vector<struct pollfd>::iterator it = _fds.begin(); it != _fds.end(); ) {
if (it->fd == fd)
it = _fds.erase(it);
else
++it;
}
_fd_events.erase(fd);
}
void PollManager::updateFd(int fd, short events) {
for (size_t i = 0; i < _fds.size(); ++i) {
if (_fds[i].fd == fd) {
_fds[i].events = events;
break;
}
}
_fd_events[fd] = events;
}
int PollManager::pollEvents(int timeout) {
return poll(_fds.data(), _fds.size(), timeout);
}
const std::vector<struct pollfd>& PollManager::getFds() const {
return _fds;
}

View file

@ -1,14 +1,14 @@
/* ************************************************************************** */
/******************************************************************************/
/* */
/* ::: :::::::: */
/* server.cpp :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: omoudni <omoudni@student.42.fr> +#+ +:+ +#+ */
/* By: omoudni <omoudni@student.42paris.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/05/13 11:11:07 by rparodi #+# #+# */
/* Updated: 2025/05/14 23:32:21 by omoudni ### ########.fr */
/* Updated: 2025/05/19 20:16:00 by omoudni ### ########.fr */
/* */
/* ************************************************************************** */
/******************************************************************************/
#include "color.hpp"
#include "server.hpp"
@ -67,17 +67,48 @@ void Server::start() {
close(server_fd);
return;
}
_pollManager.addFd(server_fd, POLLIN);
std::cout << CLR_GREEN << "Server started on port " << this->_port << CLR_RESET << std::endl;
std::cout << CLR_GREEN << "Waiting for clients..." << CLR_RESET << std::endl;
while (true) {
int client_fd = accept(server_fd, NULL, NULL);
if (client_fd == -1) {
std::cerr << CLR_RED << "Error: Failed to accept client" << CLR_RESET << std::endl;
continue;
}
std::cout << CLR_GREEN << "Client connected" << CLR_RESET << std::endl;
close(client_fd);
}
while (true) {
int n = _pollManager.pollEvents();
if (n < 0) {
std::cerr << CLR_RED << "Poll error" << CLR_RESET << std::endl;
break;
}
const std::vector<struct pollfd>& fds = _pollManager.getFds();
for (size_t i = 0; i < fds.size(); ++i) {
if (fds[i].revents & POLLIN) {
if (fds[i].fd == server_fd) {
// Accept new client
int client_fd = accept(server_fd, NULL, NULL);
if (client_fd != -1) {
_pollManager.addFd(client_fd, POLLIN);
std::cout << CLR_GREEN << "Client connected" << CLR_RESET << std::endl;
std::cout << "Welcome to IRC. Your client_id is: " << client_fd << std::endl;
} else {
std::cerr << CLR_RED << "Error: Failed to accept client" << CLR_RESET << std::endl;
}
} else {
// Handle client communication
char buffer[1024];
ssize_t bytes_received = recv(fds[i].fd, buffer, sizeof(buffer) - 1, 0);
if (bytes_received <= 0) {
// Client disconnected
// if it's negative, it means an error occurred, maybe print it with perror?
std::cout << CLR_RED << "Client disconnected" << CLR_RESET << std::endl;
close(fds[i].fd);
_pollManager.removeFd(fds[i].fd);
} else {
buffer[bytes_received] = '\0';
std::cout << "Received: " << buffer << std::endl;
// Optionally: send response to client here
}
}
}
}
}
close(server_fd);
std::cout << CLR_GREEN << "Server stopped" << CLR_RESET << std::endl;
}