89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
#include "TCPServerSocket.h"
|
|
#include "EPoll.h"
|
|
#include "Session.h"
|
|
#include "Exception.h"
|
|
|
|
namespace core {
|
|
|
|
TCPServerSocket::TCPServerSocket(EPoll &ePoll, std::string url, short int port) : TCPSocket(ePoll) {
|
|
|
|
struct sockaddr_in addr;
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = htons(port);
|
|
|
|
struct hostent *hp = gethostbyname(url.c_str());
|
|
|
|
memcpy((void *)&addr.sin_addr, hp->h_addr_list[0], hp->h_length);
|
|
|
|
setDescriptor(socket(AF_INET, SOCK_STREAM, 0));
|
|
|
|
int yes = 1;
|
|
setsockopt(getDescriptor(), SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
|
|
|
|
if(bind(getDescriptor(), (struct sockaddr *)&addr, sizeof(addr)) < 0)
|
|
throw Exception("Error on bind to socket");
|
|
|
|
if(listen(getDescriptor(), 10) < 0)
|
|
throw Exception("Error on listen to socket");
|
|
|
|
ePoll.registerSocket(this);
|
|
|
|
}
|
|
|
|
TCPServerSocket::~TCPServerSocket() {
|
|
close(getDescriptor());
|
|
}
|
|
|
|
void TCPServerSocket::init() {}
|
|
|
|
void TCPServerSocket::onDataReceived(std::string data) {
|
|
|
|
Log(LOG_DEBUG_2) << "Connection request on socket " << getDescriptor() << ".";
|
|
|
|
Session *session = accept();
|
|
sessions.push_back(session);
|
|
}
|
|
|
|
Session * TCPServerSocket::accept() {
|
|
|
|
Session *session = getSocketAccept();
|
|
session->setDescriptor(::accept(getDescriptor(), (struct sockaddr *)&session->ipAddress.address, &session->ipAddress.addressLength));
|
|
ePoll.registerSocket(session);
|
|
|
|
Log(LOG_DEBUG_2) << "Session started on socket " << session->getDescriptor() << ".";
|
|
|
|
return session;
|
|
}
|
|
|
|
Session * TCPServerSocket::getSocketAccept() {
|
|
return new Session(ePoll, *this);
|
|
}
|
|
|
|
|
|
int TCPServerSocket::processCommand(std::string command, Session *session) {
|
|
|
|
int sequence = 0;
|
|
|
|
for(auto *sessionx : sessions) {
|
|
session->out << "|" << ++sequence;
|
|
sessionx->output(session);
|
|
session->out << "|" << std::endl;
|
|
}
|
|
|
|
session->send();
|
|
}
|
|
|
|
void TCPServerSocket::removeFromSessionList(Session *session) {
|
|
std::vector<Session *>::iterator cursor;
|
|
for(cursor = sessions.begin(); cursor < sessions.end(); ++cursor)
|
|
if(*cursor == session)
|
|
sessions.erase(cursor);
|
|
}
|
|
|
|
void TCPServerSocket::sessionErrorHandler(std::string errorString, Session *session) {
|
|
throw Exception(errorString);
|
|
}
|
|
|
|
}
|