|
|
|
@ -4,12 +4,19 @@ |
|
|
|
|
* Kontrustruktor varijable tipa server, prima port i limit za ograničenje liste klijenata na čekanju |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
server::server (const ushort port, const uint limit) { |
|
|
|
|
server::server (const ushort port, const uint queue, SSL_CTX* _securefds) { |
|
|
|
|
securefds = _securefds;
|
|
|
|
|
|
|
|
|
|
addr.sin_family = AF_INET; |
|
|
|
|
addr.sin_addr.s_addr = INADDR_ANY; |
|
|
|
|
addr.sin_port = htons(port); |
|
|
|
|
|
|
|
|
|
#if _WIN32 |
|
|
|
|
if (WSAStartup(MAKEWORD(2,2),&wsa) != 0) { |
|
|
|
|
throw string("[ERROR] WSA Startup. Detail: " + to_string(WSAGetLastError())); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
sock = socket(AF_INET, SOCK_STREAM, 0);
|
|
|
|
|
if (sock <= 0) {
|
|
|
|
|
throw string("[ERROR] Unable to open TCP socket "); |
|
|
|
@ -17,20 +24,69 @@ server::server (const ushort port, const uint limit) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
int opt=1; |
|
|
|
|
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) { |
|
|
|
|
throw string("[ERROR] Unable to set REUSEADDR or REUSEPORT on socket "); |
|
|
|
|
} |
|
|
|
|
#if __linux__ |
|
|
|
|
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt))) { |
|
|
|
|
throw string("[ERROR] Unable to set REUSEADDR or REUSEPORT on socket "); |
|
|
|
|
} |
|
|
|
|
#elif _WIN32 |
|
|
|
|
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (const char*)&opt, sizeof(opt))) { |
|
|
|
|
throw string("[ERROR] Unable to set REUSEADDR or REUSEPORT on socket "); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (bind(sock, (struct sockaddr *)&addr, sizeof(struct sockaddr_in)) < 0) { |
|
|
|
|
throw string("[ERROR] Unable to bind socket "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (listen(sock, limit) < 0) { |
|
|
|
|
if (listen(sock, queue) < 0) { |
|
|
|
|
throw string("[ERROR] It is not possible to set the allowed number of waiting clients "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda za sinkroni rad s klijentima, prima pokazivač na funkciju i timeout; |
|
|
|
|
* Funkcija handlecli prima referencu tipa client - važno za definiranje funkcija koje se šalju; |
|
|
|
|
* Nije moguće proslijediti druge parametre; |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
void server::sync(void (*handlecli)(client&), const uint timeout) { |
|
|
|
|
do { |
|
|
|
|
client cli(this, timeout, securefds); |
|
|
|
|
handlecli(cli); |
|
|
|
|
} while (true); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda za asinkdorni rad s klijentima, prima limit, pokazivač na funkciju i timeout; |
|
|
|
|
* Funkcija handlecli prima referencu tipa client - važno za definiranje funkcija koje se šalju; |
|
|
|
|
* Nije moguće proslijediti druge parametre; |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
void server::async(const uint limit, void (*handlecli)(client&), const uint timeout) { |
|
|
|
|
for (uint i=0; i<limit; i++) { |
|
|
|
|
thr.push_back(thread([&](){ |
|
|
|
|
client *cli = new client(this, timeout, securefds); |
|
|
|
|
while (true) { |
|
|
|
|
try { |
|
|
|
|
handlecli(*cli); |
|
|
|
|
} catch (const ConnectionException err) { |
|
|
|
|
if (err.isInterrupted()) { |
|
|
|
|
cli->~client(); |
|
|
|
|
cli = new client(this, timeout, securefds); |
|
|
|
|
}
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
})); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for (uint i=0; i<limit; i++) { |
|
|
|
|
thr[i].join(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Destruktor varijable tipa server |
|
|
|
|
*/ |
|
|
|
@ -42,10 +98,21 @@ server::~server () { |
|
|
|
|
throw string("[ERROR] The socket is already closed ");
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
else if (close(sock) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
|
|
|
|
|
#if __linux__ |
|
|
|
|
if (close(sock) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
#elif _WIN32 |
|
|
|
|
if (closesocket(sock) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
WSACleanup(); |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
@ -109,9 +176,16 @@ secure::~secure () { |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
client::client(const string address, const ushort port, const uint timeout, SSL_CTX* securefds) { |
|
|
|
|
_timeout = timeout; |
|
|
|
|
|
|
|
|
|
sock = socket(AF_INET, SOCK_STREAM, 0); |
|
|
|
|
if (sock < 0) { |
|
|
|
|
#if _WIN32 |
|
|
|
|
if (WSAStartup(MAKEWORD(2,2),&wsa) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to set WinSock " + to_string(WSAGetLastError())); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
conn = socket(AF_INET, SOCK_STREAM, 0); |
|
|
|
|
if (conn < 0) { |
|
|
|
|
throw string("[ERROR] Unable to open TCP socket "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -121,24 +195,31 @@ client::client(const string address, const ushort port, const uint timeout, SSL_ |
|
|
|
|
addr.sin_addr.s_addr = inet_addr(_address.c_str()); |
|
|
|
|
addr.sin_port = htons(port); |
|
|
|
|
|
|
|
|
|
if (connect(sock, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)) != 0) { |
|
|
|
|
if (connect(conn, (struct sockaddr*)&addr, sizeof(struct sockaddr_in)) != 0) { |
|
|
|
|
throw string("Unable to connect to server "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
struct timeval tv; |
|
|
|
|
tv.tv_sec = timeout/1000; |
|
|
|
|
tv.tv_usec = (timeout%1000)*1000; |
|
|
|
|
#if __linux__ |
|
|
|
|
struct timeval tv; |
|
|
|
|
tv.tv_sec = 0; |
|
|
|
|
tv.tv_usec = SOCKET_TIMEOUT; |
|
|
|
|
|
|
|
|
|
if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
if (setsockopt(conn, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
#elif _WIN32 |
|
|
|
|
DWORD tv = timeout; |
|
|
|
|
if (setsockopt(conn, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
if (securefds) { |
|
|
|
|
ssl = SSL_new(securefds); |
|
|
|
|
if (!ssl) { |
|
|
|
|
throw string("[ERROR] Creating SSL object "); |
|
|
|
|
} |
|
|
|
|
SSL_set_fd(ssl, sock); |
|
|
|
|
SSL_set_fd(ssl, conn); |
|
|
|
|
|
|
|
|
|
// Perform the SSL handshake
|
|
|
|
|
if (SSL_connect(ssl) <= 0) { |
|
|
|
@ -150,65 +231,6 @@ client::client(const string address, const ushort port, const uint timeout, SSL_ |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Destruktor varijable tipa client |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
client::~client () { |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
SSL_shutdown(ssl); |
|
|
|
|
SSL_free(ssl); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (sock <= 0) { |
|
|
|
|
throw string("[ERROR] The socket is already closed ");
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
else if (close(sock) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda klase client za slanje podataka preko soketa |
|
|
|
|
* Prima string koji će biti poslan |
|
|
|
|
* Vraća logički statu poređenja psolanih karaktera i karaktera u stringu |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
bool client::tell (const string msg) { |
|
|
|
|
size_t sended = 0; |
|
|
|
|
if (ssl) { |
|
|
|
|
sended = SSL_write(ssl, msg.c_str(), msg.length()); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
sended = write(sock, msg.c_str(), msg.length()); |
|
|
|
|
} |
|
|
|
|
return sended == msg.length(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda klase client za primanje poruke preko soketa |
|
|
|
|
* Prima dozvoljeni broj karaktera koji će primiti |
|
|
|
|
* Vraća string primljene poruke |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
string client::obey (size_t byte_limit) { |
|
|
|
|
char res[byte_limit] = {0}; |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
SSL_read(ssl, res, byte_limit); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
read(sock , res, byte_limit); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return string(res); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Konstruktor varijable tipa commint |
|
|
|
|
* Prima pokazivač na inicijaliziranu varijablu tipa, port,
|
|
|
|
@ -217,22 +239,30 @@ string client::obey (size_t byte_limit) { |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
comming::comming(const server *_srv, const uint timeout, SSL_CTX* securefds) { |
|
|
|
|
client::client(const server *_srv, const uint timeout, SSL_CTX* securefds) { |
|
|
|
|
srv = _srv; |
|
|
|
|
socklen_t len = sizeof(struct sockaddr_in); |
|
|
|
|
_timeout = timeout; |
|
|
|
|
|
|
|
|
|
if ((conn = accept(srv->sock, (struct sockaddr *)&(srv->addr), (socklen_t*)&len)) < 0) { |
|
|
|
|
throw string("[ERROR] Unable to accept client connection "); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
struct timeval tv; |
|
|
|
|
tv.tv_sec = timeout/1000; |
|
|
|
|
tv.tv_usec = (timeout%1000)*1000; |
|
|
|
|
#if __linux__ |
|
|
|
|
struct timeval tv; |
|
|
|
|
tv.tv_sec = 0; |
|
|
|
|
tv.tv_usec = SOCKET_TIMEOUT; |
|
|
|
|
|
|
|
|
|
if (setsockopt(conn, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
if (setsockopt(conn, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
#elif _WIN32 |
|
|
|
|
DWORD tv = timeout; |
|
|
|
|
|
|
|
|
|
if (setsockopt(conn, SOL_SOCKET, SO_RCVTIMEO, (const char*)&tv, sizeof(tv))) { |
|
|
|
|
throw string("[ERROR] Unable to set timeout "); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
if (securefds) { |
|
|
|
|
ssl = SSL_new(securefds); |
|
|
|
@ -259,11 +289,12 @@ comming::comming(const server *_srv, const uint timeout, SSL_CTX* securefds) { |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Destruktor varijable tipa comming |
|
|
|
|
* Destruktor varijable tipa client |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
comming::~comming() { |
|
|
|
|
client::~client () { |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
SSL_shutdown(ssl); |
|
|
|
@ -274,43 +305,105 @@ comming::~comming() { |
|
|
|
|
throw string("[ERROR] The socket is already closed ");
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
else if (close(conn) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
else { |
|
|
|
|
#if __linux__ |
|
|
|
|
if (close(conn) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
#elif _WIN32 |
|
|
|
|
if (closesocket(conn) != 0) { |
|
|
|
|
throw string("[ERROR] Unable to close socket "); |
|
|
|
|
} |
|
|
|
|
//WSACleanup();
|
|
|
|
|
#endif |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda klase comming za slanje podataka preko soketa |
|
|
|
|
* Metoda klase client za slanje podataka preko soketa |
|
|
|
|
* Prima string koji će biti poslan |
|
|
|
|
* Vraća logički statu poređenja psolanih karaktera i karaktera u stringu |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
bool comming::tell (const string msg) { |
|
|
|
|
ssize_t sended = 0; |
|
|
|
|
if (ssl) { |
|
|
|
|
sended = SSL_write(ssl, msg.c_str(), msg.length()); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
sended = write(conn, msg.c_str(), msg.length()); |
|
|
|
|
|
|
|
|
|
bool client::push(const string msg) { |
|
|
|
|
size_t total_sent = 0; |
|
|
|
|
size_t msg_length = msg.length(); |
|
|
|
|
|
|
|
|
|
while (total_sent < msg_length) { |
|
|
|
|
size_t sent = 0; |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
sent = SSL_write(ssl, msg.c_str() + total_sent, msg_length - total_sent); |
|
|
|
|
} else { |
|
|
|
|
sent = send(conn, msg.c_str() + total_sent, msg_length - total_sent, 0); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (sent == -1) { |
|
|
|
|
throw string("[ERRNO] (push) - Error code: " + to_string(errno) + " Detail: " + strerror(errno)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
total_sent += sent; |
|
|
|
|
} |
|
|
|
|
return sended == msg.length(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* Metoda klase comming za primanje poruke preko soketa |
|
|
|
|
* Metoda klase client za primanje poruke preko soketa |
|
|
|
|
* Prima dozvoljeni broj karaktera koji će primiti |
|
|
|
|
* Vraća string primljene poruke |
|
|
|
|
*
|
|
|
|
|
* Funkcija baca izuzetke koji se moraju uhvatiti za pravilno rukovođenje vezom |
|
|
|
|
* Potrebno je i baciti dalje taj izuzetak ukoliko se koriste server async metode |
|
|
|
|
* PRILOG |
|
|
|
|
* ---------------------------------------------------------------- |
|
|
|
|
* try { |
|
|
|
|
fromclient = cli.pull(); |
|
|
|
|
} |
|
|
|
|
catch(const ConnectionException except) { |
|
|
|
|
if (except.isInterrupted()) { |
|
|
|
|
throw except; |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
cout << "[EXCEPT] " << except.what() << endl; |
|
|
|
|
fromclient = except.getData(); |
|
|
|
|
} |
|
|
|
|
}
|
|
|
|
|
* ----------------------------------------------------------------- |
|
|
|
|
*
|
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
string comming::obey (size_t byte_limit) { |
|
|
|
|
string client::pull(size_t byte_limit) { |
|
|
|
|
char res[byte_limit] = {0}; |
|
|
|
|
size_t total_received = 0; |
|
|
|
|
auto start = high_resolution_clock::now(); |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
SSL_read(ssl, res, byte_limit); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
read(conn , res, byte_limit); |
|
|
|
|
while (total_received < byte_limit) { |
|
|
|
|
ssize_t received = 0; |
|
|
|
|
|
|
|
|
|
if (ssl) { |
|
|
|
|
received = SSL_read(ssl, res + total_received, byte_limit - total_received); |
|
|
|
|
} else { |
|
|
|
|
received = recv(conn, res + total_received, byte_limit - total_received, 0); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
cout << "Primljeno " << received << endl; |
|
|
|
|
|
|
|
|
|
if (received == -1) { |
|
|
|
|
throw ConnectionException(strerror(errno), string(res, total_received)); |
|
|
|
|
} else if (received == 0) { |
|
|
|
|
throw ConnectionException("The socket is broken", string(res), true); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
total_received += received; |
|
|
|
|
|
|
|
|
|
auto cycle = high_resolution_clock::now(); |
|
|
|
|
if (duration_cast<milliseconds>(cycle - start).count() > _timeout) { |
|
|
|
|
cout << "TIMEOUT" << endl; |
|
|
|
|
throw ConnectionException("Timeout", string(res)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return string(res); |
|
|
|
|