2020-05-02 07:31:03 +00:00
|
|
|
#include "UnixSocketInterface.h"
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
#include "HttpResponse.h"
|
|
|
|
|
2020-05-02 07:31:03 +00:00
|
|
|
#include <iostream>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#include <netinet/in.h>
|
|
|
|
|
|
|
|
UnixSocketInterface::UnixSocketInterface()
|
2021-03-29 20:31:24 +00:00
|
|
|
: mBufferSize(1024)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
std::unique_ptr<UnixSocketInterface> UnixSocketInterface::Create()
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
return std::make_unique<UnixSocketInterface>();
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
void UnixSocketInterface::InitializeSocket(const SocketPtr& socketPtr)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
auto handle = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
socketPtr->SetHandle(handle);
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
void UnixSocketInterface::Listen(const SocketPtr& socket)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
if(socket->GetHandle() < 0)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error opening socket" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
int port = static_cast<int>(socket->GetPort());
|
2020-05-02 07:31:03 +00:00
|
|
|
struct sockaddr_in serv_addr;
|
|
|
|
memset(&serv_addr, 0, sizeof(serv_addr));
|
|
|
|
serv_addr.sin_family = AF_INET;
|
|
|
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
|
|
|
serv_addr.sin_port = htons(port);
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
int result = bind(socket->GetHandle(), (struct sockaddr *)&serv_addr, sizeof(serv_addr));
|
2020-05-02 07:31:03 +00:00
|
|
|
if(result< 0)
|
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error binding socket" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
listen(socket->GetHandle(), 5);
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
void UnixSocketInterface::Run(const SocketPtr& socket)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
if(socket->GetHandle() < 0)
|
2020-05-02 07:31:03 +00:00
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error opening socket" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
struct sockaddr_in cli_addr;
|
|
|
|
socklen_t clilen = sizeof(cli_addr);
|
|
|
|
while(true)
|
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
auto new_socket_handle = accept(socket->GetHandle(),
|
|
|
|
(struct sockaddr *) &cli_addr, &clilen);
|
2020-05-02 07:31:03 +00:00
|
|
|
if (new_socket_handle < 0)
|
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error on accept" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
char buffer[mBufferSize] = {0};
|
|
|
|
int n = read(new_socket_handle, buffer, mBufferSize);
|
|
|
|
if (n < 0)
|
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error on read" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
socket->SetMessage(buffer);
|
|
|
|
std::cout << "Here is the message: " << buffer << std::endl;
|
|
|
|
|
|
|
|
HttpResponse response;
|
|
|
|
response.SetBody("Hello world!");
|
|
|
|
|
2021-03-29 20:31:24 +00:00
|
|
|
auto response_message = response.ToString();
|
2020-05-02 07:31:03 +00:00
|
|
|
n = write(new_socket_handle, response_message.c_str(), response_message.length());
|
|
|
|
if (n < 0)
|
|
|
|
{
|
2021-03-29 20:31:24 +00:00
|
|
|
std::cerr << "Error on write" << std::endl;
|
|
|
|
return;
|
2020-05-02 07:31:03 +00:00
|
|
|
}
|
|
|
|
close(new_socket_handle);
|
|
|
|
}
|
|
|
|
}
|