-
Notifications
You must be signed in to change notification settings - Fork 0
/
network_client.h
78 lines (58 loc) · 2.14 KB
/
network_client.h
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
//
// Created by macskas on 12/9/20.
//
#ifndef NSCA_NETWORK_CLIENT_H
#define NSCA_NETWORK_CLIENT_H
#define CONNECTION_STATE_INIT_SENT 1
#include "network.h"
#include <event2/util.h>
#include <string>
#include <deque>
typedef struct {
data_packet receive_packet;
size_t packet_length;
} data_packet_pair_t;
class network_client {
private:
class network *parent{};
evutil_socket_t fd{};
struct sockaddr_storage ss{};
ev_socklen_t socklen{};
uint32_t client_id;
std::string clientIp;
uint16_t clientPort;
int read_timeout;
int write_timeout;
int connection_state;
char *mcrypt_state = nullptr;
int mcrypt_state_size = 16;
struct crypt_instance *CI;
struct bufferevent *client_bev = nullptr;
bool has_own_CI = false;
char* transmission_iv = nullptr;
int d_failed = 0;
int d_success = 0;
size_t bytes_received = 0;
size_t bytes_drained = 0;
std::deque<data_packet_pair_t *> data_packets;
public:
network_client();
~network_client();
void take_variables(network*, evutil_socket_t, struct sockaddr*, ev_socklen_t);
int init_events();
void proper_destroy();
void connection_closed();
void debug_message(const std::string&);
void writecb(bufferevent *bev);
void readcb(bufferevent *bev);
void eventcb(bufferevent *bev, short events);
static void readcb_proxy(bufferevent *, void *);
static void writecb_proxy(bufferevent *, void *);
static void eventcb_proxy(struct bufferevent *, short , void *);
void process_queue(int thread_id);
void process_queue_mcrypt(int thread_id);
void process_queue_nomcrypt(int thread_id);
void set_CI(struct crypt_instance *);
void send_receive_message(data_packet*, int thread_id);
};
#endif //NSCA_NETWORK_CLIENT_H