diff options
Diffstat (limited to 'server/src/client/client.cpp')
| -rw-r--r-- | server/src/client/client.cpp | 133 |
1 files changed, 65 insertions, 68 deletions
diff --git a/server/src/client/client.cpp b/server/src/client/client.cpp index b82b5a4..4313e85 100644 --- a/server/src/client/client.cpp +++ b/server/src/client/client.cpp @@ -2,91 +2,88 @@ #include "../util/io.h" #include "client.h" -bool tcp::client::init_ssl(SSL_CTX* server_ctx) -{ - m_ssl = SSL_new(server_ctx); - if(!m_ssl) { - io::logger->error("failed to create ssl on client {}.", m_ip); - return false; - } - - int ret = SSL_set_fd(m_ssl, m_socket); - if(ret <= 0) { - io::logger->error("failed to set descriptor on client {}.", m_ip); - return false; - } - - ret = SSL_accept(m_ssl); - - if(ret <= 0) { - int err = SSL_get_error(m_ssl, ret); - io::logger->error("client {} failed to accept ssl, return code {}", m_ip, err); - return false; - } - - return true; +bool tcp::client::init_ssl(SSL_CTX* server_ctx) { + m_ssl = SSL_new(server_ctx); + if (!m_ssl) { + io::logger->error("failed to create ssl on client {}.", m_ip); + return false; + } + + int ret = SSL_set_fd(m_ssl, m_socket); + if (ret <= 0) { + io::logger->error("failed to set descriptor on client {}.", m_ip); + return false; + } + + ret = SSL_accept(m_ssl); + + if (ret <= 0) { + int err = SSL_get_error(m_ssl, ret); + io::logger->error("client {} failed to accept ssl, return code {}", m_ip, + err); + return false; + } + + return true; } -void tcp::client::gen_session() -{ - std::random_device r; - std::default_random_engine e1(r()); - std::uniform_int_distribution<int> gen(33, 126); +void tcp::client::gen_session() { + std::random_device r; + std::default_random_engine e1(r()); + std::uniform_int_distribution<int> gen(33, 126); - for(int i = 0; i < session_id_len; i++) { - auto k = static_cast<char>(gen(e1)); - m_session_id.insert(m_session_id.end(), k); - } + for (int i = 0; i < session_id_len; i++) { + auto k = static_cast<char>(gen(e1)); + m_session_id.insert(m_session_id.end(), k); + } } -int tcp::client::stream(std::vector<char>& data) -{ - auto size = data.size(); - - auto networked_size = htonl(size); - write(&networked_size, sizeof(networked_size)); +int tcp::client::stream(std::vector<char>& data) { + auto size = data.size(); - // with 4kb chunk size, speed peaks at 90mb/s - constexpr size_t chunk_size = 4096; - size_t sent = 0; + auto networked_size = htonl(size); + write(&networked_size, sizeof(networked_size)); - while(size > 0) { - auto to_send = std::min(size, chunk_size); + // with 4kb chunk size, speed peaks at 90mb/s + constexpr size_t chunk_size = 4096; + size_t sent = 0; - int ret = write(&data[sent], to_send); - if(ret <= 0) { - break; - } + while (size > 0) { + auto to_send = std::min(size, chunk_size); - sent += ret; - size -= ret; + int ret = write(&data[sent], to_send); + if (ret <= 0) { + break; } - return sent; -} + sent += ret; + size -= ret; + } -int tcp::client::read_stream(std::vector<char>& out) -{ - size_t size; - read(&size, sizeof(size)); + return sent; +} - size = ntohl(size); - out.resize(size); +int tcp::client::read_stream(std::vector<char>& out) { + size_t size; + read(&size, sizeof(size)); - constexpr size_t chunk_size = 4096; - size_t total = 0; + size = ntohl(size); + out.resize(size); - while(size > 0) { - auto to_read = std::min(size, chunk_size); + constexpr size_t chunk_size = 4096; + size_t total = 0; - int ret = read(&out[total], to_read); - if(ret <= 0) { - break; - } + while (size > 0) { + auto to_read = std::min(size, chunk_size); - size -= ret; - total += ret; + int ret = read(&out[total], to_read); + if (ret <= 0) { + break; } - return total; + size -= ret; + total += ret; + } + + return total; }
\ No newline at end of file |