101 lines
2.1 KiB
C++
101 lines
2.1 KiB
C++
|
#include "Server.hpp"
|
||
|
|
#include "Request.hpp"
|
||
|
|
#include "Response.hpp"
|
||
|
|
|
||
|
|
#include <sys/socket.h>
|
||
|
|
#include <netinet/in.h>
|
||
|
|
#include <stdexcept>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
namespace eRPC
|
||
|
|
{
|
||
|
|
Server::Server(int port) : sockfd(-1), running(false)
|
||
|
|
{
|
||
|
|
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
|
||
|
|
{
|
||
|
|
throw std::runtime_error("Failed to create socket");
|
||
|
|
}
|
||
|
|
|
||
|
|
sockaddr_in serv_addr;
|
||
|
|
serv_addr.sin_family = AF_INET;
|
||
|
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
||
|
|
serv_addr.sin_port = htons(port);
|
||
|
|
|
||
|
|
if (bind(sockfd, (sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
|
||
|
|
{
|
||
|
|
throw std::runtime_error("Failed to bind socket");
|
||
|
|
}
|
||
|
|
|
||
|
|
if (listen(sockfd, 5) < 0)
|
||
|
|
{
|
||
|
|
throw std::runtime_error("Failed to listen on socket");
|
||
|
|
}
|
||
|
|
|
||
|
|
std::cout << "Server listening on port " << port << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
Server::~Server()
|
||
|
|
{
|
||
|
|
stop();
|
||
|
|
}
|
||
|
|
|
||
|
|
void Server::start()
|
||
|
|
{
|
||
|
|
if (running)
|
||
|
|
{
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
running = true;
|
||
|
|
|
||
|
|
while (running)
|
||
|
|
{
|
||
|
|
std::cout << "Waiting for connection" << std::endl;
|
||
|
|
|
||
|
|
int connfd = accept(sockfd, (sockaddr *)NULL, NULL);
|
||
|
|
|
||
|
|
char buffer[1024] = {0};
|
||
|
|
read(connfd, buffer, 1024);
|
||
|
|
Request request(buffer);
|
||
|
|
std::cout << "Received request:\n"
|
||
|
|
<< request.serialize() << std::endl;
|
||
|
|
|
||
|
|
Response response;
|
||
|
|
if (methods.find(request.getMethod()) != methods.end())
|
||
|
|
{
|
||
|
|
std::function<void *()> method = methods[request.getMethod()];
|
||
|
|
method();
|
||
|
|
|
||
|
|
response = Response(
|
||
|
|
request.getMsgid(),
|
||
|
|
true,
|
||
|
|
"RESULTS HERE");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
response = Response(
|
||
|
|
request.getMsgid(),
|
||
|
|
false,
|
||
|
|
"Method \"" + request.getMethod() + "\" not found");
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string serialized = response.serialize();
|
||
|
|
write(connfd, serialized.c_str(), serialized.size() + 1);
|
||
|
|
|
||
|
|
close(connfd);
|
||
|
|
}
|
||
|
|
|
||
|
|
close(sockfd);
|
||
|
|
}
|
||
|
|
|
||
|
|
void Server::stop()
|
||
|
|
{
|
||
|
|
running = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Server::bindMethod(std::string method, std::function<void *()> callback)
|
||
|
|
{
|
||
|
|
methods[method] = callback;
|
||
|
|
}
|
||
|
|
}
|