blob: 01dc1256b24223e375102a66daa0080783229ff0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#pragma once
#include "../server/packet.h"
#include "../util/enc.h"
namespace tcp {
class client {
int m_socket;
SSL* m_ssl;
time_t m_time;
std::string m_ip;
std::string m_session_id;
public:
client() : m_socket{-1} {};
client(const int& socket, const std::string_view ip)
: m_socket{std::move(socket)}, m_ip{ip}, m_ssl{nullptr} {}
~client() = default;
bool init_ssl(SSL_CTX* server_ctx);
void cleanup() {
close(m_socket);
SSL_shutdown(m_ssl);
SSL_free(m_ssl);
}
int write(const void* data, size_t size) {
//std::string str(reinterpret_cast<const char*>(data), size);
//enc::encrypt_message(str);
return SSL_write(m_ssl, data, size);
}
int write(const packet_t& packet) {
if (!packet) return 0;
return write(packet.message.data(), packet.message.size());
}
int read(void* data, size_t size) { return SSL_read(m_ssl, data, size); }
int stream(std::vector<char>& data, float *dur = nullptr);
int read_stream(std::vector<char>& out);
void gen_session();
int get_socket() { return m_socket; }
auto get_ip() { return m_ip; }
auto get_session() { return m_session_id; }
};
}; // namespace tcp
|