tom/RT_GPU
1
0
Fork 0
mirror of https://github.com/TheRedShip/RT_GPU.git synced 2025-09-27 18:48:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

clusterizer : server and client can now communicate

Browse Source
This commit is contained in:
tomoron
2025-02-22 02:07:01 +01:00
parent 960c4eb99c
commit 05ba0447d2
6 changed files with 238 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
srcs/class/Clusterizer/server.cpp
View File

@ -6,7 +6,7 @@
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2025/02/20 21:08:38 by tomoron #+# #+# */
/* Updated: 2025/02/20 22:43:00 by tomoron ### ########.fr */
/* Updated: 2025/02/21 22:19:20 by tomoron ### ########.fr */
/* */
/* ************************************************************************** */
@ -14,9 +14,12 @@
void Clusterizer::initServer(std::string port)
{
_pollfds = 0;
try
{
initServerSocket(stoi(port));
std::cout << "server initialized with socket " << _serverSocket << std::endl;
}
catch(std::exception &e)
{
@ -25,24 +28,107 @@ void Clusterizer::initServer(std::string port)
}
}
void Clusterizer::initServerSocket(uint16_t port)
void Clusterizer::initServerSocket(int port)
{
struct sockaddr_in s_addr;
if(port > 65535 || port < 0)
throw std::runtime_error("invalid port");
_serverSocket = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
if (_serverSocket < 0)
throw std::runtime_error("can't create socket");
s_addr.sin_family = AF_INET;
s_addr.sin_addr.s_addr = INADDR_ANY;
s_addr.sin_port = port >> 8 | port << 8;
s_addr.sin_port = htons(port);
if (bind(_serverSocket, (struct sockaddr *)&s_addr, \
sizeof(struct sockaddr_in)) < 0)
throw std::runtime_error("can't bind socket");
if (::listen(_serverSocket, 50) < 0)
throw std::runtime_error("can't listen on socket");
if(_serverSocket == -1)
throw std::runtime_error("unknown error");
}
void Clusterizer::updatePollfds(void)
{
if(_pollfds)
delete[] _pollfds;
_pollfds = new struct pollfd[_clients.size()];
for(auto it = _clients.begin(); it != _clients.end(); it++)
{
_pollfds[std::distance(it, _clients.begin())].fd = it->first;
_pollfds[std::distance(it, _clients.begin())].events = POLLIN;
_pollfds[std::distance(it, _clients.begin())].revents = 0;
}
}
void Clusterizer::acceptClients(void)
{
int fd;
fd = accept(_serverSocket, 0, 0);
if (fd != -1) {
std::cout << "new client :" << fd << std::endl;
_clients[fd].ready = 0;
_clients[fd].curJob = 0;
updatePollfds();
}
}
void Clusterizer::deleteClient(int fd)
{
std::map<int, t_client>::iterator it;
std::cout << "client disconnected" << std::endl;
it = _clients.find(fd);
if(it == _clients.end())
return;
_clients.erase(it);
updatePollfds();
close(fd);
}
void Clusterizer::handleBuffer(int fd, std::vector<uint8_t> &buf)
{
std::vector<uint8_t> sendBuffer;
if(buf[0] == RDY)
{
_clients[fd].ready = 1;
sendBuffer.push_back(ACK);
}
if(sendBuffer.size())
(void)write(fd, sendBuffer.data(), sendBuffer.size());
}
int Clusterizer::updateBuffer(int fd)
{
uint8_t buffer[512];
size_t ret;
ret = recv(fd, buffer, 512, 0);
if(!ret)
return(1);
_clients[fd].buffer.insert(_clients[fd].buffer.end(), buffer, buffer + ret);
handleBuffer(fd, _clients[fd].buffer);
return(0);
}
void Clusterizer::updateServer(void)
{
int recv;
acceptClients();
recv = poll(_pollfds, _clients.size(), 1);
if(!recv)
return ;
for(auto it = _clients.begin(); it != _clients.end(); it++)
{
if(_pollfds[std::distance(it, _clients.begin())].revents & POLLIN)
{
if (updateBuffer(it->first))
deleteClient(it->first);
}
}
}