aboutsummaryrefslogtreecommitdiff
path: root/server/src/client/client.h
blob: 106b2c112d2327e26e1f4bc5539c739839734c71 (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
#pragma once
#include "../server/packet.h"

namespace tcp {
constexpr uint8_t client_version = 0;

class client {
  int m_socket;
  SSL *m_ssl;

  time_t m_time;

  std::string m_ip;
  std::string m_uid;

 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(void *data, size_t size) { return SSL_write(m_ssl, data, size); }
  int write(const packet_t &packet) {
    return SSL_write(m_ssl, 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);
  int read_stream(std::vector<char> &out);

  void gen_uid();

  int get_socket() { return m_socket; }
  auto get_ip() { return m_ip; }
  auto get_uid() { return m_uid; }
};
};  // namespace tcp